@aardworx/wombat.rendering 0.8.1 → 0.9.1

package/dist/runtime/heapEffect.js

@@ -0,0 +1,529 @@
+// heapEffect — effect introspection + bucket-layout generation.
+//
+// Two stages:
+//
+//   1. `compileHeapEffect(effect)` — runs `effect.compile({target:"wgsl"})`
+//      and returns the raw stage WGSL (untouched) plus a `schema`
+//      describing the effect's surviving inputs / uniforms / outputs
+//      (post link + DCE). No name hardcoding; everything is read off
+//      `CompiledEffect.interface`.
+//
+//   2. `buildBucketLayout(schema, hasTextures, opts)` — turns a schema
+//      into a concrete per-bucket WGSL prelude + a JS-side layout
+//      object that knows where each named uniform / attribute lives
+//      (struct field offset for packing). The IR rewriter in
+//      `heapEffectIR.ts` consumes this layout to substitute uniform /
+//      attribute reads with heap loads at the IR level.
+export function compileHeapEffect(effect, fragmentOutputLayout) {
+    const compiled = effect.compile(fragmentOutputLayout !== undefined
+        ? { target: "wgsl", fragmentOutputLayout }
+        : { target: "wgsl" });
+    const vsStage = compiled.stages.find(s => s.stage === "vertex");
+    const fsStage = compiled.stages.find(s => s.stage === "fragment");
+    return {
+        rawVs: vsStage?.source ?? "",
+        rawFs: fsStage?.source ?? "",
+        schema: buildSchema(compiled.interface),
+    };
+}
+function buildSchema(iface) {
+    const attributes = iface.attributes.map(a => ({
+        name: a.name,
+        wgslType: irTypeToWgsl(a.type),
+        byteSize: a.byteSize,
+        location: a.location,
+    }));
+    // When `instanceUniforms` (or any other rewrite) converts a uniform
+    // read into a vertex-input fetch, the source module still carries the
+    // original Uniform declaration. The interface extractor surfaces it
+    // here, but the field is now satisfied by the per-instance attribute —
+    // collecting it as a uniform too would double-book the name in the
+    // drawHeader. Drop uniforms shadowed by an attribute of the same name.
+    const attributeNames = new Set(attributes.map(a => a.name));
+    const uniforms = [];
+    for (const block of iface.uniformBlocks) {
+        for (const f of block.fields) {
+            if (attributeNames.has(f.name))
+                continue;
+            uniforms.push({
+                name: f.name,
+                wgslType: irTypeToWgsl(f.type),
+                byteSize: irTypePackedSize(f.type),
+            });
+        }
+    }
+    // GLSL-style "loose" uniforms (no enclosing buffer name). On WGSL these
+    // emit as individual `var<uniform>` decls; the heap IR rewrite substitutes
+    // each `ReadInput("Uniform", name, ...)` with a per-draw heap load before
+    // WGSL emit, so the loose decls never reach the final shader.
+    for (const u of iface.uniforms) {
+        if (attributeNames.has(u.name))
+            continue;
+        uniforms.push({
+            name: u.name,
+            wgslType: irTypeToWgsl(u.type),
+            byteSize: irTypePackedSize(u.type),
+        });
+    }
+    // Inter-stage varyings: VS outputs (post link + DCE). The FS inputs
+    // mirror these — `linkCrossStage` aligns them by semantic, so we
+    // can use VS outputs as the single source of truth for VsOut.
+    const vsStage = iface.stages.find(s => s.stage === "vertex");
+    const varyings = vsStage === undefined
+        ? []
+        : vsStage.outputs.map(o => {
+            const v = { name: o.name, wgslType: irTypeToWgsl(o.type) };
+            if (o.location !== undefined)
+                v.location = o.location;
+            if (o.builtin !== undefined)
+                v.builtin = o.builtin;
+            if (o.interpolation !== undefined)
+                v.interpolation = o.interpolation;
+            return v;
+        });
+    const fragmentOutputs = iface.fragmentOutputs.map(o => ({
+        name: o.name, location: o.location, wgslType: irTypeToWgsl(o.type),
+    }));
+    // Texture/sampler bindings post-DCE. The IR Type for textures is a
+    // structured kind we don't fully model here yet — for v1 every
+    // texture maps to "texture_2d<f32>" and every sampler to plain
+    // "sampler". Specialise as multi-kind support lands in the DSL.
+    const textures = iface.textures.map(t => ({
+        name: t.name, wgslType: "texture_2d<f32>",
+    }));
+    const samplers = iface.samplers.map(s => ({
+        name: s.name, wgslType: "sampler",
+    }));
+    return { attributes, uniforms, varyings, fragmentOutputs, textures, samplers };
+}
+function irTypeToWgsl(t) {
+    switch (t.kind) {
+        case "Float": return "f32";
+        case "Int": return t.signed ? "i32" : "u32";
+        case "Bool": return "bool";
+        case "Vector": return `vec${t.dim}<${irTypeToWgsl(t.element)}>`;
+        case "Matrix": return `mat${t.cols}x${t.rows}<${irTypeToWgsl(t.element)}>`;
+        case "Array": {
+            const inner = irTypeToWgsl(t.element);
+            return t.length === "runtime" ? `array<${inner}>` : `array<${inner}, ${t.length}>`;
+        }
+        default:
+            throw new Error(`compileHeapEffect: cannot map IR Type kind '${t.kind}' to WGSL`);
+    }
+}
+function irTypePackedSize(t) {
+    switch (t.kind) {
+        case "Float": return 4;
+        case "Int": return 4;
+        case "Bool": return 4;
+        case "Vector": return t.dim * irTypePackedSize(t.element);
+        case "Matrix": return t.cols * t.rows * irTypePackedSize(t.element);
+        case "Array": {
+            if (t.length === "runtime") {
+                throw new Error("compileHeapEffect: runtime-sized arrays have no fixed packed size");
+            }
+            return t.length * irTypePackedSize(t.element);
+        }
+        default:
+            throw new Error(`compileHeapEffect: cannot size IR Type kind '${t.kind}'`);
+    }
+}
+/**
+ * First texture binding number; samplers come after the textures.
+ * Bindings 0–3 are the four heap arena views, 4–6 are megacall
+ * (drawTable, indexStorage, firstDrawInTile).
+ */
+export const HEAP_TEX_BINDING_START = 7;
+/**
+ * Number of independent pages per format. Drives both the BGL slot
+ * count (N consecutive linear bindings + N consecutive srgb bindings
+ * + 1 sampler) and the shader's `switch pageRef` ladder. Must match
+ * `ATLAS_MAX_PAGES_PER_FORMAT` in `textureAtlas/atlasPool.ts`.
+ */
+export const ATLAS_ARRAY_SIZE = 8;
+/** First atlas binding (linear texture for page 0). */
+export const ATLAS_LINEAR_BINDING_BASE = 11;
+/** First srgb atlas binding (= linear base + N). */
+export const ATLAS_SRGB_BINDING_BASE = ATLAS_LINEAR_BINDING_BASE + ATLAS_ARRAY_SIZE;
+/** Atlas sampler binding (= linear base + 2N). */
+export const ATLAS_SAMPLER_BINDING = ATLAS_LINEAR_BINDING_BASE + 2 * ATLAS_ARRAY_SIZE;
+export function buildBucketLayout(schema, _hasTextures, opts = {}) {
+    const perInstanceUniforms = opts.perInstanceUniforms ?? new Set();
+    const perInstanceAttributes = opts.perInstanceAttributes ?? new Set();
+    const atlasTextureBindings = opts.atlasTextureBindings ?? new Set();
+    const drawHeaderFields = [];
+    // Per-draw uniforms → u32 ref slots in the DrawHeader. The runtime
+    // (UniformPool) keeps one arena allocation per unique aval; sharing
+    // across draws → shared ref → shared upload.
+    let dhOff = 0;
+    for (const u of schema.uniforms) {
+        dhOff = roundUp(dhOff, 4);
+        drawHeaderFields.push({
+            name: u.name, wgslName: lowerFirst(u.name) + "Ref", wgslType: "u32",
+            byteOffset: dhOff, byteSize: 4, kind: "uniform-ref",
+            uniformWgslType: u.wgslType,
+        });
+        dhOff += 4;
+    }
+    // Attribute refs — same shape as uniform refs, also routed through the pool.
+    for (const a of schema.attributes) {
+        dhOff = roundUp(dhOff, 4);
+        drawHeaderFields.push({
+            name: a.name, wgslName: lowerFirst(a.name) + "Ref", wgslType: "u32",
+            byteOffset: dhOff, byteSize: 4, kind: "attribute-ref",
+            attributeWgslType: a.wgslType,
+        });
+        dhOff += 4;
+    }
+    // Atlas-routed texture bindings: 24 bytes / binding split into
+    // pageRef (u32) + formatBits (u32) + origin (vec2<f32>) + size
+    // (vec2<f32>). Inline values, not pool refs — the runtime packs
+    // them at addDraw time from the atlas acquisition + ISampler.
+    for (const t of schema.textures) {
+        if (!atlasTextureBindings.has(t.name))
+            continue;
+        const base = lowerFirst(t.name);
+        dhOff = roundUp(dhOff, 4);
+        drawHeaderFields.push({
+            name: `${t.name}.pageRef`, wgslName: `${base}PageRef`, wgslType: "u32",
+            byteOffset: dhOff, byteSize: 4, kind: "texture-ref",
+            textureSub: "pageRef", textureBindingName: t.name,
+        });
+        dhOff += 4;
+        drawHeaderFields.push({
+            name: `${t.name}.formatBits`, wgslName: `${base}FormatBits`, wgslType: "u32",
+            byteOffset: dhOff, byteSize: 4, kind: "texture-ref",
+            textureSub: "formatBits", textureBindingName: t.name,
+        });
+        dhOff += 4;
+        dhOff = roundUp(dhOff, 8);
+        drawHeaderFields.push({
+            name: `${t.name}.origin`, wgslName: `${base}Origin`, wgslType: "vec2<f32>",
+            byteOffset: dhOff, byteSize: 8, kind: "texture-ref",
+            textureSub: "origin", textureBindingName: t.name,
+        });
+        dhOff += 8;
+        drawHeaderFields.push({
+            name: `${t.name}.size`, wgslName: `${base}Size`, wgslType: "vec2<f32>",
+            byteOffset: dhOff, byteSize: 8, kind: "texture-ref",
+            textureSub: "size", textureBindingName: t.name,
+        });
+        dhOff += 8;
+    }
+    const drawHeaderBytes = roundUp(dhOff, 16);
+    const strideU32 = drawHeaderBytes / 4;
+    // Allocate texture + sampler bindings starting at HEAP_TEX_BINDING_START.
+    // Order: all textures, then all samplers — matches the prelude
+    // layout below and the runtime's BGL/bind-group construction.
+    // Atlas-routed bindings drop out — their data flows through the
+    // drawHeader instead of through individual texture/sampler slots.
+    let nextBinding = HEAP_TEX_BINDING_START;
+    const textureBindings = schema.textures
+        .filter(t => !atlasTextureBindings.has(t.name))
+        .map(t => ({ name: t.name, wgslType: t.wgslType, binding: nextBinding++ }));
+    // Samplers: drop the matching one when its texture is atlas-routed.
+    // Heuristic: assume position-aligned (nth sampler pairs with nth
+    // texture). Heap path's v1 surface only ever has 0..1 of each, so
+    // this collapses to "if the lone texture is atlas, drop the lone
+    // sampler too".
+    const atlasSamplerNames = new Set();
+    for (let i = 0; i < schema.textures.length; i++) {
+        const t = schema.textures[i];
+        if (atlasTextureBindings.has(t.name)) {
+            const s = schema.samplers[i];
+            if (s !== undefined)
+                atlasSamplerNames.add(s.name);
+        }
+    }
+    const samplerBindings = schema.samplers
+        .filter(s => !atlasSamplerNames.has(s.name))
+        .map(s => ({ name: s.name, wgslType: s.wgslType, binding: nextBinding++ }));
+    const isAtlasBucket = atlasTextureBindings.size > 0;
+    // Extend the schema's varyings with flat-interpolated atlas threading
+    // slots when atlas is in use — the prelude generator emits these into
+    // VsOut so user VS code can write them and FS can read them. The
+    // location numbers are picked after the existing varyings.
+    const schemaForPrelude = isAtlasBucket
+        ? extendSchemaWithAtlasVaryings(schema, atlasTextureBindings)
+        : schema;
+    const preludeWgsl = generatePrelude(schemaForPrelude, textureBindings, samplerBindings, isAtlasBucket);
+    return {
+        drawHeaderFields, drawHeaderBytes, preludeWgsl, strideU32,
+        perInstanceUniforms,
+        perInstanceAttributes,
+        textureBindings, samplerBindings,
+        atlasTextureBindings,
+    };
+}
+/**
+ * Append flat-interpolated atlas threading varyings to a schema.
+ * Locations follow the highest existing `@location(N)` so we don't
+ * collide with effect-declared varyings.
+ */
+function extendSchemaWithAtlasVaryings(schema, atlasNames) {
+    let nextLoc = 0;
+    for (const v of schema.varyings) {
+        if (v.location !== undefined)
+            nextLoc = Math.max(nextLoc, v.location + 1);
+    }
+    const extras = [];
+    for (const name of atlasNames) {
+        const v = atlasVaryingNames(name);
+        extras.push({ name: v.pageRef, wgslType: "u32", location: nextLoc++, interpolation: "flat" });
+        extras.push({ name: v.formatBits, wgslType: "u32", location: nextLoc++, interpolation: "flat" });
+        extras.push({ name: v.origin, wgslType: "vec2<f32>", location: nextLoc++, interpolation: "flat" });
+        extras.push({ name: v.size, wgslType: "vec2<f32>", location: nextLoc++, interpolation: "flat" });
+    }
+    return { ...schema, varyings: [...schema.varyings, ...extras] };
+}
+function lowerFirst(s) {
+    return s.length === 0 ? s : s[0].toLowerCase() + s.slice(1);
+}
+function roundUp(value, mult) {
+    return Math.ceil(value / mult) * mult;
+}
+// ─── Prelude generation ─────────────────────────────────────────────
+function generatePrelude(schema, textureBindings, samplerBindings, isAtlasBucket) {
+    // VsOut is generated from the schema's surviving VS outputs (post
+    // link + DCE). No hardcoded fields — what the effect declared is
+    // what ends up here.
+    const vsOutBody = schema.varyings.map(v => `  ${vsOutFieldDecl(v)}: ${v.wgslType},`).join("\n");
+    // Textures + samplers — one binding per surviving entry, starting
+    // at HEAP_TEX_BINDING_START. The user's WGSL must reference them
+    // by their schema name; the rewriter drops any DSL-emitted decls
+    // that conflict.
+    const texDecls = textureBindings.map(t => `@group(0) @binding(${t.binding}) var ${t.name}: ${t.wgslType};`).join("\n");
+    const smpDecls = samplerBindings.map(s => `@group(0) @binding(${s.binding}) var ${s.name}: ${s.wgslType};`).join("\n");
+    const samplingBlock = (texDecls.length > 0 || smpDecls.length > 0)
+        ? `\n${texDecls}${texDecls && smpDecls ? "\n" : ""}${smpDecls}\n`
+        : "";
+    const megacallDecls = `\n@group(0) @binding(4) var<storage, read> drawTable:       array<u32>;\n@group(0) @binding(5) var<storage, read> indexStorage:    array<u32>;\n@group(0) @binding(6) var<storage, read> firstDrawInTile: array<u32>;`;
+    const atlasBlock = isAtlasBucket ? generateAtlasPrelude() : "";
+    return /* wgsl */ `
+@group(0) @binding(0) var<storage, read> heapU32:    array<u32>;
+@group(0) @binding(1) var<storage, read> headersU32: array<u32>;
+@group(0) @binding(2) var<storage, read> heapF32:    array<f32>;
+@group(0) @binding(3) var<storage, read> heapV4f:    array<vec4<f32>>;${megacallDecls}
+${samplingBlock}${atlasBlock}
+struct VsOut {
+${vsOutBody}
+};
+`;
+}
+/**
+ * Atlas prelude: declares N independent `texture_2d<f32>` bindings
+ * per format (linear + srgb), plus a shared atlas sampler. The N-way
+ * `switch pageRef` ladder picks the matching texture pair for each
+ * draw. Each atlas page is its own `GPUTexture` — adding pages to a
+ * format is just a fresh allocation slotted into the next BGL entry,
+ * no GPU-side `copyTextureToTexture` required.
+ *
+ * `binding_array<texture_2d<f32>, N>` would collapse the switch into
+ * a runtime index but requires Chrome's experimental bindless
+ * feature; the N-consecutive-bindings + switch pattern is native
+ * WebGPU 1.0.
+ *
+ * Mip filtering is software (the GPU's mip walk would walk the
+ * texture's mip chain, not our embedded Iliffe pyramid). LOD is
+ * computed from screen-space derivatives of `uv * size_px` in mip-0
+ * texels-per-screen-pixel.
+ *
+ * Sample math (verified by `tests-browser/atlas-sampling-conformance.test.ts`
+ * to match native hardware bilinear within 1 LSB across clamp/repeat/
+ * mirror at any uv):
+ *   - Clamp:  atlas_p = origin_px + clamp(uv, 0, 1) * size_px
+ *   - Mirror: atlas_p = origin_px + mirrorUv(uv) * size_px
+ *   - Repeat: atlas_p = origin_px + fract(uv) * size_px, then shift ∓1
+ *             if within 0.5 atlas-pixel of the sub-rect edge, so
+ *             hardware bilinear straddles the outer wrap-gutter ring.
+ * `origin_px` / `size_px` are atlas-pixel coordinates of mip-0, NOT
+ * normalized. Mip-k rect is computed by walking the integer floor-
+ * halving sequence that matches the CPU packer's pyramid layout.
+ * Each sub-rect (incl. every mip slot) carries a 2-px gutter (inner
+ * ring = clamp-replicate edge, outer ring = wrap = opposite edge)
+ * filled by the upload kernel; without it bilinear at the edges
+ * bleeds across sub-rect boundaries.
+ */
+export function generateAtlasPrelude() {
+    return /* wgsl */ `
+${generateAtlasBindings()}
+
+fn mirrorAtlasUv(u: f32) -> f32 {
+  let t = u - floor(u * 0.5) * 2.0;
+  return 1.0 - abs(t - 1.0);
+}
+
+// One axis: produce atlas-pixel coord from logical uv per wrap mode.
+// mip_origin / mip_size are in atlas pixels for the target mip slot.
+fn atlasAxisAt(uv: f32, mip_origin: f32, mip_size: f32, mode: u32) -> f32 {
+  if (mode == 0u) {                       // clamp
+    let c = clamp(uv, 0.0, 1.0);
+    return mip_origin + c * mip_size;
+  }
+  if (mode == 2u) {                       // mirror
+    let m = mirrorAtlasUv(uv);
+    return mip_origin + m * mip_size;
+  }
+  // repeat
+  let f = uv - floor(uv);
+  var p = mip_origin + f * mip_size;
+  let dL = p - mip_origin;
+  let dR = (mip_origin + mip_size) - p;
+  if      (dL < 0.5) { p = p - 1.0; }
+  else if (dR < 0.5) { p = p + 1.0; }
+  return p;
+}
+
+// Mip-k rect within the Iliffe pyramid, in atlas pixels. Matches the
+// CPU packer's integer floor-halving (so non-pow2 sources still hit
+// the right texels). 2-px gutter between adjacent mip slots.
+const ATLAS_MIP_GAP: f32 = 4.0;
+
+fn atlasMipOriginPx(origin: vec2<f32>, size: vec2<f32>, k: u32) -> vec2<f32> {
+  if (k == 0u) { return origin; }
+  let x = origin.x + size.x + ATLAS_MIP_GAP;
+  var y = origin.y;
+  var j: u32 = 1u;
+  loop {
+    if (j >= k) { break; }
+    let mh = max(1.0, floor(size.y * exp2(-f32(j))));
+    y = y + mh + ATLAS_MIP_GAP;
+    j = j + 1u;
+  }
+  return vec2<f32>(x, y);
+}
+
+fn atlasMipSizePx(size: vec2<f32>, k: u32) -> vec2<f32> {
+  return vec2<f32>(
+    max(1.0, floor(size.x * exp2(-f32(k)))),
+    max(1.0, floor(size.y * exp2(-f32(k)))),
+  );
+}
+
+fn atlasSampleAtMip(
+  pageRef: u32, format: u32,
+  origin_px: vec2<f32>, size_px: vec2<f32>,
+  k: u32,
+  uv: vec2<f32>,
+  addrU: u32, addrV: u32,
+) -> vec4<f32> {
+  let mip_o = atlasMipOriginPx(origin_px, size_px, k);
+  let mip_s = atlasMipSizePx(size_px, k);
+  let px = atlasAxisAt(uv.x, mip_o.x, mip_s.x, addrU);
+  let py = atlasAxisAt(uv.y, mip_o.y, mip_s.y, addrV);
+  let atlasUv = vec2<f32>(px, py) / ${atlasPageSizeConst()}.0;
+${generateAtlasSwitch()}
+  return vec4<f32>(0.0);
+}
+
+fn atlasSample(
+  pageRef: u32, formatBits: u32,
+  origin_px: vec2<f32>, size_px: vec2<f32>,
+  uv: vec2<f32>,
+) -> vec4<f32> {
+  let format  = formatBits & 0x1u;
+  let numMips = (formatBits >> 1u) & 0x7u;
+  let addrU   = (formatBits >> 4u) & 0x3u;
+  let addrV   = (formatBits >> 6u) & 0x3u;
+  // LOD on pre-wrap UV. dpdx/dpdy on the wrapped value spikes at the
+  // fract/mirror seam (discontinuity → bogus huge derivative → tiny
+  // mip selected → blurred band). Multiply by size_px to get
+  // texels-per-screen-pixel.
+  let dx = dpdx(uv) * size_px;
+  let dy = dpdy(uv) * size_px;
+  let rho = max(length(dx), length(dy));
+  let maxLod = f32(max(numMips, 1u) - 1u);
+  let lod = clamp(log2(max(rho, 1e-6)), 0.0, maxLod);
+  let lo = u32(floor(lod));
+  let hi = min(lo + 1u, max(numMips, 1u) - 1u);
+  let t  = lod - f32(lo);
+  let a = atlasSampleAtMip(pageRef, format, origin_px, size_px, lo, uv, addrU, addrV);
+  if (numMips <= 1u) { return a; }
+  let b = atlasSampleAtMip(pageRef, format, origin_px, size_px, hi, uv, addrU, addrV);
+  return mix(a, b, t);
+}
+`;
+}
+function atlasPageSizeConst() {
+    // Embedded literal — divisor matches ATLAS_PAGE_SIZE in atlasPool.ts.
+    return "4096";
+}
+/**
+ * Emit `2 * ATLAS_ARRAY_SIZE + 1` binding declarations: N linear
+ * `texture_2d<f32>`, N srgb `texture_2d<f32>`, then the shared
+ * atlas sampler.
+ */
+export function generateAtlasBindings() {
+    const lines = [];
+    for (let i = 0; i < ATLAS_ARRAY_SIZE; i++) {
+        lines.push(`@group(0) @binding(${ATLAS_LINEAR_BINDING_BASE + i}) var atlasLinear${i}: texture_2d<f32>;`);
+    }
+    for (let i = 0; i < ATLAS_ARRAY_SIZE; i++) {
+        lines.push(`@group(0) @binding(${ATLAS_SRGB_BINDING_BASE + i}) var atlasSrgb${i}: texture_2d<f32>;`);
+    }
+    lines.push(`@group(0) @binding(${ATLAS_SAMPLER_BINDING}) var atlasSampler: sampler;`);
+    return lines.join("\n");
+}
+/**
+ * Emit the `switch pageRef` ladder body for `atlasSampleAtMip`. Each
+ * case picks the matching `atlasLinear<i>` / `atlasSrgb<i>` decl and
+ * blends on `format == 1u`. The fallback (page out of range) is the
+ * function-level `vec4<f32>(0.0)` after the switch.
+ */
+export function generateAtlasSwitch() {
+    const lines = ["  switch pageRef {"];
+    for (let i = 0; i < ATLAS_ARRAY_SIZE; i++) {
+        lines.push(`    case ${i}u: {`);
+        lines.push(`      let lin = textureSampleLevel(atlasLinear${i}, atlasSampler, atlasUv, 0.0);`);
+        lines.push(`      let sr  = textureSampleLevel(atlasSrgb${i},   atlasSampler, atlasUv, 0.0);`);
+        lines.push(`      return select(lin, sr, format == 1u);`);
+        lines.push(`    }`);
+    }
+    lines.push("    default: {}");
+    lines.push("  }");
+    return lines.join("\n");
+}
+/** Render one varying as the WGSL field declaration: `[@builtin(...) | @location(N)] [@interpolate(...)] name`. */
+function vsOutFieldDecl(v) {
+    if (v.builtin !== undefined)
+        return `@builtin(${v.builtin}) ${v.name}`;
+    const interp = v.interpolation !== undefined ? ` @interpolate(${v.interpolation})` : "";
+    if (v.location === undefined) {
+        throw new Error(`heapEffect: varying '${v.name}' has neither @builtin nor @location`);
+    }
+    return `@location(${v.location})${interp} ${v.name}`;
+}
+/** Atlas-texture varying names used by the VS→FS threading + FS preamble. */
+export function atlasVaryingNames(name) {
+    return {
+        pageRef: `_h_${name}PageRef`,
+        formatBits: `_h_${name}FormatBits`,
+        origin: `_h_${name}Origin`,
+        size: `_h_${name}Size`,
+    };
+}
+/**
+ * Megacall VS prelude: binary-search `drawTable` (one record per slot,
+ * 5 u32s: firstEmit, drawIdx, indexStart, indexCount, instanceCount)
+ * for the slot owning the given `emitIdx`, then split the in-record
+ * offset into `(instId, vid)` so per-RO instancing collapses to one
+ * record + one drawIndirect. Defines:
+ *   - `heap_drawIdx` — per-RO selector for header reads.
+ *   - `instId`         — instance index in [0, instanceCount-1].
+ *   - `vid`            — vertex index pulled from indexStorage.
+ *
+ * Layout: each record emits `indexCount * instanceCount` vertices.
+ * `local = emitIdx - firstEmit`, `instId = local / indexCount`,
+ * `vid = indices[indexStart + local % indexCount]`.
+ */
+export function megacallSearchPrelude() {
+    // The shared megacall values (`heap_drawIdx`, `instId`, `vid`) are
+    // declared as `let` locals inside the @vertex fn body. Wombat.shader's
+    // composed-stage helper functions (extractFusedEntry) that reference
+    // these identifiers are post-processed in `applyMegacallToEmittedVs`
+    // to receive them as `u32` parameters — module-scope `var<private>`
+    // is rejected by some WGSL parsers (Safari/WebKit), so we thread the
+    // values through helper signatures instead.
+    return `  let _tileIdx = emitIdx >> 6u;\n  var lo: u32 = firstDrawInTile[_tileIdx];\n  var hi: u32 = firstDrawInTile[_tileIdx + 1u];\n  loop {\n    if (lo >= hi) { break; }\n    let _mid = (lo + hi + 1u) >> 1u;\n    if (drawTable[_mid * 5u] <= emitIdx) { lo = _mid; } else { hi = _mid - 1u; }\n  }\n  let _slot = lo;\n  let _firstEmit  = drawTable[_slot * 5u + 0u];\n  let heap_drawIdx: u32 = drawTable[_slot * 5u + 1u];\n  let _indexStart = drawTable[_slot * 5u + 2u];\n  let _indexCount = drawTable[_slot * 5u + 3u];\n  let _local      = emitIdx - _firstEmit;\n  let instId: u32 = _local / _indexCount;\n  let vid: u32    = indexStorage[_indexStart + (_local % _indexCount)];\n`;
+}
+//# sourceMappingURL=heapEffect.js.map

package/dist/runtime/heapEffect.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"heapEffect.js","sourceRoot":"","sources":["../../src/runtime/heapEffect.ts"],"names":[],"mappings":"AAAA,gEAAgE;AAChE,EAAE;AACF,cAAc;AACd,EAAE;AACF,4EAA4E;AAC5E,kEAAkE;AAClE,qEAAqE;AACrE,qEAAqE;AACrE,mCAAmC;AACnC,EAAE;AACF,uEAAuE;AACvE,kEAAkE;AAClE,oEAAoE;AACpE,6DAA6D;AAC7D,sEAAsE;AACtE,wDAAwD;AAkExD,MAAM,UAAU,iBAAiB,CAAC,MAAc,EAAE,oBAA2C;IAC3F,MAAM,QAAQ,GAAG,MAAM,CAAC,OAAO,CAC7B,oBAAoB,KAAK,SAAS;QAChC,CAAC,CAAC,EAAE,MAAM,EAAE,MAAM,EAAE,oBAAoB,EAAE;QAC1C,CAAC,CAAC,EAAE,MAAM,EAAE,MAAM,EAAE,CACvB,CAAC;IACF,MAAM,OAAO,GAAG,QAAQ,CAAC,MAAM,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,KAAK,KAAK,QAAQ,CAAC,CAAC;IAChE,MAAM,OAAO,GAAG,QAAQ,CAAC,MAAM,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,KAAK,KAAK,UAAU,CAAC,CAAC;IAClE,OAAO;QACL,KAAK,EAAE,OAAO,EAAE,MAAM,IAAI,EAAE;QAC5B,KAAK,EAAE,OAAO,EAAE,MAAM,IAAI,EAAE;QAC5B,MAAM,EAAE,WAAW,CAAC,QAAQ,CAAC,SAAS,CAAC;KACxC,CAAC;AACJ,CAAC;AAOD,SAAS,WAAW,CAAC,KAAuB;IAC1C,MAAM,UAAU,GAAsB,KAAK,CAAC,UAAU,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC;QAC/D,IAAI,EAAE,CAAC,CAAC,IAAI;QACZ,QAAQ,EAAE,YAAY,CAAC,CAAC,CAAC,IAAI,CAAC;QAC9B,QAAQ,EAAE,CAAC,CAAC,QAAQ;QACpB,QAAQ,EAAE,CAAC,CAAC,QAAQ;KACrB,CAAC,CAAC,CAAC;IACJ,oEAAoE;IACpE,sEAAsE;IACtE,oEAAoE;IACpE,uEAAuE;IACvE,mEAAmE;IACnE,uEAAuE;IACvE,MAAM,cAAc,GAAG,IAAI,GAAG,CAAC,UAAU,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC;IAC5D,MAAM,QAAQ,GAAsB,EAAE,CAAC;IACvC,KAAK,MAAM,KAAK,IAAI,KAAK,CAAC,aAAa,EAAE,CAAC;QACxC,KAAK,MAAM,CAAC,IAAI,KAAK,CAAC,MAAM,EAAE,CAAC;YAC7B,IAAI,cAAc,CAAC,GAAG,CAAC,CAAC,CAAC,IAAI,CAAC;gBAAE,SAAS;YACzC,QAAQ,CAAC,IAAI,CAAC;gBACZ,IAAI,EAAE,CAAC,CAAC,IAAI;gBACZ,QAAQ,EAAE,YAAY,CAAC,CAAC,CAAC,IAAI,CAAC;gBAC9B,QAAQ,EAAE,gBAAgB,CAAC,CAAC,CAAC,IAAI,CAAC;aACnC,CAAC,CAAC;QACL,CAAC;IACH,CAAC;IACD,wEAAwE;IACxE,2EAA2E;IAC3E,0EAA0E;IAC1E,8DAA8D;IAC9D,KAAK,MAAM,CAAC,IAAI,KAAK,CAAC,QAAQ,EAAE,CAAC;QAC/B,IAAI,cAAc,CAAC,GAAG,CAAC,CAAC,CAAC,IAAI,CAAC;YAAE,SAAS;QACzC,QAAQ,CAAC,IAAI,CAAC;YACZ,IAAI,EAAE,CAAC,CAAC,IAAI;YACZ,QAAQ,EAAE,YAAY,CAAC,CAAC,CAAC,IAAI,CAAC;YAC9B,QAAQ,EAAE,gBAAgB,CAAC,CAAC,CAAC,IAAI,CAAC;SACnC,CAAC,CAAC;IACL,CAAC;IACD,oEAAoE;IACpE,iEAAiE;IACjE,8DAA8D;IAC9D,MAAM,OAAO,GAAG,KAAK,CAAC,MAAM,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,KAAK,KAAK,QAAQ,CAAC,CAAC;IAC7D,MAAM,QAAQ,GAAkB,OAAO,KAAK,SAAS;QACnD,CAAC,CAAC,EAAE;QACJ,CAAC,CAAC,OAAO,CAAC,OAAO,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE;YACtB,MAAM,CAAC,GAIH,EAAE,IAAI,EAAE,CAAC,CAAC,IAAI,EAAE,QAAQ,EAAE,YAAY,CAAC,CAAC,CAAC,IAAI,CAAC,EAAE,CAAC;YACrD,IAAI,CAAC,CAAC,QAAQ,KAAU,SAAS;gBAAE,CAAC,CAAC,QAAQ,GAAQ,CAAC,CAAC,QAAQ,CAAC;YAChE,IAAI,CAAC,CAAC,OAAO,KAAW,SAAS;gBAAE,CAAC,CAAC,OAAO,GAAS,CAAC,CAAC,OAAO,CAAC;YAC/D,IAAI,CAAC,CAAC,aAAa,KAAK,SAAS;gBAAE,CAAC,CAAC,aAAa,GAAG,CAAC,CAAC,aAAa,CAAC;YACrE,OAAO,CAAgB,CAAC;QAC1B,CAAC,CAAC,CAAC;IACP,MAAM,eAAe,GAAG,KAAK,CAAC,eAAe,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC;QACtD,IAAI,EAAE,CAAC,CAAC,IAAI,EAAE,QAAQ,EAAE,CAAC,CAAC,QAAQ,EAAE,QAAQ,EAAE,YAAY,CAAC,CAAC,CAAC,IAAI,CAAC;KACnE,CAAC,CAAC,CAAC;IACJ,mEAAmE;IACnE,+DAA+D;IAC/D,+DAA+D;IAC/D,gEAAgE;IAChE,MAAM,QAAQ,GAAyB,KAAK,CAAC,QAAQ,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC;QAC9D,IAAI,EAAE,CAAC,CAAC,IAAI,EAAE,QAAQ,EAAE,iBAAiB;KAC1C,CAAC,CAAC,CAAC;IACJ,MAAM,QAAQ,GAAyB,KAAK,CAAC,QAAQ,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC;QAC9D,IAAI,EAAE,CAAC,CAAC,IAAI,EAAE,QAAQ,EAAE,SAAS;KAClC,CAAC,CAAC,CAAC;IACJ,OAAO,EAAE,UAAU,EAAE,QAAQ,EAAE,QAAQ,EAAE,eAAe,EAAE,QAAQ,EAAE,QAAQ,EAAE,CAAC;AACjF,CAAC;AAED,SAAS,YAAY,CAAC,CAAS;IAC7B,QAAQ,CAAC,CAAC,IAAI,EAAE,CAAC;QACf,KAAK,OAAO,CAAC,CAAE,OAAO,KAAK,CAAC;QAC5B,KAAK,KAAK,CAAC,CAAI,OAAO,CAAC,CAAC,MAAM,CAAC,CAAC,CAAC,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC;QAC/C,KAAK,MAAM,CAAC,CAAG,OAAO,MAAM,CAAC;QAC7B,KAAK,QAAQ,CAAC,CAAC,OAAO,MAAM,CAAC,CAAC,GAAG,IAAI,YAAY,CAAC,CAAC,CAAC,OAAO,CAAC,GAAG,CAAC;QAChE,KAAK,QAAQ,CAAC,CAAC,OAAO,MAAM,CAAC,CAAC,IAAI,IAAI,CAAC,CAAC,IAAI,IAAI,YAAY,CAAC,CAAC,CAAC,OAAO,CAAC,GAAG,CAAC;QAC3E,KAAK,OAAO,CAAC,CAAC,CAAC;YACb,MAAM,KAAK,GAAG,YAAY,CAAC,CAAC,CAAC,OAAO,CAAC,CAAC;YACtC,OAAO,CAAC,CAAC,MAAM,KAAK,SAAS,CAAC,CAAC,CAAC,SAAS,KAAK,GAAG,CAAC,CAAC,CAAC,SAAS,KAAK,KAAK,CAAC,CAAC,MAAM,GAAG,CAAC;QACrF,CAAC;QACD;YACE,MAAM,IAAI,KAAK,CAAC,+CAA+C,CAAC,CAAC,IAAI,WAAW,CAAC,CAAC;IACtF,CAAC;AACH,CAAC;AAED,SAAS,gBAAgB,CAAC,CAAS;IACjC,QAAQ,CAAC,CAAC,IAAI,EAAE,CAAC;QACf,KAAK,OAAO,CAAC,CAAE,OAAO,CAAC,CAAC;QACxB,KAAK,KAAK,CAAC,CAAI,OAAO,CAAC,CAAC;QACxB,KAAK,MAAM,CAAC,CAAG,OAAO,CAAC,CAAC;QACxB,KAAK,QAAQ,CAAC,CAAC,OAAO,CAAC,CAAC,GAAG,GAAG,gBAAgB,CAAC,CAAC,CAAC,OAAO,CAAC,CAAC;QAC1D,KAAK,QAAQ,CAAC,CAAC,OAAO,CAAC,CAAC,IAAI,GAAG,CAAC,CAAC,IAAI,GAAG,gBAAgB,CAAC,CAAC,CAAC,OAAO,CAAC,CAAC;QACpE,KAAK,OAAO,CAAC,CAAC,CAAC;YACb,IAAI,CAAC,CAAC,MAAM,KAAK,SAAS,EAAE,CAAC;gBAC3B,MAAM,IAAI,KAAK,CAAC,mEAAmE,CAAC,CAAC;YACvF,CAAC;YACD,OAAO,CAAC,CAAC,MAAM,GAAG,gBAAgB,CAAC,CAAC,CAAC,OAAO,CAAC,CAAC;QAChD,CAAC;QACD;YACE,MAAM,IAAI,KAAK,CAAC,gDAAgD,CAAC,CAAC,IAAI,GAAG,CAAC,CAAC;IAC/E,CAAC;AACH,CAAC;AAwED;;;;GAIG;AACH,MAAM,CAAC,MAAM,sBAAsB,GAAG,CAAC,CAAC;AAExC;;;;;GAKG;AACH,MAAM,CAAC,MAAM,gBAAgB,GAAG,CAAC,CAAC;AAElC,uDAAuD;AACvD,MAAM,CAAC,MAAM,yBAAyB,GAAG,EAAE,CAAC;AAC5C,oDAAoD;AACpD,MAAM,CAAC,MAAM,uBAAuB,GAAG,yBAAyB,GAAG,gBAAgB,CAAC;AACpF,kDAAkD;AAClD,MAAM,CAAC,MAAM,qBAAqB,GAAG,yBAAyB,GAAG,CAAC,GAAG,gBAAgB,CAAC;AAEtF,MAAM,UAAU,iBAAiB,CAC/B,MAAwB,EACxB,YAAqB,EACrB,OAWI,EAAE;IAEN,MAAM,mBAAmB,GAAG,IAAI,CAAC,mBAAmB,IAAI,IAAI,GAAG,EAAU,CAAC;IAC1E,MAAM,qBAAqB,GAAG,IAAI,CAAC,qBAAqB,IAAI,IAAI,GAAG,EAAU,CAAC;IAC9E,MAAM,oBAAoB,GAAG,IAAI,CAAC,oBAAoB,IAAI,IAAI,GAAG,EAAU,CAAC;IAC5E,MAAM,gBAAgB,GAAsB,EAAE,CAAC;IAE/C,mEAAmE;IACnE,oEAAoE;IACpE,6CAA6C;IAC7C,IAAI,KAAK,GAAG,CAAC,CAAC;IACd,KAAK,MAAM,CAAC,IAAI,MAAM,CAAC,QAAQ,EAAE,CAAC;QAChC,KAAK,GAAG,OAAO,CAAC,KAAK,EAAE,CAAC,CAAC,CAAC;QAC1B,gBAAgB,CAAC,IAAI,CAAC;YACpB,IAAI,EAAE,CAAC,CAAC,IAAI,EAAE,QAAQ,EAAE,UAAU,CAAC,CAAC,CAAC,IAAI,CAAC,GAAG,KAAK,EAAE,QAAQ,EAAE,KAAK;YACnE,UAAU,EAAE,KAAK,EAAE,QAAQ,EAAE,CAAC,EAAE,IAAI,EAAE,aAAa;YACnD,eAAe,EAAE,CAAC,CAAC,QAAQ;SAC5B,CAAC,CAAC;QACH,KAAK,IAAI,CAAC,CAAC;IACb,CAAC;IAED,6EAA6E;IAC7E,KAAK,MAAM,CAAC,IAAI,MAAM,CAAC,UAAU,EAAE,CAAC;QAClC,KAAK,GAAG,OAAO,CAAC,KAAK,EAAE,CAAC,CAAC,CAAC;QAC1B,gBAAgB,CAAC,IAAI,CAAC;YACpB,IAAI,EAAE,CAAC,CAAC,IAAI,EAAE,QAAQ,EAAE,UAAU,CAAC,CAAC,CAAC,IAAI,CAAC,GAAG,KAAK,EAAE,QAAQ,EAAE,KAAK;YACnE,UAAU,EAAE,KAAK,EAAE,QAAQ,EAAE,CAAC,EAAE,IAAI,EAAE,eAAe;YACrD,iBAAiB,EAAE,CAAC,CAAC,QAAQ;SAC9B,CAAC,CAAC;QACH,KAAK,IAAI,CAAC,CAAC;IACb,CAAC;IAED,+DAA+D;IAC/D,+DAA+D;IAC/D,gEAAgE;IAChE,8DAA8D;IAC9D,KAAK,MAAM,CAAC,IAAI,MAAM,CAAC,QAAQ,EAAE,CAAC;QAChC,IAAI,CAAC,oBAAoB,CAAC,GAAG,CAAC,CAAC,CAAC,IAAI,CAAC;YAAE,SAAS;QAChD,MAAM,IAAI,GAAG,UAAU,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC;QAChC,KAAK,GAAG,OAAO,CAAC,KAAK,EAAE,CAAC,CAAC,CAAC;QAC1B,gBAAgB,CAAC,IAAI,CAAC;YACpB,IAAI,EAAE,GAAG,CAAC,CAAC,IAAI,UAAU,EAAE,QAAQ,EAAE,GAAG,IAAI,SAAS,EAAE,QAAQ,EAAE,KAAK;YACtE,UAAU,EAAE,KAAK,EAAE,QAAQ,EAAE,CAAC,EAAE,IAAI,EAAE,aAAa;YACnD,UAAU,EAAE,SAAS,EAAE,kBAAkB,EAAE,CAAC,CAAC,IAAI;SAClD,CAAC,CAAC;QACH,KAAK,IAAI,CAAC,CAAC;QACX,gBAAgB,CAAC,IAAI,CAAC;YACpB,IAAI,EAAE,GAAG,CAAC,CAAC,IAAI,aAAa,EAAE,QAAQ,EAAE,GAAG,IAAI,YAAY,EAAE,QAAQ,EAAE,KAAK;YAC5E,UAAU,EAAE,KAAK,EAAE,QAAQ,EAAE,CAAC,EAAE,IAAI,EAAE,aAAa;YACnD,UAAU,EAAE,YAAY,EAAE,kBAAkB,EAAE,CAAC,CAAC,IAAI;SACrD,CAAC,CAAC;QACH,KAAK,IAAI,CAAC,CAAC;QACX,KAAK,GAAG,OAAO,CAAC,KAAK,EAAE,CAAC,CAAC,CAAC;QAC1B,gBAAgB,CAAC,IAAI,CAAC;YACpB,IAAI,EAAE,GAAG,CAAC,CAAC,IAAI,SAAS,EAAE,QAAQ,EAAE,GAAG,IAAI,QAAQ,EAAE,QAAQ,EAAE,WAAW;YAC1E,UAAU,EAAE,KAAK,EAAE,QAAQ,EAAE,CAAC,EAAE,IAAI,EAAE,aAAa;YACnD,UAAU,EAAE,QAAQ,EAAE,kBAAkB,EAAE,CAAC,CAAC,IAAI;SACjD,CAAC,CAAC;QACH,KAAK,IAAI,CAAC,CAAC;QACX,gBAAgB,CAAC,IAAI,CAAC;YACpB,IAAI,EAAE,GAAG,CAAC,CAAC,IAAI,OAAO,EAAE,QAAQ,EAAE,GAAG,IAAI,MAAM,EAAE,QAAQ,EAAE,WAAW;YACtE,UAAU,EAAE,KAAK,EAAE,QAAQ,EAAE,CAAC,EAAE,IAAI,EAAE,aAAa;YACnD,UAAU,EAAE,MAAM,EAAE,kBAAkB,EAAE,CAAC,CAAC,IAAI;SAC/C,CAAC,CAAC;QACH,KAAK,IAAI,CAAC,CAAC;IACb,CAAC;IAED,MAAM,eAAe,GAAG,OAAO,CAAC,KAAK,EAAE,EAAE,CAAC,CAAC;IAC3C,MAAM,SAAS,GAAG,eAAe,GAAG,CAAC,CAAC;IAEtC,0EAA0E;IAC1E,+DAA+D;IAC/D,8DAA8D;IAC9D,gEAAgE;IAChE,kEAAkE;IAClE,IAAI,WAAW,GAAG,sBAAsB,CAAC;IACzC,MAAM,eAAe,GAAG,MAAM,CAAC,QAAQ;SACpC,MAAM,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,oBAAoB,CAAC,GAAG,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC;SAC9C,GAAG,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,EAAE,IAAI,EAAE,CAAC,CAAC,IAAI,EAAE,QAAQ,EAAE,CAAC,CAAC,QAAQ,EAAE,OAAO,EAAE,WAAW,EAAE,EAAE,CAAC,CAAC,CAAC;IAC9E,oEAAoE;IACpE,iEAAiE;IACjE,kEAAkE;IAClE,iEAAiE;IACjE,gBAAgB;IAChB,MAAM,iBAAiB,GAAG,IAAI,GAAG,EAAU,CAAC;IAC5C,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,MAAM,CAAC,QAAQ,CAAC,MAAM,EAAE,CAAC,EAAE,EAAE,CAAC;QAChD,MAAM,CAAC,GAAG,MAAM,CAAC,QAAQ,CAAC,CAAC,CAAE,CAAC;QAC9B,IAAI,oBAAoB,CAAC,GAAG,CAAC,CAAC,CAAC,IAAI,CAAC,EAAE,CAAC;YACrC,MAAM,CAAC,GAAG,MAAM,CAAC,QAAQ,CAAC,CAAC,CAAC,CAAC;YAC7B,IAAI,CAAC,KAAK,SAAS;gBAAE,iBAAiB,CAAC,GAAG,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC;QACrD,CAAC;IACH,CAAC;IACD,MAAM,eAAe,GAAG,MAAM,CAAC,QAAQ;SACpC,MAAM,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,iBAAiB,CAAC,GAAG,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC;SAC3C,GAAG,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,EAAE,IAAI,EAAE,CAAC,CAAC,IAAI,EAAE,QAAQ,EAAE,CAAC,CAAC,QAAQ,EAAE,OAAO,EAAE,WAAW,EAAE,EAAE,CAAC,CAAC,CAAC;IAE9E,MAAM,aAAa,GAAG,oBAAoB,CAAC,IAAI,GAAG,CAAC,CAAC;IACpD,sEAAsE;IACtE,sEAAsE;IACtE,iEAAiE;IACjE,2DAA2D;IAC3D,MAAM,gBAAgB,GAAG,aAAa;QACpC,CAAC,CAAC,6BAA6B,CAAC,MAAM,EAAE,oBAAoB,CAAC;QAC7D,CAAC,CAAC,MAAM,CAAC;IACX,MAAM,WAAW,GAAG,eAAe,CACjC,gBAAgB,EAAE,eAAe,EAAE,eAAe,EAClD,aAAa,CACd,CAAC;IAEF,OAAO;QACL,gBAAgB,EAAE,eAAe,EAAE,WAAW,EAAE,SAAS;QACzD,mBAAmB;QACnB,qBAAqB;QACrB,eAAe,EAAE,eAAe;QAChC,oBAAoB;KACrB,CAAC;AACJ,CAAC;AAED;;;;GAIG;AACH,SAAS,6BAA6B,CACpC,MAAwB,EACxB,UAA+B;IAE/B,IAAI,OAAO,GAAG,CAAC,CAAC;IAChB,KAAK,MAAM,CAAC,IAAI,MAAM,CAAC,QAAQ,EAAE,CAAC;QAChC,IAAI,CAAC,CAAC,QAAQ,KAAK,SAAS;YAAE,OAAO,GAAG,IAAI,CAAC,GAAG,CAAC,OAAO,EAAE,CAAC,CAAC,QAAQ,GAAG,CAAC,CAAC,CAAC;IAC5E,CAAC;IACD,MAAM,MAAM,GAAkB,EAAE,CAAC;IACjC,KAAK,MAAM,IAAI,IAAI,UAAU,EAAE,CAAC;QAC9B,MAAM,CAAC,GAAG,iBAAiB,CAAC,IAAI,CAAC,CAAC;QAClC,MAAM,CAAC,IAAI,CAAC,EAAE,IAAI,EAAE,CAAC,CAAC,OAAO,EAAK,QAAQ,EAAE,KAAK,EAAS,QAAQ,EAAE,OAAO,EAAE,EAAE,aAAa,EAAE,MAAM,EAAE,CAAC,CAAC;QACxG,MAAM,CAAC,IAAI,CAAC,EAAE,IAAI,EAAE,CAAC,CAAC,UAAU,EAAE,QAAQ,EAAE,KAAK,EAAS,QAAQ,EAAE,OAAO,EAAE,EAAE,aAAa,EAAE,MAAM,EAAE,CAAC,CAAC;QACxG,MAAM,CAAC,IAAI,CAAC,EAAE,IAAI,EAAE,CAAC,CAAC,MAAM,EAAM,QAAQ,EAAE,WAAW,EAAG,QAAQ,EAAE,OAAO,EAAE,EAAE,aAAa,EAAE,MAAM,EAAE,CAAC,CAAC;QACxG,MAAM,CAAC,IAAI,CAAC,EAAE,IAAI,EAAE,CAAC,CAAC,IAAI,EAAQ,QAAQ,EAAE,WAAW,EAAG,QAAQ,EAAE,OAAO,EAAE,EAAE,aAAa,EAAE,MAAM,EAAE,CAAC,CAAC;IAC1G,CAAC;IACD,OAAO,EAAE,GAAG,MAAM,EAAE,QAAQ,EAAE,CAAC,GAAG,MAAM,CAAC,QAAQ,EAAE,GAAG,MAAM,CAAC,EAAE,CAAC;AAClE,CAAC;AAED,SAAS,UAAU,CAAC,CAAS;IAC3B,OAAO,CAAC,CAAC,MAAM,KAAK,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAE,CAAC,WAAW,EAAE,GAAG,CAAC,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC;AAC/D,CAAC;AAED,SAAS,OAAO,CAAC,KAAa,EAAE,IAAY;IAC1C,OAAO,IAAI,CAAC,IAAI,CAAC,KAAK,GAAG,IAAI,CAAC,GAAG,IAAI,CAAC;AACxC,CAAC;AAED,uEAAuE;AAEvE,SAAS,eAAe,CACtB,MAAwB,EACxB,eAA+E,EAC/E,eAA+E,EAC/E,aAAsB;IAEtB,kEAAkE;IAClE,iEAAiE;IACjE,qBAAqB;IACrB,MAAM,SAAS,GAAG,MAAM,CAAC,QAAQ,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,CAAC,KAAK,cAAc,CAAC,CAAC,CAAC,KAAK,CAAC,CAAC,QAAQ,GAAG,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;IAEhG,kEAAkE;IAClE,iEAAiE;IACjE,iEAAiE;IACjE,iBAAiB;IACjB,MAAM,QAAQ,GAAG,eAAe,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,CACvC,sBAAsB,CAAC,CAAC,OAAO,SAAS,CAAC,CAAC,IAAI,KAAK,CAAC,CAAC,QAAQ,GAAG,CACjE,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;IACb,MAAM,QAAQ,GAAG,eAAe,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,CACvC,sBAAsB,CAAC,CAAC,OAAO,SAAS,CAAC,CAAC,IAAI,KAAK,CAAC,CAAC,QAAQ,GAAG,CACjE,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;IACb,MAAM,aAAa,GAAG,CAAC,QAAQ,CAAC,MAAM,GAAG,CAAC,IAAI,QAAQ,CAAC,MAAM,GAAG,CAAC,CAAC;QAChE,CAAC,CAAC,KAAK,QAAQ,GAAG,QAAQ,IAAI,QAAQ,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,GAAG,QAAQ,IAAI;QACjE,CAAC,CAAC,EAAE,CAAC;IAEP,MAAM,aAAa,GAAG,uNAAuN,CAAC;IAE9O,MAAM,UAAU,GAAG,aAAa,CAAC,CAAC,CAAC,oBAAoB,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;IAE/D,OAAO,UAAU,CAAA;;;;wEAIqD,aAAa;EACnF,aAAa,GAAG,UAAU;;EAE1B,SAAS;;CAEV,CAAC;AACF,CAAC;AAED;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;GAiCG;AACH,MAAM,UAAU,oBAAoB;IAClC,OAAO,UAAU,CAAA;EACjB,qBAAqB,EAAE;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;sCAiEa,kBAAkB,EAAE;EACxD,mBAAmB,EAAE;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;CA8BtB,CAAC;AACF,CAAC;AAED,SAAS,kBAAkB;IACzB,sEAAsE;IACtE,OAAO,MAAM,CAAC;AAChB,CAAC;AAED;;;;GAIG;AACH,MAAM,UAAU,qBAAqB;IACnC,MAAM,KAAK,GAAa,EAAE,CAAC;IAC3B,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,gBAAgB,EAAE,CAAC,EAAE,EAAE,CAAC;QAC1C,KAAK,CAAC,IAAI,CAAC,sBAAsB,yBAAyB,GAAG,CAAC,oBAAoB,CAAC,oBAAoB,CAAC,CAAC;IAC3G,CAAC;IACD,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,gBAAgB,EAAE,CAAC,EAAE,EAAE,CAAC;QAC1C,KAAK,CAAC,IAAI,CAAC,sBAAsB,uBAAuB,GAAG,CAAC,kBAAkB,CAAC,oBAAoB,CAAC,CAAC;IACvG,CAAC;IACD,KAAK,CAAC,IAAI,CAAC,sBAAsB,qBAAqB,8BAA8B,CAAC,CAAC;IACtF,OAAO,KAAK,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;AAC1B,CAAC;AAED;;;;;GAKG;AACH,MAAM,UAAU,mBAAmB;IACjC,MAAM,KAAK,GAAa,CAAC,oBAAoB,CAAC,CAAC;IAC/C,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,gBAAgB,EAAE,CAAC,EAAE,EAAE,CAAC;QAC1C,KAAK,CAAC,IAAI,CAAC,YAAY,CAAC,MAAM,CAAC,CAAC;QAChC,KAAK,CAAC,IAAI,CAAC,iDAAiD,CAAC,gCAAgC,CAAC,CAAC;QAC/F,KAAK,CAAC,IAAI,CAAC,+CAA+C,CAAC,kCAAkC,CAAC,CAAC;QAC/F,KAAK,CAAC,IAAI,CAAC,6CAA6C,CAAC,CAAC;QAC1D,KAAK,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC;IACtB,CAAC;IACD,KAAK,CAAC,IAAI,CAAC,iBAAiB,CAAC,CAAC;IAC9B,KAAK,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;IAClB,OAAO,KAAK,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;AAC1B,CAAC;AAED,mHAAmH;AACnH,SAAS,cAAc,CAAC,CAAc;IACpC,IAAI,CAAC,CAAC,OAAO,KAAK,SAAS;QAAE,OAAO,YAAY,CAAC,CAAC,OAAO,KAAK,CAAC,CAAC,IAAI,EAAE,CAAC;IACvE,MAAM,MAAM,GAAG,CAAC,CAAC,aAAa,KAAK,SAAS,CAAC,CAAC,CAAC,iBAAiB,CAAC,CAAC,aAAa,GAAG,CAAC,CAAC,CAAC,EAAE,CAAC;IACxF,IAAI,CAAC,CAAC,QAAQ,KAAK,SAAS,EAAE,CAAC;QAC7B,MAAM,IAAI,KAAK,CAAC,wBAAwB,CAAC,CAAC,IAAI,sCAAsC,CAAC,CAAC;IACxF,CAAC;IACD,OAAO,aAAa,CAAC,CAAC,QAAQ,IAAI,MAAM,IAAI,CAAC,CAAC,IAAI,EAAE,CAAC;AACvD,CAAC;AAED,6EAA6E;AAC7E,MAAM,UAAU,iBAAiB,CAAC,IAAY;IAG5C,OAAO;QACL,OAAO,EAAK,MAAM,IAAI,SAAS;QAC/B,UAAU,EAAE,MAAM,IAAI,YAAY;QAClC,MAAM,EAAM,MAAM,IAAI,QAAQ;QAC9B,IAAI,EAAQ,MAAM,IAAI,MAAM;KAC7B,CAAC;AACJ,CAAC;AAGD;;;;;;;;;;;;;GAaG;AACH,MAAM,UAAU,qBAAqB;IACnC,mEAAmE;IACnE,uEAAuE;IACvE,qEAAqE;IACrE,qEAAqE;IACrE,oEAAoE;IACpE,qEAAqE;IACrE,4CAA4C;IAC5C,OAAO,iqBAAiqB,CAAC;AAC3qB,CAAC"}

package/dist/runtime/heapEffectIR.d.ts

@@ -0,0 +1,34 @@
+import type { Effect, CompileOptions } from "@aardworx/wombat.shader";
+import { type BucketLayout } from "./heapEffect.js";
+/**
+ * Compile a wombat.shader Effect into heap-rendering WGSL via IR-level
+ * substitution.
+ *
+ * v1 scope:
+ *   - Adds heap storage buffers + builtin VS params.
+ *   - Substitutes VS attribute / VS uniform reads with heap loads.
+ *   - FS uniform substitution still TBD — falls back to current
+ *     regex rewriter for FS until the cross-stage threading is wired.
+ */
+/**
+ * Emission mode for `compileHeapEffectIR`.
+ *
+ *   - `"standalone"` (default): emit a self-contained pipeline. The VS
+ *     entry takes `@builtin(vertex_index) emitIdx` and runs the megacall
+ *     binary-search prelude inline; storage-buffer bindings 4..6 for
+ *     drawTable / indexStorage / firstDrawInTile are appended.
+ *   - `"family-member"`: emit a VS entry shaped to be invoked from a
+ *     family wrapper. The wrapper performs the megacall search and
+ *     passes `heap_drawIdx`, `instId`, `vid` down. The emitted entry
+ *     becomes `@vertex fn vs(heap_drawIdx: u32, instId: u32, vid: u32)`;
+ *     no megacall prelude or bindings 4..6 are emitted at this level.
+ */
+export type HeapEffectEmitMode = "standalone" | "family-member";
+export declare function compileHeapEffectIR(userEffect: Effect, layout: BucketLayout, compileOptions: CompileOptions, mode?: HeapEffectEmitMode): {
+    vs: string;
+    fs: string;
+    preludeWgsl: string;
+    vsEntry: string;
+    fsEntry: string;
+};
+//# sourceMappingURL=heapEffectIR.d.ts.map

package/dist/runtime/heapEffectIR.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"heapEffectIR.d.ts","sourceRoot":"","sources":["../../src/runtime/heapEffectIR.ts"],"names":[],"mappings":"AAoBA,OAAO,KAAK,EAAE,MAAM,EAAE,cAAc,EAAE,MAAM,yBAAyB,CAAC;AAOtE,OAAO,EAGL,KAAK,YAAY,EAClB,MAAM,iBAAiB,CAAC;AAwuBzB;;;;;;;;;GASG;AACH;;;;;;;;;;;;GAYG;AACH,MAAM,MAAM,kBAAkB,GAAG,YAAY,GAAG,eAAe,CAAC;AAEhE,wBAAgB,mBAAmB,CACjC,UAAU,EAAE,MAAM,EAClB,MAAM,EAAE,YAAY,EACpB,cAAc,EAAE,cAAc,EAC9B,IAAI,GAAE,kBAAiC,GACtC;IAAE,EAAE,EAAE,MAAM,CAAC;IAAC,EAAE,EAAE,MAAM,CAAC;IAAC,WAAW,EAAE,MAAM,CAAC;IAAC,OAAO,EAAE,MAAM,CAAC;IAAC,OAAO,EAAE,MAAM,CAAA;CAAE,CA2EnF"}