@aardworx/wombat.rendering 0.9.33 → 0.18.1

package/dist/runtime/derivedModes/kernelCodegen.d.ts

@@ -0,0 +1,90 @@
+import type { Expr, Stmt, IntrinsicEvalTable } from "@aardworx/wombat.shader/ir";
+import type { ModeAxis } from "./rule.js";
+/** One traced+rewritten axis-rule body, ready for codegen. */
+export interface RuleCodegenSpec {
+    readonly axis: ModeAxis;
+    /** Unique within `axis` within the bucket. */
+    readonly axisRuleId: number;
+    /**
+     * The rule body — already rewritten so every `ReturnValue` returns
+     * a `Const u32` PER-AXIS index into this axis's union value table.
+     */
+    readonly body: Stmt;
+    readonly intrinsics: IntrinsicEvalTable;
+    /** Per-rule WGSL helpers (concatenated into the kernel). */
+    readonly helpersWGSL: string;
+    readonly inputUniforms: ReadonlyArray<string>;
+}
+/** Per-active-axis entry inside a single combo: either picks a rule
+ *  (axis index computed at runtime by `rule_<axis>_<id>`) or a fixed
+ *  per-axis index (a compile-time constant for axes the combo has no
+ *  rule on — the index of baseDescriptor's value in this axis's
+ *  union table). */
+export type ComboAxisSource = {
+    readonly kind: "rule";
+    readonly axisRuleId: number;
+} | {
+    readonly kind: "const";
+    readonly idx: number;
+};
+export interface ComboAxis {
+    readonly axis: ModeAxis;
+    /** |axisValues[axis]| — the stride for mixed-radix encoding is the
+     *  product of cardinalities of axes after this one in `axes`. */
+    readonly cardinality: number;
+    readonly source: ComboAxisSource;
+}
+export interface ComboCodegenSpec {
+    readonly comboId: number;
+    /** Active axes (those with at least one rule in the bucket), in
+     *  canonical order. Every combo in the bucket carries the same
+     *  axis list — combos differ only in which axes have rules. */
+    readonly axes: ReadonlyArray<ComboAxis>;
+}
+export interface KernelCodegenSpec {
+    /** Distinct axis-rules registered on the bucket. */
+    readonly rules: ReadonlyArray<RuleCodegenSpec>;
+    /** Distinct combos appearing on ROs in this bucket. */
+    readonly combos: ReadonlyArray<ComboCodegenSpec>;
+    /** Total slot count = ∏ cardinality across active axes. */
+    readonly totalSlots: number;
+    /** Bucket-wide ordered list of uniform names that any rule body
+     *  reads (`ReadInput("Uniform", name)`). Each one becomes a
+     *  `refK: u32` field in the Record struct + a tail slot in master
+     *  records; rules dereference their named reads as `r.refK`. */
+    readonly uniformOrder: ReadonlyArray<string>;
+}
+/** Emit the WGSL partition kernel for the given rules + combos.
+ *  Per-record dispatch:
+ *    `switch (r.comboId) { case 0u: { return combo_0(r); } … }`
+ *  Each `combo_<id>(r)` computes per-axis indices (calling
+ *  `rule_<axis>_<axisRuleId>(r)` for axes the combo has a rule on,
+ *  or using a const for the others) and composes them into a
+ *  global slot index via mixed-radix encoding over `axes`. */
+export declare function emitPartitionKernel(spec: KernelCodegenSpec): string;
+/**
+ * Replace every `ReadInput("<scope>", "<name>")` leaf in `e` with
+ * the supplied `replacement` Expr. Used to bake the rule's
+ * `declared` ReadInput as a `Const u32` of the current declared
+ * value before codegen — the resulting body has no `declared`
+ * references, so the kernel doesn't need a per-dispatch uniform.
+ */
+export declare function substituteReadInput(e: Expr, scope: string, name: string, replacement: Expr): Expr;
+export declare function substituteReadInputInStmt(s: Stmt, scope: string, name: string, replacement: Expr): Stmt;
+/** Walk `s` and collect the names of every `ReadInput("Uniform", name)`
+ *  it contains (in DFS order, deduped). heapScene uses this to size
+ *  the bucket's `uniformOrder` + master-record uniform-ref tail. */
+export declare function collectUniformReadsInStmt(s: Stmt): string[];
+/**
+ * Rewrite the body so every `ReturnValue` returns a `Const u32` slot
+ * index (resolved by `exprToSlot`). `Conditional` expressions are
+ * recursed into — each branch is rewritten independently. Other
+ * non-Conditional return expressions are matched as a whole against
+ * `exprToSlot` (using stableStringify-content equality upstream).
+ *
+ * After this pass the kernel emits a u32 directly from
+ * `rule_<axis>(r)`; the partition entry switches on it as a slot
+ * index, no SLOT_LOOKUP table needed.
+ */
+export declare function rewriteOutputsToSlotIndices(body: Stmt, exprToSlot: (e: Expr) => number | undefined): Stmt;
+//# sourceMappingURL=kernelCodegen.d.ts.map

package/dist/runtime/derivedModes/kernelCodegen.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"kernelCodegen.d.ts","sourceRoot":"","sources":["../../../src/runtime/derivedModes/kernelCodegen.ts"],"names":[],"mappings":"AAqBA,OAAO,KAAK,EAAE,IAAI,EAAE,IAAI,EAAiB,kBAAkB,EAAE,MAAM,4BAA4B,CAAC;AAEhG,OAAO,KAAK,EAAE,QAAQ,EAAE,MAAM,WAAW,CAAC;AAE1C,8DAA8D;AAC9D,MAAM,WAAW,eAAe;IAC9B,QAAQ,CAAC,IAAI,EAAE,QAAQ,CAAC;IACxB,8CAA8C;IAC9C,QAAQ,CAAC,UAAU,EAAE,MAAM,CAAC;IAC5B;;;OAGG;IACH,QAAQ,CAAC,IAAI,EAAE,IAAI,CAAC;IACpB,QAAQ,CAAC,UAAU,EAAE,kBAAkB,CAAC;IACxC,4DAA4D;IAC5D,QAAQ,CAAC,WAAW,EAAE,MAAM,CAAC;IAC7B,QAAQ,CAAC,aAAa,EAAE,aAAa,CAAC,MAAM,CAAC,CAAC;CAC/C;AAED;;;;oBAIoB;AACpB,MAAM,MAAM,eAAe,GACvB;IAAE,QAAQ,CAAC,IAAI,EAAE,MAAM,CAAC;IAAC,QAAQ,CAAC,UAAU,EAAE,MAAM,CAAA;CAAE,GACtD;IAAE,QAAQ,CAAC,IAAI,EAAE,OAAO,CAAC;IAAC,QAAQ,CAAC,GAAG,EAAE,MAAM,CAAA;CAAE,CAAC;AAErD,MAAM,WAAW,SAAS;IACxB,QAAQ,CAAC,IAAI,EAAE,QAAQ,CAAC;IACxB;qEACiE;IACjE,QAAQ,CAAC,WAAW,EAAE,MAAM,CAAC;IAC7B,QAAQ,CAAC,MAAM,EAAE,eAAe,CAAC;CAClC;AAED,MAAM,WAAW,gBAAgB;IAC/B,QAAQ,CAAC,OAAO,EAAE,MAAM,CAAC;IACzB;;mEAE+D;IAC/D,QAAQ,CAAC,IAAI,EAAE,aAAa,CAAC,SAAS,CAAC,CAAC;CACzC;AAED,MAAM,WAAW,iBAAiB;IAChC,oDAAoD;IACpD,QAAQ,CAAC,KAAK,EAAE,aAAa,CAAC,eAAe,CAAC,CAAC;IAC/C,uDAAuD;IACvD,QAAQ,CAAC,MAAM,EAAE,aAAa,CAAC,gBAAgB,CAAC,CAAC;IACjD,2DAA2D;IAC3D,QAAQ,CAAC,UAAU,EAAE,MAAM,CAAC;IAC5B;;;oEAGgE;IAChE,QAAQ,CAAC,YAAY,EAAE,aAAa,CAAC,MAAM,CAAC,CAAC;CAC9C;AAED;;;;;;8DAM8D;AAC9D,wBAAgB,mBAAmB,CAAC,IAAI,EAAE,iBAAiB,GAAG,MAAM,CAwHnE;AA8FD;;;;;;GAMG;AACH,wBAAgB,mBAAmB,CACjC,CAAC,EAAE,IAAI,EAAE,KAAK,EAAE,MAAM,EAAE,IAAI,EAAE,MAAM,EAAE,WAAW,EAAE,IAAI,GACtD,IAAI,CAQN;AACD,wBAAgB,yBAAyB,CACvC,CAAC,EAAE,IAAI,EAAE,KAAK,EAAE,MAAM,EAAE,IAAI,EAAE,MAAM,EAAE,WAAW,EAAE,IAAI,GACtD,IAAI,CAEN;AAED;;oEAEoE;AACpE,wBAAgB,yBAAyB,CAAC,CAAC,EAAE,IAAI,GAAG,MAAM,EAAE,CAU3D;AAsCD;;;;;;;;;;GAUG;AACH,wBAAgB,2BAA2B,CACzC,IAAI,EAAE,IAAI,EACV,UAAU,EAAE,CAAC,CAAC,EAAE,IAAI,KAAK,MAAM,GAAG,SAAS,GAC1C,IAAI,CAqDN"}