oidn-web 0.3.5 → 0.5.0

package/lib/webnnUNet.js

@@ -0,0 +1,535 @@
+import { Float16Array } from '@petamoriken/float16';
+import { OIDNResourceTracker } from './resourceTracker.js';
+const interopPipelinesByDevice = new WeakMap();
+function interopPipelines(device, sourceCount) {
+    let bySourceCount = interopPipelinesByDevice.get(device);
+    if (!bySourceCount) {
+        bySourceCount = new Map();
+        interopPipelinesByDevice.set(device, bySourceCount);
+    }
+    let pipelines = bySourceCount.get(sourceCount);
+    if (!pipelines) {
+        const inputModule = device.createShaderModule({
+            label: `oidn/webnn/input-pack/${sourceCount}`,
+            code: createInputPackShader(sourceCount)
+        });
+        const outputModule = device.createShaderModule({
+            label: 'oidn/webnn/output-unpack',
+            code: createOutputUnpackShader()
+        });
+        pipelines = {
+            input: device.createComputePipeline({
+                label: `oidn/webnn/input-pack/${sourceCount}`,
+                layout: 'auto',
+                compute: { module: inputModule, entryPoint: 'main' }
+            }),
+            output: device.createComputePipeline({
+                label: 'oidn/webnn/output-unpack',
+                layout: 'auto',
+                compute: { module: outputModule, entryPoint: 'main' }
+            })
+        };
+        bySourceCount.set(sourceCount, pipelines);
+    }
+    return pipelines;
+}
+const WORKGROUP_SIZE = 8;
+function roundUp(value, alignment) {
+    return Math.ceil(value / alignment) * alignment;
+}
+function createMappedBuffer(device, label, data, usage) {
+    const buffer = device.createBuffer({
+        label,
+        size: roundUp(data.byteLength, 4),
+        usage,
+        mappedAtCreation: true
+    });
+    new Uint8Array(buffer.getMappedRange()).set(new Uint8Array(data.buffer, data.byteOffset, data.byteLength));
+    buffer.unmap();
+    return buffer;
+}
+function uniformBuffer(device, label, values) {
+    const data = new Uint32Array(roundUp(values.length, 4));
+    data.set(values);
+    return createMappedBuffer(device, label, data, GPUBufferUsage.UNIFORM);
+}
+function hasDataType(limits, op, operand, dataType) {
+    return Boolean(limits?.[op]?.[operand]?.dataTypes?.includes?.(dataType));
+}
+function tensorBytes(tensor, precision) {
+    if (precision === 'fp16' &&
+        tensor.desc.dataType === 'Float16') {
+        return new Uint8Array(tensor.data.buffer, tensor.data.byteOffset, tensor.data.byteLength);
+    }
+    if (precision === 'fp32' &&
+        tensor.desc.dataType === 'Float32') {
+        return new Uint8Array(tensor.data.buffer, tensor.data.byteOffset, tensor.data.byteLength);
+    }
+    const source = tensor.desc.dataType === 'Float32'
+        ? new Float32Array(tensor.data.buffer, tensor.data.byteOffset, tensor.data.byteLength / 4)
+        : new Float16Array(tensor.data.buffer, tensor.data.byteOffset, tensor.data.byteLength / 2);
+    const converted = precision === 'fp16'
+        ? new Float16Array(source)
+        : new Float32Array(source);
+    return new Uint8Array(converted.buffer, converted.byteOffset, converted.byteLength);
+}
+function createInputPackShader(sourceCount) {
+    const sources = Array.from({ length: sourceCount }, (_, index) => `@group(0) @binding(${index}) var<storage, read> input${index}: array<vec4<f32>>;`).join('\n');
+    const branches = Array.from({ length: sourceCount }, (_, index) => {
+        const firstChannel = index * 3;
+        return `if (channel < ${firstChannel + 3}u) {
+      return input${index}[pixel][channel - ${firstChannel}u];
+    }`;
+    }).join('\n  ');
+    return /* wgsl */ `enable f16;
+struct Params { width: u32, height: u32, channels: u32, padding: u32 }
+${sources}
+@group(0) @binding(${sourceCount}) var<storage, read_write> outputData: array<f16>;
+@group(0) @binding(${sourceCount + 1}) var<uniform> params: Params;
+
+fn readChannel(pixel: u32, channel: u32) -> f32 {
+  ${branches}
+  return 0.0;
+}
+
+@compute @workgroup_size(${WORKGROUP_SIZE}, ${WORKGROUP_SIZE}, 1)
+fn main(@builtin(global_invocation_id) gid: vec3<u32>) {
+  if (gid.x >= params.width || gid.y >= params.height || gid.z >= params.channels) {
+    return;
+  }
+  let pixel = gid.y * params.width + gid.x;
+  let outputIndex = (gid.z * params.height + gid.y) * params.width + gid.x;
+  outputData[outputIndex] = f16(readChannel(pixel, gid.z));
+}
+`;
+}
+function createOutputUnpackShader() {
+    return /* wgsl */ `enable f16;
+struct Params { width: u32, height: u32, padding0: u32, padding1: u32 }
+@group(0) @binding(0) var<storage, read> inputData: array<f16>;
+@group(0) @binding(1) var<storage, read_write> outputData: array<vec4<f32>>;
+@group(0) @binding(2) var<uniform> params: Params;
+
+@compute @workgroup_size(${WORKGROUP_SIZE}, ${WORKGROUP_SIZE}, 1)
+fn main(@builtin(global_invocation_id) gid: vec3<u32>) {
+  if (gid.x >= params.width || gid.y >= params.height) { return; }
+  let pixel = gid.y * params.width + gid.x;
+  let plane = params.width * params.height;
+  outputData[pixel] = vec4<f32>(
+    f32(inputData[pixel]),
+    f32(inputData[plane + pixel]),
+    f32(inputData[plane * 2u + pixel]),
+    0.0
+  );
+}
+`;
+}
+function resolveWebNNPrecision(device, requested) {
+    if (requested === 'fp32') {
+        throw new Error('OIDN WebNN GPU interop currently requires FP16 exportable tensors');
+    }
+    if (!device.features.has('shader-f16')) {
+        throw new Error('OIDN WebNN requires shader-f16 on the shared GPUDevice');
+    }
+    return 'fp16';
+}
+function activation(builder, operand, kind) {
+    return kind === 'relu' ? builder.relu(operand) : operand;
+}
+/** Experimental model-driven WebNN executor with FP16 WebGPU interop. */
+export class WebNNUNetExecutor {
+    _device;
+    _model;
+    precision;
+    support;
+    _context;
+    _builderConstructor;
+    _shapeCache = new Map();
+    _shapePromises = new Map();
+    _retiredExecutions = new Set();
+    _pendingCreationCount = 0;
+    _shapeCacheSize;
+    _clock = 0;
+    _inputPipeline;
+    _outputPipeline;
+    _resources = new OIDNResourceTracker();
+    _disposed = false;
+    constructor(_device, _model, options = {}) {
+        this._device = _device;
+        this._model = _model;
+        this.precision = resolveWebNNPrecision(_device, options.precision ?? 'auto');
+        this._shapeCacheSize = Math.max(1, options.shapeCacheSize ?? 2);
+        this.support = {
+            available: false,
+            fp16Conv: false,
+            gpuInterop: false
+        };
+        const pipelines = interopPipelines(_device, _model.inputChannels / 3);
+        this._inputPipeline = pipelines.input;
+        this._outputPipeline = pipelines.output;
+    }
+    async prepare() {
+        if (this._disposed)
+            throw new Error('OIDN WebNN executor is disposed');
+        const webNN = globalThis.navigator?.ml;
+        const Builder = globalThis.MLGraphBuilder;
+        if (!webNN?.createContext || typeof Builder !== 'function') {
+            this.support.reason = 'WebNN is not exposed by this browser';
+            throw new Error(this.support.reason);
+        }
+        this._builderConstructor = Builder;
+        try {
+            try {
+                // Chromium's experimental implementation only enables WebGPU tensor
+                // interop for an explicitly GPU-backed context.
+                this._context = await webNN.createContext({
+                    deviceType: 'gpu',
+                    powerPreference: 'high-performance'
+                });
+            }
+            catch {
+                this._context = await webNN.createContext({ deviceType: 'gpu' });
+            }
+            this._resources.track('ml-context', this._context);
+            if (this._disposed)
+                throw new Error('OIDN WebNN executor is disposed');
+            if (typeof this._context.createExportableTensor !== 'function' ||
+                typeof this._context.exportToGPU !== 'function') {
+                this.support.reason = 'WebNN WebGPU tensor interop is unavailable';
+                throw new Error(this.support.reason);
+            }
+            const limits = this._context.opSupportLimits?.() ?? {};
+            this.support.fp16Conv =
+                hasDataType(limits, 'conv2d', 'input', 'float16') &&
+                    hasDataType(limits, 'conv2d', 'filter', 'float16') &&
+                    hasDataType(limits, 'conv2d', 'output', 'float16');
+            if (!this.support.fp16Conv) {
+                this.support.reason = 'WebNN does not support FP16 conv2d';
+                throw new Error(this.support.reason);
+            }
+            let probe;
+            let probeBuffer;
+            try {
+                probe = this._resources.track('ml-tensor', await this._context.createExportableTensor({ dataType: 'float16', shape: [4] }, this._device));
+                probeBuffer = this._resources.track('gpu-buffer', await this._context.exportToGPU(probe));
+                this.support.gpuInterop = true;
+            }
+            catch (error) {
+                this.support.reason =
+                    `WebNN FP16 WebGPU interop failed: ${String(error)}`;
+                throw new Error(this.support.reason);
+            }
+            finally {
+                this._releaseBuffer(probeBuffer);
+                this._releaseTensor(probe);
+            }
+            this.support.available = true;
+        }
+        catch (error) {
+            this._releaseContext();
+            throw error;
+        }
+    }
+    _constant(builder, tensor) {
+        return builder.constant({
+            dataType: 'float16',
+            shape: [...tensor.desc.dims]
+        }, tensorBytes(tensor, this.precision));
+    }
+    async _createExecution(width, height) {
+        if (this._disposed)
+            throw new Error('OIDN WebNN executor is disposed');
+        const builder = new this._builderConstructor(this._context);
+        const values = new Map();
+        const shapes = new Map();
+        values.set(this._model.spec.input, builder.input('input', {
+            dataType: 'float16',
+            shape: [1, this._model.inputChannels, height, width]
+        }));
+        shapes.set(this._model.spec.input, [this._model.inputChannels, height, width]);
+        for (const node of this._model.spec.nodes) {
+            let result;
+            let shape;
+            if (node.op === 'conv2d') {
+                const inputShape = shapes.get(node.input);
+                const tensors = this._model.convTensors.get(node.id);
+                const convolution = builder.conv2d(values.get(node.input), this._constant(builder, tensors.weight), {
+                    bias: this._constant(builder, tensors.bias),
+                    padding: [1, 1, 1, 1],
+                    inputLayout: 'nchw',
+                    filterLayout: 'oihw'
+                });
+                result = activation(builder, convolution, node.activation);
+                shape = [tensors.outputChannels, inputShape[1], inputShape[2]];
+            }
+            else if (node.op === 'maxPool2d') {
+                const inputShape = shapes.get(node.input);
+                result = builder.maxPool2d(values.get(node.input), {
+                    windowDimensions: [2, 2],
+                    strides: [2, 2],
+                    padding: [0, inputShape[1] % 2, 0, inputShape[2] % 2],
+                    layout: 'nchw'
+                });
+                shape = [
+                    inputShape[0],
+                    Math.ceil(inputShape[1] / 2),
+                    Math.ceil(inputShape[2] / 2)
+                ];
+            }
+            else if (node.op === 'upsample2d') {
+                const inputShape = shapes.get(node.input);
+                result = builder.resample2d(values.get(node.input), {
+                    mode: 'nearest-neighbor',
+                    axes: [2, 3],
+                    scales: [2, 2]
+                });
+                shape = [inputShape[0], inputShape[1] * 2, inputShape[2] * 2];
+            }
+            else {
+                const inputShapes = node.inputs.map((input) => shapes.get(input));
+                if (inputShapes.some((candidate) => candidate[1] !== inputShapes[0][1] ||
+                    candidate[2] !== inputShapes[0][2])) {
+                    throw new Error(`WebNN concat ${node.id} has mismatched spatial shapes`);
+                }
+                result = builder.concat(node.inputs.map((input) => values.get(input)), 1);
+                shape = [
+                    inputShapes.reduce((sum, candidate) => sum + candidate[0], 0),
+                    inputShapes[0][1],
+                    inputShapes[0][2]
+                ];
+            }
+            values.set(node.id, result);
+            shapes.set(node.id, shape);
+        }
+        let graph;
+        let inputTensor;
+        let outputTensor;
+        let outputBuffer;
+        let inputUniform;
+        let outputUniform;
+        try {
+            graph = this._resources.track('ml-graph', await builder.build({
+                output: values.get(this._model.spec.output)
+            }));
+            inputTensor = this._resources.track('ml-tensor', await this._context.createExportableTensor({
+                dataType: 'float16',
+                shape: [1, this._model.inputChannels, height, width],
+                writable: true
+            }, this._device));
+            outputTensor = this._resources.track('ml-tensor', await this._context.createExportableTensor({
+                dataType: 'float16',
+                shape: [1, this._model.outputChannels, height, width],
+                readable: true
+            }, this._device));
+            outputBuffer = this._resources.track('gpu-buffer', this._device.createBuffer({
+                label: `oidn/webnn/output/${width}x${height}`,
+                size: width * height * 4 * 4,
+                usage: GPUBufferUsage.STORAGE |
+                    GPUBufferUsage.COPY_SRC |
+                    GPUBufferUsage.COPY_DST
+            }));
+            inputUniform = this._resources.track('gpu-buffer', uniformBuffer(this._device, `oidn/webnn/input/${width}x${height}`, [width, height, this._model.inputChannels]));
+            outputUniform = this._resources.track('gpu-buffer', uniformBuffer(this._device, `oidn/webnn/output/${width}x${height}`, [width, height]));
+            const execution = {
+                graph,
+                inputTensor,
+                outputTensor,
+                outputBuffer,
+                inputUniform,
+                outputUniform,
+                width,
+                height,
+                lastUsed: ++this._clock
+            };
+            return execution;
+        }
+        catch (error) {
+            this._releaseBuffer(outputUniform);
+            this._releaseBuffer(inputUniform);
+            this._releaseBuffer(outputBuffer);
+            this._releaseTensor(outputTensor);
+            this._releaseTensor(inputTensor);
+            this._releaseGraph(graph);
+            throw error;
+        }
+    }
+    async _execution(width, height) {
+        const key = `${width}x${height}`;
+        let execution = this._shapeCache.get(key);
+        if (!execution) {
+            let pending = this._shapePromises.get(key);
+            if (!pending) {
+                pending = (async () => {
+                    this._pendingCreationCount++;
+                    try {
+                        return await this._createExecution(width, height);
+                    }
+                    finally {
+                        this._pendingCreationCount--;
+                    }
+                })();
+                this._shapePromises.set(key, pending);
+            }
+            try {
+                execution = await pending;
+                if (this._disposed) {
+                    this._destroyExecution(execution);
+                    throw new Error('OIDN WebNN executor is disposed');
+                }
+                this._shapeCache.set(key, execution);
+            }
+            finally {
+                if (this._shapePromises.get(key) === pending) {
+                    this._shapePromises.delete(key);
+                }
+            }
+            if (this._shapeCache.size > this._shapeCacheSize) {
+                const oldest = [...this._shapeCache.entries()]
+                    .filter(([candidate]) => candidate !== key)
+                    .sort((left, right) => left[1].lastUsed - right[1].lastUsed)[0];
+                if (oldest) {
+                    this._shapeCache.delete(oldest[0]);
+                    this._retireExecution(oldest[1]);
+                }
+            }
+        }
+        execution.lastUsed = ++this._clock;
+        return execution;
+    }
+    /** Compiles common tile shapes while the host still reports model loading. */
+    async prewarm(shapes) {
+        for (const shape of shapes) {
+            await this._execution(shape.width, shape.height);
+        }
+    }
+    async execute(inputBuffers, width, height) {
+        const sourceCount = this._model.inputChannels / 3;
+        if (inputBuffers.length !== sourceCount) {
+            throw new Error(`OIDN WebNN expected ${sourceCount} input buffers, got ${inputBuffers.length}`);
+        }
+        const execution = await this._execution(width, height);
+        const inputGPUBuffer = this._resources.track('gpu-buffer', await this._context.exportToGPU(execution.inputTensor));
+        try {
+            const inputEntries = inputBuffers.map((buffer, binding) => ({ binding, resource: { buffer } }));
+            inputEntries.push({
+                binding: sourceCount,
+                resource: { buffer: inputGPUBuffer }
+            });
+            inputEntries.push({
+                binding: sourceCount + 1,
+                resource: { buffer: execution.inputUniform }
+            });
+            const inputBindGroup = this._device.createBindGroup({
+                label: 'oidn/webnn/input-bindings',
+                layout: this._inputPipeline.getBindGroupLayout(0),
+                entries: inputEntries
+            });
+            const inputEncoder = this._device.createCommandEncoder({
+                label: 'oidn/webnn/input-pack'
+            });
+            const inputPass = inputEncoder.beginComputePass();
+            inputPass.setPipeline(this._inputPipeline);
+            inputPass.setBindGroup(0, inputBindGroup);
+            inputPass.dispatchWorkgroups(Math.ceil(width / WORKGROUP_SIZE), Math.ceil(height / WORKGROUP_SIZE), this._model.inputChannels);
+            inputPass.end();
+            this._device.queue.submit([inputEncoder.finish()]);
+        }
+        finally {
+            this._releaseBuffer(inputGPUBuffer);
+        }
+        this._context.dispatch(execution.graph, { input: execution.inputTensor }, { output: execution.outputTensor });
+        const outputGPUBuffer = this._resources.track('gpu-buffer', await this._context.exportToGPU(execution.outputTensor));
+        try {
+            const outputBindGroup = this._device.createBindGroup({
+                label: 'oidn/webnn/output-bindings',
+                layout: this._outputPipeline.getBindGroupLayout(0),
+                entries: [
+                    { binding: 0, resource: { buffer: outputGPUBuffer } },
+                    { binding: 1, resource: { buffer: execution.outputBuffer } },
+                    { binding: 2, resource: { buffer: execution.outputUniform } }
+                ]
+            });
+            const outputEncoder = this._device.createCommandEncoder({
+                label: 'oidn/webnn/output-unpack'
+            });
+            const outputPass = outputEncoder.beginComputePass();
+            outputPass.setPipeline(this._outputPipeline);
+            outputPass.setBindGroup(0, outputBindGroup);
+            outputPass.dispatchWorkgroups(Math.ceil(width / WORKGROUP_SIZE), Math.ceil(height / WORKGROUP_SIZE));
+            outputPass.end();
+            this._device.queue.submit([outputEncoder.finish()]);
+        }
+        finally {
+            this._releaseBuffer(outputGPUBuffer);
+        }
+        return execution.outputBuffer;
+    }
+    async executeCPU(input, width, height) {
+        const execution = await this._execution(width, height);
+        const plane = width * height;
+        const packed = new Float16Array(plane * this._model.inputChannels);
+        for (let pixel = 0; pixel < plane; pixel++) {
+            for (let channel = 0; channel < this._model.inputChannels; channel++) {
+                packed[channel * plane + pixel] =
+                    input[pixel * this._model.inputChannels + channel];
+            }
+        }
+        this._context.writeTensor(execution.inputTensor, packed);
+        this._context.dispatch(execution.graph, { input: execution.inputTensor }, { output: execution.outputTensor });
+        const result = new Float16Array(await this._context.readTensor(execution.outputTensor));
+        const unpacked = new Float32Array(plane * this._model.outputChannels);
+        for (let pixel = 0; pixel < plane; pixel++) {
+            for (let channel = 0; channel < this._model.outputChannels; channel++) {
+                unpacked[pixel * this._model.outputChannels + channel] =
+                    result[channel * plane + pixel];
+            }
+        }
+        return unpacked;
+    }
+    _destroyExecution(execution) {
+        this._releaseGraph(execution.graph);
+        this._releaseTensor(execution.inputTensor);
+        this._releaseTensor(execution.outputTensor);
+        this._releaseBuffer(execution.outputBuffer);
+        this._releaseBuffer(execution.inputUniform);
+        this._releaseBuffer(execution.outputUniform);
+    }
+    _retireExecution(execution) {
+        this._retiredExecutions.add(execution);
+        void this._device.queue.onSubmittedWorkDone().catch(() => undefined).then(() => {
+            this._retiredExecutions.delete(execution);
+            this._destroyExecution(execution);
+        });
+    }
+    _releaseBuffer(buffer) {
+        this._resources.release('gpu-buffer', buffer, () => buffer.destroy());
+    }
+    _releaseTensor(tensor) {
+        this._resources.release('ml-tensor', tensor, () => tensor.destroy());
+    }
+    _releaseGraph(graph) {
+        this._resources.release('ml-graph', graph, () => graph.destroy?.());
+    }
+    _releaseContext() {
+        this._resources.release('ml-context', this._context, () => this._context.destroy?.());
+    }
+    getResourceInfo() {
+        return this._resources.snapshot(this._pendingCreationCount + this._retiredExecutions.size);
+    }
+    dispose() {
+        if (this._disposed)
+            return;
+        this._disposed = true;
+        for (const execution of this._shapeCache.values()) {
+            this._destroyExecution(execution);
+        }
+        this._shapeCache.clear();
+        for (const execution of this._retiredExecutions) {
+            this._destroyExecution(execution);
+        }
+        this._retiredExecutions.clear();
+        this._shapePromises.clear();
+        this._releaseContext();
+    }
+}
+//# sourceMappingURL=webnnUNet.js.map

package/lib/webnnUNet.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"webnnUNet.js","sourceRoot":"","sources":["../src/webnnUNet.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,YAAY,EAAE,MAAM,sBAAsB,CAAC;AAUpD,OAAO,EACL,mBAAmB,EAEpB,MAAM,sBAAsB,CAAC;AAiE9B,MAAM,wBAAwB,GAAG,IAAI,OAAO,EAGzC,CAAC;AAEJ,SAAS,gBAAgB,CAAC,MAAiB,EAAE,WAAmB;IAC9D,IAAI,aAAa,GAAG,wBAAwB,CAAC,GAAG,CAAC,MAAM,CAAC,CAAC;IACzD,IAAI,CAAC,aAAa,EAAE,CAAC;QACnB,aAAa,GAAG,IAAI,GAAG,EAAE,CAAC;QAC1B,wBAAwB,CAAC,GAAG,CAAC,MAAM,EAAE,aAAa,CAAC,CAAC;IACtD,CAAC;IACD,IAAI,SAAS,GAAG,aAAa,CAAC,GAAG,CAAC,WAAW,CAAC,CAAC;IAC/C,IAAI,CAAC,SAAS,EAAE,CAAC;QACf,MAAM,WAAW,GAAG,MAAM,CAAC,kBAAkB,CAAC;YAC5C,KAAK,EAAE,yBAAyB,WAAW,EAAE;YAC7C,IAAI,EAAE,qBAAqB,CAAC,WAAW,CAAC;SACzC,CAAC,CAAC;QACH,MAAM,YAAY,GAAG,MAAM,CAAC,kBAAkB,CAAC;YAC7C,KAAK,EAAE,0BAA0B;YACjC,IAAI,EAAE,wBAAwB,EAAE;SACjC,CAAC,CAAC;QACH,SAAS,GAAG;YACV,KAAK,EAAE,MAAM,CAAC,qBAAqB,CAAC;gBAClC,KAAK,EAAE,yBAAyB,WAAW,EAAE;gBAC7C,MAAM,EAAE,MAAM;gBACd,OAAO,EAAE,EAAE,MAAM,EAAE,WAAW,EAAE,UAAU,EAAE,MAAM,EAAE;aACrD,CAAC;YACF,MAAM,EAAE,MAAM,CAAC,qBAAqB,CAAC;gBACnC,KAAK,EAAE,0BAA0B;gBACjC,MAAM,EAAE,MAAM;gBACd,OAAO,EAAE,EAAE,MAAM,EAAE,YAAY,EAAE,UAAU,EAAE,MAAM,EAAE;aACtD,CAAC;SACH,CAAC;QACF,aAAa,CAAC,GAAG,CAAC,WAAW,EAAE,SAAS,CAAC,CAAC;IAC5C,CAAC;IACD,OAAO,SAAS,CAAC;AACnB,CAAC;AAcD,MAAM,cAAc,GAAG,CAAC,CAAC;AAEzB,SAAS,OAAO,CAAC,KAAa,EAAE,SAAiB;IAC/C,OAAO,IAAI,CAAC,IAAI,CAAC,KAAK,GAAG,SAAS,CAAC,GAAG,SAAS,CAAC;AAClD,CAAC;AAED,SAAS,kBAAkB,CACzB,MAAiB,EACjB,KAAa,EACb,IAAqB,EACrB,KAA0B;IAE1B,MAAM,MAAM,GAAG,MAAM,CAAC,YAAY,CAAC;QACjC,KAAK;QACL,IAAI,EAAE,OAAO,CAAC,IAAI,CAAC,UAAU,EAAE,CAAC,CAAC;QACjC,KAAK;QACL,gBAAgB,EAAE,IAAI;KACvB,CAAC,CAAC;IACH,IAAI,UAAU,CAAC,MAAM,CAAC,cAAc,EAAE,CAAC,CAAC,GAAG,CACzC,IAAI,UAAU,CAAC,IAAI,CAAC,MAAM,EAAE,IAAI,CAAC,UAAU,EAAE,IAAI,CAAC,UAAU,CAAC,CAC9D,CAAC;IACF,MAAM,CAAC,KAAK,EAAE,CAAC;IACf,OAAO,MAAM,CAAC;AAChB,CAAC;AAED,SAAS,aAAa,CACpB,MAAiB,EACjB,KAAa,EACb,MAAyB;IAEzB,MAAM,IAAI,GAAG,IAAI,WAAW,CAAC,OAAO,CAAC,MAAM,CAAC,MAAM,EAAE,CAAC,CAAC,CAAC,CAAC;IACxD,IAAI,CAAC,GAAG,CAAC,MAAM,CAAC,CAAC;IACjB,OAAO,kBAAkB,CAAC,MAAM,EAAE,KAAK,EAAE,IAAI,EAAE,cAAc,CAAC,OAAO,CAAC,CAAC;AACzE,CAAC;AAED,SAAS,WAAW,CAClB,MAA2B,EAC3B,EAAU,EACV,OAAe,EACf,QAAoB;IAEpB,OAAO,OAAO,CACZ,MAAM,EAAE,CAAC,EAAE,CAAC,EAAE,CAAC,OAAO,CAAC,EAAE,SAAS,EAAE,QAAQ,EAAE,CAAC,QAAQ,CAAC,CACzD,CAAC;AACJ,CAAC;AAED,SAAS,WAAW,CAAC,MAAkB,EAAE,SAA8B;IACrE,IACE,SAAS,KAAK,MAAM;QACpB,MAAM,CAAC,IAAI,CAAC,QAAQ,KAAK,SAAS,EAClC,CAAC;QACD,OAAO,IAAI,UAAU,CACnB,MAAM,CAAC,IAAI,CAAC,MAAM,EAClB,MAAM,CAAC,IAAI,CAAC,UAAU,EACtB,MAAM,CAAC,IAAI,CAAC,UAAU,CACvB,CAAC;IACJ,CAAC;IACD,IACE,SAAS,KAAK,MAAM;QACpB,MAAM,CAAC,IAAI,CAAC,QAAQ,KAAK,SAAS,EAClC,CAAC;QACD,OAAO,IAAI,UAAU,CACnB,MAAM,CAAC,IAAI,CAAC,MAAM,EAClB,MAAM,CAAC,IAAI,CAAC,UAAU,EACtB,MAAM,CAAC,IAAI,CAAC,UAAU,CACvB,CAAC;IACJ,CAAC;IAED,MAAM,MAAM,GAAG,MAAM,CAAC,IAAI,CAAC,QAAQ,KAAK,SAAS;QAC/C,CAAC,CAAC,IAAI,YAAY,CACd,MAAM,CAAC,IAAI,CAAC,MAAM,EAClB,MAAM,CAAC,IAAI,CAAC,UAAU,EACtB,MAAM,CAAC,IAAI,CAAC,UAAU,GAAG,CAAC,CAC3B;QACH,CAAC,CAAC,IAAI,YAAY,CACd,MAAM,CAAC,IAAI,CAAC,MAAM,EAClB,MAAM,CAAC,IAAI,CAAC,UAAU,EACtB,MAAM,CAAC,IAAI,CAAC,UAAU,GAAG,CAAC,CAC3B,CAAC;IACN,MAAM,SAAS,GAAG,SAAS,KAAK,MAAM;QACpC,CAAC,CAAC,IAAI,YAAY,CAAC,MAAM,CAAC;QAC1B,CAAC,CAAC,IAAI,YAAY,CAAC,MAAM,CAAC,CAAC;IAC7B,OAAO,IAAI,UAAU,CACnB,SAAS,CAAC,MAAM,EAChB,SAAS,CAAC,UAAU,EACpB,SAAS,CAAC,UAAU,CACrB,CAAC;AACJ,CAAC;AAED,SAAS,qBAAqB,CAAC,WAAmB;IAChD,MAAM,OAAO,GAAG,KAAK,CAAC,IAAI,CACxB,EAAE,MAAM,EAAE,WAAW,EAAE,EACvB,CAAC,CAAC,EAAE,KAAK,EAAE,EAAE,CACX,sBAAsB,KAAK,6BAA6B,KAAK,qBAAqB,CACrF,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;IACb,MAAM,QAAQ,GAAG,KAAK,CAAC,IAAI,CAAC,EAAE,MAAM,EAAE,WAAW,EAAE,EAAE,CAAC,CAAC,EAAE,KAAK,EAAE,EAAE;QAChE,MAAM,YAAY,GAAG,KAAK,GAAG,CAAC,CAAC;QAC/B,OAAO,iBAAiB,YAAY,GAAG,CAAC;oBACxB,KAAK,qBAAqB,YAAY;MACpD,CAAC;IACL,CAAC,CAAC,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC;IAChB,OAAO,UAAU,CAAC;;EAElB,OAAO;qBACY,WAAW;qBACX,WAAW,GAAG,CAAC;;;IAGhC,QAAQ;;;;2BAIe,cAAc,KAAK,cAAc;;;;;;;;;CAS3D,CAAC;AACF,CAAC;AAED,SAAS,wBAAwB;IAC/B,OAAO,UAAU,CAAC;;;;;;2BAMO,cAAc,KAAK,cAAc;;;;;;;;;;;;CAY3D,CAAC;AACF,CAAC;AAED,SAAS,qBAAqB,CAC5B,MAAiB,EACjB,SAAqC;IAErC,IAAI,SAAS,KAAK,MAAM,EAAE,CAAC;QACzB,MAAM,IAAI,KAAK,CACb,mEAAmE,CACpE,CAAC;IACJ,CAAC;IACD,IAAI,CAAC,MAAM,CAAC,QAAQ,CAAC,GAAG,CAAC,YAAY,CAAC,EAAE,CAAC;QACvC,MAAM,IAAI,KAAK,CAAC,wDAAwD,CAAC,CAAC;IAC5E,CAAC;IACD,OAAO,MAAM,CAAC;AAChB,CAAC;AAED,SAAS,UAAU,CACjB,OAA2B,EAC3B,OAAsB,EACtB,IAAkC;IAElC,OAAO,IAAI,KAAK,MAAM,CAAC,CAAC,CAAC,OAAO,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,OAAO,CAAC;AAC3D,CAAC;AAED,yEAAyE;AACzE,MAAM,OAAO,iBAAiB;IAoBlB;IACA;IApBD,SAAS,CAAsB;IAC/B,OAAO,CAAsB;IAE9B,QAAQ,CAAiB;IACzB,mBAAmB,CAEH;IAChB,WAAW,GAAG,IAAI,GAAG,EAA+B,CAAC;IACrD,cAAc,GAAG,IAAI,GAAG,EAAwC,CAAC;IACjE,kBAAkB,GAAG,IAAI,GAAG,EAAuB,CAAC;IACpD,qBAAqB,GAAG,CAAC,CAAC;IAC1B,eAAe,CAAS;IACxB,MAAM,GAAG,CAAC,CAAC;IACX,cAAc,CAAqB;IACnC,eAAe,CAAqB;IACpC,UAAU,GAAG,IAAI,mBAAmB,EAAE,CAAC;IACvC,SAAS,GAAG,KAAK,CAAC;IAE1B,YACU,OAAkB,EAClB,MAA0B,EAClC,UAA4B,EAAE;QAFtB,YAAO,GAAP,OAAO,CAAW;QAClB,WAAM,GAAN,MAAM,CAAoB;QAGlC,IAAI,CAAC,SAAS,GAAG,qBAAqB,CACpC,OAAO,EACP,OAAO,CAAC,SAAS,IAAI,MAAM,CAC5B,CAAC;QACF,IAAI,CAAC,eAAe,GAAG,IAAI,CAAC,GAAG,CAAC,CAAC,EAAE,OAAO,CAAC,cAAc,IAAI,CAAC,CAAC,CAAC;QAChE,IAAI,CAAC,OAAO,GAAG;YACb,SAAS,EAAE,KAAK;YAChB,QAAQ,EAAE,KAAK;YACf,UAAU,EAAE,KAAK;SAClB,CAAC;QAEF,MAAM,SAAS,GAAG,gBAAgB,CAChC,OAAO,EACP,MAAM,CAAC,aAAa,GAAG,CAAC,CACzB,CAAC;QACF,IAAI,CAAC,cAAc,GAAG,SAAS,CAAC,KAAK,CAAC;QACtC,IAAI,CAAC,eAAe,GAAG,SAAS,CAAC,MAAM,CAAC;IAC1C,CAAC;IAED,KAAK,CAAC,OAAO;QACX,IAAI,IAAI,CAAC,SAAS;YAAE,MAAM,IAAI,KAAK,CAAC,iCAAiC,CAAC,CAAC;QACvE,MAAM,KAAK,GAAI,UAAU,CAAC,SAAiB,EAAE,EAAE,CAAC;QAChD,MAAM,OAAO,GAAI,UAAkB,CAAC,cAAc,CAAC;QACnD,IAAI,CAAC,KAAK,EAAE,aAAa,IAAI,OAAO,OAAO,KAAK,UAAU,EAAE,CAAC;YAC3D,IAAI,CAAC,OAAO,CAAC,MAAM,GAAG,sCAAsC,CAAC;YAC7D,MAAM,IAAI,KAAK,CAAC,IAAI,CAAC,OAAO,CAAC,MAAM,CAAC,CAAC;QACvC,CAAC;QACD,IAAI,CAAC,mBAAmB,GAAG,OAAO,CAAC;QACnC,IAAI,CAAC;YACH,IAAI,CAAC;gBACH,oEAAoE;gBACpE,gDAAgD;gBAChD,IAAI,CAAC,QAAQ,GAAG,MAAM,KAAK,CAAC,aAAa,CAAC;oBACxC,UAAU,EAAE,KAAK;oBACjB,eAAe,EAAE,kBAAkB;iBACpC,CAAC,CAAC;YACL,CAAC;YAAC,MAAM,CAAC;gBACP,IAAI,CAAC,QAAQ,GAAG,MAAM,KAAK,CAAC,aAAa,CAAC,EAAE,UAAU,EAAE,KAAK,EAAE,CAAC,CAAC;YACnE,CAAC;YACD,IAAI,CAAC,UAAU,CAAC,KAAK,CAAC,YAAY,EAAE,IAAI,CAAC,QAAQ,CAAC,CAAC;YACnD,IAAI,IAAI,CAAC,SAAS;gBAAE,MAAM,IAAI,KAAK,CAAC,iCAAiC,CAAC,CAAC;YAEvE,IACE,OAAO,IAAI,CAAC,QAAQ,CAAC,sBAAsB,KAAK,UAAU;gBAC1D,OAAO,IAAI,CAAC,QAAQ,CAAC,WAAW,KAAK,UAAU,EAC/C,CAAC;gBACD,IAAI,CAAC,OAAO,CAAC,MAAM,GAAG,4CAA4C,CAAC;gBACnE,MAAM,IAAI,KAAK,CAAC,IAAI,CAAC,OAAO,CAAC,MAAM,CAAC,CAAC;YACvC,CAAC;YACD,MAAM,MAAM,GAAG,IAAI,CAAC,QAAQ,CAAC,eAAe,EAAE,EAAE,IAAI,EAAE,CAAC;YACvD,IAAI,CAAC,OAAO,CAAC,QAAQ;gBACnB,WAAW,CAAC,MAAM,EAAE,QAAQ,EAAE,OAAO,EAAE,SAAS,CAAC;oBACjD,WAAW,CAAC,MAAM,EAAE,QAAQ,EAAE,QAAQ,EAAE,SAAS,CAAC;oBAClD,WAAW,CAAC,MAAM,EAAE,QAAQ,EAAE,QAAQ,EAAE,SAAS,CAAC,CAAC;YACrD,IAAI,CAAC,IAAI,CAAC,OAAO,CAAC,QAAQ,EAAE,CAAC;gBAC3B,IAAI,CAAC,OAAO,CAAC,MAAM,GAAG,oCAAoC,CAAC;gBAC3D,MAAM,IAAI,KAAK,CAAC,IAAI,CAAC,OAAO,CAAC,MAAM,CAAC,CAAC;YACvC,CAAC;YAED,IAAI,KAA+B,CAAC;YACpC,IAAI,WAAkC,CAAC;YACvC,IAAI,CAAC;gBACH,KAAK,GAAG,IAAI,CAAC,UAAU,CAAC,KAAK,CAC3B,WAAW,EACX,MAAM,IAAI,CAAC,QAAQ,CAAC,sBAAsB,CACxC,EAAE,QAAQ,EAAE,SAAS,EAAE,KAAK,EAAE,CAAC,CAAC,CAAC,EAAE,EACnC,IAAI,CAAC,OAAO,CACb,CACF,CAAC;gBACF,WAAW,GAAG,IAAI,CAAC,UAAU,CAAC,KAAK,CACjC,YAAY,EACZ,MAAM,IAAI,CAAC,QAAQ,CAAC,WAAW,CAAC,KAAK,CAAC,CACvC,CAAC;gBACF,IAAI,CAAC,OAAO,CAAC,UAAU,GAAG,IAAI,CAAC;YACjC,CAAC;YAAC,OAAO,KAAK,EAAE,CAAC;gBACf,IAAI,CAAC,OAAO,CAAC,MAAM;oBACjB,qCAAqC,MAAM,CAAC,KAAK,CAAC,EAAE,CAAC;gBACvD,MAAM,IAAI,KAAK,CAAC,IAAI,CAAC,OAAO,CAAC,MAAM,CAAC,CAAC;YACvC,CAAC;oBAAS,CAAC;gBACT,IAAI,CAAC,cAAc,CAAC,WAAW,CAAC,CAAC;gBACjC,IAAI,CAAC,cAAc,CAAC,KAAK,CAAC,CAAC;YAC7B,CAAC;YACD,IAAI,CAAC,OAAO,CAAC,SAAS,GAAG,IAAI,CAAC;QAChC,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,IAAI,CAAC,eAAe,EAAE,CAAC;YACvB,MAAM,KAAK,CAAC;QACd,CAAC;IACH,CAAC;IAEO,SAAS,CACf,OAA2B,EAC3B,MAAkB;QAElB,OAAO,OAAO,CAAC,QAAQ,CACrB;YACE,QAAQ,EAAE,SAAS;YACnB,KAAK,EAAE,CAAC,GAAG,MAAM,CAAC,IAAI,CAAC,IAAI,CAAC;SAC7B,EACD,WAAW,CAAC,MAAM,EAAE,IAAI,CAAC,SAAS,CAAC,CACpC,CAAC;IACJ,CAAC;IAEO,KAAK,CAAC,gBAAgB,CAAC,KAAa,EAAE,MAAc;QAC1D,IAAI,IAAI,CAAC,SAAS;YAAE,MAAM,IAAI,KAAK,CAAC,iCAAiC,CAAC,CAAC;QACvE,MAAM,OAAO,GAAG,IAAI,IAAI,CAAC,mBAAmB,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC;QAC5D,MAAM,MAAM,GAAG,IAAI,GAAG,EAAyB,CAAC;QAChD,MAAM,MAAM,GAAG,IAAI,GAAG,EAAoC,CAAC;QAC3D,MAAM,CAAC,GAAG,CACR,IAAI,CAAC,MAAM,CAAC,IAAI,CAAC,KAAK,EACtB,OAAO,CAAC,KAAK,CAAC,OAAO,EAAE;YACrB,QAAQ,EAAE,SAAS;YACnB,KAAK,EAAE,CAAC,CAAC,EAAE,IAAI,CAAC,MAAM,CAAC,aAAa,EAAE,MAAM,EAAE,KAAK,CAAC;SACrD,CAAC,CACH,CAAC;QACF,MAAM,CAAC,GAAG,CAAC,IAAI,CAAC,MAAM,CAAC,IAAI,CAAC,KAAK,EAAE,CAAC,IAAI,CAAC,MAAM,CAAC,aAAa,EAAE,MAAM,EAAE,KAAK,CAAC,CAAC,CAAC;QAE/E,KAAK,MAAM,IAAI,IAAI,IAAI,CAAC,MAAM,CAAC,IAAI,CAAC,KAAK,EAAE,CAAC;YAC1C,IAAI,MAAqB,CAAC;YAC1B,IAAI,KAA+B,CAAC;YACpC,IAAI,IAAI,CAAC,EAAE,KAAK,QAAQ,EAAE,CAAC;gBACzB,MAAM,UAAU,GAAG,MAAM,CAAC,GAAG,CAAC,IAAI,CAAC,KAAK,CAAE,CAAC;gBAC3C,MAAM,OAAO,GAAG,IAAI,CAAC,MAAM,CAAC,WAAW,CAAC,GAAG,CAAC,IAAI,CAAC,EAAE,CAAE,CAAC;gBACtD,MAAM,WAAW,GAAG,OAAO,CAAC,MAAM,CAChC,MAAM,CAAC,GAAG,CAAC,IAAI,CAAC,KAAK,CAAE,EACvB,IAAI,CAAC,SAAS,CAAC,OAAO,EAAE,OAAO,CAAC,MAAM,CAAC,EACvC;oBACE,IAAI,EAAE,IAAI,CAAC,SAAS,CAAC,OAAO,EAAE,OAAO,CAAC,IAAI,CAAC;oBAC3C,OAAO,EAAE,CAAC,CAAC,EAAE,CAAC,EAAE,CAAC,EAAE,CAAC,CAAC;oBACrB,WAAW,EAAE,MAAM;oBACnB,YAAY,EAAE,MAAM;iBACrB,CACF,CAAC;gBACF,MAAM,GAAG,UAAU,CAAC,OAAO,EAAE,WAAW,EAAE,IAAI,CAAC,UAAU,CAAC,CAAC;gBAC3D,KAAK,GAAG,CAAC,OAAO,CAAC,cAAc,EAAE,UAAU,CAAC,CAAC,CAAC,EAAE,UAAU,CAAC,CAAC,CAAC,CAAC,CAAC;YACjE,CAAC;iBAAM,IAAI,IAAI,CAAC,EAAE,KAAK,WAAW,EAAE,CAAC;gBACnC,MAAM,UAAU,GAAG,MAAM,CAAC,GAAG,CAAC,IAAI,CAAC,KAAK,CAAE,CAAC;gBAC3C,MAAM,GAAG,OAAO,CAAC,SAAS,CAAC,MAAM,CAAC,GAAG,CAAC,IAAI,CAAC,KAAK,CAAE,EAAE;oBAClD,gBAAgB,EAAE,CAAC,CAAC,EAAE,CAAC,CAAC;oBACxB,OAAO,EAAE,CAAC,CAAC,EAAE,CAAC,CAAC;oBACf,OAAO,EAAE,CAAC,CAAC,EAAE,UAAU,CAAC,CAAC,CAAC,GAAG,CAAC,EAAE,CAAC,EAAE,UAAU,CAAC,CAAC,CAAC,GAAG,CAAC,CAAC;oBACrD,MAAM,EAAE,MAAM;iBACf,CAAC,CAAC;gBACH,KAAK,GAAG;oBACN,UAAU,CAAC,CAAC,CAAC;oBACb,IAAI,CAAC,IAAI,CAAC,UAAU,CAAC,CAAC,CAAC,GAAG,CAAC,CAAC;oBAC5B,IAAI,CAAC,IAAI,CAAC,UAAU,CAAC,CAAC,CAAC,GAAG,CAAC,CAAC;iBAC7B,CAAC;YACJ,CAAC;iBAAM,IAAI,IAAI,CAAC,EAAE,KAAK,YAAY,EAAE,CAAC;gBACpC,MAAM,UAAU,GAAG,MAAM,CAAC,GAAG,CAAC,IAAI,CAAC,KAAK,CAAE,CAAC;gBAC3C,MAAM,GAAG,OAAO,CAAC,UAAU,CAAC,MAAM,CAAC,GAAG,CAAC,IAAI,CAAC,KAAK,CAAE,EAAE;oBACnD,IAAI,EAAE,kBAAkB;oBACxB,IAAI,EAAE,CAAC,CAAC,EAAE,CAAC,CAAC;oBACZ,MAAM,EAAE,CAAC,CAAC,EAAE,CAAC,CAAC;iBACf,CAAC,CAAC;gBACH,KAAK,GAAG,CAAC,UAAU,CAAC,CAAC,CAAC,EAAE,UAAU,CAAC,CAAC,CAAC,GAAG,CAAC,EAAE,UAAU,CAAC,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC;YAChE,CAAC;iBAAM,CAAC;gBACN,MAAM,WAAW,GAAG,IAAI,CAAC,MAAM,CAAC,GAAG,CAAC,CAAC,KAAK,EAAE,EAAE,CAAC,MAAM,CAAC,GAAG,CAAC,KAAK,CAAE,CAAC,CAAC;gBACnE,IACE,WAAW,CAAC,IAAI,CACd,CAAC,SAAS,EAAE,EAAE,CACZ,SAAS,CAAC,CAAC,CAAC,KAAK,WAAW,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC;oBAClC,SAAS,CAAC,CAAC,CAAC,KAAK,WAAW,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CACrC,EACD,CAAC;oBACD,MAAM,IAAI,KAAK,CACb,gBAAgB,IAAI,CAAC,EAAE,gCAAgC,CACxD,CAAC;gBACJ,CAAC;gBACD,MAAM,GAAG,OAAO,CAAC,MAAM,CACrB,IAAI,CAAC,MAAM,CAAC,GAAG,CAAC,CAAC,KAAK,EAAE,EAAE,CAAC,MAAM,CAAC,GAAG,CAAC,KAAK,CAAE,CAAC,EAC9C,CAAC,CACF,CAAC;gBACF,KAAK,GAAG;oBACN,WAAW,CAAC,MAAM,CAAC,CAAC,GAAG,EAAE,SAAS,EAAE,EAAE,CAAC,GAAG,GAAG,SAAS,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC;oBAC7D,WAAW,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC;oBACjB,WAAW,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC;iBAClB,CAAC;YACJ,CAAC;YACD,MAAM,CAAC,GAAG,CAAC,IAAI,CAAC,EAAE,EAAE,MAAM,CAAC,CAAC;YAC5B,MAAM,CAAC,GAAG,CAAC,IAAI,CAAC,EAAE,EAAE,KAAK,CAAC,CAAC;QAC7B,CAAC;QAED,IAAI,KAA8B,CAAC;QACnC,IAAI,WAAqC,CAAC;QAC1C,IAAI,YAAsC,CAAC;QAC3C,IAAI,YAAmC,CAAC;QACxC,IAAI,YAAmC,CAAC;QACxC,IAAI,aAAoC,CAAC;QACzC,IAAI,CAAC;YACH,KAAK,GAAG,IAAI,CAAC,UAAU,CAAC,KAAK,CAC3B,UAAU,EACV,MAAM,OAAO,CAAC,KAAK,CAAC;gBAClB,MAAM,EAAE,MAAM,CAAC,GAAG,CAAC,IAAI,CAAC,MAAM,CAAC,IAAI,CAAC,MAAM,CAAE;aAC7C,CAAC,CACH,CAAC;YACF,WAAW,GAAG,IAAI,CAAC,UAAU,CAAC,KAAK,CACjC,WAAW,EACX,MAAM,IAAI,CAAC,QAAQ,CAAC,sBAAsB,CACxC;gBACE,QAAQ,EAAE,SAAS;gBACnB,KAAK,EAAE,CAAC,CAAC,EAAE,IAAI,CAAC,MAAM,CAAC,aAAa,EAAE,MAAM,EAAE,KAAK,CAAC;gBACpD,QAAQ,EAAE,IAAI;aACf,EACD,IAAI,CAAC,OAAO,CACb,CACF,CAAC;YACF,YAAY,GAAG,IAAI,CAAC,UAAU,CAAC,KAAK,CAClC,WAAW,EACX,MAAM,IAAI,CAAC,QAAQ,CAAC,sBAAsB,CACxC;gBACE,QAAQ,EAAE,SAAS;gBACnB,KAAK,EAAE,CAAC,CAAC,EAAE,IAAI,CAAC,MAAM,CAAC,cAAc,EAAE,MAAM,EAAE,KAAK,CAAC;gBACrD,QAAQ,EAAE,IAAI;aACf,EACD,IAAI,CAAC,OAAO,CACb,CACF,CAAC;YACF,YAAY,GAAG,IAAI,CAAC,UAAU,CAAC,KAAK,CAClC,YAAY,EACZ,IAAI,CAAC,OAAO,CAAC,YAAY,CAAC;gBACxB,KAAK,EAAE,qBAAqB,KAAK,IAAI,MAAM,EAAE;gBAC7C,IAAI,EAAE,KAAK,GAAG,MAAM,GAAG,CAAC,GAAG,CAAC;gBAC5B,KAAK,EACH,cAAc,CAAC,OAAO;oBACtB,cAAc,CAAC,QAAQ;oBACvB,cAAc,CAAC,QAAQ;aAC1B,CAAC,CACH,CAAC;YACF,YAAY,GAAG,IAAI,CAAC,UAAU,CAAC,KAAK,CAClC,YAAY,EACZ,aAAa,CACX,IAAI,CAAC,OAAO,EACZ,oBAAoB,KAAK,IAAI,MAAM,EAAE,EACrC,CAAC,KAAK,EAAE,MAAM,EAAE,IAAI,CAAC,MAAM,CAAC,aAAa,CAAC,CAC3C,CACF,CAAC;YACF,aAAa,GAAG,IAAI,CAAC,UAAU,CAAC,KAAK,CACnC,YAAY,EACZ,aAAa,CACX,IAAI,CAAC,OAAO,EACZ,qBAAqB,KAAK,IAAI,MAAM,EAAE,EACtC,CAAC,KAAK,EAAE,MAAM,CAAC,CAChB,CACF,CAAC;YACF,MAAM,SAAS,GAAwB;gBACrC,KAAK;gBACL,WAAW;gBACX,YAAY;gBACZ,YAAY;gBACZ,YAAY;gBACZ,aAAa;gBACb,KAAK;gBACL,MAAM;gBACN,QAAQ,EAAE,EAAE,IAAI,CAAC,MAAM;aACxB,CAAC;YACF,OAAO,SAAS,CAAC;QACnB,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,IAAI,CAAC,cAAc,CAAC,aAAa,CAAC,CAAC;YACnC,IAAI,CAAC,cAAc,CAAC,YAAY,CAAC,CAAC;YAClC,IAAI,CAAC,cAAc,CAAC,YAAY,CAAC,CAAC;YAClC,IAAI,CAAC,cAAc,CAAC,YAAY,CAAC,CAAC;YAClC,IAAI,CAAC,cAAc,CAAC,WAAW,CAAC,CAAC;YACjC,IAAI,CAAC,aAAa,CAAC,KAAK,CAAC,CAAC;YAC1B,MAAM,KAAK,CAAC;QACd,CAAC;IACH,CAAC;IAEO,KAAK,CAAC,UAAU,CAAC,KAAa,EAAE,MAAc;QACpD,MAAM,GAAG,GAAG,GAAG,KAAK,IAAI,MAAM,EAAE,CAAC;QACjC,IAAI,SAAS,GAAG,IAAI,CAAC,WAAW,CAAC,GAAG,CAAC,GAAG,CAAC,CAAC;QAC1C,IAAI,CAAC,SAAS,EAAE,CAAC;YACf,IAAI,OAAO,GAAG,IAAI,CAAC,cAAc,CAAC,GAAG,CAAC,GAAG,CAAC,CAAC;YAC3C,IAAI,CAAC,OAAO,EAAE,CAAC;gBACb,OAAO,GAAG,CAAC,KAAK,IAAI,EAAE;oBACpB,IAAI,CAAC,qBAAqB,EAAE,CAAC;oBAC7B,IAAI,CAAC;wBACH,OAAO,MAAM,IAAI,CAAC,gBAAgB,CAAC,KAAK,EAAE,MAAM,CAAC,CAAC;oBACpD,CAAC;4BAAS,CAAC;wBACT,IAAI,CAAC,qBAAqB,EAAE,CAAC;oBAC/B,CAAC;gBACH,CAAC,CAAC,EAAE,CAAC;gBACL,IAAI,CAAC,cAAc,CAAC,GAAG,CAAC,GAAG,EAAE,OAAO,CAAC,CAAC;YACxC,CAAC;YACD,IAAI,CAAC;gBACH,SAAS,GAAG,MAAM,OAAO,CAAC;gBAC1B,IAAI,IAAI,CAAC,SAAS,EAAE,CAAC;oBACnB,IAAI,CAAC,iBAAiB,CAAC,SAAS,CAAC,CAAC;oBAClC,MAAM,IAAI,KAAK,CAAC,iCAAiC,CAAC,CAAC;gBACrD,CAAC;gBACD,IAAI,CAAC,WAAW,CAAC,GAAG,CAAC,GAAG,EAAE,SAAS,CAAC,CAAC;YACvC,CAAC;oBAAS,CAAC;gBACT,IAAI,IAAI,CAAC,cAAc,CAAC,GAAG,CAAC,GAAG,CAAC,KAAK,OAAO,EAAE,CAAC;oBAC7C,IAAI,CAAC,cAAc,CAAC,MAAM,CAAC,GAAG,CAAC,CAAC;gBAClC,CAAC;YACH,CAAC;YACD,IAAI,IAAI,CAAC,WAAW,CAAC,IAAI,GAAG,IAAI,CAAC,eAAe,EAAE,CAAC;gBACjD,MAAM,MAAM,GAAG,CAAC,GAAG,IAAI,CAAC,WAAW,CAAC,OAAO,EAAE,CAAC;qBAC3C,MAAM,CAAC,CAAC,CAAC,SAAS,CAAC,EAAE,EAAE,CAAC,SAAS,KAAK,GAAG,CAAC;qBAC1C,IAAI,CAAC,CAAC,IAAI,EAAE,KAAK,EAAE,EAAE,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,QAAQ,GAAG,KAAK,CAAC,CAAC,CAAC,CAAC,QAAQ,CAAC,CAAC,CAAC,CAAC,CAAC;gBAClE,IAAI,MAAM,EAAE,CAAC;oBACX,IAAI,CAAC,WAAW,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC,CAAC;oBACnC,IAAI,CAAC,gBAAgB,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC,CAAC;gBACnC,CAAC;YACH,CAAC;QACH,CAAC;QACD,SAAS,CAAC,QAAQ,GAAG,EAAE,IAAI,CAAC,MAAM,CAAC;QACnC,OAAO,SAAS,CAAC;IACnB,CAAC;IAED,8EAA8E;IAC9E,KAAK,CAAC,OAAO,CAAC,MAAoD;QAChE,KAAK,MAAM,KAAK,IAAI,MAAM,EAAE,CAAC;YAC3B,MAAM,IAAI,CAAC,UAAU,CAAC,KAAK,CAAC,KAAK,EAAE,KAAK,CAAC,MAAM,CAAC,CAAC;QACnD,CAAC;IACH,CAAC;IAED,KAAK,CAAC,OAAO,CACX,YAAkC,EAClC,KAAa,EACb,MAAc;QAEd,MAAM,WAAW,GAAG,IAAI,CAAC,MAAM,CAAC,aAAa,GAAG,CAAC,CAAC;QAClD,IAAI,YAAY,CAAC,MAAM,KAAK,WAAW,EAAE,CAAC;YACxC,MAAM,IAAI,KAAK,CACb,uBAAuB,WAAW,uBAAuB,YAAY,CAAC,MAAM,EAAE,CAC/E,CAAC;QACJ,CAAC;QACD,MAAM,SAAS,GAAG,MAAM,IAAI,CAAC,UAAU,CAAC,KAAK,EAAE,MAAM,CAAC,CAAC;QACvD,MAAM,cAAc,GAAG,IAAI,CAAC,UAAU,CAAC,KAAK,CAC1C,YAAY,EACZ,MAAM,IAAI,CAAC,QAAQ,CAAC,WAAW,CAAC,SAAS,CAAC,WAAW,CAAC,CACvD,CAAC;QACF,IAAI,CAAC;YACH,MAAM,YAAY,GAAwB,YAAY,CAAC,GAAG,CACxD,CAAC,MAAM,EAAE,OAAO,EAAE,EAAE,CAAC,CAAC,EAAE,OAAO,EAAE,QAAQ,EAAE,EAAE,MAAM,EAAE,EAAE,CAAC,CACzD,CAAC;YACF,YAAY,CAAC,IAAI,CAAC;gBAChB,OAAO,EAAE,WAAW;gBACpB,QAAQ,EAAE,EAAE,MAAM,EAAE,cAAc,EAAE;aACrC,CAAC,CAAC;YACH,YAAY,CAAC,IAAI,CAAC;gBAChB,OAAO,EAAE,WAAW,GAAG,CAAC;gBACxB,QAAQ,EAAE,EAAE,MAAM,EAAE,SAAS,CAAC,YAAY,EAAE;aAC7C,CAAC,CAAC;YACH,MAAM,cAAc,GAAG,IAAI,CAAC,OAAO,CAAC,eAAe,CAAC;gBAClD,KAAK,EAAE,2BAA2B;gBAClC,MAAM,EAAE,IAAI,CAAC,cAAc,CAAC,kBAAkB,CAAC,CAAC,CAAC;gBACjD,OAAO,EAAE,YAAY;aACtB,CAAC,CAAC;YACH,MAAM,YAAY,GAAG,IAAI,CAAC,OAAO,CAAC,oBAAoB,CAAC;gBACrD,KAAK,EAAE,uBAAuB;aAC/B,CAAC,CAAC;YACH,MAAM,SAAS,GAAG,YAAY,CAAC,gBAAgB,EAAE,CAAC;YAClD,SAAS,CAAC,WAAW,CAAC,IAAI,CAAC,cAAc,CAAC,CAAC;YAC3C,SAAS,CAAC,YAAY,CAAC,CAAC,EAAE,cAAc,CAAC,CAAC;YAC1C,SAAS,CAAC,kBAAkB,CAC1B,IAAI,CAAC,IAAI,CAAC,KAAK,GAAG,cAAc,CAAC,EACjC,IAAI,CAAC,IAAI,CAAC,MAAM,GAAG,cAAc,CAAC,EAClC,IAAI,CAAC,MAAM,CAAC,aAAa,CAC1B,CAAC;YACF,SAAS,CAAC,GAAG,EAAE,CAAC;YAChB,IAAI,CAAC,OAAO,CAAC,KAAK,CAAC,MAAM,CAAC,CAAC,YAAY,CAAC,MAAM,EAAE,CAAC,CAAC,CAAC;QACrD,CAAC;gBAAS,CAAC;YACT,IAAI,CAAC,cAAc,CAAC,cAAc,CAAC,CAAC;QACtC,CAAC;QAED,IAAI,CAAC,QAAQ,CAAC,QAAQ,CACpB,SAAS,CAAC,KAAK,EACf,EAAE,KAAK,EAAE,SAAS,CAAC,WAAW,EAAE,EAChC,EAAE,MAAM,EAAE,SAAS,CAAC,YAAY,EAAE,CACnC,CAAC;QACF,MAAM,eAAe,GAAG,IAAI,CAAC,UAAU,CAAC,KAAK,CAC3C,YAAY,EACZ,MAAM,IAAI,CAAC,QAAQ,CAAC,WAAW,CAAC,SAAS,CAAC,YAAY,CAAC,CACxD,CAAC;QACF,IAAI,CAAC;YACH,MAAM,eAAe,GAAG,IAAI,CAAC,OAAO,CAAC,eAAe,CAAC;gBACnD,KAAK,EAAE,4BAA4B;gBACnC,MAAM,EAAE,IAAI,CAAC,eAAe,CAAC,kBAAkB,CAAC,CAAC,CAAC;gBAClD,OAAO,EAAE;oBACP,EAAE,OAAO,EAAE,CAAC,EAAE,QAAQ,EAAE,EAAE,MAAM,EAAE,eAAe,EAAE,EAAE;oBACrD,EAAE,OAAO,EAAE,CAAC,EAAE,QAAQ,EAAE,EAAE,MAAM,EAAE,SAAS,CAAC,YAAY,EAAE,EAAE;oBAC5D,EAAE,OAAO,EAAE,CAAC,EAAE,QAAQ,EAAE,EAAE,MAAM,EAAE,SAAS,CAAC,aAAa,EAAE,EAAE;iBAC9D;aACF,CAAC,CAAC;YACH,MAAM,aAAa,GAAG,IAAI,CAAC,OAAO,CAAC,oBAAoB,CAAC;gBACtD,KAAK,EAAE,0BAA0B;aAClC,CAAC,CAAC;YACH,MAAM,UAAU,GAAG,aAAa,CAAC,gBAAgB,EAAE,CAAC;YACpD,UAAU,CAAC,WAAW,CAAC,IAAI,CAAC,eAAe,CAAC,CAAC;YAC7C,UAAU,CAAC,YAAY,CAAC,CAAC,EAAE,eAAe,CAAC,CAAC;YAC5C,UAAU,CAAC,kBAAkB,CAC3B,IAAI,CAAC,IAAI,CAAC,KAAK,GAAG,cAAc,CAAC,EACjC,IAAI,CAAC,IAAI,CAAC,MAAM,GAAG,cAAc,CAAC,CACnC,CAAC;YACF,UAAU,CAAC,GAAG,EAAE,CAAC;YACjB,IAAI,CAAC,OAAO,CAAC,KAAK,CAAC,MAAM,CAAC,CAAC,aAAa,CAAC,MAAM,EAAE,CAAC,CAAC,CAAC;QACtD,CAAC;gBAAS,CAAC;YACT,IAAI,CAAC,cAAc,CAAC,eAAe,CAAC,CAAC;QACvC,CAAC;QACD,OAAO,SAAS,CAAC,YAAY,CAAC;IAChC,CAAC;IAED,KAAK,CAAC,UAAU,CAAC,KAAmB,EAAE,KAAa,EAAE,MAAc;QACjE,MAAM,SAAS,GAAG,MAAM,IAAI,CAAC,UAAU,CAAC,KAAK,EAAE,MAAM,CAAC,CAAC;QACvD,MAAM,KAAK,GAAG,KAAK,GAAG,MAAM,CAAC;QAC7B,MAAM,MAAM,GAAG,IAAI,YAAY,CAAC,KAAK,GAAG,IAAI,CAAC,MAAM,CAAC,aAAa,CAAC,CAAC;QACnE,KAAK,IAAI,KAAK,GAAG,CAAC,EAAE,KAAK,GAAG,KAAK,EAAE,KAAK,EAAE,EAAE,CAAC;YAC3C,KAAK,IAAI,OAAO,GAAG,CAAC,EAAE,OAAO,GAAG,IAAI,CAAC,MAAM,CAAC,aAAa,EAAE,OAAO,EAAE,EAAE,CAAC;gBACrE,MAAM,CAAC,OAAO,GAAG,KAAK,GAAG,KAAK,CAAC;oBAC7B,KAAK,CAAC,KAAK,GAAG,IAAI,CAAC,MAAM,CAAC,aAAa,GAAG,OAAO,CAAC,CAAC;YACvD,CAAC;QACH,CAAC;QACD,IAAI,CAAC,QAAQ,CAAC,WAAW,CAAC,SAAS,CAAC,WAAW,EAAE,MAAM,CAAC,CAAC;QACzD,IAAI,CAAC,QAAQ,CAAC,QAAQ,CACpB,SAAS,CAAC,KAAK,EACf,EAAE,KAAK,EAAE,SAAS,CAAC,WAAW,EAAE,EAChC,EAAE,MAAM,EAAE,SAAS,CAAC,YAAY,EAAE,CACnC,CAAC;QACF,MAAM,MAAM,GAAG,IAAI,YAAY,CAC7B,MAAM,IAAI,CAAC,QAAQ,CAAC,UAAU,CAAC,SAAS,CAAC,YAAY,CAAC,CACvD,CAAC;QACF,MAAM,QAAQ,GAAG,IAAI,YAAY,CAAC,KAAK,GAAG,IAAI,CAAC,MAAM,CAAC,cAAc,CAAC,CAAC;QACtE,KAAK,IAAI,KAAK,GAAG,CAAC,EAAE,KAAK,GAAG,KAAK,EAAE,KAAK,EAAE,EAAE,CAAC;YAC3C,KAAK,IAAI,OAAO,GAAG,CAAC,EAAE,OAAO,GAAG,IAAI,CAAC,MAAM,CAAC,cAAc,EAAE,OAAO,EAAE,EAAE,CAAC;gBACtE,QAAQ,CAAC,KAAK,GAAG,IAAI,CAAC,MAAM,CAAC,cAAc,GAAG,OAAO,CAAC;oBACpD,MAAM,CAAC,OAAO,GAAG,KAAK,GAAG,KAAK,CAAC,CAAC;YACpC,CAAC;QACH,CAAC;QACD,OAAO,QAAQ,CAAC;IAClB,CAAC;IAEO,iBAAiB,CAAC,SAA8B;QACtD,IAAI,CAAC,aAAa,CAAC,SAAS,CAAC,KAAK,CAAC,CAAC;QACpC,IAAI,CAAC,cAAc,CAAC,SAAS,CAAC,WAAW,CAAC,CAAC;QAC3C,IAAI,CAAC,cAAc,CAAC,SAAS,CAAC,YAAY,CAAC,CAAC;QAC5C,IAAI,CAAC,cAAc,CAAC,SAAS,CAAC,YAAY,CAAC,CAAC;QAC5C,IAAI,CAAC,cAAc,CAAC,SAAS,CAAC,YAAY,CAAC,CAAC;QAC5C,IAAI,CAAC,cAAc,CAAC,SAAS,CAAC,aAAa,CAAC,CAAC;IAC/C,CAAC;IAEO,gBAAgB,CAAC,SAA8B;QACrD,IAAI,CAAC,kBAAkB,CAAC,GAAG,CAAC,SAAS,CAAC,CAAC;QACvC,KAAK,IAAI,CAAC,OAAO,CAAC,KAAK,CAAC,mBAAmB,EAAE,CAAC,KAAK,CAAC,GAAG,EAAE,CAAC,SAAS,CAAC,CAAC,IAAI,CAAC,GAAG,EAAE;YAC7E,IAAI,CAAC,kBAAkB,CAAC,MAAM,CAAC,SAAS,CAAC,CAAC;YAC1C,IAAI,CAAC,iBAAiB,CAAC,SAAS,CAAC,CAAC;QACpC,CAAC,CAAC,CAAC;IACL,CAAC;IAEO,cAAc,CAAC,MAA6B;QAClD,IAAI,CAAC,UAAU,CAAC,OAAO,CAAC,YAAY,EAAE,MAAM,EAAE,GAAG,EAAE,CAAC,MAAO,CAAC,OAAO,EAAE,CAAC,CAAC;IACzE,CAAC;IAEO,cAAc,CAAC,MAAgC;QACrD,IAAI,CAAC,UAAU,CAAC,OAAO,CAAC,WAAW,EAAE,MAAM,EAAE,GAAG,EAAE,CAAC,MAAO,CAAC,OAAO,EAAE,CAAC,CAAC;IACxE,CAAC;IAEO,aAAa,CAAC,KAA8B;QAClD,IAAI,CAAC,UAAU,CAAC,OAAO,CAAC,UAAU,EAAE,KAAK,EAAE,GAAG,EAAE,CAAC,KAAM,CAAC,OAAO,EAAE,EAAE,CAAC,CAAC;IACvE,CAAC;IAEO,eAAe;QACrB,IAAI,CAAC,UAAU,CAAC,OAAO,CACrB,YAAY,EACZ,IAAI,CAAC,QAAQ,EACb,GAAG,EAAE,CAAC,IAAI,CAAC,QAAQ,CAAC,OAAO,EAAE,EAAE,CAChC,CAAC;IACJ,CAAC;IAED,eAAe;QACb,OAAO,IAAI,CAAC,UAAU,CAAC,QAAQ,CAC7B,IAAI,CAAC,qBAAqB,GAAG,IAAI,CAAC,kBAAkB,CAAC,IAAI,CAC1D,CAAC;IACJ,CAAC;IAED,OAAO;QACL,IAAI,IAAI,CAAC,SAAS;YAAE,OAAO;QAC3B,IAAI,CAAC,SAAS,GAAG,IAAI,CAAC;QACtB,KAAK,MAAM,SAAS,IAAI,IAAI,CAAC,WAAW,CAAC,MAAM,EAAE,EAAE,CAAC;YAClD,IAAI,CAAC,iBAAiB,CAAC,SAAS,CAAC,CAAC;QACpC,CAAC;QACD,IAAI,CAAC,WAAW,CAAC,KAAK,EAAE,CAAC;QACzB,KAAK,MAAM,SAAS,IAAI,IAAI,CAAC,kBAAkB,EAAE,CAAC;YAChD,IAAI,CAAC,iBAAiB,CAAC,SAAS,CAAC,CAAC;QACpC,CAAC;QACD,IAAI,CAAC,kBAAkB,CAAC,KAAK,EAAE,CAAC;QAChC,IAAI,CAAC,cAAc,CAAC,KAAK,EAAE,CAAC;QAC5B,IAAI,CAAC,eAAe,EAAE,CAAC;IACzB,CAAC;CACF"}

package/package.json

@@ -1,26 +1,37 @@
 {
   "name": "oidn-web",
-  "version": "0.3.5",
+  "version": "0.5.0",
   "description": "Run Open Image Denoiser on the Web",
   "main": "lib/main.js",
   "types": "lib/main.d.ts",
+  "files": [
+    "lib",
+    "dist",
+    "src",
+    "CHANGELOG.md"
+  ],
   "type": "module",
   "scripts": {
     "build": "vite build",
     "dev": "vite dev",
     "watch": "vite build -w",
+    "test": "npm run build:lib && npm run build && node --test tests/*.test.mjs",
+    "test:leak": "node benchmarks/leak.mjs",
+    "benchmark": "node benchmarks/compare.mjs",
+    "model:inspect": "npm run build:lib && node scripts/inspect-model.mjs",
+    "prepare": "npm run build:lib",
     "prepublishOnly": "npm run build && npm run build:lib",
-    "build:lib": "tsc",
+    "build:lib": "node -e \"require('node:fs').rmSync('lib',{recursive:true,force:true})\" && tsc",
     "watch:lib": "tsc --watch"
   },
   "author": "pissang",
   "license": "MIT",
   "dependencies": {
-    "@petamoriken/float16": "^3.8.7",
-    "@tensorflow/tfjs": "4.20.0",
-    "@tensorflow/tfjs-backend-webgpu": "4.20.0"
+    "@petamoriken/float16": "^3.8.7"
   },
   "devDependencies": {
+    "@webgpu/types": "^0.1.40",
+    "playwright-core": "^1.62.1",
     "typescript": "^5.7.3",
     "vite": "^5.2.13",
     "vite-plugin-mkcert": "^1.17.5"