@meimu/mutracer 0.0.0-stage → 0.1.0

package/lib/renderer/WGPUFullQuadPass.js

@@ -0,0 +1,389 @@
+import { scratchWGSLFunctions } from './scratchSampling.js';
+import { setTextureRegion, textureRegion } from './textureRegion.js';
+import { getTargetShaderCache } from './WGPUShaderCache.js';
+import { textureBytes } from './gpuMemory.js';
+export const fullScreenQuadVertexShaderWGSL = /*wgsl */ `
+@vertex
+fn main(
+@builtin(vertex_index) vertexIndex: u32
+) -> @builtin(position) vec4f {
+const pos = array(
+vec2(-1.0, -1.0), vec2(1.0, -1.0), vec2(-1.0, 1.0),
+vec2(-1.0, 1.0), vec2(1.0, -1.0), vec2(1.0, 1.0),
+);
+
+return vec4<f32>(pos[vertexIndex], 0.0, 1.0);
+}
+`;
+function isTextureParamsEqual(params, other) {
+    return params.format === other.format;
+}
+export function getUniformBuffer(value, type) {
+    if (value instanceof Float32Array ||
+        value instanceof Int32Array ||
+        value instanceof Uint32Array) {
+        return value;
+    }
+    if (typeof value === 'number') {
+        value = [value];
+    }
+    if (type === 'i32' || type.endsWith('i')) {
+        return new Int32Array(value);
+    }
+    else if (type === 'u32' || type.endsWith('u')) {
+        return new Uint32Array(value);
+    }
+    else {
+        return new Float32Array(value);
+    }
+}
+export class WGPUFullQuadPass {
+    _label;
+    _device;
+    _outputTextures = {};
+    _outputParams = {};
+    _pipeline;
+    _bindGroups = [];
+    _needsUpdatePipeline = true;
+    _needsUpdateOutputTextures = true;
+    /**
+     * When render to canvas
+     */
+    _renderToScreen;
+    _inputs = [];
+    _inputViewDimensions = {};
+    _outputs = [];
+    _uniforms = [];
+    _uniformsMap = {};
+    _uniformBuffers = {};
+    _samplers = {};
+    _capacityWidth = 0;
+    _capacityHeight = 0;
+    _reserveCapacity = false;
+    _regionNames = [];
+    _width = 10;
+    _height = 10;
+    _fsCode = '';
+    _fsMain;
+    _fsDefine;
+    _textureWillBeRead;
+    constructor(label, device, opts) {
+        this._label = label;
+        this._device = device;
+        this._fsMain = opts.fsMain;
+        this._fsDefine = opts.fsDefine;
+        this._inputs = opts.inputs;
+        this._inputViewDimensions = opts.inputViewDimensions || {};
+        this._outputs = opts.outputs;
+        this._uniforms = (opts.uniforms || []).concat({
+            label: 'viewportSize',
+            type: 'vec2f',
+            data: [0, 0]
+        });
+        this._samplers = opts.samplers || {};
+        this._uniformsMap = this._uniforms.reduce((map, uniform) => {
+            map[uniform.label] = uniform;
+            return map;
+        }, {});
+        this._textureWillBeRead = opts.textureWillBeRead || false;
+        this._uniforms.forEach((uniform) => {
+            const data = getUniformBuffer(uniform.data, uniform.type);
+            this._uniformBuffers[uniform.label] = device.createBuffer({
+                label: this._label,
+                size: data.byteLength,
+                usage: GPUBufferUsage.UNIFORM | GPUBufferUsage.COPY_DST
+            });
+            this._device.queue.writeBuffer(this._uniformBuffers[uniform.label], 0, data);
+        });
+    }
+    /** Keep physical outputs large enough for every render while varying the logical viewport. */
+    reserveCapacity() {
+        if (this._reserveCapacity)
+            return;
+        this._reserveCapacity = true;
+        this._capacityWidth = this._width;
+        this._capacityHeight = this._height;
+        const textures = this._inputs.filter(name => !this._inputViewDimensions[name] || this._inputViewDimensions[name] === '2d');
+        this._regionNames = textures;
+        if (textures.length) {
+            const uniform = { label: 'imageRegions', type: `array<vec4u, ${textures.length}>`, data: new Uint32Array(textures.length * 4) };
+            this._uniforms.push(uniform);
+            this._uniformsMap[uniform.label] = uniform;
+            this._uniformBuffers[uniform.label] = this._device.createBuffer({ label: `${this._label}.${uniform.label}`, size: textures.length * 16, usage: GPUBufferUsage.UNIFORM | GPUBufferUsage.COPY_DST });
+        }
+        this._needsUpdatePipeline = true;
+        this._fsCode = '';
+    }
+    setSize(width, height) {
+        width = Math.max(1, Math.ceil(width));
+        height = Math.max(1, Math.ceil(height));
+        const sizeChanged = width !== this._width || height !== this._height;
+        this._width = width;
+        this._height = height;
+        if (sizeChanged) {
+            const nextWidth = Math.max(this._capacityWidth, width);
+            const nextHeight = Math.max(this._capacityHeight, height);
+            this._needsUpdateOutputTextures ||= !this._reserveCapacity || nextWidth !== this._capacityWidth || nextHeight !== this._capacityHeight;
+            this._capacityWidth = nextWidth;
+            this._capacityHeight = nextHeight;
+            this._needsUpdatePipeline = true;
+            this.setUniform('viewportSize', [width, height]);
+        }
+    }
+    setOutputParams(outputParams) {
+        this._renderToScreen = undefined;
+        const outputsChanged = Object.keys(outputParams).some((key) => !isTextureParamsEqual(outputParams[key], this._outputParams[key] || {}));
+        this._needsUpdateOutputTextures ||= outputsChanged;
+        this._needsUpdatePipeline ||= outputsChanged;
+        if (outputsChanged)
+            this._fsCode = '';
+        this._outputParams = outputParams;
+    }
+    setRenderToScreen(screenTexture, presentationFormat) {
+        if (this._renderToScreen &&
+            this._renderToScreen.presentationFormat !== presentationFormat) {
+            this._needsUpdatePipeline = true;
+            this._fsCode = '';
+        }
+        this._renderToScreen = {
+            screenTexture,
+            presentationFormat
+        };
+    }
+    setUniform(label, data) {
+        const buffer = this._uniformBuffers[label];
+        this._device.queue.writeBuffer(buffer, 0, getUniformBuffer(data, this._uniformsMap[label].type));
+    }
+    _retainedOutputs = new Set();
+    retainOutput(texture) { if (Object.values(this._outputTextures).includes(texture))
+        this._retainedOutputs.add(texture); }
+    releaseOutput(texture) {
+        if (!this._retainedOutputs.delete(texture))
+            return;
+        if (!Object.values(this._outputTextures).includes(texture)) {
+            void this._device.queue.onSubmittedWorkDone().catch(() => undefined).then(() => texture.destroy());
+        }
+    }
+    getOutput(name) {
+        return this._outputTextures[name];
+    }
+    getMemoryBytes() {
+        return Object.values(this._uniformBuffers).reduce((sum, buffer) => sum + buffer.size, 0)
+            + Array.from(new Set([...Object.values(this._outputTextures), ...this._retainedOutputs]))
+                .reduce((sum, texture) => sum + textureBytes(texture), 0);
+    }
+    dispose() {
+        Object.keys(this._uniformBuffers).forEach((key) => {
+            this._uniformBuffers[key].destroy();
+        });
+        new Set([...Object.values(this._outputTextures), ...this._retainedOutputs]).forEach(texture => texture.destroy());
+        this._retainedOutputs.clear();
+    }
+    _createTexture(params) {
+        return this._device.createTexture({
+            label: this._label,
+            size: {
+                width: this._reserveCapacity ? this._capacityWidth : this._width,
+                height: this._reserveCapacity ? this._capacityHeight : this._height,
+                depthOrArrayLayers: 1
+            },
+            format: params.format,
+            usage: GPUTextureUsage.RENDER_ATTACHMENT |
+                GPUTextureUsage.TEXTURE_BINDING |
+                (this._textureWillBeRead ? GPUTextureUsage.COPY_SRC : 0)
+        });
+    }
+    _updateOutputTextures(excludes) {
+        const outputParams = this._outputParams;
+        for (const key in outputParams) {
+            if (excludes?.includes(key)) {
+                continue;
+            }
+            const params = outputParams[key];
+            if (!this._retainedOutputs.has(this._outputTextures[key]))
+                this._outputTextures[key]?.destroy();
+            this._outputTextures[key] = this._createTexture(params);
+        }
+    }
+    _updatePipeline() {
+        if (!this._needsUpdatePipeline) {
+            return;
+        }
+        this._needsUpdatePipeline = false;
+        const outputParams = this._outputParams;
+        const device = this._device;
+        const fsCode = this._getFullFs();
+        if (fsCode === this._fsCode) {
+            return;
+        }
+        const { screenTexture, presentationFormat } = this._renderToScreen || {};
+        this._fsCode = fsCode;
+        this._pipeline = getTargetShaderCache(device).createRenderPipeline({
+            label: this._label,
+            layout: 'auto',
+            vertex: {
+                module: getTargetShaderCache(device).createShaderModule({
+                    label: this._label,
+                    code: fullScreenQuadVertexShaderWGSL
+                }),
+                entryPoint: 'main'
+            },
+            fragment: {
+                module: getTargetShaderCache(device).createShaderModule({
+                    label: this._label,
+                    code: fsCode
+                }),
+                entryPoint: 'main',
+                targets: screenTexture
+                    ? [
+                        {
+                            format: presentationFormat
+                        }
+                    ]
+                    : this._outputs.map((key) => ({
+                        format: outputParams[key].format
+                    }))
+            },
+            primitive: {
+                topology: 'triangle-list'
+            }
+        });
+        this._updateBindGroups();
+    }
+    _getFullFs() {
+        const inputs = this._inputs;
+        const inputsLen = inputs.length;
+        const hasInputs = inputsLen > 0;
+        const fs = /* wgsl */ `
+${inputs
+            .sort()
+            .map((textureName, idx) => {
+            const viewDimension = this._inputViewDimensions[textureName];
+            const textureType = viewDimension === '3d' ? 'texture_3d' : 'texture_2d';
+            return `@group(0) @binding(${idx}) var ${textureName}: ${textureType}<f32>;`;
+        })
+            .join('\n')}
+
+${hasInputs
+            ? Object.keys(this._samplers)
+                .sort()
+                .map((samplerName, idx) => `@group(0) @binding(${idx + inputsLen}) var ${samplerName}: sampler;`)
+                .join('\n')
+            : ''}
+${this._uniforms
+            .map((uniform, idx) => `@group(${hasInputs ? 1 : 0}) @binding(${idx}) var<uniform> ${uniform.label}: ${uniform.type};`)
+            .join('\n')}
+struct oQ {
+${this._outputs
+            .map((name, idx) => `@location(${idx}) ${name}: vec4f,`)
+            .join('\n')}
+}
+${scratchWGSLFunctions(this._reserveCapacity ? this._regionNames : inputs.filter(name => !this._inputViewDimensions[name] || this._inputViewDimensions[name] === '2d'), Object.keys(this._samplers), this._reserveCapacity)}
+${this._fsDefine ?? ''}
+@fragment
+fn main(
+@builtin(position) fD: vec4f
+) -> oQ {
+let uv = vec2i(floor(fD.xy));
+let uv2 = fD.xy / viewportSize;
+var output: oQ;
+${this._fsMain}
+return output;
+}
+`;
+        return fs;
+    }
+    _updateBindGroups() {
+        const bindGroups = [];
+        const device = this._device;
+        const uniformBindGroupIndex = this._inputs.length > 0 ? 1 : 0;
+        if (this._uniforms.length > 0) {
+            bindGroups[uniformBindGroupIndex] = device.createBindGroup({
+                label: this._label,
+                layout: this._pipeline.getBindGroupLayout(uniformBindGroupIndex),
+                entries: this._uniforms.map((uniform, idx) => ({
+                    binding: idx,
+                    resource: {
+                        buffer: this._uniformBuffers[uniform.label]
+                    }
+                }))
+            });
+        }
+        this._bindGroups = bindGroups;
+    }
+    createPass(commandEncoder, inputs, 
+    /**
+     * Explict outputs. Used for buffers reuse.
+     */
+    outputs) {
+        if (this._needsUpdateOutputTextures && !this._renderToScreen) {
+            this._updateOutputTextures(Object.keys(outputs || {}));
+            this._needsUpdateOutputTextures = false;
+        }
+        if (this._regionNames.length) {
+            const values = new Uint32Array(this._regionNames.length * 4);
+            for (const [index, name] of this._regionNames.entries()) {
+                const region = textureRegion(inputs[name]);
+                values.set([region.width, region.height], index * 4);
+            }
+            this.setUniform('imageRegions', values);
+        }
+        for (const texture of Object.values(this._outputTextures))
+            setTextureRegion(texture, this._width, this._height);
+        this._updatePipeline();
+        const inputsLen = this._inputs.length;
+        if (inputsLen > 0) {
+            this._bindGroups[0] = this._device.createBindGroup({
+                label: this._label,
+                layout: this._pipeline.getBindGroupLayout(0),
+                entries: [
+                    ...this._inputs.map((textureName, idx) => ({
+                        binding: idx,
+                        resource: inputs[textureName].createView(this._inputViewDimensions[textureName]
+                            ? { dimension: this._inputViewDimensions[textureName] }
+                            : undefined)
+                    })),
+                    ...Object.keys(this._samplers).map((samplerName, idx) => ({
+                        binding: idx + inputsLen,
+                        resource: this._samplers[samplerName]
+                    }))
+                ]
+            });
+        }
+        const outputTextures = {
+            // Pick the texture from outputTextures first.
+            ...this._outputTextures,
+            ...outputs
+        };
+        // Begin the render pass
+        const renderPass = commandEncoder.beginRenderPass({
+            colorAttachments: this._renderToScreen
+                ? [
+                    {
+                        view: this._renderToScreen.screenTexture.createView(),
+                        clearValue: { r: 0, g: 0, b: 0, a: 0 },
+                        storeOp: 'store',
+                        loadOp: 'clear'
+                    }
+                ]
+                : this._outputs.map((textureName) => ({
+                    view: outputTextures[textureName].createView(),
+                    clearValue: { r: 0, g: 0, b: 0, a: 0 },
+                    loadOp: 'clear',
+                    storeOp: 'store'
+                }))
+        });
+        // Draw a full quad
+        renderPass.setViewport(0, 0, this._width, this._height, 0, 1);
+        renderPass.setScissorRect(0, 0, this._width, this._height);
+        renderPass.setPipeline(this._pipeline);
+        // Bind groups
+        this._bindGroups.forEach((bindGroup, idx) => {
+            renderPass.setBindGroup(idx, bindGroup);
+        });
+        renderPass.draw(6, 1, 0, 0);
+        // End the render pass
+        renderPass.end();
+    }
+}
+//# sourceMappingURL=WGPUFullQuadPass.js.map

package/lib/renderer/WGPUFullQuadPass.js.map

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

package/lib/renderer/WGPUHDRPeakMeter.d.ts

@@ -0,0 +1,19 @@
+/** Periodically measures the largest encoded value sent to an HDR canvas. */
+export declare class WGPUHDRPeakMeter {
+    private readonly device;
+    private readonly peakBuffer;
+    private readonly readbackBuffer;
+    private readonly pipeline;
+    private pending;
+    private disposed;
+    private lastSubmissionTime;
+    private peakDisplayValue;
+    constructor(device: GPUDevice);
+    encode(encoder: GPUCommandEncoder, inputTexture: GPUTexture, width: number, height: number, now?: number): boolean;
+    resolve(): Promise<void>;
+    getPeakDisplayValue(): number;
+    reset(): void;
+    getMemoryBytes(): number;
+    dispose(): void;
+}
+export default WGPUHDRPeakMeter;

package/lib/renderer/WGPUHDRPeakMeter.js

@@ -0,0 +1,112 @@
+const peakMeterWGSL = /* wgsl */ `
+@group(0) @binding(0) var inputTexture: texture_2d<f32>;
+@group(0) @binding(1) var<storage, read_write> peakBits: atomic<u32>;
+
+@compute @workgroup_size(16, 16)
+fn main(@builtin(global_invocation_id) id: vec3u) {
+let size = textureDimensions(inputTexture);
+if (id.x >= size.x || id.y >= size.y) { return; }
+let color = max(textureLoad(inputTexture, vec2i(id.xy), 0).rgb, vec3f(0.0));
+let peak = max(color.r, max(color.g, color.b));
+if (peak < 65504.0) {
+atomicMax(&peakBits, bitcast<u32>(peak));
+}
+}
+`;
+/** Periodically measures the largest encoded value sent to an HDR canvas. */
+export class WGPUHDRPeakMeter {
+    device;
+    peakBuffer;
+    readbackBuffer;
+    pipeline;
+    pending = false;
+    disposed = false;
+    lastSubmissionTime = -Infinity;
+    peakDisplayValue = 0;
+    constructor(device) {
+        this.device = device;
+        this.peakBuffer = device.createBuffer({
+            label: 'HDR display peak',
+            size: Uint32Array.BYTES_PER_ELEMENT,
+            usage: GPUBufferUsage.STORAGE |
+                GPUBufferUsage.COPY_SRC |
+                GPUBufferUsage.COPY_DST
+        });
+        this.readbackBuffer = device.createBuffer({
+            label: 'HDR display peak readback',
+            size: Uint32Array.BYTES_PER_ELEMENT,
+            usage: GPUBufferUsage.COPY_DST | GPUBufferUsage.MAP_READ
+        });
+        this.pipeline = device.createComputePipeline({
+            label: 'HDR display peak',
+            layout: 'auto',
+            compute: {
+                module: device.createShaderModule({
+                    label: 'HDR display peak',
+                    code: peakMeterWGSL
+                }),
+                entryPoint: 'main'
+            }
+        });
+    }
+    encode(encoder, inputTexture, width, height, now = performance.now()) {
+        if (this.pending || now - this.lastSubmissionTime < 500)
+            return false;
+        this.pending = true;
+        this.lastSubmissionTime = now;
+        encoder.clearBuffer(this.peakBuffer);
+        const bindGroup = this.device.createBindGroup({
+            label: 'HDR display peak',
+            layout: this.pipeline.getBindGroupLayout(0),
+            entries: [
+                { binding: 0, resource: inputTexture.createView() },
+                { binding: 1, resource: { buffer: this.peakBuffer } }
+            ]
+        });
+        const pass = encoder.beginComputePass({ label: 'HDR display peak' });
+        pass.setPipeline(this.pipeline);
+        pass.setBindGroup(0, bindGroup);
+        pass.dispatchWorkgroups(Math.ceil(width / 16), Math.ceil(height / 16));
+        pass.end();
+        encoder.copyBufferToBuffer(this.peakBuffer, 0, this.readbackBuffer, 0, Uint32Array.BYTES_PER_ELEMENT);
+        return true;
+    }
+    async resolve() {
+        if (!this.pending || this.disposed)
+            return;
+        try {
+            await this.readbackBuffer.mapAsync(GPUMapMode.READ);
+            const mapped = this.readbackBuffer.getMappedRange();
+            const bits = new Uint32Array(mapped)[0];
+            const raw = new ArrayBuffer(Float32Array.BYTES_PER_ELEMENT);
+            new Uint32Array(raw)[0] = bits;
+            this.peakDisplayValue = new Float32Array(raw)[0];
+            this.readbackBuffer.unmap();
+        }
+        catch {
+            if (this.readbackBuffer.mapState === 'mapped') {
+                this.readbackBuffer.unmap();
+            }
+        }
+        finally {
+            this.pending = false;
+        }
+    }
+    getPeakDisplayValue() {
+        return this.peakDisplayValue;
+    }
+    reset() {
+        this.peakDisplayValue = 0;
+        this.lastSubmissionTime = -Infinity;
+    }
+    getMemoryBytes() { return this.peakBuffer.size + this.readbackBuffer.size; }
+    dispose() {
+        this.disposed = true;
+        void this.device.queue.onSubmittedWorkDone().catch(() => undefined).then(() => {
+            this.peakBuffer.destroy();
+            this.readbackBuffer.destroy();
+        });
+    }
+}
+export default WGPUHDRPeakMeter;
+//# sourceMappingURL=WGPUHDRPeakMeter.js.map

package/lib/renderer/WGPUHDRPeakMeter.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"WGPUHDRPeakMeter.js","sourceRoot":"","sources":["../../src/renderer/WGPUHDRPeakMeter.ts"],"names":[],"mappings":"AAAA,MAAM,aAAa,GAAG,UAAU,CAAC;;;;;;;;;;;;;;CAchC,CAAC;AAEF,6EAA6E;AAC7E,MAAM,OAAO,gBAAgB;IASE;IARZ,UAAU,CAAY;IACtB,cAAc,CAAY;IAC1B,QAAQ,CAAqB;IACtC,OAAO,GAAG,KAAK,CAAC;IAChB,QAAQ,GAAG,KAAK,CAAC;IACjB,kBAAkB,GAAG,CAAC,QAAQ,CAAC;IAC/B,gBAAgB,GAAG,CAAC,CAAC;IAE7B,YAA6B,MAAiB;QAAjB,WAAM,GAAN,MAAM,CAAW;QAC5C,IAAI,CAAC,UAAU,GAAG,MAAM,CAAC,YAAY,CAAC;YACpC,KAAK,EAAE,kBAAkB;YACzB,IAAI,EAAE,WAAW,CAAC,iBAAiB;YACnC,KAAK,EACH,cAAc,CAAC,OAAO;gBACtB,cAAc,CAAC,QAAQ;gBACvB,cAAc,CAAC,QAAQ;SAC1B,CAAC,CAAC;QACH,IAAI,CAAC,cAAc,GAAG,MAAM,CAAC,YAAY,CAAC;YACxC,KAAK,EAAE,2BAA2B;YAClC,IAAI,EAAE,WAAW,CAAC,iBAAiB;YACnC,KAAK,EAAE,cAAc,CAAC,QAAQ,GAAG,cAAc,CAAC,QAAQ;SACzD,CAAC,CAAC;QACH,IAAI,CAAC,QAAQ,GAAG,MAAM,CAAC,qBAAqB,CAAC;YAC3C,KAAK,EAAE,kBAAkB;YACzB,MAAM,EAAE,MAAM;YACd,OAAO,EAAE;gBACP,MAAM,EAAE,MAAM,CAAC,kBAAkB,CAAC;oBAChC,KAAK,EAAE,kBAAkB;oBACzB,IAAI,EAAE,aAAa;iBACpB,CAAC;gBACF,UAAU,EAAE,MAAM;aACnB;SACF,CAAC,CAAC;IACL,CAAC;IAED,MAAM,CACJ,OAA0B,EAC1B,YAAwB,EACxB,KAAa,EACb,MAAc,EACd,GAAG,GAAG,WAAW,CAAC,GAAG,EAAE;QAEvB,IAAI,IAAI,CAAC,OAAO,IAAI,GAAG,GAAG,IAAI,CAAC,kBAAkB,GAAG,GAAG;YAAE,OAAO,KAAK,CAAC;QACtE,IAAI,CAAC,OAAO,GAAG,IAAI,CAAC;QACpB,IAAI,CAAC,kBAAkB,GAAG,GAAG,CAAC;QAC9B,OAAO,CAAC,WAAW,CAAC,IAAI,CAAC,UAAU,CAAC,CAAC;QACrC,MAAM,SAAS,GAAG,IAAI,CAAC,MAAM,CAAC,eAAe,CAAC;YAC5C,KAAK,EAAE,kBAAkB;YACzB,MAAM,EAAE,IAAI,CAAC,QAAQ,CAAC,kBAAkB,CAAC,CAAC,CAAC;YAC3C,OAAO,EAAE;gBACP,EAAE,OAAO,EAAE,CAAC,EAAE,QAAQ,EAAE,YAAY,CAAC,UAAU,EAAE,EAAE;gBACnD,EAAE,OAAO,EAAE,CAAC,EAAE,QAAQ,EAAE,EAAE,MAAM,EAAE,IAAI,CAAC,UAAU,EAAE,EAAE;aACtD;SACF,CAAC,CAAC;QACH,MAAM,IAAI,GAAG,OAAO,CAAC,gBAAgB,CAAC,EAAE,KAAK,EAAE,kBAAkB,EAAE,CAAC,CAAC;QACrE,IAAI,CAAC,WAAW,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC;QAChC,IAAI,CAAC,YAAY,CAAC,CAAC,EAAE,SAAS,CAAC,CAAC;QAChC,IAAI,CAAC,kBAAkB,CAAC,IAAI,CAAC,IAAI,CAAC,KAAK,GAAG,EAAE,CAAC,EAAE,IAAI,CAAC,IAAI,CAAC,MAAM,GAAG,EAAE,CAAC,CAAC,CAAC;QACvE,IAAI,CAAC,GAAG,EAAE,CAAC;QACX,OAAO,CAAC,kBAAkB,CACxB,IAAI,CAAC,UAAU,EACf,CAAC,EACD,IAAI,CAAC,cAAc,EACnB,CAAC,EACD,WAAW,CAAC,iBAAiB,CAC9B,CAAC;QACF,OAAO,IAAI,CAAC;IACd,CAAC;IAED,KAAK,CAAC,OAAO;QACX,IAAI,CAAC,IAAI,CAAC,OAAO,IAAI,IAAI,CAAC,QAAQ;YAAE,OAAO;QAC3C,IAAI,CAAC;YACH,MAAM,IAAI,CAAC,cAAc,CAAC,QAAQ,CAAC,UAAU,CAAC,IAAI,CAAC,CAAC;YACpD,MAAM,MAAM,GAAG,IAAI,CAAC,cAAc,CAAC,cAAc,EAAE,CAAC;YACpD,MAAM,IAAI,GAAG,IAAI,WAAW,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC,CAAC;YACxC,MAAM,GAAG,GAAG,IAAI,WAAW,CAAC,YAAY,CAAC,iBAAiB,CAAC,CAAC;YAC5D,IAAI,WAAW,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC,GAAG,IAAI,CAAC;YAC/B,IAAI,CAAC,gBAAgB,GAAG,IAAI,YAAY,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC,CAAC;YACjD,IAAI,CAAC,cAAc,CAAC,KAAK,EAAE,CAAC;QAC9B,CAAC;QAAC,MAAM,CAAC;YACP,IAAI,IAAI,CAAC,cAAc,CAAC,QAAQ,KAAK,QAAQ,EAAE,CAAC;gBAC9C,IAAI,CAAC,cAAc,CAAC,KAAK,EAAE,CAAC;YAC9B,CAAC;QACH,CAAC;gBAAS,CAAC;YACT,IAAI,CAAC,OAAO,GAAG,KAAK,CAAC;QACvB,CAAC;IACH,CAAC;IAED,mBAAmB;QACjB,OAAO,IAAI,CAAC,gBAAgB,CAAC;IAC/B,CAAC;IAED,KAAK;QACH,IAAI,CAAC,gBAAgB,GAAG,CAAC,CAAC;QAC1B,IAAI,CAAC,kBAAkB,GAAG,CAAC,QAAQ,CAAC;IACtC,CAAC;IAED,cAAc,KAAI,OAAO,IAAI,CAAC,UAAU,CAAC,IAAI,GAAG,IAAI,CAAC,cAAc,CAAC,IAAI,CAAC,CAAA,CAAC;IAE1E,OAAO;QACL,IAAI,CAAC,QAAQ,GAAG,IAAI,CAAC;QACrB,KAAK,IAAI,CAAC,MAAM,CAAC,KAAK,CAAC,mBAAmB,EAAE,CAAC,KAAK,CAAC,GAAG,EAAE,CAAC,SAAS,CAAC,CAAC,IAAI,CAAC,GAAG,EAAE;YAC5E,IAAI,CAAC,UAAU,CAAC,OAAO,EAAE,CAAC;YAC1B,IAAI,CAAC,cAAc,CAAC,OAAO,EAAE,CAAC;QAChC,CAAC,CAAC,CAAC;IACL,CAAC;CACF;AAED,eAAe,gBAAgB,CAAC","sourcesContent":["const peakMeterWGSL = /* wgsl */ `\n@group(0) @binding(0) var inputTexture: texture_2d<f32>;\n@group(0) @binding(1) var<storage, read_write> peakBits: atomic<u32>;\n\n@compute @workgroup_size(16, 16)\nfn main(@builtin(global_invocation_id) id: vec3u) {\n  let size = textureDimensions(inputTexture);\n  if (id.x >= size.x || id.y >= size.y) { return; }\n  let color = max(textureLoad(inputTexture, vec2i(id.xy), 0).rgb, vec3f(0.0));\n  let peak = max(color.r, max(color.g, color.b));\n  if (peak < 65504.0) {\n    atomicMax(&peakBits, bitcast<u32>(peak));\n  }\n}\n`;\n\n/** Periodically measures the largest encoded value sent to an HDR canvas. */\nexport class WGPUHDRPeakMeter {\n  private readonly peakBuffer: GPUBuffer;\n  private readonly readbackBuffer: GPUBuffer;\n  private readonly pipeline: GPUComputePipeline;\n  private pending = false;\n  private disposed = false;\n  private lastSubmissionTime = -Infinity;\n  private peakDisplayValue = 0;\n\n  constructor(private readonly device: GPUDevice) {\n    this.peakBuffer = device.createBuffer({\n      label: 'HDR display peak',\n      size: Uint32Array.BYTES_PER_ELEMENT,\n      usage:\n        GPUBufferUsage.STORAGE |\n        GPUBufferUsage.COPY_SRC |\n        GPUBufferUsage.COPY_DST\n    });\n    this.readbackBuffer = device.createBuffer({\n      label: 'HDR display peak readback',\n      size: Uint32Array.BYTES_PER_ELEMENT,\n      usage: GPUBufferUsage.COPY_DST | GPUBufferUsage.MAP_READ\n    });\n    this.pipeline = device.createComputePipeline({\n      label: 'HDR display peak',\n      layout: 'auto',\n      compute: {\n        module: device.createShaderModule({\n          label: 'HDR display peak',\n          code: peakMeterWGSL\n        }),\n        entryPoint: 'main'\n      }\n    });\n  }\n\n  encode(\n    encoder: GPUCommandEncoder,\n    inputTexture: GPUTexture,\n    width: number,\n    height: number,\n    now = performance.now()\n  ) {\n    if (this.pending || now - this.lastSubmissionTime < 500) return false;\n    this.pending = true;\n    this.lastSubmissionTime = now;\n    encoder.clearBuffer(this.peakBuffer);\n    const bindGroup = this.device.createBindGroup({\n      label: 'HDR display peak',\n      layout: this.pipeline.getBindGroupLayout(0),\n      entries: [\n        { binding: 0, resource: inputTexture.createView() },\n        { binding: 1, resource: { buffer: this.peakBuffer } }\n      ]\n    });\n    const pass = encoder.beginComputePass({ label: 'HDR display peak' });\n    pass.setPipeline(this.pipeline);\n    pass.setBindGroup(0, bindGroup);\n    pass.dispatchWorkgroups(Math.ceil(width / 16), Math.ceil(height / 16));\n    pass.end();\n    encoder.copyBufferToBuffer(\n      this.peakBuffer,\n      0,\n      this.readbackBuffer,\n      0,\n      Uint32Array.BYTES_PER_ELEMENT\n    );\n    return true;\n  }\n\n  async resolve() {\n    if (!this.pending || this.disposed) return;\n    try {\n      await this.readbackBuffer.mapAsync(GPUMapMode.READ);\n      const mapped = this.readbackBuffer.getMappedRange();\n      const bits = new Uint32Array(mapped)[0];\n      const raw = new ArrayBuffer(Float32Array.BYTES_PER_ELEMENT);\n      new Uint32Array(raw)[0] = bits;\n      this.peakDisplayValue = new Float32Array(raw)[0];\n      this.readbackBuffer.unmap();\n    } catch {\n      if (this.readbackBuffer.mapState === 'mapped') {\n        this.readbackBuffer.unmap();\n      }\n    } finally {\n      this.pending = false;\n    }\n  }\n\n  getPeakDisplayValue() {\n    return this.peakDisplayValue;\n  }\n\n  reset() {\n    this.peakDisplayValue = 0;\n    this.lastSubmissionTime = -Infinity;\n  }\n\n  getMemoryBytes() {return this.peakBuffer.size + this.readbackBuffer.size;}\n\n  dispose() {\n    this.disposed = true;\n    void this.device.queue.onSubmittedWorkDone().catch(() => undefined).then(() => {\n      this.peakBuffer.destroy();\n      this.readbackBuffer.destroy();\n    });\n  }\n}\n\nexport default WGPUHDRPeakMeter;\n"]}

package/lib/renderer/WGPUJNLMDenoiser.d.ts

@@ -0,0 +1,61 @@
+import { type JNLMFilterShape } from './jnlm.js';
+/** WebGPU passes of the JNLM denoiser described in `jnlm.ts`. */
+export declare const JNLM_WORKGROUP_SIZE = 8;
+/**
+ * Pass 1: the guide mean and the variance of that mean from the accumulated
+ * moments. Ray buffers are stored bottom-up; guides use texture rows.
+ */
+export declare const jnlmGuideWGSL: string;
+/**
+ * Pass 2: smooth the noisy per-pixel variance over 3 x 3 pixels. Pixels with
+ * a single sample take the spatial variance of their neighborhood instead.
+ */
+export declare const jnlmVarianceWGSL = "\n\nstruct JNLMParams {\n  size: vec2u,\n  colorSensitivity: f32,\n  // Pixel spacing of the search window and patches in this pass.\n  dilation: f32,\n}\n@group(0) @binding(0) var<uniform> params: JNLMParams;\n\n@group(0) @binding(1) var<storage, read> rawGuide: array<vec2f>;\n@group(0) @binding(2) var<storage, read_write> guide: array<vec2f>;\n\n@compute @workgroup_size(8, 8, 1)\nfn main(@builtin(global_invocation_id) id: vec3u) {\n  let size = params.size;\n  if (id.x >= size.x || id.y >= size.y) { return; }\n  let center = rawGuide[id.x + id.y * size.x];\n  var varianceSum = 0.0;\n  var varianceCount = 0.0;\n  var meanSum = 0.0;\n  var meanSquareSum = 0.0;\n  for (var dy = -1; dy <= 1; dy++) {\n    for (var dx = -1; dx <= 1; dx++) {\n      let q = clamp(vec2i(id.xy) + vec2i(dx, dy), vec2i(0), vec2i(size) - 1);\n      let value = rawGuide[u32(q.x) + u32(q.y) * size.x];\n      meanSum += value.x;\n      meanSquareSum += value.x * value.x;\n      if (value.y >= 0.0) {\n        varianceSum += value.y;\n        varianceCount += 1.0;\n      }\n    }\n  }\n  var variance: f32;\n  if (center.y >= 0.0) {\n    variance = varianceSum / varianceCount;\n  } else {\n    let spatialMean = meanSum / 9.0;\n    variance = max(meanSquareSum / 9.0 - spatialMean * spatialMean, 0.0) * 9.0 / 8.0;\n  }\n  guide[id.x + id.y * size.x] = vec2f(center.x, variance);\n}\n";
+/** Copies the composed image into the filter's color buffer. */
+export declare const jnlmLoadColorWGSL = "\n\nstruct JNLMParams {\n  size: vec2u,\n  colorSensitivity: f32,\n  // Pixel spacing of the search window and patches in this pass.\n  dilation: f32,\n}\n@group(0) @binding(0) var<uniform> params: JNLMParams;\n\n@group(0) @binding(1) var colorTex: texture_2d<f32>;\n@group(0) @binding(2) var<storage, read_write> outColor: array<vec4f>;\n\n@compute @workgroup_size(8, 8, 1)\nfn main(@builtin(global_invocation_id) id: vec3u) {\n  if (id.x >= params.size.x || id.y >= params.size.y) { return; }\n  outColor[id.x + id.y * params.size.x] = textureLoad(colorTex, vec2i(id.xy), 0);\n}\n";
+/**
+ * Pass 3: the joint NL-means filter of the composed image. It also filters
+ * the guide with the same weights, with the variance of the weighted mean,
+ * for a following pass. The albedo regression corrects color only, keeping
+ * uncertain fitted texture responses out of the matching guide.
+ */
+export declare function createJNLMFilterWGSL({ searchRadius, patchRadius, albedoBandwidth, normalBandwidth }?: JNLMFilterShape): string;
+export interface JNLMInputs {
+    width: number;
+    height: number;
+    /** Normalized shadow-catcher mask; red is the catcher coverage. */
+    shadowCatcher: GPUTexture;
+    /** Ray-trace auxiliary accumulation holding the guide moments. */
+    accumulation: GPUBuffer;
+    /** Ray-trace normal accumulation; its alpha is the per-pixel sample count. */
+    normalAccumulation: GPUBuffer;
+    /** Composed linear image to filter. */
+    color: GPUTexture;
+    albedo: GPUTexture;
+    normal: GPUTexture;
+    strength: number;
+    shape: JNLMFilterShape;
+}
+export declare class WGPUJNLMDenoisePass {
+    private _device;
+    private _paramsBuffers;
+    private _guidePipeline;
+    private _variancePipeline;
+    private _loadColorPipeline;
+    private _filterPipeline?;
+    private _filterShapeKey;
+    private _guides;
+    private _colors;
+    private _width;
+    private _height;
+    constructor(device: GPUDevice);
+    private _createPipeline;
+    private _getFilterPipeline;
+    reserveCapacity(): void;
+    private _ensureBuffers;
+    private _destroyBuffers;
+    /** Encodes the denoiser and returns the RGBA32F output in texture rows. */
+    encode(encoder: GPUCommandEncoder, inputs: JNLMInputs): GPUBuffer;
+    getMemoryBytes(): number;
+    dispose(): void;
+}