@meimu/mutracer 0.1.5 → 0.3.0

package/lib/renderer/WGPURayTraceOutput.d.ts

@@ -1,7 +1,9 @@
-export declare function createNormalizeRayTraceOutputWGSL(layers?: boolean, catcher?: boolean): string;
+export declare function createNormalizeRayTraceOutputWGSL(layers?: boolean, catcher?: boolean, caustics?: boolean): string;
 type RayTraceOutput = 'color' | 'normal' | 'albedo' | 'shadowCatcher' | 'foreground' | 'catcherBackground';
 type RayTraceBuffers = {
     color: GPUBuffer;
+    caustic?: GPUBuffer;
+    wideHdr?: boolean;
     normal: GPUBuffer;
     albedo: GPUBuffer;
     shadowCatcher: GPUBuffer;
@@ -23,6 +25,12 @@ export declare class WGPURayTraceOutputPass {
     private _width;
     private _height;
     private _hasCatcher;
+    private _hasCaustics;
+    private _hasWideHdr;
+    private _maskPipeline?;
+    private _wideColor?;
+    private _wideForeground?;
+    private _causticPipeline?;
     private _needsUpdatePipeline;
     constructor(device: GPUDevice, zeroTexture?: GPUTexture | undefined);
     private readonly _retainedOutputs;
@@ -37,8 +45,12 @@ export declare class WGPURayTraceOutputPass {
     setSampleTileGrid(columns: number, rows: number): void;
     createPass(encoder: GPUCommandEncoder, inputBuffers: RayTraceBuffers, separateCatcher?: boolean): void;
     private _createOutputPass;
+    private _resolveMask;
+    private _addCaustics;
     private _destroyTextures;
     getMemoryBytes(): number;
     dispose(): void;
 }
+export declare const causticCompositeWGSL = "\n@group(0)@binding(0)var base:texture_2d<f32>;\n@group(0)@binding(1)var<storage,read>caustics:array<vec4f>;\n@fragment fn main(@builtin(position)position:vec4f)->@location(0)vec4f{\n let pixel=vec2u(position.xy);let size=textureDimensions(base);\n let c=textureLoad(base,vec2i(pixel),0);let photon=caustics[pixel.x+(size.y-pixel.y-1u)*size.x];\n return vec4f(c.rgb+photon.rgb/max(photon.a,1.0),c.a);\n}\n";
+export declare const shadowMaskResolveWGSL: string;
 export {};

package/lib/renderer/WGPURayTraceOutput.js

@@ -1,48 +1,51 @@
+import { hdrResolveWGSL } from './wgsl/compose.frag.wgsl.js';
 import { getTargetShaderCache } from './WGPUShaderCache.js';
 import { textureBytes } from './gpuMemory.js';
 import shadowCatcherAccumulationWGSL from './wgsl/chunks/shadowCatcherAccumulation.wgsl.js';
 import { fullScreenQuadVertexShaderWGSL } from './WGPUFullQuadPass.js';
-export function createNormalizeRayTraceOutputWGSL(layers = false, catcher = true) {
+export function createNormalizeRayTraceOutputWGSL(layers = false, catcher = true, caustics = false) {
     return /* wgsl */ `
+${hdrResolveWGSL}
 @group(0) @binding(0) var<storage, read> colorBuffer: array<vec4f>;
 @group(0) @binding(1) var<storage, read> normalBuffer: array<vec4f>;
 @group(0) @binding(2) var<storage, read> albedoBuffer: array<vec4f>;
 ${shadowCatcherAccumulationWGSL}
 @group(0) @binding(3) var<storage, read> shadowCatcherBuffer: array<ShadowCatcherAccumulation>;
 
-@group(1) @binding(0) var<uniform> sampleCount: f32;
+${caustics ? /* wgsl */ `@group(0) @binding(4) var<storage, read> causticBuffer: array<vec4f>;
+` : ''}@group(1) @binding(0) var<uniform> sampleCount: f32;
 @group(1) @binding(1) var<uniform> sampleTileGrid: vec2u;
 
 override outputWidth: f32;
 override outputHeight: f32;
 
-struct pd {
+struct pV {
 ${layers ? `@location(0) foreground: vec4f,
 @location(1) catcherBackground: vec4f` : `@location(0) color: vec4f,
 @location(1) normal: vec4f,
 @location(2) albedo: vec4f${catcher ? ',\n  @location(3) shadowCatcher: vec4f' : ''}`}
 }
 @fragment
-fn main(@builtin(position) fH: vec4f) -> pd {
+fn main(@builtin(position) coord: vec4f) -> pV {
 let size = vec2u(u32(outputWidth), u32(outputHeight));
-let fQ = vec2u(u32(fH.x), size.y - u32(fH.y) - 1u);
-var idx = fQ.x + fQ.y * size.x;
+let pixel = vec2u(u32(coord.x), size.y - u32(coord.y) - 1u);
+var idx = pixel.x + pixel.y * size.x;
 
 
 
 
 if (sampleCount < 1.0 && normalBuffer[idx].a <= 0.0) {
-let vc = max(sampleTileGrid, vec2u(1u));
-let uM = min(
-((fQ + vc / 2u) / vc) * vc,
-((size - vec2u(1u)) / vc) * vc
+let grid = max(sampleTileGrid, vec2u(1u));
+let nearest = min(
+((pixel + grid / 2u) / grid) * grid,
+((size - vec2u(1u)) / grid) * grid
 );
-let sH = uM.x + uM.y * size.x;
-if (normalBuffer[sH].a > 0.0) {
-idx = sH;
+let tZ = nearest.x + nearest.y * size.x;
+if (normalBuffer[tZ].a > 0.0) {
+idx = tZ;
 }
 }
-var output: pd;
+var output: pV;
 if (normalBuffer[idx].a <= 0.0) {
 ${layers ? `output.foreground = vec4f(0.0);
 output.catcherBackground = vec4f(0.0);` : `output.color = vec4f(0.0);
@@ -62,20 +65,22 @@ let hit = albedo.a;
 let total = max(normal.a, 1.0);
 ${layers ? `
 let shadow = shadowCatcherBuffer[idx];
-let iV = max(hit - shadow.mask.b, 0.0);
+let iT = max(hit - shadow.mask.b, 0.0);
 
 
 output.foreground = vec4f(
 (color.rgb - shadow.background.rgb)
-/ select(total, iV, iV > 0.0),
+/ select(total, iT, iT > 0.0),
 clamp((color.a - shadow.mask.g) / total, 0.0, 1.0)
 );
 output.catcherBackground = shadow.background / total;
 ` : `output.color = vec4f(
 
 
-color.rgb / select(total, hit, hit > 0.0),
-color.a / total
+vec3f(fo(color.r,normal.a,hit,1.0,true),
+fo(color.g,normal.a,hit,1.0,true),
+fo(color.b,normal.a,hit,1.0,true)),
+fo(color.a,normal.a,hit,1.0,false)
 );
 
 output.normal = vec4f(0.0);
@@ -84,9 +89,16 @@ if (dot(normal.rgb, normal.rgb) > 0.1) {
 output.normal = vec4f(normal.rgb / total * 0.5 + 0.5, 1.0);
 }
 output.albedo = vec4f(albedo.rgb / total, clamp(hit / total, 0.0, 1.0));
-${catcher ? 'output.shadowCatcher = shadowCatcherBuffer[idx].mask / total;' : ''}
+${catcher ? `let shadow = shadowCatcherBuffer[idx];
+
+
+
+output.shadowCatcher = vec4f(shadow.mask.rgb / total,
+shadow.mask.r / total * select(0.0, shadow.moments.z / shadow.moments.w, shadow.moments.w > 0.0));` : ''}
 `}
-return output;
+${caustics ? /* wgsl */ `  let causticColor=causticBuffer[idx].rgb/max(causticBuffer[idx].a,1.0);
+${layers ? 'output.foreground = vec4f(output.foreground.rgb+causticColor,output.foreground.a);' : 'output.color = vec4f(output.color.rgb+causticColor,output.color.a);'}
+` : ''}  return output;
 }
 `;
 }
@@ -106,6 +118,12 @@ export class WGPURayTraceOutputPass {
     _width = 1;
     _height = 1;
     _hasCatcher = false;
+    _hasCaustics = false;
+    _hasWideHdr = false;
+    _maskPipeline;
+    _wideColor;
+    _wideForeground;
+    _causticPipeline;
     _needsUpdatePipeline = true;
     constructor(device, zeroTexture) {
         this.zeroTexture = zeroTexture;
@@ -136,6 +154,10 @@ export class WGPURayTraceOutputPass {
         }
     }
     getOutput(label) {
+        if (this._hasCaustics && label === 'color' && this._wideColor)
+            return this._wideColor;
+        if (this._hasCaustics && label === 'foreground' && this._wideForeground)
+            return this._wideForeground;
         return this[`_${label}Texture`] ?? this.zeroTexture;
     }
     _createTextures() {
@@ -154,7 +176,7 @@ export class WGPURayTraceOutputPass {
         this._colorTexture = device.createTexture({
             ...textureDescriptor,
             label: 'Ray Trace Color',
-            format: 'rgba16float',
+            format: this._hasWideHdr ? 'rgba32float' : 'rgba16float',
             usage: GPUTextureUsage.RENDER_ATTACHMENT |
                 GPUTextureUsage.TEXTURE_BINDING |
                 GPUTextureUsage.COPY_SRC
@@ -168,11 +190,17 @@ export class WGPURayTraceOutputPass {
             ...textureDescriptor,
             label: 'Ray Trace Albedo'
         });
+        this._wideColor = this._wideForeground = undefined;
+        if (this._hasCaustics) {
+            this._wideColor = device.createTexture({ ...textureDescriptor, label: 'Caustic HDR color', format: 'rgba32float' });
+            if (this._hasCatcher)
+                this._wideForeground = device.createTexture({ ...textureDescriptor, label: 'Caustic HDR foreground', format: 'rgba32float' });
+        }
         if (this._hasCatcher) {
             this._foregroundTexture = device.createTexture({
                 ...textureDescriptor,
                 label: 'Ray Trace Foreground',
-                format: 'rgba16float'
+                format: this._hasWideHdr ? 'rgba32float' : 'rgba16float'
             });
             this._catcherBackgroundTexture = device.createTexture({
                 ...textureDescriptor,
@@ -226,12 +254,12 @@ export class WGPURayTraceOutputPass {
             },
             fragment: {
                 module: getTargetShaderCache(device).createShaderModule({
-                    code: createNormalizeRayTraceOutputWGSL(false, this._hasCatcher)
+                    code: createNormalizeRayTraceOutputWGSL(false, this._hasCatcher && !this._hasWideHdr, false)
                 }),
                 entryPoint: 'main',
                 targets: [
                     {
-                        format: 'rgba16float'
+                        format: this._hasWideHdr ? 'rgba32float' : 'rgba16float'
                     },
                     {
                         format: 'rgba8unorm'
@@ -239,7 +267,7 @@ export class WGPURayTraceOutputPass {
                     {
                         format: 'rgba8unorm'
                     },
-                    ...(this._hasCatcher ? [{ format: 'rgba16float' }] : [])
+                    ...(this._hasCatcher && !this._hasWideHdr ? [{ format: 'rgba16float' }] : [])
                 ],
                 constants: {
                     outputWidth: this._width,
@@ -256,9 +284,9 @@ export class WGPURayTraceOutputPass {
                     entryPoint: 'main'
                 },
                 fragment: {
-                    module: getTargetShaderCache(device).createShaderModule({ code: createNormalizeRayTraceOutputWGSL(true) }),
+                    module: getTargetShaderCache(device).createShaderModule({ code: createNormalizeRayTraceOutputWGSL(true, true, false) }),
                     entryPoint: 'main',
-                    targets: [{ format: 'rgba16float' }, { format: 'rgba16float' }],
+                    targets: [{ format: this._hasWideHdr ? 'rgba32float' : 'rgba16float' }, { format: 'rgba16float' }],
                     constants: { outputWidth: this._width, outputHeight: this._height }
                 }
             });
@@ -270,6 +298,7 @@ export class WGPURayTraceOutputPass {
             !this._colorTexture) {
             this._width = width;
             this._height = height;
+            this._maskPipeline = undefined;
             this._createTextures();
             this._needsUpdatePipeline = true;
         }
@@ -281,6 +310,16 @@ export class WGPURayTraceOutputPass {
         this._device.queue.writeBuffer(this._uniformsBuffers.sampleTileGrid, 0, new Uint32Array([columns, rows]));
     }
     createPass(encoder, inputBuffers, separateCatcher = false) {
+        if (this._hasWideHdr !== !!inputBuffers.wideHdr) {
+            this._hasWideHdr = !!inputBuffers.wideHdr;
+            this._createTextures();
+            this._needsUpdatePipeline = true;
+        }
+        if (this._hasCaustics !== !!inputBuffers.caustic) {
+            this._hasCaustics = !!inputBuffers.caustic;
+            this._createTextures();
+            this._needsUpdatePipeline = true;
+        }
         if (this._hasCatcher !== separateCatcher) {
             this._hasCatcher = separateCatcher;
             this._createTextures();
@@ -290,12 +329,19 @@ export class WGPURayTraceOutputPass {
         this._createOutputPass(encoder, inputBuffers, false);
         if (separateCatcher)
             this._createOutputPass(encoder, inputBuffers, true);
+        if (this._hasWideHdr && separateCatcher)
+            this._resolveMask(encoder, inputBuffers);
+        if (inputBuffers.caustic) {
+            this._addCaustics(encoder, this._colorTexture, this._wideColor, inputBuffers.caustic);
+            if (separateCatcher)
+                this._addCaustics(encoder, this._foregroundTexture, this._wideForeground, inputBuffers.caustic);
+        }
     }
     _createOutputPass(encoder, inputBuffers, layers) {
         const pipeline = layers ? this._layersPipeline : this._pipeline;
         const textures = layers
             ? [this._foregroundTexture, this._catcherBackgroundTexture]
-            : [this._colorTexture, this._normalTexture, this._albedoTexture, ...(this._hasCatcher ? [this._shadowCatcherTexture] : [])];
+            : [this._colorTexture, this._normalTexture, this._albedoTexture, ...(this._hasCatcher && !this._hasWideHdr ? [this._shadowCatcherTexture] : [])];
         const pass = encoder.beginRenderPass({
             colorAttachments: textures.map((texture) => ({
                 view: texture.createView(),
@@ -319,7 +365,7 @@ export class WGPURayTraceOutputPass {
                     buffer,
                     ...(idx === 3 ? { offset: inputBuffers.shadowCatcherOffset ?? 0 } : {})
                 }
-            })).filter(entry => entry.binding < 3 || this._hasCatcher || layers)
+            })).filter(entry => entry.binding < 3 || (this._hasCatcher && !this._hasWideHdr) || layers).concat([])
         }));
         pass.setBindGroup(1, layers ? this._device.createBindGroup({
             layout: pipeline.getBindGroupLayout(1),
@@ -330,6 +376,31 @@ export class WGPURayTraceOutputPass {
         pass.draw(6, 1, 0, 0);
         pass.end();
     }
+    _resolveMask(encoder, input) {
+        const device = this._device;
+        this._maskPipeline ??= device.createRenderPipeline({ label: 'Wide HDR catcher mask', layout: 'auto',
+            vertex: { module: device.createShaderModule({ code: fullScreenQuadVertexShaderWGSL }), entryPoint: 'main' },
+            fragment: { module: device.createShaderModule({ code: shadowMaskResolveWGSL }), entryPoint: 'main', constants: { IMAGE_WIDTH: this._width, IMAGE_HEIGHT: this._height }, targets: [{ format: 'rgba16float' }] } });
+        const pass = encoder.beginRenderPass({ label: 'Wide HDR catcher mask', colorAttachments: [{ view: this._shadowCatcherTexture.createView(), loadOp: 'clear', storeOp: 'store' }] });
+        pass.setPipeline(this._maskPipeline);
+        pass.setBindGroup(0, device.createBindGroup({ layout: this._maskPipeline.getBindGroupLayout(0), entries: [
+                { binding: 0, resource: { buffer: input.normal } }, { binding: 1, resource: { buffer: input.shadowCatcher } },
+                { binding: 2, resource: { buffer: this._uniformsBuffers.sampleTileGrid } }, { binding: 3, resource: { buffer: this._uniformsBuffers.sampleCount } }
+            ] }));
+        pass.draw(6);
+        pass.end();
+    }
+    _addCaustics(encoder, base, target, caustics) {
+        const device = this._device;
+        this._causticPipeline ??= device.createRenderPipeline({ label: 'Caustic HDR resolve', layout: 'auto',
+            vertex: { module: device.createShaderModule({ code: fullScreenQuadVertexShaderWGSL }), entryPoint: 'main' },
+            fragment: { module: device.createShaderModule({ code: causticCompositeWGSL }), entryPoint: 'main', targets: [{ format: 'rgba32float' }] } });
+        const pass = encoder.beginRenderPass({ label: 'Caustic HDR resolve', colorAttachments: [{ view: target.createView(), loadOp: 'clear', storeOp: 'store' }] });
+        pass.setPipeline(this._causticPipeline);
+        pass.setBindGroup(0, device.createBindGroup({ layout: this._causticPipeline.getBindGroupLayout(0), entries: [{ binding: 0, resource: base.createView() }, { binding: 1, resource: { buffer: caustics } }] }));
+        pass.draw(6);
+        pass.end();
+    }
     _destroyTextures() {
         if (!this._colorTexture)
             return;
@@ -339,6 +410,7 @@ export class WGPURayTraceOutputPass {
         // far has completed; destroying them synchronously can make Dawn validate
         // the pending submit against an already-destroyed texture.
         const textures = [
+            this._wideColor, this._wideForeground,
             this._colorTexture,
             this._normalTexture,
             this._albedoTexture,
@@ -367,4 +439,35 @@ export class WGPURayTraceOutputPass {
         });
     }
 }
+// A separate color attachment preserves float32 caustics within the baseline
+// 32-byte MRT budget, including when the catcher needs its fourth attachment.
+export const causticCompositeWGSL = /* wgsl */ `
+@group(0)@binding(0)var base:texture_2d<f32>;
+@group(0)@binding(1)var<storage,read>caustics:array<vec4f>;
+@fragment fn main(@builtin(position)position:vec4f)->@location(0)vec4f{
+let pixel=vec2u(position.xy);let size=textureDimensions(base);
+let c=textureLoad(base,vec2i(pixel),0);let photon=caustics[pixel.x+(size.y-pixel.y-1u)*size.x];
+return vec4f(c.rgb+photon.rgb/max(photon.a,1.0),c.a);
+}
+`;
+export const shadowMaskResolveWGSL = /* wgsl */ `
+${shadowCatcherAccumulationWGSL}
+override IMAGE_WIDTH:u32;override IMAGE_HEIGHT:u32;
+@group(0)@binding(0)var<storage,read>normal:array<vec4f>;
+@group(0)@binding(1)var<storage,read>shadow:array<ShadowCatcherAccumulation>;
+@group(0)@binding(2)var<uniform>grid:vec2u;
+@group(0)@binding(3)var<uniform>samples:f32;
+@fragment fn main(@builtin(position)position:vec4f)->@location(0)vec4f{
+
+let size=vec2u(IMAGE_WIDTH,IMAGE_HEIGHT);
+let pixel=vec2u(u32(position.x),size.y-u32(position.y)-1u);var index=pixel.x+pixel.y*size.x;
+if(samples<1.0&&normal[index].a<=0.0){
+let tile=max(grid,vec2u(1u));let nearest=min(((pixel+tile/2u)/tile)*tile,((size-vec2u(1u))/tile)*tile);
+let candidate=nearest.x+nearest.y*size.x;if(normal[candidate].a>0.0){index=candidate;}
+}
+if(normal[index].a<=0.0){return vec4f(0.0);}
+let s=shadow[index];let total=max(normal[index].a,1.0);
+return vec4f(s.mask.rgb/total,s.mask.r/total*select(0.0,s.moments.z/s.moments.w,s.moments.w>0.0));
+}
+`;
 //# sourceMappingURL=WGPURayTraceOutput.js.map

package/lib/renderer/WGPURayTraceOutput.js.map

@@ -1 +1 @@
-{"version":3,"file":"WGPURayTraceOutput.js","sourceRoot":"","sources":["../../src/renderer/WGPURayTraceOutput.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,oBAAoB,EAAE,MAAM,sBAAsB,CAAC;AAC5D,OAAO,EAAE,YAAY,EAAE,MAAM,gBAAgB,CAAC;AAC9C,OAAO,6BAA6B,MAAM,iDAAiD,CAAC;AAC5F,OAAO,EAAE,8BAA8B,EAAE,MAAM,uBAAuB,CAAC;AAEvE,MAAM,UAAU,iCAAiC,CAAC,MAAM,GAAG,KAAK,EAAE,OAAO,GAAG,IAAI;IAC9E,OAAO,UAAU,CAAC;;;;EAIlB,6BAA6B;;;;;;;;;;EAU3B,MAAM,CAAC,CAAC,CAAC;sCAC2B,CAAC,CAAC,CAAC;;4BAEb,OAAO,CAAC,CAAC,CAAC,wCAAwC,CAAC,CAAC,CAAC,EAAE,EAAE;;;;;;;;;;;;;;;;;;;;;;;;EAwBjF,MAAM,CAAC,CAAC,CAAC;uCAC4B,CAAC,CAAC,CAAC;;;EAGxC,OAAO,CAAC,CAAC,CAAC,oCAAoC,CAAC,CAAC,CAAC,EAAE,EAAE;;;;;;;;;;;;EAYvD,MAAM,CAAC,CAAC,CAAC;;;;;;;;;;;CAWV,CAAC,CAAC,CAAC;;;;;;;;;;;;;EAaF,OAAO,CAAC,CAAC,CAAC,+DAA+D,CAAC,CAAC,CAAC,EAAE;CAC/E;;;CAGF,CAAC;AACF,CAAC;AAYD,MAAM,OAAO,sBAAsB;IAqBe;IApBxC,OAAO,CAAY;IAEnB,SAAS,CAAqB;IAC9B,eAAe,CAAqB;IACpC,kBAAkB,CAAc;IAChC,yBAAyB,CAAc;IACvC,aAAa,CAAc;IAC3B,cAAc,CAAc;IAC5B,cAAc,CAAc;IAC5B,qBAAqB,CAAc;IACnC,iBAAiB,CAAgB;IAEjC,gBAAgB,CAAC;IAEjB,MAAM,GAAG,CAAC,CAAC;IACX,OAAO,GAAG,CAAC,CAAC;IAEZ,WAAW,GAAG,KAAK,CAAC;IACpB,oBAAoB,GAAG,IAAI,CAAC;IAEpC,YAAY,MAAiB,EAAmB,WAAwB;QAAxB,gBAAW,GAAX,WAAW,CAAa;QACtE,IAAI,CAAC,OAAO,GAAG,MAAM,CAAC;QACtB,IAAI,CAAC,gBAAgB,GAAG;YACtB,WAAW,EAAE,MAAM,CAAC,YAAY,CAAC;gBAC/B,IAAI,EAAE,CAAC;gBACP,KAAK,EAAE,cAAc,CAAC,OAAO,GAAG,cAAc,CAAC,QAAQ;aACxD,CAAC;YACF,cAAc,EAAE,MAAM,CAAC,YAAY,CAAC;gBAClC,IAAI,EAAE,CAAC;gBACP,KAAK,EAAE,cAAc,CAAC,OAAO,GAAG,cAAc,CAAC,QAAQ;aACxD,CAAC;SACH,CAAC;IACJ,CAAC;IAEgB,gBAAgB,GAAG,IAAI,GAAG,EAAc,CAAC;IAC1D,YAAY,CAAC,OAAmB;QAC9B,MAAM,KAAK,GAAG,CAAC,OAAO,EAAE,QAAQ,EAAE,QAAQ,EAAE,eAAe,EAAE,YAAY,EAAE,mBAAmB,CAAU,CAAC;QACzG,IAAI,KAAK,CAAC,IAAI,CAAC,KAAK,CAAC,EAAE,CAAC,IAAI,CAAC,SAAS,CAAC,KAAK,CAAC,KAAK,OAAO,CAAC;YAAE,IAAI,CAAC,gBAAgB,CAAC,GAAG,CAAC,OAAO,CAAC,CAAC;IACjG,CAAC;IACD,aAAa,CAAC,OAAmB;QAC/B,IAAI,CAAC,IAAI,CAAC,gBAAgB,CAAC,MAAM,CAAC,OAAO,CAAC;YAAE,OAAO;QACnD,MAAM,KAAK,GAAG,CAAC,OAAO,EAAE,QAAQ,EAAE,QAAQ,EAAE,eAAe,EAAE,YAAY,EAAE,mBAAmB,CAAU,CAAC;QACzG,IAAI,CAAC,KAAK,CAAC,IAAI,CAAC,KAAK,CAAC,EAAE,CAAC,IAAI,CAAC,SAAS,CAAC,KAAK,CAAC,KAAK,OAAO,CAAC,EAAE,CAAC;YAC5D,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,KAAqB;QAE7B,OAAO,IAAI,CAAC,IAAI,KAAK,SAAS,CAAC,IAAI,IAAI,CAAC,WAAY,CAAC;IACvD,CAAC;IAEO,eAAe;QACrB,MAAM,EAAE,OAAO,EAAE,MAAM,EAAE,MAAM,EAAE,KAAK,EAAE,OAAO,EAAE,MAAM,EAAE,GAAG,IAAI,CAAC;QACjE,MAAM,iBAAiB,GAAyB;YAC9C,IAAI,EAAE;gBACJ,KAAK;gBACL,MAAM;gBACN,kBAAkB,EAAE,CAAC;aACtB;YACD,MAAM,EAAE,YAAY;YACpB,KAAK,EAAE,eAAe,CAAC,iBAAiB,GAAG,eAAe,CAAC,eAAe;gBACxE,eAAe,CAAC,QAAQ;SAC3B,CAAC;QACF,IAAI,CAAC,gBAAgB,EAAE,CAAC;QACxB,IAAI,CAAC,aAAa,GAAG,MAAM,CAAC,aAAa,CAAC;YACxC,GAAG,iBAAiB;YACpB,KAAK,EAAE,iBAAiB;YACxB,MAAM,EAAE,aAAa;YACrB,KAAK,EACH,eAAe,CAAC,iBAAiB;gBACjC,eAAe,CAAC,eAAe;gBAC/B,eAAe,CAAC,QAAQ;SAC3B,CAAC,CAAC;QACH,IAAI,CAAC,cAAc,GAAG,MAAM,CAAC,aAAa,CAAC;YACzC,GAAG,iBAAiB;YACpB,KAAK,EAAE,kBAAkB;YACzB,KAAK,EAAE,iBAAiB,CAAC,KAAK,GAAG,eAAe,CAAC,QAAQ;SAC1D,CAAC,CAAC;QACH,IAAI,CAAC,cAAc,GAAG,MAAM,CAAC,aAAa,CAAC;YACzC,GAAG,iBAAiB;YACpB,KAAK,EAAE,kBAAkB;SAC1B,CAAC,CAAC;QACH,IAAI,IAAI,CAAC,WAAW,EAAE,CAAC;YACvB,IAAI,CAAC,kBAAkB,GAAG,MAAM,CAAC,aAAa,CAAC;gBAC7C,GAAG,iBAAiB;gBACpB,KAAK,EAAE,sBAAsB;gBAC7B,MAAM,EAAE,aAAa;aACtB,CAAC,CAAC;YACH,IAAI,CAAC,yBAAyB,GAAG,MAAM,CAAC,aAAa,CAAC;gBACpD,GAAG,iBAAiB;gBACpB,KAAK,EAAE,8BAA8B;gBACrC,MAAM,EAAE,aAAa;aACtB,CAAC,CAAC;YACH,IAAI,CAAC,qBAAqB,GAAG,MAAM,CAAC,aAAa,CAAC;gBAChD,GAAG,iBAAiB;gBACpB,KAAK,EAAE,0BAA0B;gBACjC,MAAM,EAAE,aAAa;aACtB,CAAC,CAAC;QACH,CAAC;aAAM,CAAC;YACN,IAAI,CAAC,kBAAkB,GAAG,IAAI,CAAC,yBAAyB,GAAG,IAAI,CAAC,qBAAqB,GAAG,SAAU,CAAC;QACrG,CAAC;IACH,CAAC;IAEO,iBAAiB;QACvB,MAAM,MAAM,GAAG,IAAI,CAAC,OAAO,CAAC;QAC5B,IAAI,CAAC,iBAAiB,GAAG,MAAM,CAAC,eAAe,CAAC;YAC9C,MAAM,EAAE,IAAI,CAAC,SAAS,CAAC,kBAAkB,CAAC,CAAC,CAAC;YAC5C,OAAO,EAAE;gBACP;oBACE,OAAO,EAAE,CAAC;oBACV,QAAQ,EAAE;wBACR,MAAM,EAAE,IAAI,CAAC,gBAAgB,CAAC,WAAW;qBAC1C;iBACF;gBACD;oBACE,OAAO,EAAE,CAAC;oBACV,QAAQ,EAAE;wBACR,MAAM,EAAE,IAAI,CAAC,gBAAgB,CAAC,cAAc;qBAC7C;iBACF;aACF;SACF,CAAC,CAAC;IACL,CAAC;IAEO,eAAe;QACrB,IAAI,CAAC,IAAI,CAAC,oBAAoB,EAAE,CAAC;YAC/B,OAAO;QACT,CAAC;QACD,IAAI,CAAC,oBAAoB,GAAG,KAAK,CAAC;QAClC,MAAM,MAAM,GAAG,IAAI,CAAC,OAAO,CAAC;QAE5B,IAAI,CAAC,SAAS,GAAG,oBAAoB,CAAC,MAAM,CAAC,CAAC,oBAAoB,CAAC;YACjE,KAAK,EAAE,gBAAgB;YACvB,MAAM,EAAE,MAAM;YACd,MAAM,EAAE;gBACN,MAAM,EAAE,oBAAoB,CAAC,MAAM,CAAC,CAAC,kBAAkB,CAAC;oBACtD,IAAI,EAAE,8BAA8B;iBACrC,CAAC;gBACF,UAAU,EAAE,MAAM;aACnB;YACD,QAAQ,EAAE;gBACR,MAAM,EAAE,oBAAoB,CAAC,MAAM,CAAC,CAAC,kBAAkB,CAAC;oBACtD,IAAI,EAAE,iCAAiC,CAAC,KAAK,EAAE,IAAI,CAAC,WAAW,CAAC;iBACjE,CAAC;gBACF,UAAU,EAAE,MAAM;gBAClB,OAAO,EAAE;oBACP;wBACE,MAAM,EAAE,aAAa;qBACtB;oBACD;wBACE,MAAM,EAAE,YAAY;qBACrB;oBACD;wBACE,MAAM,EAAE,YAAY;qBACrB;oBACD,GAAG,CAAC,IAAI,CAAC,WAAW,CAAC,CAAC,CAAC,CAAC,EAAC,MAAM,EAAE,aAAsB,EAAC,CAAC,CAAC,CAAC,CAAC,EAAE,CAAC;iBAChE;gBACD,SAAS,EAAE;oBACT,WAAW,EAAE,IAAI,CAAC,MAAM;oBACxB,YAAY,EAAE,IAAI,CAAC,OAAO;iBAC3B;aACF;SACF,CAAC,CAAC;QAEH,IAAI,IAAI,CAAC,WAAW;YAAE,IAAI,CAAC,eAAe,GAAG,oBAAoB,CAAC,MAAM,CAAC,CAAC,oBAAoB,CAAC;gBAC7F,KAAK,EAAE,gBAAgB;gBACvB,MAAM,EAAE,MAAM;gBACd,MAAM,EAAE;oBACN,MAAM,EAAE,oBAAoB,CAAC,MAAM,CAAC,CAAC,kBAAkB,CAAC,EAAE,IAAI,EAAE,8BAA8B,EAAE,CAAC;oBACjG,UAAU,EAAE,MAAM;iBACnB;gBACD,QAAQ,EAAE;oBACR,MAAM,EAAE,oBAAoB,CAAC,MAAM,CAAC,CAAC,kBAAkB,CAAC,EAAE,IAAI,EAAE,iCAAiC,CAAC,IAAI,CAAC,EAAE,CAAC;oBAC1G,UAAU,EAAE,MAAM;oBAClB,OAAO,EAAE,CAAC,EAAE,MAAM,EAAE,aAAa,EAAE,EAAE,EAAE,MAAM,EAAE,aAAa,EAAE,CAAC;oBAC/D,SAAS,EAAE,EAAE,WAAW,EAAE,IAAI,CAAC,MAAM,EAAE,YAAY,EAAE,IAAI,CAAC,OAAO,EAAE;iBACpE;aACF,CAAC,CAAC;QACH,IAAI,CAAC,iBAAiB,EAAE,CAAC;IAC3B,CAAC;IAED,OAAO,CAAC,KAAa,EAAE,MAAc;QACnC,IACE,IAAI,CAAC,MAAM,KAAK,KAAK;YACrB,IAAI,CAAC,OAAO,KAAK,MAAM;YACvB,CAAC,IAAI,CAAC,aAAa,EACnB,CAAC;YACD,IAAI,CAAC,MAAM,GAAG,KAAK,CAAC;YACpB,IAAI,CAAC,OAAO,GAAG,MAAM,CAAC;YACtB,IAAI,CAAC,eAAe,EAAE,CAAC;YACvB,IAAI,CAAC,oBAAoB,GAAG,IAAI,CAAC;QACnC,CAAC;IACH,CAAC;IAED,cAAc,CAAC,WAAmB;QAChC,IAAI,CAAC,OAAO,CAAC,KAAK,CAAC,WAAW,CAC5B,IAAI,CAAC,gBAAgB,CAAC,WAAW,EACjC,CAAC,EACD,IAAI,YAAY,CAAC,CAAC,WAAW,CAAC,CAAC,CAChC,CAAC;IACJ,CAAC;IAED,iBAAiB,CAAC,OAAe,EAAE,IAAY;QAC7C,IAAI,CAAC,OAAO,CAAC,KAAK,CAAC,WAAW,CAC5B,IAAI,CAAC,gBAAgB,CAAC,cAAc,EACpC,CAAC,EACD,IAAI,WAAW,CAAC,CAAC,OAAO,EAAE,IAAI,CAAC,CAAC,CACjC,CAAC;IACJ,CAAC;IAED,UAAU,CACR,OAA0B,EAC1B,YAA6B,EAC7B,eAAe,GAAG,KAAK;QAEvB,IAAI,IAAI,CAAC,WAAW,KAAK,eAAe,EAAE,CAAC;YACzC,IAAI,CAAC,WAAW,GAAG,eAAe,CAAC;YAAC,IAAI,CAAC,eAAe,EAAE,CAAC;YAAC,IAAI,CAAC,oBAAoB,GAAG,IAAI,CAAC;QAC/F,CAAC;QACD,IAAI,CAAC,eAAe,EAAE,CAAC;QACvB,IAAI,CAAC,iBAAiB,CAAC,OAAO,EAAE,YAAY,EAAE,KAAK,CAAC,CAAC;QACrD,IAAI,eAAe;YAAE,IAAI,CAAC,iBAAiB,CAAC,OAAO,EAAE,YAAY,EAAE,IAAI,CAAC,CAAC;IAC3E,CAAC;IAEO,iBAAiB,CACvB,OAA0B,EAC1B,YAA6B,EAC7B,MAAe;QAEf,MAAM,QAAQ,GAAG,MAAM,CAAC,CAAC,CAAC,IAAI,CAAC,eAAe,CAAC,CAAC,CAAC,IAAI,CAAC,SAAS,CAAC;QAChE,MAAM,QAAQ,GAAG,MAAM;YACrB,CAAC,CAAC,CAAC,IAAI,CAAC,kBAAkB,EAAE,IAAI,CAAC,yBAAyB,CAAC;YAC3D,CAAC,CAAC,CAAC,IAAI,CAAC,aAAa,EAAE,IAAI,CAAC,cAAc,EAAE,IAAI,CAAC,cAAc,EAAE,GAAG,CAAC,IAAI,CAAC,WAAW,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,qBAAqB,CAAC,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC;QAC9H,MAAM,IAAI,GAAG,OAAO,CAAC,eAAe,CAAC;YACnC,gBAAgB,EAAE,QAAQ,CAAC,GAAG,CAAC,CAAC,OAAO,EAAE,EAAE,CAAC,CAAC;gBAC3C,IAAI,EAAE,OAAO,CAAC,UAAU,EAAE;gBAC1B,SAAS,EAAE,EAAE,CAAC,EAAE,CAAC,EAAE,CAAC,EAAE,CAAC,EAAE,CAAC,EAAE,CAAC,EAAE,CAAC,EAAE,CAAC,EAAE;gBACrC,OAAO,EAAE,OAAO;gBAChB,MAAM,EAAE,OAAO;aAChB,CAAC,CAAC;SACJ,CAAC,CAAC;QACH,IAAI,CAAC,WAAW,CAAC,QAAQ,CAAC,CAAC;QAE3B,IAAI,CAAC,YAAY,CACf,CAAC,EACD,IAAI,CAAC,OAAO,CAAC,eAAe,CAAC;YAC3B,KAAK,EAAE,gBAAgB;YACvB,MAAM,EAAE,QAAQ,CAAC,kBAAkB,CAAC,CAAC,CAAC;YACtC,OAAO,EAAE;gBACP,YAAY,CAAC,KAAK;gBAClB,YAAY,CAAC,MAAM;gBACnB,YAAY,CAAC,MAAM;gBACnB,YAAY,CAAC,aAAa;aAC3B,CAAC,GAAG,CAAC,CAAC,MAAM,EAAE,GAAG,EAAE,EAAE,CAAC,CAAC;gBACtB,OAAO,EAAE,GAAG;gBACZ,QAAQ,EAAE;oBACR,MAAM;oBACN,GAAG,CAAC,GAAG,KAAK,CAAC,CAAC,CAAC,CAAC,EAAE,MAAM,EAAE,YAAY,CAAC,mBAAmB,IAAI,CAAC,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;iBACxE;aACF,CAAC,CAAC,CAAC,MAAM,CAAC,KAAK,CAAC,EAAE,CAAC,KAAK,CAAC,OAAO,GAAG,CAAC,IAAI,IAAI,CAAC,WAAW,IAAI,MAAM,CAAC;SACrE,CAAC,CACH,CAAC;QACF,IAAI,CAAC,YAAY,CAAC,CAAC,EAAE,MAAM,CAAC,CAAC,CAAC,IAAI,CAAC,OAAO,CAAC,eAAe,CAAC;YACzD,MAAM,EAAE,QAAQ,CAAC,kBAAkB,CAAC,CAAC,CAAC;YACtC,OAAO,EAAE,MAAM,CAAC,MAAM,CAAC,IAAI,CAAC,gBAAgB,CAAC,CAAC,GAAG,CAAC,CAAC,MAAM,EAAE,OAAO,EAAE,EAAE,CAAC,CAAC;gBACtE,OAAO,EAAE,QAAQ,EAAE,EAAE,MAAM,EAAE;aAC9B,CAAC,CAAC;SACJ,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,iBAAiB,CAAC,CAAC;QAE7B,IAAI,CAAC,IAAI,CAAC,CAAC,EAAE,CAAC,EAAE,CAAC,EAAE,CAAC,CAAC,CAAC;QACtB,IAAI,CAAC,GAAG,EAAE,CAAC;IAEb,CAAC;IAEO,gBAAgB;QACtB,IAAI,CAAC,IAAI,CAAC,aAAa;YAAE,OAAO;QAEhC,oEAAoE;QACpE,uEAAuE;QACvE,0EAA0E;QAC1E,0EAA0E;QAC1E,2DAA2D;QAC3D,MAAM,QAAQ,GAAG;YACf,IAAI,CAAC,aAAa;YAClB,IAAI,CAAC,cAAc;YACnB,IAAI,CAAC,cAAc;YACnB,IAAI,CAAC,qBAAqB;YAC1B,IAAI,CAAC,kBAAkB;YACvB,IAAI,CAAC,yBAAyB;SAC/B,CAAC,MAAM,CAAC,CAAC,OAAO,EAAyB,EAAE,CAAC,CAAC,CAAC,OAAO,IAAI,CAAC,IAAI,CAAC,gBAAgB,CAAC,GAAG,CAAC,OAAO,CAAC,CAAC,CAAC;QAC/F,KAAK,IAAI,CAAC,OAAO,CAAC,KAAK;aACpB,mBAAmB,EAAE;aACrB,KAAK,CAAC,GAAG,EAAE,CAAC,SAAS,CAAC;aACtB,IAAI,CAAC,GAAG,EAAE,CAAC,QAAQ,CAAC,OAAO,CAAC,CAAC,OAAO,EAAE,EAAE,CAAC,OAAO,CAAC,OAAO,EAAE,CAAC,CAAC,CAAC;IAClE,CAAC;IAED,cAAc;QACZ,MAAM,QAAQ,GAAG,CAAC,OAAO,EAAE,QAAQ,EAAE,QAAQ,EAAE,eAAe,EAAE,YAAY,EAAE,mBAAmB,CAAU,CAAC;QAC5G,OAAO,MAAM,CAAC,MAAM,CAAC,IAAI,CAAC,gBAAgB,CAAC,CAAC,MAAM,CAAC,CAAC,GAAG,EAAE,MAAM,EAAE,EAAE,CAAC,GAAG,GAAG,MAAM,CAAC,IAAI,EAAE,CAAC,CAAC;cACrF,KAAK,CAAC,IAAI,CAAC,IAAI,GAAG,CAAC,CAAC,GAAG,QAAQ,CAAC,GAAG,CAAC,KAAK,CAAC,EAAE,CAAC,IAAI,CAAC,IAAI,KAAK,SAAS,CAAC,CAAC,EAAE,GAAG,IAAI,CAAC,gBAAgB,CAAC,CAAC,CAAC;iBAClG,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,QAAQ,GAAG,KAAK,CAAC,IAAI,CAAC,IAAI,CAAC,gBAAgB,CAAC,CAAC;QACnD,KAAK,IAAI,CAAC,OAAO,CAAC,KAAK,CAAC,mBAAmB,EAAE,CAAC,KAAK,CAAC,GAAG,EAAE,CAAC,SAAS,CAAC,CAAC,IAAI,CAAC,GAAG,EAAE,CAAC,QAAQ,CAAC,OAAO,CAAC,OAAO,CAAC,EAAE,CAAC,OAAO,CAAC,OAAO,EAAE,CAAC,CAAC,CAAC;QAChI,IAAI,CAAC,gBAAgB,EAAE,CAAC;QACxB,IAAI,CAAC,gBAAgB,CAAC,KAAK,EAAE,CAAC;QAC9B,MAAM,CAAC,IAAI,CAAC,IAAI,CAAC,gBAAgB,CAAC,CAAC,OAAO,CAAC,CAAC,GAAG,EAAE,EAAE;YAChD,IAAI,CAAC,gBAAwB,CAAC,GAAG,CAAC,CAAC,OAAO,EAAE,CAAC;QAChD,CAAC,CAAC,CAAC;IACL,CAAC;CACF","sourcesContent":["import { getTargetShaderCache } from './WGPUShaderCache.js';\nimport { textureBytes } from './gpuMemory.js';\nimport shadowCatcherAccumulationWGSL from './wgsl/chunks/shadowCatcherAccumulation.wgsl.js';\nimport { fullScreenQuadVertexShaderWGSL } from './WGPUFullQuadPass.js';\n\nexport function createNormalizeRayTraceOutputWGSL(layers = false, catcher = true) {\n  return /* wgsl */ `\n@group(0) @binding(0) var<storage, read> colorBuffer: array<vec4f>;\n@group(0) @binding(1) var<storage, read> normalBuffer: array<vec4f>;\n@group(0) @binding(2) var<storage, read> albedoBuffer: array<vec4f>;\n${shadowCatcherAccumulationWGSL}\n@group(0) @binding(3) var<storage, read> shadowCatcherBuffer: array<ShadowCatcherAccumulation>;\n\n@group(1) @binding(0) var<uniform> sampleCount: f32;\n@group(1) @binding(1) var<uniform> sampleTileGrid: vec2u;\n\noverride outputWidth: f32;\noverride outputHeight: f32;\n\nstruct FSOutput {\n  ${layers ? `@location(0) foreground: vec4f,\n  @location(1) catcherBackground: vec4f` : `@location(0) color: vec4f,\n  @location(1) normal: vec4f,\n  @location(2) albedo: vec4f${catcher ? ',\\n  @location(3) shadowCatcher: vec4f' : ''}`}\n}\n@fragment\nfn main(@builtin(position) coord: vec4f) -> FSOutput {\n  let size = vec2u(u32(outputWidth), u32(outputHeight));\n  let pixel = vec2u(u32(coord.x), size.y - u32(coord.y) - 1u);\n  var idx = pixel.x + pixel.y * size.x;\n  // Tiles are interleaved pixel grids, not contiguous rectangles. The first\n  // tile samples (0, 0) of every grid cell across the current camera image.\n  // Reconstruct only pixels not yet sampled, until the first sweep completes;\n  // never write these display estimates back into the accumulated ray buffers.\n  if (sampleCount < 1.0 && normalBuffer[idx].a <= 0.0) {\n    let grid = max(sampleTileGrid, vec2u(1u));\n    let nearest = min(\n      ((pixel + grid / 2u) / grid) * grid,\n      ((size - vec2u(1u)) / grid) * grid\n    );\n    let nearestIdx = nearest.x + nearest.y * size.x;\n    if (normalBuffer[nearestIdx].a > 0.0) {\n      idx = nearestIdx;\n    }\n  }\n  var output: FSOutput;\n  if (normalBuffer[idx].a <= 0.0) {\n    ${layers ? `output.foreground = vec4f(0.0);\n    output.catcherBackground = vec4f(0.0);` : `output.color = vec4f(0.0);\n    output.normal = vec4f(0.0);\n    output.albedo = vec4f(0.0);\n    ${catcher ? 'output.shadowCatcher = vec4f(0.0);' : ''}`}\n    return output;\n  }\n  let color = colorBuffer[idx];\n  let normal = normalBuffer[idx];\n  let albedo = albedoBuffer[idx];\n  let hit = albedo.a;\n  // The CPU count covers completed sweeps and is still zero during the first\n  // interleaved sample. Use the actual per-pixel count, including for misses.\n  // Empty pixels must remain finite: NaN alpha can poison float HDR output\n  // even when its canvas uses opaque compositing.\n  let total = max(normal.a, 1.0);\n  ${layers ? `\n  let shadow = shadowCatcherBuffer[idx];\n  let foregroundHit = max(hit - shadow.mask.b, 0.0);\n  // Separated background includes flat primary misses as well as catcher\n  // samples. Neither belongs in the hit-normalized geometry color/alpha.\n  output.foreground = vec4f(\n    (color.rgb - shadow.background.rgb)\n      / select(total, foregroundHit, foregroundHit > 0.0),\n    clamp((color.a - shadow.mask.g) / total, 0.0, 1.0)\n  );\n  output.catcherBackground = shadow.background / total;\n  ` : `output.color = vec4f(\n    // average the color by hit count.\n    // So the antialised edge color will be correct when having transparent background\n    color.rgb / select(total, hit, hit > 0.0),\n    color.a / total\n  );\n\n  output.normal = vec4f(0.0);\n  // Normalize to 0 - 1\n  if (dot(normal.rgb, normal.rgb) > 0.1) {\n    output.normal = vec4f(normal.rgb / total * 0.5 + 0.5, 1.0);\n  }\n  output.albedo = vec4f(albedo.rgb / total, clamp(hit / total, 0.0, 1.0));\n  ${catcher ? 'output.shadowCatcher = shadowCatcherBuffer[idx].mask / total;' : ''}\n  `}\n  return output;\n}\n`;\n}\n\ntype RayTraceOutput = 'color' | 'normal' | 'albedo' | 'shadowCatcher'\n  | 'foreground' | 'catcherBackground';\ntype RayTraceBuffers = {\n  color: GPUBuffer;\n  normal: GPUBuffer;\n  albedo: GPUBuffer;\n  shadowCatcher: GPUBuffer;\n  shadowCatcherOffset?: number;\n};\n\nexport class WGPURayTraceOutputPass {\n  private _device: GPUDevice;\n\n  private _pipeline!: GPURenderPipeline;\n  private _layersPipeline!: GPURenderPipeline;\n  private _foregroundTexture!: GPUTexture;\n  private _catcherBackgroundTexture!: GPUTexture;\n  private _colorTexture!: GPUTexture;\n  private _normalTexture!: GPUTexture;\n  private _albedoTexture!: GPUTexture;\n  private _shadowCatcherTexture!: GPUTexture;\n  private _uniformBindGroup!: GPUBindGroup;\n\n  private _uniformsBuffers;\n\n  private _width = 1;\n  private _height = 1;\n\n  private _hasCatcher = false;\n  private _needsUpdatePipeline = true;\n\n  constructor(device: GPUDevice, private readonly zeroTexture?: GPUTexture) {\n    this._device = device;\n    this._uniformsBuffers = {\n      sampleCount: device.createBuffer({\n        size: 4,\n        usage: GPUBufferUsage.UNIFORM | GPUBufferUsage.COPY_DST\n      }),\n      sampleTileGrid: device.createBuffer({\n        size: 8,\n        usage: GPUBufferUsage.UNIFORM | GPUBufferUsage.COPY_DST\n      })\n    };\n  }\n\n  private readonly _retainedOutputs = new Set<GPUTexture>();\n  retainOutput(texture: GPUTexture) {\n    const owned = ['color', 'normal', 'albedo', 'shadowCatcher', 'foreground', 'catcherBackground'] as const;\n    if (owned.some(label => this.getOutput(label) === texture)) this._retainedOutputs.add(texture);\n  }\n  releaseOutput(texture: GPUTexture) {\n    if (!this._retainedOutputs.delete(texture)) return;\n    const owned = ['color', 'normal', 'albedo', 'shadowCatcher', 'foreground', 'catcherBackground'] as const;\n    if (!owned.some(label => this.getOutput(label) === texture)) {\n      void this._device.queue.onSubmittedWorkDone().catch(() => undefined).then(() => texture.destroy());\n    }\n  }\n\n  getOutput(label: RayTraceOutput): GPUTexture {\n\n    return this[`_${label}Texture`] ?? this.zeroTexture!;\n  }\n\n  private _createTextures() {\n    const { _device: device, _width: width, _height: height } = this;\n    const textureDescriptor: GPUTextureDescriptor = {\n      size: {\n        width,\n        height,\n        depthOrArrayLayers: 1\n      },\n      format: 'rgba8unorm',\n      usage: GPUTextureUsage.RENDER_ATTACHMENT | GPUTextureUsage.TEXTURE_BINDING |\n        GPUTextureUsage.COPY_SRC\n    };\n    this._destroyTextures();\n    this._colorTexture = device.createTexture({\n      ...textureDescriptor,\n      label: 'Ray Trace Color',\n      format: 'rgba16float',\n      usage:\n        GPUTextureUsage.RENDER_ATTACHMENT |\n        GPUTextureUsage.TEXTURE_BINDING |\n        GPUTextureUsage.COPY_SRC\n    });\n    this._normalTexture = device.createTexture({\n      ...textureDescriptor,\n      label: 'Ray Trace Normal',\n      usage: textureDescriptor.usage | GPUTextureUsage.COPY_SRC\n    });\n    this._albedoTexture = device.createTexture({\n      ...textureDescriptor,\n      label: 'Ray Trace Albedo'\n    });\n    if (this._hasCatcher) {\n    this._foregroundTexture = device.createTexture({\n      ...textureDescriptor,\n      label: 'Ray Trace Foreground',\n      format: 'rgba16float'\n    });\n    this._catcherBackgroundTexture = device.createTexture({\n      ...textureDescriptor,\n      label: 'Ray Trace Catcher Background',\n      format: 'rgba16float'\n    });\n    this._shadowCatcherTexture = device.createTexture({\n      ...textureDescriptor,\n      label: 'Ray Trace Shadow Catcher',\n      format: 'rgba16float'\n    });\n    } else {\n      this._foregroundTexture = this._catcherBackgroundTexture = this._shadowCatcherTexture = undefined!;\n    }\n  }\n\n  private _updateBindGroups() {\n    const device = this._device;\n    this._uniformBindGroup = device.createBindGroup({\n      layout: this._pipeline.getBindGroupLayout(1),\n      entries: [\n        {\n          binding: 0,\n          resource: {\n            buffer: this._uniformsBuffers.sampleCount\n          }\n        },\n        {\n          binding: 1,\n          resource: {\n            buffer: this._uniformsBuffers.sampleTileGrid\n          }\n        }\n      ]\n    });\n  }\n\n  private _updatePipeline() {\n    if (!this._needsUpdatePipeline) {\n      return;\n    }\n    this._needsUpdatePipeline = false;\n    const device = this._device;\n\n    this._pipeline = getTargetShaderCache(device).createRenderPipeline({\n      label: 'rayTraceOutput',\n      layout: 'auto',\n      vertex: {\n        module: getTargetShaderCache(device).createShaderModule({\n          code: fullScreenQuadVertexShaderWGSL\n        }),\n        entryPoint: 'main'\n      },\n      fragment: {\n        module: getTargetShaderCache(device).createShaderModule({\n          code: createNormalizeRayTraceOutputWGSL(false, this._hasCatcher)\n        }),\n        entryPoint: 'main',\n        targets: [\n          {\n            format: 'rgba16float'\n          },\n          {\n            format: 'rgba8unorm'\n          },\n          {\n            format: 'rgba8unorm'\n          },\n          ...(this._hasCatcher ? [{format: 'rgba16float' as const}] : [])\n        ],\n        constants: {\n          outputWidth: this._width,\n          outputHeight: this._height\n        }\n      }\n    });\n\n    if (this._hasCatcher) this._layersPipeline = getTargetShaderCache(device).createRenderPipeline({\n      label: 'rayTraceLayers',\n      layout: 'auto',\n      vertex: {\n        module: getTargetShaderCache(device).createShaderModule({ code: fullScreenQuadVertexShaderWGSL }),\n        entryPoint: 'main'\n      },\n      fragment: {\n        module: getTargetShaderCache(device).createShaderModule({ code: createNormalizeRayTraceOutputWGSL(true) }),\n        entryPoint: 'main',\n        targets: [{ format: 'rgba16float' }, { format: 'rgba16float' }],\n        constants: { outputWidth: this._width, outputHeight: this._height }\n      }\n    });\n    this._updateBindGroups();\n  }\n\n  setSize(width: number, height: number) {\n    if (\n      this._width !== width ||\n      this._height !== height ||\n      !this._colorTexture\n    ) {\n      this._width = width;\n      this._height = height;\n      this._createTextures();\n      this._needsUpdatePipeline = true;\n    }\n  }\n\n  setSampleCount(sampleCount: number) {\n    this._device.queue.writeBuffer(\n      this._uniformsBuffers.sampleCount,\n      0,\n      new Float32Array([sampleCount])\n    );\n  }\n\n  setSampleTileGrid(columns: number, rows: number) {\n    this._device.queue.writeBuffer(\n      this._uniformsBuffers.sampleTileGrid,\n      0,\n      new Uint32Array([columns, rows])\n    );\n  }\n\n  createPass(\n    encoder: GPUCommandEncoder,\n    inputBuffers: RayTraceBuffers,\n    separateCatcher = false\n  ) {\n    if (this._hasCatcher !== separateCatcher) {\n      this._hasCatcher = separateCatcher; this._createTextures(); this._needsUpdatePipeline = true;\n    }\n    this._updatePipeline();\n    this._createOutputPass(encoder, inputBuffers, false);\n    if (separateCatcher) this._createOutputPass(encoder, inputBuffers, true);\n  }\n\n  private _createOutputPass(\n    encoder: GPUCommandEncoder,\n    inputBuffers: RayTraceBuffers,\n    layers: boolean\n  ) {\n    const pipeline = layers ? this._layersPipeline : this._pipeline;\n    const textures = layers\n      ? [this._foregroundTexture, this._catcherBackgroundTexture]\n      : [this._colorTexture, this._normalTexture, this._albedoTexture, ...(this._hasCatcher ? [this._shadowCatcherTexture] : [])];\n    const pass = encoder.beginRenderPass({\n      colorAttachments: textures.map((texture) => ({\n        view: texture.createView(),\n        loadValue: { r: 0, g: 0, b: 0, a: 0 },\n        storeOp: 'store',\n        loadOp: 'clear'\n      }))\n    });\n    pass.setPipeline(pipeline);\n\n    pass.setBindGroup(\n      0,\n      this._device.createBindGroup({\n        label: 'rayTraceOutput',\n        layout: pipeline.getBindGroupLayout(0),\n        entries: [\n          inputBuffers.color,\n          inputBuffers.normal,\n          inputBuffers.albedo,\n          inputBuffers.shadowCatcher\n        ].map((buffer, idx) => ({\n          binding: idx,\n          resource: {\n            buffer,\n            ...(idx === 3 ? { offset: inputBuffers.shadowCatcherOffset ?? 0 } : {})\n          }\n        })).filter(entry => entry.binding < 3 || this._hasCatcher || layers)\n      })\n    );\n    pass.setBindGroup(1, layers ? this._device.createBindGroup({\n      layout: pipeline.getBindGroupLayout(1),\n      entries: Object.values(this._uniformsBuffers).map((buffer, binding) => ({\n        binding, resource: { buffer }\n      }))\n    }) : this._uniformBindGroup);\n\n    pass.draw(6, 1, 0, 0);\n    pass.end();\n\n  }\n\n  private _destroyTextures() {\n    if (!this._colorTexture) return;\n\n    // These textures are also OIDN inputs. A resolution or preview-mode\n    // change can replace them immediately after a denoising command buffer\n    // was submitted. Keep the old resources alive until all work submitted so\n    // far has completed; destroying them synchronously can make Dawn validate\n    // the pending submit against an already-destroyed texture.\n    const textures = [\n      this._colorTexture,\n      this._normalTexture,\n      this._albedoTexture,\n      this._shadowCatcherTexture,\n      this._foregroundTexture,\n      this._catcherBackgroundTexture\n    ].filter((texture): texture is GPUTexture => !!texture && !this._retainedOutputs.has(texture));\n    void this._device.queue\n      .onSubmittedWorkDone()\n      .catch(() => undefined)\n      .then(() => textures.forEach((texture) => texture.destroy()));\n  }\n\n  getMemoryBytes() {\n    const textures = ['color', 'normal', 'albedo', 'shadowCatcher', 'foreground', 'catcherBackground'] as const;\n    return Object.values(this._uniformsBuffers).reduce((sum, buffer) => sum + buffer.size, 0)\n      + Array.from(new Set([...textures.map(label => this[`_${label}Texture`]), ...this._retainedOutputs]))\n        .reduce((sum, texture) => sum + textureBytes(texture), 0);\n  }\n\n  dispose() {\n    const retained = Array.from(this._retainedOutputs);\n    void this._device.queue.onSubmittedWorkDone().catch(() => undefined).then(() => retained.forEach(texture => texture.destroy()));\n    this._destroyTextures();\n    this._retainedOutputs.clear();\n    Object.keys(this._uniformsBuffers).forEach((key) => {\n      (this._uniformsBuffers as any)[key].destroy();\n    });\n  }\n}\n"]}
+{"version":3,"file":"WGPURayTraceOutput.js","sourceRoot":"","sources":["../../src/renderer/WGPURayTraceOutput.ts"],"names":[],"mappings":"AAAA,OAAO,EAAC,cAAc,EAAC,MAAM,6BAA6B,CAAC;AAC3D,OAAO,EAAE,oBAAoB,EAAE,MAAM,sBAAsB,CAAC;AAC5D,OAAO,EAAE,YAAY,EAAE,MAAM,gBAAgB,CAAC;AAC9C,OAAO,6BAA6B,MAAM,iDAAiD,CAAC;AAC5F,OAAO,EAAE,8BAA8B,EAAE,MAAM,uBAAuB,CAAC;AAEvE,MAAM,UAAU,iCAAiC,CAAC,MAAM,GAAG,KAAK,EAAE,OAAO,GAAG,IAAI,EAAE,QAAQ,GAAG,KAAK;IAChG,OAAO,UAAU,CAAC;EAClB,cAAc;;;;EAId,6BAA6B;;;EAG7B,QAAQ,CAAC,CAAC,CAAC,UAAU,CAAC;CACvB,CAAC,CAAC,CAAC,EAAE;;;;;;;EAOF,MAAM,CAAC,CAAC,CAAC;sCAC2B,CAAC,CAAC,CAAC;;4BAEb,OAAO,CAAC,CAAC,CAAC,wCAAwC,CAAC,CAAC,CAAC,EAAE,EAAE;;;;;;;;;;;;;;;;;;;;;;;;EAwBjF,MAAM,CAAC,CAAC,CAAC;uCAC4B,CAAC,CAAC,CAAC;;;EAGxC,OAAO,CAAC,CAAC,CAAC,oCAAoC,CAAC,CAAC,CAAC,EAAE,EAAE;;;;;;;;;;;;EAYvD,MAAM,CAAC,CAAC,CAAC;;;;;;;;;;;CAWV,CAAC,CAAC,CAAC;;;;;;;;;;;;;;;EAeF,OAAO,CAAC,CAAC,CAAC;;;;;mGAKyF,CAAC,CAAC,CAAC,EAAE;CACzG;EACD,QAAQ,CAAC,CAAC,CAAC,UAAU,CAAC;EACpB,MAAM,CAAC,CAAC,CAAC,oFAAoF,CAAC,CAAC,CAAC,qEAAqE;CACxK,CAAC,CAAC,CAAC,EAAE;;CAEL,CAAC;AACF,CAAC;AAcD,MAAM,OAAO,sBAAsB;IA2Be;IA1BxC,OAAO,CAAY;IAEnB,SAAS,CAAqB;IAC9B,eAAe,CAAqB;IACpC,kBAAkB,CAAc;IAChC,yBAAyB,CAAc;IACvC,aAAa,CAAc;IAC3B,cAAc,CAAc;IAC5B,cAAc,CAAc;IAC5B,qBAAqB,CAAc;IACnC,iBAAiB,CAAgB;IAEjC,gBAAgB,CAAC;IAEjB,MAAM,GAAG,CAAC,CAAC;IACX,OAAO,GAAG,CAAC,CAAC;IAEZ,WAAW,GAAG,KAAK,CAAC;IACpB,YAAY,GAAG,KAAK,CAAC;IACrB,WAAW,GAAC,KAAK,CAAC;IAClB,aAAa,CAAoB;IACjC,UAAU,CAAa;IACvB,eAAe,CAAa;IAC5B,gBAAgB,CAAoB;IACpC,oBAAoB,GAAG,IAAI,CAAC;IAEpC,YAAY,MAAiB,EAAmB,WAAwB;QAAxB,gBAAW,GAAX,WAAW,CAAa;QACtE,IAAI,CAAC,OAAO,GAAG,MAAM,CAAC;QACtB,IAAI,CAAC,gBAAgB,GAAG;YACtB,WAAW,EAAE,MAAM,CAAC,YAAY,CAAC;gBAC/B,IAAI,EAAE,CAAC;gBACP,KAAK,EAAE,cAAc,CAAC,OAAO,GAAG,cAAc,CAAC,QAAQ;aACxD,CAAC;YACF,cAAc,EAAE,MAAM,CAAC,YAAY,CAAC;gBAClC,IAAI,EAAE,CAAC;gBACP,KAAK,EAAE,cAAc,CAAC,OAAO,GAAG,cAAc,CAAC,QAAQ;aACxD,CAAC;SACH,CAAC;IACJ,CAAC;IAEgB,gBAAgB,GAAG,IAAI,GAAG,EAAc,CAAC;IAC1D,YAAY,CAAC,OAAmB;QAC9B,MAAM,KAAK,GAAG,CAAC,OAAO,EAAE,QAAQ,EAAE,QAAQ,EAAE,eAAe,EAAE,YAAY,EAAE,mBAAmB,CAAU,CAAC;QACzG,IAAI,KAAK,CAAC,IAAI,CAAC,KAAK,CAAC,EAAE,CAAC,IAAI,CAAC,SAAS,CAAC,KAAK,CAAC,KAAK,OAAO,CAAC;YAAE,IAAI,CAAC,gBAAgB,CAAC,GAAG,CAAC,OAAO,CAAC,CAAC;IACjG,CAAC;IACD,aAAa,CAAC,OAAmB;QAC/B,IAAI,CAAC,IAAI,CAAC,gBAAgB,CAAC,MAAM,CAAC,OAAO,CAAC;YAAE,OAAO;QACnD,MAAM,KAAK,GAAG,CAAC,OAAO,EAAE,QAAQ,EAAE,QAAQ,EAAE,eAAe,EAAE,YAAY,EAAE,mBAAmB,CAAU,CAAC;QACzG,IAAI,CAAC,KAAK,CAAC,IAAI,CAAC,KAAK,CAAC,EAAE,CAAC,IAAI,CAAC,SAAS,CAAC,KAAK,CAAC,KAAK,OAAO,CAAC,EAAE,CAAC;YAC5D,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,KAAqB;QAE7B,IAAG,IAAI,CAAC,YAAY,IAAE,KAAK,KAAG,OAAO,IAAE,IAAI,CAAC,UAAU;YAAC,OAAO,IAAI,CAAC,UAAU,CAAC;QAC9E,IAAG,IAAI,CAAC,YAAY,IAAE,KAAK,KAAG,YAAY,IAAE,IAAI,CAAC,eAAe;YAAC,OAAO,IAAI,CAAC,eAAe,CAAC;QAC7F,OAAO,IAAI,CAAC,IAAI,KAAK,SAAS,CAAC,IAAI,IAAI,CAAC,WAAY,CAAC;IACvD,CAAC;IAEO,eAAe;QACrB,MAAM,EAAE,OAAO,EAAE,MAAM,EAAE,MAAM,EAAE,KAAK,EAAE,OAAO,EAAE,MAAM,EAAE,GAAG,IAAI,CAAC;QACjE,MAAM,iBAAiB,GAAyB;YAC9C,IAAI,EAAE;gBACJ,KAAK;gBACL,MAAM;gBACN,kBAAkB,EAAE,CAAC;aACtB;YACD,MAAM,EAAE,YAAY;YACpB,KAAK,EAAE,eAAe,CAAC,iBAAiB,GAAG,eAAe,CAAC,eAAe;gBACxE,eAAe,CAAC,QAAQ;SAC3B,CAAC;QACF,IAAI,CAAC,gBAAgB,EAAE,CAAC;QACxB,IAAI,CAAC,aAAa,GAAG,MAAM,CAAC,aAAa,CAAC;YACxC,GAAG,iBAAiB;YACpB,KAAK,EAAE,iBAAiB;YACxB,MAAM,EAAE,IAAI,CAAC,WAAW,CAAA,CAAC,CAAA,aAAa,CAAA,CAAC,CAAA,aAAa;YACpD,KAAK,EACH,eAAe,CAAC,iBAAiB;gBACjC,eAAe,CAAC,eAAe;gBAC/B,eAAe,CAAC,QAAQ;SAC3B,CAAC,CAAC;QACH,IAAI,CAAC,cAAc,GAAG,MAAM,CAAC,aAAa,CAAC;YACzC,GAAG,iBAAiB;YACpB,KAAK,EAAE,kBAAkB;YACzB,KAAK,EAAE,iBAAiB,CAAC,KAAK,GAAG,eAAe,CAAC,QAAQ;SAC1D,CAAC,CAAC;QACH,IAAI,CAAC,cAAc,GAAG,MAAM,CAAC,aAAa,CAAC;YACzC,GAAG,iBAAiB;YACpB,KAAK,EAAE,kBAAkB;SAC1B,CAAC,CAAC;QACH,IAAI,CAAC,UAAU,GAAC,IAAI,CAAC,eAAe,GAAC,SAAS,CAAC;QAC/C,IAAG,IAAI,CAAC,YAAY,EAAC,CAAC;YACpB,IAAI,CAAC,UAAU,GAAC,MAAM,CAAC,aAAa,CAAC,EAAC,GAAG,iBAAiB,EAAC,KAAK,EAAC,mBAAmB,EAAC,MAAM,EAAC,aAAa,EAAC,CAAC,CAAC;YAC5G,IAAG,IAAI,CAAC,WAAW;gBAAC,IAAI,CAAC,eAAe,GAAC,MAAM,CAAC,aAAa,CAAC,EAAC,GAAG,iBAAiB,EAAC,KAAK,EAAC,wBAAwB,EAAC,MAAM,EAAC,aAAa,EAAC,CAAC,CAAC;QAC5I,CAAC;QACD,IAAI,IAAI,CAAC,WAAW,EAAE,CAAC;YACvB,IAAI,CAAC,kBAAkB,GAAG,MAAM,CAAC,aAAa,CAAC;gBAC7C,GAAG,iBAAiB;gBACpB,KAAK,EAAE,sBAAsB;gBAC7B,MAAM,EAAE,IAAI,CAAC,WAAW,CAAA,CAAC,CAAA,aAAa,CAAA,CAAC,CAAA,aAAa;aACrD,CAAC,CAAC;YACH,IAAI,CAAC,yBAAyB,GAAG,MAAM,CAAC,aAAa,CAAC;gBACpD,GAAG,iBAAiB;gBACpB,KAAK,EAAE,8BAA8B;gBACrC,MAAM,EAAE,aAAa;aACtB,CAAC,CAAC;YACH,IAAI,CAAC,qBAAqB,GAAG,MAAM,CAAC,aAAa,CAAC;gBAChD,GAAG,iBAAiB;gBACpB,KAAK,EAAE,0BAA0B;gBACjC,MAAM,EAAE,aAAa;aACtB,CAAC,CAAC;QACH,CAAC;aAAM,CAAC;YACN,IAAI,CAAC,kBAAkB,GAAG,IAAI,CAAC,yBAAyB,GAAG,IAAI,CAAC,qBAAqB,GAAG,SAAU,CAAC;QACrG,CAAC;IACH,CAAC;IAEO,iBAAiB;QACvB,MAAM,MAAM,GAAG,IAAI,CAAC,OAAO,CAAC;QAC5B,IAAI,CAAC,iBAAiB,GAAG,MAAM,CAAC,eAAe,CAAC;YAC9C,MAAM,EAAE,IAAI,CAAC,SAAS,CAAC,kBAAkB,CAAC,CAAC,CAAC;YAC5C,OAAO,EAAE;gBACP;oBACE,OAAO,EAAE,CAAC;oBACV,QAAQ,EAAE;wBACR,MAAM,EAAE,IAAI,CAAC,gBAAgB,CAAC,WAAW;qBAC1C;iBACF;gBACD;oBACE,OAAO,EAAE,CAAC;oBACV,QAAQ,EAAE;wBACR,MAAM,EAAE,IAAI,CAAC,gBAAgB,CAAC,cAAc;qBAC7C;iBACF;aACF;SACF,CAAC,CAAC;IACL,CAAC;IAEO,eAAe;QACrB,IAAI,CAAC,IAAI,CAAC,oBAAoB,EAAE,CAAC;YAC/B,OAAO;QACT,CAAC;QACD,IAAI,CAAC,oBAAoB,GAAG,KAAK,CAAC;QAClC,MAAM,MAAM,GAAG,IAAI,CAAC,OAAO,CAAC;QAE5B,IAAI,CAAC,SAAS,GAAG,oBAAoB,CAAC,MAAM,CAAC,CAAC,oBAAoB,CAAC;YACjE,KAAK,EAAE,gBAAgB;YACvB,MAAM,EAAE,MAAM;YACd,MAAM,EAAE;gBACN,MAAM,EAAE,oBAAoB,CAAC,MAAM,CAAC,CAAC,kBAAkB,CAAC;oBACtD,IAAI,EAAE,8BAA8B;iBACrC,CAAC;gBACF,UAAU,EAAE,MAAM;aACnB;YACD,QAAQ,EAAE;gBACR,MAAM,EAAE,oBAAoB,CAAC,MAAM,CAAC,CAAC,kBAAkB,CAAC;oBACtD,IAAI,EAAE,iCAAiC,CAAC,KAAK,EAAE,IAAI,CAAC,WAAW,IAAE,CAAC,IAAI,CAAC,WAAW,EAAE,KAAK,CAAC;iBAC3F,CAAC;gBACF,UAAU,EAAE,MAAM;gBAClB,OAAO,EAAE;oBACP;wBACE,MAAM,EAAE,IAAI,CAAC,WAAW,CAAA,CAAC,CAAA,aAAa,CAAA,CAAC,CAAA,aAAa;qBACrD;oBACD;wBACE,MAAM,EAAE,YAAY;qBACrB;oBACD;wBACE,MAAM,EAAE,YAAY;qBACrB;oBACD,GAAG,CAAC,IAAI,CAAC,WAAW,IAAE,CAAC,IAAI,CAAC,WAAW,CAAC,CAAC,CAAC,CAAC,EAAC,MAAM,EAAE,aAAsB,EAAC,CAAC,CAAC,CAAC,CAAC,EAAE,CAAC;iBACnF;gBACD,SAAS,EAAE;oBACT,WAAW,EAAE,IAAI,CAAC,MAAM;oBACxB,YAAY,EAAE,IAAI,CAAC,OAAO;iBAC3B;aACF;SACF,CAAC,CAAC;QAEH,IAAI,IAAI,CAAC,WAAW;YAAE,IAAI,CAAC,eAAe,GAAG,oBAAoB,CAAC,MAAM,CAAC,CAAC,oBAAoB,CAAC;gBAC7F,KAAK,EAAE,gBAAgB;gBACvB,MAAM,EAAE,MAAM;gBACd,MAAM,EAAE;oBACN,MAAM,EAAE,oBAAoB,CAAC,MAAM,CAAC,CAAC,kBAAkB,CAAC,EAAE,IAAI,EAAE,8BAA8B,EAAE,CAAC;oBACjG,UAAU,EAAE,MAAM;iBACnB;gBACD,QAAQ,EAAE;oBACR,MAAM,EAAE,oBAAoB,CAAC,MAAM,CAAC,CAAC,kBAAkB,CAAC,EAAE,IAAI,EAAE,iCAAiC,CAAC,IAAI,EAAE,IAAI,EAAE,KAAK,CAAC,EAAE,CAAC;oBACvH,UAAU,EAAE,MAAM;oBAClB,OAAO,EAAE,CAAC,EAAE,MAAM,EAAE,IAAI,CAAC,WAAW,CAAA,CAAC,CAAA,aAAa,CAAA,CAAC,CAAA,aAAa,EAAE,EAAE,EAAE,MAAM,EAAE,aAAa,EAAE,CAAC;oBAC9F,SAAS,EAAE,EAAE,WAAW,EAAE,IAAI,CAAC,MAAM,EAAE,YAAY,EAAE,IAAI,CAAC,OAAO,EAAE;iBACpE;aACF,CAAC,CAAC;QACH,IAAI,CAAC,iBAAiB,EAAE,CAAC;IAC3B,CAAC;IAED,OAAO,CAAC,KAAa,EAAE,MAAc;QACnC,IACE,IAAI,CAAC,MAAM,KAAK,KAAK;YACrB,IAAI,CAAC,OAAO,KAAK,MAAM;YACvB,CAAC,IAAI,CAAC,aAAa,EACnB,CAAC;YACD,IAAI,CAAC,MAAM,GAAG,KAAK,CAAC;YACpB,IAAI,CAAC,OAAO,GAAG,MAAM,CAAC;YAAA,IAAI,CAAC,aAAa,GAAC,SAAS,CAAC;YACnD,IAAI,CAAC,eAAe,EAAE,CAAC;YACvB,IAAI,CAAC,oBAAoB,GAAG,IAAI,CAAC;QACnC,CAAC;IACH,CAAC;IAED,cAAc,CAAC,WAAmB;QAChC,IAAI,CAAC,OAAO,CAAC,KAAK,CAAC,WAAW,CAC5B,IAAI,CAAC,gBAAgB,CAAC,WAAW,EACjC,CAAC,EACD,IAAI,YAAY,CAAC,CAAC,WAAW,CAAC,CAAC,CAChC,CAAC;IACJ,CAAC;IAED,iBAAiB,CAAC,OAAe,EAAE,IAAY;QAC7C,IAAI,CAAC,OAAO,CAAC,KAAK,CAAC,WAAW,CAC5B,IAAI,CAAC,gBAAgB,CAAC,cAAc,EACpC,CAAC,EACD,IAAI,WAAW,CAAC,CAAC,OAAO,EAAE,IAAI,CAAC,CAAC,CACjC,CAAC;IACJ,CAAC;IAED,UAAU,CACR,OAA0B,EAC1B,YAA6B,EAC7B,eAAe,GAAG,KAAK;QAEvB,IAAG,IAAI,CAAC,WAAW,KAAG,CAAC,CAAC,YAAY,CAAC,OAAO,EAAC,CAAC;YAAA,IAAI,CAAC,WAAW,GAAC,CAAC,CAAC,YAAY,CAAC,OAAO,CAAC;YAAA,IAAI,CAAC,eAAe,EAAE,CAAC;YAAA,IAAI,CAAC,oBAAoB,GAAC,IAAI,CAAC;QAAA,CAAC;QAC7I,IAAI,IAAI,CAAC,YAAY,KAAK,CAAC,CAAC,YAAY,CAAC,OAAO,EAAE,CAAC;YACjD,IAAI,CAAC,YAAY,GAAC,CAAC,CAAC,YAAY,CAAC,OAAO,CAAC;YAAA,IAAI,CAAC,eAAe,EAAE,CAAC;YAAA,IAAI,CAAC,oBAAoB,GAAC,IAAI,CAAC;QACjG,CAAC;QACD,IAAI,IAAI,CAAC,WAAW,KAAK,eAAe,EAAE,CAAC;YACzC,IAAI,CAAC,WAAW,GAAG,eAAe,CAAC;YAAC,IAAI,CAAC,eAAe,EAAE,CAAC;YAAC,IAAI,CAAC,oBAAoB,GAAG,IAAI,CAAC;QAC/F,CAAC;QACD,IAAI,CAAC,eAAe,EAAE,CAAC;QACvB,IAAI,CAAC,iBAAiB,CAAC,OAAO,EAAE,YAAY,EAAE,KAAK,CAAC,CAAC;QACrD,IAAI,eAAe;YAAE,IAAI,CAAC,iBAAiB,CAAC,OAAO,EAAE,YAAY,EAAE,IAAI,CAAC,CAAC;QACzE,IAAG,IAAI,CAAC,WAAW,IAAE,eAAe;YAAC,IAAI,CAAC,YAAY,CAAC,OAAO,EAAC,YAAY,CAAC,CAAC;QAC7E,IAAG,YAAY,CAAC,OAAO,EAAC,CAAC;YACvB,IAAI,CAAC,YAAY,CAAC,OAAO,EAAC,IAAI,CAAC,aAAa,EAAC,IAAI,CAAC,UAAW,EAAC,YAAY,CAAC,OAAO,CAAC,CAAC;YACpF,IAAG,eAAe;gBAAC,IAAI,CAAC,YAAY,CAAC,OAAO,EAAC,IAAI,CAAC,kBAAkB,EAAC,IAAI,CAAC,eAAgB,EAAC,YAAY,CAAC,OAAO,CAAC,CAAC;QACnH,CAAC;IACH,CAAC;IAEO,iBAAiB,CACvB,OAA0B,EAC1B,YAA6B,EAC7B,MAAe;QAEf,MAAM,QAAQ,GAAG,MAAM,CAAC,CAAC,CAAC,IAAI,CAAC,eAAe,CAAC,CAAC,CAAC,IAAI,CAAC,SAAS,CAAC;QAChE,MAAM,QAAQ,GAAG,MAAM;YACrB,CAAC,CAAC,CAAC,IAAI,CAAC,kBAAkB,EAAE,IAAI,CAAC,yBAAyB,CAAC;YAC3D,CAAC,CAAC,CAAC,IAAI,CAAC,aAAa,EAAE,IAAI,CAAC,cAAc,EAAE,IAAI,CAAC,cAAc,EAAE,GAAG,CAAC,IAAI,CAAC,WAAW,IAAE,CAAC,IAAI,CAAC,WAAW,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,qBAAqB,CAAC,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC;QACjJ,MAAM,IAAI,GAAG,OAAO,CAAC,eAAe,CAAC;YACnC,gBAAgB,EAAE,QAAQ,CAAC,GAAG,CAAC,CAAC,OAAO,EAAE,EAAE,CAAC,CAAC;gBAC3C,IAAI,EAAE,OAAO,CAAC,UAAU,EAAE;gBAC1B,SAAS,EAAE,EAAE,CAAC,EAAE,CAAC,EAAE,CAAC,EAAE,CAAC,EAAE,CAAC,EAAE,CAAC,EAAE,CAAC,EAAE,CAAC,EAAE;gBACrC,OAAO,EAAE,OAAO;gBAChB,MAAM,EAAE,OAAO;aAChB,CAAC,CAAC;SACJ,CAAC,CAAC;QACH,IAAI,CAAC,WAAW,CAAC,QAAQ,CAAC,CAAC;QAE3B,IAAI,CAAC,YAAY,CACf,CAAC,EACD,IAAI,CAAC,OAAO,CAAC,eAAe,CAAC;YAC3B,KAAK,EAAE,gBAAgB;YACvB,MAAM,EAAE,QAAQ,CAAC,kBAAkB,CAAC,CAAC,CAAC;YACtC,OAAO,EAAE;gBACP,YAAY,CAAC,KAAK;gBAClB,YAAY,CAAC,MAAM;gBACnB,YAAY,CAAC,MAAM;gBACnB,YAAY,CAAC,aAAa;aAC3B,CAAC,GAAG,CAAC,CAAC,MAAM,EAAE,GAAG,EAAE,EAAE,CAAC,CAAC;gBACtB,OAAO,EAAE,GAAG;gBACZ,QAAQ,EAAE;oBACR,MAAM;oBACN,GAAG,CAAC,GAAG,KAAK,CAAC,CAAC,CAAC,CAAC,EAAE,MAAM,EAAE,YAAY,CAAC,mBAAmB,IAAI,CAAC,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;iBACxE;aACF,CAAC,CAAC,CAAC,MAAM,CAAC,KAAK,CAAC,EAAE,CAAC,KAAK,CAAC,OAAO,GAAG,CAAC,IAAI,CAAC,IAAI,CAAC,WAAW,IAAE,CAAC,IAAI,CAAC,WAAW,CAAC,IAAI,MAAM,CAAC,CAAC,MAAM,CAC9F,EAAE,CACH;SACF,CAAC,CACH,CAAC;QACF,IAAI,CAAC,YAAY,CAAC,CAAC,EAAE,MAAM,CAAC,CAAC,CAAC,IAAI,CAAC,OAAO,CAAC,eAAe,CAAC;YACzD,MAAM,EAAE,QAAQ,CAAC,kBAAkB,CAAC,CAAC,CAAC;YACtC,OAAO,EAAE,MAAM,CAAC,MAAM,CAAC,IAAI,CAAC,gBAAgB,CAAC,CAAC,GAAG,CAAC,CAAC,MAAM,EAAE,OAAO,EAAE,EAAE,CAAC,CAAC;gBACtE,OAAO,EAAE,QAAQ,EAAE,EAAE,MAAM,EAAE;aAC9B,CAAC,CAAC;SACJ,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,iBAAiB,CAAC,CAAC;QAE7B,IAAI,CAAC,IAAI,CAAC,CAAC,EAAE,CAAC,EAAE,CAAC,EAAE,CAAC,CAAC,CAAC;QACtB,IAAI,CAAC,GAAG,EAAE,CAAC;IAEb,CAAC;IAEO,YAAY,CAAC,OAAyB,EAAC,KAAqB;QAClE,MAAM,MAAM,GAAC,IAAI,CAAC,OAAO,CAAC;QAC1B,IAAI,CAAC,aAAa,KAAG,MAAM,CAAC,oBAAoB,CAAC,EAAC,KAAK,EAAC,uBAAuB,EAAC,MAAM,EAAC,MAAM;YAC3F,MAAM,EAAC,EAAC,MAAM,EAAC,MAAM,CAAC,kBAAkB,CAAC,EAAC,IAAI,EAAC,8BAA8B,EAAC,CAAC,EAAC,UAAU,EAAC,MAAM,EAAC;YAClG,QAAQ,EAAC,EAAC,MAAM,EAAC,MAAM,CAAC,kBAAkB,CAAC,EAAC,IAAI,EAAC,qBAAqB,EAAC,CAAC,EAAC,UAAU,EAAC,MAAM,EAAC,SAAS,EAAC,EAAC,WAAW,EAAC,IAAI,CAAC,MAAM,EAAC,YAAY,EAAC,IAAI,CAAC,OAAO,EAAC,EAAC,OAAO,EAAC,CAAC,EAAC,MAAM,EAAC,aAAa,EAAC,CAAC,EAAC,EAAC,CAAC,CAAC;QAC/L,MAAM,IAAI,GAAC,OAAO,CAAC,eAAe,CAAC,EAAC,KAAK,EAAC,uBAAuB,EAAC,gBAAgB,EAAC,CAAC,EAAC,IAAI,EAAC,IAAI,CAAC,qBAAqB,CAAC,UAAU,EAAE,EAAC,MAAM,EAAC,OAAO,EAAC,OAAO,EAAC,OAAO,EAAC,CAAC,EAAC,CAAC,CAAC;QACrK,IAAI,CAAC,WAAW,CAAC,IAAI,CAAC,aAAa,CAAC,CAAC;QAAA,IAAI,CAAC,YAAY,CAAC,CAAC,EAAC,MAAM,CAAC,eAAe,CAAC,EAAC,MAAM,EAAC,IAAI,CAAC,aAAa,CAAC,kBAAkB,CAAC,CAAC,CAAC,EAAC,OAAO,EAAC;gBACvI,EAAC,OAAO,EAAC,CAAC,EAAC,QAAQ,EAAC,EAAC,MAAM,EAAC,KAAK,CAAC,MAAM,EAAC,EAAC,EAAC,EAAC,OAAO,EAAC,CAAC,EAAC,QAAQ,EAAC,EAAC,MAAM,EAAC,KAAK,CAAC,aAAa,EAAC,EAAC;gBAC5F,EAAC,OAAO,EAAC,CAAC,EAAC,QAAQ,EAAC,EAAC,MAAM,EAAC,IAAI,CAAC,gBAAgB,CAAC,cAAc,EAAC,EAAC,EAAC,EAAC,OAAO,EAAC,CAAC,EAAC,QAAQ,EAAC,EAAC,MAAM,EAAC,IAAI,CAAC,gBAAgB,CAAC,WAAW,EAAC,EAAC;aAAC,EAAC,CAAC,CAAC,CAAC;QACzI,IAAI,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;QAAA,IAAI,CAAC,GAAG,EAAE,CAAC;IAC1B,CAAC;IAEO,YAAY,CAAC,OAAyB,EAAC,IAAe,EAAC,MAAiB,EAAC,QAAkB;QACjG,MAAM,MAAM,GAAC,IAAI,CAAC,OAAO,CAAC;QAC1B,IAAI,CAAC,gBAAgB,KAAG,MAAM,CAAC,oBAAoB,CAAC,EAAC,KAAK,EAAC,qBAAqB,EAAC,MAAM,EAAC,MAAM;YAC5F,MAAM,EAAC,EAAC,MAAM,EAAC,MAAM,CAAC,kBAAkB,CAAC,EAAC,IAAI,EAAC,8BAA8B,EAAC,CAAC,EAAC,UAAU,EAAC,MAAM,EAAC;YAClG,QAAQ,EAAC,EAAC,MAAM,EAAC,MAAM,CAAC,kBAAkB,CAAC,EAAC,IAAI,EAAC,oBAAoB,EAAC,CAAC,EAAC,UAAU,EAAC,MAAM,EAAC,OAAO,EAAC,CAAC,EAAC,MAAM,EAAC,aAAa,EAAC,CAAC,EAAC,EAAC,CAAC,CAAC;QAChI,MAAM,IAAI,GAAC,OAAO,CAAC,eAAe,CAAC,EAAC,KAAK,EAAC,qBAAqB,EAAC,gBAAgB,EAAC,CAAC,EAAC,IAAI,EAAC,MAAM,CAAC,UAAU,EAAE,EAAC,MAAM,EAAC,OAAO,EAAC,OAAO,EAAC,OAAO,EAAC,CAAC,EAAC,CAAC,CAAC;QAC/I,IAAI,CAAC,WAAW,CAAC,IAAI,CAAC,gBAAgB,CAAC,CAAC;QAAA,IAAI,CAAC,YAAY,CAAC,CAAC,EAAC,MAAM,CAAC,eAAe,CAAC,EAAC,MAAM,EAAC,IAAI,CAAC,gBAAgB,CAAC,kBAAkB,CAAC,CAAC,CAAC,EAAC,OAAO,EAAC,CAAC,EAAC,OAAO,EAAC,CAAC,EAAC,QAAQ,EAAC,IAAI,CAAC,UAAU,EAAE,EAAC,EAAC,EAAC,OAAO,EAAC,CAAC,EAAC,QAAQ,EAAC,EAAC,MAAM,EAAC,QAAQ,EAAC,EAAC,CAAC,EAAC,CAAC,CAAC,CAAC;QAClO,IAAI,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;QAAA,IAAI,CAAC,GAAG,EAAE,CAAC;IAC1B,CAAC;IAEO,gBAAgB;QACtB,IAAI,CAAC,IAAI,CAAC,aAAa;YAAE,OAAO;QAEhC,oEAAoE;QACpE,uEAAuE;QACvE,0EAA0E;QAC1E,0EAA0E;QAC1E,2DAA2D;QAC3D,MAAM,QAAQ,GAAG;YACf,IAAI,CAAC,UAAU,EAAC,IAAI,CAAC,eAAe;YACpC,IAAI,CAAC,aAAa;YAClB,IAAI,CAAC,cAAc;YACnB,IAAI,CAAC,cAAc;YACnB,IAAI,CAAC,qBAAqB;YAC1B,IAAI,CAAC,kBAAkB;YACvB,IAAI,CAAC,yBAAyB;SAC/B,CAAC,MAAM,CAAC,CAAC,OAAO,EAAyB,EAAE,CAAC,CAAC,CAAC,OAAO,IAAI,CAAC,IAAI,CAAC,gBAAgB,CAAC,GAAG,CAAC,OAAO,CAAC,CAAC,CAAC;QAC/F,KAAK,IAAI,CAAC,OAAO,CAAC,KAAK;aACpB,mBAAmB,EAAE;aACrB,KAAK,CAAC,GAAG,EAAE,CAAC,SAAS,CAAC;aACtB,IAAI,CAAC,GAAG,EAAE,CAAC,QAAQ,CAAC,OAAO,CAAC,CAAC,OAAO,EAAE,EAAE,CAAC,OAAO,CAAC,OAAO,EAAE,CAAC,CAAC,CAAC;IAClE,CAAC;IAED,cAAc;QACZ,MAAM,QAAQ,GAAG,CAAC,OAAO,EAAE,QAAQ,EAAE,QAAQ,EAAE,eAAe,EAAE,YAAY,EAAE,mBAAmB,CAAU,CAAC;QAC5G,OAAO,MAAM,CAAC,MAAM,CAAC,IAAI,CAAC,gBAAgB,CAAC,CAAC,MAAM,CAAC,CAAC,GAAG,EAAE,MAAM,EAAE,EAAE,CAAC,GAAG,GAAG,MAAM,CAAC,IAAI,EAAE,CAAC,CAAC;cACrF,KAAK,CAAC,IAAI,CAAC,IAAI,GAAG,CAAC,CAAC,GAAG,QAAQ,CAAC,GAAG,CAAC,KAAK,CAAC,EAAE,CAAC,IAAI,CAAC,IAAI,KAAK,SAAS,CAAC,CAAC,EAAE,GAAG,IAAI,CAAC,gBAAgB,CAAC,CAAC,CAAC;iBAClG,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,QAAQ,GAAG,KAAK,CAAC,IAAI,CAAC,IAAI,CAAC,gBAAgB,CAAC,CAAC;QACnD,KAAK,IAAI,CAAC,OAAO,CAAC,KAAK,CAAC,mBAAmB,EAAE,CAAC,KAAK,CAAC,GAAG,EAAE,CAAC,SAAS,CAAC,CAAC,IAAI,CAAC,GAAG,EAAE,CAAC,QAAQ,CAAC,OAAO,CAAC,OAAO,CAAC,EAAE,CAAC,OAAO,CAAC,OAAO,EAAE,CAAC,CAAC,CAAC;QAChI,IAAI,CAAC,gBAAgB,EAAE,CAAC;QACxB,IAAI,CAAC,gBAAgB,CAAC,KAAK,EAAE,CAAC;QAC9B,MAAM,CAAC,IAAI,CAAC,IAAI,CAAC,gBAAgB,CAAC,CAAC,OAAO,CAAC,CAAC,GAAG,EAAE,EAAE;YAChD,IAAI,CAAC,gBAAwB,CAAC,GAAG,CAAC,CAAC,OAAO,EAAE,CAAC;QAChD,CAAC,CAAC,CAAC;IACL,CAAC;CACF;AAED,6EAA6E;AAC7E,8EAA8E;AAC9E,MAAM,CAAC,MAAM,oBAAoB,GAAC,UAAU,CAAC;;;;;;;;CAQ5C,CAAC;AAEF,MAAM,CAAC,MAAM,qBAAqB,GAAC,UAAU,CAAC;EAC5C,6BAA6B;;;;;;;;;;;;;;;;;;CAkB9B,CAAC","sourcesContent":["import {hdrResolveWGSL} from './wgsl/compose.frag.wgsl.js';\nimport { getTargetShaderCache } from './WGPUShaderCache.js';\nimport { textureBytes } from './gpuMemory.js';\nimport shadowCatcherAccumulationWGSL from './wgsl/chunks/shadowCatcherAccumulation.wgsl.js';\nimport { fullScreenQuadVertexShaderWGSL } from './WGPUFullQuadPass.js';\n\nexport function createNormalizeRayTraceOutputWGSL(layers = false, catcher = true, caustics = false) {\n  return /* wgsl */ `\n${hdrResolveWGSL}\n@group(0) @binding(0) var<storage, read> colorBuffer: array<vec4f>;\n@group(0) @binding(1) var<storage, read> normalBuffer: array<vec4f>;\n@group(0) @binding(2) var<storage, read> albedoBuffer: array<vec4f>;\n${shadowCatcherAccumulationWGSL}\n@group(0) @binding(3) var<storage, read> shadowCatcherBuffer: array<ShadowCatcherAccumulation>;\n\n${caustics ? /* wgsl */ `@group(0) @binding(4) var<storage, read> causticBuffer: array<vec4f>;\n` : ''}@group(1) @binding(0) var<uniform> sampleCount: f32;\n@group(1) @binding(1) var<uniform> sampleTileGrid: vec2u;\n\noverride outputWidth: f32;\noverride outputHeight: f32;\n\nstruct FSOutput {\n  ${layers ? `@location(0) foreground: vec4f,\n  @location(1) catcherBackground: vec4f` : `@location(0) color: vec4f,\n  @location(1) normal: vec4f,\n  @location(2) albedo: vec4f${catcher ? ',\\n  @location(3) shadowCatcher: vec4f' : ''}`}\n}\n@fragment\nfn main(@builtin(position) coord: vec4f) -> FSOutput {\n  let size = vec2u(u32(outputWidth), u32(outputHeight));\n  let pixel = vec2u(u32(coord.x), size.y - u32(coord.y) - 1u);\n  var idx = pixel.x + pixel.y * size.x;\n  // Tiles are interleaved pixel grids, not contiguous rectangles. The first\n  // tile samples (0, 0) of every grid cell across the current camera image.\n  // Reconstruct only pixels not yet sampled, until the first sweep completes;\n  // never write these display estimates back into the accumulated ray buffers.\n  if (sampleCount < 1.0 && normalBuffer[idx].a <= 0.0) {\n    let grid = max(sampleTileGrid, vec2u(1u));\n    let nearest = min(\n      ((pixel + grid / 2u) / grid) * grid,\n      ((size - vec2u(1u)) / grid) * grid\n    );\n    let nearestIdx = nearest.x + nearest.y * size.x;\n    if (normalBuffer[nearestIdx].a > 0.0) {\n      idx = nearestIdx;\n    }\n  }\n  var output: FSOutput;\n  if (normalBuffer[idx].a <= 0.0) {\n    ${layers ? `output.foreground = vec4f(0.0);\n    output.catcherBackground = vec4f(0.0);` : `output.color = vec4f(0.0);\n    output.normal = vec4f(0.0);\n    output.albedo = vec4f(0.0);\n    ${catcher ? 'output.shadowCatcher = vec4f(0.0);' : ''}`}\n    return output;\n  }\n  let color = colorBuffer[idx];\n  let normal = normalBuffer[idx];\n  let albedo = albedoBuffer[idx];\n  let hit = albedo.a;\n  // The CPU count covers completed sweeps and is still zero during the first\n  // interleaved sample. Use the actual per-pixel count, including for misses.\n  // Empty pixels must remain finite: NaN alpha can poison float HDR output\n  // even when its canvas uses opaque compositing.\n  let total = max(normal.a, 1.0);\n  ${layers ? `\n  let shadow = shadowCatcherBuffer[idx];\n  let foregroundHit = max(hit - shadow.mask.b, 0.0);\n  // Separated background includes flat primary misses as well as catcher\n  // samples. Neither belongs in the hit-normalized geometry color/alpha.\n  output.foreground = vec4f(\n    (color.rgb - shadow.background.rgb)\n      / select(total, foregroundHit, foregroundHit > 0.0),\n    clamp((color.a - shadow.mask.g) / total, 0.0, 1.0)\n  );\n  output.catcherBackground = shadow.background / total;\n  ` : `output.color = vec4f(\n    // average the color by hit count.\n    // So the antialised edge color will be correct when having transparent background\n    vec3f(hdr_resolve_component(color.r,normal.a,hit,1.0,true),\n      hdr_resolve_component(color.g,normal.a,hit,1.0,true),\n      hdr_resolve_component(color.b,normal.a,hit,1.0,true)),\n    hdr_resolve_component(color.a,normal.a,hit,1.0,false)\n  );\n\n  output.normal = vec4f(0.0);\n  // Normalize to 0 - 1\n  if (dot(normal.rgb, normal.rgb) > 0.1) {\n    output.normal = vec4f(normal.rgb / total * 0.5 + 0.5, 1.0);\n  }\n  output.albedo = vec4f(albedo.rgb / total, clamp(hit / total, 0.0, 1.0));\n  ${catcher ? `let shadow = shadowCatcherBuffer[idx];\n  // Opacity is the ratio of the summed occluded and total light sample\n  // counts; a per-sample ratio average would keep its low-sample bias. Alpha\n  // carries it scaled by coverage, as composition expects.\n  output.shadowCatcher = vec4f(shadow.mask.rgb / total,\n    shadow.mask.r / total * select(0.0, shadow.moments.z / shadow.moments.w, shadow.moments.w > 0.0));` : ''}\n  `}\n${caustics ? /* wgsl */ `  let causticColor=causticBuffer[idx].rgb/max(causticBuffer[idx].a,1.0);\n  ${layers ? 'output.foreground = vec4f(output.foreground.rgb+causticColor,output.foreground.a);' : 'output.color = vec4f(output.color.rgb+causticColor,output.color.a);'}\n` : ''}  return output;\n}\n`;\n}\n\ntype RayTraceOutput = 'color' | 'normal' | 'albedo' | 'shadowCatcher'\n  | 'foreground' | 'catcherBackground';\ntype RayTraceBuffers = {\n  color: GPUBuffer;\n  caustic?: GPUBuffer;\n  wideHdr?:boolean;\n  normal: GPUBuffer;\n  albedo: GPUBuffer;\n  shadowCatcher: GPUBuffer;\n  shadowCatcherOffset?: number;\n};\n\nexport class WGPURayTraceOutputPass {\n  private _device: GPUDevice;\n\n  private _pipeline!: GPURenderPipeline;\n  private _layersPipeline!: GPURenderPipeline;\n  private _foregroundTexture!: GPUTexture;\n  private _catcherBackgroundTexture!: GPUTexture;\n  private _colorTexture!: GPUTexture;\n  private _normalTexture!: GPUTexture;\n  private _albedoTexture!: GPUTexture;\n  private _shadowCatcherTexture!: GPUTexture;\n  private _uniformBindGroup!: GPUBindGroup;\n\n  private _uniformsBuffers;\n\n  private _width = 1;\n  private _height = 1;\n\n  private _hasCatcher = false;\n  private _hasCaustics = false;\n  private _hasWideHdr=false;\n  private _maskPipeline?:GPURenderPipeline;\n  private _wideColor?:GPUTexture;\n  private _wideForeground?:GPUTexture;\n  private _causticPipeline?:GPURenderPipeline;\n  private _needsUpdatePipeline = true;\n\n  constructor(device: GPUDevice, private readonly zeroTexture?: GPUTexture) {\n    this._device = device;\n    this._uniformsBuffers = {\n      sampleCount: device.createBuffer({\n        size: 4,\n        usage: GPUBufferUsage.UNIFORM | GPUBufferUsage.COPY_DST\n      }),\n      sampleTileGrid: device.createBuffer({\n        size: 8,\n        usage: GPUBufferUsage.UNIFORM | GPUBufferUsage.COPY_DST\n      })\n    };\n  }\n\n  private readonly _retainedOutputs = new Set<GPUTexture>();\n  retainOutput(texture: GPUTexture) {\n    const owned = ['color', 'normal', 'albedo', 'shadowCatcher', 'foreground', 'catcherBackground'] as const;\n    if (owned.some(label => this.getOutput(label) === texture)) this._retainedOutputs.add(texture);\n  }\n  releaseOutput(texture: GPUTexture) {\n    if (!this._retainedOutputs.delete(texture)) return;\n    const owned = ['color', 'normal', 'albedo', 'shadowCatcher', 'foreground', 'catcherBackground'] as const;\n    if (!owned.some(label => this.getOutput(label) === texture)) {\n      void this._device.queue.onSubmittedWorkDone().catch(() => undefined).then(() => texture.destroy());\n    }\n  }\n\n  getOutput(label: RayTraceOutput): GPUTexture {\n\n    if(this._hasCaustics&&label==='color'&&this._wideColor)return this._wideColor;\n    if(this._hasCaustics&&label==='foreground'&&this._wideForeground)return this._wideForeground;\n    return this[`_${label}Texture`] ?? this.zeroTexture!;\n  }\n\n  private _createTextures() {\n    const { _device: device, _width: width, _height: height } = this;\n    const textureDescriptor: GPUTextureDescriptor = {\n      size: {\n        width,\n        height,\n        depthOrArrayLayers: 1\n      },\n      format: 'rgba8unorm',\n      usage: GPUTextureUsage.RENDER_ATTACHMENT | GPUTextureUsage.TEXTURE_BINDING |\n        GPUTextureUsage.COPY_SRC\n    };\n    this._destroyTextures();\n    this._colorTexture = device.createTexture({\n      ...textureDescriptor,\n      label: 'Ray Trace Color',\n      format: this._hasWideHdr?'rgba32float':'rgba16float',\n      usage:\n        GPUTextureUsage.RENDER_ATTACHMENT |\n        GPUTextureUsage.TEXTURE_BINDING |\n        GPUTextureUsage.COPY_SRC\n    });\n    this._normalTexture = device.createTexture({\n      ...textureDescriptor,\n      label: 'Ray Trace Normal',\n      usage: textureDescriptor.usage | GPUTextureUsage.COPY_SRC\n    });\n    this._albedoTexture = device.createTexture({\n      ...textureDescriptor,\n      label: 'Ray Trace Albedo'\n    });\n    this._wideColor=this._wideForeground=undefined;\n    if(this._hasCaustics){\n      this._wideColor=device.createTexture({...textureDescriptor,label:'Caustic HDR color',format:'rgba32float'});\n      if(this._hasCatcher)this._wideForeground=device.createTexture({...textureDescriptor,label:'Caustic HDR foreground',format:'rgba32float'});\n    }\n    if (this._hasCatcher) {\n    this._foregroundTexture = device.createTexture({\n      ...textureDescriptor,\n      label: 'Ray Trace Foreground',\n      format: this._hasWideHdr?'rgba32float':'rgba16float'\n    });\n    this._catcherBackgroundTexture = device.createTexture({\n      ...textureDescriptor,\n      label: 'Ray Trace Catcher Background',\n      format: 'rgba16float'\n    });\n    this._shadowCatcherTexture = device.createTexture({\n      ...textureDescriptor,\n      label: 'Ray Trace Shadow Catcher',\n      format: 'rgba16float'\n    });\n    } else {\n      this._foregroundTexture = this._catcherBackgroundTexture = this._shadowCatcherTexture = undefined!;\n    }\n  }\n\n  private _updateBindGroups() {\n    const device = this._device;\n    this._uniformBindGroup = device.createBindGroup({\n      layout: this._pipeline.getBindGroupLayout(1),\n      entries: [\n        {\n          binding: 0,\n          resource: {\n            buffer: this._uniformsBuffers.sampleCount\n          }\n        },\n        {\n          binding: 1,\n          resource: {\n            buffer: this._uniformsBuffers.sampleTileGrid\n          }\n        }\n      ]\n    });\n  }\n\n  private _updatePipeline() {\n    if (!this._needsUpdatePipeline) {\n      return;\n    }\n    this._needsUpdatePipeline = false;\n    const device = this._device;\n\n    this._pipeline = getTargetShaderCache(device).createRenderPipeline({\n      label: 'rayTraceOutput',\n      layout: 'auto',\n      vertex: {\n        module: getTargetShaderCache(device).createShaderModule({\n          code: fullScreenQuadVertexShaderWGSL\n        }),\n        entryPoint: 'main'\n      },\n      fragment: {\n        module: getTargetShaderCache(device).createShaderModule({\n          code: createNormalizeRayTraceOutputWGSL(false, this._hasCatcher&&!this._hasWideHdr, false)\n        }),\n        entryPoint: 'main',\n        targets: [\n          {\n            format: this._hasWideHdr?'rgba32float':'rgba16float'\n          },\n          {\n            format: 'rgba8unorm'\n          },\n          {\n            format: 'rgba8unorm'\n          },\n          ...(this._hasCatcher&&!this._hasWideHdr ? [{format: 'rgba16float' as const}] : [])\n        ],\n        constants: {\n          outputWidth: this._width,\n          outputHeight: this._height\n        }\n      }\n    });\n\n    if (this._hasCatcher) this._layersPipeline = getTargetShaderCache(device).createRenderPipeline({\n      label: 'rayTraceLayers',\n      layout: 'auto',\n      vertex: {\n        module: getTargetShaderCache(device).createShaderModule({ code: fullScreenQuadVertexShaderWGSL }),\n        entryPoint: 'main'\n      },\n      fragment: {\n        module: getTargetShaderCache(device).createShaderModule({ code: createNormalizeRayTraceOutputWGSL(true, true, false) }),\n        entryPoint: 'main',\n        targets: [{ format: this._hasWideHdr?'rgba32float':'rgba16float' }, { format: 'rgba16float' }],\n        constants: { outputWidth: this._width, outputHeight: this._height }\n      }\n    });\n    this._updateBindGroups();\n  }\n\n  setSize(width: number, height: number) {\n    if (\n      this._width !== width ||\n      this._height !== height ||\n      !this._colorTexture\n    ) {\n      this._width = width;\n      this._height = height;this._maskPipeline=undefined;\n      this._createTextures();\n      this._needsUpdatePipeline = true;\n    }\n  }\n\n  setSampleCount(sampleCount: number) {\n    this._device.queue.writeBuffer(\n      this._uniformsBuffers.sampleCount,\n      0,\n      new Float32Array([sampleCount])\n    );\n  }\n\n  setSampleTileGrid(columns: number, rows: number) {\n    this._device.queue.writeBuffer(\n      this._uniformsBuffers.sampleTileGrid,\n      0,\n      new Uint32Array([columns, rows])\n    );\n  }\n\n  createPass(\n    encoder: GPUCommandEncoder,\n    inputBuffers: RayTraceBuffers,\n    separateCatcher = false\n  ) {\n    if(this._hasWideHdr!==!!inputBuffers.wideHdr){this._hasWideHdr=!!inputBuffers.wideHdr;this._createTextures();this._needsUpdatePipeline=true;}\n    if (this._hasCaustics !== !!inputBuffers.caustic) {\n      this._hasCaustics=!!inputBuffers.caustic;this._createTextures();this._needsUpdatePipeline=true;\n    }\n    if (this._hasCatcher !== separateCatcher) {\n      this._hasCatcher = separateCatcher; this._createTextures(); this._needsUpdatePipeline = true;\n    }\n    this._updatePipeline();\n    this._createOutputPass(encoder, inputBuffers, false);\n    if (separateCatcher) this._createOutputPass(encoder, inputBuffers, true);\n    if(this._hasWideHdr&&separateCatcher)this._resolveMask(encoder,inputBuffers);\n    if(inputBuffers.caustic){\n      this._addCaustics(encoder,this._colorTexture,this._wideColor!,inputBuffers.caustic);\n      if(separateCatcher)this._addCaustics(encoder,this._foregroundTexture,this._wideForeground!,inputBuffers.caustic);\n    }\n  }\n\n  private _createOutputPass(\n    encoder: GPUCommandEncoder,\n    inputBuffers: RayTraceBuffers,\n    layers: boolean\n  ) {\n    const pipeline = layers ? this._layersPipeline : this._pipeline;\n    const textures = layers\n      ? [this._foregroundTexture, this._catcherBackgroundTexture]\n      : [this._colorTexture, this._normalTexture, this._albedoTexture, ...(this._hasCatcher&&!this._hasWideHdr ? [this._shadowCatcherTexture] : [])];\n    const pass = encoder.beginRenderPass({\n      colorAttachments: textures.map((texture) => ({\n        view: texture.createView(),\n        loadValue: { r: 0, g: 0, b: 0, a: 0 },\n        storeOp: 'store',\n        loadOp: 'clear'\n      }))\n    });\n    pass.setPipeline(pipeline);\n\n    pass.setBindGroup(\n      0,\n      this._device.createBindGroup({\n        label: 'rayTraceOutput',\n        layout: pipeline.getBindGroupLayout(0),\n        entries: [\n          inputBuffers.color,\n          inputBuffers.normal,\n          inputBuffers.albedo,\n          inputBuffers.shadowCatcher\n        ].map((buffer, idx) => ({\n          binding: idx,\n          resource: {\n            buffer,\n            ...(idx === 3 ? { offset: inputBuffers.shadowCatcherOffset ?? 0 } : {})\n          }\n        })).filter(entry => entry.binding < 3 || (this._hasCatcher&&!this._hasWideHdr) || layers).concat(\n          []\n        )\n      })\n    );\n    pass.setBindGroup(1, layers ? this._device.createBindGroup({\n      layout: pipeline.getBindGroupLayout(1),\n      entries: Object.values(this._uniformsBuffers).map((buffer, binding) => ({\n        binding, resource: { buffer }\n      }))\n    }) : this._uniformBindGroup);\n\n    pass.draw(6, 1, 0, 0);\n    pass.end();\n\n  }\n\n  private _resolveMask(encoder:GPUCommandEncoder,input:RayTraceBuffers){\n    const device=this._device;\n    this._maskPipeline??=device.createRenderPipeline({label:'Wide HDR catcher mask',layout:'auto',\n      vertex:{module:device.createShaderModule({code:fullScreenQuadVertexShaderWGSL}),entryPoint:'main'},\n      fragment:{module:device.createShaderModule({code:shadowMaskResolveWGSL}),entryPoint:'main',constants:{IMAGE_WIDTH:this._width,IMAGE_HEIGHT:this._height},targets:[{format:'rgba16float'}]}});\n    const pass=encoder.beginRenderPass({label:'Wide HDR catcher mask',colorAttachments:[{view:this._shadowCatcherTexture.createView(),loadOp:'clear',storeOp:'store'}]});\n    pass.setPipeline(this._maskPipeline);pass.setBindGroup(0,device.createBindGroup({layout:this._maskPipeline.getBindGroupLayout(0),entries:[\n      {binding:0,resource:{buffer:input.normal}},{binding:1,resource:{buffer:input.shadowCatcher}},\n      {binding:2,resource:{buffer:this._uniformsBuffers.sampleTileGrid}},{binding:3,resource:{buffer:this._uniformsBuffers.sampleCount}}]}));\n    pass.draw(6);pass.end();\n  }\n\n  private _addCaustics(encoder:GPUCommandEncoder,base:GPUTexture,target:GPUTexture,caustics:GPUBuffer){\n    const device=this._device;\n    this._causticPipeline??=device.createRenderPipeline({label:'Caustic HDR resolve',layout:'auto',\n      vertex:{module:device.createShaderModule({code:fullScreenQuadVertexShaderWGSL}),entryPoint:'main'},\n      fragment:{module:device.createShaderModule({code:causticCompositeWGSL}),entryPoint:'main',targets:[{format:'rgba32float'}]}});\n    const pass=encoder.beginRenderPass({label:'Caustic HDR resolve',colorAttachments:[{view:target.createView(),loadOp:'clear',storeOp:'store'}]});\n    pass.setPipeline(this._causticPipeline);pass.setBindGroup(0,device.createBindGroup({layout:this._causticPipeline.getBindGroupLayout(0),entries:[{binding:0,resource:base.createView()},{binding:1,resource:{buffer:caustics}}]}));\n    pass.draw(6);pass.end();\n  }\n\n  private _destroyTextures() {\n    if (!this._colorTexture) return;\n\n    // These textures are also OIDN inputs. A resolution or preview-mode\n    // change can replace them immediately after a denoising command buffer\n    // was submitted. Keep the old resources alive until all work submitted so\n    // far has completed; destroying them synchronously can make Dawn validate\n    // the pending submit against an already-destroyed texture.\n    const textures = [\n      this._wideColor,this._wideForeground,\n      this._colorTexture,\n      this._normalTexture,\n      this._albedoTexture,\n      this._shadowCatcherTexture,\n      this._foregroundTexture,\n      this._catcherBackgroundTexture\n    ].filter((texture): texture is GPUTexture => !!texture && !this._retainedOutputs.has(texture));\n    void this._device.queue\n      .onSubmittedWorkDone()\n      .catch(() => undefined)\n      .then(() => textures.forEach((texture) => texture.destroy()));\n  }\n\n  getMemoryBytes() {\n    const textures = ['color', 'normal', 'albedo', 'shadowCatcher', 'foreground', 'catcherBackground'] as const;\n    return Object.values(this._uniformsBuffers).reduce((sum, buffer) => sum + buffer.size, 0)\n      + Array.from(new Set([...textures.map(label => this[`_${label}Texture`]), ...this._retainedOutputs]))\n        .reduce((sum, texture) => sum + textureBytes(texture), 0);\n  }\n\n  dispose() {\n    const retained = Array.from(this._retainedOutputs);\n    void this._device.queue.onSubmittedWorkDone().catch(() => undefined).then(() => retained.forEach(texture => texture.destroy()));\n    this._destroyTextures();\n    this._retainedOutputs.clear();\n    Object.keys(this._uniformsBuffers).forEach((key) => {\n      (this._uniformsBuffers as any)[key].destroy();\n    });\n  }\n}\n\n// A separate color attachment preserves float32 caustics within the baseline\n// 32-byte MRT budget, including when the catcher needs its fourth attachment.\nexport const causticCompositeWGSL=/* wgsl */ `\n@group(0)@binding(0)var base:texture_2d<f32>;\n@group(0)@binding(1)var<storage,read>caustics:array<vec4f>;\n@fragment fn main(@builtin(position)position:vec4f)->@location(0)vec4f{\n let pixel=vec2u(position.xy);let size=textureDimensions(base);\n let c=textureLoad(base,vec2i(pixel),0);let photon=caustics[pixel.x+(size.y-pixel.y-1u)*size.x];\n return vec4f(c.rgb+photon.rgb/max(photon.a,1.0),c.a);\n}\n`;\n\nexport const shadowMaskResolveWGSL=/* wgsl */ `\n${shadowCatcherAccumulationWGSL}\noverride IMAGE_WIDTH:u32;override IMAGE_HEIGHT:u32;\n@group(0)@binding(0)var<storage,read>normal:array<vec4f>;\n@group(0)@binding(1)var<storage,read>shadow:array<ShadowCatcherAccumulation>;\n@group(0)@binding(2)var<uniform>grid:vec2u;\n@group(0)@binding(3)var<uniform>samples:f32;\n@fragment fn main(@builtin(position)position:vec4f)->@location(0)vec4f{\n // Image dimensions are supplied by the actual attachment through a uniform.\n let size=vec2u(IMAGE_WIDTH,IMAGE_HEIGHT);\n let pixel=vec2u(u32(position.x),size.y-u32(position.y)-1u);var index=pixel.x+pixel.y*size.x;\n if(samples<1.0&&normal[index].a<=0.0){\n  let tile=max(grid,vec2u(1u));let nearest=min(((pixel+tile/2u)/tile)*tile,((size-vec2u(1u))/tile)*tile);\n  let candidate=nearest.x+nearest.y*size.x;if(normal[candidate].a>0.0){index=candidate;}\n }\n if(normal[index].a<=0.0){return vec4f(0.0);}\n let s=shadow[index];let total=max(normal[index].a,1.0);\n return vec4f(s.mask.rgb/total,s.mask.r/total*select(0.0,s.moments.z/s.moments.w,s.moments.w>0.0));\n}\n`;\n"]}