@meimu/mutracer 0.1.5 → 0.3.0

package/lib/renderer/WGPUAov.js

@@ -0,0 +1,239 @@
+import { aovDisplayShader } from './wgsl/aovDisplay.wgsl.js';
+import { packAovDisplay } from './aovDisplay.js';
+import { aovSnapshotShader, aovSnapshotBinding, AOV_SNAPSHOT_BATCH_SIZE } from './wgsl/aovSnapshot.wgsl.js';
+import { AovHistory, AOV_KINDS, AOV_LAYOUT, aovMask } from './aov.js';
+/** One target's optional AOV allocations and accepted frame history. */
+export class WGPUAov {
+    device;
+    history = new AovHistory();
+    width = 0;
+    height = 0;
+    samples = 0;
+    centerReady = false;
+    accumulation;
+    center;
+    previousCenter;
+    scene;
+    frame;
+    displayPipeline;
+    displayTexture;
+    resolvePipelines = new Map();
+    scenePrimitives;
+    scenePrimitiveOffset = 0;
+    snapshotGeneration = 0;
+    snapshotPool = new Map();
+    snapshotLeases = new WeakMap();
+    snapshotBuffers = new Set();
+    constructor(device) {
+        this.device = device;
+    }
+    reset() {
+        if (this.centerReady && this.center && this.previousCenter) {
+            const encoder = this.device.createCommandEncoder();
+            encoder.copyBufferToBuffer(this.center, 0, this.previousCenter, 0, this.center.size);
+            this.device.queue.submit([encoder.finish()]);
+        }
+        this.samples = 0;
+        this.centerReady = false;
+        this.frame = undefined;
+    }
+    resize(width, height) {
+        if (width === this.width && height === this.height)
+            return;
+        if (width * height * AOV_LAYOUT.accumulatedBytes > this.device.limits.maxStorageBufferBindingSize || width * height * AOV_LAYOUT.accumulatedBytes > this.device.limits.maxBufferSize)
+            throw new RangeError('AOV accumulation exceeds device buffer limits');
+        this.dispose();
+        this.history.clear();
+        this.reset();
+        this.width = width;
+        this.height = height;
+        const usage = GPUBufferUsage.STORAGE | GPUBufferUsage.COPY_DST | GPUBufferUsage.COPY_SRC;
+        this.accumulation = this.device.createBuffer({ label: 'AOV accumulation', size: width * height * AOV_LAYOUT.accumulatedBytes, usage });
+        this.previousCenter = this.device.createBuffer({ label: 'Previous AOV center', size: width * height * AOV_LAYOUT.centerBytes, usage });
+        this.center = this.device.createBuffer({ label: 'AOV center', size: width * height * AOV_LAYOUT.centerBytes, usage });
+    }
+    setScene(words, primitiveIds) {
+        const bytes = words.byteLength + (primitiveIds?.byteLength ?? 0);
+        if (bytes > this.device.limits.maxStorageBufferBindingSize)
+            throw new RangeError('AOV scene metadata exceeds device storage binding limit');
+        let replaced = false;
+        if (!this.scene || this.scene.size < bytes) {
+            this.scene?.destroy();
+            this.scene = this.device.createBuffer({ label: 'AOV identities/history', size: Math.max(16, bytes), usage: GPUBufferUsage.STORAGE | GPUBufferUsage.COPY_DST });
+            replaced = true;
+        }
+        this.device.queue.writeBuffer(this.scene, 0, words);
+        // Primitive-cache arrays are immutable. Camera, material and rigid edits
+        // update only the prefix; topology/storage replacements change the tail's
+        // identity, and instance/material counts can move its byte offset.
+        if (primitiveIds && (replaced || this.scenePrimitives !== primitiveIds || this.scenePrimitiveOffset !== words.byteLength)) {
+            this.device.queue.writeBuffer(this.scene, words.byteLength, primitiveIds);
+        }
+        this.scenePrimitives = primitiveIds;
+        this.scenePrimitiveOffset = words.byteLength;
+    }
+    get bytes() { return (this.accumulation?.size ?? 0) + (this.center?.size ?? 0) + (this.previousCenter?.size ?? 0) + (this.scene?.size ?? 0) + (this.displayTexture ? this.width * this.height * 8 : 0) + [...this.snapshotBuffers].reduce((bytes, buffer) => bytes + buffer.size, 0); }
+    snapshot(kinds) {
+        if (!this.accumulation || !this.center || !this.samples)
+            throw new Error('AOV output is not ready');
+        const device = this.device;
+        aovMask(kinds);
+        const unique = AOV_KINDS.filter(kind => kinds.includes(kind));
+        const batches = [];
+        for (let offset = 0; offset < unique.length; offset += AOV_SNAPSHOT_BATCH_SIZE) {
+            const batch = unique.slice(offset, offset + AOV_SNAPSHOT_BATCH_SIZE), key = batch.join(',');
+            let pipeline = this.resolvePipelines.get(key);
+            if (!pipeline) {
+                pipeline = device.createComputePipeline({ label: 'AOV snapshot', layout: 'auto',
+                    compute: { module: device.createShaderModule({ code: aovSnapshotShader(batch) }), entryPoint: 'main' } });
+                this.resolvePipelines.set(key, pipeline);
+            }
+            batches.push({ kinds: batch, pipeline });
+        }
+        const buffers = {};
+        if (!batches.length)
+            return buffers;
+        const parameters = device.createBuffer({ size: 16, usage: GPUBufferUsage.UNIFORM, mappedAtCreation: true });
+        new Uint32Array(parameters.getMappedRange()).set([this.width, this.height, this.samples, 0]);
+        parameters.unmap();
+        try {
+            const encoder = device.createCommandEncoder({ label: 'AOV snapshot' }), pass = encoder.beginComputePass();
+            for (const { kinds: batch, pipeline } of batches) {
+                const entries = [
+                    { binding: 0, resource: { buffer: this.accumulation } },
+                    { binding: 1, resource: { buffer: this.center } },
+                    { binding: 3, resource: { buffer: parameters } }
+                ];
+                batch.forEach((kind, index) => {
+                    const buffer = this.snapshotPool.get(kind) ?? device.createBuffer({ label: `Custom ${kind} input`, size: this.width * this.height * 16,
+                        usage: GPUBufferUsage.STORAGE | GPUBufferUsage.COPY_SRC | GPUBufferUsage.COPY_DST });
+                    this.snapshotPool.delete(kind);
+                    this.snapshotLeases.set(buffer, { kind, generation: this.snapshotGeneration });
+                    this.snapshotBuffers.add(buffer);
+                    buffers[kind] = buffer;
+                    entries.push({ binding: aovSnapshotBinding(index), resource: { buffer } });
+                });
+                pass.setPipeline(pipeline);
+                pass.setBindGroup(0, device.createBindGroup({ layout: pipeline.getBindGroupLayout(0), entries }));
+                pass.dispatchWorkgroups(Math.ceil(this.width / 8), Math.ceil(this.height / 8));
+            }
+            pass.end();
+            device.queue.submit([encoder.finish()]);
+            return buffers;
+        }
+        catch (error) {
+            // Nothing was submitted, so these unfinished outputs can be retired now.
+            Object.values(buffers).forEach(buffer => {
+                if (buffer) {
+                    this.snapshotLeases.delete(buffer);
+                    this.snapshotBuffers.delete(buffer);
+                    buffer.destroy();
+                }
+            });
+            throw error;
+        }
+        finally {
+            parameters.destroy();
+        }
+    }
+    // Call only after the consumer and its queued GPU work have finished. Each
+    // in-flight denoiser owns separate outputs, including cancelled jobs whose
+    // callbacks have not settled. Keep at most one reusable buffer per kind.
+    releaseSnapshot(buffers) {
+        for (const buffer of Object.values(buffers)) {
+            if (!buffer)
+                continue;
+            const lease = this.snapshotLeases.get(buffer);
+            if (!lease)
+                continue;
+            this.snapshotLeases.delete(buffer);
+            if (lease.generation === this.snapshotGeneration && !this.snapshotPool.has(lease.kind)) {
+                this.snapshotPool.set(lease.kind, buffer);
+            }
+            else {
+                this.snapshotBuffers.delete(buffer);
+                buffer.destroy();
+            }
+        }
+    }
+    visualize(kind, options, linear) {
+        if (!this.accumulation || !this.center || !this.samples)
+            throw new Error('AOV output is not ready');
+        const device = this.device;
+        this.displayPipeline ??= device.createComputePipeline({ label: 'AOV visualization', layout: 'auto', compute: { module: device.createShaderModule({ code: aovDisplayShader('rgba16float') }), entryPoint: 'main' } });
+        if (!this.displayTexture || this.displayTexture.width !== this.width || this.displayTexture.height !== this.height) {
+            this.displayTexture?.destroy();
+            this.displayTexture = device.createTexture({ label: 'AOV display image', size: [this.width, this.height], format: 'rgba16float', usage: GPUTextureUsage.STORAGE_BINDING | GPUTextureUsage.TEXTURE_BINDING | GPUTextureUsage.COPY_SRC });
+        }
+        const words = packAovDisplay(this.width, this.height, this.samples, kind, options, linear);
+        const parameters = device.createBuffer({ size: 48, usage: GPUBufferUsage.UNIFORM, mappedAtCreation: true });
+        new Uint32Array(parameters.getMappedRange()).set(words);
+        parameters.unmap();
+        const group = device.createBindGroup({ layout: this.displayPipeline.getBindGroupLayout(0), entries: [{ binding: 0, resource: { buffer: this.accumulation } }, { binding: 1, resource: { buffer: this.center } }, { binding: 2, resource: this.displayTexture.createView() }, { binding: 3, resource: { buffer: parameters } }] });
+        const encoder = device.createCommandEncoder(), pass = encoder.beginComputePass();
+        pass.setPipeline(this.displayPipeline);
+        pass.setBindGroup(0, group);
+        pass.dispatchWorkgroups(Math.ceil(this.width / 8), Math.ceil(this.height / 8));
+        pass.end();
+        device.queue.submit([encoder.finish()]);
+        parameters.destroy();
+        return this.displayTexture;
+    }
+    async read(kind) {
+        const center = kind === 'id' || kind === 'motion';
+        const source = center ? this.center : this.accumulation;
+        const staging = this.device.createBuffer({ size: source.size, usage: GPUBufferUsage.MAP_READ | GPUBufferUsage.COPY_DST });
+        try {
+            const encoder = this.device.createCommandEncoder();
+            encoder.copyBufferToBuffer(source, 0, staging, 0, source.size);
+            this.device.queue.submit([encoder.finish()]);
+            await staging.mapAsync(GPUMapMode.READ);
+            const f = new Float32Array(staging.getMappedRange()), u = new Uint32Array(f.buffer);
+            const data = kind === 'id' ? new Uint32Array(this.width * this.height * 4) : new Float32Array(this.width * this.height * 4);
+            for (let y = 0; y < this.height; y++)
+                for (let x = 0; x < this.width; x++) {
+                    const target = (y * this.width + x) * 4;
+                    // Tracing uses OpenGL camera coordinates; API images are top row first.
+                    const pixel = ((this.height - 1 - y) * this.width + x), o = pixel * (center ? 8 : 16);
+                    if (kind === 'id')
+                        data.set(u.subarray(o, o + 4), target);
+                    else if (kind === 'motion')
+                        data.set([f[o + 4], f[o + 5], 0, f[o + 7]], target);
+                    else {
+                        const hits = f[o + 7], coverage = hits / Math.max(1, this.samples), inv = hits > 0 ? 1 / hits : 0;
+                        if (kind === 'depth')
+                            data.set([f[o] * inv, 0, 0, coverage], target);
+                        else if (kind === 'position')
+                            data.set([f[o + 4] * inv, f[o + 5] * inv, f[o + 6] * inv, coverage], target);
+                        else if (kind === 'roughness')
+                            data.set([f[o + 1] * inv, f[o + 2] * inv, f[o + 3] * inv, coverage], target);
+                        else {
+                            const offset = kind === 'diffuseAlbedo' ? o + 8 : o + 12, valid = f[offset + 3];
+                            const scale = valid > 0 ? 1 / valid : 0;
+                            data.set([f[offset] * scale, f[offset + 1] * scale, f[offset + 2] * scale, valid / Math.max(1, this.samples)], target);
+                        }
+                    }
+                }
+            return { width: this.width, height: this.height, data };
+        }
+        finally {
+            if (staging.mapState === 'mapped')
+                staging.unmap();
+            staging.destroy();
+        }
+    }
+    dispose() {
+        const texture = this.displayTexture;
+        this.displayTexture = undefined;
+        this.scenePrimitives = undefined;
+        this.scenePrimitiveOffset = 0;
+        this.snapshotGeneration++;
+        const pooled = [...this.snapshotPool.values()];
+        this.snapshotPool.clear();
+        pooled.forEach(buffer => this.snapshotBuffers.delete(buffer));
+        const buffers = [this.accumulation, this.center, this.scene, this.previousCenter, ...pooled];
+        this.accumulation = this.center = this.scene = this.previousCenter = undefined;
+        void this.device.queue.onSubmittedWorkDone().catch(() => undefined).then(() => { texture?.destroy(); buffers.forEach(buffer => buffer?.destroy()); });
+    }
+}
+//# sourceMappingURL=WGPUAov.js.map

package/lib/renderer/WGPUAov.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"WGPUAov.js","sourceRoot":"","sources":["../../src/renderer/WGPUAov.ts"],"names":[],"mappings":"AAAA,OAAO,EAAC,gBAAgB,EAAC,MAAM,2BAA2B,CAAC;AAC3D,OAAO,EAAC,cAAc,EAAyB,MAAM,iBAAiB,CAAC;AACvE,OAAO,EAAC,iBAAiB,EAAE,kBAAkB,EAAE,uBAAuB,EAAC,MAAM,4BAA4B,CAAC;AAC1G,OAAO,EAAC,UAAU,EAAE,SAAS,EAAE,UAAU,EAAE,OAAO,EAAsC,MAAM,UAAU,CAAC;AACzG,wEAAwE;AACxE,MAAM,OAAO,OAAO;IAcG;IAbZ,OAAO,GAAG,IAAI,UAAU,EAAE,CAAC;IACpC,KAAK,GAAG,CAAC,CAAC;IAAC,MAAM,GAAG,CAAC,CAAC;IAAC,OAAO,GAAG,CAAC,CAAC;IAAC,WAAW,GAAG,KAAK,CAAC;IACxD,YAAY,CAAa;IAAC,MAAM,CAAa;IAAC,cAAc,CAAa;IAAC,KAAK,CAAa;IAC5F,KAAK,CAAoB;IACjB,eAAe,CAAsB;IACrC,cAAc,CAAc;IACnB,gBAAgB,GAAG,IAAI,GAAG,EAA8B,CAAC;IAClE,eAAe,CAAe;IAC9B,oBAAoB,GAAG,CAAC,CAAC;IACzB,kBAAkB,GAAG,CAAC,CAAC;IACd,YAAY,GAAG,IAAI,GAAG,EAAsB,CAAC;IAC7C,cAAc,GAAG,IAAI,OAAO,EAAkD,CAAC;IAC/E,eAAe,GAAG,IAAI,GAAG,EAAa,CAAC;IACxD,YAAqB,MAAiB;QAAjB,WAAM,GAAN,MAAM,CAAW;IAAG,CAAC;IAC1C,KAAK;QACH,IAAI,IAAI,CAAC,WAAW,IAAI,IAAI,CAAC,MAAM,IAAI,IAAI,CAAC,cAAc,EAAE,CAAC;YAC3D,MAAM,OAAO,GAAC,IAAI,CAAC,MAAM,CAAC,oBAAoB,EAAE,CAAC;YAAA,OAAO,CAAC,kBAAkB,CAAC,IAAI,CAAC,MAAM,EAAC,CAAC,EAAC,IAAI,CAAC,cAAc,EAAC,CAAC,EAAC,IAAI,CAAC,MAAM,CAAC,IAAI,CAAC,CAAC;YAAA,IAAI,CAAC,MAAM,CAAC,KAAK,CAAC,MAAM,CAAC,CAAC,OAAO,CAAC,MAAM,EAAE,CAAC,CAAC,CAAC;QACjL,CAAC;QACD,IAAI,CAAC,OAAO,GAAG,CAAC,CAAC;QAAC,IAAI,CAAC,WAAW,GAAG,KAAK,CAAC;QAAC,IAAI,CAAC,KAAK,GAAG,SAAS,CAAC;IAAC,CAAC;IACvE,MAAM,CAAC,KAAa,EAAE,MAAc;QAClC,IAAI,KAAK,KAAK,IAAI,CAAC,KAAK,IAAI,MAAM,KAAK,IAAI,CAAC,MAAM;YAAE,OAAO;QAC3D,IAAI,KAAK,GAAC,MAAM,GAAC,UAAU,CAAC,gBAAgB,GAAC,IAAI,CAAC,MAAM,CAAC,MAAM,CAAC,2BAA2B,IAAI,KAAK,GAAC,MAAM,GAAC,UAAU,CAAC,gBAAgB,GAAC,IAAI,CAAC,MAAM,CAAC,MAAM,CAAC,aAAa;YAAE,MAAM,IAAI,UAAU,CAAC,+CAA+C,CAAC,CAAC;QAChP,IAAI,CAAC,OAAO,EAAE,CAAC;QAAC,IAAI,CAAC,OAAO,CAAC,KAAK,EAAE,CAAC;QAAC,IAAI,CAAC,KAAK,EAAE,CAAC;QAAC,IAAI,CAAC,KAAK,GAAG,KAAK,CAAC;QAAC,IAAI,CAAC,MAAM,GAAG,MAAM,CAAC;QAC7F,MAAM,KAAK,GAAG,cAAc,CAAC,OAAO,GAAG,cAAc,CAAC,QAAQ,GAAG,cAAc,CAAC,QAAQ,CAAC;QACzF,IAAI,CAAC,YAAY,GAAG,IAAI,CAAC,MAAM,CAAC,YAAY,CAAC,EAAC,KAAK,EAAE,kBAAkB,EAAE,IAAI,EAAE,KAAK,GAAG,MAAM,GAAG,UAAU,CAAC,gBAAgB,EAAE,KAAK,EAAC,CAAC,CAAC;QACrI,IAAI,CAAC,cAAc,GAAG,IAAI,CAAC,MAAM,CAAC,YAAY,CAAC,EAAC,KAAK,EAAE,qBAAqB,EAAE,IAAI,EAAE,KAAK,GAAC,MAAM,GAAC,UAAU,CAAC,WAAW,EAAE,KAAK,EAAC,CAAC,CAAC;QACjI,IAAI,CAAC,MAAM,GAAG,IAAI,CAAC,MAAM,CAAC,YAAY,CAAC,EAAC,KAAK,EAAE,YAAY,EAAE,IAAI,EAAE,KAAK,GAAG,MAAM,GAAG,UAAU,CAAC,WAAW,EAAE,KAAK,EAAC,CAAC,CAAC;IACtH,CAAC;IACD,QAAQ,CAAC,KAAkB,EAAE,YAA0B;QACrD,MAAM,KAAK,GAAG,KAAK,CAAC,UAAU,GAAG,CAAC,YAAY,EAAE,UAAU,IAAI,CAAC,CAAC,CAAC;QACjE,IAAI,KAAK,GAAG,IAAI,CAAC,MAAM,CAAC,MAAM,CAAC,2BAA2B;YAAE,MAAM,IAAI,UAAU,CAAC,yDAAyD,CAAC,CAAC;QAC5I,IAAI,QAAQ,GAAG,KAAK,CAAC;QACrB,IAAI,CAAC,IAAI,CAAC,KAAK,IAAI,IAAI,CAAC,KAAK,CAAC,IAAI,GAAG,KAAK,EAAE,CAAC;YAC3C,IAAI,CAAC,KAAK,EAAE,OAAO,EAAE,CAAC;YACtB,IAAI,CAAC,KAAK,GAAG,IAAI,CAAC,MAAM,CAAC,YAAY,CAAC,EAAC,KAAK,EAAE,wBAAwB,EAAE,IAAI,EAAE,IAAI,CAAC,GAAG,CAAC,EAAE,EAAE,KAAK,CAAC,EAAE,KAAK,EAAE,cAAc,CAAC,OAAO,GAAG,cAAc,CAAC,QAAQ,EAAC,CAAC,CAAC;YAC7J,QAAQ,GAAG,IAAI,CAAC;QAClB,CAAC;QACD,IAAI,CAAC,MAAM,CAAC,KAAK,CAAC,WAAW,CAAC,IAAI,CAAC,KAAK,EAAE,CAAC,EAAE,KAAK,CAAC,CAAC;QACpD,yEAAyE;QACzE,0EAA0E;QAC1E,mEAAmE;QACnE,IAAI,YAAY,IAAI,CAAC,QAAQ,IAAI,IAAI,CAAC,eAAe,KAAK,YAAY,IAAI,IAAI,CAAC,oBAAoB,KAAK,KAAK,CAAC,UAAU,CAAC,EAAE,CAAC;YAC1H,IAAI,CAAC,MAAM,CAAC,KAAK,CAAC,WAAW,CAAC,IAAI,CAAC,KAAK,EAAE,KAAK,CAAC,UAAU,EAAE,YAAY,CAAC,CAAC;QAC5E,CAAC;QACD,IAAI,CAAC,eAAe,GAAG,YAAY,CAAC;QAAC,IAAI,CAAC,oBAAoB,GAAG,KAAK,CAAC,UAAU,CAAC;IACpF,CAAC;IACD,IAAI,KAAK,KAAK,OAAO,CAAC,IAAI,CAAC,YAAY,EAAE,IAAI,IAAI,CAAC,CAAC,GAAG,CAAC,IAAI,CAAC,MAAM,EAAE,IAAI,IAAI,CAAC,CAAC,GAAG,CAAC,IAAI,CAAC,cAAc,EAAE,IAAI,IAAI,CAAC,CAAC,GAAG,CAAC,IAAI,CAAC,KAAK,EAAE,IAAI,IAAI,CAAC,CAAC,GAAG,CAAC,IAAI,CAAC,cAAc,CAAC,CAAC,CAAC,IAAI,CAAC,KAAK,GAAC,IAAI,CAAC,MAAM,GAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,GAAG,CAAC,GAAG,IAAI,CAAC,eAAe,CAAC,CAAC,MAAM,CAAC,CAAC,KAAK,EAAE,MAAM,EAAE,EAAE,CAAC,KAAK,GAAG,MAAM,CAAC,IAAI,EAAE,CAAC,CAAC,CAAC,CAAC,CAAC;IACnR,QAAQ,CAAC,KAAyB;QAChC,IAAG,CAAC,IAAI,CAAC,YAAY,IAAI,CAAC,IAAI,CAAC,MAAM,IAAI,CAAC,IAAI,CAAC,OAAO;YAAE,MAAM,IAAI,KAAK,CAAC,yBAAyB,CAAC,CAAC;QACnG,MAAM,MAAM,GAAG,IAAI,CAAC,MAAM,CAAC;QAC3B,OAAO,CAAC,KAAK,CAAC,CAAC;QACf,MAAM,MAAM,GAAG,SAAS,CAAC,MAAM,CAAC,IAAI,CAAC,EAAE,CAAC,KAAK,CAAC,QAAQ,CAAC,IAAI,CAAC,CAAC,CAAC;QAC9D,MAAM,OAAO,GAAuD,EAAE,CAAC;QACvE,KAAK,IAAI,MAAM,GAAG,CAAC,EAAE,MAAM,GAAG,MAAM,CAAC,MAAM,EAAE,MAAM,IAAI,uBAAuB,EAAE,CAAC;YAC/E,MAAM,KAAK,GAAG,MAAM,CAAC,KAAK,CAAC,MAAM,EAAE,MAAM,GAAG,uBAAuB,CAAC,EAAE,GAAG,GAAG,KAAK,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC;YAC5F,IAAI,QAAQ,GAAG,IAAI,CAAC,gBAAgB,CAAC,GAAG,CAAC,GAAG,CAAC,CAAC;YAC9C,IAAI,CAAC,QAAQ,EAAE,CAAC;gBACd,QAAQ,GAAG,MAAM,CAAC,qBAAqB,CAAC,EAAC,KAAK,EAAE,cAAc,EAAE,MAAM,EAAE,MAAM;oBAC5E,OAAO,EAAE,EAAC,MAAM,EAAE,MAAM,CAAC,kBAAkB,CAAC,EAAC,IAAI,EAAE,iBAAiB,CAAC,KAAK,CAAC,EAAC,CAAC,EAAE,UAAU,EAAE,MAAM,EAAC,EAAC,CAAC,CAAC;gBACvG,IAAI,CAAC,gBAAgB,CAAC,GAAG,CAAC,GAAG,EAAE,QAAQ,CAAC,CAAC;YAC3C,CAAC;YACD,OAAO,CAAC,IAAI,CAAC,EAAC,KAAK,EAAE,KAAK,EAAE,QAAQ,EAAC,CAAC,CAAC;QACzC,CAAC;QACD,MAAM,OAAO,GAAwC,EAAE,CAAC;QACxD,IAAI,CAAC,OAAO,CAAC,MAAM;YAAE,OAAO,OAAO,CAAC;QACpC,MAAM,UAAU,GAAG,MAAM,CAAC,YAAY,CAAC,EAAC,IAAI,EAAE,EAAE,EAAE,KAAK,EAAE,cAAc,CAAC,OAAO,EAAE,gBAAgB,EAAE,IAAI,EAAC,CAAC,CAAC;QAC1G,IAAI,WAAW,CAAC,UAAU,CAAC,cAAc,EAAE,CAAC,CAAC,GAAG,CAAC,CAAC,IAAI,CAAC,KAAK,EAAE,IAAI,CAAC,MAAM,EAAE,IAAI,CAAC,OAAO,EAAE,CAAC,CAAC,CAAC,CAAC;QAAC,UAAU,CAAC,KAAK,EAAE,CAAC;QACjH,IAAI,CAAC;YACH,MAAM,OAAO,GAAG,MAAM,CAAC,oBAAoB,CAAC,EAAC,KAAK,EAAE,cAAc,EAAC,CAAC,EAAE,IAAI,GAAG,OAAO,CAAC,gBAAgB,EAAE,CAAC;YACxG,KAAK,MAAM,EAAC,KAAK,EAAE,KAAK,EAAE,QAAQ,EAAC,IAAI,OAAO,EAAE,CAAC;gBAC/C,MAAM,OAAO,GAAwB;oBACnC,EAAC,OAAO,EAAE,CAAC,EAAE,QAAQ,EAAE,EAAC,MAAM,EAAE,IAAI,CAAC,YAAY,EAAC,EAAC;oBACnD,EAAC,OAAO,EAAE,CAAC,EAAE,QAAQ,EAAE,EAAC,MAAM,EAAE,IAAI,CAAC,MAAM,EAAC,EAAC;oBAC7C,EAAC,OAAO,EAAE,CAAC,EAAE,QAAQ,EAAE,EAAC,MAAM,EAAE,UAAU,EAAC,EAAC;iBAC7C,CAAC;gBACF,KAAK,CAAC,OAAO,CAAC,CAAC,IAAI,EAAE,KAAK,EAAE,EAAE;oBAC5B,MAAM,MAAM,GAAG,IAAI,CAAC,YAAY,CAAC,GAAG,CAAC,IAAI,CAAC,IAAI,MAAM,CAAC,YAAY,CAAC,EAAC,KAAK,EAAE,UAAU,IAAI,QAAQ,EAAE,IAAI,EAAE,IAAI,CAAC,KAAK,GAAG,IAAI,CAAC,MAAM,GAAG,EAAE;wBACnI,KAAK,EAAE,cAAc,CAAC,OAAO,GAAG,cAAc,CAAC,QAAQ,GAAG,cAAc,CAAC,QAAQ,EAAC,CAAC,CAAC;oBACtF,IAAI,CAAC,YAAY,CAAC,MAAM,CAAC,IAAI,CAAC,CAAC;oBAC/B,IAAI,CAAC,cAAc,CAAC,GAAG,CAAC,MAAM,EAAE,EAAC,IAAI,EAAE,UAAU,EAAE,IAAI,CAAC,kBAAkB,EAAC,CAAC,CAAC;oBAC7E,IAAI,CAAC,eAAe,CAAC,GAAG,CAAC,MAAM,CAAC,CAAC;oBACjC,OAAO,CAAC,IAAI,CAAC,GAAG,MAAM,CAAC;oBACvB,OAAO,CAAC,IAAI,CAAC,EAAC,OAAO,EAAE,kBAAkB,CAAC,KAAK,CAAC,EAAE,QAAQ,EAAE,EAAC,MAAM,EAAC,EAAC,CAAC,CAAC;gBACzE,CAAC,CAAC,CAAC;gBACH,IAAI,CAAC,WAAW,CAAC,QAAQ,CAAC,CAAC;gBAC3B,IAAI,CAAC,YAAY,CAAC,CAAC,EAAE,MAAM,CAAC,eAAe,CAAC,EAAC,MAAM,EAAE,QAAQ,CAAC,kBAAkB,CAAC,CAAC,CAAC,EAAE,OAAO,EAAC,CAAC,CAAC,CAAC;gBAChG,IAAI,CAAC,kBAAkB,CAAC,IAAI,CAAC,IAAI,CAAC,IAAI,CAAC,KAAK,GAAG,CAAC,CAAC,EAAE,IAAI,CAAC,IAAI,CAAC,IAAI,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC,CAAC;YACjF,CAAC;YACD,IAAI,CAAC,GAAG,EAAE,CAAC;YAAC,MAAM,CAAC,KAAK,CAAC,MAAM,CAAC,CAAC,OAAO,CAAC,MAAM,EAAE,CAAC,CAAC,CAAC;YACpD,OAAO,OAAO,CAAC;QACjB,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,yEAAyE;YACzE,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC,OAAO,CAAC,MAAM,CAAC,EAAE;gBACtC,IAAI,MAAM,EAAE,CAAC;oBAAA,IAAI,CAAC,cAAc,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC;oBAAC,IAAI,CAAC,eAAe,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC;oBAAC,MAAM,CAAC,OAAO,EAAE,CAAC;gBAAA,CAAC;YAC1G,CAAC,CAAC,CAAC;YACH,MAAM,KAAK,CAAC;QACd,CAAC;gBAAS,CAAC;YAAC,UAAU,CAAC,OAAO,EAAE,CAAC;QAAC,CAAC;IACrC,CAAC;IACD,2EAA2E;IAC3E,2EAA2E;IAC3E,yEAAyE;IACzE,eAAe,CAAC,OAA4C;QAC1D,KAAK,MAAM,MAAM,IAAI,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,EAAE,CAAC;YAC5C,IAAI,CAAC,MAAM;gBAAE,SAAS;YACtB,MAAM,KAAK,GAAG,IAAI,CAAC,cAAc,CAAC,GAAG,CAAC,MAAM,CAAC,CAAC;YAC9C,IAAI,CAAC,KAAK;gBAAE,SAAS;YACrB,IAAI,CAAC,cAAc,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC;YACnC,IAAI,KAAK,CAAC,UAAU,KAAK,IAAI,CAAC,kBAAkB,IAAI,CAAC,IAAI,CAAC,YAAY,CAAC,GAAG,CAAC,KAAK,CAAC,IAAI,CAAC,EAAE,CAAC;gBACvF,IAAI,CAAC,YAAY,CAAC,GAAG,CAAC,KAAK,CAAC,IAAI,EAAE,MAAM,CAAC,CAAC;YAC5C,CAAC;iBAAM,CAAC;gBAAC,IAAI,CAAC,eAAe,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC;gBAAC,MAAM,CAAC,OAAO,EAAE,CAAC;YAAC,CAAC;QACnE,CAAC;IACH,CAAC;IACD,SAAS,CAAC,IAAa,EAAE,OAA0B,EAAE,MAAe;QAClE,IAAG,CAAC,IAAI,CAAC,YAAY,IAAI,CAAC,IAAI,CAAC,MAAM,IAAI,CAAC,IAAI,CAAC,OAAO;YAAE,MAAM,IAAI,KAAK,CAAC,yBAAyB,CAAC,CAAC;QACnG,MAAM,MAAM,GAAC,IAAI,CAAC,MAAM,CAAC;QACzB,IAAI,CAAC,eAAe,KAAK,MAAM,CAAC,qBAAqB,CAAC,EAAC,KAAK,EAAC,mBAAmB,EAAC,MAAM,EAAC,MAAM,EAAC,OAAO,EAAC,EAAC,MAAM,EAAC,MAAM,CAAC,kBAAkB,CAAC,EAAC,IAAI,EAAC,gBAAgB,CAAC,aAAa,CAAC,EAAC,CAAC,EAAC,UAAU,EAAC,MAAM,EAAC,EAAC,CAAC,CAAC;QACtM,IAAG,CAAC,IAAI,CAAC,cAAc,IAAI,IAAI,CAAC,cAAc,CAAC,KAAK,KAAG,IAAI,CAAC,KAAK,IAAI,IAAI,CAAC,cAAc,CAAC,MAAM,KAAG,IAAI,CAAC,MAAM,EAAC,CAAC;YAC7G,IAAI,CAAC,cAAc,EAAE,OAAO,EAAE,CAAC;YAC/B,IAAI,CAAC,cAAc,GAAC,MAAM,CAAC,aAAa,CAAC,EAAC,KAAK,EAAC,mBAAmB,EAAC,IAAI,EAAC,CAAC,IAAI,CAAC,KAAK,EAAC,IAAI,CAAC,MAAM,CAAC,EAAC,MAAM,EAAC,aAAa,EAAC,KAAK,EAAC,eAAe,CAAC,eAAe,GAAC,eAAe,CAAC,eAAe,GAAC,eAAe,CAAC,QAAQ,EAAC,CAAC,CAAC;QAC1N,CAAC;QACD,MAAM,KAAK,GAAC,cAAc,CAAC,IAAI,CAAC,KAAK,EAAC,IAAI,CAAC,MAAM,EAAC,IAAI,CAAC,OAAO,EAAC,IAAI,EAAC,OAAO,EAAC,MAAM,CAAC,CAAC;QACpF,MAAM,UAAU,GAAC,MAAM,CAAC,YAAY,CAAC,EAAC,IAAI,EAAC,EAAE,EAAC,KAAK,EAAC,cAAc,CAAC,OAAO,EAAC,gBAAgB,EAAC,IAAI,EAAC,CAAC,CAAC;QAAA,IAAI,WAAW,CAAC,UAAU,CAAC,cAAc,EAAE,CAAC,CAAC,GAAG,CAAC,KAAK,CAAC,CAAC;QAAA,UAAU,CAAC,KAAK,EAAE,CAAC;QAC9K,MAAM,KAAK,GAAC,MAAM,CAAC,eAAe,CAAC,EAAC,MAAM,EAAC,IAAI,CAAC,eAAe,CAAC,kBAAkB,CAAC,CAAC,CAAC,EAAC,OAAO,EAAC,CAAC,EAAC,OAAO,EAAC,CAAC,EAAC,QAAQ,EAAC,EAAC,MAAM,EAAC,IAAI,CAAC,YAAY,EAAC,EAAC,EAAC,EAAC,OAAO,EAAC,CAAC,EAAC,QAAQ,EAAC,EAAC,MAAM,EAAC,IAAI,CAAC,MAAM,EAAC,EAAC,EAAC,EAAC,OAAO,EAAC,CAAC,EAAC,QAAQ,EAAC,IAAI,CAAC,cAAc,CAAC,UAAU,EAAE,EAAC,EAAC,EAAC,OAAO,EAAC,CAAC,EAAC,QAAQ,EAAC,EAAC,MAAM,EAAC,UAAU,EAAC,EAAC,CAAC,EAAC,CAAC,CAAC;QAC3R,MAAM,OAAO,GAAC,MAAM,CAAC,oBAAoB,EAAE,EAAC,IAAI,GAAC,OAAO,CAAC,gBAAgB,EAAE,CAAC;QAAA,IAAI,CAAC,WAAW,CAAC,IAAI,CAAC,eAAe,CAAC,CAAC;QAAA,IAAI,CAAC,YAAY,CAAC,CAAC,EAAC,KAAK,CAAC,CAAC;QAAA,IAAI,CAAC,kBAAkB,CAAC,IAAI,CAAC,IAAI,CAAC,IAAI,CAAC,KAAK,GAAC,CAAC,CAAC,EAAC,IAAI,CAAC,IAAI,CAAC,IAAI,CAAC,MAAM,GAAC,CAAC,CAAC,CAAC,CAAC;QAAA,IAAI,CAAC,GAAG,EAAE,CAAC;QAAA,MAAM,CAAC,KAAK,CAAC,MAAM,CAAC,CAAC,OAAO,CAAC,MAAM,EAAE,CAAC,CAAC,CAAC;QAAA,UAAU,CAAC,OAAO,EAAE,CAAC;QAChS,OAAO,IAAI,CAAC,cAAc,CAAC;IAC7B,CAAC;IACD,KAAK,CAAC,IAAI,CAAC,IAAa;QACtB,MAAM,MAAM,GAAG,IAAI,KAAK,IAAI,IAAI,IAAI,KAAK,QAAQ,CAAC;QAClD,MAAM,MAAM,GAAG,MAAM,CAAC,CAAC,CAAC,IAAI,CAAC,MAAO,CAAC,CAAC,CAAC,IAAI,CAAC,YAAa,CAAC;QAC1D,MAAM,OAAO,GAAG,IAAI,CAAC,MAAM,CAAC,YAAY,CAAC,EAAC,IAAI,EAAE,MAAM,CAAC,IAAI,EAAE,KAAK,EAAE,cAAc,CAAC,QAAQ,GAAG,cAAc,CAAC,QAAQ,EAAC,CAAC,CAAC;QACxH,IAAI,CAAC;YACH,MAAM,OAAO,GAAG,IAAI,CAAC,MAAM,CAAC,oBAAoB,EAAE,CAAC;YAAC,OAAO,CAAC,kBAAkB,CAAC,MAAM,EAAE,CAAC,EAAE,OAAO,EAAE,CAAC,EAAE,MAAM,CAAC,IAAI,CAAC,CAAC;YACnH,IAAI,CAAC,MAAM,CAAC,KAAK,CAAC,MAAM,CAAC,CAAC,OAAO,CAAC,MAAM,EAAE,CAAC,CAAC,CAAC;YAAC,MAAM,OAAO,CAAC,QAAQ,CAAC,UAAU,CAAC,IAAI,CAAC,CAAC;YACtF,MAAM,CAAC,GAAG,IAAI,YAAY,CAAC,OAAO,CAAC,cAAc,EAAE,CAAC,EAAE,CAAC,GAAG,IAAI,WAAW,CAAC,CAAC,CAAC,MAAM,CAAC,CAAC;YACpF,MAAM,IAAI,GAAG,IAAI,KAAK,IAAI,CAAC,CAAC,CAAC,IAAI,WAAW,CAAC,IAAI,CAAC,KAAK,GAAC,IAAI,CAAC,MAAM,GAAC,CAAC,CAAC,CAAC,CAAC,CAAC,IAAI,YAAY,CAAC,IAAI,CAAC,KAAK,GAAC,IAAI,CAAC,MAAM,GAAC,CAAC,CAAC,CAAC;YACpH,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,IAAI,CAAC,MAAM,EAAE,CAAC,EAAE;gBAAE,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,IAAI,CAAC,KAAK,EAAE,CAAC,EAAE,EAAE,CAAC;oBAC1E,MAAM,MAAM,GAAG,CAAC,CAAC,GAAG,IAAI,CAAC,KAAK,GAAG,CAAC,CAAC,GAAG,CAAC,CAAC;oBACxC,wEAAwE;oBACxE,MAAM,KAAK,GAAG,CAAC,CAAC,IAAI,CAAC,MAAM,GAAC,CAAC,GAAC,CAAC,CAAC,GAAC,IAAI,CAAC,KAAK,GAAC,CAAC,CAAC,EAAE,CAAC,GAAG,KAAK,GAAG,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC;oBAC9E,IAAI,IAAI,KAAK,IAAI;wBAAE,IAAI,CAAC,GAAG,CAAC,CAAC,CAAC,QAAQ,CAAC,CAAC,EAAE,CAAC,GAAC,CAAC,CAAC,EAAE,MAAM,CAAC,CAAC;yBACnD,IAAI,IAAI,KAAK,QAAQ;wBAAE,IAAI,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC,CAAC,GAAC,CAAC,CAAC,EAAC,CAAC,CAAC,CAAC,GAAC,CAAC,CAAC,EAAC,CAAC,EAAC,CAAC,CAAC,CAAC,GAAC,CAAC,CAAC,CAAC,EAAE,MAAM,CAAC,CAAC;yBAClE,CAAC;wBACJ,MAAM,IAAI,GAAG,CAAC,CAAC,CAAC,GAAC,CAAC,CAAC,EAAE,QAAQ,GAAG,IAAI,GAAG,IAAI,CAAC,GAAG,CAAC,CAAC,EAAE,IAAI,CAAC,OAAO,CAAC,EAAE,GAAG,GAAG,IAAI,GAAG,CAAC,CAAC,CAAC,CAAC,CAAC,GAAG,IAAI,CAAC,CAAC,CAAC,CAAC,CAAC;wBAChG,IAAI,IAAI,KAAK,OAAO;4BAAE,IAAI,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,GAAC,GAAG,EAAE,CAAC,EAAE,CAAC,EAAE,QAAQ,CAAC,EAAE,MAAM,CAAC,CAAC;6BAC9D,IAAI,IAAI,KAAK,UAAU;4BAAE,IAAI,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC,CAAC,GAAC,CAAC,CAAC,GAAC,GAAG,EAAE,CAAC,CAAC,CAAC,GAAC,CAAC,CAAC,GAAC,GAAG,EAAE,CAAC,CAAC,CAAC,GAAC,CAAC,CAAC,GAAC,GAAG,EAAE,QAAQ,CAAC,EAAE,MAAM,CAAC,CAAC;6BAC1F,IAAI,IAAI,KAAK,WAAW;4BAAE,IAAI,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC,CAAC,GAAC,CAAC,CAAC,GAAC,GAAG,EAAE,CAAC,CAAC,CAAC,GAAC,CAAC,CAAC,GAAC,GAAG,EAAE,CAAC,CAAC,CAAC,GAAC,CAAC,CAAC,GAAC,GAAG,EAAE,QAAQ,CAAC,EAAE,MAAM,CAAC,CAAC;6BAC3F,CAAC;4BAAC,MAAM,MAAM,GAAG,IAAI,KAAK,eAAe,CAAC,CAAC,CAAC,CAAC,GAAC,CAAC,CAAC,CAAC,CAAC,CAAC,GAAC,EAAE,EAAE,KAAK,GAAG,CAAC,CAAC,MAAM,GAAC,CAAC,CAAC,CAAC;4BAC/E,MAAM,KAAK,GAAG,KAAK,GAAG,CAAC,CAAC,CAAC,CAAC,CAAC,GAAC,KAAK,CAAC,CAAC,CAAC,CAAC,CAAC;4BACtC,IAAI,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC,MAAM,CAAC,GAAC,KAAK,EAAE,CAAC,CAAC,MAAM,GAAC,CAAC,CAAC,GAAC,KAAK,EAAE,CAAC,CAAC,MAAM,GAAC,CAAC,CAAC,GAAC,KAAK,EAAE,KAAK,GAAG,IAAI,CAAC,GAAG,CAAC,CAAC,EAAC,IAAI,CAAC,OAAO,CAAC,CAAC,EAAE,MAAM,CAAC,CAAC;wBAAC,CAAC;oBAClH,CAAC;gBACH,CAAC;YACD,OAAO,EAAC,KAAK,EAAE,IAAI,CAAC,KAAK,EAAE,MAAM,EAAE,IAAI,CAAC,MAAM,EAAE,IAAI,EAAC,CAAC;QACxD,CAAC;gBAAS,CAAC;YAAC,IAAI,OAAO,CAAC,QAAQ,KAAK,QAAQ;gBAAE,OAAO,CAAC,KAAK,EAAE,CAAC;YAAC,OAAO,CAAC,OAAO,EAAE,CAAC;QAAC,CAAC;IACtF,CAAC;IACD,OAAO;QACL,MAAM,OAAO,GAAC,IAAI,CAAC,cAAc,CAAC;QAAA,IAAI,CAAC,cAAc,GAAC,SAAS,CAAC;QAChE,IAAI,CAAC,eAAe,GAAG,SAAS,CAAC;QAAC,IAAI,CAAC,oBAAoB,GAAG,CAAC,CAAC;QAChE,IAAI,CAAC,kBAAkB,EAAE,CAAC;QAC1B,MAAM,MAAM,GAAG,CAAC,GAAG,IAAI,CAAC,YAAY,CAAC,MAAM,EAAE,CAAC,CAAC;QAAC,IAAI,CAAC,YAAY,CAAC,KAAK,EAAE,CAAC;QAC1E,MAAM,CAAC,OAAO,CAAC,MAAM,CAAC,EAAE,CAAC,IAAI,CAAC,eAAe,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC,CAAC;QAC9D,MAAM,OAAO,GAAC,CAAC,IAAI,CAAC,YAAY,EAAC,IAAI,CAAC,MAAM,EAAC,IAAI,CAAC,KAAK,EAAC,IAAI,CAAC,cAAc,EAAC,GAAG,MAAM,CAAC,CAAC;QACvF,IAAI,CAAC,YAAY,GAAC,IAAI,CAAC,MAAM,GAAC,IAAI,CAAC,KAAK,GAAC,IAAI,CAAC,cAAc,GAAC,SAAS,CAAC;QACvE,KAAK,IAAI,CAAC,MAAM,CAAC,KAAK,CAAC,mBAAmB,EAAE,CAAC,KAAK,CAAC,GAAE,EAAE,CAAA,SAAS,CAAC,CAAC,IAAI,CAAC,GAAE,EAAE,GAAC,OAAO,EAAE,OAAO,EAAE,CAAC,CAAA,OAAO,CAAC,OAAO,CAAC,MAAM,CAAA,EAAE,CAAA,MAAM,EAAE,OAAO,EAAE,CAAC,CAAC,CAAA,CAAC,CAAC,CAAC;IAC/I,CAAC;CACF","sourcesContent":["import {aovDisplayShader} from './wgsl/aovDisplay.wgsl.js';\nimport {packAovDisplay, type AovDisplayOptions} from './aovDisplay.js';\nimport {aovSnapshotShader, aovSnapshotBinding, AOV_SNAPSHOT_BATCH_SIZE} from './wgsl/aovSnapshot.wgsl.js';\nimport {AovHistory, AOV_KINDS, AOV_LAYOUT, aovMask, type AovKind, type AovFrameSnapshot} from './aov.js';\n/** One target's optional AOV allocations and accepted frame history. */\nexport class WGPUAov {\n  readonly history = new AovHistory();\n  width = 0; height = 0; samples = 0; centerReady = false;\n  accumulation?: GPUBuffer; center?: GPUBuffer; previousCenter?: GPUBuffer; scene?: GPUBuffer;\n  frame?: AovFrameSnapshot;\n  private displayPipeline?: GPUComputePipeline;\n  private displayTexture?: GPUTexture;\n  private readonly resolvePipelines = new Map<string, GPUComputePipeline>();\n  private scenePrimitives?: Uint32Array;\n  private scenePrimitiveOffset = 0;\n  private snapshotGeneration = 0;\n  private readonly snapshotPool = new Map<AovKind, GPUBuffer>();\n  private readonly snapshotLeases = new WeakMap<GPUBuffer, {kind: AovKind; generation: number}>();\n  private readonly snapshotBuffers = new Set<GPUBuffer>();\n  constructor(readonly device: GPUDevice) {}\n  reset() {\n    if (this.centerReady && this.center && this.previousCenter) {\n      const encoder=this.device.createCommandEncoder();encoder.copyBufferToBuffer(this.center,0,this.previousCenter,0,this.center.size);this.device.queue.submit([encoder.finish()]);\n    }\n    this.samples = 0; this.centerReady = false; this.frame = undefined; }\n  resize(width: number, height: number) {\n    if (width === this.width && height === this.height) return;\n    if (width*height*AOV_LAYOUT.accumulatedBytes>this.device.limits.maxStorageBufferBindingSize || width*height*AOV_LAYOUT.accumulatedBytes>this.device.limits.maxBufferSize) throw new RangeError('AOV accumulation exceeds device buffer limits');\n    this.dispose(); this.history.clear(); this.reset(); this.width = width; this.height = height;\n    const usage = GPUBufferUsage.STORAGE | GPUBufferUsage.COPY_DST | GPUBufferUsage.COPY_SRC;\n    this.accumulation = this.device.createBuffer({label: 'AOV accumulation', size: width * height * AOV_LAYOUT.accumulatedBytes, usage});\n    this.previousCenter = this.device.createBuffer({label: 'Previous AOV center', size: width*height*AOV_LAYOUT.centerBytes, usage});\n    this.center = this.device.createBuffer({label: 'AOV center', size: width * height * AOV_LAYOUT.centerBytes, usage});\n  }\n  setScene(words: Uint32Array, primitiveIds?: Uint32Array) {\n    const bytes = words.byteLength + (primitiveIds?.byteLength ?? 0);\n    if (bytes > this.device.limits.maxStorageBufferBindingSize) throw new RangeError('AOV scene metadata exceeds device storage binding limit');\n    let replaced = false;\n    if (!this.scene || this.scene.size < bytes) {\n      this.scene?.destroy();\n      this.scene = this.device.createBuffer({label: 'AOV identities/history', size: Math.max(16, bytes), usage: GPUBufferUsage.STORAGE | GPUBufferUsage.COPY_DST});\n      replaced = true;\n    }\n    this.device.queue.writeBuffer(this.scene, 0, words);\n    // Primitive-cache arrays are immutable. Camera, material and rigid edits\n    // update only the prefix; topology/storage replacements change the tail's\n    // identity, and instance/material counts can move its byte offset.\n    if (primitiveIds && (replaced || this.scenePrimitives !== primitiveIds || this.scenePrimitiveOffset !== words.byteLength)) {\n      this.device.queue.writeBuffer(this.scene, words.byteLength, primitiveIds);\n    }\n    this.scenePrimitives = primitiveIds; this.scenePrimitiveOffset = words.byteLength;\n  }\n  get bytes() { return (this.accumulation?.size ?? 0) + (this.center?.size ?? 0) + (this.previousCenter?.size ?? 0) + (this.scene?.size ?? 0) + (this.displayTexture ? this.width*this.height*8 : 0) + [...this.snapshotBuffers].reduce((bytes, buffer) => bytes + buffer.size, 0); }\n  snapshot(kinds: readonly AovKind[]): Partial<Record<AovKind,GPUBuffer>> {\n    if(!this.accumulation || !this.center || !this.samples) throw new Error('AOV output is not ready');\n    const device = this.device;\n    aovMask(kinds);\n    const unique = AOV_KINDS.filter(kind => kinds.includes(kind));\n    const batches: {kinds: AovKind[]; pipeline: GPUComputePipeline}[] = [];\n    for (let offset = 0; offset < unique.length; offset += AOV_SNAPSHOT_BATCH_SIZE) {\n      const batch = unique.slice(offset, offset + AOV_SNAPSHOT_BATCH_SIZE), key = batch.join(',');\n      let pipeline = this.resolvePipelines.get(key);\n      if (!pipeline) {\n        pipeline = device.createComputePipeline({label: 'AOV snapshot', layout: 'auto',\n          compute: {module: device.createShaderModule({code: aovSnapshotShader(batch)}), entryPoint: 'main'}});\n        this.resolvePipelines.set(key, pipeline);\n      }\n      batches.push({kinds: batch, pipeline});\n    }\n    const buffers: Partial<Record<AovKind, GPUBuffer>> = {};\n    if (!batches.length) return buffers;\n    const parameters = device.createBuffer({size: 16, usage: GPUBufferUsage.UNIFORM, mappedAtCreation: true});\n    new Uint32Array(parameters.getMappedRange()).set([this.width, this.height, this.samples, 0]); parameters.unmap();\n    try {\n      const encoder = device.createCommandEncoder({label: 'AOV snapshot'}), pass = encoder.beginComputePass();\n      for (const {kinds: batch, pipeline} of batches) {\n        const entries: GPUBindGroupEntry[] = [\n          {binding: 0, resource: {buffer: this.accumulation}},\n          {binding: 1, resource: {buffer: this.center}},\n          {binding: 3, resource: {buffer: parameters}}\n        ];\n        batch.forEach((kind, index) => {\n          const buffer = this.snapshotPool.get(kind) ?? device.createBuffer({label: `Custom ${kind} input`, size: this.width * this.height * 16,\n            usage: GPUBufferUsage.STORAGE | GPUBufferUsage.COPY_SRC | GPUBufferUsage.COPY_DST});\n          this.snapshotPool.delete(kind);\n          this.snapshotLeases.set(buffer, {kind, generation: this.snapshotGeneration});\n          this.snapshotBuffers.add(buffer);\n          buffers[kind] = buffer;\n          entries.push({binding: aovSnapshotBinding(index), resource: {buffer}});\n        });\n        pass.setPipeline(pipeline);\n        pass.setBindGroup(0, device.createBindGroup({layout: pipeline.getBindGroupLayout(0), entries}));\n        pass.dispatchWorkgroups(Math.ceil(this.width / 8), Math.ceil(this.height / 8));\n      }\n      pass.end(); device.queue.submit([encoder.finish()]);\n      return buffers;\n    } catch (error) {\n      // Nothing was submitted, so these unfinished outputs can be retired now.\n      Object.values(buffers).forEach(buffer => {\n        if (buffer) {this.snapshotLeases.delete(buffer); this.snapshotBuffers.delete(buffer); buffer.destroy();}\n      });\n      throw error;\n    } finally { parameters.destroy(); }\n  }\n  // Call only after the consumer and its queued GPU work have finished. Each\n  // in-flight denoiser owns separate outputs, including cancelled jobs whose\n  // callbacks have not settled. Keep at most one reusable buffer per kind.\n  releaseSnapshot(buffers: Partial<Record<AovKind, GPUBuffer>>) {\n    for (const buffer of Object.values(buffers)) {\n      if (!buffer) continue;\n      const lease = this.snapshotLeases.get(buffer);\n      if (!lease) continue;\n      this.snapshotLeases.delete(buffer);\n      if (lease.generation === this.snapshotGeneration && !this.snapshotPool.has(lease.kind)) {\n        this.snapshotPool.set(lease.kind, buffer);\n      } else { this.snapshotBuffers.delete(buffer); buffer.destroy(); }\n    }\n  }\n  visualize(kind: AovKind, options: AovDisplayOptions, linear: boolean): GPUTexture {\n    if(!this.accumulation || !this.center || !this.samples) throw new Error('AOV output is not ready');\n    const device=this.device;\n    this.displayPipeline ??= device.createComputePipeline({label:'AOV visualization',layout:'auto',compute:{module:device.createShaderModule({code:aovDisplayShader('rgba16float')}),entryPoint:'main'}});\n    if(!this.displayTexture || this.displayTexture.width!==this.width || this.displayTexture.height!==this.height){\n      this.displayTexture?.destroy();\n      this.displayTexture=device.createTexture({label:'AOV display image',size:[this.width,this.height],format:'rgba16float',usage:GPUTextureUsage.STORAGE_BINDING|GPUTextureUsage.TEXTURE_BINDING|GPUTextureUsage.COPY_SRC});\n    }\n    const words=packAovDisplay(this.width,this.height,this.samples,kind,options,linear);\n    const parameters=device.createBuffer({size:48,usage:GPUBufferUsage.UNIFORM,mappedAtCreation:true});new Uint32Array(parameters.getMappedRange()).set(words);parameters.unmap();\n    const group=device.createBindGroup({layout:this.displayPipeline.getBindGroupLayout(0),entries:[{binding:0,resource:{buffer:this.accumulation}},{binding:1,resource:{buffer:this.center}},{binding:2,resource:this.displayTexture.createView()},{binding:3,resource:{buffer:parameters}}]});\n    const encoder=device.createCommandEncoder(),pass=encoder.beginComputePass();pass.setPipeline(this.displayPipeline);pass.setBindGroup(0,group);pass.dispatchWorkgroups(Math.ceil(this.width/8),Math.ceil(this.height/8));pass.end();device.queue.submit([encoder.finish()]);parameters.destroy();\n    return this.displayTexture;\n  }\n  async read(kind: AovKind) {\n    const center = kind === 'id' || kind === 'motion';\n    const source = center ? this.center! : this.accumulation!;\n    const staging = this.device.createBuffer({size: source.size, usage: GPUBufferUsage.MAP_READ | GPUBufferUsage.COPY_DST});\n    try {\n      const encoder = this.device.createCommandEncoder(); encoder.copyBufferToBuffer(source, 0, staging, 0, source.size);\n      this.device.queue.submit([encoder.finish()]); await staging.mapAsync(GPUMapMode.READ);\n      const f = new Float32Array(staging.getMappedRange()), u = new Uint32Array(f.buffer);\n      const data = kind === 'id' ? new Uint32Array(this.width*this.height*4) : new Float32Array(this.width*this.height*4);\n      for (let y = 0; y < this.height; y++) for (let x = 0; x < this.width; x++) {\n        const target = (y * this.width + x) * 4;\n        // Tracing uses OpenGL camera coordinates; API images are top row first.\n        const pixel = ((this.height-1-y)*this.width+x), o = pixel * (center ? 8 : 16);\n        if (kind === 'id') data.set(u.subarray(o, o+4), target);\n        else if (kind === 'motion') data.set([f[o+4],f[o+5],0,f[o+7]], target);\n        else {\n          const hits = f[o+7], coverage = hits / Math.max(1, this.samples), inv = hits > 0 ? 1 / hits : 0;\n          if (kind === 'depth') data.set([f[o]*inv, 0, 0, coverage], target);\n          else if (kind === 'position') data.set([f[o+4]*inv, f[o+5]*inv, f[o+6]*inv, coverage], target);\n          else if (kind === 'roughness') data.set([f[o+1]*inv, f[o+2]*inv, f[o+3]*inv, coverage], target);\n          else { const offset = kind === 'diffuseAlbedo' ? o+8 : o+12, valid = f[offset+3];\n            const scale = valid > 0 ? 1/valid : 0;\n            data.set([f[offset]*scale, f[offset+1]*scale, f[offset+2]*scale, valid / Math.max(1,this.samples)], target); }\n        }\n      }\n      return {width: this.width, height: this.height, data};\n    } finally { if (staging.mapState === 'mapped') staging.unmap(); staging.destroy(); }\n  }\n  dispose() {\n    const texture=this.displayTexture;this.displayTexture=undefined;\n    this.scenePrimitives = undefined; this.scenePrimitiveOffset = 0;\n    this.snapshotGeneration++;\n    const pooled = [...this.snapshotPool.values()]; this.snapshotPool.clear();\n    pooled.forEach(buffer => this.snapshotBuffers.delete(buffer));\n    const buffers=[this.accumulation,this.center,this.scene,this.previousCenter,...pooled];\n    this.accumulation=this.center=this.scene=this.previousCenter=undefined;\n    void this.device.queue.onSubmittedWorkDone().catch(()=>undefined).then(()=>{texture?.destroy();buffers.forEach(buffer=>buffer?.destroy());});\n  }\n}\n"]}

package/lib/renderer/WGPUBindlessMaterialData.d.ts

@@ -0,0 +1,17 @@
+import type OpenPBRMaterial from '../scene/OpenPBRMaterial.js';
+import type { SceneLight } from '../scene/Scene.js';
+import type Texture from '../scene/Texture.js';
+export declare class WGPUBindlessMaterialData {
+    readonly bufferData: import("./materialBufferData.js").BufferData;
+    readonly textures: Texture[];
+    readonly defines: Record<string, number>;
+    private readonly lightIndices;
+    private readonly displacementMaps;
+    constructor(materials: OpenPBRMaterial[], lights: SceneLight[]);
+    repackParams(materials: OpenPBRMaterial[]): void;
+    getBufferArray(): Float32Array<ArrayBufferLike>;
+    getLightTexturesIndices(): number[];
+    getDisplacementMaps(): Texture[];
+}
+export declare function textureFloatStorage(texture: Texture): boolean;
+export declare function textureColorLinear(texture: Texture): number;

package/lib/renderer/WGPUBindlessMaterialData.js

@@ -0,0 +1,48 @@
+import { isTextureDataSource } from '../scene/Texture.js';
+import { textureWrapIndex } from '../scene/TextureState.js';
+import { MATERIAL_TEXTURE_NAMES, MATERIAL_TEXTURE_SLOTS } from './materialTextureSlots.js';
+import { mergeTexturesFromMaterials } from './materialTextures.js';
+import { createMaterialBufferData, repackMaterialParams } from './materialParams.js';
+import { packMaterialBuffer } from './materialLayout.js';
+// Browser adapter intentionally keeps the shared/native MaterialData entry intact.
+export class WGPUBindlessMaterialData {
+    bufferData;
+    textures;
+    defines;
+    lightIndices = [];
+    displacementMaps;
+    constructor(materials, lights) {
+        const maps = mergeTexturesFromMaterials(materials, MATERIAL_TEXTURE_NAMES);
+        this.textures = maps.textures.slice();
+        this.displacementMaps = maps.indices.displacementMap.map(index => maps.textures[index]).filter(Boolean);
+        for (const light of lights) {
+            const texture = light.type === 'RectLight' ? light.texture : undefined;
+            this.lightIndices.push(texture ? this.textures.length : -1);
+            if (texture)
+                this.textures.push(texture);
+        }
+        this.bufferData = createMaterialBufferData(materials);
+        this.defines = { NUM_MAPS: this.textures.length };
+        for (const slot of MATERIAL_TEXTURE_SLOTS) {
+            this.bufferData[slot.index] = maps.indices[slot.property];
+            if (slot.define)
+                this.defines[slot.define] = maps.indices[slot.property].filter(index => index >= 0).length;
+        }
+        this.bufferData.mapUvChannel = this.textures.map(t => t.uvChannel);
+        this.bufferData.mapColorLinear = this.textures.map(textureColorLinear);
+        this.bufferData.mapTransform = this.textures.flatMap(t => [...t.uvScale, ...t.uvOffset]);
+        this.bufferData.mapWrap = this.textures.flatMap(t => [textureWrapIndex(t.wrapS), textureWrapIndex(t.wrapT)]);
+    }
+    repackParams(materials) { repackMaterialParams(this.bufferData, materials); }
+    getBufferArray() { return packMaterialBuffer(this.bufferData); }
+    getLightTexturesIndices() { return this.lightIndices; }
+    getDisplacementMaps() { return this.displacementMaps; }
+}
+export function textureFloatStorage(texture) {
+    const source = texture.image;
+    return isTextureDataSource(source) && !(source.data instanceof Uint8Array) && !(source.data instanceof Uint8ClampedArray);
+}
+export function textureColorLinear(texture) {
+    return Number(texture.colorSpace ? texture.colorSpace === 'linear' : textureFloatStorage(texture));
+}
+//# sourceMappingURL=WGPUBindlessMaterialData.js.map

package/lib/renderer/WGPUBindlessMaterialData.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"WGPUBindlessMaterialData.js","sourceRoot":"","sources":["../../src/renderer/WGPUBindlessMaterialData.ts"],"names":[],"mappings":"AAGA,OAAO,EAAC,mBAAmB,EAAC,MAAM,qBAAqB,CAAC;AACxD,OAAO,EAAC,gBAAgB,EAAC,MAAM,0BAA0B,CAAC;AAC1D,OAAO,EAAC,sBAAsB,EAAE,sBAAsB,EAAC,MAAM,2BAA2B,CAAC;AACzF,OAAO,EAAC,0BAA0B,EAAC,MAAM,uBAAuB,CAAC;AACjE,OAAO,EAAC,wBAAwB,EAAE,oBAAoB,EAAC,MAAM,qBAAqB,CAAC;AACnF,OAAO,EAAC,kBAAkB,EAAC,MAAM,qBAAqB,CAAC;AAEvD,mFAAmF;AACnF,MAAM,OAAO,wBAAwB;IAC1B,UAAU,CAAC;IACX,QAAQ,CAAY;IACpB,OAAO,CAAyB;IACxB,YAAY,GAAa,EAAE,CAAC;IAC5B,gBAAgB,CAAY;IAC7C,YAAY,SAA4B,EAAE,MAAoB;QAC5D,MAAM,IAAI,GAAG,0BAA0B,CAAC,SAAS,EAAE,sBAAsB,CAAC,CAAC;QAC3E,IAAI,CAAC,QAAQ,GAAG,IAAI,CAAC,QAAQ,CAAC,KAAK,EAAE,CAAC;QACtC,IAAI,CAAC,gBAAgB,GAAG,IAAI,CAAC,OAAO,CAAC,eAAe,CAAC,GAAG,CAAC,KAAK,CAAC,EAAE,CAAC,IAAI,CAAC,QAAQ,CAAC,KAAK,CAAC,CAAC,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC;QACxG,KAAK,MAAM,KAAK,IAAI,MAAM,EAAE,CAAC;YAC3B,MAAM,OAAO,GAAG,KAAK,CAAC,IAAI,KAAK,WAAW,CAAC,CAAC,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,SAAS,CAAC;YACvE,IAAI,CAAC,YAAY,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC,CAAC,IAAI,CAAC,QAAQ,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC;YAC5D,IAAI,OAAO;gBAAE,IAAI,CAAC,QAAQ,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC;QAC3C,CAAC;QACD,IAAI,CAAC,UAAU,GAAG,wBAAwB,CAAC,SAAS,CAAC,CAAC;QACtD,IAAI,CAAC,OAAO,GAAG,EAAC,QAAQ,EAAE,IAAI,CAAC,QAAQ,CAAC,MAAM,EAAC,CAAC;QAChD,KAAK,MAAM,IAAI,IAAI,sBAAsB,EAAE,CAAC;YAC1C,IAAI,CAAC,UAAU,CAAC,IAAI,CAAC,KAAK,CAAC,GAAG,IAAI,CAAC,OAAO,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC;YAC1D,IAAI,IAAI,CAAC,MAAM;gBAAE,IAAI,CAAC,OAAO,CAAC,IAAI,CAAC,MAAM,CAAC,GAAG,IAAI,CAAC,OAAO,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC,MAAM,CAAC,KAAK,CAAC,EAAE,CAAC,KAAK,IAAI,CAAC,CAAC,CAAC,MAAM,CAAC;QAC9G,CAAC;QACD,IAAI,CAAC,UAAU,CAAC,YAAY,GAAG,IAAI,CAAC,QAAQ,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,SAAS,CAAC,CAAC;QACnE,IAAI,CAAC,UAAU,CAAC,cAAc,GAAG,IAAI,CAAC,QAAQ,CAAC,GAAG,CAAC,kBAAkB,CAAC,CAAC;QACvE,IAAI,CAAC,UAAU,CAAC,YAAY,GAAG,IAAI,CAAC,QAAQ,CAAC,OAAO,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,GAAG,CAAC,CAAC,OAAO,EAAE,GAAG,CAAC,CAAC,QAAQ,CAAC,CAAC,CAAC;QACzF,IAAI,CAAC,UAAU,CAAC,OAAO,GAAG,IAAI,CAAC,QAAQ,CAAC,OAAO,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,gBAAgB,CAAC,CAAC,CAAC,KAAK,CAAC,EAAE,gBAAgB,CAAC,CAAC,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC;IAC/G,CAAC;IACD,YAAY,CAAC,SAA4B,IAAI,oBAAoB,CAAC,IAAI,CAAC,UAAU,EAAE,SAAS,CAAC,CAAC,CAAC,CAAC;IAChG,cAAc,KAAK,OAAO,kBAAkB,CAAC,IAAI,CAAC,UAAU,CAAC,CAAC,CAAC,CAAC;IAChE,uBAAuB,KAAK,OAAO,IAAI,CAAC,YAAY,CAAC,CAAC,CAAC;IACvD,mBAAmB,KAAK,OAAO,IAAI,CAAC,gBAAgB,CAAC,CAAC,CAAC;CACxD;AACD,MAAM,UAAU,mBAAmB,CAAC,OAAgB;IAClD,MAAM,MAAM,GAAG,OAAO,CAAC,KAAK,CAAC;IAC7B,OAAO,mBAAmB,CAAC,MAAM,CAAC,IAAI,CAAC,CAAC,MAAM,CAAC,IAAI,YAAY,UAAU,CAAC,IAAI,CAAC,CAAC,MAAM,CAAC,IAAI,YAAY,iBAAiB,CAAC,CAAC;AAC5H,CAAC;AACD,MAAM,UAAU,kBAAkB,CAAC,OAAgB;IACjD,OAAO,MAAM,CAAC,OAAO,CAAC,UAAU,CAAC,CAAC,CAAC,OAAO,CAAC,UAAU,KAAK,QAAQ,CAAC,CAAC,CAAC,mBAAmB,CAAC,OAAO,CAAC,CAAC,CAAC;AACrG,CAAC","sourcesContent":["import type OpenPBRMaterial from '../scene/OpenPBRMaterial.js';\nimport type {SceneLight} from '../scene/Scene.js';\nimport type Texture from '../scene/Texture.js';\nimport {isTextureDataSource} from '../scene/Texture.js';\nimport {textureWrapIndex} from '../scene/TextureState.js';\nimport {MATERIAL_TEXTURE_NAMES, MATERIAL_TEXTURE_SLOTS} from './materialTextureSlots.js';\nimport {mergeTexturesFromMaterials} from './materialTextures.js';\nimport {createMaterialBufferData, repackMaterialParams} from './materialParams.js';\nimport {packMaterialBuffer} from './materialLayout.js';\n\n// Browser adapter intentionally keeps the shared/native MaterialData entry intact.\nexport class WGPUBindlessMaterialData {\n  readonly bufferData;\n  readonly textures: Texture[];\n  readonly defines: Record<string, number>;\n  private readonly lightIndices: number[] = [];\n  private readonly displacementMaps: Texture[];\n  constructor(materials: OpenPBRMaterial[], lights: SceneLight[]) {\n    const maps = mergeTexturesFromMaterials(materials, MATERIAL_TEXTURE_NAMES);\n    this.textures = maps.textures.slice();\n    this.displacementMaps = maps.indices.displacementMap.map(index => maps.textures[index]).filter(Boolean);\n    for (const light of lights) {\n      const texture = light.type === 'RectLight' ? light.texture : undefined;\n      this.lightIndices.push(texture ? this.textures.length : -1);\n      if (texture) this.textures.push(texture);\n    }\n    this.bufferData = createMaterialBufferData(materials);\n    this.defines = {NUM_MAPS: this.textures.length};\n    for (const slot of MATERIAL_TEXTURE_SLOTS) {\n      this.bufferData[slot.index] = maps.indices[slot.property];\n      if (slot.define) this.defines[slot.define] = maps.indices[slot.property].filter(index => index >= 0).length;\n    }\n    this.bufferData.mapUvChannel = this.textures.map(t => t.uvChannel);\n    this.bufferData.mapColorLinear = this.textures.map(textureColorLinear);\n    this.bufferData.mapTransform = this.textures.flatMap(t => [...t.uvScale, ...t.uvOffset]);\n    this.bufferData.mapWrap = this.textures.flatMap(t => [textureWrapIndex(t.wrapS), textureWrapIndex(t.wrapT)]);\n  }\n  repackParams(materials: OpenPBRMaterial[]) { repackMaterialParams(this.bufferData, materials); }\n  getBufferArray() { return packMaterialBuffer(this.bufferData); }\n  getLightTexturesIndices() { return this.lightIndices; }\n  getDisplacementMaps() { return this.displacementMaps; }\n}\nexport function textureFloatStorage(texture: Texture) {\n  const source = texture.image;\n  return isTextureDataSource(source) && !(source.data instanceof Uint8Array) && !(source.data instanceof Uint8ClampedArray);\n}\nexport function textureColorLinear(texture: Texture) {\n  return Number(texture.colorSpace ? texture.colorSpace === 'linear' : textureFloatStorage(texture));\n}\n"]}

package/lib/renderer/WGPUBindlessTextures.d.ts

@@ -0,0 +1,47 @@
+import type Texture from '../scene/Texture.js';
+import { type ResourceTable } from './wgpuBindless.js';
+export interface TextureExtent {
+    width: number;
+    height: number;
+}
+export declare function planIndependentTextures(textures: readonly Texture[], budget: number, dimension: number): {
+    sizes: {
+        width: number;
+        height: number;
+    }[];
+    scale: number;
+    floatStorage: boolean;
+    bytesPerPixel: number;
+    estimatedBytes: number;
+};
+export declare function independentTexturePixels(texture: Texture, size: TextureExtent, floatStorage: boolean): Uint8Array | Uint16Array;
+export declare function packIndependentTextureInfo(textures: readonly Texture[], sizes: TextureExtent[], capacity: number): ArrayBuffer;
+/** Browser scene resources; callers wait for retirement before preparing another generation. */
+export declare class WGPUBindlessTextures {
+    private device;
+    table?: ResourceTable;
+    metadata?: GPUBuffer;
+    retirement?: Promise<void>;
+    private entries;
+    private logical;
+    private samplers;
+    private fallback;
+    private disposed;
+    private generation;
+    private metadataCapacity;
+    private retiredBytes;
+    private destroyRetired?;
+    private stats;
+    constructor(device: GPUDevice);
+    get revision(): number;
+    get diagnostics(): {
+        metadataBytes: number;
+        resourceTableCapacity: number;
+        retiredBytes: number;
+        textureCount: number;
+        estimatedBytes: number;
+        scale: number;
+    };
+    update(textures: Texture[], budget: number): void;
+    dispose(): void;
+}

package/lib/renderer/WGPUBindlessTextures.js

@@ -0,0 +1,180 @@
+import { isProceduralSkyImage } from '../scene/skyParameters.js';
+import { isTextureDataSource } from '../scene/Texture.js';
+import { textureWrapIndex } from '../scene/TextureState.js';
+import { createCanvas } from '../util/canvas.js';
+import { writeRawAtlasPlacement } from './textureAtlasPixels.js';
+import { float16Bits } from './float16.js';
+import { textureColorLinear, textureFloatStorage } from './WGPUBindlessMaterialData.js';
+import { bindlessLimitReason, createResourceTable, FIRST_TEXTURE_SLOT, RESOURCE_TABLE_LIMIT } from './wgpuBindless.js';
+export function planIndependentTextures(textures, budget, dimension) {
+    const floatStorage = textures.some(textureFloatStorage), bytesPerPixel = floatStorage ? 8 : 4;
+    const indices = [...new Map(textures.map((t, i) => [t, i])).values()];
+    for (const { image } of textures)
+        if (!Number.isSafeInteger(image.width) || !Number.isSafeInteger(image.height) || image.width < 1 || image.height < 1) {
+            throw new Error('Material textures require positive integer dimensions.');
+        }
+    const extents = (scale) => textures.map(t => ({ width: Math.max(1, Math.floor(t.image.width * scale)), height: Math.max(1, Math.floor(t.image.height * scale)) }));
+    const bytes = (sizes) => indices.reduce((sum, i) => sum + sizes[i].width * sizes[i].height * bytesPerPixel, 0);
+    if (indices.length * bytesPerPixel > budget)
+        throw new Error('Material textures cannot fit the budget even at 1×1.');
+    let scale = textures.reduce((s, t) => Math.min(s, dimension / Math.max(t.image.width, t.image.height)), 1);
+    if (bytes(extents(scale)) > budget) {
+        let low = 0, high = scale;
+        for (let i = 0; i < 40; i++) {
+            const mid = (low + high) / 2;
+            if (bytes(extents(mid)) <= budget)
+                low = mid;
+            else
+                high = mid;
+        }
+        scale = low;
+    }
+    const sizes = extents(scale);
+    return { sizes, scale, floatStorage, bytesPerPixel, estimatedBytes: bytes(sizes) };
+}
+export function independentTexturePixels(texture, size, floatStorage) {
+    const data = floatStorage ? new Uint16Array(size.width * size.height * 4) : new Uint8Array(size.width * size.height * 4);
+    if (isTextureDataSource(texture.image)) {
+        writeRawAtlasPlacement(data, size.width, { index: 0, layer: 0, x: 0, y: 0, ...size }, { image: texture.image, flipY: texture.flipY }, floatStorage);
+    }
+    else {
+        const canvas = createCanvas(size.width, size.height), context = canvas.getContext('2d');
+        if (!context)
+            throw new Error('Unable to prepare material texture pixels.');
+        if (texture.flipY) {
+            context.translate(0, size.height);
+            context.scale(1, -1);
+        }
+        if (isProceduralSkyImage(texture.image))
+            throw new Error('ProceduralSky is only supported as an environment or background.');
+        context.drawImage(texture.image, 0, 0, size.width, size.height);
+        const pixels = context.getImageData(0, 0, size.width, size.height).data;
+        for (let i = 0; i < pixels.length; i++)
+            data[i] = floatStorage ? float16Bits(pixels[i] / 255) : pixels[i];
+    }
+    return data;
+}
+export function packIndependentTextureInfo(textures, sizes, capacity) {
+    const data = new ArrayBuffer(capacity * 48), f32 = new Float32Array(data), u32 = new Uint32Array(data);
+    textures.forEach((t, i) => {
+        const base = i * 12;
+        f32.set([...t.uvScale, ...t.uvOffset], base);
+        u32.set([FIRST_TEXTURE_SLOT + i, 1 + textureWrapIndex(t.wrapS) * 3 + textureWrapIndex(t.wrapT), t.uvChannel, textureColorLinear(t), sizes[i].width, sizes[i].height], base + 4);
+    });
+    return data;
+}
+/** Browser scene resources; callers wait for retirement before preparing another generation. */
+export class WGPUBindlessTextures {
+    device;
+    table;
+    metadata;
+    retirement;
+    entries = new Map();
+    logical = [];
+    samplers;
+    fallback;
+    disposed = false;
+    generation = 0;
+    metadataCapacity = 0;
+    retiredBytes = 0;
+    destroyRetired;
+    stats = { textureCount: 0, estimatedBytes: 0, scale: 1 };
+    constructor(device) {
+        this.device = device;
+        const modes = ['repeat', 'clamp-to-edge', 'mirror-repeat'];
+        this.samplers = modes.flatMap(addressModeU => modes.map(addressModeV => device.createSampler({ minFilter: 'linear', magFilter: 'linear', addressModeU, addressModeV })));
+        this.fallback = device.createTexture({ size: [1, 1], format: 'rgba8unorm', usage: GPUTextureUsage.TEXTURE_BINDING | GPUTextureUsage.COPY_DST });
+        device.queue.writeTexture({ texture: this.fallback }, new Uint8Array([255, 255, 255, 255]), { bytesPerRow: 4 }, [1, 1]);
+    }
+    get revision() { return this.generation; }
+    get diagnostics() { return { ...this.stats, metadataBytes: this.metadata?.size ?? 0, resourceTableCapacity: this.table?.size ?? 0, retiredBytes: this.retiredBytes }; }
+    update(textures, budget) {
+        if (this.disposed)
+            throw new Error('Bindless textures are disposed.');
+        if (this.retirement)
+            throw new Error('Previous resource-table generation has not retired.');
+        const reason = bindlessLimitReason(this.device, textures.length);
+        if (reason)
+            throw new Error(reason);
+        const plan = planIndependentTextures(textures, budget, this.device.limits.maxTextureDimension2D);
+        const format = plan.floatStorage ? 'rgba16float' : 'rgba8unorm';
+        const next = new Map(), created = [];
+        let newTable, newBuffer;
+        try {
+            textures.forEach((texture, i) => {
+                if (next.has(texture))
+                    return;
+                const size = plan.sizes[i], old = this.entries.get(texture);
+                const reusable = old && old.width === size.width && old.height === size.height && old.format === format;
+                const gpu = reusable ? old.gpu : this.device.createTexture({ label: 'Material texture', size: [size.width, size.height], format, usage: GPUTextureUsage.TEXTURE_BINDING | GPUTextureUsage.COPY_DST });
+                if (!reusable)
+                    created.push(gpu);
+                if (!reusable || old.image !== texture.image || old.version !== texture.version || old.flipY !== texture.flipY || old.sourceWidth !== texture.image.width || old.sourceHeight !== texture.image.height) {
+                    const pixels = independentTexturePixels(texture, size, plan.floatStorage);
+                    this.device.queue.writeTexture({ texture: gpu }, pixels, { bytesPerRow: size.width * plan.bytesPerPixel }, [size.width, size.height]);
+                }
+                next.set(texture, { gpu, image: texture.image, version: texture.version, sourceWidth: texture.image.width, sourceHeight: texture.image.height, flipY: texture.flipY, format, ...size });
+            });
+            const logical = textures.map(t => next.get(t).gpu);
+            const changed = !this.table || logical.length !== this.logical.length || logical.some((t, i) => t !== this.logical[i]);
+            if (changed) {
+                const size = Math.min(RESOURCE_TABLE_LIMIT, 2 ** Math.ceil(Math.log2(Math.max(16, textures.length + FIRST_TEXTURE_SLOT))));
+                newTable = createResourceTable(this.device, size);
+                newTable.update(0, this.fallback.createView());
+                this.samplers.forEach((sampler, i) => newTable.update(i + 1, sampler));
+                logical.forEach((gpu, i) => newTable.update(FIRST_TEXTURE_SLOT + i, gpu.createView()));
+            }
+            const maxCapacity = Math.floor(Math.min(this.device.limits.maxStorageBufferBindingSize, this.device.limits.maxBufferSize) / 48);
+            const capacity = Math.min(maxCapacity, Math.max(this.metadataCapacity, 2 ** Math.ceil(Math.log2(Math.max(1, textures.length)))));
+            if (!this.metadata || capacity !== this.metadataCapacity)
+                newBuffer = this.device.createBuffer({ label: 'Material texture metadata', size: capacity * 48, usage: GPUBufferUsage.STORAGE | GPUBufferUsage.COPY_DST });
+            this.device.queue.writeBuffer(newBuffer ?? this.metadata, 0, packIndependentTextureInfo(textures, plan.sizes, capacity));
+            const oldTable = newTable ? this.table : undefined, oldBuffer = newBuffer ? this.metadata : undefined;
+            const retained = new Set(logical);
+            const removed = [...new Set([...this.entries.values()].map(e => e.gpu))].filter(t => !retained.has(t));
+            this.entries = next;
+            this.logical = logical;
+            if (newTable)
+                this.table = newTable;
+            if (newBuffer)
+                this.metadata = newBuffer;
+            this.metadataCapacity = capacity;
+            this.generation++;
+            this.stats = { textureCount: textures.length, estimatedBytes: [...new Set(logical)].reduce((n, t) => n + t.width * t.height * plan.bytesPerPixel, 0), scale: plan.scale };
+            if (oldTable || oldBuffer || removed.length) {
+                this.retiredBytes = (oldBuffer?.size ?? 0) + removed.reduce((n, t) => n + t.width * t.height * (t.format === 'rgba16float' ? 8 : 4), 0);
+                const destroy = () => { oldTable?.destroy(); oldBuffer?.destroy(); removed.forEach(t => t.destroy()); this.retiredBytes = 0; };
+                this.destroyRetired = destroy;
+                this.retirement = this.device.queue.onSubmittedWorkDone().catch(() => { }).then(() => {
+                    if (this.destroyRetired === destroy) {
+                        destroy();
+                        this.destroyRetired = undefined;
+                        this.retirement = undefined;
+                    }
+                });
+            }
+        }
+        catch (error) {
+            newTable?.destroy();
+            newBuffer?.destroy();
+            created.forEach(t => t.destroy());
+            throw error;
+        }
+    }
+    dispose() {
+        if (this.disposed)
+            return;
+        this.disposed = true;
+        this.generation++;
+        this.destroyRetired?.();
+        this.destroyRetired = undefined;
+        this.retirement = undefined;
+        this.table?.destroy();
+        this.metadata?.destroy();
+        this.fallback.destroy();
+        new Set([...this.entries.values()].map(e => e.gpu)).forEach(t => t.destroy());
+        this.entries.clear();
+        this.logical = [];
+    }
+}
+//# sourceMappingURL=WGPUBindlessTextures.js.map