renderer_webgpu.cpp 149 KB

12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455565758596061626364656667686970717273747576777879808182838485868788899091929394959697989910010110210310410510610710810911011111211311411511611711811912012112212312412512612712812913013113213313413513613713813914014114214314414514614714814915015115215315415515615715815916016116216316416516616716816917017117217317417517617717817918018118218318418518618718818919019119219319419519619719819920020120220320420520620720820921021121221321421521621721821922022122222322422522622722822923023123223323423523623723823924024124224324424524624724824925025125225325425525625725825926026126226326426526626726826927027127227327427527627727827928028128228328428528628728828929029129229329429529629729829930030130230330430530630730830931031131231331431531631731831932032132232332432532632732832933033133233333433533633733833934034134234334434534634734834935035135235335435535635735835936036136236336436536636736836937037137237337437537637737837938038138238338438538638738838939039139239339439539639739839940040140240340440540640740840941041141241341441541641741841942042142242342442542642742842943043143243343443543643743843944044144244344444544644744844945045145245345445545645745845946046146246346446546646746846947047147247347447547647747847948048148248348448548648748848949049149249349449549649749849950050150250350450550650750850951051151251351451551651751851952052152252352452552652752852953053153253353453553653753853954054154254354454554654754854955055155255355455555655755855956056156256356456556656756856957057157257357457557657757857958058158258358458558658758858959059159259359459559659759859960060160260360460560660760860961061161261361461561661761861962062162262362462562662762862963063163263363463563663763863964064164264364464564664764864965065165265365465565665765865966066166266366466566666766866967067167267367467567667767867968068168268368468568668768868969069169269369469569669769869970070170270370470570670770870971071171271371471571671771871972072172272372472572672772872973073173273373473573673773873974074174274374474574674774874975075175275375475575675775875976076176276376476576676776876977077177277377477577677777877978078178278378478578678778878979079179279379479579679779879980080180280380480580680780880981081181281381481581681781881982082182282382482582682782882983083183283383483583683783883984084184284384484584684784884985085185285385485585685785885986086186286386486586686786886987087187287387487587687787887988088188288388488588688788888989089189289389489589689789889990090190290390490590690790890991091191291391491591691791891992092192292392492592692792892993093193293393493593693793893994094194294394494594694794894995095195295395495595695795895996096196296396496596696796896997097197297397497597697797897998098198298398498598698798898999099199299399499599699799899910001001100210031004100510061007100810091010101110121013101410151016101710181019102010211022102310241025102610271028102910301031103210331034103510361037103810391040104110421043104410451046104710481049105010511052105310541055105610571058105910601061106210631064106510661067106810691070107110721073107410751076107710781079108010811082108310841085108610871088108910901091109210931094109510961097109810991100110111021103110411051106110711081109111011111112111311141115111611171118111911201121112211231124112511261127112811291130113111321133113411351136113711381139114011411142114311441145114611471148114911501151115211531154115511561157115811591160116111621163116411651166116711681169117011711172117311741175117611771178117911801181118211831184118511861187118811891190119111921193119411951196119711981199120012011202120312041205120612071208120912101211121212131214121512161217121812191220122112221223122412251226122712281229123012311232123312341235123612371238123912401241124212431244124512461247124812491250125112521253125412551256125712581259126012611262126312641265126612671268126912701271127212731274127512761277127812791280128112821283128412851286128712881289129012911292129312941295129612971298129913001301130213031304130513061307130813091310131113121313131413151316131713181319132013211322132313241325132613271328132913301331133213331334133513361337133813391340134113421343134413451346134713481349135013511352135313541355135613571358135913601361136213631364136513661367136813691370137113721373137413751376137713781379138013811382138313841385138613871388138913901391139213931394139513961397139813991400140114021403140414051406140714081409141014111412141314141415141614171418141914201421142214231424142514261427142814291430143114321433143414351436143714381439144014411442144314441445144614471448144914501451145214531454145514561457145814591460146114621463146414651466146714681469147014711472147314741475147614771478147914801481148214831484148514861487148814891490149114921493149414951496149714981499150015011502150315041505150615071508150915101511151215131514151515161517151815191520152115221523152415251526152715281529153015311532153315341535153615371538153915401541154215431544154515461547154815491550155115521553155415551556155715581559156015611562156315641565156615671568156915701571157215731574157515761577157815791580158115821583158415851586158715881589159015911592159315941595159615971598159916001601160216031604160516061607160816091610161116121613161416151616161716181619162016211622162316241625162616271628162916301631163216331634163516361637163816391640164116421643164416451646164716481649165016511652165316541655165616571658165916601661166216631664166516661667166816691670167116721673167416751676167716781679168016811682168316841685168616871688168916901691169216931694169516961697169816991700170117021703170417051706170717081709171017111712171317141715171617171718171917201721172217231724172517261727172817291730173117321733173417351736173717381739174017411742174317441745174617471748174917501751175217531754175517561757175817591760176117621763176417651766176717681769177017711772177317741775177617771778177917801781178217831784178517861787178817891790179117921793179417951796179717981799180018011802180318041805180618071808180918101811181218131814181518161817181818191820182118221823182418251826182718281829183018311832183318341835183618371838183918401841184218431844184518461847184818491850185118521853185418551856185718581859186018611862186318641865186618671868186918701871187218731874187518761877187818791880188118821883188418851886188718881889189018911892189318941895189618971898189919001901190219031904190519061907190819091910191119121913191419151916191719181919192019211922192319241925192619271928192919301931193219331934193519361937193819391940194119421943194419451946194719481949195019511952195319541955195619571958195919601961196219631964196519661967196819691970197119721973197419751976197719781979198019811982198319841985198619871988198919901991199219931994199519961997199819992000200120022003200420052006200720082009201020112012201320142015201620172018201920202021202220232024202520262027202820292030203120322033203420352036203720382039204020412042204320442045204620472048204920502051205220532054205520562057205820592060206120622063206420652066206720682069207020712072207320742075207620772078207920802081208220832084208520862087208820892090209120922093209420952096209720982099210021012102210321042105210621072108210921102111211221132114211521162117211821192120212121222123212421252126212721282129213021312132213321342135213621372138213921402141214221432144214521462147214821492150215121522153215421552156215721582159216021612162216321642165216621672168216921702171217221732174217521762177217821792180218121822183218421852186218721882189219021912192219321942195219621972198219922002201220222032204220522062207220822092210221122122213221422152216221722182219222022212222222322242225222622272228222922302231223222332234223522362237223822392240224122422243224422452246224722482249225022512252225322542255225622572258225922602261226222632264226522662267226822692270227122722273227422752276227722782279228022812282228322842285228622872288228922902291229222932294229522962297229822992300230123022303230423052306230723082309231023112312231323142315231623172318231923202321232223232324232523262327232823292330233123322333233423352336233723382339234023412342234323442345234623472348234923502351235223532354235523562357235823592360236123622363236423652366236723682369237023712372237323742375237623772378237923802381238223832384238523862387238823892390239123922393239423952396239723982399240024012402240324042405240624072408240924102411241224132414241524162417241824192420242124222423242424252426242724282429243024312432243324342435243624372438243924402441244224432444244524462447244824492450245124522453245424552456245724582459246024612462246324642465246624672468246924702471247224732474247524762477247824792480248124822483248424852486248724882489249024912492249324942495249624972498249925002501250225032504250525062507250825092510251125122513251425152516251725182519252025212522252325242525252625272528252925302531253225332534253525362537253825392540254125422543254425452546254725482549255025512552255325542555255625572558255925602561256225632564256525662567256825692570257125722573257425752576257725782579258025812582258325842585258625872588258925902591259225932594259525962597259825992600260126022603260426052606260726082609261026112612261326142615261626172618261926202621262226232624262526262627262826292630263126322633263426352636263726382639264026412642264326442645264626472648264926502651265226532654265526562657265826592660266126622663266426652666266726682669267026712672267326742675267626772678267926802681268226832684268526862687268826892690269126922693269426952696269726982699270027012702270327042705270627072708270927102711271227132714271527162717271827192720272127222723272427252726272727282729273027312732273327342735273627372738273927402741274227432744274527462747274827492750275127522753275427552756275727582759276027612762276327642765276627672768276927702771277227732774277527762777277827792780278127822783278427852786278727882789279027912792279327942795279627972798279928002801280228032804280528062807280828092810281128122813281428152816281728182819282028212822282328242825282628272828282928302831283228332834283528362837283828392840284128422843284428452846284728482849285028512852285328542855285628572858285928602861286228632864286528662867286828692870287128722873287428752876287728782879288028812882288328842885288628872888288928902891289228932894289528962897289828992900290129022903290429052906290729082909291029112912291329142915291629172918291929202921292229232924292529262927292829292930293129322933293429352936293729382939294029412942294329442945294629472948294929502951295229532954295529562957295829592960296129622963296429652966296729682969297029712972297329742975297629772978297929802981298229832984298529862987298829892990299129922993299429952996299729982999300030013002300330043005300630073008300930103011301230133014301530163017301830193020302130223023302430253026302730283029303030313032303330343035303630373038303930403041304230433044304530463047304830493050305130523053305430553056305730583059306030613062306330643065306630673068306930703071307230733074307530763077307830793080308130823083308430853086308730883089309030913092309330943095309630973098309931003101310231033104310531063107310831093110311131123113311431153116311731183119312031213122312331243125312631273128312931303131313231333134313531363137313831393140314131423143314431453146314731483149315031513152315331543155315631573158315931603161316231633164316531663167316831693170317131723173317431753176317731783179318031813182318331843185318631873188318931903191319231933194319531963197319831993200320132023203320432053206320732083209321032113212321332143215321632173218321932203221322232233224322532263227322832293230323132323233323432353236323732383239324032413242324332443245324632473248324932503251325232533254325532563257325832593260326132623263326432653266326732683269327032713272327332743275327632773278327932803281328232833284328532863287328832893290329132923293329432953296329732983299330033013302330333043305330633073308330933103311331233133314331533163317331833193320332133223323332433253326332733283329333033313332333333343335333633373338333933403341334233433344334533463347334833493350335133523353335433553356335733583359336033613362336333643365336633673368336933703371337233733374337533763377337833793380338133823383338433853386338733883389339033913392339333943395339633973398339934003401340234033404340534063407340834093410341134123413341434153416341734183419342034213422342334243425342634273428342934303431343234333434343534363437343834393440344134423443344434453446344734483449345034513452345334543455345634573458345934603461346234633464346534663467346834693470347134723473347434753476347734783479348034813482348334843485348634873488348934903491349234933494349534963497349834993500350135023503350435053506350735083509351035113512351335143515351635173518351935203521352235233524352535263527352835293530353135323533353435353536353735383539354035413542354335443545354635473548354935503551355235533554355535563557355835593560356135623563356435653566356735683569357035713572357335743575357635773578357935803581358235833584358535863587358835893590359135923593359435953596359735983599360036013602360336043605360636073608360936103611361236133614361536163617361836193620362136223623362436253626362736283629363036313632363336343635363636373638363936403641364236433644364536463647364836493650365136523653365436553656365736583659366036613662366336643665366636673668366936703671367236733674367536763677367836793680368136823683368436853686368736883689369036913692369336943695369636973698369937003701370237033704370537063707370837093710371137123713371437153716371737183719372037213722372337243725372637273728372937303731373237333734373537363737373837393740374137423743374437453746374737483749375037513752375337543755375637573758375937603761376237633764376537663767376837693770377137723773377437753776377737783779378037813782378337843785378637873788378937903791379237933794379537963797379837993800380138023803380438053806380738083809381038113812381338143815381638173818381938203821382238233824382538263827382838293830383138323833383438353836383738383839384038413842384338443845384638473848384938503851385238533854385538563857385838593860386138623863386438653866386738683869387038713872387338743875387638773878387938803881388238833884388538863887388838893890389138923893389438953896389738983899390039013902390339043905390639073908390939103911391239133914391539163917391839193920392139223923392439253926392739283929393039313932393339343935393639373938393939403941394239433944394539463947394839493950395139523953395439553956395739583959396039613962396339643965396639673968396939703971397239733974397539763977397839793980398139823983398439853986398739883989399039913992399339943995399639973998399940004001400240034004400540064007400840094010401140124013401440154016401740184019402040214022402340244025402640274028402940304031403240334034403540364037403840394040404140424043404440454046404740484049405040514052405340544055405640574058405940604061406240634064406540664067406840694070407140724073407440754076407740784079408040814082408340844085408640874088408940904091409240934094409540964097409840994100410141024103410441054106410741084109411041114112411341144115411641174118411941204121412241234124412541264127412841294130413141324133413441354136413741384139414041414142414341444145414641474148414941504151415241534154415541564157415841594160416141624163416441654166416741684169417041714172417341744175417641774178417941804181418241834184418541864187418841894190419141924193419441954196419741984199420042014202420342044205420642074208420942104211421242134214421542164217421842194220422142224223422442254226422742284229423042314232423342344235423642374238423942404241424242434244424542464247424842494250425142524253425442554256425742584259426042614262426342644265426642674268426942704271427242734274427542764277427842794280428142824283428442854286428742884289429042914292429342944295429642974298429943004301430243034304430543064307430843094310431143124313431443154316431743184319432043214322432343244325432643274328432943304331433243334334433543364337433843394340434143424343434443454346434743484349435043514352435343544355435643574358435943604361436243634364436543664367436843694370437143724373437443754376437743784379438043814382438343844385438643874388438943904391439243934394439543964397439843994400440144024403440444054406440744084409441044114412441344144415441644174418441944204421442244234424442544264427442844294430443144324433443444354436443744384439444044414442444344444445444644474448444944504451445244534454445544564457445844594460446144624463446444654466446744684469447044714472447344744475447644774478447944804481448244834484448544864487448844894490449144924493449444954496449744984499450045014502450345044505450645074508450945104511451245134514451545164517451845194520452145224523452445254526452745284529453045314532453345344535453645374538453945404541454245434544454545464547454845494550455145524553455445554556455745584559456045614562456345644565456645674568456945704571457245734574457545764577457845794580458145824583458445854586458745884589459045914592459345944595459645974598459946004601460246034604460546064607460846094610461146124613461446154616461746184619462046214622462346244625462646274628462946304631463246334634463546364637463846394640464146424643464446454646464746484649465046514652465346544655465646574658465946604661466246634664466546664667466846694670467146724673467446754676467746784679468046814682468346844685468646874688468946904691469246934694469546964697469846994700470147024703470447054706470747084709471047114712471347144715471647174718471947204721472247234724472547264727472847294730473147324733473447354736473747384739474047414742474347444745474647474748474947504751475247534754475547564757475847594760476147624763476447654766476747684769477047714772477347744775477647774778477947804781478247834784478547864787478847894790479147924793479447954796479747984799480048014802480348044805480648074808480948104811481248134814481548164817481848194820482148224823482448254826482748284829483048314832483348344835483648374838483948404841484248434844484548464847484848494850485148524853485448554856485748584859486048614862486348644865486648674868486948704871487248734874487548764877487848794880488148824883488448854886488748884889489048914892489348944895489648974898489949004901490249034904490549064907490849094910491149124913491449154916491749184919492049214922492349244925492649274928
  1. /*
  2. * Copyright 2011-2019 Branimir Karadzic. All rights reserved.
  3. * License: https://github.com/bkaradzic/bgfx#license-bsd-2-clause
  4. */
  5. #include "bgfx_p.h"
  6. //#define DAWN_ENABLE_BACKEND_D3D12
  7. #define DAWN_ENABLE_BACKEND_VULKAN
  8. #if BGFX_CONFIG_RENDERER_WEBGPU
  9. # include "renderer_webgpu.h"
  10. # include "renderer.h"
  11. # include "debug_renderdoc.h"
  12. # if !BX_PLATFORM_EMSCRIPTEN
  13. # ifdef DAWN_ENABLE_BACKEND_D3D12
  14. # include <dawn_native/D3D12Backend.h>
  15. # endif // !BX_PLATFORM_EMSCRIPTEN
  16. # ifdef DAWN_ENABLE_BACKEND_VULKAN
  17. # include <dawn_native/VulkanBackend.h>
  18. # endif // DAWN_ENABLE_BACKEND_VULKAN
  19. # include <dawn_native/DawnNative.h>
  20. # include <dawn/dawn_wsi.h>
  21. # include <dawn/dawn_proc.h>
  22. # else
  23. # include <emscripten/emscripten.h>
  24. # include <emscripten/html5.h>
  25. # endif // !BX_PLATFORM_EMSCRIPTEN
  26. namespace bgfx { namespace webgpu
  27. {
  28. // TODO (hugoam) cleanup
  29. template <class T>
  30. T defaultDescriptor() { return T(); }
  31. template <> wgpu::BlendDescriptor defaultDescriptor() { return { wgpu::BlendOperation::Add, wgpu::BlendFactor::One, wgpu::BlendFactor::Zero }; }
  32. template <> wgpu::ColorStateDescriptor defaultDescriptor() { return { NULL, wgpu::TextureFormat::RGBA8Unorm, defaultDescriptor<wgpu::BlendDescriptor>(), defaultDescriptor<wgpu::BlendDescriptor>(), wgpu::ColorWriteMask::All }; }
  33. template <> wgpu::StencilStateFaceDescriptor defaultDescriptor() { return { wgpu::CompareFunction::Always, wgpu::StencilOperation::Keep, wgpu::StencilOperation::Keep, wgpu::StencilOperation::Keep }; }
  34. template <> wgpu::VertexStateDescriptor defaultDescriptor() { return { NULL, wgpu::IndexFormat::Uint32, 0, NULL }; }
  35. template <> wgpu::VertexBufferLayoutDescriptor defaultDescriptor() { return { 0, wgpu::InputStepMode::Vertex, 0, NULL }; }
  36. template <> wgpu::VertexAttributeDescriptor defaultDescriptor() { return { wgpu::VertexFormat::Float, 0, 0 }; }
  37. template <> wgpu::RasterizationStateDescriptor defaultDescriptor() { return { NULL, wgpu::FrontFace::CCW, wgpu::CullMode::None, 0, 0.f, 0.f }; }
  38. template <> wgpu::ProgrammableStageDescriptor defaultDescriptor() { return { NULL, {}, "main" }; }
  39. template <> wgpu::DepthStencilStateDescriptor defaultDescriptor() { return { NULL, wgpu::TextureFormat::Depth24PlusStencil8, false, wgpu::CompareFunction::Always, defaultDescriptor<wgpu::StencilStateFaceDescriptor>(), defaultDescriptor<wgpu::StencilStateFaceDescriptor>(), 0xff, 0xff }; }
  40. template <> wgpu::PipelineLayoutDescriptor defaultDescriptor() { return { NULL, "", 0, NULL }; }
  41. template <> wgpu::TextureViewDescriptor defaultDescriptor() { return {}; }
  42. template <> wgpu::RenderPassColorAttachmentDescriptor defaultDescriptor() { return { {}, {}, wgpu::LoadOp::Clear, wgpu::StoreOp::Store, { 0.0f, 0.0f, 0.0f, 0.0f } }; }
  43. template <> wgpu::RenderPassDepthStencilAttachmentDescriptor defaultDescriptor() { return { {}, wgpu::LoadOp::Clear, wgpu::StoreOp::Store, 1.0f, wgpu::LoadOp::Clear, wgpu::StoreOp::Store, 0 }; }
  44. RenderPassDescriptor::RenderPassDescriptor()
  45. {
  46. depthStencilAttachment = defaultDescriptor<wgpu::RenderPassDepthStencilAttachmentDescriptor>();
  47. for(uint32_t i = 0; i < kMaxColorAttachments; ++i)
  48. {
  49. colorAttachments[i] = defaultDescriptor<wgpu::RenderPassColorAttachmentDescriptor>();
  50. }
  51. desc = defaultDescriptor<wgpu::RenderPassDescriptor>();
  52. //desc.colorAttachmentCount = colorAttachmentCount;
  53. desc.colorAttachments = colorAttachments;
  54. desc.colorAttachmentCount = 1; // TODO (hugoam) set it properly everywhere
  55. }
  56. VertexStateDescriptor::VertexStateDescriptor()
  57. {
  58. for(uint32_t i = 0; i < kMaxVertexInputs; ++i)
  59. {
  60. vertexBuffers[i] = defaultDescriptor<wgpu::VertexBufferLayoutDescriptor>();
  61. }
  62. for (uint32_t i = 0; i < kMaxVertexAttributes; ++i)
  63. {
  64. attributes[i] = defaultDescriptor<wgpu::VertexAttributeDescriptor>();
  65. }
  66. vertexBuffers[0].attributes = &attributes[0];
  67. //vertexBuffers[0].attributeCount = numAttributes;
  68. desc = defaultDescriptor<wgpu::VertexStateDescriptor>();
  69. desc.vertexBuffers = vertexBuffers;
  70. //desc.vertexBufferCount = numVertexBuffers;
  71. }
  72. RenderPipelineDescriptor::RenderPipelineDescriptor()
  73. {
  74. //vertexStage = defaultDescriptor<wgpu::ProgrammableStageDescriptor>();
  75. fragmentStage = defaultDescriptor<wgpu::ProgrammableStageDescriptor>();
  76. rasterizationState = defaultDescriptor<wgpu::RasterizationStateDescriptor>();
  77. depthStencilState = defaultDescriptor<wgpu::DepthStencilStateDescriptor>();
  78. for(uint32_t i = 0; i < kMaxColorAttachments; ++i)
  79. {
  80. colorStates[i] = defaultDescriptor<wgpu::ColorStateDescriptor>();
  81. }
  82. desc = defaultDescriptor<wgpu::RenderPipelineDescriptor>();
  83. desc.primitiveTopology = wgpu::PrimitiveTopology::TriangleList;
  84. desc.sampleCount = 1;
  85. desc.colorStateCount = 1;
  86. //wgpu::VertexStateDescriptor inputState = inputState.descriptor();
  87. desc.vertexStage = defaultDescriptor<wgpu::ProgrammableStageDescriptor>();
  88. desc.fragmentStage = &fragmentStage;
  89. //desc.vertexState = &inputState;
  90. desc.rasterizationState = &rasterizationState;
  91. desc.depthStencilState = NULL;
  92. desc.colorStates = colorStates;
  93. }
  94. // TODO (hugoam) cleanup (end)
  95. static char s_viewName[BGFX_CONFIG_MAX_VIEWS][BGFX_CONFIG_MAX_VIEW_NAME];
  96. inline void setViewType(ViewId _view, const bx::StringView _str)
  97. {
  98. if (BX_ENABLED(BGFX_CONFIG_DEBUG_ANNOTATION || BGFX_CONFIG_PROFILER) )
  99. {
  100. bx::memCopy(&s_viewName[_view][3], _str.getPtr(), _str.getLength() );
  101. }
  102. }
  103. struct PrimInfo
  104. {
  105. wgpu::PrimitiveTopology m_type;
  106. uint32_t m_min;
  107. uint32_t m_div;
  108. uint32_t m_sub;
  109. };
  110. static const PrimInfo s_primInfo[] =
  111. {
  112. { wgpu::PrimitiveTopology::TriangleList, 3, 3, 0 },
  113. { wgpu::PrimitiveTopology::TriangleStrip, 3, 1, 2 },
  114. { wgpu::PrimitiveTopology::LineList, 2, 2, 0 },
  115. { wgpu::PrimitiveTopology::LineStrip, 2, 1, 1 },
  116. { wgpu::PrimitiveTopology::PointList, 1, 1, 0 },
  117. };
  118. BX_STATIC_ASSERT(Topology::Count == BX_COUNTOF(s_primInfo) );
  119. static const wgpu::VertexFormat s_attribType[][4][2] =
  120. {
  121. { // Uint8
  122. { wgpu::VertexFormat::UChar2, wgpu::VertexFormat::UChar2Norm },
  123. { wgpu::VertexFormat::UChar2, wgpu::VertexFormat::UChar2Norm },
  124. { wgpu::VertexFormat::UChar4, wgpu::VertexFormat::UChar4Norm },
  125. { wgpu::VertexFormat::UChar4, wgpu::VertexFormat::UChar4Norm },
  126. },
  127. { // Uint10
  128. { wgpu::VertexFormat::UShort2, wgpu::VertexFormat::UShort2Norm },
  129. { wgpu::VertexFormat::UShort2, wgpu::VertexFormat::UShort2Norm },
  130. { wgpu::VertexFormat::UShort4, wgpu::VertexFormat::UShort4Norm },
  131. { wgpu::VertexFormat::UShort4, wgpu::VertexFormat::UShort4Norm },
  132. },
  133. { // Int16
  134. { wgpu::VertexFormat::Short2, wgpu::VertexFormat::Short2Norm },
  135. { wgpu::VertexFormat::Short2, wgpu::VertexFormat::Short2Norm },
  136. { wgpu::VertexFormat::Short4, wgpu::VertexFormat::Short4Norm },
  137. { wgpu::VertexFormat::Short4, wgpu::VertexFormat::Short4Norm },
  138. },
  139. { // Half
  140. { wgpu::VertexFormat::Half2, wgpu::VertexFormat::Half2 },
  141. { wgpu::VertexFormat::Half2, wgpu::VertexFormat::Half2 },
  142. { wgpu::VertexFormat::Half4, wgpu::VertexFormat::Half4 },
  143. { wgpu::VertexFormat::Half4, wgpu::VertexFormat::Half4 },
  144. },
  145. { // Float
  146. { wgpu::VertexFormat::Float, wgpu::VertexFormat::Float },
  147. { wgpu::VertexFormat::Float2, wgpu::VertexFormat::Float2 },
  148. { wgpu::VertexFormat::Float3, wgpu::VertexFormat::Float3 },
  149. { wgpu::VertexFormat::Float4, wgpu::VertexFormat::Float4 },
  150. },
  151. };
  152. BX_STATIC_ASSERT(AttribType::Count == BX_COUNTOF(s_attribType) );
  153. static const wgpu::CullMode s_cullMode[] =
  154. {
  155. wgpu::CullMode::None,
  156. wgpu::CullMode::Front,
  157. wgpu::CullMode::Back,
  158. wgpu::CullMode::None,
  159. };
  160. static const wgpu::BlendFactor s_blendFactor[][2] =
  161. {
  162. { wgpu::BlendFactor(0), wgpu::BlendFactor(0) }, // ignored
  163. { wgpu::BlendFactor::Zero, wgpu::BlendFactor::Zero }, // ZERO
  164. { wgpu::BlendFactor::One, wgpu::BlendFactor::One }, // ONE
  165. { wgpu::BlendFactor::SrcColor, wgpu::BlendFactor::SrcAlpha }, // SRC_COLOR
  166. { wgpu::BlendFactor::OneMinusSrcColor, wgpu::BlendFactor::OneMinusSrcAlpha }, // INV_SRC_COLOR
  167. { wgpu::BlendFactor::SrcAlpha, wgpu::BlendFactor::SrcAlpha }, // SRC_ALPHA
  168. { wgpu::BlendFactor::OneMinusSrcAlpha, wgpu::BlendFactor::OneMinusSrcAlpha }, // INV_SRC_ALPHA
  169. { wgpu::BlendFactor::DstAlpha, wgpu::BlendFactor::DstAlpha }, // DST_ALPHA
  170. { wgpu::BlendFactor::OneMinusDstAlpha, wgpu::BlendFactor::OneMinusDstAlpha }, // INV_DST_ALPHA
  171. { wgpu::BlendFactor::DstColor, wgpu::BlendFactor::DstAlpha }, // DST_COLOR
  172. { wgpu::BlendFactor::OneMinusDstColor, wgpu::BlendFactor::OneMinusDstAlpha }, // INV_DST_COLOR
  173. { wgpu::BlendFactor::SrcAlphaSaturated, wgpu::BlendFactor::One }, // SRC_ALPHA_SAT
  174. { wgpu::BlendFactor::BlendColor, wgpu::BlendFactor::BlendColor }, // FACTOR
  175. { wgpu::BlendFactor::OneMinusBlendColor, wgpu::BlendFactor::OneMinusBlendColor }, // INV_FACTOR
  176. };
  177. static const wgpu::BlendOperation s_blendEquation[] =
  178. {
  179. wgpu::BlendOperation::Add,
  180. wgpu::BlendOperation::Subtract,
  181. wgpu::BlendOperation::ReverseSubtract,
  182. wgpu::BlendOperation::Min,
  183. wgpu::BlendOperation::Max,
  184. };
  185. static const wgpu::CompareFunction s_cmpFunc[] =
  186. {
  187. wgpu::CompareFunction::Always, // ignored
  188. wgpu::CompareFunction::Less,
  189. wgpu::CompareFunction::LessEqual,
  190. wgpu::CompareFunction::Equal,
  191. wgpu::CompareFunction::GreaterEqual,
  192. wgpu::CompareFunction::Greater,
  193. wgpu::CompareFunction::NotEqual,
  194. wgpu::CompareFunction::Never,
  195. wgpu::CompareFunction::Always,
  196. };
  197. static const wgpu::StencilOperation s_stencilOp[] =
  198. {
  199. wgpu::StencilOperation::Zero,
  200. wgpu::StencilOperation::Keep,
  201. wgpu::StencilOperation::Replace,
  202. wgpu::StencilOperation::IncrementWrap,
  203. wgpu::StencilOperation::IncrementClamp,
  204. wgpu::StencilOperation::DecrementWrap,
  205. wgpu::StencilOperation::DecrementClamp,
  206. wgpu::StencilOperation::Invert,
  207. };
  208. static const wgpu::AddressMode s_textureAddress[] =
  209. {
  210. wgpu::AddressMode::Repeat,
  211. wgpu::AddressMode::MirrorRepeat,
  212. wgpu::AddressMode::ClampToEdge,
  213. wgpu::AddressMode(0), // Border ? ClampToZero ?
  214. };
  215. static const wgpu::FilterMode s_textureFilterMinMag[] =
  216. {
  217. wgpu::FilterMode::Linear,
  218. wgpu::FilterMode::Nearest,
  219. wgpu::FilterMode::Linear,
  220. };
  221. static const wgpu::FilterMode s_textureFilterMip[] =
  222. {
  223. wgpu::FilterMode::Linear,
  224. wgpu::FilterMode::Nearest,
  225. };
  226. struct TextureFormatInfo
  227. {
  228. wgpu::TextureFormat m_fmt;
  229. wgpu::TextureFormat m_fmtSrgb;
  230. };
  231. static TextureFormatInfo s_textureFormat[] =
  232. {
  233. { wgpu::TextureFormat::BC1RGBAUnorm, wgpu::TextureFormat::BC1RGBAUnormSrgb }, // BC1
  234. { wgpu::TextureFormat::BC2RGBAUnorm, wgpu::TextureFormat::BC2RGBAUnormSrgb }, // BC2
  235. { wgpu::TextureFormat::BC3RGBAUnorm, wgpu::TextureFormat::BC3RGBAUnormSrgb }, // BC3
  236. { wgpu::TextureFormat::BC4RUnorm, wgpu::TextureFormat::Undefined }, // BC4 // BC4RSnorm ??
  237. { wgpu::TextureFormat::BC5RGUnorm, wgpu::TextureFormat::Undefined }, // BC5 // BC5RGSnorm ??
  238. { wgpu::TextureFormat::BC6HRGBUfloat, wgpu::TextureFormat::Undefined }, // BC6H // BC6HRGBSfloat ??
  239. { wgpu::TextureFormat::BC7RGBAUnorm, wgpu::TextureFormat::BC7RGBAUnormSrgb }, // BC7
  240. { wgpu::TextureFormat::Undefined, wgpu::TextureFormat::Undefined }, // ETC1
  241. { wgpu::TextureFormat::Undefined, wgpu::TextureFormat::Undefined }, // ETC2
  242. { wgpu::TextureFormat::Undefined, wgpu::TextureFormat::Undefined }, // ETC2A
  243. { wgpu::TextureFormat::Undefined, wgpu::TextureFormat::Undefined }, // ETC2A1
  244. { wgpu::TextureFormat::Undefined, wgpu::TextureFormat::Undefined }, // PTC12
  245. { wgpu::TextureFormat::Undefined, wgpu::TextureFormat::Undefined }, // PTC14
  246. { wgpu::TextureFormat::Undefined, wgpu::TextureFormat::Undefined }, // PTC12A
  247. { wgpu::TextureFormat::Undefined, wgpu::TextureFormat::Undefined }, // PTC14A
  248. { wgpu::TextureFormat::Undefined, wgpu::TextureFormat::Undefined }, // PTC22
  249. { wgpu::TextureFormat::Undefined, wgpu::TextureFormat::Undefined }, // PTC24
  250. { wgpu::TextureFormat::Undefined, wgpu::TextureFormat::Undefined }, // ATC
  251. { wgpu::TextureFormat::Undefined, wgpu::TextureFormat::Undefined }, // ATCE
  252. { wgpu::TextureFormat::Undefined, wgpu::TextureFormat::Undefined }, // ATCI
  253. { wgpu::TextureFormat::Undefined, wgpu::TextureFormat::Undefined }, // ASTC4x4
  254. { wgpu::TextureFormat::Undefined, wgpu::TextureFormat::Undefined }, // ASTC5x5
  255. { wgpu::TextureFormat::Undefined, wgpu::TextureFormat::Undefined }, // ASTC6x6
  256. { wgpu::TextureFormat::Undefined, wgpu::TextureFormat::Undefined }, // ASTC8x5
  257. { wgpu::TextureFormat::Undefined, wgpu::TextureFormat::Undefined }, // ASTC8x6
  258. { wgpu::TextureFormat::Undefined, wgpu::TextureFormat::Undefined }, // ASTC10x5
  259. { wgpu::TextureFormat::Undefined, wgpu::TextureFormat::Undefined }, // Unknown
  260. { wgpu::TextureFormat::Undefined, wgpu::TextureFormat::Undefined }, // R1
  261. { wgpu::TextureFormat::Undefined, wgpu::TextureFormat::Undefined }, // A8
  262. { wgpu::TextureFormat::R8Unorm, wgpu::TextureFormat::Undefined }, // R8
  263. { wgpu::TextureFormat::R8Sint, wgpu::TextureFormat::Undefined }, // R8I
  264. { wgpu::TextureFormat::R8Uint, wgpu::TextureFormat::Undefined }, // R8U
  265. { wgpu::TextureFormat::R8Snorm, wgpu::TextureFormat::Undefined }, // R8S
  266. { wgpu::TextureFormat::Undefined, wgpu::TextureFormat::Undefined }, // R16
  267. { wgpu::TextureFormat::R16Sint, wgpu::TextureFormat::Undefined }, // R16I
  268. { wgpu::TextureFormat::R16Uint, wgpu::TextureFormat::Undefined }, // R16U
  269. { wgpu::TextureFormat::R16Float, wgpu::TextureFormat::Undefined }, // R16F
  270. { wgpu::TextureFormat::Undefined, wgpu::TextureFormat::Undefined }, // R16S
  271. { wgpu::TextureFormat::R32Sint, wgpu::TextureFormat::Undefined }, // R32I
  272. { wgpu::TextureFormat::R32Uint, wgpu::TextureFormat::Undefined }, // R32U
  273. { wgpu::TextureFormat::R32Float, wgpu::TextureFormat::Undefined }, // R32F
  274. { wgpu::TextureFormat::RG8Unorm, wgpu::TextureFormat::Undefined }, // RG8
  275. { wgpu::TextureFormat::RG8Sint, wgpu::TextureFormat::Undefined }, // RG8I
  276. { wgpu::TextureFormat::RG8Uint, wgpu::TextureFormat::Undefined }, // RG8U
  277. { wgpu::TextureFormat::RG8Snorm, wgpu::TextureFormat::Undefined }, // RG8S
  278. { wgpu::TextureFormat::Undefined, wgpu::TextureFormat::Undefined }, // RG16
  279. { wgpu::TextureFormat::RG16Sint, wgpu::TextureFormat::Undefined }, // RG16I
  280. { wgpu::TextureFormat::RG16Uint, wgpu::TextureFormat::Undefined }, // RG16U
  281. { wgpu::TextureFormat::RG16Float, wgpu::TextureFormat::Undefined }, // RG16F
  282. { wgpu::TextureFormat::Undefined, wgpu::TextureFormat::Undefined }, // RG16S
  283. { wgpu::TextureFormat::RG32Sint, wgpu::TextureFormat::Undefined }, // RG32I
  284. { wgpu::TextureFormat::RG32Uint, wgpu::TextureFormat::Undefined }, // RG32U
  285. { wgpu::TextureFormat::RG32Float, wgpu::TextureFormat::Undefined }, // RG32F
  286. { wgpu::TextureFormat::Undefined, wgpu::TextureFormat::Undefined }, // RGB8
  287. { wgpu::TextureFormat::Undefined, wgpu::TextureFormat::Undefined }, // RGB8I
  288. { wgpu::TextureFormat::Undefined, wgpu::TextureFormat::Undefined }, // RGB8U
  289. { wgpu::TextureFormat::Undefined, wgpu::TextureFormat::Undefined }, // RGB8S
  290. { wgpu::TextureFormat::Undefined, wgpu::TextureFormat::Undefined }, // RGB9E5F
  291. { wgpu::TextureFormat::BGRA8Unorm, wgpu::TextureFormat::BGRA8UnormSrgb }, // BGRA8
  292. { wgpu::TextureFormat::RGBA8Unorm, wgpu::TextureFormat::RGBA8UnormSrgb }, // RGBA8
  293. { wgpu::TextureFormat::RGBA8Sint, wgpu::TextureFormat::Undefined }, // RGBA8I
  294. { wgpu::TextureFormat::RGBA8Uint, wgpu::TextureFormat::Undefined }, // RGBA8U
  295. { wgpu::TextureFormat::RGBA8Snorm, wgpu::TextureFormat::Undefined }, // RGBA8S
  296. { wgpu::TextureFormat::Undefined, wgpu::TextureFormat::Undefined }, // RGBA16
  297. { wgpu::TextureFormat::RGBA16Sint, wgpu::TextureFormat::Undefined }, // RGBA16I
  298. { wgpu::TextureFormat::RGBA16Uint, wgpu::TextureFormat::Undefined }, // RGBA16U
  299. { wgpu::TextureFormat::RGBA16Float, wgpu::TextureFormat::Undefined }, // RGBA16F
  300. { wgpu::TextureFormat::Undefined, wgpu::TextureFormat::Undefined }, // RGBA16S
  301. { wgpu::TextureFormat::RGBA32Sint, wgpu::TextureFormat::Undefined }, // RGBA32I
  302. { wgpu::TextureFormat::RGBA32Uint, wgpu::TextureFormat::Undefined }, // RGBA32U
  303. { wgpu::TextureFormat::RGBA32Float, wgpu::TextureFormat::Undefined }, // RGBA32F
  304. { wgpu::TextureFormat::Undefined, wgpu::TextureFormat::Undefined }, // R5G6B5
  305. { wgpu::TextureFormat::Undefined, wgpu::TextureFormat::Undefined }, // RGBA4
  306. { wgpu::TextureFormat::Undefined, wgpu::TextureFormat::Undefined }, // RGB5A1
  307. { wgpu::TextureFormat::RGB10A2Unorm, wgpu::TextureFormat::Undefined }, // RGB10A2
  308. { wgpu::TextureFormat::RG11B10Float, wgpu::TextureFormat::Undefined }, // RG11B10F
  309. { wgpu::TextureFormat::Undefined, wgpu::TextureFormat::Undefined }, // UnknownDepth
  310. { wgpu::TextureFormat::Undefined, wgpu::TextureFormat::Undefined }, // D16
  311. { wgpu::TextureFormat::Depth24Plus, wgpu::TextureFormat::Undefined }, // D24
  312. { wgpu::TextureFormat::Depth24PlusStencil8, wgpu::TextureFormat::Undefined }, // D24S8
  313. { wgpu::TextureFormat::Undefined, wgpu::TextureFormat::Undefined }, // D32
  314. { wgpu::TextureFormat::Undefined, wgpu::TextureFormat::Undefined }, // D16F
  315. { wgpu::TextureFormat::Undefined, wgpu::TextureFormat::Undefined }, // D24F
  316. { wgpu::TextureFormat::Depth32Float, wgpu::TextureFormat::Undefined }, // D32F
  317. { wgpu::TextureFormat::Undefined, wgpu::TextureFormat::Undefined }, // D0S8
  318. };
  319. BX_STATIC_ASSERT(TextureFormat::Count == BX_COUNTOF(s_textureFormat));
  320. int32_t s_msaa[] =
  321. {
  322. 1,
  323. 2,
  324. 4,
  325. 8,
  326. 16,
  327. };
  328. struct RendererContextWgpu;
  329. static RendererContextWgpu* s_renderWgpu;
  330. static bool s_ignoreError = false;
  331. #if !BX_PLATFORM_EMSCRIPTEN
  332. DawnSwapChainImplementation(*createSwapChain)(wgpu::Device device, void* nwh);
  333. # if defined(DAWN_ENABLE_BACKEND_D3D12)
  334. DawnSwapChainImplementation CreateSwapChainD3D12(wgpu::Device device, void* nwh)
  335. {
  336. HWND win32Window = (HWND)nwh;
  337. return dawn_native::d3d12::CreateNativeSwapChainImpl(device.Get(), win32Window);
  338. }
  339. # endif // defined(DAWN_ENABLE_BACKEND_D3D12)
  340. # if defined(DAWN_ENABLE_BACKEND_VULKAN)
  341. DawnSwapChainImplementation CreateSwapChainVulkan(wgpu::Device device, void* nwh)
  342. {
  343. VkInstance instance = dawn_native::vulkan::GetInstance(device.Get());
  344. PFN_vkCreateWin32SurfaceKHR vkCreateWin32SurfaceKHR = (PFN_vkCreateWin32SurfaceKHR)dawn_native::vulkan::GetInstanceProcAddr(device.Get(), "vkCreateWin32SurfaceKHR");
  345. VkSurfaceKHR surface;
  346. # if BX_PLATFORM_WINDOWS
  347. // Copied from renderer_vk.cpp -> needs refactor
  348. {
  349. VkWin32SurfaceCreateInfoKHR sci;
  350. sci.sType = VK_STRUCTURE_TYPE_WIN32_SURFACE_CREATE_INFO_KHR;
  351. sci.pNext = NULL;
  352. sci.flags = 0;
  353. sci.hinstance = (HINSTANCE)GetModuleHandle(NULL);
  354. sci.hwnd = (HWND)nwh;
  355. VkResult result = vkCreateWin32SurfaceKHR(instance, &sci, NULL, &surface);
  356. }
  357. # endif // BX_PLATFORM_WINDOWS
  358. return dawn_native::vulkan::CreateNativeSwapChainImpl(device.Get(), surface);
  359. }
  360. # endif // defined(DAWN_ENABLE_BACKEND_VULKAN)
  361. #endif // !BX_PLATFORM_EMSCRIPTEN
  362. struct RendererContextWgpu : public RendererContextI
  363. {
  364. RendererContextWgpu()
  365. : m_frameIndex(0)
  366. , m_numWindows(0)
  367. , m_rtMsaa(false)
  368. , m_capture(NULL)
  369. , m_captureSize(0)
  370. {
  371. bx::memSet(&m_windows, 0xff, sizeof(m_windows) );
  372. }
  373. ~RendererContextWgpu()
  374. {
  375. }
  376. bool init(const Init& _init)
  377. {
  378. BX_UNUSED(_init);
  379. BX_TRACE("Init.");
  380. if (_init.debug
  381. || _init.profile)
  382. {
  383. m_renderDocDll = loadRenderDoc();
  384. }
  385. setGraphicsDebuggerPresent(NULL != m_renderDocDll);
  386. m_fbh.idx = kInvalidHandle;
  387. bx::memSet(m_uniforms, 0, sizeof(m_uniforms) );
  388. bx::memSet(&m_resolution, 0, sizeof(m_resolution) );
  389. #if !BX_PLATFORM_EMSCRIPTEN
  390. // Default to D3D12, Metal, Vulkan, OpenGL in that order as D3D12 and Metal are the preferred on
  391. // their respective platforms, and Vulkan is preferred to OpenGL
  392. # if defined(DAWN_ENABLE_BACKEND_D3D12)
  393. static dawn_native::BackendType backendType = dawn_native::BackendType::D3D12;
  394. # elif defined(DAWN_ENABLE_BACKEND_METAL)
  395. static dawn_native::BackendType backendType = dawn_native::BackendType::Metal;
  396. # elif defined(DAWN_ENABLE_BACKEND_OPENGL)
  397. static dawn_native::BackendType backendType = dawn_native::BackendType::OpenGL;
  398. # elif defined(DAWN_ENABLE_BACKEND_VULKAN)
  399. static dawn_native::BackendType backendType = dawn_native::BackendType::Vulkan;
  400. # else
  401. # error "Unknown platform."
  402. # endif // defined(DAWN_ENABLE_BACKEND_*)
  403. if (BX_ENABLED(BGFX_CONFIG_DEBUG))
  404. {
  405. m_instance.EnableBackendValidation(true);
  406. }
  407. m_instance.DiscoverDefaultAdapters();
  408. dawn_native::Adapter backendAdapter;
  409. std::vector<dawn_native::Adapter> adapters = m_instance.GetAdapters();
  410. for (dawn_native::Adapter& adapter : adapters)
  411. {
  412. if (adapter.GetBackendType() == backendType)
  413. {
  414. backendAdapter = adapter;
  415. break;
  416. }
  417. }
  418. //BX_ASSERT(adapterIt != adapters.end());
  419. WGPUDevice backendDevice = backendAdapter.CreateDevice();
  420. DawnProcTable backendProcs = dawn_native::GetProcs();
  421. using CreateSwapChain = DawnSwapChainImplementation (*)(wgpu::Device device, void* nwh);
  422. # if defined(DAWN_ENABLE_BACKEND_D3D12)
  423. createSwapChain = CreateSwapChainD3D12;
  424. # elif defined(DAWN_ENABLE_BACKEND_METAL)
  425. createSwapChain = CreateSwapChainMetal;
  426. # elif defined(DAWN_ENABLE_BACKEND_NULL)
  427. createSwapChain = CreateSwapChainNull;
  428. # elif defined(DAWN_ENABLE_BACKEND_OPENGL)
  429. createSwapChain = CreateSwapChainOpenGL;
  430. # elif defined(DAWN_ENABLE_BACKEND_VULKAN)
  431. createSwapChain = CreateSwapChainVulkan;
  432. # endif // defined(DAWN_ENABLE_BACKEND_*)
  433. // Choose whether to use the backend procs and devices directly, or set up the wire.
  434. WGPUDevice cDevice = backendDevice;
  435. DawnProcTable procs = backendProcs;
  436. dawnProcSetProcs(&procs);
  437. m_device = wgpu::Device::Acquire(cDevice);
  438. #else
  439. m_device = wgpu::Device(emscripten_webgpu_get_device());
  440. #endif // !BX_PLATFORM_EMSCRIPTEN
  441. auto PrintDeviceError = [](WGPUErrorType errorType, const char* message, void*) {
  442. BX_UNUSED(errorType);
  443. if (s_ignoreError)
  444. {
  445. BX_TRACE("Device error: %s", message);
  446. }
  447. else
  448. {
  449. BX_ASSERT(false, "Device error: %s", message);
  450. }
  451. s_ignoreError = false;
  452. };
  453. m_device.SetUncapturedErrorCallback(PrintDeviceError, NULL);
  454. if (!m_device)
  455. {
  456. BX_WARN(!m_device, "Unable to create WebGPU device.");
  457. return false;
  458. }
  459. bool success = m_mainFrameBuffer.create(
  460. 0
  461. , g_platformData.nwh
  462. , _init.resolution.width
  463. , _init.resolution.height
  464. , TextureFormat::Unknown
  465. , TextureFormat::UnknownDepth
  466. );
  467. m_numWindows = 1;
  468. if (!success)
  469. {
  470. return false;
  471. }
  472. m_queue = m_device.GetDefaultQueue();
  473. m_cmd.init(m_queue);
  474. //BGFX_FATAL(NULL != m_cmd.m_commandQueue, Fatal::UnableToInitialize, "Unable to create Metal device.");
  475. for (uint8_t ii = 0; ii < WEBGPU_MAX_FRAMES_IN_FLIGHT; ++ii)
  476. {
  477. BX_TRACE("Create scratch buffer %d", ii);
  478. m_scratchBuffers[ii].create(BGFX_CONFIG_MAX_DRAW_CALLS * 128);
  479. m_bindStateCache[ii].create(); // (1024);
  480. }
  481. for (uint8_t ii = 0; ii < WEBGPU_NUM_UNIFORM_BUFFERS; ++ii)
  482. {
  483. bool mapped = true; // ii == WEBGPU_NUM_UNIFORM_BUFFERS - 1;
  484. m_uniformBuffers[ii].create(BGFX_CONFIG_MAX_DRAW_CALLS * 128, mapped);
  485. }
  486. g_caps.supported |= (0
  487. | BGFX_CAPS_ALPHA_TO_COVERAGE
  488. | BGFX_CAPS_BLEND_INDEPENDENT
  489. | BGFX_CAPS_FRAGMENT_DEPTH
  490. | BGFX_CAPS_INDEX32
  491. | BGFX_CAPS_INSTANCING
  492. // | BGFX_CAPS_OCCLUSION_QUERY
  493. | BGFX_CAPS_SWAP_CHAIN
  494. | BGFX_CAPS_TEXTURE_2D_ARRAY
  495. // | BGFX_CAPS_TEXTURE_3D
  496. | BGFX_CAPS_TEXTURE_BLIT
  497. | BGFX_CAPS_TEXTURE_COMPARE_ALL
  498. | BGFX_CAPS_TEXTURE_COMPARE_LEQUAL
  499. | BGFX_CAPS_TEXTURE_READ_BACK
  500. | BGFX_CAPS_VERTEX_ATTRIB_HALF
  501. | BGFX_CAPS_VERTEX_ATTRIB_UINT10
  502. | BGFX_CAPS_COMPUTE
  503. );
  504. g_caps.limits.maxTextureSize = 8192;
  505. g_caps.limits.maxFBAttachments = 4;
  506. g_caps.supported |= BGFX_CAPS_TEXTURE_CUBE_ARRAY;
  507. g_caps.supported |= BGFX_CAPS_DRAW_INDIRECT;
  508. g_caps.limits.maxTextureLayers = 2048;
  509. g_caps.limits.maxVertexStreams = BGFX_CONFIG_MAX_VERTEX_STREAMS;
  510. // Maximum number of entries in the buffer argument table, per graphics or compute function are 31.
  511. // It is decremented by 1 because 1 entry is used for uniforms.
  512. g_caps.limits.maxComputeBindings = bx::uint32_min(30, BGFX_MAX_COMPUTE_BINDINGS);
  513. for (uint32_t ii = 0; ii < TextureFormat::Count; ++ii)
  514. {
  515. uint16_t support = 0;
  516. support |= wgpu::TextureFormat::Undefined != s_textureFormat[ii].m_fmt
  517. ? BGFX_CAPS_FORMAT_TEXTURE_2D
  518. | BGFX_CAPS_FORMAT_TEXTURE_3D
  519. | BGFX_CAPS_FORMAT_TEXTURE_CUBE
  520. | BGFX_CAPS_FORMAT_TEXTURE_VERTEX
  521. : BGFX_CAPS_FORMAT_TEXTURE_NONE
  522. ;
  523. support |= wgpu::TextureFormat::Undefined != s_textureFormat[ii].m_fmtSrgb
  524. ? BGFX_CAPS_FORMAT_TEXTURE_2D_SRGB
  525. | BGFX_CAPS_FORMAT_TEXTURE_3D_SRGB
  526. | BGFX_CAPS_FORMAT_TEXTURE_CUBE_SRGB
  527. | BGFX_CAPS_FORMAT_TEXTURE_VERTEX
  528. : BGFX_CAPS_FORMAT_TEXTURE_NONE
  529. ;
  530. if (!bimg::isCompressed(bimg::TextureFormat::Enum(ii) ) )
  531. {
  532. support |= 0
  533. | BGFX_CAPS_FORMAT_TEXTURE_FRAMEBUFFER
  534. // | BGFX_CAPS_FORMAT_TEXTURE_FRAMEBUFFER_MSAA
  535. ;
  536. }
  537. g_caps.formats[ii] = support;
  538. }
  539. g_caps.formats[TextureFormat::A8 ] &= ~(BGFX_CAPS_FORMAT_TEXTURE_FRAMEBUFFER | BGFX_CAPS_FORMAT_TEXTURE_FRAMEBUFFER_MSAA);
  540. g_caps.formats[TextureFormat::RG32I ] &= ~(BGFX_CAPS_FORMAT_TEXTURE_FRAMEBUFFER_MSAA);
  541. g_caps.formats[TextureFormat::RG32U ] &= ~(BGFX_CAPS_FORMAT_TEXTURE_FRAMEBUFFER_MSAA);
  542. g_caps.formats[TextureFormat::RGBA32I] &= ~(BGFX_CAPS_FORMAT_TEXTURE_FRAMEBUFFER_MSAA);
  543. g_caps.formats[TextureFormat::RGBA32U] &= ~(BGFX_CAPS_FORMAT_TEXTURE_FRAMEBUFFER_MSAA);
  544. g_caps.formats[TextureFormat::ETC2 ] =
  545. g_caps.formats[TextureFormat::ETC2A ] =
  546. g_caps.formats[TextureFormat::ETC2A1] =
  547. g_caps.formats[TextureFormat::PTC12 ] =
  548. g_caps.formats[TextureFormat::PTC14 ] =
  549. g_caps.formats[TextureFormat::PTC12A] =
  550. g_caps.formats[TextureFormat::PTC14A] =
  551. g_caps.formats[TextureFormat::R5G6B5] =
  552. g_caps.formats[TextureFormat::RGBA4 ] =
  553. g_caps.formats[TextureFormat::RGB5A1] = BGFX_CAPS_FORMAT_TEXTURE_NONE;
  554. g_caps.formats[TextureFormat::RGB9E5F] &= ~(BGFX_CAPS_FORMAT_TEXTURE_FRAMEBUFFER | BGFX_CAPS_FORMAT_TEXTURE_FRAMEBUFFER_MSAA);
  555. // disable compressed formats
  556. for (uint32_t ii = 0; ii < TextureFormat::Unknown; ++ii)
  557. {
  558. s_textureFormat[ii].m_fmt = wgpu::TextureFormat::Undefined;
  559. }
  560. for (uint32_t ii = 0; ii < TextureFormat::Count; ++ii)
  561. {
  562. if (BGFX_CAPS_FORMAT_TEXTURE_NONE == g_caps.formats[ii])
  563. {
  564. s_textureFormat[ii].m_fmt = wgpu::TextureFormat::Undefined;
  565. s_textureFormat[ii].m_fmtSrgb = wgpu::TextureFormat::Undefined;
  566. }
  567. }
  568. for (uint32_t ii = 1, last = 0; ii < BX_COUNTOF(s_msaa); ++ii)
  569. {
  570. // TODO (hugoam)
  571. //const int32_t sampleCount = 1; //1<<ii;
  572. //if (m_device.supportsTextureSampleCount(sampleCount) )
  573. //{
  574. // s_msaa[ii] = sampleCount;
  575. // last = ii;
  576. //}
  577. //else
  578. {
  579. s_msaa[ii] = s_msaa[last];
  580. }
  581. }
  582. // Init reserved part of view name.
  583. for (uint32_t ii = 0; ii < BGFX_CONFIG_MAX_VIEWS; ++ii)
  584. {
  585. bx::snprintf(s_viewName[ii], BGFX_CONFIG_MAX_VIEW_NAME_RESERVED+1, "%3d ", ii);
  586. }
  587. m_gpuTimer.init();
  588. g_internalData.context = &m_device;
  589. return true;
  590. }
  591. void shutdown()
  592. {
  593. m_gpuTimer.shutdown();
  594. m_pipelineStateCache.invalidate();
  595. for (uint32_t ii = 0; ii < BX_COUNTOF(m_shaders); ++ii)
  596. {
  597. m_shaders[ii].destroy();
  598. }
  599. for (uint32_t ii = 0; ii < BX_COUNTOF(m_textures); ++ii)
  600. {
  601. m_textures[ii].destroy();
  602. }
  603. captureFinish();
  604. m_mainFrameBuffer.destroy();
  605. for (uint32_t ii = 0; ii < BX_COUNTOF(m_scratchBuffers); ++ii)
  606. {
  607. m_scratchBuffers[ii].destroy();
  608. }
  609. m_cmd.shutdown();
  610. }
  611. RendererType::Enum getRendererType() const override
  612. {
  613. return RendererType::WebGPU;
  614. }
  615. const char* getRendererName() const override
  616. {
  617. return BGFX_RENDERER_WEBGPU_NAME;
  618. }
  619. void createIndexBuffer(IndexBufferHandle _handle, const Memory* _mem, uint16_t _flags) override
  620. {
  621. m_indexBuffers[_handle.idx].create(_mem->size, _mem->data, _flags);
  622. }
  623. void destroyIndexBuffer(IndexBufferHandle _handle) override
  624. {
  625. m_indexBuffers[_handle.idx].destroy();
  626. }
  627. void createVertexLayout(VertexLayoutHandle _handle, const VertexLayout& _decl) override
  628. {
  629. VertexLayout& decl = m_vertexDecls[_handle.idx];
  630. bx::memCopy(&decl, &_decl, sizeof(VertexLayout) );
  631. dump(decl);
  632. }
  633. void destroyVertexLayout(VertexLayoutHandle /*_handle*/) override
  634. {
  635. }
  636. void createVertexBuffer(VertexBufferHandle _handle, const Memory* _mem, VertexLayoutHandle _declHandle, uint16_t _flags) override
  637. {
  638. m_vertexBuffers[_handle.idx].create(_mem->size, _mem->data, _declHandle, _flags);
  639. }
  640. void destroyVertexBuffer(VertexBufferHandle _handle) override
  641. {
  642. m_vertexBuffers[_handle.idx].destroy();
  643. }
  644. void createDynamicIndexBuffer(IndexBufferHandle _handle, uint32_t _size, uint16_t _flags) override
  645. {
  646. m_indexBuffers[_handle.idx].create(_size, NULL, _flags);
  647. }
  648. void updateDynamicIndexBuffer(IndexBufferHandle _handle, uint32_t _offset, uint32_t _size, const Memory* _mem) override
  649. {
  650. m_indexBuffers[_handle.idx].update(_offset, bx::uint32_min(_size, _mem->size), _mem->data);
  651. }
  652. void destroyDynamicIndexBuffer(IndexBufferHandle _handle) override
  653. {
  654. m_indexBuffers[_handle.idx].destroy();
  655. }
  656. void createDynamicVertexBuffer(VertexBufferHandle _handle, uint32_t _size, uint16_t _flags) override
  657. {
  658. VertexLayoutHandle decl = BGFX_INVALID_HANDLE;
  659. m_vertexBuffers[_handle.idx].create(_size, NULL, decl, _flags);
  660. }
  661. void updateDynamicVertexBuffer(VertexBufferHandle _handle, uint32_t _offset, uint32_t _size, const Memory* _mem) override
  662. {
  663. m_vertexBuffers[_handle.idx].update(_offset, bx::uint32_min(_size, _mem->size), _mem->data);
  664. }
  665. void destroyDynamicVertexBuffer(VertexBufferHandle _handle) override
  666. {
  667. m_vertexBuffers[_handle.idx].destroy();
  668. }
  669. void createShader(ShaderHandle _handle, const Memory* _mem) override
  670. {
  671. m_shaders[_handle.idx].create(_handle, _mem);
  672. }
  673. void destroyShader(ShaderHandle _handle) override
  674. {
  675. m_shaders[_handle.idx].destroy();
  676. }
  677. void createProgram(ProgramHandle _handle, ShaderHandle _vsh, ShaderHandle _fsh) override
  678. {
  679. m_program[_handle.idx].create(&m_shaders[_vsh.idx], isValid(_fsh) ? &m_shaders[_fsh.idx] : NULL);
  680. }
  681. void destroyProgram(ProgramHandle _handle) override
  682. {
  683. m_program[_handle.idx].destroy();
  684. }
  685. void* createTexture(TextureHandle _handle, const Memory* _mem, uint64_t _flags, uint8_t _skip) override
  686. {
  687. m_textures[_handle.idx].create(_handle, _mem, _flags, _skip);
  688. return NULL;
  689. }
  690. void updateTextureBegin(TextureHandle /*_handle*/, uint8_t /*_side*/, uint8_t /*_mip*/) override
  691. {
  692. }
  693. void updateTexture(TextureHandle _handle, uint8_t _side, uint8_t _mip, const Rect& _rect, uint16_t _z, uint16_t _depth, uint16_t _pitch, const Memory* _mem) override
  694. {
  695. m_textures[_handle.idx].update(_side, _mip, _rect, _z, _depth, _pitch, _mem);
  696. }
  697. void updateTextureEnd() override
  698. {
  699. }
  700. void readback(ReadbackWgpu& readback, const TextureWgpu& texture, void* _data)
  701. {
  702. m_cmd.kick(false, true);
  703. m_cmd.beginRender();
  704. if (readback.m_mapped)
  705. return;
  706. BX_ASSERT(readback.m_mip<texture.m_numMips,"Invalid mip: %d num mips:", readback.m_mip,texture.m_numMips);
  707. uint32_t srcWidth = bx::uint32_max(1, texture.m_width >> readback.m_mip);
  708. uint32_t srcHeight = bx::uint32_max(1, texture.m_height >> readback.m_mip);
  709. const uint32_t bpp = bimg::getBitsPerPixel(bimg::TextureFormat::Enum(texture.m_textureFormat));
  710. const uint32_t pitch = srcWidth * bpp / 8;
  711. const uint32_t dstpitch = bx::strideAlign(pitch, kMinBufferOffsetAlignment);
  712. // TODO move inside ReadbackWgpu::create
  713. if (!readback.m_buffer)
  714. {
  715. wgpu::BufferDescriptor desc;
  716. desc.size = dstpitch * srcHeight;
  717. desc.usage = wgpu::BufferUsage::CopyDst | wgpu::BufferUsage::MapRead;
  718. readback.m_buffer = m_device.CreateBuffer(&desc);
  719. }
  720. wgpu::TextureCopyView textureCopyView;
  721. textureCopyView.texture = texture.m_ptr;
  722. textureCopyView.origin = { 0, 0, 0 };
  723. wgpu::BufferCopyView bufferCopyView;
  724. bufferCopyView.buffer = readback.m_buffer;
  725. bufferCopyView.bytesPerRow = dstpitch;
  726. bufferCopyView.rowsPerImage = srcHeight;
  727. wgpu::Extent3D extent3D = { srcWidth, srcHeight, 1 };
  728. getBlitCommandEncoder().CopyTextureToBuffer(&textureCopyView, &bufferCopyView, &extent3D);
  729. auto finish = [](WGPUBufferMapAsyncStatus status, void const* data, uint64_t dataLength, void* userdata)
  730. {
  731. if(status == WGPUBufferMapAsyncStatus_Success)
  732. static_cast<ReadbackWgpu*>(userdata)->readback(data, dataLength);
  733. };
  734. m_cmd.finish();
  735. m_cmd.kick(true);
  736. readback.m_mapped = true;
  737. readback.m_data = _data;
  738. readback.m_size = pitch * srcHeight;
  739. readback.m_buffer.MapReadAsync(finish, &readback);
  740. }
  741. void readTexture(TextureHandle _handle, void* _data, uint8_t _mip) override
  742. {
  743. TextureWgpu& texture = m_textures[_handle.idx];
  744. readback(texture.m_readback, texture, _data);
  745. }
  746. void resizeTexture(TextureHandle _handle, uint16_t _width, uint16_t _height, uint8_t _numMips, uint16_t _numLayers) override
  747. {
  748. TextureWgpu& texture = m_textures[_handle.idx];
  749. uint32_t size = sizeof(uint32_t) + sizeof(TextureCreate);
  750. const Memory* mem = alloc(size);
  751. bx::StaticMemoryBlockWriter writer(mem->data, mem->size);
  752. uint32_t magic = BGFX_CHUNK_MAGIC_TEX;
  753. bx::write(&writer, magic);
  754. TextureCreate tc;
  755. tc.m_width = _width;
  756. tc.m_height = _height;
  757. tc.m_depth = 0;
  758. tc.m_numLayers = _numLayers;
  759. tc.m_numMips = _numMips;
  760. tc.m_format = TextureFormat::Enum(texture.m_requestedFormat);
  761. tc.m_cubeMap = false;
  762. tc.m_mem = NULL;
  763. bx::write(&writer, tc);
  764. texture.destroy();
  765. texture.create(_handle, mem, texture.m_flags, 0);
  766. release(mem);
  767. }
  768. void overrideInternal(TextureHandle _handle, uintptr_t _ptr) override
  769. {
  770. BX_UNUSED(_handle, _ptr);
  771. }
  772. uintptr_t getInternal(TextureHandle _handle) override
  773. {
  774. BX_UNUSED(_handle);
  775. return 0;
  776. }
  777. void destroyTexture(TextureHandle _handle) override
  778. {
  779. m_textures[_handle.idx].destroy();
  780. }
  781. void createFrameBuffer(FrameBufferHandle _handle, uint8_t _num, const Attachment* _attachment) override
  782. {
  783. m_frameBuffers[_handle.idx].create(_num, _attachment);
  784. }
  785. void createFrameBuffer(FrameBufferHandle _handle, void* _nwh, uint32_t _width, uint32_t _height, TextureFormat::Enum _format, TextureFormat::Enum _depthFormat) override
  786. {
  787. for (uint32_t ii = 0, num = m_numWindows; ii < num; ++ii)
  788. {
  789. FrameBufferHandle handle = m_windows[ii];
  790. if (isValid(handle)
  791. && m_frameBuffers[handle.idx].m_nwh == _nwh)
  792. {
  793. destroyFrameBuffer(handle);
  794. }
  795. }
  796. uint16_t denseIdx = m_numWindows++;
  797. m_windows[denseIdx] = _handle;
  798. FrameBufferWgpu& fb = m_frameBuffers[_handle.idx];
  799. fb.create(denseIdx, _nwh, _width, _height, _format, _depthFormat);
  800. fb.m_swapChain->resize(m_frameBuffers[_handle.idx], _width, _height, 0);
  801. }
  802. void destroyFrameBuffer(FrameBufferHandle _handle) override
  803. {
  804. uint16_t denseIdx = m_frameBuffers[_handle.idx].destroy();
  805. if (UINT16_MAX != denseIdx)
  806. {
  807. --m_numWindows;
  808. if (m_numWindows > 1)
  809. {
  810. FrameBufferHandle handle = m_windows[m_numWindows];
  811. m_windows[m_numWindows] = {kInvalidHandle};
  812. if (m_numWindows != denseIdx)
  813. {
  814. m_windows[denseIdx] = handle;
  815. m_frameBuffers[handle.idx].m_denseIdx = denseIdx;
  816. }
  817. }
  818. }
  819. }
  820. void createUniform(UniformHandle _handle, UniformType::Enum _type, uint16_t _num, const char* _name) override
  821. {
  822. if (NULL != m_uniforms[_handle.idx])
  823. {
  824. BX_FREE(g_allocator, m_uniforms[_handle.idx]);
  825. }
  826. uint32_t size = bx::alignUp(g_uniformTypeSize[_type]*_num, 16);
  827. void* data = BX_ALLOC(g_allocator, size);
  828. bx::memSet(data, 0, size);
  829. m_uniforms[_handle.idx] = data;
  830. m_uniformReg.add(_handle, _name);
  831. }
  832. void destroyUniform(UniformHandle _handle) override
  833. {
  834. BX_FREE(g_allocator, m_uniforms[_handle.idx]);
  835. m_uniforms[_handle.idx] = NULL;
  836. m_uniformReg.remove(_handle);
  837. }
  838. void requestScreenShotPre(const char* _filePath)
  839. {
  840. BX_UNUSED(_filePath);
  841. //m_saveScreenshot = true;
  842. }
  843. void requestScreenShot(FrameBufferHandle _handle, const char* _filePath) override
  844. {
  845. BX_UNUSED(_handle); BX_UNUSED(_filePath);
  846. }
  847. void updateViewName(ViewId _id, const char* _name) override
  848. {
  849. bx::strCopy(
  850. &s_viewName[_id][BGFX_CONFIG_MAX_VIEW_NAME_RESERVED]
  851. , BX_COUNTOF(s_viewName[0])-BGFX_CONFIG_MAX_VIEW_NAME_RESERVED
  852. , _name
  853. );
  854. }
  855. void updateUniform(uint16_t _loc, const void* _data, uint32_t _size) override
  856. {
  857. bx::memCopy(m_uniforms[_loc], _data, _size);
  858. }
  859. void invalidateOcclusionQuery(OcclusionQueryHandle _handle) override
  860. {
  861. BX_UNUSED(_handle);
  862. }
  863. void setMarker(const char* _marker, uint16_t _len) override
  864. {
  865. BX_UNUSED(_len);
  866. if (BX_ENABLED(BGFX_CONFIG_DEBUG_ANNOTATION) )
  867. {
  868. m_renderEncoder.InsertDebugMarker(_marker);
  869. }
  870. }
  871. virtual void setName(Handle _handle, const char* _name, uint16_t _len) override
  872. {
  873. BX_UNUSED(_handle); BX_UNUSED(_name); BX_UNUSED(_len);
  874. BX_UNUSED(_len);
  875. switch (_handle.type)
  876. {
  877. case Handle::IndexBuffer:
  878. m_indexBuffers[_handle.idx].m_label.clear();
  879. m_indexBuffers[_handle.idx].m_label.append(_name);
  880. break;
  881. case Handle::Shader:
  882. m_shaders[_handle.idx].m_label.clear();
  883. m_shaders[_handle.idx].m_label.append(_name);
  884. break;
  885. case Handle::Texture:
  886. m_textures[_handle.idx].m_label.clear();
  887. m_textures[_handle.idx].m_label.append(_name);
  888. break;
  889. case Handle::VertexBuffer:
  890. m_vertexBuffers[_handle.idx].m_label.clear();
  891. m_vertexBuffers[_handle.idx].m_label.append(_name);
  892. break;
  893. default:
  894. BX_ASSERT(false, "Invalid handle type?! %d", _handle.type);
  895. break;
  896. }
  897. }
  898. void submitBlit(BlitState& _bs, uint16_t _view);
  899. void submit(Frame* _render, ClearQuad& _clearQuad, TextVideoMemBlitter& _textVideoMemBlitter) override;
  900. void blitSetup(TextVideoMemBlitter& _blitter) override
  901. {
  902. BX_UNUSED(_blitter);
  903. }
  904. void blitRender(TextVideoMemBlitter& _blitter, uint32_t _numIndices) override
  905. {
  906. const uint32_t numVertices = _numIndices*4/6;
  907. if (0 < numVertices)
  908. {
  909. m_indexBuffers [_blitter.m_ib->handle.idx].update(
  910. 0
  911. , bx::strideAlign(_numIndices*2, 4)
  912. , _blitter.m_ib->data
  913. , true
  914. );
  915. m_vertexBuffers[_blitter.m_vb->handle.idx].update(
  916. 0
  917. , numVertices*_blitter.m_layout.m_stride
  918. , _blitter.m_vb->data
  919. , true
  920. );
  921. endEncoding();
  922. uint32_t width = m_resolution.width;
  923. uint32_t height = m_resolution.height;
  924. FrameBufferHandle fbh = BGFX_INVALID_HANDLE;
  925. uint64_t state = 0
  926. | BGFX_STATE_WRITE_RGB
  927. | BGFX_STATE_WRITE_A
  928. | BGFX_STATE_DEPTH_TEST_ALWAYS
  929. ;
  930. PipelineStateWgpu* pso = getPipelineState(
  931. state
  932. , 0
  933. , 0
  934. , fbh
  935. , _blitter.m_vb->layoutHandle
  936. , false
  937. , _blitter.m_program
  938. , 0
  939. );
  940. RenderPassDescriptor renderPassDescriptor;
  941. wgpu::RenderPassColorAttachmentDescriptor& color = renderPassDescriptor.colorAttachments[0];
  942. setFrameBuffer(renderPassDescriptor, fbh);
  943. color.loadOp = wgpu::LoadOp::Load;
  944. color.storeOp = wgpu::StoreOp::Store;
  945. // NULL != renderPassDescriptor.colorAttachments[0].resolveTexture
  946. // ? wgpu::StoreOp::MultisampleResolve
  947. // : wgpu::StoreOp::Store
  948. //;
  949. wgpu::RenderPassEncoder rce = m_cmd.m_renderEncoder.BeginRenderPass(&renderPassDescriptor.desc);
  950. m_renderEncoder = rce;
  951. rce.SetViewport(0.0f, 0.0f, (float)width, (float)height, 0.0f, 1.0f);
  952. rce.SetScissorRect(0.0f, 0.0f, (float)width, (float)height);
  953. rce.SetPipeline(pso->m_rps);
  954. ProgramWgpu& program = m_program[_blitter.m_program.idx];
  955. ScratchBufferWgpu& scratchBuffer = m_scratchBuffers[0];
  956. BindStateCacheWgpu& bindStates = m_bindStateCache[0];
  957. float proj[16];
  958. bx::mtxOrtho(proj, 0.0f, (float)width, (float)height, 0.0f, 0.0f, 1000.0f, 0.0f, false);
  959. PredefinedUniform& predefined = program.m_predefined[0];
  960. uint8_t flags = predefined.m_type;
  961. setShaderUniform(flags, predefined.m_loc, proj, 4);
  962. BX_ASSERT(program.m_vsh->m_size > 0, "Not supposed to happen");
  963. const uint32_t voffset = scratchBuffer.write(m_vsScratch, program.m_vsh->m_gpuSize);
  964. const uint32_t fsize = (NULL != program.m_fsh ? program.m_fsh->m_gpuSize : 0);
  965. BX_ASSERT(fsize == 0, "Not supposed to happen");
  966. TextureWgpu& texture = m_textures[_blitter.m_texture.idx];
  967. BindingsWgpu b;
  968. BindStateWgpu& bindState = allocBindState(program, bindStates, b, scratchBuffer);
  969. wgpu::BindGroupEntry& textureEntry = b.m_entries[b.numEntries++];
  970. textureEntry.binding = program.m_textures[0].binding;
  971. textureEntry.textureView = texture.m_ptr.CreateView();
  972. wgpu::BindGroupEntry& samplerEntry = b.m_entries[b.numEntries++];
  973. samplerEntry.binding = program.m_samplers[0].binding;
  974. samplerEntry.sampler = 0 == (BGFX_SAMPLER_INTERNAL_DEFAULT & state)
  975. ? getSamplerState(state)
  976. : texture.m_sampler;
  977. bindGroups(program, bindState, b);
  978. uint32_t numOffset = 1;
  979. uint32_t offsets[1] = { voffset };
  980. bindProgram(rce, program, bindState, numOffset, offsets);
  981. VertexBufferWgpu& vb = m_vertexBuffers[_blitter.m_vb->handle.idx];
  982. rce.SetVertexBuffer(0, vb.m_ptr);
  983. rce.SetIndexBuffer(m_indexBuffers[_blitter.m_ib->handle.idx].m_ptr);
  984. rce.DrawIndexed(_numIndices, 1, 0, 0, 0);
  985. }
  986. }
  987. bool isDeviceRemoved() override
  988. {
  989. return false;
  990. }
  991. void flip() override
  992. {
  993. for (uint32_t ii = 0, num = m_numWindows; ii < num; ++ii)
  994. {
  995. FrameBufferWgpu& frameBuffer = ii == 0 ? m_mainFrameBuffer : m_frameBuffers[m_windows[ii].idx];
  996. if (NULL != frameBuffer.m_swapChain)
  997. //&& frameBuffer.m_swapChain->m_drawable)
  998. {
  999. SwapChainWgpu& swapChain = *frameBuffer.m_swapChain;
  1000. swapChain.flip();
  1001. }
  1002. }
  1003. m_cmd.m_stagingEncoder = NULL;
  1004. m_cmd.m_renderEncoder = NULL;
  1005. }
  1006. void updateResolution(const Resolution& _resolution)
  1007. {
  1008. m_resolution = _resolution;
  1009. return; // TODO (hugoam)
  1010. m_mainFrameBuffer.m_swapChain->m_maxAnisotropy = !!(_resolution.reset & BGFX_RESET_MAXANISOTROPY)
  1011. ? 16
  1012. : 1
  1013. ;
  1014. const uint32_t maskFlags = ~(0
  1015. | BGFX_RESET_MAXANISOTROPY
  1016. | BGFX_RESET_DEPTH_CLAMP
  1017. | BGFX_RESET_SUSPEND
  1018. );
  1019. if (m_resolution.width != _resolution.width
  1020. || m_resolution.height != _resolution.height
  1021. || (m_resolution.reset&maskFlags) != (_resolution.reset&maskFlags) )
  1022. {
  1023. wgpu::TextureFormat prevMetalLayerPixelFormat; // = m_mainFrameBuffer.m_swapChain->m_metalLayer.pixelFormat;
  1024. BX_UNUSED(prevMetalLayerPixelFormat);
  1025. m_resolution = _resolution;
  1026. m_resolution.reset &= ~BGFX_RESET_INTERNAL_FORCE;
  1027. m_mainFrameBuffer.m_swapChain->resize(m_mainFrameBuffer, _resolution.width, _resolution.height, _resolution.reset);
  1028. for (uint32_t ii = 0; ii < BX_COUNTOF(m_frameBuffers); ++ii)
  1029. {
  1030. m_frameBuffers[ii].postReset();
  1031. }
  1032. updateCapture();
  1033. m_textVideoMem.resize(false, _resolution.width, _resolution.height);
  1034. m_textVideoMem.clear();
  1035. //if (prevMetalLayerPixelFormat != m_mainFrameBuffer.m_swapChain->m_metalLayer.pixelFormat)
  1036. {
  1037. //MTL_RELEASE(m_screenshotBlitRenderPipelineState)
  1038. //reset(m_renderPipelineDescriptor);
  1039. //m_renderPipelineDescriptor.colorAttachments[0].pixelFormat = m_mainFrameBuffer.m_swapChain->m_metalLayer.pixelFormat;
  1040. //m_renderPipelineDescriptor.vertexFunction = m_screenshotBlitProgram.m_vsh->m_function;
  1041. //m_renderPipelineDescriptor.fragmentFunction = m_screenshotBlitProgram.m_fsh->m_function;
  1042. //m_screenshotBlitRenderPipelineState = m_device.newRenderPipelineStateWithDescriptor(m_renderPipelineDescriptor);
  1043. }
  1044. }
  1045. }
  1046. void invalidateCompute()
  1047. {
  1048. if (m_computeEncoder)
  1049. {
  1050. m_computeEncoder.EndPass();
  1051. m_computeEncoder = NULL;
  1052. }
  1053. }
  1054. void updateCapture()
  1055. {
  1056. }
  1057. void capture()
  1058. {
  1059. }
  1060. void captureFinish()
  1061. {
  1062. }
  1063. BindStateWgpu& allocBindState(const ProgramWgpu& program, BindStateCacheWgpu& bindStates, BindingsWgpu& bindings, ScratchBufferWgpu& scratchBuffer)
  1064. {
  1065. BindStateWgpu& bindState = bindStates.m_bindStates[bindStates.m_currentBindState];
  1066. bindStates.m_currentBindState++;
  1067. bindState.numOffset = program.m_numUniforms;
  1068. // first two bindings are always uniform buffer (vertex/fragment)
  1069. bindings.m_entries[0].binding = 0;
  1070. bindings.m_entries[0].offset = 0;
  1071. bindings.m_entries[0].size = program.m_vsh->m_gpuSize;
  1072. bindings.m_entries[0].buffer = scratchBuffer.m_buffer;
  1073. bindings.numEntries++;
  1074. if (NULL != program.m_fsh
  1075. && 0 < program.m_fsh->m_gpuSize)
  1076. {
  1077. bindings.m_entries[1].binding = 48;
  1078. bindings.m_entries[1].offset = 0;
  1079. bindings.m_entries[1].size = program.m_fsh->m_gpuSize;
  1080. bindings.m_entries[1].buffer = scratchBuffer.m_buffer;
  1081. bindings.numEntries++;
  1082. }
  1083. return bindState;
  1084. }
  1085. void bindGroups(const ProgramWgpu& program, BindStateWgpu& bindState, BindingsWgpu& bindings)
  1086. {
  1087. wgpu::BindGroupDescriptor bindGroupDesc;
  1088. bindGroupDesc.layout = program.m_bindGroupLayout;
  1089. bindGroupDesc.entryCount = bindings.numEntries;
  1090. bindGroupDesc.entries = bindings.m_entries;
  1091. bindState.m_bindGroup = m_device.CreateBindGroup(&bindGroupDesc);
  1092. }
  1093. template <class Encoder>
  1094. void bindProgram(Encoder& encoder, const ProgramWgpu& program, BindStateWgpu& bindState, uint32_t numOffset, uint32_t* offsets)
  1095. {
  1096. BX_ASSERT(bindState.numOffset == numOffset, "We're obviously doing something wrong");
  1097. encoder.SetBindGroup(0, bindState.m_bindGroup, numOffset, offsets);
  1098. }
  1099. BindStateWgpu& allocAndFillBindState(const ProgramWgpu& program, BindStateCacheWgpu& bindStates, ScratchBufferWgpu& scratchBuffer, const RenderBind& renderBind)
  1100. {
  1101. BindingsWgpu b;
  1102. BindStateWgpu& bindState = allocBindState(program, bindStates, b, scratchBuffer);
  1103. for (uint8_t stage = 0; stage < BGFX_CONFIG_MAX_TEXTURE_SAMPLERS; ++stage)
  1104. {
  1105. const Binding& bind = renderBind.m_bind[stage];
  1106. const BindInfo& bindInfo = program.m_bindInfo[stage];
  1107. bool isUsed = isValid(program.m_bindInfo[stage].m_uniform);
  1108. BX_ASSERT(!isUsed || kInvalidHandle != bind.m_idx, "All expected bindings must be bound with WebGPU");
  1109. if (kInvalidHandle != bind.m_idx)
  1110. {
  1111. switch (bind.m_type)
  1112. {
  1113. case Binding::Image:
  1114. {
  1115. TextureWgpu& texture = m_textures[bind.m_idx];
  1116. wgpu::BindGroupEntry& entry = b.m_entries[b.numEntries++];
  1117. entry.binding = bindInfo.m_binding;
  1118. entry.textureView = texture.getTextureMipLevel(bind.m_mip);
  1119. }
  1120. break;
  1121. case Binding::Texture:
  1122. {
  1123. // apparently bgfx allows to set a texture to a stage that a program does not even use
  1124. if (isUsed)
  1125. {
  1126. TextureWgpu& texture = m_textures[bind.m_idx];
  1127. uint32_t flags = bind.m_samplerFlags;
  1128. wgpu::TextureViewDescriptor viewDesc = defaultDescriptor<wgpu::TextureViewDescriptor>();
  1129. viewDesc.dimension = program.m_textures[bindInfo.m_index].viewDimension;
  1130. wgpu::BindGroupEntry& textureEntry = b.m_entries[b.numEntries++];
  1131. textureEntry.binding = bindInfo.m_binding;
  1132. //textureEntry.textureView = texture.m_ptr.CreateView();
  1133. textureEntry.textureView = texture.m_ptr.CreateView(&viewDesc);
  1134. wgpu::BindGroupEntry& samplerEntry = b.m_entries[b.numEntries++];
  1135. samplerEntry.binding = bindInfo.m_binding + 16;
  1136. samplerEntry.sampler = 0 == (BGFX_SAMPLER_INTERNAL_DEFAULT & flags)
  1137. ? getSamplerState(flags)
  1138. : texture.m_sampler;
  1139. }
  1140. }
  1141. break;
  1142. case Binding::IndexBuffer:
  1143. case Binding::VertexBuffer:
  1144. {
  1145. const BufferWgpu& buffer = Binding::IndexBuffer == bind.m_type
  1146. ? m_indexBuffers[bind.m_idx]
  1147. : m_vertexBuffers[bind.m_idx]
  1148. ;
  1149. wgpu::BindGroupEntry& entry = b.m_entries[b.numEntries++];
  1150. entry.binding = bindInfo.m_binding;
  1151. entry.offset = 0;
  1152. entry.size = buffer.m_size;
  1153. entry.buffer = buffer.m_ptr;
  1154. }
  1155. break;
  1156. }
  1157. }
  1158. }
  1159. bindGroups(program, bindState, b);
  1160. return bindState;
  1161. };
  1162. void setShaderUniform(uint8_t _flags, uint32_t _regIndex, const void* _val, uint32_t _numRegs)
  1163. {
  1164. if(_flags&kUniformFragmentBit)
  1165. {
  1166. bx::memCopy(&m_fsScratch[_regIndex], _val, _numRegs * 16);
  1167. }
  1168. else
  1169. {
  1170. bx::memCopy(&m_vsScratch[_regIndex], _val, _numRegs * 16);
  1171. }
  1172. }
  1173. void setShaderUniform4f(uint8_t _flags, uint32_t _loc, const void* _val, uint32_t _numRegs)
  1174. {
  1175. setShaderUniform(_flags, _loc, _val, _numRegs);
  1176. }
  1177. void setShaderUniform4x4f(uint8_t _flags, uint32_t _loc, const void* _val, uint32_t _numRegs)
  1178. {
  1179. setShaderUniform(_flags, _loc, _val, _numRegs);
  1180. }
  1181. void commitShaderConstants(ScratchBufferWgpu& _scratchBuffer, const ProgramWgpu& _program, uint32_t _vertexOffset, uint32_t _fragmentOffset)
  1182. {
  1183. const uint32_t size = _program.m_vsh->m_gpuSize;
  1184. if (0 != size)
  1185. _scratchBuffer.write(m_vsScratch, size);
  1186. if(NULL != _program.m_fsh)
  1187. {
  1188. const uint32_t size = _program.m_fsh->m_gpuSize;
  1189. if(0 != size)
  1190. _scratchBuffer.write(m_fsScratch, size);
  1191. }
  1192. }
  1193. void commit(UniformBuffer& _uniformBuffer)
  1194. {
  1195. _uniformBuffer.reset();
  1196. for (;;)
  1197. {
  1198. uint32_t opcode = _uniformBuffer.read();
  1199. if (UniformType::End == opcode)
  1200. {
  1201. break;
  1202. }
  1203. UniformType::Enum type;
  1204. uint16_t loc;
  1205. uint16_t num;
  1206. uint16_t copy;
  1207. UniformBuffer::decodeOpcode(opcode, type, loc, num, copy);
  1208. const char* data;
  1209. if (copy)
  1210. {
  1211. data = _uniformBuffer.read(g_uniformTypeSize[type]*num);
  1212. }
  1213. else
  1214. {
  1215. UniformHandle handle;
  1216. bx::memCopy(&handle, _uniformBuffer.read(sizeof(UniformHandle) ), sizeof(UniformHandle) );
  1217. data = (const char*)m_uniforms[handle.idx];
  1218. }
  1219. switch ( (uint32_t)type)
  1220. {
  1221. case UniformType::Mat3:
  1222. case UniformType::Mat3|kUniformFragmentBit:
  1223. {
  1224. float* value = (float*)data;
  1225. for (uint32_t ii = 0, count = num/3; ii < count; ++ii, loc += 3*16, value += 9)
  1226. {
  1227. Matrix4 mtx;
  1228. mtx.un.val[ 0] = value[0];
  1229. mtx.un.val[ 1] = value[1];
  1230. mtx.un.val[ 2] = value[2];
  1231. mtx.un.val[ 3] = 0.0f;
  1232. mtx.un.val[ 4] = value[3];
  1233. mtx.un.val[ 5] = value[4];
  1234. mtx.un.val[ 6] = value[5];
  1235. mtx.un.val[ 7] = 0.0f;
  1236. mtx.un.val[ 8] = value[6];
  1237. mtx.un.val[ 9] = value[7];
  1238. mtx.un.val[10] = value[8];
  1239. mtx.un.val[11] = 0.0f;
  1240. setShaderUniform(uint8_t(type), loc, &mtx.un.val[0], 3);
  1241. }
  1242. }
  1243. break;
  1244. case UniformType::Sampler:
  1245. case UniformType::Sampler | kUniformFragmentBit:
  1246. case UniformType::Vec4:
  1247. case UniformType::Vec4 | kUniformFragmentBit:
  1248. case UniformType::Mat4:
  1249. case UniformType::Mat4 | kUniformFragmentBit:
  1250. {
  1251. setShaderUniform(uint8_t(type), loc, data, num);
  1252. }
  1253. break;
  1254. case UniformType::End:
  1255. break;
  1256. default:
  1257. BX_TRACE("%4d: INVALID 0x%08x, t %d, l %d, n %d, c %d", _uniformBuffer.getPos(), opcode, type, loc, num, copy);
  1258. break;
  1259. }
  1260. }
  1261. }
  1262. void clearQuad(ClearQuad& _clearQuad, const Rect& _rect, const Clear& _clear, const float _palette[][4])
  1263. {
  1264. uint32_t width;
  1265. uint32_t height;
  1266. if (isValid(m_fbh) )
  1267. {
  1268. const FrameBufferWgpu& fb = m_frameBuffers[m_fbh.idx];
  1269. width = fb.m_width;
  1270. height = fb.m_height;
  1271. }
  1272. else
  1273. {
  1274. width = m_resolution.width;
  1275. height = m_resolution.height;
  1276. }
  1277. uint64_t state = 0;
  1278. state |= _clear.m_flags & BGFX_CLEAR_COLOR ? BGFX_STATE_WRITE_RGB|BGFX_STATE_WRITE_A : 0;
  1279. state |= _clear.m_flags & BGFX_CLEAR_DEPTH ? BGFX_STATE_DEPTH_TEST_ALWAYS|BGFX_STATE_WRITE_Z : 0;
  1280. state |= BGFX_STATE_PT_TRISTRIP;
  1281. uint64_t stencil = 0;
  1282. stencil |= _clear.m_flags & BGFX_CLEAR_STENCIL ? 0
  1283. | BGFX_STENCIL_TEST_ALWAYS
  1284. | BGFX_STENCIL_FUNC_REF(_clear.m_stencil)
  1285. | BGFX_STENCIL_FUNC_RMASK(0xff)
  1286. | BGFX_STENCIL_OP_FAIL_S_REPLACE
  1287. | BGFX_STENCIL_OP_FAIL_Z_REPLACE
  1288. | BGFX_STENCIL_OP_PASS_Z_REPLACE
  1289. : 0
  1290. ;
  1291. uint32_t numMrt = 1;
  1292. FrameBufferHandle fbh = m_fbh;
  1293. if (isValid(fbh) && m_frameBuffers[fbh.idx].m_swapChain == NULL)
  1294. {
  1295. const FrameBufferWgpu& fb = m_frameBuffers[fbh.idx];
  1296. numMrt = bx::uint32_max(1, fb.m_num);
  1297. }
  1298. wgpu::RenderPassEncoder rce = m_renderEncoder;
  1299. ProgramHandle programHandle = _clearQuad.m_program[numMrt-1];
  1300. const VertexLayout* decl = &_clearQuad.m_layout;
  1301. const PipelineStateWgpu* pso = getPipelineState(
  1302. state
  1303. , stencil
  1304. , 0
  1305. , fbh
  1306. , 1
  1307. , &decl
  1308. , false
  1309. , programHandle
  1310. , 0
  1311. );
  1312. rce.SetPipeline(pso->m_rps);
  1313. float mrtClearColor[BGFX_CONFIG_MAX_FRAME_BUFFER_ATTACHMENTS][4];
  1314. float mrtClearDepth[4] = { _clear.m_depth };
  1315. if (BGFX_CLEAR_COLOR_USE_PALETTE & _clear.m_flags)
  1316. {
  1317. for (uint32_t ii = 0; ii < numMrt; ++ii)
  1318. {
  1319. uint8_t index = (uint8_t)bx::uint32_min(BGFX_CONFIG_MAX_COLOR_PALETTE-1, _clear.m_index[ii]);
  1320. bx::memCopy(mrtClearColor[ii], _palette[index], 16);
  1321. }
  1322. }
  1323. else
  1324. {
  1325. float rgba[4] =
  1326. {
  1327. _clear.m_index[0]*1.0f/255.0f,
  1328. _clear.m_index[1]*1.0f/255.0f,
  1329. _clear.m_index[2]*1.0f/255.0f,
  1330. _clear.m_index[3]*1.0f/255.0f,
  1331. };
  1332. for (uint32_t ii = 0; ii < numMrt; ++ii)
  1333. {
  1334. bx::memCopy( mrtClearColor[ii]
  1335. , rgba
  1336. , 16
  1337. );
  1338. }
  1339. }
  1340. ProgramWgpu& program = m_program[programHandle.idx];
  1341. ScratchBufferWgpu& scratchBuffer = m_scratchBuffers[0];
  1342. BindStateCacheWgpu& bindStates = m_bindStateCache[0];
  1343. BindingsWgpu b;
  1344. BindStateWgpu& bindState = allocBindState(program, bindStates, b, scratchBuffer);
  1345. const uint32_t voffset = scratchBuffer.write(mrtClearDepth, sizeof(mrtClearDepth), program.m_vsh->m_gpuSize);
  1346. const uint32_t foffset = scratchBuffer.write(mrtClearColor, sizeof(mrtClearColor), program.m_fsh->m_gpuSize);
  1347. uint32_t numOffset = 2;
  1348. uint32_t offsets[2] = { voffset, foffset };
  1349. bindGroups(program, bindState, b);
  1350. const VertexBufferWgpu& vb = m_vertexBuffers[_clearQuad.m_vb.idx];
  1351. bindProgram(rce, program, bindState, numOffset, offsets);
  1352. rce.SetViewport(_rect.m_x, _rect.m_y, _rect.m_width, _rect.m_height, 0.0f, 1.0f);
  1353. rce.SetScissorRect(_rect.m_x, _rect.m_y, _rect.m_width, _rect.m_height);
  1354. rce.SetVertexBuffer(0, vb.m_ptr);
  1355. rce.Draw(4, 1, 0, 0);
  1356. }
  1357. wgpu::TextureViewDescriptor attachmentView(const Attachment& _at, const TextureWgpu& _texture)
  1358. {
  1359. bool _resolve = bool(_texture.m_ptrMsaa);
  1360. BX_UNUSED(_resolve);
  1361. wgpu::TextureViewDescriptor desc;
  1362. if (1 < _texture.m_numSides)
  1363. {
  1364. desc.baseArrayLayer = _at.layer;
  1365. }
  1366. desc.baseMipLevel = _at.mip;
  1367. desc.arrayLayerCount = 1;
  1368. desc.mipLevelCount = 1;
  1369. if (_texture.m_type == TextureWgpu::Texture3D)
  1370. {
  1371. desc.dimension = wgpu::TextureViewDimension::e3D;
  1372. }
  1373. return desc;
  1374. }
  1375. void setFrameBuffer(RenderPassDescriptor& _renderPassDescriptor, FrameBufferHandle _fbh, bool _msaa = true)
  1376. {
  1377. if (!isValid(_fbh)
  1378. || m_frameBuffers[_fbh.idx].m_swapChain)
  1379. {
  1380. SwapChainWgpu* swapChain = !isValid(_fbh)
  1381. ? m_mainFrameBuffer.m_swapChain
  1382. : m_frameBuffers[_fbh.idx].m_swapChain
  1383. ;
  1384. _renderPassDescriptor.colorAttachments[0] = defaultDescriptor<wgpu::RenderPassColorAttachmentDescriptor>();
  1385. _renderPassDescriptor.desc.colorAttachmentCount = 1;
  1386. // Force 1 array layers for attachments
  1387. wgpu::TextureViewDescriptor desc;
  1388. desc.arrayLayerCount = 1;
  1389. if (swapChain->m_backBufferColorMsaa)
  1390. {
  1391. _renderPassDescriptor.colorAttachments[0].attachment = swapChain->m_backBufferColorMsaa.CreateView(&desc);
  1392. _renderPassDescriptor.colorAttachments[0].resolveTarget = swapChain->current();
  1393. }
  1394. else
  1395. {
  1396. _renderPassDescriptor.colorAttachments[0].attachment = swapChain->current();
  1397. }
  1398. _renderPassDescriptor.depthStencilAttachment = defaultDescriptor<wgpu::RenderPassDepthStencilAttachmentDescriptor>();
  1399. _renderPassDescriptor.depthStencilAttachment.attachment = swapChain->m_backBufferDepth.CreateView();
  1400. _renderPassDescriptor.desc.depthStencilAttachment = &_renderPassDescriptor.depthStencilAttachment;
  1401. }
  1402. else
  1403. {
  1404. FrameBufferWgpu& frameBuffer = m_frameBuffers[_fbh.idx];
  1405. _renderPassDescriptor.desc.colorAttachmentCount = frameBuffer.m_num;
  1406. for (uint32_t ii = 0; ii < frameBuffer.m_num; ++ii)
  1407. {
  1408. const TextureWgpu& texture = m_textures[frameBuffer.m_colorHandle[ii].idx];
  1409. const wgpu::TextureViewDescriptor desc = attachmentView(frameBuffer.m_colorAttachment[ii], texture);
  1410. _renderPassDescriptor.colorAttachments[ii] = defaultDescriptor<wgpu::RenderPassColorAttachmentDescriptor>();
  1411. _renderPassDescriptor.colorAttachments[ii].attachment = texture.m_ptrMsaa
  1412. ? texture.m_ptrMsaa.CreateView(&desc)
  1413. : texture.m_ptr.CreateView(&desc)
  1414. ;
  1415. _renderPassDescriptor.colorAttachments[ii].resolveTarget = texture.m_ptrMsaa
  1416. ? texture.m_ptr.CreateView(&desc)
  1417. : wgpu::TextureView()
  1418. ;
  1419. }
  1420. if (isValid(frameBuffer.m_depthHandle) )
  1421. {
  1422. const TextureWgpu& texture = m_textures[frameBuffer.m_depthHandle.idx];
  1423. const wgpu::TextureViewDescriptor desc = attachmentView(frameBuffer.m_depthAttachment, texture);
  1424. _renderPassDescriptor.depthStencilAttachment = defaultDescriptor<wgpu::RenderPassDepthStencilAttachmentDescriptor>();
  1425. _renderPassDescriptor.depthStencilAttachment.attachment = texture.m_ptrMsaa
  1426. ? texture.m_ptrMsaa.CreateView(&desc)
  1427. : texture.m_ptr.CreateView(&desc)
  1428. ;
  1429. _renderPassDescriptor.desc.depthStencilAttachment = &_renderPassDescriptor.depthStencilAttachment;
  1430. }
  1431. }
  1432. m_fbh = _fbh;
  1433. m_rtMsaa = _msaa;
  1434. }
  1435. void setDepthStencilState(wgpu::DepthStencilStateDescriptor& desc, uint64_t _state, uint64_t _stencil = 0)
  1436. {
  1437. const uint32_t fstencil = unpackStencil(0, _stencil);
  1438. const uint32_t func = (_state&BGFX_STATE_DEPTH_TEST_MASK) >> BGFX_STATE_DEPTH_TEST_SHIFT;
  1439. desc.depthWriteEnabled = !!(BGFX_STATE_WRITE_Z & _state);
  1440. desc.depthCompare = s_cmpFunc[func];
  1441. uint32_t bstencil = unpackStencil(1, _stencil);
  1442. const uint32_t frontAndBack = bstencil != BGFX_STENCIL_NONE && bstencil != fstencil;
  1443. bstencil = frontAndBack ? bstencil : fstencil;
  1444. desc.stencilFront = defaultDescriptor<wgpu::StencilStateFaceDescriptor>();
  1445. desc.stencilBack = defaultDescriptor<wgpu::StencilStateFaceDescriptor>();
  1446. if (0 != _stencil)
  1447. {
  1448. // TODO (hugoam)
  1449. const uint32_t readMask = (fstencil&BGFX_STENCIL_FUNC_RMASK_MASK)>>BGFX_STENCIL_FUNC_RMASK_SHIFT;
  1450. const uint32_t writeMask = 0xff;
  1451. desc.stencilReadMask = readMask;
  1452. desc.stencilWriteMask = writeMask;
  1453. desc.stencilFront.failOp = s_stencilOp[(fstencil&BGFX_STENCIL_OP_FAIL_S_MASK)>>BGFX_STENCIL_OP_FAIL_S_SHIFT];
  1454. desc.stencilFront.depthFailOp = s_stencilOp[(fstencil&BGFX_STENCIL_OP_FAIL_Z_MASK)>>BGFX_STENCIL_OP_FAIL_Z_SHIFT];
  1455. desc.stencilFront.passOp = s_stencilOp[(fstencil&BGFX_STENCIL_OP_PASS_Z_MASK)>>BGFX_STENCIL_OP_PASS_Z_SHIFT];
  1456. desc.stencilFront.compare = s_cmpFunc[(fstencil&BGFX_STENCIL_TEST_MASK)>>BGFX_STENCIL_TEST_SHIFT];
  1457. desc.stencilBack.failOp = s_stencilOp[(bstencil&BGFX_STENCIL_OP_FAIL_S_MASK)>>BGFX_STENCIL_OP_FAIL_S_SHIFT];
  1458. desc.stencilBack.depthFailOp = s_stencilOp[(bstencil&BGFX_STENCIL_OP_FAIL_Z_MASK)>>BGFX_STENCIL_OP_FAIL_Z_SHIFT];
  1459. desc.stencilBack.passOp = s_stencilOp[(bstencil&BGFX_STENCIL_OP_PASS_Z_MASK)>>BGFX_STENCIL_OP_PASS_Z_SHIFT];
  1460. desc.stencilBack.compare = s_cmpFunc[(bstencil&BGFX_STENCIL_TEST_MASK)>>BGFX_STENCIL_TEST_SHIFT];
  1461. }
  1462. }
  1463. RenderPassStateWgpu* getRenderPassState(bgfx::FrameBufferHandle fbh, bool clear, Clear clr)
  1464. {
  1465. bx::HashMurmur2A murmur;
  1466. murmur.begin();
  1467. murmur.add(fbh.idx);
  1468. murmur.add(clear);
  1469. murmur.add(&clr, sizeof(clr));
  1470. uint32_t hash = murmur.end();
  1471. RenderPassStateWgpu* rps = m_renderPassStateCache.find(hash);
  1472. if (NULL == rps)
  1473. {
  1474. rps = BX_NEW(g_allocator, RenderPassStateWgpu);
  1475. m_renderPassStateCache.add(hash, rps);
  1476. }
  1477. return rps;
  1478. }
  1479. PipelineStateWgpu* getPipelineState(
  1480. uint64_t _state
  1481. , uint64_t _stencil
  1482. , uint32_t _rgba
  1483. , FrameBufferHandle _fbh
  1484. , uint8_t _numStreams
  1485. , const VertexLayout** _vertexDecls
  1486. , bool _index32
  1487. , ProgramHandle _program
  1488. , uint8_t _numInstanceData
  1489. )
  1490. {
  1491. _state &= 0
  1492. | BGFX_STATE_WRITE_RGB
  1493. | BGFX_STATE_WRITE_A
  1494. | BGFX_STATE_WRITE_Z
  1495. | BGFX_STATE_DEPTH_TEST_MASK
  1496. | BGFX_STATE_BLEND_MASK
  1497. | BGFX_STATE_BLEND_EQUATION_MASK
  1498. | BGFX_STATE_BLEND_INDEPENDENT
  1499. | BGFX_STATE_BLEND_ALPHA_TO_COVERAGE
  1500. | BGFX_STATE_CULL_MASK
  1501. | BGFX_STATE_MSAA
  1502. | BGFX_STATE_LINEAA
  1503. | BGFX_STATE_CONSERVATIVE_RASTER
  1504. | BGFX_STATE_PT_MASK
  1505. ;
  1506. const bool independentBlendEnable = !!(BGFX_STATE_BLEND_INDEPENDENT & _state);
  1507. const ProgramWgpu& program = m_program[_program.idx];
  1508. bx::HashMurmur2A murmur;
  1509. murmur.begin();
  1510. murmur.add(_state);
  1511. murmur.add(_stencil);
  1512. murmur.add(independentBlendEnable ? _rgba : 0);
  1513. murmur.add(_numInstanceData);
  1514. FrameBufferWgpu& frameBuffer = !isValid(_fbh) ? m_mainFrameBuffer : m_frameBuffers[_fbh.idx];
  1515. murmur.add(frameBuffer.m_pixelFormatHash);
  1516. murmur.add(program.m_vsh->m_hash);
  1517. if (NULL != program.m_fsh)
  1518. {
  1519. murmur.add(program.m_fsh->m_hash);
  1520. }
  1521. for (uint8_t ii = 0; ii < _numStreams; ++ii)
  1522. {
  1523. murmur.add(_vertexDecls[ii]->m_hash);
  1524. }
  1525. uint32_t hash = murmur.end();
  1526. PipelineStateWgpu* pso = m_pipelineStateCache.find(hash);
  1527. if (NULL == pso)
  1528. {
  1529. pso = BX_NEW(g_allocator, PipelineStateWgpu);
  1530. //pd.alphaToCoverageEnabled = !!(BGFX_STATE_BLEND_ALPHA_TO_COVERAGE & _state);
  1531. RenderPipelineDescriptor& pd = pso->m_rpd;
  1532. uint32_t frameBufferAttachment = 1;
  1533. uint32_t sampleCount = 1;
  1534. if (!isValid(_fbh)
  1535. || s_renderWgpu->m_frameBuffers[_fbh.idx].m_swapChain)
  1536. {
  1537. SwapChainWgpu& swapChain = !isValid(_fbh)
  1538. ? *s_renderWgpu->m_mainFrameBuffer.m_swapChain
  1539. : *s_renderWgpu->m_frameBuffers[_fbh.idx].m_swapChain
  1540. ;
  1541. sampleCount = swapChain.m_backBufferColorMsaa
  1542. ? swapChain.m_sampleCount
  1543. : 1
  1544. ;
  1545. pd.colorStates[0].format = swapChain.m_colorFormat;
  1546. pd.depthStencilState.format = swapChain.m_depthFormat;
  1547. pd.desc.depthStencilState = &pd.depthStencilState;
  1548. }
  1549. else
  1550. {
  1551. frameBufferAttachment = frameBuffer.m_num;
  1552. for (uint32_t ii = 0; ii < frameBuffer.m_num; ++ii)
  1553. {
  1554. const TextureWgpu& texture = m_textures[frameBuffer.m_colorHandle[ii].idx];
  1555. sampleCount = texture.m_ptrMsaa
  1556. ? texture.m_sampleCount
  1557. : 1
  1558. ;
  1559. pd.colorStates[ii].format = s_textureFormat[texture.m_textureFormat].m_fmt;
  1560. }
  1561. pd.desc.colorStateCount = frameBuffer.m_num;
  1562. if (isValid(frameBuffer.m_depthHandle) )
  1563. {
  1564. const TextureWgpu& texture = m_textures[frameBuffer.m_depthHandle.idx];
  1565. pd.depthStencilState.format = s_textureFormat[texture.m_textureFormat].m_fmt;
  1566. pd.desc.depthStencilState = &pd.depthStencilState;
  1567. }
  1568. }
  1569. const uint32_t blend = uint32_t( (_state&BGFX_STATE_BLEND_MASK )>>BGFX_STATE_BLEND_SHIFT);
  1570. const uint32_t equation = uint32_t( (_state&BGFX_STATE_BLEND_EQUATION_MASK)>>BGFX_STATE_BLEND_EQUATION_SHIFT);
  1571. const uint32_t srcRGB = (blend )&0xf;
  1572. const uint32_t dstRGB = (blend>> 4)&0xf;
  1573. const uint32_t srcA = (blend>> 8)&0xf;
  1574. const uint32_t dstA = (blend>>12)&0xf;
  1575. const uint32_t equRGB = (equation )&0x7;
  1576. const uint32_t equA = (equation>>3)&0x7;
  1577. wgpu::ColorWriteMask writeMask = wgpu::ColorWriteMask::None;
  1578. writeMask |= (_state&BGFX_STATE_WRITE_R) ? wgpu::ColorWriteMask::Red : wgpu::ColorWriteMask::None;
  1579. writeMask |= (_state&BGFX_STATE_WRITE_G) ? wgpu::ColorWriteMask::Green : wgpu::ColorWriteMask::None;
  1580. writeMask |= (_state&BGFX_STATE_WRITE_B) ? wgpu::ColorWriteMask::Blue : wgpu::ColorWriteMask::None;
  1581. writeMask |= (_state&BGFX_STATE_WRITE_A) ? wgpu::ColorWriteMask::Alpha : wgpu::ColorWriteMask::None;
  1582. for (uint32_t ii = 0; ii < (independentBlendEnable ? 1 : frameBufferAttachment); ++ii)
  1583. {
  1584. wgpu::ColorStateDescriptor& drt = pd.colorStates[ii]; // = pd.colorAttachments[ii];
  1585. if(!(BGFX_STATE_BLEND_MASK & _state))
  1586. {
  1587. drt.colorBlend = defaultDescriptor<wgpu::BlendDescriptor>();
  1588. drt.alphaBlend = defaultDescriptor<wgpu::BlendDescriptor>();
  1589. }
  1590. else
  1591. {
  1592. drt.colorBlend.srcFactor = s_blendFactor[srcRGB][0];
  1593. drt.colorBlend.dstFactor = s_blendFactor[dstRGB][0];
  1594. drt.colorBlend.operation = s_blendEquation[equRGB];
  1595. drt.alphaBlend.srcFactor = s_blendFactor[srcA][1];
  1596. drt.alphaBlend.dstFactor = s_blendFactor[dstA][1];
  1597. drt.alphaBlend.operation = s_blendEquation[equA];
  1598. }
  1599. drt.writeMask = writeMask;
  1600. }
  1601. if (independentBlendEnable)
  1602. {
  1603. for (uint32_t ii = 1, rgba = _rgba; ii < frameBufferAttachment; ++ii, rgba >>= 11)
  1604. {
  1605. wgpu::ColorStateDescriptor drt = pd.colorStates[ii]; // = pd.colorAttachments[ii];
  1606. //drt.blendingEnabled = 0 != (rgba&0x7ff);
  1607. const uint32_t src = (rgba )&0xf;
  1608. const uint32_t dst = (rgba>>4)&0xf;
  1609. const uint32_t equationIndex = (rgba>>8)&0x7;
  1610. drt.colorBlend.srcFactor = s_blendFactor[src][0];
  1611. drt.colorBlend.dstFactor = s_blendFactor[dst][0];
  1612. drt.colorBlend.operation = s_blendEquation[equationIndex];
  1613. drt.alphaBlend.srcFactor = s_blendFactor[src][1];
  1614. drt.alphaBlend.dstFactor = s_blendFactor[dst][1];
  1615. drt.alphaBlend.operation = s_blendEquation[equationIndex];
  1616. drt.writeMask = writeMask;
  1617. }
  1618. }
  1619. pd.desc.vertexStage.module = program.m_vsh->m_module;
  1620. pd.fragmentStage.module = program.m_fsh != NULL ? program.m_fsh->m_module : wgpu::ShaderModule();
  1621. setDepthStencilState(pd.depthStencilState, _state, _stencil);
  1622. const uint64_t cull = _state & BGFX_STATE_CULL_MASK;
  1623. const uint8_t cullIndex = uint8_t(cull >> BGFX_STATE_CULL_SHIFT);
  1624. pd.rasterizationState.cullMode = s_cullMode[cullIndex];
  1625. pd.rasterizationState.frontFace = (_state & BGFX_STATE_FRONT_CCW) ? wgpu::FrontFace::CCW : wgpu::FrontFace::CW;
  1626. // pd.desc = m_renderPipelineDescriptor;
  1627. pd.desc.sampleCount = sampleCount;
  1628. wgpu::PipelineLayoutDescriptor layout = defaultDescriptor<wgpu::PipelineLayoutDescriptor>();
  1629. layout.bindGroupLayouts = &program.m_bindGroupLayout;
  1630. layout.bindGroupLayoutCount = 1;
  1631. BX_TRACE("Creating WebGPU render pipeline layout for program %s", program.m_vsh->name());
  1632. pd.desc.layout = m_device.CreatePipelineLayout(&layout);
  1633. // TODO (hugoam) this should be cached too ?
  1634. //uint32_t ref = (_state&BGFX_STATE_ALPHA_REF_MASK) >> BGFX_STATE_ALPHA_REF_SHIFT;
  1635. //viewState.m_alphaRef = ref / 255.0f;
  1636. const uint64_t primType = _state & BGFX_STATE_PT_MASK;
  1637. uint8_t primIndex = uint8_t(primType >> BGFX_STATE_PT_SHIFT);
  1638. PrimInfo prim = s_primInfo[primIndex];
  1639. pd.desc.primitiveTopology = prim.m_type;
  1640. VertexStateDescriptor input;
  1641. input.desc.vertexBufferCount = 0;
  1642. wgpu::VertexBufferLayoutDescriptor* inputBinding = input.vertexBuffers;
  1643. wgpu::VertexAttributeDescriptor* inputAttrib = input.attributes;
  1644. auto fillVertexDecl = [&](const ShaderWgpu* _vsh, const VertexLayout& _decl)
  1645. {
  1646. input.desc.vertexBufferCount += 1;
  1647. inputBinding->arrayStride = _decl.m_stride;
  1648. inputBinding->stepMode = wgpu::InputStepMode::Vertex;
  1649. inputBinding->attributes = inputAttrib;
  1650. uint32_t numAttribs = 0;
  1651. for(uint32_t attr = 0; attr < Attrib::Count; ++attr)
  1652. {
  1653. if(UINT16_MAX != _decl.m_attributes[attr])
  1654. {
  1655. if(UINT8_MAX == _vsh->m_attrRemap[attr])
  1656. continue;
  1657. inputAttrib->shaderLocation = _vsh->m_attrRemap[attr];
  1658. if(0 == _decl.m_attributes[attr])
  1659. {
  1660. inputAttrib->format = wgpu::VertexFormat::Float3;
  1661. inputAttrib->offset = 0;
  1662. }
  1663. else
  1664. {
  1665. uint8_t num;
  1666. AttribType::Enum type;
  1667. bool normalized;
  1668. bool asInt;
  1669. _decl.decode(Attrib::Enum(attr), num, type, normalized, asInt);
  1670. inputAttrib->format = s_attribType[type][num-1][normalized];
  1671. inputAttrib->offset = _decl.m_offset[attr];
  1672. }
  1673. ++inputAttrib;
  1674. ++numAttribs;
  1675. }
  1676. }
  1677. inputBinding->attributeCount = numAttribs;
  1678. inputBinding++;
  1679. return numAttribs;
  1680. };
  1681. //bool attrSet[Attrib::Count] = {};
  1682. uint16_t unsettedAttr[Attrib::Count];
  1683. bx::memCopy(unsettedAttr, program.m_vsh->m_attrMask, sizeof(uint16_t) * Attrib::Count);
  1684. uint8_t stream = 0;
  1685. for (; stream < _numStreams; ++stream)
  1686. {
  1687. VertexLayout layout;
  1688. bx::memCopy(&layout, _vertexDecls[stream], sizeof(VertexLayout));
  1689. const uint16_t* attrMask = program.m_vsh->m_attrMask;
  1690. for (uint32_t ii = 0; ii < Attrib::Count; ++ii)
  1691. {
  1692. Attrib::Enum iiattr = Attrib::Enum(ii);
  1693. uint16_t mask = attrMask[ii];
  1694. uint16_t attr = (layout.m_attributes[ii] & mask);
  1695. if (attr == 0)
  1696. {
  1697. layout.m_attributes[ii] = UINT16_MAX;
  1698. }
  1699. if (unsettedAttr[ii] && attr != UINT16_MAX)
  1700. {
  1701. unsettedAttr[ii] = 0;
  1702. }
  1703. }
  1704. fillVertexDecl(program.m_vsh, layout);
  1705. }
  1706. for (uint32_t ii = 0; ii < Attrib::Count; ++ii)
  1707. {
  1708. Attrib::Enum iiattr = Attrib::Enum(ii);
  1709. if (0 < unsettedAttr[ii])
  1710. {
  1711. //uint32_t numAttribs = input.vertexBuffers[stream].attributeCount;
  1712. //uint32_t numAttribs = inputBinding->attributeCount;
  1713. //wgpu::VertexBufferLayoutDescriptor* inputAttrib = const_cast<VkVertexInputAttributeDescription*>(_vertexInputState.pVertexAttributeDescriptions + numAttribs);
  1714. inputAttrib->shaderLocation = program.m_vsh->m_attrRemap[ii];
  1715. //inputAttrib->binding = 0;
  1716. inputAttrib->format = wgpu::VertexFormat::Float3; // VK_FORMAT_R32G32B32_SFLOAT;
  1717. inputAttrib->offset = 0;
  1718. input.vertexBuffers[stream-1].attributeCount++;
  1719. ++inputAttrib;
  1720. }
  1721. }
  1722. // TODO (hugoam) WebGPU will crash whenever we are not supplying the correct number of attributes (which depends on the stride passed to bgfx::allocInstanceDataBuffer)
  1723. // so we need to know the number of live instance attributes in the shader and if they aren't all supplied:
  1724. // - fail the pipeline state creation
  1725. // - bind dummy attributes
  1726. if (0 < _numInstanceData)
  1727. {
  1728. uint32_t numBindings = input.desc.vertexBufferCount; // == stream+1 // .vertexBindingDescriptionCount;
  1729. uint32_t firstAttrib = input.vertexBuffers[stream-1].attributeCount;
  1730. uint32_t numAttribs = firstAttrib;
  1731. inputBinding->arrayStride = _numInstanceData * 16;
  1732. inputBinding->stepMode = wgpu::InputStepMode::Instance;
  1733. for (uint32_t inst = 0; inst < _numInstanceData; ++inst)
  1734. {
  1735. inputAttrib->shaderLocation = numAttribs;
  1736. inputAttrib->format = wgpu::VertexFormat::Float4;
  1737. inputAttrib->offset = inst * 16;
  1738. ++numAttribs;
  1739. ++inputAttrib;
  1740. }
  1741. input.desc.vertexBufferCount = numBindings + 1;
  1742. input.vertexBuffers[stream].attributeCount = numAttribs - firstAttrib;
  1743. input.vertexBuffers[stream].attributes = &input.attributes[firstAttrib];
  1744. }
  1745. input.desc.indexFormat = _index32 ? wgpu::IndexFormat::Uint32 : wgpu::IndexFormat::Uint16;
  1746. pd.desc.vertexState = &input.desc;
  1747. BX_TRACE("Creating WebGPU render pipeline state for program %s", program.m_vsh->name());
  1748. pso->m_rps = m_device.CreateRenderPipeline(&pd.desc);
  1749. m_pipelineStateCache.add(hash, pso);
  1750. }
  1751. return pso;
  1752. }
  1753. PipelineStateWgpu* getPipelineState(
  1754. uint64_t _state
  1755. , uint64_t _stencil
  1756. , uint32_t _rgba
  1757. , FrameBufferHandle _fbh
  1758. , VertexLayoutHandle _declHandle
  1759. , bool _index32
  1760. , ProgramHandle _program
  1761. , uint8_t _numInstanceData
  1762. )
  1763. {
  1764. const VertexLayout* decl = &m_vertexDecls[_declHandle.idx];
  1765. return getPipelineState(
  1766. _state
  1767. , _stencil
  1768. , _rgba
  1769. , _fbh
  1770. , 1
  1771. , &decl
  1772. , _index32
  1773. , _program
  1774. , _numInstanceData
  1775. );
  1776. }
  1777. PipelineStateWgpu* getComputePipelineState(ProgramHandle _program)
  1778. {
  1779. ProgramWgpu& program = m_program[_program.idx];
  1780. if (NULL == program.m_computePS)
  1781. {
  1782. PipelineStateWgpu* pso = BX_NEW(g_allocator, PipelineStateWgpu);
  1783. program.m_computePS = pso;
  1784. wgpu::PipelineLayoutDescriptor layout = defaultDescriptor<wgpu::PipelineLayoutDescriptor>();
  1785. layout.bindGroupLayouts = &program.m_bindGroupLayout;
  1786. layout.bindGroupLayoutCount = 1;
  1787. BX_TRACE("Creating WebGPU render pipeline layout for program %s", program.m_vsh->name());
  1788. pso->m_layout = m_device.CreatePipelineLayout(&layout);
  1789. wgpu::ComputePipelineDescriptor desc;
  1790. desc.layout = pso->m_layout;
  1791. desc.computeStage = { NULL, program.m_vsh->m_module, "main" };
  1792. BX_TRACE("Creating WebGPU render pipeline state for program %s", program.m_vsh->name());
  1793. pso->m_cps = m_device.CreateComputePipeline(&desc);
  1794. }
  1795. return program.m_computePS;
  1796. }
  1797. wgpu::Sampler getSamplerState(uint32_t _flags)
  1798. {
  1799. _flags &= BGFX_SAMPLER_BITS_MASK;
  1800. SamplerStateWgpu* sampler = m_samplerStateCache.find(_flags);
  1801. if (NULL == sampler)
  1802. {
  1803. sampler = BX_NEW(g_allocator, SamplerStateWgpu);
  1804. wgpu::SamplerDescriptor desc;
  1805. desc.addressModeU = s_textureAddress[(_flags&BGFX_SAMPLER_U_MASK)>>BGFX_SAMPLER_U_SHIFT];
  1806. desc.addressModeV = s_textureAddress[(_flags&BGFX_SAMPLER_V_MASK)>>BGFX_SAMPLER_V_SHIFT];
  1807. desc.addressModeW = s_textureAddress[(_flags&BGFX_SAMPLER_W_MASK)>>BGFX_SAMPLER_W_SHIFT];
  1808. desc.minFilter = s_textureFilterMinMag[(_flags&BGFX_SAMPLER_MIN_MASK)>>BGFX_SAMPLER_MIN_SHIFT];
  1809. desc.magFilter = s_textureFilterMinMag[(_flags&BGFX_SAMPLER_MAG_MASK)>>BGFX_SAMPLER_MAG_SHIFT];
  1810. desc.mipmapFilter = s_textureFilterMip[(_flags&BGFX_SAMPLER_MIP_MASK)>>BGFX_SAMPLER_MIP_SHIFT];
  1811. desc.lodMinClamp = 0;
  1812. desc.lodMaxClamp = bx::kFloatMax;
  1813. const uint32_t cmpFunc = (_flags&BGFX_SAMPLER_COMPARE_MASK)>>BGFX_SAMPLER_COMPARE_SHIFT;
  1814. desc.compare = 0 == cmpFunc
  1815. ? wgpu::CompareFunction::Undefined
  1816. : s_cmpFunc[cmpFunc]
  1817. ;
  1818. sampler->m_sampler = s_renderWgpu->m_device.CreateSampler(&desc);
  1819. m_samplerStateCache.add(_flags, sampler);
  1820. }
  1821. return sampler->m_sampler;
  1822. }
  1823. wgpu::CommandEncoder& getRenderEncoder()
  1824. {
  1825. if (!m_cmd.m_renderEncoder)
  1826. m_cmd.beginRender();
  1827. return m_cmd.m_renderEncoder;
  1828. }
  1829. wgpu::CommandEncoder& getStagingEncoder()
  1830. {
  1831. if (!m_cmd.m_stagingEncoder)
  1832. m_cmd.beginStaging();
  1833. return m_cmd.m_stagingEncoder;
  1834. }
  1835. wgpu::CommandEncoder& getBlitCommandEncoder()
  1836. {
  1837. if (m_renderEncoder || m_computeEncoder)
  1838. endEncoding();
  1839. return getRenderEncoder();
  1840. }
  1841. wgpu::RenderPassEncoder renderPass(bgfx::Frame* _render, bgfx::FrameBufferHandle fbh, bool clear, Clear clr, const char* name = NULL)
  1842. {
  1843. RenderPassStateWgpu* rps = s_renderWgpu->getRenderPassState(fbh, clear, clr);
  1844. RenderPassDescriptor& renderPassDescriptor = rps->m_rpd;
  1845. renderPassDescriptor.desc.label = name;
  1846. setFrameBuffer(renderPassDescriptor, fbh);
  1847. if(clear)
  1848. {
  1849. for(uint32_t ii = 0; ii < g_caps.limits.maxFBAttachments; ++ii)
  1850. {
  1851. wgpu::RenderPassColorAttachmentDescriptor& color = renderPassDescriptor.colorAttachments[ii];
  1852. if(0 != (BGFX_CLEAR_COLOR & clr.m_flags))
  1853. {
  1854. if(0 != (BGFX_CLEAR_COLOR_USE_PALETTE & clr.m_flags))
  1855. {
  1856. uint8_t index = (uint8_t)bx::uint32_min(BGFX_CONFIG_MAX_COLOR_PALETTE - 1, clr.m_index[ii]);
  1857. const float* rgba = _render->m_colorPalette[index];
  1858. const float rr = rgba[0];
  1859. const float gg = rgba[1];
  1860. const float bb = rgba[2];
  1861. const float aa = rgba[3];
  1862. color.clearColor = { rr, gg, bb, aa };
  1863. }
  1864. else
  1865. {
  1866. float rr = clr.m_index[0] * 1.0f / 255.0f;
  1867. float gg = clr.m_index[1] * 1.0f / 255.0f;
  1868. float bb = clr.m_index[2] * 1.0f / 255.0f;
  1869. float aa = clr.m_index[3] * 1.0f / 255.0f;
  1870. color.clearColor = { rr, gg, bb, aa };
  1871. }
  1872. color.loadOp = wgpu::LoadOp::Clear;
  1873. }
  1874. else
  1875. {
  1876. color.loadOp = wgpu::LoadOp::Load;
  1877. }
  1878. //desc.storeOp = desc.attachment.sampleCount > 1 ? wgpu::StoreOp::MultisampleResolve : wgpu::StoreOp::Store;
  1879. color.storeOp = wgpu::StoreOp::Store;
  1880. }
  1881. wgpu::RenderPassDepthStencilAttachmentDescriptor& depthStencil = renderPassDescriptor.depthStencilAttachment;
  1882. if(depthStencil.attachment)
  1883. {
  1884. depthStencil.clearDepth = clr.m_depth;
  1885. depthStencil.depthLoadOp = 0 != (BGFX_CLEAR_DEPTH & clr.m_flags)
  1886. ? wgpu::LoadOp::Clear
  1887. : wgpu::LoadOp::Load
  1888. ;
  1889. depthStencil.depthStoreOp = m_mainFrameBuffer.m_swapChain->m_backBufferColorMsaa
  1890. ? wgpu::StoreOp(0) //wgpu::StoreOp::DontCare
  1891. : wgpu::StoreOp::Store
  1892. ;
  1893. depthStencil.clearStencil = clr.m_stencil;
  1894. depthStencil.stencilLoadOp = 0 != (BGFX_CLEAR_STENCIL & clr.m_flags)
  1895. ? wgpu::LoadOp::Clear
  1896. : wgpu::LoadOp::Load
  1897. ;
  1898. depthStencil.stencilStoreOp = m_mainFrameBuffer.m_swapChain->m_backBufferColorMsaa
  1899. ? wgpu::StoreOp(0) //wgpu::StoreOp::DontCare
  1900. : wgpu::StoreOp::Store
  1901. ;
  1902. }
  1903. }
  1904. else
  1905. {
  1906. for(uint32_t ii = 0; ii < g_caps.limits.maxFBAttachments; ++ii)
  1907. {
  1908. wgpu::RenderPassColorAttachmentDescriptor& color = renderPassDescriptor.colorAttachments[ii];
  1909. if(color.attachment)
  1910. {
  1911. color.loadOp = wgpu::LoadOp::Load;
  1912. }
  1913. }
  1914. wgpu::RenderPassDepthStencilAttachmentDescriptor& depthStencil = renderPassDescriptor.depthStencilAttachment;
  1915. if(depthStencil.attachment)
  1916. {
  1917. depthStencil.depthLoadOp = wgpu::LoadOp::Load;
  1918. depthStencil.depthStoreOp = wgpu::StoreOp::Store;
  1919. depthStencil.stencilLoadOp = wgpu::LoadOp::Load;
  1920. depthStencil.stencilStoreOp = wgpu::StoreOp::Store;
  1921. }
  1922. }
  1923. wgpu::RenderPassEncoder rce = m_cmd.m_renderEncoder.BeginRenderPass(&renderPassDescriptor.desc);
  1924. m_renderEncoder = rce;
  1925. return rce;
  1926. }
  1927. void endEncoding()
  1928. {
  1929. if (m_renderEncoder)
  1930. {
  1931. m_renderEncoder.EndPass();
  1932. m_renderEncoder = NULL;
  1933. }
  1934. if (m_computeEncoder)
  1935. {
  1936. m_computeEncoder.EndPass();
  1937. m_computeEncoder = NULL;
  1938. }
  1939. }
  1940. void* m_renderDocDll;
  1941. #if !BX_PLATFORM_EMSCRIPTEN
  1942. dawn_native::Instance m_instance;
  1943. #endif
  1944. wgpu::Device m_device;
  1945. wgpu::Queue m_queue;
  1946. TimerQueryWgpu m_gpuTimer;
  1947. CommandQueueWgpu m_cmd;
  1948. StagingBufferWgpu m_uniformBuffers[WEBGPU_NUM_UNIFORM_BUFFERS];
  1949. ScratchBufferWgpu m_scratchBuffers[WEBGPU_MAX_FRAMES_IN_FLIGHT];
  1950. BindStateCacheWgpu m_bindStateCache[WEBGPU_MAX_FRAMES_IN_FLIGHT];
  1951. uint8_t m_frameIndex;
  1952. uint16_t m_numWindows;
  1953. FrameBufferHandle m_windows[BGFX_CONFIG_MAX_FRAME_BUFFERS];
  1954. IndexBufferWgpu m_indexBuffers[BGFX_CONFIG_MAX_INDEX_BUFFERS];
  1955. VertexBufferWgpu m_vertexBuffers[BGFX_CONFIG_MAX_VERTEX_BUFFERS];
  1956. ShaderWgpu m_shaders[BGFX_CONFIG_MAX_SHADERS];
  1957. ProgramWgpu m_program[BGFX_CONFIG_MAX_PROGRAMS];
  1958. TextureWgpu m_textures[BGFX_CONFIG_MAX_TEXTURES];
  1959. ReadbackWgpu m_readbacks[BGFX_CONFIG_MAX_TEXTURES];
  1960. FrameBufferWgpu m_mainFrameBuffer;
  1961. FrameBufferWgpu m_frameBuffers[BGFX_CONFIG_MAX_FRAME_BUFFERS];
  1962. VertexLayout m_vertexDecls[BGFX_CONFIG_MAX_VERTEX_LAYOUTS];
  1963. UniformRegistry m_uniformReg;
  1964. void* m_uniforms[BGFX_CONFIG_MAX_UNIFORMS];
  1965. //StateCacheT<BindStateWgpu*> m_bindStateCache;
  1966. StateCacheT<RenderPassStateWgpu*> m_renderPassStateCache;
  1967. StateCacheT<PipelineStateWgpu*> m_pipelineStateCache;
  1968. StateCacheT<SamplerStateWgpu*> m_samplerStateCache;
  1969. TextVideoMem m_textVideoMem;
  1970. uint8_t m_fsScratch[64 << 10];
  1971. uint8_t m_vsScratch[64 << 10];
  1972. FrameBufferHandle m_fbh;
  1973. bool m_rtMsaa;
  1974. Resolution m_resolution;
  1975. void* m_capture;
  1976. uint32_t m_captureSize;
  1977. wgpu::RenderPassEncoder m_renderEncoder;
  1978. wgpu::ComputePassEncoder m_computeEncoder;
  1979. };
  1980. RendererContextI* rendererCreate(const Init& _init)
  1981. {
  1982. s_renderWgpu = BX_NEW(g_allocator, RendererContextWgpu);
  1983. if (!s_renderWgpu->init(_init) )
  1984. {
  1985. BX_DELETE(g_allocator, s_renderWgpu);
  1986. s_renderWgpu = NULL;
  1987. }
  1988. return s_renderWgpu;
  1989. }
  1990. void rendererDestroy()
  1991. {
  1992. s_renderWgpu->shutdown();
  1993. BX_DELETE(g_allocator, s_renderWgpu);
  1994. s_renderWgpu = NULL;
  1995. }
  1996. void writeString(bx::WriterI* _writer, const char* _str)
  1997. {
  1998. bx::write(_writer, _str, (int32_t)bx::strLen(_str) );
  1999. }
  2000. void ShaderWgpu::create(ShaderHandle _handle, const Memory* _mem)
  2001. {
  2002. m_handle = _handle;
  2003. BX_TRACE("Creating shader %s", getName(_handle));
  2004. bx::MemoryReader reader(_mem->data, _mem->size);
  2005. uint32_t magic;
  2006. bx::read(&reader, magic);
  2007. wgpu::ShaderStage shaderStage;
  2008. if (isShaderType(magic, 'C'))
  2009. {
  2010. shaderStage = wgpu::ShaderStage::Compute;
  2011. }
  2012. else if (isShaderType(magic, 'F'))
  2013. {
  2014. shaderStage = wgpu::ShaderStage::Fragment;
  2015. }
  2016. else if (isShaderType(magic, 'G'))
  2017. {
  2018. //shaderStage = wgpu::ShaderStage::Geometry;
  2019. }
  2020. else if (isShaderType(magic, 'V'))
  2021. {
  2022. shaderStage = wgpu::ShaderStage::Vertex;
  2023. }
  2024. m_stage = shaderStage;
  2025. uint32_t hashIn;
  2026. bx::read(&reader, hashIn);
  2027. uint32_t hashOut;
  2028. if (isShaderVerLess(magic, 6) )
  2029. {
  2030. hashOut = hashIn;
  2031. }
  2032. else
  2033. {
  2034. bx::read(&reader, hashOut);
  2035. }
  2036. uint16_t count;
  2037. bx::read(&reader, count);
  2038. m_numPredefined = 0;
  2039. m_numUniforms = count;
  2040. BX_TRACE("%s Shader consts %d"
  2041. , getShaderTypeName(magic)
  2042. , count
  2043. );
  2044. const bool fragment = isShaderType(magic, 'F');
  2045. uint8_t fragmentBit = fragment ? kUniformFragmentBit : 0;
  2046. BX_ASSERT(!isShaderVerLess(magic, 7), "WebGPU backend supports only shader binary version >= 7");
  2047. if (0 < count)
  2048. {
  2049. for (uint32_t ii = 0; ii < count; ++ii)
  2050. {
  2051. uint8_t nameSize = 0;
  2052. bx::read(&reader, nameSize);
  2053. char name[256];
  2054. bx::read(&reader, &name, nameSize);
  2055. name[nameSize] = '\0';
  2056. uint8_t type = 0;
  2057. bx::read(&reader, type);
  2058. uint8_t num;
  2059. bx::read(&reader, num);
  2060. uint16_t regIndex;
  2061. bx::read(&reader, regIndex);
  2062. uint16_t regCount;
  2063. bx::read(&reader, regCount);
  2064. uint8_t texComponent;
  2065. bx::read(&reader, texComponent);
  2066. uint8_t texDimension;
  2067. bx::read(&reader, texDimension);
  2068. const char* kind = "invalid";
  2069. PredefinedUniform::Enum predefined = nameToPredefinedUniformEnum(name);
  2070. if (PredefinedUniform::Count != predefined)
  2071. {
  2072. kind = "predefined";
  2073. m_predefined[m_numPredefined].m_loc = regIndex;
  2074. m_predefined[m_numPredefined].m_count = regCount;
  2075. m_predefined[m_numPredefined].m_type = uint8_t(predefined|fragmentBit);
  2076. m_numPredefined++;
  2077. }
  2078. else if (UniformType::End == (~kUniformMask & type))
  2079. {
  2080. // regCount is used for descriptor type
  2081. const bool buffer = regCount == VK_DESCRIPTOR_TYPE_STORAGE_BUFFER;
  2082. const bool readonly = (type & kUniformReadOnlyBit) != 0;
  2083. const uint8_t stage = regIndex - (buffer ? 16 : 32) - (fragment ? 48 : 0);
  2084. m_bindInfo[stage].m_index = m_numBuffers;
  2085. m_bindInfo[stage].m_binding = regIndex;
  2086. m_bindInfo[stage].m_uniform = { 0 };
  2087. m_buffers[m_numBuffers] = wgpu::BindGroupLayoutBinding();
  2088. m_buffers[m_numBuffers].binding = regIndex;
  2089. m_buffers[m_numBuffers].visibility = shaderStage;
  2090. if (buffer)
  2091. {
  2092. m_buffers[m_numBuffers].type = readonly
  2093. ? wgpu::BindingType::ReadonlyStorageBuffer
  2094. : wgpu::BindingType::StorageBuffer;
  2095. }
  2096. else
  2097. {
  2098. m_buffers[m_numBuffers].type = readonly
  2099. ? wgpu::BindingType::ReadonlyStorageTexture
  2100. : wgpu::BindingType::WriteonlyStorageTexture;
  2101. }
  2102. m_numBuffers++;
  2103. kind = "storage";
  2104. }
  2105. else if (UniformType::Sampler == (~kUniformMask & type))
  2106. {
  2107. const UniformRegInfo* info = s_renderWgpu->m_uniformReg.find(name);
  2108. BX_ASSERT(NULL != info, "User defined uniform '%s' is not found, it won't be set.", name);
  2109. const uint8_t stage = regIndex - 16 - (fragment ? 48 : 0);
  2110. m_bindInfo[stage].m_index = m_numSamplers;
  2111. m_bindInfo[stage].m_binding = regIndex;
  2112. m_bindInfo[stage].m_uniform = info->m_handle;
  2113. m_textures[m_numSamplers] = wgpu::BindGroupLayoutBinding();
  2114. m_textures[m_numSamplers].binding = regIndex;
  2115. m_textures[m_numSamplers].visibility = shaderStage;
  2116. m_textures[m_numSamplers].type = wgpu::BindingType::SampledTexture;
  2117. m_textures[m_numSamplers].viewDimension = wgpu::TextureViewDimension(texDimension);
  2118. m_textures[m_numSamplers].textureComponentType = wgpu::TextureComponentType(texComponent);
  2119. const bool comparisonSampler = (type & kUniformCompareBit) != 0;
  2120. m_samplers[m_numSamplers] = wgpu::BindGroupLayoutBinding();
  2121. m_samplers[m_numSamplers].binding = regIndex + 16;
  2122. m_samplers[m_numSamplers].visibility = shaderStage;
  2123. m_samplers[m_numSamplers].type = comparisonSampler
  2124. ? wgpu::BindingType::ComparisonSampler
  2125. : wgpu::BindingType::Sampler;
  2126. m_numSamplers++;
  2127. kind = "sampler";
  2128. }
  2129. else
  2130. {
  2131. const UniformRegInfo* info = s_renderWgpu->m_uniformReg.find(name);
  2132. BX_ASSERT(NULL != info, "User defined uniform '%s' is not found, it won't be set.", name);
  2133. if(NULL == m_constantBuffer)
  2134. {
  2135. m_constantBuffer = UniformBuffer::create(1024);
  2136. }
  2137. kind = "user";
  2138. m_constantBuffer->writeUniformHandle((UniformType::Enum)(type | fragmentBit), regIndex, info->m_handle, regCount);
  2139. }
  2140. BX_TRACE("\t%s: %s (%s), r.index %3d, r.count %2d"
  2141. , kind
  2142. , name
  2143. , getUniformTypeName(UniformType::Enum(type&~kUniformMask) )
  2144. , regIndex
  2145. , regCount
  2146. );
  2147. BX_UNUSED(kind);
  2148. }
  2149. if (NULL != m_constantBuffer)
  2150. {
  2151. m_constantBuffer->finish();
  2152. }
  2153. }
  2154. uint32_t shaderSize;
  2155. bx::read(&reader, shaderSize);
  2156. BX_TRACE("Shader body is at %lld size %u remaining %lld", reader.getPos(), shaderSize, reader.remaining());
  2157. const uint32_t* code = (const uint32_t*)reader.getDataPtr();
  2158. bx::skip(&reader, shaderSize+1);
  2159. m_code = (uint32_t*)BX_ALLOC(g_allocator, shaderSize);
  2160. m_codeSize = shaderSize;
  2161. bx::memCopy(m_code, code, shaderSize);
  2162. // TODO (hugoam) delete this
  2163. BX_TRACE("First word %08" PRIx32, code[0]);
  2164. uint8_t numAttrs = 0;
  2165. bx::read(&reader, numAttrs);
  2166. m_numAttrs = numAttrs;
  2167. bx::memSet(m_attrMask, 0, sizeof(m_attrMask));
  2168. bx::memSet(m_attrRemap, UINT8_MAX, sizeof(m_attrRemap));
  2169. for(uint8_t ii = 0; ii < numAttrs; ++ii)
  2170. {
  2171. uint16_t id;
  2172. bx::read(&reader, id);
  2173. auto toString = [](Attrib::Enum attr)
  2174. {
  2175. if (attr == Attrib::Position) return "Position";
  2176. else if (attr == Attrib::Normal) return "Normal";
  2177. else if (attr == Attrib::Tangent) return "Tangent";
  2178. else if (attr == Attrib::Bitangent) return "Bitangent";
  2179. else if (attr == Attrib::Color0) return "Color0";
  2180. else if (attr == Attrib::Color1) return "Color1";
  2181. else if (attr == Attrib::Color2) return "Color2";
  2182. else if (attr == Attrib::Color3) return "Color3";
  2183. else if (attr == Attrib::Indices) return "Indices";
  2184. else if (attr == Attrib::Weight) return "Weight";
  2185. else if (attr == Attrib::TexCoord0) return "TexCoord0";
  2186. else if (attr == Attrib::TexCoord1) return "TexCoord1";
  2187. else if (attr == Attrib::TexCoord2) return "TexCoord2";
  2188. else if (attr == Attrib::TexCoord3) return "TexCoord3";
  2189. else if (attr == Attrib::TexCoord4) return "TexCoord4";
  2190. else if (attr == Attrib::TexCoord5) return "TexCoord5";
  2191. else if (attr == Attrib::TexCoord6) return "TexCoord6";
  2192. else if (attr == Attrib::TexCoord7) return "TexCoord7";
  2193. return "Invalid";
  2194. };
  2195. Attrib::Enum attr = idToAttrib(id);
  2196. if(Attrib::Count != attr)
  2197. {
  2198. m_attrMask[attr] = UINT16_MAX;
  2199. m_attrRemap[attr] = ii;
  2200. BX_TRACE("\tattrib: %s (%i) at index %i", toString(attr), attr, ii);
  2201. }
  2202. }
  2203. wgpu::ShaderModuleSPIRVDescriptor spirv;
  2204. spirv.code = m_code;
  2205. spirv.codeSize = shaderSize / 4;
  2206. wgpu::ShaderModuleDescriptor desc;
  2207. desc.label = getName(_handle);
  2208. desc.nextInChain = &spirv;
  2209. m_module = s_renderWgpu->m_device.CreateShaderModule(&desc);
  2210. BGFX_FATAL(m_module
  2211. , bgfx::Fatal::InvalidShader
  2212. , "Failed to create %s shader."
  2213. , getShaderTypeName(magic)
  2214. );
  2215. bx::HashMurmur2A murmur;
  2216. murmur.begin();
  2217. murmur.add(hashIn);
  2218. murmur.add(hashOut);
  2219. murmur.add(code, shaderSize);
  2220. murmur.add(numAttrs);
  2221. murmur.add(m_attrMask, numAttrs);
  2222. m_hash = murmur.end();
  2223. auto roundUp = [](auto value, auto multiple)
  2224. {
  2225. return ((value + multiple - 1) / multiple) * multiple;
  2226. };
  2227. bx::read(&reader, m_size);
  2228. const uint32_t align = kMinBufferOffsetAlignment;
  2229. m_gpuSize = bx::strideAlign(m_size, align);
  2230. BX_TRACE("shader size %d (used=%d) (prev=%d)", (int)m_size, (int)m_gpuSize, (int)bx::strideAlign(roundUp(m_size, 4), align));
  2231. }
  2232. void ProgramWgpu::create(const ShaderWgpu* _vsh, const ShaderWgpu* _fsh)
  2233. {
  2234. BX_ASSERT(_vsh->m_module, "Vertex shader doesn't exist.");
  2235. m_vsh = _vsh;
  2236. m_fsh = _fsh;
  2237. m_gpuSize = _vsh->m_gpuSize + (_fsh ? _fsh->m_gpuSize : 0);
  2238. //BX_ASSERT(NULL != _vsh->m_code, "Vertex shader doesn't exist.");
  2239. m_vsh = _vsh;
  2240. bx::memCopy(&m_predefined[0], _vsh->m_predefined, _vsh->m_numPredefined * sizeof(PredefinedUniform));
  2241. m_numPredefined = _vsh->m_numPredefined;
  2242. if(NULL != _fsh)
  2243. {
  2244. //BX_ASSERT(NULL != _fsh->m_code, "Fragment shader doesn't exist.");
  2245. m_fsh = _fsh;
  2246. bx::memCopy(&m_predefined[m_numPredefined], _fsh->m_predefined, _fsh->m_numPredefined * sizeof(PredefinedUniform));
  2247. m_numPredefined += _fsh->m_numPredefined;
  2248. }
  2249. wgpu::BindGroupLayoutEntry bindings[2 + BGFX_CONFIG_MAX_TEXTURE_SAMPLERS * 3];
  2250. m_numUniforms = 0 + (_vsh->m_size > 0 ? 1 : 0) + (NULL != _fsh && _fsh->m_size > 0 ? 1 : 0);
  2251. uint8_t numBindings = 0;
  2252. // bind uniform buffer at slot 0
  2253. bindings[numBindings].binding = 0;
  2254. bindings[numBindings].visibility = _vsh->m_stage;
  2255. bindings[numBindings].type = wgpu::BindingType::UniformBuffer;
  2256. bindings[numBindings].hasDynamicOffset = true;
  2257. numBindings++;
  2258. if (m_numUniforms > 1)
  2259. {
  2260. bindings[numBindings].binding = 48;
  2261. bindings[numBindings].visibility = wgpu::ShaderStage::Fragment;
  2262. bindings[numBindings].type = wgpu::BindingType::UniformBuffer;
  2263. bindings[numBindings].hasDynamicOffset = true;
  2264. numBindings++;
  2265. }
  2266. uint8_t numSamplers = 0;
  2267. for (uint32_t ii = 0; ii < _vsh->m_numSamplers; ++ii)
  2268. {
  2269. m_textures[ii] = _vsh->m_textures[ii];
  2270. m_samplers[ii] = _vsh->m_samplers[ii];
  2271. bindings[numBindings++] = _vsh->m_textures[ii];
  2272. bindings[numBindings++] = _vsh->m_samplers[ii];
  2273. }
  2274. numSamplers += _vsh->m_numSamplers;
  2275. if (NULL != _fsh)
  2276. {
  2277. for (uint32_t ii = 0; ii < _fsh->m_numSamplers; ++ii)
  2278. {
  2279. m_textures[numSamplers + ii] = _fsh->m_textures[ii];
  2280. m_samplers[numSamplers + ii] = _fsh->m_samplers[ii];
  2281. bindings[numBindings++] = _fsh->m_textures[ii];
  2282. bindings[numBindings++] = _fsh->m_samplers[ii];
  2283. }
  2284. numSamplers += _fsh->m_numSamplers;
  2285. }
  2286. for (uint8_t stage = 0; stage < BGFX_CONFIG_MAX_TEXTURE_SAMPLERS; ++stage)
  2287. {
  2288. if (isValid(m_vsh->m_bindInfo[stage].m_uniform))
  2289. {
  2290. m_bindInfo[stage] = m_vsh->m_bindInfo[stage];
  2291. }
  2292. else if (NULL != m_fsh && isValid(m_fsh->m_bindInfo[stage].m_uniform))
  2293. {
  2294. m_bindInfo[stage] = m_fsh->m_bindInfo[stage];
  2295. m_bindInfo[stage].m_index += _vsh->m_numSamplers;
  2296. }
  2297. }
  2298. m_numSamplers = numSamplers;
  2299. for (uint32_t ii = 0; ii < _vsh->m_numBuffers; ++ii)
  2300. {
  2301. m_buffers[ii] = _vsh->m_buffers[ii];
  2302. bindings[numBindings++] = _vsh->m_buffers[ii];
  2303. }
  2304. m_numBuffers = _vsh->m_numBuffers;
  2305. BX_ASSERT(m_numUniforms + m_numSamplers * 2 + m_numBuffers == numBindings, "");
  2306. wgpu::BindGroupLayoutDescriptor bindGroupDesc;
  2307. bindGroupDesc.entryCount = numBindings;
  2308. bindGroupDesc.entries = bindings;
  2309. m_bindGroupLayout = s_renderWgpu->m_device.CreateBindGroupLayout(&bindGroupDesc);
  2310. bx::HashMurmur2A murmur;
  2311. murmur.begin();
  2312. murmur.add(m_numUniforms);
  2313. murmur.add(m_textures, sizeof(wgpu::BindGroupLayoutEntry) * numSamplers);
  2314. murmur.add(m_samplers, sizeof(wgpu::BindGroupLayoutEntry) * numSamplers);
  2315. murmur.add(m_buffers, sizeof(wgpu::BindGroupLayoutEntry) * m_numBuffers);
  2316. m_bindGroupLayoutHash = murmur.end();
  2317. }
  2318. void ProgramWgpu::destroy()
  2319. {
  2320. m_vsh = NULL;
  2321. m_fsh = NULL;
  2322. if ( NULL != m_computePS )
  2323. {
  2324. BX_DELETE(g_allocator, m_computePS);
  2325. m_computePS = NULL;
  2326. }
  2327. }
  2328. void BufferWgpu::create(uint32_t _size, void* _data, uint16_t _flags, uint16_t _stride, bool _vertex)
  2329. {
  2330. BX_UNUSED(_stride);
  2331. m_size = _size;
  2332. m_flags = _flags;
  2333. m_vertex = _vertex;
  2334. const uint32_t paddedSize = bx::strideAlign(_size, 4);
  2335. bool storage = m_flags & BGFX_BUFFER_COMPUTE_READ_WRITE;
  2336. bool indirect = m_flags & BGFX_BUFFER_DRAW_INDIRECT;
  2337. wgpu::BufferDescriptor desc;
  2338. desc.size = paddedSize;
  2339. desc.usage = _vertex ? wgpu::BufferUsage::Vertex : wgpu::BufferUsage::Index;
  2340. desc.usage |= (storage || indirect) ? wgpu::BufferUsage::Storage : wgpu::BufferUsage::None;
  2341. desc.usage |= indirect ? wgpu::BufferUsage::Indirect : wgpu::BufferUsage::None;
  2342. desc.usage |= NULL == _data ? wgpu::BufferUsage::CopyDst : wgpu::BufferUsage::None;
  2343. if(NULL != _data)
  2344. {
  2345. wgpu::CreateBufferMappedResult mapped = s_renderWgpu->m_device.CreateBufferMapped(&desc);
  2346. m_ptr = mapped.buffer;
  2347. bx::memCopy(mapped.data, _data, _size);
  2348. mapped.buffer.Unmap();
  2349. }
  2350. else
  2351. {
  2352. m_ptr = s_renderWgpu->m_device.CreateBuffer(&desc);
  2353. }
  2354. }
  2355. void BufferWgpu::update(uint32_t _offset, uint32_t _size, void* _data, bool _discard)
  2356. {
  2357. wgpu::CommandEncoder& bce = s_renderWgpu->getBlitCommandEncoder();
  2358. if (!m_vertex && !_discard)
  2359. {
  2360. if ( m_dynamic == NULL )
  2361. {
  2362. m_dynamic = (uint8_t*)BX_ALLOC(g_allocator, m_size);
  2363. }
  2364. bx::memCopy(m_dynamic + _offset, _data, _size);
  2365. uint32_t start = _offset & 4;
  2366. uint32_t end = bx::strideAlign(_offset + _size, 4);
  2367. wgpu::BufferDescriptor desc;
  2368. desc.size = end - start;
  2369. desc.usage = wgpu::BufferUsage::CopyDst | wgpu::BufferUsage::CopySrc;
  2370. wgpu::CreateBufferMappedResult mapped = s_renderWgpu->m_device.CreateBufferMapped(&desc);
  2371. wgpu::Buffer staging = mapped.buffer;
  2372. bx::memCopy(mapped.data, m_dynamic, end - start);
  2373. mapped.buffer.Unmap();
  2374. // TODO pad to 4 bytes
  2375. bce.CopyBufferToBuffer(staging, 0, m_ptr, start, end - start);
  2376. s_renderWgpu->m_cmd.release(staging);
  2377. }
  2378. else
  2379. {
  2380. wgpu::BufferDescriptor desc;
  2381. desc.size = _size;
  2382. desc.usage = wgpu::BufferUsage::CopyDst | wgpu::BufferUsage::CopySrc;
  2383. wgpu::CreateBufferMappedResult mapped = s_renderWgpu->m_device.CreateBufferMapped(&desc);
  2384. wgpu::Buffer staging = mapped.buffer;
  2385. bx::memCopy(mapped.data, _data, _size);
  2386. mapped.buffer.Unmap();
  2387. bce.CopyBufferToBuffer(staging, 0, m_ptr, _offset, _size);
  2388. s_renderWgpu->m_cmd.release(staging);
  2389. }
  2390. }
  2391. void VertexBufferWgpu::create(uint32_t _size, void* _data, VertexLayoutHandle _layoutHandle, uint16_t _flags)
  2392. {
  2393. m_layoutHandle = _layoutHandle;
  2394. uint16_t stride = isValid(_layoutHandle)
  2395. ? s_renderWgpu->m_vertexDecls[_layoutHandle.idx].m_stride
  2396. : 0
  2397. ;
  2398. BufferWgpu::create(_size, _data, _flags, stride, true);
  2399. }
  2400. void TextureWgpu::create(TextureHandle _handle, const Memory* _mem, uint64_t _flags, uint8_t _skip)
  2401. {
  2402. m_handle = _handle;
  2403. m_sampler = s_renderWgpu->getSamplerState(uint32_t(_flags) );
  2404. bimg::ImageContainer imageContainer;
  2405. if (bimg::imageParse(imageContainer, _mem->data, _mem->size) )
  2406. {
  2407. const bimg::ImageBlockInfo& blockInfo = getBlockInfo(bimg::TextureFormat::Enum(imageContainer.m_format) );
  2408. const uint8_t startLod = bx::min<uint8_t>(_skip, imageContainer.m_numMips-1);
  2409. bimg::TextureInfo ti;
  2410. bimg::imageGetSize(
  2411. &ti
  2412. , uint16_t(imageContainer.m_width >>startLod)
  2413. , uint16_t(imageContainer.m_height>>startLod)
  2414. , uint16_t(imageContainer.m_depth >>startLod)
  2415. , imageContainer.m_cubeMap
  2416. , 1 < imageContainer.m_numMips
  2417. , imageContainer.m_numLayers
  2418. , imageContainer.m_format
  2419. );
  2420. ti.numMips = bx::min<uint8_t>(imageContainer.m_numMips-startLod, ti.numMips);
  2421. m_flags = _flags;
  2422. m_width = ti.width;
  2423. m_height = ti.height;
  2424. m_depth = ti.depth;
  2425. m_numLayers = ti.numLayers;
  2426. m_numMips = ti.numMips;
  2427. m_numSides = ti.numLayers * (imageContainer.m_cubeMap ? 6 : 1);
  2428. m_requestedFormat = TextureFormat::Enum(imageContainer.m_format);
  2429. m_textureFormat = getViableTextureFormat(imageContainer);
  2430. if (m_requestedFormat == bgfx::TextureFormat::D16)
  2431. m_textureFormat = bgfx::TextureFormat::D32F;
  2432. const bool compressed = bimg::isCompressed(bimg::TextureFormat::Enum(imageContainer.m_format));
  2433. if (compressed)
  2434. m_textureFormat = bgfx::TextureFormat::BGRA8;
  2435. const bool convert = m_textureFormat != m_requestedFormat;
  2436. const uint8_t bpp = bimg::getBitsPerPixel(bimg::TextureFormat::Enum(m_textureFormat) );
  2437. wgpu::TextureDescriptor desc = defaultDescriptor<wgpu::TextureDescriptor>();
  2438. //desc.label = getName(_handle);
  2439. if (1 < ti.numLayers)
  2440. {
  2441. if (imageContainer.m_cubeMap)
  2442. {
  2443. m_type = TextureCube;
  2444. desc.dimension = wgpu::TextureDimension::e2D;
  2445. }
  2446. else
  2447. {
  2448. m_type = Texture2D;
  2449. desc.dimension = wgpu::TextureDimension::e2D;
  2450. }
  2451. }
  2452. else if (imageContainer.m_cubeMap)
  2453. {
  2454. m_type = TextureCube;
  2455. desc.dimension = wgpu::TextureDimension::e2D;
  2456. }
  2457. else if (1 < imageContainer.m_depth)
  2458. {
  2459. m_type = Texture3D;
  2460. desc.dimension = wgpu::TextureDimension::e3D;
  2461. }
  2462. else
  2463. {
  2464. m_type = Texture2D;
  2465. desc.dimension = wgpu::TextureDimension::e2D;
  2466. }
  2467. const uint16_t numSides = ti.numLayers * (imageContainer.m_cubeMap ? 6 : 1);
  2468. const uint32_t numSrd = numSides * ti.numMips;
  2469. const bool writeOnly = 0 != (_flags&BGFX_TEXTURE_RT_WRITE_ONLY);
  2470. const bool computeWrite = 0 != (_flags&BGFX_TEXTURE_COMPUTE_WRITE);
  2471. const bool renderTarget = 0 != (_flags&BGFX_TEXTURE_RT_MASK);
  2472. const bool srgb = 0 != (_flags&BGFX_TEXTURE_SRGB);
  2473. BX_TRACE("Texture %3d: %s (requested: %s), layers %d, %dx%d%s RT[%c], WO[%c], CW[%c], sRGB[%c]"
  2474. , this - s_renderWgpu->m_textures
  2475. , getName( (TextureFormat::Enum)m_textureFormat)
  2476. , getName( (TextureFormat::Enum)m_requestedFormat)
  2477. , ti.numLayers
  2478. , ti.width
  2479. , ti.height
  2480. , imageContainer.m_cubeMap ? "x6" : ""
  2481. , renderTarget ? 'x' : ' '
  2482. , writeOnly ? 'x' : ' '
  2483. , computeWrite ? 'x' : ' '
  2484. , srgb ? 'x' : ' '
  2485. );
  2486. const uint32_t msaaQuality = bx::uint32_satsub( (_flags&BGFX_TEXTURE_RT_MSAA_MASK)>>BGFX_TEXTURE_RT_MSAA_SHIFT, 1);
  2487. const int32_t sampleCount = s_msaa[msaaQuality];
  2488. wgpu::TextureFormat format = wgpu::TextureFormat::Undefined;
  2489. if (srgb)
  2490. {
  2491. format = s_textureFormat[m_textureFormat].m_fmtSrgb;
  2492. BX_WARN(format != wgpu::TextureFormat::Undefined
  2493. , "sRGB not supported for texture format %d"
  2494. , m_textureFormat
  2495. );
  2496. }
  2497. if (format == wgpu::TextureFormat::Undefined)
  2498. {
  2499. // not swizzled and not sRGB, or sRGB unsupported
  2500. format = s_textureFormat[m_textureFormat].m_fmt;
  2501. }
  2502. desc.format = format;
  2503. desc.size.width = m_width;
  2504. desc.size.height = m_height;
  2505. desc.size.depth = bx::uint32_max(1,imageContainer.m_depth);
  2506. desc.mipLevelCount = m_numMips;
  2507. desc.sampleCount = 1;
  2508. desc.arrayLayerCount = m_numSides;
  2509. desc.usage = wgpu::TextureUsage::Sampled;
  2510. desc.usage |= wgpu::TextureUsage::CopyDst;
  2511. desc.usage |= wgpu::TextureUsage::CopySrc;
  2512. if (computeWrite)
  2513. {
  2514. desc.usage |= wgpu::TextureUsage::Storage;
  2515. }
  2516. if (renderTarget)
  2517. {
  2518. desc.usage |= wgpu::TextureUsage::OutputAttachment;
  2519. }
  2520. m_ptr = s_renderWgpu->m_device.CreateTexture(&desc);
  2521. if (sampleCount > 1)
  2522. {
  2523. desc.sampleCount = sampleCount;
  2524. m_ptrMsaa = s_renderWgpu->m_device.CreateTexture(&desc);
  2525. }
  2526. // decode images
  2527. struct ImageInfo
  2528. {
  2529. uint8_t* data;
  2530. uint32_t width;
  2531. uint32_t height;
  2532. uint32_t depth;
  2533. uint32_t pitch;
  2534. uint32_t slice;
  2535. uint32_t size;
  2536. uint8_t mipLevel;
  2537. uint8_t layer;
  2538. };
  2539. ImageInfo* imageInfos = (ImageInfo*)BX_ALLOC(g_allocator, sizeof(ImageInfo) * numSrd);
  2540. bx::memSet(imageInfos, 0, sizeof(ImageInfo) * numSrd);
  2541. uint32_t alignment = 1; // tightly aligned buffer
  2542. uint32_t kk = 0;
  2543. for (uint8_t side = 0; side < numSides; ++side)
  2544. {
  2545. for (uint8_t lod = 0; lod < ti.numMips; ++lod)
  2546. {
  2547. bimg::ImageMip mip;
  2548. if (bimg::imageGetRawData(imageContainer, side, lod + startLod, _mem->data, _mem->size, mip))
  2549. {
  2550. if (convert)
  2551. {
  2552. const uint32_t pitch = bx::strideAlign(bx::max<uint32_t>(mip.m_width, 4) * bpp / 8, alignment);
  2553. const uint32_t slice = bx::strideAlign(bx::max<uint32_t>(mip.m_height, 4) * pitch, alignment);
  2554. const uint32_t size = slice * mip.m_depth;
  2555. uint8_t* temp = (uint8_t*)BX_ALLOC(g_allocator, size);
  2556. bimg::imageDecodeToBgra8(
  2557. g_allocator
  2558. , temp
  2559. , mip.m_data
  2560. , mip.m_width
  2561. , mip.m_height
  2562. , pitch
  2563. , mip.m_format
  2564. );
  2565. imageInfos[kk].data = temp;
  2566. imageInfos[kk].width = mip.m_width;
  2567. imageInfos[kk].height = mip.m_height;
  2568. imageInfos[kk].depth = mip.m_depth;
  2569. imageInfos[kk].pitch = pitch;
  2570. imageInfos[kk].slice = slice;
  2571. imageInfos[kk].size = size;
  2572. imageInfos[kk].mipLevel = lod;
  2573. imageInfos[kk].layer = side;
  2574. }
  2575. else if (compressed)
  2576. {
  2577. const uint32_t pitch = bx::strideAlign((mip.m_width / blockInfo.blockWidth) * mip.m_blockSize, alignment);
  2578. const uint32_t slice = bx::strideAlign((mip.m_height / blockInfo.blockHeight) * pitch, alignment);
  2579. const uint32_t size = slice * mip.m_depth;
  2580. uint8_t* temp = (uint8_t*)BX_ALLOC(g_allocator, size);
  2581. bimg::imageCopy(
  2582. temp
  2583. , mip.m_height / blockInfo.blockHeight
  2584. , (mip.m_width / blockInfo.blockWidth) * mip.m_blockSize
  2585. , mip.m_depth
  2586. , mip.m_data
  2587. , pitch
  2588. );
  2589. imageInfos[kk].data = temp;
  2590. imageInfos[kk].width = mip.m_width;
  2591. imageInfos[kk].height = mip.m_height;
  2592. imageInfos[kk].depth = mip.m_depth;
  2593. imageInfos[kk].pitch = pitch;
  2594. imageInfos[kk].slice = slice;
  2595. imageInfos[kk].size = size;
  2596. imageInfos[kk].mipLevel = lod;
  2597. imageInfos[kk].layer = side;
  2598. }
  2599. else
  2600. {
  2601. const uint32_t pitch = bx::strideAlign(mip.m_width * mip.m_bpp / 8, alignment);
  2602. const uint32_t slice = bx::strideAlign(mip.m_height * pitch, alignment);
  2603. const uint32_t size = slice * mip.m_depth;
  2604. uint8_t* temp = (uint8_t*)BX_ALLOC(g_allocator, size);
  2605. bimg::imageCopy(temp
  2606. , mip.m_height
  2607. , mip.m_width * mip.m_bpp / 8
  2608. , mip.m_depth
  2609. , mip.m_data
  2610. , pitch
  2611. );
  2612. imageInfos[kk].data = temp;
  2613. imageInfos[kk].width = mip.m_width;
  2614. imageInfos[kk].height = mip.m_height;
  2615. imageInfos[kk].depth = mip.m_depth;
  2616. imageInfos[kk].pitch = pitch;
  2617. imageInfos[kk].slice = slice;
  2618. imageInfos[kk].size = size;
  2619. imageInfos[kk].mipLevel = lod;
  2620. imageInfos[kk].layer = side;
  2621. }
  2622. }
  2623. ++kk;
  2624. }
  2625. }
  2626. uint32_t totalMemSize = 0;
  2627. for (uint32_t ii = 0; ii < numSrd; ++ii)
  2628. {
  2629. const uint32_t dstpitch = bx::strideAlign(imageInfos[ii].pitch, kMinBufferOffsetAlignment);
  2630. totalMemSize += dstpitch * imageInfos[ii].height;
  2631. //totalMemSize += imageInfos[ii].size;
  2632. }
  2633. wgpu::Buffer stagingBuffer;
  2634. if (totalMemSize > 0)
  2635. {
  2636. wgpu::BufferDescriptor staginBufferDesc;
  2637. staginBufferDesc.size = totalMemSize;
  2638. staginBufferDesc.usage = wgpu::BufferUsage::CopyDst | wgpu::BufferUsage::CopySrc;
  2639. wgpu::CreateBufferMappedResult mapped = s_renderWgpu->m_device.CreateBufferMapped(&staginBufferDesc);
  2640. stagingBuffer = mapped.buffer;
  2641. uint64_t offset = 0;
  2642. for (uint32_t ii = 0; ii < numSrd; ++ii)
  2643. {
  2644. const uint32_t dstpitch = bx::strideAlign(imageInfos[ii].pitch, kMinBufferOffsetAlignment);
  2645. const uint8_t* src = (uint8_t*)imageInfos[ii].data;
  2646. uint8_t* dst = (uint8_t*)mapped.data;
  2647. for (uint32_t yy = 0; yy < imageInfos[ii].height; ++yy, src += imageInfos[ii].pitch, offset += dstpitch)
  2648. {
  2649. bx::memCopy(dst + offset, src, imageInfos[ii].pitch);
  2650. }
  2651. //bx::memCopy(dst + offset, imageInfos[ii].data, imageInfos[ii].size);
  2652. //offset += imageInfos[ii].size;
  2653. }
  2654. mapped.buffer.Unmap();
  2655. }
  2656. wgpu::BufferCopyView* bufferCopyView = (wgpu::BufferCopyView*)BX_ALLOC(g_allocator, sizeof(wgpu::BufferCopyView) * numSrd);
  2657. wgpu::TextureCopyView* textureCopyView = (wgpu::TextureCopyView*)BX_ALLOC(g_allocator, sizeof(wgpu::TextureCopyView) * numSrd);
  2658. wgpu::Extent3D* textureCopySize = (wgpu::Extent3D*)BX_ALLOC(g_allocator, sizeof(wgpu::Extent3D) * numSrd);
  2659. uint64_t offset = 0;
  2660. for (uint32_t ii = 0; ii < numSrd; ++ii)
  2661. {
  2662. const uint32_t dstpitch = bx::strideAlign(imageInfos[ii].pitch, kMinBufferOffsetAlignment);
  2663. uint32_t idealWidth = bx::max<uint32_t>(1, m_width >> imageInfos[ii].mipLevel);
  2664. uint32_t idealHeight = bx::max<uint32_t>(1, m_height >> imageInfos[ii].mipLevel);
  2665. BX_PLACEMENT_NEW(&bufferCopyView[ii], wgpu::BufferCopyView)();
  2666. BX_PLACEMENT_NEW(&textureCopyView[ii], wgpu::TextureCopyView)();
  2667. BX_PLACEMENT_NEW(&textureCopySize[ii], wgpu::Extent3D)();
  2668. bufferCopyView[ii].buffer = stagingBuffer;
  2669. bufferCopyView[ii].offset = offset;
  2670. bufferCopyView[ii].bytesPerRow = dstpitch; // assume that image data are tightly aligned
  2671. bufferCopyView[ii].rowsPerImage = 0; // assume that image data are tightly aligned
  2672. textureCopyView[ii].texture = m_ptr;
  2673. //textureCopyView[ii].imageSubresource.aspectMask = m_vkTextureAspect;
  2674. textureCopyView[ii].mipLevel = imageInfos[ii].mipLevel;
  2675. textureCopyView[ii].arrayLayer = imageInfos[ii].layer;
  2676. //textureCopyView[ii].layerCount = 1;
  2677. textureCopyView[ii].origin = { 0, 0, 0 };
  2678. textureCopySize[ii] = { idealWidth, idealHeight, imageInfos[ii].depth };
  2679. offset += dstpitch * imageInfos[ii].height;
  2680. //offset += imageInfos[ii].size;
  2681. }
  2682. if (stagingBuffer)
  2683. {
  2684. wgpu::CommandEncoder encoder = s_renderWgpu->getBlitCommandEncoder();
  2685. //wgpu::CommandEncoder encoder = s_renderWgpu->m_cmd.m_encoder;
  2686. for (uint32_t ii = 0; ii < numSrd; ++ii)
  2687. {
  2688. encoder.CopyBufferToTexture(&bufferCopyView[ii], &textureCopyView[ii], &textureCopySize[ii]);
  2689. }
  2690. }
  2691. else
  2692. {
  2693. //VkCommandBuffer commandBuffer = s_renderVK->beginNewCommand();
  2694. //setImageMemoryBarrier(
  2695. // commandBuffer
  2696. // , (m_flags & BGFX_TEXTURE_COMPUTE_WRITE
  2697. // ? VK_IMAGE_LAYOUT_GENERAL
  2698. // : VK_IMAGE_LAYOUT_SHADER_READ_ONLY_OPTIMAL
  2699. // )
  2700. //);
  2701. //s_renderVK->submitCommandAndWait(commandBuffer);
  2702. }
  2703. //vkFreeMemory(device, stagingDeviceMem, allocatorCb);
  2704. //vkDestroy(stagingBuffer);
  2705. BX_FREE(g_allocator, bufferCopyView);
  2706. BX_FREE(g_allocator, textureCopyView);
  2707. BX_FREE(g_allocator, textureCopySize);
  2708. for (uint32_t ii = 0; ii < numSrd; ++ii)
  2709. {
  2710. BX_FREE(g_allocator, imageInfos[ii].data);
  2711. }
  2712. BX_FREE(g_allocator, imageInfos);
  2713. }
  2714. }
  2715. void TextureWgpu::update(uint8_t _side, uint8_t _mip, const Rect& _rect, uint16_t _z, uint16_t _depth, uint16_t _pitch, const Memory* _mem)
  2716. {
  2717. BX_UNUSED(_side); BX_UNUSED(_mip); BX_UNUSED(_depth); BX_UNUSED(_z);
  2718. const uint32_t bpp = bimg::getBitsPerPixel(bimg::TextureFormat::Enum(m_textureFormat) );
  2719. const uint32_t rectpitch = _rect.m_width*bpp/8;
  2720. const uint32_t srcpitch = UINT16_MAX == _pitch ? rectpitch : _pitch;
  2721. const uint32_t slice = ( (m_type == Texture3D) ? 0 : _side + _z * (m_type == TextureCube ? 6 : 1) );
  2722. const uint16_t zz = (m_type == Texture3D) ? _z : 0 ;
  2723. const bool convert = m_textureFormat != m_requestedFormat;
  2724. uint8_t* data = _mem->data;
  2725. uint8_t* temp = NULL;
  2726. if (convert)
  2727. {
  2728. temp = (uint8_t*)BX_ALLOC(g_allocator, rectpitch*_rect.m_height);
  2729. bimg::imageDecodeToBgra8(
  2730. g_allocator
  2731. , temp
  2732. , data
  2733. , _rect.m_width
  2734. , _rect.m_height
  2735. , srcpitch
  2736. , bimg::TextureFormat::Enum(m_requestedFormat)
  2737. );
  2738. data = temp;
  2739. }
  2740. const uint32_t dstpitch = bx::strideAlign(rectpitch, kMinBufferOffsetAlignment);
  2741. wgpu::BufferDescriptor desc;
  2742. desc.size = dstpitch * _rect.m_height;
  2743. desc.usage = wgpu::BufferUsage::CopyDst | wgpu::BufferUsage::CopySrc;
  2744. wgpu::CreateBufferMappedResult mapped = s_renderWgpu->m_device.CreateBufferMapped(&desc);
  2745. wgpu::Buffer staging = mapped.buffer;
  2746. const uint8_t* src = (uint8_t*)data;
  2747. uint8_t* dst = (uint8_t*)mapped.data;
  2748. uint64_t offset = 0;
  2749. for (uint32_t yy = 0; yy < _rect.m_height; ++yy, src += srcpitch, offset += dstpitch)
  2750. {
  2751. const uint32_t size = bx::strideAlign(rectpitch, 4);
  2752. bx::memCopy(dst + offset, src, size);
  2753. }
  2754. mapped.buffer.Unmap();
  2755. wgpu::BufferCopyView srcView;
  2756. srcView.buffer = staging;
  2757. srcView.offset = 0;
  2758. srcView.bytesPerRow = dstpitch;
  2759. srcView.rowsPerImage = 0;
  2760. wgpu::TextureCopyView destView;
  2761. destView.texture = m_ptr;
  2762. destView.mipLevel = _mip;
  2763. destView.arrayLayer = _side;
  2764. destView.origin = { _rect.m_x, _rect.m_y, zz };
  2765. //destView.origin = { _rect.m_x, _rect.m_y, _z };
  2766. wgpu::Extent3D destExtent = { _rect.m_width, _rect.m_height, _depth };
  2767. //region.imageSubresource.aspectMask = m_vkTextureAspect;
  2768. wgpu::CommandEncoder encoder = s_renderWgpu->getBlitCommandEncoder();
  2769. //wgpu::CommandEncoder encoder = s_renderWgpu->m_cmd.m_encoder;
  2770. encoder.CopyBufferToTexture(&srcView, &destView, &destExtent);
  2771. //wgpu::CommandBuffer copy = encoder.Finish();
  2772. //wgpu::Queue queue = s_renderWgpu->m_queue;
  2773. //queue.Submit(1, &copy);
  2774. //staging.Destroy();
  2775. if (NULL != temp)
  2776. {
  2777. BX_FREE(g_allocator, temp);
  2778. }
  2779. }
  2780. void BindStateWgpu::clear()
  2781. {
  2782. m_bindGroup = NULL;
  2783. }
  2784. void StagingBufferWgpu::create(uint32_t _size, bool mapped)
  2785. {
  2786. wgpu::BufferDescriptor desc;
  2787. desc.size = BGFX_CONFIG_MAX_DRAW_CALLS * 128;
  2788. desc.usage = wgpu::BufferUsage::MapWrite | wgpu::BufferUsage::CopySrc;
  2789. if (mapped)
  2790. {
  2791. wgpu::CreateBufferMappedResult mapped = s_renderWgpu->m_device.CreateBufferMapped(&desc);
  2792. m_buffer = mapped.buffer;
  2793. m_data = mapped.data;
  2794. m_size = mapped.dataLength;
  2795. }
  2796. else
  2797. {
  2798. m_buffer = s_renderWgpu->m_device.CreateBuffer(&desc);
  2799. map();
  2800. }
  2801. }
  2802. void StagingBufferWgpu::map()
  2803. {
  2804. auto ready = [](WGPUBufferMapAsyncStatus status, void* data, uint64_t dataLength, void* userdata)
  2805. {
  2806. if (status == WGPUBufferMapAsyncStatus_Success)
  2807. static_cast<StagingBufferWgpu*>(userdata)->mapped(data, dataLength);
  2808. };
  2809. m_buffer.MapWriteAsync(ready, this);
  2810. }
  2811. void StagingBufferWgpu::unmap()
  2812. {
  2813. m_data = NULL;
  2814. m_size = 0;
  2815. m_buffer.Unmap();
  2816. }
  2817. void StagingBufferWgpu::destroy()
  2818. {
  2819. m_buffer = NULL;
  2820. }
  2821. void StagingBufferWgpu::mapped(void* _data, uint64_t _size)
  2822. {
  2823. m_data = _data;
  2824. m_size = _size;
  2825. }
  2826. void ScratchBufferWgpu::create(uint32_t _size)
  2827. {
  2828. m_offset = 0;
  2829. m_size = _size;
  2830. wgpu::BufferDescriptor desc;
  2831. desc.size = BGFX_CONFIG_MAX_DRAW_CALLS * 128;
  2832. desc.usage = wgpu::BufferUsage::CopyDst | wgpu::BufferUsage::Uniform;
  2833. m_buffer = s_renderWgpu->m_device.CreateBuffer(&desc);
  2834. }
  2835. void ScratchBufferWgpu::destroy()
  2836. {
  2837. }
  2838. void ScratchBufferWgpu::begin()
  2839. {
  2840. for (uint8_t ii = 0; ii < WEBGPU_NUM_UNIFORM_BUFFERS; ++ii)
  2841. {
  2842. if (NULL != s_renderWgpu->m_uniformBuffers[ii].m_data)
  2843. {
  2844. m_staging = &s_renderWgpu->m_uniformBuffers[ii];
  2845. break;
  2846. }
  2847. }
  2848. BX_ASSERT(NULL != m_staging, "No available mapped uniform buffer");
  2849. }
  2850. uint32_t ScratchBufferWgpu::write(void* data, uint64_t _size, uint64_t _offset)
  2851. {
  2852. BX_ASSERT(nullptr != m_staging, "Cannot write uniforms outside of begin()/submit() calls");
  2853. BX_ASSERT(m_size > m_offset + _offset, "Out-of-bounds scratch buffer write");
  2854. uint32_t offset = m_offset;
  2855. bx::memCopy((void*)((uint8_t*)m_staging->m_data + offset), data, _size);
  2856. m_offset += _offset;
  2857. return offset;
  2858. }
  2859. uint32_t ScratchBufferWgpu::write(void* data, uint64_t _size)
  2860. {
  2861. BX_ASSERT(nullptr != m_staging, "Cannot write uniforms outside of begin()/submit() calls");
  2862. BX_ASSERT(m_size > m_offset + _size, "Out-of-bounds scratch buffer write");
  2863. uint32_t offset = m_offset;
  2864. bx::memCopy((void*)((uint8_t*)m_staging->m_data + offset), data, _size);
  2865. m_offset += _size;
  2866. return offset;
  2867. }
  2868. void ScratchBufferWgpu::submit()
  2869. {
  2870. m_staging->unmap();
  2871. if (m_offset != 0)
  2872. {
  2873. wgpu::CommandEncoder& bce = s_renderWgpu->getStagingEncoder();
  2874. bce.CopyBufferToBuffer(m_staging->m_buffer, 0, m_buffer, 0, m_offset);
  2875. }
  2876. }
  2877. void ScratchBufferWgpu::release()
  2878. {
  2879. m_staging->map();
  2880. m_staging = NULL;
  2881. m_offset = 0;
  2882. }
  2883. void BindStateCacheWgpu::create() //(uint32_t _maxBindGroups)
  2884. {
  2885. //m_maxBindStates = 1024; // _maxBindStates;
  2886. m_currentBindState = 0;
  2887. }
  2888. void BindStateCacheWgpu::destroy()
  2889. {
  2890. reset();
  2891. }
  2892. void BindStateCacheWgpu::reset()
  2893. {
  2894. for (size_t i = 0; i < m_currentBindState; ++i)
  2895. {
  2896. m_bindStates[i] = {};
  2897. }
  2898. m_currentBindState = 0;
  2899. }
  2900. wgpu::TextureView TextureWgpu::getTextureMipLevel(int _mip)
  2901. {
  2902. if (_mip >= 0
  2903. && _mip < m_numMips
  2904. && m_ptr)
  2905. {
  2906. if (!m_ptrMips[_mip])
  2907. {
  2908. wgpu::TextureViewDescriptor desc;
  2909. desc.baseMipLevel = _mip;
  2910. desc.mipLevelCount = 1;
  2911. desc.format = s_textureFormat[m_textureFormat].m_fmt;
  2912. if (TextureCube == m_type)
  2913. {
  2914. //desc.dimension = MTLTextureType2DArray;
  2915. desc.baseArrayLayer = 0;
  2916. desc.arrayLayerCount = m_numLayers * 6;
  2917. }
  2918. else
  2919. {
  2920. desc.baseArrayLayer = 0;
  2921. desc.arrayLayerCount = m_numLayers;
  2922. }
  2923. m_ptrMips[_mip] = m_ptr.CreateView(&desc);
  2924. }
  2925. return m_ptrMips[_mip];
  2926. }
  2927. return wgpu::TextureView();
  2928. }
  2929. void SwapChainWgpu::init(wgpu::Device _device, void* _nwh, uint32_t _width, uint32_t _height)
  2930. {
  2931. BX_UNUSED(_nwh);
  2932. wgpu::SwapChainDescriptor desc;
  2933. desc.usage = wgpu::TextureUsage::OutputAttachment;
  2934. desc.width = _width;
  2935. desc.height = _height;
  2936. #if !BX_PLATFORM_EMSCRIPTEN
  2937. m_impl = createSwapChain(_device, _nwh);
  2938. desc.presentMode = wgpu::PresentMode::Immediate;
  2939. desc.format = wgpu::TextureFormat::RGBA8Unorm;
  2940. desc.implementation = reinterpret_cast<uint64_t>(&m_impl);
  2941. m_swapChain = _device.CreateSwapChain(nullptr, &desc);
  2942. #else
  2943. wgpu::SurfaceDescriptorFromHTMLCanvasId canvasDesc{};
  2944. canvasDesc.id = "canvas";
  2945. wgpu::SurfaceDescriptor surfDesc{};
  2946. surfDesc.nextInChain = &canvasDesc;
  2947. wgpu::Surface surface = wgpu::Instance().CreateSurface(&surfDesc);
  2948. desc.presentMode = wgpu::PresentMode::Fifo;
  2949. desc.format = wgpu::TextureFormat::BGRA8Unorm;
  2950. m_swapChain = _device.CreateSwapChain(surface, &desc);
  2951. #endif
  2952. m_colorFormat = desc.format;
  2953. m_depthFormat = wgpu::TextureFormat::Depth24PlusStencil8;
  2954. }
  2955. void SwapChainWgpu::resize(FrameBufferWgpu& _frameBuffer, uint32_t _width, uint32_t _height, uint32_t _flags)
  2956. {
  2957. BX_TRACE("SwapChainWgpu::resize");
  2958. const int32_t sampleCount = s_msaa[(_flags&BGFX_RESET_MSAA_MASK)>>BGFX_RESET_MSAA_SHIFT];
  2959. wgpu::TextureFormat format = (_flags & BGFX_RESET_SRGB_BACKBUFFER)
  2960. #ifdef DAWN_ENABLE_BACKEND_VULKAN
  2961. ? wgpu::TextureFormat::BGRA8UnormSrgb
  2962. : wgpu::TextureFormat::BGRA8Unorm
  2963. #else
  2964. ? wgpu::TextureFormat::RGBA8UnormSrgb
  2965. : wgpu::TextureFormat::RGBA8Unorm
  2966. #endif
  2967. ;
  2968. #if !BX_PLATFORM_EMSCRIPTEN
  2969. m_swapChain.Configure(format, wgpu::TextureUsage::OutputAttachment, _width, _height);
  2970. #endif
  2971. m_colorFormat = format;
  2972. m_depthFormat = wgpu::TextureFormat::Depth24PlusStencil8;
  2973. bx::HashMurmur2A murmur;
  2974. murmur.begin();
  2975. murmur.add(1);
  2976. murmur.add((uint32_t)m_colorFormat);
  2977. murmur.add((uint32_t)m_depthFormat);
  2978. murmur.add((uint32_t)sampleCount);
  2979. _frameBuffer.m_pixelFormatHash = murmur.end();
  2980. wgpu::TextureDescriptor desc;
  2981. desc.dimension = wgpu::TextureDimension::e2D;
  2982. desc.size.width = _width;
  2983. desc.size.height = _height;
  2984. desc.size.depth = 1;
  2985. desc.mipLevelCount = 1;
  2986. desc.sampleCount = sampleCount;
  2987. desc.arrayLayerCount = 1;
  2988. desc.usage = wgpu::TextureUsage::OutputAttachment;
  2989. if (m_backBufferDepth)
  2990. {
  2991. m_backBufferDepth.Destroy();
  2992. }
  2993. desc.format = wgpu::TextureFormat::Depth24PlusStencil8;
  2994. m_backBufferDepth = s_renderWgpu->m_device.CreateTexture(&desc);
  2995. if (sampleCount > 1)
  2996. {
  2997. if (m_backBufferColorMsaa)
  2998. {
  2999. m_backBufferColorMsaa.Destroy();
  3000. }
  3001. desc.format = m_colorFormat;
  3002. desc.sampleCount = sampleCount;
  3003. m_backBufferColorMsaa = s_renderWgpu->m_device.CreateTexture(&desc);
  3004. }
  3005. }
  3006. void SwapChainWgpu::flip()
  3007. {
  3008. m_drawable = m_swapChain.GetCurrentTextureView();
  3009. }
  3010. wgpu::TextureView SwapChainWgpu::current()
  3011. {
  3012. if (!m_drawable)
  3013. m_drawable = m_swapChain.GetCurrentTextureView();
  3014. return m_drawable;
  3015. }
  3016. void FrameBufferWgpu::create(uint8_t _num, const Attachment* _attachment)
  3017. {
  3018. m_swapChain = NULL;
  3019. m_denseIdx = UINT16_MAX;
  3020. m_num = 0;
  3021. m_width = 0;
  3022. m_height = 0;
  3023. for (uint32_t ii = 0; ii < _num; ++ii)
  3024. {
  3025. const Attachment& at = _attachment[ii];
  3026. TextureHandle handle = at.handle;
  3027. if (isValid(handle) )
  3028. {
  3029. const TextureWgpu& texture = s_renderWgpu->m_textures[handle.idx];
  3030. if (0 == m_width)
  3031. {
  3032. m_width = texture.m_width;
  3033. m_height = texture.m_height;
  3034. }
  3035. if (bimg::isDepth(bimg::TextureFormat::Enum(texture.m_textureFormat) ) )
  3036. {
  3037. m_depthHandle = handle;
  3038. m_depthAttachment = at;
  3039. }
  3040. else
  3041. {
  3042. m_colorHandle[m_num] = handle;
  3043. m_colorAttachment[m_num] = at;
  3044. m_num++;
  3045. }
  3046. }
  3047. }
  3048. bx::HashMurmur2A murmur;
  3049. murmur.begin();
  3050. murmur.add(m_num);
  3051. for (uint32_t ii = 0; ii < m_num; ++ii)
  3052. {
  3053. const TextureWgpu& texture = s_renderWgpu->m_textures[m_colorHandle[ii].idx];
  3054. murmur.add(uint32_t(s_textureFormat[texture.m_textureFormat].m_fmt) );
  3055. }
  3056. if (!isValid(m_depthHandle) )
  3057. {
  3058. murmur.add(uint32_t(wgpu::TextureFormat::Undefined) );
  3059. }
  3060. else
  3061. {
  3062. const TextureWgpu& depthTexture = s_renderWgpu->m_textures[m_depthHandle.idx];
  3063. murmur.add(uint32_t(s_textureFormat[depthTexture.m_textureFormat].m_fmt) );
  3064. }
  3065. murmur.add(1); // SampleCount
  3066. m_pixelFormatHash = murmur.end();
  3067. }
  3068. bool FrameBufferWgpu::create(uint16_t _denseIdx, void* _nwh, uint32_t _width, uint32_t _height, TextureFormat::Enum _format, TextureFormat::Enum _depthFormat)
  3069. {
  3070. BX_UNUSED(_format, _depthFormat);
  3071. m_swapChain = BX_NEW(g_allocator, SwapChainWgpu);
  3072. m_num = 0;
  3073. m_width = _width;
  3074. m_height = _height;
  3075. m_nwh = _nwh;
  3076. m_denseIdx = _denseIdx;
  3077. m_swapChain->init(s_renderWgpu->m_device, _nwh, _width, _height);
  3078. m_swapChain->resize(*this, _width, _height, 0);
  3079. return m_swapChain->m_swapChain != NULL;
  3080. }
  3081. void FrameBufferWgpu::postReset()
  3082. {
  3083. }
  3084. uint16_t FrameBufferWgpu::destroy()
  3085. {
  3086. if (NULL != m_swapChain)
  3087. {
  3088. BX_DELETE(g_allocator, m_swapChain);
  3089. m_swapChain = NULL;
  3090. }
  3091. m_num = 0;
  3092. m_nwh = NULL;
  3093. m_depthHandle.idx = kInvalidHandle;
  3094. uint16_t denseIdx = m_denseIdx;
  3095. m_denseIdx = UINT16_MAX;
  3096. return denseIdx;
  3097. }
  3098. void CommandQueueWgpu::init(wgpu::Queue _queue)
  3099. {
  3100. m_queue = _queue;
  3101. #if BGFX_CONFIG_MULTITHREADED
  3102. //m_framesSemaphore.post(WEBGPU_MAX_FRAMES_IN_FLIGHT);
  3103. #endif
  3104. }
  3105. void CommandQueueWgpu::shutdown()
  3106. {
  3107. finish(true);
  3108. }
  3109. void CommandQueueWgpu::beginRender()
  3110. {
  3111. m_renderEncoder = s_renderWgpu->m_device.CreateCommandEncoder();
  3112. }
  3113. void CommandQueueWgpu::beginStaging()
  3114. {
  3115. m_stagingEncoder = s_renderWgpu->m_device.CreateCommandEncoder();
  3116. }
  3117. inline void commandBufferFinishedCallback(void* _data)
  3118. {
  3119. #if BGFX_CONFIG_MULTITHREADED
  3120. CommandQueueWgpu* queue = (CommandQueueWgpu*)_data;
  3121. if (queue)
  3122. {
  3123. //queue->m_framesSemaphore.post();
  3124. }
  3125. #else
  3126. BX_UNUSED(_data);
  3127. #endif
  3128. }
  3129. void CommandQueueWgpu::kick(bool _endFrame, bool _waitForFinish)
  3130. {
  3131. if (m_renderEncoder)
  3132. {
  3133. if (_endFrame)
  3134. {
  3135. m_releaseWriteIndex = (m_releaseWriteIndex + 1) % WEBGPU_MAX_FRAMES_IN_FLIGHT;
  3136. //m_encoder.addCompletedHandler(commandBufferFinishedCallback, this);
  3137. }
  3138. if (m_stagingEncoder)
  3139. {
  3140. wgpu::CommandBuffer commands = m_stagingEncoder.Finish();
  3141. m_queue.Submit(1, &commands);
  3142. }
  3143. wgpu::CommandBuffer commands = m_renderEncoder.Finish();
  3144. m_queue.Submit(1, &commands);
  3145. if (_waitForFinish)
  3146. {
  3147. #if BGFX_CONFIG_MULTITHREADED
  3148. //m_framesSemaphore.post();
  3149. #endif
  3150. }
  3151. m_stagingEncoder = NULL;
  3152. m_renderEncoder = NULL;
  3153. }
  3154. }
  3155. void CommandQueueWgpu::finish(bool _finishAll)
  3156. {
  3157. if (_finishAll)
  3158. {
  3159. uint32_t count = m_renderEncoder
  3160. ? 2
  3161. : 3
  3162. ;
  3163. for (uint32_t ii = 0; ii < count; ++ii)
  3164. {
  3165. consume();
  3166. }
  3167. #if BGFX_CONFIG_MULTITHREADED
  3168. //m_framesSemaphore.post(count);
  3169. #endif
  3170. }
  3171. else
  3172. {
  3173. consume();
  3174. }
  3175. }
  3176. void CommandQueueWgpu::release(wgpu::Buffer _buffer)
  3177. {
  3178. m_release[m_releaseWriteIndex].push_back(_buffer);
  3179. }
  3180. void CommandQueueWgpu::consume()
  3181. {
  3182. #if BGFX_CONFIG_MULTITHREADED
  3183. //m_framesSemaphore.wait();
  3184. #endif
  3185. m_releaseReadIndex = (m_releaseReadIndex + 1) % WEBGPU_MAX_FRAMES_IN_FLIGHT;
  3186. for (wgpu::Buffer& buffer : m_release[m_releaseReadIndex])
  3187. {
  3188. buffer.Destroy();
  3189. }
  3190. m_release[m_releaseReadIndex].clear();
  3191. }
  3192. void TimerQueryWgpu::init()
  3193. {
  3194. m_frequency = bx::getHPFrequency();
  3195. }
  3196. void TimerQueryWgpu::shutdown()
  3197. {
  3198. }
  3199. uint32_t TimerQueryWgpu::begin(uint32_t _resultIdx)
  3200. {
  3201. BX_UNUSED(_resultIdx);
  3202. return 0;
  3203. }
  3204. void TimerQueryWgpu::end(uint32_t _idx)
  3205. {
  3206. BX_UNUSED(_idx);
  3207. }
  3208. #if 0
  3209. static void setTimestamp(void* _data)
  3210. {
  3211. *( (int64_t*)_data) = bx::getHPCounter();
  3212. }
  3213. #endif
  3214. void TimerQueryWgpu::addHandlers(wgpu::CommandBuffer& _commandBuffer)
  3215. {
  3216. BX_UNUSED(_commandBuffer);
  3217. while (0 == m_control.reserve(1) )
  3218. {
  3219. m_control.consume(1);
  3220. }
  3221. //uint32_t offset = m_control.m_current;
  3222. //_commandBuffer.addScheduledHandler(setTimestamp, &m_result[offset].m_begin);
  3223. //_commandBuffer.addCompletedHandler(setTimestamp, &m_result[offset].m_end);
  3224. m_control.commit(1);
  3225. }
  3226. bool TimerQueryWgpu::get()
  3227. {
  3228. if (0 != m_control.available() )
  3229. {
  3230. uint32_t offset = m_control.m_read;
  3231. m_begin = m_result[offset].m_begin;
  3232. m_end = m_result[offset].m_end;
  3233. m_elapsed = m_end - m_begin;
  3234. m_control.consume(1);
  3235. return true;
  3236. }
  3237. return false;
  3238. }
  3239. void RendererContextWgpu::submitBlit(BlitState& _bs, uint16_t _view)
  3240. {
  3241. if (!_bs.hasItem(_view) )
  3242. {
  3243. return;
  3244. }
  3245. endEncoding();
  3246. wgpu::CommandEncoder& bce = getBlitCommandEncoder();
  3247. while (_bs.hasItem(_view) )
  3248. {
  3249. const BlitItem& blit = _bs.advance();
  3250. const TextureWgpu& src = m_textures[blit.m_src.idx];
  3251. const TextureWgpu& dst = m_textures[blit.m_dst.idx];
  3252. uint32_t srcWidth = bx::uint32_min(src.m_width, blit.m_srcX + blit.m_width) - blit.m_srcX;
  3253. uint32_t srcHeight = bx::uint32_min(src.m_height, blit.m_srcY + blit.m_height) - blit.m_srcY;
  3254. uint32_t srcDepth = bx::uint32_min(src.m_depth, blit.m_srcZ + blit.m_depth) - blit.m_srcZ;
  3255. uint32_t dstWidth = bx::uint32_min(dst.m_width, blit.m_dstX + blit.m_width) - blit.m_dstX;
  3256. uint32_t dstHeight = bx::uint32_min(dst.m_height, blit.m_dstY + blit.m_height) - blit.m_dstY;
  3257. uint32_t dstDepth = bx::uint32_min(dst.m_depth, blit.m_dstZ + blit.m_depth) - blit.m_dstZ;
  3258. uint32_t width = bx::uint32_min(srcWidth, dstWidth);
  3259. uint32_t height = bx::uint32_min(srcHeight, dstHeight);
  3260. uint32_t depth = bx::uint32_min(srcDepth, dstDepth);
  3261. bool readBack = !!(dst.m_flags & BGFX_TEXTURE_READ_BACK);
  3262. wgpu::TextureCopyView srcView;
  3263. srcView.texture = src.m_ptr;
  3264. srcView.origin = { blit.m_srcX, blit.m_srcY, 0 };
  3265. srcView.mipLevel = blit.m_srcMip;
  3266. srcView.arrayLayer = blit.m_srcZ;
  3267. wgpu::TextureCopyView dstView;
  3268. dstView.texture = dst.m_ptr;
  3269. dstView.origin = { blit.m_dstX, blit.m_dstY, 0 };
  3270. dstView.mipLevel = blit.m_dstMip;
  3271. dstView.arrayLayer = blit.m_dstZ;
  3272. if (depth == 0)
  3273. {
  3274. wgpu::Extent3D copyExtent = { width, height, 1 };
  3275. bce.CopyTextureToTexture(&srcView, &dstView, &copyExtent);
  3276. }
  3277. else
  3278. {
  3279. wgpu::Extent3D copyExtent = { width, height, depth };
  3280. bce.CopyTextureToTexture(&srcView, &dstView, &copyExtent);
  3281. }
  3282. if (readBack)
  3283. {
  3284. //bce..synchronizeTexture(dst.m_ptr, 0, blit.m_dstMip);
  3285. }
  3286. }
  3287. //if (bce)
  3288. //{
  3289. // bce.endEncoding();
  3290. // bce = 0;
  3291. //}
  3292. }
  3293. void RendererContextWgpu::submit(Frame* _render, ClearQuad& _clearQuad, TextVideoMemBlitter& _textVideoMemBlitter)
  3294. {
  3295. if(_render->m_capture)
  3296. {
  3297. renderDocTriggerCapture();
  3298. }
  3299. m_cmd.finish(false);
  3300. if (!m_cmd.m_renderEncoder)
  3301. {
  3302. m_cmd.beginRender();
  3303. }
  3304. BGFX_WEBGPU_PROFILER_BEGIN_LITERAL("rendererSubmit", kColorFrame);
  3305. int64_t timeBegin = bx::getHPCounter();
  3306. int64_t captureElapsed = 0;
  3307. //m_gpuTimer.addHandlers(m_encoder);
  3308. updateResolution(_render->m_resolution);
  3309. m_frameIndex = 0; // (m_frameIndex + 1) % WEBGPU_MAX_FRAMES_IN_FLIGHT;
  3310. ScratchBufferWgpu& scratchBuffer = m_scratchBuffers[m_frameIndex];
  3311. scratchBuffer.begin();
  3312. BindStateCacheWgpu& bindStates = m_bindStateCache[m_frameIndex];
  3313. bindStates.reset();
  3314. if (0 < _render->m_iboffset)
  3315. {
  3316. BGFX_PROFILER_SCOPE("bgfx/Update transient index buffer", kColorResource);
  3317. TransientIndexBuffer* ib = _render->m_transientIb;
  3318. m_indexBuffers[ib->handle.idx].update(0, bx::strideAlign(_render->m_iboffset,4), ib->data, true);
  3319. }
  3320. if (0 < _render->m_vboffset)
  3321. {
  3322. BGFX_PROFILER_SCOPE("bgfx/Update transient vertex buffer", kColorResource);
  3323. TransientVertexBuffer* vb = _render->m_transientVb;
  3324. m_vertexBuffers[vb->handle.idx].update(0, bx::strideAlign(_render->m_vboffset,4), vb->data, true);
  3325. }
  3326. _render->sort();
  3327. RenderDraw currentState;
  3328. currentState.clear();
  3329. currentState.m_stateFlags = BGFX_STATE_NONE;
  3330. currentState.m_stencil = packStencil(BGFX_STENCIL_NONE, BGFX_STENCIL_NONE);
  3331. RenderBind currentBind;
  3332. currentBind.clear();
  3333. static ViewState viewState;
  3334. viewState.reset(_render);
  3335. uint32_t blendFactor = 0;
  3336. //bool wireframe = !!(_render->m_debug&BGFX_DEBUG_WIREFRAME);
  3337. ProgramHandle currentProgram = BGFX_INVALID_HANDLE;
  3338. uint32_t currentBindHash = 0;
  3339. uint32_t currentBindLayoutHash = 0;
  3340. BindStateWgpu* previousBindState = NULL;
  3341. SortKey key;
  3342. uint16_t view = UINT16_MAX;
  3343. FrameBufferHandle fbh = { BGFX_CONFIG_MAX_FRAME_BUFFERS };
  3344. BlitState bs(_render);
  3345. const uint64_t primType = 0;
  3346. uint8_t primIndex = uint8_t(primType >> BGFX_STATE_PT_SHIFT);
  3347. PrimInfo prim = s_primInfo[primIndex];
  3348. const uint32_t maxComputeBindings = g_caps.limits.maxComputeBindings;
  3349. // TODO store this
  3350. static wgpu::RenderPassEncoder rce;
  3351. PipelineStateWgpu* currentPso = NULL;
  3352. bool wasCompute = false;
  3353. bool viewHasScissor = false;
  3354. Rect viewScissorRect;
  3355. viewScissorRect.clear();
  3356. uint32_t statsNumPrimsSubmitted[BX_COUNTOF(s_primInfo)] = {};
  3357. uint32_t statsNumPrimsRendered[BX_COUNTOF(s_primInfo)] = {};
  3358. uint32_t statsNumInstances[BX_COUNTOF(s_primInfo)] = {};
  3359. uint32_t statsNumDrawIndirect[BX_COUNTOF(s_primInfo)] = {};
  3360. uint32_t statsNumIndices = 0;
  3361. uint32_t statsKeyType[2] = {};
  3362. Profiler<TimerQueryWgpu> profiler(
  3363. _render
  3364. , m_gpuTimer
  3365. , s_viewName
  3366. );
  3367. if (0 == (_render->m_debug & BGFX_DEBUG_IFH))
  3368. {
  3369. viewState.m_rect = _render->m_view[0].m_rect;
  3370. int32_t numItems = _render->m_numRenderItems;
  3371. for (int32_t item = 0; item < numItems;)
  3372. {
  3373. const uint64_t encodedKey = _render->m_sortKeys[item];
  3374. const bool isCompute = key.decode(encodedKey, _render->m_viewRemap);
  3375. statsKeyType[isCompute]++;
  3376. const bool viewChanged = 0
  3377. || key.m_view != view
  3378. || item == numItems
  3379. ;
  3380. const uint32_t itemIdx = _render->m_sortValues[item];
  3381. const RenderItem& renderItem = _render->m_renderItem[itemIdx];
  3382. const RenderBind& renderBind = _render->m_renderItemBind[itemIdx];
  3383. ++item;
  3384. if (viewChanged
  3385. || (!isCompute && wasCompute))
  3386. {
  3387. view = key.m_view;
  3388. currentProgram = BGFX_INVALID_HANDLE;
  3389. if (item > 1)
  3390. {
  3391. profiler.end();
  3392. }
  3393. BGFX_WEBGPU_PROFILER_END();
  3394. setViewType(view, " ");
  3395. BGFX_WEBGPU_PROFILER_BEGIN(view, kColorView);
  3396. profiler.begin(view);
  3397. viewState.m_rect = _render->m_view[view].m_rect;
  3398. submitBlit(bs, view);
  3399. if (!isCompute)
  3400. {
  3401. const Rect& scissorRect = _render->m_view[view].m_scissor;
  3402. viewHasScissor = !scissorRect.isZero();
  3403. viewScissorRect = viewHasScissor ? scissorRect : viewState.m_rect;
  3404. Clear& clr = _render->m_view[view].m_clear;
  3405. Rect viewRect = viewState.m_rect;
  3406. bool clearWithRenderPass = false;
  3407. if (!m_renderEncoder
  3408. || fbh.idx != _render->m_view[view].m_fbh.idx)
  3409. {
  3410. endEncoding();
  3411. fbh = _render->m_view[view].m_fbh;
  3412. uint32_t width = m_resolution.width;
  3413. uint32_t height = m_resolution.height;
  3414. if (isValid(fbh))
  3415. {
  3416. FrameBufferWgpu& frameBuffer = m_frameBuffers[fbh.idx];
  3417. width = frameBuffer.m_width;
  3418. height = frameBuffer.m_height;
  3419. }
  3420. clearWithRenderPass = true
  3421. && 0 == viewRect.m_x
  3422. && 0 == viewRect.m_y
  3423. && width == viewRect.m_width
  3424. && height == viewRect.m_height
  3425. ;
  3426. rce = renderPass(_render, fbh, clearWithRenderPass, clr, s_viewName[view]);
  3427. }
  3428. else if (BX_ENABLED(BGFX_CONFIG_DEBUG_ANNOTATION))
  3429. {
  3430. rce.PopDebugGroup();
  3431. }
  3432. if (BX_ENABLED(BGFX_CONFIG_DEBUG_ANNOTATION))
  3433. {
  3434. rce.PushDebugGroup(s_viewName[view]);
  3435. }
  3436. //rce.setTriangleFillMode(wireframe ? MTLTriangleFillModeLines : MTLTriangleFillModeFill);
  3437. const Rect& rect = viewState.m_rect;
  3438. rce.SetViewport(rect.m_x, rect.m_y, rect.m_width, rect.m_height, 0.0f, 1.0f);
  3439. rce.SetScissorRect(rect.m_x, rect.m_y, rect.m_width, rect.m_height);
  3440. if (BGFX_CLEAR_NONE != (clr.m_flags & BGFX_CLEAR_MASK)
  3441. && !clearWithRenderPass)
  3442. {
  3443. clearQuad(_clearQuad, viewState.m_rect, clr, _render->m_colorPalette);
  3444. }
  3445. }
  3446. }
  3447. if (isCompute)
  3448. {
  3449. if (!wasCompute)
  3450. {
  3451. wasCompute = true;
  3452. endEncoding();
  3453. rce = NULL;
  3454. setViewType(view, "C");
  3455. BGFX_WEBGPU_PROFILER_END();
  3456. BGFX_WEBGPU_PROFILER_BEGIN(view, kColorCompute);
  3457. m_computeEncoder = m_cmd.m_renderEncoder.BeginComputePass();
  3458. }
  3459. else if (viewChanged)
  3460. {
  3461. if (BX_ENABLED(BGFX_CONFIG_DEBUG_ANNOTATION))
  3462. {
  3463. m_computeEncoder.PopDebugGroup();
  3464. }
  3465. endEncoding();
  3466. m_computeEncoder = m_cmd.m_renderEncoder.BeginComputePass();
  3467. }
  3468. if (viewChanged)
  3469. {
  3470. if (BX_ENABLED(BGFX_CONFIG_DEBUG_ANNOTATION))
  3471. {
  3472. s_viewName[view][3] = L'C';
  3473. m_computeEncoder.PushDebugGroup(s_viewName[view]);
  3474. s_viewName[view][3] = L' ';
  3475. }
  3476. }
  3477. const RenderCompute& compute = renderItem.compute;
  3478. bool programChanged = false;
  3479. bool constantsChanged = compute.m_uniformBegin < compute.m_uniformEnd;
  3480. rendererUpdateUniforms(this, _render->m_uniformBuffer[compute.m_uniformIdx], compute.m_uniformBegin, compute.m_uniformEnd);
  3481. if (key.m_program.idx != currentProgram.idx)
  3482. {
  3483. currentProgram = key.m_program;
  3484. currentPso = getComputePipelineState(currentProgram);
  3485. if (NULL == currentPso)
  3486. {
  3487. currentProgram = BGFX_INVALID_HANDLE;
  3488. continue;
  3489. }
  3490. m_computeEncoder.SetPipeline(currentPso->m_cps);
  3491. programChanged =
  3492. constantsChanged = true;
  3493. }
  3494. if (!isValid(currentProgram)
  3495. || NULL == currentPso)
  3496. BX_WARN(false, "Invalid program / No PSO");
  3497. const ProgramWgpu& program = m_program[currentProgram.idx];
  3498. if (constantsChanged)
  3499. {
  3500. UniformBuffer* vcb = program.m_vsh->m_constantBuffer;
  3501. if (NULL != vcb)
  3502. {
  3503. commit(*vcb);
  3504. }
  3505. }
  3506. viewState.setPredefined<4>(this, view, program, _render, compute);
  3507. uint32_t numOffset = 1;
  3508. uint32_t offsets[2] = { scratchBuffer.m_offset, 0 };
  3509. if (program.m_vsh->m_size > 0)
  3510. {
  3511. scratchBuffer.write(m_vsScratch, program.m_vsh->m_gpuSize);
  3512. }
  3513. BindStateWgpu& bindState = allocAndFillBindState(program, bindStates, scratchBuffer, renderBind);
  3514. bindProgram(m_computeEncoder, program, bindState, numOffset, offsets);
  3515. if (isValid(compute.m_indirectBuffer))
  3516. {
  3517. const VertexBufferWgpu& vb = m_vertexBuffers[compute.m_indirectBuffer.idx];
  3518. uint32_t numDrawIndirect = UINT16_MAX == compute.m_numIndirect
  3519. ? vb.m_size/BGFX_CONFIG_DRAW_INDIRECT_STRIDE
  3520. : compute.m_numIndirect
  3521. ;
  3522. uint32_t args = compute.m_startIndirect * BGFX_CONFIG_DRAW_INDIRECT_STRIDE;
  3523. for (uint32_t ii = 0; ii < numDrawIndirect; ++ii)
  3524. {
  3525. m_computeEncoder.DispatchIndirect(
  3526. vb.m_ptr
  3527. , args
  3528. );
  3529. args += BGFX_CONFIG_DRAW_INDIRECT_STRIDE;
  3530. }
  3531. }
  3532. else
  3533. {
  3534. m_computeEncoder.Dispatch(compute.m_numX, compute.m_numY, compute.m_numZ);
  3535. }
  3536. continue;
  3537. }
  3538. bool resetState = viewChanged || wasCompute;
  3539. if (wasCompute)
  3540. {
  3541. wasCompute = false;
  3542. currentProgram = BGFX_INVALID_HANDLE;
  3543. setViewType(view, " ");
  3544. BGFX_WEBGPU_PROFILER_END();
  3545. BGFX_WEBGPU_PROFILER_BEGIN(view, kColorDraw);
  3546. }
  3547. const RenderDraw& draw = renderItem.draw;
  3548. // TODO (hugoam)
  3549. //const bool depthWrite = !!(BGFX_STATE_WRITE_Z & draw.m_stateFlags);
  3550. const uint64_t newFlags = draw.m_stateFlags;
  3551. uint64_t changedFlags = currentState.m_stateFlags ^ draw.m_stateFlags;
  3552. currentState.m_stateFlags = newFlags;
  3553. const uint64_t newStencil = draw.m_stencil;
  3554. uint64_t changedStencil = (currentState.m_stencil ^ draw.m_stencil) & BGFX_STENCIL_FUNC_REF_MASK;
  3555. currentState.m_stencil = newStencil;
  3556. if (resetState)
  3557. {
  3558. wasCompute = false;
  3559. currentState.clear();
  3560. currentState.m_scissor = !draw.m_scissor;
  3561. changedFlags = BGFX_STATE_MASK;
  3562. changedStencil = packStencil(BGFX_STENCIL_MASK, BGFX_STENCIL_MASK);
  3563. currentState.m_stateFlags = newFlags;
  3564. currentState.m_stencil = newStencil;
  3565. currentBind.clear();
  3566. currentProgram = BGFX_INVALID_HANDLE;
  3567. const uint64_t pt = newFlags & BGFX_STATE_PT_MASK;
  3568. primIndex = uint8_t(pt >> BGFX_STATE_PT_SHIFT);
  3569. }
  3570. if (prim.m_type != s_primInfo[primIndex].m_type)
  3571. {
  3572. prim = s_primInfo[primIndex];
  3573. }
  3574. uint16_t scissor = draw.m_scissor;
  3575. if (currentState.m_scissor != scissor)
  3576. {
  3577. currentState.m_scissor = scissor;
  3578. if (UINT16_MAX == scissor)
  3579. {
  3580. if (viewHasScissor)
  3581. {
  3582. const auto& r = viewScissorRect;
  3583. rce.SetScissorRect(r.m_x, r.m_y, r.m_width, r.m_height);
  3584. }
  3585. else
  3586. { // can't disable: set to view rect
  3587. const auto& r = viewState.m_rect;
  3588. rce.SetScissorRect(r.m_x, r.m_y, r.m_width, r.m_height);
  3589. }
  3590. }
  3591. else
  3592. {
  3593. Rect scissorRect;
  3594. scissorRect.setIntersect(viewScissorRect, _render->m_frameCache.m_rectCache.m_cache[scissor]);
  3595. const auto& r = scissorRect;
  3596. if (r.m_width == 0 || r.m_height == 0)
  3597. {
  3598. continue;
  3599. }
  3600. rce.SetScissorRect(r.m_x, r.m_y, r.m_width, r.m_height);
  3601. }
  3602. }
  3603. if (0 != changedStencil)
  3604. {
  3605. const uint32_t fstencil = unpackStencil(0, draw.m_stencil);
  3606. const uint32_t ref = (fstencil & BGFX_STENCIL_FUNC_REF_MASK) >> BGFX_STENCIL_FUNC_REF_SHIFT;
  3607. rce.SetStencilReference(ref);
  3608. }
  3609. if ((0 | BGFX_STATE_PT_MASK) & changedFlags)
  3610. {
  3611. const uint64_t pt = newFlags & BGFX_STATE_PT_MASK;
  3612. primIndex = uint8_t(pt >> BGFX_STATE_PT_SHIFT);
  3613. if (prim.m_type != s_primInfo[primIndex].m_type)
  3614. {
  3615. prim = s_primInfo[primIndex];
  3616. }
  3617. }
  3618. if (blendFactor != draw.m_rgba
  3619. && !(newFlags & BGFX_STATE_BLEND_INDEPENDENT))
  3620. {
  3621. const uint32_t rgba = draw.m_rgba;
  3622. float rr = ((rgba >> 24)) / 255.0f;
  3623. float gg = ((rgba >> 16) & 0xff) / 255.0f;
  3624. float bb = ((rgba >> 8) & 0xff) / 255.0f;
  3625. float aa = ((rgba) & 0xff) / 255.0f;
  3626. wgpu::Color color = { rr, gg, bb, aa };
  3627. rce.SetBlendColor(&color);
  3628. blendFactor = draw.m_rgba;
  3629. }
  3630. bool programChanged = false;
  3631. bool constantsChanged = draw.m_uniformBegin < draw.m_uniformEnd;
  3632. rendererUpdateUniforms(this, _render->m_uniformBuffer[draw.m_uniformIdx], draw.m_uniformBegin, draw.m_uniformEnd);
  3633. bool vertexStreamChanged = hasVertexStreamChanged(currentState, draw);
  3634. if (key.m_program.idx != currentProgram.idx
  3635. || vertexStreamChanged
  3636. || (0
  3637. | BGFX_STATE_BLEND_MASK
  3638. | BGFX_STATE_BLEND_EQUATION_MASK
  3639. | BGFX_STATE_WRITE_RGB
  3640. | BGFX_STATE_WRITE_A
  3641. | BGFX_STATE_BLEND_INDEPENDENT
  3642. | BGFX_STATE_MSAA
  3643. | BGFX_STATE_BLEND_ALPHA_TO_COVERAGE
  3644. ) & changedFlags
  3645. || ((blendFactor != draw.m_rgba) && !!(newFlags & BGFX_STATE_BLEND_INDEPENDENT)))
  3646. {
  3647. currentProgram = key.m_program;
  3648. currentState.m_streamMask = draw.m_streamMask;
  3649. currentState.m_instanceDataBuffer.idx = draw.m_instanceDataBuffer.idx;
  3650. currentState.m_instanceDataOffset = draw.m_instanceDataOffset;
  3651. currentState.m_instanceDataStride = draw.m_instanceDataStride;
  3652. const VertexLayout* decls[BGFX_CONFIG_MAX_VERTEX_STREAMS];
  3653. uint32_t numVertices = draw.m_numVertices;
  3654. uint8_t numStreams = 0;
  3655. for (uint32_t idx = 0, streamMask = draw.m_streamMask
  3656. ; 0 != streamMask
  3657. ; streamMask >>= 1, idx += 1, ++numStreams
  3658. )
  3659. {
  3660. const uint32_t ntz = bx::uint32_cnttz(streamMask);
  3661. streamMask >>= ntz;
  3662. idx += ntz;
  3663. currentState.m_stream[idx].m_layoutHandle = draw.m_stream[idx].m_layoutHandle;
  3664. currentState.m_stream[idx].m_handle = draw.m_stream[idx].m_handle;
  3665. currentState.m_stream[idx].m_startVertex = draw.m_stream[idx].m_startVertex;
  3666. const uint16_t handle = draw.m_stream[idx].m_handle.idx;
  3667. const VertexBufferWgpu& vb = m_vertexBuffers[handle];
  3668. const uint16_t decl = isValid(draw.m_stream[idx].m_layoutHandle)
  3669. ? draw.m_stream[idx].m_layoutHandle.idx
  3670. : vb.m_layoutHandle.idx;
  3671. const VertexLayout& vertexDecl = m_vertexDecls[decl];
  3672. const uint32_t stride = vertexDecl.m_stride;
  3673. decls[numStreams] = &vertexDecl;
  3674. numVertices = bx::uint32_min(UINT32_MAX == draw.m_numVertices
  3675. ? vb.m_size / stride
  3676. : draw.m_numVertices
  3677. , numVertices
  3678. );
  3679. const uint32_t offset = draw.m_stream[idx].m_startVertex * stride;
  3680. rce.SetVertexBuffer(idx, vb.m_ptr, offset);
  3681. }
  3682. bool index32 = false;
  3683. if (isValid(draw.m_indexBuffer))
  3684. {
  3685. const IndexBufferWgpu& ib = m_indexBuffers[draw.m_indexBuffer.idx];
  3686. index32 = 0 != (ib.m_flags & BGFX_BUFFER_INDEX32);
  3687. }
  3688. currentState.m_numVertices = numVertices;
  3689. if (!isValid(currentProgram))
  3690. {
  3691. continue;
  3692. }
  3693. else
  3694. {
  3695. currentPso = NULL;
  3696. if (0 < numStreams)
  3697. {
  3698. currentPso = getPipelineState(
  3699. newFlags
  3700. , newStencil
  3701. , draw.m_rgba
  3702. , fbh
  3703. , numStreams
  3704. , decls
  3705. , index32
  3706. , currentProgram
  3707. , uint8_t(draw.m_instanceDataStride / 16)
  3708. );
  3709. }
  3710. if (NULL == currentPso)
  3711. {
  3712. currentProgram = BGFX_INVALID_HANDLE;
  3713. continue;
  3714. }
  3715. rce.SetPipeline(currentPso->m_rps);
  3716. }
  3717. if (isValid(draw.m_instanceDataBuffer))
  3718. {
  3719. const VertexBufferWgpu& inst = m_vertexBuffers[draw.m_instanceDataBuffer.idx];
  3720. rce.SetVertexBuffer(numStreams/*+1*/, inst.m_ptr, draw.m_instanceDataOffset);
  3721. }
  3722. programChanged =
  3723. constantsChanged = true;
  3724. }
  3725. if (isValid(currentProgram))
  3726. {
  3727. const ProgramWgpu& program = m_program[currentProgram.idx];
  3728. if (constantsChanged)
  3729. {
  3730. UniformBuffer* vcb = program.m_vsh->m_constantBuffer;
  3731. if (NULL != vcb)
  3732. {
  3733. commit(*vcb);
  3734. }
  3735. }
  3736. if (constantsChanged)
  3737. {
  3738. UniformBuffer* fcb = program.m_fsh->m_constantBuffer;
  3739. if (NULL != fcb)
  3740. {
  3741. commit(*fcb);
  3742. }
  3743. }
  3744. viewState.setPredefined<4>(this, view, program, _render, draw);
  3745. bool hasPredefined = 0 < program.m_numPredefined;
  3746. uint32_t numOffset = 0;
  3747. uint32_t offsets[2] = { 0, 0 };
  3748. if (constantsChanged
  3749. || hasPredefined)
  3750. {
  3751. //viewState.setPredefined<4>(this, view, program, _render, draw, programChanged || viewChanged);
  3752. //commitShaderConstants(scratchBuffer, program, voffset, foffset);
  3753. const uint32_t vsize = program.m_vsh->m_gpuSize;
  3754. const uint32_t fsize = (NULL != program.m_fsh ? program.m_fsh->m_gpuSize : 0);
  3755. if (program.m_vsh->m_size > 0)
  3756. {
  3757. offsets[numOffset++] = scratchBuffer.write(m_vsScratch, vsize);
  3758. }
  3759. if (fsize > 0)
  3760. {
  3761. offsets[numOffset++] = scratchBuffer.write(m_fsScratch, fsize);
  3762. }
  3763. }
  3764. uint32_t bindHash = bx::hash<bx::HashMurmur2A>(renderBind.m_bind, sizeof(renderBind.m_bind));
  3765. if (currentBindHash != bindHash
  3766. || currentBindLayoutHash != program.m_bindGroupLayoutHash)
  3767. {
  3768. currentBindHash = bindHash;
  3769. currentBindLayoutHash = program.m_bindGroupLayoutHash;
  3770. previousBindState = &bindStates.m_bindStates[bindStates.m_currentBindState];
  3771. allocAndFillBindState(program, bindStates, scratchBuffer, renderBind);
  3772. }
  3773. BindStateWgpu& bindState = bindStates.m_bindStates[bindStates.m_currentBindState-1];
  3774. bindProgram(rce, program, bindState, numOffset, offsets);
  3775. }
  3776. if (0 != currentState.m_streamMask)
  3777. {
  3778. uint32_t numVertices = draw.m_numVertices;
  3779. if (UINT32_MAX == numVertices)
  3780. {
  3781. const VertexBufferWgpu& vb = m_vertexBuffers[currentState.m_stream[0].m_handle.idx];
  3782. uint16_t decl = !isValid(vb.m_layoutHandle) ? draw.m_stream[0].m_layoutHandle.idx : vb.m_layoutHandle.idx;
  3783. const VertexLayout& vertexDecl = m_vertexDecls[decl];
  3784. numVertices = vb.m_size/vertexDecl.m_stride;
  3785. }
  3786. uint32_t numIndices = 0;
  3787. uint32_t numPrimsSubmitted = 0;
  3788. uint32_t numInstances = 0;
  3789. uint32_t numPrimsRendered = 0;
  3790. uint32_t numDrawIndirect = 0;
  3791. if (isValid(draw.m_indirectBuffer) )
  3792. {
  3793. const VertexBufferWgpu& vb = m_vertexBuffers[draw.m_indirectBuffer.idx];
  3794. if (isValid(draw.m_indexBuffer) )
  3795. {
  3796. const IndexBufferWgpu& ib = m_indexBuffers[draw.m_indexBuffer.idx];
  3797. numDrawIndirect = UINT16_MAX == draw.m_numIndirect
  3798. ? vb.m_size/BGFX_CONFIG_DRAW_INDIRECT_STRIDE
  3799. : draw.m_numIndirect
  3800. ;
  3801. for (uint32_t ii = 0; ii < numDrawIndirect; ++ii)
  3802. {
  3803. rce.SetIndexBuffer(ib.m_ptr, 0);
  3804. rce.DrawIndexedIndirect(vb.m_ptr, (draw.m_startIndirect + ii)* BGFX_CONFIG_DRAW_INDIRECT_STRIDE);
  3805. }
  3806. }
  3807. else
  3808. {
  3809. numDrawIndirect = UINT16_MAX == draw.m_numIndirect
  3810. ? vb.m_size/BGFX_CONFIG_DRAW_INDIRECT_STRIDE
  3811. : draw.m_numIndirect
  3812. ;
  3813. for (uint32_t ii = 0; ii < numDrawIndirect; ++ii)
  3814. {
  3815. rce.DrawIndirect(vb.m_ptr, (draw.m_startIndirect + ii)* BGFX_CONFIG_DRAW_INDIRECT_STRIDE);
  3816. }
  3817. }
  3818. }
  3819. else
  3820. {
  3821. if (isValid(draw.m_indexBuffer) )
  3822. {
  3823. const IndexBufferWgpu& ib = m_indexBuffers[draw.m_indexBuffer.idx];
  3824. const uint32_t indexSize = 0 == (ib.m_flags & BGFX_BUFFER_INDEX32) ? 2 : 4;
  3825. if (UINT32_MAX == draw.m_numIndices)
  3826. {
  3827. numIndices = ib.m_size/indexSize;
  3828. numPrimsSubmitted = numIndices/prim.m_div - prim.m_sub;
  3829. numInstances = draw.m_numInstances;
  3830. numPrimsRendered = numPrimsSubmitted*draw.m_numInstances;
  3831. rce.SetIndexBuffer(ib.m_ptr, 0);
  3832. rce.DrawIndexed(numIndices, draw.m_numInstances, 0, 0, 0);
  3833. }
  3834. else if (prim.m_min <= draw.m_numIndices)
  3835. {
  3836. numIndices = draw.m_numIndices;
  3837. numPrimsSubmitted = numIndices/prim.m_div - prim.m_sub;
  3838. numInstances = draw.m_numInstances;
  3839. numPrimsRendered = numPrimsSubmitted*draw.m_numInstances;
  3840. rce.SetIndexBuffer(ib.m_ptr, 0);
  3841. rce.DrawIndexed(numIndices, numInstances, draw.m_startIndex, 0, 0);
  3842. }
  3843. }
  3844. else
  3845. {
  3846. numPrimsSubmitted = numVertices/prim.m_div - prim.m_sub;
  3847. numInstances = draw.m_numInstances;
  3848. numPrimsRendered = numPrimsSubmitted*draw.m_numInstances;
  3849. rce.Draw(numVertices, draw.m_numInstances, 0, 0);
  3850. }
  3851. }
  3852. statsNumPrimsSubmitted[primIndex] += numPrimsSubmitted;
  3853. statsNumPrimsRendered[primIndex] += numPrimsRendered;
  3854. statsNumInstances[primIndex] += numInstances;
  3855. statsNumDrawIndirect[primIndex] += numDrawIndirect;
  3856. statsNumIndices += numIndices;
  3857. }
  3858. }
  3859. if (wasCompute)
  3860. {
  3861. invalidateCompute();
  3862. setViewType(view, "C");
  3863. BGFX_WEBGPU_PROFILER_END();
  3864. BGFX_WEBGPU_PROFILER_BEGIN(view, kColorCompute);
  3865. }
  3866. submitBlit(bs, BGFX_CONFIG_MAX_VIEWS);
  3867. if (0 < _render->m_numRenderItems)
  3868. {
  3869. captureElapsed = -bx::getHPCounter();
  3870. capture();
  3871. rce = m_renderEncoder;
  3872. captureElapsed += bx::getHPCounter();
  3873. profiler.end();
  3874. }
  3875. }
  3876. if (BX_ENABLED(BGFX_CONFIG_DEBUG_ANNOTATION) )
  3877. {
  3878. if (0 < _render->m_numRenderItems)
  3879. {
  3880. rce.PopDebugGroup();
  3881. }
  3882. }
  3883. BGFX_WEBGPU_PROFILER_END();
  3884. int64_t timeEnd = bx::getHPCounter();
  3885. int64_t frameTime = timeEnd - timeBegin;
  3886. static int64_t min = frameTime;
  3887. static int64_t max = frameTime;
  3888. min = bx::min<int64_t>(min, frameTime);
  3889. max = bx::max<int64_t>(max, frameTime);
  3890. static uint32_t maxGpuLatency = 0;
  3891. static double maxGpuElapsed = 0.0f;
  3892. double elapsedGpuMs = 0.0;
  3893. do
  3894. {
  3895. double toGpuMs = 1000.0 / double(m_gpuTimer.m_frequency);
  3896. elapsedGpuMs = m_gpuTimer.m_elapsed * toGpuMs;
  3897. maxGpuElapsed = elapsedGpuMs > maxGpuElapsed ? elapsedGpuMs : maxGpuElapsed;
  3898. }
  3899. while (m_gpuTimer.get() );
  3900. maxGpuLatency = bx::uint32_imax(maxGpuLatency, m_gpuTimer.m_control.available()-1);
  3901. const int64_t timerFreq = bx::getHPFrequency();
  3902. Stats& perfStats = _render->m_perfStats;
  3903. perfStats.cpuTimeBegin = timeBegin;
  3904. perfStats.cpuTimeEnd = timeEnd;
  3905. perfStats.cpuTimerFreq = timerFreq;
  3906. perfStats.gpuTimeBegin = m_gpuTimer.m_begin;
  3907. perfStats.gpuTimeEnd = m_gpuTimer.m_end;
  3908. perfStats.gpuTimerFreq = m_gpuTimer.m_frequency;
  3909. perfStats.numDraw = statsKeyType[0];
  3910. perfStats.numCompute = statsKeyType[1];
  3911. perfStats.numBlit = _render->m_numBlitItems;
  3912. perfStats.maxGpuLatency = maxGpuLatency;
  3913. bx::memCopy(perfStats.numPrims, statsNumPrimsRendered, sizeof(perfStats.numPrims) );
  3914. perfStats.gpuMemoryMax = -INT64_MAX;
  3915. perfStats.gpuMemoryUsed = -INT64_MAX;
  3916. //rce.setTriangleFillMode(MTLTriangleFillModeFill);
  3917. if (_render->m_debug & (BGFX_DEBUG_IFH|BGFX_DEBUG_STATS) )
  3918. {
  3919. rce = renderPass(_render, BGFX_INVALID_HANDLE, false, Clear());
  3920. if (BX_ENABLED(BGFX_CONFIG_DEBUG_ANNOTATION))
  3921. {
  3922. rce.PushDebugGroup("debugstats");
  3923. }
  3924. TextVideoMem& tvm = m_textVideoMem;
  3925. static int64_t next = timeEnd;
  3926. if (timeEnd >= next)
  3927. {
  3928. next = timeEnd + timerFreq;
  3929. double freq = double(timerFreq);
  3930. double toMs = 1000.0/freq;
  3931. tvm.clear();
  3932. uint16_t pos = 0;
  3933. tvm.printf(0, pos++, BGFX_CONFIG_DEBUG ? 0x8c : 0x8f
  3934. , " %s / " BX_COMPILER_NAME " / " BX_CPU_NAME " / " BX_ARCH_NAME " / " BX_PLATFORM_NAME " "
  3935. , getRendererName()
  3936. );
  3937. pos = 10;
  3938. tvm.printf(10, pos++, 0x8b, " Frame: %7.3f, % 7.3f \x1f, % 7.3f \x1e [ms] / % 6.2f FPS "
  3939. , double(frameTime)*toMs
  3940. , double(min)*toMs
  3941. , double(max)*toMs
  3942. , freq/frameTime
  3943. );
  3944. const uint32_t msaa = (m_resolution.reset&BGFX_RESET_MSAA_MASK)>>BGFX_RESET_MSAA_SHIFT;
  3945. tvm.printf(10, pos++, 0x8b, " Reset flags: [%c] vsync, [%c] MSAAx%d, [%c] MaxAnisotropy "
  3946. , !!(m_resolution.reset&BGFX_RESET_VSYNC) ? '\xfe' : ' '
  3947. , 0 != msaa ? '\xfe' : ' '
  3948. , 1<<msaa
  3949. , !!(m_resolution.reset&BGFX_RESET_MAXANISOTROPY) ? '\xfe' : ' '
  3950. );
  3951. double elapsedCpuMs = double(frameTime)*toMs;
  3952. tvm.printf(10, pos++, 0x8b, " Submitted: %4d (draw %4d, compute %4d) / CPU %3.4f [ms] %c GPU %3.4f [ms] (latency %d)"
  3953. , _render->m_numRenderItems
  3954. , statsKeyType[0]
  3955. , statsKeyType[1]
  3956. , elapsedCpuMs
  3957. , elapsedCpuMs > maxGpuElapsed ? '>' : '<'
  3958. , maxGpuElapsed
  3959. , maxGpuLatency
  3960. );
  3961. maxGpuLatency = 0;
  3962. maxGpuElapsed = 0.0;
  3963. for (uint32_t ii = 0; ii < Topology::Count; ++ii)
  3964. {
  3965. tvm.printf(10, pos++, 0x8b, " %10s: %7d (#inst: %5d), submitted: %7d"
  3966. , getName(Topology::Enum(ii) )
  3967. , statsNumPrimsRendered[ii]
  3968. , statsNumInstances[ii]
  3969. , statsNumPrimsSubmitted[ii]
  3970. );
  3971. }
  3972. tvm.printf(10, pos++, 0x8b, " Indices: %7d ", statsNumIndices);
  3973. // tvm.printf(10, pos++, 0x8b, " Uniform size: %7d, Max: %7d ", _render->m_uniformEnd, _render->m_uniformMax);
  3974. tvm.printf(10, pos++, 0x8b, " DVB size: %7d ", _render->m_vboffset);
  3975. tvm.printf(10, pos++, 0x8b, " DIB size: %7d ", _render->m_iboffset);
  3976. pos++;
  3977. double captureMs = double(captureElapsed)*toMs;
  3978. tvm.printf(10, pos++, 0x8b, " Capture: %3.4f [ms]", captureMs);
  3979. uint8_t attr[2] = { 0x8c, 0x8a };
  3980. uint8_t attrIndex = _render->m_waitSubmit < _render->m_waitRender;
  3981. tvm.printf(10, pos++, attr[attrIndex &1], " Submit wait: %3.4f [ms]", _render->m_waitSubmit*toMs);
  3982. tvm.printf(10, pos++, attr[(attrIndex+1)&1], " Render wait: %3.4f [ms]", _render->m_waitRender*toMs);
  3983. min = frameTime;
  3984. max = frameTime;
  3985. }
  3986. blit(this, _textVideoMemBlitter, tvm);
  3987. rce = m_renderEncoder;
  3988. if (BX_ENABLED(BGFX_CONFIG_DEBUG_ANNOTATION))
  3989. {
  3990. rce.PopDebugGroup();
  3991. }
  3992. }
  3993. else if (_render->m_debug & BGFX_DEBUG_TEXT)
  3994. {
  3995. if (BX_ENABLED(BGFX_CONFIG_DEBUG_ANNOTATION))
  3996. {
  3997. rce.PushDebugGroup("debugtext");
  3998. }
  3999. blit(this, _textVideoMemBlitter, _render->m_textVideoMem);
  4000. rce = m_renderEncoder;
  4001. if (BX_ENABLED(BGFX_CONFIG_DEBUG_ANNOTATION))
  4002. {
  4003. rce.PopDebugGroup();
  4004. }
  4005. }
  4006. endEncoding();
  4007. scratchBuffer.submit();
  4008. m_cmd.kick(true);
  4009. scratchBuffer.release();
  4010. #if !BX_PLATFORM_EMSCRIPTEN
  4011. for (uint32_t ii = 0, num = m_numWindows; ii < num; ++ii)
  4012. {
  4013. FrameBufferWgpu& frameBuffer = ii == 0 ? m_mainFrameBuffer : m_frameBuffers[m_windows[ii].idx];
  4014. if (NULL != frameBuffer.m_swapChain
  4015. && frameBuffer.m_swapChain->m_drawable)
  4016. {
  4017. SwapChainWgpu& swapChain = *frameBuffer.m_swapChain;
  4018. swapChain.m_swapChain.Present();
  4019. }
  4020. }
  4021. #endif
  4022. }
  4023. } /* namespace webgpu */ } // namespace bgfx
  4024. #else
  4025. namespace bgfx { namespace webgpu
  4026. {
  4027. RendererContextI* rendererCreate(const Init& _init)
  4028. {
  4029. BX_UNUSED(_init);
  4030. return NULL;
  4031. }
  4032. void rendererDestroy()
  4033. {
  4034. }
  4035. } /* namespace webgpu */ } // namespace bgfx
  4036. #endif // BGFX_CONFIG_RENDERER_WEBGPU