hama-js 1.3.3 → 1.3.4

package/dist/browser/browser.js

@@ -1,24 +1,99 @@
 import { InferenceSession, Tensor } from "onnxruntime-web";
-import { decodeIdsToResult, encodeText } from "./tokenizer.js";
+import { decodeIdsToResult, decoderIds, encodeText } from "./tokenizer.js";
 const DEFAULT_MODEL_URL = new URL("./assets/g2p_fp16.onnx", import.meta.url).toString();
+const DEFAULT_ENCODER_URL = new URL("./assets/encoder.onnx", import.meta.url).toString();
+const DEFAULT_DECODER_STEP_URL = new URL("./assets/decoder_step.onnx", import.meta.url).toString();
+const resolveName = (available, primary, ...fallbacks) => {
+    if (available.includes(primary))
+        return primary;
+    for (const fallback of fallbacks) {
+        if (available.includes(fallback))
+            return fallback;
+    }
+    const matches = available.filter((name) => name.startsWith(`${primary}.`) || name.startsWith(primary));
+    if (matches.length === 1)
+        return matches[0];
+    if (matches.length > 1) {
+        const numeric = matches.filter((name) => name.startsWith(`${primary}.`));
+        if (numeric.length === 1)
+            return numeric[0];
+        return matches[0];
+    }
+    throw new Error(`Could not resolve ONNX tensor name for '${primary}'. Available: ${available.join(", ")}`);
+};
 export class G2PBrowserModel {
     constructor(options) {
         this.options = options;
     }
     static async create(options = {}) {
+        if ((options.encoderUrl === undefined) !== (options.decoderStepUrl === undefined)) {
+            throw new Error("encoderUrl and decoderStepUrl must be provided together");
+        }
         const opts = {
             modelUrl: options.modelUrl ?? DEFAULT_MODEL_URL,
+            encoderUrl: options.encoderUrl ?? DEFAULT_ENCODER_URL,
+            decoderStepUrl: options.decoderStepUrl ?? DEFAULT_DECODER_STEP_URL,
             maxInputLen: options.maxInputLen ?? 128,
             // Retained for API compatibility; autoregressive ONNX sets output length in-graph.
             maxOutputLen: options.maxOutputLen ?? 32,
         };
         const model = new G2PBrowserModel(opts);
-        model.session = await InferenceSession.create(opts.modelUrl, {
-            executionProviders: ["wasm"],
-        });
+        const useExplicitSplit = options.encoderUrl !== undefined && options.decoderStepUrl !== undefined;
+        if (useExplicitSplit) {
+            const [encoderSession, decoderStepSession] = await Promise.all([
+                InferenceSession.create(opts.encoderUrl, { executionProviders: ["wasm"] }),
+                InferenceSession.create(opts.decoderStepUrl, { executionProviders: ["wasm"] }),
+            ]);
+            model.encoderSession = encoderSession;
+            model.decoderStepSession = decoderStepSession;
+            return model;
+        }
+        try {
+            const [encoderSession, decoderStepSession] = await Promise.all([
+                InferenceSession.create(opts.encoderUrl, { executionProviders: ["wasm"] }),
+                InferenceSession.create(opts.decoderStepUrl, { executionProviders: ["wasm"] }),
+            ]);
+            model.encoderSession = encoderSession;
+            model.decoderStepSession = decoderStepSession;
+        }
+        catch {
+            model.session = await InferenceSession.create(opts.modelUrl, {
+                executionProviders: ["wasm"],
+            });
+        }
         return model;
     }
-    async predict(text) {
+    async predict(text, options = {}) {
+        const splitDelimiter = options.splitDelimiter ?? /\s+/u;
+        const outputDelimiter = options.outputDelimiter ?? " ";
+        const segments = splitSegments(text, splitDelimiter);
+        if (segments.length === 0) {
+            return this.predictSingle(text, 0);
+        }
+        const results = await Promise.all(segments.map(async (segment) => this.predictSingle(segment.text, codePointOffset(text, segment.startCodeUnit))));
+        const ipaParts = [];
+        const alignments = [];
+        for (let i = 0; i < results.length; i++) {
+            if (i > 0)
+                ipaParts.push(outputDelimiter);
+            ipaParts.push(results[i].ipa);
+            for (const alignment of results[i].alignments) {
+                alignments.push({
+                    phoneme: alignment.phoneme,
+                    phonemeIndex: alignments.length,
+                    charIndex: alignment.charIndex,
+                });
+            }
+        }
+        return { ipa: ipaParts.join(""), alignments };
+    }
+    async predictSingle(text, baseCharIndex) {
+        if (this.encoderSession && this.decoderStepSession) {
+            return this.predictSingleSplit(text, baseCharIndex);
+        }
+        if (!this.session) {
+            throw new Error("No ONNX session initialized");
+        }
         const encoded = encodeText(text, this.options.maxInputLen);
         const inputIds = BigInt64Array.from(encoded.ids);
         const inputLengths = new BigInt64Array([BigInt(encoded.length || 1)]);
@@ -29,7 +104,149 @@ export class G2PBrowserModel {
         const outputs = await this.session.run(feeds);
         const decoded = outputs.decoded_ids.data;
         const attn = outputs.attn_indices.data;
-        return decodeIdsToResult(decoded, attn, encoded.positionMap);
+        const result = decodeIdsToResult(decoded, attn, encoded.positionMap);
+        return {
+            ipa: result.ipa,
+            alignments: result.alignments.map((alignment, idx) => ({
+                phoneme: alignment.phoneme,
+                phonemeIndex: idx,
+                charIndex: alignment.charIndex < 0 ? alignment.charIndex : alignment.charIndex + baseCharIndex,
+            })),
+        };
+    }
+    async predictSingleSplit(text, baseCharIndex) {
+        if (!this.encoderSession || !this.decoderStepSession) {
+            throw new Error("Split ONNX sessions are not initialized");
+        }
+        const encoded = encodeText(text, this.options.maxInputLen);
+        const inputIds = BigInt64Array.from(encoded.ids);
+        const inputLengths = new BigInt64Array([BigInt(encoded.length || 1)]);
+        const encoderFeeds = {
+            input_ids: new Tensor("int64", inputIds, [1, this.options.maxInputLen]),
+            input_lengths: new Tensor("int64", inputLengths, [1]),
+        };
+        const encoderOutputs = await this.encoderSession.run(encoderFeeds);
+        const encoderOutputNames = this.encoderSession.outputNames;
+        const decoderInputNames = this.decoderStepSession.inputNames;
+        const decoderOutputNames = this.decoderStepSession.outputNames;
+        const encNames = {
+            encoder_outputs: resolveName(encoderOutputNames, "encoder_outputs"),
+            projected_keys: resolveName(encoderOutputNames, "projected_keys"),
+            encoder_mask: resolveName(encoderOutputNames, "encoder_mask"),
+            hidden: resolveName(encoderOutputNames, "hidden"),
+            prev_attn: resolveName(encoderOutputNames, "prev_attn"),
+        };
+        const decIn = {
+            decoder_input_ids: resolveName(decoderInputNames, "decoder_input_ids"),
+            encoder_outputs: resolveName(decoderInputNames, "encoder_outputs"),
+            projected_keys: resolveName(decoderInputNames, "projected_keys"),
+            encoder_mask: resolveName(decoderInputNames, "encoder_mask"),
+            prev_attn: resolveName(decoderInputNames, "prev_attn", "prev_attn_in"),
+            hidden: resolveName(decoderInputNames, "hidden", "hidden_in"),
+            positions: resolveName(decoderInputNames, "positions"),
+        };
+        const decOut = {
+            next_token_ids: resolveName(decoderOutputNames, "next_token_ids"),
+            hidden: resolveName(decoderOutputNames, "hidden_out", "hidden"),
+            prev_attn: resolveName(decoderOutputNames, "prev_attn_out", "prev_attn"),
+            attn_argmax: resolveName(decoderOutputNames, "attn_argmax"),
+        };
+        const encoderStates = {
+            encoder_outputs: encoderOutputs[encNames.encoder_outputs],
+            projected_keys: encoderOutputs[encNames.projected_keys],
+            encoder_mask: encoderOutputs[encNames.encoder_mask],
+            hidden: encoderOutputs[encNames.hidden],
+            prev_attn: encoderOutputs[encNames.prev_attn],
+        };
+        const srcLen = Number(encoderStates.encoder_outputs.dims[1] ?? 0);
+        const positions = new Float32Array(srcLen);
+        for (let i = 0; i < srcLen; i++)
+            positions[i] = i;
+        const positionsTensor = new Tensor("float32", positions, [1, srcLen]);
+        let decoderInput = new Tensor("int64", new BigInt64Array([BigInt(decoderIds.sos)]), [1, 1]);
+        let hidden = encoderStates.hidden;
+        let prevAttn = encoderStates.prev_attn;
+        const decoded = [];
+        const attnIndices = [];
+        for (let step = 0; step < this.options.maxOutputLen; step++) {
+            const stepOutputs = await this.decoderStepSession.run({
+                [decIn.decoder_input_ids]: decoderInput,
+                [decIn.encoder_outputs]: encoderStates.encoder_outputs,
+                [decIn.projected_keys]: encoderStates.projected_keys,
+                [decIn.encoder_mask]: encoderStates.encoder_mask,
+                [decIn.prev_attn]: prevAttn,
+                [decIn.hidden]: hidden,
+                [decIn.positions]: positionsTensor,
+            });
+            const nextToken = firstInt64(stepOutputs[decOut.next_token_ids]);
+            const attnIdx = firstInt64(stepOutputs[decOut.attn_argmax]);
+            decoded.push(nextToken);
+            attnIndices.push(attnIdx);
+            hidden = stepOutputs[decOut.hidden];
+            prevAttn = stepOutputs[decOut.prev_attn];
+            decoderInput = new Tensor("int64", new BigInt64Array([nextToken]), [1, 1]);
+            if (nextToken === BigInt(decoderIds.eos)) {
+                break;
+            }
+        }
+        const result = decodeIdsToResult(decoded, attnIndices, encoded.positionMap);
+        return {
+            ipa: result.ipa,
+            alignments: result.alignments.map((alignment, idx) => ({
+                phoneme: alignment.phoneme,
+                phonemeIndex: idx,
+                charIndex: alignment.charIndex < 0 ? alignment.charIndex : alignment.charIndex + baseCharIndex,
+            })),
+        };
     }
 }
+const firstInt64 = (tensor) => {
+    const data = tensor.data;
+    const value = data[0];
+    return typeof value === "bigint" ? value : BigInt(Math.trunc(Number(value)));
+};
+const escapeRegex = (value) => value.replace(/[.*+?^${}()|[\]\\]/g, "\\$&");
+const toGlobalRegex = (delimiter) => {
+    if (typeof delimiter === "string") {
+        if (delimiter.length === 0) {
+            throw new Error("splitDelimiter must not be an empty string");
+        }
+        return new RegExp(escapeRegex(delimiter), "gu");
+    }
+    const flags = delimiter.flags.includes("g") ? delimiter.flags : `${delimiter.flags}g`;
+    return new RegExp(delimiter.source, flags.includes("u") ? flags : `${flags}u`);
+};
+const splitSegments = (text, delimiter) => {
+    if (delimiter === null) {
+        return [{ text, startCodeUnit: 0 }];
+    }
+    const regex = toGlobalRegex(delimiter);
+    if (regex.test("")) {
+        throw new Error("splitDelimiter must not match an empty string");
+    }
+    regex.lastIndex = 0;
+    const segments = [];
+    let start = 0;
+    for (const match of text.matchAll(regex)) {
+        const end = match.index ?? 0;
+        if (end > start) {
+            segments.push({ text: text.slice(start, end), startCodeUnit: start });
+        }
+        start = end + match[0].length;
+    }
+    if (start < text.length) {
+        segments.push({ text: text.slice(start), startCodeUnit: start });
+    }
+    return segments;
+};
+const codePointOffset = (text, codeUnitOffset) => {
+    let codePointIndex = 0;
+    let i = 0;
+    while (i < codeUnitOffset) {
+        const cp = text.codePointAt(i) ?? 0;
+        i += cp > 0xffff ? 2 : 1;
+        codePointIndex += 1;
+    }
+    return codePointIndex;
+};
 //# sourceMappingURL=browser.js.map

package/dist/browser/browser.js.map

@@ -1 +1 @@
-{"version":3,"file":"browser.js","sourceRoot":"","sources":["../../src/browser.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,gBAAgB,EAAE,MAAM,EAAE,MAAM,iBAAiB,CAAC;AAE3D,OAAO,EAAE,iBAAiB,EAAE,UAAU,EAAa,MAAM,gBAAgB,CAAC;AAQ1E,MAAM,iBAAiB,GAAG,IAAI,GAAG,CAAC,wBAAwB,EAAE,MAAM,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC,QAAQ,EAAE,CAAC;AAExF,MAAM,OAAO,eAAe;IAI1B,YAAoB,OAAiC;QACnD,IAAI,CAAC,OAAO,GAAG,OAAO,CAAC;IACzB,CAAC;IAED,MAAM,CAAC,KAAK,CAAC,MAAM,CAAC,UAA0B,EAAE;QAC9C,MAAM,IAAI,GAA6B;YACrC,QAAQ,EAAE,OAAO,CAAC,QAAQ,IAAI,iBAAiB;YAC/C,WAAW,EAAE,OAAO,CAAC,WAAW,IAAI,GAAG;YACvC,mFAAmF;YACnF,YAAY,EAAE,OAAO,CAAC,YAAY,IAAI,EAAE;SACzC,CAAC;QACF,MAAM,KAAK,GAAG,IAAI,eAAe,CAAC,IAAI,CAAC,CAAC;QACxC,KAAK,CAAC,OAAO,GAAG,MAAM,gBAAgB,CAAC,MAAM,CAAC,IAAI,CAAC,QAAQ,EAAE;YAC3D,kBAAkB,EAAE,CAAC,MAAM,CAAC;SAC7B,CAAC,CAAC;QACH,OAAO,KAAK,CAAC;IACf,CAAC;IAED,KAAK,CAAC,OAAO,CAAC,IAAY;QACxB,MAAM,OAAO,GAAG,UAAU,CAAC,IAAI,EAAE,IAAI,CAAC,OAAO,CAAC,WAAW,CAAC,CAAC;QAC3D,MAAM,QAAQ,GAAG,aAAa,CAAC,IAAI,CAAC,OAAO,CAAC,GAAG,CAAC,CAAC;QACjD,MAAM,YAAY,GAAG,IAAI,aAAa,CAAC,CAAC,MAAM,CAAC,OAAO,CAAC,MAAM,IAAI,CAAC,CAAC,CAAC,CAAC,CAAC;QAEtE,MAAM,KAAK,GAA2B;YACpC,SAAS,EAAE,IAAI,MAAM,CAAC,OAAO,EAAE,QAAQ,EAAE,CAAC,CAAC,EAAE,IAAI,CAAC,OAAO,CAAC,WAAW,CAAC,CAAC;YACvE,aAAa,EAAE,IAAI,MAAM,CAAC,OAAO,EAAE,YAAY,EAAE,CAAC,CAAC,CAAC,CAAC;SACtD,CAAC;QAEF,MAAM,OAAO,GAAG,MAAM,IAAI,CAAC,OAAO,CAAC,GAAG,CAAC,KAAK,CAAC,CAAC;QAC9C,MAAM,OAAO,GAAG,OAAO,CAAC,WAAW,CAAC,IAAqB,CAAC;QAC1D,MAAM,IAAI,GAAG,OAAO,CAAC,YAAY,CAAC,IAAqB,CAAC;QACxD,OAAO,iBAAiB,CAAC,OAAO,EAAE,IAAI,EAAE,OAAO,CAAC,WAAW,CAAC,CAAC;IAC/D,CAAC;CACF"}
+{"version":3,"file":"browser.js","sourceRoot":"","sources":["../../src/browser.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,gBAAgB,EAAE,MAAM,EAAE,MAAM,iBAAiB,CAAC;AAE3D,OAAO,EAAE,iBAAiB,EAAE,UAAU,EAAE,UAAU,EAAa,MAAM,gBAAgB,CAAC;AAetF,MAAM,iBAAiB,GAAG,IAAI,GAAG,CAAC,wBAAwB,EAAE,MAAM,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC,QAAQ,EAAE,CAAC;AACxF,MAAM,mBAAmB,GAAG,IAAI,GAAG,CAAC,uBAAuB,EAAE,MAAM,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC,QAAQ,EAAE,CAAC;AACzF,MAAM,wBAAwB,GAAG,IAAI,GAAG,CAAC,4BAA4B,EAAE,MAAM,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC,QAAQ,EAAE,CAAC;AAEnG,MAAM,WAAW,GAAG,CAAC,SAA4B,EAAE,OAAe,EAAE,GAAG,SAAmB,EAAU,EAAE;IACpG,IAAI,SAAS,CAAC,QAAQ,CAAC,OAAO,CAAC;QAAE,OAAO,OAAO,CAAC;IAChD,KAAK,MAAM,QAAQ,IAAI,SAAS,EAAE,CAAC;QACjC,IAAI,SAAS,CAAC,QAAQ,CAAC,QAAQ,CAAC;YAAE,OAAO,QAAQ,CAAC;IACpD,CAAC;IACD,MAAM,OAAO,GAAG,SAAS,CAAC,MAAM,CAC9B,CAAC,IAAI,EAAE,EAAE,CAAC,IAAI,CAAC,UAAU,CAAC,GAAG,OAAO,GAAG,CAAC,IAAI,IAAI,CAAC,UAAU,CAAC,OAAO,CAAC,CACrE,CAAC;IACF,IAAI,OAAO,CAAC,MAAM,KAAK,CAAC;QAAE,OAAO,OAAO,CAAC,CAAC,CAAC,CAAC;IAC5C,IAAI,OAAO,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;QACvB,MAAM,OAAO,GAAG,OAAO,CAAC,MAAM,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,IAAI,CAAC,UAAU,CAAC,GAAG,OAAO,GAAG,CAAC,CAAC,CAAC;QACzE,IAAI,OAAO,CAAC,MAAM,KAAK,CAAC;YAAE,OAAO,OAAO,CAAC,CAAC,CAAC,CAAC;QAC5C,OAAO,OAAO,CAAC,CAAC,CAAC,CAAC;IACpB,CAAC;IACD,MAAM,IAAI,KAAK,CAAC,2CAA2C,OAAO,iBAAiB,SAAS,CAAC,IAAI,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC;AAC7G,CAAC,CAAC;AAEF,MAAM,OAAO,eAAe;IAM1B,YAAoB,OAAiC;QACnD,IAAI,CAAC,OAAO,GAAG,OAAO,CAAC;IACzB,CAAC;IAED,MAAM,CAAC,KAAK,CAAC,MAAM,CAAC,UAA0B,EAAE;QAC9C,IAAI,CAAC,OAAO,CAAC,UAAU,KAAK,SAAS,CAAC,KAAK,CAAC,OAAO,CAAC,cAAc,KAAK,SAAS,CAAC,EAAE,CAAC;YAClF,MAAM,IAAI,KAAK,CAAC,yDAAyD,CAAC,CAAC;QAC7E,CAAC;QACD,MAAM,IAAI,GAA6B;YACrC,QAAQ,EAAE,OAAO,CAAC,QAAQ,IAAI,iBAAiB;YAC/C,UAAU,EAAE,OAAO,CAAC,UAAU,IAAI,mBAAmB;YACrD,cAAc,EAAE,OAAO,CAAC,cAAc,IAAI,wBAAwB;YAClE,WAAW,EAAE,OAAO,CAAC,WAAW,IAAI,GAAG;YACvC,mFAAmF;YACnF,YAAY,EAAE,OAAO,CAAC,YAAY,IAAI,EAAE;SACzC,CAAC;QACF,MAAM,KAAK,GAAG,IAAI,eAAe,CAAC,IAAI,CAAC,CAAC;QACxC,MAAM,gBAAgB,GAAG,OAAO,CAAC,UAAU,KAAK,SAAS,IAAI,OAAO,CAAC,cAAc,KAAK,SAAS,CAAC;QAClG,IAAI,gBAAgB,EAAE,CAAC;YACrB,MAAM,CAAC,cAAc,EAAE,kBAAkB,CAAC,GAAG,MAAM,OAAO,CAAC,GAAG,CAAC;gBAC7D,gBAAgB,CAAC,MAAM,CAAC,IAAI,CAAC,UAAU,EAAE,EAAE,kBAAkB,EAAE,CAAC,MAAM,CAAC,EAAE,CAAC;gBAC1E,gBAAgB,CAAC,MAAM,CAAC,IAAI,CAAC,cAAc,EAAE,EAAE,kBAAkB,EAAE,CAAC,MAAM,CAAC,EAAE,CAAC;aAC/E,CAAC,CAAC;YACH,KAAK,CAAC,cAAc,GAAG,cAAc,CAAC;YACtC,KAAK,CAAC,kBAAkB,GAAG,kBAAkB,CAAC;YAC9C,OAAO,KAAK,CAAC;QACf,CAAC;QAED,IAAI,CAAC;YACH,MAAM,CAAC,cAAc,EAAE,kBAAkB,CAAC,GAAG,MAAM,OAAO,CAAC,GAAG,CAAC;gBAC7D,gBAAgB,CAAC,MAAM,CAAC,IAAI,CAAC,UAAU,EAAE,EAAE,kBAAkB,EAAE,CAAC,MAAM,CAAC,EAAE,CAAC;gBAC1E,gBAAgB,CAAC,MAAM,CAAC,IAAI,CAAC,cAAc,EAAE,EAAE,kBAAkB,EAAE,CAAC,MAAM,CAAC,EAAE,CAAC;aAC/E,CAAC,CAAC;YACH,KAAK,CAAC,cAAc,GAAG,cAAc,CAAC;YACtC,KAAK,CAAC,kBAAkB,GAAG,kBAAkB,CAAC;QAChD,CAAC;QAAC,MAAM,CAAC;YACP,KAAK,CAAC,OAAO,GAAG,MAAM,gBAAgB,CAAC,MAAM,CAAC,IAAI,CAAC,QAAQ,EAAE;gBAC3D,kBAAkB,EAAE,CAAC,MAAM,CAAC;aAC7B,CAAC,CAAC;QACL,CAAC;QACD,OAAO,KAAK,CAAC;IACf,CAAC;IAED,KAAK,CAAC,OAAO,CAAC,IAAY,EAAE,UAAiC,EAAE;QAC7D,MAAM,cAAc,GAAG,OAAO,CAAC,cAAc,IAAI,MAAM,CAAC;QACxD,MAAM,eAAe,GAAG,OAAO,CAAC,eAAe,IAAI,GAAG,CAAC;QACvD,MAAM,QAAQ,GAAG,aAAa,CAAC,IAAI,EAAE,cAAc,CAAC,CAAC;QACrD,IAAI,QAAQ,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;YAC1B,OAAO,IAAI,CAAC,aAAa,CAAC,IAAI,EAAE,CAAC,CAAC,CAAC;QACrC,CAAC;QAED,MAAM,OAAO,GAAG,MAAM,OAAO,CAAC,GAAG,CAC/B,QAAQ,CAAC,GAAG,CAAC,KAAK,EAAE,OAAO,EAAE,EAAE,CAC7B,IAAI,CAAC,aAAa,CAAC,OAAO,CAAC,IAAI,EAAE,eAAe,CAAC,IAAI,EAAE,OAAO,CAAC,aAAa,CAAC,CAAC,CAC/E,CACF,CAAC;QAEF,MAAM,QAAQ,GAAa,EAAE,CAAC;QAC9B,MAAM,UAAU,GAA4B,EAAE,CAAC;QAC/C,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,OAAO,CAAC,MAAM,EAAE,CAAC,EAAE,EAAE,CAAC;YACxC,IAAI,CAAC,GAAG,CAAC;gBAAE,QAAQ,CAAC,IAAI,CAAC,eAAe,CAAC,CAAC;YAC1C,QAAQ,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,GAAG,CAAC,CAAC;YAC9B,KAAK,MAAM,SAAS,IAAI,OAAO,CAAC,CAAC,CAAC,CAAC,UAAU,EAAE,CAAC;gBAC9C,UAAU,CAAC,IAAI,CAAC;oBACd,OAAO,EAAE,SAAS,CAAC,OAAO;oBAC1B,YAAY,EAAE,UAAU,CAAC,MAAM;oBAC/B,SAAS,EAAE,SAAS,CAAC,SAAS;iBAC/B,CAAC,CAAC;YACL,CAAC;QACH,CAAC;QACD,OAAO,EAAE,GAAG,EAAE,QAAQ,CAAC,IAAI,CAAC,EAAE,CAAC,EAAE,UAAU,EAAE,CAAC;IAChD,CAAC;IAEO,KAAK,CAAC,aAAa,CAAC,IAAY,EAAE,aAAqB;QAC7D,IAAI,IAAI,CAAC,cAAc,IAAI,IAAI,CAAC,kBAAkB,EAAE,CAAC;YACnD,OAAO,IAAI,CAAC,kBAAkB,CAAC,IAAI,EAAE,aAAa,CAAC,CAAC;QACtD,CAAC;QACD,IAAI,CAAC,IAAI,CAAC,OAAO,EAAE,CAAC;YAClB,MAAM,IAAI,KAAK,CAAC,6BAA6B,CAAC,CAAC;QACjD,CAAC;QACD,MAAM,OAAO,GAAG,UAAU,CAAC,IAAI,EAAE,IAAI,CAAC,OAAO,CAAC,WAAW,CAAC,CAAC;QAC3D,MAAM,QAAQ,GAAG,aAAa,CAAC,IAAI,CAAC,OAAO,CAAC,GAAG,CAAC,CAAC;QACjD,MAAM,YAAY,GAAG,IAAI,aAAa,CAAC,CAAC,MAAM,CAAC,OAAO,CAAC,MAAM,IAAI,CAAC,CAAC,CAAC,CAAC,CAAC;QAEtE,MAAM,KAAK,GAA2B;YACpC,SAAS,EAAE,IAAI,MAAM,CAAC,OAAO,EAAE,QAAQ,EAAE,CAAC,CAAC,EAAE,IAAI,CAAC,OAAO,CAAC,WAAW,CAAC,CAAC;YACvE,aAAa,EAAE,IAAI,MAAM,CAAC,OAAO,EAAE,YAAY,EAAE,CAAC,CAAC,CAAC,CAAC;SACtD,CAAC;QAEF,MAAM,OAAO,GAAG,MAAM,IAAI,CAAC,OAAO,CAAC,GAAG,CAAC,KAAK,CAAC,CAAC;QAC9C,MAAM,OAAO,GAAG,OAAO,CAAC,WAAW,CAAC,IAAqB,CAAC;QAC1D,MAAM,IAAI,GAAG,OAAO,CAAC,YAAY,CAAC,IAAqB,CAAC;QACxD,MAAM,MAAM,GAAG,iBAAiB,CAAC,OAAO,EAAE,IAAI,EAAE,OAAO,CAAC,WAAW,CAAC,CAAC;QACrE,OAAO;YACL,GAAG,EAAE,MAAM,CAAC,GAAG;YACf,UAAU,EAAE,MAAM,CAAC,UAAU,CAAC,GAAG,CAAC,CAAC,SAAS,EAAE,GAAG,EAAE,EAAE,CAAC,CAAC;gBACrD,OAAO,EAAE,SAAS,CAAC,OAAO;gBAC1B,YAAY,EAAE,GAAG;gBACjB,SAAS,EACP,SAAS,CAAC,SAAS,GAAG,CAAC,CAAC,CAAC,CAAC,SAAS,CAAC,SAAS,CAAC,CAAC,CAAC,SAAS,CAAC,SAAS,GAAG,aAAa;aACtF,CAAC,CAAC;SACJ,CAAC;IACJ,CAAC;IAEO,KAAK,CAAC,kBAAkB,CAAC,IAAY,EAAE,aAAqB;QAClE,IAAI,CAAC,IAAI,CAAC,cAAc,IAAI,CAAC,IAAI,CAAC,kBAAkB,EAAE,CAAC;YACrD,MAAM,IAAI,KAAK,CAAC,yCAAyC,CAAC,CAAC;QAC7D,CAAC;QAED,MAAM,OAAO,GAAG,UAAU,CAAC,IAAI,EAAE,IAAI,CAAC,OAAO,CAAC,WAAW,CAAC,CAAC;QAC3D,MAAM,QAAQ,GAAG,aAAa,CAAC,IAAI,CAAC,OAAO,CAAC,GAAG,CAAC,CAAC;QACjD,MAAM,YAAY,GAAG,IAAI,aAAa,CAAC,CAAC,MAAM,CAAC,OAAO,CAAC,MAAM,IAAI,CAAC,CAAC,CAAC,CAAC,CAAC;QACtE,MAAM,YAAY,GAA2B;YAC3C,SAAS,EAAE,IAAI,MAAM,CAAC,OAAO,EAAE,QAAQ,EAAE,CAAC,CAAC,EAAE,IAAI,CAAC,OAAO,CAAC,WAAW,CAAC,CAAC;YACvE,aAAa,EAAE,IAAI,MAAM,CAAC,OAAO,EAAE,YAAY,EAAE,CAAC,CAAC,CAAC,CAAC;SACtD,CAAC;QAEF,MAAM,cAAc,GAAG,MAAM,IAAI,CAAC,cAAc,CAAC,GAAG,CAAC,YAAY,CAAC,CAAC;QACnE,MAAM,kBAAkB,GAAG,IAAI,CAAC,cAAc,CAAC,WAAW,CAAC;QAC3D,MAAM,iBAAiB,GAAG,IAAI,CAAC,kBAAkB,CAAC,UAAU,CAAC;QAC7D,MAAM,kBAAkB,GAAG,IAAI,CAAC,kBAAkB,CAAC,WAAW,CAAC;QAC/D,MAAM,QAAQ,GAAG;YACf,eAAe,EAAE,WAAW,CAAC,kBAAkB,EAAE,iBAAiB,CAAC;YACnE,cAAc,EAAE,WAAW,CAAC,kBAAkB,EAAE,gBAAgB,CAAC;YACjE,YAAY,EAAE,WAAW,CAAC,kBAAkB,EAAE,cAAc,CAAC;YAC7D,MAAM,EAAE,WAAW,CAAC,kBAAkB,EAAE,QAAQ,CAAC;YACjD,SAAS,EAAE,WAAW,CAAC,kBAAkB,EAAE,WAAW,CAAC;SACxD,CAAC;QACF,MAAM,KAAK,GAAG;YACZ,iBAAiB,EAAE,WAAW,CAAC,iBAAiB,EAAE,mBAAmB,CAAC;YACtE,eAAe,EAAE,WAAW,CAAC,iBAAiB,EAAE,iBAAiB,CAAC;YAClE,cAAc,EAAE,WAAW,CAAC,iBAAiB,EAAE,gBAAgB,CAAC;YAChE,YAAY,EAAE,WAAW,CAAC,iBAAiB,EAAE,cAAc,CAAC;YAC5D,SAAS,EAAE,WAAW,CAAC,iBAAiB,EAAE,WAAW,EAAE,cAAc,CAAC;YACtE,MAAM,EAAE,WAAW,CAAC,iBAAiB,EAAE,QAAQ,EAAE,WAAW,CAAC;YAC7D,SAAS,EAAE,WAAW,CAAC,iBAAiB,EAAE,WAAW,CAAC;SACvD,CAAC;QACF,MAAM,MAAM,GAAG;YACb,cAAc,EAAE,WAAW,CAAC,kBAAkB,EAAE,gBAAgB,CAAC;YACjE,MAAM,EAAE,WAAW,CAAC,kBAAkB,EAAE,YAAY,EAAE,QAAQ,CAAC;YAC/D,SAAS,EAAE,WAAW,CAAC,kBAAkB,EAAE,eAAe,EAAE,WAAW,CAAC;YACxE,WAAW,EAAE,WAAW,CAAC,kBAAkB,EAAE,aAAa,CAAC;SAC5D,CAAC;QACF,MAAM,aAAa,GAAG;YACpB,eAAe,EAAE,cAAc,CAAC,QAAQ,CAAC,eAAe,CAAW;YACnE,cAAc,EAAE,cAAc,CAAC,QAAQ,CAAC,cAAc,CAAW;YACjE,YAAY,EAAE,cAAc,CAAC,QAAQ,CAAC,YAAY,CAAW;YAC7D,MAAM,EAAE,cAAc,CAAC,QAAQ,CAAC,MAAM,CAAW;YACjD,SAAS,EAAE,cAAc,CAAC,QAAQ,CAAC,SAAS,CAAW;SACxD,CAAC;QAEF,MAAM,MAAM,GAAG,MAAM,CAAC,aAAa,CAAC,eAAe,CAAC,IAAI,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC;QAClE,MAAM,SAAS,GAAG,IAAI,YAAY,CAAC,MAAM,CAAC,CAAC;QAC3C,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,MAAM,EAAE,CAAC,EAAE;YAAE,SAAS,CAAC,CAAC,CAAC,GAAG,CAAC,CAAC;QAClD,MAAM,eAAe,GAAG,IAAI,MAAM,CAAC,SAAS,EAAE,SAAS,EAAE,CAAC,CAAC,EAAE,MAAM,CAAC,CAAC,CAAC;QAEtE,IAAI,YAAY,GAAG,IAAI,MAAM,CAAC,OAAO,EAAE,IAAI,aAAa,CAAC,CAAC,MAAM,CAAC,UAAU,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,CAAC;QAC5F,IAAI,MAAM,GAAG,aAAa,CAAC,MAAM,CAAC;QAClC,IAAI,QAAQ,GAAG,aAAa,CAAC,SAAS,CAAC;QAEvC,MAAM,OAAO,GAAa,EAAE,CAAC;QAC7B,MAAM,WAAW,GAAa,EAAE,CAAC;QAEjC,KAAK,IAAI,IAAI,GAAG,CAAC,EAAE,IAAI,GAAG,IAAI,CAAC,OAAO,CAAC,YAAY,EAAE,IAAI,EAAE,EAAE,CAAC;YAC5D,MAAM,WAAW,GAAG,MAAM,IAAI,CAAC,kBAAkB,CAAC,GAAG,CAAC;gBACpD,CAAC,KAAK,CAAC,iBAAiB,CAAC,EAAE,YAAY;gBACvC,CAAC,KAAK,CAAC,eAAe,CAAC,EAAE,aAAa,CAAC,eAAe;gBACtD,CAAC,KAAK,CAAC,cAAc,CAAC,EAAE,aAAa,CAAC,cAAc;gBACpD,CAAC,KAAK,CAAC,YAAY,CAAC,EAAE,aAAa,CAAC,YAAY;gBAChD,CAAC,KAAK,CAAC,SAAS,CAAC,EAAE,QAAQ;gBAC3B,CAAC,KAAK,CAAC,MAAM,CAAC,EAAE,MAAM;gBACtB,CAAC,KAAK,CAAC,SAAS,CAAC,EAAE,eAAe;aACnC,CAAC,CAAC;YAEH,MAAM,SAAS,GAAG,UAAU,CAAC,WAAW,CAAC,MAAM,CAAC,cAAc,CAAW,CAAC,CAAC;YAC3E,MAAM,OAAO,GAAG,UAAU,CAAC,WAAW,CAAC,MAAM,CAAC,WAAW,CAAW,CAAC,CAAC;YACtE,OAAO,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC;YACxB,WAAW,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC;YAE1B,MAAM,GAAG,WAAW,CAAC,MAAM,CAAC,MAAM,CAAW,CAAC;YAC9C,QAAQ,GAAG,WAAW,CAAC,MAAM,CAAC,SAAS,CAAW,CAAC;YACnD,YAAY,GAAG,IAAI,MAAM,CAAC,OAAO,EAAE,IAAI,aAAa,CAAC,CAAC,SAAS,CAAC,CAAC,EAAE,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,CAAC;YAE3E,IAAI,SAAS,KAAK,MAAM,CAAC,UAAU,CAAC,GAAG,CAAC,EAAE,CAAC;gBACzC,MAAM;YACR,CAAC;QACH,CAAC;QAED,MAAM,MAAM,GAAG,iBAAiB,CAAC,OAAO,EAAE,WAAW,EAAE,OAAO,CAAC,WAAW,CAAC,CAAC;QAC5E,OAAO;YACL,GAAG,EAAE,MAAM,CAAC,GAAG;YACf,UAAU,EAAE,MAAM,CAAC,UAAU,CAAC,GAAG,CAAC,CAAC,SAAS,EAAE,GAAG,EAAE,EAAE,CAAC,CAAC;gBACrD,OAAO,EAAE,SAAS,CAAC,OAAO;gBAC1B,YAAY,EAAE,GAAG;gBACjB,SAAS,EACP,SAAS,CAAC,SAAS,GAAG,CAAC,CAAC,CAAC,CAAC,SAAS,CAAC,SAAS,CAAC,CAAC,CAAC,SAAS,CAAC,SAAS,GAAG,aAAa;aACtF,CAAC,CAAC;SACJ,CAAC;IACJ,CAAC;CACF;AAED,MAAM,UAAU,GAAG,CAAC,MAAc,EAAU,EAAE;IAC5C,MAAM,IAAI,GAAG,MAAM,CAAC,IAAkC,CAAC;IACvD,MAAM,KAAK,GAAG,IAAI,CAAC,CAAC,CAAC,CAAC;IACtB,OAAO,OAAO,KAAK,KAAK,QAAQ,CAAC,CAAC,CAAC,KAAK,CAAC,CAAC,CAAC,MAAM,CAAC,IAAI,CAAC,KAAK,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC;AAC/E,CAAC,CAAC;AAEF,MAAM,WAAW,GAAG,CAAC,KAAa,EAAU,EAAE,CAAC,KAAK,CAAC,OAAO,CAAC,qBAAqB,EAAE,MAAM,CAAC,CAAC;AAE5F,MAAM,aAAa,GAAG,CAAC,SAA0B,EAAU,EAAE;IAC3D,IAAI,OAAO,SAAS,KAAK,QAAQ,EAAE,CAAC;QAClC,IAAI,SAAS,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;YAC3B,MAAM,IAAI,KAAK,CAAC,4CAA4C,CAAC,CAAC;QAChE,CAAC;QACD,OAAO,IAAI,MAAM,CAAC,WAAW,CAAC,SAAS,CAAC,EAAE,IAAI,CAAC,CAAC;IAClD,CAAC;IACD,MAAM,KAAK,GAAG,SAAS,CAAC,KAAK,CAAC,QAAQ,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC,SAAS,CAAC,KAAK,CAAC,CAAC,CAAC,GAAG,SAAS,CAAC,KAAK,GAAG,CAAC;IACtF,OAAO,IAAI,MAAM,CAAC,SAAS,CAAC,MAAM,EAAE,KAAK,CAAC,QAAQ,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC,KAAK,CAAC,CAAC,CAAC,GAAG,KAAK,GAAG,CAAC,CAAC;AACjF,CAAC,CAAC;AAEF,MAAM,aAAa,GAAG,CACpB,IAAY,EACZ,SAAiC,EACe,EAAE;IAClD,IAAI,SAAS,KAAK,IAAI,EAAE,CAAC;QACvB,OAAO,CAAC,EAAE,IAAI,EAAE,aAAa,EAAE,CAAC,EAAE,CAAC,CAAC;IACtC,CAAC;IACD,MAAM,KAAK,GAAG,aAAa,CAAC,SAAS,CAAC,CAAC;IACvC,IAAI,KAAK,CAAC,IAAI,CAAC,EAAE,CAAC,EAAE,CAAC;QACnB,MAAM,IAAI,KAAK,CAAC,+CAA+C,CAAC,CAAC;IACnE,CAAC;IACD,KAAK,CAAC,SAAS,GAAG,CAAC,CAAC;IAEpB,MAAM,QAAQ,GAAmD,EAAE,CAAC;IACpE,IAAI,KAAK,GAAG,CAAC,CAAC;IACd,KAAK,MAAM,KAAK,IAAI,IAAI,CAAC,QAAQ,CAAC,KAAK,CAAC,EAAE,CAAC;QACzC,MAAM,GAAG,GAAG,KAAK,CAAC,KAAK,IAAI,CAAC,CAAC;QAC7B,IAAI,GAAG,GAAG,KAAK,EAAE,CAAC;YAChB,QAAQ,CAAC,IAAI,CAAC,EAAE,IAAI,EAAE,IAAI,CAAC,KAAK,CAAC,KAAK,EAAE,GAAG,CAAC,EAAE,aAAa,EAAE,KAAK,EAAE,CAAC,CAAC;QACxE,CAAC;QACD,KAAK,GAAG,GAAG,GAAG,KAAK,CAAC,CAAC,CAAC,CAAC,MAAM,CAAC;IAChC,CAAC;IACD,IAAI,KAAK,GAAG,IAAI,CAAC,MAAM,EAAE,CAAC;QACxB,QAAQ,CAAC,IAAI,CAAC,EAAE,IAAI,EAAE,IAAI,CAAC,KAAK,CAAC,KAAK,CAAC,EAAE,aAAa,EAAE,KAAK,EAAE,CAAC,CAAC;IACnE,CAAC;IACD,OAAO,QAAQ,CAAC;AAClB,CAAC,CAAC;AAEF,MAAM,eAAe,GAAG,CAAC,IAAY,EAAE,cAAsB,EAAU,EAAE;IACvE,IAAI,cAAc,GAAG,CAAC,CAAC;IACvB,IAAI,CAAC,GAAG,CAAC,CAAC;IACV,OAAO,CAAC,GAAG,cAAc,EAAE,CAAC;QAC1B,MAAM,EAAE,GAAG,IAAI,CAAC,WAAW,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC;QACpC,CAAC,IAAI,EAAE,GAAG,MAAM,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC;QACzB,cAAc,IAAI,CAAC,CAAC;IACtB,CAAC;IACD,OAAO,cAAc,CAAC;AACxB,CAAC,CAAC"}

package/dist/node/asr.js

@@ -0,0 +1,340 @@
+import fs from "node:fs";
+import path from "node:path";
+import { fileURLToPath } from "node:url";
+import { InferenceSession, Tensor } from "onnxruntime-node";
+import vocabData from "./assets/g2p_vocab.json";
+const __dirname = path.dirname(fileURLToPath(import.meta.url));
+const defaultWaveformModelPath = path.join(__dirname, "assets", "asr_waveform_fp16.onnx");
+const resolveDefaultAsrModelPath = () => {
+    if (fs.existsSync(defaultWaveformModelPath)) {
+        return defaultWaveformModelPath;
+    }
+    throw new Error(`Missing waveform ASR asset: ${defaultWaveformModelPath}`);
+};
+const resolveName = (available, primary, ...fallbacks) => {
+    if (available.includes(primary))
+        return primary;
+    for (const fallback of fallbacks) {
+        if (available.includes(fallback))
+            return fallback;
+    }
+    const matches = available.filter((name) => name.startsWith(`${primary}.`) || name.startsWith(primary));
+    if (matches.length === 1)
+        return matches[0];
+    if (matches.length > 1) {
+        const numeric = matches.filter((name) => name.startsWith(`${primary}.`));
+        if (numeric.length === 1)
+            return numeric[0];
+        return matches[0];
+    }
+    throw new Error(`Could not resolve ONNX tensor name for '${primary}'. Available: ${available.join(", ")}`);
+};
+const scalarInt = (tensor) => {
+    const data = tensor.data;
+    const raw = data[0];
+    return typeof raw === "bigint" ? Number(raw) : Math.trunc(Number(raw));
+};
+export const decodeCtcTokens = (frameTokenIds, decoderTokens, options) => {
+    const collapseRepeats = options.collapseRepeats ?? true;
+    const tokenIds = [];
+    let prev = -1;
+    for (const tokenId of frameTokenIds) {
+        if (collapseRepeats && tokenId === prev) {
+            continue;
+        }
+        prev = tokenId;
+        if (tokenId === options.blankId) {
+            continue;
+        }
+        tokenIds.push(tokenId);
+    }
+    const rawTokens = tokenIds.map((tokenId) => decoderTokens[tokenId] ?? "<unk>");
+    const phonemes = rawTokens.filter((token) => token !== options.wordBoundaryToken);
+    const words = [];
+    let current = [];
+    for (const token of rawTokens) {
+        if (token === options.wordBoundaryToken) {
+            if (current.length > 0) {
+                words.push(current);
+                current = [];
+            }
+            continue;
+        }
+        current.push(token);
+    }
+    if (current.length > 0) {
+        words.push(current);
+    }
+    return { tokenIds, phonemes, words };
+};
+const loadDecoderTokens = (vocabPath) => {
+    let vocab;
+    if (vocabPath) {
+        const raw = fs.readFileSync(vocabPath, "utf-8");
+        vocab = JSON.parse(raw);
+    }
+    else {
+        vocab = vocabData;
+    }
+    if (!Array.isArray(vocab.decoder) || vocab.decoder.length === 0) {
+        throw new Error("Invalid vocabulary JSON: missing decoder token list");
+    }
+    return [...vocab.decoder, "<wb>", "<blank>"];
+};
+export class ASRNodeModel {
+    constructor(session, options) {
+        this.session = session;
+        this.sampleRate = options.sampleRate;
+        this.wordBoundaryToken = options.wordBoundaryToken;
+        this.blankBias = options.blankBias;
+        this.unkBias = options.unkBias;
+        this.collapseRepeats = options.collapseRepeats;
+        this.decoderTokens = loadDecoderTokens(options.vocabPath);
+        this.blankId = this.decoderTokens.indexOf(options.blankToken);
+        if (this.blankId < 0) {
+            throw new Error(`blank token '${options.blankToken}' not found in decoder tokens`);
+        }
+        const unk = this.decoderTokens.indexOf(options.unkToken);
+        this.unkId = unk >= 0 ? unk : null;
+        const inputNames = this.session.inputNames;
+        const hasWaveform = inputNames.includes("waveform")
+            || inputNames.some((name) => name.startsWith("waveform."));
+        if (!hasWaveform) {
+            throw new Error(`Unsupported ASR ONNX inputs: ${inputNames.join(", ")}. `
+                + "hama ASR requires a waveform-input model with waveform/waveform_lengths.");
+        }
+        this.waveformInputName = resolveName(inputNames, "waveform");
+        this.waveformLengthsInputName = resolveName(inputNames, "waveform_lengths", "waveform_length");
+        this.logProbsOutputName = resolveName(this.session.outputNames, "log_probs");
+        this.outLengthsOutputName = resolveName(this.session.outputNames, "out_lengths");
+    }
+    static async create(options = {}) {
+        const opts = {
+            modelPath: options.modelPath ?? resolveDefaultAsrModelPath(),
+            vocabPath: options.vocabPath ?? "",
+            sampleRate: options.sampleRate ?? 16000,
+            blankToken: options.blankToken ?? "<blank>",
+            unkToken: options.unkToken ?? "<unk>",
+            wordBoundaryToken: options.wordBoundaryToken ?? "<wb>",
+            blankBias: options.blankBias ?? -0.1,
+            unkBias: options.unkBias ?? 0.0,
+            collapseRepeats: options.collapseRepeats ?? true,
+        };
+        const session = await InferenceSession.create(opts.modelPath, {
+            graphOptimizationLevel: "disabled",
+        });
+        return new ASRNodeModel(session, opts);
+    }
+    get inputFormat() {
+        return "waveform";
+    }
+    async transcribeWavFile(wavPath) {
+        const { waveform, sampleRate } = readWavMono(wavPath);
+        return this.transcribeWaveform(waveform, sampleRate);
+    }
+    async transcribeWaveform(waveform, sampleRate) {
+        const mono = toFloat32Mono(waveform);
+        const resampled = sampleRate === this.sampleRate
+            ? mono
+            : resampleLinear(mono, sampleRate, this.sampleRate);
+        const waveformTensor = new Tensor("float32", resampled, [1, resampled.length]);
+        const lengthTensor = new Tensor("int64", BigInt64Array.from([BigInt(resampled.length)]), [1]);
+        const outputs = await this.session.run({
+            [this.waveformInputName]: waveformTensor,
+            [this.waveformLengthsInputName]: lengthTensor,
+        });
+        const logProbs = outputs[this.logProbsOutputName];
+        const outLengths = outputs[this.outLengthsOutputName];
+        const numFrames = Math.max(0, Math.min(scalarInt(outLengths), Number(logProbs.dims[1] ?? 0)));
+        const frameTokenIds = this.argmaxFrames(logProbs, numFrames);
+        const decoded = decodeCtcTokens(frameTokenIds, this.decoderTokens, {
+            blankId: this.blankId,
+            wordBoundaryToken: this.wordBoundaryToken,
+            collapseRepeats: this.collapseRepeats,
+        });
+        return {
+            phonemes: decoded.phonemes,
+            phonemeText: decoded.phonemes.join(" "),
+            wordPhonemeText: decoded.words.map((word) => word.join(" ")).join(" | "),
+            tokenIds: decoded.tokenIds,
+            frameTokenIds,
+            numFrames,
+        };
+    }
+    argmaxFrames(logProbs, numFrames) {
+        const dims = logProbs.dims.map(Number);
+        if (dims.length !== 3 || dims[0] !== 1) {
+            throw new Error(`Expected log_probs shape [1, T, C], got [${dims.join(", ")}]`);
+        }
+        const classes = dims[2];
+        const data = logProbs.data;
+        const out = [];
+        for (let t = 0; t < numFrames; t++) {
+            let bestIdx = 0;
+            let bestScore = -Infinity;
+            const base = t * classes;
+            for (let c = 0; c < classes; c++) {
+                const raw = Number(data[base + c]);
+                const score = raw
+                    + (c === this.blankId ? this.blankBias : 0.0)
+                    + (this.unkId !== null && c === this.unkId ? this.unkBias : 0.0);
+                if (score > bestScore) {
+                    bestScore = score;
+                    bestIdx = c;
+                }
+            }
+            out.push(bestIdx);
+        }
+        return out;
+    }
+}
+const readWavMono = (wavPath) => {
+    const buf = fs.readFileSync(wavPath);
+    if (buf.length < 44) {
+        throw new Error(`Invalid WAV file: too short (${wavPath})`);
+    }
+    const riff = buf.toString("ascii", 0, 4);
+    const wave = buf.toString("ascii", 8, 12);
+    if (riff !== "RIFF" || wave !== "WAVE") {
+        throw new Error(`Invalid WAV header: expected RIFF/WAVE (${wavPath})`);
+    }
+    let offset = 12;
+    let format = 0;
+    let channels = 0;
+    let sampleRate = 0;
+    let bitsPerSample = 0;
+    let dataOffset = -1;
+    let dataSize = 0;
+    while (offset + 8 <= buf.length) {
+        const chunkId = buf.toString("ascii", offset, offset + 4);
+        const chunkSize = buf.readUInt32LE(offset + 4);
+        const chunkDataOffset = offset + 8;
+        const next = chunkDataOffset + chunkSize + (chunkSize % 2);
+        if (chunkId === "fmt " && chunkSize >= 16) {
+            format = buf.readUInt16LE(chunkDataOffset);
+            channels = buf.readUInt16LE(chunkDataOffset + 2);
+            sampleRate = buf.readUInt32LE(chunkDataOffset + 4);
+            bitsPerSample = buf.readUInt16LE(chunkDataOffset + 14);
+        }
+        else if (chunkId === "data") {
+            dataOffset = chunkDataOffset;
+            dataSize = chunkSize;
+            break;
+        }
+        offset = next;
+    }
+    if (dataOffset < 0 || dataSize <= 0) {
+        throw new Error(`Invalid WAV: missing data chunk (${wavPath})`);
+    }
+    if (channels <= 0 || sampleRate <= 0) {
+        throw new Error(`Invalid WAV: malformed fmt chunk (${wavPath})`);
+    }
+    let samplesPerChannel = 0;
+    let interleaved;
+    if (format === 1) {
+        if (bitsPerSample === 8) {
+            samplesPerChannel = Math.floor(dataSize / channels);
+            interleaved = new Float32Array(samplesPerChannel * channels);
+            for (let i = 0; i < interleaved.length; i++) {
+                const v = buf.readUInt8(dataOffset + i);
+                interleaved[i] = (v - 128) / 128;
+            }
+        }
+        else if (bitsPerSample === 16) {
+            const bytesPerSample = 2;
+            samplesPerChannel = Math.floor(dataSize / (channels * bytesPerSample));
+            interleaved = new Float32Array(samplesPerChannel * channels);
+            for (let i = 0; i < interleaved.length; i++) {
+                const v = buf.readInt16LE(dataOffset + i * bytesPerSample);
+                interleaved[i] = v / 32768;
+            }
+        }
+        else if (bitsPerSample === 24) {
+            const bytesPerSample = 3;
+            samplesPerChannel = Math.floor(dataSize / (channels * bytesPerSample));
+            interleaved = new Float32Array(samplesPerChannel * channels);
+            for (let i = 0; i < interleaved.length; i++) {
+                const p = dataOffset + i * bytesPerSample;
+                let v = buf[p] | (buf[p + 1] << 8) | (buf[p + 2] << 16);
+                if (v & 0x800000)
+                    v -= 0x1000000;
+                interleaved[i] = v / 8388608;
+            }
+        }
+        else if (bitsPerSample === 32) {
+            const bytesPerSample = 4;
+            samplesPerChannel = Math.floor(dataSize / (channels * bytesPerSample));
+            interleaved = new Float32Array(samplesPerChannel * channels);
+            for (let i = 0; i < interleaved.length; i++) {
+                const v = buf.readInt32LE(dataOffset + i * bytesPerSample);
+                interleaved[i] = v / 2147483648;
+            }
+        }
+        else {
+            throw new Error(`Unsupported PCM WAV bits_per_sample=${bitsPerSample}`);
+        }
+    }
+    else if (format === 3 && bitsPerSample === 32) {
+        const bytesPerSample = 4;
+        samplesPerChannel = Math.floor(dataSize / (channels * bytesPerSample));
+        interleaved = new Float32Array(samplesPerChannel * channels);
+        for (let i = 0; i < interleaved.length; i++) {
+            interleaved[i] = buf.readFloatLE(dataOffset + i * bytesPerSample);
+        }
+    }
+    else {
+        throw new Error(`Unsupported WAV format=${format} bits_per_sample=${bitsPerSample}`);
+    }
+    if (channels === 1) {
+        return { waveform: clampUnit(interleaved), sampleRate };
+    }
+    const mono = new Float32Array(samplesPerChannel);
+    for (let i = 0; i < samplesPerChannel; i++) {
+        let acc = 0;
+        for (let ch = 0; ch < channels; ch++) {
+            acc += interleaved[i * channels + ch];
+        }
+        mono[i] = acc / channels;
+    }
+    return { waveform: clampUnit(mono), sampleRate };
+};
+const toFloat32Mono = (waveform) => {
+    if (waveform instanceof Float32Array) {
+        return clampUnit(waveform);
+    }
+    if (!Array.isArray(waveform)) {
+        throw new Error("waveform must be Float32Array or number[]");
+    }
+    const out = new Float32Array(waveform.length);
+    for (let i = 0; i < waveform.length; i++) {
+        out[i] = Number(waveform[i]);
+    }
+    return clampUnit(out);
+};
+const clampUnit = (arr) => {
+    const out = new Float32Array(arr.length);
+    for (let i = 0; i < arr.length; i++) {
+        const v = arr[i];
+        out[i] = v < -1 ? -1 : v > 1 ? 1 : v;
+    }
+    return out;
+};
+const resampleLinear = (waveform, srcRate, dstRate) => {
+    if (srcRate === dstRate)
+        return waveform;
+    if (waveform.length === 0)
+        return waveform;
+    const duration = (waveform.length - 1) / srcRate;
+    const dstLen = Math.max(1, Math.round(duration * dstRate) + 1);
+    const out = new Float32Array(dstLen);
+    for (let i = 0; i < dstLen; i++) {
+        const t = i / dstRate;
+        const srcPos = t * srcRate;
+        const i0 = Math.floor(srcPos);
+        const i1 = Math.min(waveform.length - 1, i0 + 1);
+        const a = srcPos - i0;
+        out[i] = waveform[i0] * (1 - a) + waveform[i1] * a;
+    }
+    return out;
+};
+//# sourceMappingURL=asr.js.map

package/dist/node/asr.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"asr.js","sourceRoot":"","sources":["../../src/asr.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,MAAM,SAAS,CAAC;AACzB,OAAO,IAAI,MAAM,WAAW,CAAC;AAC7B,OAAO,EAAE,aAAa,EAAE,MAAM,UAAU,CAAC;AACzC,OAAO,EAAE,gBAAgB,EAAE,MAAM,EAAE,MAAM,kBAAkB,CAAC;AAE5D,OAAO,SAAS,MAAM,yBAAyB,CAAC;AAiChD,MAAM,SAAS,GAAG,IAAI,CAAC,OAAO,CAAC,aAAa,CAAC,MAAM,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC,CAAC;AAC/D,MAAM,wBAAwB,GAAG,IAAI,CAAC,IAAI,CAAC,SAAS,EAAE,QAAQ,EAAE,wBAAwB,CAAC,CAAC;AAE1F,MAAM,0BAA0B,GAAG,GAAW,EAAE;IAC9C,IAAI,EAAE,CAAC,UAAU,CAAC,wBAAwB,CAAC,EAAE,CAAC;QAC5C,OAAO,wBAAwB,CAAC;IAClC,CAAC;IACD,MAAM,IAAI,KAAK,CAAC,+BAA+B,wBAAwB,EAAE,CAAC,CAAC;AAC7E,CAAC,CAAC;AAEF,MAAM,WAAW,GAAG,CAAC,SAA4B,EAAE,OAAe,EAAE,GAAG,SAAmB,EAAU,EAAE;IACpG,IAAI,SAAS,CAAC,QAAQ,CAAC,OAAO,CAAC;QAAE,OAAO,OAAO,CAAC;IAChD,KAAK,MAAM,QAAQ,IAAI,SAAS,EAAE,CAAC;QACjC,IAAI,SAAS,CAAC,QAAQ,CAAC,QAAQ,CAAC;YAAE,OAAO,QAAQ,CAAC;IACpD,CAAC;IACD,MAAM,OAAO,GAAG,SAAS,CAAC,MAAM,CAC9B,CAAC,IAAI,EAAE,EAAE,CAAC,IAAI,CAAC,UAAU,CAAC,GAAG,OAAO,GAAG,CAAC,IAAI,IAAI,CAAC,UAAU,CAAC,OAAO,CAAC,CACrE,CAAC;IACF,IAAI,OAAO,CAAC,MAAM,KAAK,CAAC;QAAE,OAAO,OAAO,CAAC,CAAC,CAAC,CAAC;IAC5C,IAAI,OAAO,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;QACvB,MAAM,OAAO,GAAG,OAAO,CAAC,MAAM,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,IAAI,CAAC,UAAU,CAAC,GAAG,OAAO,GAAG,CAAC,CAAC,CAAC;QACzE,IAAI,OAAO,CAAC,MAAM,KAAK,CAAC;YAAE,OAAO,OAAO,CAAC,CAAC,CAAC,CAAC;QAC5C,OAAO,OAAO,CAAC,CAAC,CAAC,CAAC;IACpB,CAAC;IACD,MAAM,IAAI,KAAK,CAAC,2CAA2C,OAAO,iBAAiB,SAAS,CAAC,IAAI,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC;AAC7G,CAAC,CAAC;AAEF,MAAM,SAAS,GAAG,CAAC,MAAc,EAAU,EAAE;IAC3C,MAAM,IAAI,GAAG,MAAM,CAAC,IAAkC,CAAC;IACvD,MAAM,GAAG,GAAG,IAAI,CAAC,CAAC,CAAC,CAAC;IACpB,OAAO,OAAO,GAAG,KAAK,QAAQ,CAAC,CAAC,CAAC,MAAM,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,MAAM,CAAC,GAAG,CAAC,CAAC,CAAC;AACzE,CAAC,CAAC;AAEF,MAAM,CAAC,MAAM,eAAe,GAAG,CAC7B,aAAgC,EAChC,aAAgC,EAChC,OAAyB,EACsC,EAAE;IACjE,MAAM,eAAe,GAAG,OAAO,CAAC,eAAe,IAAI,IAAI,CAAC;IAExD,MAAM,QAAQ,GAAa,EAAE,CAAC;IAC9B,IAAI,IAAI,GAAG,CAAC,CAAC,CAAC;IACd,KAAK,MAAM,OAAO,IAAI,aAAa,EAAE,CAAC;QACpC,IAAI,eAAe,IAAI,OAAO,KAAK,IAAI,EAAE,CAAC;YACxC,SAAS;QACX,CAAC;QACD,IAAI,GAAG,OAAO,CAAC;QACf,IAAI,OAAO,KAAK,OAAO,CAAC,OAAO,EAAE,CAAC;YAChC,SAAS;QACX,CAAC;QACD,QAAQ,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC;IACzB,CAAC;IAED,MAAM,SAAS,GAAG,QAAQ,CAAC,GAAG,CAAC,CAAC,OAAO,EAAE,EAAE,CAAC,aAAa,CAAC,OAAO,CAAC,IAAI,OAAO,CAAC,CAAC;IAC/E,MAAM,QAAQ,GAAG,SAAS,CAAC,MAAM,CAAC,CAAC,KAAK,EAAE,EAAE,CAAC,KAAK,KAAK,OAAO,CAAC,iBAAiB,CAAC,CAAC;IAClF,MAAM,KAAK,GAAe,EAAE,CAAC;IAC7B,IAAI,OAAO,GAAa,EAAE,CAAC;IAC3B,KAAK,MAAM,KAAK,IAAI,SAAS,EAAE,CAAC;QAC9B,IAAI,KAAK,KAAK,OAAO,CAAC,iBAAiB,EAAE,CAAC;YACxC,IAAI,OAAO,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;gBACvB,KAAK,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC;gBACpB,OAAO,GAAG,EAAE,CAAC;YACf,CAAC;YACD,SAAS;QACX,CAAC;QACD,OAAO,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;IACtB,CAAC;IACD,IAAI,OAAO,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;QACvB,KAAK,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC;IACtB,CAAC;IAED,OAAO,EAAE,QAAQ,EAAE,QAAQ,EAAE,KAAK,EAAE,CAAC;AACvC,CAAC,CAAC;AAEF,MAAM,iBAAiB,GAAG,CAAC,SAAkB,EAAY,EAAE;IACzD,IAAI,KAAqB,CAAC;IAC1B,IAAI,SAAS,EAAE,CAAC;QACd,MAAM,GAAG,GAAG,EAAE,CAAC,YAAY,CAAC,SAAS,EAAE,OAAO,CAAC,CAAC;QAChD,KAAK,GAAG,IAAI,CAAC,KAAK,CAAC,GAAG,CAAmB,CAAC;IAC5C,CAAC;SAAM,CAAC;QACN,KAAK,GAAG,SAA2B,CAAC;IACtC,CAAC;IACD,IAAI,CAAC,KAAK,CAAC,OAAO,CAAC,KAAK,CAAC,OAAO,CAAC,IAAI,KAAK,CAAC,OAAO,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;QAChE,MAAM,IAAI,KAAK,CAAC,qDAAqD,CAAC,CAAC;IACzE,CAAC;IACD,OAAO,CAAC,GAAG,KAAK,CAAC,OAAO,EAAE,MAAM,EAAE,SAAS,CAAC,CAAC;AAC/C,CAAC,CAAC;AAEF,MAAM,OAAO,YAAY;IAevB,YAAoB,OAAyB,EAAE,OAAiC;QAC9E,IAAI,CAAC,OAAO,GAAG,OAAO,CAAC;QACvB,IAAI,CAAC,UAAU,GAAG,OAAO,CAAC,UAAU,CAAC;QACrC,IAAI,CAAC,iBAAiB,GAAG,OAAO,CAAC,iBAAiB,CAAC;QACnD,IAAI,CAAC,SAAS,GAAG,OAAO,CAAC,SAAS,CAAC;QACnC,IAAI,CAAC,OAAO,GAAG,OAAO,CAAC,OAAO,CAAC;QAC/B,IAAI,CAAC,eAAe,GAAG,OAAO,CAAC,eAAe,CAAC;QAC/C,IAAI,CAAC,aAAa,GAAG,iBAAiB,CAAC,OAAO,CAAC,SAAS,CAAC,CAAC;QAC1D,IAAI,CAAC,OAAO,GAAG,IAAI,CAAC,aAAa,CAAC,OAAO,CAAC,OAAO,CAAC,UAAU,CAAC,CAAC;QAC9D,IAAI,IAAI,CAAC,OAAO,GAAG,CAAC,EAAE,CAAC;YACrB,MAAM,IAAI,KAAK,CAAC,gBAAgB,OAAO,CAAC,UAAU,+BAA+B,CAAC,CAAC;QACrF,CAAC;QACD,MAAM,GAAG,GAAG,IAAI,CAAC,aAAa,CAAC,OAAO,CAAC,OAAO,CAAC,QAAQ,CAAC,CAAC;QACzD,IAAI,CAAC,KAAK,GAAG,GAAG,IAAI,CAAC,CAAC,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,IAAI,CAAC;QAEnC,MAAM,UAAU,GAAG,IAAI,CAAC,OAAO,CAAC,UAAU,CAAC;QAC3C,MAAM,WAAW,GACf,UAAU,CAAC,QAAQ,CAAC,UAAU,CAAC;eAC5B,UAAU,CAAC,IAAI,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,IAAI,CAAC,UAAU,CAAC,WAAW,CAAC,CAAC,CAAC;QAC7D,IAAI,CAAC,WAAW,EAAE,CAAC;YACjB,MAAM,IAAI,KAAK,CACb,gCAAgC,UAAU,CAAC,IAAI,CAAC,IAAI,CAAC,IAAI;kBACvD,0EAA0E,CAC7E,CAAC;QACJ,CAAC;QACD,IAAI,CAAC,iBAAiB,GAAG,WAAW,CAAC,UAAU,EAAE,UAAU,CAAC,CAAC;QAC7D,IAAI,CAAC,wBAAwB,GAAG,WAAW,CAAC,UAAU,EAAE,kBAAkB,EAAE,iBAAiB,CAAC,CAAC;QAC/F,IAAI,CAAC,kBAAkB,GAAG,WAAW,CAAC,IAAI,CAAC,OAAO,CAAC,WAAW,EAAE,WAAW,CAAC,CAAC;QAC7E,IAAI,CAAC,oBAAoB,GAAG,WAAW,CAAC,IAAI,CAAC,OAAO,CAAC,WAAW,EAAE,aAAa,CAAC,CAAC;IACnF,CAAC;IAED,MAAM,CAAC,KAAK,CAAC,MAAM,CAAC,UAA0B,EAAE;QAC9C,MAAM,IAAI,GAA6B;YACrC,SAAS,EAAE,OAAO,CAAC,SAAS,IAAI,0BAA0B,EAAE;YAC5D,SAAS,EAAE,OAAO,CAAC,SAAS,IAAI,EAAE;YAClC,UAAU,EAAE,OAAO,CAAC,UAAU,IAAI,KAAK;YACvC,UAAU,EAAE,OAAO,CAAC,UAAU,IAAI,SAAS;YAC3C,QAAQ,EAAE,OAAO,CAAC,QAAQ,IAAI,OAAO;YACrC,iBAAiB,EAAE,OAAO,CAAC,iBAAiB,IAAI,MAAM;YACtD,SAAS,EAAE,OAAO,CAAC,SAAS,IAAI,CAAC,GAAG;YACpC,OAAO,EAAE,OAAO,CAAC,OAAO,IAAI,GAAG;YAC/B,eAAe,EAAE,OAAO,CAAC,eAAe,IAAI,IAAI;SACjD,CAAC;QACF,MAAM,OAAO,GAAG,MAAM,gBAAgB,CAAC,MAAM,CAAC,IAAI,CAAC,SAAS,EAAE;YAC5D,sBAAsB,EAAE,UAAU;SACnC,CAAC,CAAC;QACH,OAAO,IAAI,YAAY,CAAC,OAAO,EAAE,IAAI,CAAC,CAAC;IACzC,CAAC;IAED,IAAI,WAAW;QACb,OAAO,UAAU,CAAC;IACpB,CAAC;IAED,KAAK,CAAC,iBAAiB,CAAC,OAAe;QACrC,MAAM,EAAE,QAAQ,EAAE,UAAU,EAAE,GAAG,WAAW,CAAC,OAAO,CAAC,CAAC;QACtD,OAAO,IAAI,CAAC,kBAAkB,CAAC,QAAQ,EAAE,UAAU,CAAC,CAAC;IACvD,CAAC;IAED,KAAK,CAAC,kBAAkB,CACtB,QAA0C,EAC1C,UAAkB;QAElB,MAAM,IAAI,GAAG,aAAa,CAAC,QAAQ,CAAC,CAAC;QACrC,MAAM,SAAS,GACb,UAAU,KAAK,IAAI,CAAC,UAAU;YAC5B,CAAC,CAAC,IAAI;YACN,CAAC,CAAC,cAAc,CAAC,IAAI,EAAE,UAAU,EAAE,IAAI,CAAC,UAAU,CAAC,CAAC;QACxD,MAAM,cAAc,GAAG,IAAI,MAAM,CAAC,SAAS,EAAE,SAAS,EAAE,CAAC,CAAC,EAAE,SAAS,CAAC,MAAM,CAAC,CAAC,CAAC;QAC/E,MAAM,YAAY,GAAG,IAAI,MAAM,CAAC,OAAO,EAAE,aAAa,CAAC,IAAI,CAAC,CAAC,MAAM,CAAC,SAAS,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,CAAC,CAAC;QAC9F,MAAM,OAAO,GAAG,MAAM,IAAI,CAAC,OAAO,CAAC,GAAG,CAAC;YACrC,CAAC,IAAI,CAAC,iBAAiB,CAAC,EAAE,cAAc;YACxC,CAAC,IAAI,CAAC,wBAAwB,CAAC,EAAE,YAAY;SAC9C,CAAC,CAAC;QAEH,MAAM,QAAQ,GAAG,OAAO,CAAC,IAAI,CAAC,kBAAkB,CAAW,CAAC;QAC5D,MAAM,UAAU,GAAG,OAAO,CAAC,IAAI,CAAC,oBAAoB,CAAW,CAAC;QAChE,MAAM,SAAS,GAAG,IAAI,CAAC,GAAG,CAAC,CAAC,EAAE,IAAI,CAAC,GAAG,CAAC,SAAS,CAAC,UAAU,CAAC,EAAE,MAAM,CAAC,QAAQ,CAAC,IAAI,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,CAAC;QAC9F,MAAM,aAAa,GAAG,IAAI,CAAC,YAAY,CAAC,QAAQ,EAAE,SAAS,CAAC,CAAC;QAC7D,MAAM,OAAO,GAAG,eAAe,CAAC,aAAa,EAAE,IAAI,CAAC,aAAa,EAAE;YACjE,OAAO,EAAE,IAAI,CAAC,OAAO;YACrB,iBAAiB,EAAE,IAAI,CAAC,iBAAiB;YACzC,eAAe,EAAE,IAAI,CAAC,eAAe;SACtC,CAAC,CAAC;QACH,OAAO;YACL,QAAQ,EAAE,OAAO,CAAC,QAAQ;YAC1B,WAAW,EAAE,OAAO,CAAC,QAAQ,CAAC,IAAI,CAAC,GAAG,CAAC;YACvC,eAAe,EAAE,OAAO,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,IAAI,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC;YACxE,QAAQ,EAAE,OAAO,CAAC,QAAQ;YAC1B,aAAa;YACb,SAAS;SACV,CAAC;IACJ,CAAC;IAEO,YAAY,CAAC,QAAgB,EAAE,SAAiB;QACtD,MAAM,IAAI,GAAG,QAAQ,CAAC,IAAI,CAAC,GAAG,CAAC,MAAM,CAAC,CAAC;QACvC,IAAI,IAAI,CAAC,MAAM,KAAK,CAAC,IAAI,IAAI,CAAC,CAAC,CAAC,KAAK,CAAC,EAAE,CAAC;YACvC,MAAM,IAAI,KAAK,CAAC,4CAA4C,IAAI,CAAC,IAAI,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC;QAClF,CAAC;QACD,MAAM,OAAO,GAAG,IAAI,CAAC,CAAC,CAAC,CAAC;QACxB,MAAM,IAAI,GAAG,QAAQ,CAAC,IAA+B,CAAC;QAEtD,MAAM,GAAG,GAAa,EAAE,CAAC;QACzB,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,SAAS,EAAE,CAAC,EAAE,EAAE,CAAC;YACnC,IAAI,OAAO,GAAG,CAAC,CAAC;YAChB,IAAI,SAAS,GAAG,CAAC,QAAQ,CAAC;YAC1B,MAAM,IAAI,GAAG,CAAC,GAAG,OAAO,CAAC;YACzB,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,OAAO,EAAE,CAAC,EAAE,EAAE,CAAC;gBACjC,MAAM,GAAG,GAAG,MAAM,CAAC,IAAI,CAAC,IAAI,GAAG,CAAC,CAAC,CAAC,CAAC;gBACnC,MAAM,KAAK,GACT,GAAG;sBACD,CAAC,CAAC,KAAK,IAAI,CAAC,OAAO,CAAC,CAAC,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC,CAAC,GAAG,CAAC;sBAC3C,CAAC,IAAI,CAAC,KAAK,KAAK,IAAI,IAAI,CAAC,KAAK,IAAI,CAAC,KAAK,CAAC,CAAC,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC,CAAC,GAAG,CAAC,CAAC;gBACnE,IAAI,KAAK,GAAG,SAAS,EAAE,CAAC;oBACtB,SAAS,GAAG,KAAK,CAAC;oBAClB,OAAO,GAAG,CAAC,CAAC;gBACd,CAAC;YACH,CAAC;YACD,GAAG,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC;QACpB,CAAC;QACD,OAAO,GAAG,CAAC;IACb,CAAC;CACF;AAED,MAAM,WAAW,GAAG,CAAC,OAAe,EAAkD,EAAE;IACtF,MAAM,GAAG,GAAG,EAAE,CAAC,YAAY,CAAC,OAAO,CAAC,CAAC;IACrC,IAAI,GAAG,CAAC,MAAM,GAAG,EAAE,EAAE,CAAC;QACpB,MAAM,IAAI,KAAK,CAAC,gCAAgC,OAAO,GAAG,CAAC,CAAC;IAC9D,CAAC;IAED,MAAM,IAAI,GAAG,GAAG,CAAC,QAAQ,CAAC,OAAO,EAAE,CAAC,EAAE,CAAC,CAAC,CAAC;IACzC,MAAM,IAAI,GAAG,GAAG,CAAC,QAAQ,CAAC,OAAO,EAAE,CAAC,EAAE,EAAE,CAAC,CAAC;IAC1C,IAAI,IAAI,KAAK,MAAM,IAAI,IAAI,KAAK,MAAM,EAAE,CAAC;QACvC,MAAM,IAAI,KAAK,CAAC,2CAA2C,OAAO,GAAG,CAAC,CAAC;IACzE,CAAC;IAED,IAAI,MAAM,GAAG,EAAE,CAAC;IAChB,IAAI,MAAM,GAAG,CAAC,CAAC;IACf,IAAI,QAAQ,GAAG,CAAC,CAAC;IACjB,IAAI,UAAU,GAAG,CAAC,CAAC;IACnB,IAAI,aAAa,GAAG,CAAC,CAAC;IACtB,IAAI,UAAU,GAAG,CAAC,CAAC,CAAC;IACpB,IAAI,QAAQ,GAAG,CAAC,CAAC;IAEjB,OAAO,MAAM,GAAG,CAAC,IAAI,GAAG,CAAC,MAAM,EAAE,CAAC;QAChC,MAAM,OAAO,GAAG,GAAG,CAAC,QAAQ,CAAC,OAAO,EAAE,MAAM,EAAE,MAAM,GAAG,CAAC,CAAC,CAAC;QAC1D,MAAM,SAAS,GAAG,GAAG,CAAC,YAAY,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC;QAC/C,MAAM,eAAe,GAAG,MAAM,GAAG,CAAC,CAAC;QACnC,MAAM,IAAI,GAAG,eAAe,GAAG,SAAS,GAAG,CAAC,SAAS,GAAG,CAAC,CAAC,CAAC;QAE3D,IAAI,OAAO,KAAK,MAAM,IAAI,SAAS,IAAI,EAAE,EAAE,CAAC;YAC1C,MAAM,GAAG,GAAG,CAAC,YAAY,CAAC,eAAe,CAAC,CAAC;YAC3C,QAAQ,GAAG,GAAG,CAAC,YAAY,CAAC,eAAe,GAAG,CAAC,CAAC,CAAC;YACjD,UAAU,GAAG,GAAG,CAAC,YAAY,CAAC,eAAe,GAAG,CAAC,CAAC,CAAC;YACnD,aAAa,GAAG,GAAG,CAAC,YAAY,CAAC,eAAe,GAAG,EAAE,CAAC,CAAC;QACzD,CAAC;aAAM,IAAI,OAAO,KAAK,MAAM,EAAE,CAAC;YAC9B,UAAU,GAAG,eAAe,CAAC;YAC7B,QAAQ,GAAG,SAAS,CAAC;YACrB,MAAM;QACR,CAAC;QACD,MAAM,GAAG,IAAI,CAAC;IAChB,CAAC;IAED,IAAI,UAAU,GAAG,CAAC,IAAI,QAAQ,IAAI,CAAC,EAAE,CAAC;QACpC,MAAM,IAAI,KAAK,CAAC,oCAAoC,OAAO,GAAG,CAAC,CAAC;IAClE,CAAC;IACD,IAAI,QAAQ,IAAI,CAAC,IAAI,UAAU,IAAI,CAAC,EAAE,CAAC;QACrC,MAAM,IAAI,KAAK,CAAC,qCAAqC,OAAO,GAAG,CAAC,CAAC;IACnE,CAAC;IAED,IAAI,iBAAiB,GAAG,CAAC,CAAC;IAC1B,IAAI,WAAyB,CAAC;IAE9B,IAAI,MAAM,KAAK,CAAC,EAAE,CAAC;QACjB,IAAI,aAAa,KAAK,CAAC,EAAE,CAAC;YACxB,iBAAiB,GAAG,IAAI,CAAC,KAAK,CAAC,QAAQ,GAAG,QAAQ,CAAC,CAAC;YACpD,WAAW,GAAG,IAAI,YAAY,CAAC,iBAAiB,GAAG,QAAQ,CAAC,CAAC;YAC7D,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,WAAW,CAAC,MAAM,EAAE,CAAC,EAAE,EAAE,CAAC;gBAC5C,MAAM,CAAC,GAAG,GAAG,CAAC,SAAS,CAAC,UAAU,GAAG,CAAC,CAAC,CAAC;gBACxC,WAAW,CAAC,CAAC,CAAC,GAAG,CAAC,CAAC,GAAG,GAAG,CAAC,GAAG,GAAG,CAAC;YACnC,CAAC;QACH,CAAC;aAAM,IAAI,aAAa,KAAK,EAAE,EAAE,CAAC;YAChC,MAAM,cAAc,GAAG,CAAC,CAAC;YACzB,iBAAiB,GAAG,IAAI,CAAC,KAAK,CAAC,QAAQ,GAAG,CAAC,QAAQ,GAAG,cAAc,CAAC,CAAC,CAAC;YACvE,WAAW,GAAG,IAAI,YAAY,CAAC,iBAAiB,GAAG,QAAQ,CAAC,CAAC;YAC7D,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,WAAW,CAAC,MAAM,EAAE,CAAC,EAAE,EAAE,CAAC;gBAC5C,MAAM,CAAC,GAAG,GAAG,CAAC,WAAW,CAAC,UAAU,GAAG,CAAC,GAAG,cAAc,CAAC,CAAC;gBAC3D,WAAW,CAAC,CAAC,CAAC,GAAG,CAAC,GAAG,KAAK,CAAC;YAC7B,CAAC;QACH,CAAC;aAAM,IAAI,aAAa,KAAK,EAAE,EAAE,CAAC;YAChC,MAAM,cAAc,GAAG,CAAC,CAAC;YACzB,iBAAiB,GAAG,IAAI,CAAC,KAAK,CAAC,QAAQ,GAAG,CAAC,QAAQ,GAAG,cAAc,CAAC,CAAC,CAAC;YACvE,WAAW,GAAG,IAAI,YAAY,CAAC,iBAAiB,GAAG,QAAQ,CAAC,CAAC;YAC7D,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,WAAW,CAAC,MAAM,EAAE,CAAC,EAAE,EAAE,CAAC;gBAC5C,MAAM,CAAC,GAAG,UAAU,GAAG,CAAC,GAAG,cAAc,CAAC;gBAC1C,IAAI,CAAC,GAAG,GAAG,CAAC,CAAC,CAAC,GAAG,CAAC,GAAG,CAAC,CAAC,GAAG,CAAC,CAAC,IAAI,CAAC,CAAC,GAAG,CAAC,GAAG,CAAC,CAAC,GAAG,CAAC,CAAC,IAAI,EAAE,CAAC,CAAC;gBACxD,IAAI,CAAC,GAAG,QAAQ;oBAAE,CAAC,IAAI,SAAS,CAAC;gBACjC,WAAW,CAAC,CAAC,CAAC,GAAG,CAAC,GAAG,OAAO,CAAC;YAC/B,CAAC;QACH,CAAC;aAAM,IAAI,aAAa,KAAK,EAAE,EAAE,CAAC;YAChC,MAAM,cAAc,GAAG,CAAC,CAAC;YACzB,iBAAiB,GAAG,IAAI,CAAC,KAAK,CAAC,QAAQ,GAAG,CAAC,QAAQ,GAAG,cAAc,CAAC,CAAC,CAAC;YACvE,WAAW,GAAG,IAAI,YAAY,CAAC,iBAAiB,GAAG,QAAQ,CAAC,CAAC;YAC7D,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,WAAW,CAAC,MAAM,EAAE,CAAC,EAAE,EAAE,CAAC;gBAC5C,MAAM,CAAC,GAAG,GAAG,CAAC,WAAW,CAAC,UAAU,GAAG,CAAC,GAAG,cAAc,CAAC,CAAC;gBAC3D,WAAW,CAAC,CAAC,CAAC,GAAG,CAAC,GAAG,UAAU,CAAC;YAClC,CAAC;QACH,CAAC;aAAM,CAAC;YACN,MAAM,IAAI,KAAK,CAAC,uCAAuC,aAAa,EAAE,CAAC,CAAC;QAC1E,CAAC;IACH,CAAC;SAAM,IAAI,MAAM,KAAK,CAAC,IAAI,aAAa,KAAK,EAAE,EAAE,CAAC;QAChD,MAAM,cAAc,GAAG,CAAC,CAAC;QACzB,iBAAiB,GAAG,IAAI,CAAC,KAAK,CAAC,QAAQ,GAAG,CAAC,QAAQ,GAAG,cAAc,CAAC,CAAC,CAAC;QACvE,WAAW,GAAG,IAAI,YAAY,CAAC,iBAAiB,GAAG,QAAQ,CAAC,CAAC;QAC7D,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,WAAW,CAAC,MAAM,EAAE,CAAC,EAAE,EAAE,CAAC;YAC5C,WAAW,CAAC,CAAC,CAAC,GAAG,GAAG,CAAC,WAAW,CAAC,UAAU,GAAG,CAAC,GAAG,cAAc,CAAC,CAAC;QACpE,CAAC;IACH,CAAC;SAAM,CAAC;QACN,MAAM,IAAI,KAAK,CAAC,0BAA0B,MAAM,oBAAoB,aAAa,EAAE,CAAC,CAAC;IACvF,CAAC;IAED,IAAI,QAAQ,KAAK,CAAC,EAAE,CAAC;QACnB,OAAO,EAAE,QAAQ,EAAE,SAAS,CAAC,WAAW,CAAC,EAAE,UAAU,EAAE,CAAC;IAC1D,CAAC;IAED,MAAM,IAAI,GAAG,IAAI,YAAY,CAAC,iBAAiB,CAAC,CAAC;IACjD,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,iBAAiB,EAAE,CAAC,EAAE,EAAE,CAAC;QAC3C,IAAI,GAAG,GAAG,CAAC,CAAC;QACZ,KAAK,IAAI,EAAE,GAAG,CAAC,EAAE,EAAE,GAAG,QAAQ,EAAE,EAAE,EAAE,EAAE,CAAC;YACrC,GAAG,IAAI,WAAW,CAAC,CAAC,GAAG,QAAQ,GAAG,EAAE,CAAC,CAAC;QACxC,CAAC;QACD,IAAI,CAAC,CAAC,CAAC,GAAG,GAAG,GAAG,QAAQ,CAAC;IAC3B,CAAC;IACD,OAAO,EAAE,QAAQ,EAAE,SAAS,CAAC,IAAI,CAAC,EAAE,UAAU,EAAE,CAAC;AACnD,CAAC,CAAC;AAEF,MAAM,aAAa,GAAG,CAAC,QAA0C,EAAgB,EAAE;IACjF,IAAI,QAAQ,YAAY,YAAY,EAAE,CAAC;QACrC,OAAO,SAAS,CAAC,QAAQ,CAAC,CAAC;IAC7B,CAAC;IACD,IAAI,CAAC,KAAK,CAAC,OAAO,CAAC,QAAQ,CAAC,EAAE,CAAC;QAC7B,MAAM,IAAI,KAAK,CAAC,2CAA2C,CAAC,CAAC;IAC/D,CAAC;IACD,MAAM,GAAG,GAAG,IAAI,YAAY,CAAC,QAAQ,CAAC,MAAM,CAAC,CAAC;IAC9C,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,QAAQ,CAAC,MAAM,EAAE,CAAC,EAAE,EAAE,CAAC;QACzC,GAAG,CAAC,CAAC,CAAC,GAAG,MAAM,CAAC,QAAQ,CAAC,CAAC,CAAC,CAAC,CAAC;IAC/B,CAAC;IACD,OAAO,SAAS,CAAC,GAAG,CAAC,CAAC;AACxB,CAAC,CAAC;AAEF,MAAM,SAAS,GAAG,CAAC,GAAiB,EAAgB,EAAE;IACpD,MAAM,GAAG,GAAG,IAAI,YAAY,CAAC,GAAG,CAAC,MAAM,CAAC,CAAC;IACzC,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,GAAG,CAAC,MAAM,EAAE,CAAC,EAAE,EAAE,CAAC;QACpC,MAAM,CAAC,GAAG,GAAG,CAAC,CAAC,CAAC,CAAC;QACjB,GAAG,CAAC,CAAC,CAAC,GAAG,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC;IACvC,CAAC;IACD,OAAO,GAAG,CAAC;AACb,CAAC,CAAC;AAEF,MAAM,cAAc,GAAG,CAAC,QAAsB,EAAE,OAAe,EAAE,OAAe,EAAgB,EAAE;IAChG,IAAI,OAAO,KAAK,OAAO;QAAE,OAAO,QAAQ,CAAC;IACzC,IAAI,QAAQ,CAAC,MAAM,KAAK,CAAC;QAAE,OAAO,QAAQ,CAAC;IAC3C,MAAM,QAAQ,GAAG,CAAC,QAAQ,CAAC,MAAM,GAAG,CAAC,CAAC,GAAG,OAAO,CAAC;IACjD,MAAM,MAAM,GAAG,IAAI,CAAC,GAAG,CAAC,CAAC,EAAE,IAAI,CAAC,KAAK,CAAC,QAAQ,GAAG,OAAO,CAAC,GAAG,CAAC,CAAC,CAAC;IAC/D,MAAM,GAAG,GAAG,IAAI,YAAY,CAAC,MAAM,CAAC,CAAC;IACrC,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,MAAM,EAAE,CAAC,EAAE,EAAE,CAAC;QAChC,MAAM,CAAC,GAAG,CAAC,GAAG,OAAO,CAAC;QACtB,MAAM,MAAM,GAAG,CAAC,GAAG,OAAO,CAAC;QAC3B,MAAM,EAAE,GAAG,IAAI,CAAC,KAAK,CAAC,MAAM,CAAC,CAAC;QAC9B,MAAM,EAAE,GAAG,IAAI,CAAC,GAAG,CAAC,QAAQ,CAAC,MAAM,GAAG,CAAC,EAAE,EAAE,GAAG,CAAC,CAAC,CAAC;QACjD,MAAM,CAAC,GAAG,MAAM,GAAG,EAAE,CAAC;QACtB,GAAG,CAAC,CAAC,CAAC,GAAG,QAAQ,CAAC,EAAE,CAAC,GAAG,CAAC,CAAC,GAAG,CAAC,CAAC,GAAG,QAAQ,CAAC,EAAE,CAAC,GAAG,CAAC,CAAC;IACrD,CAAC;IACD,OAAO,GAAG,CAAC;AACb,CAAC,CAAC"}