oidn-web 0.1.0

package/lib/hdr.js

@@ -0,0 +1,340 @@
+// https://github.com/bbbbx/hdr.js/blob/main/src/hdr.ts
+function frexp(v) {
+    v = Number(v);
+    const result = { f: v, e: 0 };
+    if (v !== 0 && Number.isFinite(v)) {
+        const absV = Math.abs(v);
+        const log2 = Math.log2 ||
+            function log2(n) {
+                return Math.log(n) * Math.LOG2E;
+            };
+        // Math.pow(2, -exp) === Infinity when exp <= -1024
+        let e = Math.max(-1023, Math.floor(log2(absV)) + 1);
+        let f = absV * Math.pow(2.0, -e);
+        while (f >= 1.0) {
+            f *= 0.5;
+            e++;
+        }
+        while (f < 0.5) {
+            f *= 2;
+            e--;
+        }
+        if (v < 0) {
+            f = -f;
+        }
+        result.f = f;
+        result.e = e;
+    }
+    return result;
+}
+function ldexp(f, e) {
+    return f * Math.pow(2.0, e);
+}
+/**
+ * Convert 4 byte uint8 buffer to 3 channels float data
+ * @param rgbe input uint8 buffer
+ * @param float output float data
+ */
+function rgbe2float(rgbe, float) {
+    if (rgbe[3] !== 0) {
+        const f1 = ldexp(1.0, rgbe[3] - (128 + 8));
+        float[0] = rgbe[0] * f1;
+        float[1] = rgbe[1] * f1;
+        float[2] = rgbe[2] * f1;
+    }
+    else {
+        float[0] = float[1] = float[2] = 0.0;
+    }
+}
+/**
+ * Convert 3 channels float data to 4 byte uint8 buffer
+ * @param float input float data
+ * @param rgbe output uint8 buffer
+ */
+function float2rgbe(float, rgbe) {
+    const red = float[0];
+    const green = float[1];
+    const blue = float[2];
+    const v = Math.max(red, Math.max(green, blue));
+    if (v < 1e-32) {
+        rgbe[0] = rgbe[1] = rgbe[2] = rgbe[3] = 0;
+    }
+    else {
+        const { f, e } = frexp(v);
+        const s = (f / v) * 256.0;
+        rgbe[0] = red * s;
+        rgbe[1] = green * s;
+        rgbe[2] = blue * s;
+        rgbe[3] = e + 128;
+    }
+}
+/**
+ * Write float data to RGBE(.hdr) file buffer
+ * @param w image width
+ * @param h image height
+ * @param data float data, RGB 3 channels.
+ * @returns file buffer
+ */
+function write_hdr(w, h, data) {
+    const s = [];
+    const comp = 3;
+    write_hdr_core(s, w, h, comp, data);
+    return new Uint8Array(s);
+}
+function write_hdr_core(s, x, y, comp, data) {
+    if (y <= 0 || x <= 0 || !data) {
+        return;
+    }
+    const header = '#?RADIANCE\n' +
+        'FORMAT=32-bit_rle_rgbe\n' +
+        'EXPOSURE=1.0\n' +
+        '\n' +
+        '-Y ' +
+        y +
+        ' +X ' +
+        x +
+        '\n';
+    header.split('').forEach((c) => {
+        const charCode = c.charCodeAt(0);
+        s.push(charCode);
+    });
+    const scratch = new Uint8Array(x * 4);
+    const stbi__flip_vertically_on_write = false;
+    for (let i = 0; i < y; i++) {
+        write_hdr_scanline(s, x, comp, scratch, data.subarray(comp * x * (stbi__flip_vertically_on_write ? y - 1 - i : i)));
+    }
+}
+function write_hdr_scanline(s, width, ncomp, scratch, scanline) {
+    const scanlineheader = new Uint8Array([2, 2, 0, 0]);
+    const rgbe = new Uint8Array(4);
+    const linear = new Float32Array(3);
+    let x;
+    scanlineheader[2] = (width & 0xff00) >> 8;
+    scanlineheader[3] = width & 0x00ff;
+    /* skip RLE for images too small or large */
+    if (width < 8 || width >= 32768) {
+        for (x = 0; x < width; x++) {
+            switch (ncomp) {
+                case 4: /* fallthrough */
+                case 3:
+                    linear[2] = scanline[x * ncomp + 2];
+                    linear[1] = scanline[x * ncomp + 1];
+                    linear[0] = scanline[x * ncomp + 0];
+                    break;
+                default:
+                    linear[0] = linear[1] = linear[2] = scanline[x * ncomp + 0];
+                    break;
+            }
+            float2rgbe(linear, rgbe);
+            s.push(rgbe[0], rgbe[1], rgbe[2], rgbe[3]);
+        }
+    }
+    else {
+        let c, r;
+        /* encode into scratch buffer */
+        for (x = 0; x < width; x++) {
+            switch (ncomp) {
+                case 4: /* fallthrough */
+                case 3:
+                    linear[2] = scanline[x * ncomp + 2];
+                    linear[1] = scanline[x * ncomp + 1];
+                    linear[0] = scanline[x * ncomp + 0];
+                    break;
+                default:
+                    linear[0] = linear[1] = linear[2] = scanline[x * ncomp + 0];
+                    break;
+            }
+            float2rgbe(linear, rgbe);
+            scratch[x + width * 0] = rgbe[0];
+            scratch[x + width * 1] = rgbe[1];
+            scratch[x + width * 2] = rgbe[2];
+            scratch[x + width * 3] = rgbe[3];
+        }
+        s.push(scanlineheader[0], scanlineheader[1], scanlineheader[2], scanlineheader[3]);
+        /* RLE each component separately */
+        for (c = 0; c < 4; c++) {
+            const comp = scratch.subarray(width * c);
+            x = 0;
+            while (x < width) {
+                // find first run
+                r = x;
+                while (r + 2 < width) {
+                    if (comp[r] === comp[r + 1] && comp[r] === comp[r + 2])
+                        break;
+                    ++r;
+                }
+                if (r + 2 >= width)
+                    r = width;
+                // dump up to first run
+                while (x < r) {
+                    let len = r - x;
+                    if (len > 128)
+                        len = 128;
+                    write_dump_data(s, len, comp.subarray(x));
+                    x += len;
+                }
+                // if there's a run, output it
+                if (r + 2 < width) {
+                    // same test as what we break out of in search loop, so only true if we break'd
+                    // find next byte after run
+                    while (r < width && comp[r] == comp[x])
+                        ++r;
+                    // output run up to r
+                    while (x < r) {
+                        let len = r - x;
+                        if (len > 127)
+                            len = 127;
+                        write_run_data(s, len, comp[x]);
+                        x += len;
+                    }
+                }
+            }
+        }
+    }
+}
+function write_dump_data(s, length, data) {
+    const lengthbyte = length & 0xff;
+    if (!(length <= 128))
+        throw new Error('length is greater than 128');
+    s.push(lengthbyte);
+    for (let i = 0; i < length; i++) {
+        s.push(data[i]);
+    }
+}
+function write_run_data(s, length, databyte) {
+    const lengthbyte = (length + 128) & 0xff;
+    if (!(length + 128 <= 255))
+        throw new Error('length is greater than 128');
+    s.push(lengthbyte, databyte);
+}
+function testMagic(uint8, signature) {
+    const l = signature.length;
+    for (let i = 0; i < l; i++) {
+        if (uint8[i] !== signature.charCodeAt(i)) {
+            return false;
+        }
+    }
+    return true;
+}
+function isHdr(uint8) {
+    let r = testMagic(uint8, '#?RADIANCE\n');
+    if (!r) {
+        r = testMagic(uint8, '#?RGBE\n');
+    }
+    return r;
+}
+const MAX_HEADER_LENGTH = 1024 * 10;
+const MAX_DIMENSIONS = 1 << 24;
+/**
+ * Read a RGBE(.hdr) file from buffer
+ * @param uint8 RGBE(.hdr) file buffer
+ * @returns Failure reason or resolved data.
+ */
+function read_hdr(uint8) {
+    let header = '';
+    let pos = 0;
+    if (!isHdr(uint8)) {
+        return 'Corrupt HDR image.';
+    }
+    // read header
+    while (!header.match(/\n\n[^\n]+\n/g) && pos < MAX_HEADER_LENGTH) {
+        header += String.fromCharCode(uint8[pos++]);
+    }
+    // check format
+    const format = header.match(/FORMAT=(.*)$/m)?.[1];
+    if (format !== '32-bit_rle_rgbe') {
+        return 'Unsupported HDR format: ' + format;
+    }
+    // parse resolution
+    const rez = header.split(/\n/).reverse()[1].split(' ');
+    if (rez[0] !== '-Y' || rez[2] !== '+X') {
+        return 'Unsupported HDR format';
+    }
+    const width = Number.parseFloat(rez[3]);
+    const height = Number.parseFloat(rez[1]);
+    if (width > MAX_DIMENSIONS || height > MAX_DIMENSIONS) {
+        return 'Very large image (corrupt?)';
+    }
+    let i, j;
+    let c1 = uint8[pos];
+    let c2 = uint8[pos + 1];
+    let len = uint8[pos + 2];
+    // not run-length encoded, so we have to actually use THIS data as a decoded
+    // pixel (note this can't be a valid pixel--one of RGB must be >= 128)
+    const notRLE = c1 !== 2 || c2 !== 2 || !!(len & 0x80); // not run-length encoded
+    const hdrData = new Float32Array(width * height * 3);
+    if (width < 8 || width >= 32768 || notRLE) {
+        // 32768: 2^15
+        // Read flat data
+        for (j = 0; j < height; ++j) {
+            for (i = 0; i < width; ++i) {
+                const rgbe = uint8.subarray(pos, pos + 4);
+                pos += 4;
+                const start = (j * width + i) * 3;
+                rgbe2float(rgbe, hdrData.subarray(start, start + 3));
+            }
+        }
+    }
+    else {
+        // Read RLE-encoded data
+        let scanline;
+        let c1;
+        let c2;
+        let len;
+        for (let j = 0; j < height; j++) {
+            c1 = uint8[pos++];
+            c2 = uint8[pos++];
+            len = uint8[pos++];
+            if (c1 !== 2 || c2 !== 2 || len & 0x80) {
+                return 'Invalid scanline';
+            }
+            len = len << 8;
+            len |= uint8[pos++];
+            if (len !== width) {
+                return 'invalid decoded scanline length';
+            }
+            if (!scanline) {
+                scanline = new Uint8Array(width * 4);
+            }
+            let count;
+            let value;
+            for (let k = 0; k < 4; k++) {
+                let nLeft;
+                i = 0;
+                while ((nLeft = width - i) > 0) {
+                    count = uint8[pos++];
+                    if (count > 128) {
+                        // is RUN
+                        value = uint8[pos++];
+                        count -= 128;
+                        if (count > nLeft) {
+                            return 'bad RLE data in HDR';
+                        }
+                        for (let z = 0; z < count; z++) {
+                            scanline[i++ * 4 + k] = value;
+                        }
+                    }
+                    else {
+                        // is DUMP
+                        if (count > nLeft) {
+                            return 'bad RLE data in HDR';
+                        }
+                        for (let z = 0; z < count; z++) {
+                            scanline[i++ * 4 + k] = uint8[pos++];
+                        }
+                    }
+                }
+            }
+            for (let i = 0; i < width; i++) {
+                rgbe2float(scanline.subarray(i * 4), hdrData.subarray((j * width + i) * 3));
+            }
+        }
+    }
+    return {
+        rgbFloat: hdrData,
+        width: width,
+        height: height
+    };
+}
+export { read_hdr as readHDR, write_hdr as writeHDR, float2rgbe, rgbe2float };
+//# sourceMappingURL=hdr.js.map

package/lib/hdr.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"hdr.js","sourceRoot":"","sources":["../src/hdr.ts"],"names":[],"mappings":"AAAA,uDAAuD;AACvD,SAAS,KAAK,CAAC,CAAS;IACtB,CAAC,GAAG,MAAM,CAAC,CAAC,CAAC,CAAC;IACd,MAAM,MAAM,GAAG,EAAE,CAAC,EAAE,CAAC,EAAE,CAAC,EAAE,CAAC,EAAE,CAAC;IAE9B,IAAI,CAAC,KAAK,CAAC,IAAI,MAAM,CAAC,QAAQ,CAAC,CAAC,CAAC,EAAE,CAAC;QAClC,MAAM,IAAI,GAAG,IAAI,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC;QACzB,MAAM,IAAI,GACR,IAAI,CAAC,IAAI;YACT,SAAS,IAAI,CAAC,CAAC;gBACb,OAAO,IAAI,CAAC,GAAG,CAAC,CAAC,CAAC,GAAG,IAAI,CAAC,KAAK,CAAC;YAClC,CAAC,CAAC;QAEJ,mDAAmD;QACnD,IAAI,CAAC,GAAG,IAAI,CAAC,GAAG,CAAC,CAAC,IAAI,EAAE,IAAI,CAAC,KAAK,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC;QACpD,IAAI,CAAC,GAAG,IAAI,GAAG,IAAI,CAAC,GAAG,CAAC,GAAG,EAAE,CAAC,CAAC,CAAC,CAAC;QAEjC,OAAO,CAAC,IAAI,GAAG,EAAE,CAAC;YAChB,CAAC,IAAI,GAAG,CAAC;YACT,CAAC,EAAE,CAAC;QACN,CAAC;QACD,OAAO,CAAC,GAAG,GAAG,EAAE,CAAC;YACf,CAAC,IAAI,CAAC,CAAC;YACP,CAAC,EAAE,CAAC;QACN,CAAC;QAED,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC;YACV,CAAC,GAAG,CAAC,CAAC,CAAC;QACT,CAAC;QAED,MAAM,CAAC,CAAC,GAAG,CAAC,CAAC;QACb,MAAM,CAAC,CAAC,GAAG,CAAC,CAAC;IACf,CAAC;IAED,OAAO,MAAM,CAAC;AAChB,CAAC;AAED,SAAS,KAAK,CAAC,CAAS,EAAE,CAAS;IACjC,OAAO,CAAC,GAAG,IAAI,CAAC,GAAG,CAAC,GAAG,EAAE,CAAC,CAAC,CAAC;AAC9B,CAAC;AAED;;;;GAIG;AACH,SAAS,UAAU,CAAC,IAAgB,EAAE,KAAmB;IACvD,IAAI,IAAI,CAAC,CAAC,CAAC,KAAK,CAAC,EAAE,CAAC;QAClB,MAAM,EAAE,GAAG,KAAK,CAAC,GAAG,EAAE,IAAI,CAAC,CAAC,CAAC,GAAG,CAAC,GAAG,GAAG,CAAC,CAAC,CAAC,CAAC;QAC3C,KAAK,CAAC,CAAC,CAAC,GAAG,IAAI,CAAC,CAAC,CAAC,GAAG,EAAE,CAAC;QACxB,KAAK,CAAC,CAAC,CAAC,GAAG,IAAI,CAAC,CAAC,CAAC,GAAG,EAAE,CAAC;QACxB,KAAK,CAAC,CAAC,CAAC,GAAG,IAAI,CAAC,CAAC,CAAC,GAAG,EAAE,CAAC;IAC1B,CAAC;SAAM,CAAC;QACN,KAAK,CAAC,CAAC,CAAC,GAAG,KAAK,CAAC,CAAC,CAAC,GAAG,KAAK,CAAC,CAAC,CAAC,GAAG,GAAG,CAAC;IACvC,CAAC;AACH,CAAC;AAED;;;;GAIG;AACH,SAAS,UAAU,CAAC,KAAmB,EAAE,IAAgB;IACvD,MAAM,GAAG,GAAG,KAAK,CAAC,CAAC,CAAC,CAAC;IACrB,MAAM,KAAK,GAAG,KAAK,CAAC,CAAC,CAAC,CAAC;IACvB,MAAM,IAAI,GAAG,KAAK,CAAC,CAAC,CAAC,CAAC;IACtB,MAAM,CAAC,GAAG,IAAI,CAAC,GAAG,CAAC,GAAG,EAAE,IAAI,CAAC,GAAG,CAAC,KAAK,EAAE,IAAI,CAAC,CAAC,CAAC;IAC/C,IAAI,CAAC,GAAG,KAAK,EAAE,CAAC;QACd,IAAI,CAAC,CAAC,CAAC,GAAG,IAAI,CAAC,CAAC,CAAC,GAAG,IAAI,CAAC,CAAC,CAAC,GAAG,IAAI,CAAC,CAAC,CAAC,GAAG,CAAC,CAAC;IAC5C,CAAC;SAAM,CAAC;QACN,MAAM,EAAE,CAAC,EAAE,CAAC,EAAE,GAAG,KAAK,CAAC,CAAC,CAAC,CAAC;QAC1B,MAAM,CAAC,GAAG,CAAC,CAAC,GAAG,CAAC,CAAC,GAAG,KAAK,CAAC;QAC1B,IAAI,CAAC,CAAC,CAAC,GAAG,GAAG,GAAG,CAAC,CAAC;QAClB,IAAI,CAAC,CAAC,CAAC,GAAG,KAAK,GAAG,CAAC,CAAC;QACpB,IAAI,CAAC,CAAC,CAAC,GAAG,IAAI,GAAG,CAAC,CAAC;QACnB,IAAI,CAAC,CAAC,CAAC,GAAG,CAAC,GAAG,GAAG,CAAC;IACpB,CAAC;AACH,CAAC;AAED;;;;;;GAMG;AACH,SAAS,SAAS,CAAC,CAAS,EAAE,CAAS,EAAE,IAAkB;IACzD,MAAM,CAAC,GAAa,EAAE,CAAC;IACvB,MAAM,IAAI,GAAG,CAAC,CAAC;IACf,cAAc,CAAC,CAAC,EAAE,CAAC,EAAE,CAAC,EAAE,IAAI,EAAE,IAAI,CAAC,CAAC;IACpC,OAAO,IAAI,UAAU,CAAC,CAAC,CAAC,CAAC;AAC3B,CAAC;AAED,SAAS,cAAc,CACrB,CAAW,EACX,CAAS,EACT,CAAS,EACT,IAAY,EACZ,IAAkB;IAElB,IAAI,CAAC,IAAI,CAAC,IAAI,CAAC,IAAI,CAAC,IAAI,CAAC,IAAI,EAAE,CAAC;QAC9B,OAAO;IACT,CAAC;IAED,MAAM,MAAM,GACV,cAAc;QACd,0BAA0B;QAC1B,gBAAgB;QAChB,IAAI;QACJ,KAAK;QACL,CAAC;QACD,MAAM;QACN,CAAC;QACD,IAAI,CAAC;IACP,MAAM,CAAC,KAAK,CAAC,EAAE,CAAC,CAAC,OAAO,CAAC,CAAC,CAAC,EAAE,EAAE;QAC7B,MAAM,QAAQ,GAAG,CAAC,CAAC,UAAU,CAAC,CAAC,CAAC,CAAC;QACjC,CAAC,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC;IACnB,CAAC,CAAC,CAAC;IAEH,MAAM,OAAO,GAAG,IAAI,UAAU,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC;IACtC,MAAM,8BAA8B,GAAG,KAAK,CAAC;IAC7C,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,CAAC,EAAE,CAAC,EAAE,EAAE,CAAC;QAC3B,kBAAkB,CAChB,CAAC,EACD,CAAC,EACD,IAAI,EACJ,OAAO,EACP,IAAI,CAAC,QAAQ,CAAC,IAAI,GAAG,CAAC,GAAG,CAAC,8BAA8B,CAAC,CAAC,CAAC,CAAC,GAAG,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAC3E,CAAC;IACJ,CAAC;AACH,CAAC;AAED,SAAS,kBAAkB,CACzB,CAAW,EACX,KAAa,EACb,KAAa,EACb,OAAmB,EACnB,QAAsB;IAEtB,MAAM,cAAc,GAAG,IAAI,UAAU,CAAC,CAAC,CAAC,EAAE,CAAC,EAAE,CAAC,EAAE,CAAC,CAAC,CAAC,CAAC;IACpD,MAAM,IAAI,GAAG,IAAI,UAAU,CAAC,CAAC,CAAC,CAAC;IAC/B,MAAM,MAAM,GAAG,IAAI,YAAY,CAAC,CAAC,CAAC,CAAC;IACnC,IAAI,CAAC,CAAC;IAEN,cAAc,CAAC,CAAC,CAAC,GAAG,CAAC,KAAK,GAAG,MAAM,CAAC,IAAI,CAAC,CAAC;IAC1C,cAAc,CAAC,CAAC,CAAC,GAAG,KAAK,GAAG,MAAM,CAAC;IAEnC,4CAA4C;IAC5C,IAAI,KAAK,GAAG,CAAC,IAAI,KAAK,IAAI,KAAK,EAAE,CAAC;QAChC,KAAK,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,KAAK,EAAE,CAAC,EAAE,EAAE,CAAC;YAC3B,QAAQ,KAAK,EAAE,CAAC;gBACd,KAAK,CAAC,CAAC,CAAC,iBAAiB;gBACzB,KAAK,CAAC;oBACJ,MAAM,CAAC,CAAC,CAAC,GAAG,QAAQ,CAAC,CAAC,GAAG,KAAK,GAAG,CAAC,CAAC,CAAC;oBACpC,MAAM,CAAC,CAAC,CAAC,GAAG,QAAQ,CAAC,CAAC,GAAG,KAAK,GAAG,CAAC,CAAC,CAAC;oBACpC,MAAM,CAAC,CAAC,CAAC,GAAG,QAAQ,CAAC,CAAC,GAAG,KAAK,GAAG,CAAC,CAAC,CAAC;oBACpC,MAAM;gBACR;oBACE,MAAM,CAAC,CAAC,CAAC,GAAG,MAAM,CAAC,CAAC,CAAC,GAAG,MAAM,CAAC,CAAC,CAAC,GAAG,QAAQ,CAAC,CAAC,GAAG,KAAK,GAAG,CAAC,CAAC,CAAC;oBAC5D,MAAM;YACV,CAAC;YACD,UAAU,CAAC,MAAM,EAAE,IAAI,CAAC,CAAC;YACzB,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,IAAI,CAAC,CAAC,CAAC,EAAE,IAAI,CAAC,CAAC,CAAC,EAAE,IAAI,CAAC,CAAC,CAAC,CAAC,CAAC;QAC7C,CAAC;IACH,CAAC;SAAM,CAAC;QACN,IAAI,CAAC,EAAE,CAAC,CAAC;QACT,gCAAgC;QAChC,KAAK,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,KAAK,EAAE,CAAC,EAAE,EAAE,CAAC;YAC3B,QAAQ,KAAK,EAAE,CAAC;gBACd,KAAK,CAAC,CAAC,CAAC,iBAAiB;gBACzB,KAAK,CAAC;oBACJ,MAAM,CAAC,CAAC,CAAC,GAAG,QAAQ,CAAC,CAAC,GAAG,KAAK,GAAG,CAAC,CAAC,CAAC;oBACpC,MAAM,CAAC,CAAC,CAAC,GAAG,QAAQ,CAAC,CAAC,GAAG,KAAK,GAAG,CAAC,CAAC,CAAC;oBACpC,MAAM,CAAC,CAAC,CAAC,GAAG,QAAQ,CAAC,CAAC,GAAG,KAAK,GAAG,CAAC,CAAC,CAAC;oBACpC,MAAM;gBACR;oBACE,MAAM,CAAC,CAAC,CAAC,GAAG,MAAM,CAAC,CAAC,CAAC,GAAG,MAAM,CAAC,CAAC,CAAC,GAAG,QAAQ,CAAC,CAAC,GAAG,KAAK,GAAG,CAAC,CAAC,CAAC;oBAC5D,MAAM;YACV,CAAC;YACD,UAAU,CAAC,MAAM,EAAE,IAAI,CAAC,CAAC;YACzB,OAAO,CAAC,CAAC,GAAG,KAAK,GAAG,CAAC,CAAC,GAAG,IAAI,CAAC,CAAC,CAAC,CAAC;YACjC,OAAO,CAAC,CAAC,GAAG,KAAK,GAAG,CAAC,CAAC,GAAG,IAAI,CAAC,CAAC,CAAC,CAAC;YACjC,OAAO,CAAC,CAAC,GAAG,KAAK,GAAG,CAAC,CAAC,GAAG,IAAI,CAAC,CAAC,CAAC,CAAC;YACjC,OAAO,CAAC,CAAC,GAAG,KAAK,GAAG,CAAC,CAAC,GAAG,IAAI,CAAC,CAAC,CAAC,CAAC;QACnC,CAAC;QAED,CAAC,CAAC,IAAI,CACJ,cAAc,CAAC,CAAC,CAAC,EACjB,cAAc,CAAC,CAAC,CAAC,EACjB,cAAc,CAAC,CAAC,CAAC,EACjB,cAAc,CAAC,CAAC,CAAC,CAClB,CAAC;QAEF,mCAAmC;QACnC,KAAK,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,CAAC,EAAE,CAAC,EAAE,EAAE,CAAC;YACvB,MAAM,IAAI,GAAG,OAAO,CAAC,QAAQ,CAAC,KAAK,GAAG,CAAC,CAAC,CAAC;YAEzC,CAAC,GAAG,CAAC,CAAC;YACN,OAAO,CAAC,GAAG,KAAK,EAAE,CAAC;gBACjB,iBAAiB;gBACjB,CAAC,GAAG,CAAC,CAAC;gBACN,OAAO,CAAC,GAAG,CAAC,GAAG,KAAK,EAAE,CAAC;oBACrB,IAAI,IAAI,CAAC,CAAC,CAAC,KAAK,IAAI,CAAC,CAAC,GAAG,CAAC,CAAC,IAAI,IAAI,CAAC,CAAC,CAAC,KAAK,IAAI,CAAC,CAAC,GAAG,CAAC,CAAC;wBAAE,MAAM;oBAC9D,EAAE,CAAC,CAAC;gBACN,CAAC;gBACD,IAAI,CAAC,GAAG,CAAC,IAAI,KAAK;oBAAE,CAAC,GAAG,KAAK,CAAC;gBAC9B,uBAAuB;gBACvB,OAAO,CAAC,GAAG,CAAC,EAAE,CAAC;oBACb,IAAI,GAAG,GAAG,CAAC,GAAG,CAAC,CAAC;oBAChB,IAAI,GAAG,GAAG,GAAG;wBAAE,GAAG,GAAG,GAAG,CAAC;oBACzB,eAAe,CAAC,CAAC,EAAE,GAAG,EAAE,IAAI,CAAC,QAAQ,CAAC,CAAC,CAAC,CAAC,CAAC;oBAC1C,CAAC,IAAI,GAAG,CAAC;gBACX,CAAC;gBACD,8BAA8B;gBAC9B,IAAI,CAAC,GAAG,CAAC,GAAG,KAAK,EAAE,CAAC;oBAClB,+EAA+E;oBAC/E,2BAA2B;oBAC3B,OAAO,CAAC,GAAG,KAAK,IAAI,IAAI,CAAC,CAAC,CAAC,IAAI,IAAI,CAAC,CAAC,CAAC;wBAAE,EAAE,CAAC,CAAC;oBAC5C,qBAAqB;oBACrB,OAAO,CAAC,GAAG,CAAC,EAAE,CAAC;wBACb,IAAI,GAAG,GAAG,CAAC,GAAG,CAAC,CAAC;wBAChB,IAAI,GAAG,GAAG,GAAG;4BAAE,GAAG,GAAG,GAAG,CAAC;wBACzB,cAAc,CAAC,CAAC,EAAE,GAAG,EAAE,IAAI,CAAC,CAAC,CAAC,CAAC,CAAC;wBAChC,CAAC,IAAI,GAAG,CAAC;oBACX,CAAC;gBACH,CAAC;YACH,CAAC;QACH,CAAC;IACH,CAAC;AACH,CAAC;AAED,SAAS,eAAe,CAAC,CAAW,EAAE,MAAc,EAAE,IAAgB;IACpE,MAAM,UAAU,GAAG,MAAM,GAAG,IAAI,CAAC;IACjC,IAAI,CAAC,CAAC,MAAM,IAAI,GAAG,CAAC;QAAE,MAAM,IAAI,KAAK,CAAC,4BAA4B,CAAC,CAAC;IACpE,CAAC,CAAC,IAAI,CAAC,UAAU,CAAC,CAAC;IACnB,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,MAAM,EAAE,CAAC,EAAE,EAAE,CAAC;QAChC,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,CAAC;IAClB,CAAC;AACH,CAAC;AAED,SAAS,cAAc,CAAC,CAAW,EAAE,MAAc,EAAE,QAAgB;IACnE,MAAM,UAAU,GAAG,CAAC,MAAM,GAAG,GAAG,CAAC,GAAG,IAAI,CAAC;IACzC,IAAI,CAAC,CAAC,MAAM,GAAG,GAAG,IAAI,GAAG,CAAC;QAAE,MAAM,IAAI,KAAK,CAAC,4BAA4B,CAAC,CAAC;IAC1E,CAAC,CAAC,IAAI,CAAC,UAAU,EAAE,QAAQ,CAAC,CAAC;AAC/B,CAAC;AAED,SAAS,SAAS,CAAC,KAAiB,EAAE,SAAiB;IACrD,MAAM,CAAC,GAAG,SAAS,CAAC,MAAM,CAAC;IAC3B,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,CAAC,EAAE,CAAC,EAAE,EAAE,CAAC;QAC3B,IAAI,KAAK,CAAC,CAAC,CAAC,KAAK,SAAS,CAAC,UAAU,CAAC,CAAC,CAAC,EAAE,CAAC;YACzC,OAAO,KAAK,CAAC;QACf,CAAC;IACH,CAAC;IACD,OAAO,IAAI,CAAC;AACd,CAAC;AAED,SAAS,KAAK,CAAC,KAAiB;IAC9B,IAAI,CAAC,GAAG,SAAS,CAAC,KAAK,EAAE,cAAc,CAAC,CAAC;IACzC,IAAI,CAAC,CAAC,EAAE,CAAC;QACP,CAAC,GAAG,SAAS,CAAC,KAAK,EAAE,UAAU,CAAC,CAAC;IACnC,CAAC;IACD,OAAO,CAAC,CAAC;AACX,CAAC;AAED,MAAM,iBAAiB,GAAG,IAAI,GAAG,EAAE,CAAC;AACpC,MAAM,cAAc,GAAG,CAAC,IAAI,EAAE,CAAC;AAE/B;;;;GAIG;AACH,SAAS,QAAQ,CAAC,KAAiB;IAOjC,IAAI,MAAM,GAAG,EAAE,CAAC;IAChB,IAAI,GAAG,GAAG,CAAC,CAAC;IAEZ,IAAI,CAAC,KAAK,CAAC,KAAK,CAAC,EAAE,CAAC;QAClB,OAAO,oBAAoB,CAAC;IAC9B,CAAC;IAED,cAAc;IACd,OAAO,CAAC,MAAM,CAAC,KAAK,CAAC,eAAe,CAAC,IAAI,GAAG,GAAG,iBAAiB,EAAE,CAAC;QACjE,MAAM,IAAI,MAAM,CAAC,YAAY,CAAC,KAAK,CAAC,GAAG,EAAE,CAAC,CAAC,CAAC;IAC9C,CAAC;IAED,eAAe;IACf,MAAM,MAAM,GAAG,MAAM,CAAC,KAAK,CAAC,eAAe,CAAC,EAAE,CAAC,CAAC,CAAC,CAAC;IAClD,IAAI,MAAM,KAAK,iBAAiB,EAAE,CAAC;QACjC,OAAO,0BAA0B,GAAG,MAAM,CAAC;IAC7C,CAAC;IAED,mBAAmB;IACnB,MAAM,GAAG,GAAa,MAAM,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC,OAAO,EAAE,CAAC,CAAC,CAAC,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC;IACjE,IAAI,GAAG,CAAC,CAAC,CAAC,KAAK,IAAI,IAAI,GAAG,CAAC,CAAC,CAAC,KAAK,IAAI,EAAE,CAAC;QACvC,OAAO,wBAAwB,CAAC;IAClC,CAAC;IACD,MAAM,KAAK,GAAG,MAAM,CAAC,UAAU,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC,CAAC;IACxC,MAAM,MAAM,GAAG,MAAM,CAAC,UAAU,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC,CAAC;IACzC,IAAI,KAAK,GAAG,cAAc,IAAI,MAAM,GAAG,cAAc,EAAE,CAAC;QACtD,OAAO,6BAA6B,CAAC;IACvC,CAAC;IAED,IAAI,CAAC,EAAE,CAAC,CAAC;IACT,IAAI,EAAE,GAAW,KAAK,CAAC,GAAG,CAAC,CAAC;IAC5B,IAAI,EAAE,GAAW,KAAK,CAAC,GAAG,GAAG,CAAC,CAAC,CAAC;IAChC,IAAI,GAAG,GAAW,KAAK,CAAC,GAAG,GAAG,CAAC,CAAC,CAAC;IAEjC,4EAA4E;IAC5E,sEAAsE;IACtE,MAAM,MAAM,GAAY,EAAE,KAAK,CAAC,IAAI,EAAE,KAAK,CAAC,IAAI,CAAC,CAAC,CAAC,GAAG,GAAG,IAAI,CAAC,CAAC,CAAC,yBAAyB;IAEzF,MAAM,OAAO,GAAG,IAAI,YAAY,CAAC,KAAK,GAAG,MAAM,GAAG,CAAC,CAAC,CAAC;IACrD,IAAI,KAAK,GAAG,CAAC,IAAI,KAAK,IAAI,KAAK,IAAI,MAAM,EAAE,CAAC;QAC1C,cAAc;QACd,iBAAiB;QACjB,KAAK,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,MAAM,EAAE,EAAE,CAAC,EAAE,CAAC;YAC5B,KAAK,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,KAAK,EAAE,EAAE,CAAC,EAAE,CAAC;gBAC3B,MAAM,IAAI,GAAG,KAAK,CAAC,QAAQ,CAAC,GAAG,EAAE,GAAG,GAAG,CAAC,CAAC,CAAC;gBAC1C,GAAG,IAAI,CAAC,CAAC;gBACT,MAAM,KAAK,GAAG,CAAC,CAAC,GAAG,KAAK,GAAG,CAAC,CAAC,GAAG,CAAC,CAAC;gBAClC,UAAU,CAAC,IAAI,EAAE,OAAO,CAAC,QAAQ,CAAC,KAAK,EAAE,KAAK,GAAG,CAAC,CAAC,CAAC,CAAC;YACvD,CAAC;QACH,CAAC;IACH,CAAC;SAAM,CAAC;QACN,wBAAwB;QACxB,IAAI,QAAgC,CAAC;QACrC,IAAI,EAAU,CAAC;QACf,IAAI,EAAU,CAAC;QACf,IAAI,GAAW,CAAC;QAChB,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,MAAM,EAAE,CAAC,EAAE,EAAE,CAAC;YAChC,EAAE,GAAG,KAAK,CAAC,GAAG,EAAE,CAAC,CAAC;YAClB,EAAE,GAAG,KAAK,CAAC,GAAG,EAAE,CAAC,CAAC;YAClB,GAAG,GAAG,KAAK,CAAC,GAAG,EAAE,CAAC,CAAC;YACnB,IAAI,EAAE,KAAK,CAAC,IAAI,EAAE,KAAK,CAAC,IAAI,GAAG,GAAG,IAAI,EAAE,CAAC;gBACvC,OAAO,kBAAkB,CAAC;YAC5B,CAAC;YAED,GAAG,GAAG,GAAG,IAAI,CAAC,CAAC;YACf,GAAG,IAAI,KAAK,CAAC,GAAG,EAAE,CAAC,CAAC;YACpB,IAAI,GAAG,KAAK,KAAK,EAAE,CAAC;gBAClB,OAAO,iCAAiC,CAAC;YAC3C,CAAC;YACD,IAAI,CAAC,QAAQ,EAAE,CAAC;gBACd,QAAQ,GAAG,IAAI,UAAU,CAAC,KAAK,GAAG,CAAC,CAAC,CAAC;YACvC,CAAC;YAED,IAAI,KAAa,CAAC;YAClB,IAAI,KAAa,CAAC;YAClB,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,CAAC,EAAE,CAAC,EAAE,EAAE,CAAC;gBAC3B,IAAI,KAAa,CAAC;gBAClB,CAAC,GAAG,CAAC,CAAC;gBACN,OAAO,CAAC,KAAK,GAAG,KAAK,GAAG,CAAC,CAAC,GAAG,CAAC,EAAE,CAAC;oBAC/B,KAAK,GAAG,KAAK,CAAC,GAAG,EAAE,CAAC,CAAC;oBACrB,IAAI,KAAK,GAAG,GAAG,EAAE,CAAC;wBAChB,SAAS;wBACT,KAAK,GAAG,KAAK,CAAC,GAAG,EAAE,CAAC,CAAC;wBACrB,KAAK,IAAI,GAAG,CAAC;wBACb,IAAI,KAAK,GAAG,KAAK,EAAE,CAAC;4BAClB,OAAO,qBAAqB,CAAC;wBAC/B,CAAC;wBACD,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,KAAK,EAAE,CAAC,EAAE,EAAE,CAAC;4BAC/B,QAAQ,CAAC,CAAC,EAAE,GAAG,CAAC,GAAG,CAAC,CAAC,GAAG,KAAK,CAAC;wBAChC,CAAC;oBACH,CAAC;yBAAM,CAAC;wBACN,UAAU;wBACV,IAAI,KAAK,GAAG,KAAK,EAAE,CAAC;4BAClB,OAAO,qBAAqB,CAAC;wBAC/B,CAAC;wBACD,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,KAAK,EAAE,CAAC,EAAE,EAAE,CAAC;4BAC/B,QAAQ,CAAC,CAAC,EAAE,GAAG,CAAC,GAAG,CAAC,CAAC,GAAG,KAAK,CAAC,GAAG,EAAE,CAAC,CAAC;wBACvC,CAAC;oBACH,CAAC;gBACH,CAAC;YACH,CAAC;YAED,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,KAAK,EAAE,CAAC,EAAE,EAAE,CAAC;gBAC/B,UAAU,CACR,QAAQ,CAAC,QAAQ,CAAC,CAAC,GAAG,CAAC,CAAC,EACxB,OAAO,CAAC,QAAQ,CAAC,CAAC,CAAC,GAAG,KAAK,GAAG,CAAC,CAAC,GAAG,CAAC,CAAC,CACtC,CAAC;YACJ,CAAC;QACH,CAAC;IACH,CAAC;IAED,OAAO;QACL,QAAQ,EAAE,OAAO;QACjB,KAAK,EAAE,KAAK;QACZ,MAAM,EAAE,MAAM;KACf,CAAC;AACJ,CAAC;AAED,OAAO,EAAE,QAAQ,IAAI,OAAO,EAAE,SAAS,IAAI,QAAQ,EAAE,UAAU,EAAE,UAAU,EAAE,CAAC"}

package/lib/helper.d.ts

@@ -0,0 +1 @@
+export declare function profileAndLogKernelCode(execute: () => void, disabled?: boolean): void;

package/lib/helper.js

@@ -0,0 +1,27 @@
+import * as tfjs from '@tensorflow/tfjs-core';
+export function profileAndLogKernelCode(execute, disabled = true) {
+    if (disabled) {
+        execute();
+        return;
+    }
+    tfjs
+        .profile(() => {
+        execute();
+    })
+        .then((res) => {
+        const kernelNames = Array.from(new Set(res.kernels.map((k) => k.name).filter((name) => !name.endsWith('_op'))));
+        function nameToConfig(name) {
+            return `${name[0].toLowerCase()}${name.slice(1)}Config`;
+        }
+        const importCode = kernelNames.map((name) => `import { ${nameToConfig(name)} } from '@tensorflow/tfjs-backend-webgpu/dist/kernels/${name}';`);
+        const configCode = kernelNames.map((name) => `${nameToConfig(name)},`);
+        const code = `
+  ${importCode.join('\n')}
+  const kernelConfigs: KernelConfig[] = [
+  ${configCode.join('\n')}
+  ]
+  `;
+        console.log(code);
+    });
+}
+//# sourceMappingURL=helper.js.map

package/lib/helper.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"helper.js","sourceRoot":"","sources":["../src/helper.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,IAAI,MAAM,uBAAuB,CAAC;AAC9C,MAAM,UAAU,uBAAuB,CAAC,OAAmB,EAAE,QAAQ,GAAG,IAAI;IAC1E,IAAI,QAAQ,EAAE,CAAC;QACb,OAAO,EAAE,CAAC;QACV,OAAO;IACT,CAAC;IACD,IAAI;SACD,OAAO,CAAC,GAAG,EAAE;QACZ,OAAO,EAAE,CAAC;IACZ,CAAC,CAAC;SACD,IAAI,CAAC,CAAC,GAAG,EAAE,EAAE;QACZ,MAAM,WAAW,GAAG,KAAK,CAAC,IAAI,CAC5B,IAAI,GAAG,CACL,GAAG,CAAC,OAAO,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,MAAM,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,CAAC,IAAI,CAAC,QAAQ,CAAC,KAAK,CAAC,CAAC,CACvE,CACF,CAAC;QACF,SAAS,YAAY,CAAC,IAAY;YAChC,OAAO,GAAG,IAAI,CAAC,CAAC,CAAC,CAAC,WAAW,EAAE,GAAG,IAAI,CAAC,KAAK,CAAC,CAAC,CAAC,QAAQ,CAAC;QAC1D,CAAC;QACD,MAAM,UAAU,GAAG,WAAW,CAAC,GAAG,CAChC,CAAC,IAAI,EAAE,EAAE,CACP,YAAY,YAAY,CACtB,IAAI,CACL,yDAAyD,IAAI,IAAI,CACrE,CAAC;QACF,MAAM,UAAU,GAAG,WAAW,CAAC,GAAG,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,GAAG,YAAY,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC;QACvE,MAAM,IAAI,GAAG;IACf,UAAU,CAAC,IAAI,CAAC,IAAI,CAAC;;IAErB,UAAU,CAAC,IAAI,CAAC,IAAI,CAAC;;GAEtB,CAAC;QACE,OAAO,CAAC,GAAG,CAAC,IAAI,CAAC,CAAC;IACpB,CAAC,CAAC,CAAC;AACP,CAAC"}

package/lib/kernels.d.ts

@@ -0,0 +1 @@
+export {};

package/lib/kernels.js

@@ -0,0 +1,26 @@
+import { registerKernel } from '@tensorflow/tfjs-core/dist/kernel_registry';
+import { mirrorPadConfig } from '@tensorflow/tfjs-backend-webgpu/dist/kernels/MirrorPad';
+import { padV2Config } from '@tensorflow/tfjs-backend-webgpu/dist/kernels/PadV2';
+import { sliceConfig } from '@tensorflow/tfjs-backend-webgpu/dist/kernels/Slice';
+import { fusedConv2DConfig } from '@tensorflow/tfjs-backend-webgpu/dist/kernels/FusedConv2D';
+import { maxPoolConfig } from '@tensorflow/tfjs-backend-webgpu/dist/kernels/MaxPool';
+import { resizeNearestNeighborConfig } from '@tensorflow/tfjs-backend-webgpu/dist/kernels/ResizeNearestNeighbor';
+import { concatConfig } from '@tensorflow/tfjs-backend-webgpu/dist/kernels/Concat';
+import { identityConfig } from '@tensorflow/tfjs-backend-webgpu/dist/kernels/Identity';
+const kernelConfigs = [
+    mirrorPadConfig,
+    padV2Config,
+    sliceConfig,
+    fusedConv2DConfig,
+    maxPoolConfig,
+    resizeNearestNeighborConfig,
+    concatConfig,
+    identityConfig
+];
+for (const kernelConfig of kernelConfigs) {
+    registerKernel({
+        ...kernelConfig,
+        backendName: 'webgpu-oidn'
+    });
+}
+//# sourceMappingURL=kernels.js.map

package/lib/kernels.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"kernels.js","sourceRoot":"","sources":["../src/kernels.ts"],"names":[],"mappings":"AAAA,OAAO,EAEL,cAAc,EACf,MAAM,4CAA4C,CAAC;AAEpD,OAAO,EAAE,eAAe,EAAE,MAAM,wDAAwD,CAAC;AACzF,OAAO,EAAE,WAAW,EAAE,MAAM,oDAAoD,CAAC;AACjF,OAAO,EAAE,WAAW,EAAE,MAAM,oDAAoD,CAAC;AACjF,OAAO,EAAE,iBAAiB,EAAE,MAAM,0DAA0D,CAAC;AAC7F,OAAO,EAAE,aAAa,EAAE,MAAM,sDAAsD,CAAC;AACrF,OAAO,EAAE,2BAA2B,EAAE,MAAM,oEAAoE,CAAC;AACjH,OAAO,EAAE,YAAY,EAAE,MAAM,qDAAqD,CAAC;AACnF,OAAO,EAAE,cAAc,EAAE,MAAM,uDAAuD,CAAC;AAEvF,MAAM,aAAa,GAAmB;IACpC,eAAe;IACf,WAAW;IACX,WAAW;IACX,iBAAiB;IACjB,aAAa;IACb,2BAA2B;IAC3B,YAAY;IACZ,cAAc;CACf,CAAC;AAEF,KAAK,MAAM,YAAY,IAAI,aAAa,EAAE,CAAC;IACzC,cAAc,CAAC;QACb,GAAG,YAAY;QACf,WAAW,EAAE,aAAa;KAC3B,CAAC,CAAC;AACL,CAAC"}

package/lib/main.d.ts

@@ -0,0 +1,20 @@
+/// <reference types="@webgpu/types/dist" />
+import { parseTZA } from './tza';
+import UNet from './UNet';
+export { parseTZA, UNet };
+export declare function initUNetFromBuffer(tzaBuffer: ArrayBuffer, backendParams?: {
+    device: GPUDevice;
+    adapterInfo: GPUAdapterInfo;
+}, opts?: {
+    aux?: boolean;
+    hdr?: boolean;
+    maxTileSize?: number;
+}): Promise<UNet>;
+export declare function initUNetFromURL(modelPath: string, backendParams?: {
+    device: GPUDevice;
+    adapterInfo: GPUAdapterInfo;
+}, opts?: {
+    aux?: boolean;
+    hdr?: boolean;
+    maxTileSize?: number;
+}): Promise<UNet>;

package/lib/main.js

@@ -0,0 +1,20 @@
+import { parseTZA } from './tza';
+import UNet from './UNet';
+import { initWebGPUBackend, initWebGPUBackendWithDevice } from './backend';
+export { parseTZA, UNet };
+export async function initUNetFromBuffer(tzaBuffer, backendParams, opts) {
+    const backend = await (backendParams
+        ? initWebGPUBackendWithDevice(backendParams.device, backendParams.adapterInfo)
+        : initWebGPUBackend());
+    const tensors = parseTZA(tzaBuffer);
+    const unet = new UNet(tensors, backend, opts);
+    return unet;
+}
+export async function initUNetFromURL(modelPath, backendParams, opts) {
+    return fetch(modelPath)
+        .then((res) => res.arrayBuffer())
+        .then((ab) => {
+        return initUNetFromBuffer(ab, backendParams, opts);
+    });
+}
+//# sourceMappingURL=main.js.map

package/lib/main.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"main.js","sourceRoot":"","sources":["../src/main.ts"],"names":[],"mappings":"AACA,OAAO,EAAE,QAAQ,EAAE,MAAM,OAAO,CAAC;AACjC,OAAO,IAAI,MAAM,QAAQ,CAAC;AAC1B,OAAO,EAAE,iBAAiB,EAAE,2BAA2B,EAAE,MAAM,WAAW,CAAC;AAE3E,OAAO,EAAE,QAAQ,EAAE,IAAI,EAAE,CAAC;AAE1B,MAAM,CAAC,KAAK,UAAU,kBAAkB,CACtC,SAAsB,EACtB,aAAkE,EAClE,IAIC;IAED,MAAM,OAAO,GAAG,MAAM,CAAC,aAAa;QAClC,CAAC,CAAC,2BAA2B,CACzB,aAAa,CAAC,MAAM,EACpB,aAAa,CAAC,WAAW,CAC1B;QACH,CAAC,CAAC,iBAAiB,EAAE,CAAC,CAAC;IACzB,MAAM,OAAO,GAAG,QAAQ,CAAC,SAAS,CAAC,CAAC;IACpC,MAAM,IAAI,GAAG,IAAI,IAAI,CAAC,OAAO,EAAE,OAAQ,EAAE,IAAI,CAAC,CAAC;IAC/C,OAAO,IAAI,CAAC;AACd,CAAC;AAED,MAAM,CAAC,KAAK,UAAU,eAAe,CACnC,SAAiB,EACjB,aAAkE,EAClE,IAIC;IAED,OAAO,KAAK,CAAC,SAAS,CAAC;SACpB,IAAI,CAAC,CAAC,GAAG,EAAE,EAAE,CAAC,GAAG,CAAC,WAAW,EAAE,CAAC;SAChC,IAAI,CAAC,CAAC,EAAE,EAAE,EAAE;QACX,OAAO,kBAAkB,CAAC,EAAE,EAAE,aAAa,EAAE,IAAI,CAAC,CAAC;IACrD,CAAC,CAAC,CAAC;AACP,CAAC"}

package/lib/process.d.ts

@@ -0,0 +1,39 @@
+/// <reference types="@webgpu/types/dist" />
+export declare class Tile {
+    x: number;
+    y: number;
+    width: number;
+    height: number;
+    constructor(x: number, y: number, width: number, height: number);
+}
+export declare function avgLogLum({ data, channels }: {
+    data: Float32Array;
+    channels: number;
+}): number;
+export declare function hdrTransferFuncCPU({ data, channels, inputScale }: {
+    data: Float32Array;
+    channels: number;
+    inputScale: number;
+}): Float32Array;
+export declare function hdrTransferFuncInverseCPU({ data, channels, inputScale }: {
+    data: Float32Array;
+    channels: number;
+    inputScale: number;
+}): Float32Array;
+export declare class GPUDataProcess {
+    private _device;
+    private _inputPass;
+    private _outputPass;
+    constructor(_device: GPUDevice);
+    setImageSize(w: number, h: number): void;
+    setInputTile(tile: Tile): void;
+    setOutputTile(dstTile: Tile, srcTile: Tile): void;
+    forward(colorBuffer: GPUBuffer, albedoBuffer: GPUBuffer, normalBuffer: GPUBuffer): {
+        color: GPUBuffer;
+        albedo: GPUBuffer;
+        normal: GPUBuffer;
+    };
+    inverse(buffer: GPUBuffer, raw: GPUBuffer): GPUBuffer;
+    copyInputDataToOutput(inputColorBuffer: GPUBuffer): void;
+    dispose(): void;
+}