@meimu/mutracer 0.1.5 → 0.2.0

package/lib/renderer/WGPUBindlessTextures.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"WGPUBindlessTextures.js","sourceRoot":"","sources":["../../src/renderer/WGPUBindlessTextures.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,oBAAoB,EAAE,MAAM,2BAA2B,CAAC;AAEjE,OAAO,EAAC,mBAAmB,EAAC,MAAM,qBAAqB,CAAC;AACxD,OAAO,EAAC,gBAAgB,EAAC,MAAM,0BAA0B,CAAC;AAC1D,OAAO,EAAC,YAAY,EAAC,MAAM,mBAAmB,CAAC;AAC/C,OAAO,EAAC,sBAAsB,EAAC,MAAM,yBAAyB,CAAC;AAC/D,OAAO,EAAC,WAAW,EAAC,MAAM,cAAc,CAAC;AACzC,OAAO,EAAC,kBAAkB,EAAE,mBAAmB,EAAC,MAAM,+BAA+B,CAAC;AACtF,OAAO,EAAC,mBAAmB,EAAE,mBAAmB,EAAE,kBAAkB,EAAE,oBAAoB,EAAqB,MAAM,mBAAmB,CAAC;AAGzI,MAAM,UAAU,uBAAuB,CAAC,QAA4B,EAAE,MAAc,EAAE,SAAiB;IACrG,MAAM,YAAY,GAAG,QAAQ,CAAC,IAAI,CAAC,mBAAmB,CAAC,EAAE,aAAa,GAAG,YAAY,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC;IAC9F,MAAM,OAAO,GAAG,CAAC,GAAG,IAAI,GAAG,CAAC,QAAQ,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,CAAC,CAAC,MAAM,EAAE,CAAC,CAAC;IACtE,KAAK,MAAM,EAAC,KAAK,EAAC,IAAI,QAAQ;QAAE,IAAI,CAAC,MAAM,CAAC,aAAa,CAAC,KAAK,CAAC,KAAK,CAAC,IAAI,CAAC,MAAM,CAAC,aAAa,CAAC,KAAK,CAAC,MAAM,CAAC,IAAI,KAAK,CAAC,KAAK,GAAG,CAAC,IAAI,KAAK,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;YACrJ,MAAM,IAAI,KAAK,CAAC,wDAAwD,CAAC,CAAC;QAC5E,CAAC;IACD,MAAM,OAAO,GAAG,CAAC,KAAa,EAAE,EAAE,CAAC,QAAQ,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,EAAC,KAAK,EAAE,IAAI,CAAC,GAAG,CAAC,CAAC,EAAE,IAAI,CAAC,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,KAAK,GAAG,KAAK,CAAC,CAAC,EAAE,MAAM,EAAE,IAAI,CAAC,GAAG,CAAC,CAAC,EAAE,IAAI,CAAC,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,MAAM,GAAG,KAAK,CAAC,CAAC,EAAC,CAAC,CAAC,CAAC;IACzK,MAAM,KAAK,GAAG,CAAC,KAAsB,EAAE,EAAE,CAAC,OAAO,CAAC,MAAM,CAAC,CAAC,GAAG,EAAE,CAAC,EAAE,EAAE,CAAC,GAAG,GAAG,KAAK,CAAC,CAAC,CAAC,CAAC,KAAK,GAAG,KAAK,CAAC,CAAC,CAAC,CAAC,MAAM,GAAG,aAAa,EAAE,CAAC,CAAC,CAAC;IAChI,IAAI,OAAO,CAAC,MAAM,GAAG,aAAa,GAAG,MAAM;QAAE,MAAM,IAAI,KAAK,CAAC,sDAAsD,CAAC,CAAC;IACrH,IAAI,KAAK,GAAG,QAAQ,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,CAAC,EAAE,EAAE,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC,EAAE,SAAS,GAAG,IAAI,CAAC,GAAG,CAAC,CAAC,CAAC,KAAK,CAAC,KAAK,EAAE,CAAC,CAAC,KAAK,CAAC,MAAM,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC;IAC3G,IAAI,KAAK,CAAC,OAAO,CAAC,KAAK,CAAC,CAAC,GAAG,MAAM,EAAE,CAAC;QACnC,IAAI,GAAG,GAAG,CAAC,EAAE,IAAI,GAAG,KAAK,CAAC;QAC1B,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,EAAE,EAAE,CAAC,EAAE,EAAE,CAAC;YAAA,MAAM,GAAG,GAAG,CAAC,GAAG,GAAG,IAAI,CAAC,GAAG,CAAC,CAAC;YAAC,IAAI,KAAK,CAAC,OAAO,CAAC,GAAG,CAAC,CAAC,IAAI,MAAM;gBAAE,GAAG,GAAG,GAAG,CAAC;;gBAAM,IAAI,GAAG,GAAG,CAAC;QAAA,CAAC;QAC3H,KAAK,GAAG,GAAG,CAAC;IACd,CAAC;IACD,MAAM,KAAK,GAAG,OAAO,CAAC,KAAK,CAAC,CAAC;IAC7B,OAAO,EAAC,KAAK,EAAE,KAAK,EAAE,YAAY,EAAE,aAAa,EAAE,cAAc,EAAE,KAAK,CAAC,KAAK,CAAC,EAAC,CAAC;AACnF,CAAC;AACD,MAAM,UAAU,wBAAwB,CAAC,OAAgB,EAAE,IAAmB,EAAE,YAAqB;IACnG,MAAM,IAAI,GAAG,YAAY,CAAC,CAAC,CAAC,IAAI,WAAW,CAAC,IAAI,CAAC,KAAK,GAAG,IAAI,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC,CAAC,CAAC,IAAI,UAAU,CAAC,IAAI,CAAC,KAAK,GAAG,IAAI,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC;IACzH,IAAI,mBAAmB,CAAC,OAAO,CAAC,KAAK,CAAC,EAAE,CAAC;QACvC,sBAAsB,CAAC,IAAI,EAAE,IAAI,CAAC,KAAK,EAAE,EAAC,KAAK,EAAE,CAAC,EAAE,KAAK,EAAE,CAAC,EAAE,CAAC,EAAE,CAAC,EAAE,CAAC,EAAE,CAAC,EAAE,GAAG,IAAI,EAAC,EAAE,EAAC,KAAK,EAAE,OAAO,CAAC,KAAK,EAAE,KAAK,EAAE,OAAO,CAAC,KAAK,EAAC,EAAE,YAAY,CAAC,CAAC;IAClJ,CAAC;SAAM,CAAC;QACN,MAAM,MAAM,GAAG,YAAY,CAAC,IAAI,CAAC,KAAK,EAAE,IAAI,CAAC,MAAM,CAAC,EAAE,OAAO,GAAG,MAAM,CAAC,UAAU,CAAC,IAAI,CAAC,CAAC;QACxF,IAAI,CAAC,OAAO;YAAE,MAAM,IAAI,KAAK,CAAC,4CAA4C,CAAC,CAAC;QAC5E,IAAI,OAAO,CAAC,KAAK,EAAE,CAAC;YAAA,OAAO,CAAC,SAAS,CAAC,CAAC,EAAE,IAAI,CAAC,MAAM,CAAC,CAAC;YAAC,OAAO,CAAC,KAAK,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,CAAC;QAAA,CAAC;QAC7E,IAAI,oBAAoB,CAAC,OAAO,CAAC,KAAK,CAAC;YAAE,MAAM,IAAI,KAAK,CAAC,kEAAkE,CAAC,CAAC;QAC7H,OAAO,CAAC,SAAS,CAAC,OAAO,CAAC,KAAK,EAAE,CAAC,EAAE,CAAC,EAAE,IAAI,CAAC,KAAK,EAAE,IAAI,CAAC,MAAM,CAAC,CAAC;QAChE,MAAM,MAAM,GAAG,OAAO,CAAC,YAAY,CAAC,CAAC,EAAE,CAAC,EAAE,IAAI,CAAC,KAAK,EAAE,IAAI,CAAC,MAAM,CAAC,CAAC,IAAI,CAAC;QACxE,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,MAAM,CAAC,MAAM,EAAE,CAAC,EAAE;YAAE,IAAI,CAAC,CAAC,CAAC,GAAG,YAAY,CAAC,CAAC,CAAC,WAAW,CAAC,MAAM,CAAC,CAAC,CAAC,GAAG,GAAG,CAAC,CAAC,CAAC,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC;IAC5G,CAAC;IACD,OAAO,IAAI,CAAC;AACd,CAAC;AACD,MAAM,UAAU,0BAA0B,CAAC,QAA4B,EAAE,KAAsB,EAAE,QAAgB;IAC/G,MAAM,IAAI,GAAG,IAAI,WAAW,CAAC,QAAQ,GAAG,EAAE,CAAC,EAAE,GAAG,GAAG,IAAI,YAAY,CAAC,IAAI,CAAC,EAAE,GAAG,GAAG,IAAI,WAAW,CAAC,IAAI,CAAC,CAAC;IACvG,QAAQ,CAAC,OAAO,CAAC,CAAC,CAAC,EAAE,CAAC,EAAE,EAAE;QACxB,MAAM,IAAI,GAAG,CAAC,GAAG,EAAE,CAAC;QACpB,GAAG,CAAC,GAAG,CAAC,CAAC,GAAG,CAAC,CAAC,OAAO,EAAE,GAAG,CAAC,CAAC,QAAQ,CAAC,EAAE,IAAI,CAAC,CAAC;QAC7C,GAAG,CAAC,GAAG,CAAC,CAAC,kBAAkB,GAAG,CAAC,EAAE,CAAC,GAAG,gBAAgB,CAAC,CAAC,CAAC,KAAK,CAAC,GAAG,CAAC,GAAG,gBAAgB,CAAC,CAAC,CAAC,KAAK,CAAC,EAAE,CAAC,CAAC,SAAS,EAAE,kBAAkB,CAAC,CAAC,CAAC,EAAE,KAAK,CAAC,CAAC,CAAC,CAAC,KAAK,EAAE,KAAK,CAAC,CAAC,CAAC,CAAC,MAAM,CAAC,EAAE,IAAI,GAAG,CAAC,CAAC,CAAC;IAClL,CAAC,CAAC,CAAC;IACH,OAAO,IAAI,CAAC;AACd,CAAC;AAKD,gGAAgG;AAChG,MAAM,OAAO,oBAAoB;IAcX;IAbpB,KAAK,CAAiB;IACtB,QAAQ,CAAa;IACrB,UAAU,CAAiB;IACnB,OAAO,GAAG,IAAI,GAAG,EAAyB,CAAC;IAC3C,OAAO,GAAiB,EAAE,CAAC;IAC3B,QAAQ,CAAe;IACvB,QAAQ,CAAa;IACrB,QAAQ,GAAG,KAAK,CAAC;IACjB,UAAU,GAAG,CAAC,CAAC;IACf,gBAAgB,GAAG,CAAC,CAAC;IACrB,YAAY,GAAG,CAAC,CAAC;IACjB,cAAc,CAAc;IAC5B,KAAK,GAAG,EAAC,YAAY,EAAE,CAAC,EAAE,cAAc,EAAE,CAAC,EAAE,KAAK,EAAE,CAAC,EAAC,CAAC;IAC/D,YAAoB,MAAiB;QAAjB,WAAM,GAAN,MAAM,CAAW;QACnC,MAAM,KAAK,GAAqB,CAAC,QAAQ,EAAE,eAAe,EAAE,eAAe,CAAC,CAAC;QAC7E,IAAI,CAAC,QAAQ,GAAG,KAAK,CAAC,OAAO,CAAC,YAAY,CAAC,EAAE,CAAC,KAAK,CAAC,GAAG,CAAC,YAAY,CAAC,EAAE,CAAC,MAAM,CAAC,aAAa,CAAC,EAAC,SAAS,EAAE,QAAQ,EAAE,SAAS,EAAE,QAAQ,EAAE,YAAY,EAAE,YAAY,EAAC,CAAC,CAAC,CAAC,CAAC;QACvK,IAAI,CAAC,QAAQ,GAAG,MAAM,CAAC,aAAa,CAAC,EAAC,IAAI,EAAE,CAAC,CAAC,EAAC,CAAC,CAAC,EAAE,MAAM,EAAE,YAAY,EAAE,KAAK,EAAE,eAAe,CAAC,eAAe,GAAG,eAAe,CAAC,QAAQ,EAAC,CAAC,CAAC;QAC7I,MAAM,CAAC,KAAK,CAAC,YAAY,CAAC,EAAC,OAAO,EAAE,IAAI,CAAC,QAAQ,EAAC,EAAE,IAAI,UAAU,CAAC,CAAC,GAAG,EAAC,GAAG,EAAC,GAAG,EAAC,GAAG,CAAC,CAAC,EAAE,EAAC,WAAW,EAAE,CAAC,EAAC,EAAE,CAAC,CAAC,EAAC,CAAC,CAAC,CAAC,CAAC;IAClH,CAAC;IACD,IAAI,QAAQ,KAAK,OAAO,IAAI,CAAC,UAAU,CAAC,CAAC,CAAC;IAC1C,IAAI,WAAW,KAAK,OAAO,EAAC,GAAG,IAAI,CAAC,KAAK,EAAE,aAAa,EAAE,IAAI,CAAC,QAAQ,EAAE,IAAI,IAAI,CAAC,EAAE,qBAAqB,EAAE,IAAI,CAAC,KAAK,EAAE,IAAI,IAAI,CAAC,EAAE,YAAY,EAAE,IAAI,CAAC,YAAY,EAAC,CAAC,CAAC,CAAC;IACrK,MAAM,CAAC,QAAmB,EAAE,MAAc;QACxC,IAAI,IAAI,CAAC,QAAQ;YAAE,MAAM,IAAI,KAAK,CAAC,iCAAiC,CAAC,CAAC;QACtE,IAAI,IAAI,CAAC,UAAU;YAAE,MAAM,IAAI,KAAK,CAAC,qDAAqD,CAAC,CAAC;QAC5F,MAAM,MAAM,GAAG,mBAAmB,CAAC,IAAI,CAAC,MAAM,EAAE,QAAQ,CAAC,MAAM,CAAC,CAAC;QACjE,IAAI,MAAM;YAAE,MAAM,IAAI,KAAK,CAAC,MAAM,CAAC,CAAC;QACpC,MAAM,IAAI,GAAG,uBAAuB,CAAC,QAAQ,EAAE,MAAM,EAAE,IAAI,CAAC,MAAM,CAAC,MAAM,CAAC,qBAAqB,CAAC,CAAC;QACjG,MAAM,MAAM,GAAG,IAAI,CAAC,YAAY,CAAC,CAAC,CAAC,aAAa,CAAC,CAAC,CAAC,YAAY,CAAC;QAChE,MAAM,IAAI,GAAG,IAAI,GAAG,EAAyB,EAAE,OAAO,GAAiB,EAAE,CAAC;QAC1E,IAAI,QAAmC,EAAE,SAAgC,CAAC;QAC1E,IAAI,CAAC;YACH,QAAQ,CAAC,OAAO,CAAC,CAAC,OAAO,EAAE,CAAC,EAAE,EAAE;gBAC9B,IAAI,IAAI,CAAC,GAAG,CAAC,OAAO,CAAC;oBAAE,OAAO;gBAC9B,MAAM,IAAI,GAAG,IAAI,CAAC,KAAK,CAAC,CAAC,CAAC,EAAE,GAAG,GAAG,IAAI,CAAC,OAAO,CAAC,GAAG,CAAC,OAAO,CAAC,CAAC;gBAC5D,MAAM,QAAQ,GAAG,GAAG,IAAI,GAAG,CAAC,KAAK,KAAK,IAAI,CAAC,KAAK,IAAI,GAAG,CAAC,MAAM,KAAK,IAAI,CAAC,MAAM,IAAI,GAAG,CAAC,MAAM,KAAK,MAAM,CAAC;gBACxG,MAAM,GAAG,GAAG,QAAQ,CAAC,CAAC,CAAC,GAAG,CAAC,GAAG,CAAC,CAAC,CAAC,IAAI,CAAC,MAAM,CAAC,aAAa,CAAC,EAAC,KAAK,EAAE,kBAAkB,EAAE,IAAI,EAAE,CAAC,IAAI,CAAC,KAAK,EAAE,IAAI,CAAC,MAAM,CAAC,EAAE,MAAM,EAAE,KAAK,EAAE,eAAe,CAAC,eAAe,GAAG,eAAe,CAAC,QAAQ,EAAC,CAAC,CAAC;gBACpM,IAAI,CAAC,QAAQ;oBAAE,OAAO,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC;gBACjC,IAAI,CAAC,QAAQ,IAAI,GAAG,CAAC,KAAK,KAAK,OAAO,CAAC,KAAK,IAAI,GAAG,CAAC,OAAO,KAAK,OAAO,CAAC,OAAO,IAAI,GAAG,CAAC,KAAK,KAAK,OAAO,CAAC,KAAK,IAAI,GAAG,CAAC,WAAW,KAAK,OAAO,CAAC,KAAK,CAAC,KAAK,IAAI,GAAG,CAAC,YAAY,KAAK,OAAO,CAAC,KAAK,CAAC,MAAM,EAAE,CAAC;oBACvM,MAAM,MAAM,GAAG,wBAAwB,CAAC,OAAO,EAAE,IAAI,EAAE,IAAI,CAAC,YAAY,CAAC,CAAC;oBAC1E,IAAI,CAAC,MAAM,CAAC,KAAK,CAAC,YAAY,CAAC,EAAC,OAAO,EAAE,GAAG,EAAC,EAAE,MAAM,EAAE,EAAC,WAAW,EAAE,IAAI,CAAC,KAAK,GAAG,IAAI,CAAC,aAAa,EAAC,EAAE,CAAC,IAAI,CAAC,KAAK,EAAE,IAAI,CAAC,MAAM,CAAC,CAAC,CAAC;gBACpI,CAAC;gBACD,IAAI,CAAC,GAAG,CAAC,OAAO,EAAE,EAAC,GAAG,EAAE,KAAK,EAAE,OAAO,CAAC,KAAK,EAAE,OAAO,EAAE,OAAO,CAAC,OAAO,EAAE,WAAW,EAAE,OAAO,CAAC,KAAK,CAAC,KAAK,EAAE,YAAY,EAAE,OAAO,CAAC,KAAK,CAAC,MAAM,EAAE,KAAK,EAAE,OAAO,CAAC,KAAK,EAAE,MAAM,EAAE,GAAG,IAAI,EAAC,CAAC,CAAC;YACxL,CAAC,CAAC,CAAC;YACH,MAAM,OAAO,GAAG,QAAQ,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC,CAAE,CAAC,GAAG,CAAC,CAAC;YACpD,MAAM,OAAO,GAAG,CAAC,IAAI,CAAC,KAAK,IAAI,OAAO,CAAC,MAAM,KAAK,IAAI,CAAC,OAAO,CAAC,MAAM,IAAI,OAAO,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,CAAC,EAAE,EAAE,CAAC,CAAC,KAAK,IAAI,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,CAAC;YACvH,IAAI,OAAO,EAAE,CAAC;gBACZ,MAAM,IAAI,GAAG,IAAI,CAAC,GAAG,CAAC,oBAAoB,EAAE,CAAC,IAAI,IAAI,CAAC,IAAI,CAAC,IAAI,CAAC,IAAI,CAAC,IAAI,CAAC,GAAG,CAAC,EAAE,EAAE,QAAQ,CAAC,MAAM,GAAG,kBAAkB,CAAC,CAAC,CAAC,CAAC,CAAC;gBAC3H,QAAQ,GAAG,mBAAmB,CAAC,IAAI,CAAC,MAAM,EAAE,IAAI,CAAC,CAAC;gBAAC,QAAQ,CAAC,MAAM,CAAC,CAAC,EAAE,IAAI,CAAC,QAAQ,CAAC,UAAU,EAAE,CAAC,CAAC;gBAClG,IAAI,CAAC,QAAQ,CAAC,OAAO,CAAC,CAAC,OAAO,EAAE,CAAC,EAAE,EAAE,CAAC,QAAS,CAAC,MAAM,CAAC,CAAC,GAAG,CAAC,EAAE,OAAO,CAAC,CAAC,CAAC;gBACxE,OAAO,CAAC,OAAO,CAAC,CAAC,GAAG,EAAE,CAAC,EAAE,EAAE,CAAC,QAAS,CAAC,MAAM,CAAC,kBAAkB,GAAG,CAAC,EAAE,GAAG,CAAC,UAAU,EAAE,CAAC,CAAC,CAAC;YAC1F,CAAC;YACD,MAAM,WAAW,GAAG,IAAI,CAAC,KAAK,CAAC,IAAI,CAAC,GAAG,CAAC,IAAI,CAAC,MAAM,CAAC,MAAM,CAAC,2BAA2B,EAAE,IAAI,CAAC,MAAM,CAAC,MAAM,CAAC,aAAa,CAAC,GAAG,EAAE,CAAC,CAAC;YAChI,MAAM,QAAQ,GAAG,IAAI,CAAC,GAAG,CAAC,WAAW,EAAE,IAAI,CAAC,GAAG,CAAC,IAAI,CAAC,gBAAgB,EAAE,CAAC,IAAI,IAAI,CAAC,IAAI,CAAC,IAAI,CAAC,IAAI,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC,EAAE,QAAQ,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC;YACjI,IAAI,CAAC,IAAI,CAAC,QAAQ,IAAI,QAAQ,KAAK,IAAI,CAAC,gBAAgB;gBAAE,SAAS,GAAG,IAAI,CAAC,MAAM,CAAC,YAAY,CAAC,EAAC,KAAK,EAAE,2BAA2B,EAAE,IAAI,EAAE,QAAQ,GAAG,EAAE,EAAE,KAAK,EAAE,cAAc,CAAC,OAAO,GAAG,cAAc,CAAC,QAAQ,EAAC,CAAC,CAAC;YACnN,IAAI,CAAC,MAAM,CAAC,KAAK,CAAC,WAAW,CAAC,SAAS,IAAI,IAAI,CAAC,QAAS,EAAE,CAAC,EAAE,0BAA0B,CAAC,QAAQ,EAAE,IAAI,CAAC,KAAK,EAAE,QAAQ,CAAC,CAAC,CAAC;YAC1H,MAAM,QAAQ,GAAG,QAAQ,CAAC,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC,CAAC,SAAS,EAAE,SAAS,GAAG,SAAS,CAAC,CAAC,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC,CAAC,SAAS,CAAC;YACtG,MAAM,QAAQ,GAAG,IAAI,GAAG,CAAC,OAAO,CAAC,CAAC;YAClC,MAAM,OAAO,GAAG,CAAC,GAAG,IAAI,GAAG,CAAC,CAAC,GAAG,IAAI,CAAC,OAAO,CAAC,MAAM,EAAE,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,QAAQ,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC,CAAC;YACvG,IAAI,CAAC,OAAO,GAAG,IAAI,CAAC;YAAC,IAAI,CAAC,OAAO,GAAG,OAAO,CAAC;YAC5C,IAAI,QAAQ;gBAAE,IAAI,CAAC,KAAK,GAAG,QAAQ,CAAC;YACpC,IAAI,SAAS;gBAAE,IAAI,CAAC,QAAQ,GAAG,SAAS,CAAC;YACzC,IAAI,CAAC,gBAAgB,GAAG,QAAQ,CAAC;YAAC,IAAI,CAAC,UAAU,EAAE,CAAC;YACpD,IAAI,CAAC,KAAK,GAAG,EAAC,YAAY,EAAE,QAAQ,CAAC,MAAM,EAAE,cAAc,EAAE,CAAC,GAAG,IAAI,GAAG,CAAC,OAAO,CAAC,CAAC,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,CAAC,EAAE,EAAE,CAAC,CAAC,GAAG,CAAC,CAAC,KAAK,GAAG,CAAC,CAAC,MAAM,GAAG,IAAI,CAAC,aAAa,EAAE,CAAC,CAAC,EAAE,KAAK,EAAE,IAAI,CAAC,KAAK,EAAC,CAAC;YACxK,IAAI,QAAQ,IAAI,SAAS,IAAI,OAAO,CAAC,MAAM,EAAE,CAAC;gBAC5C,IAAI,CAAC,YAAY,GAAG,CAAC,SAAS,EAAE,IAAI,IAAI,CAAC,CAAC,GAAG,OAAO,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,CAAC,EAAE,EAAE,CAAC,CAAC,GAAG,CAAC,CAAC,KAAK,GAAG,CAAC,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC,MAAM,KAAK,aAAa,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC;gBACxI,MAAM,OAAO,GAAG,GAAG,EAAE,GAAE,QAAQ,EAAE,OAAO,EAAE,CAAC,CAAC,SAAS,EAAE,OAAO,EAAE,CAAC,CAAC,OAAO,CAAC,OAAO,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,OAAO,EAAE,CAAC,CAAC,CAAC,IAAI,CAAC,YAAY,GAAG,CAAC,CAAC,CAAA,CAAC,CAAC;gBAC7H,IAAI,CAAC,cAAc,GAAG,OAAO,CAAC;gBAC9B,IAAI,CAAC,UAAU,GAAG,IAAI,CAAC,MAAM,CAAC,KAAK,CAAC,mBAAmB,EAAE,CAAC,KAAK,CAAC,GAAG,EAAE,GAAE,CAAC,CAAC,CAAC,IAAI,CAAC,GAAG,EAAE;oBAClF,IAAI,IAAI,CAAC,cAAc,KAAK,OAAO,EAAE,CAAC;wBAAA,OAAO,EAAE,CAAC;wBAAC,IAAI,CAAC,cAAc,GAAG,SAAS,CAAC;wBAAC,IAAI,CAAC,UAAU,GAAG,SAAS,CAAC;oBAAA,CAAC;gBACjH,CAAC,CAAC,CAAC;YACL,CAAC;QACH,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YAAA,QAAQ,EAAE,OAAO,EAAE,CAAC;YAAC,SAAS,EAAE,OAAO,EAAE,CAAC;YAAC,OAAO,CAAC,OAAO,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,OAAO,EAAE,CAAC,CAAC;YAAC,MAAM,KAAK,CAAC;QAAA,CAAC;IAC9G,CAAC;IACD,OAAO;QACL,IAAI,IAAI,CAAC,QAAQ;YAAE,OAAO;QAC1B,IAAI,CAAC,QAAQ,GAAG,IAAI,CAAC;QAAC,IAAI,CAAC,UAAU,EAAE,CAAC;QACxC,IAAI,CAAC,cAAc,EAAE,EAAE,CAAC;QAAC,IAAI,CAAC,cAAc,GAAG,SAAS,CAAC;QAAC,IAAI,CAAC,UAAU,GAAG,SAAS,CAAC;QACtF,IAAI,CAAC,KAAK,EAAE,OAAO,EAAE,CAAC;QAAC,IAAI,CAAC,QAAQ,EAAE,OAAO,EAAE,CAAC;QAAC,IAAI,CAAC,QAAQ,CAAC,OAAO,EAAE,CAAC;QACzE,IAAI,GAAG,CAAC,CAAC,GAAG,IAAI,CAAC,OAAO,CAAC,MAAM,EAAE,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,OAAO,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,OAAO,EAAE,CAAC,CAAC;QAC9E,IAAI,CAAC,OAAO,CAAC,KAAK,EAAE,CAAC;QAAC,IAAI,CAAC,OAAO,GAAG,EAAE,CAAC;IAC1C,CAAC;CACF","sourcesContent":["import { isProceduralSkyImage } from '../scene/skyParameters.js';\nimport type Texture from '../scene/Texture.js';\nimport {isTextureDataSource} from '../scene/Texture.js';\nimport {textureWrapIndex} from '../scene/TextureState.js';\nimport {createCanvas} from '../util/canvas.js';\nimport {writeRawAtlasPlacement} from './textureAtlasPixels.js';\nimport {float16Bits} from './float16.js';\nimport {textureColorLinear, textureFloatStorage} from './WGPUBindlessMaterialData.js';\nimport {bindlessLimitReason, createResourceTable, FIRST_TEXTURE_SLOT, RESOURCE_TABLE_LIMIT, type ResourceTable} from './wgpuBindless.js';\n\nexport interface TextureExtent {width: number; height: number;}\nexport function planIndependentTextures(textures: readonly Texture[], budget: number, dimension: number) {\n  const floatStorage = textures.some(textureFloatStorage), bytesPerPixel = floatStorage ? 8 : 4;\n  const indices = [...new Map(textures.map((t, i) => [t, i])).values()];\n  for (const {image} of textures) if (!Number.isSafeInteger(image.width) || !Number.isSafeInteger(image.height) || image.width < 1 || image.height < 1) {\n    throw new Error('Material textures require positive integer dimensions.');\n  }\n  const extents = (scale: number) => textures.map(t => ({width: Math.max(1, Math.floor(t.image.width * scale)), height: Math.max(1, Math.floor(t.image.height * scale))}));\n  const bytes = (sizes: TextureExtent[]) => indices.reduce((sum, i) => sum + sizes[i].width * sizes[i].height * bytesPerPixel, 0);\n  if (indices.length * bytesPerPixel > budget) throw new Error('Material textures cannot fit the budget even at 1×1.');\n  let scale = textures.reduce((s, t) => Math.min(s, dimension / Math.max(t.image.width, t.image.height)), 1);\n  if (bytes(extents(scale)) > budget) {\n    let low = 0, high = scale;\n    for (let i = 0; i < 40; i++) {const mid = (low + high) / 2; if (bytes(extents(mid)) <= budget) low = mid; else high = mid;}\n    scale = low;\n  }\n  const sizes = extents(scale);\n  return {sizes, scale, floatStorage, bytesPerPixel, estimatedBytes: bytes(sizes)};\n}\nexport function independentTexturePixels(texture: Texture, size: TextureExtent, floatStorage: boolean): Uint8Array | Uint16Array {\n  const data = floatStorage ? new Uint16Array(size.width * size.height * 4) : new Uint8Array(size.width * size.height * 4);\n  if (isTextureDataSource(texture.image)) {\n    writeRawAtlasPlacement(data, size.width, {index: 0, layer: 0, x: 0, y: 0, ...size}, {image: texture.image, flipY: texture.flipY}, floatStorage);\n  } else {\n    const canvas = createCanvas(size.width, size.height), context = canvas.getContext('2d');\n    if (!context) throw new Error('Unable to prepare material texture pixels.');\n    if (texture.flipY) {context.translate(0, size.height); context.scale(1, -1);}\n    if (isProceduralSkyImage(texture.image)) throw new Error('ProceduralSky is only supported as an environment or background.');\n    context.drawImage(texture.image, 0, 0, size.width, size.height);\n    const pixels = context.getImageData(0, 0, size.width, size.height).data;\n    for (let i = 0; i < pixels.length; i++) data[i] = floatStorage ? float16Bits(pixels[i] / 255) : pixels[i];\n  }\n  return data;\n}\nexport function packIndependentTextureInfo(textures: readonly Texture[], sizes: TextureExtent[], capacity: number) {\n  const data = new ArrayBuffer(capacity * 48), f32 = new Float32Array(data), u32 = new Uint32Array(data);\n  textures.forEach((t, i) => {\n    const base = i * 12;\n    f32.set([...t.uvScale, ...t.uvOffset], base);\n    u32.set([FIRST_TEXTURE_SLOT + i, 1 + textureWrapIndex(t.wrapS) * 3 + textureWrapIndex(t.wrapT), t.uvChannel, textureColorLinear(t), sizes[i].width, sizes[i].height], base + 4);\n  });\n  return data;\n}\ninterface TextureEntry {\n  gpu: GPUTexture; image: Texture['image']; version: number; width: number; height: number;\n  sourceWidth: number; sourceHeight: number; flipY: boolean; format: GPUTextureFormat;\n}\n/** Browser scene resources; callers wait for retirement before preparing another generation. */\nexport class WGPUBindlessTextures {\n  table?: ResourceTable;\n  metadata?: GPUBuffer;\n  retirement?: Promise<void>;\n  private entries = new Map<Texture, TextureEntry>();\n  private logical: GPUTexture[] = [];\n  private samplers: GPUSampler[];\n  private fallback: GPUTexture;\n  private disposed = false;\n  private generation = 0;\n  private metadataCapacity = 0;\n  private retiredBytes = 0;\n  private destroyRetired?: () => void;\n  private stats = {textureCount: 0, estimatedBytes: 0, scale: 1};\n  constructor(private device: GPUDevice) {\n    const modes: GPUAddressMode[] = ['repeat', 'clamp-to-edge', 'mirror-repeat'];\n    this.samplers = modes.flatMap(addressModeU => modes.map(addressModeV => device.createSampler({minFilter: 'linear', magFilter: 'linear', addressModeU, addressModeV})));\n    this.fallback = device.createTexture({size: [1,1], format: 'rgba8unorm', usage: GPUTextureUsage.TEXTURE_BINDING | GPUTextureUsage.COPY_DST});\n    device.queue.writeTexture({texture: this.fallback}, new Uint8Array([255,255,255,255]), {bytesPerRow: 4}, [1,1]);\n  }\n  get revision() { return this.generation; }\n  get diagnostics() { return {...this.stats, metadataBytes: this.metadata?.size ?? 0, resourceTableCapacity: this.table?.size ?? 0, retiredBytes: this.retiredBytes}; }\n  update(textures: Texture[], budget: number) {\n    if (this.disposed) throw new Error('Bindless textures are disposed.');\n    if (this.retirement) throw new Error('Previous resource-table generation has not retired.');\n    const reason = bindlessLimitReason(this.device, textures.length);\n    if (reason) throw new Error(reason);\n    const plan = planIndependentTextures(textures, budget, this.device.limits.maxTextureDimension2D);\n    const format = plan.floatStorage ? 'rgba16float' : 'rgba8unorm';\n    const next = new Map<Texture, TextureEntry>(), created: GPUTexture[] = [];\n    let newTable: ResourceTable | undefined, newBuffer: GPUBuffer | undefined;\n    try {\n      textures.forEach((texture, i) => {\n        if (next.has(texture)) return;\n        const size = plan.sizes[i], old = this.entries.get(texture);\n        const reusable = old && old.width === size.width && old.height === size.height && old.format === format;\n        const gpu = reusable ? old.gpu : this.device.createTexture({label: 'Material texture', size: [size.width, size.height], format, usage: GPUTextureUsage.TEXTURE_BINDING | GPUTextureUsage.COPY_DST});\n        if (!reusable) created.push(gpu);\n        if (!reusable || old.image !== texture.image || old.version !== texture.version || old.flipY !== texture.flipY || old.sourceWidth !== texture.image.width || old.sourceHeight !== texture.image.height) {\n          const pixels = independentTexturePixels(texture, size, plan.floatStorage);\n          this.device.queue.writeTexture({texture: gpu}, pixels, {bytesPerRow: size.width * plan.bytesPerPixel}, [size.width, size.height]);\n        }\n        next.set(texture, {gpu, image: texture.image, version: texture.version, sourceWidth: texture.image.width, sourceHeight: texture.image.height, flipY: texture.flipY, format, ...size});\n      });\n      const logical = textures.map(t => next.get(t)!.gpu);\n      const changed = !this.table || logical.length !== this.logical.length || logical.some((t, i) => t !== this.logical[i]);\n      if (changed) {\n        const size = Math.min(RESOURCE_TABLE_LIMIT, 2 ** Math.ceil(Math.log2(Math.max(16, textures.length + FIRST_TEXTURE_SLOT))));\n        newTable = createResourceTable(this.device, size); newTable.update(0, this.fallback.createView());\n        this.samplers.forEach((sampler, i) => newTable!.update(i + 1, sampler));\n        logical.forEach((gpu, i) => newTable!.update(FIRST_TEXTURE_SLOT + i, gpu.createView()));\n      }\n      const maxCapacity = Math.floor(Math.min(this.device.limits.maxStorageBufferBindingSize, this.device.limits.maxBufferSize) / 48);\n      const capacity = Math.min(maxCapacity, Math.max(this.metadataCapacity, 2 ** Math.ceil(Math.log2(Math.max(1, textures.length)))));\n      if (!this.metadata || capacity !== this.metadataCapacity) newBuffer = this.device.createBuffer({label: 'Material texture metadata', size: capacity * 48, usage: GPUBufferUsage.STORAGE | GPUBufferUsage.COPY_DST});\n      this.device.queue.writeBuffer(newBuffer ?? this.metadata!, 0, packIndependentTextureInfo(textures, plan.sizes, capacity));\n      const oldTable = newTable ? this.table : undefined, oldBuffer = newBuffer ? this.metadata : undefined;\n      const retained = new Set(logical);\n      const removed = [...new Set([...this.entries.values()].map(e => e.gpu))].filter(t => !retained.has(t));\n      this.entries = next; this.logical = logical;\n      if (newTable) this.table = newTable;\n      if (newBuffer) this.metadata = newBuffer;\n      this.metadataCapacity = capacity; this.generation++;\n      this.stats = {textureCount: textures.length, estimatedBytes: [...new Set(logical)].reduce((n, t) => n + t.width * t.height * plan.bytesPerPixel, 0), scale: plan.scale};\n      if (oldTable || oldBuffer || removed.length) {\n        this.retiredBytes = (oldBuffer?.size ?? 0) + removed.reduce((n, t) => n + t.width * t.height * (t.format === 'rgba16float' ? 8 : 4), 0);\n        const destroy = () => {oldTable?.destroy(); oldBuffer?.destroy(); removed.forEach(t => t.destroy()); this.retiredBytes = 0;};\n        this.destroyRetired = destroy;\n        this.retirement = this.device.queue.onSubmittedWorkDone().catch(() => {}).then(() => {\n          if (this.destroyRetired === destroy) {destroy(); this.destroyRetired = undefined; this.retirement = undefined;}\n        });\n      }\n    } catch (error) {newTable?.destroy(); newBuffer?.destroy(); created.forEach(t => t.destroy()); throw error;}\n  }\n  dispose() {\n    if (this.disposed) return;\n    this.disposed = true; this.generation++;\n    this.destroyRetired?.(); this.destroyRetired = undefined; this.retirement = undefined;\n    this.table?.destroy(); this.metadata?.destroy(); this.fallback.destroy();\n    new Set([...this.entries.values()].map(e => e.gpu)).forEach(t => t.destroy());\n    this.entries.clear(); this.logical = [];\n  }\n}\n"]}

package/lib/renderer/WGPUBloomPass.js

@@ -1,5 +1,6 @@
+import { BLOOM_PLAN, BLOOM_LEVELS, bloomBlurSize, bloomBlendWeight, bloomBrightDefine, bloomBrightMain, bloomDownDefine, bloomDownMain, bloomBlurDefine, bloomBlurMain, bloomBlendMain } from './wgsl/bloom.wgsl.js';
 import { WGPUFullQuadPass } from './WGPUFullQuadPass.js';
-const COUNT = 5;
+const COUNT = BLOOM_LEVELS - 1;
 const linearSamplerDescriptor = {
     magFilter: 'linear',
     minFilter: 'linear'
@@ -20,36 +21,8 @@ function createBrightPass(device) {
                 data: 1
             }
         ],
-        fsDefine: /*wgsl*/ `
-fn sD(a: vec4f, b: vec4f, c: vec4f) -> vec4f {
-return a + b + c - min(min(a, b), c) - max(max(a, b), c);
-}
-`,
-        fsMain: /*wgsl*/ `
-${''
-        // Use median filter to reduce noise
-        // https://github.com/keijiro/KinoBloom/blob/master/Assets/Kino/Bloom/Shader/Bloom.cginc#L96
-        }
-let d = vec3i(1, 1, 0);
-
-var eA = load_colorTex(uv, 0);
-let s1 = load_colorTex(uv - d.xz, 0);
-let s2 = load_colorTex(uv + d.xz, 0);
-let s3 = load_colorTex(uv - d.zy, 0);
-let s4 = load_colorTex(uv + d.zy, 0);
-eA = sD(sD(eA, s1, s2), s3, s4);
-
-
-let lum = dot(eA.rgb, vec3f(0.2125, 0.7154, 0.0721));
-var color: vec4f;
-if (lum > threshold && eA.a > 0.0) {
-color = vec4f(eA.rgb * scale, eA.a);
-} else {
-
-color = vec4f(vec3(0.0), eA.a);
-}
-output.color = color;
-`
+        fsDefine: bloomBrightDefine,
+        fsMain: bloomBrightMain
     });
 }
 function createDownsamplePass(device) {
@@ -60,20 +33,8 @@ function createDownsamplePass(device) {
         samplers: {
             linearSampler: device.createSampler(linearSamplerDescriptor)
         },
-        fsDefine: /*wgsl*/ `
-fn vY(c: vec3f) -> f32 {
-return max(max(c.r, c.g), c.b);
-}
-`,
-        fsMain: /*wgsl*/ `
-let size = vec2f(colorTexRegion());
-let d = vec4f(-1.0, -1.0, 1.0, 1.0) / size.xyxy;
-${''}
-output.color = sample_colorTex_linearSampler(uv2 + d.xy) * 0.25
-+ sample_colorTex_linearSampler(uv2 + d.zy) * 0.25
-+ sample_colorTex_linearSampler(uv2 + d.xw) * 0.25
-+ sample_colorTex_linearSampler(uv2 + d.zw) * 0.25;
-`
+        fsDefine: bloomDownDefine,
+        fsMain: bloomDownMain
     });
 }
 function createGaussianPass(device) {
@@ -95,22 +56,8 @@ function createGaussianPass(device) {
         samplers: {
             linearSampler: device.createSampler(linearSamplerDescriptor)
         },
-        fsDefine: /*wgsl*/ `
-const wc = array<f32, 9>(0.07, 0.09, 0.12, 0.14, 0.16, 0.14, 0.12, 0.09, 0.07);
-`,
-        fsMain: /*wgsl*/ `
-let vR = vec2f(colorTexRegion());
-let off = direction / vR * size;
-
-var sum = vec4f(0.0);
-var tg = 0.0;
-for (var i = 0; i < 9; i++) {
-let w = wc[i];
-sum += sample_colorTex_linearSampler(uv2 + off * f32(i - 4)) * w;
-tg += w;
-}
-output.color = sum / max(tg, 0.01);
-`
+        fsDefine: bloomBlurDefine,
+        fsMain: bloomBlurMain
     });
 }
 function createBlendPass(device) {
@@ -127,11 +74,7 @@ function createBlendPass(device) {
         samplers: {
             linearSampler: device.createSampler(linearSamplerDescriptor)
         },
-        fsMain: /*wgsl*/ `
-let vZ = sample_colorTex1_linearSampler(uv2);
-let wa = sample_colorTex2_linearSampler(uv2);
-output.color = mix(vZ, wa, weight);
-`
+        fsMain: bloomBlendMain
     });
 }
 class WGPUBloomPass {
@@ -188,44 +131,21 @@ class WGPUBloomPass {
         this._blendPasses.forEach((blendPass) => {
             blendPass.setUniform('weight', 
             // Use weight to control radius
-            1 - Math.max(Math.min(radius ** 0.3 * 0.8, 0.8), 0));
+            bloomBlendWeight(radius));
         });
         [...this._blurHPasses, ...this._blurVPasses].forEach((blurPass) => {
             //Some magic fomular
-            blurPass.setUniform('size', Math.max(Math.min(radius * 2, 1.0), 0.2));
+            blurPass.setUniform('size', bloomBlurSize(radius));
         });
     }
     createPass(encoder, colorTex) {
-        const brightPass = this._brightPass;
-        const downSamplePasses = this._downsamplePasses;
-        const blendPasses = this._blendPasses;
-        const blurHPasses = this._blurHPasses;
-        const blurVPasses = this._blurVPasses;
-        brightPass.createPass(encoder, { colorTex });
-        // Downsample
-        for (let i = 0; i <= COUNT; i++) {
-            const colorTex1 = (i === 0 ? brightPass : downSamplePasses[i - 1]).getOutput('color');
-            downSamplePasses[i].createPass(encoder, { colorTex: colorTex1 });
-        }
-        // Upsample with gaussian and blending
-        for (let i = COUNT; i >= 0; i--) {
-            // Blur
-            blurHPasses[i].createPass(encoder, {
-                colorTex: downSamplePasses[i].getOutput('color')
-            });
-            blurVPasses[i].createPass(encoder, {
-                colorTex: blurHPasses[i].getOutput('color')
-            });
+        const passes = [this._brightPass, ...this._downsamplePasses, ...this._blurHPasses,
+            ...this._blurVPasses, ...this._blendPasses];
+        const source = (slot) => slot === passes.length ? colorTex : passes[slot].getOutput('color');
+        for (const [, a, b, target] of BLOOM_PLAN) {
+            passes[target].createPass(encoder, { colorTex: source(a), colorTex1: source(a), colorTex2: source(b) });
         }
-        for (let i = COUNT - 1; i >= 0; i--) {
-            blendPasses[i].createPass(encoder, {
-                colorTex1: (i === COUNT - 1
-                    ? blurVPasses[COUNT]
-                    : blendPasses[i + 1]).getOutput('color'),
-                colorTex2: blurVPasses[i].getOutput('color')
-            });
-        }
-        return blendPasses[0].getOutput('color');
+        return this._blendPasses[0].getOutput('color');
     }
     getMemoryBytes() {
         return [this._brightPass, ...this._downsamplePasses, ...this._blurHPasses, ...this._blurVPasses, ...this._blendPasses]

package/lib/renderer/WGPUBloomPass.js.map

@@ -1 +1 @@
-{"version":3,"file":"WGPUBloomPass.js","sourceRoot":"","sources":["../../src/renderer/WGPUBloomPass.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,gBAAgB,EAAE,MAAM,uBAAuB,CAAC;AAEzD,MAAM,KAAK,GAAG,CAAC,CAAC;AAEhB,MAAM,uBAAuB,GAAG;IAC9B,SAAS,EAAE,QAAQ;IACnB,SAAS,EAAE,QAAQ;CACW,CAAC;AAEjC,SAAS,gBAAgB,CAAC,MAAiB;IACzC,OAAO,IAAI,gBAAgB,CAAC,QAAQ,EAAE,MAAM,EAAE;QAC5C,MAAM,EAAE,CAAC,UAAU,CAAC;QACpB,OAAO,EAAE,CAAC,OAAO,CAAC;QAClB,QAAQ,EAAE;YACR;gBACE,KAAK,EAAE,WAAW;gBAClB,IAAI,EAAE,KAAK;gBACX,IAAI,EAAE,CAAC;aACR;YACD;gBACE,KAAK,EAAE,OAAO;gBACd,IAAI,EAAE,KAAK;gBACX,IAAI,EAAE,CAAC;aACR;SACF;QACD,QAAQ,EAAE,QAAQ,CAAC;;;;CAItB;QACG,MAAM,EAAE,QAAQ,CAAC;EAEnB,EAAE;QACF,oCAAoC;QACpC,4FAA4F;QAC9F;;;;;;;;;;;;;;;;;;;;CAoBC;KACE,CAAC,CAAC;AACL,CAAC;AAED,SAAS,oBAAoB,CAAC,MAAiB;IAC7C,OAAO,IAAI,gBAAgB,CAAC,YAAY,EAAE,MAAM,EAAE;QAChD,MAAM,EAAE,CAAC,UAAU,CAAC;QACpB,OAAO,EAAE,CAAC,OAAO,CAAC;QAClB,QAAQ,EAAE,EAAE;QACZ,QAAQ,EAAE;YACR,aAAa,EAAE,MAAM,CAAC,aAAa,CAAC,uBAAuB,CAAC;SAC7D;QACD,QAAQ,EAAE,QAAQ,CAAC;;;;CAItB;QACG,MAAM,EAAE,QAAQ,CAAC;;;EAInB,EACF;;;;;CAKC;KACE,CAAC,CAAC;AACL,CAAC;AAED,SAAS,kBAAkB,CAAC,MAAiB;IAC3C,OAAO,IAAI,gBAAgB,CAAC,UAAU,EAAE,MAAM,EAAE;QAC9C,MAAM,EAAE,CAAC,UAAU,CAAC;QACpB,OAAO,EAAE,CAAC,OAAO,CAAC;QAClB,QAAQ,EAAE;YACR;gBACE,KAAK,EAAE,WAAW;gBAClB,IAAI,EAAE,OAAO;gBACb,IAAI,EAAE,CAAC,CAAC,EAAE,CAAC,CAAC;aACb;YACD;gBACE,KAAK,EAAE,MAAM;gBACb,IAAI,EAAE,KAAK;gBACX,IAAI,EAAE,GAAG;aACV;SACF;QACD,QAAQ,EAAE;YACR,aAAa,EAAE,MAAM,CAAC,aAAa,CAAC,uBAAuB,CAAC;SAC7D;QACD,QAAQ,EAAE,QAAQ,CAAC;;CAEtB;QACG,MAAM,EAAE,QAAQ,CAAC;;;;;;;;;;;;CAYpB;KACE,CAAC,CAAC;AACL,CAAC;AAED,SAAS,eAAe,CAAC,MAAiB;IACxC,OAAO,IAAI,gBAAgB,CAAC,OAAO,EAAE,MAAM,EAAE;QAC3C,MAAM,EAAE,CAAC,WAAW,EAAE,WAAW,CAAC;QAClC,OAAO,EAAE,CAAC,OAAO,CAAC;QAClB,QAAQ,EAAE;YACR;gBACE,KAAK,EAAE,QAAQ;gBACf,IAAI,EAAE,KAAK;gBACX,IAAI,EAAE,GAAG;aACV;SACF;QAED,QAAQ,EAAE;YACR,aAAa,EAAE,MAAM,CAAC,aAAa,CAAC,uBAAuB,CAAC;SAC7D;QACD,MAAM,EAAE,QAAQ,CAAC;;;;CAIpB;KACE,CAAC,CAAC;AACL,CAAC;AAED,MAAM,aAAa;IACT,WAAW,CAAC;IACZ,iBAAiB,GAA8C,EAAE,CAAC;IAClE,YAAY,GAA4C,EAAE,CAAC;IAC3D,YAAY,GAA4C,EAAE,CAAC;IAC3D,YAAY,GAAyC,EAAE,CAAC;IAEhE,YAAY,MAAiB;QAC3B,IAAI,CAAC,WAAW,GAAG,gBAAgB,CAAC,MAAM,CAAC,CAAC;QAC5C,MAAM,kBAAkB,GAAG;YACzB,KAAK,EAAE;gBACL,MAAM,EAAE,aAAa;aACtB;SACO,CAAC;QACX,IAAI,CAAC,WAAW,CAAC,eAAe,CAAC,kBAAkB,CAAC,CAAC;QACrD,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,IAAI,KAAK,EAAE,CAAC,EAAE,EAAE,CAAC;YAChC,MAAM,cAAc,GAAG,oBAAoB,CAAC,MAAM,CAAC,CAAC;YACpD,MAAM,SAAS,GAAG,kBAAkB,CAAC,MAAM,CAAC,CAAC;YAC7C,MAAM,SAAS,GAAG,kBAAkB,CAAC,MAAM,CAAC,CAAC;YAC7C,cAAc,CAAC,eAAe,CAAC,kBAAkB,CAAC,CAAC;YACnD,SAAS,CAAC,eAAe,CAAC,kBAAkB,CAAC,CAAC;YAC9C,SAAS,CAAC,eAAe,CAAC,kBAAkB,CAAC,CAAC;YAC9C,SAAS,CAAC,UAAU,CAAC,WAAW,EAAE,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,CAAC;YAC1C,SAAS,CAAC,UAAU,CAAC,WAAW,EAAE,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,CAAC;YAC1C,IAAI,CAAC,iBAAiB,CAAC,IAAI,CAAC,cAAc,CAAC,CAAC;YAC5C,IAAI,CAAC,YAAY,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC;YAClC,IAAI,CAAC,YAAY,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC;YAClC,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC;gBACV,MAAM,SAAS,GAAG,eAAe,CAAC,MAAM,CAAC,CAAC;gBAC1C,SAAS,CAAC,eAAe,CAAC,kBAAkB,CAAC,CAAC;gBAC9C,IAAI,CAAC,YAAY,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC;YACpC,CAAC;QACH,CAAC;IACH,CAAC;IAED,eAAe;QACb,CAAC,IAAI,CAAC,WAAW,EAAE,GAAG,IAAI,CAAC,iBAAiB,EAAE,GAAG,IAAI,CAAC,YAAY,EAAE,GAAG,IAAI,CAAC,YAAY,EAAE,GAAG,IAAI,CAAC,YAAY,CAAC,CAAC,OAAO,CAAC,IAAI,CAAC,EAAE,CAAC,IAAI,CAAC,eAAe,EAAE,CAAC,CAAC;IAC1J,CAAC;IAED,OAAO,CAAC,KAAa,EAAE,MAAc;QACnC,IAAI,CAAC,WAAW,CAAC,OAAO,CAAC,KAAK,EAAE,MAAM,CAAC,CAAC;QACxC,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,IAAI,KAAK,EAAE,CAAC,EAAE,EAAE,CAAC;YAChC,IAAI,CAAC,iBAAiB,CAAC,CAAC,CAAC,CAAC,OAAO,CAAC,KAAK,IAAI,CAAC,CAAC,GAAG,CAAC,CAAC,EAAE,MAAM,IAAI,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC;YACvE,IAAI,CAAC,YAAY,CAAC,CAAC,CAAC,CAAC,OAAO,CAAC,KAAK,IAAI,CAAC,CAAC,GAAG,CAAC,CAAC,EAAE,MAAM,IAAI,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC;YAClE,IAAI,CAAC,YAAY,CAAC,CAAC,CAAC,CAAC,OAAO,CAAC,KAAK,IAAI,CAAC,EAAE,MAAM,IAAI,CAAC,CAAC,CAAC;YACtD,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC;gBACV,IAAI,CAAC,YAAY,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,OAAO,CAAC,KAAK,IAAI,CAAC,CAAC,GAAG,CAAC,CAAC,EAAE,MAAM,IAAI,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC;YACxE,CAAC;QACH,CAAC;IACH,CAAC;IAED,kBAAkB,CAAC,SAAiB;QAClC,IAAI,CAAC,WAAW,CAAC,UAAU,CAAC,WAAW,EAAE,SAAS,CAAC,CAAC;IACtD,CAAC;IAED,SAAS,CAAC,MAAc;QACtB,IAAI,CAAC,YAAY,CAAC,OAAO,CAAC,CAAC,SAAS,EAAE,EAAE;YACtC,SAAS,CAAC,UAAU,CAClB,QAAQ;YACR,+BAA+B;YAC/B,CAAC,GAAG,IAAI,CAAC,GAAG,CAAC,IAAI,CAAC,GAAG,CAAC,MAAM,IAAI,GAAG,GAAG,GAAG,EAAE,GAAG,CAAC,EAAE,CAAC,CAAC,CACpD,CAAC;QACJ,CAAC,CAAC,CAAC;QACH,CAAC,GAAG,IAAI,CAAC,YAAY,EAAE,GAAG,IAAI,CAAC,YAAY,CAAC,CAAC,OAAO,CAAC,CAAC,QAAQ,EAAE,EAAE;YAChE,oBAAoB;YACpB,QAAQ,CAAC,UAAU,CAAC,MAAM,EAAE,IAAI,CAAC,GAAG,CAAC,IAAI,CAAC,GAAG,CAAC,MAAM,GAAG,CAAC,EAAE,GAAG,CAAC,EAAE,GAAG,CAAC,CAAC,CAAC;QACxE,CAAC,CAAC,CAAC;IACL,CAAC;IAED,UAAU,CAAC,OAA0B,EAAE,QAAoB;QACzD,MAAM,UAAU,GAAG,IAAI,CAAC,WAAW,CAAC;QACpC,MAAM,gBAAgB,GAAG,IAAI,CAAC,iBAAiB,CAAC;QAChD,MAAM,WAAW,GAAG,IAAI,CAAC,YAAY,CAAC;QACtC,MAAM,WAAW,GAAG,IAAI,CAAC,YAAY,CAAC;QACtC,MAAM,WAAW,GAAG,IAAI,CAAC,YAAY,CAAC;QACtC,UAAU,CAAC,UAAU,CAAC,OAAO,EAAE,EAAE,QAAQ,EAAE,CAAC,CAAC;QAC7C,aAAa;QACb,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,IAAI,KAAK,EAAE,CAAC,EAAE,EAAE,CAAC;YAChC,MAAM,SAAS,GAAG,CAChB,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,UAAU,CAAC,CAAC,CAAC,gBAAgB,CAAC,CAAC,GAAG,CAAC,CAAC,CAC/C,CAAC,SAAS,CAAC,OAAO,CAAC,CAAC;YACrB,gBAAgB,CAAC,CAAC,CAAC,CAAC,UAAU,CAAC,OAAO,EAAE,EAAE,QAAQ,EAAE,SAAS,EAAE,CAAC,CAAC;QACnE,CAAC;QAED,sCAAsC;QACtC,KAAK,IAAI,CAAC,GAAG,KAAK,EAAE,CAAC,IAAI,CAAC,EAAE,CAAC,EAAE,EAAE,CAAC;YAChC,OAAO;YACP,WAAW,CAAC,CAAC,CAAC,CAAC,UAAU,CAAC,OAAO,EAAE;gBACjC,QAAQ,EAAE,gBAAgB,CAAC,CAAC,CAAC,CAAC,SAAS,CAAC,OAAO,CAAC;aACjD,CAAC,CAAC;YACH,WAAW,CAAC,CAAC,CAAC,CAAC,UAAU,CAAC,OAAO,EAAE;gBACjC,QAAQ,EAAE,WAAW,CAAC,CAAC,CAAC,CAAC,SAAS,CAAC,OAAO,CAAC;aAC5C,CAAC,CAAC;QACL,CAAC;QACD,KAAK,IAAI,CAAC,GAAG,KAAK,GAAG,CAAC,EAAE,CAAC,IAAI,CAAC,EAAE,CAAC,EAAE,EAAE,CAAC;YACpC,WAAW,CAAC,CAAC,CAAC,CAAC,UAAU,CAAC,OAAO,EAAE;gBACjC,SAAS,EAAE,CAAC,CAAC,KAAK,KAAK,GAAG,CAAC;oBACzB,CAAC,CAAC,WAAW,CAAC,KAAK,CAAC;oBACpB,CAAC,CAAC,WAAW,CAAC,CAAC,GAAG,CAAC,CAAC,CACrB,CAAC,SAAS,CAAC,OAAO,CAAC;gBACpB,SAAS,EAAE,WAAW,CAAC,CAAC,CAAC,CAAC,SAAS,CAAC,OAAO,CAAC;aAC7C,CAAC,CAAC;QACL,CAAC;QAED,OAAO,WAAW,CAAC,CAAC,CAAC,CAAC,SAAS,CAAC,OAAO,CAAC,CAAC;IAC3C,CAAC;IAED,cAAc;QACZ,OAAO,CAAC,IAAI,CAAC,WAAW,EAAE,GAAG,IAAI,CAAC,iBAAiB,EAAE,GAAG,IAAI,CAAC,YAAY,EAAE,GAAG,IAAI,CAAC,YAAY,EAAE,GAAG,IAAI,CAAC,YAAY,CAAC;aACnH,MAAM,CAAC,CAAC,GAAG,EAAE,IAAI,EAAE,EAAE,CAAC,GAAG,GAAG,IAAI,CAAC,cAAc,EAAE,EAAE,CAAC,CAAC,CAAC;IAC3D,CAAC;IAED,OAAO;QACL,IAAI,CAAC,WAAW,CAAC,OAAO,EAAE,CAAC;QAC3B,CAAC,GAAG,IAAI,CAAC,iBAAiB,EAAE,GAAG,IAAI,CAAC,YAAY,EAAE,GAAG,IAAI,CAAC,YAAY,EAAE,GAAG,IAAI,CAAC,YAAY,CAAC,CAAC,OAAO,CAAC,IAAI,CAAC,EAAE,CAAC,IAAI,CAAC,OAAO,EAAE,CAAC,CAAC;IAChI,CAAC;IAED,SAAS;QACP,OAAO,IAAI,CAAC,YAAY,CAAC,CAAC,CAAC,CAAC,SAAS,CAAC,OAAO,CAAC,CAAC;IACjD,CAAC;CACF;AAED,eAAe,aAAa,CAAC","sourcesContent":["import { WGPUFullQuadPass } from './WGPUFullQuadPass.js';\n\nconst COUNT = 5;\n\nconst linearSamplerDescriptor = {\n  magFilter: 'linear',\n  minFilter: 'linear'\n} satisfies GPUSamplerDescriptor;\n\nfunction createBrightPass(device: GPUDevice) {\n  return new WGPUFullQuadPass('bright', device, {\n    inputs: ['colorTex'],\n    outputs: ['color'],\n    uniforms: [\n      {\n        label: 'threshold',\n        type: 'f32',\n        data: 1\n      },\n      {\n        label: 'scale',\n        type: 'f32',\n        data: 1\n      }\n    ],\n    fsDefine: /*wgsl*/ `\nfn median(a: vec4f, b: vec4f, c: vec4f) -> vec4f {\nreturn a + b + c - min(min(a, b), c) - max(max(a, b), c);\n}\n`,\n    fsMain: /*wgsl*/ `\n${\n  ''\n  // Use median filter to reduce noise\n  // https://github.com/keijiro/KinoBloom/blob/master/Assets/Kino/Bloom/Shader/Bloom.cginc#L96\n}\nlet d = vec3i(1, 1, 0);\n\nvar texel = load_colorTex(uv, 0);\nlet s1 = load_colorTex(uv - d.xz, 0);\nlet s2 = load_colorTex(uv + d.xz, 0);\nlet s3 = load_colorTex(uv - d.zy, 0);\nlet s4 = load_colorTex(uv + d.zy, 0);\ntexel = median(median(texel, s1, s2), s3, s4);\n\n// Premultiplied Alpha\nlet lum = dot(texel.rgb, vec3f(0.2125, 0.7154, 0.0721));\nvar color: vec4f;\nif (lum > threshold && texel.a > 0.0) {\n  color = vec4f(texel.rgb * scale, texel.a);\n} else {\n  // Bleed the alpha to avoid sharp edge when using alpha\n  color = vec4f(vec3(0.0), texel.a);\n}\noutput.color = color;\n`\n  });\n}\n\nfunction createDownsamplePass(device: GPUDevice) {\n  return new WGPUFullQuadPass('downsample', device, {\n    inputs: ['colorTex'],\n    outputs: ['color'],\n    uniforms: [],\n    samplers: {\n      linearSampler: device.createSampler(linearSamplerDescriptor)\n    },\n    fsDefine: /*wgsl*/ `\nfn brightness(c: vec3f) -> f32 {\n  return max(max(c.r, c.g), c.b);\n}\n`,\n    fsMain: /*wgsl*/ `\nlet size = vec2f(colorTexRegion());\nlet d = vec4f(-1.0, -1.0, 1.0, 1.0) / size.xyxy;\n${\n  ''\n}\noutput.color = sample_colorTex_linearSampler(uv2 + d.xy) * 0.25\n  + sample_colorTex_linearSampler(uv2 + d.zy) * 0.25\n  + sample_colorTex_linearSampler(uv2 + d.xw) * 0.25\n  + sample_colorTex_linearSampler(uv2 + d.zw) * 0.25;\n`\n  });\n}\n\nfunction createGaussianPass(device: GPUDevice) {\n  return new WGPUFullQuadPass('gaussian', device, {\n    inputs: ['colorTex'],\n    outputs: ['color'],\n    uniforms: [\n      {\n        label: 'direction',\n        type: 'vec2f',\n        data: [1, 0]\n      },\n      {\n        label: 'size',\n        type: 'f32',\n        data: 1.0\n      }\n    ],\n    samplers: {\n      linearSampler: device.createSampler(linearSamplerDescriptor)\n    },\n    fsDefine: /*wgsl*/ `\nconst kernel = array<f32, 9>(0.07, 0.09, 0.12, 0.14, 0.16, 0.14, 0.12, 0.09, 0.07);\n`,\n    fsMain: /*wgsl*/ `\nlet texSize = vec2f(colorTexRegion());\nlet off = direction / texSize * size;\n\nvar sum = vec4f(0.0);\nvar weightAll = 0.0;\nfor (var i = 0; i < 9; i++) {\n  let w = kernel[i];\n  sum += sample_colorTex_linearSampler(uv2 + off * f32(i - 4)) * w;\n  weightAll += w;\n}\noutput.color = sum / max(weightAll, 0.01);\n`\n  });\n}\n\nfunction createBlendPass(device: GPUDevice) {\n  return new WGPUFullQuadPass('blend', device, {\n    inputs: ['colorTex1', 'colorTex2'],\n    outputs: ['color'],\n    uniforms: [\n      {\n        label: 'weight',\n        type: 'f32',\n        data: 0.3\n      }\n    ],\n\n    samplers: {\n      linearSampler: device.createSampler(linearSamplerDescriptor)\n    },\n    fsMain: /*wgsl*/ `\nlet color1 = sample_colorTex1_linearSampler(uv2);\nlet color2 = sample_colorTex2_linearSampler(uv2);\noutput.color = mix(color1, color2, weight);\n`\n  });\n}\n\nclass WGPUBloomPass {\n  private _brightPass;\n  private _downsamplePasses: ReturnType<typeof createDownsamplePass>[] = [];\n  private _blurHPasses: ReturnType<typeof createGaussianPass>[] = [];\n  private _blurVPasses: ReturnType<typeof createGaussianPass>[] = [];\n  private _blendPasses: ReturnType<typeof createBlendPass>[] = [];\n\n  constructor(device: GPUDevice) {\n    this._brightPass = createBrightPass(device);\n    const commonOutputParams = {\n      color: {\n        format: 'rgba16float'\n      }\n    } as const;\n    this._brightPass.setOutputParams(commonOutputParams);\n    for (let i = 0; i <= COUNT; i++) {\n      const downsamplePass = createDownsamplePass(device);\n      const blurHPass = createGaussianPass(device);\n      const blurVPass = createGaussianPass(device);\n      downsamplePass.setOutputParams(commonOutputParams);\n      blurHPass.setOutputParams(commonOutputParams);\n      blurVPass.setOutputParams(commonOutputParams);\n      blurHPass.setUniform('direction', [1, 0]);\n      blurVPass.setUniform('direction', [0, 1]);\n      this._downsamplePasses.push(downsamplePass);\n      this._blurHPasses.push(blurHPass);\n      this._blurVPasses.push(blurVPass);\n      if (i > 0) {\n        const blendPass = createBlendPass(device);\n        blendPass.setOutputParams(commonOutputParams);\n        this._blendPasses.push(blendPass);\n      }\n    }\n  }\n\n  reserveCapacity() {\n    [this._brightPass, ...this._downsamplePasses, ...this._blurHPasses, ...this._blurVPasses, ...this._blendPasses].forEach(pass => pass.reserveCapacity());\n  }\n\n  setSize(width: number, height: number) {\n    this._brightPass.setSize(width, height);\n    for (let i = 0; i <= COUNT; i++) {\n      this._downsamplePasses[i].setSize(width >> (i + 1), height >> (i + 1));\n      this._blurHPasses[i].setSize(width >> (i + 1), height >> (i + 1));\n      this._blurVPasses[i].setSize(width >> i, height >> i);\n      if (i > 0) {\n        this._blendPasses[i - 1].setSize(width >> (i - 1), height >> (i - 1));\n      }\n    }\n  }\n\n  setBrightThreshold(threshold: number) {\n    this._brightPass.setUniform('threshold', threshold);\n  }\n\n  setRadius(radius: number) {\n    this._blendPasses.forEach((blendPass) => {\n      blendPass.setUniform(\n        'weight',\n        // Use weight to control radius\n        1 - Math.max(Math.min(radius ** 0.3 * 0.8, 0.8), 0)\n      );\n    });\n    [...this._blurHPasses, ...this._blurVPasses].forEach((blurPass) => {\n      //Some magic fomular\n      blurPass.setUniform('size', Math.max(Math.min(radius * 2, 1.0), 0.2));\n    });\n  }\n\n  createPass(encoder: GPUCommandEncoder, colorTex: GPUTexture) {\n    const brightPass = this._brightPass;\n    const downSamplePasses = this._downsamplePasses;\n    const blendPasses = this._blendPasses;\n    const blurHPasses = this._blurHPasses;\n    const blurVPasses = this._blurVPasses;\n    brightPass.createPass(encoder, { colorTex });\n    // Downsample\n    for (let i = 0; i <= COUNT; i++) {\n      const colorTex1 = (\n        i === 0 ? brightPass : downSamplePasses[i - 1]\n      ).getOutput('color');\n      downSamplePasses[i].createPass(encoder, { colorTex: colorTex1 });\n    }\n\n    // Upsample with gaussian and blending\n    for (let i = COUNT; i >= 0; i--) {\n      // Blur\n      blurHPasses[i].createPass(encoder, {\n        colorTex: downSamplePasses[i].getOutput('color')\n      });\n      blurVPasses[i].createPass(encoder, {\n        colorTex: blurHPasses[i].getOutput('color')\n      });\n    }\n    for (let i = COUNT - 1; i >= 0; i--) {\n      blendPasses[i].createPass(encoder, {\n        colorTex1: (i === COUNT - 1\n          ? blurVPasses[COUNT]\n          : blendPasses[i + 1]\n        ).getOutput('color'),\n        colorTex2: blurVPasses[i].getOutput('color')\n      });\n    }\n\n    return blendPasses[0].getOutput('color');\n  }\n\n  getMemoryBytes() {\n    return [this._brightPass, ...this._downsamplePasses, ...this._blurHPasses, ...this._blurVPasses, ...this._blendPasses]\n      .reduce((sum, pass) => sum + pass.getMemoryBytes(), 0);\n  }\n\n  dispose() {\n    this._brightPass.dispose();\n    [...this._downsamplePasses, ...this._blurHPasses, ...this._blurVPasses, ...this._blendPasses].forEach(pass => pass.dispose());\n  }\n\n  getOutput() {\n    return this._blendPasses[0].getOutput('color');\n  }\n}\n\nexport default WGPUBloomPass;\n"]}
+{"version":3,"file":"WGPUBloomPass.js","sourceRoot":"","sources":["../../src/renderer/WGPUBloomPass.ts"],"names":[],"mappings":"AAAA,OAAO,EAAC,UAAU,EAAE,YAAY,EAAE,aAAa,EAAE,gBAAgB,EAAE,iBAAiB,EAAE,eAAe,EAAE,eAAe,EAAE,aAAa,EAAE,eAAe,EAAE,aAAa,EAAE,cAAc,EAAC,MAAM,sBAAsB,CAAC;AACnN,OAAO,EAAE,gBAAgB,EAAE,MAAM,uBAAuB,CAAC;AAEzD,MAAM,KAAK,GAAG,YAAY,GAAG,CAAC,CAAC;AAE/B,MAAM,uBAAuB,GAAG;IAC9B,SAAS,EAAE,QAAQ;IACnB,SAAS,EAAE,QAAQ;CACW,CAAC;AAEjC,SAAS,gBAAgB,CAAC,MAAiB;IACzC,OAAO,IAAI,gBAAgB,CAAC,QAAQ,EAAE,MAAM,EAAE;QAC5C,MAAM,EAAE,CAAC,UAAU,CAAC;QACpB,OAAO,EAAE,CAAC,OAAO,CAAC;QAClB,QAAQ,EAAE;YACR;gBACE,KAAK,EAAE,WAAW;gBAClB,IAAI,EAAE,KAAK;gBACX,IAAI,EAAE,CAAC;aACR;YACD;gBACE,KAAK,EAAE,OAAO;gBACd,IAAI,EAAE,KAAK;gBACX,IAAI,EAAE,CAAC;aACR;SACF;QACD,QAAQ,EAAE,iBAAiB;QAC3B,MAAM,EAAE,eAAe;KACxB,CAAC,CAAC;AACL,CAAC;AAED,SAAS,oBAAoB,CAAC,MAAiB;IAC7C,OAAO,IAAI,gBAAgB,CAAC,YAAY,EAAE,MAAM,EAAE;QAChD,MAAM,EAAE,CAAC,UAAU,CAAC;QACpB,OAAO,EAAE,CAAC,OAAO,CAAC;QAClB,QAAQ,EAAE,EAAE;QACZ,QAAQ,EAAE;YACR,aAAa,EAAE,MAAM,CAAC,aAAa,CAAC,uBAAuB,CAAC;SAC7D;QACD,QAAQ,EAAE,eAAe;QACzB,MAAM,EAAE,aAAa;KACtB,CAAC,CAAC;AACL,CAAC;AAED,SAAS,kBAAkB,CAAC,MAAiB;IAC3C,OAAO,IAAI,gBAAgB,CAAC,UAAU,EAAE,MAAM,EAAE;QAC9C,MAAM,EAAE,CAAC,UAAU,CAAC;QACpB,OAAO,EAAE,CAAC,OAAO,CAAC;QAClB,QAAQ,EAAE;YACR;gBACE,KAAK,EAAE,WAAW;gBAClB,IAAI,EAAE,OAAO;gBACb,IAAI,EAAE,CAAC,CAAC,EAAE,CAAC,CAAC;aACb;YACD;gBACE,KAAK,EAAE,MAAM;gBACb,IAAI,EAAE,KAAK;gBACX,IAAI,EAAE,GAAG;aACV;SACF;QACD,QAAQ,EAAE;YACR,aAAa,EAAE,MAAM,CAAC,aAAa,CAAC,uBAAuB,CAAC;SAC7D;QACD,QAAQ,EAAE,eAAe;QACzB,MAAM,EAAE,aAAa;KACtB,CAAC,CAAC;AACL,CAAC;AAED,SAAS,eAAe,CAAC,MAAiB;IACxC,OAAO,IAAI,gBAAgB,CAAC,OAAO,EAAE,MAAM,EAAE;QAC3C,MAAM,EAAE,CAAC,WAAW,EAAE,WAAW,CAAC;QAClC,OAAO,EAAE,CAAC,OAAO,CAAC;QAClB,QAAQ,EAAE;YACR;gBACE,KAAK,EAAE,QAAQ;gBACf,IAAI,EAAE,KAAK;gBACX,IAAI,EAAE,GAAG;aACV;SACF;QAED,QAAQ,EAAE;YACR,aAAa,EAAE,MAAM,CAAC,aAAa,CAAC,uBAAuB,CAAC;SAC7D;QACD,MAAM,EAAE,cAAc;KACvB,CAAC,CAAC;AACL,CAAC;AAED,MAAM,aAAa;IACT,WAAW,CAAC;IACZ,iBAAiB,GAA8C,EAAE,CAAC;IAClE,YAAY,GAA4C,EAAE,CAAC;IAC3D,YAAY,GAA4C,EAAE,CAAC;IAC3D,YAAY,GAAyC,EAAE,CAAC;IAEhE,YAAY,MAAiB;QAC3B,IAAI,CAAC,WAAW,GAAG,gBAAgB,CAAC,MAAM,CAAC,CAAC;QAC5C,MAAM,kBAAkB,GAAG;YACzB,KAAK,EAAE;gBACL,MAAM,EAAE,aAAa;aACtB;SACO,CAAC;QACX,IAAI,CAAC,WAAW,CAAC,eAAe,CAAC,kBAAkB,CAAC,CAAC;QACrD,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,IAAI,KAAK,EAAE,CAAC,EAAE,EAAE,CAAC;YAChC,MAAM,cAAc,GAAG,oBAAoB,CAAC,MAAM,CAAC,CAAC;YACpD,MAAM,SAAS,GAAG,kBAAkB,CAAC,MAAM,CAAC,CAAC;YAC7C,MAAM,SAAS,GAAG,kBAAkB,CAAC,MAAM,CAAC,CAAC;YAC7C,cAAc,CAAC,eAAe,CAAC,kBAAkB,CAAC,CAAC;YACnD,SAAS,CAAC,eAAe,CAAC,kBAAkB,CAAC,CAAC;YAC9C,SAAS,CAAC,eAAe,CAAC,kBAAkB,CAAC,CAAC;YAC9C,SAAS,CAAC,UAAU,CAAC,WAAW,EAAE,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,CAAC;YAC1C,SAAS,CAAC,UAAU,CAAC,WAAW,EAAE,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,CAAC;YAC1C,IAAI,CAAC,iBAAiB,CAAC,IAAI,CAAC,cAAc,CAAC,CAAC;YAC5C,IAAI,CAAC,YAAY,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC;YAClC,IAAI,CAAC,YAAY,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC;YAClC,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC;gBACV,MAAM,SAAS,GAAG,eAAe,CAAC,MAAM,CAAC,CAAC;gBAC1C,SAAS,CAAC,eAAe,CAAC,kBAAkB,CAAC,CAAC;gBAC9C,IAAI,CAAC,YAAY,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC;YACpC,CAAC;QACH,CAAC;IACH,CAAC;IAED,eAAe;QACb,CAAC,IAAI,CAAC,WAAW,EAAE,GAAG,IAAI,CAAC,iBAAiB,EAAE,GAAG,IAAI,CAAC,YAAY,EAAE,GAAG,IAAI,CAAC,YAAY,EAAE,GAAG,IAAI,CAAC,YAAY,CAAC,CAAC,OAAO,CAAC,IAAI,CAAC,EAAE,CAAC,IAAI,CAAC,eAAe,EAAE,CAAC,CAAC;IAC1J,CAAC;IAED,OAAO,CAAC,KAAa,EAAE,MAAc;QACnC,IAAI,CAAC,WAAW,CAAC,OAAO,CAAC,KAAK,EAAE,MAAM,CAAC,CAAC;QACxC,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,IAAI,KAAK,EAAE,CAAC,EAAE,EAAE,CAAC;YAChC,IAAI,CAAC,iBAAiB,CAAC,CAAC,CAAC,CAAC,OAAO,CAAC,KAAK,IAAI,CAAC,CAAC,GAAG,CAAC,CAAC,EAAE,MAAM,IAAI,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC;YACvE,IAAI,CAAC,YAAY,CAAC,CAAC,CAAC,CAAC,OAAO,CAAC,KAAK,IAAI,CAAC,CAAC,GAAG,CAAC,CAAC,EAAE,MAAM,IAAI,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC;YAClE,IAAI,CAAC,YAAY,CAAC,CAAC,CAAC,CAAC,OAAO,CAAC,KAAK,IAAI,CAAC,EAAE,MAAM,IAAI,CAAC,CAAC,CAAC;YACtD,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC;gBACV,IAAI,CAAC,YAAY,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,OAAO,CAAC,KAAK,IAAI,CAAC,CAAC,GAAG,CAAC,CAAC,EAAE,MAAM,IAAI,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC;YACxE,CAAC;QACH,CAAC;IACH,CAAC;IAED,kBAAkB,CAAC,SAAiB;QAClC,IAAI,CAAC,WAAW,CAAC,UAAU,CAAC,WAAW,EAAE,SAAS,CAAC,CAAC;IACtD,CAAC;IAED,SAAS,CAAC,MAAc;QACtB,IAAI,CAAC,YAAY,CAAC,OAAO,CAAC,CAAC,SAAS,EAAE,EAAE;YACtC,SAAS,CAAC,UAAU,CAClB,QAAQ;YACR,+BAA+B;YAC/B,gBAAgB,CAAC,MAAM,CAAC,CACzB,CAAC;QACJ,CAAC,CAAC,CAAC;QACH,CAAC,GAAG,IAAI,CAAC,YAAY,EAAE,GAAG,IAAI,CAAC,YAAY,CAAC,CAAC,OAAO,CAAC,CAAC,QAAQ,EAAE,EAAE;YAChE,oBAAoB;YACpB,QAAQ,CAAC,UAAU,CAAC,MAAM,EAAE,aAAa,CAAC,MAAM,CAAC,CAAC,CAAC;QACrD,CAAC,CAAC,CAAC;IACL,CAAC;IAED,UAAU,CAAC,OAA0B,EAAE,QAAoB;QACzD,MAAM,MAAM,GAAG,CAAC,IAAI,CAAC,WAAW,EAAE,GAAG,IAAI,CAAC,iBAAiB,EAAE,GAAG,IAAI,CAAC,YAAY;YAC/E,GAAG,IAAI,CAAC,YAAY,EAAE,GAAG,IAAI,CAAC,YAAY,CAAC,CAAC;QAC9C,MAAM,MAAM,GAAG,CAAC,IAAY,EAAE,EAAE,CAAC,IAAI,KAAK,MAAM,CAAC,MAAM,CAAC,CAAC,CAAC,QAAQ,CAAC,CAAC,CAAC,MAAM,CAAC,IAAI,CAAC,CAAC,SAAS,CAAC,OAAO,CAAC,CAAC;QACrG,KAAK,MAAM,CAAC,EAAE,CAAC,EAAE,CAAC,EAAE,MAAM,CAAC,IAAI,UAAU,EAAE,CAAC;YAC1C,MAAM,CAAC,MAAM,CAAC,CAAC,UAAU,CAAC,OAAO,EAC/B,EAAC,QAAQ,EAAE,MAAM,CAAC,CAAC,CAAC,EAAE,SAAS,EAAE,MAAM,CAAC,CAAC,CAAC,EAAE,SAAS,EAAE,MAAM,CAAC,CAAC,CAAC,EAAC,CAAC,CAAC;QACvE,CAAC;QACD,OAAO,IAAI,CAAC,YAAY,CAAC,CAAC,CAAC,CAAC,SAAS,CAAC,OAAO,CAAC,CAAC;IACjD,CAAC;IAED,cAAc;QACZ,OAAO,CAAC,IAAI,CAAC,WAAW,EAAE,GAAG,IAAI,CAAC,iBAAiB,EAAE,GAAG,IAAI,CAAC,YAAY,EAAE,GAAG,IAAI,CAAC,YAAY,EAAE,GAAG,IAAI,CAAC,YAAY,CAAC;aACnH,MAAM,CAAC,CAAC,GAAG,EAAE,IAAI,EAAE,EAAE,CAAC,GAAG,GAAG,IAAI,CAAC,cAAc,EAAE,EAAE,CAAC,CAAC,CAAC;IAC3D,CAAC;IAED,OAAO;QACL,IAAI,CAAC,WAAW,CAAC,OAAO,EAAE,CAAC;QAC3B,CAAC,GAAG,IAAI,CAAC,iBAAiB,EAAE,GAAG,IAAI,CAAC,YAAY,EAAE,GAAG,IAAI,CAAC,YAAY,EAAE,GAAG,IAAI,CAAC,YAAY,CAAC,CAAC,OAAO,CAAC,IAAI,CAAC,EAAE,CAAC,IAAI,CAAC,OAAO,EAAE,CAAC,CAAC;IAChI,CAAC;IAED,SAAS;QACP,OAAO,IAAI,CAAC,YAAY,CAAC,CAAC,CAAC,CAAC,SAAS,CAAC,OAAO,CAAC,CAAC;IACjD,CAAC;CACF;AAED,eAAe,aAAa,CAAC","sourcesContent":["import {BLOOM_PLAN, BLOOM_LEVELS, bloomBlurSize, bloomBlendWeight, bloomBrightDefine, bloomBrightMain, bloomDownDefine, bloomDownMain, bloomBlurDefine, bloomBlurMain, bloomBlendMain} from './wgsl/bloom.wgsl.js';\nimport { WGPUFullQuadPass } from './WGPUFullQuadPass.js';\n\nconst COUNT = BLOOM_LEVELS - 1;\n\nconst linearSamplerDescriptor = {\n  magFilter: 'linear',\n  minFilter: 'linear'\n} satisfies GPUSamplerDescriptor;\n\nfunction createBrightPass(device: GPUDevice) {\n  return new WGPUFullQuadPass('bright', device, {\n    inputs: ['colorTex'],\n    outputs: ['color'],\n    uniforms: [\n      {\n        label: 'threshold',\n        type: 'f32',\n        data: 1\n      },\n      {\n        label: 'scale',\n        type: 'f32',\n        data: 1\n      }\n    ],\n    fsDefine: bloomBrightDefine,\n    fsMain: bloomBrightMain\n  });\n}\n\nfunction createDownsamplePass(device: GPUDevice) {\n  return new WGPUFullQuadPass('downsample', device, {\n    inputs: ['colorTex'],\n    outputs: ['color'],\n    uniforms: [],\n    samplers: {\n      linearSampler: device.createSampler(linearSamplerDescriptor)\n    },\n    fsDefine: bloomDownDefine,\n    fsMain: bloomDownMain\n  });\n}\n\nfunction createGaussianPass(device: GPUDevice) {\n  return new WGPUFullQuadPass('gaussian', device, {\n    inputs: ['colorTex'],\n    outputs: ['color'],\n    uniforms: [\n      {\n        label: 'direction',\n        type: 'vec2f',\n        data: [1, 0]\n      },\n      {\n        label: 'size',\n        type: 'f32',\n        data: 1.0\n      }\n    ],\n    samplers: {\n      linearSampler: device.createSampler(linearSamplerDescriptor)\n    },\n    fsDefine: bloomBlurDefine,\n    fsMain: bloomBlurMain\n  });\n}\n\nfunction createBlendPass(device: GPUDevice) {\n  return new WGPUFullQuadPass('blend', device, {\n    inputs: ['colorTex1', 'colorTex2'],\n    outputs: ['color'],\n    uniforms: [\n      {\n        label: 'weight',\n        type: 'f32',\n        data: 0.3\n      }\n    ],\n\n    samplers: {\n      linearSampler: device.createSampler(linearSamplerDescriptor)\n    },\n    fsMain: bloomBlendMain\n  });\n}\n\nclass WGPUBloomPass {\n  private _brightPass;\n  private _downsamplePasses: ReturnType<typeof createDownsamplePass>[] = [];\n  private _blurHPasses: ReturnType<typeof createGaussianPass>[] = [];\n  private _blurVPasses: ReturnType<typeof createGaussianPass>[] = [];\n  private _blendPasses: ReturnType<typeof createBlendPass>[] = [];\n\n  constructor(device: GPUDevice) {\n    this._brightPass = createBrightPass(device);\n    const commonOutputParams = {\n      color: {\n        format: 'rgba16float'\n      }\n    } as const;\n    this._brightPass.setOutputParams(commonOutputParams);\n    for (let i = 0; i <= COUNT; i++) {\n      const downsamplePass = createDownsamplePass(device);\n      const blurHPass = createGaussianPass(device);\n      const blurVPass = createGaussianPass(device);\n      downsamplePass.setOutputParams(commonOutputParams);\n      blurHPass.setOutputParams(commonOutputParams);\n      blurVPass.setOutputParams(commonOutputParams);\n      blurHPass.setUniform('direction', [1, 0]);\n      blurVPass.setUniform('direction', [0, 1]);\n      this._downsamplePasses.push(downsamplePass);\n      this._blurHPasses.push(blurHPass);\n      this._blurVPasses.push(blurVPass);\n      if (i > 0) {\n        const blendPass = createBlendPass(device);\n        blendPass.setOutputParams(commonOutputParams);\n        this._blendPasses.push(blendPass);\n      }\n    }\n  }\n\n  reserveCapacity() {\n    [this._brightPass, ...this._downsamplePasses, ...this._blurHPasses, ...this._blurVPasses, ...this._blendPasses].forEach(pass => pass.reserveCapacity());\n  }\n\n  setSize(width: number, height: number) {\n    this._brightPass.setSize(width, height);\n    for (let i = 0; i <= COUNT; i++) {\n      this._downsamplePasses[i].setSize(width >> (i + 1), height >> (i + 1));\n      this._blurHPasses[i].setSize(width >> (i + 1), height >> (i + 1));\n      this._blurVPasses[i].setSize(width >> i, height >> i);\n      if (i > 0) {\n        this._blendPasses[i - 1].setSize(width >> (i - 1), height >> (i - 1));\n      }\n    }\n  }\n\n  setBrightThreshold(threshold: number) {\n    this._brightPass.setUniform('threshold', threshold);\n  }\n\n  setRadius(radius: number) {\n    this._blendPasses.forEach((blendPass) => {\n      blendPass.setUniform(\n        'weight',\n        // Use weight to control radius\n        bloomBlendWeight(radius)\n      );\n    });\n    [...this._blurHPasses, ...this._blurVPasses].forEach((blurPass) => {\n      //Some magic fomular\n      blurPass.setUniform('size', bloomBlurSize(radius));\n    });\n  }\n\n  createPass(encoder: GPUCommandEncoder, colorTex: GPUTexture) {\n    const passes = [this._brightPass, ...this._downsamplePasses, ...this._blurHPasses,\n      ...this._blurVPasses, ...this._blendPasses];\n    const source = (slot: number) => slot === passes.length ? colorTex : passes[slot].getOutput('color');\n    for (const [, a, b, target] of BLOOM_PLAN) {\n      passes[target].createPass(encoder,\n        {colorTex: source(a), colorTex1: source(a), colorTex2: source(b)});\n    }\n    return this._blendPasses[0].getOutput('color');\n  }\n\n  getMemoryBytes() {\n    return [this._brightPass, ...this._downsamplePasses, ...this._blurHPasses, ...this._blurVPasses, ...this._blendPasses]\n      .reduce((sum, pass) => sum + pass.getMemoryBytes(), 0);\n  }\n\n  dispose() {\n    this._brightPass.dispose();\n    [...this._downsamplePasses, ...this._blurHPasses, ...this._blurVPasses, ...this._blendPasses].forEach(pass => pass.dispose());\n  }\n\n  getOutput() {\n    return this._blendPasses[0].getOutput('color');\n  }\n}\n\nexport default WGPUBloomPass;\n"]}

package/lib/renderer/WGPUComputePass.d.ts

@@ -1,4 +1,4 @@
-import { Uniform } from './WGPUFullQuadPass.js';
+import { type Uniform } from './WGPUFullQuadPass.js';
 export type { Uniform };
 export declare const WORKGROUP_SIZE = 8;
 export interface WGPUComputePassInput {

package/lib/renderer/WGPUComputePass.js.map

@@ -1 +1 @@
-{"version":3,"file":"WGPUComputePass.js","sourceRoot":"","sources":["../../src/renderer/WGPUComputePass.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,oBAAoB,EAAE,MAAM,sBAAsB,CAAC;AAC5D,gCAAgC;AAChC,OAAO,EAAE,gBAAgB,EAAW,MAAM,uBAAuB,CAAC;AAIlE,SAAS,oBAAoB,CAC3B,MAA6B,EAC7B,KAA4B;IAE5B,OAAO,CACL,MAAM,CAAC,QAAQ,KAAK,KAAK,CAAC,QAAQ;QAClC,MAAM,CAAC,IAAI,KAAK,KAAK,CAAC,IAAI;QAC1B,MAAM,CAAC,KAAK,KAAK,KAAK,CAAC,KAAK;QAC5B,MAAM,CAAC,MAAM,KAAK,KAAK,CAAC,MAAM,CAC/B,CAAC;AACJ,CAAC;AAED,MAAM,CAAC,MAAM,cAAc,GAAG,CAAC,CAAC;AAiBhC,MAAM,OAAO,eAAe;IAClB,MAAM,CAAC;IAEP,OAAO,CAAC;IACR,cAAc,GAA8B,EAAE,CAAC;IAC/C,aAAa,GAA0C,EAAE,CAAC;IAE1D,SAAS,CAAsB;IAC/B,WAAW,GAAmB,EAAE,CAAC;IACjC,oBAAoB,GAAG,IAAI,CAAC;IAE5B,OAAO,GAAa,EAAE,CAAC;IACvB,QAAQ,GAAa,EAAE,CAAC;IACxB,SAAS,GAAc,EAAE,CAAC;IAC1B,YAAY,GAA4B,EAAE,CAAC;IAC3C,eAAe,GAA8B,EAAE,CAAC;IAEhD,cAAc,GAAG,CAAC,CAAC;IACnB,eAAe,GAAG,CAAC,CAAC;IACpB,gBAAgB,GAAG,KAAK,CAAC;IACzB,MAAM,GAAG,EAAE,CAAC;IACZ,OAAO,GAAG,EAAE,CAAC;IAEb,UAAU,CAAU;IACpB,WAAW,CAAU;IAErB,cAAc,GAAG,CAAC,cAAc,EAAE,cAAc,CAAC,CAAC;IAElD,OAAO,GAAG,EAAE,CAAC;IACb,OAAO,CAAC;IACR,SAAS,CAAC;IAEV,aAAa,GAAG;QACtB,MAAM,EAAE,CAAC;QACT,QAAQ,EAAE,CAAC;QACX,OAAO,EAAE,CAAC;KACX,CAAC;IAEF,YACE,KAAa,EACb,MAAiB,EACjB,IAMC;QAED,IAAI,CAAC,MAAM,GAAG,KAAK,CAAC;QACpB,IAAI,CAAC,OAAO,GAAG,MAAM,CAAC;QACtB,IAAI,CAAC,OAAO,GAAG,IAAI,CAAC,MAAM,CAAC;QAC3B,IAAI,CAAC,SAAS,GAAG,IAAI,CAAC,QAAQ,CAAC;QAC/B,IAAI,CAAC,OAAO,GAAG,IAAI,CAAC,MAAM,CAAC;QAC3B,IAAI,CAAC,QAAQ,GAAG,IAAI,CAAC,OAAO,CAAC;QAC7B,MAAM,QAAQ,GAAG,IAAI,CAAC,QAAQ,IAAI,EAAE,CAAC;QACrC,IAAI,CAAC,SAAS,GAAG,QAAQ,CAAC;QAE1B,IAAI,CAAC,YAAY,GAAG,IAAI,CAAC,SAAS,CAAC,MAAM,CAAC,CAAC,GAAG,EAAE,OAAO,EAAE,EAAE;YACzD,GAAG,CAAC,OAAO,CAAC,KAAK,CAAC,GAAG,OAAO,CAAC;YAC7B,OAAO,GAAG,CAAC;QACb,CAAC,EAAE,EAA6B,CAAC,CAAC;QAClC,QAAQ,CAAC,OAAO,CAAC,CAAC,OAAO,EAAE,EAAE;YAC3B,MAAM,IAAI,GAAG,gBAAgB,CAAC,OAAO,CAAC,IAAI,EAAE,OAAO,CAAC,IAAI,CAAC,CAAC;YAC1D,IAAI,CAAC,eAAe,CAAC,OAAO,CAAC,KAAK,CAAC,GAAG,MAAM,CAAC,YAAY,CAAC;gBACxD,KAAK,EAAE,IAAI,CAAC,MAAM,GAAG,GAAG,GAAG,OAAO,CAAC,KAAK;gBACxC,IAAI,EAAE,IAAI,CAAC,UAAU;gBACrB,KAAK,EAAE,cAAc,CAAC,OAAO,GAAG,cAAc,CAAC,QAAQ;aACxD,CAAC,CAAC;YACH,IAAI,CAAC,OAAO,CAAC,KAAK,CAAC,WAAW,CAC5B,IAAI,CAAC,eAAe,CAAC,OAAO,CAAC,KAAK,CAAC,EACnC,CAAC,EACD,IAAI,CACL,CAAC;QACJ,CAAC,CAAC,CAAC;IACL,CAAC;IAED,SAAS,CAAC,EAAE,QAAQ,EAAE,MAAM,EAAwC;QAClE,IAAI,CAAC,SAAS,GAAG,QAAQ,CAAC;QAC1B,IAAI,CAAC,OAAO,GAAG,MAAM,CAAC;QACtB,IAAI,CAAC,oBAAoB,GAAG,IAAI,CAAC;IACnC,CAAC;IAED,eAAe,KAAK,IAAI,CAAC,gBAAgB,GAAG,IAAI,CAAC,CAAC,IAAI,CAAC,cAAc,GAAG,IAAI,CAAC,MAAM,CAAC,CAAC,IAAI,CAAC,eAAe,GAAG,IAAI,CAAC,OAAO,CAAC,CAAC,CAAC;IAC3H,OAAO,CAAC,KAAa,EAAE,MAAc;QACnC,KAAK,GAAG,IAAI,CAAC,GAAG,CAAC,CAAC,EAAE,IAAI,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC,CAAC;QACtC,MAAM,GAAG,IAAI,CAAC,GAAG,CAAC,CAAC,EAAE,IAAI,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC,CAAC;QACxC,MAAM,WAAW,GAAG,KAAK,KAAK,IAAI,CAAC,MAAM,IAAI,MAAM,KAAK,IAAI,CAAC,OAAO,CAAC;QACrE,IAAI,CAAC,MAAM,GAAG,KAAK,CAAC;QACpB,IAAI,CAAC,OAAO,GAAG,MAAM,CAAC;QACtB,IAAI,WAAW,EAAE,CAAC;YAChB,IAAI,CAAC,oBAAoB,GAAG,IAAI,CAAC;YACjC,MAAM,SAAS,GAAG,IAAI,CAAC,GAAG,CAAC,IAAI,CAAC,cAAc,EAAE,KAAK,CAAC,EAAE,UAAU,GAAG,IAAI,CAAC,GAAG,CAAC,IAAI,CAAC,eAAe,EAAE,MAAM,CAAC,CAAC;YAC5G,MAAM,eAAe,GAAG,SAAS,KAAK,IAAI,CAAC,cAAc,IAAI,UAAU,KAAK,IAAI,CAAC,eAAe,CAAC;YACjG,IAAI,CAAC,cAAc,GAAG,SAAS,CAAC;YAAC,IAAI,CAAC,eAAe,GAAG,UAAU,CAAC;YACnE,IAAI,IAAI,CAAC,gBAAgB,IAAI,CAAC,eAAe;gBAAE,OAAO;YACtD,yBAAyB;YACzB,KAAK,MAAM,GAAG,IAAI,IAAI,CAAC,cAAc,EAAE,CAAC;gBACtC,IAAI,CAAC,cAAc,CAAC,GAAG,CAAC,CAAC,OAAO,EAAE,CAAC;YACrC,CAAC;YACD,IAAI,CAAC,cAAc,GAAG,EAAE,CAAC;QAC3B,CAAC;IACH,CAAC;IAED,cAAc,CAAC,KAAa,EAAE,MAAc;QAC1C,KAAK,GAAG,IAAI,CAAC,GAAG,CAAC,CAAC,EAAE,IAAI,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC,CAAC;QACtC,MAAM,GAAG,IAAI,CAAC,GAAG,CAAC,CAAC,EAAE,IAAI,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC,CAAC;QACxC,IAAI,CAAC,UAAU,GAAG,KAAK,CAAC;QACxB,IAAI,CAAC,WAAW,GAAG,MAAM,CAAC;IAC5B,CAAC;IAED,eAAe,CAAC,YAA8C;QAC5D,2BAA2B;QAC3B,IAAI,CAAC,+BAA+B,CAAC,YAAY,CAAC,CAAC;QACnD,IAAI,CAAC,aAAa,GAAG,YAAY,CAAC;QAClC,IAAI,CAAC,oBAAoB,GAAG,IAAI,CAAC;IACnC,CAAC;IAED,UAAU,CAAC,KAAa,EAAE,IAAqB;QAC7C,MAAM,MAAM,GAAG,IAAI,CAAC,eAAe,CAAC,KAAK,CAAC,CAAC;QAC3C,IAAI,CAAC,OAAO,CAAC,KAAK,CAAC,WAAW,CAC5B,MAAM,EACN,CAAC,EACD,gBAAgB,CAAC,IAAI,EAAE,IAAI,CAAC,YAAY,CAAC,KAAK,CAAC,CAAC,IAAI,CAAC,CACtD,CAAC;IACJ,CAAC;IAED,SAAS,CAAC,IAAO;QACf,MAAM,aAAa,GAAG,IAAI,CAAC,cAAc,CAAC;QAC1C,IAAI,CAAC,aAAa,CAAC,IAAI,CAAC,EAAE,CAAC;YACzB,aAAa,CAAC,IAAI,CAAC,GAAG,IAAI,CAAC,aAAa,CACtC,IAAI,EACJ,IAAI,CAAC,aAAa,CAAC,IAAI,CAAC,IAAI,EAAE,CAC/B,CAAC;QACJ,CAAC;QACD,OAAO,aAAa,CAAC,IAAI,CAAC,CAAC;IAC7B,CAAC;IAED,cAAc;QACZ,OAAO,CAAC,GAAG,MAAM,CAAC,MAAM,CAAC,IAAI,CAAC,eAAe,CAAC,EAAE,GAAG,MAAM,CAAC,MAAM,CAAC,IAAI,CAAC,cAAc,CAAC,CAAC;aACnF,MAAM,CAAC,CAAC,GAAG,EAAE,MAAM,EAAE,EAAE,CAAC,GAAG,GAAG,MAAM,CAAC,IAAI,EAAE,CAAC,CAAC,CAAC;IACnD,CAAC;IAED,OAAO;QACL,MAAM,CAAC,IAAI,CAAC,IAAI,CAAC,eAAe,CAAC,CAAC,OAAO,CAAC,CAAC,GAAG,EAAE,EAAE;YAC/C,IAAI,CAAC,eAAuB,CAAC,GAAG,CAAC,CAAC,OAAO,EAAE,CAAC;QAC/C,CAAC,CAAC,CAAC;QACH,MAAM,CAAC,IAAI,CAAC,IAAI,CAAC,cAAc,CAAC,CAAC,OAAO,CAAC,CAAC,GAAG,EAAE,EAAE;YAC9C,IAAI,CAAC,cAAsB,CAAC,GAAG,CAAC,CAAC,OAAO,EAAE,CAAC;QAC9C,CAAC,CAAC,CAAC;IACL,CAAC;IAED,gBAAgB,CAAC,KAAa,EAAE,MAAc;QAC5C,IAAI,CAAC,oBAAoB;YACvB,KAAK,KAAK,IAAI,CAAC,cAAc,CAAC,CAAC,CAAC,IAAI,MAAM,KAAK,IAAI,CAAC,cAAc,CAAC,CAAC,CAAC,CAAC;QACxE,IAAI,CAAC,cAAc,GAAG,CAAC,KAAK,EAAE,MAAM,CAAC,CAAC;IACxC,CAAC;IAEO,uBAAuB;QAC7B,OAAO;YACL,IAAI,CAAC,GAAG,CAAC,IAAI,CAAC,cAAc,CAAC,CAAC,CAAC,EAAE,IAAI,CAAC,UAAU,IAAI,IAAI,CAAC,MAAM,CAAC;YAChE,IAAI,CAAC,GAAG,CAAC,IAAI,CAAC,cAAc,CAAC,CAAC,CAAC,EAAE,IAAI,CAAC,WAAW,IAAI,IAAI,CAAC,OAAO,CAAC;SACnE,CAAC;IACJ,CAAC;IAEO,aAAa,CAAC,KAAa,EAAE,MAA6B;QAChE,MAAM,QAAQ,GAAG,MAAM,CAAC,QAAQ,IAAI,CAAC,CAAC;QACtC,MAAM,UAAU,GACd,CAAC,IAAI,CAAC,gBAAgB,CAAC,CAAC,CAAC,IAAI,CAAC,cAAc,GAAG,IAAI,CAAC,eAAe,CAAC,CAAC,CAAC,IAAI,CAAC,MAAM,GAAG,IAAI,CAAC,OAAO,CAAC,GAAG,QAAQ,GAAG,CAAC,MAAM,CAAC,KAAK,IAAI,CAAC,CAAC,GAAG,CAAC,CAAC;QACzI,OAAO,IAAI,CAAC,OAAO,CAAC,YAAY,CAAC;YAC/B,KAAK,EAAE,IAAI,CAAC,MAAM,GAAG,GAAG,GAAG,KAAK;YAChC,yCAAyC;YACzC,IAAI,EAAE,IAAI,CAAC,GAAG,CAAC,UAAU,EAAE,EAAE,CAAC;YAC9B,KAAK,EAAE,cAAc,CAAC,OAAO,GAAG,cAAc,CAAC,QAAQ;SACxD,CAAC,CAAC;IACL,CAAC;IAEO,+BAA+B,CAAC,SAA8B;QACpE,MAAM,aAAa,GAAG,IAAI,CAAC,cAAc,CAAC;QAC1C,MAAM,SAAS,GAAG,IAAI,CAAC,aAAa,CAAC;QACrC,KAAK,MAAM,GAAG,IAAI,aAAa,EAAE,CAAC;YAChC,IAAI,CAAC,oBAAoB,CAAC,SAAS,CAAC,GAAG,CAAC,IAAI,EAAE,EAAE,SAAS,CAAC,GAAG,CAAC,IAAI,EAAE,CAAC,EAAE,CAAC;gBACtE,aAAa,CAAC,GAAG,CAAC,CAAC,OAAO,EAAE,CAAC;gBAC7B,OAAO,aAAa,CAAC,GAAG,CAAC,CAAC;YAC5B,CAAC;QACH,CAAC;IACH,CAAC;IAEO,oBAAoB,CAAC,QAAmB;QAC9C,MAAM,aAAa,GAAG,IAAI,CAAC,cAAc,CAAC;QAC1C,MAAM,MAAM,GAAG,IAAI,CAAC,aAAa,CAAC;QAClC,KAAK,MAAM,GAAG,IAAI,MAAM,EAAE,CAAC;YACzB,IAAI,CAAC,aAAa,CAAC,GAAG,CAAC,IAAI,CAAC,QAAQ,EAAE,QAAQ,CAAC,GAAG,CAAC,EAAE,CAAC;gBACpD,aAAa,CAAC,GAAG,CAAC,GAAG,IAAI,CAAC,aAAa,CAAC,GAAG,EAAE,MAAM,CAAC,GAAG,CAAC,CAAC,CAAC;YAC5D,CAAC;QACH,CAAC;IACH,CAAC;IAEO,eAAe,CACrB,WAAiD,EACjD,UAAgD;QAEhD,IAAI,CAAC,IAAI,CAAC,oBAAoB,EAAE,CAAC;YAC/B,OAAO;QACT,CAAC;QACD,IAAI,CAAC,oBAAoB,GAAG,KAAK,CAAC;QAClC,MAAM,MAAM,GAAG,IAAI,CAAC,OAAO,CAAC;QAC5B,MAAM,MAAM,GAAG,IAAI,CAAC,UAAU,CAAC,WAAW,EAAE,UAAU,CAAC,CAAC;QACxD,IAAI,MAAM,KAAK,IAAI,CAAC,OAAO,EAAE,CAAC;YAC5B,OAAO;QACT,CAAC;QACD,IAAI,CAAC,OAAO,GAAG,MAAM,CAAC;QACtB,IAAI,CAAC,SAAS,GAAG,oBAAoB,CAAC,MAAM,CAAC,CAAC,qBAAqB,CAAC;YAClE,KAAK,EAAE,IAAI,CAAC,MAAM;YAClB,MAAM,EAAE,MAAM;YACd,OAAO,EAAE;gBACP,MAAM,EAAE,oBAAoB,CAAC,MAAM,CAAC,CAAC,kBAAkB,CAAC;oBACtD,KAAK,EAAE,IAAI,CAAC,MAAM;oBAClB,IAAI,EAAE,MAAM;iBACb,CAAC;gBACF,UAAU,EAAE,MAAM;aACnB;SACF,CAAC,CAAC;QACH,IAAI,CAAC,iBAAiB,EAAE,CAAC;IAC3B,CAAC;IAEO,UAAU,CAChB,WAAiD,EACjD,UAAgD;QAEhD,MAAM,MAAM,GAAG,IAAI,CAAC,OAAO,CAAC;QAC5B,MAAM,QAAQ,GAAG,IAAI,CAAC,SAAS,CAAC;QAChC,IAAI,MAAM,GAAG,CAAC,CAAC;QACf,MAAM,YAAY,GAAG,CAAC,IAAI,CAAC,aAAa,GAAG;YACzC,MAAM,EAAE,CAAC;YACT,QAAQ,EAAE,CAAC;YACX,OAAO,EAAE,CAAC;SACX,CAAC,CAAC;QACH,MAAM,aAAa,GAAG,IAAI,CAAC,uBAAuB,EAAE,CAAC;QAErD,IAAI,MAAM,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;YACtB,MAAM,EAAE,CAAC;QACX,CAAC;QACD,IAAI,QAAQ,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;YACxB,YAAY,CAAC,QAAQ,GAAG,MAAM,CAAC;YAC/B,MAAM,EAAE,CAAC;QACX,CAAC;QAED,SAAS,WAAW,CAAC,MAA6B;YAChD,MAAM,EAAE,IAAI,GAAG,KAAK,EAAE,QAAQ,EAAE,GAAG,MAAM,CAAC;YAC1C,IAAI,IAAI,GAAG,QAAQ,KAAK,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,MAAM,QAAQ,GAAG,IAAI,CAAC,CAAC,CAAC,EAAE,CAAC;YAC9D,IAAI,MAAM,CAAC,MAAM,EAAE,CAAC;gBAClB,IAAI,GAAG,UAAU,IAAI,GAAG,CAAC;YAC3B,CAAC;YACD,OAAO,IAAI,CAAC;QACd,CAAC;QACD,YAAY,CAAC,OAAO,GAAG,MAAM,CAAC;QAC9B,MAAM,EAAE,GAAG,UAAU,CAAC;EACxB,MAAM;aACL,IAAI,EAAE;aACN,GAAG,CAAC,CAAC,SAAS,EAAE,GAAG,EAAE,EAAE;YACtB,MAAM,aAAa,GAAG,UAAU,YAAY,CAAC,MAAM,cAAc,GAAG,IAAI,CAAC;YACzE,MAAM,OAAO,GAAG,MAAM,SAAS,EAAE,CAAC;YAClC,6BAA6B;YAC7B,OAAO,UAAU,CAAC,SAAS,CAAC,KAAK,SAAS;gBACxC,CAAC,CAAC,GAAG,aAAa,QAAQ,OAAO,oBAAoB;gBACrD,CAAC,CAAC,GAAG,aAAa,uBAAuB,OAAO,WAAW,WAAW,CAClE,WAAW,CAAC,SAAS,CAAC,CACvB,IAAI,CAAC;QACZ,CAAC,CAAC;aACD,IAAI,CAAC,IAAI,CAAC;EACX,IAAI,CAAC,SAAS;aACb,GAAG,CACF,CAAC,OAAO,EAAE,GAAG,EAAE,EAAE,CACf,UAAU,YAAY,CAAC,QAAQ,cAAc,GAAG,kBAAkB,OAAO,CAAC,KAAK,KAAK,OAAO,CAAC,IAAI,GAAG,CACtG;aACA,IAAI,CAAC,IAAI,CAAC;;EAEX,IAAI,CAAC,QAAQ;aACZ,GAAG,CAAC,CAAC,IAAI,EAAE,GAAG,EAAE,EAAE;YACjB,OAAO,UACL,YAAY,CAAC,OACf,cAAc,GAAG,kCAAkC,IAAI,WAAW,WAAW,CAC3E,IAAI,CAAC,aAAa,CAAC,IAAI,CAAC,IAAI,EAAE,CAC/B,IAAI,CAAC;QACR,CAAC,CAAC;aACD,IAAI,CAAC,IAAI,CAAC;EACX,IAAI,CAAC,SAAS,IAAI,EAAE;2BACK,aAAa,CAAC,CAAC,CAAC,KAAK,aAAa,CAAC,CAAC,CAAC;;EAE9D,IAAI,CAAC,OAAO;;CAEb,CAAC;QAEE,OAAO,EAAE,CAAC;IACZ,CAAC;IAEO,iBAAiB;QACvB,MAAM,UAAU,GAAmB,EAAE,CAAC;QACtC,MAAM,MAAM,GAAG,IAAI,CAAC,OAAO,CAAC;QAC5B,MAAM,YAAY,GAAG,IAAI,CAAC,aAAa,CAAC;QAExC,IAAI,IAAI,CAAC,SAAS,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;YAC9B,UAAU,CAAC,YAAY,CAAC,QAAQ,CAAC,GAAG,MAAM,CAAC,eAAe,CAAC;gBACzD,KAAK,EAAE,IAAI,CAAC,MAAM,GAAG,WAAW;gBAChC,MAAM,EAAE,IAAI,CAAC,SAAS,CAAC,kBAAkB,CAAC,YAAY,CAAC,QAAQ,CAAC;gBAChE,OAAO,EAAE,IAAI,CAAC,SAAS,CAAC,GAAG,CACzB,CAAC,OAAO,EAAE,GAAG,EAAE,EAAE,CACf,CAAC;oBACC,OAAO,EAAE,GAAG;oBACZ,QAAQ,EAAE;wBACR,MAAM,EAAE,IAAI,CAAC,eAAe,CAAC,OAAO,CAAC,KAAK,CAAC;qBAC5C;iBACoB,CAAA,CAC1B;aACF,CAAC,CAAC;QACL,CAAC;QAED,IAAI,CAAC,WAAW,GAAG,UAAU,CAAC;IAChC,CAAC;IAED,UAAU,CACR,cAAiC,EACjC,MAAuC;IAEvC;;OAEG;IACH,OAA8B;QAE9B,IAAI,CAAC,oBAAoB,CAAC,MAAM,CAAC,IAAI,CAAC,OAAO,IAAI,EAAE,CAAC,CAAC,CAAC;QACtD,MAAM,aAAa,GAAG;YACpB,GAAG,IAAI,CAAC,cAAc;YACtB,GAAG,OAAO;SACkB,CAAC;QAE/B,MAAM,UAAU,GAAG,IAAI,CAAC,OAAO,CAAC,MAAM,CAAC,CAAC,GAAG,EAAE,SAAS,EAAE,EAAE;YACxD,GAAG,CAAC,SAAS,CAAC,GAAG,MAAM,CAAC,SAAc,CAAC,CAAC,MAAM,CAAC,CAAC,CAAC,QAAQ,CAAC,CAAC,CAAC,SAAS,CAAC;YACtE,OAAO,GAAG,CAAC;QACb,CAAC,EAAE,EAA0C,CAAC,CAAC;QAE/C,IAAI,CAAC,eAAe,CAAC,MAAM,EAAE,UAAU,CAAC,CAAC;QAEzC,MAAM,YAAY,GAAG,IAAI,CAAC,aAAa,CAAC;QACxC,IAAI,IAAI,CAAC,OAAO,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;YAC5B,IAAI,CAAC,WAAW,CAAC,YAAY,CAAC,MAAM,CAAC,GAAG,IAAI,CAAC,OAAO,CAAC,eAAe,CAAC;gBACnE,KAAK,EAAE,IAAI,CAAC,MAAM,GAAG,SAAS;gBAC9B,MAAM,EAAE,IAAI,CAAC,SAAS,CAAC,kBAAkB,CAAC,YAAY,CAAC,MAAM,CAAC;gBAC9D,OAAO,EAAE,IAAI,CAAC,OAAO,CAAC,GAAG,CAAC,CAAC,SAAS,EAAE,GAAG,EAAE,EAAE,CAAC,CAAC;oBAC7C,OAAO,EAAE,GAAG;oBACZ,QAAQ,EAAE,MAAM,CAAC,SAAc,CAAC,CAAC,MAAM;wBACrC,CAAC,CAAC;4BACE,MAAM,EAAE,MAAM,CAAC,SAAc,CAAC,CAAC,MAAO;yBACvC;wBACH,CAAC,CAAC,MAAM,CAAC,SAAc,CAAC,CAAC,OAAQ,CAAC,UAAU,EAAE;iBACjD,CAAC,CAAC;aACJ,CAAC,CAAC;QACL,CAAC;QAED,UAAU;QACV,IAAI,CAAC,WAAW,CAAC,YAAY,CAAC,OAAO,CAAC,GAAG,IAAI,CAAC,OAAO,CAAC,eAAe,CAAC;YACpE,KAAK,EAAE,IAAI,CAAC,MAAM,GAAG,UAAU;YAC/B,MAAM,EAAE,IAAI,CAAC,SAAS,CAAC,kBAAkB,CAAC,YAAY,CAAC,OAAO,CAAC;YAC/D,OAAO,EAAE,IAAI,CAAC,QAAQ,CAAC,GAAG,CAAC,CAAC,UAAU,EAAE,GAAG,EAAE,EAAE,CAAC,CAAC;gBAC/C,OAAO,EAAE,GAAG;gBACZ,QAAQ,EAAE;oBACR,MAAM,EAAE,aAAa,CAAC,UAAU,CAAC;iBAClC;aACF,CAAC,CAAC;SACJ,CAAC,CAAC;QAEH,wBAAwB;QACxB,MAAM,WAAW,GAAG,cAAc,CAAC,gBAAgB,EAAE,CAAC;QAEtD,mBAAmB;QACnB,WAAW,CAAC,WAAW,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC;QACxC,cAAc;QACd,IAAI,CAAC,WAAW,CAAC,OAAO,CAAC,CAAC,SAAS,EAAE,GAAG,EAAE,EAAE;YAC1C,WAAW,CAAC,YAAY,CAAC,GAAG,EAAE,SAAS,CAAC,CAAC;QAC3C,CAAC,CAAC,CAAC;QAEH,MAAM,aAAa,GAAG,IAAI,CAAC,uBAAuB,EAAE,CAAC;QACrD,WAAW,CAAC,kBAAkB,CAC5B,IAAI,CAAC,IAAI,CAAC,CAAC,IAAI,CAAC,UAAU,IAAI,IAAI,CAAC,MAAM,CAAC,GAAG,aAAa,CAAC,CAAC,CAAC,CAAC,EAC9D,IAAI,CAAC,IAAI,CAAC,CAAC,IAAI,CAAC,WAAW,IAAI,IAAI,CAAC,OAAO,CAAC,GAAG,aAAa,CAAC,CAAC,CAAC,CAAC,EAChE,CAAC,CACF,CAAC;QACF,sBAAsB;QACtB,WAAW,CAAC,GAAG,EAAE,CAAC;IACpB,CAAC;CACF","sourcesContent":["import { getTargetShaderCache } from './WGPUShaderCache.js';\n// TODO output read write buffer\nimport { getUniformBuffer, Uniform } from './WGPUFullQuadPass.js';\n\nexport type { Uniform };\n\nfunction isStorageParamsEqual(\n  params: WGPUComputePassOutput,\n  other: WGPUComputePassOutput\n) {\n  return (\n    params.channels === other.channels &&\n    params.type === other.type &&\n    params.count === other.count &&\n    params.atomic === other.atomic\n  );\n}\n\nexport const WORKGROUP_SIZE = 8;\n\nexport interface WGPUComputePassInput {\n  buffer?: GPUBuffer;\n  texture?: GPUTexture;\n  channels: number;\n  type?: 'u32' | 'i32' | 'f32';\n}\n\nexport interface WGPUComputePassOutput {\n  channels?: number;\n  // Default: 1. Count per element.\n  count?: number;\n  atomic?: boolean;\n  type?: 'u32' | 'i32' | 'f32';\n}\n\nexport class WGPUComputePass<I extends string, O extends string> {\n  private _label;\n\n  private _device;\n  private _outputBuffers: Record<string, GPUBuffer> = {};\n  private _outputParams: Record<string, WGPUComputePassOutput> = {};\n\n  private _pipeline!: GPUComputePipeline;\n  private _bindGroups: GPUBindGroup[] = [];\n  private _needsUpdatePipeline = true;\n\n  private _inputs: string[] = [];\n  private _outputs: string[] = [];\n  private _uniforms: Uniform[] = [];\n  private _uniformsMap: Record<string, Uniform> = {};\n  private _uniformBuffers: Record<string, GPUBuffer> = {};\n\n  private _capacityWidth = 0;\n  private _capacityHeight = 0;\n  private _reserveCapacity = false;\n  private _width = 10;\n  private _height = 10;\n\n  private _execWidth?: number;\n  private _execHeight?: number;\n\n  private _workGroupSize = [WORKGROUP_SIZE, WORKGROUP_SIZE];\n\n  private _csCode = '';\n  private _csMain;\n  private _csDefine;\n\n  private _groupOffsets = {\n    inputs: 0,\n    uniforms: 1,\n    outputs: 2\n  };\n\n  constructor(\n    label: string,\n    device: GPUDevice,\n    opts: {\n      inputs: I[];\n      outputs: O[];\n      csMain: string;\n      csDefine?: string;\n      uniforms?: Uniform[];\n    }\n  ) {\n    this._label = label;\n    this._device = device;\n    this._csMain = opts.csMain;\n    this._csDefine = opts.csDefine;\n    this._inputs = opts.inputs;\n    this._outputs = opts.outputs;\n    const uniforms = opts.uniforms || [];\n    this._uniforms = uniforms;\n\n    this._uniformsMap = this._uniforms.reduce((map, uniform) => {\n      map[uniform.label] = uniform;\n      return map;\n    }, {} as Record<string, Uniform>);\n    uniforms.forEach((uniform) => {\n      const data = getUniformBuffer(uniform.data, uniform.type);\n      this._uniformBuffers[uniform.label] = device.createBuffer({\n        label: this._label + '.' + uniform.label,\n        size: data.byteLength,\n        usage: GPUBufferUsage.UNIFORM | GPUBufferUsage.COPY_DST\n      });\n      this._device.queue.writeBuffer(\n        this._uniformBuffers[uniform.label],\n        0,\n        data\n      );\n    });\n  }\n\n  setCSCode({ csDefine, csMain }: { csDefine: string; csMain: string }) {\n    this._csDefine = csDefine;\n    this._csMain = csMain;\n    this._needsUpdatePipeline = true;\n  }\n\n  reserveCapacity() { this._reserveCapacity = true; this._capacityWidth = this._width; this._capacityHeight = this._height; }\n  setSize(width: number, height: number) {\n    width = Math.max(1, Math.ceil(width));\n    height = Math.max(1, Math.ceil(height));\n    const sizeChanged = width !== this._width || height !== this._height;\n    this._width = width;\n    this._height = height;\n    if (sizeChanged) {\n      this._needsUpdatePipeline = true;\n      const nextWidth = Math.max(this._capacityWidth, width), nextHeight = Math.max(this._capacityHeight, height);\n      const capacityChanged = nextWidth !== this._capacityWidth || nextHeight !== this._capacityHeight;\n      this._capacityWidth = nextWidth; this._capacityHeight = nextHeight;\n      if (this._reserveCapacity && !capacityChanged) return;\n      // Invalidate all buffers\n      for (const key in this._outputBuffers) {\n        this._outputBuffers[key].destroy();\n      }\n      this._outputBuffers = {};\n    }\n  }\n\n  setExecuteSize(width: number, height: number) {\n    width = Math.max(1, Math.ceil(width));\n    height = Math.max(1, Math.ceil(height));\n    this._execWidth = width;\n    this._execHeight = height;\n  }\n\n  setOutputParams(outputParams: Record<O, WGPUComputePassOutput>) {\n    // TODO lazy update buffers\n    this._invalidateChangedOutputBuffers(outputParams);\n    this._outputParams = outputParams;\n    this._needsUpdatePipeline = true;\n  }\n\n  setUniform(label: string, data: Uniform['data']) {\n    const buffer = this._uniformBuffers[label];\n    this._device.queue.writeBuffer(\n      buffer,\n      0,\n      getUniformBuffer(data, this._uniformsMap[label].type)\n    );\n  }\n\n  getOutput(name: O) {\n    const outputBuffers = this._outputBuffers;\n    if (!outputBuffers[name]) {\n      outputBuffers[name] = this._createBuffer(\n        name,\n        this._outputParams[name] || {}\n      );\n    }\n    return outputBuffers[name];\n  }\n\n  getMemoryBytes() {\n    return [...Object.values(this._uniformBuffers), ...Object.values(this._outputBuffers)]\n      .reduce((sum, buffer) => sum + buffer.size, 0);\n  }\n\n  dispose() {\n    Object.keys(this._uniformBuffers).forEach((key) => {\n      (this._uniformBuffers as any)[key].destroy();\n    });\n    Object.keys(this._outputBuffers).forEach((key) => {\n      (this._outputBuffers as any)[key].destroy();\n    });\n  }\n\n  setWorkGroupSize(width: number, height: number) {\n    this._needsUpdatePipeline =\n      width !== this._workGroupSize[0] || height !== this._workGroupSize[1];\n    this._workGroupSize = [width, height];\n  }\n\n  private _getActualWorkgroupSize() {\n    return [\n      Math.min(this._workGroupSize[0], this._execWidth ?? this._width),\n      Math.min(this._workGroupSize[1], this._execHeight ?? this._height)\n    ];\n  }\n\n  private _createBuffer(label: string, params: WGPUComputePassOutput) {\n    const channels = params.channels ?? 4;\n    const byteLength =\n      (this._reserveCapacity ? this._capacityWidth * this._capacityHeight : this._width * this._height) * channels * (params.count ?? 1) * 4;\n    return this._device.createBuffer({\n      label: this._label + '.' + label,\n      // webgpu needs buffer at least 80 bytes.\n      size: Math.max(byteLength, 80),\n      usage: GPUBufferUsage.STORAGE | GPUBufferUsage.COPY_SRC\n    });\n  }\n\n  private _invalidateChangedOutputBuffers(newParams: Record<string, any>) {\n    const outputBuffers = this._outputBuffers;\n    const oldParams = this._outputParams;\n    for (const key in outputBuffers) {\n      if (!isStorageParamsEqual(newParams[key] || {}, oldParams[key] || {})) {\n        outputBuffers[key].destroy();\n        delete outputBuffers[key];\n      }\n    }\n  }\n\n  private _updateOutputBuffers(excludes?: string[]) {\n    const outputBuffers = this._outputBuffers;\n    const params = this._outputParams;\n    for (const key in params) {\n      if (!outputBuffers[key] && !excludes?.includes(key)) {\n        outputBuffers[key] = this._createBuffer(key, params[key]);\n      }\n    }\n  }\n\n  private _updatePipeline(\n    inputParams: Record<string, WGPUComputePassInput>,\n    inputTypes: Record<string, 'texture' | 'buffer'>\n  ) {\n    if (!this._needsUpdatePipeline) {\n      return;\n    }\n    this._needsUpdatePipeline = false;\n    const device = this._device;\n    const csCode = this._getFullCs(inputParams, inputTypes);\n    if (csCode === this._csCode) {\n      return;\n    }\n    this._csCode = csCode;\n    this._pipeline = getTargetShaderCache(device).createComputePipeline({\n      label: this._label,\n      layout: 'auto',\n      compute: {\n        module: getTargetShaderCache(device).createShaderModule({\n          label: this._label,\n          code: csCode\n        }),\n        entryPoint: 'main'\n      }\n    });\n    this._updateBindGroups();\n  }\n\n  private _getFullCs(\n    inputParams: Record<string, WGPUComputePassInput>,\n    inputTypes: Record<string, 'texture' | 'buffer'>\n  ) {\n    const inputs = this._inputs;\n    const uniforms = this._uniforms;\n    let offset = 0;\n    const groupOffsets = (this._groupOffsets = {\n      inputs: 0,\n      uniforms: 0,\n      outputs: 0\n    });\n    const workgroupSize = this._getActualWorkgroupSize();\n\n    if (inputs.length > 0) {\n      offset++;\n    }\n    if (uniforms.length > 0) {\n      groupOffsets.uniforms = offset;\n      offset++;\n    }\n\n    function getTypeCode(params: WGPUComputePassOutput) {\n      const { type = 'f32', channels } = params;\n      let code = channels === 1 ? type : `vec${channels}${type[0]}`;\n      if (params.atomic) {\n        code = `atomic<${code}>`;\n      }\n      return code;\n    }\n    groupOffsets.outputs = offset;\n    const cs = /* wgsl */ `\n${inputs\n  .sort()\n  .map((inputName, idx) => {\n    const bindingPrefix = `@group(${groupOffsets.inputs}) @binding(${idx}) `;\n    const varName = `in_${inputName}`;\n    // TODO more channels option.\n    return inputTypes[inputName] === 'texture'\n      ? `${bindingPrefix} var ${varName}: texture_2d<f32>;`\n      : `${bindingPrefix} var<storage, read> ${varName}: array<${getTypeCode(\n          inputParams[inputName]\n        )}>;`;\n  })\n  .join('\\n')}\n${this._uniforms\n  .map(\n    (uniform, idx) =>\n      `@group(${groupOffsets.uniforms}) @binding(${idx}) var<uniform> ${uniform.label}: ${uniform.type};`\n  )\n  .join('\\n')}\n\n${this._outputs\n  .map((name, idx) => {\n    return `@group(${\n      groupOffsets.outputs\n    }) @binding(${idx}) var<storage, read_write> out_${name}: array<${getTypeCode(\n      this._outputParams[name] || {}\n    )}>;`;\n  })\n  .join('\\n')}\n${this._csDefine ?? ''}\n@compute @workgroup_size(${workgroupSize[0]}, ${workgroupSize[1]}, 1)\nfn main(@builtin(global_invocation_id) globalId: vec3u) {\n${this._csMain}\n}\n`;\n\n    return cs;\n  }\n\n  private _updateBindGroups() {\n    const bindGroups: GPUBindGroup[] = [];\n    const device = this._device;\n    const groupOffsets = this._groupOffsets;\n\n    if (this._uniforms.length > 0) {\n      bindGroups[groupOffsets.uniforms] = device.createBindGroup({\n        label: this._label + '.uniforms',\n        layout: this._pipeline.getBindGroupLayout(groupOffsets.uniforms),\n        entries: this._uniforms.map(\n          (uniform, idx) =>\n            ({\n              binding: idx,\n              resource: {\n                buffer: this._uniformBuffers[uniform.label]\n              }\n            } as GPUBindGroupEntry)\n        )\n      });\n    }\n\n    this._bindGroups = bindGroups;\n  }\n\n  createPass(\n    commandEncoder: GPUCommandEncoder,\n    inputs: Record<I, WGPUComputePassInput>,\n\n    /**\n     * Explict outputs. Used for buffers reuse.\n     */\n    outputs?: Record<O, GPUBuffer>\n  ) {\n    this._updateOutputBuffers(Object.keys(outputs || {}));\n    const outputBuffers = {\n      ...this._outputBuffers,\n      ...outputs\n    } as Record<string, GPUBuffer>;\n\n    const inputTypes = this._inputs.reduce((obj, inputName) => {\n      obj[inputName] = inputs[inputName as I].buffer ? 'buffer' : 'texture';\n      return obj;\n    }, {} as Record<string, 'texture' | 'buffer'>);\n\n    this._updatePipeline(inputs, inputTypes);\n\n    const groupOffsets = this._groupOffsets;\n    if (this._inputs.length > 0) {\n      this._bindGroups[groupOffsets.inputs] = this._device.createBindGroup({\n        label: this._label + '.inputs',\n        layout: this._pipeline.getBindGroupLayout(groupOffsets.inputs),\n        entries: this._inputs.map((inputName, idx) => ({\n          binding: idx,\n          resource: inputs[inputName as I].buffer\n            ? {\n                buffer: inputs[inputName as I].buffer!\n              }\n            : inputs[inputName as I].texture!.createView()\n        }))\n      });\n    }\n\n    // Outputs\n    this._bindGroups[groupOffsets.outputs] = this._device.createBindGroup({\n      label: this._label + '.outputs',\n      layout: this._pipeline.getBindGroupLayout(groupOffsets.outputs),\n      entries: this._outputs.map((bufferName, idx) => ({\n        binding: idx,\n        resource: {\n          buffer: outputBuffers[bufferName]\n        }\n      }))\n    });\n\n    // Begin the render pass\n    const computePass = commandEncoder.beginComputePass();\n\n    // Draw a full quad\n    computePass.setPipeline(this._pipeline);\n    // Bind groups\n    this._bindGroups.forEach((bindGroup, idx) => {\n      computePass.setBindGroup(idx, bindGroup);\n    });\n\n    const workgroupSize = this._getActualWorkgroupSize();\n    computePass.dispatchWorkgroups(\n      Math.ceil((this._execWidth ?? this._width) / workgroupSize[0]),\n      Math.ceil((this._execHeight ?? this._height) / workgroupSize[1]),\n      1\n    );\n    // End the render pass\n    computePass.end();\n  }\n}\n"]}
+{"version":3,"file":"WGPUComputePass.js","sourceRoot":"","sources":["../../src/renderer/WGPUComputePass.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,oBAAoB,EAAE,MAAM,sBAAsB,CAAC;AAC5D,gCAAgC;AAChC,OAAO,EAAE,gBAAgB,EAAgB,MAAM,uBAAuB,CAAC;AAIvE,SAAS,oBAAoB,CAC3B,MAA6B,EAC7B,KAA4B;IAE5B,OAAO,CACL,MAAM,CAAC,QAAQ,KAAK,KAAK,CAAC,QAAQ;QAClC,MAAM,CAAC,IAAI,KAAK,KAAK,CAAC,IAAI;QAC1B,MAAM,CAAC,KAAK,KAAK,KAAK,CAAC,KAAK;QAC5B,MAAM,CAAC,MAAM,KAAK,KAAK,CAAC,MAAM,CAC/B,CAAC;AACJ,CAAC;AAED,MAAM,CAAC,MAAM,cAAc,GAAG,CAAC,CAAC;AAiBhC,MAAM,OAAO,eAAe;IAClB,MAAM,CAAC;IAEP,OAAO,CAAC;IACR,cAAc,GAA8B,EAAE,CAAC;IAC/C,aAAa,GAA0C,EAAE,CAAC;IAE1D,SAAS,CAAsB;IAC/B,WAAW,GAAmB,EAAE,CAAC;IACjC,oBAAoB,GAAG,IAAI,CAAC;IAE5B,OAAO,GAAa,EAAE,CAAC;IACvB,QAAQ,GAAa,EAAE,CAAC;IACxB,SAAS,GAAc,EAAE,CAAC;IAC1B,YAAY,GAA4B,EAAE,CAAC;IAC3C,eAAe,GAA8B,EAAE,CAAC;IAEhD,cAAc,GAAG,CAAC,CAAC;IACnB,eAAe,GAAG,CAAC,CAAC;IACpB,gBAAgB,GAAG,KAAK,CAAC;IACzB,MAAM,GAAG,EAAE,CAAC;IACZ,OAAO,GAAG,EAAE,CAAC;IAEb,UAAU,CAAU;IACpB,WAAW,CAAU;IAErB,cAAc,GAAG,CAAC,cAAc,EAAE,cAAc,CAAC,CAAC;IAElD,OAAO,GAAG,EAAE,CAAC;IACb,OAAO,CAAC;IACR,SAAS,CAAC;IAEV,aAAa,GAAG;QACtB,MAAM,EAAE,CAAC;QACT,QAAQ,EAAE,CAAC;QACX,OAAO,EAAE,CAAC;KACX,CAAC;IAEF,YACE,KAAa,EACb,MAAiB,EACjB,IAMC;QAED,IAAI,CAAC,MAAM,GAAG,KAAK,CAAC;QACpB,IAAI,CAAC,OAAO,GAAG,MAAM,CAAC;QACtB,IAAI,CAAC,OAAO,GAAG,IAAI,CAAC,MAAM,CAAC;QAC3B,IAAI,CAAC,SAAS,GAAG,IAAI,CAAC,QAAQ,CAAC;QAC/B,IAAI,CAAC,OAAO,GAAG,IAAI,CAAC,MAAM,CAAC;QAC3B,IAAI,CAAC,QAAQ,GAAG,IAAI,CAAC,OAAO,CAAC;QAC7B,MAAM,QAAQ,GAAG,IAAI,CAAC,QAAQ,IAAI,EAAE,CAAC;QACrC,IAAI,CAAC,SAAS,GAAG,QAAQ,CAAC;QAE1B,IAAI,CAAC,YAAY,GAAG,IAAI,CAAC,SAAS,CAAC,MAAM,CAAC,CAAC,GAAG,EAAE,OAAO,EAAE,EAAE;YACzD,GAAG,CAAC,OAAO,CAAC,KAAK,CAAC,GAAG,OAAO,CAAC;YAC7B,OAAO,GAAG,CAAC;QACb,CAAC,EAAE,EAA6B,CAAC,CAAC;QAClC,QAAQ,CAAC,OAAO,CAAC,CAAC,OAAO,EAAE,EAAE;YAC3B,MAAM,IAAI,GAAG,gBAAgB,CAAC,OAAO,CAAC,IAAI,EAAE,OAAO,CAAC,IAAI,CAAC,CAAC;YAC1D,IAAI,CAAC,eAAe,CAAC,OAAO,CAAC,KAAK,CAAC,GAAG,MAAM,CAAC,YAAY,CAAC;gBACxD,KAAK,EAAE,IAAI,CAAC,MAAM,GAAG,GAAG,GAAG,OAAO,CAAC,KAAK;gBACxC,IAAI,EAAE,IAAI,CAAC,UAAU;gBACrB,KAAK,EAAE,cAAc,CAAC,OAAO,GAAG,cAAc,CAAC,QAAQ;aACxD,CAAC,CAAC;YACH,IAAI,CAAC,OAAO,CAAC,KAAK,CAAC,WAAW,CAC5B,IAAI,CAAC,eAAe,CAAC,OAAO,CAAC,KAAK,CAAC,EACnC,CAAC,EACD,IAAI,CACL,CAAC;QACJ,CAAC,CAAC,CAAC;IACL,CAAC;IAED,SAAS,CAAC,EAAE,QAAQ,EAAE,MAAM,EAAwC;QAClE,IAAI,CAAC,SAAS,GAAG,QAAQ,CAAC;QAC1B,IAAI,CAAC,OAAO,GAAG,MAAM,CAAC;QACtB,IAAI,CAAC,oBAAoB,GAAG,IAAI,CAAC;IACnC,CAAC;IAED,eAAe,KAAK,IAAI,CAAC,gBAAgB,GAAG,IAAI,CAAC,CAAC,IAAI,CAAC,cAAc,GAAG,IAAI,CAAC,MAAM,CAAC,CAAC,IAAI,CAAC,eAAe,GAAG,IAAI,CAAC,OAAO,CAAC,CAAC,CAAC;IAC3H,OAAO,CAAC,KAAa,EAAE,MAAc;QACnC,KAAK,GAAG,IAAI,CAAC,GAAG,CAAC,CAAC,EAAE,IAAI,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC,CAAC;QACtC,MAAM,GAAG,IAAI,CAAC,GAAG,CAAC,CAAC,EAAE,IAAI,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC,CAAC;QACxC,MAAM,WAAW,GAAG,KAAK,KAAK,IAAI,CAAC,MAAM,IAAI,MAAM,KAAK,IAAI,CAAC,OAAO,CAAC;QACrE,IAAI,CAAC,MAAM,GAAG,KAAK,CAAC;QACpB,IAAI,CAAC,OAAO,GAAG,MAAM,CAAC;QACtB,IAAI,WAAW,EAAE,CAAC;YAChB,IAAI,CAAC,oBAAoB,GAAG,IAAI,CAAC;YACjC,MAAM,SAAS,GAAG,IAAI,CAAC,GAAG,CAAC,IAAI,CAAC,cAAc,EAAE,KAAK,CAAC,EAAE,UAAU,GAAG,IAAI,CAAC,GAAG,CAAC,IAAI,CAAC,eAAe,EAAE,MAAM,CAAC,CAAC;YAC5G,MAAM,eAAe,GAAG,SAAS,KAAK,IAAI,CAAC,cAAc,IAAI,UAAU,KAAK,IAAI,CAAC,eAAe,CAAC;YACjG,IAAI,CAAC,cAAc,GAAG,SAAS,CAAC;YAAC,IAAI,CAAC,eAAe,GAAG,UAAU,CAAC;YACnE,IAAI,IAAI,CAAC,gBAAgB,IAAI,CAAC,eAAe;gBAAE,OAAO;YACtD,yBAAyB;YACzB,KAAK,MAAM,GAAG,IAAI,IAAI,CAAC,cAAc,EAAE,CAAC;gBACtC,IAAI,CAAC,cAAc,CAAC,GAAG,CAAC,CAAC,OAAO,EAAE,CAAC;YACrC,CAAC;YACD,IAAI,CAAC,cAAc,GAAG,EAAE,CAAC;QAC3B,CAAC;IACH,CAAC;IAED,cAAc,CAAC,KAAa,EAAE,MAAc;QAC1C,KAAK,GAAG,IAAI,CAAC,GAAG,CAAC,CAAC,EAAE,IAAI,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC,CAAC;QACtC,MAAM,GAAG,IAAI,CAAC,GAAG,CAAC,CAAC,EAAE,IAAI,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC,CAAC;QACxC,IAAI,CAAC,UAAU,GAAG,KAAK,CAAC;QACxB,IAAI,CAAC,WAAW,GAAG,MAAM,CAAC;IAC5B,CAAC;IAED,eAAe,CAAC,YAA8C;QAC5D,2BAA2B;QAC3B,IAAI,CAAC,+BAA+B,CAAC,YAAY,CAAC,CAAC;QACnD,IAAI,CAAC,aAAa,GAAG,YAAY,CAAC;QAClC,IAAI,CAAC,oBAAoB,GAAG,IAAI,CAAC;IACnC,CAAC;IAED,UAAU,CAAC,KAAa,EAAE,IAAqB;QAC7C,MAAM,MAAM,GAAG,IAAI,CAAC,eAAe,CAAC,KAAK,CAAC,CAAC;QAC3C,IAAI,CAAC,OAAO,CAAC,KAAK,CAAC,WAAW,CAC5B,MAAM,EACN,CAAC,EACD,gBAAgB,CAAC,IAAI,EAAE,IAAI,CAAC,YAAY,CAAC,KAAK,CAAC,CAAC,IAAI,CAAC,CACtD,CAAC;IACJ,CAAC;IAED,SAAS,CAAC,IAAO;QACf,MAAM,aAAa,GAAG,IAAI,CAAC,cAAc,CAAC;QAC1C,IAAI,CAAC,aAAa,CAAC,IAAI,CAAC,EAAE,CAAC;YACzB,aAAa,CAAC,IAAI,CAAC,GAAG,IAAI,CAAC,aAAa,CACtC,IAAI,EACJ,IAAI,CAAC,aAAa,CAAC,IAAI,CAAC,IAAI,EAAE,CAC/B,CAAC;QACJ,CAAC;QACD,OAAO,aAAa,CAAC,IAAI,CAAC,CAAC;IAC7B,CAAC;IAED,cAAc;QACZ,OAAO,CAAC,GAAG,MAAM,CAAC,MAAM,CAAC,IAAI,CAAC,eAAe,CAAC,EAAE,GAAG,MAAM,CAAC,MAAM,CAAC,IAAI,CAAC,cAAc,CAAC,CAAC;aACnF,MAAM,CAAC,CAAC,GAAG,EAAE,MAAM,EAAE,EAAE,CAAC,GAAG,GAAG,MAAM,CAAC,IAAI,EAAE,CAAC,CAAC,CAAC;IACnD,CAAC;IAED,OAAO;QACL,MAAM,CAAC,IAAI,CAAC,IAAI,CAAC,eAAe,CAAC,CAAC,OAAO,CAAC,CAAC,GAAG,EAAE,EAAE;YAC/C,IAAI,CAAC,eAAuB,CAAC,GAAG,CAAC,CAAC,OAAO,EAAE,CAAC;QAC/C,CAAC,CAAC,CAAC;QACH,MAAM,CAAC,IAAI,CAAC,IAAI,CAAC,cAAc,CAAC,CAAC,OAAO,CAAC,CAAC,GAAG,EAAE,EAAE;YAC9C,IAAI,CAAC,cAAsB,CAAC,GAAG,CAAC,CAAC,OAAO,EAAE,CAAC;QAC9C,CAAC,CAAC,CAAC;IACL,CAAC;IAED,gBAAgB,CAAC,KAAa,EAAE,MAAc;QAC5C,IAAI,CAAC,oBAAoB;YACvB,KAAK,KAAK,IAAI,CAAC,cAAc,CAAC,CAAC,CAAC,IAAI,MAAM,KAAK,IAAI,CAAC,cAAc,CAAC,CAAC,CAAC,CAAC;QACxE,IAAI,CAAC,cAAc,GAAG,CAAC,KAAK,EAAE,MAAM,CAAC,CAAC;IACxC,CAAC;IAEO,uBAAuB;QAC7B,OAAO;YACL,IAAI,CAAC,GAAG,CAAC,IAAI,CAAC,cAAc,CAAC,CAAC,CAAC,EAAE,IAAI,CAAC,UAAU,IAAI,IAAI,CAAC,MAAM,CAAC;YAChE,IAAI,CAAC,GAAG,CAAC,IAAI,CAAC,cAAc,CAAC,CAAC,CAAC,EAAE,IAAI,CAAC,WAAW,IAAI,IAAI,CAAC,OAAO,CAAC;SACnE,CAAC;IACJ,CAAC;IAEO,aAAa,CAAC,KAAa,EAAE,MAA6B;QAChE,MAAM,QAAQ,GAAG,MAAM,CAAC,QAAQ,IAAI,CAAC,CAAC;QACtC,MAAM,UAAU,GACd,CAAC,IAAI,CAAC,gBAAgB,CAAC,CAAC,CAAC,IAAI,CAAC,cAAc,GAAG,IAAI,CAAC,eAAe,CAAC,CAAC,CAAC,IAAI,CAAC,MAAM,GAAG,IAAI,CAAC,OAAO,CAAC,GAAG,QAAQ,GAAG,CAAC,MAAM,CAAC,KAAK,IAAI,CAAC,CAAC,GAAG,CAAC,CAAC;QACzI,OAAO,IAAI,CAAC,OAAO,CAAC,YAAY,CAAC;YAC/B,KAAK,EAAE,IAAI,CAAC,MAAM,GAAG,GAAG,GAAG,KAAK;YAChC,yCAAyC;YACzC,IAAI,EAAE,IAAI,CAAC,GAAG,CAAC,UAAU,EAAE,EAAE,CAAC;YAC9B,KAAK,EAAE,cAAc,CAAC,OAAO,GAAG,cAAc,CAAC,QAAQ;SACxD,CAAC,CAAC;IACL,CAAC;IAEO,+BAA+B,CAAC,SAA8B;QACpE,MAAM,aAAa,GAAG,IAAI,CAAC,cAAc,CAAC;QAC1C,MAAM,SAAS,GAAG,IAAI,CAAC,aAAa,CAAC;QACrC,KAAK,MAAM,GAAG,IAAI,aAAa,EAAE,CAAC;YAChC,IAAI,CAAC,oBAAoB,CAAC,SAAS,CAAC,GAAG,CAAC,IAAI,EAAE,EAAE,SAAS,CAAC,GAAG,CAAC,IAAI,EAAE,CAAC,EAAE,CAAC;gBACtE,aAAa,CAAC,GAAG,CAAC,CAAC,OAAO,EAAE,CAAC;gBAC7B,OAAO,aAAa,CAAC,GAAG,CAAC,CAAC;YAC5B,CAAC;QACH,CAAC;IACH,CAAC;IAEO,oBAAoB,CAAC,QAAmB;QAC9C,MAAM,aAAa,GAAG,IAAI,CAAC,cAAc,CAAC;QAC1C,MAAM,MAAM,GAAG,IAAI,CAAC,aAAa,CAAC;QAClC,KAAK,MAAM,GAAG,IAAI,MAAM,EAAE,CAAC;YACzB,IAAI,CAAC,aAAa,CAAC,GAAG,CAAC,IAAI,CAAC,QAAQ,EAAE,QAAQ,CAAC,GAAG,CAAC,EAAE,CAAC;gBACpD,aAAa,CAAC,GAAG,CAAC,GAAG,IAAI,CAAC,aAAa,CAAC,GAAG,EAAE,MAAM,CAAC,GAAG,CAAC,CAAC,CAAC;YAC5D,CAAC;QACH,CAAC;IACH,CAAC;IAEO,eAAe,CACrB,WAAiD,EACjD,UAAgD;QAEhD,IAAI,CAAC,IAAI,CAAC,oBAAoB,EAAE,CAAC;YAC/B,OAAO;QACT,CAAC;QACD,IAAI,CAAC,oBAAoB,GAAG,KAAK,CAAC;QAClC,MAAM,MAAM,GAAG,IAAI,CAAC,OAAO,CAAC;QAC5B,MAAM,MAAM,GAAG,IAAI,CAAC,UAAU,CAAC,WAAW,EAAE,UAAU,CAAC,CAAC;QACxD,IAAI,MAAM,KAAK,IAAI,CAAC,OAAO,EAAE,CAAC;YAC5B,OAAO;QACT,CAAC;QACD,IAAI,CAAC,OAAO,GAAG,MAAM,CAAC;QACtB,IAAI,CAAC,SAAS,GAAG,oBAAoB,CAAC,MAAM,CAAC,CAAC,qBAAqB,CAAC;YAClE,KAAK,EAAE,IAAI,CAAC,MAAM;YAClB,MAAM,EAAE,MAAM;YACd,OAAO,EAAE;gBACP,MAAM,EAAE,oBAAoB,CAAC,MAAM,CAAC,CAAC,kBAAkB,CAAC;oBACtD,KAAK,EAAE,IAAI,CAAC,MAAM;oBAClB,IAAI,EAAE,MAAM;iBACb,CAAC;gBACF,UAAU,EAAE,MAAM;aACnB;SACF,CAAC,CAAC;QACH,IAAI,CAAC,iBAAiB,EAAE,CAAC;IAC3B,CAAC;IAEO,UAAU,CAChB,WAAiD,EACjD,UAAgD;QAEhD,MAAM,MAAM,GAAG,IAAI,CAAC,OAAO,CAAC;QAC5B,MAAM,QAAQ,GAAG,IAAI,CAAC,SAAS,CAAC;QAChC,IAAI,MAAM,GAAG,CAAC,CAAC;QACf,MAAM,YAAY,GAAG,CAAC,IAAI,CAAC,aAAa,GAAG;YACzC,MAAM,EAAE,CAAC;YACT,QAAQ,EAAE,CAAC;YACX,OAAO,EAAE,CAAC;SACX,CAAC,CAAC;QACH,MAAM,aAAa,GAAG,IAAI,CAAC,uBAAuB,EAAE,CAAC;QAErD,IAAI,MAAM,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;YACtB,MAAM,EAAE,CAAC;QACX,CAAC;QACD,IAAI,QAAQ,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;YACxB,YAAY,CAAC,QAAQ,GAAG,MAAM,CAAC;YAC/B,MAAM,EAAE,CAAC;QACX,CAAC;QAED,SAAS,WAAW,CAAC,MAA6B;YAChD,MAAM,EAAE,IAAI,GAAG,KAAK,EAAE,QAAQ,EAAE,GAAG,MAAM,CAAC;YAC1C,IAAI,IAAI,GAAG,QAAQ,KAAK,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,MAAM,QAAQ,GAAG,IAAI,CAAC,CAAC,CAAC,EAAE,CAAC;YAC9D,IAAI,MAAM,CAAC,MAAM,EAAE,CAAC;gBAClB,IAAI,GAAG,UAAU,IAAI,GAAG,CAAC;YAC3B,CAAC;YACD,OAAO,IAAI,CAAC;QACd,CAAC;QACD,YAAY,CAAC,OAAO,GAAG,MAAM,CAAC;QAC9B,MAAM,EAAE,GAAG,UAAU,CAAC;EACxB,MAAM;aACL,IAAI,EAAE;aACN,GAAG,CAAC,CAAC,SAAS,EAAE,GAAG,EAAE,EAAE;YACtB,MAAM,aAAa,GAAG,UAAU,YAAY,CAAC,MAAM,cAAc,GAAG,IAAI,CAAC;YACzE,MAAM,OAAO,GAAG,MAAM,SAAS,EAAE,CAAC;YAClC,6BAA6B;YAC7B,OAAO,UAAU,CAAC,SAAS,CAAC,KAAK,SAAS;gBACxC,CAAC,CAAC,GAAG,aAAa,QAAQ,OAAO,oBAAoB;gBACrD,CAAC,CAAC,GAAG,aAAa,uBAAuB,OAAO,WAAW,WAAW,CAClE,WAAW,CAAC,SAAS,CAAC,CACvB,IAAI,CAAC;QACZ,CAAC,CAAC;aACD,IAAI,CAAC,IAAI,CAAC;EACX,IAAI,CAAC,SAAS;aACb,GAAG,CACF,CAAC,OAAO,EAAE,GAAG,EAAE,EAAE,CACf,UAAU,YAAY,CAAC,QAAQ,cAAc,GAAG,kBAAkB,OAAO,CAAC,KAAK,KAAK,OAAO,CAAC,IAAI,GAAG,CACtG;aACA,IAAI,CAAC,IAAI,CAAC;;EAEX,IAAI,CAAC,QAAQ;aACZ,GAAG,CAAC,CAAC,IAAI,EAAE,GAAG,EAAE,EAAE;YACjB,OAAO,UACL,YAAY,CAAC,OACf,cAAc,GAAG,kCAAkC,IAAI,WAAW,WAAW,CAC3E,IAAI,CAAC,aAAa,CAAC,IAAI,CAAC,IAAI,EAAE,CAC/B,IAAI,CAAC;QACR,CAAC,CAAC;aACD,IAAI,CAAC,IAAI,CAAC;EACX,IAAI,CAAC,SAAS,IAAI,EAAE;2BACK,aAAa,CAAC,CAAC,CAAC,KAAK,aAAa,CAAC,CAAC,CAAC;;EAE9D,IAAI,CAAC,OAAO;;CAEb,CAAC;QAEE,OAAO,EAAE,CAAC;IACZ,CAAC;IAEO,iBAAiB;QACvB,MAAM,UAAU,GAAmB,EAAE,CAAC;QACtC,MAAM,MAAM,GAAG,IAAI,CAAC,OAAO,CAAC;QAC5B,MAAM,YAAY,GAAG,IAAI,CAAC,aAAa,CAAC;QAExC,IAAI,IAAI,CAAC,SAAS,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;YAC9B,UAAU,CAAC,YAAY,CAAC,QAAQ,CAAC,GAAG,MAAM,CAAC,eAAe,CAAC;gBACzD,KAAK,EAAE,IAAI,CAAC,MAAM,GAAG,WAAW;gBAChC,MAAM,EAAE,IAAI,CAAC,SAAS,CAAC,kBAAkB,CAAC,YAAY,CAAC,QAAQ,CAAC;gBAChE,OAAO,EAAE,IAAI,CAAC,SAAS,CAAC,GAAG,CACzB,CAAC,OAAO,EAAE,GAAG,EAAE,EAAE,CACf,CAAC;oBACC,OAAO,EAAE,GAAG;oBACZ,QAAQ,EAAE;wBACR,MAAM,EAAE,IAAI,CAAC,eAAe,CAAC,OAAO,CAAC,KAAK,CAAC;qBAC5C;iBACoB,CAAA,CAC1B;aACF,CAAC,CAAC;QACL,CAAC;QAED,IAAI,CAAC,WAAW,GAAG,UAAU,CAAC;IAChC,CAAC;IAED,UAAU,CACR,cAAiC,EACjC,MAAuC;IAEvC;;OAEG;IACH,OAA8B;QAE9B,IAAI,CAAC,oBAAoB,CAAC,MAAM,CAAC,IAAI,CAAC,OAAO,IAAI,EAAE,CAAC,CAAC,CAAC;QACtD,MAAM,aAAa,GAAG;YACpB,GAAG,IAAI,CAAC,cAAc;YACtB,GAAG,OAAO;SACkB,CAAC;QAE/B,MAAM,UAAU,GAAG,IAAI,CAAC,OAAO,CAAC,MAAM,CAAC,CAAC,GAAG,EAAE,SAAS,EAAE,EAAE;YACxD,GAAG,CAAC,SAAS,CAAC,GAAG,MAAM,CAAC,SAAc,CAAC,CAAC,MAAM,CAAC,CAAC,CAAC,QAAQ,CAAC,CAAC,CAAC,SAAS,CAAC;YACtE,OAAO,GAAG,CAAC;QACb,CAAC,EAAE,EAA0C,CAAC,CAAC;QAE/C,IAAI,CAAC,eAAe,CAAC,MAAM,EAAE,UAAU,CAAC,CAAC;QAEzC,MAAM,YAAY,GAAG,IAAI,CAAC,aAAa,CAAC;QACxC,IAAI,IAAI,CAAC,OAAO,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;YAC5B,IAAI,CAAC,WAAW,CAAC,YAAY,CAAC,MAAM,CAAC,GAAG,IAAI,CAAC,OAAO,CAAC,eAAe,CAAC;gBACnE,KAAK,EAAE,IAAI,CAAC,MAAM,GAAG,SAAS;gBAC9B,MAAM,EAAE,IAAI,CAAC,SAAS,CAAC,kBAAkB,CAAC,YAAY,CAAC,MAAM,CAAC;gBAC9D,OAAO,EAAE,IAAI,CAAC,OAAO,CAAC,GAAG,CAAC,CAAC,SAAS,EAAE,GAAG,EAAE,EAAE,CAAC,CAAC;oBAC7C,OAAO,EAAE,GAAG;oBACZ,QAAQ,EAAE,MAAM,CAAC,SAAc,CAAC,CAAC,MAAM;wBACrC,CAAC,CAAC;4BACE,MAAM,EAAE,MAAM,CAAC,SAAc,CAAC,CAAC,MAAO;yBACvC;wBACH,CAAC,CAAC,MAAM,CAAC,SAAc,CAAC,CAAC,OAAQ,CAAC,UAAU,EAAE;iBACjD,CAAC,CAAC;aACJ,CAAC,CAAC;QACL,CAAC;QAED,UAAU;QACV,IAAI,CAAC,WAAW,CAAC,YAAY,CAAC,OAAO,CAAC,GAAG,IAAI,CAAC,OAAO,CAAC,eAAe,CAAC;YACpE,KAAK,EAAE,IAAI,CAAC,MAAM,GAAG,UAAU;YAC/B,MAAM,EAAE,IAAI,CAAC,SAAS,CAAC,kBAAkB,CAAC,YAAY,CAAC,OAAO,CAAC;YAC/D,OAAO,EAAE,IAAI,CAAC,QAAQ,CAAC,GAAG,CAAC,CAAC,UAAU,EAAE,GAAG,EAAE,EAAE,CAAC,CAAC;gBAC/C,OAAO,EAAE,GAAG;gBACZ,QAAQ,EAAE;oBACR,MAAM,EAAE,aAAa,CAAC,UAAU,CAAC;iBAClC;aACF,CAAC,CAAC;SACJ,CAAC,CAAC;QAEH,wBAAwB;QACxB,MAAM,WAAW,GAAG,cAAc,CAAC,gBAAgB,EAAE,CAAC;QAEtD,mBAAmB;QACnB,WAAW,CAAC,WAAW,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC;QACxC,cAAc;QACd,IAAI,CAAC,WAAW,CAAC,OAAO,CAAC,CAAC,SAAS,EAAE,GAAG,EAAE,EAAE;YAC1C,WAAW,CAAC,YAAY,CAAC,GAAG,EAAE,SAAS,CAAC,CAAC;QAC3C,CAAC,CAAC,CAAC;QAEH,MAAM,aAAa,GAAG,IAAI,CAAC,uBAAuB,EAAE,CAAC;QACrD,WAAW,CAAC,kBAAkB,CAC5B,IAAI,CAAC,IAAI,CAAC,CAAC,IAAI,CAAC,UAAU,IAAI,IAAI,CAAC,MAAM,CAAC,GAAG,aAAa,CAAC,CAAC,CAAC,CAAC,EAC9D,IAAI,CAAC,IAAI,CAAC,CAAC,IAAI,CAAC,WAAW,IAAI,IAAI,CAAC,OAAO,CAAC,GAAG,aAAa,CAAC,CAAC,CAAC,CAAC,EAChE,CAAC,CACF,CAAC;QACF,sBAAsB;QACtB,WAAW,CAAC,GAAG,EAAE,CAAC;IACpB,CAAC;CACF","sourcesContent":["import { getTargetShaderCache } from './WGPUShaderCache.js';\n// TODO output read write buffer\nimport { getUniformBuffer, type Uniform } from './WGPUFullQuadPass.js';\n\nexport type { Uniform };\n\nfunction isStorageParamsEqual(\n  params: WGPUComputePassOutput,\n  other: WGPUComputePassOutput\n) {\n  return (\n    params.channels === other.channels &&\n    params.type === other.type &&\n    params.count === other.count &&\n    params.atomic === other.atomic\n  );\n}\n\nexport const WORKGROUP_SIZE = 8;\n\nexport interface WGPUComputePassInput {\n  buffer?: GPUBuffer;\n  texture?: GPUTexture;\n  channels: number;\n  type?: 'u32' | 'i32' | 'f32';\n}\n\nexport interface WGPUComputePassOutput {\n  channels?: number;\n  // Default: 1. Count per element.\n  count?: number;\n  atomic?: boolean;\n  type?: 'u32' | 'i32' | 'f32';\n}\n\nexport class WGPUComputePass<I extends string, O extends string> {\n  private _label;\n\n  private _device;\n  private _outputBuffers: Record<string, GPUBuffer> = {};\n  private _outputParams: Record<string, WGPUComputePassOutput> = {};\n\n  private _pipeline!: GPUComputePipeline;\n  private _bindGroups: GPUBindGroup[] = [];\n  private _needsUpdatePipeline = true;\n\n  private _inputs: string[] = [];\n  private _outputs: string[] = [];\n  private _uniforms: Uniform[] = [];\n  private _uniformsMap: Record<string, Uniform> = {};\n  private _uniformBuffers: Record<string, GPUBuffer> = {};\n\n  private _capacityWidth = 0;\n  private _capacityHeight = 0;\n  private _reserveCapacity = false;\n  private _width = 10;\n  private _height = 10;\n\n  private _execWidth?: number;\n  private _execHeight?: number;\n\n  private _workGroupSize = [WORKGROUP_SIZE, WORKGROUP_SIZE];\n\n  private _csCode = '';\n  private _csMain;\n  private _csDefine;\n\n  private _groupOffsets = {\n    inputs: 0,\n    uniforms: 1,\n    outputs: 2\n  };\n\n  constructor(\n    label: string,\n    device: GPUDevice,\n    opts: {\n      inputs: I[];\n      outputs: O[];\n      csMain: string;\n      csDefine?: string;\n      uniforms?: Uniform[];\n    }\n  ) {\n    this._label = label;\n    this._device = device;\n    this._csMain = opts.csMain;\n    this._csDefine = opts.csDefine;\n    this._inputs = opts.inputs;\n    this._outputs = opts.outputs;\n    const uniforms = opts.uniforms || [];\n    this._uniforms = uniforms;\n\n    this._uniformsMap = this._uniforms.reduce((map, uniform) => {\n      map[uniform.label] = uniform;\n      return map;\n    }, {} as Record<string, Uniform>);\n    uniforms.forEach((uniform) => {\n      const data = getUniformBuffer(uniform.data, uniform.type);\n      this._uniformBuffers[uniform.label] = device.createBuffer({\n        label: this._label + '.' + uniform.label,\n        size: data.byteLength,\n        usage: GPUBufferUsage.UNIFORM | GPUBufferUsage.COPY_DST\n      });\n      this._device.queue.writeBuffer(\n        this._uniformBuffers[uniform.label],\n        0,\n        data\n      );\n    });\n  }\n\n  setCSCode({ csDefine, csMain }: { csDefine: string; csMain: string }) {\n    this._csDefine = csDefine;\n    this._csMain = csMain;\n    this._needsUpdatePipeline = true;\n  }\n\n  reserveCapacity() { this._reserveCapacity = true; this._capacityWidth = this._width; this._capacityHeight = this._height; }\n  setSize(width: number, height: number) {\n    width = Math.max(1, Math.ceil(width));\n    height = Math.max(1, Math.ceil(height));\n    const sizeChanged = width !== this._width || height !== this._height;\n    this._width = width;\n    this._height = height;\n    if (sizeChanged) {\n      this._needsUpdatePipeline = true;\n      const nextWidth = Math.max(this._capacityWidth, width), nextHeight = Math.max(this._capacityHeight, height);\n      const capacityChanged = nextWidth !== this._capacityWidth || nextHeight !== this._capacityHeight;\n      this._capacityWidth = nextWidth; this._capacityHeight = nextHeight;\n      if (this._reserveCapacity && !capacityChanged) return;\n      // Invalidate all buffers\n      for (const key in this._outputBuffers) {\n        this._outputBuffers[key].destroy();\n      }\n      this._outputBuffers = {};\n    }\n  }\n\n  setExecuteSize(width: number, height: number) {\n    width = Math.max(1, Math.ceil(width));\n    height = Math.max(1, Math.ceil(height));\n    this._execWidth = width;\n    this._execHeight = height;\n  }\n\n  setOutputParams(outputParams: Record<O, WGPUComputePassOutput>) {\n    // TODO lazy update buffers\n    this._invalidateChangedOutputBuffers(outputParams);\n    this._outputParams = outputParams;\n    this._needsUpdatePipeline = true;\n  }\n\n  setUniform(label: string, data: Uniform['data']) {\n    const buffer = this._uniformBuffers[label];\n    this._device.queue.writeBuffer(\n      buffer,\n      0,\n      getUniformBuffer(data, this._uniformsMap[label].type)\n    );\n  }\n\n  getOutput(name: O) {\n    const outputBuffers = this._outputBuffers;\n    if (!outputBuffers[name]) {\n      outputBuffers[name] = this._createBuffer(\n        name,\n        this._outputParams[name] || {}\n      );\n    }\n    return outputBuffers[name];\n  }\n\n  getMemoryBytes() {\n    return [...Object.values(this._uniformBuffers), ...Object.values(this._outputBuffers)]\n      .reduce((sum, buffer) => sum + buffer.size, 0);\n  }\n\n  dispose() {\n    Object.keys(this._uniformBuffers).forEach((key) => {\n      (this._uniformBuffers as any)[key].destroy();\n    });\n    Object.keys(this._outputBuffers).forEach((key) => {\n      (this._outputBuffers as any)[key].destroy();\n    });\n  }\n\n  setWorkGroupSize(width: number, height: number) {\n    this._needsUpdatePipeline =\n      width !== this._workGroupSize[0] || height !== this._workGroupSize[1];\n    this._workGroupSize = [width, height];\n  }\n\n  private _getActualWorkgroupSize() {\n    return [\n      Math.min(this._workGroupSize[0], this._execWidth ?? this._width),\n      Math.min(this._workGroupSize[1], this._execHeight ?? this._height)\n    ];\n  }\n\n  private _createBuffer(label: string, params: WGPUComputePassOutput) {\n    const channels = params.channels ?? 4;\n    const byteLength =\n      (this._reserveCapacity ? this._capacityWidth * this._capacityHeight : this._width * this._height) * channels * (params.count ?? 1) * 4;\n    return this._device.createBuffer({\n      label: this._label + '.' + label,\n      // webgpu needs buffer at least 80 bytes.\n      size: Math.max(byteLength, 80),\n      usage: GPUBufferUsage.STORAGE | GPUBufferUsage.COPY_SRC\n    });\n  }\n\n  private _invalidateChangedOutputBuffers(newParams: Record<string, any>) {\n    const outputBuffers = this._outputBuffers;\n    const oldParams = this._outputParams;\n    for (const key in outputBuffers) {\n      if (!isStorageParamsEqual(newParams[key] || {}, oldParams[key] || {})) {\n        outputBuffers[key].destroy();\n        delete outputBuffers[key];\n      }\n    }\n  }\n\n  private _updateOutputBuffers(excludes?: string[]) {\n    const outputBuffers = this._outputBuffers;\n    const params = this._outputParams;\n    for (const key in params) {\n      if (!outputBuffers[key] && !excludes?.includes(key)) {\n        outputBuffers[key] = this._createBuffer(key, params[key]);\n      }\n    }\n  }\n\n  private _updatePipeline(\n    inputParams: Record<string, WGPUComputePassInput>,\n    inputTypes: Record<string, 'texture' | 'buffer'>\n  ) {\n    if (!this._needsUpdatePipeline) {\n      return;\n    }\n    this._needsUpdatePipeline = false;\n    const device = this._device;\n    const csCode = this._getFullCs(inputParams, inputTypes);\n    if (csCode === this._csCode) {\n      return;\n    }\n    this._csCode = csCode;\n    this._pipeline = getTargetShaderCache(device).createComputePipeline({\n      label: this._label,\n      layout: 'auto',\n      compute: {\n        module: getTargetShaderCache(device).createShaderModule({\n          label: this._label,\n          code: csCode\n        }),\n        entryPoint: 'main'\n      }\n    });\n    this._updateBindGroups();\n  }\n\n  private _getFullCs(\n    inputParams: Record<string, WGPUComputePassInput>,\n    inputTypes: Record<string, 'texture' | 'buffer'>\n  ) {\n    const inputs = this._inputs;\n    const uniforms = this._uniforms;\n    let offset = 0;\n    const groupOffsets = (this._groupOffsets = {\n      inputs: 0,\n      uniforms: 0,\n      outputs: 0\n    });\n    const workgroupSize = this._getActualWorkgroupSize();\n\n    if (inputs.length > 0) {\n      offset++;\n    }\n    if (uniforms.length > 0) {\n      groupOffsets.uniforms = offset;\n      offset++;\n    }\n\n    function getTypeCode(params: WGPUComputePassOutput) {\n      const { type = 'f32', channels } = params;\n      let code = channels === 1 ? type : `vec${channels}${type[0]}`;\n      if (params.atomic) {\n        code = `atomic<${code}>`;\n      }\n      return code;\n    }\n    groupOffsets.outputs = offset;\n    const cs = /* wgsl */ `\n${inputs\n  .sort()\n  .map((inputName, idx) => {\n    const bindingPrefix = `@group(${groupOffsets.inputs}) @binding(${idx}) `;\n    const varName = `in_${inputName}`;\n    // TODO more channels option.\n    return inputTypes[inputName] === 'texture'\n      ? `${bindingPrefix} var ${varName}: texture_2d<f32>;`\n      : `${bindingPrefix} var<storage, read> ${varName}: array<${getTypeCode(\n          inputParams[inputName]\n        )}>;`;\n  })\n  .join('\\n')}\n${this._uniforms\n  .map(\n    (uniform, idx) =>\n      `@group(${groupOffsets.uniforms}) @binding(${idx}) var<uniform> ${uniform.label}: ${uniform.type};`\n  )\n  .join('\\n')}\n\n${this._outputs\n  .map((name, idx) => {\n    return `@group(${\n      groupOffsets.outputs\n    }) @binding(${idx}) var<storage, read_write> out_${name}: array<${getTypeCode(\n      this._outputParams[name] || {}\n    )}>;`;\n  })\n  .join('\\n')}\n${this._csDefine ?? ''}\n@compute @workgroup_size(${workgroupSize[0]}, ${workgroupSize[1]}, 1)\nfn main(@builtin(global_invocation_id) globalId: vec3u) {\n${this._csMain}\n}\n`;\n\n    return cs;\n  }\n\n  private _updateBindGroups() {\n    const bindGroups: GPUBindGroup[] = [];\n    const device = this._device;\n    const groupOffsets = this._groupOffsets;\n\n    if (this._uniforms.length > 0) {\n      bindGroups[groupOffsets.uniforms] = device.createBindGroup({\n        label: this._label + '.uniforms',\n        layout: this._pipeline.getBindGroupLayout(groupOffsets.uniforms),\n        entries: this._uniforms.map(\n          (uniform, idx) =>\n            ({\n              binding: idx,\n              resource: {\n                buffer: this._uniformBuffers[uniform.label]\n              }\n            } as GPUBindGroupEntry)\n        )\n      });\n    }\n\n    this._bindGroups = bindGroups;\n  }\n\n  createPass(\n    commandEncoder: GPUCommandEncoder,\n    inputs: Record<I, WGPUComputePassInput>,\n\n    /**\n     * Explict outputs. Used for buffers reuse.\n     */\n    outputs?: Record<O, GPUBuffer>\n  ) {\n    this._updateOutputBuffers(Object.keys(outputs || {}));\n    const outputBuffers = {\n      ...this._outputBuffers,\n      ...outputs\n    } as Record<string, GPUBuffer>;\n\n    const inputTypes = this._inputs.reduce((obj, inputName) => {\n      obj[inputName] = inputs[inputName as I].buffer ? 'buffer' : 'texture';\n      return obj;\n    }, {} as Record<string, 'texture' | 'buffer'>);\n\n    this._updatePipeline(inputs, inputTypes);\n\n    const groupOffsets = this._groupOffsets;\n    if (this._inputs.length > 0) {\n      this._bindGroups[groupOffsets.inputs] = this._device.createBindGroup({\n        label: this._label + '.inputs',\n        layout: this._pipeline.getBindGroupLayout(groupOffsets.inputs),\n        entries: this._inputs.map((inputName, idx) => ({\n          binding: idx,\n          resource: inputs[inputName as I].buffer\n            ? {\n                buffer: inputs[inputName as I].buffer!\n              }\n            : inputs[inputName as I].texture!.createView()\n        }))\n      });\n    }\n\n    // Outputs\n    this._bindGroups[groupOffsets.outputs] = this._device.createBindGroup({\n      label: this._label + '.outputs',\n      layout: this._pipeline.getBindGroupLayout(groupOffsets.outputs),\n      entries: this._outputs.map((bufferName, idx) => ({\n        binding: idx,\n        resource: {\n          buffer: outputBuffers[bufferName]\n        }\n      }))\n    });\n\n    // Begin the render pass\n    const computePass = commandEncoder.beginComputePass();\n\n    // Draw a full quad\n    computePass.setPipeline(this._pipeline);\n    // Bind groups\n    this._bindGroups.forEach((bindGroup, idx) => {\n      computePass.setBindGroup(idx, bindGroup);\n    });\n\n    const workgroupSize = this._getActualWorkgroupSize();\n    computePass.dispatchWorkgroups(\n      Math.ceil((this._execWidth ?? this._width) / workgroupSize[0]),\n      Math.ceil((this._execHeight ?? this._height) / workgroupSize[1]),\n      1\n    );\n    // End the render pass\n    computePass.end();\n  }\n}\n"]}

package/lib/renderer/WGPUEnvMapDistribution.d.ts

@@ -0,0 +1,12 @@
+/** Builds the existing CDF/PDF texture layout without CPU readback or upload. */
+export declare class WGPUEnvMapDistribution {
+    private readonly device;
+    private readonly layout;
+    private readonly pipelines;
+    private pixels?;
+    private rows?;
+    static supported(device: GPUDevice, width: number, height: number): boolean;
+    constructor(device: GPUDevice);
+    build(input: GPUTexture, output: GPUTexture): void;
+    dispose(): void;
+}

package/lib/renderer/WGPUEnvMapDistribution.js

@@ -0,0 +1,150 @@
+const distributionWGSL = /* wgsl */ `
+@group(0) @binding(0) var radiance: texture_2d<f32>;
+
+@group(0) @binding(1) var<storage, read_write> pixels: array<vec2f>;
+@group(0) @binding(2) var<storage, read_write> rows: array<vec2f>;
+@group(0) @binding(3) var output: texture_storage_2d<rg32float, write>;
+var<workgroup> scan: array<f32, 256>;
+var<workgroup> tw: f32;
+
+fn rl(pZ: u32) {
+workgroupBarrier();
+for (var offset = 1u; offset < 256u; offset *= 2u) {
+var previous = 0.0;
+if (pZ >= offset) { previous = scan[pZ - offset]; }
+workgroupBarrier();
+scan[pZ] += previous;
+workgroupBarrier();
+}
+}
+
+@compute @workgroup_size(256)
+fn scanRows(@builtin(workgroup_id) group: vec3u, @builtin(local_invocation_index) pZ: u32) {
+let size = textureDimensions(radiance);
+let y = group.x;
+if (pZ == 0u) { tw = 0.0; }
+workgroupBarrier();
+for (var start = 0u; start < size.x; start += 256u) {
+let x = start + pZ;
+var weight = 0.0;
+if (x < size.x) {
+
+var maximum = 0.0;
+for (var dy = -1; dy <= 1; dy++) {
+for (var dx = -1; dx <= 1; dx++) {
+let p = clamp(vec2i(i32(x) + dx, i32(y) + dy), vec2i(0), vec2i(size) - 1);
+maximum = max(maximum, dot(textureLoad(radiance, p, 0).rgb, vec3f(0.2126, 0.7152, 0.0722)));
+}
+}
+weight = max(maximum, 1e-12) * sin(3.141592653589793 * (f32(y) + 0.5) / f32(size.y));
+}
+scan[pZ] = weight / f32(size.x);
+rl(pZ);
+if (x < size.x) { pixels[y * size.x + x] = vec2f(tw + scan[pZ], weight); }
+workgroupBarrier();
+if (pZ == 0u) { tw += scan[255]; }
+workgroupBarrier();
+}
+if (pZ == 0u) { rows[y] = vec2f(0.0, tw); }
+}
+
+@compute @workgroup_size(256)
+fn scanMarginal(@builtin(local_invocation_index) pZ: u32) {
+let height = textureDimensions(radiance).y;
+if (pZ == 0u) { tw = 0.0; }
+workgroupBarrier();
+for (var start = 0u; start < height; start += 256u) {
+let y = start + pZ;
+var integral = 0.0;
+if (y < height) { integral = rows[y].y; }
+scan[pZ] = integral / f32(height);
+rl(pZ);
+if (y < height) { rows[y].x = tw + scan[pZ]; }
+workgroupBarrier();
+if (pZ == 0u) { tw += scan[255]; }
+workgroupBarrier();
+}
+}
+
+@compute @workgroup_size(16, 16)
+fn normalize(@builtin(global_invocation_id) id: vec3u) {
+let size = textureDimensions(radiance);
+let x = id.x;
+let y = id.y;
+if (x >= size.x + 2u || y >= size.y + 1u) { return; }
+var value = vec2f(0.0);
+if (x == 0u) {
+let total = rows[size.y - 1u].x;
+if (y > 0u) { value.x = rows[y - 1u].x / total; }
+if (y < size.y) { value.y = rows[y].y / total; }
+} else if (y < size.y) {
+let integral = rows[y].y;
+if (x > 1u) { value.x = pixels[y * size.x + x - 2u].x / integral; }
+if (x <= size.x) { value.y = pixels[y * size.x + x - 1u].y / integral; }
+}
+textureStore(output, vec2i(id.xy), vec4f(value, 0.0, 0.0));
+}
+`;
+/** Builds the existing CDF/PDF texture layout without CPU readback or upload. */
+export class WGPUEnvMapDistribution {
+    device;
+    layout;
+    pipelines;
+    pixels;
+    rows;
+    static supported(device, width, height) {
+        // Large HDRIs can exceed a single storage binding; retain the CPU path there.
+        return width * height * 8 <= Math.min(device.limits.maxStorageBufferBindingSize, device.limits.maxBufferSize);
+    }
+    constructor(device) {
+        this.device = device;
+        this.layout = device.createBindGroupLayout({
+            label: 'Environment distribution',
+            entries: [
+                { binding: 0, visibility: GPUShaderStage.COMPUTE, texture: { sampleType: 'unfilterable-float' } },
+                { binding: 1, visibility: GPUShaderStage.COMPUTE, buffer: { type: 'storage' } },
+                { binding: 2, visibility: GPUShaderStage.COMPUTE, buffer: { type: 'storage' } },
+                { binding: 3, visibility: GPUShaderStage.COMPUTE, storageTexture: { access: 'write-only', format: 'rg32float' } }
+            ]
+        });
+        const module = device.createShaderModule({ label: 'Environment distribution', code: distributionWGSL });
+        const layout = device.createPipelineLayout({ bindGroupLayouts: [this.layout] });
+        this.pipelines = ['scanRows', 'scanMarginal', 'normalize'].map(entryPoint => device.createComputePipeline({
+            label: `Environment distribution ${entryPoint}`, layout, compute: { module, entryPoint }
+        }));
+    }
+    build(input, output) {
+        const { width, height } = input;
+        if (this.pixels?.size !== width * height * 8 || this.rows?.size !== height * 8) {
+            this.dispose();
+            this.pixels = this.device.createBuffer({ label: 'Environment row scans', size: width * height * 8, usage: GPUBufferUsage.STORAGE });
+            this.rows = this.device.createBuffer({ label: 'Environment marginal scan', size: height * 8, usage: GPUBufferUsage.STORAGE });
+        }
+        const group = this.device.createBindGroup({
+            layout: this.layout,
+            entries: [
+                { binding: 0, resource: input.createView() },
+                { binding: 1, resource: { buffer: this.pixels } },
+                { binding: 2, resource: { buffer: this.rows } },
+                { binding: 3, resource: output.createView() }
+            ]
+        });
+        const encoder = this.device.createCommandEncoder({ label: 'Environment distribution' });
+        const workgroups = [[height, 1], [1, 1], [Math.ceil((width + 2) / 16), Math.ceil((height + 1) / 16)]];
+        this.pipelines.forEach((pipeline, index) => {
+            const pass = encoder.beginComputePass();
+            pass.setPipeline(pipeline);
+            pass.setBindGroup(0, group);
+            pass.dispatchWorkgroups(...workgroups[index]);
+            pass.end();
+        });
+        this.device.queue.submit([encoder.finish()]);
+    }
+    dispose() {
+        this.pixels?.destroy();
+        this.rows?.destroy();
+        this.pixels = undefined;
+        this.rows = undefined;
+    }
+}
+//# sourceMappingURL=WGPUEnvMapDistribution.js.map

package/lib/renderer/WGPUEnvMapDistribution.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"WGPUEnvMapDistribution.js","sourceRoot":"","sources":["../../src/renderer/WGPUEnvMapDistribution.ts"],"names":[],"mappings":"AAAA,MAAM,gBAAgB,GAAG,UAAU,CAAC;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;CAsFnC,CAAC;AAEF,iFAAiF;AACjF,MAAM,OAAO,sBAAsB;IAWJ;IAVZ,MAAM,CAAqB;IAC3B,SAAS,CAAuB;IACzC,MAAM,CAAa;IACnB,IAAI,CAAa;IAEzB,MAAM,CAAC,SAAS,CAAC,MAAiB,EAAE,KAAa,EAAE,MAAc;QAC/D,8EAA8E;QAC9E,OAAO,KAAK,GAAG,MAAM,GAAG,CAAC,IAAI,IAAI,CAAC,GAAG,CAAC,MAAM,CAAC,MAAM,CAAC,2BAA2B,EAAE,MAAM,CAAC,MAAM,CAAC,aAAa,CAAC,CAAC;IAChH,CAAC;IAED,YAA6B,MAAiB;QAAjB,WAAM,GAAN,MAAM,CAAW;QAC5C,IAAI,CAAC,MAAM,GAAG,MAAM,CAAC,qBAAqB,CAAC;YACzC,KAAK,EAAE,0BAA0B;YACjC,OAAO,EAAE;gBACP,EAAE,OAAO,EAAE,CAAC,EAAE,UAAU,EAAE,cAAc,CAAC,OAAO,EAAE,OAAO,EAAE,EAAE,UAAU,EAAE,oBAAoB,EAAE,EAAE;gBACjG,EAAE,OAAO,EAAE,CAAC,EAAE,UAAU,EAAE,cAAc,CAAC,OAAO,EAAE,MAAM,EAAE,EAAE,IAAI,EAAE,SAAS,EAAE,EAAE;gBAC/E,EAAE,OAAO,EAAE,CAAC,EAAE,UAAU,EAAE,cAAc,CAAC,OAAO,EAAE,MAAM,EAAE,EAAE,IAAI,EAAE,SAAS,EAAE,EAAE;gBAC/E,EAAE,OAAO,EAAE,CAAC,EAAE,UAAU,EAAE,cAAc,CAAC,OAAO,EAAE,cAAc,EAAE,EAAE,MAAM,EAAE,YAAY,EAAE,MAAM,EAAE,WAAW,EAAE,EAAE;aAClH;SACF,CAAC,CAAC;QACH,MAAM,MAAM,GAAG,MAAM,CAAC,kBAAkB,CAAC,EAAE,KAAK,EAAE,0BAA0B,EAAE,IAAI,EAAE,gBAAgB,EAAE,CAAC,CAAC;QACxG,MAAM,MAAM,GAAG,MAAM,CAAC,oBAAoB,CAAC,EAAE,gBAAgB,EAAE,CAAC,IAAI,CAAC,MAAM,CAAC,EAAE,CAAC,CAAC;QAChF,IAAI,CAAC,SAAS,GAAG,CAAC,UAAU,EAAE,cAAc,EAAE,WAAW,CAAC,CAAC,GAAG,CAAC,UAAU,CAAC,EAAE,CAAC,MAAM,CAAC,qBAAqB,CAAC;YACxG,KAAK,EAAE,4BAA4B,UAAU,EAAE,EAAE,MAAM,EAAE,OAAO,EAAE,EAAE,MAAM,EAAE,UAAU,EAAE;SACzF,CAAC,CAAC,CAAC;IACN,CAAC;IAED,KAAK,CAAC,KAAiB,EAAE,MAAkB;QACzC,MAAM,EAAE,KAAK,EAAE,MAAM,EAAE,GAAG,KAAK,CAAC;QAChC,IAAI,IAAI,CAAC,MAAM,EAAE,IAAI,KAAK,KAAK,GAAG,MAAM,GAAG,CAAC,IAAI,IAAI,CAAC,IAAI,EAAE,IAAI,KAAK,MAAM,GAAG,CAAC,EAAE,CAAC;YAC/E,IAAI,CAAC,OAAO,EAAE,CAAC;YACf,IAAI,CAAC,MAAM,GAAG,IAAI,CAAC,MAAM,CAAC,YAAY,CAAC,EAAE,KAAK,EAAE,uBAAuB,EAAE,IAAI,EAAE,KAAK,GAAG,MAAM,GAAG,CAAC,EAAE,KAAK,EAAE,cAAc,CAAC,OAAO,EAAE,CAAC,CAAC;YACpI,IAAI,CAAC,IAAI,GAAG,IAAI,CAAC,MAAM,CAAC,YAAY,CAAC,EAAE,KAAK,EAAE,2BAA2B,EAAE,IAAI,EAAE,MAAM,GAAG,CAAC,EAAE,KAAK,EAAE,cAAc,CAAC,OAAO,EAAE,CAAC,CAAC;QAChI,CAAC;QACD,MAAM,KAAK,GAAG,IAAI,CAAC,MAAM,CAAC,eAAe,CAAC;YACxC,MAAM,EAAE,IAAI,CAAC,MAAM;YACnB,OAAO,EAAE;gBACP,EAAE,OAAO,EAAE,CAAC,EAAE,QAAQ,EAAE,KAAK,CAAC,UAAU,EAAE,EAAE;gBAC5C,EAAE,OAAO,EAAE,CAAC,EAAE,QAAQ,EAAE,EAAE,MAAM,EAAE,IAAI,CAAC,MAAO,EAAE,EAAE;gBAClD,EAAE,OAAO,EAAE,CAAC,EAAE,QAAQ,EAAE,EAAE,MAAM,EAAE,IAAI,CAAC,IAAK,EAAE,EAAE;gBAChD,EAAE,OAAO,EAAE,CAAC,EAAE,QAAQ,EAAE,MAAM,CAAC,UAAU,EAAE,EAAE;aAC9C;SACF,CAAC,CAAC;QACH,MAAM,OAAO,GAAG,IAAI,CAAC,MAAM,CAAC,oBAAoB,CAAC,EAAE,KAAK,EAAE,0BAA0B,EAAE,CAAC,CAAC;QACxF,MAAM,UAAU,GAAG,CAAC,CAAC,MAAM,EAAE,CAAC,CAAC,EAAE,CAAC,CAAC,EAAE,CAAC,CAAC,EAAE,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC,KAAK,GAAG,CAAC,CAAC,GAAG,EAAE,CAAC,EAAE,IAAI,CAAC,IAAI,CAAC,CAAC,MAAM,GAAG,CAAC,CAAC,GAAG,EAAE,CAAC,CAAC,CAAC,CAAC;QACtG,IAAI,CAAC,SAAS,CAAC,OAAO,CAAC,CAAC,QAAQ,EAAE,KAAK,EAAE,EAAE;YACzC,MAAM,IAAI,GAAG,OAAO,CAAC,gBAAgB,EAAE,CAAC;YACxC,IAAI,CAAC,WAAW,CAAC,QAAQ,CAAC,CAAC;YAC3B,IAAI,CAAC,YAAY,CAAC,CAAC,EAAE,KAAK,CAAC,CAAC;YAC5B,IAAI,CAAC,kBAAkB,CAAC,GAAG,UAAU,CAAC,KAAK,CAAqB,CAAC,CAAC;YAClE,IAAI,CAAC,GAAG,EAAE,CAAC;QACb,CAAC,CAAC,CAAC;QACH,IAAI,CAAC,MAAM,CAAC,KAAK,CAAC,MAAM,CAAC,CAAC,OAAO,CAAC,MAAM,EAAE,CAAC,CAAC,CAAC;IAC/C,CAAC;IAED,OAAO;QACL,IAAI,CAAC,MAAM,EAAE,OAAO,EAAE,CAAC;QACvB,IAAI,CAAC,IAAI,EAAE,OAAO,EAAE,CAAC;QACrB,IAAI,CAAC,MAAM,GAAG,SAAS,CAAC;QACxB,IAAI,CAAC,IAAI,GAAG,SAAS,CAAC;IACxB,CAAC;CACF","sourcesContent":["const distributionWGSL = /* wgsl */ `\n@group(0) @binding(0) var radiance: texture_2d<f32>;\n// Per-texel inclusive row integral and unnormalized PDF, then per-row CDF/PDF.\n@group(0) @binding(1) var<storage, read_write> pixels: array<vec2f>;\n@group(0) @binding(2) var<storage, read_write> rows: array<vec2f>;\n@group(0) @binding(3) var output: texture_storage_2d<rg32float, write>;\nvar<workgroup> scan: array<f32, 256>;\nvar<workgroup> carry: f32;\n\nfn inclusiveScan(lane: u32) {\n  workgroupBarrier();\n  for (var offset = 1u; offset < 256u; offset *= 2u) {\n    var previous = 0.0;\n    if (lane >= offset) { previous = scan[lane - offset]; }\n    workgroupBarrier();\n    scan[lane] += previous;\n    workgroupBarrier();\n  }\n}\n\n@compute @workgroup_size(256)\nfn scanRows(@builtin(workgroup_id) group: vec3u, @builtin(local_invocation_index) lane: u32) {\n  let size = textureDimensions(radiance);\n  let y = group.x;\n  if (lane == 0u) { carry = 0.0; }\n  workgroupBarrier();\n  for (var start = 0u; start < size.x; start += 256u) {\n    let x = start + lane;\n    var weight = 0.0;\n    if (x < size.x) {\n      // Match the CPU proposal's clamped 3×3 footprint around filtered HDR radiance.\n      var maximum = 0.0;\n      for (var dy = -1; dy <= 1; dy++) {\n        for (var dx = -1; dx <= 1; dx++) {\n          let p = clamp(vec2i(i32(x) + dx, i32(y) + dy), vec2i(0), vec2i(size) - 1);\n          maximum = max(maximum, dot(textureLoad(radiance, p, 0).rgb, vec3f(0.2126, 0.7152, 0.0722)));\n        }\n      }\n      weight = max(maximum, 1e-12) * sin(3.141592653589793 * (f32(y) + 0.5) / f32(size.y));\n    }\n    scan[lane] = weight / f32(size.x);\n    inclusiveScan(lane);\n    if (x < size.x) { pixels[y * size.x + x] = vec2f(carry + scan[lane], weight); }\n    workgroupBarrier();\n    if (lane == 0u) { carry += scan[255]; }\n    workgroupBarrier();\n  }\n  if (lane == 0u) { rows[y] = vec2f(0.0, carry); }\n}\n\n@compute @workgroup_size(256)\nfn scanMarginal(@builtin(local_invocation_index) lane: u32) {\n  let height = textureDimensions(radiance).y;\n  if (lane == 0u) { carry = 0.0; }\n  workgroupBarrier();\n  for (var start = 0u; start < height; start += 256u) {\n    let y = start + lane;\n    var integral = 0.0;\n    if (y < height) { integral = rows[y].y; }\n    scan[lane] = integral / f32(height);\n    inclusiveScan(lane);\n    if (y < height) { rows[y].x = carry + scan[lane]; }\n    workgroupBarrier();\n    if (lane == 0u) { carry += scan[255]; }\n    workgroupBarrier();\n  }\n}\n\n@compute @workgroup_size(16, 16)\nfn normalize(@builtin(global_invocation_id) id: vec3u) {\n  let size = textureDimensions(radiance);\n  let x = id.x;\n  let y = id.y;\n  if (x >= size.x + 2u || y >= size.y + 1u) { return; }\n  var value = vec2f(0.0);\n  if (x == 0u) {\n    let total = rows[size.y - 1u].x;\n    if (y > 0u) { value.x = rows[y - 1u].x / total; }\n    if (y < size.y) { value.y = rows[y].y / total; }\n  } else if (y < size.y) {\n    let integral = rows[y].y;\n    if (x > 1u) { value.x = pixels[y * size.x + x - 2u].x / integral; }\n    if (x <= size.x) { value.y = pixels[y * size.x + x - 1u].y / integral; }\n  }\n  textureStore(output, vec2i(id.xy), vec4f(value, 0.0, 0.0));\n}\n`;\n\n/** Builds the existing CDF/PDF texture layout without CPU readback or upload. */\nexport class WGPUEnvMapDistribution {\n  private readonly layout: GPUBindGroupLayout;\n  private readonly pipelines: GPUComputePipeline[];\n  private pixels?: GPUBuffer;\n  private rows?: GPUBuffer;\n\n  static supported(device: GPUDevice, width: number, height: number) {\n    // Large HDRIs can exceed a single storage binding; retain the CPU path there.\n    return width * height * 8 <= Math.min(device.limits.maxStorageBufferBindingSize, device.limits.maxBufferSize);\n  }\n\n  constructor(private readonly device: GPUDevice) {\n    this.layout = device.createBindGroupLayout({\n      label: 'Environment distribution',\n      entries: [\n        { binding: 0, visibility: GPUShaderStage.COMPUTE, texture: { sampleType: 'unfilterable-float' } },\n        { binding: 1, visibility: GPUShaderStage.COMPUTE, buffer: { type: 'storage' } },\n        { binding: 2, visibility: GPUShaderStage.COMPUTE, buffer: { type: 'storage' } },\n        { binding: 3, visibility: GPUShaderStage.COMPUTE, storageTexture: { access: 'write-only', format: 'rg32float' } }\n      ]\n    });\n    const module = device.createShaderModule({ label: 'Environment distribution', code: distributionWGSL });\n    const layout = device.createPipelineLayout({ bindGroupLayouts: [this.layout] });\n    this.pipelines = ['scanRows', 'scanMarginal', 'normalize'].map(entryPoint => device.createComputePipeline({\n      label: `Environment distribution ${entryPoint}`, layout, compute: { module, entryPoint }\n    }));\n  }\n\n  build(input: GPUTexture, output: GPUTexture) {\n    const { width, height } = input;\n    if (this.pixels?.size !== width * height * 8 || this.rows?.size !== height * 8) {\n      this.dispose();\n      this.pixels = this.device.createBuffer({ label: 'Environment row scans', size: width * height * 8, usage: GPUBufferUsage.STORAGE });\n      this.rows = this.device.createBuffer({ label: 'Environment marginal scan', size: height * 8, usage: GPUBufferUsage.STORAGE });\n    }\n    const group = this.device.createBindGroup({\n      layout: this.layout,\n      entries: [\n        { binding: 0, resource: input.createView() },\n        { binding: 1, resource: { buffer: this.pixels! } },\n        { binding: 2, resource: { buffer: this.rows! } },\n        { binding: 3, resource: output.createView() }\n      ]\n    });\n    const encoder = this.device.createCommandEncoder({ label: 'Environment distribution' });\n    const workgroups = [[height, 1], [1, 1], [Math.ceil((width + 2) / 16), Math.ceil((height + 1) / 16)]];\n    this.pipelines.forEach((pipeline, index) => {\n      const pass = encoder.beginComputePass();\n      pass.setPipeline(pipeline);\n      pass.setBindGroup(0, group);\n      pass.dispatchWorkgroups(...workgroups[index] as [number, number]);\n      pass.end();\n    });\n    this.device.queue.submit([encoder.finish()]);\n  }\n\n  dispose() {\n    this.pixels?.destroy();\n    this.rows?.destroy();\n    this.pixels = undefined;\n    this.rows = undefined;\n  }\n}\n"]}