@meimu/mutracer 0.1.5 → 0.2.0

package/lib/renderer/WGPUFullQuadPass.js

@@ -131,16 +131,17 @@ export class WGPUFullQuadPass {
         }
     }
     setOutputParams(outputParams) {
+        const wasScreen = !!this._renderToScreen;
         this._renderToScreen = undefined;
         const outputsChanged = Object.keys(outputParams).some((key) => !isTextureParamsEqual(outputParams[key], this._outputParams[key] || {}));
         this._needsUpdateOutputTextures ||= outputsChanged;
-        this._needsUpdatePipeline ||= outputsChanged;
-        if (outputsChanged)
+        this._needsUpdatePipeline ||= outputsChanged || wasScreen;
+        if (outputsChanged || wasScreen)
             this._fsCode = '';
         this._outputParams = outputParams;
     }
     setRenderToScreen(screenTexture, presentationFormat) {
-        if (this._renderToScreen &&
+        if (!this._renderToScreen ||
             this._renderToScreen.presentationFormat !== presentationFormat) {
             this._needsUpdatePipeline = true;
             this._fsCode = '';
@@ -273,7 +274,7 @@ ${hasInputs
 ${this._uniforms
             .map((uniform, idx) => `@group(${hasInputs ? 1 : 0}) @binding(${idx}) var<uniform> ${uniform.label}: ${uniform.type};`)
             .join('\n')}
-struct pd {
+struct qa {
 ${this._outputs
             .map((name, idx) => `@location(${idx}) ${name}: vec4f,`)
             .join('\n')}
@@ -282,11 +283,11 @@ ${scratchWGSLFunctions(this._reserveCapacity ? this._regionNames : inputs.filter
 ${this._fsDefine ?? ''}
 @fragment
 fn main(
-@builtin(position) fH: vec4f
-) -> pd {
-let uv = vec2i(floor(fH.xy));
-let uv2 = fH.xy / viewportSize;
-var output: pd;
+@builtin(position) coord: vec4f
+) -> qa {
+let uv = vec2i(floor(coord.xy));
+let uv2 = coord.xy / viewportSize;
+var output: qa;
 ${this._fsMain}
 return output;
 }

package/lib/renderer/WGPUFullQuadPass.js.map

@@ -1 +1 @@
-{"version":3,"file":"WGPUFullQuadPass.js","sourceRoot":"","sources":["../../src/renderer/WGPUFullQuadPass.ts"],"names":[],"mappings":"AAAA,OAAO,EAAC,oBAAoB,EAAC,MAAM,sBAAsB,CAAC;AAC1D,OAAO,EAAC,gBAAgB,EAAE,aAAa,EAAC,MAAM,oBAAoB,CAAC;AACnE,OAAO,EAAE,oBAAoB,EAAE,MAAM,sBAAsB,CAAC;AAC5D,OAAO,EAAE,YAAY,EAAE,MAAM,gBAAgB,CAAC;AAC9C,MAAM,CAAC,MAAM,8BAA8B,GAAG,SAAS,CAAC;;;;;;;;;;;;CAYvD,CAAC;AAEF,SAAS,oBAAoB,CAC3B,MAA8B,EAC9B,KAA6B;IAE7B,OAAO,MAAM,CAAC,MAAM,KAAK,KAAK,CAAC,MAAM,CAAC;AACxC,CAAC;AAED,MAAM,UAAU,gBAAgB,CAC9B,KAAU,EACV,IAAqB;IAErB,IACE,KAAK,YAAY,YAAY;QAC7B,KAAK,YAAY,UAAU;QAC3B,KAAK,YAAY,WAAW,EAC5B,CAAC;QACD,OAAO,KAAK,CAAC;IACf,CAAC;IAED,IAAI,OAAO,KAAK,KAAK,QAAQ,EAAE,CAAC;QAC9B,KAAK,GAAG,CAAC,KAAK,CAAC,CAAC;IAClB,CAAC;IAED,IAAI,IAAI,KAAK,KAAK,IAAI,IAAI,CAAC,QAAQ,CAAC,GAAG,CAAC,EAAE,CAAC;QACzC,OAAO,IAAI,UAAU,CAAC,KAAK,CAAC,CAAC;IAC/B,CAAC;SAAM,IAAI,IAAI,KAAK,KAAK,IAAI,IAAI,CAAC,QAAQ,CAAC,GAAG,CAAC,EAAE,CAAC;QAChD,OAAO,IAAI,WAAW,CAAC,KAAK,CAAC,CAAC;IAChC,CAAC;SAAM,CAAC;QACN,OAAO,IAAI,YAAY,CAAC,KAAK,CAAC,CAAC;IACjC,CAAC;AACH,CAAC;AAwBD,MAAM,OAAO,gBAAgB;IACnB,MAAM,CAAC;IAEP,OAAO,CAAC;IACR,eAAe,GAA+B,EAAE,CAAC;IACjD,aAAa,GAA2C,EAAE,CAAC;IAE3D,SAAS,CAAqB;IAC9B,WAAW,GAAmB,EAAE,CAAC;IACjC,oBAAoB,GAAG,IAAI,CAAC;IAC5B,0BAA0B,GAAG,IAAI,CAAC;IAC1C;;OAEG;IACK,eAAe,CAGrB;IACM,OAAO,GAAa,EAAE,CAAC;IACvB,oBAAoB,GAExB,EAAE,CAAC;IACC,QAAQ,GAAa,EAAE,CAAC;IACxB,SAAS,GAAc,EAAE,CAAC;IAC1B,YAAY,GAA4B,EAAE,CAAC;IAC3C,eAAe,GAA8B,EAAE,CAAC;IAEhD,SAAS,GAA+B,EAAE,CAAC;IAE3C,cAAc,GAAG,CAAC,CAAC;IACnB,eAAe,GAAG,CAAC,CAAC;IACpB,gBAAgB,GAAG,KAAK,CAAC;IACzB,YAAY,GAAa,EAAE,CAAC;IAC5B,MAAM,GAAG,EAAE,CAAC;IACZ,OAAO,GAAG,EAAE,CAAC;IAEb,OAAO,GAAG,EAAE,CAAC;IACJ,OAAO,CAAC;IACR,SAAS,CAAC;IAEnB,kBAAkB,CAAC;IAE3B,YACE,KAAa,EACb,MAAiB,EACjB,IASC;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,oBAAoB,GAAG,IAAI,CAAC,mBAAmB,IAAI,EAAE,CAAC;QAC3D,IAAI,CAAC,QAAQ,GAAG,IAAI,CAAC,OAAO,CAAC;QAC7B,IAAI,CAAC,SAAS,GAAG,CAAC,IAAI,CAAC,QAAQ,IAAI,EAAE,CAAC,CAAC,MAAM,CAAC;YAC5C,KAAK,EAAE,cAAc;YACrB,IAAI,EAAE,OAAO;YACb,IAAI,EAAE,CAAC,CAAC,EAAE,CAAC,CAAC;SACb,CAAC,CAAC;QACH,IAAI,CAAC,SAAS,GAAG,IAAI,CAAC,QAAQ,IAAI,EAAE,CAAC;QACrC,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,IAAI,CAAC,kBAAkB,GAAG,IAAI,CAAC,iBAAiB,IAAI,KAAK,CAAC;QAE1D,IAAI,CAAC,SAAS,CAAC,OAAO,CAAC,CAAC,OAAO,EAAE,EAAE;YACjC,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;gBAClB,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,8FAA8F;IAC9F,eAAe;QACb,IAAI,IAAI,CAAC,gBAAgB;YAAE,OAAO;QAClC,IAAI,CAAC,gBAAgB,GAAG,IAAI,CAAC;QAC7B,IAAI,CAAC,cAAc,GAAG,IAAI,CAAC,MAAM,CAAC;QAAC,IAAI,CAAC,eAAe,GAAG,IAAI,CAAC,OAAO,CAAC;QACvE,MAAM,QAAQ,GAAG,IAAI,CAAC,OAAO,CAAC,MAAM,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC,IAAI,CAAC,oBAAoB,CAAC,IAAS,CAAC,IAAI,IAAI,CAAC,oBAAoB,CAAC,IAAS,CAAC,KAAK,IAAI,CAAC,CAAC;QACrI,IAAI,CAAC,YAAY,GAAG,QAAQ,CAAC;QAC7B,IAAI,QAAQ,CAAC,MAAM,EAAE,CAAC;YACpB,MAAM,OAAO,GAAY,EAAC,KAAK,EAAE,cAAc,EAAE,IAAI,EAAE,gBAAgB,QAAQ,CAAC,MAAM,GAAG,EAAE,IAAI,EAAE,IAAI,WAAW,CAAC,QAAQ,CAAC,MAAM,GAAG,CAAC,CAAC,EAAC,CAAC;YACvI,IAAI,CAAC,SAAS,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC;YAAC,IAAI,CAAC,YAAY,CAAC,OAAO,CAAC,KAAK,CAAC,GAAG,OAAO,CAAC;YACzE,IAAI,CAAC,eAAe,CAAC,OAAO,CAAC,KAAK,CAAC,GAAG,IAAI,CAAC,OAAO,CAAC,YAAY,CAAC,EAAC,KAAK,EAAE,GAAG,IAAI,CAAC,MAAM,IAAI,OAAO,CAAC,KAAK,EAAE,EAAE,IAAI,EAAE,QAAQ,CAAC,MAAM,GAAG,EAAE,EAAE,KAAK,EAAE,cAAc,CAAC,OAAO,GAAG,cAAc,CAAC,QAAQ,EAAC,CAAC,CAAC;QACnM,CAAC;QACD,IAAI,CAAC,oBAAoB,GAAG,IAAI,CAAC;QAAC,IAAI,CAAC,OAAO,GAAG,EAAE,CAAC;IACtD,CAAC;IACD,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,MAAM,SAAS,GAAG,IAAI,CAAC,GAAG,CAAC,IAAI,CAAC,cAAc,EAAE,KAAK,CAAC,CAAC;YACvD,MAAM,UAAU,GAAG,IAAI,CAAC,GAAG,CAAC,IAAI,CAAC,eAAe,EAAE,MAAM,CAAC,CAAC;YAC1D,IAAI,CAAC,0BAA0B,KAAK,CAAC,IAAI,CAAC,gBAAgB,IAAI,SAAS,KAAK,IAAI,CAAC,cAAc,IAAI,UAAU,KAAK,IAAI,CAAC,eAAe,CAAC;YACvI,IAAI,CAAC,cAAc,GAAG,SAAS,CAAC;YAAC,IAAI,CAAC,eAAe,GAAG,UAAU,CAAC;YACnE,IAAI,CAAC,oBAAoB,GAAG,IAAI,CAAC;YAEjC,IAAI,CAAC,UAAU,CAAC,cAAc,EAAE,CAAC,KAAK,EAAE,MAAM,CAAC,CAAC,CAAC;QACnD,CAAC;IACH,CAAC;IAED,eAAe,CAAC,YAA+C;QAC7D,IAAI,CAAC,eAAe,GAAG,SAAS,CAAC;QACjC,MAAM,cAAc,GAAG,MAAM,CAAC,IAAI,CAAC,YAAY,CAAC,CAAC,IAAI,CACnD,CAAC,GAAG,EAAE,EAAE,CACN,CAAC,oBAAoB,CAClB,YAAoB,CAAC,GAAG,CAAC,EAC1B,IAAI,CAAC,aAAa,CAAC,GAAG,CAAC,IAAK,EAA6B,CAC1D,CACJ,CAAC;QACF,IAAI,CAAC,0BAA0B,KAAK,cAAc,CAAC;QACnD,IAAI,CAAC,oBAAoB,KAAK,cAAc,CAAC;QAC7C,IAAI,cAAc;YAAE,IAAI,CAAC,OAAO,GAAG,EAAE,CAAC;QACtC,IAAI,CAAC,aAAa,GAAG,YAAY,CAAC;IACpC,CAAC;IAED,iBAAiB,CACf,aAAyB,EACzB,kBAAoC;QAEpC,IACE,IAAI,CAAC,eAAe;YACpB,IAAI,CAAC,eAAe,CAAC,kBAAkB,KAAK,kBAAkB,EAC9D,CAAC;YACD,IAAI,CAAC,oBAAoB,GAAG,IAAI,CAAC;YACjC,IAAI,CAAC,OAAO,GAAG,EAAE,CAAC;QACpB,CAAC;QACD,IAAI,CAAC,eAAe,GAAG;YACrB,aAAa;YACb,kBAAkB;SACnB,CAAC;IACJ,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;IAEgB,gBAAgB,GAAG,IAAI,GAAG,EAAc,CAAC;IAC1D,YAAY,CAAC,OAAmB,IAAI,IAAI,MAAM,CAAC,MAAM,CAAC,IAAI,CAAC,eAAe,CAAC,CAAC,QAAQ,CAAC,OAAO,CAAC;QAAE,IAAI,CAAC,gBAAgB,CAAC,GAAG,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC;IACpI,aAAa,CAAC,OAAmB;QAC/B,IAAI,CAAC,IAAI,CAAC,gBAAgB,CAAC,MAAM,CAAC,OAAO,CAAC;YAAE,OAAO;QACnD,IAAI,CAAC,MAAM,CAAC,MAAM,CAAC,IAAI,CAAC,eAAe,CAAC,CAAC,QAAQ,CAAC,OAAO,CAAC,EAAE,CAAC;YAC3D,KAAK,IAAI,CAAC,OAAO,CAAC,KAAK,CAAC,mBAAmB,EAAE,CAAC,KAAK,CAAC,GAAG,EAAE,CAAC,SAAS,CAAC,CAAC,IAAI,CAAC,GAAG,EAAE,CAAC,OAAO,CAAC,OAAO,EAAE,CAAC,CAAC;QACrG,CAAC;IACH,CAAC;IAED,SAAS,CAAC,IAAO;QACf,OAAO,IAAI,CAAC,eAAe,CAAC,IAAI,CAAC,CAAC;IACpC,CAAC;IAED,cAAc;QACZ,OAAO,MAAM,CAAC,MAAM,CAAC,IAAI,CAAC,eAAe,CAAC,CAAC,MAAM,CAAC,CAAC,GAAG,EAAE,MAAM,EAAE,EAAE,CAAC,GAAG,GAAG,MAAM,CAAC,IAAI,EAAE,CAAC,CAAC;cACpF,KAAK,CAAC,IAAI,CAAC,IAAI,GAAG,CAAC,CAAC,GAAG,MAAM,CAAC,MAAM,CAAC,IAAI,CAAC,eAAe,CAAC,EAAE,GAAG,IAAI,CAAC,gBAAgB,CAAC,CAAC,CAAC;iBACtF,MAAM,CAAC,CAAC,GAAG,EAAE,OAAO,EAAE,EAAE,CAAC,GAAG,GAAG,YAAY,CAAC,OAAO,CAAC,EAAE,CAAC,CAAC,CAAC;IAChE,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,IAAI,GAAG,CAAC,CAAC,GAAG,MAAM,CAAC,MAAM,CAAC,IAAI,CAAC,eAAe,CAAC,EAAE,GAAG,IAAI,CAAC,gBAAgB,CAAC,CAAC,CAAC,OAAO,CAAC,OAAO,CAAC,EAAE,CAAC,OAAO,CAAC,OAAO,EAAE,CAAC,CAAC;QAClH,IAAI,CAAC,gBAAgB,CAAC,KAAK,EAAE,CAAC;IAChC,CAAC;IAEO,cAAc,CAAC,MAA8B;QACnD,OAAO,IAAI,CAAC,OAAO,CAAC,aAAa,CAAC;YAChC,KAAK,EAAE,IAAI,CAAC,MAAM;YAClB,IAAI,EAAE;gBACJ,KAAK,EAAE,IAAI,CAAC,gBAAgB,CAAC,CAAC,CAAC,IAAI,CAAC,cAAc,CAAC,CAAC,CAAC,IAAI,CAAC,MAAM;gBAChE,MAAM,EAAE,IAAI,CAAC,gBAAgB,CAAC,CAAC,CAAC,IAAI,CAAC,eAAe,CAAC,CAAC,CAAC,IAAI,CAAC,OAAO;gBACnE,kBAAkB,EAAE,CAAC;aACtB;YACD,MAAM,EAAE,MAAM,CAAC,MAAM;YACrB,KAAK,EACH,eAAe,CAAC,iBAAiB;gBACjC,eAAe,CAAC,eAAe;gBAC/B,CAAC,IAAI,CAAC,kBAAkB,CAAC,CAAC,CAAC,eAAe,CAAC,QAAQ,CAAC,CAAC,CAAC,CAAC,CAAC;SAC3D,CAAC,CAAC;IACL,CAAC;IAEO,qBAAqB,CAAC,QAAmB;QAC/C,MAAM,YAAY,GAAG,IAAI,CAAC,aAAa,CAAC;QACxC,KAAK,MAAM,GAAG,IAAI,YAAY,EAAE,CAAC;YAC/B,IAAI,QAAQ,EAAE,QAAQ,CAAC,GAAG,CAAC,EAAE,CAAC;gBAC5B,SAAS;YACX,CAAC;YACD,MAAM,MAAM,GAAG,YAAY,CAAC,GAAG,CAAC,CAAC;YACjC,IAAI,CAAC,IAAI,CAAC,gBAAgB,CAAC,GAAG,CAAC,IAAI,CAAC,eAAe,CAAC,GAAG,CAAC,CAAC;gBAAE,IAAI,CAAC,eAAe,CAAC,GAAG,CAAC,EAAE,OAAO,EAAE,CAAC;YAChG,IAAI,CAAC,eAAe,CAAC,GAAG,CAAC,GAAG,IAAI,CAAC,cAAc,CAAC,MAAM,CAAC,CAAC;QAC1D,CAAC;IACH,CAAC;IAEO,eAAe;QACrB,IAAI,CAAC,IAAI,CAAC,oBAAoB,EAAE,CAAC;YAC/B,OAAO;QACT,CAAC;QACD,IAAI,CAAC,oBAAoB,GAAG,KAAK,CAAC;QAElC,MAAM,YAAY,GAAG,IAAI,CAAC,aAAa,CAAC;QACxC,MAAM,MAAM,GAAG,IAAI,CAAC,OAAO,CAAC;QAC5B,MAAM,MAAM,GAAG,IAAI,CAAC,UAAU,EAAE,CAAC;QACjC,IAAI,MAAM,KAAK,IAAI,CAAC,OAAO,EAAE,CAAC;YAC5B,OAAO;QACT,CAAC;QACD,MAAM,EAAE,aAAa,EAAE,kBAAkB,EAAE,GAAG,IAAI,CAAC,eAAe,IAAI,EAAE,CAAC;QACzE,IAAI,CAAC,OAAO,GAAG,MAAM,CAAC;QACtB,IAAI,CAAC,SAAS,GAAG,oBAAoB,CAAC,MAAM,CAAC,CAAC,oBAAoB,CAAC;YACjE,KAAK,EAAE,IAAI,CAAC,MAAM;YAClB,MAAM,EAAE,MAAM;YACd,MAAM,EAAE;gBACN,MAAM,EAAE,oBAAoB,CAAC,MAAM,CAAC,CAAC,kBAAkB,CAAC;oBACtD,KAAK,EAAE,IAAI,CAAC,MAAM;oBAClB,IAAI,EAAE,8BAA8B;iBACrC,CAAC;gBACF,UAAU,EAAE,MAAM;aACnB;YACD,QAAQ,EAAE;gBACR,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;gBAClB,OAAO,EAAE,aAAa;oBACpB,CAAC,CAAC;wBACE;4BACE,MAAM,EAAE,kBAAmB;yBAC5B;qBACF;oBACH,CAAC,CAAC,IAAI,CAAC,QAAQ,CAAC,GAAG,CAAC,CAAC,GAAG,EAAE,EAAE,CAAC,CAAC;wBAC1B,MAAM,EAAG,YAAoB,CAAC,GAAG,CAAC,CAAC,MAAM;qBAC1C,CAAC,CAAC;aACR;YACD,SAAS,EAAE;gBACT,QAAQ,EAAE,eAAe;aAC1B;SACF,CAAC,CAAC;QACH,IAAI,CAAC,iBAAiB,EAAE,CAAC;IAC3B,CAAC;IAEO,UAAU;QAChB,MAAM,MAAM,GAAG,IAAI,CAAC,OAAO,CAAC;QAC5B,MAAM,SAAS,GAAG,MAAM,CAAC,MAAM,CAAC;QAChC,MAAM,SAAS,GAAG,SAAS,GAAG,CAAC,CAAC;QAChC,MAAM,EAAE,GAAG,UAAU,CAAC;EACxB,MAAM;aACL,IAAI,EAAE;aACN,GAAG,CACF,CAAC,WAAW,EAAE,GAAG,EAAE,EAAE;YACnB,MAAM,aAAa,GAAG,IAAI,CAAC,oBAAoB,CAAC,WAAgB,CAAC,CAAC;YAClE,MAAM,WAAW,GAAG,aAAa,KAAK,IAAI,CAAC,CAAC,CAAC,YAAY,CAAC,CAAC,CAAC,YAAY,CAAC;YACzE,OAAO,sBAAsB,GAAG,SAAS,WAAW,KAAK,WAAW,QAAQ,CAAC;QAC/E,CAAC,CACF;aACA,IAAI,CAAC,IAAI,CAAC;;EAGX,SAAS;YACP,CAAC,CAAC,MAAM,CAAC,IAAI,CAAC,IAAI,CAAC,SAAS,CAAC;iBACxB,IAAI,EAAE;iBACN,GAAG,CACF,CAAC,WAAW,EAAE,GAAG,EAAE,EAAE,CACnB,sBACE,GAAG,GAAG,SACR,SAAS,WAAW,YAAY,CACnC;iBACA,IAAI,CAAC,IAAI,CAAC;YACf,CAAC,CAAC,EACN;EACE,IAAI,CAAC,SAAS;aACb,GAAG,CACF,CAAC,OAAO,EAAE,GAAG,EAAE,EAAE,CACf,UAAU,SAAS,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,cAAc,GAAG,kBAC1C,OAAO,CAAC,KACV,KAAK,OAAO,CAAC,IAAI,GAAG,CACvB;aACA,IAAI,CAAC,IAAI,CAAC;;EAEX,IAAI,CAAC,QAAQ;aACZ,GAAG,CAAC,CAAC,IAAI,EAAE,GAAG,EAAE,EAAE,CAAC,aAAa,GAAG,KAAK,IAAI,UAAU,CAAC;aACvD,IAAI,CAAC,IAAI,CAAC;;EAEX,oBAAoB,CACpB,IAAI,CAAC,gBAAgB,CAAC,CAAC,CAAC,IAAI,CAAC,YAAY,CAAC,CAAC,CAAC,MAAM,CAAC,MAAM,CAAC,IAAI,CAAC,EAAE,CAC/D,CAAC,IAAI,CAAC,oBAAoB,CAAC,IAAS,CAAC,IAAI,IAAI,CAAC,oBAAoB,CAAC,IAAS,CAAC,KAAK,IAAI,CAAC,EACzF,MAAM,CAAC,IAAI,CAAC,IAAI,CAAC,SAAS,CAAC,EAAE,IAAI,CAAC,gBAAgB,CACnD;EACC,IAAI,CAAC,SAAS,IAAI,EAAE;;;;;;;;EAQpB,IAAI,CAAC,OAAO;;;CAGb,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,qBAAqB,GAAG,IAAI,CAAC,OAAO,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC;QAC9D,IAAI,IAAI,CAAC,SAAS,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;YAC9B,UAAU,CAAC,qBAAqB,CAAC,GAAG,MAAM,CAAC,eAAe,CAAC;gBACzD,KAAK,EAAE,IAAI,CAAC,MAAM;gBAClB,MAAM,EAAE,IAAI,CAAC,SAAS,CAAC,kBAAkB,CAAC,qBAAqB,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;iBAC2B,CAAA,CACjC;aACF,CAAC,CAAC;QACL,CAAC;QAED,IAAI,CAAC,WAAW,GAAG,UAAU,CAAC;IAChC,CAAC;IAED,UAAU,CACR,cAAiC,EACjC,MAA6B;IAE7B;;OAEG;IACH,OAA+B;QAE/B,IAAI,IAAI,CAAC,0BAA0B,IAAI,CAAC,IAAI,CAAC,eAAe,EAAE,CAAC;YAC7D,IAAI,CAAC,qBAAqB,CAAC,MAAM,CAAC,IAAI,CAAC,OAAO,IAAI,EAAE,CAAC,CAAC,CAAC;YACvD,IAAI,CAAC,0BAA0B,GAAG,KAAK,CAAC;QAC1C,CAAC;QAED,IAAI,IAAI,CAAC,YAAY,CAAC,MAAM,EAAE,CAAC;YAC7B,MAAM,MAAM,GAAG,IAAI,WAAW,CAAC,IAAI,CAAC,YAAY,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC;YAC7D,KAAK,MAAM,CAAC,KAAK,EAAE,IAAI,CAAC,IAAI,IAAI,CAAC,YAAY,CAAC,OAAO,EAAE,EAAE,CAAC;gBACxD,MAAM,MAAM,GAAG,aAAa,CAAC,MAAM,CAAC,IAAS,CAAC,CAAC,CAAC;gBAChD,MAAM,CAAC,GAAG,CAAC,CAAC,MAAM,CAAC,KAAK,EAAE,MAAM,CAAC,MAAM,CAAC,EAAE,KAAK,GAAG,CAAC,CAAC,CAAC;YACvD,CAAC;YACD,IAAI,CAAC,UAAU,CAAC,cAAc,EAAE,MAAM,CAAC,CAAC;QAC1C,CAAC;QACD,KAAK,MAAM,OAAO,IAAI,MAAM,CAAC,MAAM,CAAC,IAAI,CAAC,eAAe,CAAC;YAAE,gBAAgB,CAAC,OAAO,EAAE,IAAI,CAAC,MAAM,EAAE,IAAI,CAAC,OAAO,CAAC,CAAC;QAChH,IAAI,CAAC,eAAe,EAAE,CAAC;QAEvB,MAAM,SAAS,GAAG,IAAI,CAAC,OAAO,CAAC,MAAM,CAAC;QACtC,IAAI,SAAS,GAAG,CAAC,EAAE,CAAC;YAClB,IAAI,CAAC,WAAW,CAAC,CAAC,CAAC,GAAG,IAAI,CAAC,OAAO,CAAC,eAAe,CAAC;gBACjD,KAAK,EAAE,IAAI,CAAC,MAAM;gBAClB,MAAM,EAAE,IAAI,CAAC,SAAS,CAAC,kBAAkB,CAAC,CAAC,CAAC;gBAC5C,OAAO,EAAE;oBACP,GAAG,IAAI,CAAC,OAAO,CAAC,GAAG,CAAC,CAAC,WAAW,EAAE,GAAG,EAAE,EAAE,CAAC,CAAC;wBACzC,OAAO,EAAE,GAAG;wBACZ,QAAQ,EAAE,MAAM,CAAC,WAAgB,CAAC,CAAC,UAAU,CAC3C,IAAI,CAAC,oBAAoB,CAAC,WAAgB,CAAC;4BACzC,CAAC,CAAC,EAAE,SAAS,EAAE,IAAI,CAAC,oBAAoB,CAAC,WAAgB,CAAC,EAAE;4BAC5D,CAAC,CAAC,SAAS,CACd;qBACF,CAAC,CAAC;oBACH,GAAG,MAAM,CAAC,IAAI,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC,GAAG,CAAC,CAAC,WAAW,EAAE,GAAG,EAAE,EAAE,CAAC,CAAC;wBACxD,OAAO,EAAE,GAAG,GAAG,SAAS;wBACxB,QAAQ,EAAE,IAAI,CAAC,SAAS,CAAC,WAAW,CAAC;qBACtC,CAAC,CAAC;iBACJ;aACF,CAAC,CAAC;QACL,CAAC;QAED,MAAM,cAAc,GAAG;YACrB,8CAA8C;YAC9C,GAAG,IAAI,CAAC,eAAe;YACvB,GAAG,OAAO;SACmB,CAAC;QAChC,wBAAwB;QACxB,MAAM,UAAU,GAAG,cAAc,CAAC,eAAe,CAAC;YAChD,gBAAgB,EAAE,IAAI,CAAC,eAAe;gBACpC,CAAC,CAAC;oBACE;wBACE,IAAI,EAAE,IAAI,CAAC,eAAe,CAAC,aAAa,CAAC,UAAU,EAAE;wBACrD,UAAU,EAAE,EAAE,CAAC,EAAE,CAAC,EAAE,CAAC,EAAE,CAAC,EAAE,CAAC,EAAE,CAAC,EAAE,CAAC,EAAE,CAAC,EAAE;wBACtC,OAAO,EAAE,OAAO;wBAChB,MAAM,EAAE,OAAO;qBAChB;iBACF;gBACH,CAAC,CAAC,IAAI,CAAC,QAAQ,CAAC,GAAG,CACf,CAAC,WAAW,EAAE,EAAE,CACd,CAAC;oBACC,IAAI,EAAE,cAAc,CAAC,WAAW,CAAC,CAAC,UAAU,EAAE;oBAC9C,UAAU,EAAE,EAAE,CAAC,EAAE,CAAC,EAAE,CAAC,EAAE,CAAC,EAAE,CAAC,EAAE,CAAC,EAAE,CAAC,EAAE,CAAC,EAAE;oBACtC,MAAM,EAAE,OAAO;oBACf,OAAO,EAAE,OAAO;iBACuB,CAAA,CAC5C;SACN,CAAC,CAAC;QAEH,mBAAmB;QACnB,UAAU,CAAC,WAAW,CAAC,CAAC,EAAE,CAAC,EAAE,IAAI,CAAC,MAAM,EAAE,IAAI,CAAC,OAAO,EAAE,CAAC,EAAE,CAAC,CAAC,CAAC;QAC9D,UAAU,CAAC,cAAc,CAAC,CAAC,EAAE,CAAC,EAAE,IAAI,CAAC,MAAM,EAAE,IAAI,CAAC,OAAO,CAAC,CAAC;QAC3D,UAAU,CAAC,WAAW,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC;QACvC,cAAc;QACd,IAAI,CAAC,WAAW,CAAC,OAAO,CAAC,CAAC,SAAS,EAAE,GAAG,EAAE,EAAE;YAC1C,UAAU,CAAC,YAAY,CAAC,GAAG,EAAE,SAAS,CAAC,CAAC;QAC1C,CAAC,CAAC,CAAC;QACH,UAAU,CAAC,IAAI,CAAC,CAAC,EAAE,CAAC,EAAE,CAAC,EAAE,CAAC,CAAC,CAAC;QAC5B,sBAAsB;QACtB,UAAU,CAAC,GAAG,EAAE,CAAC;IACnB,CAAC;CACF","sourcesContent":["import {scratchWGSLFunctions} from './scratchSampling.js';\nimport {setTextureRegion, textureRegion} from './textureRegion.js';\nimport { getTargetShaderCache } from './WGPUShaderCache.js';\nimport { textureBytes } from './gpuMemory.js';\nexport const fullScreenQuadVertexShaderWGSL = /*wgsl */ `\n@vertex\nfn main(\n  @builtin(vertex_index) vertexIndex: u32\n) -> @builtin(position) vec4f {\n  const pos = array(\n    vec2(-1.0, -1.0), vec2(1.0, -1.0), vec2(-1.0, 1.0),\n    vec2(-1.0, 1.0), vec2(1.0, -1.0), vec2(1.0, 1.0),\n  );\n\n  return vec4<f32>(pos[vertexIndex], 0.0, 1.0);\n}\n`;\n\nfunction isTextureParamsEqual(\n  params: WGPUFullQuadPassOutput,\n  other: WGPUFullQuadPassOutput\n) {\n  return params.format === other.format;\n}\n\nexport function getUniformBuffer(\n  value: any,\n  type: Uniform['type']\n): Float32Array | Int32Array | Uint32Array {\n  if (\n    value instanceof Float32Array ||\n    value instanceof Int32Array ||\n    value instanceof Uint32Array\n  ) {\n    return value;\n  }\n\n  if (typeof value === 'number') {\n    value = [value];\n  }\n\n  if (type === 'i32' || type.endsWith('i')) {\n    return new Int32Array(value);\n  } else if (type === 'u32' || type.endsWith('u')) {\n    return new Uint32Array(value);\n  } else {\n    return new Float32Array(value);\n  }\n}\n\nexport interface Uniform {\n  label: string;\n  type:\n    | `array<vec4u, ${number}>`\n    | 'f32'\n    | 'i32'\n    | 'u32'\n    | 'vec3f'\n    | 'vec4f'\n    | 'vec2f'\n    | 'vec2i'\n    | 'vec3i'\n    | 'vec4i'\n    | 'vec2u'\n    | 'vec3u'\n    | 'vec4u';\n  data: Float32Array | Int32Array | Uint32Array | number | number[];\n}\n\nexport interface WGPUFullQuadPassOutput {\n  format: GPUTextureFormat;\n}\nexport class WGPUFullQuadPass<I extends string, O extends string> {\n  private _label;\n\n  private _device;\n  private _outputTextures: Record<string, GPUTexture> = {};\n  private _outputParams: Record<string, WGPUFullQuadPassOutput> = {};\n\n  private _pipeline!: GPURenderPipeline;\n  private _bindGroups: GPUBindGroup[] = [];\n  private _needsUpdatePipeline = true;\n  private _needsUpdateOutputTextures = true;\n  /**\n   * When render to canvas\n   */\n  private _renderToScreen?: {\n    screenTexture: GPUTexture;\n    presentationFormat: GPUTextureFormat;\n  };\n  private _inputs: string[] = [];\n  private _inputViewDimensions: Partial<\n    Record<I, GPUTextureViewDimension>\n  > = {};\n  private _outputs: string[] = [];\n  private _uniforms: Uniform[] = [];\n  private _uniformsMap: Record<string, Uniform> = {};\n  private _uniformBuffers: Record<string, GPUBuffer> = {};\n\n  private _samplers: Record<string, GPUSampler> = {};\n\n  private _capacityWidth = 0;\n  private _capacityHeight = 0;\n  private _reserveCapacity = false;\n  private _regionNames: string[] = [];\n  private _width = 10;\n  private _height = 10;\n\n  private _fsCode = '';\n  private readonly _fsMain;\n  private readonly _fsDefine;\n\n  private _textureWillBeRead;\n\n  constructor(\n    label: string,\n    device: GPUDevice,\n    opts: {\n      inputs: I[];\n      inputViewDimensions?: Partial<Record<I, GPUTextureViewDimension>>;\n      outputs: O[];\n      samplers?: Record<string, GPUSampler>;\n      fsMain: string;\n      fsDefine?: string;\n      uniforms?: Uniform[];\n      textureWillBeRead?: boolean;\n    }\n  ) {\n    this._label = label;\n    this._device = device;\n    this._fsMain = opts.fsMain;\n    this._fsDefine = opts.fsDefine;\n    this._inputs = opts.inputs;\n    this._inputViewDimensions = opts.inputViewDimensions || {};\n    this._outputs = opts.outputs;\n    this._uniforms = (opts.uniforms || []).concat({\n      label: 'viewportSize',\n      type: 'vec2f',\n      data: [0, 0]\n    });\n    this._samplers = opts.samplers || {};\n    this._uniformsMap = this._uniforms.reduce((map, uniform) => {\n      map[uniform.label] = uniform;\n      return map;\n    }, {} as Record<string, Uniform>);\n    this._textureWillBeRead = opts.textureWillBeRead || false;\n\n    this._uniforms.forEach((uniform) => {\n      const data = getUniformBuffer(uniform.data, uniform.type);\n      this._uniformBuffers[uniform.label] = device.createBuffer({\n        label: this._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  /** Keep physical outputs large enough for every render while varying the logical viewport. */\n  reserveCapacity() {\n    if (this._reserveCapacity) return;\n    this._reserveCapacity = true;\n    this._capacityWidth = this._width; this._capacityHeight = this._height;\n    const textures = this._inputs.filter(name => !this._inputViewDimensions[name as I] || this._inputViewDimensions[name as I] === '2d');\n    this._regionNames = textures;\n    if (textures.length) {\n      const uniform: Uniform = {label: 'imageRegions', type: `array<vec4u, ${textures.length}>`, data: new Uint32Array(textures.length * 4)};\n      this._uniforms.push(uniform); this._uniformsMap[uniform.label] = uniform;\n      this._uniformBuffers[uniform.label] = this._device.createBuffer({label: `${this._label}.${uniform.label}`, size: textures.length * 16, usage: GPUBufferUsage.UNIFORM | GPUBufferUsage.COPY_DST});\n    }\n    this._needsUpdatePipeline = true; this._fsCode = '';\n  }\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      const nextWidth = Math.max(this._capacityWidth, width);\n      const nextHeight = Math.max(this._capacityHeight, height);\n      this._needsUpdateOutputTextures ||= !this._reserveCapacity || nextWidth !== this._capacityWidth || nextHeight !== this._capacityHeight;\n      this._capacityWidth = nextWidth; this._capacityHeight = nextHeight;\n      this._needsUpdatePipeline = true;\n\n      this.setUniform('viewportSize', [width, height]);\n    }\n  }\n\n  setOutputParams(outputParams: Record<O, WGPUFullQuadPassOutput>) {\n    this._renderToScreen = undefined;\n    const outputsChanged = Object.keys(outputParams).some(\n      (key) =>\n        !isTextureParamsEqual(\n          (outputParams as any)[key],\n          this._outputParams[key] || ({} as WGPUFullQuadPassOutput)\n        )\n    );\n    this._needsUpdateOutputTextures ||= outputsChanged;\n    this._needsUpdatePipeline ||= outputsChanged;\n    if (outputsChanged) this._fsCode = '';\n    this._outputParams = outputParams;\n  }\n\n  setRenderToScreen(\n    screenTexture: GPUTexture,\n    presentationFormat: GPUTextureFormat\n  ) {\n    if (\n      this._renderToScreen &&\n      this._renderToScreen.presentationFormat !== presentationFormat\n    ) {\n      this._needsUpdatePipeline = true;\n      this._fsCode = '';\n    }\n    this._renderToScreen = {\n      screenTexture,\n      presentationFormat\n    };\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  private readonly _retainedOutputs = new Set<GPUTexture>();\n  retainOutput(texture: GPUTexture) { if (Object.values(this._outputTextures).includes(texture)) this._retainedOutputs.add(texture); }\n  releaseOutput(texture: GPUTexture) {\n    if (!this._retainedOutputs.delete(texture)) return;\n    if (!Object.values(this._outputTextures).includes(texture)) {\n      void this._device.queue.onSubmittedWorkDone().catch(() => undefined).then(() => texture.destroy());\n    }\n  }\n\n  getOutput(name: O) {\n    return this._outputTextures[name];\n  }\n\n  getMemoryBytes() {\n    return Object.values(this._uniformBuffers).reduce((sum, buffer) => sum + buffer.size, 0)\n      + Array.from(new Set([...Object.values(this._outputTextures), ...this._retainedOutputs]))\n        .reduce((sum, texture) => sum + textureBytes(texture), 0);\n  }\n\n  dispose() {\n    Object.keys(this._uniformBuffers).forEach((key) => {\n      (this._uniformBuffers as any)[key].destroy();\n    });\n    new Set([...Object.values(this._outputTextures), ...this._retainedOutputs]).forEach(texture => texture.destroy());\n    this._retainedOutputs.clear();\n  }\n\n  private _createTexture(params: WGPUFullQuadPassOutput) {\n    return this._device.createTexture({\n      label: this._label,\n      size: {\n        width: this._reserveCapacity ? this._capacityWidth : this._width,\n        height: this._reserveCapacity ? this._capacityHeight : this._height,\n        depthOrArrayLayers: 1\n      },\n      format: params.format,\n      usage:\n        GPUTextureUsage.RENDER_ATTACHMENT |\n        GPUTextureUsage.TEXTURE_BINDING |\n        (this._textureWillBeRead ? GPUTextureUsage.COPY_SRC : 0)\n    });\n  }\n\n  private _updateOutputTextures(excludes?: string[]) {\n    const outputParams = this._outputParams;\n    for (const key in outputParams) {\n      if (excludes?.includes(key)) {\n        continue;\n      }\n      const params = outputParams[key];\n      if (!this._retainedOutputs.has(this._outputTextures[key])) this._outputTextures[key]?.destroy();\n      this._outputTextures[key] = this._createTexture(params);\n    }\n  }\n\n  private _updatePipeline() {\n    if (!this._needsUpdatePipeline) {\n      return;\n    }\n    this._needsUpdatePipeline = false;\n\n    const outputParams = this._outputParams;\n    const device = this._device;\n    const fsCode = this._getFullFs();\n    if (fsCode === this._fsCode) {\n      return;\n    }\n    const { screenTexture, presentationFormat } = this._renderToScreen || {};\n    this._fsCode = fsCode;\n    this._pipeline = getTargetShaderCache(device).createRenderPipeline({\n      label: this._label,\n      layout: 'auto',\n      vertex: {\n        module: getTargetShaderCache(device).createShaderModule({\n          label: this._label,\n          code: fullScreenQuadVertexShaderWGSL\n        }),\n        entryPoint: 'main'\n      },\n      fragment: {\n        module: getTargetShaderCache(device).createShaderModule({\n          label: this._label,\n          code: fsCode\n        }),\n        entryPoint: 'main',\n        targets: screenTexture\n          ? [\n              {\n                format: presentationFormat!\n              }\n            ]\n          : this._outputs.map((key) => ({\n              format: (outputParams as any)[key].format\n            }))\n      },\n      primitive: {\n        topology: 'triangle-list'\n      }\n    });\n    this._updateBindGroups();\n  }\n\n  private _getFullFs() {\n    const inputs = this._inputs;\n    const inputsLen = inputs.length;\n    const hasInputs = inputsLen > 0;\n    const fs = /* wgsl */ `\n${inputs\n  .sort()\n  .map(\n    (textureName, idx) => {\n      const viewDimension = this._inputViewDimensions[textureName as I];\n      const textureType = viewDimension === '3d' ? 'texture_3d' : 'texture_2d';\n      return `@group(0) @binding(${idx}) var ${textureName}: ${textureType}<f32>;`;\n    }\n  )\n  .join('\\n')}\n\n${\n  hasInputs\n    ? Object.keys(this._samplers)\n        .sort()\n        .map(\n          (samplerName, idx) =>\n            `@group(0) @binding(${\n              idx + inputsLen\n            }) var ${samplerName}: sampler;`\n        )\n        .join('\\n')\n    : ''\n}\n${this._uniforms\n  .map(\n    (uniform, idx) =>\n      `@group(${hasInputs ? 1 : 0}) @binding(${idx}) var<uniform> ${\n        uniform.label\n      }: ${uniform.type};`\n  )\n  .join('\\n')}\nstruct FSOutput {\n${this._outputs\n  .map((name, idx) => `@location(${idx}) ${name}: vec4f,`)\n  .join('\\n')}\n}\n${scratchWGSLFunctions(\n  this._reserveCapacity ? this._regionNames : inputs.filter(name =>\n    !this._inputViewDimensions[name as I] || this._inputViewDimensions[name as I] === '2d'),\n  Object.keys(this._samplers), this._reserveCapacity\n)}\n${this._fsDefine ?? ''}\n@fragment\nfn main(\n  @builtin(position) coord: vec4f\n) -> FSOutput {\n  let uv = vec2i(floor(coord.xy));\n  let uv2 = coord.xy / viewportSize;\n  var output: FSOutput;\n${this._fsMain}\n  return output;\n}\n`;\n\n    return fs;\n  }\n\n  private _updateBindGroups() {\n    const bindGroups: GPUBindGroup[] = [];\n    const device = this._device;\n    const uniformBindGroupIndex = this._inputs.length > 0 ? 1 : 0;\n    if (this._uniforms.length > 0) {\n      bindGroups[uniformBindGroupIndex] = device.createBindGroup({\n        label: this._label,\n        layout: this._pipeline.getBindGroupLayout(uniformBindGroupIndex),\n        entries: this._uniforms.map(\n          (uniform, idx) =>\n            ({\n              binding: idx,\n              resource: {\n                buffer: this._uniformBuffers[uniform.label]\n              }\n            } satisfies GPUBindGroupEntry)\n        )\n      });\n    }\n\n    this._bindGroups = bindGroups;\n  }\n\n  createPass(\n    commandEncoder: GPUCommandEncoder,\n    inputs: Record<I, GPUTexture>,\n\n    /**\n     * Explict outputs. Used for buffers reuse.\n     */\n    outputs?: Record<O, GPUTexture>\n  ) {\n    if (this._needsUpdateOutputTextures && !this._renderToScreen) {\n      this._updateOutputTextures(Object.keys(outputs || {}));\n      this._needsUpdateOutputTextures = false;\n    }\n\n    if (this._regionNames.length) {\n      const values = new Uint32Array(this._regionNames.length * 4);\n      for (const [index, name] of this._regionNames.entries()) {\n        const region = textureRegion(inputs[name as I]);\n        values.set([region.width, region.height], index * 4);\n      }\n      this.setUniform('imageRegions', values);\n    }\n    for (const texture of Object.values(this._outputTextures)) setTextureRegion(texture, this._width, this._height);\n    this._updatePipeline();\n\n    const inputsLen = this._inputs.length;\n    if (inputsLen > 0) {\n      this._bindGroups[0] = this._device.createBindGroup({\n        label: this._label,\n        layout: this._pipeline.getBindGroupLayout(0),\n        entries: [\n          ...this._inputs.map((textureName, idx) => ({\n            binding: idx,\n            resource: inputs[textureName as I].createView(\n              this._inputViewDimensions[textureName as I]\n                ? { dimension: this._inputViewDimensions[textureName as I] }\n                : undefined\n            )\n          })),\n          ...Object.keys(this._samplers).map((samplerName, idx) => ({\n            binding: idx + inputsLen,\n            resource: this._samplers[samplerName]\n          }))\n        ]\n      });\n    }\n\n    const outputTextures = {\n      // Pick the texture from outputTextures first.\n      ...this._outputTextures,\n      ...outputs\n    } as Record<string, GPUTexture>;\n    // Begin the render pass\n    const renderPass = commandEncoder.beginRenderPass({\n      colorAttachments: this._renderToScreen\n        ? [\n            {\n              view: this._renderToScreen.screenTexture.createView(),\n              clearValue: { r: 0, g: 0, b: 0, a: 0 },\n              storeOp: 'store',\n              loadOp: 'clear'\n            }\n          ]\n        : this._outputs.map(\n            (textureName) =>\n              ({\n                view: outputTextures[textureName].createView(),\n                clearValue: { r: 0, g: 0, b: 0, a: 0 },\n                loadOp: 'clear',\n                storeOp: 'store'\n              } satisfies GPURenderPassColorAttachment)\n          )\n    });\n\n    // Draw a full quad\n    renderPass.setViewport(0, 0, this._width, this._height, 0, 1);\n    renderPass.setScissorRect(0, 0, this._width, this._height);\n    renderPass.setPipeline(this._pipeline);\n    // Bind groups\n    this._bindGroups.forEach((bindGroup, idx) => {\n      renderPass.setBindGroup(idx, bindGroup);\n    });\n    renderPass.draw(6, 1, 0, 0);\n    // End the render pass\n    renderPass.end();\n  }\n}\n"]}
+{"version":3,"file":"WGPUFullQuadPass.js","sourceRoot":"","sources":["../../src/renderer/WGPUFullQuadPass.ts"],"names":[],"mappings":"AAAA,OAAO,EAAC,oBAAoB,EAAC,MAAM,sBAAsB,CAAC;AAC1D,OAAO,EAAC,gBAAgB,EAAE,aAAa,EAAC,MAAM,oBAAoB,CAAC;AACnE,OAAO,EAAE,oBAAoB,EAAE,MAAM,sBAAsB,CAAC;AAC5D,OAAO,EAAE,YAAY,EAAE,MAAM,gBAAgB,CAAC;AAC9C,MAAM,CAAC,MAAM,8BAA8B,GAAG,SAAS,CAAC;;;;;;;;;;;;CAYvD,CAAC;AAEF,SAAS,oBAAoB,CAC3B,MAA8B,EAC9B,KAA6B;IAE7B,OAAO,MAAM,CAAC,MAAM,KAAK,KAAK,CAAC,MAAM,CAAC;AACxC,CAAC;AAED,MAAM,UAAU,gBAAgB,CAC9B,KAAU,EACV,IAAqB;IAErB,IACE,KAAK,YAAY,YAAY;QAC7B,KAAK,YAAY,UAAU;QAC3B,KAAK,YAAY,WAAW,EAC5B,CAAC;QACD,OAAO,KAAK,CAAC;IACf,CAAC;IAED,IAAI,OAAO,KAAK,KAAK,QAAQ,EAAE,CAAC;QAC9B,KAAK,GAAG,CAAC,KAAK,CAAC,CAAC;IAClB,CAAC;IAED,IAAI,IAAI,KAAK,KAAK,IAAI,IAAI,CAAC,QAAQ,CAAC,GAAG,CAAC,EAAE,CAAC;QACzC,OAAO,IAAI,UAAU,CAAC,KAAK,CAAC,CAAC;IAC/B,CAAC;SAAM,IAAI,IAAI,KAAK,KAAK,IAAI,IAAI,CAAC,QAAQ,CAAC,GAAG,CAAC,EAAE,CAAC;QAChD,OAAO,IAAI,WAAW,CAAC,KAAK,CAAC,CAAC;IAChC,CAAC;SAAM,CAAC;QACN,OAAO,IAAI,YAAY,CAAC,KAAK,CAAC,CAAC;IACjC,CAAC;AACH,CAAC;AAwBD,MAAM,OAAO,gBAAgB;IACnB,MAAM,CAAC;IAEP,OAAO,CAAC;IACR,eAAe,GAA+B,EAAE,CAAC;IACjD,aAAa,GAA2C,EAAE,CAAC;IAE3D,SAAS,CAAqB;IAC9B,WAAW,GAAmB,EAAE,CAAC;IACjC,oBAAoB,GAAG,IAAI,CAAC;IAC5B,0BAA0B,GAAG,IAAI,CAAC;IAC1C;;OAEG;IACK,eAAe,CAGrB;IACM,OAAO,GAAa,EAAE,CAAC;IACvB,oBAAoB,GAExB,EAAE,CAAC;IACC,QAAQ,GAAa,EAAE,CAAC;IACxB,SAAS,GAAc,EAAE,CAAC;IAC1B,YAAY,GAA4B,EAAE,CAAC;IAC3C,eAAe,GAA8B,EAAE,CAAC;IAEhD,SAAS,GAA+B,EAAE,CAAC;IAE3C,cAAc,GAAG,CAAC,CAAC;IACnB,eAAe,GAAG,CAAC,CAAC;IACpB,gBAAgB,GAAG,KAAK,CAAC;IACzB,YAAY,GAAa,EAAE,CAAC;IAC5B,MAAM,GAAG,EAAE,CAAC;IACZ,OAAO,GAAG,EAAE,CAAC;IAEb,OAAO,GAAG,EAAE,CAAC;IACJ,OAAO,CAAC;IACR,SAAS,CAAC;IAEnB,kBAAkB,CAAC;IAE3B,YACE,KAAa,EACb,MAAiB,EACjB,IASC;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,oBAAoB,GAAG,IAAI,CAAC,mBAAmB,IAAI,EAAE,CAAC;QAC3D,IAAI,CAAC,QAAQ,GAAG,IAAI,CAAC,OAAO,CAAC;QAC7B,IAAI,CAAC,SAAS,GAAG,CAAC,IAAI,CAAC,QAAQ,IAAI,EAAE,CAAC,CAAC,MAAM,CAAC;YAC5C,KAAK,EAAE,cAAc;YACrB,IAAI,EAAE,OAAO;YACb,IAAI,EAAE,CAAC,CAAC,EAAE,CAAC,CAAC;SACb,CAAC,CAAC;QACH,IAAI,CAAC,SAAS,GAAG,IAAI,CAAC,QAAQ,IAAI,EAAE,CAAC;QACrC,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,IAAI,CAAC,kBAAkB,GAAG,IAAI,CAAC,iBAAiB,IAAI,KAAK,CAAC;QAE1D,IAAI,CAAC,SAAS,CAAC,OAAO,CAAC,CAAC,OAAO,EAAE,EAAE;YACjC,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;gBAClB,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,8FAA8F;IAC9F,eAAe;QACb,IAAI,IAAI,CAAC,gBAAgB;YAAE,OAAO;QAClC,IAAI,CAAC,gBAAgB,GAAG,IAAI,CAAC;QAC7B,IAAI,CAAC,cAAc,GAAG,IAAI,CAAC,MAAM,CAAC;QAAC,IAAI,CAAC,eAAe,GAAG,IAAI,CAAC,OAAO,CAAC;QACvE,MAAM,QAAQ,GAAG,IAAI,CAAC,OAAO,CAAC,MAAM,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC,IAAI,CAAC,oBAAoB,CAAC,IAAS,CAAC,IAAI,IAAI,CAAC,oBAAoB,CAAC,IAAS,CAAC,KAAK,IAAI,CAAC,CAAC;QACrI,IAAI,CAAC,YAAY,GAAG,QAAQ,CAAC;QAC7B,IAAI,QAAQ,CAAC,MAAM,EAAE,CAAC;YACpB,MAAM,OAAO,GAAY,EAAC,KAAK,EAAE,cAAc,EAAE,IAAI,EAAE,gBAAgB,QAAQ,CAAC,MAAM,GAAG,EAAE,IAAI,EAAE,IAAI,WAAW,CAAC,QAAQ,CAAC,MAAM,GAAG,CAAC,CAAC,EAAC,CAAC;YACvI,IAAI,CAAC,SAAS,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC;YAAC,IAAI,CAAC,YAAY,CAAC,OAAO,CAAC,KAAK,CAAC,GAAG,OAAO,CAAC;YACzE,IAAI,CAAC,eAAe,CAAC,OAAO,CAAC,KAAK,CAAC,GAAG,IAAI,CAAC,OAAO,CAAC,YAAY,CAAC,EAAC,KAAK,EAAE,GAAG,IAAI,CAAC,MAAM,IAAI,OAAO,CAAC,KAAK,EAAE,EAAE,IAAI,EAAE,QAAQ,CAAC,MAAM,GAAG,EAAE,EAAE,KAAK,EAAE,cAAc,CAAC,OAAO,GAAG,cAAc,CAAC,QAAQ,EAAC,CAAC,CAAC;QACnM,CAAC;QACD,IAAI,CAAC,oBAAoB,GAAG,IAAI,CAAC;QAAC,IAAI,CAAC,OAAO,GAAG,EAAE,CAAC;IACtD,CAAC;IACD,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,MAAM,SAAS,GAAG,IAAI,CAAC,GAAG,CAAC,IAAI,CAAC,cAAc,EAAE,KAAK,CAAC,CAAC;YACvD,MAAM,UAAU,GAAG,IAAI,CAAC,GAAG,CAAC,IAAI,CAAC,eAAe,EAAE,MAAM,CAAC,CAAC;YAC1D,IAAI,CAAC,0BAA0B,KAAK,CAAC,IAAI,CAAC,gBAAgB,IAAI,SAAS,KAAK,IAAI,CAAC,cAAc,IAAI,UAAU,KAAK,IAAI,CAAC,eAAe,CAAC;YACvI,IAAI,CAAC,cAAc,GAAG,SAAS,CAAC;YAAC,IAAI,CAAC,eAAe,GAAG,UAAU,CAAC;YACnE,IAAI,CAAC,oBAAoB,GAAG,IAAI,CAAC;YAEjC,IAAI,CAAC,UAAU,CAAC,cAAc,EAAE,CAAC,KAAK,EAAE,MAAM,CAAC,CAAC,CAAC;QACnD,CAAC;IACH,CAAC;IAED,eAAe,CAAC,YAA+C;QAC7D,MAAM,SAAS,GAAG,CAAC,CAAC,IAAI,CAAC,eAAe,CAAC;QACzC,IAAI,CAAC,eAAe,GAAG,SAAS,CAAC;QACjC,MAAM,cAAc,GAAG,MAAM,CAAC,IAAI,CAAC,YAAY,CAAC,CAAC,IAAI,CACnD,CAAC,GAAG,EAAE,EAAE,CACN,CAAC,oBAAoB,CAClB,YAAoB,CAAC,GAAG,CAAC,EAC1B,IAAI,CAAC,aAAa,CAAC,GAAG,CAAC,IAAK,EAA6B,CAC1D,CACJ,CAAC;QACF,IAAI,CAAC,0BAA0B,KAAK,cAAc,CAAC;QACnD,IAAI,CAAC,oBAAoB,KAAK,cAAc,IAAI,SAAS,CAAC;QAC1D,IAAI,cAAc,IAAI,SAAS;YAAE,IAAI,CAAC,OAAO,GAAG,EAAE,CAAC;QACnD,IAAI,CAAC,aAAa,GAAG,YAAY,CAAC;IACpC,CAAC;IAED,iBAAiB,CACf,aAAyB,EACzB,kBAAoC;QAEpC,IACE,CAAC,IAAI,CAAC,eAAe;YACrB,IAAI,CAAC,eAAe,CAAC,kBAAkB,KAAK,kBAAkB,EAC9D,CAAC;YACD,IAAI,CAAC,oBAAoB,GAAG,IAAI,CAAC;YACjC,IAAI,CAAC,OAAO,GAAG,EAAE,CAAC;QACpB,CAAC;QACD,IAAI,CAAC,eAAe,GAAG;YACrB,aAAa;YACb,kBAAkB;SACnB,CAAC;IACJ,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;IAEgB,gBAAgB,GAAG,IAAI,GAAG,EAAc,CAAC;IAC1D,YAAY,CAAC,OAAmB,IAAI,IAAI,MAAM,CAAC,MAAM,CAAC,IAAI,CAAC,eAAe,CAAC,CAAC,QAAQ,CAAC,OAAO,CAAC;QAAE,IAAI,CAAC,gBAAgB,CAAC,GAAG,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC;IACpI,aAAa,CAAC,OAAmB;QAC/B,IAAI,CAAC,IAAI,CAAC,gBAAgB,CAAC,MAAM,CAAC,OAAO,CAAC;YAAE,OAAO;QACnD,IAAI,CAAC,MAAM,CAAC,MAAM,CAAC,IAAI,CAAC,eAAe,CAAC,CAAC,QAAQ,CAAC,OAAO,CAAC,EAAE,CAAC;YAC3D,KAAK,IAAI,CAAC,OAAO,CAAC,KAAK,CAAC,mBAAmB,EAAE,CAAC,KAAK,CAAC,GAAG,EAAE,CAAC,SAAS,CAAC,CAAC,IAAI,CAAC,GAAG,EAAE,CAAC,OAAO,CAAC,OAAO,EAAE,CAAC,CAAC;QACrG,CAAC;IACH,CAAC;IAED,SAAS,CAAC,IAAO;QACf,OAAO,IAAI,CAAC,eAAe,CAAC,IAAI,CAAC,CAAC;IACpC,CAAC;IAED,cAAc;QACZ,OAAO,MAAM,CAAC,MAAM,CAAC,IAAI,CAAC,eAAe,CAAC,CAAC,MAAM,CAAC,CAAC,GAAG,EAAE,MAAM,EAAE,EAAE,CAAC,GAAG,GAAG,MAAM,CAAC,IAAI,EAAE,CAAC,CAAC;cACpF,KAAK,CAAC,IAAI,CAAC,IAAI,GAAG,CAAC,CAAC,GAAG,MAAM,CAAC,MAAM,CAAC,IAAI,CAAC,eAAe,CAAC,EAAE,GAAG,IAAI,CAAC,gBAAgB,CAAC,CAAC,CAAC;iBACtF,MAAM,CAAC,CAAC,GAAG,EAAE,OAAO,EAAE,EAAE,CAAC,GAAG,GAAG,YAAY,CAAC,OAAO,CAAC,EAAE,CAAC,CAAC,CAAC;IAChE,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,IAAI,GAAG,CAAC,CAAC,GAAG,MAAM,CAAC,MAAM,CAAC,IAAI,CAAC,eAAe,CAAC,EAAE,GAAG,IAAI,CAAC,gBAAgB,CAAC,CAAC,CAAC,OAAO,CAAC,OAAO,CAAC,EAAE,CAAC,OAAO,CAAC,OAAO,EAAE,CAAC,CAAC;QAClH,IAAI,CAAC,gBAAgB,CAAC,KAAK,EAAE,CAAC;IAChC,CAAC;IAEO,cAAc,CAAC,MAA8B;QACnD,OAAO,IAAI,CAAC,OAAO,CAAC,aAAa,CAAC;YAChC,KAAK,EAAE,IAAI,CAAC,MAAM;YAClB,IAAI,EAAE;gBACJ,KAAK,EAAE,IAAI,CAAC,gBAAgB,CAAC,CAAC,CAAC,IAAI,CAAC,cAAc,CAAC,CAAC,CAAC,IAAI,CAAC,MAAM;gBAChE,MAAM,EAAE,IAAI,CAAC,gBAAgB,CAAC,CAAC,CAAC,IAAI,CAAC,eAAe,CAAC,CAAC,CAAC,IAAI,CAAC,OAAO;gBACnE,kBAAkB,EAAE,CAAC;aACtB;YACD,MAAM,EAAE,MAAM,CAAC,MAAM;YACrB,KAAK,EACH,eAAe,CAAC,iBAAiB;gBACjC,eAAe,CAAC,eAAe;gBAC/B,CAAC,IAAI,CAAC,kBAAkB,CAAC,CAAC,CAAC,eAAe,CAAC,QAAQ,CAAC,CAAC,CAAC,CAAC,CAAC;SAC3D,CAAC,CAAC;IACL,CAAC;IAEO,qBAAqB,CAAC,QAAmB;QAC/C,MAAM,YAAY,GAAG,IAAI,CAAC,aAAa,CAAC;QACxC,KAAK,MAAM,GAAG,IAAI,YAAY,EAAE,CAAC;YAC/B,IAAI,QAAQ,EAAE,QAAQ,CAAC,GAAG,CAAC,EAAE,CAAC;gBAC5B,SAAS;YACX,CAAC;YACD,MAAM,MAAM,GAAG,YAAY,CAAC,GAAG,CAAC,CAAC;YACjC,IAAI,CAAC,IAAI,CAAC,gBAAgB,CAAC,GAAG,CAAC,IAAI,CAAC,eAAe,CAAC,GAAG,CAAC,CAAC;gBAAE,IAAI,CAAC,eAAe,CAAC,GAAG,CAAC,EAAE,OAAO,EAAE,CAAC;YAChG,IAAI,CAAC,eAAe,CAAC,GAAG,CAAC,GAAG,IAAI,CAAC,cAAc,CAAC,MAAM,CAAC,CAAC;QAC1D,CAAC;IACH,CAAC;IAEO,eAAe;QACrB,IAAI,CAAC,IAAI,CAAC,oBAAoB,EAAE,CAAC;YAC/B,OAAO;QACT,CAAC;QACD,IAAI,CAAC,oBAAoB,GAAG,KAAK,CAAC;QAElC,MAAM,YAAY,GAAG,IAAI,CAAC,aAAa,CAAC;QACxC,MAAM,MAAM,GAAG,IAAI,CAAC,OAAO,CAAC;QAC5B,MAAM,MAAM,GAAG,IAAI,CAAC,UAAU,EAAE,CAAC;QACjC,IAAI,MAAM,KAAK,IAAI,CAAC,OAAO,EAAE,CAAC;YAC5B,OAAO;QACT,CAAC;QACD,MAAM,EAAE,aAAa,EAAE,kBAAkB,EAAE,GAAG,IAAI,CAAC,eAAe,IAAI,EAAE,CAAC;QACzE,IAAI,CAAC,OAAO,GAAG,MAAM,CAAC;QACtB,IAAI,CAAC,SAAS,GAAG,oBAAoB,CAAC,MAAM,CAAC,CAAC,oBAAoB,CAAC;YACjE,KAAK,EAAE,IAAI,CAAC,MAAM;YAClB,MAAM,EAAE,MAAM;YACd,MAAM,EAAE;gBACN,MAAM,EAAE,oBAAoB,CAAC,MAAM,CAAC,CAAC,kBAAkB,CAAC;oBACtD,KAAK,EAAE,IAAI,CAAC,MAAM;oBAClB,IAAI,EAAE,8BAA8B;iBACrC,CAAC;gBACF,UAAU,EAAE,MAAM;aACnB;YACD,QAAQ,EAAE;gBACR,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;gBAClB,OAAO,EAAE,aAAa;oBACpB,CAAC,CAAC;wBACE;4BACE,MAAM,EAAE,kBAAmB;yBAC5B;qBACF;oBACH,CAAC,CAAC,IAAI,CAAC,QAAQ,CAAC,GAAG,CAAC,CAAC,GAAG,EAAE,EAAE,CAAC,CAAC;wBAC1B,MAAM,EAAG,YAAoB,CAAC,GAAG,CAAC,CAAC,MAAM;qBAC1C,CAAC,CAAC;aACR;YACD,SAAS,EAAE;gBACT,QAAQ,EAAE,eAAe;aAC1B;SACF,CAAC,CAAC;QACH,IAAI,CAAC,iBAAiB,EAAE,CAAC;IAC3B,CAAC;IAEO,UAAU;QAChB,MAAM,MAAM,GAAG,IAAI,CAAC,OAAO,CAAC;QAC5B,MAAM,SAAS,GAAG,MAAM,CAAC,MAAM,CAAC;QAChC,MAAM,SAAS,GAAG,SAAS,GAAG,CAAC,CAAC;QAChC,MAAM,EAAE,GAAG,UAAU,CAAC;EACxB,MAAM;aACL,IAAI,EAAE;aACN,GAAG,CACF,CAAC,WAAW,EAAE,GAAG,EAAE,EAAE;YACnB,MAAM,aAAa,GAAG,IAAI,CAAC,oBAAoB,CAAC,WAAgB,CAAC,CAAC;YAClE,MAAM,WAAW,GAAG,aAAa,KAAK,IAAI,CAAC,CAAC,CAAC,YAAY,CAAC,CAAC,CAAC,YAAY,CAAC;YACzE,OAAO,sBAAsB,GAAG,SAAS,WAAW,KAAK,WAAW,QAAQ,CAAC;QAC/E,CAAC,CACF;aACA,IAAI,CAAC,IAAI,CAAC;;EAGX,SAAS;YACP,CAAC,CAAC,MAAM,CAAC,IAAI,CAAC,IAAI,CAAC,SAAS,CAAC;iBACxB,IAAI,EAAE;iBACN,GAAG,CACF,CAAC,WAAW,EAAE,GAAG,EAAE,EAAE,CACnB,sBACE,GAAG,GAAG,SACR,SAAS,WAAW,YAAY,CACnC;iBACA,IAAI,CAAC,IAAI,CAAC;YACf,CAAC,CAAC,EACN;EACE,IAAI,CAAC,SAAS;aACb,GAAG,CACF,CAAC,OAAO,EAAE,GAAG,EAAE,EAAE,CACf,UAAU,SAAS,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,cAAc,GAAG,kBAC1C,OAAO,CAAC,KACV,KAAK,OAAO,CAAC,IAAI,GAAG,CACvB;aACA,IAAI,CAAC,IAAI,CAAC;;EAEX,IAAI,CAAC,QAAQ;aACZ,GAAG,CAAC,CAAC,IAAI,EAAE,GAAG,EAAE,EAAE,CAAC,aAAa,GAAG,KAAK,IAAI,UAAU,CAAC;aACvD,IAAI,CAAC,IAAI,CAAC;;EAEX,oBAAoB,CACpB,IAAI,CAAC,gBAAgB,CAAC,CAAC,CAAC,IAAI,CAAC,YAAY,CAAC,CAAC,CAAC,MAAM,CAAC,MAAM,CAAC,IAAI,CAAC,EAAE,CAC/D,CAAC,IAAI,CAAC,oBAAoB,CAAC,IAAS,CAAC,IAAI,IAAI,CAAC,oBAAoB,CAAC,IAAS,CAAC,KAAK,IAAI,CAAC,EACzF,MAAM,CAAC,IAAI,CAAC,IAAI,CAAC,SAAS,CAAC,EAAE,IAAI,CAAC,gBAAgB,CACnD;EACC,IAAI,CAAC,SAAS,IAAI,EAAE;;;;;;;;EAQpB,IAAI,CAAC,OAAO;;;CAGb,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,qBAAqB,GAAG,IAAI,CAAC,OAAO,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC;QAC9D,IAAI,IAAI,CAAC,SAAS,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;YAC9B,UAAU,CAAC,qBAAqB,CAAC,GAAG,MAAM,CAAC,eAAe,CAAC;gBACzD,KAAK,EAAE,IAAI,CAAC,MAAM;gBAClB,MAAM,EAAE,IAAI,CAAC,SAAS,CAAC,kBAAkB,CAAC,qBAAqB,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;iBAC2B,CAAA,CACjC;aACF,CAAC,CAAC;QACL,CAAC;QAED,IAAI,CAAC,WAAW,GAAG,UAAU,CAAC;IAChC,CAAC;IAED,UAAU,CACR,cAAiC,EACjC,MAA6B;IAE7B;;OAEG;IACH,OAA+B;QAE/B,IAAI,IAAI,CAAC,0BAA0B,IAAI,CAAC,IAAI,CAAC,eAAe,EAAE,CAAC;YAC7D,IAAI,CAAC,qBAAqB,CAAC,MAAM,CAAC,IAAI,CAAC,OAAO,IAAI,EAAE,CAAC,CAAC,CAAC;YACvD,IAAI,CAAC,0BAA0B,GAAG,KAAK,CAAC;QAC1C,CAAC;QAED,IAAI,IAAI,CAAC,YAAY,CAAC,MAAM,EAAE,CAAC;YAC7B,MAAM,MAAM,GAAG,IAAI,WAAW,CAAC,IAAI,CAAC,YAAY,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC;YAC7D,KAAK,MAAM,CAAC,KAAK,EAAE,IAAI,CAAC,IAAI,IAAI,CAAC,YAAY,CAAC,OAAO,EAAE,EAAE,CAAC;gBACxD,MAAM,MAAM,GAAG,aAAa,CAAC,MAAM,CAAC,IAAS,CAAC,CAAC,CAAC;gBAChD,MAAM,CAAC,GAAG,CAAC,CAAC,MAAM,CAAC,KAAK,EAAE,MAAM,CAAC,MAAM,CAAC,EAAE,KAAK,GAAG,CAAC,CAAC,CAAC;YACvD,CAAC;YACD,IAAI,CAAC,UAAU,CAAC,cAAc,EAAE,MAAM,CAAC,CAAC;QAC1C,CAAC;QACD,KAAK,MAAM,OAAO,IAAI,MAAM,CAAC,MAAM,CAAC,IAAI,CAAC,eAAe,CAAC;YAAE,gBAAgB,CAAC,OAAO,EAAE,IAAI,CAAC,MAAM,EAAE,IAAI,CAAC,OAAO,CAAC,CAAC;QAChH,IAAI,CAAC,eAAe,EAAE,CAAC;QAEvB,MAAM,SAAS,GAAG,IAAI,CAAC,OAAO,CAAC,MAAM,CAAC;QACtC,IAAI,SAAS,GAAG,CAAC,EAAE,CAAC;YAClB,IAAI,CAAC,WAAW,CAAC,CAAC,CAAC,GAAG,IAAI,CAAC,OAAO,CAAC,eAAe,CAAC;gBACjD,KAAK,EAAE,IAAI,CAAC,MAAM;gBAClB,MAAM,EAAE,IAAI,CAAC,SAAS,CAAC,kBAAkB,CAAC,CAAC,CAAC;gBAC5C,OAAO,EAAE;oBACP,GAAG,IAAI,CAAC,OAAO,CAAC,GAAG,CAAC,CAAC,WAAW,EAAE,GAAG,EAAE,EAAE,CAAC,CAAC;wBACzC,OAAO,EAAE,GAAG;wBACZ,QAAQ,EAAE,MAAM,CAAC,WAAgB,CAAC,CAAC,UAAU,CAC3C,IAAI,CAAC,oBAAoB,CAAC,WAAgB,CAAC;4BACzC,CAAC,CAAC,EAAE,SAAS,EAAE,IAAI,CAAC,oBAAoB,CAAC,WAAgB,CAAC,EAAE;4BAC5D,CAAC,CAAC,SAAS,CACd;qBACF,CAAC,CAAC;oBACH,GAAG,MAAM,CAAC,IAAI,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC,GAAG,CAAC,CAAC,WAAW,EAAE,GAAG,EAAE,EAAE,CAAC,CAAC;wBACxD,OAAO,EAAE,GAAG,GAAG,SAAS;wBACxB,QAAQ,EAAE,IAAI,CAAC,SAAS,CAAC,WAAW,CAAC;qBACtC,CAAC,CAAC;iBACJ;aACF,CAAC,CAAC;QACL,CAAC;QAED,MAAM,cAAc,GAAG;YACrB,8CAA8C;YAC9C,GAAG,IAAI,CAAC,eAAe;YACvB,GAAG,OAAO;SACmB,CAAC;QAChC,wBAAwB;QACxB,MAAM,UAAU,GAAG,cAAc,CAAC,eAAe,CAAC;YAChD,gBAAgB,EAAE,IAAI,CAAC,eAAe;gBACpC,CAAC,CAAC;oBACE;wBACE,IAAI,EAAE,IAAI,CAAC,eAAe,CAAC,aAAa,CAAC,UAAU,EAAE;wBACrD,UAAU,EAAE,EAAE,CAAC,EAAE,CAAC,EAAE,CAAC,EAAE,CAAC,EAAE,CAAC,EAAE,CAAC,EAAE,CAAC,EAAE,CAAC,EAAE;wBACtC,OAAO,EAAE,OAAO;wBAChB,MAAM,EAAE,OAAO;qBAChB;iBACF;gBACH,CAAC,CAAC,IAAI,CAAC,QAAQ,CAAC,GAAG,CACf,CAAC,WAAW,EAAE,EAAE,CACd,CAAC;oBACC,IAAI,EAAE,cAAc,CAAC,WAAW,CAAC,CAAC,UAAU,EAAE;oBAC9C,UAAU,EAAE,EAAE,CAAC,EAAE,CAAC,EAAE,CAAC,EAAE,CAAC,EAAE,CAAC,EAAE,CAAC,EAAE,CAAC,EAAE,CAAC,EAAE;oBACtC,MAAM,EAAE,OAAO;oBACf,OAAO,EAAE,OAAO;iBACuB,CAAA,CAC5C;SACN,CAAC,CAAC;QAEH,mBAAmB;QACnB,UAAU,CAAC,WAAW,CAAC,CAAC,EAAE,CAAC,EAAE,IAAI,CAAC,MAAM,EAAE,IAAI,CAAC,OAAO,EAAE,CAAC,EAAE,CAAC,CAAC,CAAC;QAC9D,UAAU,CAAC,cAAc,CAAC,CAAC,EAAE,CAAC,EAAE,IAAI,CAAC,MAAM,EAAE,IAAI,CAAC,OAAO,CAAC,CAAC;QAC3D,UAAU,CAAC,WAAW,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC;QACvC,cAAc;QACd,IAAI,CAAC,WAAW,CAAC,OAAO,CAAC,CAAC,SAAS,EAAE,GAAG,EAAE,EAAE;YAC1C,UAAU,CAAC,YAAY,CAAC,GAAG,EAAE,SAAS,CAAC,CAAC;QAC1C,CAAC,CAAC,CAAC;QACH,UAAU,CAAC,IAAI,CAAC,CAAC,EAAE,CAAC,EAAE,CAAC,EAAE,CAAC,CAAC,CAAC;QAC5B,sBAAsB;QACtB,UAAU,CAAC,GAAG,EAAE,CAAC;IACnB,CAAC;CACF","sourcesContent":["import {scratchWGSLFunctions} from './scratchSampling.js';\nimport {setTextureRegion, textureRegion} from './textureRegion.js';\nimport { getTargetShaderCache } from './WGPUShaderCache.js';\nimport { textureBytes } from './gpuMemory.js';\nexport const fullScreenQuadVertexShaderWGSL = /*wgsl */ `\n@vertex\nfn main(\n  @builtin(vertex_index) vertexIndex: u32\n) -> @builtin(position) vec4f {\n  const pos = array(\n    vec2(-1.0, -1.0), vec2(1.0, -1.0), vec2(-1.0, 1.0),\n    vec2(-1.0, 1.0), vec2(1.0, -1.0), vec2(1.0, 1.0),\n  );\n\n  return vec4<f32>(pos[vertexIndex], 0.0, 1.0);\n}\n`;\n\nfunction isTextureParamsEqual(\n  params: WGPUFullQuadPassOutput,\n  other: WGPUFullQuadPassOutput\n) {\n  return params.format === other.format;\n}\n\nexport function getUniformBuffer(\n  value: any,\n  type: Uniform['type']\n): Float32Array | Int32Array | Uint32Array {\n  if (\n    value instanceof Float32Array ||\n    value instanceof Int32Array ||\n    value instanceof Uint32Array\n  ) {\n    return value;\n  }\n\n  if (typeof value === 'number') {\n    value = [value];\n  }\n\n  if (type === 'i32' || type.endsWith('i')) {\n    return new Int32Array(value);\n  } else if (type === 'u32' || type.endsWith('u')) {\n    return new Uint32Array(value);\n  } else {\n    return new Float32Array(value);\n  }\n}\n\nexport interface Uniform {\n  label: string;\n  type:\n    | `array<vec4u, ${number}>`\n    | 'f32'\n    | 'i32'\n    | 'u32'\n    | 'vec3f'\n    | 'vec4f'\n    | 'vec2f'\n    | 'vec2i'\n    | 'vec3i'\n    | 'vec4i'\n    | 'vec2u'\n    | 'vec3u'\n    | 'vec4u';\n  data: Float32Array | Int32Array | Uint32Array | number | number[];\n}\n\nexport interface WGPUFullQuadPassOutput {\n  format: GPUTextureFormat;\n}\nexport class WGPUFullQuadPass<I extends string, O extends string> {\n  private _label;\n\n  private _device;\n  private _outputTextures: Record<string, GPUTexture> = {};\n  private _outputParams: Record<string, WGPUFullQuadPassOutput> = {};\n\n  private _pipeline!: GPURenderPipeline;\n  private _bindGroups: GPUBindGroup[] = [];\n  private _needsUpdatePipeline = true;\n  private _needsUpdateOutputTextures = true;\n  /**\n   * When render to canvas\n   */\n  private _renderToScreen?: {\n    screenTexture: GPUTexture;\n    presentationFormat: GPUTextureFormat;\n  };\n  private _inputs: string[] = [];\n  private _inputViewDimensions: Partial<\n    Record<I, GPUTextureViewDimension>\n  > = {};\n  private _outputs: string[] = [];\n  private _uniforms: Uniform[] = [];\n  private _uniformsMap: Record<string, Uniform> = {};\n  private _uniformBuffers: Record<string, GPUBuffer> = {};\n\n  private _samplers: Record<string, GPUSampler> = {};\n\n  private _capacityWidth = 0;\n  private _capacityHeight = 0;\n  private _reserveCapacity = false;\n  private _regionNames: string[] = [];\n  private _width = 10;\n  private _height = 10;\n\n  private _fsCode = '';\n  private readonly _fsMain;\n  private readonly _fsDefine;\n\n  private _textureWillBeRead;\n\n  constructor(\n    label: string,\n    device: GPUDevice,\n    opts: {\n      inputs: I[];\n      inputViewDimensions?: Partial<Record<I, GPUTextureViewDimension>>;\n      outputs: O[];\n      samplers?: Record<string, GPUSampler>;\n      fsMain: string;\n      fsDefine?: string;\n      uniforms?: Uniform[];\n      textureWillBeRead?: boolean;\n    }\n  ) {\n    this._label = label;\n    this._device = device;\n    this._fsMain = opts.fsMain;\n    this._fsDefine = opts.fsDefine;\n    this._inputs = opts.inputs;\n    this._inputViewDimensions = opts.inputViewDimensions || {};\n    this._outputs = opts.outputs;\n    this._uniforms = (opts.uniforms || []).concat({\n      label: 'viewportSize',\n      type: 'vec2f',\n      data: [0, 0]\n    });\n    this._samplers = opts.samplers || {};\n    this._uniformsMap = this._uniforms.reduce((map, uniform) => {\n      map[uniform.label] = uniform;\n      return map;\n    }, {} as Record<string, Uniform>);\n    this._textureWillBeRead = opts.textureWillBeRead || false;\n\n    this._uniforms.forEach((uniform) => {\n      const data = getUniformBuffer(uniform.data, uniform.type);\n      this._uniformBuffers[uniform.label] = device.createBuffer({\n        label: this._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  /** Keep physical outputs large enough for every render while varying the logical viewport. */\n  reserveCapacity() {\n    if (this._reserveCapacity) return;\n    this._reserveCapacity = true;\n    this._capacityWidth = this._width; this._capacityHeight = this._height;\n    const textures = this._inputs.filter(name => !this._inputViewDimensions[name as I] || this._inputViewDimensions[name as I] === '2d');\n    this._regionNames = textures;\n    if (textures.length) {\n      const uniform: Uniform = {label: 'imageRegions', type: `array<vec4u, ${textures.length}>`, data: new Uint32Array(textures.length * 4)};\n      this._uniforms.push(uniform); this._uniformsMap[uniform.label] = uniform;\n      this._uniformBuffers[uniform.label] = this._device.createBuffer({label: `${this._label}.${uniform.label}`, size: textures.length * 16, usage: GPUBufferUsage.UNIFORM | GPUBufferUsage.COPY_DST});\n    }\n    this._needsUpdatePipeline = true; this._fsCode = '';\n  }\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      const nextWidth = Math.max(this._capacityWidth, width);\n      const nextHeight = Math.max(this._capacityHeight, height);\n      this._needsUpdateOutputTextures ||= !this._reserveCapacity || nextWidth !== this._capacityWidth || nextHeight !== this._capacityHeight;\n      this._capacityWidth = nextWidth; this._capacityHeight = nextHeight;\n      this._needsUpdatePipeline = true;\n\n      this.setUniform('viewportSize', [width, height]);\n    }\n  }\n\n  setOutputParams(outputParams: Record<O, WGPUFullQuadPassOutput>) {\n    const wasScreen = !!this._renderToScreen;\n    this._renderToScreen = undefined;\n    const outputsChanged = Object.keys(outputParams).some(\n      (key) =>\n        !isTextureParamsEqual(\n          (outputParams as any)[key],\n          this._outputParams[key] || ({} as WGPUFullQuadPassOutput)\n        )\n    );\n    this._needsUpdateOutputTextures ||= outputsChanged;\n    this._needsUpdatePipeline ||= outputsChanged || wasScreen;\n    if (outputsChanged || wasScreen) this._fsCode = '';\n    this._outputParams = outputParams;\n  }\n\n  setRenderToScreen(\n    screenTexture: GPUTexture,\n    presentationFormat: GPUTextureFormat\n  ) {\n    if (\n      !this._renderToScreen ||\n      this._renderToScreen.presentationFormat !== presentationFormat\n    ) {\n      this._needsUpdatePipeline = true;\n      this._fsCode = '';\n    }\n    this._renderToScreen = {\n      screenTexture,\n      presentationFormat\n    };\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  private readonly _retainedOutputs = new Set<GPUTexture>();\n  retainOutput(texture: GPUTexture) { if (Object.values(this._outputTextures).includes(texture)) this._retainedOutputs.add(texture); }\n  releaseOutput(texture: GPUTexture) {\n    if (!this._retainedOutputs.delete(texture)) return;\n    if (!Object.values(this._outputTextures).includes(texture)) {\n      void this._device.queue.onSubmittedWorkDone().catch(() => undefined).then(() => texture.destroy());\n    }\n  }\n\n  getOutput(name: O) {\n    return this._outputTextures[name];\n  }\n\n  getMemoryBytes() {\n    return Object.values(this._uniformBuffers).reduce((sum, buffer) => sum + buffer.size, 0)\n      + Array.from(new Set([...Object.values(this._outputTextures), ...this._retainedOutputs]))\n        .reduce((sum, texture) => sum + textureBytes(texture), 0);\n  }\n\n  dispose() {\n    Object.keys(this._uniformBuffers).forEach((key) => {\n      (this._uniformBuffers as any)[key].destroy();\n    });\n    new Set([...Object.values(this._outputTextures), ...this._retainedOutputs]).forEach(texture => texture.destroy());\n    this._retainedOutputs.clear();\n  }\n\n  private _createTexture(params: WGPUFullQuadPassOutput) {\n    return this._device.createTexture({\n      label: this._label,\n      size: {\n        width: this._reserveCapacity ? this._capacityWidth : this._width,\n        height: this._reserveCapacity ? this._capacityHeight : this._height,\n        depthOrArrayLayers: 1\n      },\n      format: params.format,\n      usage:\n        GPUTextureUsage.RENDER_ATTACHMENT |\n        GPUTextureUsage.TEXTURE_BINDING |\n        (this._textureWillBeRead ? GPUTextureUsage.COPY_SRC : 0)\n    });\n  }\n\n  private _updateOutputTextures(excludes?: string[]) {\n    const outputParams = this._outputParams;\n    for (const key in outputParams) {\n      if (excludes?.includes(key)) {\n        continue;\n      }\n      const params = outputParams[key];\n      if (!this._retainedOutputs.has(this._outputTextures[key])) this._outputTextures[key]?.destroy();\n      this._outputTextures[key] = this._createTexture(params);\n    }\n  }\n\n  private _updatePipeline() {\n    if (!this._needsUpdatePipeline) {\n      return;\n    }\n    this._needsUpdatePipeline = false;\n\n    const outputParams = this._outputParams;\n    const device = this._device;\n    const fsCode = this._getFullFs();\n    if (fsCode === this._fsCode) {\n      return;\n    }\n    const { screenTexture, presentationFormat } = this._renderToScreen || {};\n    this._fsCode = fsCode;\n    this._pipeline = getTargetShaderCache(device).createRenderPipeline({\n      label: this._label,\n      layout: 'auto',\n      vertex: {\n        module: getTargetShaderCache(device).createShaderModule({\n          label: this._label,\n          code: fullScreenQuadVertexShaderWGSL\n        }),\n        entryPoint: 'main'\n      },\n      fragment: {\n        module: getTargetShaderCache(device).createShaderModule({\n          label: this._label,\n          code: fsCode\n        }),\n        entryPoint: 'main',\n        targets: screenTexture\n          ? [\n              {\n                format: presentationFormat!\n              }\n            ]\n          : this._outputs.map((key) => ({\n              format: (outputParams as any)[key].format\n            }))\n      },\n      primitive: {\n        topology: 'triangle-list'\n      }\n    });\n    this._updateBindGroups();\n  }\n\n  private _getFullFs() {\n    const inputs = this._inputs;\n    const inputsLen = inputs.length;\n    const hasInputs = inputsLen > 0;\n    const fs = /* wgsl */ `\n${inputs\n  .sort()\n  .map(\n    (textureName, idx) => {\n      const viewDimension = this._inputViewDimensions[textureName as I];\n      const textureType = viewDimension === '3d' ? 'texture_3d' : 'texture_2d';\n      return `@group(0) @binding(${idx}) var ${textureName}: ${textureType}<f32>;`;\n    }\n  )\n  .join('\\n')}\n\n${\n  hasInputs\n    ? Object.keys(this._samplers)\n        .sort()\n        .map(\n          (samplerName, idx) =>\n            `@group(0) @binding(${\n              idx + inputsLen\n            }) var ${samplerName}: sampler;`\n        )\n        .join('\\n')\n    : ''\n}\n${this._uniforms\n  .map(\n    (uniform, idx) =>\n      `@group(${hasInputs ? 1 : 0}) @binding(${idx}) var<uniform> ${\n        uniform.label\n      }: ${uniform.type};`\n  )\n  .join('\\n')}\nstruct FSOutput {\n${this._outputs\n  .map((name, idx) => `@location(${idx}) ${name}: vec4f,`)\n  .join('\\n')}\n}\n${scratchWGSLFunctions(\n  this._reserveCapacity ? this._regionNames : inputs.filter(name =>\n    !this._inputViewDimensions[name as I] || this._inputViewDimensions[name as I] === '2d'),\n  Object.keys(this._samplers), this._reserveCapacity\n)}\n${this._fsDefine ?? ''}\n@fragment\nfn main(\n  @builtin(position) coord: vec4f\n) -> FSOutput {\n  let uv = vec2i(floor(coord.xy));\n  let uv2 = coord.xy / viewportSize;\n  var output: FSOutput;\n${this._fsMain}\n  return output;\n}\n`;\n\n    return fs;\n  }\n\n  private _updateBindGroups() {\n    const bindGroups: GPUBindGroup[] = [];\n    const device = this._device;\n    const uniformBindGroupIndex = this._inputs.length > 0 ? 1 : 0;\n    if (this._uniforms.length > 0) {\n      bindGroups[uniformBindGroupIndex] = device.createBindGroup({\n        label: this._label,\n        layout: this._pipeline.getBindGroupLayout(uniformBindGroupIndex),\n        entries: this._uniforms.map(\n          (uniform, idx) =>\n            ({\n              binding: idx,\n              resource: {\n                buffer: this._uniformBuffers[uniform.label]\n              }\n            } satisfies GPUBindGroupEntry)\n        )\n      });\n    }\n\n    this._bindGroups = bindGroups;\n  }\n\n  createPass(\n    commandEncoder: GPUCommandEncoder,\n    inputs: Record<I, GPUTexture>,\n\n    /**\n     * Explict outputs. Used for buffers reuse.\n     */\n    outputs?: Record<O, GPUTexture>\n  ) {\n    if (this._needsUpdateOutputTextures && !this._renderToScreen) {\n      this._updateOutputTextures(Object.keys(outputs || {}));\n      this._needsUpdateOutputTextures = false;\n    }\n\n    if (this._regionNames.length) {\n      const values = new Uint32Array(this._regionNames.length * 4);\n      for (const [index, name] of this._regionNames.entries()) {\n        const region = textureRegion(inputs[name as I]);\n        values.set([region.width, region.height], index * 4);\n      }\n      this.setUniform('imageRegions', values);\n    }\n    for (const texture of Object.values(this._outputTextures)) setTextureRegion(texture, this._width, this._height);\n    this._updatePipeline();\n\n    const inputsLen = this._inputs.length;\n    if (inputsLen > 0) {\n      this._bindGroups[0] = this._device.createBindGroup({\n        label: this._label,\n        layout: this._pipeline.getBindGroupLayout(0),\n        entries: [\n          ...this._inputs.map((textureName, idx) => ({\n            binding: idx,\n            resource: inputs[textureName as I].createView(\n              this._inputViewDimensions[textureName as I]\n                ? { dimension: this._inputViewDimensions[textureName as I] }\n                : undefined\n            )\n          })),\n          ...Object.keys(this._samplers).map((samplerName, idx) => ({\n            binding: idx + inputsLen,\n            resource: this._samplers[samplerName]\n          }))\n        ]\n      });\n    }\n\n    const outputTextures = {\n      // Pick the texture from outputTextures first.\n      ...this._outputTextures,\n      ...outputs\n    } as Record<string, GPUTexture>;\n    // Begin the render pass\n    const renderPass = commandEncoder.beginRenderPass({\n      colorAttachments: this._renderToScreen\n        ? [\n            {\n              view: this._renderToScreen.screenTexture.createView(),\n              clearValue: { r: 0, g: 0, b: 0, a: 0 },\n              storeOp: 'store',\n              loadOp: 'clear'\n            }\n          ]\n        : this._outputs.map(\n            (textureName) =>\n              ({\n                view: outputTextures[textureName].createView(),\n                clearValue: { r: 0, g: 0, b: 0, a: 0 },\n                loadOp: 'clear',\n                storeOp: 'store'\n              } satisfies GPURenderPassColorAttachment)\n          )\n    });\n\n    // Draw a full quad\n    renderPass.setViewport(0, 0, this._width, this._height, 0, 1);\n    renderPass.setScissorRect(0, 0, this._width, this._height);\n    renderPass.setPipeline(this._pipeline);\n    // Bind groups\n    this._bindGroups.forEach((bindGroup, idx) => {\n      renderPass.setBindGroup(idx, bindGroup);\n    });\n    renderPass.draw(6, 1, 0, 0);\n    // End the render pass\n    renderPass.end();\n  }\n}\n"]}

package/lib/renderer/WGPUJNLMDenoiser.d.ts

@@ -1,25 +1,6 @@
 import { type JNLMFilterShape } from './jnlm.js';
+export { createJNLMFilterWGSL, jnlmGuideWGSL, jnlmVarianceWGSL, jnlmLoadColorWGSL, JNLM_WORKGROUP_SIZE } from './wgsl/jnlm.compute.js';
 /** WebGPU passes of the JNLM denoiser described in `jnlm.ts`. */
-export declare const JNLM_WORKGROUP_SIZE = 8;
-/**
- * Pass 1: the guide mean and the variance of that mean from the accumulated
- * moments. Ray buffers are stored bottom-up; guides use texture rows.
- */
-export declare const jnlmGuideWGSL: string;
-/**
- * Pass 2: smooth the noisy per-pixel variance over 3 x 3 pixels. Pixels with
- * a single sample take the spatial variance of their neighborhood instead.
- */
-export declare const jnlmVarianceWGSL = "\n\nstruct JNLMParams {\n  size: vec2u,\n  colorSensitivity: f32,\n  // Pixel spacing of the search window and patches in this pass.\n  dilation: f32,\n}\n@group(0) @binding(0) var<uniform> params: JNLMParams;\n\n@group(0) @binding(1) var<storage, read> rawGuide: array<vec2f>;\n@group(0) @binding(2) var<storage, read_write> guide: array<vec2f>;\n\n@compute @workgroup_size(8, 8, 1)\nfn main(@builtin(global_invocation_id) id: vec3u) {\n  let size = params.size;\n  if (id.x >= size.x || id.y >= size.y) { return; }\n  let center = rawGuide[id.x + id.y * size.x];\n  var varianceSum = 0.0;\n  var varianceCount = 0.0;\n  var meanSum = 0.0;\n  var meanSquareSum = 0.0;\n  for (var dy = -1; dy <= 1; dy++) {\n    for (var dx = -1; dx <= 1; dx++) {\n      let q = clamp(vec2i(id.xy) + vec2i(dx, dy), vec2i(0), vec2i(size) - 1);\n      let value = rawGuide[u32(q.x) + u32(q.y) * size.x];\n      meanSum += value.x;\n      meanSquareSum += value.x * value.x;\n      if (value.y >= 0.0) {\n        varianceSum += value.y;\n        varianceCount += 1.0;\n      }\n    }\n  }\n  var variance: f32;\n  if (center.y >= 0.0) {\n    variance = varianceSum / varianceCount;\n  } else {\n    let spatialMean = meanSum / 9.0;\n    variance = max(meanSquareSum / 9.0 - spatialMean * spatialMean, 0.0) * 9.0 / 8.0;\n  }\n  guide[id.x + id.y * size.x] = vec2f(center.x, variance);\n}\n";
-/** Copies the composed image into the filter's color buffer. */
-export declare const jnlmLoadColorWGSL = "\n\nstruct JNLMParams {\n  size: vec2u,\n  colorSensitivity: f32,\n  // Pixel spacing of the search window and patches in this pass.\n  dilation: f32,\n}\n@group(0) @binding(0) var<uniform> params: JNLMParams;\n\n@group(0) @binding(1) var colorTex: texture_2d<f32>;\n@group(0) @binding(2) var<storage, read_write> outColor: array<vec4f>;\n\n@compute @workgroup_size(8, 8, 1)\nfn main(@builtin(global_invocation_id) id: vec3u) {\n  if (id.x >= params.size.x || id.y >= params.size.y) { return; }\n  outColor[id.x + id.y * params.size.x] = textureLoad(colorTex, vec2i(id.xy), 0);\n}\n";
-/**
- * Pass 3: the joint NL-means filter of the composed image. It also filters
- * the guide with the same weights, with the variance of the weighted mean,
- * for a following pass. The albedo regression corrects color only, keeping
- * uncertain fitted texture responses out of the matching guide.
- */
-export declare function createJNLMFilterWGSL({ searchRadius, patchRadius, albedoBandwidth, normalBandwidth }?: JNLMFilterShape): string;
 export interface JNLMInputs {
     width: number;
     height: number;

package/lib/renderer/WGPUJNLMDenoiser.js

@@ -1,267 +1,7 @@
 import { getTargetShaderCache } from './WGPUShaderCache.js';
-import shadowCatcherAccumulationWGSL from './wgsl/chunks/shadowCatcherAccumulation.wgsl.js';
-import { JNLM_DEFAULT_COLOR_SENSITIVITY, JNLM_DEFAULT_SHAPE, JNLM_ALBEDO_FLOOR, JNLM_ALBEDO_REGULARIZATION, JNLM_COVERAGE_BANDWIDTH, JNLM_MAX_ITERATIONS, jnlmPassDilation } from './jnlm.js';
-/** WebGPU passes of the JNLM denoiser described in `jnlm.ts`. */
-export const JNLM_WORKGROUP_SIZE = 8;
-const paramsWGSL = /* wgsl */ `
-struct JNLMParams {
-size: vec2u,
-colorSensitivity: f32,
-
-dilation: f32,
-}
-@group(0) @binding(0) var<uniform> params: JNLMParams;
-`;
-/**
- * Pass 1: the guide mean and the variance of that mean from the accumulated
- * moments. Ray buffers are stored bottom-up; guides use texture rows.
- */
-export const jnlmGuideWGSL = /* wgsl */ `
-${paramsWGSL}
-${shadowCatcherAccumulationWGSL}
-@group(0) @binding(1) var<storage, read> accumulation: array<ShadowCatcherAccumulation>;
-@group(0) @binding(2) var<storage, read> normalBuffer: array<vec4f>;
-@group(0) @binding(3) var<storage, read_write> rawGuide: array<vec2f>;
-
-@compute @workgroup_size(${JNLM_WORKGROUP_SIZE}, ${JNLM_WORKGROUP_SIZE}, 1)
-fn main(@builtin(global_invocation_id) id: vec3u) {
-let size = params.size;
-if (id.x >= size.x || id.y >= size.y) { return; }
-let ua = id.x + (size.y - 1u - id.y) * size.x;
-let n = normalBuffer[ua].a;
-let moments = accumulation[ua].moments;
-let mean = moments.x / max(n, 1.0);
-
-var variance = -1.0;
-if (n >= 2.0) {
-variance = max(moments.y - n * mean * mean, 0.0) / (n * (n - 1.0));
-}
-rawGuide[id.x + id.y * size.x] = vec2f(mean, variance);
-}
-`;
-/**
- * Pass 2: smooth the noisy per-pixel variance over 3 x 3 pixels. Pixels with
- * a single sample take the spatial variance of their neighborhood instead.
- */
-export const jnlmVarianceWGSL = /* wgsl */ `
-${paramsWGSL}
-@group(0) @binding(1) var<storage, read> rawGuide: array<vec2f>;
-@group(0) @binding(2) var<storage, read_write> guide: array<vec2f>;
-
-@compute @workgroup_size(${JNLM_WORKGROUP_SIZE}, ${JNLM_WORKGROUP_SIZE}, 1)
-fn main(@builtin(global_invocation_id) id: vec3u) {
-let size = params.size;
-if (id.x >= size.x || id.y >= size.y) { return; }
-let center = rawGuide[id.x + id.y * size.x];
-var lu = 0.0;
-var jc = 0.0;
-var mY = 0.0;
-var iX = 0.0;
-for (var dy = -1; dy <= 1; dy++) {
-for (var dx = -1; dx <= 1; dx++) {
-let q = clamp(vec2i(id.xy) + vec2i(dx, dy), vec2i(0), vec2i(size) - 1);
-let value = rawGuide[u32(q.x) + u32(q.y) * size.x];
-mY += value.x;
-iX += value.x * value.x;
-if (value.y >= 0.0) {
-lu += value.y;
-jc += 1.0;
-}
-}
-}
-var variance: f32;
-if (center.y >= 0.0) {
-variance = lu / jc;
-} else {
-let lt = mY / 9.0;
-variance = max(iX / 9.0 - lt * lt, 0.0) * 9.0 / 8.0;
-}
-guide[id.x + id.y * size.x] = vec2f(center.x, variance);
-}
-`;
-/** Copies the composed image into the filter's color buffer. */
-export const jnlmLoadColorWGSL = /* wgsl */ `
-${paramsWGSL}
-@group(0) @binding(1) var colorTex: texture_2d<f32>;
-@group(0) @binding(2) var<storage, read_write> outColor: array<vec4f>;
-
-@compute @workgroup_size(${JNLM_WORKGROUP_SIZE}, ${JNLM_WORKGROUP_SIZE}, 1)
-fn main(@builtin(global_invocation_id) id: vec3u) {
-if (id.x >= params.size.x || id.y >= params.size.y) { return; }
-outColor[id.x + id.y * params.size.x] = textureLoad(colorTex, vec2i(id.xy), 0);
-}
-`;
-/**
- * Pass 3: the joint NL-means filter of the composed image. It also filters
- * the guide with the same weights, with the variance of the weighted mean,
- * for a following pass. The albedo regression corrects color only, keeping
- * uncertain fitted texture responses out of the matching guide.
- */
-export function createJNLMFilterWGSL({ searchRadius, patchRadius, albedoBandwidth, normalBandwidth } = JNLM_DEFAULT_SHAPE) {
-    const patchSize = (2 * patchRadius + 1) ** 2;
-    return /* wgsl */ `
-${paramsWGSL}
-@group(0) @binding(1) var<storage, read> guide: array<vec2f>;
-@group(0) @binding(2) var<storage, read> color: array<vec4f>;
-@group(0) @binding(3) var albedoTex: texture_2d<f32>;
-@group(0) @binding(4) var normalTex: texture_2d<f32>;
-@group(0) @binding(5) var<storage, read_write> outColor: array<vec4f>;
-@group(0) @binding(6) var<storage, read_write> outGuide: array<vec2f>;
-@group(0) @binding(7) var shadowCatcherTex: texture_2d<f32>;
-
-const SEARCH_RADIUS = ${searchRadius};
-const PATCH_RADIUS = ${patchRadius};
-const PATCH_SIZE = ${patchSize};
-
-const VARIANCE_CANCELLATION = 1.0;
-
-const INV_ALBEDO_BANDWIDTH2 = ${(1 / albedoBandwidth ** 2).toFixed(6)};
-const INV_COVERAGE_BANDWIDTH2 = ${(1 / JNLM_COVERAGE_BANDWIDTH ** 2).toFixed(6)};
-const INV_NORMAL_BANDWIDTH2 = ${(1 / normalBandwidth ** 2).toFixed(6)};
-
-
-const CENTER_WEIGHT_MIN = 0.01;
-
-fn hg(p: vec2i) -> u32 {
-let q = clamp(p, vec2i(0), vec2i(params.size) - 1);
-return u32(q.x) + u32(q.y) * params.size.x;
-}
-
-fn lw(q: vec2i, bs: vec4f, gT: vec3f) -> f32 {
-let albedo = textureLoad(albedoTex, q, 0);
-
-
-let pf = max((albedo.rgb + bs.rgb) * 0.5, vec3f(${JNLM_ALBEDO_FLOOR}));
-let ld = (albedo.rgb - bs.rgb) / pf;
-let iU = albedo.a - bs.a;
-let lp = textureLoad(normalTex, q, 0).rgb - gT;
-return max(
-dot(ld, ld) * INV_ALBEDO_BANDWIDTH2
-+ iU * iU * INV_COVERAGE_BANDWIDTH2,
-dot(lp, lp) * INV_NORMAL_BANDWIDTH2
-);
-}
-
-fn premultiplied(c: vec4f) -> vec4f {
-return vec4f(c.rgb * c.a, c.a);
-}
-
-@compute @workgroup_size(${JNLM_WORKGROUP_SIZE}, ${JNLM_WORKGROUP_SIZE}, 1)
-fn main(@builtin(global_invocation_id) id: vec3u) {
-let size = vec2i(params.size);
-let p = vec2i(id.xy);
-if (p.x >= size.x || p.y >= size.y) { return; }
-let step = i32(params.dilation);
-let bs = textureLoad(albedoTex, p, 0);
-let fD = textureLoad(normalTex, p, 0);
-let gT = fD.rgb;
-let index = hg(p);
-
-
-if (fD.a <= 0.0 && textureLoad(shadowCatcherTex, p, 0).r <= 0.0) {
-outColor[index] = color[index];
-outGuide[index] = guide[index];
-return;
-}
-
-var lf: array<vec2f, PATCH_SIZE>;
-var k = 0;
-for (var dy = -PATCH_RADIUS; dy <= PATCH_RADIUS; dy++) {
-for (var dx = -PATCH_RADIUS; dx <= PATCH_RADIUS; dx++) {
-lf[k] = guide[hg(p + vec2i(dx, dy) * step)];
-k++;
-}
-}
-let k2 = params.colorSensitivity * params.colorSensitivity;
-
-var sum = vec4f(0.0);
-let gW = max(bs.rgb, vec3f(${JNLM_ALBEDO_FLOOR}));
-var mp = vec3f(0.0);
-var gy = vec3f(0.0);
-var hs = vec3f(0.0);
-var ps = 0.0;
-var gA = 0.0;
-var nU = 0.0;
-var kM = 0.0;
-for (var sy = -SEARCH_RADIUS; sy <= SEARCH_RADIUS; sy++) {
-for (var sx = -SEARCH_RADIUS; sx <= SEARCH_RADIUS; sx++) {
-let q = p + vec2i(sx, sy) * step;
-if (
-(sx == 0 && sy == 0) ||
-q.x < 0 || q.y < 0 || q.x >= size.x || q.y >= size.y
-) { continue; }
-
-let features = lw(q, bs, gT);
-if (features > 16.0) { continue; }
-
-var fY = 0.0;
-k = 0;
-for (var dy = -PATCH_RADIUS; dy <= PATCH_RADIUS; dy++) {
-for (var dx = -PATCH_RADIUS; dx <= PATCH_RADIUS; dx++) {
-let a = lf[k];
-let b = guide[hg(q + vec2i(dx, dy) * step)];
-let qU = a.x - b.x;
-fY += (qU * qU
-- VARIANCE_CANCELLATION * (a.y + min(a.y, b.y)))
-/ (1e-10 + k2 * (a.y + b.y));
-k++;
-}
-}
-fY /= f32(PATCH_SIZE);
-
-let weight = exp(-max(max(fY, 0.0), features));
-let index = hg(q);
-let c = color[index];
-let g = guide[index];
-
-let a = (textureLoad(albedoTex, q, 0).rgb - bs.rgb) / gW;
-let value = premultiplied(c);
-sum += value * weight;
-mp += a * weight;
-gy += a * a * weight;
-hs += a * value.rgb * weight;
-ps += g.x * weight;
-gA += g.y * weight * weight;
-nU += weight;
-kM = max(kM, weight);
-}
-}
-
-
-
-let center = color[index];
-let le = guide[index];
-let dG = max(kM, CENTER_WEIGHT_MIN);
-let totalWeight = nU + dG;
-let tM = premultiplied(center);
-let mean = (sum + tM * dG) / totalWeight;
-
-
-let ez = mp / totalWeight;
-let lA = max(gy / totalWeight - ez * ez, vec3f(0.0));
-let qu = hs / totalWeight - ez * mean.rgb;
-
-
-
-let qE = bs.rgb + ez * gW;
-let oz = mean.rgb * gW / max(qE, vec3f(${JNLM_ALBEDO_FLOOR}));
-let vB = clamp(qu / (lA + vec3f(${JNLM_ALBEDO_REGULARIZATION})), vec3f(0.0), oz);
-let rD = mean.rgb - vB * ez;
-let filtered = vec4f(max(rD, vec3f(0.0)), max(mean.a, 0.0));
-let rgb = select(
-center.rgb,
-filtered.rgb / max(filtered.a, 1e-6),
-filtered.a > 1e-6
-);
-outColor[index] = vec4f(rgb, filtered.a);
-outGuide[index] = vec2f(
-(ps + le.x * dG) / totalWeight,
-(gA + le.y * dG * dG)
-/ (totalWeight * totalWeight)
-);
-}
-`;
-}
+import { JNLM_MAX_ITERATIONS } from './jnlm.js';
+import { JNLM_WORKGROUP_SIZE, jnlmGuideWGSL, jnlmVarianceWGSL, jnlmLoadColorWGSL, createJNLMFilterWGSL, packJNLMParameters } from './wgsl/jnlm.compute.js';
+export { createJNLMFilterWGSL, jnlmGuideWGSL, jnlmVarianceWGSL, jnlmLoadColorWGSL, JNLM_WORKGROUP_SIZE } from './wgsl/jnlm.compute.js';
 const MAX_ITERATIONS = JNLM_MAX_ITERATIONS;
 export class WGPUJNLMDenoisePass {
     _device;
@@ -337,12 +77,9 @@ export class WGPUJNLMDenoisePass {
         const { width, height, shape } = inputs;
         this._ensureBuffers(width, height);
         const iterations = Math.min(Math.max(Math.round(shape.iterations), 1), MAX_ITERATIONS);
-        const sensitivity = JNLM_DEFAULT_COLOR_SENSITIVITY * Math.max(inputs.strength, 1e-3);
         this._paramsBuffers.forEach((buffer, i) => {
-            const params = new ArrayBuffer(16);
-            new Uint32Array(params, 0, 2).set([width, height]);
             // Buffer i (from 1) is filter pass i - 1; index 0 is the prepasses.
-            new Float32Array(params, 8, 2).set([sensitivity, jnlmPassDilation(Math.max(i - 1, 0))]);
+            const params = packJNLMParameters(width, height, inputs.strength, Math.max(i - 1, 0));
             this._device.queue.writeBuffer(buffer, 0, params);
         });
         const device = this._device;

package/lib/renderer/WGPUJNLMDenoiser.js.map

@@ -1 +1 @@
-{"version":3,"file":"WGPUJNLMDenoiser.js","sourceRoot":"","sources":["../../src/renderer/WGPUJNLMDenoiser.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,oBAAoB,EAAE,MAAM,sBAAsB,CAAC;AAC5D,OAAO,6BAA6B,MAAM,iDAAiD,CAAC;AAE5F,OAAO,EACL,8BAA8B,EAC9B,kBAAkB,EAClB,iBAAiB,EACjB,0BAA0B,EAC1B,uBAAuB,EACvB,mBAAmB,EACnB,gBAAgB,EAEjB,MAAM,WAAW,CAAC;AAEnB,iEAAiE;AAEjE,MAAM,CAAC,MAAM,mBAAmB,GAAG,CAAC,CAAC;AAErC,MAAM,UAAU,GAAG,UAAU,CAAC;;;;;;;;CAQ7B,CAAC;AAEF;;;GAGG;AACH,MAAM,CAAC,MAAM,aAAa,GAAG,UAAU,CAAC;EACtC,UAAU;EACV,6BAA6B;;;;;2BAKJ,mBAAmB,KAAK,mBAAmB;;;;;;;;;;;;;;;CAerE,CAAC;AAEF;;;GAGG;AACH,MAAM,CAAC,MAAM,gBAAgB,GAAG,UAAU,CAAC;EACzC,UAAU;;;;2BAIe,mBAAmB,KAAK,mBAAmB;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;CA8BrE,CAAC;AAEF,gEAAgE;AAChE,MAAM,CAAC,MAAM,iBAAiB,GAAG,UAAU,CAAC;EAC1C,UAAU;;;;2BAIe,mBAAmB,KAAK,mBAAmB;;;;;CAKrE,CAAC;AAEF;;;;;GAKG;AACH,MAAM,UAAU,oBAAoB,CAAC,EACnC,YAAY,EAAE,WAAW,EAAE,eAAe,EAAE,eAAe,KACxC,kBAAkB;IACrC,MAAM,SAAS,GAAG,CAAC,CAAC,GAAG,WAAW,GAAG,CAAC,CAAC,IAAI,CAAC,CAAC;IAC7C,OAAO,UAAU,CAAC;EAClB,UAAU;;;;;;;;;wBASY,YAAY;uBACb,WAAW;qBACb,SAAS;;;;gCAIE,CAAC,CAAC,GAAG,eAAe,IAAI,CAAC,CAAC,CAAC,OAAO,CAAC,CAAC,CAAC;kCACnC,CAAC,CAAC,GAAG,uBAAuB,IAAI,CAAC,CAAC,CAAC,OAAO,CAAC,CAAC,CAAC;gCAC/C,CAAC,CAAC,GAAG,eAAe,IAAI,CAAC,CAAC,CAAC,OAAO,CAAC,CAAC,CAAC;;;;;;;;;;;;;;kDAcE,iBAAiB;;;;;;;;;;;;;;;2BAe7D,mBAAmB,KAAK,mBAAmB;;;;;;;;;;;;;;;;;;;;;;;;;;;;;6BA6BnB,iBAAiB;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;yCAqEG,iBAAiB;kCAC5B,0BAA0B;;;;;;;;;;;;;;;CAerF,CAAC;AACF,CAAC;AAmBD,MAAM,cAAc,GAAG,mBAAmB,CAAC;AAE3C,MAAM,OAAO,mBAAmB;IACtB,OAAO,CAAY;IACnB,cAAc,GAAgB,EAAE,CAAC;IACjC,cAAc,CAAqB;IACnC,iBAAiB,CAAqB;IACtC,kBAAkB,CAAqB;IACvC,eAAe,CAAsB;IACrC,eAAe,GAAG,EAAE,CAAC;IACrB,OAAO,GAAgB,EAAE,CAAC;IAC1B,OAAO,GAAgB,EAAE,CAAC;IAC1B,MAAM,GAAG,CAAC,CAAC;IACX,OAAO,GAAG,CAAC,CAAC;IAEpB,YAAY,MAAiB;QAC3B,IAAI,CAAC,OAAO,GAAG,MAAM,CAAC;QACtB,sEAAsE;QACtE,iCAAiC;QACjC,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,IAAI,cAAc,EAAE,CAAC,EAAE,EAAE,CAAC;YACzC,IAAI,CAAC,cAAc,CAAC,IAAI,CAAC,MAAM,CAAC,YAAY,CAAC;gBAC3C,KAAK,EAAE,aAAa;gBACpB,IAAI,EAAE,EAAE;gBACR,KAAK,EAAE,cAAc,CAAC,OAAO,GAAG,cAAc,CAAC,QAAQ;aACxD,CAAC,CAAC,CAAC;QACN,CAAC;QACD,IAAI,CAAC,cAAc,GAAG,IAAI,CAAC,eAAe,CAAC,YAAY,EAAE,aAAa,CAAC,CAAC;QACxE,IAAI,CAAC,iBAAiB,GAAG,IAAI,CAAC,eAAe,CAAC,eAAe,EAAE,gBAAgB,CAAC,CAAC;QACjF,IAAI,CAAC,kBAAkB,GAAG,IAAI,CAAC,eAAe,CAAC,iBAAiB,EAAE,iBAAiB,CAAC,CAAC;IACvF,CAAC;IAEO,eAAe,CAAC,KAAa,EAAE,IAAY;QACjD,OAAO,oBAAoB,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC,qBAAqB,CAAC;YAC9D,KAAK;YACL,MAAM,EAAE,MAAM;YACd,OAAO,EAAE;gBACP,MAAM,EAAE,oBAAoB,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC,kBAAkB,CAAC,EAAE,KAAK,EAAE,IAAI,EAAE,CAAC;gBAC9E,UAAU,EAAE,MAAM;aACnB;SACF,CAAC,CAAC;IACL,CAAC;IAEO,kBAAkB,CAAC,KAAsB;QAC/C,MAAM,GAAG,GAAG,IAAI,CAAC,SAAS,CAAC,KAAK,CAAC,CAAC;QAClC,IAAI,CAAC,IAAI,CAAC,eAAe,IAAI,GAAG,KAAK,IAAI,CAAC,eAAe,EAAE,CAAC;YAC1D,IAAI,CAAC,eAAe,GAAG,IAAI,CAAC,eAAe,CAAC,aAAa,EAAE,oBAAoB,CAAC,KAAK,CAAC,CAAC,CAAC;YACxF,IAAI,CAAC,eAAe,GAAG,GAAG,CAAC;QAC7B,CAAC;QACD,OAAO,IAAI,CAAC,eAAe,CAAC;IAC9B,CAAC;IAED,eAAe,KAAI,CAAC;IACZ,cAAc,CAAC,KAAa,EAAE,MAAc;QAClD,IAAI,KAAK,IAAI,IAAI,CAAC,MAAM,IAAI,MAAM,IAAI,IAAI,CAAC,OAAO,IAAI,IAAI,CAAC,OAAO,CAAC,MAAM;YAAE,OAAO;QAClF,KAAK,GAAG,IAAI,CAAC,GAAG,CAAC,KAAK,EAAE,IAAI,CAAC,MAAM,CAAC,CAAC;QAAC,MAAM,GAAG,IAAI,CAAC,GAAG,CAAC,MAAM,EAAE,IAAI,CAAC,OAAO,CAAC,CAAC;QAC9E,IAAI,CAAC,eAAe,EAAE,CAAC;QACvB,MAAM,MAAM,GAAG,KAAK,GAAG,MAAM,CAAC;QAC9B,MAAM,YAAY,GAAG,CAAC,KAAa,EAAE,KAAa,EAAE,EAAE,CACpD,IAAI,CAAC,OAAO,CAAC,YAAY,CAAC;YACxB,KAAK;YACL,IAAI,EAAE,MAAM,GAAG,KAAK;YACpB,KAAK,EAAE,cAAc,CAAC,OAAO,GAAG,cAAc,CAAC,QAAQ;SACxD,CAAC,CAAC;QACL,IAAI,CAAC,OAAO,GAAG,CAAC,YAAY,CAAC,cAAc,EAAE,CAAC,CAAC,EAAE,YAAY,CAAC,cAAc,EAAE,CAAC,CAAC,CAAC,CAAC;QAClF,IAAI,CAAC,OAAO,GAAG,CAAC,YAAY,CAAC,cAAc,EAAE,EAAE,CAAC,EAAE,YAAY,CAAC,cAAc,EAAE,EAAE,CAAC,CAAC,CAAC;QACpF,IAAI,CAAC,MAAM,GAAG,KAAK,CAAC;QACpB,IAAI,CAAC,OAAO,GAAG,MAAM,CAAC;IACxB,CAAC;IAEO,eAAe;QACrB,KAAK,MAAM,MAAM,IAAI,CAAC,GAAG,IAAI,CAAC,OAAO,EAAE,GAAG,IAAI,CAAC,OAAO,CAAC;YAAE,MAAM,CAAC,OAAO,EAAE,CAAC;QAC1E,IAAI,CAAC,OAAO,GAAG,EAAE,CAAC;QAClB,IAAI,CAAC,OAAO,GAAG,EAAE,CAAC;IACpB,CAAC;IAED,2EAA2E;IAC3E,MAAM,CAAC,OAA0B,EAAE,MAAkB;QACnD,MAAM,EAAE,KAAK,EAAE,MAAM,EAAE,KAAK,EAAE,GAAG,MAAM,CAAC;QACxC,IAAI,CAAC,cAAc,CAAC,KAAK,EAAE,MAAM,CAAC,CAAC;QACnC,MAAM,UAAU,GAAG,IAAI,CAAC,GAAG,CAAC,IAAI,CAAC,GAAG,CAAC,IAAI,CAAC,KAAK,CAAC,KAAK,CAAC,UAAU,CAAC,EAAE,CAAC,CAAC,EAAE,cAAc,CAAC,CAAC;QACvF,MAAM,WAAW,GAAG,8BAA8B,GAAG,IAAI,CAAC,GAAG,CAAC,MAAM,CAAC,QAAQ,EAAE,IAAI,CAAC,CAAC;QACrF,IAAI,CAAC,cAAc,CAAC,OAAO,CAAC,CAAC,MAAM,EAAE,CAAC,EAAE,EAAE;YACxC,MAAM,MAAM,GAAG,IAAI,WAAW,CAAC,EAAE,CAAC,CAAC;YACnC,IAAI,WAAW,CAAC,MAAM,EAAE,CAAC,EAAE,CAAC,CAAC,CAAC,GAAG,CAAC,CAAC,KAAK,EAAE,MAAM,CAAC,CAAC,CAAC;YACnD,oEAAoE;YACpE,IAAI,YAAY,CAAC,MAAM,EAAE,CAAC,EAAE,CAAC,CAAC,CAAC,GAAG,CAAC,CAAC,WAAW,EAAE,gBAAgB,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC,GAAG,CAAC,EAAE,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC;YACxF,IAAI,CAAC,OAAO,CAAC,KAAK,CAAC,WAAW,CAAC,MAAM,EAAE,CAAC,EAAE,MAAM,CAAC,CAAC;QACpD,CAAC,CAAC,CAAC;QAEH,MAAM,MAAM,GAAG,IAAI,CAAC,OAAO,CAAC;QAC5B,MAAM,cAAc,GAAG,IAAI,CAAC,kBAAkB,CAAC,KAAK,CAAC,CAAC;QACtD,MAAM,IAAI,GAAG,CACX,QAA4B,EAC5B,YAAuB,EACvB,SAA+B,EAC/B,EAAE,CACF,MAAM,CAAC,eAAe,CAAC;YACrB,MAAM,EAAE,QAAQ,CAAC,kBAAkB,CAAC,CAAC,CAAC;YACtC,OAAO,EAAE;gBACP,EAAE,OAAO,EAAE,CAAC,EAAE,QAAQ,EAAE,EAAE,MAAM,EAAE,YAAY,EAAE,EAAE;gBAClD,GAAG,SAAS,CAAC,GAAG,CAAC,CAAC,QAAQ,EAAE,CAAC,EAAE,EAAE,CAAC,CAAC,EAAE,OAAO,EAAE,CAAC,GAAG,CAAC,EAAE,QAAQ,EAAE,CAAC,CAAC;aAClE;SACF,CAAC,CAAC;QACL,MAAM,CAAC,MAAM,EAAE,MAAM,CAAC,GAAG,IAAI,CAAC,OAAO,CAAC;QACtC,MAAM,CAAC,MAAM,EAAE,MAAM,CAAC,GAAG,IAAI,CAAC,OAAO,CAAC;QACtC,MAAM,OAAO,GAAG,IAAI,CAAC,cAAc,CAAC,CAAC,CAAC,CAAC;QACvC,MAAM,MAAM,GAAyC;YACnD,CAAC,IAAI,CAAC,cAAc,EAAE,IAAI,CAAC,IAAI,CAAC,cAAc,EAAE,OAAO,EAAE;oBACvD,EAAE,MAAM,EAAE,MAAM,CAAC,YAAY,EAAE;oBAC/B,EAAE,MAAM,EAAE,MAAM,CAAC,kBAAkB,EAAE;oBACrC,EAAE,MAAM,EAAE,MAAM,EAAE;iBACnB,CAAC,CAAC;YACH,CAAC,IAAI,CAAC,iBAAiB,EAAE,IAAI,CAAC,IAAI,CAAC,iBAAiB,EAAE,OAAO,EAAE;oBAC7D,EAAE,MAAM,EAAE,MAAM,EAAE;oBAClB,EAAE,MAAM,EAAE,MAAM,EAAE;iBACnB,CAAC,CAAC;YACH,CAAC,IAAI,CAAC,kBAAkB,EAAE,IAAI,CAAC,IAAI,CAAC,kBAAkB,EAAE,OAAO,EAAE;oBAC/D,MAAM,CAAC,KAAK,CAAC,UAAU,EAAE;oBACzB,EAAE,MAAM,EAAE,MAAM,EAAE;iBACnB,CAAC,CAAC;SACJ,CAAC;QACF,MAAM,UAAU,GAAG,MAAM,CAAC,MAAM,CAAC,UAAU,EAAE,CAAC;QAC9C,MAAM,UAAU,GAAG,MAAM,CAAC,MAAM,CAAC,UAAU,EAAE,CAAC;QAC9C,MAAM,iBAAiB,GAAG,MAAM,CAAC,aAAa,CAAC,UAAU,EAAE,CAAC;QAC5D,IAAI,MAAM,GAAG,CAAC,MAAM,EAAE,MAAM,CAAC,CAAC;QAC9B,IAAI,MAAM,GAAG,CAAC,MAAM,EAAE,MAAM,CAAC,CAAC;QAC9B,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,IAAI,UAAU,EAAE,CAAC,EAAE,EAAE,CAAC;YACrC,MAAM,CAAC,IAAI,CAAC,CAAC,cAAc,EAAE,IAAI,CAAC,cAAc,EAAE,IAAI,CAAC,cAAc,CAAC,CAAC,CAAC,EAAE;oBACxE,EAAE,MAAM,EAAE,MAAM,CAAC,CAAC,CAAC,EAAE;oBACrB,EAAE,MAAM,EAAE,MAAM,CAAC,CAAC,CAAC,EAAE;oBACrB,UAAU;oBACV,UAAU;oBACV,EAAE,MAAM,EAAE,MAAM,CAAC,CAAC,CAAC,EAAE;oBACrB,EAAE,MAAM,EAAE,MAAM,CAAC,CAAC,CAAC,EAAE;oBACrB,iBAAiB;iBAClB,CAAC,CAAC,CAAC,CAAC;YACL,MAAM,GAAG,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,MAAM,CAAC,CAAC,CAAC,CAAC,CAAC;YAChC,MAAM,GAAG,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,MAAM,CAAC,CAAC,CAAC,CAAC,CAAC;QAClC,CAAC;QACD,MAAM,IAAI,GAAG,OAAO,CAAC,gBAAgB,CAAC,EAAE,KAAK,EAAE,cAAc,EAAE,CAAC,CAAC;QACjE,KAAK,MAAM,CAAC,QAAQ,EAAE,SAAS,CAAC,IAAI,MAAM,EAAE,CAAC;YAC3C,IAAI,CAAC,WAAW,CAAC,QAAQ,CAAC,CAAC;YAC3B,IAAI,CAAC,YAAY,CAAC,CAAC,EAAE,SAAS,CAAC,CAAC;YAChC,IAAI,CAAC,kBAAkB,CACrB,IAAI,CAAC,IAAI,CAAC,KAAK,GAAG,mBAAmB,CAAC,EACtC,IAAI,CAAC,IAAI,CAAC,MAAM,GAAG,mBAAmB,CAAC,CACxC,CAAC;QACJ,CAAC;QACD,IAAI,CAAC,GAAG,EAAE,CAAC;QACX,OAAO,MAAM,CAAC,CAAC,CAAC,CAAC;IACnB,CAAC;IAED,cAAc;QACZ,OAAO,CAAC,GAAG,IAAI,CAAC,cAAc,EAAE,GAAG,IAAI,CAAC,OAAO,EAAE,GAAG,IAAI,CAAC,OAAO,CAAC;aAC9D,MAAM,CAAC,CAAC,GAAG,EAAE,MAAM,EAAE,EAAE,CAAC,GAAG,GAAG,MAAM,CAAC,IAAI,EAAE,CAAC,CAAC,CAAC;IACnD,CAAC;IAED,OAAO;QACL,KAAK,MAAM,MAAM,IAAI,IAAI,CAAC,cAAc;YAAE,MAAM,CAAC,OAAO,EAAE,CAAC;QAC3D,IAAI,CAAC,eAAe,EAAE,CAAC;IACzB,CAAC;CACF","sourcesContent":["import { getTargetShaderCache } from './WGPUShaderCache.js';\nimport shadowCatcherAccumulationWGSL from './wgsl/chunks/shadowCatcherAccumulation.wgsl.js';\n\nimport {\n  JNLM_DEFAULT_COLOR_SENSITIVITY,\n  JNLM_DEFAULT_SHAPE,\n  JNLM_ALBEDO_FLOOR,\n  JNLM_ALBEDO_REGULARIZATION,\n  JNLM_COVERAGE_BANDWIDTH,\n  JNLM_MAX_ITERATIONS,\n  jnlmPassDilation,\n  type JNLMFilterShape\n} from './jnlm.js';\n\n/** WebGPU passes of the JNLM denoiser described in `jnlm.ts`. */\n\nexport const JNLM_WORKGROUP_SIZE = 8;\n\nconst paramsWGSL = /* wgsl */ `\nstruct JNLMParams {\n  size: vec2u,\n  colorSensitivity: f32,\n  // Pixel spacing of the search window and patches in this pass.\n  dilation: f32,\n}\n@group(0) @binding(0) var<uniform> params: JNLMParams;\n`;\n\n/**\n * Pass 1: the guide mean and the variance of that mean from the accumulated\n * moments. Ray buffers are stored bottom-up; guides use texture rows.\n */\nexport const jnlmGuideWGSL = /* wgsl */ `\n${paramsWGSL}\n${shadowCatcherAccumulationWGSL}\n@group(0) @binding(1) var<storage, read> accumulation: array<ShadowCatcherAccumulation>;\n@group(0) @binding(2) var<storage, read> normalBuffer: array<vec4f>;\n@group(0) @binding(3) var<storage, read_write> rawGuide: array<vec2f>;\n\n@compute @workgroup_size(${JNLM_WORKGROUP_SIZE}, ${JNLM_WORKGROUP_SIZE}, 1)\nfn main(@builtin(global_invocation_id) id: vec3u) {\n  let size = params.size;\n  if (id.x >= size.x || id.y >= size.y) { return; }\n  let rayIndex = id.x + (size.y - 1u - id.y) * size.x;\n  let n = normalBuffer[rayIndex].a;\n  let moments = accumulation[rayIndex].moments;\n  let mean = moments.x / max(n, 1.0);\n  // Variance of the mean, or -1 when one sample cannot estimate it.\n  var variance = -1.0;\n  if (n >= 2.0) {\n    variance = max(moments.y - n * mean * mean, 0.0) / (n * (n - 1.0));\n  }\n  rawGuide[id.x + id.y * size.x] = vec2f(mean, variance);\n}\n`;\n\n/**\n * Pass 2: smooth the noisy per-pixel variance over 3 x 3 pixels. Pixels with\n * a single sample take the spatial variance of their neighborhood instead.\n */\nexport const jnlmVarianceWGSL = /* wgsl */ `\n${paramsWGSL}\n@group(0) @binding(1) var<storage, read> rawGuide: array<vec2f>;\n@group(0) @binding(2) var<storage, read_write> guide: array<vec2f>;\n\n@compute @workgroup_size(${JNLM_WORKGROUP_SIZE}, ${JNLM_WORKGROUP_SIZE}, 1)\nfn main(@builtin(global_invocation_id) id: vec3u) {\n  let size = params.size;\n  if (id.x >= size.x || id.y >= size.y) { return; }\n  let center = rawGuide[id.x + id.y * size.x];\n  var varianceSum = 0.0;\n  var varianceCount = 0.0;\n  var meanSum = 0.0;\n  var meanSquareSum = 0.0;\n  for (var dy = -1; dy <= 1; dy++) {\n    for (var dx = -1; dx <= 1; dx++) {\n      let q = clamp(vec2i(id.xy) + vec2i(dx, dy), vec2i(0), vec2i(size) - 1);\n      let value = rawGuide[u32(q.x) + u32(q.y) * size.x];\n      meanSum += value.x;\n      meanSquareSum += value.x * value.x;\n      if (value.y >= 0.0) {\n        varianceSum += value.y;\n        varianceCount += 1.0;\n      }\n    }\n  }\n  var variance: f32;\n  if (center.y >= 0.0) {\n    variance = varianceSum / varianceCount;\n  } else {\n    let spatialMean = meanSum / 9.0;\n    variance = max(meanSquareSum / 9.0 - spatialMean * spatialMean, 0.0) * 9.0 / 8.0;\n  }\n  guide[id.x + id.y * size.x] = vec2f(center.x, variance);\n}\n`;\n\n/** Copies the composed image into the filter's color buffer. */\nexport const jnlmLoadColorWGSL = /* wgsl */ `\n${paramsWGSL}\n@group(0) @binding(1) var colorTex: texture_2d<f32>;\n@group(0) @binding(2) var<storage, read_write> outColor: array<vec4f>;\n\n@compute @workgroup_size(${JNLM_WORKGROUP_SIZE}, ${JNLM_WORKGROUP_SIZE}, 1)\nfn main(@builtin(global_invocation_id) id: vec3u) {\n  if (id.x >= params.size.x || id.y >= params.size.y) { return; }\n  outColor[id.x + id.y * params.size.x] = textureLoad(colorTex, vec2i(id.xy), 0);\n}\n`;\n\n/**\n * Pass 3: the joint NL-means filter of the composed image. It also filters\n * the guide with the same weights, with the variance of the weighted mean,\n * for a following pass. The albedo regression corrects color only, keeping\n * uncertain fitted texture responses out of the matching guide.\n */\nexport function createJNLMFilterWGSL({\n  searchRadius, patchRadius, albedoBandwidth, normalBandwidth\n}: JNLMFilterShape = JNLM_DEFAULT_SHAPE) {\n  const patchSize = (2 * patchRadius + 1) ** 2;\n  return /* wgsl */ `\n${paramsWGSL}\n@group(0) @binding(1) var<storage, read> guide: array<vec2f>;\n@group(0) @binding(2) var<storage, read> color: array<vec4f>;\n@group(0) @binding(3) var albedoTex: texture_2d<f32>;\n@group(0) @binding(4) var normalTex: texture_2d<f32>;\n@group(0) @binding(5) var<storage, read_write> outColor: array<vec4f>;\n@group(0) @binding(6) var<storage, read_write> outGuide: array<vec2f>;\n@group(0) @binding(7) var shadowCatcherTex: texture_2d<f32>;\n\nconst SEARCH_RADIUS = ${searchRadius};\nconst PATCH_RADIUS = ${patchRadius};\nconst PATCH_SIZE = ${patchSize};\n// Variance cancellation strength alpha of the paper.\nconst VARIANCE_CANCELLATION = 1.0;\n// Inverse squared feature bandwidths. Normals are encoded as n * 0.5 + 0.5.\nconst INV_ALBEDO_BANDWIDTH2 = ${(1 / albedoBandwidth ** 2).toFixed(6)};\nconst INV_COVERAGE_BANDWIDTH2 = ${(1 / JNLM_COVERAGE_BANDWIDTH ** 2).toFixed(6)};\nconst INV_NORMAL_BANDWIDTH2 = ${(1 / normalBandwidth ** 2).toFixed(6)};\n// Lower bound of the center weight, so a pixel unlike all of its neighbors\n// (a thin highlight) is kept rather than replaced by them.\nconst CENTER_WEIGHT_MIN = 0.01;\n\nfn pixelIndex(p: vec2i) -> u32 {\n  let q = clamp(p, vec2i(0), vec2i(params.size) - 1);\n  return u32(q.x) + u32(q.y) * params.size.x;\n}\n\nfn featureDistance(q: vec2i, centerAlbedo: vec4f, centerNormal: vec3f) -> f32 {\n  let albedo = textureLoad(albedoTex, q, 0);\n  // Absolute differences in linear albedo erase low-contrast dark textures.\n  // Measure RGB contrast relative to both pixels; keep coverage absolute.\n  let albedoScale = max((albedo.rgb + centerAlbedo.rgb) * 0.5, vec3f(${JNLM_ALBEDO_FLOOR}));\n  let albedoDelta = (albedo.rgb - centerAlbedo.rgb) / albedoScale;\n  let coverageDelta = albedo.a - centerAlbedo.a;\n  let normalDelta = textureLoad(normalTex, q, 0).rgb - centerNormal;\n  return max(\n    dot(albedoDelta, albedoDelta) * INV_ALBEDO_BANDWIDTH2\n      + coverageDelta * coverageDelta * INV_COVERAGE_BANDWIDTH2,\n    dot(normalDelta, normalDelta) * INV_NORMAL_BANDWIDTH2\n  );\n}\n\nfn premultiplied(c: vec4f) -> vec4f {\n  return vec4f(c.rgb * c.a, c.a);\n}\n\n@compute @workgroup_size(${JNLM_WORKGROUP_SIZE}, ${JNLM_WORKGROUP_SIZE}, 1)\nfn main(@builtin(global_invocation_id) id: vec3u) {\n  let size = vec2i(params.size);\n  let p = vec2i(id.xy);\n  if (p.x >= size.x || p.y >= size.y) { return; }\n  let step = i32(params.dilation);\n  let centerAlbedo = textureLoad(albedoTex, p, 0);\n  let centerNormalTexel = textureLoad(normalTex, p, 0);\n  let centerNormal = centerNormalTexel.rgb;\n  let index = pixelIndex(p);\n  // Primary misses show the background directly. It has no features that\n  // protect its detail and no noise beyond antialiasing, so keep it.\n  if (centerNormalTexel.a <= 0.0 && textureLoad(shadowCatcherTex, p, 0).r <= 0.0) {\n    outColor[index] = color[index];\n    outGuide[index] = guide[index];\n    return;\n  }\n\n  var centerPatch: array<vec2f, PATCH_SIZE>;\n  var k = 0;\n  for (var dy = -PATCH_RADIUS; dy <= PATCH_RADIUS; dy++) {\n    for (var dx = -PATCH_RADIUS; dx <= PATCH_RADIUS; dx++) {\n      centerPatch[k] = guide[pixelIndex(p + vec2i(dx, dy) * step)];\n      k++;\n    }\n  }\n  let k2 = params.colorSensitivity * params.colorSensitivity;\n\n  var sum = vec4f(0.0);\n  let featureScale = max(centerAlbedo.rgb, vec3f(${JNLM_ALBEDO_FLOOR}));\n  var featureSum = vec3f(0.0);\n  var featureSquareSum = vec3f(0.0);\n  var featureColorSum = vec3f(0.0);\n  var guideSum = 0.0;\n  var guideVarianceSum = 0.0;\n  var weightSum = 0.0;\n  var maxWeight = 0.0;\n  for (var sy = -SEARCH_RADIUS; sy <= SEARCH_RADIUS; sy++) {\n    for (var sx = -SEARCH_RADIUS; sx <= SEARCH_RADIUS; sx++) {\n      let q = p + vec2i(sx, sy) * step;\n      if (\n        (sx == 0 && sy == 0) ||\n        q.x < 0 || q.y < 0 || q.x >= size.x || q.y >= size.y\n      ) { continue; }\n\n      let features = featureDistance(q, centerAlbedo, centerNormal);\n      if (features > 16.0) { continue; }\n\n      var colorDistance = 0.0;\n      k = 0;\n      for (var dy = -PATCH_RADIUS; dy <= PATCH_RADIUS; dy++) {\n        for (var dx = -PATCH_RADIUS; dx <= PATCH_RADIUS; dx++) {\n          let a = centerPatch[k];\n          let b = guide[pixelIndex(q + vec2i(dx, dy) * step)];\n          let delta = a.x - b.x;\n          colorDistance += (delta * delta\n              - VARIANCE_CANCELLATION * (a.y + min(a.y, b.y)))\n            / (1e-10 + k2 * (a.y + b.y));\n          k++;\n        }\n      }\n      colorDistance /= f32(PATCH_SIZE);\n\n      let weight = exp(-max(max(colorDistance, 0.0), features));\n      let index = pixelIndex(q);\n      let c = color[index];\n      let g = guide[index];\n      // Filter premultiplied color so coverage and color stay consistent.\n      let a = (textureLoad(albedoTex, q, 0).rgb - centerAlbedo.rgb) / featureScale;\n      let value = premultiplied(c);\n      sum += value * weight;\n      featureSum += a * weight;\n      featureSquareSum += a * a * weight;\n      featureColorSum += a * value.rgb * weight;\n      guideSum += g.x * weight;\n      guideVarianceSum += g.y * weight * weight;\n      weightSum += weight;\n      maxWeight = max(maxWeight, weight);\n    }\n  }\n\n  // Weight the center like its most similar neighbor (Buades et al. 2005)\n  // instead of 1, so isolated outliers are not kept unfiltered.\n  let center = color[index];\n  let centerGuide = guide[index];\n  let centerWeight = max(maxWeight, CENTER_WEIGHT_MIN);\n  let totalWeight = weightSum + centerWeight;\n  let centerValue = premultiplied(center);\n  let mean = (sum + centerValue * centerWeight) / totalWeight;\n  // Fit each color channel against relative albedo with the same JNLM weights.\n  // Centering at this pixel avoids cancellation when the feature is constant.\n  let meanFeature = featureSum / totalWeight;\n  let featureVariance = max(featureSquareSum / totalWeight - meanFeature * meanFeature, vec3f(0.0));\n  let covariance = featureColorSum / totalWeight - meanFeature * mean.rgb;\n  // A 1% contrast ridge stabilizes flat guides. Bound the response between no\n  // texture dependence and multiplicative albedo modulation; specular color\n  // with an unrelated albedo falls back to the weighted mean.\n  let meanAlbedo = centerAlbedo.rgb + meanFeature * featureScale;\n  let maximumSlope = mean.rgb * featureScale / max(meanAlbedo, vec3f(${JNLM_ALBEDO_FLOOR}));\n  let slope = clamp(covariance / (featureVariance + vec3f(${JNLM_ALBEDO_REGULARIZATION})), vec3f(0.0), maximumSlope);\n  let corrected = mean.rgb - slope * meanFeature;\n  let filtered = vec4f(max(corrected, vec3f(0.0)), max(mean.a, 0.0));\n  let rgb = select(\n    center.rgb,\n    filtered.rgb / max(filtered.a, 1e-6),\n    filtered.a > 1e-6\n  );\n  outColor[index] = vec4f(rgb, filtered.a);\n  outGuide[index] = vec2f(\n    (guideSum + centerGuide.x * centerWeight) / totalWeight,\n    (guideVarianceSum + centerGuide.y * centerWeight * centerWeight)\n      / (totalWeight * totalWeight)\n  );\n}\n`;\n}\n\nexport interface JNLMInputs {\n  width: number;\n  height: number;\n  /** Normalized shadow-catcher mask; red is the catcher coverage. */\n  shadowCatcher: GPUTexture;\n  /** Ray-trace auxiliary accumulation holding the guide moments. */\n  accumulation: GPUBuffer;\n  /** Ray-trace normal accumulation; its alpha is the per-pixel sample count. */\n  normalAccumulation: GPUBuffer;\n  /** Composed linear image to filter. */\n  color: GPUTexture;\n  albedo: GPUTexture;\n  normal: GPUTexture;\n  strength: number;\n  shape: JNLMFilterShape;\n}\n\nconst MAX_ITERATIONS = JNLM_MAX_ITERATIONS;\n\nexport class WGPUJNLMDenoisePass {\n  private _device: GPUDevice;\n  private _paramsBuffers: GPUBuffer[] = [];\n  private _guidePipeline: GPUComputePipeline;\n  private _variancePipeline: GPUComputePipeline;\n  private _loadColorPipeline: GPUComputePipeline;\n  private _filterPipeline?: GPUComputePipeline;\n  private _filterShapeKey = '';\n  private _guides: GPUBuffer[] = [];\n  private _colors: GPUBuffer[] = [];\n  private _width = 0;\n  private _height = 0;\n\n  constructor(device: GPUDevice) {\n    this._device = device;\n    // One uniform buffer per pass: every pass of an encoder sees the last\n    // writeBuffer() of a shared one.\n    for (let i = 0; i <= MAX_ITERATIONS; i++) {\n      this._paramsBuffers.push(device.createBuffer({\n        label: 'JNLM params',\n        size: 16,\n        usage: GPUBufferUsage.UNIFORM | GPUBufferUsage.COPY_DST\n      }));\n    }\n    this._guidePipeline = this._createPipeline('JNLM guide', jnlmGuideWGSL);\n    this._variancePipeline = this._createPipeline('JNLM variance', jnlmVarianceWGSL);\n    this._loadColorPipeline = this._createPipeline('JNLM load color', jnlmLoadColorWGSL);\n  }\n\n  private _createPipeline(label: string, code: string) {\n    return getTargetShaderCache(this._device).createComputePipeline({\n      label,\n      layout: 'auto',\n      compute: {\n        module: getTargetShaderCache(this._device).createShaderModule({ label, code }),\n        entryPoint: 'main'\n      }\n    });\n  }\n\n  private _getFilterPipeline(shape: JNLMFilterShape) {\n    const key = JSON.stringify(shape);\n    if (!this._filterPipeline || key !== this._filterShapeKey) {\n      this._filterPipeline = this._createPipeline('JNLM filter', createJNLMFilterWGSL(shape));\n      this._filterShapeKey = key;\n    }\n    return this._filterPipeline;\n  }\n\n  reserveCapacity() {}\n  private _ensureBuffers(width: number, height: number) {\n    if (width <= this._width && height <= this._height && this._colors.length) return;\n    width = Math.max(width, this._width); height = Math.max(height, this._height);\n    this._destroyBuffers();\n    const pixels = width * height;\n    const createBuffer = (label: string, bytes: number) =>\n      this._device.createBuffer({\n        label,\n        size: pixels * bytes,\n        usage: GPUBufferUsage.STORAGE | GPUBufferUsage.COPY_SRC\n      });\n    this._guides = [createBuffer('JNLM guide A', 8), createBuffer('JNLM guide B', 8)];\n    this._colors = [createBuffer('JNLM color A', 16), createBuffer('JNLM color B', 16)];\n    this._width = width;\n    this._height = height;\n  }\n\n  private _destroyBuffers() {\n    for (const buffer of [...this._guides, ...this._colors]) buffer.destroy();\n    this._guides = [];\n    this._colors = [];\n  }\n\n  /** Encodes the denoiser and returns the RGBA32F output in texture rows. */\n  encode(encoder: GPUCommandEncoder, inputs: JNLMInputs): GPUBuffer {\n    const { width, height, shape } = inputs;\n    this._ensureBuffers(width, height);\n    const iterations = Math.min(Math.max(Math.round(shape.iterations), 1), MAX_ITERATIONS);\n    const sensitivity = JNLM_DEFAULT_COLOR_SENSITIVITY * Math.max(inputs.strength, 1e-3);\n    this._paramsBuffers.forEach((buffer, i) => {\n      const params = new ArrayBuffer(16);\n      new Uint32Array(params, 0, 2).set([width, height]);\n      // Buffer i (from 1) is filter pass i - 1; index 0 is the prepasses.\n      new Float32Array(params, 8, 2).set([sensitivity, jnlmPassDilation(Math.max(i - 1, 0))]);\n      this._device.queue.writeBuffer(buffer, 0, params);\n    });\n\n    const device = this._device;\n    const filterPipeline = this._getFilterPipeline(shape);\n    const bind = (\n      pipeline: GPUComputePipeline,\n      paramsBuffer: GPUBuffer,\n      resources: GPUBindingResource[]\n    ) =>\n      device.createBindGroup({\n        layout: pipeline.getBindGroupLayout(0),\n        entries: [\n          { binding: 0, resource: { buffer: paramsBuffer } },\n          ...resources.map((resource, i) => ({ binding: i + 1, resource }))\n        ]\n      });\n    const [guideA, guideB] = this._guides;\n    const [colorA, colorB] = this._colors;\n    const prepass = this._paramsBuffers[0];\n    const passes: [GPUComputePipeline, GPUBindGroup][] = [\n      [this._guidePipeline, bind(this._guidePipeline, prepass, [\n        { buffer: inputs.accumulation },\n        { buffer: inputs.normalAccumulation },\n        { buffer: guideB }\n      ])],\n      [this._variancePipeline, bind(this._variancePipeline, prepass, [\n        { buffer: guideB },\n        { buffer: guideA }\n      ])],\n      [this._loadColorPipeline, bind(this._loadColorPipeline, prepass, [\n        inputs.color.createView(),\n        { buffer: colorA }\n      ])]\n    ];\n    const albedoView = inputs.albedo.createView();\n    const normalView = inputs.normal.createView();\n    const shadowCatcherView = inputs.shadowCatcher.createView();\n    let guides = [guideA, guideB];\n    let colors = [colorA, colorB];\n    for (let i = 1; i <= iterations; i++) {\n      passes.push([filterPipeline, bind(filterPipeline, this._paramsBuffers[i], [\n        { buffer: guides[0] },\n        { buffer: colors[0] },\n        albedoView,\n        normalView,\n        { buffer: colors[1] },\n        { buffer: guides[1] },\n        shadowCatcherView\n      ])]);\n      guides = [guides[1], guides[0]];\n      colors = [colors[1], colors[0]];\n    }\n    const pass = encoder.beginComputePass({ label: 'JNLM denoise' });\n    for (const [pipeline, bindGroup] of passes) {\n      pass.setPipeline(pipeline);\n      pass.setBindGroup(0, bindGroup);\n      pass.dispatchWorkgroups(\n        Math.ceil(width / JNLM_WORKGROUP_SIZE),\n        Math.ceil(height / JNLM_WORKGROUP_SIZE)\n      );\n    }\n    pass.end();\n    return colors[0];\n  }\n\n  getMemoryBytes() {\n    return [...this._paramsBuffers, ...this._guides, ...this._colors]\n      .reduce((sum, buffer) => sum + buffer.size, 0);\n  }\n\n  dispose() {\n    for (const buffer of this._paramsBuffers) buffer.destroy();\n    this._destroyBuffers();\n  }\n}\n"]}
+{"version":3,"file":"WGPUJNLMDenoiser.js","sourceRoot":"","sources":["../../src/renderer/WGPUJNLMDenoiser.ts"],"names":[],"mappings":"AAAA,OAAO,EAAC,oBAAoB,EAAC,MAAM,sBAAsB,CAAC;AAC1D,OAAO,EAAC,mBAAmB,EAAuB,MAAM,WAAW,CAAC;AACpE,OAAO,EACL,mBAAmB,EAAE,aAAa,EAAE,gBAAgB,EAAE,iBAAiB,EACvE,oBAAoB,EAAE,kBAAkB,EACzC,MAAM,wBAAwB,CAAC;AAChC,OAAO,EAAC,oBAAoB,EAAE,aAAa,EAAE,gBAAgB,EAAE,iBAAiB,EAAE,mBAAmB,EAAC,MAAM,wBAAwB,CAAC;AAqBrI,MAAM,cAAc,GAAG,mBAAmB,CAAC;AAE3C,MAAM,OAAO,mBAAmB;IACtB,OAAO,CAAY;IACnB,cAAc,GAAgB,EAAE,CAAC;IACjC,cAAc,CAAqB;IACnC,iBAAiB,CAAqB;IACtC,kBAAkB,CAAqB;IACvC,eAAe,CAAsB;IACrC,eAAe,GAAG,EAAE,CAAC;IACrB,OAAO,GAAgB,EAAE,CAAC;IAC1B,OAAO,GAAgB,EAAE,CAAC;IAC1B,MAAM,GAAG,CAAC,CAAC;IACX,OAAO,GAAG,CAAC,CAAC;IAEpB,YAAY,MAAiB;QAC3B,IAAI,CAAC,OAAO,GAAG,MAAM,CAAC;QACtB,sEAAsE;QACtE,iCAAiC;QACjC,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,IAAI,cAAc,EAAE,CAAC,EAAE,EAAE,CAAC;YACzC,IAAI,CAAC,cAAc,CAAC,IAAI,CAAC,MAAM,CAAC,YAAY,CAAC;gBAC3C,KAAK,EAAE,aAAa;gBACpB,IAAI,EAAE,EAAE;gBACR,KAAK,EAAE,cAAc,CAAC,OAAO,GAAG,cAAc,CAAC,QAAQ;aACxD,CAAC,CAAC,CAAC;QACN,CAAC;QACD,IAAI,CAAC,cAAc,GAAG,IAAI,CAAC,eAAe,CAAC,YAAY,EAAE,aAAa,CAAC,CAAC;QACxE,IAAI,CAAC,iBAAiB,GAAG,IAAI,CAAC,eAAe,CAAC,eAAe,EAAE,gBAAgB,CAAC,CAAC;QACjF,IAAI,CAAC,kBAAkB,GAAG,IAAI,CAAC,eAAe,CAAC,iBAAiB,EAAE,iBAAiB,CAAC,CAAC;IACvF,CAAC;IAEO,eAAe,CAAC,KAAa,EAAE,IAAY;QACjD,OAAO,oBAAoB,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC,qBAAqB,CAAC;YAC9D,KAAK;YACL,MAAM,EAAE,MAAM;YACd,OAAO,EAAE;gBACP,MAAM,EAAE,oBAAoB,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC,kBAAkB,CAAC,EAAE,KAAK,EAAE,IAAI,EAAE,CAAC;gBAC9E,UAAU,EAAE,MAAM;aACnB;SACF,CAAC,CAAC;IACL,CAAC;IAEO,kBAAkB,CAAC,KAAsB;QAC/C,MAAM,GAAG,GAAG,IAAI,CAAC,SAAS,CAAC,KAAK,CAAC,CAAC;QAClC,IAAI,CAAC,IAAI,CAAC,eAAe,IAAI,GAAG,KAAK,IAAI,CAAC,eAAe,EAAE,CAAC;YAC1D,IAAI,CAAC,eAAe,GAAG,IAAI,CAAC,eAAe,CAAC,aAAa,EAAE,oBAAoB,CAAC,KAAK,CAAC,CAAC,CAAC;YACxF,IAAI,CAAC,eAAe,GAAG,GAAG,CAAC;QAC7B,CAAC;QACD,OAAO,IAAI,CAAC,eAAe,CAAC;IAC9B,CAAC;IAED,eAAe,KAAI,CAAC;IACZ,cAAc,CAAC,KAAa,EAAE,MAAc;QAClD,IAAI,KAAK,IAAI,IAAI,CAAC,MAAM,IAAI,MAAM,IAAI,IAAI,CAAC,OAAO,IAAI,IAAI,CAAC,OAAO,CAAC,MAAM;YAAE,OAAO;QAClF,KAAK,GAAG,IAAI,CAAC,GAAG,CAAC,KAAK,EAAE,IAAI,CAAC,MAAM,CAAC,CAAC;QAAC,MAAM,GAAG,IAAI,CAAC,GAAG,CAAC,MAAM,EAAE,IAAI,CAAC,OAAO,CAAC,CAAC;QAC9E,IAAI,CAAC,eAAe,EAAE,CAAC;QACvB,MAAM,MAAM,GAAG,KAAK,GAAG,MAAM,CAAC;QAC9B,MAAM,YAAY,GAAG,CAAC,KAAa,EAAE,KAAa,EAAE,EAAE,CACpD,IAAI,CAAC,OAAO,CAAC,YAAY,CAAC;YACxB,KAAK;YACL,IAAI,EAAE,MAAM,GAAG,KAAK;YACpB,KAAK,EAAE,cAAc,CAAC,OAAO,GAAG,cAAc,CAAC,QAAQ;SACxD,CAAC,CAAC;QACL,IAAI,CAAC,OAAO,GAAG,CAAC,YAAY,CAAC,cAAc,EAAE,CAAC,CAAC,EAAE,YAAY,CAAC,cAAc,EAAE,CAAC,CAAC,CAAC,CAAC;QAClF,IAAI,CAAC,OAAO,GAAG,CAAC,YAAY,CAAC,cAAc,EAAE,EAAE,CAAC,EAAE,YAAY,CAAC,cAAc,EAAE,EAAE,CAAC,CAAC,CAAC;QACpF,IAAI,CAAC,MAAM,GAAG,KAAK,CAAC;QACpB,IAAI,CAAC,OAAO,GAAG,MAAM,CAAC;IACxB,CAAC;IAEO,eAAe;QACrB,KAAK,MAAM,MAAM,IAAI,CAAC,GAAG,IAAI,CAAC,OAAO,EAAE,GAAG,IAAI,CAAC,OAAO,CAAC;YAAE,MAAM,CAAC,OAAO,EAAE,CAAC;QAC1E,IAAI,CAAC,OAAO,GAAG,EAAE,CAAC;QAClB,IAAI,CAAC,OAAO,GAAG,EAAE,CAAC;IACpB,CAAC;IAED,2EAA2E;IAC3E,MAAM,CAAC,OAA0B,EAAE,MAAkB;QACnD,MAAM,EAAE,KAAK,EAAE,MAAM,EAAE,KAAK,EAAE,GAAG,MAAM,CAAC;QACxC,IAAI,CAAC,cAAc,CAAC,KAAK,EAAE,MAAM,CAAC,CAAC;QACnC,MAAM,UAAU,GAAG,IAAI,CAAC,GAAG,CAAC,IAAI,CAAC,GAAG,CAAC,IAAI,CAAC,KAAK,CAAC,KAAK,CAAC,UAAU,CAAC,EAAE,CAAC,CAAC,EAAE,cAAc,CAAC,CAAC;QACvF,IAAI,CAAC,cAAc,CAAC,OAAO,CAAC,CAAC,MAAM,EAAE,CAAC,EAAE,EAAE;YACxC,oEAAoE;YACpE,MAAM,MAAM,GAAG,kBAAkB,CAAC,KAAK,EAAE,MAAM,EAAE,MAAM,CAAC,QAAQ,EAAE,IAAI,CAAC,GAAG,CAAC,CAAC,GAAG,CAAC,EAAE,CAAC,CAAC,CAAC,CAAC;YACtF,IAAI,CAAC,OAAO,CAAC,KAAK,CAAC,WAAW,CAAC,MAAM,EAAE,CAAC,EAAE,MAAM,CAAC,CAAC;QACpD,CAAC,CAAC,CAAC;QAEH,MAAM,MAAM,GAAG,IAAI,CAAC,OAAO,CAAC;QAC5B,MAAM,cAAc,GAAG,IAAI,CAAC,kBAAkB,CAAC,KAAK,CAAC,CAAC;QACtD,MAAM,IAAI,GAAG,CACX,QAA4B,EAC5B,YAAuB,EACvB,SAA+B,EAC/B,EAAE,CACF,MAAM,CAAC,eAAe,CAAC;YACrB,MAAM,EAAE,QAAQ,CAAC,kBAAkB,CAAC,CAAC,CAAC;YACtC,OAAO,EAAE;gBACP,EAAE,OAAO,EAAE,CAAC,EAAE,QAAQ,EAAE,EAAE,MAAM,EAAE,YAAY,EAAE,EAAE;gBAClD,GAAG,SAAS,CAAC,GAAG,CAAC,CAAC,QAAQ,EAAE,CAAC,EAAE,EAAE,CAAC,CAAC,EAAE,OAAO,EAAE,CAAC,GAAG,CAAC,EAAE,QAAQ,EAAE,CAAC,CAAC;aAClE;SACF,CAAC,CAAC;QACL,MAAM,CAAC,MAAM,EAAE,MAAM,CAAC,GAAG,IAAI,CAAC,OAAO,CAAC;QACtC,MAAM,CAAC,MAAM,EAAE,MAAM,CAAC,GAAG,IAAI,CAAC,OAAO,CAAC;QACtC,MAAM,OAAO,GAAG,IAAI,CAAC,cAAc,CAAC,CAAC,CAAC,CAAC;QACvC,MAAM,MAAM,GAAyC;YACnD,CAAC,IAAI,CAAC,cAAc,EAAE,IAAI,CAAC,IAAI,CAAC,cAAc,EAAE,OAAO,EAAE;oBACvD,EAAE,MAAM,EAAE,MAAM,CAAC,YAAY,EAAE;oBAC/B,EAAE,MAAM,EAAE,MAAM,CAAC,kBAAkB,EAAE;oBACrC,EAAE,MAAM,EAAE,MAAM,EAAE;iBACnB,CAAC,CAAC;YACH,CAAC,IAAI,CAAC,iBAAiB,EAAE,IAAI,CAAC,IAAI,CAAC,iBAAiB,EAAE,OAAO,EAAE;oBAC7D,EAAE,MAAM,EAAE,MAAM,EAAE;oBAClB,EAAE,MAAM,EAAE,MAAM,EAAE;iBACnB,CAAC,CAAC;YACH,CAAC,IAAI,CAAC,kBAAkB,EAAE,IAAI,CAAC,IAAI,CAAC,kBAAkB,EAAE,OAAO,EAAE;oBAC/D,MAAM,CAAC,KAAK,CAAC,UAAU,EAAE;oBACzB,EAAE,MAAM,EAAE,MAAM,EAAE;iBACnB,CAAC,CAAC;SACJ,CAAC;QACF,MAAM,UAAU,GAAG,MAAM,CAAC,MAAM,CAAC,UAAU,EAAE,CAAC;QAC9C,MAAM,UAAU,GAAG,MAAM,CAAC,MAAM,CAAC,UAAU,EAAE,CAAC;QAC9C,MAAM,iBAAiB,GAAG,MAAM,CAAC,aAAa,CAAC,UAAU,EAAE,CAAC;QAC5D,IAAI,MAAM,GAAG,CAAC,MAAM,EAAE,MAAM,CAAC,CAAC;QAC9B,IAAI,MAAM,GAAG,CAAC,MAAM,EAAE,MAAM,CAAC,CAAC;QAC9B,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,IAAI,UAAU,EAAE,CAAC,EAAE,EAAE,CAAC;YACrC,MAAM,CAAC,IAAI,CAAC,CAAC,cAAc,EAAE,IAAI,CAAC,cAAc,EAAE,IAAI,CAAC,cAAc,CAAC,CAAC,CAAC,EAAE;oBACxE,EAAE,MAAM,EAAE,MAAM,CAAC,CAAC,CAAC,EAAE;oBACrB,EAAE,MAAM,EAAE,MAAM,CAAC,CAAC,CAAC,EAAE;oBACrB,UAAU;oBACV,UAAU;oBACV,EAAE,MAAM,EAAE,MAAM,CAAC,CAAC,CAAC,EAAE;oBACrB,EAAE,MAAM,EAAE,MAAM,CAAC,CAAC,CAAC,EAAE;oBACrB,iBAAiB;iBAClB,CAAC,CAAC,CAAC,CAAC;YACL,MAAM,GAAG,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,MAAM,CAAC,CAAC,CAAC,CAAC,CAAC;YAChC,MAAM,GAAG,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,MAAM,CAAC,CAAC,CAAC,CAAC,CAAC;QAClC,CAAC;QACD,MAAM,IAAI,GAAG,OAAO,CAAC,gBAAgB,CAAC,EAAE,KAAK,EAAE,cAAc,EAAE,CAAC,CAAC;QACjE,KAAK,MAAM,CAAC,QAAQ,EAAE,SAAS,CAAC,IAAI,MAAM,EAAE,CAAC;YAC3C,IAAI,CAAC,WAAW,CAAC,QAAQ,CAAC,CAAC;YAC3B,IAAI,CAAC,YAAY,CAAC,CAAC,EAAE,SAAS,CAAC,CAAC;YAChC,IAAI,CAAC,kBAAkB,CACrB,IAAI,CAAC,IAAI,CAAC,KAAK,GAAG,mBAAmB,CAAC,EACtC,IAAI,CAAC,IAAI,CAAC,MAAM,GAAG,mBAAmB,CAAC,CACxC,CAAC;QACJ,CAAC;QACD,IAAI,CAAC,GAAG,EAAE,CAAC;QACX,OAAO,MAAM,CAAC,CAAC,CAAC,CAAC;IACnB,CAAC;IAED,cAAc;QACZ,OAAO,CAAC,GAAG,IAAI,CAAC,cAAc,EAAE,GAAG,IAAI,CAAC,OAAO,EAAE,GAAG,IAAI,CAAC,OAAO,CAAC;aAC9D,MAAM,CAAC,CAAC,GAAG,EAAE,MAAM,EAAE,EAAE,CAAC,GAAG,GAAG,MAAM,CAAC,IAAI,EAAE,CAAC,CAAC,CAAC;IACnD,CAAC;IAED,OAAO;QACL,KAAK,MAAM,MAAM,IAAI,IAAI,CAAC,cAAc;YAAE,MAAM,CAAC,OAAO,EAAE,CAAC;QAC3D,IAAI,CAAC,eAAe,EAAE,CAAC;IACzB,CAAC;CACF","sourcesContent":["import {getTargetShaderCache} from './WGPUShaderCache.js';\nimport {JNLM_MAX_ITERATIONS, type JNLMFilterShape} from './jnlm.js';\nimport {\n  JNLM_WORKGROUP_SIZE, jnlmGuideWGSL, jnlmVarianceWGSL, jnlmLoadColorWGSL,\n  createJNLMFilterWGSL, packJNLMParameters\n} from './wgsl/jnlm.compute.js';\nexport {createJNLMFilterWGSL, jnlmGuideWGSL, jnlmVarianceWGSL, jnlmLoadColorWGSL, JNLM_WORKGROUP_SIZE} from './wgsl/jnlm.compute.js';\n\n/** WebGPU passes of the JNLM denoiser described in `jnlm.ts`. */\n\nexport interface JNLMInputs {\n  width: number;\n  height: number;\n  /** Normalized shadow-catcher mask; red is the catcher coverage. */\n  shadowCatcher: GPUTexture;\n  /** Ray-trace auxiliary accumulation holding the guide moments. */\n  accumulation: GPUBuffer;\n  /** Ray-trace normal accumulation; its alpha is the per-pixel sample count. */\n  normalAccumulation: GPUBuffer;\n  /** Composed linear image to filter. */\n  color: GPUTexture;\n  albedo: GPUTexture;\n  normal: GPUTexture;\n  strength: number;\n  shape: JNLMFilterShape;\n}\n\nconst MAX_ITERATIONS = JNLM_MAX_ITERATIONS;\n\nexport class WGPUJNLMDenoisePass {\n  private _device: GPUDevice;\n  private _paramsBuffers: GPUBuffer[] = [];\n  private _guidePipeline: GPUComputePipeline;\n  private _variancePipeline: GPUComputePipeline;\n  private _loadColorPipeline: GPUComputePipeline;\n  private _filterPipeline?: GPUComputePipeline;\n  private _filterShapeKey = '';\n  private _guides: GPUBuffer[] = [];\n  private _colors: GPUBuffer[] = [];\n  private _width = 0;\n  private _height = 0;\n\n  constructor(device: GPUDevice) {\n    this._device = device;\n    // One uniform buffer per pass: every pass of an encoder sees the last\n    // writeBuffer() of a shared one.\n    for (let i = 0; i <= MAX_ITERATIONS; i++) {\n      this._paramsBuffers.push(device.createBuffer({\n        label: 'JNLM params',\n        size: 16,\n        usage: GPUBufferUsage.UNIFORM | GPUBufferUsage.COPY_DST\n      }));\n    }\n    this._guidePipeline = this._createPipeline('JNLM guide', jnlmGuideWGSL);\n    this._variancePipeline = this._createPipeline('JNLM variance', jnlmVarianceWGSL);\n    this._loadColorPipeline = this._createPipeline('JNLM load color', jnlmLoadColorWGSL);\n  }\n\n  private _createPipeline(label: string, code: string) {\n    return getTargetShaderCache(this._device).createComputePipeline({\n      label,\n      layout: 'auto',\n      compute: {\n        module: getTargetShaderCache(this._device).createShaderModule({ label, code }),\n        entryPoint: 'main'\n      }\n    });\n  }\n\n  private _getFilterPipeline(shape: JNLMFilterShape) {\n    const key = JSON.stringify(shape);\n    if (!this._filterPipeline || key !== this._filterShapeKey) {\n      this._filterPipeline = this._createPipeline('JNLM filter', createJNLMFilterWGSL(shape));\n      this._filterShapeKey = key;\n    }\n    return this._filterPipeline;\n  }\n\n  reserveCapacity() {}\n  private _ensureBuffers(width: number, height: number) {\n    if (width <= this._width && height <= this._height && this._colors.length) return;\n    width = Math.max(width, this._width); height = Math.max(height, this._height);\n    this._destroyBuffers();\n    const pixels = width * height;\n    const createBuffer = (label: string, bytes: number) =>\n      this._device.createBuffer({\n        label,\n        size: pixels * bytes,\n        usage: GPUBufferUsage.STORAGE | GPUBufferUsage.COPY_SRC\n      });\n    this._guides = [createBuffer('JNLM guide A', 8), createBuffer('JNLM guide B', 8)];\n    this._colors = [createBuffer('JNLM color A', 16), createBuffer('JNLM color B', 16)];\n    this._width = width;\n    this._height = height;\n  }\n\n  private _destroyBuffers() {\n    for (const buffer of [...this._guides, ...this._colors]) buffer.destroy();\n    this._guides = [];\n    this._colors = [];\n  }\n\n  /** Encodes the denoiser and returns the RGBA32F output in texture rows. */\n  encode(encoder: GPUCommandEncoder, inputs: JNLMInputs): GPUBuffer {\n    const { width, height, shape } = inputs;\n    this._ensureBuffers(width, height);\n    const iterations = Math.min(Math.max(Math.round(shape.iterations), 1), MAX_ITERATIONS);\n    this._paramsBuffers.forEach((buffer, i) => {\n      // Buffer i (from 1) is filter pass i - 1; index 0 is the prepasses.\n      const params = packJNLMParameters(width, height, inputs.strength, Math.max(i - 1, 0));\n      this._device.queue.writeBuffer(buffer, 0, params);\n    });\n\n    const device = this._device;\n    const filterPipeline = this._getFilterPipeline(shape);\n    const bind = (\n      pipeline: GPUComputePipeline,\n      paramsBuffer: GPUBuffer,\n      resources: GPUBindingResource[]\n    ) =>\n      device.createBindGroup({\n        layout: pipeline.getBindGroupLayout(0),\n        entries: [\n          { binding: 0, resource: { buffer: paramsBuffer } },\n          ...resources.map((resource, i) => ({ binding: i + 1, resource }))\n        ]\n      });\n    const [guideA, guideB] = this._guides;\n    const [colorA, colorB] = this._colors;\n    const prepass = this._paramsBuffers[0];\n    const passes: [GPUComputePipeline, GPUBindGroup][] = [\n      [this._guidePipeline, bind(this._guidePipeline, prepass, [\n        { buffer: inputs.accumulation },\n        { buffer: inputs.normalAccumulation },\n        { buffer: guideB }\n      ])],\n      [this._variancePipeline, bind(this._variancePipeline, prepass, [\n        { buffer: guideB },\n        { buffer: guideA }\n      ])],\n      [this._loadColorPipeline, bind(this._loadColorPipeline, prepass, [\n        inputs.color.createView(),\n        { buffer: colorA }\n      ])]\n    ];\n    const albedoView = inputs.albedo.createView();\n    const normalView = inputs.normal.createView();\n    const shadowCatcherView = inputs.shadowCatcher.createView();\n    let guides = [guideA, guideB];\n    let colors = [colorA, colorB];\n    for (let i = 1; i <= iterations; i++) {\n      passes.push([filterPipeline, bind(filterPipeline, this._paramsBuffers[i], [\n        { buffer: guides[0] },\n        { buffer: colors[0] },\n        albedoView,\n        normalView,\n        { buffer: colors[1] },\n        { buffer: guides[1] },\n        shadowCatcherView\n      ])]);\n      guides = [guides[1], guides[0]];\n      colors = [colors[1], colors[0]];\n    }\n    const pass = encoder.beginComputePass({ label: 'JNLM denoise' });\n    for (const [pipeline, bindGroup] of passes) {\n      pass.setPipeline(pipeline);\n      pass.setBindGroup(0, bindGroup);\n      pass.dispatchWorkgroups(\n        Math.ceil(width / JNLM_WORKGROUP_SIZE),\n        Math.ceil(height / JNLM_WORKGROUP_SIZE)\n      );\n    }\n    pass.end();\n    return colors[0];\n  }\n\n  getMemoryBytes() {\n    return [...this._paramsBuffers, ...this._guides, ...this._colors]\n      .reduce((sum, buffer) => sum + buffer.size, 0);\n  }\n\n  dispose() {\n    for (const buffer of this._paramsBuffers) buffer.destroy();\n    this._destroyBuffers();\n  }\n}\n"]}