@woosh/meep-engine 3.40.0 → 3.42.0

package/src/shade/renderer/water/surface/shader_water_wave_domain.js

@@ -1,83 +0,0 @@
-import { ComputeShader } from "../../shader/ComputeShader.js";
-import { CodeChunk } from "../../shader/compiler/CodeChunk.js";
-import { ShaderResourceSetDescriptor } from "../../shader/resource/ShaderResourceSetDescriptor.js";
-import { WebGPUArray } from "../../shader/type/WebGPUArray.js";
-import { GEOMETRY_METADATA_STRUCT } from "../../geometry/GEOMETRY_METADATA_STRUCT.js";
-import { MESHLET_METADATA_STRUCT } from "../../geometry/meshlet/encoding/MESHLET_METADATA_STRUCT.js";
-import { BVH2_NODE_STRUCT } from "../../shader/chunk/bvh/BVH2_NODE_STRUCT.js";
-import { BVH_TRAVERSAL_STACK_SIZE } from "../../shader/chunk/bvh/BVH_TRAVERSAL_STACK_SIZE.js";
-import { chunk_ray_query_blas_nearest } from "../../shader/chunk/geometry/ray/chunk_ray_query_blas_nearest.js";
-import { chunk_folded_workgroup_index } from "../../shader/chunk/grid/chunk_folded_workgroup_index.js";
-import { f32 } from "../../shader/type/WEBGPU_WGSL_PRIMITIVE_TYPES.js";
-import { WATER_WAVES_TABLE as table } from "../medium/chunk_water_bodies.js";
-import { WATER_WAVE_BATCH_STRUCT } from "./shader_water_wave_batch.js";
-
-const resources = new ShaderResourceSetDescriptor();
-resources.createGroup().addUniform('batch', WATER_WAVE_BATCH_STRUCT)
-    .addStorageBuffer('scene_database', WebGPUArray.u32)
-    .addStorageBuffer('geometries', WebGPUArray.from(GEOMETRY_METADATA_STRUCT))
-    .addStorageBuffer('blas_lookup', WebGPUArray.u32)
-    .addStorageBuffer('blas_data', WebGPUArray.from(BVH2_NODE_STRUCT))
-    .addStorageBuffer('meshlet_metadata', WebGPUArray.from(MESHLET_METADATA_STRUCT))
-    .addStorageBuffer('meshlet_data', WebGPUArray.from(f32))
-    .addStorageBuffer('domain', WebGPUArray.u32, true);
-
-/** Bake the authored footprint once per new wave row, using the existing geometry BLAS. */
-export const shader_water_wave_domain = ComputeShader.from({ label: 'Water / authored simulation domain', resources,
-    body: CodeChunk.from(/* language=WGSL */ `
-@compute @workgroup_size(128)
-fn main(@builtin(workgroup_id) group: vec3u, @builtin(local_invocation_index) lane: u32) {
-    let i = folded_workgroup_index(group) * 128u + lane;
-    let n = batch.resolution;
-    let row = i / (n * n);
-    if (row >= batch.capacity) { return; }
-    let params = ${table.marshalling_method_read}(&scene_database, row);
-    if (params.node == 0u || params.node == ~0u) { domain[i] = 0u; return; }
-    if (params.first_write == 0u && batch.apply_impulses == 0u) { return; }
-    let geometry = geometries[params.geometry];
-    let cell = vec2u(i % n, (i / n) % n);
-    let xz = mix(params.extent.xy, params.extent.zw, vec2f(cell) / f32(n - 1u));
-    let margin = max(1e-3, (geometry.bounding_box[4] - geometry.bounding_box[1]) * 1e-4);
-    let top = geometry.bounding_box[4] + margin;
-    var stack: array<u32, ${BVH_TRAVERSAL_STACK_SIZE}>;
-    let hit = ray_query_blas_nearest(Ray(vec3f(xz.x, top, xz.y), vec3f(0.0, -1.0, 0.0),
-        top - geometry.bounding_box[1] + margin), params.geometry, &stack, 0u);
-    domain[i] = select(0u, 1u, hit.t >= 0.0);
-}
-`, [table.chunk_read, chunk_folded_workgroup_index, chunk_ray_query_blas_nearest]) });
-
-const bounds_resources = new ShaderResourceSetDescriptor();
-bounds_resources.createGroup().addUniform('batch', WATER_WAVE_BATCH_STRUCT)
-    .addStorageBuffer('scene_database', WebGPUArray.u32, true)
-    .addStorageBuffer('state', WebGPUArray.from(f32));
-
-/** Bilinear reconstruction cannot leave the extrema of its samples. Include the rest surface. */
-export const shader_water_wave_bounds = ComputeShader.from({ label: 'Water / displacement bounds', resources: bounds_resources,
-    body: CodeChunk.from(/* language=WGSL */ `
-var<workgroup> lower: array<f32, 128>;
-var<workgroup> upper: array<f32, 128>;
-@compute @workgroup_size(128)
-fn main(@builtin(workgroup_id) group: vec3u, @builtin(local_invocation_index) lane: u32) {
-    let row = folded_workgroup_index(group);
-    if (row >= batch.capacity) { return; }
-    let params = ${table.marshalling_method_read_rw}(&scene_database, row);
-    let occupied = params.node != 0u && params.node != ~0u;
-    let size = select(0u, batch.resolution * batch.resolution, occupied);
-    var lo = 0.0;
-    var hi = 0.0;
-    for (var cell = lane; cell < size; cell += 128u) {
-        let h = state[row * size + cell];
-        lo = min(lo, h); hi = max(hi, h);
-    }
-    lower[lane] = lo; upper[lane] = hi;
-    workgroupBarrier();
-    for (var stride = 64u; stride != 0u; stride /= 2u) {
-        if (lane < stride) {
-            lower[lane] = min(lower[lane], lower[lane + stride]);
-            upper[lane] = max(upper[lane], upper[lane + stride]);
-        }
-        workgroupBarrier();
-    }
-    if (lane == 0u && occupied) { ${table.marshalling_method_write_field('displacement_bounds')}(&scene_database, row, vec2f(lower[0], upper[0])); }
-}
-`, [table.chunk_read_rw, table.wgsl_gen_write_field_code('displacement_bounds'), chunk_folded_workgroup_index]) });