renderer_webgpu.cpp 150 KB

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