@scriptc/compiler 0.4.1 → 0.4.3

package/dist/ir/paired-while-loops.js

@@ -0,0 +1,335 @@
+import { matchIntegerPairedForLoop } from "./integer-loops.js";
+import { analyzeIntegerRanges } from "./integer-ranges.js";
+import { F64, } from "./ir.js";
+import { everyExprChild, everyStmtChild, mapExprChildren, mapStmtChildren } from "./traverse.js";
+/** Two-counter while loops run as two-counter for loops.
+ *
+ *   let lo = a, hi = b;
+ *   while (lo < hi) { ...; lo++; hi--; }
+ *
+ * runs as `for (let l = lo, h = hi; l < h; l++, h--) { ... }`, whose
+ * counters the backend stores as integers (matchIntegerPairedForLoop). The
+ * loop qualifies when:
+ * - its condition compares the two counters (`<`, `<=`, `>` or `>=`);
+ * - its body ends with one update of each counter;
+ * - no `continue` in the body binds to it (a for loop's `continue` runs the
+ *   update, a while loop's skips the body's last statements);
+ * - the function mentions neither counter outside the loop, except in one
+ *   declaration or as a parameter;
+ * - every entry starts from the declared or passed values: a declaration is
+ *   an earlier statement of the loop's own statement list, which runs before
+ *   each entry, and a loop over a parameter is in no other loop.
+ *
+ * The for loop's counters are new locals that start from the counters'
+ * values on entry, so the declarations stay where they are and nothing
+ * between them and the loop moves. The original counters are not read
+ * after the loop, so they need no final value. A loop that the paired match
+ * would refuse is left as it is. */
+export function runPairedWhileLoopsAsFor(mod) {
+    let changed = false;
+    const functions = mod.functions.map((fn) => {
+        const next = rewriteFunction(fn);
+        if (next !== fn)
+            changed = true;
+        return next;
+    });
+    return changed ? { ...mod, functions } : mod;
+}
+/** The local a statement steps: `x++`, `x--` or `x = ...`. */
+function updatedLocal(s) {
+    if (s.kind === "assign")
+        return s.localId;
+    if (s.kind === "exprStmt" && s.expr.kind === "incDec")
+        return s.expr.localId;
+    return null;
+}
+/** The counters of a loop with the qualifying condition and trailing
+ * updates, or null. */
+function loopCounters(loop, locals) {
+    const cond = loop.cond;
+    if (cond.kind !== "bin")
+        return null;
+    if (cond.op !== "<" && cond.op !== "<=" && cond.op !== ">" && cond.op !== ">=")
+        return null;
+    if (cond.left.kind !== "varRef" || cond.right.kind !== "varRef")
+        return null;
+    const counters = [cond.left.localId, cond.right.localId];
+    if (counters[0] === counters[1])
+        return null;
+    const n = loop.body.length;
+    if (n < 2)
+        return null;
+    const updated = [updatedLocal(loop.body[n - 2]), updatedLocal(loop.body[n - 1])];
+    if (!counters.every((id) => updated.includes(id)))
+        return null;
+    for (const id of counters) {
+        const local = locals.get(id);
+        if (local?.type.kind !== "f64" || !local.mutable || local.boxed || local.tdz)
+            return null;
+    }
+    return counters;
+}
+function isLoop(s) {
+    return s.kind === "while" || s.kind === "doWhile" || s.kind === "for" || s.kind === "forOf";
+}
+/** True when a `continue` in the body binds to the loop: an unlabeled one
+ * outside nested loops, or one that names a label of the loop. */
+function continuesLoop(loop) {
+    const labels = loop.labels ?? [];
+    let depth = 0;
+    let found = false;
+    const expr = (e) => !found && everyExprChild(e, expr, stmt);
+    const stmt = (s) => {
+        if (found)
+            return false;
+        if (s.kind === "continue") {
+            if (s.label === undefined ? depth === 0 : labels.includes(s.label))
+                found = true;
+            return !found;
+        }
+        const nested = isLoop(s);
+        if (nested)
+            depth++;
+        const ok = everyStmtChild(s, expr, stmt);
+        if (nested)
+            depth--;
+        return ok;
+    };
+    loop.body.every(stmt);
+    return found;
+}
+/** Mentions of each id in `node`, in any field (a string literal equal to
+ * an id counts too, which only refuses more loops). */
+function countMentions(node, ids, mentions) {
+    if (typeof node === "string") {
+        if (ids.has(node))
+            mentions.set(node, (mentions.get(node) ?? 0) + 1);
+    }
+    else if (Array.isArray(node)) {
+        for (const item of node)
+            countMentions(item, ids, mentions);
+    }
+    else if (node !== null && typeof node === "object") {
+        const record = node;
+        for (const key in record)
+            if (key !== "loc" && key !== "type")
+                countMentions(record[key], ids, mentions);
+    }
+}
+function findCandidates(fn, locals) {
+    const candidates = [];
+    let depth = 0;
+    const list = (body) => {
+        body.forEach((s, i) => {
+            if (s.kind === "while") {
+                const counters = loopCounters(s, locals);
+                if (counters !== null && !continuesLoop(s))
+                    candidates.push({ loop: s, counters, before: body.slice(0, i), nested: depth > 0 });
+            }
+            visit(s);
+        });
+    };
+    // Statement lists inside expressions are not searched: their loops stay.
+    const visit = (s) => {
+        const loop = isLoop(s);
+        if (loop)
+            depth++;
+        switch (s.kind) {
+            case "block":
+            case "while":
+            case "doWhile":
+            case "for":
+            case "forOf":
+                list(s.body);
+                break;
+            case "if":
+                list(s.then);
+                if (s.else_ !== null)
+                    list(s.else_);
+                break;
+            case "switch":
+                for (const c of s.cases)
+                    list(c.body);
+                break;
+            case "tryCatch":
+                list(s.tryBody);
+                if (s.catchBody !== null)
+                    list(s.catchBody);
+                if (s.finallyBody !== null)
+                    list(s.finallyBody);
+                break;
+            default:
+                break;
+        }
+        if (loop)
+            depth--;
+    };
+    list(fn.body);
+    return candidates;
+}
+function rewriteFunction(fn) {
+    const locals = new Map(fn.locals.map((l) => [l.id, l]));
+    const candidates = findCandidates(fn, locals);
+    if (candidates.length === 0)
+        return fn;
+    const ids = new Set();
+    for (const c of candidates)
+        for (const id of c.counters)
+            ids.add(id);
+    const total = new Map();
+    countMentions(fn.body, ids, total);
+    const params = new Set(fn.params.map((p) => p.localId));
+    // A counter this function reaches through its environment.
+    const outer = new Map();
+    countMentions([fn.captures ?? [], fn.classCaptures ?? []], ids, outer);
+    let ranges = null;
+    const used = new Set(fn.locals.map((l) => l.id));
+    const added = [];
+    const rewritten = new Map();
+    for (const { loop, counters, before, nested } of candidates) {
+        const inside = new Map();
+        countMentions(loop, new Set(counters), inside);
+        // Each counter's start: its declaration's initializer, or null for a
+        // parameter.
+        const starts = new Map();
+        for (const id of counters) {
+            if ((outer.get(id) ?? 0) !== 0)
+                break;
+            const elsewhere = (total.get(id) ?? 0) - (inside.get(id) ?? 0);
+            if (params.has(id)) {
+                if (elsewhere === 0 && !nested)
+                    starts.set(id, null);
+                continue;
+            }
+            const decl = before.find((s) => s.kind === "varDecl" && s.localId === id);
+            if (elsewhere === 1 && decl?.kind === "varDecl" && decl.init !== null)
+                starts.set(id, decl.init);
+        }
+        if (starts.size !== counters.length)
+            continue;
+        const renamed = new Map();
+        const fresh = [];
+        for (const id of counters) {
+            const local = locals.get(id);
+            if (local === undefined)
+                break;
+            let k = 0;
+            while (used.has(`${local.name}.${k}`))
+                k++;
+            const next = `${local.name}.${k}`;
+            used.add(next);
+            renamed.set(id, next);
+            fresh.push({ ...local, id: next });
+        }
+        if (renamed.size !== counters.length) {
+            for (const local of fresh)
+                used.delete(local.id);
+            continue;
+        }
+        const candidate = pairedForLoop(loop, renamed);
+        const forLocals = new Map(locals);
+        for (const local of fresh)
+            forLocals.set(local.id, local);
+        // A start reads its counter on entry, which holds the initializer's
+        // value, with that value's range.
+        ranges ??= analyzeIntegerRanges(fn);
+        const startRanges = new Map();
+        if (candidate.init?.kind === "block")
+            for (const decl of candidate.init.body) {
+                if (decl.kind !== "varDecl" || decl.init?.kind !== "varRef")
+                    continue;
+                const init = starts.get(decl.init.localId);
+                if (!init)
+                    continue;
+                const literal = init.kind === "numLit" && Number.isSafeInteger(init.value) && !Object.is(init.value, -0);
+                startRanges.set(decl.init, literal ? { min: init.value, max: init.value } : (ranges.get(init) ?? null));
+            }
+        if (matchIntegerPairedForLoop(candidate, forLocals, startRanges) === null) {
+            for (const local of fresh)
+                used.delete(local.id);
+            continue;
+        }
+        rewritten.set(loop, renamed);
+        added.push(...fresh);
+    }
+    if (rewritten.size === 0)
+        return fn;
+    const expr = (e) => mapExprChildren(e, expr, stmt);
+    const stmt = (s) => {
+        const next = mapStmtChildren(s, expr, stmt);
+        const renamed = s.kind === "while" ? rewritten.get(s) : undefined;
+        // The body may hold rewritten loops of its own.
+        return renamed === undefined ? next : pairedForLoop(next, renamed);
+    };
+    return { ...fn, body: fn.body.map(stmt), locals: [...fn.locals, ...added] };
+}
+/** The for loop over the renamed counters: their starts read the original
+ * counters, and the body's two trailing updates become the update. */
+function pairedForLoop(loop, renamed) {
+    // Each copy spreads one node kind.
+    const expr = (e) => {
+        const next = mapExprChildren(e, expr, stmt);
+        if (next.kind === "varRef") {
+            const id = renamed.get(next.localId);
+            if (id !== undefined)
+                return { ...next, localId: id };
+        }
+        else if (next.kind === "incDec") {
+            const id = renamed.get(next.localId);
+            if (id !== undefined)
+                return { ...next, localId: id };
+        }
+        else if (next.kind === "assignExpr") {
+            const id = renamed.get(next.localId);
+            if (id !== undefined)
+                return { ...next, localId: id };
+        }
+        return next;
+    };
+    const stmt = (s) => {
+        const next = mapStmtChildren(s, expr, stmt);
+        if (next.kind === "varDecl") {
+            const id = renamed.get(next.localId);
+            if (id !== undefined)
+                return { ...next, localId: id };
+        }
+        else if (next.kind === "assign") {
+            const id = renamed.get(next.localId);
+            if (id !== undefined)
+                return { ...next, localId: id };
+        }
+        else if (next.kind === "forOf") {
+            const id = renamed.get(next.localId);
+            if (id !== undefined)
+                return { ...next, localId: id };
+        }
+        else if (next.kind === "rethrow") {
+            const id = renamed.get(next.localId);
+            if (id !== undefined)
+                return { ...next, localId: id };
+        }
+        return next;
+    };
+    const loc = loop.loc;
+    const n = loop.body.length;
+    const declarations = [];
+    for (const [id, next] of renamed)
+        declarations.push({
+            kind: "varDecl",
+            localId: next,
+            init: { kind: "varRef", localId: id, type: F64, loc },
+            loc,
+        });
+    const result = {
+        kind: "for",
+        init: { kind: "block", body: declarations, loc },
+        cond: expr(loop.cond),
+        update: { kind: "block", body: loop.body.slice(n - 2).map(stmt), loc },
+        body: loop.body.slice(0, n - 2).map(stmt),
+        loc,
+    };
+    if (loop.labels !== undefined)
+        result.labels = loop.labels;
+    return result;
+}
+//# sourceMappingURL=paired-while-loops.js.map

package/dist/ir/paired-while-loops.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"paired-while-loops.js","sourceRoot":"","sources":["../../src/ir/paired-while-loops.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,yBAAyB,EAAE,MAAM,oBAAoB,CAAC;AAC/D,OAAO,EAAE,oBAAoB,EAAqB,MAAM,qBAAqB,CAAC;AAC9E,OAAO,EACL,GAAG,GAMJ,MAAM,SAAS,CAAC;AACjB,OAAO,EAAE,cAAc,EAAE,cAAc,EAAE,eAAe,EAAE,eAAe,EAAE,MAAM,eAAe,CAAC;AAIjG;;;;;;;;;;;;;;;;;;;;;;oCAsBoC;AACpC,MAAM,UAAU,wBAAwB,CAAC,GAAa;IACpD,IAAI,OAAO,GAAG,KAAK,CAAC;IACpB,MAAM,SAAS,GAAG,GAAG,CAAC,SAAS,CAAC,GAAG,CAAC,CAAC,EAAE,EAAE,EAAE;QACzC,MAAM,IAAI,GAAG,eAAe,CAAC,EAAE,CAAC,CAAC;QACjC,IAAI,IAAI,KAAK,EAAE;YAAE,OAAO,GAAG,IAAI,CAAC;QAChC,OAAO,IAAI,CAAC;IACd,CAAC,CAAC,CAAC;IACH,OAAO,OAAO,CAAC,CAAC,CAAC,EAAE,GAAG,GAAG,EAAE,SAAS,EAAE,CAAC,CAAC,CAAC,GAAG,CAAC;AAC/C,CAAC;AAED,8DAA8D;AAC9D,SAAS,YAAY,CAAC,CAAS;IAC7B,IAAI,CAAC,CAAC,IAAI,KAAK,QAAQ;QAAE,OAAO,CAAC,CAAC,OAAO,CAAC;IAC1C,IAAI,CAAC,CAAC,IAAI,KAAK,UAAU,IAAI,CAAC,CAAC,IAAI,CAAC,IAAI,KAAK,QAAQ;QAAE,OAAO,CAAC,CAAC,IAAI,CAAC,OAAO,CAAC;IAC7E,OAAO,IAAI,CAAC;AACd,CAAC;AAED;uBACuB;AACvB,SAAS,YAAY,CAAC,IAAe,EAAE,MAAoC;IACzE,MAAM,IAAI,GAAG,IAAI,CAAC,IAAI,CAAC;IACvB,IAAI,IAAI,CAAC,IAAI,KAAK,KAAK;QAAE,OAAO,IAAI,CAAC;IACrC,IAAI,IAAI,CAAC,EAAE,KAAK,GAAG,IAAI,IAAI,CAAC,EAAE,KAAK,IAAI,IAAI,IAAI,CAAC,EAAE,KAAK,GAAG,IAAI,IAAI,CAAC,EAAE,KAAK,IAAI;QAAE,OAAO,IAAI,CAAC;IAC5F,IAAI,IAAI,CAAC,IAAI,CAAC,IAAI,KAAK,QAAQ,IAAI,IAAI,CAAC,KAAK,CAAC,IAAI,KAAK,QAAQ;QAAE,OAAO,IAAI,CAAC;IAC7E,MAAM,QAAQ,GAAG,CAAC,IAAI,CAAC,IAAI,CAAC,OAAO,EAAE,IAAI,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC;IACzD,IAAI,QAAQ,CAAC,CAAC,CAAC,KAAK,QAAQ,CAAC,CAAC,CAAC;QAAE,OAAO,IAAI,CAAC;IAC7C,MAAM,CAAC,GAAG,IAAI,CAAC,IAAI,CAAC,MAAM,CAAC;IAC3B,IAAI,CAAC,GAAG,CAAC;QAAE,OAAO,IAAI,CAAC;IACvB,MAAM,OAAO,GAAG,CAAC,YAAY,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC,GAAG,CAAC,CAAE,CAAC,EAAE,YAAY,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC,GAAG,CAAC,CAAE,CAAC,CAAC,CAAC;IACnF,IAAI,CAAC,QAAQ,CAAC,KAAK,CAAC,CAAC,EAAE,EAAE,EAAE,CAAC,OAAO,CAAC,QAAQ,CAAC,EAAE,CAAC,CAAC;QAAE,OAAO,IAAI,CAAC;IAC/D,KAAK,MAAM,EAAE,IAAI,QAAQ,EAAE,CAAC;QAC1B,MAAM,KAAK,GAAG,MAAM,CAAC,GAAG,CAAC,EAAE,CAAC,CAAC;QAC7B,IAAI,KAAK,EAAE,IAAI,CAAC,IAAI,KAAK,KAAK,IAAI,CAAC,KAAK,CAAC,OAAO,IAAI,KAAK,CAAC,KAAK,IAAI,KAAK,CAAC,GAAG;YAAE,OAAO,IAAI,CAAC;IAC5F,CAAC;IACD,OAAO,QAAQ,CAAC;AAClB,CAAC;AAED,SAAS,MAAM,CAAC,CAAS;IACvB,OAAO,CAAC,CAAC,IAAI,KAAK,OAAO,IAAI,CAAC,CAAC,IAAI,KAAK,SAAS,IAAI,CAAC,CAAC,IAAI,KAAK,KAAK,IAAI,CAAC,CAAC,IAAI,KAAK,OAAO,CAAC;AAC9F,CAAC;AAED;kEACkE;AAClE,SAAS,aAAa,CAAC,IAAe;IACpC,MAAM,MAAM,GAAG,IAAI,CAAC,MAAM,IAAI,EAAE,CAAC;IACjC,IAAI,KAAK,GAAG,CAAC,CAAC;IACd,IAAI,KAAK,GAAG,KAAK,CAAC;IAClB,MAAM,IAAI,GAAG,CAAC,CAAS,EAAW,EAAE,CAAC,CAAC,KAAK,IAAI,cAAc,CAAC,CAAC,EAAE,IAAI,EAAE,IAAI,CAAC,CAAC;IAC7E,MAAM,IAAI,GAAG,CAAC,CAAS,EAAW,EAAE;QAClC,IAAI,KAAK;YAAE,OAAO,KAAK,CAAC;QACxB,IAAI,CAAC,CAAC,IAAI,KAAK,UAAU,EAAE,CAAC;YAC1B,IAAI,CAAC,CAAC,KAAK,KAAK,SAAS,CAAC,CAAC,CAAC,KAAK,KAAK,CAAC,CAAC,CAAC,CAAC,MAAM,CAAC,QAAQ,CAAC,CAAC,CAAC,KAAK,CAAC;gBAAE,KAAK,GAAG,IAAI,CAAC;YACjF,OAAO,CAAC,KAAK,CAAC;QAChB,CAAC;QACD,MAAM,MAAM,GAAG,MAAM,CAAC,CAAC,CAAC,CAAC;QACzB,IAAI,MAAM;YAAE,KAAK,EAAE,CAAC;QACpB,MAAM,EAAE,GAAG,cAAc,CAAC,CAAC,EAAE,IAAI,EAAE,IAAI,CAAC,CAAC;QACzC,IAAI,MAAM;YAAE,KAAK,EAAE,CAAC;QACpB,OAAO,EAAE,CAAC;IACZ,CAAC,CAAC;IACF,IAAI,CAAC,IAAI,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC;IACtB,OAAO,KAAK,CAAC;AACf,CAAC;AAED;uDACuD;AACvD,SAAS,aAAa,CACpB,IAAa,EACb,GAAwB,EACxB,QAA6B;IAE7B,IAAI,OAAO,IAAI,KAAK,QAAQ,EAAE,CAAC;QAC7B,IAAI,GAAG,CAAC,GAAG,CAAC,IAAI,CAAC;YAAE,QAAQ,CAAC,GAAG,CAAC,IAAI,EAAE,CAAC,QAAQ,CAAC,GAAG,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC;IACvE,CAAC;SAAM,IAAI,KAAK,CAAC,OAAO,CAAC,IAAI,CAAC,EAAE,CAAC;QAC/B,KAAK,MAAM,IAAI,IAAI,IAAI;YAAE,aAAa,CAAC,IAAI,EAAE,GAAG,EAAE,QAAQ,CAAC,CAAC;IAC9D,CAAC;SAAM,IAAI,IAAI,KAAK,IAAI,IAAI,OAAO,IAAI,KAAK,QAAQ,EAAE,CAAC;QACrD,MAAM,MAAM,GAAG,IAA+B,CAAC;QAC/C,KAAK,MAAM,GAAG,IAAI,MAAM;YACtB,IAAI,GAAG,KAAK,KAAK,IAAI,GAAG,KAAK,MAAM;gBAAE,aAAa,CAAC,MAAM,CAAC,GAAG,CAAC,EAAE,GAAG,EAAE,QAAQ,CAAC,CAAC;IACnF,CAAC;AACH,CAAC;AAWD,SAAS,cAAc,CAAC,EAAc,EAAE,MAAoC;IAC1E,MAAM,UAAU,GAAgB,EAAE,CAAC;IACnC,IAAI,KAAK,GAAG,CAAC,CAAC;IACd,MAAM,IAAI,GAAG,CAAC,IAAuB,EAAQ,EAAE;QAC7C,IAAI,CAAC,OAAO,CAAC,CAAC,CAAC,EAAE,CAAC,EAAE,EAAE;YACpB,IAAI,CAAC,CAAC,IAAI,KAAK,OAAO,EAAE,CAAC;gBACvB,MAAM,QAAQ,GAAG,YAAY,CAAC,CAAC,EAAE,MAAM,CAAC,CAAC;gBACzC,IAAI,QAAQ,KAAK,IAAI,IAAI,CAAC,aAAa,CAAC,CAAC,CAAC;oBACxC,UAAU,CAAC,IAAI,CAAC,EAAE,IAAI,EAAE,CAAC,EAAE,QAAQ,EAAE,MAAM,EAAE,IAAI,CAAC,KAAK,CAAC,CAAC,EAAE,CAAC,CAAC,EAAE,MAAM,EAAE,KAAK,GAAG,CAAC,EAAE,CAAC,CAAC;YACxF,CAAC;YACD,KAAK,CAAC,CAAC,CAAC,CAAC;QACX,CAAC,CAAC,CAAC;IACL,CAAC,CAAC;IACF,yEAAyE;IACzE,MAAM,KAAK,GAAG,CAAC,CAAS,EAAQ,EAAE;QAChC,MAAM,IAAI,GAAG,MAAM,CAAC,CAAC,CAAC,CAAC;QACvB,IAAI,IAAI;YAAE,KAAK,EAAE,CAAC;QAClB,QAAQ,CAAC,CAAC,IAAI,EAAE,CAAC;YACf,KAAK,OAAO,CAAC;YACb,KAAK,OAAO,CAAC;YACb,KAAK,SAAS,CAAC;YACf,KAAK,KAAK,CAAC;YACX,KAAK,OAAO;gBACV,IAAI,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC;gBACb,MAAM;YACR,KAAK,IAAI;gBACP,IAAI,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC;gBACb,IAAI,CAAC,CAAC,KAAK,KAAK,IAAI;oBAAE,IAAI,CAAC,CAAC,CAAC,KAAK,CAAC,CAAC;gBACpC,MAAM;YACR,KAAK,QAAQ;gBACX,KAAK,MAAM,CAAC,IAAI,CAAC,CAAC,KAAK;oBAAE,IAAI,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC;gBACtC,MAAM;YACR,KAAK,UAAU;gBACb,IAAI,CAAC,CAAC,CAAC,OAAO,CAAC,CAAC;gBAChB,IAAI,CAAC,CAAC,SAAS,KAAK,IAAI;oBAAE,IAAI,CAAC,CAAC,CAAC,SAAS,CAAC,CAAC;gBAC5C,IAAI,CAAC,CAAC,WAAW,KAAK,IAAI;oBAAE,IAAI,CAAC,CAAC,CAAC,WAAW,CAAC,CAAC;gBAChD,MAAM;YACR;gBACE,MAAM;QACV,CAAC;QACD,IAAI,IAAI;YAAE,KAAK,EAAE,CAAC;IACpB,CAAC,CAAC;IACF,IAAI,CAAC,EAAE,CAAC,IAAI,CAAC,CAAC;IACd,OAAO,UAAU,CAAC;AACpB,CAAC;AAED,SAAS,eAAe,CAAC,EAAc;IACrC,MAAM,MAAM,GAAG,IAAI,GAAG,CAAC,EAAE,CAAC,MAAM,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,CAAC,CAAC;IACxD,MAAM,UAAU,GAAG,cAAc,CAAC,EAAE,EAAE,MAAM,CAAC,CAAC;IAC9C,IAAI,UAAU,CAAC,MAAM,KAAK,CAAC;QAAE,OAAO,EAAE,CAAC;IAEvC,MAAM,GAAG,GAAG,IAAI,GAAG,EAAU,CAAC;IAC9B,KAAK,MAAM,CAAC,IAAI,UAAU;QAAE,KAAK,MAAM,EAAE,IAAI,CAAC,CAAC,QAAQ;YAAE,GAAG,CAAC,GAAG,CAAC,EAAE,CAAC,CAAC;IACrE,MAAM,KAAK,GAAG,IAAI,GAAG,EAAkB,CAAC;IACxC,aAAa,CAAC,EAAE,CAAC,IAAI,EAAE,GAAG,EAAE,KAAK,CAAC,CAAC;IACnC,MAAM,MAAM,GAAG,IAAI,GAAG,CAAC,EAAE,CAAC,MAAM,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,OAAO,CAAC,CAAC,CAAC;IACxD,2DAA2D;IAC3D,MAAM,KAAK,GAAG,IAAI,GAAG,EAAkB,CAAC;IACxC,aAAa,CAAC,CAAC,EAAE,CAAC,QAAQ,IAAI,EAAE,EAAE,EAAE,CAAC,aAAa,IAAI,EAAE,CAAC,EAAE,GAAG,EAAE,KAAK,CAAC,CAAC;IAEvE,IAAI,MAAM,GAAoD,IAAI,CAAC;IACnE,MAAM,IAAI,GAAG,IAAI,GAAG,CAAC,EAAE,CAAC,MAAM,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC;IACjD,MAAM,KAAK,GAAc,EAAE,CAAC;IAC5B,MAAM,SAAS,GAAG,IAAI,GAAG,EAAkC,CAAC;IAC5D,KAAK,MAAM,EAAE,IAAI,EAAE,QAAQ,EAAE,MAAM,EAAE,MAAM,EAAE,IAAI,UAAU,EAAE,CAAC;QAC5D,MAAM,MAAM,GAAG,IAAI,GAAG,EAAkB,CAAC;QACzC,aAAa,CAAC,IAAI,EAAE,IAAI,GAAG,CAAC,QAAQ,CAAC,EAAE,MAAM,CAAC,CAAC;QAC/C,qEAAqE;QACrE,aAAa;QACb,MAAM,MAAM,GAAG,IAAI,GAAG,EAAyB,CAAC;QAChD,KAAK,MAAM,EAAE,IAAI,QAAQ,EAAE,CAAC;YAC1B,IAAI,CAAC,KAAK,CAAC,GAAG,CAAC,EAAE,CAAC,IAAI,CAAC,CAAC,KAAK,CAAC;gBAAE,MAAM;YACtC,MAAM,SAAS,GAAG,CAAC,KAAK,CAAC,GAAG,CAAC,EAAE,CAAC,IAAI,CAAC,CAAC,GAAG,CAAC,MAAM,CAAC,GAAG,CAAC,EAAE,CAAC,IAAI,CAAC,CAAC,CAAC;YAC/D,IAAI,MAAM,CAAC,GAAG,CAAC,EAAE,CAAC,EAAE,CAAC;gBACnB,IAAI,SAAS,KAAK,CAAC,IAAI,CAAC,MAAM;oBAAE,MAAM,CAAC,GAAG,CAAC,EAAE,EAAE,IAAI,CAAC,CAAC;gBACrD,SAAS;YACX,CAAC;YACD,MAAM,IAAI,GAAG,MAAM,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,IAAI,KAAK,SAAS,IAAI,CAAC,CAAC,OAAO,KAAK,EAAE,CAAC,CAAC;YAC1E,IAAI,SAAS,KAAK,CAAC,IAAI,IAAI,EAAE,IAAI,KAAK,SAAS,IAAI,IAAI,CAAC,IAAI,KAAK,IAAI;gBACnE,MAAM,CAAC,GAAG,CAAC,EAAE,EAAE,IAAI,CAAC,IAAI,CAAC,CAAC;QAC9B,CAAC;QACD,IAAI,MAAM,CAAC,IAAI,KAAK,QAAQ,CAAC,MAAM;YAAE,SAAS;QAC9C,MAAM,OAAO,GAAG,IAAI,GAAG,EAAkB,CAAC;QAC1C,MAAM,KAAK,GAAc,EAAE,CAAC;QAC5B,KAAK,MAAM,EAAE,IAAI,QAAQ,EAAE,CAAC;YAC1B,MAAM,KAAK,GAAG,MAAM,CAAC,GAAG,CAAC,EAAE,CAAC,CAAC;YAC7B,IAAI,KAAK,KAAK,SAAS;gBAAE,MAAM;YAC/B,IAAI,CAAC,GAAG,CAAC,CAAC;YACV,OAAO,IAAI,CAAC,GAAG,CAAC,GAAG,KAAK,CAAC,IAAI,IAAI,CAAC,EAAE,CAAC;gBAAE,CAAC,EAAE,CAAC;YAC3C,MAAM,IAAI,GAAG,GAAG,KAAK,CAAC,IAAI,IAAI,CAAC,EAAE,CAAC;YAClC,IAAI,CAAC,GAAG,CAAC,IAAI,CAAC,CAAC;YACf,OAAO,CAAC,GAAG,CAAC,EAAE,EAAE,IAAI,CAAC,CAAC;YACtB,KAAK,CAAC,IAAI,CAAC,EAAE,GAAG,KAAK,EAAE,EAAE,EAAE,IAAI,EAAE,CAAC,CAAC;QACrC,CAAC;QACD,IAAI,OAAO,CAAC,IAAI,KAAK,QAAQ,CAAC,MAAM,EAAE,CAAC;YACrC,KAAK,MAAM,KAAK,IAAI,KAAK;gBAAE,IAAI,CAAC,MAAM,CAAC,KAAK,CAAC,EAAE,CAAC,CAAC;YACjD,SAAS;QACX,CAAC;QACD,MAAM,SAAS,GAAG,aAAa,CAAC,IAAI,EAAE,OAAO,CAAC,CAAC;QAC/C,MAAM,SAAS,GAAG,IAAI,GAAG,CAAC,MAAM,CAAC,CAAC;QAClC,KAAK,MAAM,KAAK,IAAI,KAAK;YAAE,SAAS,CAAC,GAAG,CAAC,KAAK,CAAC,EAAE,EAAE,KAAK,CAAC,CAAC;QAC1D,oEAAoE;QACpE,kCAAkC;QAClC,MAAM,KAAK,oBAAoB,CAAC,EAAE,CAAC,CAAC;QACpC,MAAM,WAAW,GAAG,IAAI,GAAG,EAA+B,CAAC;QAC3D,IAAI,SAAS,CAAC,IAAI,EAAE,IAAI,KAAK,OAAO;YAClC,KAAK,MAAM,IAAI,IAAI,SAAS,CAAC,IAAI,CAAC,IAAI,EAAE,CAAC;gBACvC,IAAI,IAAI,CAAC,IAAI,KAAK,SAAS,IAAI,IAAI,CAAC,IAAI,EAAE,IAAI,KAAK,QAAQ;oBAAE,SAAS;gBACtE,MAAM,IAAI,GAAG,MAAM,CAAC,GAAG,CAAC,IAAI,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC;gBAC3C,IAAI,CAAC,IAAI;oBAAE,SAAS;gBACpB,MAAM,OAAO,GACX,IAAI,CAAC,IAAI,KAAK,QAAQ,IAAI,MAAM,CAAC,aAAa,CAAC,IAAI,CAAC,KAAK,CAAC,IAAI,CAAC,MAAM,CAAC,EAAE,CAAC,IAAI,CAAC,KAAK,EAAE,CAAC,CAAC,CAAC,CAAC;gBAC3F,WAAW,CAAC,GAAG,CACb,IAAI,CAAC,IAAI,EACT,OAAO,CAAC,CAAC,CAAC,EAAE,GAAG,EAAE,IAAI,CAAC,KAAK,EAAE,GAAG,EAAE,IAAI,CAAC,KAAK,EAAE,CAAC,CAAC,CAAC,CAAC,MAAM,CAAC,GAAG,CAAC,IAAI,CAAC,IAAI,IAAI,CAAC,CAC5E,CAAC;YACJ,CAAC;QACH,IAAI,yBAAyB,CAAC,SAAS,EAAE,SAAS,EAAE,WAAW,CAAC,KAAK,IAAI,EAAE,CAAC;YAC1E,KAAK,MAAM,KAAK,IAAI,KAAK;gBAAE,IAAI,CAAC,MAAM,CAAC,KAAK,CAAC,EAAE,CAAC,CAAC;YACjD,SAAS;QACX,CAAC;QACD,SAAS,CAAC,GAAG,CAAC,IAAI,EAAE,OAAO,CAAC,CAAC;QAC7B,KAAK,CAAC,IAAI,CAAC,GAAG,KAAK,CAAC,CAAC;IACvB,CAAC;IACD,IAAI,SAAS,CAAC,IAAI,KAAK,CAAC;QAAE,OAAO,EAAE,CAAC;IAEpC,MAAM,IAAI,GAAG,CAAC,CAAS,EAAU,EAAE,CAAC,eAAe,CAAC,CAAC,EAAE,IAAI,EAAE,IAAI,CAAC,CAAC;IACnE,MAAM,IAAI,GAAG,CAAC,CAAS,EAAU,EAAE;QACjC,MAAM,IAAI,GAAG,eAAe,CAAC,CAAC,EAAE,IAAI,EAAE,IAAI,CAAC,CAAC;QAC5C,MAAM,OAAO,GAAG,CAAC,CAAC,IAAI,KAAK,OAAO,CAAC,CAAC,CAAC,SAAS,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,SAAS,CAAC;QAClE,gDAAgD;QAChD,OAAO,OAAO,KAAK,SAAS,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,aAAa,CAAC,IAAiB,EAAE,OAAO,CAAC,CAAC;IAClF,CAAC,CAAC;IACF,OAAO,EAAE,GAAG,EAAE,EAAE,IAAI,EAAE,EAAE,CAAC,IAAI,CAAC,GAAG,CAAC,IAAI,CAAC,EAAE,MAAM,EAAE,CAAC,GAAG,EAAE,CAAC,MAAM,EAAE,GAAG,KAAK,CAAC,EAAE,CAAC;AAC9E,CAAC;AAED;sEACsE;AACtE,SAAS,aAAa,CACpB,IAAe,EACf,OAAoC;IAEpC,mCAAmC;IACnC,MAAM,IAAI,GAAG,CAAC,CAAS,EAAU,EAAE;QACjC,MAAM,IAAI,GAAG,eAAe,CAAC,CAAC,EAAE,IAAI,EAAE,IAAI,CAAC,CAAC;QAC5C,IAAI,IAAI,CAAC,IAAI,KAAK,QAAQ,EAAE,CAAC;YAC3B,MAAM,EAAE,GAAG,OAAO,CAAC,GAAG,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC;YACrC,IAAI,EAAE,KAAK,SAAS;gBAAE,OAAO,EAAE,GAAG,IAAI,EAAE,OAAO,EAAE,EAAE,EAAE,CAAC;QACxD,CAAC;aAAM,IAAI,IAAI,CAAC,IAAI,KAAK,QAAQ,EAAE,CAAC;YAClC,MAAM,EAAE,GAAG,OAAO,CAAC,GAAG,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC;YACrC,IAAI,EAAE,KAAK,SAAS;gBAAE,OAAO,EAAE,GAAG,IAAI,EAAE,OAAO,EAAE,EAAE,EAAE,CAAC;QACxD,CAAC;aAAM,IAAI,IAAI,CAAC,IAAI,KAAK,YAAY,EAAE,CAAC;YACtC,MAAM,EAAE,GAAG,OAAO,CAAC,GAAG,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC;YACrC,IAAI,EAAE,KAAK,SAAS;gBAAE,OAAO,EAAE,GAAG,IAAI,EAAE,OAAO,EAAE,EAAE,EAAE,CAAC;QACxD,CAAC;QACD,OAAO,IAAI,CAAC;IACd,CAAC,CAAC;IACF,MAAM,IAAI,GAAG,CAAC,CAAS,EAAU,EAAE;QACjC,MAAM,IAAI,GAAG,eAAe,CAAC,CAAC,EAAE,IAAI,EAAE,IAAI,CAAC,CAAC;QAC5C,IAAI,IAAI,CAAC,IAAI,KAAK,SAAS,EAAE,CAAC;YAC5B,MAAM,EAAE,GAAG,OAAO,CAAC,GAAG,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC;YACrC,IAAI,EAAE,KAAK,SAAS;gBAAE,OAAO,EAAE,GAAG,IAAI,EAAE,OAAO,EAAE,EAAE,EAAE,CAAC;QACxD,CAAC;aAAM,IAAI,IAAI,CAAC,IAAI,KAAK,QAAQ,EAAE,CAAC;YAClC,MAAM,EAAE,GAAG,OAAO,CAAC,GAAG,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC;YACrC,IAAI,EAAE,KAAK,SAAS;gBAAE,OAAO,EAAE,GAAG,IAAI,EAAE,OAAO,EAAE,EAAE,EAAE,CAAC;QACxD,CAAC;aAAM,IAAI,IAAI,CAAC,IAAI,KAAK,OAAO,EAAE,CAAC;YACjC,MAAM,EAAE,GAAG,OAAO,CAAC,GAAG,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC;YACrC,IAAI,EAAE,KAAK,SAAS;gBAAE,OAAO,EAAE,GAAG,IAAI,EAAE,OAAO,EAAE,EAAE,EAAE,CAAC;QACxD,CAAC;aAAM,IAAI,IAAI,CAAC,IAAI,KAAK,SAAS,EAAE,CAAC;YACnC,MAAM,EAAE,GAAG,OAAO,CAAC,GAAG,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC;YACrC,IAAI,EAAE,KAAK,SAAS;gBAAE,OAAO,EAAE,GAAG,IAAI,EAAE,OAAO,EAAE,EAAE,EAAE,CAAC;QACxD,CAAC;QACD,OAAO,IAAI,CAAC;IACd,CAAC,CAAC;IACF,MAAM,GAAG,GAAG,IAAI,CAAC,GAAG,CAAC;IACrB,MAAM,CAAC,GAAG,IAAI,CAAC,IAAI,CAAC,MAAM,CAAC;IAC3B,MAAM,YAAY,GAAa,EAAE,CAAC;IAClC,KAAK,MAAM,CAAC,EAAE,EAAE,IAAI,CAAC,IAAI,OAAO;QAC9B,YAAY,CAAC,IAAI,CAAC;YAChB,IAAI,EAAE,SAAS;YACf,OAAO,EAAE,IAAI;YACb,IAAI,EAAE,EAAE,IAAI,EAAE,QAAQ,EAAE,OAAO,EAAE,EAAE,EAAE,IAAI,EAAE,GAAG,EAAE,GAAG,EAAE;YACrD,GAAG;SACJ,CAAC,CAAC;IACL,MAAM,MAAM,GAA6B;QACvC,IAAI,EAAE,KAAK;QACX,IAAI,EAAE,EAAE,IAAI,EAAE,OAAO,EAAE,IAAI,EAAE,YAAY,EAAE,GAAG,EAAE;QAChD,IAAI,EAAE,IAAI,CAAC,IAAI,CAAC,IAAI,CAAC;QACrB,MAAM,EAAE,EAAE,IAAI,EAAE,OAAO,EAAE,IAAI,EAAE,IAAI,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,GAAG,CAAC,IAAI,CAAC,EAAE,GAAG,EAAE;QACtE,IAAI,EAAE,IAAI,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC,EAAE,CAAC,GAAG,CAAC,CAAC,CAAC,GAAG,CAAC,IAAI,CAAC;QACzC,GAAG;KACJ,CAAC;IACF,IAAI,IAAI,CAAC,MAAM,KAAK,SAAS;QAAE,MAAM,CAAC,MAAM,GAAG,IAAI,CAAC,MAAM,CAAC;IAC3D,OAAO,MAAM,CAAC;AAChB,CAAC"}

package/dist/ir/paired-while-loops.test.d.ts

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

package/dist/ir/paired-while-loops.test.js

@@ -0,0 +1,163 @@
+import { expect, test } from "vitest";
+import { BOOL, F64, arrayOf, } from "./ir.js";
+import { runPairedWhileLoopsAsFor } from "./paired-while-loops.js";
+const loc = { file: "paired-while-loops.ts", start: 0, end: 0 };
+const number = (value) => ({ kind: "numLit", value, type: F64, loc });
+const ref = (localId) => ({ kind: "varRef", localId, type: F64, loc });
+const array = { kind: "varRef", localId: "a", type: arrayOf(F64), loc };
+const element = (index) => ({
+    kind: "exprStmt",
+    expr: { kind: "arrayGet", arr: array, index, type: F64, loc },
+    loc,
+});
+const step = (id, op) => ({
+    kind: "assign",
+    localId: id,
+    value: { kind: "bin", op, left: ref(id), right: number(1), type: F64, loc },
+    loc,
+});
+const declare = (id, init) => ({ kind: "varDecl", localId: id, init, loc });
+const number_ = (id) => ({
+    id,
+    name: id.split(".")[0],
+    type: F64,
+    mutable: true,
+});
+function whileLoop(body = [element(ref("lo.0")), element(ref("hi.0"))], updates = [step("lo.0", "+"), step("hi.0", "-")]) {
+    return {
+        kind: "while",
+        cond: { kind: "bin", op: "<", left: ref("lo.0"), right: ref("hi.0"), type: BOOL, loc },
+        body: [...body, ...updates],
+        loc,
+    };
+}
+function program(body, extra = {}) {
+    const fn = {
+        name: "f",
+        params: [{ localId: "n.0", name: "n", type: F64 }],
+        returnType: F64,
+        locals: [
+            number_("n.0"),
+            number_("lo.0"),
+            number_("hi.0"),
+            { id: "a", name: "a", type: arrayOf(F64), mutable: false },
+        ],
+        body,
+        loc,
+        ...extra,
+    };
+    return { functions: [fn], globals: [] };
+}
+const rewritten = (mod) => runPairedWhileLoopsAsFor(mod).functions[0].body;
+test("a two-counter while loop runs as a for loop over new counters", () => {
+    const mod = program([declare("lo.0", number(0)), declare("hi.0", ref("n.0")), whileLoop()]);
+    const out = runPairedWhileLoopsAsFor(mod).functions[0];
+    // The declarations stay; the loop's counters are new locals that start
+    // from them, and the trailing updates are the for loop's update.
+    expect(out.body.slice(0, 2)).toEqual(mod.functions[0].body.slice(0, 2));
+    expect(out.body[2]).toMatchObject({
+        kind: "for",
+        init: {
+            kind: "block",
+            body: [
+                { kind: "varDecl", localId: "lo.1", init: { kind: "varRef", localId: "lo.0" } },
+                { kind: "varDecl", localId: "hi.1", init: { kind: "varRef", localId: "hi.0" } },
+            ],
+        },
+        cond: { op: "<", left: { localId: "lo.1" }, right: { localId: "hi.1" } },
+        update: { kind: "block", body: [{ localId: "lo.1" }, { localId: "hi.1" }] },
+        body: [
+            { expr: { kind: "arrayGet", index: { localId: "lo.1" } } },
+            { expr: { kind: "arrayGet", index: { localId: "hi.1" } } },
+        ],
+    });
+    expect(out.locals.map((l) => l.id)).toEqual(["n.0", "lo.0", "hi.0", "a", "lo.1", "hi.1"]);
+});
+test("parameters, reversed comparisons, labels and either update order qualify", () => {
+    const loop = whileLoop(undefined, [step("hi.0", "-"), step("lo.0", "+")]);
+    loop.cond = { kind: "bin", op: ">=", left: ref("hi.0"), right: ref("lo.0"), type: BOOL, loc };
+    loop.labels = ["outer"];
+    const mod = program([declare("lo.0", number(0)), loop], {
+        params: [{ localId: "hi.0", name: "hi", type: F64 }],
+    });
+    expect(rewritten(mod)[1]).toMatchObject({
+        kind: "for",
+        labels: ["outer"],
+        cond: { op: ">=", left: { localId: "hi.1" }, right: { localId: "lo.1" } },
+        update: { body: [{ localId: "hi.1" }, { localId: "lo.1" }] },
+    });
+});
+test("loops the for form would change, or whose counters are used elsewhere, stay", () => {
+    const decls = [declare("lo.0", number(0)), declare("hi.0", ref("n.0"))];
+    const kept = (body, extra = {}) => {
+        const mod = program(body, extra);
+        expect(runPairedWhileLoopsAsFor(mod)).toBe(mod);
+    };
+    // A continue that binds to the loop skips the trailing updates.
+    kept([...decls, whileLoop([{ kind: "continue", loc }, element(ref("lo.0"))])]);
+    const labeled = whileLoop([{ kind: "continue", label: "outer", loc }, element(ref("lo.0"))]);
+    labeled.labels = ["outer"];
+    kept([...decls, labeled]);
+    // A counter read after the loop, written before it, or declared twice.
+    kept([...decls, whileLoop(), { kind: "return", value: ref("lo.0"), loc }]);
+    kept([...decls, step("hi.0", "-"), whileLoop()]);
+    kept([...decls, declare("lo.0", number(1)), whileLoop()]);
+    // A declaration outside an enclosing loop: a later entry would resume
+    // from the values the previous run left.
+    kept([
+        ...decls,
+        {
+            kind: "while",
+            cond: { kind: "boolLit", value: true, type: BOOL, loc },
+            body: [whileLoop()],
+            loc,
+        },
+    ]);
+    // A parameter counter inside another loop, for the same reason.
+    const outerLoop = {
+        kind: "while",
+        cond: { kind: "boolLit", value: true, type: BOOL, loc },
+        body: [whileLoop()],
+        loc,
+    };
+    kept([outerLoop], {
+        params: [
+            { localId: "lo.0", name: "lo", type: F64 },
+            { localId: "hi.0", name: "hi", type: F64 },
+        ],
+    });
+    // A counter that was never declared, or that this function captures.
+    kept([declare("lo.0", number(0)), whileLoop()]);
+    kept([...decls, whileLoop()], { captures: [{ localId: "hi.0", name: "hi", type: F64 }] });
+    // The body writes a counter, a counter is boxed, or no counter is an index.
+    kept([...decls, whileLoop([element(ref("lo.0")), step("hi.0", "-")])]);
+    const boxed = program([...decls, whileLoop()]);
+    boxed.functions[0].locals[2] = { ...number_("hi.0"), boxed: true };
+    expect(runPairedWhileLoopsAsFor(boxed)).toBe(boxed);
+    kept([...decls, whileLoop([])]);
+});
+test("a continue in a nested loop keeps the outer loop eligible", () => {
+    const inner = {
+        kind: "while",
+        cond: { kind: "boolLit", value: false, type: BOOL, loc },
+        body: [{ kind: "continue", loc }],
+        loc,
+    };
+    // Literal starts have known ranges, so the loop needs no start test and
+    // may hold a nested loop.
+    const mod = program([
+        declare("lo.0", number(0)),
+        declare("hi.0", number(9)),
+        whileLoop([inner, element(ref("lo.0"))]),
+    ]);
+    expect(rewritten(mod)[2]).toMatchObject({ kind: "for", body: [{ kind: "while" }, {}] });
+    // A start without a known range is tested at run time, on a loop that
+    // may not nest another.
+    const guarded = program([
+        declare("lo.0", number(0)),
+        declare("hi.0", ref("n.0")),
+        whileLoop([inner, element(ref("lo.0"))]),
+    ]);
+    expect(runPairedWhileLoopsAsFor(guarded)).toBe(guarded);
+});
+//# sourceMappingURL=paired-while-loops.test.js.map

package/dist/ir/paired-while-loops.test.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"paired-while-loops.test.js","sourceRoot":"","sources":["../../src/ir/paired-while-loops.test.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,MAAM,EAAE,IAAI,EAAE,MAAM,QAAQ,CAAC;AACtC,OAAO,EACL,IAAI,EACJ,GAAG,EACH,OAAO,GAMR,MAAM,SAAS,CAAC;AACjB,OAAO,EAAE,wBAAwB,EAAE,MAAM,yBAAyB,CAAC;AAEnE,MAAM,GAAG,GAAG,EAAE,IAAI,EAAE,uBAAuB,EAAE,KAAK,EAAE,CAAC,EAAE,GAAG,EAAE,CAAC,EAAE,CAAC;AAChE,MAAM,MAAM,GAAG,CAAC,KAAa,EAAU,EAAE,CAAC,CAAC,EAAE,IAAI,EAAE,QAAQ,EAAE,KAAK,EAAE,IAAI,EAAE,GAAG,EAAE,GAAG,EAAE,CAAC,CAAC;AACtF,MAAM,GAAG,GAAG,CAAC,OAAe,EAAU,EAAE,CAAC,CAAC,EAAE,IAAI,EAAE,QAAQ,EAAE,OAAO,EAAE,IAAI,EAAE,GAAG,EAAE,GAAG,EAAE,CAAC,CAAC;AACvF,MAAM,KAAK,GAAG,EAAE,IAAI,EAAE,QAAQ,EAAE,OAAO,EAAE,GAAG,EAAE,IAAI,EAAE,OAAO,CAAC,GAAG,CAAC,EAAE,GAAG,EAAW,CAAC;AACjF,MAAM,OAAO,GAAG,CAAC,KAAa,EAAU,EAAE,CAAC,CAAC;IAC1C,IAAI,EAAE,UAAU;IAChB,IAAI,EAAE,EAAE,IAAI,EAAE,UAAU,EAAE,GAAG,EAAE,KAAK,EAAE,KAAK,EAAE,IAAI,EAAE,GAAG,EAAE,GAAG,EAAE;IAC7D,GAAG;CACJ,CAAC,CAAC;AACH,MAAM,IAAI,GAAG,CAAC,EAAU,EAAE,EAAa,EAAU,EAAE,CAAC,CAAC;IACnD,IAAI,EAAE,QAAQ;IACd,OAAO,EAAE,EAAE;IACX,KAAK,EAAE,EAAE,IAAI,EAAE,KAAK,EAAE,EAAE,EAAE,IAAI,EAAE,GAAG,CAAC,EAAE,CAAC,EAAE,KAAK,EAAE,MAAM,CAAC,CAAC,CAAC,EAAE,IAAI,EAAE,GAAG,EAAE,GAAG,EAAE;IAC3E,GAAG;CACJ,CAAC,CAAC;AACH,MAAM,OAAO,GAAG,CAAC,EAAU,EAAE,IAAY,EAAU,EAAE,CAAC,CAAC,EAAE,IAAI,EAAE,SAAS,EAAE,OAAO,EAAE,EAAE,EAAE,IAAI,EAAE,GAAG,EAAE,CAAC,CAAC;AACpG,MAAM,OAAO,GAAG,CAAC,EAAU,EAAW,EAAE,CAAC,CAAC;IACxC,EAAE;IACF,IAAI,EAAE,EAAE,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC,CAAC,CAAE;IACvB,IAAI,EAAE,GAAG;IACT,OAAO,EAAE,IAAI;CACd,CAAC,CAAC;AAEH,SAAS,SAAS,CAChB,IAAI,GAAa,CAAC,OAAO,CAAC,GAAG,CAAC,MAAM,CAAC,CAAC,EAAE,OAAO,CAAC,GAAG,CAAC,MAAM,CAAC,CAAC,CAAC,EAC7D,OAAO,GAAa,CAAC,IAAI,CAAC,MAAM,EAAE,GAAG,CAAC,EAAE,IAAI,CAAC,MAAM,EAAE,GAAG,CAAC,CAAC;IAE1D,OAAO;QACL,IAAI,EAAE,OAAO;QACb,IAAI,EAAE,EAAE,IAAI,EAAE,KAAK,EAAE,EAAE,EAAE,GAAG,EAAE,IAAI,EAAE,GAAG,CAAC,MAAM,CAAC,EAAE,KAAK,EAAE,GAAG,CAAC,MAAM,CAAC,EAAE,IAAI,EAAE,IAAI,EAAE,GAAG,EAAE;QACtF,IAAI,EAAE,CAAC,GAAG,IAAI,EAAE,GAAG,OAAO,CAAC;QAC3B,GAAG;KACJ,CAAC;AACJ,CAAC;AAED,SAAS,OAAO,CAAC,IAAc,EAAE,KAAK,GAAwB,EAAE;IAC9D,MAAM,EAAE,GAAe;QACrB,IAAI,EAAE,GAAG;QACT,MAAM,EAAE,CAAC,EAAE,OAAO,EAAE,KAAK,EAAE,IAAI,EAAE,GAAG,EAAE,IAAI,EAAE,GAAG,EAAE,CAAC;QAClD,UAAU,EAAE,GAAG;QACf,MAAM,EAAE;YACN,OAAO,CAAC,KAAK,CAAC;YACd,OAAO,CAAC,MAAM,CAAC;YACf,OAAO,CAAC,MAAM,CAAC;YACf,EAAE,EAAE,EAAE,GAAG,EAAE,IAAI,EAAE,GAAG,EAAE,IAAI,EAAE,OAAO,CAAC,GAAG,CAAC,EAAE,OAAO,EAAE,KAAK,EAAE;SAC3D;QACD,IAAI;QACJ,GAAG;QACH,GAAG,KAAK;KACT,CAAC;IACF,OAAO,EAAE,SAAS,EAAE,CAAC,EAAE,CAAC,EAAE,OAAO,EAAE,EAAE,EAAyB,CAAC;AACjE,CAAC;AAED,MAAM,SAAS,GAAG,CAAC,GAAa,EAAY,EAAE,CAAC,wBAAwB,CAAC,GAAG,CAAC,CAAC,SAAS,CAAC,CAAC,CAAE,CAAC,IAAI,CAAC;AAEhG,IAAI,CAAC,+DAA+D,EAAE,GAAG,EAAE;IACzE,MAAM,GAAG,GAAG,OAAO,CAAC,CAAC,OAAO,CAAC,MAAM,EAAE,MAAM,CAAC,CAAC,CAAC,CAAC,EAAE,OAAO,CAAC,MAAM,EAAE,GAAG,CAAC,KAAK,CAAC,CAAC,EAAE,SAAS,EAAE,CAAC,CAAC,CAAC;IAC5F,MAAM,GAAG,GAAG,wBAAwB,CAAC,GAAG,CAAC,CAAC,SAAS,CAAC,CAAC,CAAE,CAAC;IACxD,uEAAuE;IACvE,iEAAiE;IACjE,MAAM,CAAC,GAAG,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,CAAC,OAAO,CAAC,GAAG,CAAC,SAAS,CAAC,CAAC,CAAE,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,CAAC;IACzE,MAAM,CAAC,GAAG,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,CAAC,aAAa,CAAC;QAChC,IAAI,EAAE,KAAK;QACX,IAAI,EAAE;YACJ,IAAI,EAAE,OAAO;YACb,IAAI,EAAE;gBACJ,EAAE,IAAI,EAAE,SAAS,EAAE,OAAO,EAAE,MAAM,EAAE,IAAI,EAAE,EAAE,IAAI,EAAE,QAAQ,EAAE,OAAO,EAAE,MAAM,EAAE,EAAE;gBAC/E,EAAE,IAAI,EAAE,SAAS,EAAE,OAAO,EAAE,MAAM,EAAE,IAAI,EAAE,EAAE,IAAI,EAAE,QAAQ,EAAE,OAAO,EAAE,MAAM,EAAE,EAAE;aAChF;SACF;QACD,IAAI,EAAE,EAAE,EAAE,EAAE,GAAG,EAAE,IAAI,EAAE,EAAE,OAAO,EAAE,MAAM,EAAE,EAAE,KAAK,EAAE,EAAE,OAAO,EAAE,MAAM,EAAE,EAAE;QACxE,MAAM,EAAE,EAAE,IAAI,EAAE,OAAO,EAAE,IAAI,EAAE,CAAC,EAAE,OAAO,EAAE,MAAM,EAAE,EAAE,EAAE,OAAO,EAAE,MAAM,EAAE,CAAC,EAAE;QAC3E,IAAI,EAAE;YACJ,EAAE,IAAI,EAAE,EAAE,IAAI,EAAE,UAAU,EAAE,KAAK,EAAE,EAAE,OAAO,EAAE,MAAM,EAAE,EAAE,EAAE;YAC1D,EAAE,IAAI,EAAE,EAAE,IAAI,EAAE,UAAU,EAAE,KAAK,EAAE,EAAE,OAAO,EAAE,MAAM,EAAE,EAAE,EAAE;SAC3D;KACF,CAAC,CAAC;IACH,MAAM,CAAC,GAAG,CAAC,MAAM,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,OAAO,CAAC,CAAC,KAAK,EAAE,MAAM,EAAE,MAAM,EAAE,GAAG,EAAE,MAAM,EAAE,MAAM,CAAC,CAAC,CAAC;AAC5F,CAAC,CAAC,CAAC;AAEH,IAAI,CAAC,0EAA0E,EAAE,GAAG,EAAE;IACpF,MAAM,IAAI,GAAG,SAAS,CAAC,SAAS,EAAE,CAAC,IAAI,CAAC,MAAM,EAAE,GAAG,CAAC,EAAE,IAAI,CAAC,MAAM,EAAE,GAAG,CAAC,CAAC,CAAC,CAAC;IAC1E,IAAI,CAAC,IAAI,GAAG,EAAE,IAAI,EAAE,KAAK,EAAE,EAAE,EAAE,IAAI,EAAE,IAAI,EAAE,GAAG,CAAC,MAAM,CAAC,EAAE,KAAK,EAAE,GAAG,CAAC,MAAM,CAAC,EAAE,IAAI,EAAE,IAAI,EAAE,GAAG,EAAE,CAAC;IAC9F,IAAI,CAAC,MAAM,GAAG,CAAC,OAAO,CAAC,CAAC;IACxB,MAAM,GAAG,GAAG,OAAO,CAAC,CAAC,OAAO,CAAC,MAAM,EAAE,MAAM,CAAC,CAAC,CAAC,CAAC,EAAE,IAAI,CAAC,EAAE;QACtD,MAAM,EAAE,CAAC,EAAE,OAAO,EAAE,MAAM,EAAE,IAAI,EAAE,IAAI,EAAE,IAAI,EAAE,GAAG,EAAE,CAAC;KACrD,CAAC,CAAC;IACH,MAAM,CAAC,SAAS,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,aAAa,CAAC;QACtC,IAAI,EAAE,KAAK;QACX,MAAM,EAAE,CAAC,OAAO,CAAC;QACjB,IAAI,EAAE,EAAE,EAAE,EAAE,IAAI,EAAE,IAAI,EAAE,EAAE,OAAO,EAAE,MAAM,EAAE,EAAE,KAAK,EAAE,EAAE,OAAO,EAAE,MAAM,EAAE,EAAE;QACzE,MAAM,EAAE,EAAE,IAAI,EAAE,CAAC,EAAE,OAAO,EAAE,MAAM,EAAE,EAAE,EAAE,OAAO,EAAE,MAAM,EAAE,CAAC,EAAE;KAC7D,CAAC,CAAC;AACL,CAAC,CAAC,CAAC;AAEH,IAAI,CAAC,6EAA6E,EAAE,GAAG,EAAE;IACvF,MAAM,KAAK,GAAG,CAAC,OAAO,CAAC,MAAM,EAAE,MAAM,CAAC,CAAC,CAAC,CAAC,EAAE,OAAO,CAAC,MAAM,EAAE,GAAG,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC;IACxE,MAAM,IAAI,GAAG,CAAC,IAAc,EAAE,KAAK,GAAwB,EAAE,EAAQ,EAAE;QACrE,MAAM,GAAG,GAAG,OAAO,CAAC,IAAI,EAAE,KAAK,CAAC,CAAC;QACjC,MAAM,CAAC,wBAAwB,CAAC,GAAG,CAAC,CAAC,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC;IAClD,CAAC,CAAC;IACF,gEAAgE;IAChE,IAAI,CAAC,CAAC,GAAG,KAAK,EAAE,SAAS,CAAC,CAAC,EAAE,IAAI,EAAE,UAAU,EAAE,GAAG,EAAE,EAAE,OAAO,CAAC,GAAG,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC;IAC/E,MAAM,OAAO,GAAG,SAAS,CAAC,CAAC,EAAE,IAAI,EAAE,UAAU,EAAE,KAAK,EAAE,OAAO,EAAE,GAAG,EAAE,EAAE,OAAO,CAAC,GAAG,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC,CAAC;IAC7F,OAAO,CAAC,MAAM,GAAG,CAAC,OAAO,CAAC,CAAC;IAC3B,IAAI,CAAC,CAAC,GAAG,KAAK,EAAE,OAAO,CAAC,CAAC,CAAC;IAC1B,uEAAuE;IACvE,IAAI,CAAC,CAAC,GAAG,KAAK,EAAE,SAAS,EAAE,EAAE,EAAE,IAAI,EAAE,QAAQ,EAAE,KAAK,EAAE,GAAG,CAAC,MAAM,CAAC,EAAE,GAAG,EAAE,CAAC,CAAC,CAAC;IAC3E,IAAI,CAAC,CAAC,GAAG,KAAK,EAAE,IAAI,CAAC,MAAM,EAAE,GAAG,CAAC,EAAE,SAAS,EAAE,CAAC,CAAC,CAAC;IACjD,IAAI,CAAC,CAAC,GAAG,KAAK,EAAE,OAAO,CAAC,MAAM,EAAE,MAAM,CAAC,CAAC,CAAC,CAAC,EAAE,SAAS,EAAE,CAAC,CAAC,CAAC;IAC1D,sEAAsE;IACtE,yCAAyC;IACzC,IAAI,CAAC;QACH,GAAG,KAAK;QACR;YACE,IAAI,EAAE,OAAO;YACb,IAAI,EAAE,EAAE,IAAI,EAAE,SAAS,EAAE,KAAK,EAAE,IAAI,EAAE,IAAI,EAAE,IAAI,EAAE,GAAG,EAAE;YACvD,IAAI,EAAE,CAAC,SAAS,EAAE,CAAC;YACnB,GAAG;SACJ;KACF,CAAC,CAAC;IACH,gEAAgE;IAChE,MAAM,SAAS,GAAW;QACxB,IAAI,EAAE,OAAO;QACb,IAAI,EAAE,EAAE,IAAI,EAAE,SAAS,EAAE,KAAK,EAAE,IAAI,EAAE,IAAI,EAAE,IAAI,EAAE,GAAG,EAAE;QACvD,IAAI,EAAE,CAAC,SAAS,EAAE,CAAC;QACnB,GAAG;KACJ,CAAC;IACF,IAAI,CAAC,CAAC,SAAS,CAAC,EAAE;QAChB,MAAM,EAAE;YACN,EAAE,OAAO,EAAE,MAAM,EAAE,IAAI,EAAE,IAAI,EAAE,IAAI,EAAE,GAAG,EAAE;YAC1C,EAAE,OAAO,EAAE,MAAM,EAAE,IAAI,EAAE,IAAI,EAAE,IAAI,EAAE,GAAG,EAAE;SAC3C;KACF,CAAC,CAAC;IACH,qEAAqE;IACrE,IAAI,CAAC,CAAC,OAAO,CAAC,MAAM,EAAE,MAAM,CAAC,CAAC,CAAC,CAAC,EAAE,SAAS,EAAE,CAAC,CAAC,CAAC;IAChD,IAAI,CAAC,CAAC,GAAG,KAAK,EAAE,SAAS,EAAE,CAAC,EAAE,EAAE,QAAQ,EAAE,CAAC,EAAE,OAAO,EAAE,MAAM,EAAE,IAAI,EAAE,IAAI,EAAE,IAAI,EAAE,GAAG,EAAE,CAAC,EAAE,CAAC,CAAC;IAC1F,4EAA4E;IAC5E,IAAI,CAAC,CAAC,GAAG,KAAK,EAAE,SAAS,CAAC,CAAC,OAAO,CAAC,GAAG,CAAC,MAAM,CAAC,CAAC,EAAE,IAAI,CAAC,MAAM,EAAE,GAAG,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC;IACvE,MAAM,KAAK,GAAG,OAAO,CAAC,CAAC,GAAG,KAAK,EAAE,SAAS,EAAE,CAAC,CAAC,CAAC;IAC/C,KAAK,CAAC,SAAS,CAAC,CAAC,CAAE,CAAC,MAAM,CAAC,CAAC,CAAC,GAAG,EAAE,GAAG,OAAO,CAAC,MAAM,CAAC,EAAE,KAAK,EAAE,IAAI,EAAE,CAAC;IACpE,MAAM,CAAC,wBAAwB,CAAC,KAAK,CAAC,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;IACpD,IAAI,CAAC,CAAC,GAAG,KAAK,EAAE,SAAS,CAAC,EAAE,CAAC,CAAC,CAAC,CAAC;AAClC,CAAC,CAAC,CAAC;AAEH,IAAI,CAAC,2DAA2D,EAAE,GAAG,EAAE;IACrE,MAAM,KAAK,GAAW;QACpB,IAAI,EAAE,OAAO;QACb,IAAI,EAAE,EAAE,IAAI,EAAE,SAAS,EAAE,KAAK,EAAE,KAAK,EAAE,IAAI,EAAE,IAAI,EAAE,GAAG,EAAE;QACxD,IAAI,EAAE,CAAC,EAAE,IAAI,EAAE,UAAU,EAAE,GAAG,EAAE,CAAC;QACjC,GAAG;KACJ,CAAC;IACF,wEAAwE;IACxE,0BAA0B;IAC1B,MAAM,GAAG,GAAG,OAAO,CAAC;QAClB,OAAO,CAAC,MAAM,EAAE,MAAM,CAAC,CAAC,CAAC,CAAC;QAC1B,OAAO,CAAC,MAAM,EAAE,MAAM,CAAC,CAAC,CAAC,CAAC;QAC1B,SAAS,CAAC,CAAC,KAAK,EAAE,OAAO,CAAC,GAAG,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC;KACzC,CAAC,CAAC;IACH,MAAM,CAAC,SAAS,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,aAAa,CAAC,EAAE,IAAI,EAAE,KAAK,EAAE,IAAI,EAAE,CAAC,EAAE,IAAI,EAAE,OAAO,EAAE,EAAE,EAAE,CAAC,EAAE,CAAC,CAAC;IACxF,sEAAsE;IACtE,wBAAwB;IACxB,MAAM,OAAO,GAAG,OAAO,CAAC;QACtB,OAAO,CAAC,MAAM,EAAE,MAAM,CAAC,CAAC,CAAC,CAAC;QAC1B,OAAO,CAAC,MAAM,EAAE,GAAG,CAAC,KAAK,CAAC,CAAC;QAC3B,SAAS,CAAC,CAAC,KAAK,EAAE,OAAO,CAAC,GAAG,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC;KACzC,CAAC,CAAC;IACH,MAAM,CAAC,wBAAwB,CAAC,OAAO,CAAC,CAAC,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC;AAC1D,CAAC,CAAC,CAAC"}

package/dist/ir/parameter-versions.d.ts

@@ -0,0 +1,25 @@
+import { type IrModule } from "./ir.js";
+/** Versions of functions whose number parameters bound their loops.
+ *
+ * A parameter that every call site proves an int32 already reaches the
+ * function's i32 entry (int32-abi.ts). One that some caller passes as an
+ * unproven number (a value read from a message, a field, a parse) keeps
+ * its double, so the loops it bounds compare a counter with a double and
+ * the values derived from it stay doubles.
+ *
+ * A parameter qualifies when it bounds a numeric loop (`numericLoop`).
+ * Such a function gets a copy of its body under a new name, and its own
+ * body starts by calling the copy when every such parameter holds an
+ * int32: `(p | 0) === p` (an integer in the int32 range, not NaN) and not
+ * -0 (`1 / p` is not -Infinity). The copy then receives `p | 0`, which is
+ * the same double, so it computes what the body would; its only caller
+ * proves the parameter an int32, so it gets the i32 entry. Other values
+ * run the body as before.
+ *
+ * Only plain synchronous functions qualify: not class members, no
+ * captures, not the module entry, not dynamic-tier, at most
+ * MAX_FUNCTION_NODES nodes, not calling themselves, and not holding the
+ * only call site of another function (`keepsSingleBody`). The caller
+ * skips modules that observe call stacks, where the copy's frame could be
+ * seen. */
+export declare function versionLoopBoundParameters(mod: IrModule, symbolName?: (name: string) => string): IrModule;