as-labs 0.1.0

package/dist/branch-hinting/transform/index.js

@@ -0,0 +1,446 @@
+import { decode } from "@webassemblyjs/wasm-parser";
+import { CallExpression, IdentifierExpression, ParenthesizedExpression, getFunctionName, } from "assemblyscript/dist/assemblyscript.js";
+import { Transform } from "assemblyscript/dist/transform.js";
+const CUSTOM_SECTION_NAME = "metadata.code.branch_hint";
+export default class BranchHintTransform extends Transform {
+    functionHintsByDeclaration = new Map();
+    functionHintsByInternalName = new Map();
+    afterParse(parser) {
+        for (const source of parser.sources) {
+            if (source.sourceKind === 2 /* SourceKind.Library */)
+                continue;
+            if (source.internalPath.startsWith("~lib/"))
+                continue;
+            for (const statement of source.statements) {
+                this.visitNode(statement, null);
+            }
+        }
+    }
+    afterInitialize(program) {
+        for (const [declaration, hints] of this.functionHintsByDeclaration) {
+            const element = program.elementsByDeclaration.get(declaration);
+            if (!element || !element.internalName || hints.length === 0)
+                continue;
+            this.functionHintsByInternalName.set(element.internalName, hints);
+        }
+    }
+    afterCompile(module) {
+        if (this.functionHintsByInternalName.size === 0)
+            return;
+        const originalEmitBinary = module.emitBinary.bind(module);
+        module.emitBinary = ((...args) => {
+            const result = originalEmitBinary(...args);
+            result.binary = injectBranchHints(result.binary, module, this.functionHintsByInternalName);
+            return result;
+        });
+    }
+    visitNode(node, currentFunction) {
+        if (!node)
+            return;
+        if (Array.isArray(node)) {
+            for (const item of node)
+                this.visitNode(item, currentFunction);
+            return;
+        }
+        switch (node.kind) {
+            case 55 /* NodeKind.FunctionDeclaration */:
+                this.visitFunctionLike(node, node);
+                return;
+            case 58 /* NodeKind.MethodDeclaration */:
+                this.visitFunctionLike(node, node);
+                return;
+            case 41 /* NodeKind.If */:
+                this.visitConditionalStatement(node, currentFunction, [node.ifTrue, node.ifFalse]);
+                return;
+            case 49 /* NodeKind.While */:
+                this.visitConditionalStatement(node, currentFunction, [node.body]);
+                return;
+            case 33 /* NodeKind.Do */:
+                this.visitConditionalStatement(node, currentFunction, [node.body]);
+                return;
+            case 39 /* NodeKind.For */:
+                this.visitConditionalStatement(node, currentFunction, [node.initializer, node.incrementor, node.body]);
+                return;
+            case 30 /* NodeKind.Block */:
+                this.visitNode(node.statements, currentFunction);
+                return;
+            case 44 /* NodeKind.Switch */:
+                this.visitNode(node.condition, currentFunction);
+                this.visitNode(node.cases, currentFunction);
+                return;
+            case 64 /* NodeKind.SwitchCase */:
+                this.visitNode(node.label, currentFunction);
+                this.visitNode(node.statements, currentFunction);
+                return;
+            case 38 /* NodeKind.Expression */:
+                this.visitNode(node.expression, currentFunction);
+                return;
+            case 43 /* NodeKind.Return */:
+                this.visitNode(node.value, currentFunction);
+                return;
+            case 47 /* NodeKind.Variable */:
+                this.visitNode(node.declarations, currentFunction);
+                return;
+            case 61 /* NodeKind.VariableDeclaration */:
+                this.visitNode(node.initializer, currentFunction);
+                return;
+            case 9 /* NodeKind.Call */:
+                this.visitNode(node.expression, currentFunction);
+                this.visitNode(node.args, currentFunction);
+                return;
+            case 8 /* NodeKind.Binary */:
+                this.visitNode(node.left, currentFunction);
+                this.visitNode(node.right, currentFunction);
+                return;
+            case 28 /* NodeKind.UnaryPrefix */:
+            case 27 /* NodeKind.UnaryPostfix */:
+            case 20 /* NodeKind.Parenthesized */:
+            case 7 /* NodeKind.Assertion */:
+                this.visitNode(node.operand ?? node.expression, currentFunction);
+                return;
+            case 22 /* NodeKind.Ternary */:
+                this.visitNode(node.condition, currentFunction);
+                this.visitNode(node.ifThen, currentFunction);
+                this.visitNode(node.ifElse, currentFunction);
+                return;
+            case 40 /* NodeKind.ForOf */:
+                this.visitNode(node.variable, currentFunction);
+                this.visitNode(node.iterable, currentFunction);
+                this.visitNode(node.body, currentFunction);
+                return;
+            case 46 /* NodeKind.Try */:
+                this.visitNode(node.bodyStatements, currentFunction);
+                this.visitNode(node.catchStatements, currentFunction);
+                this.visitNode(node.finallyStatements, currentFunction);
+                return;
+            case 45 /* NodeKind.Throw */:
+                this.visitNode(node.value, currentFunction);
+                return;
+            case 48 /* NodeKind.Void */:
+                this.visitNode(node.expression, currentFunction);
+                return;
+            default:
+                return;
+        }
+    }
+    visitFunctionLike(node, declaration) {
+        if (!node.body)
+            return;
+        this.visitNode(node.body, declaration);
+    }
+    visitConditionalStatement(node, currentFunction, extraChildren) {
+        if (!currentFunction)
+            return;
+        const hint = unwrapHint(node.condition);
+        if (hint) {
+            node.condition = hint.expression;
+            let hints = this.functionHintsByDeclaration.get(currentFunction);
+            if (!hints) {
+                hints = [];
+                this.functionHintsByDeclaration.set(currentFunction, hints);
+            }
+            hints.push({ value: hint.value });
+        }
+        else {
+            this.visitNode(node.condition, currentFunction);
+        }
+        for (const child of extraChildren) {
+            this.visitNode(child, currentFunction);
+        }
+    }
+}
+function unwrapHint(expression) {
+    let current = expression;
+    while (current instanceof ParenthesizedExpression) {
+        current = current.expression;
+    }
+    if (!(current instanceof CallExpression))
+        return null;
+    const callee = current.expression;
+    if (!(callee instanceof IdentifierExpression))
+        return null;
+    if (current.args.length !== 1)
+        return null;
+    if (callee.text === "likely") {
+        return { value: 1, expression: current.args[0] };
+    }
+    if (callee.text === "unlikely") {
+        return { value: 0, expression: current.args[0] };
+    }
+    return null;
+}
+function injectBranchHints(binary, module, functionHintsByInternalName) {
+    const importedFunctionCount = countImportedFunctions(binary);
+    const functionOrder = [];
+    for (let i = 0, n = module.getNumFunctions(); i < n; i++) {
+        const name = getFunctionName(module.getFunctionByIndex(i));
+        functionOrder.push(name ?? "");
+    }
+    const conditionalOffsets = collectConditionalOffsets(binary);
+    const functionHints = [];
+    for (let definedIndex = 0; definedIndex < conditionalOffsets.length; definedIndex++) {
+        const internalName = functionOrder[importedFunctionCount + definedIndex];
+        const wanted = functionHintsByInternalName.get(internalName);
+        if (!wanted || wanted.length === 0)
+            continue;
+        const available = conditionalOffsets[definedIndex] || [];
+        const matched = [];
+        const count = Math.min(wanted.length, available.length);
+        for (let i = 0; i < count; i++) {
+            matched.push({
+                offset: available[i],
+                value: wanted[i].value,
+            });
+        }
+        if (matched.length === 0)
+            continue;
+        functionHints.push({
+            functionIndex: importedFunctionCount + definedIndex,
+            hints: matched,
+        });
+    }
+    if (functionHints.length === 0)
+        return binary;
+    const payload = encodeBranchHintPayload(functionHints);
+    const customSection = encodeCustomSection(CUSTOM_SECTION_NAME, payload);
+    return insertCustomSectionBeforeCode(binary, customSection, CUSTOM_SECTION_NAME);
+}
+function countImportedFunctions(binary) {
+    const sections = parseSections(binary);
+    const importSection = sections.find((section) => section.id === 2);
+    if (!importSection)
+        return 0;
+    let offset = importSection.payloadOffset;
+    const end = importSection.end;
+    const countInfo = readU32(binary, offset);
+    offset = countInfo.nextOffset;
+    let functionImports = 0;
+    for (let i = 0; i < countInfo.value && offset < end; i++) {
+        const moduleName = readName(binary, offset);
+        offset = moduleName.nextOffset;
+        const fieldName = readName(binary, offset);
+        offset = fieldName.nextOffset;
+        const kind = binary[offset++];
+        switch (kind) {
+            case 0:
+                functionImports++;
+                offset = readU32(binary, offset).nextOffset;
+                break;
+            case 1:
+                offset = skipTableType(binary, offset);
+                break;
+            case 2:
+                offset = skipMemoryType(binary, offset);
+                break;
+            case 3:
+                offset = skipGlobalType(binary, offset);
+                break;
+            case 4:
+                offset = skipTagType(binary, offset);
+                break;
+            default:
+                throw new Error(`Unsupported import kind ${kind}`);
+        }
+    }
+    return functionImports;
+}
+function collectConditionalOffsets(binary) {
+    const ast = decode(binary, {
+        dump: false,
+        ignoreCodeSection: false,
+        ignoreDataSection: true,
+    });
+    const functions = [];
+    walk(ast, (node) => {
+        if (node?.type === "Func") {
+            functions.push(node);
+        }
+    });
+    return functions.map((func) => {
+        const bodySize = readU32(binary, func.loc.start.column);
+        const localsOffset = bodySize.nextOffset;
+        const offsets = [];
+        walk(func.body, (node) => {
+            if (!node || !node.loc)
+                return;
+            if (node.type === "IfInstruction") {
+                offsets.push(node.loc.start.column - localsOffset - 1);
+            }
+            else if (node.type === "Instr" && node.id === "br_if") {
+                offsets.push(node.loc.start.column - localsOffset);
+            }
+        });
+        return offsets.filter((offset) => offset >= 0);
+    });
+}
+function walk(node, visit) {
+    if (!node || typeof node !== "object")
+        return;
+    if (Array.isArray(node)) {
+        for (const item of node)
+            walk(item, visit);
+        return;
+    }
+    visit(node);
+    for (const [key, value] of Object.entries(node)) {
+        if (key === "loc")
+            continue;
+        walk(value, visit);
+    }
+}
+function encodeBranchHintPayload(functionHints) {
+    const bytes = [];
+    writeU32(bytes, functionHints.length);
+    for (const { functionIndex, hints } of functionHints) {
+        writeU32(bytes, functionIndex);
+        writeU32(bytes, hints.length);
+        for (const hint of hints) {
+            writeU32(bytes, hint.offset);
+            writeU32(bytes, 1);
+            writeU32(bytes, hint.value);
+        }
+    }
+    return Uint8Array.from(bytes);
+}
+function encodeCustomSection(name, payload) {
+    const nameBytes = new TextEncoder().encode(name);
+    const contents = [];
+    writeU32(contents, nameBytes.length);
+    contents.push(...nameBytes);
+    contents.push(...payload);
+    const section = [];
+    section.push(0);
+    writeU32(section, contents.length);
+    section.push(...contents);
+    return Uint8Array.from(section);
+}
+function insertCustomSectionBeforeCode(binary, customSection, sectionName) {
+    const sections = parseSections(binary);
+    const filtered = sections.filter((section) => {
+        if (section.id !== 0)
+            return true;
+        return readCustomSectionName(binary, section) !== sectionName;
+    });
+    const codeSection = filtered.find((section) => section.id === 10);
+    if (!codeSection) {
+        throw new Error("WebAssembly module has no code section");
+    }
+    const output = [];
+    output.push(...binary.slice(0, 8));
+    for (const section of filtered) {
+        if (section.start === codeSection.start) {
+            output.push(...customSection);
+        }
+        output.push(...binary.slice(section.start, section.end));
+    }
+    return Uint8Array.from(output);
+}
+function parseSections(binary) {
+    const sections = [];
+    let offset = 8;
+    while (offset < binary.length) {
+        const start = offset;
+        const id = binary[offset++];
+        const size = readU32(binary, offset);
+        offset = size.nextOffset;
+        const payloadOffset = offset;
+        const end = payloadOffset + size.value;
+        sections.push({
+            id,
+            start,
+            payloadOffset,
+            end,
+        });
+        offset = end;
+    }
+    return sections;
+}
+function readCustomSectionName(binary, section) {
+    const name = readName(binary, section.payloadOffset);
+    return name.value;
+}
+function readName(binary, offset) {
+    const length = readU32(binary, offset);
+    const start = length.nextOffset;
+    const end = start + length.value;
+    return {
+        value: new TextDecoder().decode(binary.slice(start, end)),
+        nextOffset: end,
+    };
+}
+function readU32(binary, offset) {
+    let value = 0;
+    let shift = 0;
+    let current = offset;
+    while (true) {
+        const byte = binary[current++];
+        value |= (byte & 0x7f) << shift;
+        if ((byte & 0x80) === 0) {
+            return { value: value >>> 0, nextOffset: current };
+        }
+        shift += 7;
+    }
+}
+function skipTableType(binary, offset) {
+    offset = skipReferenceType(binary, offset);
+    return skipLimits(binary, offset);
+}
+function skipMemoryType(binary, offset) {
+    return skipLimits(binary, offset);
+}
+function skipGlobalType(binary, offset) {
+    offset = skipValueType(binary, offset);
+    return offset + 1;
+}
+function skipTagType(binary, offset) {
+    offset += 1;
+    return readU32(binary, offset).nextOffset;
+}
+function skipLimits(binary, offset) {
+    const flags = binary[offset++];
+    offset = readU32(binary, offset).nextOffset;
+    if (flags & 0x01) {
+        offset = readU32(binary, offset).nextOffset;
+    }
+    if (flags & 0x04) {
+        offset = readU32(binary, offset).nextOffset;
+    }
+    return offset;
+}
+function skipReferenceType(binary, offset) {
+    const type = binary[offset];
+    if (type === 0x63 || type === 0x64) {
+        return offset + 1;
+    }
+    return skipHeapType(binary, offset + 1);
+}
+function skipValueType(binary, offset) {
+    const type = binary[offset];
+    if (type === 0x7f ||
+        type === 0x7e ||
+        type === 0x7d ||
+        type === 0x7c ||
+        type === 0x7b ||
+        type === 0x70 ||
+        type === 0x6f) {
+        return offset + 1;
+    }
+    return skipReferenceType(binary, offset);
+}
+function skipHeapType(binary, offset) {
+    const type = binary[offset];
+    if (type <= 0x7f)
+        return offset + 1;
+    return readU32(binary, offset).nextOffset;
+}
+function writeU32(target, value) {
+    let current = value >>> 0;
+    do {
+        let byte = current & 0x7f;
+        current >>>= 7;
+        if (current !== 0)
+            byte |= 0x80;
+        target.push(byte);
+    } while (current !== 0);
+}
+//# sourceMappingURL=index.js.map

package/dist/branch-hinting/transform/index.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"index.js","sourceRoot":"","sources":["../../../branch-hinting/transform/index.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,MAAM,EAAE,MAAM,4BAA4B,CAAC;AACpD,OAAO,EACL,cAAc,EAEd,oBAAoB,EAIpB,uBAAuB,EAEvB,eAAe,GAChB,MAAM,uCAAuC,CAAC;AAO/C,OAAO,EAAE,SAAS,EAAE,MAAM,kCAAkC,CAAC;AAE7D,MAAM,mBAAmB,GAAG,2BAA2B,CAAC;AA8CxD,MAAM,CAAC,OAAO,OAAO,mBAAoB,SAAQ,SAAS;IAChD,0BAA0B,GAAG,IAAI,GAAG,EAGzC,CAAC;IACI,2BAA2B,GAAG,IAAI,GAAG,EAAwB,CAAC;IAEtE,UAAU,CAAC,MAA6B;QACtC,KAAK,MAAM,MAAM,IAAI,MAAM,CAAC,OAAO,EAAE,CAAC;YACpC,IAAI,MAAM,CAAC,UAAU,+BAAuB;gBAAE,SAAS;YACvD,IAAI,MAAM,CAAC,YAAY,CAAC,UAAU,CAAC,OAAO,CAAC;gBAAE,SAAS;YACtD,KAAK,MAAM,SAAS,IAAI,MAAM,CAAC,UAAU,EAAE,CAAC;gBAC1C,IAAI,CAAC,SAAS,CAAC,SAAS,EAAE,IAAI,CAAC,CAAC;YAClC,CAAC;QACH,CAAC;IACH,CAAC;IAED,eAAe,CAAC,OAAgB;QAC9B,KAAK,MAAM,CAAC,WAAW,EAAE,KAAK,CAAC,IAAI,IAAI,CAAC,0BAA0B,EAAE,CAAC;YACnE,MAAM,OAAO,GAAG,OAAO,CAAC,qBAAqB,CAAC,GAAG,CAAC,WAAW,CAAC,CAAC;YAC/D,IAAI,CAAC,OAAO,IAAI,CAAC,OAAO,CAAC,YAAY,IAAI,KAAK,CAAC,MAAM,KAAK,CAAC;gBAAE,SAAS;YACtE,IAAI,CAAC,2BAA2B,CAAC,GAAG,CAAC,OAAO,CAAC,YAAY,EAAE,KAAK,CAAC,CAAC;QACpE,CAAC;IACH,CAAC;IAED,YAAY,CAAC,MAAuB;QAClC,IAAI,IAAI,CAAC,2BAA2B,CAAC,IAAI,KAAK,CAAC;YAAE,OAAO;QAExD,MAAM,kBAAkB,GAAG,MAAM,CAAC,UAAU,CAAC,IAAI,CAAC,MAAM,CAAkB,CAAC;QAC3E,MAAM,CAAC,UAAU,GAAG,CAAC,CAAC,GAAG,IAAe,EAAE,EAAE;YAC1C,MAAM,MAAM,GAAG,kBAAkB,CAAC,GAAG,IAAI,CAAC,CAAC;YAC3C,MAAM,CAAC,MAAM,GAAG,iBAAiB,CAC/B,MAAM,CAAC,MAAM,EACb,MAAM,EACN,IAAI,CAAC,2BAA2B,CACjC,CAAC;YACF,OAAO,MAAM,CAAC;QAChB,CAAC,CAA6B,CAAC;IACjC,CAAC;IAEO,SAAS,CACf,IAAsC,EACtC,eAA+D;QAE/D,IAAI,CAAC,IAAI;YAAE,OAAO;QAClB,IAAI,KAAK,CAAC,OAAO,CAAC,IAAI,CAAC,EAAE,CAAC;YACxB,KAAK,MAAM,IAAI,IAAI,IAAI;gBAAE,IAAI,CAAC,SAAS,CAAC,IAAI,EAAE,eAAe,CAAC,CAAC;YAC/D,OAAO;QACT,CAAC;QAED,QAAQ,IAAI,CAAC,IAAI,EAAE,CAAC;YAClB;gBACE,IAAI,CAAC,iBAAiB,CAAC,IAA2B,EAAE,IAA2B,CAAC,CAAC;gBACjF,OAAO;YACT;gBACE,IAAI,CAAC,iBAAiB,CAAC,IAAyB,EAAE,IAAyB,CAAC,CAAC;gBAC7E,OAAO;YACT;gBACE,IAAI,CAAC,yBAAyB,CAC5B,IAAuC,EACvC,eAAe,EACf,CAAE,IAAY,CAAC,MAAM,EAAG,IAAY,CAAC,OAAO,CAAC,CAC9C,CAAC;gBACF,OAAO;YACT;gBACE,IAAI,CAAC,yBAAyB,CAC5B,IAAuC,EACvC,eAAe,EACf,CAAE,IAAY,CAAC,IAAI,CAAC,CACrB,CAAC;gBACF,OAAO;YACT;gBACE,IAAI,CAAC,yBAAyB,CAC5B,IAAuC,EACvC,eAAe,EACf,CAAE,IAAY,CAAC,IAAI,CAAC,CACrB,CAAC;gBACF,OAAO;YACT;gBACE,IAAI,CAAC,yBAAyB,CAC5B,IAAuC,EACvC,eAAe,EACf,CAAE,IAAY,CAAC,WAAW,EAAG,IAAY,CAAC,WAAW,EAAG,IAAY,CAAC,IAAI,CAAC,CAC3E,CAAC;gBACF,OAAO;YACT;gBACE,IAAI,CAAC,SAAS,CAAE,IAAY,CAAC,UAAU,EAAE,eAAe,CAAC,CAAC;gBAC1D,OAAO;YACT;gBACE,IAAI,CAAC,SAAS,CAAE,IAAY,CAAC,SAAS,EAAE,eAAe,CAAC,CAAC;gBACzD,IAAI,CAAC,SAAS,CAAE,IAAY,CAAC,KAAK,EAAE,eAAe,CAAC,CAAC;gBACrD,OAAO;YACT;gBACE,IAAI,CAAC,SAAS,CAAE,IAAY,CAAC,KAAK,EAAE,eAAe,CAAC,CAAC;gBACrD,IAAI,CAAC,SAAS,CAAE,IAAY,CAAC,UAAU,EAAE,eAAe,CAAC,CAAC;gBAC1D,OAAO;YACT;gBACE,IAAI,CAAC,SAAS,CAAE,IAAY,CAAC,UAAU,EAAE,eAAe,CAAC,CAAC;gBAC1D,OAAO;YACT;gBACE,IAAI,CAAC,SAAS,CAAE,IAAY,CAAC,KAAK,EAAE,eAAe,CAAC,CAAC;gBACrD,OAAO;YACT;gBACE,IAAI,CAAC,SAAS,CAAE,IAAY,CAAC,YAAY,EAAE,eAAe,CAAC,CAAC;gBAC5D,OAAO;YACT;gBACE,IAAI,CAAC,SAAS,CAAE,IAAY,CAAC,WAAW,EAAE,eAAe,CAAC,CAAC;gBAC3D,OAAO;YACT;gBACE,IAAI,CAAC,SAAS,CAAE,IAAY,CAAC,UAAU,EAAE,eAAe,CAAC,CAAC;gBAC1D,IAAI,CAAC,SAAS,CAAE,IAAY,CAAC,IAAI,EAAE,eAAe,CAAC,CAAC;gBACpD,OAAO;YACT;gBACE,IAAI,CAAC,SAAS,CAAE,IAAY,CAAC,IAAI,EAAE,eAAe,CAAC,CAAC;gBACpD,IAAI,CAAC,SAAS,CAAE,IAAY,CAAC,KAAK,EAAE,eAAe,CAAC,CAAC;gBACrD,OAAO;YACT,mCAA0B;YAC1B,oCAA2B;YAC3B,qCAA4B;YAC5B;gBACE,IAAI,CAAC,SAAS,CAAE,IAAY,CAAC,OAAO,IAAK,IAAY,CAAC,UAAU,EAAE,eAAe,CAAC,CAAC;gBACnF,OAAO;YACT;gBACE,IAAI,CAAC,SAAS,CAAE,IAAY,CAAC,SAAS,EAAE,eAAe,CAAC,CAAC;gBACzD,IAAI,CAAC,SAAS,CAAE,IAAY,CAAC,MAAM,EAAE,eAAe,CAAC,CAAC;gBACtD,IAAI,CAAC,SAAS,CAAE,IAAY,CAAC,MAAM,EAAE,eAAe,CAAC,CAAC;gBACtD,OAAO;YACT;gBACE,IAAI,CAAC,SAAS,CAAE,IAAY,CAAC,QAAQ,EAAE,eAAe,CAAC,CAAC;gBACxD,IAAI,CAAC,SAAS,CAAE,IAAY,CAAC,QAAQ,EAAE,eAAe,CAAC,CAAC;gBACxD,IAAI,CAAC,SAAS,CAAE,IAAY,CAAC,IAAI,EAAE,eAAe,CAAC,CAAC;gBACpD,OAAO;YACT;gBACE,IAAI,CAAC,SAAS,CAAE,IAAY,CAAC,cAAc,EAAE,eAAe,CAAC,CAAC;gBAC9D,IAAI,CAAC,SAAS,CAAE,IAAY,CAAC,eAAe,EAAE,eAAe,CAAC,CAAC;gBAC/D,IAAI,CAAC,SAAS,CAAE,IAAY,CAAC,iBAAiB,EAAE,eAAe,CAAC,CAAC;gBACjE,OAAO;YACT;gBACE,IAAI,CAAC,SAAS,CAAE,IAAY,CAAC,KAAK,EAAE,eAAe,CAAC,CAAC;gBACrD,OAAO;YACT;gBACE,IAAI,CAAC,SAAS,CAAE,IAAY,CAAC,UAAU,EAAE,eAAe,CAAC,CAAC;gBAC1D,OAAO;YACT;gBACE,OAAO;QACX,CAAC;IACH,CAAC;IAEO,iBAAiB,CACvB,IAA6C,EAC7C,WAAoD;QAEpD,IAAI,CAAC,IAAI,CAAC,IAAI;YAAE,OAAO;QACvB,IAAI,CAAC,SAAS,CAAC,IAAI,CAAC,IAAI,EAAE,WAAW,CAAC,CAAC;IACzC,CAAC;IAEO,yBAAyB,CAC/B,IAA0B,EAC1B,eAA+D,EAC/D,aAAsD;QAEtD,IAAI,CAAC,eAAe;YAAE,OAAO;QAE7B,MAAM,IAAI,GAAG,UAAU,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC;QACxC,IAAI,IAAI,EAAE,CAAC;YACT,IAAI,CAAC,SAAS,GAAG,IAAI,CAAC,UAAU,CAAC;YACjC,IAAI,KAAK,GAAG,IAAI,CAAC,0BAA0B,CAAC,GAAG,CAAC,eAAe,CAAC,CAAC;YACjE,IAAI,CAAC,KAAK,EAAE,CAAC;gBACX,KAAK,GAAG,EAAE,CAAC;gBACX,IAAI,CAAC,0BAA0B,CAAC,GAAG,CAAC,eAAe,EAAE,KAAK,CAAC,CAAC;YAC9D,CAAC;YACD,KAAK,CAAC,IAAI,CAAC,EAAE,KAAK,EAAE,IAAI,CAAC,KAAK,EAAE,CAAC,CAAC;QACpC,CAAC;aAAM,CAAC;YACN,IAAI,CAAC,SAAS,CAAC,IAAI,CAAC,SAAS,EAAE,eAAe,CAAC,CAAC;QAClD,CAAC;QAED,KAAK,MAAM,KAAK,IAAI,aAAa,EAAE,CAAC;YAClC,IAAI,CAAC,SAAS,CAAC,KAAK,EAAE,eAAe,CAAC,CAAC;QACzC,CAAC;IACH,CAAC;CACF;AAED,SAAS,UAAU,CAAC,UAAsB;IACxC,IAAI,OAAO,GAAe,UAAU,CAAC;IACrC,OAAO,OAAO,YAAY,uBAAuB,EAAE,CAAC;QAClD,OAAO,GAAG,OAAO,CAAC,UAAU,CAAC;IAC/B,CAAC;IACD,IAAI,CAAC,CAAC,OAAO,YAAY,cAAc,CAAC;QAAE,OAAO,IAAI,CAAC;IAEtD,MAAM,MAAM,GAAG,OAAO,CAAC,UAAU,CAAC;IAClC,IAAI,CAAC,CAAC,MAAM,YAAY,oBAAoB,CAAC;QAAE,OAAO,IAAI,CAAC;IAC3D,IAAI,OAAO,CAAC,IAAI,CAAC,MAAM,KAAK,CAAC;QAAE,OAAO,IAAI,CAAC;IAE3C,IAAI,MAAM,CAAC,IAAI,KAAK,QAAQ,EAAE,CAAC;QAC7B,OAAO,EAAE,KAAK,EAAE,CAAC,EAAE,UAAU,EAAE,OAAO,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,CAAC;IACnD,CAAC;IACD,IAAI,MAAM,CAAC,IAAI,KAAK,UAAU,EAAE,CAAC;QAC/B,OAAO,EAAE,KAAK,EAAE,CAAC,EAAE,UAAU,EAAE,OAAO,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,CAAC;IACnD,CAAC;IACD,OAAO,IAAI,CAAC;AACd,CAAC;AAED,SAAS,iBAAiB,CACxB,MAAkB,EAClB,MAAuB,EACvB,2BAAsD;IAEtD,MAAM,qBAAqB,GAAG,sBAAsB,CAAC,MAAM,CAAC,CAAC;IAC7D,MAAM,aAAa,GAAa,EAAE,CAAC;IACnC,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,MAAM,CAAC,eAAe,EAAE,EAAE,CAAC,GAAG,CAAC,EAAE,CAAC,EAAE,EAAE,CAAC;QACzD,MAAM,IAAI,GAAG,eAAe,CAAC,MAAM,CAAC,kBAAkB,CAAC,CAAC,CAAC,CAAC,CAAC;QAC3D,aAAa,CAAC,IAAI,CAAC,IAAI,IAAI,EAAE,CAAC,CAAC;IACjC,CAAC;IAED,MAAM,kBAAkB,GAAG,yBAAyB,CAAC,MAAM,CAAC,CAAC;IAC7D,MAAM,aAAa,GAAwB,EAAE,CAAC;IAE9C,KAAK,IAAI,YAAY,GAAG,CAAC,EAAE,YAAY,GAAG,kBAAkB,CAAC,MAAM,EAAE,YAAY,EAAE,EAAE,CAAC;QACpF,MAAM,YAAY,GAAG,aAAa,CAAC,qBAAqB,GAAG,YAAY,CAAC,CAAC;QACzE,MAAM,MAAM,GAAG,2BAA2B,CAAC,GAAG,CAAC,YAAY,CAAC,CAAC;QAC7D,IAAI,CAAC,MAAM,IAAI,MAAM,CAAC,MAAM,KAAK,CAAC;YAAE,SAAS;QAE7C,MAAM,SAAS,GAAG,kBAAkB,CAAC,YAAY,CAAC,IAAI,EAAE,CAAC;QACzD,MAAM,OAAO,GAAmB,EAAE,CAAC;QACnC,MAAM,KAAK,GAAG,IAAI,CAAC,GAAG,CAAC,MAAM,CAAC,MAAM,EAAE,SAAS,CAAC,MAAM,CAAC,CAAC;QACxD,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,KAAK,EAAE,CAAC,EAAE,EAAE,CAAC;YAC/B,OAAO,CAAC,IAAI,CAAC;gBACX,MAAM,EAAE,SAAS,CAAC,CAAC,CAAC;gBACpB,KAAK,EAAE,MAAM,CAAC,CAAC,CAAC,CAAC,KAAK;aACvB,CAAC,CAAC;QACL,CAAC;QACD,IAAI,OAAO,CAAC,MAAM,KAAK,CAAC;YAAE,SAAS;QAEnC,aAAa,CAAC,IAAI,CAAC;YACjB,aAAa,EAAE,qBAAqB,GAAG,YAAY;YACnD,KAAK,EAAE,OAAO;SACf,CAAC,CAAC;IACL,CAAC;IAED,IAAI,aAAa,CAAC,MAAM,KAAK,CAAC;QAAE,OAAO,MAAM,CAAC;IAE9C,MAAM,OAAO,GAAG,uBAAuB,CAAC,aAAa,CAAC,CAAC;IACvD,MAAM,aAAa,GAAG,mBAAmB,CAAC,mBAAmB,EAAE,OAAO,CAAC,CAAC;IACxE,OAAO,6BAA6B,CAAC,MAAM,EAAE,aAAa,EAAE,mBAAmB,CAAC,CAAC;AACnF,CAAC;AAED,SAAS,sBAAsB,CAAC,MAAkB;IAChD,MAAM,QAAQ,GAAG,aAAa,CAAC,MAAM,CAAC,CAAC;IACvC,MAAM,aAAa,GAAG,QAAQ,CAAC,IAAI,CAAC,CAAC,OAAO,EAAE,EAAE,CAAC,OAAO,CAAC,EAAE,KAAK,CAAC,CAAC,CAAC;IACnE,IAAI,CAAC,aAAa;QAAE,OAAO,CAAC,CAAC;IAE7B,IAAI,MAAM,GAAG,aAAa,CAAC,aAAa,CAAC;IACzC,MAAM,GAAG,GAAG,aAAa,CAAC,GAAG,CAAC;IAC9B,MAAM,SAAS,GAAG,OAAO,CAAC,MAAM,EAAE,MAAM,CAAC,CAAC;IAC1C,MAAM,GAAG,SAAS,CAAC,UAAU,CAAC;IAC9B,IAAI,eAAe,GAAG,CAAC,CAAC;IAExB,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,SAAS,CAAC,KAAK,IAAI,MAAM,GAAG,GAAG,EAAE,CAAC,EAAE,EAAE,CAAC;QACzD,MAAM,UAAU,GAAG,QAAQ,CAAC,MAAM,EAAE,MAAM,CAAC,CAAC;QAC5C,MAAM,GAAG,UAAU,CAAC,UAAU,CAAC;QAC/B,MAAM,SAAS,GAAG,QAAQ,CAAC,MAAM,EAAE,MAAM,CAAC,CAAC;QAC3C,MAAM,GAAG,SAAS,CAAC,UAAU,CAAC;QAC9B,MAAM,IAAI,GAAG,MAAM,CAAC,MAAM,EAAE,CAAW,CAAC;QAExC,QAAQ,IAAI,EAAE,CAAC;YACb,KAAK,CAAC;gBACJ,eAAe,EAAE,CAAC;gBAClB,MAAM,GAAG,OAAO,CAAC,MAAM,EAAE,MAAM,CAAC,CAAC,UAAU,CAAC;gBAC5C,MAAM;YACR,KAAK,CAAC;gBACJ,MAAM,GAAG,aAAa,CAAC,MAAM,EAAE,MAAM,CAAC,CAAC;gBACvC,MAAM;YACR,KAAK,CAAC;gBACJ,MAAM,GAAG,cAAc,CAAC,MAAM,EAAE,MAAM,CAAC,CAAC;gBACxC,MAAM;YACR,KAAK,CAAC;gBACJ,MAAM,GAAG,cAAc,CAAC,MAAM,EAAE,MAAM,CAAC,CAAC;gBACxC,MAAM;YACR,KAAK,CAAC;gBACJ,MAAM,GAAG,WAAW,CAAC,MAAM,EAAE,MAAM,CAAC,CAAC;gBACrC,MAAM;YACR;gBACE,MAAM,IAAI,KAAK,CAAC,2BAA2B,IAAI,EAAE,CAAC,CAAC;QACvD,CAAC;IACH,CAAC;IAED,OAAO,eAAe,CAAC;AACzB,CAAC;AAED,SAAS,yBAAyB,CAAC,MAAkB;IACnD,MAAM,GAAG,GAAG,MAAM,CAAC,MAAM,EAAE;QACzB,IAAI,EAAE,KAAK;QACX,iBAAiB,EAAE,KAAK;QACxB,iBAAiB,EAAE,IAAI;KACxB,CAAC,CAAC;IAEH,MAAM,SAAS,GAAU,EAAE,CAAC;IAC5B,IAAI,CAAC,GAAG,EAAE,CAAC,IAAS,EAAE,EAAE;QACtB,IAAI,IAAI,EAAE,IAAI,KAAK,MAAM,EAAE,CAAC;YAC1B,SAAS,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;QACvB,CAAC;IACH,CAAC,CAAC,CAAC;IAEH,OAAO,SAAS,CAAC,GAAG,CAAC,CAAC,IAAI,EAAE,EAAE;QAC5B,MAAM,QAAQ,GAAG,OAAO,CAAC,MAAM,EAAE,IAAI,CAAC,GAAG,CAAC,KAAK,CAAC,MAAgB,CAAC,CAAC;QAClE,MAAM,YAAY,GAAG,QAAQ,CAAC,UAAU,CAAC;QACzC,MAAM,OAAO,GAAa,EAAE,CAAC;QAE7B,IAAI,CAAC,IAAI,CAAC,IAAI,EAAE,CAAC,IAAS,EAAE,EAAE;YAC5B,IAAI,CAAC,IAAI,IAAI,CAAC,IAAI,CAAC,GAAG;gBAAE,OAAO;YAC/B,IAAI,IAAI,CAAC,IAAI,KAAK,eAAe,EAAE,CAAC;gBAClC,OAAO,CAAC,IAAI,CAAE,IAAI,CAAC,GAAG,CAAC,KAAK,CAAC,MAAiB,GAAG,YAAY,GAAG,CAAC,CAAC,CAAC;YACrE,CAAC;iBAAM,IAAI,IAAI,CAAC,IAAI,KAAK,OAAO,IAAI,IAAI,CAAC,EAAE,KAAK,OAAO,EAAE,CAAC;gBACxD,OAAO,CAAC,IAAI,CAAE,IAAI,CAAC,GAAG,CAAC,KAAK,CAAC,MAAiB,GAAG,YAAY,CAAC,CAAC;YACjE,CAAC;QACH,CAAC,CAAC,CAAC;QAEH,OAAO,OAAO,CAAC,MAAM,CAAC,CAAC,MAAM,EAAE,EAAE,CAAC,MAAM,IAAI,CAAC,CAAC,CAAC;IACjD,CAAC,CAAC,CAAC;AACL,CAAC;AAED,SAAS,IAAI,CAAC,IAAa,EAAE,KAA8B;IACzD,IAAI,CAAC,IAAI,IAAI,OAAO,IAAI,KAAK,QAAQ;QAAE,OAAO;IAC9C,IAAI,KAAK,CAAC,OAAO,CAAC,IAAI,CAAC,EAAE,CAAC;QACxB,KAAK,MAAM,IAAI,IAAI,IAAI;YAAE,IAAI,CAAC,IAAI,EAAE,KAAK,CAAC,CAAC;QAC3C,OAAO;IACT,CAAC;IAED,KAAK,CAAC,IAAI,CAAC,CAAC;IACZ,KAAK,MAAM,CAAC,GAAG,EAAE,KAAK,CAAC,IAAI,MAAM,CAAC,OAAO,CAAC,IAAI,CAAC,EAAE,CAAC;QAChD,IAAI,GAAG,KAAK,KAAK;YAAE,SAAS;QAC5B,IAAI,CAAC,KAAK,EAAE,KAAK,CAAC,CAAC;IACrB,CAAC;AACH,CAAC;AAED,SAAS,uBAAuB,CAAC,aAAkC;IACjE,MAAM,KAAK,GAAa,EAAE,CAAC;IAC3B,QAAQ,CAAC,KAAK,EAAE,aAAa,CAAC,MAAM,CAAC,CAAC;IAEtC,KAAK,MAAM,EAAE,aAAa,EAAE,KAAK,EAAE,IAAI,aAAa,EAAE,CAAC;QACrD,QAAQ,CAAC,KAAK,EAAE,aAAa,CAAC,CAAC;QAC/B,QAAQ,CAAC,KAAK,EAAE,KAAK,CAAC,MAAM,CAAC,CAAC;QAC9B,KAAK,MAAM,IAAI,IAAI,KAAK,EAAE,CAAC;YACzB,QAAQ,CAAC,KAAK,EAAE,IAAI,CAAC,MAAM,CAAC,CAAC;YAC7B,QAAQ,CAAC,KAAK,EAAE,CAAC,CAAC,CAAC;YACnB,QAAQ,CAAC,KAAK,EAAE,IAAI,CAAC,KAAK,CAAC,CAAC;QAC9B,CAAC;IACH,CAAC;IAED,OAAO,UAAU,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;AAChC,CAAC;AAED,SAAS,mBAAmB,CAAC,IAAY,EAAE,OAAmB;IAC5D,MAAM,SAAS,GAAG,IAAI,WAAW,EAAE,CAAC,MAAM,CAAC,IAAI,CAAC,CAAC;IACjD,MAAM,QAAQ,GAAa,EAAE,CAAC;IAC9B,QAAQ,CAAC,QAAQ,EAAE,SAAS,CAAC,MAAM,CAAC,CAAC;IACrC,QAAQ,CAAC,IAAI,CAAC,GAAG,SAAS,CAAC,CAAC;IAC5B,QAAQ,CAAC,IAAI,CAAC,GAAG,OAAO,CAAC,CAAC;IAE1B,MAAM,OAAO,GAAa,EAAE,CAAC;IAC7B,OAAO,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;IAChB,QAAQ,CAAC,OAAO,EAAE,QAAQ,CAAC,MAAM,CAAC,CAAC;IACnC,OAAO,CAAC,IAAI,CAAC,GAAG,QAAQ,CAAC,CAAC;IAC1B,OAAO,UAAU,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC;AAClC,CAAC;AAED,SAAS,6BAA6B,CACpC,MAAkB,EAClB,aAAyB,EACzB,WAAmB;IAEnB,MAAM,QAAQ,GAAG,aAAa,CAAC,MAAM,CAAC,CAAC;IACvC,MAAM,QAAQ,GAAG,QAAQ,CAAC,MAAM,CAAC,CAAC,OAAO,EAAE,EAAE;QAC3C,IAAI,OAAO,CAAC,EAAE,KAAK,CAAC;YAAE,OAAO,IAAI,CAAC;QAClC,OAAO,qBAAqB,CAAC,MAAM,EAAE,OAAO,CAAC,KAAK,WAAW,CAAC;IAChE,CAAC,CAAC,CAAC;IAEH,MAAM,WAAW,GAAG,QAAQ,CAAC,IAAI,CAAC,CAAC,OAAO,EAAE,EAAE,CAAC,OAAO,CAAC,EAAE,KAAK,EAAE,CAAC,CAAC;IAClE,IAAI,CAAC,WAAW,EAAE,CAAC;QACjB,MAAM,IAAI,KAAK,CAAC,wCAAwC,CAAC,CAAC;IAC5D,CAAC;IAED,MAAM,MAAM,GAAa,EAAE,CAAC;IAC5B,MAAM,CAAC,IAAI,CAAC,GAAG,MAAM,CAAC,KAAK,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,CAAC;IAEnC,KAAK,MAAM,OAAO,IAAI,QAAQ,EAAE,CAAC;QAC/B,IAAI,OAAO,CAAC,KAAK,KAAK,WAAW,CAAC,KAAK,EAAE,CAAC;YACxC,MAAM,CAAC,IAAI,CAAC,GAAG,aAAa,CAAC,CAAC;QAChC,CAAC;QACD,MAAM,CAAC,IAAI,CAAC,GAAG,MAAM,CAAC,KAAK,CAAC,OAAO,CAAC,KAAK,EAAE,OAAO,CAAC,GAAG,CAAC,CAAC,CAAC;IAC3D,CAAC;IAED,OAAO,UAAU,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC;AACjC,CAAC;AAED,SAAS,aAAa,CAAC,MAAkB;IACvC,MAAM,QAAQ,GAAkB,EAAE,CAAC;IACnC,IAAI,MAAM,GAAG,CAAC,CAAC;IAEf,OAAO,MAAM,GAAG,MAAM,CAAC,MAAM,EAAE,CAAC;QAC9B,MAAM,KAAK,GAAG,MAAM,CAAC;QACrB,MAAM,EAAE,GAAG,MAAM,CAAC,MAAM,EAAE,CAAW,CAAC;QACtC,MAAM,IAAI,GAAG,OAAO,CAAC,MAAM,EAAE,MAAM,CAAC,CAAC;QACrC,MAAM,GAAG,IAAI,CAAC,UAAU,CAAC;QACzB,MAAM,aAAa,GAAG,MAAM,CAAC;QAC7B,MAAM,GAAG,GAAG,aAAa,GAAG,IAAI,CAAC,KAAK,CAAC;QACvC,QAAQ,CAAC,IAAI,CAAC;YACZ,EAAE;YACF,KAAK;YACL,aAAa;YACb,GAAG;SACJ,CAAC,CAAC;QACH,MAAM,GAAG,GAAG,CAAC;IACf,CAAC;IAED,OAAO,QAAQ,CAAC;AAClB,CAAC;AAED,SAAS,qBAAqB,CAAC,MAAkB,EAAE,OAAoB;IACrE,MAAM,IAAI,GAAG,QAAQ,CAAC,MAAM,EAAE,OAAO,CAAC,aAAa,CAAC,CAAC;IACrD,OAAO,IAAI,CAAC,KAAK,CAAC;AACpB,CAAC;AAED,SAAS,QAAQ,CAAC,MAAkB,EAAE,MAAc;IAClD,MAAM,MAAM,GAAG,OAAO,CAAC,MAAM,EAAE,MAAM,CAAC,CAAC;IACvC,MAAM,KAAK,GAAG,MAAM,CAAC,UAAU,CAAC;IAChC,MAAM,GAAG,GAAG,KAAK,GAAG,MAAM,CAAC,KAAK,CAAC;IACjC,OAAO;QACL,KAAK,EAAE,IAAI,WAAW,EAAE,CAAC,MAAM,CAAC,MAAM,CAAC,KAAK,CAAC,KAAK,EAAE,GAAG,CAAC,CAAC;QACzD,UAAU,EAAE,GAAG;KAChB,CAAC;AACJ,CAAC;AAED,SAAS,OAAO,CAAC,MAAkB,EAAE,MAAc;IACjD,IAAI,KAAK,GAAG,CAAC,CAAC;IACd,IAAI,KAAK,GAAG,CAAC,CAAC;IACd,IAAI,OAAO,GAAG,MAAM,CAAC;IAErB,OAAO,IAAI,EAAE,CAAC;QACZ,MAAM,IAAI,GAAG,MAAM,CAAC,OAAO,EAAE,CAAW,CAAC;QACzC,KAAK,IAAI,CAAC,IAAI,GAAG,IAAI,CAAC,IAAI,KAAK,CAAC;QAChC,IAAI,CAAC,IAAI,GAAG,IAAI,CAAC,KAAK,CAAC,EAAE,CAAC;YACxB,OAAO,EAAE,KAAK,EAAE,KAAK,KAAK,CAAC,EAAE,UAAU,EAAE,OAAO,EAAE,CAAC;QACrD,CAAC;QACD,KAAK,IAAI,CAAC,CAAC;IACb,CAAC;AACH,CAAC;AAED,SAAS,aAAa,CAAC,MAAkB,EAAE,MAAc;IACvD,MAAM,GAAG,iBAAiB,CAAC,MAAM,EAAE,MAAM,CAAC,CAAC;IAC3C,OAAO,UAAU,CAAC,MAAM,EAAE,MAAM,CAAC,CAAC;AACpC,CAAC;AAED,SAAS,cAAc,CAAC,MAAkB,EAAE,MAAc;IACxD,OAAO,UAAU,CAAC,MAAM,EAAE,MAAM,CAAC,CAAC;AACpC,CAAC;AAED,SAAS,cAAc,CAAC,MAAkB,EAAE,MAAc;IACxD,MAAM,GAAG,aAAa,CAAC,MAAM,EAAE,MAAM,CAAC,CAAC;IACvC,OAAO,MAAM,GAAG,CAAC,CAAC;AACpB,CAAC;AAED,SAAS,WAAW,CAAC,MAAkB,EAAE,MAAc;IACrD,MAAM,IAAI,CAAC,CAAC;IACZ,OAAO,OAAO,CAAC,MAAM,EAAE,MAAM,CAAC,CAAC,UAAU,CAAC;AAC5C,CAAC;AAED,SAAS,UAAU,CAAC,MAAkB,EAAE,MAAc;IACpD,MAAM,KAAK,GAAG,MAAM,CAAC,MAAM,EAAE,CAAW,CAAC;IACzC,MAAM,GAAG,OAAO,CAAC,MAAM,EAAE,MAAM,CAAC,CAAC,UAAU,CAAC;IAC5C,IAAI,KAAK,GAAG,IAAI,EAAE,CAAC;QACjB,MAAM,GAAG,OAAO,CAAC,MAAM,EAAE,MAAM,CAAC,CAAC,UAAU,CAAC;IAC9C,CAAC;IACD,IAAI,KAAK,GAAG,IAAI,EAAE,CAAC;QACjB,MAAM,GAAG,OAAO,CAAC,MAAM,EAAE,MAAM,CAAC,CAAC,UAAU,CAAC;IAC9C,CAAC;IACD,OAAO,MAAM,CAAC;AAChB,CAAC;AAED,SAAS,iBAAiB,CAAC,MAAkB,EAAE,MAAc;IAC3D,MAAM,IAAI,GAAG,MAAM,CAAC,MAAM,CAAW,CAAC;IACtC,IAAI,IAAI,KAAK,IAAI,IAAI,IAAI,KAAK,IAAI,EAAE,CAAC;QACnC,OAAO,MAAM,GAAG,CAAC,CAAC;IACpB,CAAC;IACD,OAAO,YAAY,CAAC,MAAM,EAAE,MAAM,GAAG,CAAC,CAAC,CAAC;AAC1C,CAAC;AAED,SAAS,aAAa,CAAC,MAAkB,EAAE,MAAc;IACvD,MAAM,IAAI,GAAG,MAAM,CAAC,MAAM,CAAW,CAAC;IACtC,IACE,IAAI,KAAK,IAAI;QACb,IAAI,KAAK,IAAI;QACb,IAAI,KAAK,IAAI;QACb,IAAI,KAAK,IAAI;QACb,IAAI,KAAK,IAAI;QACb,IAAI,KAAK,IAAI;QACb,IAAI,KAAK,IAAI,EACb,CAAC;QACD,OAAO,MAAM,GAAG,CAAC,CAAC;IACpB,CAAC;IACD,OAAO,iBAAiB,CAAC,MAAM,EAAE,MAAM,CAAC,CAAC;AAC3C,CAAC;AAED,SAAS,YAAY,CAAC,MAAkB,EAAE,MAAc;IACtD,MAAM,IAAI,GAAG,MAAM,CAAC,MAAM,CAAW,CAAC;IACtC,IAAI,IAAI,IAAI,IAAI;QAAE,OAAO,MAAM,GAAG,CAAC,CAAC;IACpC,OAAO,OAAO,CAAC,MAAM,EAAE,MAAM,CAAC,CAAC,UAAU,CAAC;AAC5C,CAAC;AAED,SAAS,QAAQ,CAAC,MAAgB,EAAE,KAAa;IAC/C,IAAI,OAAO,GAAG,KAAK,KAAK,CAAC,CAAC;IAC1B,GAAG,CAAC;QACF,IAAI,IAAI,GAAG,OAAO,GAAG,IAAI,CAAC;QAC1B,OAAO,MAAM,CAAC,CAAC;QACf,IAAI,OAAO,KAAK,CAAC;YAAE,IAAI,IAAI,IAAI,CAAC;QAChC,MAAM,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;IACpB,CAAC,QAAQ,OAAO,KAAK,CAAC,EAAE;AAC1B,CAAC"}

package/dist/transform/index.d.ts

@@ -0,0 +1,10 @@
+import type { Parser, Program } from "assemblyscript/dist/assemblyscript.js";
+import type binaryen from "assemblyscript/lib/binaryen.js";
+import { Transform } from "assemblyscript/dist/transform.js";
+export default class LabsTransform extends Transform {
+    private readonly transforms;
+    constructor();
+    afterParse(parser: Parser): Promise<void>;
+    afterInitialize(program: Program): Promise<void>;
+    afterCompile(module: binaryen.Module): Promise<void>;
+}

package/dist/transform/index.js

@@ -0,0 +1,25 @@
+import { Transform } from "assemblyscript/dist/transform.js";
+import BranchHintingTransform from "../branch-hinting/transform/index.js";
+export default class LabsTransform extends Transform {
+    transforms;
+    constructor() {
+        super();
+        this.transforms = [new BranchHintingTransform()];
+    }
+    async afterParse(parser) {
+        for (const transform of this.transforms) {
+            await transform.afterParse?.(parser);
+        }
+    }
+    async afterInitialize(program) {
+        for (const transform of this.transforms) {
+            await transform.afterInitialize?.(program);
+        }
+    }
+    async afterCompile(module) {
+        for (const transform of this.transforms) {
+            await transform.afterCompile?.(module);
+        }
+    }
+}
+//# sourceMappingURL=index.js.map

package/dist/transform/index.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"index.js","sourceRoot":"","sources":["../../transform/index.ts"],"names":[],"mappings":"AAEA,OAAO,EAAE,SAAS,EAAE,MAAM,kCAAkC,CAAC;AAC7D,OAAO,sBAAsB,MAAM,sCAAsC,CAAC;AAQ1E,MAAM,CAAC,OAAO,OAAO,aAAc,SAAQ,SAAS;IACjC,UAAU,CAAkB;IAE7C;QACE,KAAK,EAAE,CAAC;QACR,IAAI,CAAC,UAAU,GAAG,CAAC,IAAI,sBAAsB,EAAE,CAAC,CAAC;IACnD,CAAC;IAED,KAAK,CAAC,UAAU,CAAC,MAAc;QAC7B,KAAK,MAAM,SAAS,IAAI,IAAI,CAAC,UAAU,EAAE,CAAC;YACxC,MAAM,SAAS,CAAC,UAAU,EAAE,CAAC,MAAM,CAAC,CAAC;QACvC,CAAC;IACH,CAAC;IAED,KAAK,CAAC,eAAe,CAAC,OAAgB;QACpC,KAAK,MAAM,SAAS,IAAI,IAAI,CAAC,UAAU,EAAE,CAAC;YACxC,MAAM,SAAS,CAAC,eAAe,EAAE,CAAC,OAAO,CAAC,CAAC;QAC7C,CAAC;IACH,CAAC;IAED,KAAK,CAAC,YAAY,CAAC,MAAuB;QACxC,KAAK,MAAM,SAAS,IAAI,IAAI,CAAC,UAAU,EAAE,CAAC;YACxC,MAAM,SAAS,CAAC,YAAY,EAAE,CAAC,MAAM,CAAC,CAAC;QACzC,CAAC;IACH,CAAC;CACF"}

package/index.ts

@@ -0,0 +1 @@
+export { likely, unlikely } from "./branch-hinting/index";

package/package.json

@@ -0,0 +1,55 @@
+{
+  "name": "as-labs",
+  "version": "0.1.0",
+  "description": "Experimental AssemblyScript APIs and transforms",
+  "type": "module",
+  "main": "./dist/transform/index.js",
+  "types": "./index.ts",
+  "homepage": "https://github.com/JairusSW/as-extensions",
+  "bugs": {
+    "url": "https://github.com/JairusSW/as-extensions/issues"
+  },
+  "funding": "https://github.com/sponsors/JairusSW",
+  "keywords": [
+    "assemblyscript",
+    "webassembly",
+    "wasm",
+    "transform",
+    "experimental",
+    "branch-hinting"
+  ],
+  "files": [
+    "branch-hinting",
+    "dist",
+    "transform",
+    "index.ts",
+    "README.md",
+    "CHANGELOG.md",
+    "LICENSE"
+  ],
+  "scripts": {
+    "build": "tsc -p tsconfig.json",
+    "test:unit": "vitest run test/branch-hints.test.js",
+    "test:as": "ast test",
+    "test": "npm run build && npm run test:unit && npm run test:as",
+    "release:check": "npm run test && npm pack --dry-run",
+    "prepublishOnly": "npm run test"
+  },
+  "repository": {
+    "type": "git",
+    "url": "git+https://github.com/JairusSW/as-extensions.git"
+  },
+  "license": "MIT",
+  "dependencies": {
+    "@webassemblyjs/ast": "^1.14.1",
+    "@webassemblyjs/wasm-parser": "^1.14.1",
+    "assemblyscript": "^0.28.12"
+  },
+  "devDependencies": {
+    "@assemblyscript/wasi-shim": "^0.1.0",
+    "as-test": "^1.0.7",
+    "typescript": "^6.0.2",
+    "vitest": "^4.1.2",
+    "wabt": "^1.0.39"
+  }
+}

package/transform/index.ts

@@ -0,0 +1,37 @@
+import type { Parser, Program } from "assemblyscript/dist/assemblyscript.js";
+import type binaryen from "assemblyscript/lib/binaryen.js";
+import { Transform } from "assemblyscript/dist/transform.js";
+import BranchHintingTransform from "../branch-hinting/transform/index.js";
+
+type TransformHook = {
+  afterParse?(parser: Parser): void | Promise<void>;
+  afterInitialize?(program: Program): void | Promise<void>;
+  afterCompile?(module: binaryen.Module): void | Promise<void>;
+};
+
+export default class LabsTransform extends Transform {
+  private readonly transforms: TransformHook[];
+
+  constructor() {
+    super();
+    this.transforms = [new BranchHintingTransform()];
+  }
+
+  async afterParse(parser: Parser): Promise<void> {
+    for (const transform of this.transforms) {
+      await transform.afterParse?.(parser);
+    }
+  }
+
+  async afterInitialize(program: Program): Promise<void> {
+    for (const transform of this.transforms) {
+      await transform.afterInitialize?.(program);
+    }
+  }
+
+  async afterCompile(module: binaryen.Module): Promise<void> {
+    for (const transform of this.transforms) {
+      await transform.afterCompile?.(module);
+    }
+  }
+}

package/transform/types.d.ts

@@ -0,0 +1,3 @@
+declare module "@webassemblyjs/wasm-parser" {
+  export function decode(bytes: Uint8Array, options?: Record<string, unknown>): unknown;
+}