oidn-web 0.1.0

package/lib/UNet.d.ts

@@ -0,0 +1,54 @@
+/// <reference types="@webgpu/types/dist" />
+import { HostTensor } from './tza';
+import { Tile } from './process';
+import type { WebGPUBackend } from '@tensorflow/tfjs-backend-webgpu';
+interface HDRImageData {
+    data: Float32Array;
+    width: number;
+    height: number;
+}
+interface GPUImageData {
+    data: GPUBuffer;
+    width: number;
+    height: number;
+}
+declare class UNet {
+    private _tensors;
+    private _backend;
+    private _tfModel;
+    private _device;
+    private _tileWidth;
+    private _tileHeight;
+    private _tileOverlapX;
+    private _tileOverlapY;
+    private _aux;
+    private _hdr;
+    private _dataProcessGPU?;
+    private _maxTileSize;
+    constructor(_tensors: Map<string, HostTensor>, _backend: WebGPUBackend, opts?: {
+        aux?: boolean;
+        hdr?: boolean;
+        maxTileSize?: number;
+    });
+    private _createConv;
+    private _createConcatConv;
+    private _createPooling;
+    private _addUpsamplingLayer;
+    getDevice(): GPUDevice | undefined;
+    buildModel(): void;
+    private _updateModel;
+    private _getTileSizeWithOverlap;
+    private _processImageData;
+    private _readTile;
+    private _writeTile;
+    private _executeTile;
+    progressiveExecute<T extends ImageData | HDRImageData | GPUImageData>({ color, albedo, normal, done, progress }: {
+        color: T;
+        albedo?: ImageData | GPUImageData;
+        normal?: ImageData | GPUImageData;
+        done: (outputData: T) => void;
+        progress?: (outputData: T, tileData: T | undefined, tile: Tile, currentIdx: number, totalIdx: number) => void;
+    }): () => void;
+    dispose(): void;
+}
+export default UNet;

package/lib/UNet.js

@@ -0,0 +1,467 @@
+import { tensor } from '@tensorflow/tfjs-core/dist/ops/tensor';
+import { tensor1d } from '@tensorflow/tfjs-core/dist/ops/tensor1d';
+import { mirrorPad } from '@tensorflow/tfjs-core/dist/ops/mirror_pad';
+import { pad4d } from '@tensorflow/tfjs-core/dist/ops/pad4d';
+import { slice4d } from '@tensorflow/tfjs-core/dist/ops/slice4d';
+import { concat4d } from '@tensorflow/tfjs-core/dist/ops/concat_4d';
+import { Conv2D, UpSampling2D } from '@tensorflow/tfjs-layers/dist/layers/convolutional';
+import { MaxPooling2D } from '@tensorflow/tfjs-layers/dist/layers/pooling';
+import { Concatenate } from '@tensorflow/tfjs-layers/dist/layers/merge';
+import { LayersModel } from '@tensorflow/tfjs-layers/dist/engine/training';
+import { Input as TFInput } from '@tensorflow/tfjs-layers/dist/engine/input_layer';
+import { Float16Array } from '@petamoriken/float16';
+import { GPUDataProcess, Tile, avgLogLum, hdrTransferFuncCPU, hdrTransferFuncInverseCPU } from './process';
+// import { profileAndLogKernelCode } from './helper';
+function getTensorData(ubytes, type) {
+    const buffer = ubytes.buffer;
+    if (type === 'Float32') {
+        return new Float32Array(ubytes.buffer);
+    }
+    const float16Data = new Float16Array(buffer);
+    const float32Data = new Float32Array(float16Data.length);
+    for (let i = 0; i < float32Data.length; ++i) {
+        float32Data[i] = float16Data[i];
+    }
+    return float32Data;
+}
+function changeWeightShapes(weightData, dims) {
+    const [O, C, H, W] = dims;
+    const reorderedWeightData = new Float32Array(weightData.length);
+    for (let o = 0; o < O; ++o) {
+        for (let c = 0; c < C; ++c) {
+            for (let h = 0; h < H; ++h) {
+                for (let w = 0; w < W; ++w) {
+                    // Change OCHW to HWCO
+                    const idx = o * C * H * W + c * H * W + h * W + w;
+                    const idx2 = h * W * C * O + w * C * O + c * O + o;
+                    reorderedWeightData[idx2] = weightData[idx];
+                }
+            }
+        }
+    }
+    return reorderedWeightData;
+}
+function roundUp(a, b) {
+    return Math.ceil(a / b) * b;
+}
+// Returns the smallest integer larger than or equal to a which has remainder c when divided by b
+function roundUp2(a, b, c) {
+    return Math.ceil((a - c) / b) * b + c;
+}
+function isGPUImageData(data) {
+    return data.data instanceof GPUBuffer;
+}
+const receptiveField = 174; // receptive field in pixels
+// TODO metal is 32?
+const minTileAlignment = 1;
+const tileAlignment = 16; // required spatial alignment in pixels (padding may be necessary)
+const defaultTileOverlap = roundUp(receptiveField / 2, tileAlignment);
+class UNet {
+    _tensors;
+    _backend;
+    _tfModel;
+    _device;
+    // TODO calculate the tile size from memory size
+    // https://github.com/RenderKit/oidn/blob/713ec7838ba650f99e0a896549c0dca5eeb3652d/core/unet_filter.cpp#L287
+    _tileWidth = 0;
+    _tileHeight = 0;
+    _tileOverlapX = 0;
+    _tileOverlapY = 0;
+    _aux = false;
+    _hdr = false;
+    _dataProcessGPU;
+    _maxTileSize;
+    constructor(_tensors, _backend, opts = {}) {
+        this._tensors = _tensors;
+        this._backend = _backend;
+        this._aux = opts.aux || false;
+        this._hdr = opts.hdr || false;
+        this._maxTileSize = opts.maxTileSize ?? 512;
+        this._device = this._backend.device;
+    }
+    _createConv(name, source, activation) {
+        const unetWeightTensor = this._tensors.get(name + '.weight');
+        const unetBiasTensor = this._tensors.get(name + '.bias');
+        const weightDims = unetWeightTensor.desc.dims;
+        const weightTensor = tensor(changeWeightShapes(getTensorData(unetWeightTensor.data, unetWeightTensor.desc.dataType), weightDims), [weightDims[2], weightDims[3], weightDims[1], weightDims[0]], 'float32');
+        const biasTensor = tensor1d(getTensorData(unetBiasTensor.data, unetBiasTensor.desc.dataType), 'float32');
+        // TODO whats the purpose of padded dims ?
+        const convLayer = new Conv2D({
+            name,
+            filters: unetWeightTensor.desc.dims[0],
+            kernelSize: unetWeightTensor.desc.dims.slice(2, 4),
+            useBias: true,
+            activation,
+            padding: 'same',
+            weights: [weightTensor, biasTensor],
+            trainable: false
+        });
+        return convLayer.apply(source);
+    }
+    _createConcatConv(name, source1, source2) {
+        //https://github.com/RenderKit/oidn/blob/713ec7838ba650f99e0a896549c0dca5eeb3652d/training/model.py#L40
+        return this._createConv(name, 
+        // Concat on the channel
+        new Concatenate({ trainable: false, axis: 3 }).apply([
+            // convLayer.apply(source2) as SymbolicTensor,
+            source1,
+            source2
+        ]), 'relu');
+    }
+    _createPooling(source) {
+        // https://github.com/RenderKit/oidn/blob/713ec7838ba650f99e0a896549c0dca5eeb3652d/training/model.py#L33
+        return new MaxPooling2D({
+            name: source.name + '/pooling',
+            poolSize: [2, 2],
+            strides: [2, 2],
+            padding: 'same',
+            trainable: false
+        }).apply(source);
+    }
+    _addUpsamplingLayer(source) {
+        return new UpSampling2D({
+            name: source.name + '/upsampling',
+            size: [2, 2],
+            trainable: false
+        }).apply(source);
+    }
+    getDevice() {
+        return this._device;
+    }
+    buildModel() {
+        const aux = this._aux;
+        const channels = 3 + (aux ? 6 : 0);
+        const tileSize = this._getTileSizeWithOverlap();
+        // TODO input process transferFunc
+        // TODO input shape
+        const input = TFInput({
+            shape: [tileSize.height, tileSize.width, channels],
+            dtype: 'float32'
+        });
+        const encConv0 = this._createConv('enc_conv0', input, 'relu');
+        const pool1 = this._createPooling(this._createConv('enc_conv1', encConv0, 'relu'));
+        const pool2 = this._createPooling(this._createConv('enc_conv2', pool1, 'relu'));
+        const pool3 = this._createPooling(this._createConv('enc_conv3', pool2, 'relu'));
+        const pool4 = this._createPooling(this._createConv('enc_conv4', pool3, 'relu'));
+        const encConv5a = this._createConv('enc_conv5a', pool4, 'relu');
+        const upsample4 = this._addUpsamplingLayer(this._createConv('enc_conv5b', encConv5a, 'relu'));
+        const decConv4a = this._createConcatConv('dec_conv4a', upsample4, pool3);
+        const upsample3 = this._addUpsamplingLayer(this._createConv('dec_conv4b', decConv4a, 'relu'));
+        const decConv3a = this._createConcatConv('dec_conv3a', upsample3, pool2);
+        const upsample2 = this._addUpsamplingLayer(this._createConv('dec_conv3b', decConv3a, 'relu'));
+        const decConv2a = this._createConcatConv('dec_conv2a', upsample2, pool1);
+        const upsample1 = this._addUpsamplingLayer(this._createConv('dec_conv2b', decConv2a, 'relu'));
+        const decConv1a = this._createConcatConv('dec_conv1a', upsample1, input);
+        const decConv1b = this._createConv('dec_conv1b', decConv1a, 'relu');
+        const decConv0 = this._createConv('dec_conv0', decConv1b, 'relu');
+        this._tfModel = new LayersModel({
+            inputs: [input],
+            // TODO output process transferFunc
+            outputs: decConv0
+        });
+    }
+    _updateModel(width, height) {
+        const maxTileSize = this._maxTileSize;
+        let tileWidth = maxTileSize;
+        let tileHeight = maxTileSize;
+        let tileOverlapX = defaultTileOverlap;
+        let tileOverlapY = defaultTileOverlap;
+        if (width < maxTileSize + defaultTileOverlap * 2) {
+            tileWidth = roundUp(width, tileAlignment);
+            tileOverlapX = 0;
+        }
+        if (height < maxTileSize + defaultTileOverlap * 2) {
+            tileHeight = roundUp(height, tileAlignment);
+            tileOverlapY = 0;
+        }
+        if (tileWidth !== this._tileWidth ||
+            tileHeight !== this._tileHeight ||
+            tileOverlapX !== this._tileOverlapX ||
+            tileOverlapY !== this._tileOverlapY ||
+            !this._tfModel) {
+            this._tileWidth = tileWidth;
+            this._tileHeight = tileHeight;
+            this._tileOverlapX = tileOverlapX;
+            this._tileOverlapY = tileOverlapY;
+            if (this._tfModel) {
+                this._tfModel.dispose();
+            }
+            this.buildModel();
+        }
+    }
+    _getTileSizeWithOverlap() {
+        return {
+            width: this._tileWidth + 2 * this._tileOverlapX,
+            height: this._tileHeight + 2 * this._tileOverlapY
+        };
+    }
+    _processImageData(color, albedo, normal, isHDR) {
+        const rawData = color.data;
+        const pixelsCount = rawData.length / 4;
+        const channels = this._aux ? 9 : 3;
+        const tensorData = new Float32Array(pixelsCount * channels);
+        if ((albedo && !normal) || (normal && !albedo)) {
+            throw new Error('Normal map and albedo map are both required');
+        }
+        if (albedo && normal) {
+            if (albedo.width !== normal.width ||
+                albedo.height !== normal.height ||
+                color.width !== albedo.width ||
+                color.height !== albedo.height) {
+                throw new Error('Image size mismatch');
+            }
+        }
+        const albedoData = albedo?.data;
+        const normalData = normal?.data;
+        for (let i = 0; i < rawData.length; i += 4) {
+            const i2 = (i / 4) * channels;
+            for (let c = 0; c < 3; c++) {
+                if (isHDR) {
+                    tensorData[i2 + c] = rawData[i + c];
+                }
+                else {
+                    tensorData[i2 + c] = rawData[i + c] / 255;
+                }
+                if (albedoData) {
+                    tensorData[i2 + c + 3] = albedoData[i + c] / 255;
+                }
+                if (normalData) {
+                    tensorData[i2 + c + 6] = normalData[i + c] / 255;
+                }
+            }
+        }
+        return tensorData;
+    }
+    _readTile(data, channels, srcTile, width) {
+        const tileData = new Float32Array(srcTile.width * srcTile.height * channels);
+        for (let y = 0; y < srcTile.height; y++) {
+            for (let x = 0; x < srcTile.width; x++) {
+                const i2 = ((y + srcTile.y) * width + (x + srcTile.x)) * channels;
+                const i1 = (y * srcTile.width + x) * channels;
+                for (let c = 0; c < channels; c++) {
+                    tileData[i1 + c] = data[i2 + c];
+                }
+            }
+        }
+        return tileData;
+    }
+    _writeTile(imageData, srcTile, dstTile, srcTileData, srcWidth, isHDR) {
+        const { data: outImageData, width } = imageData;
+        const dx = dstTile.x - srcTile.x;
+        const dy = dstTile.y - srcTile.y;
+        for (let y = 0; y < dstTile.height; y++) {
+            for (let x = 0; x < dstTile.width; x++) {
+                const i1 = ((y + dy) * srcWidth + x + dx) * 3;
+                const i2 = ((y + dstTile.y) * width + (x + dstTile.x)) * 4;
+                for (let c = 0; c < 3; c++) {
+                    if (isHDR) {
+                        outImageData[i2 + c] = srcTileData[i1 + c];
+                    }
+                    else {
+                        outImageData[i2 + c] = Math.min(Math.max(srcTileData[i1 + c] * 255, 0), 255);
+                    }
+                }
+                imageData.data[i2 + 3] = isHDR ? 1 : 255;
+            }
+        }
+    }
+    _executeTile(inputData, outputTileData, outputImageData, i, j, width, height, isHDR) {
+        const channels = this._aux ? 9 : 3;
+        const tileOverlapX = this._tileOverlapX;
+        const tileOverlapY = this._tileOverlapY;
+        let srcTileSize = this._getTileSizeWithOverlap();
+        let dstTileSize = { width: this._tileWidth, height: this._tileHeight };
+        let srcX0 = i > 0 ? i * dstTileSize.width - tileOverlapX : 0;
+        let srcX1 = Math.min(srcX0 + srcTileSize.width, width);
+        srcX0 = Math.max(srcX1 - srcTileSize.width, 0);
+        let srcY0 = j > 0 ? j * dstTileSize.height - tileOverlapY : 0;
+        let srcY1 = Math.min(srcY0 + srcTileSize.height, height);
+        srcY0 = Math.max(srcY1 - srcTileSize.height, 0);
+        const srcTileWidth = Math.min(srcTileSize.width, width);
+        const srcTileHeight = Math.min(srcTileSize.height, height);
+        const needsResize = width < dstTileSize.width || height < dstTileSize.height;
+        const srcTile = new Tile(srcX0, srcY0, srcTileWidth, srcTileHeight);
+        let tileTensor;
+        let inputScale = 1;
+        const device = this._device;
+        let dataProcessGPU = this._dataProcessGPU;
+        if (inputData instanceof Float32Array) {
+            let tileData = this._readTile(inputData, channels, srcTile, width);
+            if (isHDR) {
+                inputScale = avgLogLum({
+                    data: tileData,
+                    channels: 9
+                });
+                tileData = hdrTransferFuncCPU({
+                    data: tileData,
+                    channels: 9,
+                    inputScale
+                });
+            }
+            tileTensor = tensor(tileData, [1, srcTileHeight, srcTileWidth, channels], 'float32');
+        }
+        else {
+            if (!isHDR) {
+                throw new Error('Only hdr is supported for webgpu data.');
+            }
+            if (!dataProcessGPU) {
+                dataProcessGPU = this._dataProcessGPU = new GPUDataProcess(device);
+            }
+            dataProcessGPU.setImageSize(width, height);
+            dataProcessGPU.setInputTile(srcTile);
+            // Display the noisy input instead of prev denoised result
+            if (i === 0 && j === 0) {
+                dataProcessGPU.copyInputDataToOutput(inputData.color);
+            }
+            const { color, albedo, normal } = dataProcessGPU.forward(inputData.color, inputData.albedo, inputData.normal);
+            const shape = [1, srcTileHeight, srcTileWidth, 4];
+            tileTensor = concat4d([color, albedo, normal].map((buffer) => {
+                const tmp = tensor({ buffer, zeroCopy: true }, shape);
+                const ret = slice4d(tmp, [0, 0, 0, 0], [1, srcTileHeight, srcTileWidth, 3]);
+                tmp.dispose();
+                return ret;
+            }), 3);
+        }
+        // We need resize if input size is smaller than tile size. And is rounded up.
+        if (needsResize) {
+            const rawTileTensor = tileTensor;
+            tileTensor = mirrorPad(rawTileTensor, [
+                [0, 0],
+                [0, srcTileSize.height - height],
+                [0, srcTileSize.width - width],
+                [0, 0]
+            ], 'reflect');
+            rawTileTensor.dispose();
+        }
+        const outputTensor = this._tfModel.predict(tileTensor);
+        tileTensor.dispose();
+        const dstWidth = Math.min(dstTileSize.width, width);
+        const dstHeight = Math.min(dstTileSize.height, height);
+        const dstTile = new Tile(i * dstWidth, j * dstHeight, dstWidth, dstHeight);
+        dstTile.width = Math.min(dstTile.width, width - dstTile.x);
+        dstTile.height = Math.min(dstTile.height, height - dstTile.y);
+        if (inputData instanceof Float32Array) {
+            let denoisedData = outputTensor.dataSync();
+            if (isHDR) {
+                denoisedData = hdrTransferFuncInverseCPU({
+                    data: denoisedData,
+                    channels: 3,
+                    inputScale
+                });
+            }
+            this._writeTile(outputImageData, srcTile, dstTile, denoisedData, srcTileSize.width, isHDR);
+            for (let y = 0; y < dstHeight; y++) {
+                for (let x = 0; x < dstWidth; x++) {
+                    const i1 = (y * dstWidth + x) * 4;
+                    const i2 = ((y + dstTile.y) * width + (x + dstTile.x)) * 4;
+                    for (let c = 0; c < 4; c++) {
+                        outputTileData.data[i1 + c] = outputImageData.data[i2 + c];
+                    }
+                }
+            }
+            outputTensor.dispose();
+        }
+        else {
+            dataProcessGPU.setOutputTile(dstTile, srcTile);
+            // IMPORTANT
+            // storage buffer has alignment. that 3 channels still needs 16 bytes data.
+            // So we need to pad it to 4 channels.
+            const outputTensor4Channnels = pad4d(outputTensor, [
+                [0, 0],
+                [0, 0],
+                [0, 0],
+                [0, 1]
+            ]);
+            const outBuffer = dataProcessGPU.inverse(outputTensor4Channnels.dataToGPU().buffer, inputData.color);
+            outputTensor.dispose();
+            outputTensor4Channnels.dispose();
+            return outBuffer;
+        }
+    }
+    progressiveExecute({ color, albedo, normal, done, progress }) {
+        if (this._aux && (!albedo || !normal)) {
+            throw new Error('Normal map and albedo map are both required');
+        }
+        if (!this._aux) {
+            if (albedo || normal) {
+                throw new Error('Normal map and albedo map are not required');
+            }
+        }
+        const width = color.width;
+        const height = color.height;
+        this._updateModel(width, height);
+        // TODO should fixed to be hdr when UNet is created.
+        // weights of hdr and ldr is different
+        const hdr = this._hdr || false;
+        let rawData;
+        if (!isGPUImageData(color)) {
+            rawData = this._processImageData(color, albedo, normal, hdr);
+        }
+        const tileWidth = this._tileWidth;
+        const tileHeight = this._tileHeight;
+        const tileCountH = Math.ceil(height / tileHeight);
+        const tileCountW = Math.ceil(width / tileWidth);
+        function makeImageData(width, height) {
+            return hdr
+                ? {
+                    data: new Float32Array(width * height * 4),
+                    width,
+                    height
+                }
+                : new ImageData(width, height);
+        }
+        const outputImageData = isGPUImageData(color)
+            ? undefined
+            : makeImageData(width, height);
+        const outputTileData = isGPUImageData(color)
+            ? undefined
+            : makeImageData(Math.min(tileWidth, width), Math.min(tileHeight, height));
+        let aborted = false;
+        const executeTile = (i, j) => {
+            if (aborted) {
+                return;
+            }
+            let resGPUBuffer;
+            // profileAndLogKernelCode(() => {
+            resGPUBuffer = this._executeTile(isGPUImageData(color)
+                ? {
+                    color: color.data,
+                    albedo: albedo.data,
+                    normal: normal.data
+                }
+                : rawData, outputTileData, outputImageData, i, j, width, height, hdr);
+            // }, true);
+            const output = outputImageData || {
+                data: resGPUBuffer,
+                width,
+                height
+            };
+            progress?.(output, 
+            // Is undefined if using webgpu buffer
+            outputTileData, new Tile(i * tileWidth, j * tileHeight, tileWidth, tileHeight), i + j * tileCountW, tileCountW * tileCountH);
+            if (i + 1 < tileCountW || j + 1 < tileCountH) {
+                requestAnimationFrame(() => {
+                    if (i + 1 < tileCountW) {
+                        executeTile(i + 1, j);
+                    }
+                    else if (j + 1 < tileCountH) {
+                        executeTile(0, j + 1);
+                    }
+                });
+            }
+            else {
+                done(output);
+            }
+        };
+        executeTile(0, 0);
+        return () => {
+            aborted = true;
+        };
+    }
+    dispose() {
+        this._tfModel?.dispose();
+        this._dataProcessGPU?.dispose();
+    }
+}
+export default UNet;
+//# sourceMappingURL=UNet.js.map

package/lib/UNet.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"UNet.js","sourceRoot":"","sources":["../src/UNet.ts"],"names":[],"mappings":"AAGA,OAAO,EAAE,MAAM,EAAE,MAAM,uCAAuC,CAAC;AAC/D,OAAO,EAAE,QAAQ,EAAE,MAAM,yCAAyC,CAAC;AACnE,OAAO,EAAE,SAAS,EAAE,MAAM,2CAA2C,CAAC;AACtE,OAAO,EAAE,KAAK,EAAE,MAAM,sCAAsC,CAAC;AAC7D,OAAO,EAAE,OAAO,EAAE,MAAM,wCAAwC,CAAC;AACjE,OAAO,EAAE,QAAQ,EAAE,MAAM,0CAA0C,CAAC;AACpE,OAAO,EACL,MAAM,EACN,YAAY,EACb,MAAM,mDAAmD,CAAC;AAC3D,OAAO,EAAE,YAAY,EAAE,MAAM,6CAA6C,CAAC;AAC3E,OAAO,EAAE,WAAW,EAAE,MAAM,2CAA2C,CAAC;AACxE,OAAO,EAAE,WAAW,EAAE,MAAM,8CAA8C,CAAC;AAC3E,OAAO,EAAE,KAAK,IAAI,OAAO,EAAE,MAAM,iDAAiD,CAAC;AAEnF,OAAO,EAAE,YAAY,EAAE,MAAM,sBAAsB,CAAC;AACpD,OAAO,EACL,cAAc,EACd,IAAI,EACJ,SAAS,EACT,kBAAkB,EAClB,yBAAyB,EAC1B,MAAM,WAAW,CAAC;AAEnB,sDAAsD;AAEtD,SAAS,aAAa,CACpB,MAAkB,EAClB,IAAoC;IAEpC,MAAM,MAAM,GAAG,MAAM,CAAC,MAAM,CAAC;IAC7B,IAAI,IAAI,KAAK,SAAS,EAAE,CAAC;QACvB,OAAO,IAAI,YAAY,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC;IACzC,CAAC;IACD,MAAM,WAAW,GAAG,IAAI,YAAY,CAAC,MAAM,CAAC,CAAC;IAC7C,MAAM,WAAW,GAAG,IAAI,YAAY,CAAC,WAAW,CAAC,MAAM,CAAC,CAAC;IACzD,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,WAAW,CAAC,MAAM,EAAE,EAAE,CAAC,EAAE,CAAC;QAC5C,WAAW,CAAC,CAAC,CAAC,GAAG,WAAW,CAAC,CAAC,CAAC,CAAC;IAClC,CAAC;IACD,OAAO,WAAW,CAAC;AACrB,CAAC;AAED,SAAS,kBAAkB,CAAC,UAAwB,EAAE,IAAc;IAClE,MAAM,CAAC,CAAC,EAAE,CAAC,EAAE,CAAC,EAAE,CAAC,CAAC,GAAG,IAAI,CAAC;IAC1B,MAAM,mBAAmB,GAAG,IAAI,YAAY,CAAC,UAAU,CAAC,MAAM,CAAC,CAAC;IAChE,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,CAAC,EAAE,EAAE,CAAC,EAAE,CAAC;QAC3B,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,CAAC,EAAE,EAAE,CAAC,EAAE,CAAC;YAC3B,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,CAAC,EAAE,EAAE,CAAC,EAAE,CAAC;gBAC3B,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,CAAC,EAAE,EAAE,CAAC,EAAE,CAAC;oBAC3B,sBAAsB;oBACtB,MAAM,GAAG,GAAG,CAAC,GAAG,CAAC,GAAG,CAAC,GAAG,CAAC,GAAG,CAAC,GAAG,CAAC,GAAG,CAAC,GAAG,CAAC,GAAG,CAAC,GAAG,CAAC,CAAC;oBAClD,MAAM,IAAI,GAAG,CAAC,GAAG,CAAC,GAAG,CAAC,GAAG,CAAC,GAAG,CAAC,GAAG,CAAC,GAAG,CAAC,GAAG,CAAC,GAAG,CAAC,GAAG,CAAC,CAAC;oBACnD,mBAAmB,CAAC,IAAI,CAAC,GAAG,UAAU,CAAC,GAAG,CAAC,CAAC;gBAC9C,CAAC;YACH,CAAC;QACH,CAAC;IACH,CAAC;IACD,OAAO,mBAAmB,CAAC;AAC7B,CAAC;AAcD,SAAS,OAAO,CAAC,CAAS,EAAE,CAAS;IACnC,OAAO,IAAI,CAAC,IAAI,CAAC,CAAC,GAAG,CAAC,CAAC,GAAG,CAAC,CAAC;AAC9B,CAAC;AACD,iGAAiG;AACjG,SAAS,QAAQ,CAAC,CAAS,EAAE,CAAS,EAAE,CAAS;IAC/C,OAAO,IAAI,CAAC,IAAI,CAAC,CAAC,CAAC,GAAG,CAAC,CAAC,GAAG,CAAC,CAAC,GAAG,CAAC,GAAG,CAAC,CAAC;AACxC,CAAC;AAED,SAAS,cAAc,CACrB,IAA6C;IAE7C,OAAO,IAAI,CAAC,IAAI,YAAY,SAAS,CAAC;AACxC,CAAC;AAED,MAAM,cAAc,GAAG,GAAG,CAAC,CAAC,4BAA4B;AACxD,oBAAoB;AACpB,MAAM,gBAAgB,GAAG,CAAC,CAAC;AAE3B,MAAM,aAAa,GAAG,EAAE,CAAC,CAAC,kEAAkE;AAE5F,MAAM,kBAAkB,GAAG,OAAO,CAAC,cAAc,GAAG,CAAC,EAAE,aAAa,CAAC,CAAC;AACtE,MAAM,IAAI;IAoBE;IACA;IApBF,QAAQ,CAA0B;IAClC,OAAO,CAAwB;IAEvC,gDAAgD;IAChD,4GAA4G;IACpG,UAAU,GAAG,CAAC,CAAC;IACf,WAAW,GAAG,CAAC,CAAC;IAEhB,aAAa,GAAG,CAAC,CAAC;IAClB,aAAa,GAAG,CAAC,CAAC;IAElB,IAAI,GAAG,KAAK,CAAC;IACb,IAAI,GAAG,KAAK,CAAC;IAEb,eAAe,CAAkB;IAEjC,YAAY,CAAC;IAErB,YACU,QAAiC,EACjC,QAAuB,EAC/B,OAII,EAAE;QANE,aAAQ,GAAR,QAAQ,CAAyB;QACjC,aAAQ,GAAR,QAAQ,CAAe;QAO/B,IAAI,CAAC,IAAI,GAAG,IAAI,CAAC,GAAG,IAAI,KAAK,CAAC;QAC9B,IAAI,CAAC,IAAI,GAAG,IAAI,CAAC,GAAG,IAAI,KAAK,CAAC;QAE9B,IAAI,CAAC,YAAY,GAAG,IAAI,CAAC,WAAW,IAAI,GAAG,CAAC;QAE5C,IAAI,CAAC,OAAO,GAAG,IAAI,CAAC,QAAQ,CAAC,MAAM,CAAC;IACtC,CAAC;IAEO,WAAW,CACjB,IAAY,EACZ,MAAsB,EACtB,UAAmB;QAEnB,MAAM,gBAAgB,GAAG,IAAI,CAAC,QAAQ,CAAC,GAAG,CAAC,IAAI,GAAG,SAAS,CAAE,CAAC;QAC9D,MAAM,cAAc,GAAG,IAAI,CAAC,QAAQ,CAAC,GAAG,CAAC,IAAI,GAAG,OAAO,CAAE,CAAC;QAC1D,MAAM,UAAU,GAAG,gBAAgB,CAAC,IAAI,CAAC,IAAI,CAAC;QAC9C,MAAM,YAAY,GAAG,MAAM,CACzB,kBAAkB,CAChB,aAAa,CAAC,gBAAgB,CAAC,IAAI,EAAE,gBAAgB,CAAC,IAAI,CAAC,QAAQ,CAAC,EACpE,UAAU,CACX,EACD,CAAC,UAAU,CAAC,CAAC,CAAC,EAAE,UAAU,CAAC,CAAC,CAAC,EAAE,UAAU,CAAC,CAAC,CAAC,EAAE,UAAU,CAAC,CAAC,CAAC,CAAC,EAC5D,SAAS,CACV,CAAC;QACF,MAAM,UAAU,GAAG,QAAQ,CACzB,aAAa,CAAC,cAAc,CAAC,IAAI,EAAE,cAAc,CAAC,IAAI,CAAC,QAAQ,CAAC,EAChE,SAAS,CACV,CAAC;QACF,0CAA0C;QAC1C,MAAM,SAAS,GAAG,IAAI,MAAM,CAAC;YAC3B,IAAI;YACJ,OAAO,EAAE,gBAAgB,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC,CAAC;YACtC,UAAU,EAAE,gBAAgB,CAAC,IAAI,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC,EAAE,CAAC,CAAqB;YACtE,OAAO,EAAE,IAAI;YACb,UAAU;YACV,OAAO,EAAE,MAAM;YACf,OAAO,EAAE,CAAC,YAAY,EAAE,UAAU,CAAC;YACnC,SAAS,EAAE,KAAK;SACjB,CAAC,CAAC;QAEH,OAAO,SAAS,CAAC,KAAK,CAAC,MAAM,CAAmB,CAAC;IACnD,CAAC;IAEO,iBAAiB,CACvB,IAAY,EACZ,OAAuB,EACvB,OAAuB;QAEvB,uGAAuG;QACvG,OAAO,IAAI,CAAC,WAAW,CACrB,IAAI;QACJ,wBAAwB;QACxB,IAAI,WAAW,CAAC,EAAE,SAAS,EAAE,KAAK,EAAE,IAAI,EAAE,CAAC,EAAE,CAAC,CAAC,KAAK,CAAC;YACnD,8CAA8C;YAC9C,OAAO;YACP,OAAO;SACR,CAAmB,EACpB,MAAM,CACW,CAAC;IACtB,CAAC;IAEO,cAAc,CAAC,MAAsB;QAC3C,wGAAwG;QACxG,OAAO,IAAI,YAAY,CAAC;YACtB,IAAI,EAAE,MAAM,CAAC,IAAI,GAAG,UAAU;YAC9B,QAAQ,EAAE,CAAC,CAAC,EAAE,CAAC,CAAC;YAChB,OAAO,EAAE,CAAC,CAAC,EAAE,CAAC,CAAC;YACf,OAAO,EAAE,MAAM;YACf,SAAS,EAAE,KAAK;SACjB,CAAC,CAAC,KAAK,CAAC,MAAM,CAAmB,CAAC;IACrC,CAAC;IAEO,mBAAmB,CAAC,MAAsB;QAChD,OAAO,IAAI,YAAY,CAAC;YACtB,IAAI,EAAE,MAAM,CAAC,IAAI,GAAG,aAAa;YACjC,IAAI,EAAE,CAAC,CAAC,EAAE,CAAC,CAAC;YACZ,SAAS,EAAE,KAAK;SACjB,CAAC,CAAC,KAAK,CAAC,MAAM,CAAmB,CAAC;IACrC,CAAC;IAED,SAAS;QACP,OAAO,IAAI,CAAC,OAAO,CAAC;IACtB,CAAC;IAED,UAAU;QACR,MAAM,GAAG,GAAG,IAAI,CAAC,IAAI,CAAC;QACtB,MAAM,QAAQ,GAAG,CAAC,GAAG,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC;QACnC,MAAM,QAAQ,GAAG,IAAI,CAAC,uBAAuB,EAAE,CAAC;QAEhD,kCAAkC;QAClC,mBAAmB;QACnB,MAAM,KAAK,GAAG,OAAO,CAAC;YACpB,KAAK,EAAE,CAAC,QAAQ,CAAC,MAAM,EAAE,QAAQ,CAAC,KAAK,EAAE,QAAQ,CAAC;YAClD,KAAK,EAAE,SAAS;SACjB,CAAC,CAAC;QAEH,MAAM,QAAQ,GAAG,IAAI,CAAC,WAAW,CAAC,WAAW,EAAE,KAAK,EAAE,MAAM,CAAC,CAAC;QAC9D,MAAM,KAAK,GAAG,IAAI,CAAC,cAAc,CAC/B,IAAI,CAAC,WAAW,CAAC,WAAW,EAAE,QAAQ,EAAE,MAAM,CAAC,CAChD,CAAC;QACF,MAAM,KAAK,GAAG,IAAI,CAAC,cAAc,CAC/B,IAAI,CAAC,WAAW,CAAC,WAAW,EAAE,KAAK,EAAE,MAAM,CAAC,CAC7C,CAAC;QACF,MAAM,KAAK,GAAG,IAAI,CAAC,cAAc,CAC/B,IAAI,CAAC,WAAW,CAAC,WAAW,EAAE,KAAK,EAAE,MAAM,CAAC,CAC7C,CAAC;QACF,MAAM,KAAK,GAAG,IAAI,CAAC,cAAc,CAC/B,IAAI,CAAC,WAAW,CAAC,WAAW,EAAE,KAAK,EAAE,MAAM,CAAC,CAC7C,CAAC;QACF,MAAM,SAAS,GAAG,IAAI,CAAC,WAAW,CAAC,YAAY,EAAE,KAAK,EAAE,MAAM,CAAC,CAAC;QAChE,MAAM,SAAS,GAAG,IAAI,CAAC,mBAAmB,CACxC,IAAI,CAAC,WAAW,CAAC,YAAY,EAAE,SAAS,EAAE,MAAM,CAAC,CAClD,CAAC;QACF,MAAM,SAAS,GAAG,IAAI,CAAC,iBAAiB,CAAC,YAAY,EAAE,SAAS,EAAE,KAAK,CAAC,CAAC;QACzE,MAAM,SAAS,GAAG,IAAI,CAAC,mBAAmB,CACxC,IAAI,CAAC,WAAW,CAAC,YAAY,EAAE,SAAS,EAAE,MAAM,CAAC,CAClD,CAAC;QACF,MAAM,SAAS,GAAG,IAAI,CAAC,iBAAiB,CAAC,YAAY,EAAE,SAAS,EAAE,KAAK,CAAC,CAAC;QACzE,MAAM,SAAS,GAAG,IAAI,CAAC,mBAAmB,CACxC,IAAI,CAAC,WAAW,CAAC,YAAY,EAAE,SAAS,EAAE,MAAM,CAAC,CAClD,CAAC;QACF,MAAM,SAAS,GAAG,IAAI,CAAC,iBAAiB,CAAC,YAAY,EAAE,SAAS,EAAE,KAAK,CAAC,CAAC;QACzE,MAAM,SAAS,GAAG,IAAI,CAAC,mBAAmB,CACxC,IAAI,CAAC,WAAW,CAAC,YAAY,EAAE,SAAS,EAAE,MAAM,CAAC,CAClD,CAAC;QACF,MAAM,SAAS,GAAG,IAAI,CAAC,iBAAiB,CAAC,YAAY,EAAE,SAAS,EAAE,KAAK,CAAC,CAAC;QACzE,MAAM,SAAS,GAAG,IAAI,CAAC,WAAW,CAAC,YAAY,EAAE,SAAS,EAAE,MAAM,CAAC,CAAC;QACpE,MAAM,QAAQ,GAAG,IAAI,CAAC,WAAW,CAAC,WAAW,EAAE,SAAS,EAAE,MAAM,CAAC,CAAC;QAElE,IAAI,CAAC,QAAQ,GAAG,IAAI,WAAW,CAAC;YAC9B,MAAM,EAAE,CAAC,KAAK,CAAC;YACf,mCAAmC;YACnC,OAAO,EAAE,QAAQ;SAClB,CAAC,CAAC;IACL,CAAC;IAEO,YAAY,CAAC,KAAa,EAAE,MAAc;QAChD,MAAM,WAAW,GAAG,IAAI,CAAC,YAAY,CAAC;QACtC,IAAI,SAAS,GAAG,WAAW,CAAC;QAC5B,IAAI,UAAU,GAAG,WAAW,CAAC;QAC7B,IAAI,YAAY,GAAG,kBAAkB,CAAC;QACtC,IAAI,YAAY,GAAG,kBAAkB,CAAC;QAEtC,IAAI,KAAK,GAAG,WAAW,GAAG,kBAAkB,GAAG,CAAC,EAAE,CAAC;YACjD,SAAS,GAAG,OAAO,CAAC,KAAK,EAAE,aAAa,CAAC,CAAC;YAC1C,YAAY,GAAG,CAAC,CAAC;QACnB,CAAC;QACD,IAAI,MAAM,GAAG,WAAW,GAAG,kBAAkB,GAAG,CAAC,EAAE,CAAC;YAClD,UAAU,GAAG,OAAO,CAAC,MAAM,EAAE,aAAa,CAAC,CAAC;YAC5C,YAAY,GAAG,CAAC,CAAC;QACnB,CAAC;QAED,IACE,SAAS,KAAK,IAAI,CAAC,UAAU;YAC7B,UAAU,KAAK,IAAI,CAAC,WAAW;YAC/B,YAAY,KAAK,IAAI,CAAC,aAAa;YACnC,YAAY,KAAK,IAAI,CAAC,aAAa;YACnC,CAAC,IAAI,CAAC,QAAQ,EACd,CAAC;YACD,IAAI,CAAC,UAAU,GAAG,SAAS,CAAC;YAC5B,IAAI,CAAC,WAAW,GAAG,UAAU,CAAC;YAC9B,IAAI,CAAC,aAAa,GAAG,YAAY,CAAC;YAClC,IAAI,CAAC,aAAa,GAAG,YAAY,CAAC;YAElC,IAAI,IAAI,CAAC,QAAQ,EAAE,CAAC;gBAClB,IAAI,CAAC,QAAQ,CAAC,OAAO,EAAE,CAAC;YAC1B,CAAC;YAED,IAAI,CAAC,UAAU,EAAE,CAAC;QACpB,CAAC;IACH,CAAC;IAEO,uBAAuB;QAC7B,OAAO;YACL,KAAK,EAAE,IAAI,CAAC,UAAU,GAAG,CAAC,GAAG,IAAI,CAAC,aAAa;YAC/C,MAAM,EAAE,IAAI,CAAC,WAAW,GAAG,CAAC,GAAG,IAAI,CAAC,aAAa;SAClD,CAAC;IACJ,CAAC;IAEO,iBAAiB,CACvB,KAA+B,EAC/B,MAA6B,EAC7B,MAA6B,EAC7B,KAAc;QAEd,MAAM,OAAO,GAAG,KAAK,CAAC,IAAI,CAAC;QAC3B,MAAM,WAAW,GAAG,OAAO,CAAC,MAAM,GAAG,CAAC,CAAC;QACvC,MAAM,QAAQ,GAAG,IAAI,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC;QACnC,MAAM,UAAU,GAAG,IAAI,YAAY,CAAC,WAAW,GAAG,QAAQ,CAAC,CAAC;QAE5D,IAAI,CAAC,MAAM,IAAI,CAAC,MAAM,CAAC,IAAI,CAAC,MAAM,IAAI,CAAC,MAAM,CAAC,EAAE,CAAC;YAC/C,MAAM,IAAI,KAAK,CAAC,6CAA6C,CAAC,CAAC;QACjE,CAAC;QACD,IAAI,MAAM,IAAI,MAAM,EAAE,CAAC;YACrB,IACE,MAAM,CAAC,KAAK,KAAK,MAAM,CAAC,KAAK;gBAC7B,MAAM,CAAC,MAAM,KAAK,MAAM,CAAC,MAAM;gBAC/B,KAAK,CAAC,KAAK,KAAK,MAAM,CAAC,KAAK;gBAC5B,KAAK,CAAC,MAAM,KAAK,MAAM,CAAC,MAAM,EAC9B,CAAC;gBACD,MAAM,IAAI,KAAK,CAAC,qBAAqB,CAAC,CAAC;YACzC,CAAC;QACH,CAAC;QAED,MAAM,UAAU,GAAG,MAAM,EAAE,IAAI,CAAC;QAChC,MAAM,UAAU,GAAG,MAAM,EAAE,IAAI,CAAC;QAChC,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,OAAO,CAAC,MAAM,EAAE,CAAC,IAAI,CAAC,EAAE,CAAC;YAC3C,MAAM,EAAE,GAAG,CAAC,CAAC,GAAG,CAAC,CAAC,GAAG,QAAQ,CAAC;YAE9B,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,CAAC,EAAE,CAAC,EAAE,EAAE,CAAC;gBAC3B,IAAI,KAAK,EAAE,CAAC;oBACV,UAAU,CAAC,EAAE,GAAG,CAAC,CAAC,GAAG,OAAO,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC;gBACtC,CAAC;qBAAM,CAAC;oBACN,UAAU,CAAC,EAAE,GAAG,CAAC,CAAC,GAAG,OAAO,CAAC,CAAC,GAAG,CAAC,CAAC,GAAG,GAAG,CAAC;gBAC5C,CAAC;gBACD,IAAI,UAAU,EAAE,CAAC;oBACf,UAAU,CAAC,EAAE,GAAG,CAAC,GAAG,CAAC,CAAC,GAAG,UAAU,CAAC,CAAC,GAAG,CAAC,CAAC,GAAG,GAAG,CAAC;gBACnD,CAAC;gBACD,IAAI,UAAU,EAAE,CAAC;oBACf,UAAU,CAAC,EAAE,GAAG,CAAC,GAAG,CAAC,CAAC,GAAG,UAAU,CAAC,CAAC,GAAG,CAAC,CAAC,GAAG,GAAG,CAAC;gBACnD,CAAC;YACH,CAAC;QACH,CAAC;QAED,OAAO,UAAU,CAAC;IACpB,CAAC;IAEO,SAAS,CACf,IAAkB,EAClB,QAAgB,EAChB,OAAa,EACb,KAAa;QAEb,MAAM,QAAQ,GAAG,IAAI,YAAY,CAC/B,OAAO,CAAC,KAAK,GAAG,OAAO,CAAC,MAAM,GAAG,QAAQ,CAC1C,CAAC;QACF,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,OAAO,CAAC,MAAM,EAAE,CAAC,EAAE,EAAE,CAAC;YACxC,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,OAAO,CAAC,KAAK,EAAE,CAAC,EAAE,EAAE,CAAC;gBACvC,MAAM,EAAE,GAAG,CAAC,CAAC,CAAC,GAAG,OAAO,CAAC,CAAC,CAAC,GAAG,KAAK,GAAG,CAAC,CAAC,GAAG,OAAO,CAAC,CAAC,CAAC,CAAC,GAAG,QAAQ,CAAC;gBAClE,MAAM,EAAE,GAAG,CAAC,CAAC,GAAG,OAAO,CAAC,KAAK,GAAG,CAAC,CAAC,GAAG,QAAQ,CAAC;gBAE9C,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,QAAQ,EAAE,CAAC,EAAE,EAAE,CAAC;oBAClC,QAAQ,CAAC,EAAE,GAAG,CAAC,CAAC,GAAG,IAAI,CAAC,EAAE,GAAG,CAAC,CAAC,CAAC;gBAClC,CAAC;YACH,CAAC;QACH,CAAC;QACD,OAAO,QAAQ,CAAC;IAClB,CAAC;IAEO,UAAU,CAChB,SAAmC,EACnC,OAAa,EACb,OAAa,EACb,WAAyB,EACzB,QAAgB,EAChB,KAAc;QAEd,MAAM,EAAE,IAAI,EAAE,YAAY,EAAE,KAAK,EAAE,GAAG,SAAS,CAAC;QAChD,MAAM,EAAE,GAAG,OAAO,CAAC,CAAC,GAAG,OAAO,CAAC,CAAC,CAAC;QACjC,MAAM,EAAE,GAAG,OAAO,CAAC,CAAC,GAAG,OAAO,CAAC,CAAC,CAAC;QACjC,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,OAAO,CAAC,MAAM,EAAE,CAAC,EAAE,EAAE,CAAC;YACxC,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,OAAO,CAAC,KAAK,EAAE,CAAC,EAAE,EAAE,CAAC;gBACvC,MAAM,EAAE,GAAG,CAAC,CAAC,CAAC,GAAG,EAAE,CAAC,GAAG,QAAQ,GAAG,CAAC,GAAG,EAAE,CAAC,GAAG,CAAC,CAAC;gBAC9C,MAAM,EAAE,GAAG,CAAC,CAAC,CAAC,GAAG,OAAO,CAAC,CAAC,CAAC,GAAG,KAAK,GAAG,CAAC,CAAC,GAAG,OAAO,CAAC,CAAC,CAAC,CAAC,GAAG,CAAC,CAAC;gBAE3D,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,CAAC,EAAE,CAAC,EAAE,EAAE,CAAC;oBAC3B,IAAI,KAAK,EAAE,CAAC;wBACV,YAAY,CAAC,EAAE,GAAG,CAAC,CAAC,GAAG,WAAW,CAAC,EAAE,GAAG,CAAC,CAAC,CAAC;oBAC7C,CAAC;yBAAM,CAAC;wBACN,YAAY,CAAC,EAAE,GAAG,CAAC,CAAC,GAAG,IAAI,CAAC,GAAG,CAC7B,IAAI,CAAC,GAAG,CAAC,WAAW,CAAC,EAAE,GAAG,CAAC,CAAC,GAAG,GAAG,EAAE,CAAC,CAAC,EACtC,GAAG,CACJ,CAAC;oBACJ,CAAC;gBACH,CAAC;gBACD,SAAS,CAAC,IAAI,CAAC,EAAE,GAAG,CAAC,CAAC,GAAG,KAAK,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,GAAG,CAAC;YAC3C,CAAC;QACH,CAAC;IACH,CAAC;IAEO,YAAY,CAClB,SAOK,EACL,cAAoD,EACpD,eAAqD,EACrD,CAAS,EACT,CAAS,EACT,KAAa,EACb,MAAc,EACd,KAAc;QAEd,MAAM,QAAQ,GAAG,IAAI,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC;QACnC,MAAM,YAAY,GAAG,IAAI,CAAC,aAAa,CAAC;QACxC,MAAM,YAAY,GAAG,IAAI,CAAC,aAAa,CAAC;QACxC,IAAI,WAAW,GAAG,IAAI,CAAC,uBAAuB,EAAE,CAAC;QACjD,IAAI,WAAW,GAAG,EAAE,KAAK,EAAE,IAAI,CAAC,UAAU,EAAE,MAAM,EAAE,IAAI,CAAC,WAAW,EAAE,CAAC;QAEvE,IAAI,KAAK,GAAG,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC,CAAC,GAAG,WAAW,CAAC,KAAK,GAAG,YAAY,CAAC,CAAC,CAAC,CAAC,CAAC;QAC7D,IAAI,KAAK,GAAG,IAAI,CAAC,GAAG,CAAC,KAAK,GAAG,WAAW,CAAC,KAAK,EAAE,KAAK,CAAC,CAAC;QACvD,KAAK,GAAG,IAAI,CAAC,GAAG,CAAC,KAAK,GAAG,WAAW,CAAC,KAAK,EAAE,CAAC,CAAC,CAAC;QAE/C,IAAI,KAAK,GAAG,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC,CAAC,GAAG,WAAW,CAAC,MAAM,GAAG,YAAY,CAAC,CAAC,CAAC,CAAC,CAAC;QAC9D,IAAI,KAAK,GAAG,IAAI,CAAC,GAAG,CAAC,KAAK,GAAG,WAAW,CAAC,MAAM,EAAE,MAAM,CAAC,CAAC;QACzD,KAAK,GAAG,IAAI,CAAC,GAAG,CAAC,KAAK,GAAG,WAAW,CAAC,MAAM,EAAE,CAAC,CAAC,CAAC;QAEhD,MAAM,YAAY,GAAG,IAAI,CAAC,GAAG,CAAC,WAAW,CAAC,KAAK,EAAE,KAAK,CAAC,CAAC;QACxD,MAAM,aAAa,GAAG,IAAI,CAAC,GAAG,CAAC,WAAW,CAAC,MAAM,EAAE,MAAM,CAAC,CAAC;QAC3D,MAAM,WAAW,GACf,KAAK,GAAG,WAAW,CAAC,KAAK,IAAI,MAAM,GAAG,WAAW,CAAC,MAAM,CAAC;QAC3D,MAAM,OAAO,GAAG,IAAI,IAAI,CAAC,KAAK,EAAE,KAAK,EAAE,YAAY,EAAE,aAAa,CAAC,CAAC;QAEpE,IAAI,UAAmB,CAAC;QACxB,IAAI,UAAU,GAAG,CAAC,CAAC;QACnB,MAAM,MAAM,GAAG,IAAI,CAAC,OAAQ,CAAC;QAC7B,IAAI,cAAc,GAAG,IAAI,CAAC,eAAe,CAAC;QAE1C,IAAI,SAAS,YAAY,YAAY,EAAE,CAAC;YACtC,IAAI,QAAQ,GAAG,IAAI,CAAC,SAAS,CAAC,SAAS,EAAE,QAAQ,EAAE,OAAO,EAAE,KAAK,CAAC,CAAC;YACnE,IAAI,KAAK,EAAE,CAAC;gBACV,UAAU,GAAG,SAAS,CAAC;oBACrB,IAAI,EAAE,QAAQ;oBACd,QAAQ,EAAE,CAAC;iBACZ,CAAC,CAAC;gBACH,QAAQ,GAAG,kBAAkB,CAAC;oBAC5B,IAAI,EAAE,QAAQ;oBACd,QAAQ,EAAE,CAAC;oBACX,UAAU;iBACX,CAAC,CAAC;YACL,CAAC;YACD,UAAU,GAAG,MAAM,CACjB,QAAQ,EACR,CAAC,CAAC,EAAE,aAAa,EAAE,YAAY,EAAE,QAAQ,CAAC,EAC1C,SAAS,CACE,CAAC;QAChB,CAAC;aAAM,CAAC;YACN,IAAI,CAAC,KAAK,EAAE,CAAC;gBACX,MAAM,IAAI,KAAK,CAAC,wCAAwC,CAAC,CAAC;YAC5D,CAAC;YACD,IAAI,CAAC,cAAc,EAAE,CAAC;gBACpB,cAAc,GAAG,IAAI,CAAC,eAAe,GAAG,IAAI,cAAc,CAAC,MAAM,CAAC,CAAC;YACrE,CAAC;YACD,cAAc,CAAC,YAAY,CAAC,KAAK,EAAE,MAAM,CAAC,CAAC;YAC3C,cAAc,CAAC,YAAY,CAAC,OAAO,CAAC,CAAC;YACrC,0DAA0D;YAC1D,IAAI,CAAC,KAAK,CAAC,IAAI,CAAC,KAAK,CAAC,EAAE,CAAC;gBACvB,cAAc,CAAC,qBAAqB,CAAC,SAAS,CAAC,KAAK,CAAC,CAAC;YACxD,CAAC;YACD,MAAM,EAAE,KAAK,EAAE,MAAM,EAAE,MAAM,EAAE,GAAG,cAAc,CAAC,OAAO,CACtD,SAAS,CAAC,KAAK,EACf,SAAS,CAAC,MAAM,EAChB,SAAS,CAAC,MAAM,CACjB,CAAC;YACF,MAAM,KAAK,GAAG,CAAC,CAAC,EAAE,aAAa,EAAE,YAAY,EAAE,CAAC,CAAQ,CAAC;YAEzD,UAAU,GAAG,QAAQ,CACnB,CAAC,KAAK,EAAE,MAAM,EAAE,MAAM,CAAC,CAAC,GAAG,CAAC,CAAC,MAAM,EAAE,EAAE;gBACrC,MAAM,GAAG,GAAG,MAAM,CAAC,EAAE,MAAM,EAAE,QAAQ,EAAE,IAAI,EAAE,EAAE,KAAK,CAAa,CAAC;gBAClE,MAAM,GAAG,GAAG,OAAO,CACjB,GAAG,EACH,CAAC,CAAC,EAAE,CAAC,EAAE,CAAC,EAAE,CAAC,CAAC,EACZ,CAAC,CAAC,EAAE,aAAa,EAAE,YAAY,EAAE,CAAC,CAAC,CACpC,CAAC;gBACF,GAAG,CAAC,OAAO,EAAE,CAAC;gBACd,OAAO,GAAG,CAAC;YACb,CAAC,CAAC,EACF,CAAC,CACF,CAAC;QACJ,CAAC;QACD,6EAA6E;QAC7E,IAAI,WAAW,EAAE,CAAC;YAChB,MAAM,aAAa,GAAG,UAAU,CAAC;YACjC,UAAU,GAAG,SAAS,CACpB,aAAa,EACb;gBACE,CAAC,CAAC,EAAE,CAAC,CAAC;gBACN,CAAC,CAAC,EAAE,WAAW,CAAC,MAAM,GAAG,MAAM,CAAC;gBAChC,CAAC,CAAC,EAAE,WAAW,CAAC,KAAK,GAAG,KAAK,CAAC;gBAC9B,CAAC,CAAC,EAAE,CAAC,CAAC;aACP,EACD,SAAS,CACV,CAAC;YACF,aAAa,CAAC,OAAO,EAAE,CAAC;QAC1B,CAAC;QAED,MAAM,YAAY,GAAG,IAAI,CAAC,QAAS,CAAC,OAAO,CAAC,UAAU,CAAW,CAAC;QAClE,UAAU,CAAC,OAAO,EAAE,CAAC;QAErB,MAAM,QAAQ,GAAG,IAAI,CAAC,GAAG,CAAC,WAAW,CAAC,KAAK,EAAE,KAAK,CAAC,CAAC;QACpD,MAAM,SAAS,GAAG,IAAI,CAAC,GAAG,CAAC,WAAW,CAAC,MAAM,EAAE,MAAM,CAAC,CAAC;QACvD,MAAM,OAAO,GAAG,IAAI,IAAI,CAAC,CAAC,GAAG,QAAQ,EAAE,CAAC,GAAG,SAAS,EAAE,QAAQ,EAAE,SAAS,CAAC,CAAC;QAC3E,OAAO,CAAC,KAAK,GAAG,IAAI,CAAC,GAAG,CAAC,OAAO,CAAC,KAAK,EAAE,KAAK,GAAG,OAAO,CAAC,CAAC,CAAC,CAAC;QAC3D,OAAO,CAAC,MAAM,GAAG,IAAI,CAAC,GAAG,CAAC,OAAO,CAAC,MAAM,EAAE,MAAM,GAAG,OAAO,CAAC,CAAC,CAAC,CAAC;QAE9D,IAAI,SAAS,YAAY,YAAY,EAAE,CAAC;YACtC,IAAI,YAAY,GAAG,YAAY,CAAC,QAAQ,EAAE,CAAC;YAC3C,IAAI,KAAK,EAAE,CAAC;gBACV,YAAY,GAAG,yBAAyB,CAAC;oBACvC,IAAI,EAAE,YAA4B;oBAClC,QAAQ,EAAE,CAAC;oBACX,UAAU;iBACX,CAAC,CAAC;YACL,CAAC;YAED,IAAI,CAAC,UAAU,CACb,eAAgB,EAChB,OAAO,EACP,OAAO,EACP,YAA4B,EAC5B,WAAW,CAAC,KAAK,EACjB,KAAK,CACN,CAAC;YAEF,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,SAAS,EAAE,CAAC,EAAE,EAAE,CAAC;gBACnC,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,QAAQ,EAAE,CAAC,EAAE,EAAE,CAAC;oBAClC,MAAM,EAAE,GAAG,CAAC,CAAC,GAAG,QAAQ,GAAG,CAAC,CAAC,GAAG,CAAC,CAAC;oBAClC,MAAM,EAAE,GAAG,CAAC,CAAC,CAAC,GAAG,OAAO,CAAC,CAAC,CAAC,GAAG,KAAK,GAAG,CAAC,CAAC,GAAG,OAAO,CAAC,CAAC,CAAC,CAAC,GAAG,CAAC,CAAC;oBAC3D,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,CAAC,EAAE,CAAC,EAAE,EAAE,CAAC;wBAC3B,cAAe,CAAC,IAAI,CAAC,EAAE,GAAG,CAAC,CAAC,GAAG,eAAgB,CAAC,IAAI,CAAC,EAAE,GAAG,CAAC,CAAC,CAAC;oBAC/D,CAAC;gBACH,CAAC;YACH,CAAC;YAED,YAAY,CAAC,OAAO,EAAE,CAAC;QACzB,CAAC;aAAM,CAAC;YACN,cAAe,CAAC,aAAa,CAAC,OAAO,EAAE,OAAO,CAAC,CAAC;YAChD,YAAY;YACZ,2EAA2E;YAC3E,sCAAsC;YACtC,MAAM,sBAAsB,GAAG,KAAK,CAAC,YAAwB,EAAE;gBAC7D,CAAC,CAAC,EAAE,CAAC,CAAC;gBACN,CAAC,CAAC,EAAE,CAAC,CAAC;gBACN,CAAC,CAAC,EAAE,CAAC,CAAC;gBACN,CAAC,CAAC,EAAE,CAAC,CAAC;aACP,CAAC,CAAC;YACH,MAAM,SAAS,GAAG,cAAe,CAAC,OAAO,CACvC,sBAAsB,CAAC,SAAS,EAAE,CAAC,MAAO,EAC1C,SAAS,CAAC,KAAK,CAChB,CAAC;YACF,YAAY,CAAC,OAAO,EAAE,CAAC;YACvB,sBAAsB,CAAC,OAAO,EAAE,CAAC;YACjC,OAAO,SAAS,CAAC;QACnB,CAAC;IACH,CAAC;IAED,kBAAkB,CAAoD,EACpE,KAAK,EACL,MAAM,EACN,MAAM,EACN,IAAI,EACJ,QAAQ,EAaT;QACC,IAAI,IAAI,CAAC,IAAI,IAAI,CAAC,CAAC,MAAM,IAAI,CAAC,MAAM,CAAC,EAAE,CAAC;YACtC,MAAM,IAAI,KAAK,CAAC,6CAA6C,CAAC,CAAC;QACjE,CAAC;QAED,IAAI,CAAC,IAAI,CAAC,IAAI,EAAE,CAAC;YACf,IAAI,MAAM,IAAI,MAAM,EAAE,CAAC;gBACrB,MAAM,IAAI,KAAK,CAAC,4CAA4C,CAAC,CAAC;YAChE,CAAC;QACH,CAAC;QAED,MAAM,KAAK,GAAG,KAAK,CAAC,KAAK,CAAC;QAC1B,MAAM,MAAM,GAAG,KAAK,CAAC,MAAM,CAAC;QAC5B,IAAI,CAAC,YAAY,CAAC,KAAK,EAAE,MAAM,CAAC,CAAC;QAEjC,oDAAoD;QACpD,sCAAsC;QAEtC,MAAM,GAAG,GAAG,IAAI,CAAC,IAAI,IAAI,KAAK,CAAC;QAC/B,IAAI,OAAqB,CAAC;QAC1B,IAAI,CAAC,cAAc,CAAC,KAAK,CAAC,EAAE,CAAC;YAC3B,OAAO,GAAG,IAAI,CAAC,iBAAiB,CAC9B,KAAK,EACL,MAAmB,EACnB,MAAmB,EACnB,GAAG,CACJ,CAAC;QACJ,CAAC;QACD,MAAM,SAAS,GAAG,IAAI,CAAC,UAAU,CAAC;QAClC,MAAM,UAAU,GAAG,IAAI,CAAC,WAAW,CAAC;QACpC,MAAM,UAAU,GAAG,IAAI,CAAC,IAAI,CAAC,MAAM,GAAG,UAAU,CAAC,CAAC;QAClD,MAAM,UAAU,GAAG,IAAI,CAAC,IAAI,CAAC,KAAK,GAAG,SAAS,CAAC,CAAC;QAEhD,SAAS,aAAa,CAAC,KAAa,EAAE,MAAc;YAClD,OAAO,GAAG;gBACR,CAAC,CAAC;oBACE,IAAI,EAAE,IAAI,YAAY,CAAC,KAAK,GAAG,MAAM,GAAG,CAAC,CAAC;oBAC1C,KAAK;oBACL,MAAM;iBACP;gBACH,CAAC,CAAC,IAAI,SAAS,CAAC,KAAK,EAAE,MAAM,CAAC,CAAC;QACnC,CAAC;QAED,MAAM,eAAe,GAAG,cAAc,CAAC,KAAK,CAAC;YAC3C,CAAC,CAAC,SAAS;YACX,CAAC,CAAC,aAAa,CAAC,KAAK,EAAE,MAAM,CAAC,CAAC;QACjC,MAAM,cAAc,GAAG,cAAc,CAAC,KAAK,CAAC;YAC1C,CAAC,CAAC,SAAS;YACX,CAAC,CAAC,aAAa,CAAC,IAAI,CAAC,GAAG,CAAC,SAAS,EAAE,KAAK,CAAC,EAAE,IAAI,CAAC,GAAG,CAAC,UAAU,EAAE,MAAM,CAAC,CAAC,CAAC;QAE5E,IAAI,OAAO,GAAG,KAAK,CAAC;QAEpB,MAAM,WAAW,GAAG,CAAC,CAAS,EAAE,CAAS,EAAE,EAAE;YAC3C,IAAI,OAAO,EAAE,CAAC;gBACZ,OAAO;YACT,CAAC;YACD,IAAI,YAAY,CAAC;YACjB,kCAAkC;YAClC,YAAY,GAAG,IAAI,CAAC,YAAY,CAC9B,cAAc,CAAC,KAAK,CAAC;gBACnB,CAAC,CAAC;oBACE,KAAK,EAAE,KAAK,CAAC,IAAI;oBACjB,MAAM,EAAG,MAAuB,CAAC,IAAI;oBACrC,MAAM,EAAG,MAAuB,CAAC,IAAI;iBACtC;gBACH,CAAC,CAAC,OAAO,EACX,cAAc,EACd,eAAe,EACf,CAAC,EACD,CAAC,EACD,KAAK,EACL,MAAM,EACN,GAAG,CACJ,CAAC;YACF,YAAY;YACZ,MAAM,MAAM,GAAG,eAAe,IAAI;gBAChC,IAAI,EAAE,YAAY;gBAClB,KAAK;gBACL,MAAM;aACP,CAAC;YACF,QAAQ,EAAE,CACR,MAAW;YACX,sCAAsC;YACtC,cAA+B,EAC/B,IAAI,IAAI,CAAC,CAAC,GAAG,SAAS,EAAE,CAAC,GAAG,UAAU,EAAE,SAAS,EAAE,UAAU,CAAC,EAC9D,CAAC,GAAG,CAAC,GAAG,UAAU,EAClB,UAAU,GAAG,UAAU,CACxB,CAAC;YAEF,IAAI,CAAC,GAAG,CAAC,GAAG,UAAU,IAAI,CAAC,GAAG,CAAC,GAAG,UAAU,EAAE,CAAC;gBAC7C,qBAAqB,CAAC,GAAG,EAAE;oBACzB,IAAI,CAAC,GAAG,CAAC,GAAG,UAAU,EAAE,CAAC;wBACvB,WAAW,CAAC,CAAC,GAAG,CAAC,EAAE,CAAC,CAAC,CAAC;oBACxB,CAAC;yBAAM,IAAI,CAAC,GAAG,CAAC,GAAG,UAAU,EAAE,CAAC;wBAC9B,WAAW,CAAC,CAAC,EAAE,CAAC,GAAG,CAAC,CAAC,CAAC;oBACxB,CAAC;gBACH,CAAC,CAAC,CAAC;YACL,CAAC;iBAAM,CAAC;gBACN,IAAI,CAAC,MAAW,CAAC,CAAC;YACpB,CAAC;QACH,CAAC,CAAC;QAEF,WAAW,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC;QAElB,OAAO,GAAG,EAAE;YACV,OAAO,GAAG,IAAI,CAAC;QACjB,CAAC,CAAC;IACJ,CAAC;IAED,OAAO;QACL,IAAI,CAAC,QAAQ,EAAE,OAAO,EAAE,CAAC;QACzB,IAAI,CAAC,eAAe,EAAE,OAAO,EAAE,CAAC;IAClC,CAAC;CACF;AAED,eAAe,IAAI,CAAC"}

package/lib/WGPUComputePass.d.ts

@@ -0,0 +1,53 @@
+/// <reference types="@webgpu/types/dist" />
+export declare const WORKGROUP_SIZE = 8;
+export interface Uniform {
+    label: string;
+    type: string;
+    data: Float32Array | Int32Array | Uint32Array;
+}
+export interface WGPUComputePassInput {
+    buffer: GPUBuffer;
+    channels: number;
+}
+export interface WGPUComputePassOutput {
+    channels: number;
+}
+export declare class WGPUComputePass<I extends string, O extends string> {
+    private _label;
+    private _device;
+    private _outputBuffers;
+    private _pipeline;
+    private _bindGroups;
+    private _needsUpdatePipeline;
+    private _inputs;
+    private _outputs;
+    private _uniforms;
+    private _uniformBuffers;
+    private _width;
+    private _height;
+    private _execWidth?;
+    private _execHeight?;
+    private _csCode;
+    private _csMain;
+    private _csDefine;
+    constructor(label: string, device: GPUDevice, opts: {
+        inputs: I[];
+        outputs: O[];
+        csMain: string;
+        csDefine?: string;
+        uniforms: Uniform[];
+    });
+    setSize(width: number, height: number): void;
+    setExecuteSize(width: number, height: number): void;
+    setOutputParams(outputParams: Record<O, WGPUComputePassOutput>): void;
+    setUniform(label: string, data: Float32Array | Int32Array | Uint32Array): void;
+    getOutputBuffer(name: O): GPUBuffer;
+    dispose(): void;
+    private _createBuffer;
+    private _resizeOutputBuffers;
+    private _updateOutputBuffers;
+    private _updatePipeline;
+    private _getFullCs;
+    private _updateBindGroups;
+    createPass(commandEncoder: GPUCommandEncoder, inputBuffers: Record<I, WGPUComputePassInput>): void;
+}