oidn-webgpu 0.0.0-stage → 0.6.0

package/lib/native/recordingDevice.d.ts

@@ -0,0 +1,125 @@
+import type { OIDNGPUAdapterInfo, OIDNGPUBindGroup, OIDNGPUBindGroupEntry, OIDNGPUBindGroupLayout, OIDNGPUBuffer, OIDNGPUCommandBuffer, OIDNGPUCommandEncoder, OIDNGPUComputePipeline, OIDNGPUDevice, OIDNGPUFeatures, OIDNGPULimits, OIDNGPUQuerySet, OIDNGPUQueue, OIDNGPUShaderModule } from '../gpuTypes.js';
+/**
+ * An OIDNGPUDevice that executes nothing: it records every resource and queue
+ * operation so that a native host can create the same resources once and
+ * replay the recorded commands with its own GPU API.
+ */
+export declare class RecordedBuffer implements OIDNGPUBuffer {
+    readonly id: number;
+    readonly size: number;
+    readonly usage: number;
+    readonly label: string;
+    mapState: string;
+    data?: Uint8Array;
+    destroyed: boolean;
+    constructor(id: number, size: number, usage: number, label: string, mapped: boolean);
+    getMappedRange(): ArrayBuffer;
+    unmap(): void;
+    destroy(): void;
+}
+export declare class RecordedShaderModule implements OIDNGPUShaderModule {
+    readonly id: number;
+    readonly label: string;
+    readonly code: string;
+    constructor(id: number, label: string, code: string);
+}
+export declare class RecordedBindGroupLayout implements OIDNGPUBindGroupLayout {
+    readonly label: string;
+    readonly pipeline: number;
+    readonly group: number;
+    constructor(label: string, pipeline: number, group: number);
+}
+export declare class RecordedPipeline implements OIDNGPUComputePipeline {
+    readonly id: number;
+    readonly label: string;
+    readonly shader: number;
+    readonly entryPoint: string;
+    constructor(id: number, label: string, shader: number, entryPoint: string);
+    getBindGroupLayout(index: number): OIDNGPUBindGroupLayout;
+}
+export interface RecordedBinding {
+    binding: number;
+    buffer: number;
+    offset: number;
+    /** Zero binds the rest of the buffer. */
+    size: number;
+}
+export declare class RecordedBindGroup implements OIDNGPUBindGroup {
+    readonly id: number;
+    readonly label: string;
+    readonly pipeline: number;
+    readonly group: number;
+    readonly entries: RecordedBinding[];
+    constructor(id: number, label: string, pipeline: number, group: number, entries: RecordedBinding[]);
+}
+export interface RecordedCommand {
+    /** 0 = dispatch, 1 = buffer copy. */
+    kind: number;
+    pipeline: number;
+    bindGroup: number;
+    x: number;
+    y: number;
+    z: number;
+    source: number;
+    sourceOffset: number;
+    destination: number;
+    destinationOffset: number;
+    size: number;
+}
+export declare class RecordedCommandBuffer implements OIDNGPUCommandBuffer {
+    readonly label: string;
+    readonly commands: RecordedCommand[];
+    constructor(label: string, commands: RecordedCommand[]);
+}
+export interface RecordedQueueOperation {
+    /** 0 = submit, 1 = buffer write. */
+    kind: number;
+    commands: RecordedCommand[];
+    buffer: number;
+    offset: number;
+    data?: Uint8Array;
+}
+export interface RecordingDeviceDescriptor {
+    features: string[];
+    limits: OIDNGPULimits;
+    adapterInfo: OIDNGPUAdapterInfo;
+}
+export declare class RecordingDevice implements OIDNGPUDevice {
+    readonly features: OIDNGPUFeatures;
+    readonly limits: OIDNGPULimits;
+    readonly adapterInfo: OIDNGPUAdapterInfo;
+    readonly queue: OIDNGPUQueue;
+    readonly buffers: RecordedBuffer[];
+    readonly shaders: RecordedShaderModule[];
+    readonly pipelines: RecordedPipeline[];
+    readonly bindGroups: RecordedBindGroup[];
+    operations: RecordedQueueOperation[];
+    constructor(descriptor: RecordingDeviceDescriptor);
+    createBuffer(descriptor: {
+        label?: string;
+        size: number;
+        usage: number;
+        mappedAtCreation?: boolean;
+    }): OIDNGPUBuffer;
+    createShaderModule(descriptor: {
+        label?: string;
+        code: string;
+    }): OIDNGPUShaderModule;
+    createComputePipeline(descriptor: {
+        label?: string;
+        layout: 'auto';
+        compute: {
+            module: OIDNGPUShaderModule;
+            entryPoint: string;
+        };
+    }): OIDNGPUComputePipeline;
+    createBindGroup(descriptor: {
+        label?: string;
+        layout: OIDNGPUBindGroupLayout;
+        entries: OIDNGPUBindGroupEntry[];
+    }): OIDNGPUBindGroup;
+    createCommandEncoder(descriptor?: {
+        label?: string;
+    }): OIDNGPUCommandEncoder;
+    createQuerySet(): OIDNGPUQuerySet;
+}

package/lib/native/recordingDevice.js

@@ -0,0 +1,247 @@
+/**
+ * An OIDNGPUDevice that executes nothing: it records every resource and queue
+ * operation so that a native host can create the same resources once and
+ * replay the recorded commands with its own GPU API.
+ */
+export class RecordedBuffer {
+    id;
+    size;
+    usage;
+    label;
+    mapState = 'unmapped';
+    data;
+    destroyed = false;
+    constructor(id, size, usage, label, mapped) {
+        this.id = id;
+        this.size = size;
+        this.usage = usage;
+        this.label = label;
+        if (mapped) {
+            this.data = new Uint8Array(size);
+            this.mapState = 'mapped';
+        }
+    }
+    getMappedRange() {
+        if (!this.data)
+            throw new Error(`Recorded buffer ${this.label} is not mapped`);
+        return this.data.buffer;
+    }
+    unmap() {
+        this.mapState = 'unmapped';
+    }
+    destroy() {
+        this.destroyed = true;
+    }
+}
+export class RecordedShaderModule {
+    id;
+    label;
+    code;
+    constructor(id, label, code) {
+        this.id = id;
+        this.label = label;
+        this.code = code;
+    }
+}
+export class RecordedBindGroupLayout {
+    label;
+    pipeline;
+    group;
+    constructor(label, pipeline, group) {
+        this.label = label;
+        this.pipeline = pipeline;
+        this.group = group;
+    }
+}
+export class RecordedPipeline {
+    id;
+    label;
+    shader;
+    entryPoint;
+    constructor(id, label, shader, entryPoint) {
+        this.id = id;
+        this.label = label;
+        this.shader = shader;
+        this.entryPoint = entryPoint;
+    }
+    getBindGroupLayout(index) {
+        return new RecordedBindGroupLayout(this.label, this.id, index);
+    }
+}
+export class RecordedBindGroup {
+    id;
+    label;
+    pipeline;
+    group;
+    entries;
+    constructor(id, label, pipeline, group, entries) {
+        this.id = id;
+        this.label = label;
+        this.pipeline = pipeline;
+        this.group = group;
+        this.entries = entries;
+    }
+}
+export class RecordedCommandBuffer {
+    label;
+    commands;
+    constructor(label, commands) {
+        this.label = label;
+        this.commands = commands;
+    }
+}
+function command(kind) {
+    return {
+        kind,
+        pipeline: -1,
+        bindGroup: -1,
+        x: 0,
+        y: 0,
+        z: 0,
+        source: -1,
+        sourceOffset: 0,
+        destination: -1,
+        destinationOffset: 0,
+        size: 0
+    };
+}
+class RecordingPass {
+    _commands;
+    _pipeline = -1;
+    _bindGroup = -1;
+    constructor(_commands) {
+        this._commands = _commands;
+    }
+    setPipeline(pipeline) {
+        this._pipeline = pipeline.id;
+    }
+    setBindGroup(index, bindGroup) {
+        if (index !== 0)
+            throw new Error('Recorded passes support bind group 0 only');
+        this._bindGroup = bindGroup.id;
+    }
+    dispatchWorkgroups(x, y, z) {
+        const recorded = command(0);
+        recorded.pipeline = this._pipeline;
+        recorded.bindGroup = this._bindGroup;
+        recorded.x = x;
+        recorded.y = y ?? 1;
+        recorded.z = z ?? 1;
+        this._commands.push(recorded);
+    }
+    end() { }
+}
+class RecordingEncoder {
+    _label;
+    _commands = [];
+    constructor(_label) {
+        this._label = _label;
+    }
+    beginComputePass() {
+        return new RecordingPass(this._commands);
+    }
+    copyBufferToBuffer(source, sourceOffset, destination, destinationOffset, size) {
+        const recorded = command(1);
+        recorded.source = source.id;
+        recorded.sourceOffset = sourceOffset;
+        recorded.destination = destination.id;
+        recorded.destinationOffset = destinationOffset;
+        recorded.size = size;
+        this._commands.push(recorded);
+    }
+    resolveQuerySet() {
+        throw new Error('Recorded devices do not support timestamp queries');
+    }
+    finish() {
+        return new RecordedCommandBuffer(this._label, this._commands);
+    }
+}
+class RecordingQueue {
+    _device;
+    constructor(_device) {
+        this._device = _device;
+    }
+    submit(commandBuffers) {
+        const commands = [];
+        for (const commandBuffer of commandBuffers) {
+            for (const recorded of commandBuffer.commands) {
+                commands.push(recorded);
+            }
+        }
+        this._device.operations.push({ kind: 0, commands, buffer: -1, offset: 0 });
+    }
+    writeBuffer(buffer, bufferOffset, data) {
+        const bytes = new Uint8Array(data.byteLength);
+        bytes.set(data);
+        this._device.operations.push({
+            kind: 1,
+            commands: [],
+            buffer: buffer.id,
+            offset: bufferOffset,
+            data: bytes
+        });
+    }
+}
+class RecordedFeatures {
+    _names;
+    constructor(_names) {
+        this._names = _names;
+    }
+    has(name) {
+        return this._names.includes(name);
+    }
+}
+class RecordedQuerySet {
+    destroy() { }
+}
+export class RecordingDevice {
+    features;
+    limits;
+    adapterInfo;
+    queue;
+    buffers = [];
+    shaders = [];
+    pipelines = [];
+    bindGroups = [];
+    operations = [];
+    constructor(descriptor) {
+        this.features = new RecordedFeatures(descriptor.features);
+        this.limits = descriptor.limits;
+        this.adapterInfo = descriptor.adapterInfo;
+        this.queue = new RecordingQueue(this);
+    }
+    createBuffer(descriptor) {
+        const buffer = new RecordedBuffer(this.buffers.length, descriptor.size, descriptor.usage, descriptor.label ?? '', descriptor.mappedAtCreation === true);
+        this.buffers.push(buffer);
+        return buffer;
+    }
+    createShaderModule(descriptor) {
+        const module = new RecordedShaderModule(this.shaders.length, descriptor.label ?? '', descriptor.code);
+        this.shaders.push(module);
+        return module;
+    }
+    createComputePipeline(descriptor) {
+        const pipeline = new RecordedPipeline(this.pipelines.length, descriptor.label ?? '', descriptor.compute.module.id, descriptor.compute.entryPoint);
+        this.pipelines.push(pipeline);
+        return pipeline;
+    }
+    createBindGroup(descriptor) {
+        const layout = descriptor.layout;
+        const entries = descriptor.entries.map((entry) => ({
+            binding: entry.binding,
+            buffer: entry.resource.buffer.id,
+            offset: entry.resource.offset ?? 0,
+            size: entry.resource.size ?? 0
+        }));
+        const bindGroup = new RecordedBindGroup(this.bindGroups.length, descriptor.label ?? '', layout.pipeline, layout.group, entries);
+        this.bindGroups.push(bindGroup);
+        return bindGroup;
+    }
+    createCommandEncoder(descriptor) {
+        return new RecordingEncoder(descriptor?.label ?? '');
+    }
+    createQuerySet() {
+        return new RecordedQuerySet();
+    }
+}
+//# sourceMappingURL=recordingDevice.js.map

package/lib/native/recordingDevice.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"recordingDevice.js","sourceRoot":"","sources":["../../src/native/recordingDevice.ts"],"names":[],"mappings":"AAkBA;;;;GAIG;AAEH,MAAM,OAAO,cAAc;IAMd;IACA;IACA;IACA;IARX,QAAQ,GAAG,UAAU,CAAC;IACtB,IAAI,CAAc;IAClB,SAAS,GAAG,KAAK,CAAC;IAElB,YACW,EAAU,EACV,IAAY,EACZ,KAAa,EACb,KAAa,EACtB,MAAe;QAJN,OAAE,GAAF,EAAE,CAAQ;QACV,SAAI,GAAJ,IAAI,CAAQ;QACZ,UAAK,GAAL,KAAK,CAAQ;QACb,UAAK,GAAL,KAAK,CAAQ;QAGtB,IAAI,MAAM,EAAE,CAAC;YACX,IAAI,CAAC,IAAI,GAAG,IAAI,UAAU,CAAC,IAAI,CAAC,CAAC;YACjC,IAAI,CAAC,QAAQ,GAAG,QAAQ,CAAC;QAC3B,CAAC;IACH,CAAC;IAED,cAAc;QACZ,IAAI,CAAC,IAAI,CAAC,IAAI;YAAE,MAAM,IAAI,KAAK,CAAC,mBAAmB,IAAI,CAAC,KAAK,gBAAgB,CAAC,CAAC;QAC/E,OAAO,IAAI,CAAC,IAAI,CAAC,MAAqB,CAAC;IACzC,CAAC;IAED,KAAK;QACH,IAAI,CAAC,QAAQ,GAAG,UAAU,CAAC;IAC7B,CAAC;IAED,OAAO;QACL,IAAI,CAAC,SAAS,GAAG,IAAI,CAAC;IACxB,CAAC;CACF;AAED,MAAM,OAAO,oBAAoB;IACV;IAAqB;IAAwB;IAAlE,YAAqB,EAAU,EAAW,KAAa,EAAW,IAAY;QAAzD,OAAE,GAAF,EAAE,CAAQ;QAAW,UAAK,GAAL,KAAK,CAAQ;QAAW,SAAI,GAAJ,IAAI,CAAQ;IAAG,CAAC;CACnF;AAED,MAAM,OAAO,uBAAuB;IACb;IAAwB;IAA2B;IAAxE,YAAqB,KAAa,EAAW,QAAgB,EAAW,KAAa;QAAhE,UAAK,GAAL,KAAK,CAAQ;QAAW,aAAQ,GAAR,QAAQ,CAAQ;QAAW,UAAK,GAAL,KAAK,CAAQ;IAAG,CAAC;CAC1F;AAED,MAAM,OAAO,gBAAgB;IAEhB;IACA;IACA;IACA;IAJX,YACW,EAAU,EACV,KAAa,EACb,MAAc,EACd,UAAkB;QAHlB,OAAE,GAAF,EAAE,CAAQ;QACV,UAAK,GAAL,KAAK,CAAQ;QACb,WAAM,GAAN,MAAM,CAAQ;QACd,eAAU,GAAV,UAAU,CAAQ;IAC1B,CAAC;IAEJ,kBAAkB,CAAC,KAAa;QAC9B,OAAO,IAAI,uBAAuB,CAAC,IAAI,CAAC,KAAK,EAAE,IAAI,CAAC,EAAE,EAAE,KAAK,CAAC,CAAC;IACjE,CAAC;CACF;AAUD,MAAM,OAAO,iBAAiB;IAEjB;IACA;IACA;IACA;IACA;IALX,YACW,EAAU,EACV,KAAa,EACb,QAAgB,EAChB,KAAa,EACb,OAA0B;QAJ1B,OAAE,GAAF,EAAE,CAAQ;QACV,UAAK,GAAL,KAAK,CAAQ;QACb,aAAQ,GAAR,QAAQ,CAAQ;QAChB,UAAK,GAAL,KAAK,CAAQ;QACb,YAAO,GAAP,OAAO,CAAmB;IAClC,CAAC;CACL;AAiBD,MAAM,OAAO,qBAAqB;IACX;IAAwB;IAA7C,YAAqB,KAAa,EAAW,QAA2B;QAAnD,UAAK,GAAL,KAAK,CAAQ;QAAW,aAAQ,GAAR,QAAQ,CAAmB;IAAG,CAAC;CAC7E;AAWD,SAAS,OAAO,CAAC,IAAY;IAC3B,OAAO;QACL,IAAI;QACJ,QAAQ,EAAE,CAAC,CAAC;QACZ,SAAS,EAAE,CAAC,CAAC;QACb,CAAC,EAAE,CAAC;QACJ,CAAC,EAAE,CAAC;QACJ,CAAC,EAAE,CAAC;QACJ,MAAM,EAAE,CAAC,CAAC;QACV,YAAY,EAAE,CAAC;QACf,WAAW,EAAE,CAAC,CAAC;QACf,iBAAiB,EAAE,CAAC;QACpB,IAAI,EAAE,CAAC;KACR,CAAC;AACJ,CAAC;AAED,MAAM,aAAa;IAIG;IAHZ,SAAS,GAAG,CAAC,CAAC,CAAC;IACf,UAAU,GAAG,CAAC,CAAC,CAAC;IAExB,YAAoB,SAA4B;QAA5B,cAAS,GAAT,SAAS,CAAmB;IAAG,CAAC;IAEpD,WAAW,CAAC,QAAgC;QAC1C,IAAI,CAAC,SAAS,GAAI,QAA6B,CAAC,EAAE,CAAC;IACrD,CAAC;IAED,YAAY,CAAC,KAAa,EAAE,SAA2B;QACrD,IAAI,KAAK,KAAK,CAAC;YAAE,MAAM,IAAI,KAAK,CAAC,2CAA2C,CAAC,CAAC;QAC9E,IAAI,CAAC,UAAU,GAAI,SAA+B,CAAC,EAAE,CAAC;IACxD,CAAC;IAED,kBAAkB,CAAC,CAAS,EAAE,CAAU,EAAE,CAAU;QAClD,MAAM,QAAQ,GAAG,OAAO,CAAC,CAAC,CAAC,CAAC;QAC5B,QAAQ,CAAC,QAAQ,GAAG,IAAI,CAAC,SAAS,CAAC;QACnC,QAAQ,CAAC,SAAS,GAAG,IAAI,CAAC,UAAU,CAAC;QACrC,QAAQ,CAAC,CAAC,GAAG,CAAC,CAAC;QACf,QAAQ,CAAC,CAAC,GAAG,CAAC,IAAI,CAAC,CAAC;QACpB,QAAQ,CAAC,CAAC,GAAG,CAAC,IAAI,CAAC,CAAC;QACpB,IAAI,CAAC,SAAS,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC;IAChC,CAAC;IAED,GAAG,KAAI,CAAC;CACT;AAED,MAAM,gBAAgB;IAGA;IAFZ,SAAS,GAAsB,EAAE,CAAC;IAE1C,YAAoB,MAAc;QAAd,WAAM,GAAN,MAAM,CAAQ;IAAG,CAAC;IAEtC,gBAAgB;QACd,OAAO,IAAI,aAAa,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC;IAC3C,CAAC;IAED,kBAAkB,CAChB,MAAqB,EACrB,YAAoB,EACpB,WAA0B,EAC1B,iBAAyB,EACzB,IAAY;QAEZ,MAAM,QAAQ,GAAG,OAAO,CAAC,CAAC,CAAC,CAAC;QAC5B,QAAQ,CAAC,MAAM,GAAI,MAAyB,CAAC,EAAE,CAAC;QAChD,QAAQ,CAAC,YAAY,GAAG,YAAY,CAAC;QACrC,QAAQ,CAAC,WAAW,GAAI,WAA8B,CAAC,EAAE,CAAC;QAC1D,QAAQ,CAAC,iBAAiB,GAAG,iBAAiB,CAAC;QAC/C,QAAQ,CAAC,IAAI,GAAG,IAAI,CAAC;QACrB,IAAI,CAAC,SAAS,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC;IAChC,CAAC;IAED,eAAe;QACb,MAAM,IAAI,KAAK,CAAC,mDAAmD,CAAC,CAAC;IACvE,CAAC;IAED,MAAM;QACJ,OAAO,IAAI,qBAAqB,CAAC,IAAI,CAAC,MAAM,EAAE,IAAI,CAAC,SAAS,CAAC,CAAC;IAChE,CAAC;CACF;AAED,MAAM,cAAc;IACE;IAApB,YAAoB,OAAwB;QAAxB,YAAO,GAAP,OAAO,CAAiB;IAAG,CAAC;IAEhD,MAAM,CAAC,cAAsC;QAC3C,MAAM,QAAQ,GAAsB,EAAE,CAAC;QACvC,KAAK,MAAM,aAAa,IAAI,cAAc,EAAE,CAAC;YAC3C,KAAK,MAAM,QAAQ,IAAK,aAAuC,CAAC,QAAQ,EAAE,CAAC;gBACzE,QAAQ,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC;YAC1B,CAAC;QACH,CAAC;QACD,IAAI,CAAC,OAAO,CAAC,UAAU,CAAC,IAAI,CAAC,EAAE,IAAI,EAAE,CAAC,EAAE,QAAQ,EAAE,MAAM,EAAE,CAAC,CAAC,EAAE,MAAM,EAAE,CAAC,EAAE,CAAC,CAAC;IAC7E,CAAC;IAED,WAAW,CACT,MAAqB,EACrB,YAAoB,EACpB,IAAgB;QAEhB,MAAM,KAAK,GAAG,IAAI,UAAU,CAAC,IAAI,CAAC,UAAU,CAAC,CAAC;QAC9C,KAAK,CAAC,GAAG,CAAC,IAAI,CAAC,CAAC;QAChB,IAAI,CAAC,OAAO,CAAC,UAAU,CAAC,IAAI,CAAC;YAC3B,IAAI,EAAE,CAAC;YACP,QAAQ,EAAE,EAAE;YACZ,MAAM,EAAG,MAAyB,CAAC,EAAE;YACrC,MAAM,EAAE,YAAY;YACpB,IAAI,EAAE,KAAK;SACZ,CAAC,CAAC;IACL,CAAC;CAEF;AAED,MAAM,gBAAgB;IACA;IAApB,YAAoB,MAAgB;QAAhB,WAAM,GAAN,MAAM,CAAU;IAAG,CAAC;IAExC,GAAG,CAAC,IAAY;QACd,OAAO,IAAI,CAAC,MAAM,CAAC,QAAQ,CAAC,IAAI,CAAC,CAAC;IACpC,CAAC;CACF;AAED,MAAM,gBAAgB;IACpB,OAAO,KAAI,CAAC;CACb;AAQD,MAAM,OAAO,eAAe;IACjB,QAAQ,CAAkB;IAC1B,MAAM,CAAgB;IACtB,WAAW,CAAqB;IAChC,KAAK,CAAe;IACpB,OAAO,GAAqB,EAAE,CAAC;IAC/B,OAAO,GAA2B,EAAE,CAAC;IACrC,SAAS,GAAuB,EAAE,CAAC;IACnC,UAAU,GAAwB,EAAE,CAAC;IAC9C,UAAU,GAA6B,EAAE,CAAC;IAE1C,YAAY,UAAqC;QAC/C,IAAI,CAAC,QAAQ,GAAG,IAAI,gBAAgB,CAAC,UAAU,CAAC,QAAQ,CAAC,CAAC;QAC1D,IAAI,CAAC,MAAM,GAAG,UAAU,CAAC,MAAM,CAAC;QAChC,IAAI,CAAC,WAAW,GAAG,UAAU,CAAC,WAAW,CAAC;QAC1C,IAAI,CAAC,KAAK,GAAG,IAAI,cAAc,CAAC,IAAI,CAAC,CAAC;IACxC,CAAC;IAED,YAAY,CAAC,UAKZ;QACC,MAAM,MAAM,GAAG,IAAI,cAAc,CAC/B,IAAI,CAAC,OAAO,CAAC,MAAM,EACnB,UAAU,CAAC,IAAI,EACf,UAAU,CAAC,KAAK,EAChB,UAAU,CAAC,KAAK,IAAI,EAAE,EACtB,UAAU,CAAC,gBAAgB,KAAK,IAAI,CACrC,CAAC;QACF,IAAI,CAAC,OAAO,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC;QAC1B,OAAO,MAAM,CAAC;IAChB,CAAC;IAED,kBAAkB,CAAC,UAA4C;QAC7D,MAAM,MAAM,GAAG,IAAI,oBAAoB,CACrC,IAAI,CAAC,OAAO,CAAC,MAAM,EACnB,UAAU,CAAC,KAAK,IAAI,EAAE,EACtB,UAAU,CAAC,IAAI,CAChB,CAAC;QACF,IAAI,CAAC,OAAO,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC;QAC1B,OAAO,MAAM,CAAC;IAChB,CAAC;IAED,qBAAqB,CAAC,UAIrB;QACC,MAAM,QAAQ,GAAG,IAAI,gBAAgB,CACnC,IAAI,CAAC,SAAS,CAAC,MAAM,EACrB,UAAU,CAAC,KAAK,IAAI,EAAE,EACrB,UAAU,CAAC,OAAO,CAAC,MAA+B,CAAC,EAAE,EACtD,UAAU,CAAC,OAAO,CAAC,UAAU,CAC9B,CAAC;QACF,IAAI,CAAC,SAAS,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC;QAC9B,OAAO,QAAQ,CAAC;IAClB,CAAC;IAED,eAAe,CAAC,UAIf;QACC,MAAM,MAAM,GAAG,UAAU,CAAC,MAAiC,CAAC;QAC5D,MAAM,OAAO,GAAsB,UAAU,CAAC,OAAO,CAAC,GAAG,CAAC,CAAC,KAAK,EAAE,EAAE,CAAC,CAAC;YACpE,OAAO,EAAE,KAAK,CAAC,OAAO;YACtB,MAAM,EAAG,KAAK,CAAC,QAAQ,CAAC,MAAyB,CAAC,EAAE;YACpD,MAAM,EAAE,KAAK,CAAC,QAAQ,CAAC,MAAM,IAAI,CAAC;YAClC,IAAI,EAAE,KAAK,CAAC,QAAQ,CAAC,IAAI,IAAI,CAAC;SAC/B,CAAC,CAAC,CAAC;QACJ,MAAM,SAAS,GAAG,IAAI,iBAAiB,CACrC,IAAI,CAAC,UAAU,CAAC,MAAM,EACtB,UAAU,CAAC,KAAK,IAAI,EAAE,EACtB,MAAM,CAAC,QAAQ,EACf,MAAM,CAAC,KAAK,EACZ,OAAO,CACR,CAAC;QACF,IAAI,CAAC,UAAU,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC;QAChC,OAAO,SAAS,CAAC;IACnB,CAAC;IAED,oBAAoB,CAAC,UAA+B;QAClD,OAAO,IAAI,gBAAgB,CAAC,UAAU,EAAE,KAAK,IAAI,EAAE,CAAC,CAAC;IACvD,CAAC;IAED,cAAc;QACZ,OAAO,IAAI,gBAAgB,EAAE,CAAC;IAChC,CAAC;CACF"}

package/lib/nativeUNet.d.ts

@@ -0,0 +1,25 @@
+import { type OIDNGPUBuffer, type OIDNGPUQuerySet } from './gpuTypes.js';
+import type { ValidatedUNetModel } from './modelSpec';
+import { NativeUNetCore, type CachedExecution, type NativeSubgroupMatrixConfig, type NativeUNetExecutionProfile, type NativeUNetOptions, type NativeUNetPrecision, type NativeUNetPrecisionSetting } from './nativeUNetCore.js';
+export * from './nativeUNetCore.js';
+export declare function resolveNativeUNetPrecision(device: GPUDevice, requested?: NativeUNetPrecisionSetting): NativeUNetPrecision;
+export declare function selectSubgroupMatrixConfig(device: GPUDevice, accumulation?: 'f32' | 'f16'): NativeSubgroupMatrixConfig | undefined;
+/**
+ * Native, model-driven OIDN U-Net executor for browser WebGPU devices. The
+ * device must also provide the asynchronous WebGPU operations.
+ */
+export declare class NativeUNetExecutor extends NativeUNetCore {
+    private _lastExecutionProfile?;
+    constructor(device: GPUDevice, model: ValidatedUNetModel, options?: NativeUNetOptions);
+    execute(inputBuffers: readonly GPUBuffer[], width: number, height: number): GPUBuffer;
+    private get _asyncDevice();
+    private get _pipelinePromises();
+    getLastExecutionProfile(): Promise<NativeUNetExecutionProfile> | undefined;
+    private _pipelineAsync;
+    protected _retireExecution(execution: CachedExecution): void;
+    /** Compiles all shape-independent kernels before the model reports ready. */
+    prepare(): Promise<void>;
+    protected _readProfile(querySet: OIDNGPUQuerySet, resolveBuffer: OIDNGPUBuffer, readbackBuffer: OIDNGPUBuffer, labels: string[]): void;
+    /** Executes interleaved CPU image data through the native GPU runtime. */
+    executeCPU(interleavedInput: Float32Array, width: number, height: number): Promise<Float32Array>;
+}

package/lib/nativeUNet.js

@@ -0,0 +1,184 @@
+import { optimizeModelGraph } from './graphOptimizer.js';
+import { BufferUsage, MapMode } from './gpuTypes.js';
+import { createInputPackShader, NativeUNetCore, resolveNativeUNetPrecision as resolveCorePrecision, selectSubgroupMatrixConfig as selectCoreSubgroupMatrixConfig } from './nativeUNetCore.js';
+// The browser declarations below shadow the core's OIDNGPUDevice signatures.
+export * from './nativeUNetCore.js';
+// A GPUDevice provides every OIDNGPUAsyncDevice operation at runtime, but its
+// branded WebGPU types (GPUBindGroup, Iterable<GPUCommandBuffer>) are not
+// structurally assignable, so the browser API converts at this boundary.
+function asOIDNDevice(device) {
+    return device;
+}
+export function resolveNativeUNetPrecision(device, requested) {
+    return resolveCorePrecision(asOIDNDevice(device), requested);
+}
+export function selectSubgroupMatrixConfig(device, accumulation) {
+    return selectCoreSubgroupMatrixConfig(asOIDNDevice(device), accumulation);
+}
+const pendingPipelinesByDevice = new WeakMap();
+/**
+ * Native, model-driven OIDN U-Net executor for browser WebGPU devices. The
+ * device must also provide the asynchronous WebGPU operations.
+ */
+export class NativeUNetExecutor extends NativeUNetCore {
+    _lastExecutionProfile;
+    constructor(device, model, options) {
+        super(asOIDNDevice(device), model, options);
+    }
+    // Inputs and outputs are buffers of the browser device, so they are GPUBuffers.
+    execute(inputBuffers, width, height) {
+        return super.execute(inputBuffers, width, height);
+    }
+    get _asyncDevice() {
+        return this._device;
+    }
+    get _pipelinePromises() {
+        let pending = pendingPipelinesByDevice.get(this._device);
+        if (!pending) {
+            pending = new Map();
+            pendingPipelinesByDevice.set(this._device, pending);
+        }
+        return pending;
+    }
+    getLastExecutionProfile() {
+        return this._lastExecutionProfile;
+    }
+    _pipelineAsync(key, code) {
+        const ready = this._pipelineCache.get(key);
+        if (ready)
+            return Promise.resolve(ready);
+        const pending = this._pipelinePromises.get(key);
+        if (pending)
+            return pending;
+        const promise = this._asyncDevice.createComputePipelineAsync({
+            label: `oidn/${key}`,
+            layout: 'auto',
+            compute: {
+                module: this._device.createShaderModule({
+                    label: `oidn/${key}`,
+                    code
+                }),
+                entryPoint: 'main'
+            }
+        }).then((pipeline) => {
+            this._pipelineCache.set(key, pipeline);
+            this._pipelinePromises.delete(key);
+            return pipeline;
+        }, (error) => {
+            this._pipelinePromises.delete(key);
+            throw error;
+        });
+        this._pipelinePromises.set(key, promise);
+        return promise;
+    }
+    _retireExecution(execution) {
+        // Commands using an evicted plan may still be submitted. Defer actual
+        // destruction until all work currently on the shared queue completes.
+        this._retiredExecutions.add(execution);
+        void this._asyncDevice.queue.onSubmittedWorkDone()
+            .catch(() => undefined)
+            .then(() => {
+            this._retiredExecutions.delete(execution);
+            this._destroyExecution(execution);
+        });
+    }
+    /** Compiles all shape-independent kernels before the model reports ready. */
+    async prepare() {
+        if (this._disposed)
+            throw new Error('Native OIDN executor is disposed');
+        const graph = optimizeModelGraph(this._model, { fuseConvPool: false });
+        const sourceCount = this._model.inputChannels / 3;
+        const specs = [
+            {
+                key: `pack/${this.precision}/${sourceCount}`,
+                code: createInputPackShader(this.precision, sourceCount)
+            },
+            ...graph.nodes.map((node) => this._nodePipelineSpec(node, node.id === graph.spec.output))
+        ];
+        await Promise.all(specs.map(({ key, code }) => this._pipelineAsync(key, code)));
+    }
+    _readProfile(querySet, resolveBuffer, readbackBuffer, labels) {
+        this._lastExecutionProfile = (async () => {
+            try {
+                await readbackBuffer.mapAsync(MapMode.READ);
+                // Little-endian u64 nanoseconds as u32 pairs; differences stay exact.
+                const words = new Uint32Array(readbackBuffer.getMappedRange());
+                const timestamp = (query) => words[query * 2 + 1] * 4294967296 + words[query * 2];
+                const layers = labels.map((id, index) => ({
+                    id,
+                    durationMs: (timestamp(index * 2 + 1) - timestamp(index * 2)) / 1_000_000
+                }));
+                return {
+                    totalMs: layers.reduce((sum, layer) => sum + layer.durationMs, 0),
+                    layers
+                };
+            }
+            finally {
+                if (readbackBuffer.mapState === 'mapped') {
+                    readbackBuffer.unmap();
+                }
+                this._releaseQuerySet(querySet);
+                this._releaseBuffer(resolveBuffer);
+                this._releaseBuffer(readbackBuffer);
+                this._profileOperations--;
+            }
+        })();
+    }
+    /** Executes interleaved CPU image data through the native GPU runtime. */
+    async executeCPU(interleavedInput, width, height) {
+        const expectedLength = width * height * this._model.inputChannels;
+        if (interleavedInput.length !== expectedLength) {
+            throw new Error(`Native OIDN CPU input has ${interleavedInput.length} values, expected ${expectedLength}`);
+        }
+        const execution = this._execution(width, height);
+        const sourceCount = this._model.inputChannels / 3;
+        const pixelCount = width * height;
+        if (!execution.cpuInputBuffers) {
+            execution.cpuInputBuffers = Array.from({ length: sourceCount }, (_, index) => {
+                const buffer = this._device.createBuffer({
+                    label: `oidn/cpu-input/${width}x${height}/${index}`,
+                    size: pixelCount * 16,
+                    usage: BufferUsage.STORAGE | BufferUsage.COPY_DST
+                });
+                this._resources.track('gpu-buffer', buffer);
+                execution.ownedBuffers.push(buffer);
+                return buffer;
+            });
+            execution.cpuReadbackBuffer = this._device.createBuffer({
+                label: `oidn/cpu-readback/${width}x${height}`,
+                size: pixelCount * 16,
+                usage: BufferUsage.COPY_DST | BufferUsage.MAP_READ
+            });
+            this._resources.track('gpu-buffer', execution.cpuReadbackBuffer);
+            execution.ownedBuffers.push(execution.cpuReadbackBuffer);
+        }
+        for (let source = 0; source < sourceCount; source++) {
+            const upload = new Float32Array(pixelCount * 4);
+            for (let pixel = 0; pixel < pixelCount; pixel++) {
+                const inputOffset = pixel * this._model.inputChannels + source * 3;
+                const outputOffset = pixel * 4;
+                upload[outputOffset] = interleavedInput[inputOffset];
+                upload[outputOffset + 1] = interleavedInput[inputOffset + 1];
+                upload[outputOffset + 2] = interleavedInput[inputOffset + 2];
+            }
+            this._device.queue.writeBuffer(execution.cpuInputBuffers[source], 0, new Uint8Array(upload.buffer, upload.byteOffset, upload.byteLength));
+        }
+        const output = super.execute(execution.cpuInputBuffers, width, height);
+        const encoder = this._device.createCommandEncoder({
+            label: `oidn/cpu-readback/${width}x${height}`
+        });
+        encoder.copyBufferToBuffer(output, 0, execution.cpuReadbackBuffer, 0, pixelCount * 16);
+        this._device.queue.submit([encoder.finish()]);
+        await execution.cpuReadbackBuffer.mapAsync(MapMode.READ);
+        const rgba = new Float32Array(execution.cpuReadbackBuffer.getMappedRange());
+        const rgb = new Float32Array(pixelCount * 3);
+        for (let pixel = 0; pixel < pixelCount; pixel++) {
+            rgb[pixel * 3] = rgba[pixel * 4];
+            rgb[pixel * 3 + 1] = rgba[pixel * 4 + 1];
+            rgb[pixel * 3 + 2] = rgba[pixel * 4 + 2];
+        }
+        execution.cpuReadbackBuffer.unmap();
+        return rgb;
+    }
+}
+//# sourceMappingURL=nativeUNet.js.map

package/lib/nativeUNet.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"nativeUNet.js","sourceRoot":"","sources":["../src/nativeUNet.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,kBAAkB,EAAE,MAAM,qBAAqB,CAAC;AACzD,OAAO,EACL,WAAW,EACX,OAAO,EAOR,MAAM,eAAe,CAAC;AAEvB,OAAO,EACL,qBAAqB,EACrB,cAAc,EACd,0BAA0B,IAAI,oBAAoB,EAClD,0BAA0B,IAAI,8BAA8B,EAQ7D,MAAM,qBAAqB,CAAC;AAE7B,6EAA6E;AAC7E,cAAc,qBAAqB,CAAC;AAEpC,8EAA8E;AAC9E,0EAA0E;AAC1E,yEAAyE;AACzE,SAAS,YAAY,CAAC,MAAiB;IACrC,OAAO,MAAuC,CAAC;AACjD,CAAC;AAED,MAAM,UAAU,0BAA0B,CACxC,MAAiB,EACjB,SAAsC;IAEtC,OAAO,oBAAoB,CAAC,YAAY,CAAC,MAAM,CAAC,EAAE,SAAS,CAAC,CAAC;AAC/D,CAAC;AAED,MAAM,UAAU,0BAA0B,CACxC,MAAiB,EACjB,YAA4B;IAE5B,OAAO,8BAA8B,CAAC,YAAY,CAAC,MAAM,CAAC,EAAE,YAAY,CAAC,CAAC;AAC5E,CAAC;AAED,MAAM,wBAAwB,GAAG,IAAI,OAAO,EAGzC,CAAC;AAEJ;;;GAGG;AACH,MAAM,OAAO,kBAAmB,SAAQ,cAAc;IAC5C,qBAAqB,CAAuC;IAEpE,YACE,MAAiB,EACjB,KAAyB,EACzB,OAA2B;QAE3B,KAAK,CAAC,YAAY,CAAC,MAAM,CAAC,EAAE,KAAK,EAAE,OAAO,CAAC,CAAC;IAC9C,CAAC;IAED,gFAAgF;IACvE,OAAO,CACd,YAAkC,EAClC,KAAa,EACb,MAAc;QAEd,OAAO,KAAK,CAAC,OAAO,CAAC,YAAY,EAAE,KAAK,EAAE,MAAM,CAAyB,CAAC;IAC5E,CAAC;IAED,IAAY,YAAY;QACtB,OAAO,IAAI,CAAC,OAA6B,CAAC;IAC5C,CAAC;IAED,IAAY,iBAAiB;QAC3B,IAAI,OAAO,GAAG,wBAAwB,CAAC,GAAG,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC;QACzD,IAAI,CAAC,OAAO,EAAE,CAAC;YACb,OAAO,GAAG,IAAI,GAAG,EAAE,CAAC;YACpB,wBAAwB,CAAC,GAAG,CAAC,IAAI,CAAC,OAAO,EAAE,OAAO,CAAC,CAAC;QACtD,CAAC;QACD,OAAO,OAAO,CAAC;IACjB,CAAC;IAED,uBAAuB;QACrB,OAAO,IAAI,CAAC,qBAAqB,CAAC;IACpC,CAAC;IAEO,cAAc,CAAC,GAAW,EAAE,IAAY;QAC9C,MAAM,KAAK,GAAG,IAAI,CAAC,cAAc,CAAC,GAAG,CAAC,GAAG,CAAC,CAAC;QAC3C,IAAI,KAAK;YAAE,OAAO,OAAO,CAAC,OAAO,CAAC,KAAK,CAAC,CAAC;QACzC,MAAM,OAAO,GAAG,IAAI,CAAC,iBAAiB,CAAC,GAAG,CAAC,GAAG,CAAC,CAAC;QAChD,IAAI,OAAO;YAAE,OAAO,OAAO,CAAC;QAC5B,MAAM,OAAO,GAAG,IAAI,CAAC,YAAY,CAAC,0BAA0B,CAAC;YAC3D,KAAK,EAAE,QAAQ,GAAG,EAAE;YACpB,MAAM,EAAE,MAAM;YACd,OAAO,EAAE;gBACP,MAAM,EAAE,IAAI,CAAC,OAAO,CAAC,kBAAkB,CAAC;oBACtC,KAAK,EAAE,QAAQ,GAAG,EAAE;oBACpB,IAAI;iBACL,CAAC;gBACF,UAAU,EAAE,MAAM;aACnB;SACF,CAAC,CAAC,IAAI,CAAC,CAAC,QAAgC,EAAE,EAAE;YAC3C,IAAI,CAAC,cAAc,CAAC,GAAG,CAAC,GAAG,EAAE,QAAQ,CAAC,CAAC;YACvC,IAAI,CAAC,iBAAiB,CAAC,MAAM,CAAC,GAAG,CAAC,CAAC;YACnC,OAAO,QAAQ,CAAC;QAClB,CAAC,EAAE,CAAC,KAAc,EAAE,EAAE;YACpB,IAAI,CAAC,iBAAiB,CAAC,MAAM,CAAC,GAAG,CAAC,CAAC;YACnC,MAAM,KAAK,CAAC;QACd,CAAC,CAAC,CAAC;QACH,IAAI,CAAC,iBAAiB,CAAC,GAAG,CAAC,GAAG,EAAE,OAAO,CAAC,CAAC;QACzC,OAAO,OAAO,CAAC;IACjB,CAAC;IAEkB,gBAAgB,CAAC,SAA0B;QAC5D,sEAAsE;QACtE,sEAAsE;QACtE,IAAI,CAAC,kBAAkB,CAAC,GAAG,CAAC,SAAS,CAAC,CAAC;QACvC,KAAK,IAAI,CAAC,YAAY,CAAC,KAAK,CAAC,mBAAmB,EAAE;aAC/C,KAAK,CAAC,GAAG,EAAE,CAAC,SAAS,CAAC;aACtB,IAAI,CAAC,GAAG,EAAE;YACT,IAAI,CAAC,kBAAkB,CAAC,MAAM,CAAC,SAAS,CAAC,CAAC;YAC1C,IAAI,CAAC,iBAAiB,CAAC,SAAS,CAAC,CAAC;QACpC,CAAC,CAAC,CAAC;IACP,CAAC;IAED,6EAA6E;IAC7E,KAAK,CAAC,OAAO;QACX,IAAI,IAAI,CAAC,SAAS;YAAE,MAAM,IAAI,KAAK,CAAC,kCAAkC,CAAC,CAAC;QACxE,MAAM,KAAK,GAAG,kBAAkB,CAAC,IAAI,CAAC,MAAM,EAAE,EAAE,YAAY,EAAE,KAAK,EAAE,CAAC,CAAC;QACvE,MAAM,WAAW,GAAG,IAAI,CAAC,MAAM,CAAC,aAAa,GAAG,CAAC,CAAC;QAClD,MAAM,KAAK,GAAyB;YAClC;gBACE,GAAG,EAAE,QAAQ,IAAI,CAAC,SAAS,IAAI,WAAW,EAAE;gBAC5C,IAAI,EAAE,qBAAqB,CAAC,IAAI,CAAC,SAAS,EAAE,WAAW,CAAC;aACzD;YACD,GAAG,KAAK,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC,IAAI,EAAE,EAAE,CAC1B,IAAI,CAAC,iBAAiB,CAAC,IAAI,EAAE,IAAI,CAAC,EAAE,KAAK,KAAK,CAAC,IAAI,CAAC,MAAM,CAAC,CAC5D;SACF,CAAC;QACF,MAAM,OAAO,CAAC,GAAG,CACf,KAAK,CAAC,GAAG,CAAC,CAAC,EAAE,GAAG,EAAE,IAAI,EAAE,EAAE,EAAE,CAAC,IAAI,CAAC,cAAc,CAAC,GAAG,EAAE,IAAI,CAAC,CAAC,CAC7D,CAAC;IACJ,CAAC;IAEkB,YAAY,CAC7B,QAAyB,EACzB,aAA4B,EAC5B,cAA6B,EAC7B,MAAgB;QAEhB,IAAI,CAAC,qBAAqB,GAAG,CAAC,KAAK,IAAI,EAAE;YACvC,IAAI,CAAC;gBACH,MAAO,cAAwC,CAAC,QAAQ,CAAC,OAAO,CAAC,IAAI,CAAC,CAAC;gBACvE,sEAAsE;gBACtE,MAAM,KAAK,GAAG,IAAI,WAAW,CAAC,cAAc,CAAC,cAAc,EAAE,CAAC,CAAC;gBAC/D,MAAM,SAAS,GAAG,CAAC,KAAa,EAAE,EAAE,CAClC,KAAK,CAAC,KAAK,GAAG,CAAC,GAAG,CAAC,CAAC,GAAG,UAAU,GAAG,KAAK,CAAC,KAAK,GAAG,CAAC,CAAC,CAAC;gBACvD,MAAM,MAAM,GAAG,MAAM,CAAC,GAAG,CAAC,CAAC,EAAE,EAAE,KAAK,EAAE,EAAE,CAAC,CAAC;oBACxC,EAAE;oBACF,UAAU,EACR,CAAC,SAAS,CAAC,KAAK,GAAG,CAAC,GAAG,CAAC,CAAC,GAAG,SAAS,CAAC,KAAK,GAAG,CAAC,CAAC,CAAC,GAAG,SAAS;iBAChE,CAAC,CAAC,CAAC;gBACJ,OAAO;oBACL,OAAO,EAAE,MAAM,CAAC,MAAM,CACpB,CAAC,GAAG,EAAE,KAAK,EAAE,EAAE,CAAC,GAAG,GAAG,KAAK,CAAC,UAAU,EACtC,CAAC,CACF;oBACD,MAAM;iBACP,CAAC;YACJ,CAAC;oBAAS,CAAC;gBACT,IAAI,cAAc,CAAC,QAAQ,KAAK,QAAQ,EAAE,CAAC;oBACzC,cAAc,CAAC,KAAK,EAAE,CAAC;gBACzB,CAAC;gBACD,IAAI,CAAC,gBAAgB,CAAC,QAAQ,CAAC,CAAC;gBAChC,IAAI,CAAC,cAAc,CAAC,aAAa,CAAC,CAAC;gBACnC,IAAI,CAAC,cAAc,CAAC,cAAc,CAAC,CAAC;gBACpC,IAAI,CAAC,kBAAkB,EAAE,CAAC;YAC5B,CAAC;QACH,CAAC,CAAC,EAAE,CAAC;IACP,CAAC;IAED,0EAA0E;IAC1E,KAAK,CAAC,UAAU,CACd,gBAA8B,EAC9B,KAAa,EACb,MAAc;QAEd,MAAM,cAAc,GAAG,KAAK,GAAG,MAAM,GAAG,IAAI,CAAC,MAAM,CAAC,aAAa,CAAC;QAClE,IAAI,gBAAgB,CAAC,MAAM,KAAK,cAAc,EAAE,CAAC;YAC/C,MAAM,IAAI,KAAK,CACb,6BAA6B,gBAAgB,CAAC,MAAM,qBAAqB,cAAc,EAAE,CAC1F,CAAC;QACJ,CAAC;QACD,MAAM,SAAS,GAAG,IAAI,CAAC,UAAU,CAAC,KAAK,EAAE,MAAM,CAAC,CAAC;QACjD,MAAM,WAAW,GAAG,IAAI,CAAC,MAAM,CAAC,aAAa,GAAG,CAAC,CAAC;QAClD,MAAM,UAAU,GAAG,KAAK,GAAG,MAAM,CAAC;QAClC,IAAI,CAAC,SAAS,CAAC,eAAe,EAAE,CAAC;YAC/B,SAAS,CAAC,eAAe,GAAG,KAAK,CAAC,IAAI,CAAC,EAAE,MAAM,EAAE,WAAW,EAAE,EAAE,CAAC,CAAC,EAAE,KAAK,EAAE,EAAE;gBAC3E,MAAM,MAAM,GAAG,IAAI,CAAC,OAAO,CAAC,YAAY,CAAC;oBACvC,KAAK,EAAE,kBAAkB,KAAK,IAAI,MAAM,IAAI,KAAK,EAAE;oBACnD,IAAI,EAAE,UAAU,GAAG,EAAE;oBACrB,KAAK,EAAE,WAAW,CAAC,OAAO,GAAG,WAAW,CAAC,QAAQ;iBAClD,CAAC,CAAC;gBACH,IAAI,CAAC,UAAU,CAAC,KAAK,CAAC,YAAY,EAAE,MAAM,CAAC,CAAC;gBAC5C,SAAS,CAAC,YAAY,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC;gBACpC,OAAO,MAAM,CAAC;YAChB,CAAC,CAAC,CAAC;YACH,SAAS,CAAC,iBAAiB,GAAG,IAAI,CAAC,OAAO,CAAC,YAAY,CAAC;gBACtD,KAAK,EAAE,qBAAqB,KAAK,IAAI,MAAM,EAAE;gBAC7C,IAAI,EAAE,UAAU,GAAG,EAAE;gBACrB,KAAK,EAAE,WAAW,CAAC,QAAQ,GAAG,WAAW,CAAC,QAAQ;aACnD,CAAC,CAAC;YACH,IAAI,CAAC,UAAU,CAAC,KAAK,CAAC,YAAY,EAAE,SAAS,CAAC,iBAAiB,CAAC,CAAC;YACjE,SAAS,CAAC,YAAY,CAAC,IAAI,CAAC,SAAS,CAAC,iBAAiB,CAAC,CAAC;QAC3D,CAAC;QAED,KAAK,IAAI,MAAM,GAAG,CAAC,EAAE,MAAM,GAAG,WAAW,EAAE,MAAM,EAAE,EAAE,CAAC;YACpD,MAAM,MAAM,GAAG,IAAI,YAAY,CAAC,UAAU,GAAG,CAAC,CAAC,CAAC;YAChD,KAAK,IAAI,KAAK,GAAG,CAAC,EAAE,KAAK,GAAG,UAAU,EAAE,KAAK,EAAE,EAAE,CAAC;gBAChD,MAAM,WAAW,GACf,KAAK,GAAG,IAAI,CAAC,MAAM,CAAC,aAAa,GAAG,MAAM,GAAG,CAAC,CAAC;gBACjD,MAAM,YAAY,GAAG,KAAK,GAAG,CAAC,CAAC;gBAC/B,MAAM,CAAC,YAAY,CAAC,GAAG,gBAAgB,CAAC,WAAW,CAAC,CAAC;gBACrD,MAAM,CAAC,YAAY,GAAG,CAAC,CAAC,GAAG,gBAAgB,CAAC,WAAW,GAAG,CAAC,CAAC,CAAC;gBAC7D,MAAM,CAAC,YAAY,GAAG,CAAC,CAAC,GAAG,gBAAgB,CAAC,WAAW,GAAG,CAAC,CAAC,CAAC;YAC/D,CAAC;YACD,IAAI,CAAC,OAAO,CAAC,KAAK,CAAC,WAAW,CAC5B,SAAS,CAAC,eAAe,CAAC,MAAM,CAAC,EACjC,CAAC,EACD,IAAI,UAAU,CAAC,MAAM,CAAC,MAAM,EAAE,MAAM,CAAC,UAAU,EAAE,MAAM,CAAC,UAAU,CAAC,CACpE,CAAC;QACJ,CAAC;QAED,MAAM,MAAM,GAAG,KAAK,CAAC,OAAO,CAAC,SAAS,CAAC,eAAe,EAAE,KAAK,EAAE,MAAM,CAAC,CAAC;QACvE,MAAM,OAAO,GAAG,IAAI,CAAC,OAAO,CAAC,oBAAoB,CAAC;YAChD,KAAK,EAAE,qBAAqB,KAAK,IAAI,MAAM,EAAE;SAC9C,CAAC,CAAC;QACH,OAAO,CAAC,kBAAkB,CACxB,MAAM,EACN,CAAC,EACD,SAAS,CAAC,iBAAkB,EAC5B,CAAC,EACD,UAAU,GAAG,EAAE,CAChB,CAAC;QACF,IAAI,CAAC,OAAO,CAAC,KAAK,CAAC,MAAM,CAAC,CAAC,OAAO,CAAC,MAAM,EAAE,CAAC,CAAC,CAAC;QAC9C,MAAO,SAAS,CAAC,iBAA2C,CAAC,QAAQ,CAAC,OAAO,CAAC,IAAI,CAAC,CAAC;QACpF,MAAM,IAAI,GAAG,IAAI,YAAY,CAC3B,SAAS,CAAC,iBAAkB,CAAC,cAAc,EAAE,CAC9C,CAAC;QACF,MAAM,GAAG,GAAG,IAAI,YAAY,CAAC,UAAU,GAAG,CAAC,CAAC,CAAC;QAC7C,KAAK,IAAI,KAAK,GAAG,CAAC,EAAE,KAAK,GAAG,UAAU,EAAE,KAAK,EAAE,EAAE,CAAC;YAChD,GAAG,CAAC,KAAK,GAAG,CAAC,CAAC,GAAG,IAAI,CAAC,KAAK,GAAG,CAAC,CAAC,CAAC;YACjC,GAAG,CAAC,KAAK,GAAG,CAAC,GAAG,CAAC,CAAC,GAAG,IAAI,CAAC,KAAK,GAAG,CAAC,GAAG,CAAC,CAAC,CAAC;YACzC,GAAG,CAAC,KAAK,GAAG,CAAC,GAAG,CAAC,CAAC,GAAG,IAAI,CAAC,KAAK,GAAG,CAAC,GAAG,CAAC,CAAC,CAAC;QAC3C,CAAC;QACD,SAAS,CAAC,iBAAkB,CAAC,KAAK,EAAE,CAAC;QACrC,OAAO,GAAG,CAAC;IACb,CAAC;CACF"}