@woosh/meep-engine 3.11.3 → 3.12.0

package/src/shade/renderer/rasterize/cull/hzb/shader_hzb_meshlet_filter_1way_full.js

@@ -1,117 +1,116 @@
-import { u32, vec2u } from "../../../shader/type/WEBGPU_WGSL_PRIMITIVE_TYPES.js";
-import { WebGPUArray } from "../../../shader/type/WebGPUArray.js";
-import { CAMERA_UNIFORM_STRUCT } from "../../../camera/CAMERA_UNIFORM_STRUCT.js";
-import { MESHLET_METADATA_STRUCT } from "../../../geometry/meshlet/encoding/MESHLET_METADATA_STRUCT.js";
-import { chunk_scene_database_access } from "../../../scene/chunk_scene_database_access.js";
-import {
-    chunk_query_depth_from_screen_space_bb
-} from "../../../shader/chunk/depth/chunk_query_depth_from_screen_space_bb.js";
-import { AABB3_STRUCT } from "../../../shader/chunk/geometry/aabb3/AABB3_STRUCT.js";
-import { chunk_aabb3_intersects_frustum } from "../../../shader/chunk/geometry/aabb3/chunk_aabb3_intersects_frustum.js";
-import { chunk_aabb3_project } from "../../../shader/chunk/geometry/aabb3/chunk_aabb3_project.js";
-import { chunk_aabb3_project_perspective } from "../../../shader/chunk/geometry/aabb3/chunk_aabb3_project_perspective.js";
-import { chunk_array_to_aabb3 } from "../../../shader/chunk/geometry/aabb3/chunk_array_to_aabb3.js";
-import {
-    chunk_aabb2_clip_overlaps_texel_centers
-} from "../../../shader/chunk/texture/chunk_aabb2_clip_overlaps_texel_centers.js";
-import { CodeChunk } from "../../../shader/compiler/CodeChunk.js";
-import { COMPUTE_SHADER_DEFAULT_GROUP_SIZE } from "../../../shader/COMPUTE_SHADER_DEFAULT_GROUP_SIZE.js";
-import { ComputeShader } from "../../../shader/ComputeShader.js";
-import { ShaderResourceSetDescriptor } from "../../../shader/resource/ShaderResourceSetDescriptor.js";
-import { MESHLET_COLLECTION_STRUCT } from "../../MESHLET_COLLECTION_STRUCT.js";
-import { MESHLET_COLLECTION_WRITE_STRUCT } from "../../MESHLET_COLLECTION_WRITE_STRUCT.js";
-
-const resources = new ShaderResourceSetDescriptor();
-
-resources.createGroup()
-    .addUniform("camera", CAMERA_UNIFORM_STRUCT)
-    .addUniform("resolution", vec2u)
-    .addStorageBuffer("scene_database", WebGPUArray.from(u32))
-    .addStorageBuffer("meshlet_metadata", WebGPUArray.from(MESHLET_METADATA_STRUCT))
-    .addTexture("texture_hzb")
-
-
-resources.createGroup()
-    .addStorageBuffer("input", MESHLET_COLLECTION_STRUCT)
-    .addStorageBuffer("output", MESHLET_COLLECTION_WRITE_STRUCT, true)
-
-
-const body = CodeChunk.from(
-// language=WGSL
-    `
-@compute @workgroup_size(${COMPUTE_SHADER_DEFAULT_GROUP_SIZE})
-fn unpack(@builtin(global_invocation_id) global_id : vec3<u32>){
-    
-    let input_index = global_id.x;
-    
-    if(input_index >= input.count){
-        return;
-    }
-    
-    let meshlet = input.elements[input_index];
-
-    let mesh = scene_read_mesh(&scene_database, meshlet.mesh);
-    let node = scene_read_node(&scene_database, mesh.node);
-
-    let meshlet_index = meshlet.index;
-
-    let bounds_array = meshlet_metadata[ meshlet_index ].bounds_box;
-
-    let meshlet_bounds = array_to_aabb3(bounds_array);
-
-    let world_bounds = aabb3_project(meshlet_bounds, node.global);
-    
-    if( !aabb3_intersects_frustum(world_bounds, camera.frustum) ){
-        // outside of view frustum
-        return ;
-    }
-    
-    var view_bounds: ${AABB3_STRUCT.wgsl_ref};
-    
-    let valid_bounds = aabb3_project_perspective(&view_bounds, world_bounds, camera.view_projection_matrix);
-        
-    let would_draw = aabb2_clip_overlaps_texel_centers(
-        view_bounds.min.xy,
-        view_bounds.max.xy,
-        resolution
-    );
-    
-    if(valid_bounds && !would_draw){
-        // cull, doesn't result in any draw
-        return;
-    }
-        
-    if(
-        valid_bounds && query_depth_from_screen_space_bb(view_bounds, texture_hzb) < 0.0
-    ){
-        // filter failed
-        return;
-    }
-
-    let write_offset = atomicAdd(&output.count, 1u);
-    
-    if(write_offset < arrayLength(&output.elements)){
-        output.elements[ write_offset ] = meshlet;
-    }
-}`,
-    [
-        AABB3_STRUCT.declaration_chunk,
-        chunk_array_to_aabb3,
-        chunk_aabb3_project,
-        chunk_aabb3_project_perspective,
-        chunk_query_depth_from_screen_space_bb,
-        chunk_aabb2_clip_overlaps_texel_centers,
-        chunk_aabb3_intersects_frustum,
-        chunk_scene_database_access,
-    ]);
-
-/**
- * Differs from {@link shader_hzb_meshlet_filter_1way} by using a full set of culling filters, the other one assumes some pre-filtering and only does HZB
- * @type {ComputeShader}
- */
-export const shader_hzb_meshlet_filter_1way_full = ComputeShader.from({
-    label: 'HZB / Filter Meshlets - full',
-    resources,
-    body,
-    group_size: [COMPUTE_SHADER_DEFAULT_GROUP_SIZE],
+import { u32, vec2u } from "../../../shader/type/WEBGPU_WGSL_PRIMITIVE_TYPES.js";
+import { WebGPUArray } from "../../../shader/type/WebGPUArray.js";
+import { CAMERA_UNIFORM_STRUCT } from "../../../camera/CAMERA_UNIFORM_STRUCT.js";
+import { MESHLET_METADATA_STRUCT } from "../../../geometry/meshlet/encoding/MESHLET_METADATA_STRUCT.js";
+import { chunk_scene_database_access } from "../../../scene/chunk_scene_database_access.js";
+import {
+    chunk_query_depth_from_screen_space_bb
+} from "../../../shader/chunk/depth/chunk_query_depth_from_screen_space_bb.js";
+import { AABB3_STRUCT } from "../../../shader/chunk/geometry/aabb3/AABB3_STRUCT.js";
+import { chunk_aabb3_intersects_frustum } from "../../../shader/chunk/geometry/aabb3/chunk_aabb3_intersects_frustum.js";
+import { chunk_aabb3_project } from "../../../shader/chunk/geometry/aabb3/chunk_aabb3_project.js";
+import { chunk_aabb3_project_perspective } from "../../../shader/chunk/geometry/aabb3/chunk_aabb3_project_perspective.js";
+import { chunk_array_to_aabb3 } from "../../../shader/chunk/geometry/aabb3/chunk_array_to_aabb3.js";
+import {
+    chunk_aabb2_clip_overlaps_texel_centers
+} from "../../../shader/chunk/texture/chunk_aabb2_clip_overlaps_texel_centers.js";
+import { CodeChunk } from "../../../shader/compiler/CodeChunk.js";
+import { COMPUTE_SHADER_DEFAULT_GROUP_SIZE } from "../../../shader/COMPUTE_SHADER_DEFAULT_GROUP_SIZE.js";
+import { ComputeShader } from "../../../shader/ComputeShader.js";
+import { ShaderResourceSetDescriptor } from "../../../shader/resource/ShaderResourceSetDescriptor.js";
+import { MESHLET_COLLECTION_STRUCT } from "../../MESHLET_COLLECTION_STRUCT.js";
+import { MESHLET_COLLECTION_WRITE_STRUCT } from "../../MESHLET_COLLECTION_WRITE_STRUCT.js";
+
+const resources = new ShaderResourceSetDescriptor();
+
+resources.createGroup()
+    .addUniform("camera", CAMERA_UNIFORM_STRUCT)
+    .addUniform("resolution", vec2u)
+    .addStorageBuffer("scene_database", WebGPUArray.from(u32))
+    .addStorageBuffer("meshlet_metadata", WebGPUArray.from(MESHLET_METADATA_STRUCT))
+    .addTexture("texture_hzb")
+
+
+resources.createGroup()
+    .addStorageBuffer("input", MESHLET_COLLECTION_STRUCT)
+    .addStorageBuffer("output", MESHLET_COLLECTION_WRITE_STRUCT, true)
+
+
+const body = CodeChunk.from(
+// language=WGSL
+    `
+@compute @workgroup_size(${COMPUTE_SHADER_DEFAULT_GROUP_SIZE})
+fn unpack(@builtin(global_invocation_id) global_id : vec3<u32>){
+    
+    let input_index = global_id.x;
+    
+    if(input_index >= input.count){
+        return;
+    }
+    
+    let meshlet = input.elements[input_index];
+
+    let mesh = scene_read_mesh(&scene_database, meshlet.mesh);
+    let node = scene_read_node(&scene_database, mesh.node);
+
+    let meshlet_index = meshlet.index;
+
+    let bounds_array = meshlet_metadata[ meshlet_index ].bounds_box;
+
+    let meshlet_bounds = array_to_aabb3(bounds_array);
+
+    let world_bounds = aabb3_project(meshlet_bounds, node.global);
+    
+    if( !aabb3_intersects_frustum(world_bounds, camera.frustum) ){
+        // outside of view frustum
+        return ;
+    }
+    
+    var view_bounds: ${AABB3_STRUCT.wgsl_ref};
+    
+    let valid_bounds = aabb3_project_perspective(&view_bounds, world_bounds, camera.view_projection_matrix);
+        
+    let would_draw = aabb2_clip_overlaps_texel_centers(
+        view_bounds.min.xy,
+        view_bounds.max.xy,
+        resolution
+    );
+    
+    if(valid_bounds && !would_draw){
+        // cull, doesn't result in any draw
+        return;
+    }
+        
+    if(
+        valid_bounds && query_depth_from_screen_space_bb(view_bounds, texture_hzb) < 0.0
+    ){
+        // filter failed
+        return;
+    }
+
+    let write_offset = atomicAdd(&output.count, 1u);
+    
+    if(write_offset < arrayLength(&output.elements)){
+        output.elements[ write_offset ] = meshlet;
+    }
+}`,
+    [
+        AABB3_STRUCT.declaration_chunk,
+        chunk_array_to_aabb3,
+        chunk_aabb3_project,
+        chunk_aabb3_project_perspective,
+        chunk_query_depth_from_screen_space_bb,
+        chunk_aabb2_clip_overlaps_texel_centers,
+        chunk_aabb3_intersects_frustum,
+        chunk_scene_database_access,
+    ]);
+
+/**
+ * Differs from {@link shader_hzb_meshlet_filter_1way} by using a full set of culling filters, the other one assumes some pre-filtering and only does HZB
+ * @type {ComputeShader}
+ */
+export const shader_hzb_meshlet_filter_1way_full = ComputeShader.from({
+    label: 'HZB / Filter Meshlets - full',
+    resources,
+    body,
 });

package/src/shade/renderer/rasterize/expand/bucket/mesh/shader_instances_classify_bucket.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"shader_instances_classify_bucket.d.ts","sourceRoot":"","sources":["../../../../../../../../src/shade/renderer/rasterize/expand/bucket/mesh/shader_instances_classify_bucket.js"],"names":[],"mappings":"AAsIA,6DAKE;8BA1H4B,qCAAqC"}
+{"version":3,"file":"shader_instances_classify_bucket.d.ts","sourceRoot":"","sources":["../../../../../../../../src/shade/renderer/rasterize/expand/bucket/mesh/shader_instances_classify_bucket.js"],"names":[],"mappings":"AAsIA,6DAIE;8BAzH4B,qCAAqC"}

package/src/shade/renderer/rasterize/expand/bucket/mesh/shader_instances_classify_bucket.js

@@ -1,140 +1,139 @@
-import { assert } from "../../../../../../core/assert.js";
-import { atomic_u32 } from "../../../../shader/type/WEBGPU_WGSL_PRIMITIVE_TYPES.js";
-import { WebGPUArray } from "../../../../shader/type/WebGPUArray.js";
-import { WebGPUStruct } from "../../../../shader/type/WebGPUStruct.js";
-import { GPU_DATABASE_BUFFER_ELEMENT_TYPE } from "../../../../buffer/table/GPU_DATABASE_BUFFER_ELEMENT_TYPE.js";
-import { CAMERA_UNIFORM_STRUCT } from "../../../../camera/CAMERA_UNIFORM_STRUCT.js";
-import { MATERIAL_METADATA_STRUCT } from "../../../../material/MATERIAL_METADATA_STRUCT.js";
-import { chunk_scene_database_access, MESHES_TABLE } from "../../../../scene/chunk_scene_database_access.js";
-import {
-    chunk_aabb3_intersects_frustum
-} from "../../../../shader/chunk/geometry/aabb3/chunk_aabb3_intersects_frustum.js";
-import { chunk_array_to_aabb3 } from "../../../../shader/chunk/geometry/aabb3/chunk_array_to_aabb3.js";
-import {
-    chunk_sphere_intersects_frustum
-} from "../../../../shader/chunk/geometry/sphere/chunk_sphere_intersects_frustum.js";
-import { CodeChunk } from "../../../../shader/compiler/CodeChunk.js";
-import { COMPUTE_SHADER_DEFAULT_GROUP_SIZE } from "../../../../shader/COMPUTE_SHADER_DEFAULT_GROUP_SIZE.js";
-import { ComputeShader } from "../../../../shader/ComputeShader.js";
-import { ShaderResourceSetDescriptor } from "../../../../shader/resource/ShaderResourceSetDescriptor.js";
-import { chunk_rasterization_material_bucket } from "../chunk_rasterization_material_bucket.js";
-import { RASTERIZATION_BUCKET_LIMIT } from "../RASTERIZATION_BUCKET_LIMIT.js";
-
-const GROUP_SIZE = COMPUTE_SHADER_DEFAULT_GROUP_SIZE;
-
-// Bucket reduction at the tail relies on lanes [0, RASTERIZATION_BUCKET_LIMIT)
-// covering every bucket in one pass.
-assert.lessThanOrEqual(RASTERIZATION_BUCKET_LIMIT, GROUP_SIZE);
-
-const PAGE_ITERATOR_PREFIX = MESHES_TABLE.page_iterator_symbol_prefix(GROUP_SIZE);
-
-const OUTPUT_STRUCT = WebGPUStruct.from({
-    total: atomic_u32,
-    counts: WebGPUArray.from(atomic_u32, RASTERIZATION_BUCKET_LIMIT),
-    elements: WebGPUArray.u32
-});
-
-const resources = new ShaderResourceSetDescriptor();
-
-resources.createGroup()
-    .addUniform("camera_current", CAMERA_UNIFORM_STRUCT)
-    .addStorageBuffer("output", OUTPUT_STRUCT, true)
-    .addStorageBuffer("scene_database", WebGPUArray.from(GPU_DATABASE_BUFFER_ELEMENT_TYPE))
-    .addStorageBuffer("materials", WebGPUArray.from(MATERIAL_METADATA_STRUCT))
-;
-
-// Per-page dispatch — see graph_mesh_frustum_cull for the iteration
-// shape. Adds a workgroup-shared bucket counter so per-bucket
-// atomicAdds collapse to one global update per (workgroup, bucket)
-// pair instead of one per visible mesh.
-const body = CodeChunk.from(
-    //language=WGSL
-    `
-var<workgroup> wg_buckets: array<atomic<u32>, ${RASTERIZATION_BUCKET_LIMIT}>;
-var<workgroup> wg_visible_count: atomic<u32>;
-var<workgroup> wg_global_offset: u32;
-
-@compute @workgroup_size(${GROUP_SIZE})
-fn main(
-    @builtin(workgroup_id) wg_id: vec3<u32>,
-    @builtin(local_invocation_index) local_id: u32,
-){
-    if !${PAGE_ITERATOR_PREFIX}_setup(&scene_database, wg_id.x, local_id) {
-        return;
-    }
-
-    let slot_in_page = ${PAGE_ITERATOR_PREFIX}_slot_in_page(local_id);
-    var local_offset = ~0u;
-
-    if slot_in_page < ${PAGE_ITERATOR_PREFIX}_ELEMENTS_PER_PAGE && ${PAGE_ITERATOR_PREFIX}_is_occupied(slot_in_page) {
-        let mesh = ${PAGE_ITERATOR_PREFIX}_read(&scene_database, slot_in_page);
-
-        if sphere_intersects_frustum(mesh.bounding_sphere, camera_current.frustum) {
-            let mesh_bounds = array_to_aabb3(mesh.bounding_box);
-            if aabb3_intersects_frustum(mesh_bounds, camera_current.frustum) {
-                // TODO cull by pixel overlap (see texture_space_bounds_overlaps_pixel_centers)
-
-                let material = materials[mesh.material];
-
-                let bucket_index = rasterization_material_bucket(
-                    material.transparency_mode,
-                    material.draw_mode,
-                    material.draw_side
-                );
-
-                atomicAdd(&wg_buckets[bucket_index], 1u);
-                local_offset = atomicAdd(&wg_visible_count, 1u);
-            }
-        }
-    }
-
-    workgroupBarrier();
-
-    // Lane 0 claims the workgroup's contiguous slice of output.elements.
-    if local_id == 0u {
-        let total = atomicLoad(&wg_visible_count);
-        if total > 0u {
-            wg_global_offset = atomicAdd(&output.total, total);
-        }
-    }
-
-    // Bucket reduction — one lane per bucket pushes its workgroup-local
-    // count to the global counts array. Independent of the lane-0 work
-    // above, so they run concurrently in this region between barriers.
-    if local_id < ${RASTERIZATION_BUCKET_LIMIT}u {
-        let bucket_index = local_id;
-        let count = atomicLoad(&wg_buckets[bucket_index]);
-        if count > 0u {
-            atomicAdd(&output.counts[bucket_index], count);
-        }
-    }
-
-    workgroupBarrier();
-
-    if local_offset == ~0u { return; }
-
-    let mesh_index = ${PAGE_ITERATOR_PREFIX}_slot_to_index(slot_in_page);
-    let output_index = wg_global_offset + local_offset;
-    let output_size = arrayLength(&output.elements);
-
-    if output_index < output_size {
-        output.elements[output_index] = mesh_index;
-    }
-}
-    `,
-    [
-        chunk_rasterization_material_bucket,
-        chunk_aabb3_intersects_frustum,
-        chunk_sphere_intersects_frustum,
-        chunk_array_to_aabb3,
-        chunk_scene_database_access,
-        MESHES_TABLE.chunk_iterate(GROUP_SIZE),
-    ]
-)
-
-export const shader_instances_classify_bucket = ComputeShader.from({
-    label: "classify instance bucket",
-    resources,
-    body,
-    group_size: [GROUP_SIZE],
-})
+import { assert } from "../../../../../../core/assert.js";
+import { atomic_u32 } from "../../../../shader/type/WEBGPU_WGSL_PRIMITIVE_TYPES.js";
+import { WebGPUArray } from "../../../../shader/type/WebGPUArray.js";
+import { WebGPUStruct } from "../../../../shader/type/WebGPUStruct.js";
+import { GPU_DATABASE_BUFFER_ELEMENT_TYPE } from "../../../../buffer/table/GPU_DATABASE_BUFFER_ELEMENT_TYPE.js";
+import { CAMERA_UNIFORM_STRUCT } from "../../../../camera/CAMERA_UNIFORM_STRUCT.js";
+import { MATERIAL_METADATA_STRUCT } from "../../../../material/MATERIAL_METADATA_STRUCT.js";
+import { chunk_scene_database_access, MESHES_TABLE } from "../../../../scene/chunk_scene_database_access.js";
+import {
+    chunk_aabb3_intersects_frustum
+} from "../../../../shader/chunk/geometry/aabb3/chunk_aabb3_intersects_frustum.js";
+import { chunk_array_to_aabb3 } from "../../../../shader/chunk/geometry/aabb3/chunk_array_to_aabb3.js";
+import {
+    chunk_sphere_intersects_frustum
+} from "../../../../shader/chunk/geometry/sphere/chunk_sphere_intersects_frustum.js";
+import { CodeChunk } from "../../../../shader/compiler/CodeChunk.js";
+import { COMPUTE_SHADER_DEFAULT_GROUP_SIZE } from "../../../../shader/COMPUTE_SHADER_DEFAULT_GROUP_SIZE.js";
+import { ComputeShader } from "../../../../shader/ComputeShader.js";
+import { ShaderResourceSetDescriptor } from "../../../../shader/resource/ShaderResourceSetDescriptor.js";
+import { chunk_rasterization_material_bucket } from "../chunk_rasterization_material_bucket.js";
+import { RASTERIZATION_BUCKET_LIMIT } from "../RASTERIZATION_BUCKET_LIMIT.js";
+
+const GROUP_SIZE = COMPUTE_SHADER_DEFAULT_GROUP_SIZE;
+
+// Bucket reduction at the tail relies on lanes [0, RASTERIZATION_BUCKET_LIMIT)
+// covering every bucket in one pass.
+assert.lessThanOrEqual(RASTERIZATION_BUCKET_LIMIT, GROUP_SIZE);
+
+const PAGE_ITERATOR_PREFIX = MESHES_TABLE.page_iterator_symbol_prefix(GROUP_SIZE);
+
+const OUTPUT_STRUCT = WebGPUStruct.from({
+    total: atomic_u32,
+    counts: WebGPUArray.from(atomic_u32, RASTERIZATION_BUCKET_LIMIT),
+    elements: WebGPUArray.u32
+});
+
+const resources = new ShaderResourceSetDescriptor();
+
+resources.createGroup()
+    .addUniform("camera_current", CAMERA_UNIFORM_STRUCT)
+    .addStorageBuffer("output", OUTPUT_STRUCT, true)
+    .addStorageBuffer("scene_database", WebGPUArray.from(GPU_DATABASE_BUFFER_ELEMENT_TYPE))
+    .addStorageBuffer("materials", WebGPUArray.from(MATERIAL_METADATA_STRUCT))
+;
+
+// Per-page dispatch — see graph_mesh_frustum_cull for the iteration
+// shape. Adds a workgroup-shared bucket counter so per-bucket
+// atomicAdds collapse to one global update per (workgroup, bucket)
+// pair instead of one per visible mesh.
+const body = CodeChunk.from(
+    //language=WGSL
+    `
+var<workgroup> wg_buckets: array<atomic<u32>, ${RASTERIZATION_BUCKET_LIMIT}>;
+var<workgroup> wg_visible_count: atomic<u32>;
+var<workgroup> wg_global_offset: u32;
+
+@compute @workgroup_size(${GROUP_SIZE})
+fn main(
+    @builtin(workgroup_id) wg_id: vec3<u32>,
+    @builtin(local_invocation_index) local_id: u32,
+){
+    if !${PAGE_ITERATOR_PREFIX}_setup(&scene_database, wg_id.x, local_id) {
+        return;
+    }
+
+    let slot_in_page = ${PAGE_ITERATOR_PREFIX}_slot_in_page(local_id);
+    var local_offset = ~0u;
+
+    if slot_in_page < ${PAGE_ITERATOR_PREFIX}_ELEMENTS_PER_PAGE && ${PAGE_ITERATOR_PREFIX}_is_occupied(slot_in_page) {
+        let mesh = ${PAGE_ITERATOR_PREFIX}_read(&scene_database, slot_in_page);
+
+        if sphere_intersects_frustum(mesh.bounding_sphere, camera_current.frustum) {
+            let mesh_bounds = array_to_aabb3(mesh.bounding_box);
+            if aabb3_intersects_frustum(mesh_bounds, camera_current.frustum) {
+                // TODO cull by pixel overlap (see texture_space_bounds_overlaps_pixel_centers)
+
+                let material = materials[mesh.material];
+
+                let bucket_index = rasterization_material_bucket(
+                    material.transparency_mode,
+                    material.draw_mode,
+                    material.draw_side
+                );
+
+                atomicAdd(&wg_buckets[bucket_index], 1u);
+                local_offset = atomicAdd(&wg_visible_count, 1u);
+            }
+        }
+    }
+
+    workgroupBarrier();
+
+    // Lane 0 claims the workgroup's contiguous slice of output.elements.
+    if local_id == 0u {
+        let total = atomicLoad(&wg_visible_count);
+        if total > 0u {
+            wg_global_offset = atomicAdd(&output.total, total);
+        }
+    }
+
+    // Bucket reduction — one lane per bucket pushes its workgroup-local
+    // count to the global counts array. Independent of the lane-0 work
+    // above, so they run concurrently in this region between barriers.
+    if local_id < ${RASTERIZATION_BUCKET_LIMIT}u {
+        let bucket_index = local_id;
+        let count = atomicLoad(&wg_buckets[bucket_index]);
+        if count > 0u {
+            atomicAdd(&output.counts[bucket_index], count);
+        }
+    }
+
+    workgroupBarrier();
+
+    if local_offset == ~0u { return; }
+
+    let mesh_index = ${PAGE_ITERATOR_PREFIX}_slot_to_index(slot_in_page);
+    let output_index = wg_global_offset + local_offset;
+    let output_size = arrayLength(&output.elements);
+
+    if output_index < output_size {
+        output.elements[output_index] = mesh_index;
+    }
+}
+    `,
+    [
+        chunk_rasterization_material_bucket,
+        chunk_aabb3_intersects_frustum,
+        chunk_sphere_intersects_frustum,
+        chunk_array_to_aabb3,
+        chunk_scene_database_access,
+        MESHES_TABLE.chunk_iterate(GROUP_SIZE),
+    ]
+)
+
+export const shader_instances_classify_bucket = ComputeShader.from({
+    label: "classify instance bucket",
+    resources,
+    body,
+})

package/src/shade/renderer/rasterize/expand/bucket/mesh/shader_instances_count_by_rasterization_bucket.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"shader_instances_count_by_rasterization_bucket.d.ts","sourceRoot":"","sources":["../../../../../../../../src/shade/renderer/rasterize/expand/bucket/mesh/shader_instances_count_by_rasterization_bucket.js"],"names":[],"mappings":"AAuFA,2EAKG;8BAnF2B,qCAAqC"}
+{"version":3,"file":"shader_instances_count_by_rasterization_bucket.d.ts","sourceRoot":"","sources":["../../../../../../../../src/shade/renderer/rasterize/expand/bucket/mesh/shader_instances_count_by_rasterization_bucket.js"],"names":[],"mappings":"AAuFA,2EAIG;8BAlF2B,qCAAqC"}