@meimu/mutracer 0.0.0-stage → 0.1.1

package/lib/renderer/WGPURayTraceOutput.js

@@ -0,0 +1,368 @@
+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) {
+    return /* wgsl */ `
+@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;
+@group(1) @binding(1) var<uniform> sampleTileGrid: vec2u;
+
+override outputWidth: f32;
+override outputHeight: f32;
+
+struct oQ {
+${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) fD: vec4f) -> oQ {
+let size = vec2u(u32(outputWidth), u32(outputHeight));
+let fM = vec2u(u32(fD.x), size.y - u32(fD.y) - 1u);
+var idx = fM.x + fM.y * size.x;
+
+
+
+
+if (sampleCount < 1.0 && normalBuffer[idx].a <= 0.0) {
+let uO = max(sampleTileGrid, vec2u(1u));
+let uy = min(
+((fM + uO / 2u) / uO) * uO,
+((size - vec2u(1u)) / uO) * uO
+);
+let sr = uy.x + uy.y * size.x;
+if (normalBuffer[sr].a > 0.0) {
+idx = sr;
+}
+}
+var output: oQ;
+if (normalBuffer[idx].a <= 0.0) {
+${layers ? `output.foreground = vec4f(0.0);
+output.catcherBackground = vec4f(0.0);` : `output.color = vec4f(0.0);
+output.normal = vec4f(0.0);
+output.albedo = vec4f(0.0);
+${catcher ? 'output.shadowCatcher = vec4f(0.0);' : ''}`}
+return output;
+}
+let color = colorBuffer[idx];
+let normal = normalBuffer[idx];
+let albedo = albedoBuffer[idx];
+let hit = albedo.a;
+
+
+
+
+let total = max(normal.a, 1.0);
+${layers ? `
+let shadow = shadowCatcherBuffer[idx];
+let iP = max(hit - shadow.mask.b, 0.0);
+output.foreground = vec4f(
+(color.rgb - shadow.background.rgb)
+/ select(total, iP, iP > 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
+);
+
+output.normal = vec4f(0.0);
+
+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;' : ''}
+`}
+return output;
+}
+`;
+}
+export class WGPURayTraceOutputPass {
+    zeroTexture;
+    _device;
+    _pipeline;
+    _layersPipeline;
+    _foregroundTexture;
+    _catcherBackgroundTexture;
+    _colorTexture;
+    _normalTexture;
+    _albedoTexture;
+    _shadowCatcherTexture;
+    _uniformBindGroup;
+    _uniformsBuffers;
+    _width = 1;
+    _height = 1;
+    _hasCatcher = false;
+    _needsUpdatePipeline = true;
+    constructor(device, zeroTexture) {
+        this.zeroTexture = zeroTexture;
+        this._device = device;
+        this._uniformsBuffers = {
+            sampleCount: device.createBuffer({
+                size: 4,
+                usage: GPUBufferUsage.UNIFORM | GPUBufferUsage.COPY_DST
+            }),
+            sampleTileGrid: device.createBuffer({
+                size: 8,
+                usage: GPUBufferUsage.UNIFORM | GPUBufferUsage.COPY_DST
+            })
+        };
+    }
+    _retainedOutputs = new Set();
+    retainOutput(texture) {
+        const owned = ['color', 'normal', 'albedo', 'shadowCatcher', 'foreground', 'catcherBackground'];
+        if (owned.some(label => this.getOutput(label) === texture))
+            this._retainedOutputs.add(texture);
+    }
+    releaseOutput(texture) {
+        if (!this._retainedOutputs.delete(texture))
+            return;
+        const owned = ['color', 'normal', 'albedo', 'shadowCatcher', 'foreground', 'catcherBackground'];
+        if (!owned.some(label => this.getOutput(label) === texture)) {
+            void this._device.queue.onSubmittedWorkDone().catch(() => undefined).then(() => texture.destroy());
+        }
+    }
+    getOutput(label) {
+        return this[`_${label}Texture`] ?? this.zeroTexture;
+    }
+    _createTextures() {
+        const { _device: device, _width: width, _height: height } = this;
+        const textureDescriptor = {
+            size: {
+                width,
+                height,
+                depthOrArrayLayers: 1
+            },
+            format: 'rgba8unorm',
+            usage: GPUTextureUsage.RENDER_ATTACHMENT | GPUTextureUsage.TEXTURE_BINDING |
+                GPUTextureUsage.COPY_SRC
+        };
+        this._destroyTextures();
+        this._colorTexture = device.createTexture({
+            ...textureDescriptor,
+            label: 'Ray Trace Color',
+            format: 'rgba16float',
+            usage: GPUTextureUsage.RENDER_ATTACHMENT |
+                GPUTextureUsage.TEXTURE_BINDING |
+                GPUTextureUsage.COPY_SRC
+        });
+        this._normalTexture = device.createTexture({
+            ...textureDescriptor,
+            label: 'Ray Trace Normal',
+            usage: textureDescriptor.usage | GPUTextureUsage.COPY_SRC
+        });
+        this._albedoTexture = device.createTexture({
+            ...textureDescriptor,
+            label: 'Ray Trace Albedo'
+        });
+        if (this._hasCatcher) {
+            this._foregroundTexture = device.createTexture({
+                ...textureDescriptor,
+                label: 'Ray Trace Foreground',
+                format: 'rgba16float'
+            });
+            this._catcherBackgroundTexture = device.createTexture({
+                ...textureDescriptor,
+                label: 'Ray Trace Catcher Background',
+                format: 'rgba16float'
+            });
+            this._shadowCatcherTexture = device.createTexture({
+                ...textureDescriptor,
+                label: 'Ray Trace Shadow Catcher',
+                format: 'rgba16float'
+            });
+        }
+        else {
+            this._foregroundTexture = this._catcherBackgroundTexture = this._shadowCatcherTexture = undefined;
+        }
+    }
+    _updateBindGroups() {
+        const device = this._device;
+        this._uniformBindGroup = device.createBindGroup({
+            layout: this._pipeline.getBindGroupLayout(1),
+            entries: [
+                {
+                    binding: 0,
+                    resource: {
+                        buffer: this._uniformsBuffers.sampleCount
+                    }
+                },
+                {
+                    binding: 1,
+                    resource: {
+                        buffer: this._uniformsBuffers.sampleTileGrid
+                    }
+                }
+            ]
+        });
+    }
+    _updatePipeline() {
+        if (!this._needsUpdatePipeline) {
+            return;
+        }
+        this._needsUpdatePipeline = false;
+        const device = this._device;
+        this._pipeline = getTargetShaderCache(device).createRenderPipeline({
+            label: 'rayTraceOutput',
+            layout: 'auto',
+            vertex: {
+                module: getTargetShaderCache(device).createShaderModule({
+                    code: fullScreenQuadVertexShaderWGSL
+                }),
+                entryPoint: 'main'
+            },
+            fragment: {
+                module: getTargetShaderCache(device).createShaderModule({
+                    code: createNormalizeRayTraceOutputWGSL(false, this._hasCatcher)
+                }),
+                entryPoint: 'main',
+                targets: [
+                    {
+                        format: 'rgba16float'
+                    },
+                    {
+                        format: 'rgba8unorm'
+                    },
+                    {
+                        format: 'rgba8unorm'
+                    },
+                    ...(this._hasCatcher ? [{ format: 'rgba16float' }] : [])
+                ],
+                constants: {
+                    outputWidth: this._width,
+                    outputHeight: this._height
+                }
+            }
+        });
+        if (this._hasCatcher)
+            this._layersPipeline = getTargetShaderCache(device).createRenderPipeline({
+                label: 'rayTraceLayers',
+                layout: 'auto',
+                vertex: {
+                    module: getTargetShaderCache(device).createShaderModule({ code: fullScreenQuadVertexShaderWGSL }),
+                    entryPoint: 'main'
+                },
+                fragment: {
+                    module: getTargetShaderCache(device).createShaderModule({ code: createNormalizeRayTraceOutputWGSL(true) }),
+                    entryPoint: 'main',
+                    targets: [{ format: 'rgba16float' }, { format: 'rgba16float' }],
+                    constants: { outputWidth: this._width, outputHeight: this._height }
+                }
+            });
+        this._updateBindGroups();
+    }
+    setSize(width, height) {
+        if (this._width !== width ||
+            this._height !== height ||
+            !this._colorTexture) {
+            this._width = width;
+            this._height = height;
+            this._createTextures();
+            this._needsUpdatePipeline = true;
+        }
+    }
+    setSampleCount(sampleCount) {
+        this._device.queue.writeBuffer(this._uniformsBuffers.sampleCount, 0, new Float32Array([sampleCount]));
+    }
+    setSampleTileGrid(columns, rows) {
+        this._device.queue.writeBuffer(this._uniformsBuffers.sampleTileGrid, 0, new Uint32Array([columns, rows]));
+    }
+    createPass(encoder, inputBuffers, separateCatcher = false) {
+        if (this._hasCatcher !== separateCatcher) {
+            this._hasCatcher = separateCatcher;
+            this._createTextures();
+            this._needsUpdatePipeline = true;
+        }
+        this._updatePipeline();
+        this._createOutputPass(encoder, inputBuffers, false);
+        if (separateCatcher)
+            this._createOutputPass(encoder, inputBuffers, true);
+    }
+    _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] : [])];
+        const pass = encoder.beginRenderPass({
+            colorAttachments: textures.map((texture) => ({
+                view: texture.createView(),
+                loadValue: { r: 0, g: 0, b: 0, a: 0 },
+                storeOp: 'store',
+                loadOp: 'clear'
+            }))
+        });
+        pass.setPipeline(pipeline);
+        pass.setBindGroup(0, this._device.createBindGroup({
+            label: 'rayTraceOutput',
+            layout: pipeline.getBindGroupLayout(0),
+            entries: [
+                inputBuffers.color,
+                inputBuffers.normal,
+                inputBuffers.albedo,
+                inputBuffers.shadowCatcher
+            ].map((buffer, idx) => ({
+                binding: idx,
+                resource: {
+                    buffer,
+                    ...(idx === 3 ? { offset: inputBuffers.shadowCatcherOffset ?? 0 } : {})
+                }
+            })).filter(entry => entry.binding < 3 || this._hasCatcher || layers)
+        }));
+        pass.setBindGroup(1, layers ? this._device.createBindGroup({
+            layout: pipeline.getBindGroupLayout(1),
+            entries: Object.values(this._uniformsBuffers).map((buffer, binding) => ({
+                binding, resource: { buffer }
+            }))
+        }) : this._uniformBindGroup);
+        pass.draw(6, 1, 0, 0);
+        pass.end();
+    }
+    _destroyTextures() {
+        if (!this._colorTexture)
+            return;
+        // These textures are also OIDN inputs. A resolution or preview-mode
+        // change can replace them immediately after a denoising command buffer
+        // was submitted. Keep the old resources alive until all work submitted so
+        // far has completed; destroying them synchronously can make Dawn validate
+        // the pending submit against an already-destroyed texture.
+        const textures = [
+            this._colorTexture,
+            this._normalTexture,
+            this._albedoTexture,
+            this._shadowCatcherTexture,
+            this._foregroundTexture,
+            this._catcherBackgroundTexture
+        ].filter((texture) => !!texture && !this._retainedOutputs.has(texture));
+        void this._device.queue
+            .onSubmittedWorkDone()
+            .catch(() => undefined)
+            .then(() => textures.forEach((texture) => texture.destroy()));
+    }
+    getMemoryBytes() {
+        const textures = ['color', 'normal', 'albedo', 'shadowCatcher', 'foreground', 'catcherBackground'];
+        return Object.values(this._uniformsBuffers).reduce((sum, buffer) => sum + buffer.size, 0)
+            + Array.from(new Set([...textures.map(label => this[`_${label}Texture`]), ...this._retainedOutputs]))
+                .reduce((sum, texture) => sum + textureBytes(texture), 0);
+    }
+    dispose() {
+        const retained = Array.from(this._retainedOutputs);
+        void this._device.queue.onSubmittedWorkDone().catch(() => undefined).then(() => retained.forEach(texture => texture.destroy()));
+        this._destroyTextures();
+        this._retainedOutputs.clear();
+        Object.keys(this._uniformsBuffers).forEach((key) => {
+            this._uniformsBuffers[key].destroy();
+        });
+    }
+}
+//# sourceMappingURL=WGPURayTraceOutput.js.map

package/lib/renderer/WGPURayTraceOutput.js.map

@@ -0,0 +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;;;;;;;;;CASV,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  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"]}