@meimu/mutracer 0.0.0-stage → 0.1.0

package/lib/renderer/WGPUCanvasSurface.js

@@ -0,0 +1,48 @@
+import { createCanvas } from '../util/canvas.js';
+/** One canvas surface; composing or exporting an image never reconfigures it. */
+export class WGPUCanvasSurface {
+    device;
+    standardFormat;
+    context;
+    hdr = false;
+    compatibleHDR;
+    constructor(canvas, device, standardFormat) {
+        this.device = device;
+        this.standardFormat = standardFormat;
+        this.context = canvas.getContext('webgpu');
+        this.context.configure({ device, format: standardFormat, alphaMode: 'premultiplied' });
+    }
+    get format() { return this.hdr ? 'rgba16float' : this.standardFormat; }
+    get supportsHDR() {
+        if (this.compatibleHDR !== undefined)
+            return this.compatibleHDR;
+        this.compatibleHDR = false;
+        let probe = null;
+        try {
+            probe = createCanvas().getContext('webgpu');
+            if (!probe)
+                return false;
+            probe.configure({ device: this.device, format: 'rgba16float', alphaMode: 'premultiplied', colorSpace: 'srgb', toneMapping: { mode: 'extended' } });
+            const applied = probe.getConfiguration?.();
+            this.compatibleHDR = applied?.format === 'rgba16float' && applied.toneMapping?.mode === 'extended';
+        }
+        catch {
+            this.compatibleHDR = false;
+        }
+        finally {
+            probe?.unconfigure?.();
+        }
+        return this.compatibleHDR;
+    }
+    configure(enabled) {
+        const active = enabled && this.supportsHDR;
+        if (active === this.hdr)
+            return;
+        this.hdr = active;
+        this.context.configure(active
+            ? { device: this.device, format: 'rgba16float', alphaMode: 'premultiplied', colorSpace: 'srgb', toneMapping: { mode: 'extended' } }
+            : { device: this.device, format: this.standardFormat, alphaMode: 'premultiplied' });
+    }
+    dispose() { this.context.unconfigure?.(); }
+}
+//# sourceMappingURL=WGPUCanvasSurface.js.map

package/lib/renderer/WGPUCanvasSurface.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"WGPUCanvasSurface.js","sourceRoot":"","sources":["../../src/renderer/WGPUCanvasSurface.ts"],"names":[],"mappings":"AAAA,OAAO,EAAC,YAAY,EAAC,MAAM,mBAAmB,CAAC;AAG/C,iFAAiF;AACjF,MAAM,OAAO,iBAAiB;IAI8C;IAAoC;IAHrG,OAAO,CAAqB;IACrC,GAAG,GAAG,KAAK,CAAC;IACJ,aAAa,CAAW;IAChC,YAAY,MAA2C,EAAmB,MAAiB,EAAmB,cAAgC;QAApE,WAAM,GAAN,MAAM,CAAW;QAAmB,mBAAc,GAAd,cAAc,CAAkB;QAC5I,IAAI,CAAC,OAAO,GAAG,MAAM,CAAC,UAAU,CAAC,QAAQ,CAAwB,CAAC;QAClE,IAAI,CAAC,OAAO,CAAC,SAAS,CAAC,EAAC,MAAM,EAAE,MAAM,EAAE,cAAc,EAAE,SAAS,EAAE,eAAe,EAAC,CAAC,CAAC;IACvF,CAAC;IACD,IAAI,MAAM,KAAI,OAAO,IAAI,CAAC,GAAG,CAAC,CAAC,CAAC,aAAsB,CAAC,CAAC,CAAC,IAAI,CAAC,cAAc,CAAC,CAAA,CAAC;IAC9E,IAAI,WAAW;QACb,IAAI,IAAI,CAAC,aAAa,KAAK,SAAS;YAAE,OAAO,IAAI,CAAC,aAAa,CAAC;QAChE,IAAI,CAAC,aAAa,GAAG,KAAK,CAAC;QAC3B,IAAI,KAAK,GAA8B,IAAI,CAAC;QAC5C,IAAI,CAAC;YACH,KAAK,GAAG,YAAY,EAAE,CAAC,UAAU,CAAC,QAAQ,CAA8B,CAAC;YACzE,IAAI,CAAC,KAAK;gBAAE,OAAO,KAAK,CAAC;YACzB,KAAK,CAAC,SAAS,CAAC,EAAC,MAAM,EAAE,IAAI,CAAC,MAAM,EAAE,MAAM,EAAE,aAAa,EAAE,SAAS,EAAE,eAAe,EAAE,UAAU,EAAE,MAAM,EAAE,WAAW,EAAE,EAAC,IAAI,EAAE,UAAU,EAAC,EAAqB,CAAC,CAAC;YACnK,MAAM,OAAO,GAAG,KAAK,CAAC,gBAAgB,EAAE,EAAE,CAAC;YAC3C,IAAI,CAAC,aAAa,GAAG,OAAO,EAAE,MAAM,KAAK,aAAa,IAAI,OAAO,CAAC,WAAW,EAAE,IAAI,KAAK,UAAU,CAAC;QACrG,CAAC;QAAC,MAAM,CAAC;YAAA,IAAI,CAAC,aAAa,GAAG,KAAK,CAAC;QAAA,CAAC;gBAC7B,CAAC;YAAA,KAAK,EAAE,WAAW,EAAE,EAAE,CAAC;QAAA,CAAC;QACjC,OAAO,IAAI,CAAC,aAAa,CAAC;IAC5B,CAAC;IACD,SAAS,CAAC,OAAgB;QACxB,MAAM,MAAM,GAAG,OAAO,IAAI,IAAI,CAAC,WAAW,CAAC;QAC3C,IAAI,MAAM,KAAK,IAAI,CAAC,GAAG;YAAE,OAAO;QAChC,IAAI,CAAC,GAAG,GAAG,MAAM,CAAC;QAClB,IAAI,CAAC,OAAO,CAAC,SAAS,CAAC,MAAM;YAC3B,CAAC,CAAC,EAAC,MAAM,EAAE,IAAI,CAAC,MAAM,EAAE,MAAM,EAAE,aAAa,EAAE,SAAS,EAAE,eAAe,EAAE,UAAU,EAAE,MAAM,EAAE,WAAW,EAAE,EAAC,IAAI,EAAE,UAAU,EAAC,EAAqB;YACnJ,CAAC,CAAC,EAAC,MAAM,EAAE,IAAI,CAAC,MAAM,EAAE,MAAM,EAAE,IAAI,CAAC,cAAc,EAAE,SAAS,EAAE,eAAe,EAAC,CAAC,CAAC;IACtF,CAAC;IACD,OAAO,KAAI,IAAI,CAAC,OAAO,CAAC,WAAW,EAAE,EAAE,CAAC,CAAA,CAAC;CAC1C","sourcesContent":["import {createCanvas} from '../util/canvas.js';\ntype HDRConfiguration = GPUCanvasConfiguration & {toneMapping: {mode: 'extended'}};\ntype InspectableContext = GPUCanvasContext & {getConfiguration?: () => GPUCanvasConfiguration & {toneMapping?: {mode?: string}}};\n/** One canvas surface; composing or exporting an image never reconfigures it. */\nexport class WGPUCanvasSurface {\n  readonly context: InspectableContext;\n  hdr = false;\n  private compatibleHDR?: boolean;\n  constructor(canvas: HTMLCanvasElement | OffscreenCanvas, private readonly device: GPUDevice, private readonly standardFormat: GPUTextureFormat) {\n    this.context = canvas.getContext('webgpu')! as InspectableContext;\n    this.context.configure({device, format: standardFormat, alphaMode: 'premultiplied'});\n  }\n  get format() {return this.hdr ? 'rgba16float' as const : this.standardFormat;}\n  get supportsHDR() {\n    if (this.compatibleHDR !== undefined) return this.compatibleHDR;\n    this.compatibleHDR = false;\n    let probe: InspectableContext | null = null;\n    try {\n      probe = createCanvas().getContext('webgpu') as InspectableContext | null;\n      if (!probe) return false;\n      probe.configure({device: this.device, format: 'rgba16float', alphaMode: 'premultiplied', colorSpace: 'srgb', toneMapping: {mode: 'extended'}} as HDRConfiguration);\n      const applied = probe.getConfiguration?.();\n      this.compatibleHDR = applied?.format === 'rgba16float' && applied.toneMapping?.mode === 'extended';\n    } catch {this.compatibleHDR = false;}\n    finally {probe?.unconfigure?.();}\n    return this.compatibleHDR;\n  }\n  configure(enabled: boolean) {\n    const active = enabled && this.supportsHDR;\n    if (active === this.hdr) return;\n    this.hdr = active;\n    this.context.configure(active\n      ? {device: this.device, format: 'rgba16float', alphaMode: 'premultiplied', colorSpace: 'srgb', toneMapping: {mode: 'extended'}} as HDRConfiguration\n      : {device: this.device, format: this.standardFormat, alphaMode: 'premultiplied'});\n  }\n  dispose() {this.context.unconfigure?.();}\n}\n"]}

package/lib/renderer/WGPUComputePass.d.ts

@@ -0,0 +1,73 @@
+import { Uniform } from './WGPUFullQuadPass.js';
+export type { Uniform };
+export declare const WORKGROUP_SIZE = 8;
+export interface WGPUComputePassInput {
+    buffer?: GPUBuffer;
+    texture?: GPUTexture;
+    channels: number;
+    type?: 'u32' | 'i32' | 'f32';
+}
+export interface WGPUComputePassOutput {
+    channels?: number;
+    count?: number;
+    atomic?: boolean;
+    type?: 'u32' | 'i32' | 'f32';
+}
+export declare class WGPUComputePass<I extends string, O extends string> {
+    private _label;
+    private _device;
+    private _outputBuffers;
+    private _outputParams;
+    private _pipeline;
+    private _bindGroups;
+    private _needsUpdatePipeline;
+    private _inputs;
+    private _outputs;
+    private _uniforms;
+    private _uniformsMap;
+    private _uniformBuffers;
+    private _capacityWidth;
+    private _capacityHeight;
+    private _reserveCapacity;
+    private _width;
+    private _height;
+    private _execWidth?;
+    private _execHeight?;
+    private _workGroupSize;
+    private _csCode;
+    private _csMain;
+    private _csDefine;
+    private _groupOffsets;
+    constructor(label: string, device: GPUDevice, opts: {
+        inputs: I[];
+        outputs: O[];
+        csMain: string;
+        csDefine?: string;
+        uniforms?: Uniform[];
+    });
+    setCSCode({ csDefine, csMain }: {
+        csDefine: string;
+        csMain: string;
+    }): void;
+    reserveCapacity(): void;
+    setSize(width: number, height: number): void;
+    setExecuteSize(width: number, height: number): void;
+    setOutputParams(outputParams: Record<O, WGPUComputePassOutput>): void;
+    setUniform(label: string, data: Uniform['data']): void;
+    getOutput(name: O): GPUBuffer;
+    getMemoryBytes(): number;
+    dispose(): void;
+    setWorkGroupSize(width: number, height: number): void;
+    private _getActualWorkgroupSize;
+    private _createBuffer;
+    private _invalidateChangedOutputBuffers;
+    private _updateOutputBuffers;
+    private _updatePipeline;
+    private _getFullCs;
+    private _updateBindGroups;
+    createPass(commandEncoder: GPUCommandEncoder, inputs: Record<I, WGPUComputePassInput>, 
+    /**
+     * Explict outputs. Used for buffers reuse.
+     */
+    outputs?: Record<O, GPUBuffer>): void;
+}

package/lib/renderer/WGPUComputePass.js

@@ -0,0 +1,317 @@
+import { getTargetShaderCache } from './WGPUShaderCache.js';
+// TODO output read write buffer
+import { getUniformBuffer } from './WGPUFullQuadPass.js';
+function isStorageParamsEqual(params, other) {
+    return (params.channels === other.channels &&
+        params.type === other.type &&
+        params.count === other.count &&
+        params.atomic === other.atomic);
+}
+export const WORKGROUP_SIZE = 8;
+export class WGPUComputePass {
+    _label;
+    _device;
+    _outputBuffers = {};
+    _outputParams = {};
+    _pipeline;
+    _bindGroups = [];
+    _needsUpdatePipeline = true;
+    _inputs = [];
+    _outputs = [];
+    _uniforms = [];
+    _uniformsMap = {};
+    _uniformBuffers = {};
+    _capacityWidth = 0;
+    _capacityHeight = 0;
+    _reserveCapacity = false;
+    _width = 10;
+    _height = 10;
+    _execWidth;
+    _execHeight;
+    _workGroupSize = [WORKGROUP_SIZE, WORKGROUP_SIZE];
+    _csCode = '';
+    _csMain;
+    _csDefine;
+    _groupOffsets = {
+        inputs: 0,
+        uniforms: 1,
+        outputs: 2
+    };
+    constructor(label, device, opts) {
+        this._label = label;
+        this._device = device;
+        this._csMain = opts.csMain;
+        this._csDefine = opts.csDefine;
+        this._inputs = opts.inputs;
+        this._outputs = opts.outputs;
+        const uniforms = opts.uniforms || [];
+        this._uniforms = uniforms;
+        this._uniformsMap = this._uniforms.reduce((map, uniform) => {
+            map[uniform.label] = uniform;
+            return map;
+        }, {});
+        uniforms.forEach((uniform) => {
+            const data = getUniformBuffer(uniform.data, uniform.type);
+            this._uniformBuffers[uniform.label] = device.createBuffer({
+                label: this._label + '.' + uniform.label,
+                size: data.byteLength,
+                usage: GPUBufferUsage.UNIFORM | GPUBufferUsage.COPY_DST
+            });
+            this._device.queue.writeBuffer(this._uniformBuffers[uniform.label], 0, data);
+        });
+    }
+    setCSCode({ csDefine, csMain }) {
+        this._csDefine = csDefine;
+        this._csMain = csMain;
+        this._needsUpdatePipeline = true;
+    }
+    reserveCapacity() { this._reserveCapacity = true; this._capacityWidth = this._width; this._capacityHeight = this._height; }
+    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) {
+            this._needsUpdatePipeline = true;
+            const nextWidth = Math.max(this._capacityWidth, width), nextHeight = Math.max(this._capacityHeight, height);
+            const capacityChanged = nextWidth !== this._capacityWidth || nextHeight !== this._capacityHeight;
+            this._capacityWidth = nextWidth;
+            this._capacityHeight = nextHeight;
+            if (this._reserveCapacity && !capacityChanged)
+                return;
+            // Invalidate all buffers
+            for (const key in this._outputBuffers) {
+                this._outputBuffers[key].destroy();
+            }
+            this._outputBuffers = {};
+        }
+    }
+    setExecuteSize(width, height) {
+        width = Math.max(1, Math.ceil(width));
+        height = Math.max(1, Math.ceil(height));
+        this._execWidth = width;
+        this._execHeight = height;
+    }
+    setOutputParams(outputParams) {
+        // TODO lazy update buffers
+        this._invalidateChangedOutputBuffers(outputParams);
+        this._outputParams = outputParams;
+        this._needsUpdatePipeline = true;
+    }
+    setUniform(label, data) {
+        const buffer = this._uniformBuffers[label];
+        this._device.queue.writeBuffer(buffer, 0, getUniformBuffer(data, this._uniformsMap[label].type));
+    }
+    getOutput(name) {
+        const outputBuffers = this._outputBuffers;
+        if (!outputBuffers[name]) {
+            outputBuffers[name] = this._createBuffer(name, this._outputParams[name] || {});
+        }
+        return outputBuffers[name];
+    }
+    getMemoryBytes() {
+        return [...Object.values(this._uniformBuffers), ...Object.values(this._outputBuffers)]
+            .reduce((sum, buffer) => sum + buffer.size, 0);
+    }
+    dispose() {
+        Object.keys(this._uniformBuffers).forEach((key) => {
+            this._uniformBuffers[key].destroy();
+        });
+        Object.keys(this._outputBuffers).forEach((key) => {
+            this._outputBuffers[key].destroy();
+        });
+    }
+    setWorkGroupSize(width, height) {
+        this._needsUpdatePipeline =
+            width !== this._workGroupSize[0] || height !== this._workGroupSize[1];
+        this._workGroupSize = [width, height];
+    }
+    _getActualWorkgroupSize() {
+        return [
+            Math.min(this._workGroupSize[0], this._execWidth ?? this._width),
+            Math.min(this._workGroupSize[1], this._execHeight ?? this._height)
+        ];
+    }
+    _createBuffer(label, params) {
+        const channels = params.channels ?? 4;
+        const byteLength = (this._reserveCapacity ? this._capacityWidth * this._capacityHeight : this._width * this._height) * channels * (params.count ?? 1) * 4;
+        return this._device.createBuffer({
+            label: this._label + '.' + label,
+            // webgpu needs buffer at least 80 bytes.
+            size: Math.max(byteLength, 80),
+            usage: GPUBufferUsage.STORAGE | GPUBufferUsage.COPY_SRC
+        });
+    }
+    _invalidateChangedOutputBuffers(newParams) {
+        const outputBuffers = this._outputBuffers;
+        const oldParams = this._outputParams;
+        for (const key in outputBuffers) {
+            if (!isStorageParamsEqual(newParams[key] || {}, oldParams[key] || {})) {
+                outputBuffers[key].destroy();
+                delete outputBuffers[key];
+            }
+        }
+    }
+    _updateOutputBuffers(excludes) {
+        const outputBuffers = this._outputBuffers;
+        const params = this._outputParams;
+        for (const key in params) {
+            if (!outputBuffers[key] && !excludes?.includes(key)) {
+                outputBuffers[key] = this._createBuffer(key, params[key]);
+            }
+        }
+    }
+    _updatePipeline(inputParams, inputTypes) {
+        if (!this._needsUpdatePipeline) {
+            return;
+        }
+        this._needsUpdatePipeline = false;
+        const device = this._device;
+        const csCode = this._getFullCs(inputParams, inputTypes);
+        if (csCode === this._csCode) {
+            return;
+        }
+        this._csCode = csCode;
+        this._pipeline = getTargetShaderCache(device).createComputePipeline({
+            label: this._label,
+            layout: 'auto',
+            compute: {
+                module: getTargetShaderCache(device).createShaderModule({
+                    label: this._label,
+                    code: csCode
+                }),
+                entryPoint: 'main'
+            }
+        });
+        this._updateBindGroups();
+    }
+    _getFullCs(inputParams, inputTypes) {
+        const inputs = this._inputs;
+        const uniforms = this._uniforms;
+        let offset = 0;
+        const groupOffsets = (this._groupOffsets = {
+            inputs: 0,
+            uniforms: 0,
+            outputs: 0
+        });
+        const workgroupSize = this._getActualWorkgroupSize();
+        if (inputs.length > 0) {
+            offset++;
+        }
+        if (uniforms.length > 0) {
+            groupOffsets.uniforms = offset;
+            offset++;
+        }
+        function getTypeCode(params) {
+            const { type = 'f32', channels } = params;
+            let code = channels === 1 ? type : `vec${channels}${type[0]}`;
+            if (params.atomic) {
+                code = `atomic<${code}>`;
+            }
+            return code;
+        }
+        groupOffsets.outputs = offset;
+        const cs = /* wgsl */ `
+${inputs
+            .sort()
+            .map((inputName, idx) => {
+            const bindingPrefix = `@group(${groupOffsets.inputs}) @binding(${idx}) `;
+            const varName = `in_${inputName}`;
+            // TODO more channels option.
+            return inputTypes[inputName] === 'texture'
+                ? `${bindingPrefix} var ${varName}: texture_2d<f32>;`
+                : `${bindingPrefix} var<storage, read> ${varName}: array<${getTypeCode(inputParams[inputName])}>;`;
+        })
+            .join('\n')}
+${this._uniforms
+            .map((uniform, idx) => `@group(${groupOffsets.uniforms}) @binding(${idx}) var<uniform> ${uniform.label}: ${uniform.type};`)
+            .join('\n')}
+
+${this._outputs
+            .map((name, idx) => {
+            return `@group(${groupOffsets.outputs}) @binding(${idx}) var<storage, read_write> out_${name}: array<${getTypeCode(this._outputParams[name] || {})}>;`;
+        })
+            .join('\n')}
+${this._csDefine ?? ''}
+@compute @workgroup_size(${workgroupSize[0]}, ${workgroupSize[1]}, 1)
+fn main(@builtin(global_invocation_id) globalId: vec3u) {
+${this._csMain}
+}
+`;
+        return cs;
+    }
+    _updateBindGroups() {
+        const bindGroups = [];
+        const device = this._device;
+        const groupOffsets = this._groupOffsets;
+        if (this._uniforms.length > 0) {
+            bindGroups[groupOffsets.uniforms] = device.createBindGroup({
+                label: this._label + '.uniforms',
+                layout: this._pipeline.getBindGroupLayout(groupOffsets.uniforms),
+                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) {
+        this._updateOutputBuffers(Object.keys(outputs || {}));
+        const outputBuffers = {
+            ...this._outputBuffers,
+            ...outputs
+        };
+        const inputTypes = this._inputs.reduce((obj, inputName) => {
+            obj[inputName] = inputs[inputName].buffer ? 'buffer' : 'texture';
+            return obj;
+        }, {});
+        this._updatePipeline(inputs, inputTypes);
+        const groupOffsets = this._groupOffsets;
+        if (this._inputs.length > 0) {
+            this._bindGroups[groupOffsets.inputs] = this._device.createBindGroup({
+                label: this._label + '.inputs',
+                layout: this._pipeline.getBindGroupLayout(groupOffsets.inputs),
+                entries: this._inputs.map((inputName, idx) => ({
+                    binding: idx,
+                    resource: inputs[inputName].buffer
+                        ? {
+                            buffer: inputs[inputName].buffer
+                        }
+                        : inputs[inputName].texture.createView()
+                }))
+            });
+        }
+        // Outputs
+        this._bindGroups[groupOffsets.outputs] = this._device.createBindGroup({
+            label: this._label + '.outputs',
+            layout: this._pipeline.getBindGroupLayout(groupOffsets.outputs),
+            entries: this._outputs.map((bufferName, idx) => ({
+                binding: idx,
+                resource: {
+                    buffer: outputBuffers[bufferName]
+                }
+            }))
+        });
+        // Begin the render pass
+        const computePass = commandEncoder.beginComputePass();
+        // Draw a full quad
+        computePass.setPipeline(this._pipeline);
+        // Bind groups
+        this._bindGroups.forEach((bindGroup, idx) => {
+            computePass.setBindGroup(idx, bindGroup);
+        });
+        const workgroupSize = this._getActualWorkgroupSize();
+        computePass.dispatchWorkgroups(Math.ceil((this._execWidth ?? this._width) / workgroupSize[0]), Math.ceil((this._execHeight ?? this._height) / workgroupSize[1]), 1);
+        // End the render pass
+        computePass.end();
+    }
+}
+//# sourceMappingURL=WGPUComputePass.js.map

package/lib/renderer/WGPUComputePass.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"WGPUComputePass.js","sourceRoot":"","sources":["../../src/renderer/WGPUComputePass.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,oBAAoB,EAAE,MAAM,sBAAsB,CAAC;AAC5D,gCAAgC;AAChC,OAAO,EAAE,gBAAgB,EAAW,MAAM,uBAAuB,CAAC;AAIlE,SAAS,oBAAoB,CAC3B,MAA6B,EAC7B,KAA4B;IAE5B,OAAO,CACL,MAAM,CAAC,QAAQ,KAAK,KAAK,CAAC,QAAQ;QAClC,MAAM,CAAC,IAAI,KAAK,KAAK,CAAC,IAAI;QAC1B,MAAM,CAAC,KAAK,KAAK,KAAK,CAAC,KAAK;QAC5B,MAAM,CAAC,MAAM,KAAK,KAAK,CAAC,MAAM,CAC/B,CAAC;AACJ,CAAC;AAED,MAAM,CAAC,MAAM,cAAc,GAAG,CAAC,CAAC;AAiBhC,MAAM,OAAO,eAAe;IAClB,MAAM,CAAC;IAEP,OAAO,CAAC;IACR,cAAc,GAA8B,EAAE,CAAC;IAC/C,aAAa,GAA0C,EAAE,CAAC;IAE1D,SAAS,CAAsB;IAC/B,WAAW,GAAmB,EAAE,CAAC;IACjC,oBAAoB,GAAG,IAAI,CAAC;IAE5B,OAAO,GAAa,EAAE,CAAC;IACvB,QAAQ,GAAa,EAAE,CAAC;IACxB,SAAS,GAAc,EAAE,CAAC;IAC1B,YAAY,GAA4B,EAAE,CAAC;IAC3C,eAAe,GAA8B,EAAE,CAAC;IAEhD,cAAc,GAAG,CAAC,CAAC;IACnB,eAAe,GAAG,CAAC,CAAC;IACpB,gBAAgB,GAAG,KAAK,CAAC;IACzB,MAAM,GAAG,EAAE,CAAC;IACZ,OAAO,GAAG,EAAE,CAAC;IAEb,UAAU,CAAU;IACpB,WAAW,CAAU;IAErB,cAAc,GAAG,CAAC,cAAc,EAAE,cAAc,CAAC,CAAC;IAElD,OAAO,GAAG,EAAE,CAAC;IACb,OAAO,CAAC;IACR,SAAS,CAAC;IAEV,aAAa,GAAG;QACtB,MAAM,EAAE,CAAC;QACT,QAAQ,EAAE,CAAC;QACX,OAAO,EAAE,CAAC;KACX,CAAC;IAEF,YACE,KAAa,EACb,MAAiB,EACjB,IAMC;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,QAAQ,GAAG,IAAI,CAAC,OAAO,CAAC;QAC7B,MAAM,QAAQ,GAAG,IAAI,CAAC,QAAQ,IAAI,EAAE,CAAC;QACrC,IAAI,CAAC,SAAS,GAAG,QAAQ,CAAC;QAE1B,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,QAAQ,CAAC,OAAO,CAAC,CAAC,OAAO,EAAE,EAAE;YAC3B,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,GAAG,GAAG,GAAG,OAAO,CAAC,KAAK;gBACxC,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,SAAS,CAAC,EAAE,QAAQ,EAAE,MAAM,EAAwC;QAClE,IAAI,CAAC,SAAS,GAAG,QAAQ,CAAC;QAC1B,IAAI,CAAC,OAAO,GAAG,MAAM,CAAC;QACtB,IAAI,CAAC,oBAAoB,GAAG,IAAI,CAAC;IACnC,CAAC;IAED,eAAe,KAAK,IAAI,CAAC,gBAAgB,GAAG,IAAI,CAAC,CAAC,IAAI,CAAC,cAAc,GAAG,IAAI,CAAC,MAAM,CAAC,CAAC,IAAI,CAAC,eAAe,GAAG,IAAI,CAAC,OAAO,CAAC,CAAC,CAAC;IAC3H,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,IAAI,CAAC,oBAAoB,GAAG,IAAI,CAAC;YACjC,MAAM,SAAS,GAAG,IAAI,CAAC,GAAG,CAAC,IAAI,CAAC,cAAc,EAAE,KAAK,CAAC,EAAE,UAAU,GAAG,IAAI,CAAC,GAAG,CAAC,IAAI,CAAC,eAAe,EAAE,MAAM,CAAC,CAAC;YAC5G,MAAM,eAAe,GAAG,SAAS,KAAK,IAAI,CAAC,cAAc,IAAI,UAAU,KAAK,IAAI,CAAC,eAAe,CAAC;YACjG,IAAI,CAAC,cAAc,GAAG,SAAS,CAAC;YAAC,IAAI,CAAC,eAAe,GAAG,UAAU,CAAC;YACnE,IAAI,IAAI,CAAC,gBAAgB,IAAI,CAAC,eAAe;gBAAE,OAAO;YACtD,yBAAyB;YACzB,KAAK,MAAM,GAAG,IAAI,IAAI,CAAC,cAAc,EAAE,CAAC;gBACtC,IAAI,CAAC,cAAc,CAAC,GAAG,CAAC,CAAC,OAAO,EAAE,CAAC;YACrC,CAAC;YACD,IAAI,CAAC,cAAc,GAAG,EAAE,CAAC;QAC3B,CAAC;IACH,CAAC;IAED,cAAc,CAAC,KAAa,EAAE,MAAc;QAC1C,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,IAAI,CAAC,UAAU,GAAG,KAAK,CAAC;QACxB,IAAI,CAAC,WAAW,GAAG,MAAM,CAAC;IAC5B,CAAC;IAED,eAAe,CAAC,YAA8C;QAC5D,2BAA2B;QAC3B,IAAI,CAAC,+BAA+B,CAAC,YAAY,CAAC,CAAC;QACnD,IAAI,CAAC,aAAa,GAAG,YAAY,CAAC;QAClC,IAAI,CAAC,oBAAoB,GAAG,IAAI,CAAC;IACnC,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;IAED,SAAS,CAAC,IAAO;QACf,MAAM,aAAa,GAAG,IAAI,CAAC,cAAc,CAAC;QAC1C,IAAI,CAAC,aAAa,CAAC,IAAI,CAAC,EAAE,CAAC;YACzB,aAAa,CAAC,IAAI,CAAC,GAAG,IAAI,CAAC,aAAa,CACtC,IAAI,EACJ,IAAI,CAAC,aAAa,CAAC,IAAI,CAAC,IAAI,EAAE,CAC/B,CAAC;QACJ,CAAC;QACD,OAAO,aAAa,CAAC,IAAI,CAAC,CAAC;IAC7B,CAAC;IAED,cAAc;QACZ,OAAO,CAAC,GAAG,MAAM,CAAC,MAAM,CAAC,IAAI,CAAC,eAAe,CAAC,EAAE,GAAG,MAAM,CAAC,MAAM,CAAC,IAAI,CAAC,cAAc,CAAC,CAAC;aACnF,MAAM,CAAC,CAAC,GAAG,EAAE,MAAM,EAAE,EAAE,CAAC,GAAG,GAAG,MAAM,CAAC,IAAI,EAAE,CAAC,CAAC,CAAC;IACnD,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,MAAM,CAAC,IAAI,CAAC,IAAI,CAAC,cAAc,CAAC,CAAC,OAAO,CAAC,CAAC,GAAG,EAAE,EAAE;YAC9C,IAAI,CAAC,cAAsB,CAAC,GAAG,CAAC,CAAC,OAAO,EAAE,CAAC;QAC9C,CAAC,CAAC,CAAC;IACL,CAAC;IAED,gBAAgB,CAAC,KAAa,EAAE,MAAc;QAC5C,IAAI,CAAC,oBAAoB;YACvB,KAAK,KAAK,IAAI,CAAC,cAAc,CAAC,CAAC,CAAC,IAAI,MAAM,KAAK,IAAI,CAAC,cAAc,CAAC,CAAC,CAAC,CAAC;QACxE,IAAI,CAAC,cAAc,GAAG,CAAC,KAAK,EAAE,MAAM,CAAC,CAAC;IACxC,CAAC;IAEO,uBAAuB;QAC7B,OAAO;YACL,IAAI,CAAC,GAAG,CAAC,IAAI,CAAC,cAAc,CAAC,CAAC,CAAC,EAAE,IAAI,CAAC,UAAU,IAAI,IAAI,CAAC,MAAM,CAAC;YAChE,IAAI,CAAC,GAAG,CAAC,IAAI,CAAC,cAAc,CAAC,CAAC,CAAC,EAAE,IAAI,CAAC,WAAW,IAAI,IAAI,CAAC,OAAO,CAAC;SACnE,CAAC;IACJ,CAAC;IAEO,aAAa,CAAC,KAAa,EAAE,MAA6B;QAChE,MAAM,QAAQ,GAAG,MAAM,CAAC,QAAQ,IAAI,CAAC,CAAC;QACtC,MAAM,UAAU,GACd,CAAC,IAAI,CAAC,gBAAgB,CAAC,CAAC,CAAC,IAAI,CAAC,cAAc,GAAG,IAAI,CAAC,eAAe,CAAC,CAAC,CAAC,IAAI,CAAC,MAAM,GAAG,IAAI,CAAC,OAAO,CAAC,GAAG,QAAQ,GAAG,CAAC,MAAM,CAAC,KAAK,IAAI,CAAC,CAAC,GAAG,CAAC,CAAC;QACzI,OAAO,IAAI,CAAC,OAAO,CAAC,YAAY,CAAC;YAC/B,KAAK,EAAE,IAAI,CAAC,MAAM,GAAG,GAAG,GAAG,KAAK;YAChC,yCAAyC;YACzC,IAAI,EAAE,IAAI,CAAC,GAAG,CAAC,UAAU,EAAE,EAAE,CAAC;YAC9B,KAAK,EAAE,cAAc,CAAC,OAAO,GAAG,cAAc,CAAC,QAAQ;SACxD,CAAC,CAAC;IACL,CAAC;IAEO,+BAA+B,CAAC,SAA8B;QACpE,MAAM,aAAa,GAAG,IAAI,CAAC,cAAc,CAAC;QAC1C,MAAM,SAAS,GAAG,IAAI,CAAC,aAAa,CAAC;QACrC,KAAK,MAAM,GAAG,IAAI,aAAa,EAAE,CAAC;YAChC,IAAI,CAAC,oBAAoB,CAAC,SAAS,CAAC,GAAG,CAAC,IAAI,EAAE,EAAE,SAAS,CAAC,GAAG,CAAC,IAAI,EAAE,CAAC,EAAE,CAAC;gBACtE,aAAa,CAAC,GAAG,CAAC,CAAC,OAAO,EAAE,CAAC;gBAC7B,OAAO,aAAa,CAAC,GAAG,CAAC,CAAC;YAC5B,CAAC;QACH,CAAC;IACH,CAAC;IAEO,oBAAoB,CAAC,QAAmB;QAC9C,MAAM,aAAa,GAAG,IAAI,CAAC,cAAc,CAAC;QAC1C,MAAM,MAAM,GAAG,IAAI,CAAC,aAAa,CAAC;QAClC,KAAK,MAAM,GAAG,IAAI,MAAM,EAAE,CAAC;YACzB,IAAI,CAAC,aAAa,CAAC,GAAG,CAAC,IAAI,CAAC,QAAQ,EAAE,QAAQ,CAAC,GAAG,CAAC,EAAE,CAAC;gBACpD,aAAa,CAAC,GAAG,CAAC,GAAG,IAAI,CAAC,aAAa,CAAC,GAAG,EAAE,MAAM,CAAC,GAAG,CAAC,CAAC,CAAC;YAC5D,CAAC;QACH,CAAC;IACH,CAAC;IAEO,eAAe,CACrB,WAAiD,EACjD,UAAgD;QAEhD,IAAI,CAAC,IAAI,CAAC,oBAAoB,EAAE,CAAC;YAC/B,OAAO;QACT,CAAC;QACD,IAAI,CAAC,oBAAoB,GAAG,KAAK,CAAC;QAClC,MAAM,MAAM,GAAG,IAAI,CAAC,OAAO,CAAC;QAC5B,MAAM,MAAM,GAAG,IAAI,CAAC,UAAU,CAAC,WAAW,EAAE,UAAU,CAAC,CAAC;QACxD,IAAI,MAAM,KAAK,IAAI,CAAC,OAAO,EAAE,CAAC;YAC5B,OAAO;QACT,CAAC;QACD,IAAI,CAAC,OAAO,GAAG,MAAM,CAAC;QACtB,IAAI,CAAC,SAAS,GAAG,oBAAoB,CAAC,MAAM,CAAC,CAAC,qBAAqB,CAAC;YAClE,KAAK,EAAE,IAAI,CAAC,MAAM;YAClB,MAAM,EAAE,MAAM;YACd,OAAO,EAAE;gBACP,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;aACnB;SACF,CAAC,CAAC;QACH,IAAI,CAAC,iBAAiB,EAAE,CAAC;IAC3B,CAAC;IAEO,UAAU,CAChB,WAAiD,EACjD,UAAgD;QAEhD,MAAM,MAAM,GAAG,IAAI,CAAC,OAAO,CAAC;QAC5B,MAAM,QAAQ,GAAG,IAAI,CAAC,SAAS,CAAC;QAChC,IAAI,MAAM,GAAG,CAAC,CAAC;QACf,MAAM,YAAY,GAAG,CAAC,IAAI,CAAC,aAAa,GAAG;YACzC,MAAM,EAAE,CAAC;YACT,QAAQ,EAAE,CAAC;YACX,OAAO,EAAE,CAAC;SACX,CAAC,CAAC;QACH,MAAM,aAAa,GAAG,IAAI,CAAC,uBAAuB,EAAE,CAAC;QAErD,IAAI,MAAM,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;YACtB,MAAM,EAAE,CAAC;QACX,CAAC;QACD,IAAI,QAAQ,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;YACxB,YAAY,CAAC,QAAQ,GAAG,MAAM,CAAC;YAC/B,MAAM,EAAE,CAAC;QACX,CAAC;QAED,SAAS,WAAW,CAAC,MAA6B;YAChD,MAAM,EAAE,IAAI,GAAG,KAAK,EAAE,QAAQ,EAAE,GAAG,MAAM,CAAC;YAC1C,IAAI,IAAI,GAAG,QAAQ,KAAK,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,MAAM,QAAQ,GAAG,IAAI,CAAC,CAAC,CAAC,EAAE,CAAC;YAC9D,IAAI,MAAM,CAAC,MAAM,EAAE,CAAC;gBAClB,IAAI,GAAG,UAAU,IAAI,GAAG,CAAC;YAC3B,CAAC;YACD,OAAO,IAAI,CAAC;QACd,CAAC;QACD,YAAY,CAAC,OAAO,GAAG,MAAM,CAAC;QAC9B,MAAM,EAAE,GAAG,UAAU,CAAC;EACxB,MAAM;aACL,IAAI,EAAE;aACN,GAAG,CAAC,CAAC,SAAS,EAAE,GAAG,EAAE,EAAE;YACtB,MAAM,aAAa,GAAG,UAAU,YAAY,CAAC,MAAM,cAAc,GAAG,IAAI,CAAC;YACzE,MAAM,OAAO,GAAG,MAAM,SAAS,EAAE,CAAC;YAClC,6BAA6B;YAC7B,OAAO,UAAU,CAAC,SAAS,CAAC,KAAK,SAAS;gBACxC,CAAC,CAAC,GAAG,aAAa,QAAQ,OAAO,oBAAoB;gBACrD,CAAC,CAAC,GAAG,aAAa,uBAAuB,OAAO,WAAW,WAAW,CAClE,WAAW,CAAC,SAAS,CAAC,CACvB,IAAI,CAAC;QACZ,CAAC,CAAC;aACD,IAAI,CAAC,IAAI,CAAC;EACX,IAAI,CAAC,SAAS;aACb,GAAG,CACF,CAAC,OAAO,EAAE,GAAG,EAAE,EAAE,CACf,UAAU,YAAY,CAAC,QAAQ,cAAc,GAAG,kBAAkB,OAAO,CAAC,KAAK,KAAK,OAAO,CAAC,IAAI,GAAG,CACtG;aACA,IAAI,CAAC,IAAI,CAAC;;EAEX,IAAI,CAAC,QAAQ;aACZ,GAAG,CAAC,CAAC,IAAI,EAAE,GAAG,EAAE,EAAE;YACjB,OAAO,UACL,YAAY,CAAC,OACf,cAAc,GAAG,kCAAkC,IAAI,WAAW,WAAW,CAC3E,IAAI,CAAC,aAAa,CAAC,IAAI,CAAC,IAAI,EAAE,CAC/B,IAAI,CAAC;QACR,CAAC,CAAC;aACD,IAAI,CAAC,IAAI,CAAC;EACX,IAAI,CAAC,SAAS,IAAI,EAAE;2BACK,aAAa,CAAC,CAAC,CAAC,KAAK,aAAa,CAAC,CAAC,CAAC;;EAE9D,IAAI,CAAC,OAAO;;CAEb,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,YAAY,GAAG,IAAI,CAAC,aAAa,CAAC;QAExC,IAAI,IAAI,CAAC,SAAS,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;YAC9B,UAAU,CAAC,YAAY,CAAC,QAAQ,CAAC,GAAG,MAAM,CAAC,eAAe,CAAC;gBACzD,KAAK,EAAE,IAAI,CAAC,MAAM,GAAG,WAAW;gBAChC,MAAM,EAAE,IAAI,CAAC,SAAS,CAAC,kBAAkB,CAAC,YAAY,CAAC,QAAQ,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;iBACoB,CAAA,CAC1B;aACF,CAAC,CAAC;QACL,CAAC;QAED,IAAI,CAAC,WAAW,GAAG,UAAU,CAAC;IAChC,CAAC;IAED,UAAU,CACR,cAAiC,EACjC,MAAuC;IAEvC;;OAEG;IACH,OAA8B;QAE9B,IAAI,CAAC,oBAAoB,CAAC,MAAM,CAAC,IAAI,CAAC,OAAO,IAAI,EAAE,CAAC,CAAC,CAAC;QACtD,MAAM,aAAa,GAAG;YACpB,GAAG,IAAI,CAAC,cAAc;YACtB,GAAG,OAAO;SACkB,CAAC;QAE/B,MAAM,UAAU,GAAG,IAAI,CAAC,OAAO,CAAC,MAAM,CAAC,CAAC,GAAG,EAAE,SAAS,EAAE,EAAE;YACxD,GAAG,CAAC,SAAS,CAAC,GAAG,MAAM,CAAC,SAAc,CAAC,CAAC,MAAM,CAAC,CAAC,CAAC,QAAQ,CAAC,CAAC,CAAC,SAAS,CAAC;YACtE,OAAO,GAAG,CAAC;QACb,CAAC,EAAE,EAA0C,CAAC,CAAC;QAE/C,IAAI,CAAC,eAAe,CAAC,MAAM,EAAE,UAAU,CAAC,CAAC;QAEzC,MAAM,YAAY,GAAG,IAAI,CAAC,aAAa,CAAC;QACxC,IAAI,IAAI,CAAC,OAAO,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;YAC5B,IAAI,CAAC,WAAW,CAAC,YAAY,CAAC,MAAM,CAAC,GAAG,IAAI,CAAC,OAAO,CAAC,eAAe,CAAC;gBACnE,KAAK,EAAE,IAAI,CAAC,MAAM,GAAG,SAAS;gBAC9B,MAAM,EAAE,IAAI,CAAC,SAAS,CAAC,kBAAkB,CAAC,YAAY,CAAC,MAAM,CAAC;gBAC9D,OAAO,EAAE,IAAI,CAAC,OAAO,CAAC,GAAG,CAAC,CAAC,SAAS,EAAE,GAAG,EAAE,EAAE,CAAC,CAAC;oBAC7C,OAAO,EAAE,GAAG;oBACZ,QAAQ,EAAE,MAAM,CAAC,SAAc,CAAC,CAAC,MAAM;wBACrC,CAAC,CAAC;4BACE,MAAM,EAAE,MAAM,CAAC,SAAc,CAAC,CAAC,MAAO;yBACvC;wBACH,CAAC,CAAC,MAAM,CAAC,SAAc,CAAC,CAAC,OAAQ,CAAC,UAAU,EAAE;iBACjD,CAAC,CAAC;aACJ,CAAC,CAAC;QACL,CAAC;QAED,UAAU;QACV,IAAI,CAAC,WAAW,CAAC,YAAY,CAAC,OAAO,CAAC,GAAG,IAAI,CAAC,OAAO,CAAC,eAAe,CAAC;YACpE,KAAK,EAAE,IAAI,CAAC,MAAM,GAAG,UAAU;YAC/B,MAAM,EAAE,IAAI,CAAC,SAAS,CAAC,kBAAkB,CAAC,YAAY,CAAC,OAAO,CAAC;YAC/D,OAAO,EAAE,IAAI,CAAC,QAAQ,CAAC,GAAG,CAAC,CAAC,UAAU,EAAE,GAAG,EAAE,EAAE,CAAC,CAAC;gBAC/C,OAAO,EAAE,GAAG;gBACZ,QAAQ,EAAE;oBACR,MAAM,EAAE,aAAa,CAAC,UAAU,CAAC;iBAClC;aACF,CAAC,CAAC;SACJ,CAAC,CAAC;QAEH,wBAAwB;QACxB,MAAM,WAAW,GAAG,cAAc,CAAC,gBAAgB,EAAE,CAAC;QAEtD,mBAAmB;QACnB,WAAW,CAAC,WAAW,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC;QACxC,cAAc;QACd,IAAI,CAAC,WAAW,CAAC,OAAO,CAAC,CAAC,SAAS,EAAE,GAAG,EAAE,EAAE;YAC1C,WAAW,CAAC,YAAY,CAAC,GAAG,EAAE,SAAS,CAAC,CAAC;QAC3C,CAAC,CAAC,CAAC;QAEH,MAAM,aAAa,GAAG,IAAI,CAAC,uBAAuB,EAAE,CAAC;QACrD,WAAW,CAAC,kBAAkB,CAC5B,IAAI,CAAC,IAAI,CAAC,CAAC,IAAI,CAAC,UAAU,IAAI,IAAI,CAAC,MAAM,CAAC,GAAG,aAAa,CAAC,CAAC,CAAC,CAAC,EAC9D,IAAI,CAAC,IAAI,CAAC,CAAC,IAAI,CAAC,WAAW,IAAI,IAAI,CAAC,OAAO,CAAC,GAAG,aAAa,CAAC,CAAC,CAAC,CAAC,EAChE,CAAC,CACF,CAAC;QACF,sBAAsB;QACtB,WAAW,CAAC,GAAG,EAAE,CAAC;IACpB,CAAC;CACF","sourcesContent":["import { getTargetShaderCache } from './WGPUShaderCache.js';\n// TODO output read write buffer\nimport { getUniformBuffer, Uniform } from './WGPUFullQuadPass.js';\n\nexport type { Uniform };\n\nfunction isStorageParamsEqual(\n  params: WGPUComputePassOutput,\n  other: WGPUComputePassOutput\n) {\n  return (\n    params.channels === other.channels &&\n    params.type === other.type &&\n    params.count === other.count &&\n    params.atomic === other.atomic\n  );\n}\n\nexport const WORKGROUP_SIZE = 8;\n\nexport interface WGPUComputePassInput {\n  buffer?: GPUBuffer;\n  texture?: GPUTexture;\n  channels: number;\n  type?: 'u32' | 'i32' | 'f32';\n}\n\nexport interface WGPUComputePassOutput {\n  channels?: number;\n  // Default: 1. Count per element.\n  count?: number;\n  atomic?: boolean;\n  type?: 'u32' | 'i32' | 'f32';\n}\n\nexport class WGPUComputePass<I extends string, O extends string> {\n  private _label;\n\n  private _device;\n  private _outputBuffers: Record<string, GPUBuffer> = {};\n  private _outputParams: Record<string, WGPUComputePassOutput> = {};\n\n  private _pipeline!: GPUComputePipeline;\n  private _bindGroups: GPUBindGroup[] = [];\n  private _needsUpdatePipeline = true;\n\n  private _inputs: string[] = [];\n  private _outputs: string[] = [];\n  private _uniforms: Uniform[] = [];\n  private _uniformsMap: Record<string, Uniform> = {};\n  private _uniformBuffers: Record<string, GPUBuffer> = {};\n\n  private _capacityWidth = 0;\n  private _capacityHeight = 0;\n  private _reserveCapacity = false;\n  private _width = 10;\n  private _height = 10;\n\n  private _execWidth?: number;\n  private _execHeight?: number;\n\n  private _workGroupSize = [WORKGROUP_SIZE, WORKGROUP_SIZE];\n\n  private _csCode = '';\n  private _csMain;\n  private _csDefine;\n\n  private _groupOffsets = {\n    inputs: 0,\n    uniforms: 1,\n    outputs: 2\n  };\n\n  constructor(\n    label: string,\n    device: GPUDevice,\n    opts: {\n      inputs: I[];\n      outputs: O[];\n      csMain: string;\n      csDefine?: string;\n      uniforms?: Uniform[];\n    }\n  ) {\n    this._label = label;\n    this._device = device;\n    this._csMain = opts.csMain;\n    this._csDefine = opts.csDefine;\n    this._inputs = opts.inputs;\n    this._outputs = opts.outputs;\n    const uniforms = opts.uniforms || [];\n    this._uniforms = uniforms;\n\n    this._uniformsMap = this._uniforms.reduce((map, uniform) => {\n      map[uniform.label] = uniform;\n      return map;\n    }, {} as Record<string, Uniform>);\n    uniforms.forEach((uniform) => {\n      const data = getUniformBuffer(uniform.data, uniform.type);\n      this._uniformBuffers[uniform.label] = device.createBuffer({\n        label: this._label + '.' + uniform.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  setCSCode({ csDefine, csMain }: { csDefine: string; csMain: string }) {\n    this._csDefine = csDefine;\n    this._csMain = csMain;\n    this._needsUpdatePipeline = true;\n  }\n\n  reserveCapacity() { this._reserveCapacity = true; this._capacityWidth = this._width; this._capacityHeight = this._height; }\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      this._needsUpdatePipeline = true;\n      const nextWidth = Math.max(this._capacityWidth, width), nextHeight = Math.max(this._capacityHeight, height);\n      const capacityChanged = nextWidth !== this._capacityWidth || nextHeight !== this._capacityHeight;\n      this._capacityWidth = nextWidth; this._capacityHeight = nextHeight;\n      if (this._reserveCapacity && !capacityChanged) return;\n      // Invalidate all buffers\n      for (const key in this._outputBuffers) {\n        this._outputBuffers[key].destroy();\n      }\n      this._outputBuffers = {};\n    }\n  }\n\n  setExecuteSize(width: number, height: number) {\n    width = Math.max(1, Math.ceil(width));\n    height = Math.max(1, Math.ceil(height));\n    this._execWidth = width;\n    this._execHeight = height;\n  }\n\n  setOutputParams(outputParams: Record<O, WGPUComputePassOutput>) {\n    // TODO lazy update buffers\n    this._invalidateChangedOutputBuffers(outputParams);\n    this._outputParams = outputParams;\n    this._needsUpdatePipeline = true;\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  getOutput(name: O) {\n    const outputBuffers = this._outputBuffers;\n    if (!outputBuffers[name]) {\n      outputBuffers[name] = this._createBuffer(\n        name,\n        this._outputParams[name] || {}\n      );\n    }\n    return outputBuffers[name];\n  }\n\n  getMemoryBytes() {\n    return [...Object.values(this._uniformBuffers), ...Object.values(this._outputBuffers)]\n      .reduce((sum, buffer) => sum + buffer.size, 0);\n  }\n\n  dispose() {\n    Object.keys(this._uniformBuffers).forEach((key) => {\n      (this._uniformBuffers as any)[key].destroy();\n    });\n    Object.keys(this._outputBuffers).forEach((key) => {\n      (this._outputBuffers as any)[key].destroy();\n    });\n  }\n\n  setWorkGroupSize(width: number, height: number) {\n    this._needsUpdatePipeline =\n      width !== this._workGroupSize[0] || height !== this._workGroupSize[1];\n    this._workGroupSize = [width, height];\n  }\n\n  private _getActualWorkgroupSize() {\n    return [\n      Math.min(this._workGroupSize[0], this._execWidth ?? this._width),\n      Math.min(this._workGroupSize[1], this._execHeight ?? this._height)\n    ];\n  }\n\n  private _createBuffer(label: string, params: WGPUComputePassOutput) {\n    const channels = params.channels ?? 4;\n    const byteLength =\n      (this._reserveCapacity ? this._capacityWidth * this._capacityHeight : this._width * this._height) * channels * (params.count ?? 1) * 4;\n    return this._device.createBuffer({\n      label: this._label + '.' + label,\n      // webgpu needs buffer at least 80 bytes.\n      size: Math.max(byteLength, 80),\n      usage: GPUBufferUsage.STORAGE | GPUBufferUsage.COPY_SRC\n    });\n  }\n\n  private _invalidateChangedOutputBuffers(newParams: Record<string, any>) {\n    const outputBuffers = this._outputBuffers;\n    const oldParams = this._outputParams;\n    for (const key in outputBuffers) {\n      if (!isStorageParamsEqual(newParams[key] || {}, oldParams[key] || {})) {\n        outputBuffers[key].destroy();\n        delete outputBuffers[key];\n      }\n    }\n  }\n\n  private _updateOutputBuffers(excludes?: string[]) {\n    const outputBuffers = this._outputBuffers;\n    const params = this._outputParams;\n    for (const key in params) {\n      if (!outputBuffers[key] && !excludes?.includes(key)) {\n        outputBuffers[key] = this._createBuffer(key, params[key]);\n      }\n    }\n  }\n\n  private _updatePipeline(\n    inputParams: Record<string, WGPUComputePassInput>,\n    inputTypes: Record<string, 'texture' | 'buffer'>\n  ) {\n    if (!this._needsUpdatePipeline) {\n      return;\n    }\n    this._needsUpdatePipeline = false;\n    const device = this._device;\n    const csCode = this._getFullCs(inputParams, inputTypes);\n    if (csCode === this._csCode) {\n      return;\n    }\n    this._csCode = csCode;\n    this._pipeline = getTargetShaderCache(device).createComputePipeline({\n      label: this._label,\n      layout: 'auto',\n      compute: {\n        module: getTargetShaderCache(device).createShaderModule({\n          label: this._label,\n          code: csCode\n        }),\n        entryPoint: 'main'\n      }\n    });\n    this._updateBindGroups();\n  }\n\n  private _getFullCs(\n    inputParams: Record<string, WGPUComputePassInput>,\n    inputTypes: Record<string, 'texture' | 'buffer'>\n  ) {\n    const inputs = this._inputs;\n    const uniforms = this._uniforms;\n    let offset = 0;\n    const groupOffsets = (this._groupOffsets = {\n      inputs: 0,\n      uniforms: 0,\n      outputs: 0\n    });\n    const workgroupSize = this._getActualWorkgroupSize();\n\n    if (inputs.length > 0) {\n      offset++;\n    }\n    if (uniforms.length > 0) {\n      groupOffsets.uniforms = offset;\n      offset++;\n    }\n\n    function getTypeCode(params: WGPUComputePassOutput) {\n      const { type = 'f32', channels } = params;\n      let code = channels === 1 ? type : `vec${channels}${type[0]}`;\n      if (params.atomic) {\n        code = `atomic<${code}>`;\n      }\n      return code;\n    }\n    groupOffsets.outputs = offset;\n    const cs = /* wgsl */ `\n${inputs\n  .sort()\n  .map((inputName, idx) => {\n    const bindingPrefix = `@group(${groupOffsets.inputs}) @binding(${idx}) `;\n    const varName = `in_${inputName}`;\n    // TODO more channels option.\n    return inputTypes[inputName] === 'texture'\n      ? `${bindingPrefix} var ${varName}: texture_2d<f32>;`\n      : `${bindingPrefix} var<storage, read> ${varName}: array<${getTypeCode(\n          inputParams[inputName]\n        )}>;`;\n  })\n  .join('\\n')}\n${this._uniforms\n  .map(\n    (uniform, idx) =>\n      `@group(${groupOffsets.uniforms}) @binding(${idx}) var<uniform> ${uniform.label}: ${uniform.type};`\n  )\n  .join('\\n')}\n\n${this._outputs\n  .map((name, idx) => {\n    return `@group(${\n      groupOffsets.outputs\n    }) @binding(${idx}) var<storage, read_write> out_${name}: array<${getTypeCode(\n      this._outputParams[name] || {}\n    )}>;`;\n  })\n  .join('\\n')}\n${this._csDefine ?? ''}\n@compute @workgroup_size(${workgroupSize[0]}, ${workgroupSize[1]}, 1)\nfn main(@builtin(global_invocation_id) globalId: vec3u) {\n${this._csMain}\n}\n`;\n\n    return cs;\n  }\n\n  private _updateBindGroups() {\n    const bindGroups: GPUBindGroup[] = [];\n    const device = this._device;\n    const groupOffsets = this._groupOffsets;\n\n    if (this._uniforms.length > 0) {\n      bindGroups[groupOffsets.uniforms] = device.createBindGroup({\n        label: this._label + '.uniforms',\n        layout: this._pipeline.getBindGroupLayout(groupOffsets.uniforms),\n        entries: this._uniforms.map(\n          (uniform, idx) =>\n            ({\n              binding: idx,\n              resource: {\n                buffer: this._uniformBuffers[uniform.label]\n              }\n            } as GPUBindGroupEntry)\n        )\n      });\n    }\n\n    this._bindGroups = bindGroups;\n  }\n\n  createPass(\n    commandEncoder: GPUCommandEncoder,\n    inputs: Record<I, WGPUComputePassInput>,\n\n    /**\n     * Explict outputs. Used for buffers reuse.\n     */\n    outputs?: Record<O, GPUBuffer>\n  ) {\n    this._updateOutputBuffers(Object.keys(outputs || {}));\n    const outputBuffers = {\n      ...this._outputBuffers,\n      ...outputs\n    } as Record<string, GPUBuffer>;\n\n    const inputTypes = this._inputs.reduce((obj, inputName) => {\n      obj[inputName] = inputs[inputName as I].buffer ? 'buffer' : 'texture';\n      return obj;\n    }, {} as Record<string, 'texture' | 'buffer'>);\n\n    this._updatePipeline(inputs, inputTypes);\n\n    const groupOffsets = this._groupOffsets;\n    if (this._inputs.length > 0) {\n      this._bindGroups[groupOffsets.inputs] = this._device.createBindGroup({\n        label: this._label + '.inputs',\n        layout: this._pipeline.getBindGroupLayout(groupOffsets.inputs),\n        entries: this._inputs.map((inputName, idx) => ({\n          binding: idx,\n          resource: inputs[inputName as I].buffer\n            ? {\n                buffer: inputs[inputName as I].buffer!\n              }\n            : inputs[inputName as I].texture!.createView()\n        }))\n      });\n    }\n\n    // Outputs\n    this._bindGroups[groupOffsets.outputs] = this._device.createBindGroup({\n      label: this._label + '.outputs',\n      layout: this._pipeline.getBindGroupLayout(groupOffsets.outputs),\n      entries: this._outputs.map((bufferName, idx) => ({\n        binding: idx,\n        resource: {\n          buffer: outputBuffers[bufferName]\n        }\n      }))\n    });\n\n    // Begin the render pass\n    const computePass = commandEncoder.beginComputePass();\n\n    // Draw a full quad\n    computePass.setPipeline(this._pipeline);\n    // Bind groups\n    this._bindGroups.forEach((bindGroup, idx) => {\n      computePass.setBindGroup(idx, bindGroup);\n    });\n\n    const workgroupSize = this._getActualWorkgroupSize();\n    computePass.dispatchWorkgroups(\n      Math.ceil((this._execWidth ?? this._width) / workgroupSize[0]),\n      Math.ceil((this._execHeight ?? this._height) / workgroupSize[1]),\n      1\n    );\n    // End the render pass\n    computePass.end();\n  }\n}\n"]}

package/lib/renderer/WGPUFullQuadPass.d.ts

@@ -0,0 +1,73 @@
+export declare const fullScreenQuadVertexShaderWGSL = "\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";
+export declare function getUniformBuffer(value: any, type: Uniform['type']): Float32Array | Int32Array | Uint32Array;
+export interface Uniform {
+    label: string;
+    type: `array<vec4u, ${number}>` | 'f32' | 'i32' | 'u32' | 'vec3f' | 'vec4f' | 'vec2f' | 'vec2i' | 'vec3i' | 'vec4i' | 'vec2u' | 'vec3u' | 'vec4u';
+    data: Float32Array | Int32Array | Uint32Array | number | number[];
+}
+export interface WGPUFullQuadPassOutput {
+    format: GPUTextureFormat;
+}
+export declare class WGPUFullQuadPass<I extends string, O extends string> {
+    private _label;
+    private _device;
+    private _outputTextures;
+    private _outputParams;
+    private _pipeline;
+    private _bindGroups;
+    private _needsUpdatePipeline;
+    private _needsUpdateOutputTextures;
+    /**
+     * When render to canvas
+     */
+    private _renderToScreen?;
+    private _inputs;
+    private _inputViewDimensions;
+    private _outputs;
+    private _uniforms;
+    private _uniformsMap;
+    private _uniformBuffers;
+    private _samplers;
+    private _capacityWidth;
+    private _capacityHeight;
+    private _reserveCapacity;
+    private _regionNames;
+    private _width;
+    private _height;
+    private _fsCode;
+    private readonly _fsMain;
+    private readonly _fsDefine;
+    private _textureWillBeRead;
+    constructor(label: string, device: GPUDevice, opts: {
+        inputs: I[];
+        inputViewDimensions?: Partial<Record<I, GPUTextureViewDimension>>;
+        outputs: O[];
+        samplers?: Record<string, GPUSampler>;
+        fsMain: string;
+        fsDefine?: string;
+        uniforms?: Uniform[];
+        textureWillBeRead?: boolean;
+    });
+    /** Keep physical outputs large enough for every render while varying the logical viewport. */
+    reserveCapacity(): void;
+    setSize(width: number, height: number): void;
+    setOutputParams(outputParams: Record<O, WGPUFullQuadPassOutput>): void;
+    setRenderToScreen(screenTexture: GPUTexture, presentationFormat: GPUTextureFormat): void;
+    setUniform(label: string, data: Uniform['data']): void;
+    private readonly _retainedOutputs;
+    retainOutput(texture: GPUTexture): void;
+    releaseOutput(texture: GPUTexture): void;
+    getOutput(name: O): GPUTexture;
+    getMemoryBytes(): number;
+    dispose(): void;
+    private _createTexture;
+    private _updateOutputTextures;
+    private _updatePipeline;
+    private _getFullFs;
+    private _updateBindGroups;
+    createPass(commandEncoder: GPUCommandEncoder, inputs: Record<I, GPUTexture>, 
+    /**
+     * Explict outputs. Used for buffers reuse.
+     */
+    outputs?: Record<O, GPUTexture>): void;
+}