@woosh/meep-engine 3.55.0 → 3.57.0

package/src/shade/renderer/fluid/mac/shader_fluid_mac_gradient.js

@@ -1,181 +1,229 @@
-import { WebGPUExtensionType } from "../../../descriptor/WebGPUExtensionType.js";
-import { chunk_grid3d_to_index } from "../../shader/chunk/grid/chunk_grid3d_to_index.js";
-import { chunk_grid_tile_halo_f32 } from "../../shader/chunk/grid/chunk_grid_tile_halo_f32.js";
-import { GRID_TILE_SIZE } from "../../shader/chunk/grid/chunk_grid_tile_window.js";
-import { GRID_TILE_WINDOW_STRUCT } from "../../shader/chunk/grid/GRID_TILE_WINDOW_STRUCT.js";
-import { ComputeShader } from "../../shader/ComputeShader.js";
-import { CodeChunk } from "../../shader/compiler/CodeChunk.js";
-import { ShaderResourceSetDescriptor } from "../../shader/resource/ShaderResourceSetDescriptor.js";
-import { f32, u32 } from "../../shader/type/WEBGPU_WGSL_PRIMITIVE_TYPES.js";
-import { WebGPUArray } from "../../shader/type/WebGPUArray.js";
-import { chunk_fluid_pressure_ghost } from "../pressure/chunk_fluid_pressure_ghost.js";
-import { chunk_v3_mac_subtract_pressure_gradient } from "./chunk_v3_mac_subtract_pressure_gradient.js";
-import { FLUID_MAC_STEP_STRUCT } from "./FLUID_MAC_STEP_STRUCT.js";
-
-const resources = new ShaderResourceSetDescriptor();
-
-resources.createGroup()
-    .addUniform("step", FLUID_MAC_STEP_STRUCT)
-    .addTexture("velocity_u", "unfilterable-float", "3d")
-    .addTexture("velocity_v", "unfilterable-float", "3d")
-    .addTexture("velocity_w", "unfilterable-float", "3d")
-    .addStorageBuffer("pressure", WebGPUArray.from(f32), false)
-    .addStorageBuffer("ghost_pressure", WebGPUArray.from(f32), false)
-    .addStorageBuffer("solid", WebGPUArray.from(u32), false)
-    .addStorageTexture("out_u", "r32float", "3d", "write-only")
-    .addStorageTexture("out_v", "r32float", "3d", "write-only")
-    .addStorageTexture("out_w", "r32float", "3d", "write-only");
-
-const body = CodeChunk.from(
-    //language=WGSL
-    `
-const FLUID_MAC_TILE_PRESSURE = 0u;
-const FLUID_MAC_TILE_SOLID = 1u;
-
-var<workgroup> fluid_mac_tile_pressure: array<f32, GRID_TILE_HALO_CELLS>;
-// 1.0 where a cell is solid
-var<workgroup> fluid_mac_tile_solid: array<f32, GRID_TILE_HALO_CELLS>;
-
-// the window-local cell of this workgroup's first interior cell
-var<private> fluid_mac_tile_origin: vec3<i32>;
-
-fn grid_tile_f32_fetch_inside(source: u32, storage_cell: vec3<u32>) -> f32 {
-    if (source == FLUID_MAC_TILE_PRESSURE) {
-        return pressure[grid3d_to_index(storage_cell, step.cells.period.xy)];
-    }
-
-    // the solid flags are window-local, and their tile loads through a window that does not wrap
-    return select(0.0, 1.0, (solid[grid3d_to_index(storage_cell, step.cells.size.xy)] & 1u) != 0u);
-}
-
-fn fluid_pressure_ghost_corner() -> vec3<i32> {
-    return step.cells.corner;
-}
-
-fn fluid_pressure_ghost_size() -> vec3<u32> {
-    return step.cells.size;
-}
-
-// beyond the window: pressure is the Dirichlet ghost at the ghost cell's centre, the parent's, and cells are fluid
-fn grid_tile_f32_fetch_outside(source: u32, world_cell: vec3<i32>) -> f32 {
-    if (source == FLUID_MAC_TILE_PRESSURE) {
-        return fluid_pressure_ghost(world_cell);
-    }
-
-    return 0.0;
-}
-
-fn mac_pressure(cell: vec3<i32>) -> f32 {
-    return grid_tile_f32_read(&fluid_mac_tile_pressure, cell - fluid_mac_tile_origin);
-}
-
-fn mac_cell_solid(cell: vec3<i32>) -> bool {
-    return grid_tile_f32_read(&fluid_mac_tile_solid, cell - fluid_mac_tile_origin) != 0.0;
-}
-
-// the advected velocity is a transient set, stored window-local; only faces inside the window are read
-fn mac_velocity_u(face: vec3<i32>) -> f32 {
-    return textureLoad(velocity_u, vec3<u32>(face), 0).r;
-}
-
-fn mac_velocity_v(face: vec3<i32>) -> f32 {
-    return textureLoad(velocity_v, vec3<u32>(face), 0).r;
-}
-
-fn mac_velocity_w(face: vec3<i32>) -> f32 {
-    return textureLoad(velocity_w, vec3<u32>(face), 0).r;
-}
-
-// the projected velocity is the persistent set, stored through its toroidal windows
-fn fluid_mac_gradient_store_u(face: vec3<i32>) {
-    let storage = grid_tile_window_storage(face, step.faces_u.corner_storage, step.faces_u.period);
-    textureStore(out_u, storage, vec4<f32>(v3_mac_subtract_pressure_gradient_u(face, step.dt_over_h), 0.0, 0.0, 0.0));
-}
-
-fn fluid_mac_gradient_store_v(face: vec3<i32>) {
-    let storage = grid_tile_window_storage(face, step.faces_v.corner_storage, step.faces_v.period);
-    textureStore(out_v, storage, vec4<f32>(v3_mac_subtract_pressure_gradient_v(face, step.dt_over_h), 0.0, 0.0, 0.0));
-}
-
-fn fluid_mac_gradient_store_w(face: vec3<i32>) {
-    let storage = grid_tile_window_storage(face, step.faces_w.corner_storage, step.faces_w.period);
-    textureStore(out_w, storage, vec4<f32>(v3_mac_subtract_pressure_gradient_w(face, step.dt_over_h), 0.0, 0.0, 0.0));
-}
-
-@compute @workgroup_size(${GRID_TILE_SIZE.join(', ')})
-fn main(
-    @builtin(workgroup_id) group: vec3<u32>,
-    @builtin(local_invocation_id) local: vec3<u32>,
-    @builtin(local_invocation_index) lane: u32,
-) {
-    fluid_mac_tile_origin = vec3<i32>(group) * GRID_TILE_SIZE;
-
-    let cells_local = ${GRID_TILE_WINDOW_STRUCT.wgsl_ref}(step.cells.corner, vec3<u32>(0u), step.cells.size, step.cells.size);
-
-    grid_tile_f32_load(&fluid_mac_tile_pressure, step.cells, FLUID_MAC_TILE_PRESSURE, fluid_mac_tile_origin, lane);
-    grid_tile_f32_load(&fluid_mac_tile_solid, cells_local, FLUID_MAC_TILE_SOLID, fluid_mac_tile_origin, lane);
-
-    let cell = fluid_mac_tile_origin + vec3<i32>(local);
-    let last = vec3<i32>(step.cells.size) - vec3<i32>(1);
-
-    // each cell's -x, -y and -z faces
-    fluid_mac_gradient_store_u(cell);
-    fluid_mac_gradient_store_v(cell);
-    fluid_mac_gradient_store_w(cell);
-
-    // the +boundary face planes, owned by the last tile along each axis; their outer cell is in the halo
-    if (cell.x == last.x) {
-        fluid_mac_gradient_store_u(cell + vec3<i32>(1, 0, 0));
-    }
-    if (cell.y == last.y) {
-        fluid_mac_gradient_store_v(cell + vec3<i32>(0, 1, 0));
-    }
-    if (cell.z == last.z) {
-        fluid_mac_gradient_store_w(cell + vec3<i32>(0, 0, 1));
-    }
-}
-`, [
-        GRID_TILE_WINDOW_STRUCT.declaration_chunk,
-        chunk_grid_tile_halo_f32,
-        chunk_grid3d_to_index,
-        chunk_v3_mac_subtract_pressure_gradient,
-        chunk_fluid_pressure_ghost,
-    ]
-);
-
-body.addExtension(WebGPUExtensionType.Subgroups);
-
-/**
- * The projection's last pass over one cascade window (GPU_FLUID_CASCADES_PROPOSAL_2026_10_16.md §8,
- * pass 6, without the publish): every face of the advected velocity has the pressure gradient taken
- * from it, `u − (dt/h)·(p_R − p_L)`, a pinned face passes through, and the result is written into
- * the persistent velocity.
- *
- * A tile pass: 8 × 8 × 4 cells a workgroup, dispatched `(nx / 8, ny / 8, nz / 4)`. The workgroup loads
- * the pressure and the solid flags as f32 tiles with a one-cell halo into workgroup memory
- * (`chunk_grid_tile_halo_f32`, one barrier; 2 × 2,400 bytes); beyond the window the halo holds the
- * Dirichlet ghost pressure, the parent's at the ghost cell's centre (`chunk_fluid_pressure_ghost`), and
- * fluid. Each invocation writes its cell's −x, −y and −z faces, and the
- * last tile along an axis the window's +boundary face plane, whose outer cells are in the halo.
- *
- * Bindings:
- * - `step`: {@link FLUID_MAC_STEP_STRUCT};
- * - `ghost_pressure`: the step's pressure ghosts, the parent's pressure one cell past the window;
- * - `velocity_u`, `velocity_v`, `velocity_w`: the advected velocity, `r32float` 3-D textures,
- *   window-local;
- * - `pressure`: kinematic pressure, m²/s², `array<f32>`, toroidal through `step.cells`;
- * - `solid`: the window-local solid flags, `array<u32>`, bit 0;
- * - `out_u`, `out_v`, `out_w`: written, the persistent velocity, `r32float` 3-D storage textures,
- *   toroidal through `step.faces_*`.
- *
- * Reads: 2 × 600 halo words a workgroup, 4.7 a cell, where reading each face's two cells' pressure
- * and flags straight from the buffers would take 12; then 12 workgroup-memory reads and 3 texels a
- * cell (each face's own).
- *
- * @type {ComputeShader}
- */
-export const shader_fluid_mac_gradient = ComputeShader.from({
-    label: "Fluid / MAC gradient",
-    body,
-    resources,
-});
+import { WebGPUExtensionType } from "../../../descriptor/WebGPUExtensionType.js";
+import { chunk_grid3d_to_index } from "../../shader/chunk/grid/chunk_grid3d_to_index.js";
+import { chunk_grid_tile_halo_f32 } from "../../shader/chunk/grid/chunk_grid_tile_halo_f32.js";
+import { GRID_TILE_SIZE } from "../../shader/chunk/grid/chunk_grid_tile_window.js";
+import { ComputeShader } from "../../shader/ComputeShader.js";
+import { CodeChunk } from "../../shader/compiler/CodeChunk.js";
+import { ShaderResourceSetDescriptor } from "../../shader/resource/ShaderResourceSetDescriptor.js";
+import { f32, u32 } from "../../shader/type/WEBGPU_WGSL_PRIMITIVE_TYPES.js";
+import { WebGPUArray } from "../../shader/type/WebGPUArray.js";
+import { chunk_fluid_pressure_ghost } from "../pressure/chunk_fluid_pressure_ghost.js";
+import { chunk_v3_mac_subtract_pressure_gradient } from "./chunk_v3_mac_subtract_pressure_gradient.js";
+import { FLUID_MAC_STEP_STRUCT } from "./FLUID_MAC_STEP_STRUCT.js";
+
+const resources = new ShaderResourceSetDescriptor();
+
+resources.createGroup()
+    .addUniform("step", FLUID_MAC_STEP_STRUCT)
+    .addTexture("velocity_u", "unfilterable-float", "3d")
+    .addTexture("velocity_v", "unfilterable-float", "3d")
+    .addTexture("velocity_w", "unfilterable-float", "3d")
+    .addStorageBuffer("pressure", WebGPUArray.from(f32), false)
+    .addStorageBuffer("ghost_pressure", WebGPUArray.from(f32), false)
+    .addStorageBuffer("solid", WebGPUArray.from(u32), false)
+    .addStorageTexture("out_u", "r32float", "3d", "write-only")
+    .addStorageTexture("out_v", "r32float", "3d", "write-only")
+    .addStorageTexture("out_w", "r32float", "3d", "write-only");
+
+const body = CodeChunk.from(
+    //language=WGSL
+    `
+const FLUID_MAC_SOLID_SENTINEL_BITS = 0x7f7fffffu;
+
+// holds pressure for fluid cells and FLUID_MAC_SOLID_SENTINEL_BITS for solid cells
+var<workgroup> fluid_mac_tile_pressure: array<f32, GRID_TILE_HALO_CELLS>;
+
+// the window-local cell of this workgroup's first interior cell
+var<private> fluid_mac_tile_origin: vec3<i32>;
+
+fn fluid_pressure_ghost_corner() -> vec3<i32> {
+    return step.cells.corner;
+}
+
+fn fluid_pressure_ghost_size() -> vec3<u32> {
+    return step.cells.size;
+}
+
+fn fluid_mac_fetch_cell_inside(window_local: vec3<i32>) -> f32 {
+    let local_u32 = vec3<u32>(window_local);
+    if ((solid[grid3d_to_index(local_u32, step.cells.size.xy)] & 1u) != 0u) {
+        return bitcast<f32>(FLUID_MAC_SOLID_SENTINEL_BITS);
+    }
+
+    let storage_cell = grid_tile_window_storage(window_local, step.cells.corner_storage, step.cells.period);
+    let p = pressure[grid3d_to_index(storage_cell, step.cells.period.xy)];
+
+    return select(p, bitcast<f32>(0x7f7ffffeu), bitcast<u32>(p) == FLUID_MAC_SOLID_SENTINEL_BITS);
+}
+
+fn fluid_mac_fetch_cell_outside(window_local: vec3<i32>) -> f32 {
+    let p = fluid_pressure_ghost(step.cells.corner + window_local);
+
+    return select(p, bitcast<f32>(0x7f7ffffeu), bitcast<u32>(p) == FLUID_MAC_SOLID_SENTINEL_BITS);
+}
+
+fn grid_tile_f32_fetch_inside(source: u32, storage_cell: vec3<u32>) -> f32 {
+    return fluid_mac_fetch_cell_inside(vec3<i32>(storage_cell));
+}
+
+fn grid_tile_f32_fetch_outside(source: u32, world_cell: vec3<i32>) -> f32 {
+    return fluid_mac_fetch_cell_outside(world_cell - step.cells.corner);
+}
+
+fn mac_pressure(cell: vec3<i32>) -> f32 {
+    return grid_tile_f32_read(&fluid_mac_tile_pressure, cell - fluid_mac_tile_origin);
+}
+
+fn mac_cell_solid(cell: vec3<i32>) -> bool {
+    return bitcast<u32>(grid_tile_f32_read(&fluid_mac_tile_pressure, cell - fluid_mac_tile_origin)) == FLUID_MAC_SOLID_SENTINEL_BITS;
+}
+
+// the advected velocity is a transient set, stored window-local; only faces inside the window are read
+fn mac_velocity_u(face: vec3<i32>) -> f32 {
+    return textureLoad(velocity_u, vec3<u32>(face), 0).r;
+}
+
+fn mac_velocity_v(face: vec3<i32>) -> f32 {
+    return textureLoad(velocity_v, vec3<u32>(face), 0).r;
+}
+
+fn mac_velocity_w(face: vec3<i32>) -> f32 {
+    return textureLoad(velocity_w, vec3<u32>(face), 0).r;
+}
+
+// the projected velocity is the persistent set, stored through its toroidal windows
+fn fluid_mac_gradient_store_u(face: vec3<i32>) {
+    let storage = grid_tile_window_storage(face, step.faces_u.corner_storage, step.faces_u.period);
+    textureStore(out_u, storage, vec4<f32>(v3_mac_subtract_pressure_gradient_u(face, step.dt_over_h), 0.0, 0.0, 0.0));
+}
+
+fn fluid_mac_gradient_store_v(face: vec3<i32>) {
+    let storage = grid_tile_window_storage(face, step.faces_v.corner_storage, step.faces_v.period);
+    textureStore(out_v, storage, vec4<f32>(v3_mac_subtract_pressure_gradient_v(face, step.dt_over_h), 0.0, 0.0, 0.0));
+}
+
+fn fluid_mac_gradient_store_w(face: vec3<i32>) {
+    let storage = grid_tile_window_storage(face, step.faces_w.corner_storage, step.faces_w.period);
+    textureStore(out_w, storage, vec4<f32>(v3_mac_subtract_pressure_gradient_w(face, step.dt_over_h), 0.0, 0.0, 0.0));
+}
+
+@compute @workgroup_size(${GRID_TILE_SIZE.join(', ')})
+fn main(
+    @builtin(workgroup_id) group: vec3<u32>,
+    @builtin(local_invocation_id) local: vec3<u32>,
+    @builtin(local_invocation_index) lane: u32,
+) {
+    let origin = vec3<i32>(group) * GRID_TILE_SIZE;
+    fluid_mac_tile_origin = origin;
+
+    let size_i = vec3<i32>(step.cells.size);
+    let last = size_i - vec3<i32>(1);
+    let local_i = vec3<i32>(local);
+    let cell = origin + local_i;
+
+    // 1. Every invocation loads its own interior cell (always inside the window).
+    grid_tile_f32_write(&fluid_mac_tile_pressure, local_i, fluid_mac_fetch_cell_inside(cell));
+
+    // 2. Load only the 6 axial halo faces that can be read by this workgroup (1 cell per lane, 0 loops):
+    //    lanes 0..127 load the -x (32), -y (32), and -z (64) faces;
+    //    lanes 128..255 load the +x (32), +y (32), and +z (64) ghost faces only on the last tile along that axis.
+    if (lane < 64u) {
+        let is_y = (lane & 32u) != 0u;
+        let a = i32(lane & 7u);
+        let b = i32((lane >> 3u) & 3u);
+        let halo_local = select(vec3<i32>(-1, a, b), vec3<i32>(a, -1, b), is_y);
+        let halo_cell = origin + halo_local;
+        let inside = select(halo_cell.x >= 0, halo_cell.y >= 0, is_y);
+        var val: f32;
+        if (inside) {
+            val = fluid_mac_fetch_cell_inside(halo_cell);
+        } else {
+            val = fluid_mac_fetch_cell_outside(halo_cell);
+        }
+        grid_tile_f32_write(&fluid_mac_tile_pressure, halo_local, val);
+    } else if (lane < 128u) {
+        let halo_local = vec3<i32>(i32(lane & 7u), i32((lane >> 3u) & 7u), -1);
+        let halo_cell = origin + halo_local;
+        var val: f32;
+        if (halo_cell.z >= 0) {
+            val = fluid_mac_fetch_cell_inside(halo_cell);
+        } else {
+            val = fluid_mac_fetch_cell_outside(halo_cell);
+        }
+        grid_tile_f32_write(&fluid_mac_tile_pressure, halo_local, val);
+    } else if (lane < 160u) {
+        if (origin.x + 8 == size_i.x) {
+            let halo_local = vec3<i32>(8, i32(lane & 7u), i32((lane >> 3u) & 3u));
+            grid_tile_f32_write(&fluid_mac_tile_pressure, halo_local, fluid_mac_fetch_cell_outside(origin + halo_local));
+        }
+    } else if (lane < 192u) {
+        if (origin.y + 8 == size_i.y) {
+            let halo_local = vec3<i32>(i32(lane & 7u), 8, i32((lane >> 3u) & 3u));
+            grid_tile_f32_write(&fluid_mac_tile_pressure, halo_local, fluid_mac_fetch_cell_outside(origin + halo_local));
+        }
+    } else {
+        if (origin.z + 4 == size_i.z) {
+            let halo_local = vec3<i32>(i32(lane & 7u), i32((lane >> 3u) & 7u), 4);
+            grid_tile_f32_write(&fluid_mac_tile_pressure, halo_local, fluid_mac_fetch_cell_outside(origin + halo_local));
+        }
+    }
+
+    workgroupBarrier();
+
+    // each cell's -x, -y and -z faces
+    fluid_mac_gradient_store_u(cell);
+    fluid_mac_gradient_store_v(cell);
+    fluid_mac_gradient_store_w(cell);
+
+    // the +boundary face planes, owned by the last tile along each axis; their outer cell is in the halo
+    if (cell.x == last.x) {
+        fluid_mac_gradient_store_u(cell + vec3<i32>(1, 0, 0));
+    }
+    if (cell.y == last.y) {
+        fluid_mac_gradient_store_v(cell + vec3<i32>(0, 1, 0));
+    }
+    if (cell.z == last.z) {
+        fluid_mac_gradient_store_w(cell + vec3<i32>(0, 0, 1));
+    }
+}
+`, [
+        chunk_grid_tile_halo_f32,
+        chunk_grid3d_to_index,
+        chunk_v3_mac_subtract_pressure_gradient,
+        chunk_fluid_pressure_ghost,
+    ]
+);
+
+body.addExtension(WebGPUExtensionType.Subgroups);
+
+/**
+ * The projection's last pass over one cascade window (GPU_FLUID_CASCADES_PROPOSAL_2026_10_16.md §8,
+ * pass 6, without the publish): every face of the advected velocity has the pressure gradient taken
+ * from it, `u − (dt/h)·(p_R − p_L)`, a pinned face passes through, and the result is written into
+ * the persistent velocity.
+ *
+ * A tile pass: 8 × 8 × 4 cells a workgroup, dispatched `(nx / 8, ny / 8, nz / 4)`. The workgroup loads
+ * the pressure and solid flags as a single f32 tile with a one-cell halo into workgroup memory
+ * (`chunk_grid_tile_halo_f32`, one barrier; 2,400 bytes), encoding solid cells with a sentinel value;
+ * beyond the window the halo holds the Dirichlet ghost pressure, the parent's at the ghost cell's
+ * centre (`chunk_fluid_pressure_ghost`), and fluid. Each invocation writes its cell's −x, −y and −z
+ * faces, and the last tile along an axis the window's +boundary face plane, whose outer cells are in
+ * the halo.
+ *
+ * Bindings:
+ * - `step`: {@link FLUID_MAC_STEP_STRUCT};
+ * - `ghost_pressure`: the step's pressure ghosts, the parent's pressure one cell past the window;
+ * - `velocity_u`, `velocity_v`, `velocity_w`: the advected velocity, `r32float` 3-D textures,
+ *   window-local;
+ * - `pressure`: kinematic pressure, m²/s², `array<f32>`, toroidal through `step.cells`;
+ * - `solid`: the window-local solid flags, `array<u32>`, bit 0;
+ * - `out_u`, `out_v`, `out_w`: written, the persistent velocity, `r32float` 3-D storage textures,
+ *   toroidal through `step.faces_*`.
+ *
+ * @type {ComputeShader}
+ */
+export const shader_fluid_mac_gradient = ComputeShader.from({
+    label: "Fluid / MAC gradient",
+    body,
+    resources,
+});

package/src/shade/renderer/fluid/pressure/GPUFluidPressureSolve.d.ts

@@ -9,15 +9,15 @@
  * changes, never per frame. Each `record_*` adds one graph pass that dispatches everything; each
  * `encode_*` dispatches the same into the encoder of a pass the caller records itself:
  *
- * - **Weights**, once a solve: the finest level's words from the solid flags, then each coarse level's
- *   from the one above (`shader_fluid_mg_level0_weights`, `shader_fluid_mg_coarse_weights`). The SOR
- *   arm needs the finest only.
+ * - **Weights**: the finest level's words are written by `shader_fluid_mac_divergence` when fused into a
+ *   cascade step (`level0_weights_ready: true`) or by `shader_fluid_mg_level0_weights` when standalone;
+ *   each coarse level's words are written by `shader_fluid_mg_residual_restrict` as the V-cycle descends.
  * - **{@link record_v_cycles}**, V-cycles of `3L − 2` dispatches over `L` levels: on each level above
- *   the coarsest a smoother of {@link FLUID_MG_SMOOTH_SWEEPS} sweeps and a residual + restriction; the
- *   coarsest level in one workgroup; then each level back up, one smoother whose load prolongs the
- *   level below in. At 64×32×64, four levels: 10 dispatches a V-cycle, plus 4 for the weights.
+ *   the coarsest a smoother of {@link FLUID_MG_SMOOTH_SWEEPS} sweeps and a residual + restriction +
+ *   coarse weights; the coarsest level in one workgroup; then each level back up, one smoother whose
+ *   load prolongs the level below in. At 64×32×64, four levels: 10 dispatches a V-cycle.
  * - **{@link record_sor}**, red-black SOR on the finest level at `sor_optimal_omega`, one colour a
- *   dispatch: `2K` dispatches for `K` sweeps, plus 1 for the weights.
+ *   dispatch: `2K` dispatches for `K` sweeps.
  *
  * A smoother never writes the buffer it reads, so the finest level goes pressure → scratch → pressure.
  *
@@ -55,6 +55,15 @@ export class GPUFluidPressureSolve {
      * @returns {GPUBuffer|null}
      */
     get nonfinite_counter(): any;
+    /**
+     * Ensure level transients exist for `size` on `device` and return the finest level's weight words
+     * buffer so `shader_fluid_mac_divergence` can write it directly.
+     *
+     * @param {GPUDevice} device
+     * @param {ArrayLike<number>} size
+     * @returns {GPUBuffer}
+     */
+    level0_words(device: GPUDevice, size: ArrayLike<number>): GPUBuffer;
     /**
      * Record V-cycles into a pass of `graph`. Nothing is recorded for 0 cycles.
      *
@@ -159,5 +168,6 @@ export type FluidPressureSolveInputs = {
     ghost_pressure: GPUBuffer;
     cycles?: number;
     sweeps?: number;
+    level0_weights_ready?: boolean;
 };
 //# sourceMappingURL=GPUFluidPressureSolve.d.ts.map

package/src/shade/renderer/fluid/pressure/GPUFluidPressureSolve.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"GPUFluidPressureSolve.d.ts","sourceRoot":"","sources":["../../../../../../src/shade/renderer/fluid/pressure/GPUFluidPressureSolve.js"],"names":[],"mappings":"AAwEA;;;;;;;;;;;;;;;;;;;;;;;;;;;;;GA6BG;AACH;IACI;;;;;OAKG;IACH,iCAFW,OAAO,6CAA6C,EAAE,eAAe,QAU/E;IAiKD;;;;OAIG;IACH,+BAJW;QAAC,KAAK,EAAE,qBAAqB,CAAC;QAAC,GAAG,EAAE,MAAM,CAAC;QAAC,KAAK,EAAE,MAAM,CAAC;QAAC,MAAM,EAAE,wBAAwB,CAAA;KAAC,4BAE5F;QAAC,OAAO,EAAE,sBAAsB,CAAA;KAAC,QAI3C;IAzID;;;;OAIG;IACH,6BAEC;IAED;;;;;;;;;;;;;;;;;OAiBG;IACH;QAdsG,KAAK,EAAhG,OAAO,8DAA8D,EAAE,UAAU;QAC9D,QAAQ;QAEiG,MAAM,EAAlI;YAAC,MAAM,EAAE,UAAU,MAAM,CAAC,CAAC;YAAC,cAAc,EAAE,UAAU,MAAM,CAAC,CAAC;YAAC,IAAI,EAAE,UAAU,MAAM,CAAC,CAAC;YAAC,MAAM,EAAE,UAAU,MAAM,CAAC,CAAA;SAAC;QAG/F,GAAG;QACH,KAAK;QACR,CAAC,EAAjB,MAAM;QACa,cAAc;QAGjB,MAAM,EAAtB,MAAM;aAMhB;IAED;;;;;;;OAOG;IACH,yBAHW,sBAAsB,UACtB,wBAAwB,QASlC;IAED;;;;;;OAMG;IACH,oBAHW,sBAAsB,UACtB,wBAAwB,QASlC;IAED;;;;;;;;;;;;;OAaG;IACH;QATsG,KAAK,EAAhG,OAAO,8DAA8D,EAAE,UAAU;QAC9D,QAAQ;QACiG,MAAM,EAAlI;YAAC,MAAM,EAAE,UAAU,MAAM,CAAC,CAAC;YAAC,cAAc,EAAE,UAAU,MAAM,CAAC,CAAC;YAAC,IAAI,EAAE,UAAU,MAAM,CAAC,CAAC;YAAC,MAAM,EAAE,UAAU,MAAM,CAAC,CAAA;SAAC;QAC/F,GAAG;QACH,KAAK;QACR,CAAC,EAAjB,MAAM;QACa,cAAc;QACjB,MAAM,EAAtB,MAAM;aAMhB;IA0RD;;OAEG;IACH,gBA0BC;;CACJ;qCA7gBY,OAAO,2CAA2C,EAAE,sBAAsB;;;;;uCAO1E;IAAC,QAAQ,YAAY;IAAC,MAAM,EAAE;QAAC,MAAM,EAAE,UAAU,MAAM,CAAC,CAAC;QAAC,cAAc,EAAE,UAAU,MAAM,CAAC,CAAC;QAAC,IAAI,EAAE,UAAU,MAAM,CAAC,CAAC;QAAC,MAAM,EAAE,UAAU,MAAM,CAAC,CAAA;KAAC,CAAC;IAAC,GAAG,YAAY;IAAC,KAAK,YAAY;IAAC,CAAC,EAAE,MAAM,CAAC;IAAC,cAAc,YAAY;IAAC,MAAM,CAAC,EAAE,MAAM,CAAC;IAAC,MAAM,CAAC,EAAE,MAAM,CAAA;CAAC"}
+{"version":3,"file":"GPUFluidPressureSolve.d.ts","sourceRoot":"","sources":["../../../../../../src/shade/renderer/fluid/pressure/GPUFluidPressureSolve.js"],"names":[],"mappings":"AAwEA;;;;;;;;;;;;;;;;;;;;;;;;;;;;;GA6BG;AACH;IACI;;;;;OAKG;IACH,iCAFW,OAAO,6CAA6C,EAAE,eAAe,QAU/E;IA+KD;;;;OAIG;IACH,+BAJW;QAAC,KAAK,EAAE,qBAAqB,CAAC;QAAC,GAAG,EAAE,MAAM,CAAC;QAAC,KAAK,EAAE,MAAM,CAAC;QAAC,MAAM,EAAE,wBAAwB,CAAA;KAAC,4BAE5F;QAAC,OAAO,EAAE,sBAAsB,CAAA;KAAC,QAI3C;IAvJD;;;;OAIG;IACH,6BAEC;IAED;;;;;;;OAOG;IACH,sCAHW,UAAU,MAAM,CAAC,aAO3B;IAED;;;;;;;;;;;;;;;;;OAiBG;IACH;QAdsG,KAAK,EAAhG,OAAO,8DAA8D,EAAE,UAAU;QAC9D,QAAQ;QAEiG,MAAM,EAAlI;YAAC,MAAM,EAAE,UAAU,MAAM,CAAC,CAAC;YAAC,cAAc,EAAE,UAAU,MAAM,CAAC,CAAC;YAAC,IAAI,EAAE,UAAU,MAAM,CAAC,CAAC;YAAC,MAAM,EAAE,UAAU,MAAM,CAAC,CAAA;SAAC;QAG/F,GAAG;QACH,KAAK;QACR,CAAC,EAAjB,MAAM;QACa,cAAc;QAGjB,MAAM,EAAtB,MAAM;aAMhB;IAED;;;;;;;OAOG;IACH,yBAHW,sBAAsB,UACtB,wBAAwB,QASlC;IAED;;;;;;OAMG;IACH,oBAHW,sBAAsB,UACtB,wBAAwB,QASlC;IAED;;;;;;;;;;;;;OAaG;IACH;QATsG,KAAK,EAAhG,OAAO,8DAA8D,EAAE,UAAU;QAC9D,QAAQ;QACiG,MAAM,EAAlI;YAAC,MAAM,EAAE,UAAU,MAAM,CAAC,CAAC;YAAC,cAAc,EAAE,UAAU,MAAM,CAAC,CAAC;YAAC,IAAI,EAAE,UAAU,MAAM,CAAC,CAAC;YAAC,MAAM,EAAE,UAAU,MAAM,CAAC,CAAA;SAAC;QAC/F,GAAG;QACH,KAAK;QACR,CAAC,EAAjB,MAAM;QACa,cAAc;QACjB,MAAM,EAAtB,MAAM;aAMhB;IAyRD;;OAEG;IACH,gBA0BC;;CACJ;qCA1hBY,OAAO,2CAA2C,EAAE,sBAAsB;;;;;uCAO1E;IAAC,QAAQ,YAAY;IAAC,MAAM,EAAE;QAAC,MAAM,EAAE,UAAU,MAAM,CAAC,CAAC;QAAC,cAAc,EAAE,UAAU,MAAM,CAAC,CAAC;QAAC,IAAI,EAAE,UAAU,MAAM,CAAC,CAAC;QAAC,MAAM,EAAE,UAAU,MAAM,CAAC,CAAA;KAAC,CAAC;IAAC,GAAG,YAAY;IAAC,KAAK,YAAY;IAAC,CAAC,EAAE,MAAM,CAAC;IAAC,cAAc,YAAY;IAAC,MAAM,CAAC,EAAE,MAAM,CAAC;IAAC,MAAM,CAAC,EAAE,MAAM,CAAC;IAAC,oBAAoB,CAAC,EAAE,OAAO,CAAA;CAAC"}