hama-js 1.3.2 → 1.3.4

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"}

package/dist/node/index.js

@@ -1,27 +1,104 @@
+import fs from "node:fs";
 import path from "node:path";
 import { fileURLToPath } from "node:url";
 import { InferenceSession, Tensor } from "onnxruntime-node";
-import { decodeIdsToResult, encodeText } from "./tokenizer.js";
+import { decodeIdsToResult, decoderIds, encodeText } from "./tokenizer.js";
 const __dirname = path.dirname(fileURLToPath(import.meta.url));
 const defaultModelPath = path.join(__dirname, "assets", "g2p_fp16.onnx");
+const defaultEncoderModelPath = path.join(__dirname, "assets", "encoder.onnx");
+const defaultDecoderStepModelPath = path.join(__dirname, "assets", "decoder_step.onnx");
+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 G2PNodeModel {
-    constructor(session, opts) {
-        this.session = session;
+    constructor(sessions, opts) {
+        this.session = sessions.session;
+        this.encoderSession = sessions.encoderSession;
+        this.decoderStepSession = sessions.decoderStepSession;
         this.maxInputLen = opts.maxInputLen;
         this.maxOutputLen = opts.maxOutputLen;
     }
     static async create(options = {}) {
+        if ((options.encoderModelPath === undefined) !== (options.decoderStepModelPath === undefined)) {
+            throw new Error("encoderModelPath and decoderStepModelPath must be provided together");
+        }
+        let resolvedModelPath = options.modelPath ?? defaultModelPath;
+        let resolvedEncoderPath = options.encoderModelPath ?? defaultEncoderModelPath;
+        let resolvedDecoderStepPath = options.decoderStepModelPath ?? defaultDecoderStepModelPath;
+        if (options.modelPath &&
+            fs.existsSync(options.modelPath) &&
+            fs.statSync(options.modelPath).isDirectory()) {
+            resolvedEncoderPath = path.join(options.modelPath, "encoder.onnx");
+            resolvedDecoderStepPath = path.join(options.modelPath, "decoder_step.onnx");
+            resolvedModelPath = path.join(options.modelPath, "g2p_fp16.onnx");
+        }
         const opts = {
-            modelPath: options.modelPath ?? defaultModelPath,
+            modelPath: resolvedModelPath,
+            encoderModelPath: resolvedEncoderPath,
+            decoderStepModelPath: resolvedDecoderStepPath,
             maxInputLen: options.maxInputLen ?? 128,
+            // Retained for API compatibility; autoregressive ONNX sets output length in-graph.
             maxOutputLen: options.maxOutputLen ?? 32,
         };
-        const session = await InferenceSession.create(opts.modelPath, {
-            graphOptimizationLevel: "disabled",
-        });
-        return new G2PNodeModel(session, opts);
+        const sessionOptions = { graphOptimizationLevel: "disabled" };
+        const shouldUseSplit = (options.encoderModelPath !== undefined && options.decoderStepModelPath !== undefined) ||
+            (fs.existsSync(opts.encoderModelPath) && fs.existsSync(opts.decoderStepModelPath));
+        if (shouldUseSplit) {
+            const [encoderSession, decoderStepSession] = await Promise.all([
+                InferenceSession.create(opts.encoderModelPath, sessionOptions),
+                InferenceSession.create(opts.decoderStepModelPath, sessionOptions),
+            ]);
+            return new G2PNodeModel({ encoderSession, decoderStepSession }, opts);
+        }
+        const session = await InferenceSession.create(opts.modelPath, sessionOptions);
+        return new G2PNodeModel({ session }, opts);
+    }
+    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 predict(text) {
+    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.maxInputLen);
         const inputIds = new BigInt64Array(encoded.ids);
         const inputLengths = new BigInt64Array([BigInt(encoded.length || 1)]);
@@ -32,7 +109,150 @@ export class G2PNodeModel {
         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.maxInputLen);
+        const inputIds = new BigInt64Array(encoded.ids);
+        const inputLengths = new BigInt64Array([BigInt(encoded.length || 1)]);
+        const encoderFeeds = {
+            input_ids: new Tensor("int64", inputIds, [1, this.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.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;
+};
+export { ASRNodeModel, decodeCtcTokens, } from "./asr.js";
 //# sourceMappingURL=index.js.map