@woosh/meep-engine 3.39.0 → 3.41.0

package/src/shade/renderer/shadow/vsm/vms_pipeline_compute_coarse_bounds.js

@@ -1,184 +0,0 @@
-import { UINT32_MAX } from "../../../../core/binary/UINT32_MAX.js";
-import { BufferBindingLayout } from "../../../descriptor/binding/BufferBindingLayout.js";
-import { ComputePipelineDescriptor } from "../../../descriptor/pipeline/compute/ComputePipelineDescriptor.js";
-import { u32, vec2u } from "../../shader/type/WEBGPU_WGSL_PRIMITIVE_TYPES.js";
-import { WebGPUArray } from "../../shader/type/WebGPUArray.js";
-import { WebGPUStruct } from "../../shader/type/WebGPUStruct.js";
-import { chunk_split_by_2 } from "../../shader/chunk/binary/chunk_split_by_2.js";
-import { VSM_UNIFORMS_STRUCT } from "./VSM_UNIFORMS_STRUCT.js";
-
-export const STRUCT_AABB2_UINT32 = WebGPUStruct.from({
-    min: vec2u,
-    max: vec2u,
-}, "AABB2_UINT32");
-
-//language=WGSL
-const vsm_shader_compute_coarse_bounds = `
-
-${chunk_split_by_2}
-
-struct VitrualTextureTileAddress{
-    lod: u32,
-    x: u32,
-    y: u32
-}
-
-fn vt_tile_compute_texel_bounds( tile: VitrualTextureTileAddress , virtual_resolution:u32)-> ${(STRUCT_AABB2_UINT32.wgsl_ref)}{
-    let tile_size = virtual_resolution >> tile.lod; 
-    
-    let bb_min = vec2(tile.x * tile_size, tile.y * tile_size);
-    return ${(STRUCT_AABB2_UINT32.wgsl_ref)}(
-        bb_min,
-        bb_min+tile_size,
-    );
-}
-
-fn vt_tile_index_to_address(index: u32) -> VitrualTextureTileAddress{
-    for(var lod = 0u; lod < 32u; lod++){
-       
-        let resolution = 1u << lod;
-    
-        let mip_mask = resolution - 1;
-        
-        let index_offset = split_by_2(mip_mask);
-
-        let next_mip_mask = (mip_mask << 1u) | 0x1;
-
-        let index_offset_next = split_by_2(next_mip_mask);
-
-        if (index >= index_offset_next) {
-            continue;
-        }
-
-        let local_index = index - index_offset;
-
-        let y = (local_index / resolution) | 0;
-        let x = local_index % resolution;
-        
-        return VitrualTextureTileAddress(lod, x, y);
-    }
-    
-    return VitrualTextureTileAddress(0,0,0);
-}
-
-${(STRUCT_AABB2_UINT32.wgsl_declaration)}
-${(VSM_UNIFORMS_STRUCT.wgsl_declaration)}
-
-
-struct Output{
-    bounds: array<atomic<u32>,4>
-}
-
-@group(0) @binding(0) var<uniform> vsm: ${(VSM_UNIFORMS_STRUCT.wgsl_ref)};
-@group(0) @binding(1) var<storage, read> residence: array<u32>;
-
-@group(1) @binding(0) var<storage, read_write> output: Output;
-
-@compute @workgroup_size(256,1,1)
-fn main(@builtin(global_invocation_id) global_id : vec3<u32>){
-    let max_mip_level = countTrailingZeros(vsm.resolution_in_tiles); // fast log2 alternative
-    
-    let total_tiles = split_by_2((1u << (max_mip_level + 1u)) - 1);
-    
-    let tile_index = global_id.x;
-    
-    if(tile_index >= total_tiles){
-        return;
-    }
-    
-    let usage_count = residence[tile_index];
-    
-    if(usage_count == 0u){
-        return;
-    }
-    
-    let virtual_resolution = vsm.resolution_in_tiles*vsm.tile_resolution;
-    
-    let tile = vt_tile_index_to_address(tile_index);
-    // compute tile bounds
-    let tile_bounds = vt_tile_compute_texel_bounds(tile, virtual_resolution);
-    
-    atomicMin(&output.bounds[0], tile_bounds.min.x);
-    atomicMin(&output.bounds[1], tile_bounds.min.y);
-    
-    atomicMax(&output.bounds[2], tile_bounds.max.x);
-    atomicMax(&output.bounds[3], tile_bounds.max.y);
-}
-`;
-
-export const vms_pipeline_compute_coarse_bounds = ComputePipelineDescriptor.simplified({
-    label: "VSM / Compute Coarse Bounds",
-    code: vsm_shader_compute_coarse_bounds,
-    bindings: [
-        [
-            BufferBindingLayout.uniform,
-            BufferBindingLayout.readOnlyStorage,
-        ],
-        [
-            BufferBindingLayout.storage
-        ]
-    ]
-});
-
-
-export class VSMCoarseBoundsPassData{
-    bounds = -1
-    residence = -1
-
-    /**
-     * @type {VirtualShadowMap}
-     */
-    map
-}
-
-const INITIAL_BOUNDS_AABB2 = new Uint32Array([UINT32_MAX, UINT32_MAX, 0, 0]);
-
-/**
- *
- * @param {VSMCoarseBoundsPassData} data
- * @param {FramePassResources} resources
- * @param {GraphicsExecutionContext} context
- */
-export function vsm_coarse_bounds_pass(data, resources, context) {
-
-    const encoder = context.encoder;
-
-    const buffer_texture_bounds_coarse = resources.get(data.bounds);
-
-    const map = data.map;
-
-    // Write exactly four words at this point in the command stream. A pooled transient buffer
-    // may be larger than the bounds and is not necessarily usable as a copy source.
-    encoder.writeValueBuffer(
-        buffer_texture_bounds_coarse, 0,
-        WebGPUArray.from(u32, 4), INITIAL_BOUNDS_AABB2
-    );
-
-    /**
-     * @type {GPUBuffer}
-     */
-    const residence_buffer = resources.get(data.residence);
-
-    const bounds_pass = encoder.constructComputePass({
-        pipeline: vms_pipeline_compute_coarse_bounds,
-        bindings: [
-            [
-                { buffer: map.buffer_uniforms },
-                { buffer: residence_buffer }
-            ],
-            [
-                { buffer: buffer_texture_bounds_coarse, size: STRUCT_AABB2_UINT32.size }
-            ]
-        ]
-    });
-
-    bounds_pass.dispatchWorkgroups(Math.ceil(map.total_tile_count / 256));
-
-    bounds_pass.end();
-
-    // const debug_buffer = this.#debug_buffer;
-    // encoder.copyBufferToBuffer(buffer_texture_bounds_coarse, 0, debug_buffer, 0, STRUCT_AABB2_UINT32.size);
-    //
-    // const device = this.#graphics.device;
-    // debug_inspect_coarse_bounds_buffer(debug_buffer, device, map);
-}

package/src/shade/renderer/shadow/vsm/vsm_residence_pipeline.d.ts

@@ -1,25 +0,0 @@
-/**
- *
- * @param {VSMTileUsagePassData} data
- * @param {FramePassResources} resources
- * @param {GraphicsExecutionContext} context
- */
-export function vsm_time_usage_data(data: VSMTileUsagePassData, resources: FramePassResources, context: GraphicsExecutionContext): void;
-export const vsm_residence_pipeline: ComputePipelineDescriptor;
-export class VSMTileUsagePassData {
-    residence: number;
-    /**
-     * @type {VirtualShadowMap}
-     */
-    map: VirtualShadowMap;
-    /**
-     * @type {RenderTarget}
-     */
-    viz: RenderTarget;
-    /**
-     * @type {GPUViewContext}
-     */
-    main_view: GPUViewContext;
-}
-import { ComputePipelineDescriptor } from "../../../descriptor/pipeline/compute/ComputePipelineDescriptor.js";
-//# sourceMappingURL=vsm_residence_pipeline.d.ts.map

package/src/shade/renderer/shadow/vsm/vsm_residence_pipeline.d.ts.map

@@ -1 +0,0 @@
-{"version":3,"file":"vsm_residence_pipeline.d.ts","sourceRoot":"","sources":["../../../../../../src/shade/renderer/shadow/vsm/vsm_residence_pipeline.js"],"names":[],"mappings":"AAyKA;;;;;GAKG;AACH,0CAJW,oBAAoB,0EAyD9B;AA9FD,+DAeE;AAEF;IACI,kBAAc;IACd;;OAEG;IACH,sBAAG;IAEH;;OAEG;IACH,kBAAG;IAEH;;OAEG;IACH,0BAAS;CACZ;0CArKyC,mEAAmE"}

package/src/shade/renderer/shadow/vsm/vsm_residence_pipeline.js

@@ -1,230 +0,0 @@
-import { BufferBindingLayout } from "../../../descriptor/binding/BufferBindingLayout.js";
-import { TextureBindingLayout } from "../../../descriptor/binding/TextureBindingLayout.js";
-import { ComputePipelineDescriptor } from "../../../descriptor/pipeline/compute/ComputePipelineDescriptor.js";
-import { CAMERA_UNIFORM_STRUCT } from "../../camera/CAMERA_UNIFORM_STRUCT.js";
-import { chunk_split_by_2 } from "../../shader/chunk/binary/chunk_split_by_2.js";
-import { chunk_project_position_from_depth } from "../../shader/chunk/camera/chunk_project_position_from_depth.js";
-import { chunk_v3_matrix4_project } from "../../shader/chunk/geometry/vec3/chunk_v3_matrix4_project.js";
-import { chunk_ndc_to_uv } from "../../shader/chunk/texture/chunk_ndc_to_uv.js";
-import { CodeChunk } from "../../shader/compiler/CodeChunk.js";
-import { VSM_RESIDENCE_WORKGROUP_TILE_RESOLUTION } from "./VSM_RESIDENCE_WORKGROUP_TILE_RESOLUTION.js";
-import { VSM_UNIFORMS_STRUCT } from "./VSM_UNIFORMS_STRUCT.js";
-
-const shader_code = CodeChunk.from(
-    //language=WGSL
-    `
-
-${(VSM_UNIFORMS_STRUCT.wgsl_declaration)}
-
-${(CAMERA_UNIFORM_STRUCT.wgsl_declaration)}
-
-struct Ocupancy{
-    counts: array<atomic<u32>>
-}
-
-@group(0) @binding(0) var<uniform> vsm: ${VSM_UNIFORMS_STRUCT.wgsl_ref};
-@group(0) @binding(1) var<uniform> main_camera: ${CAMERA_UNIFORM_STRUCT.wgsl_ref};
-@group(0) @binding(2) var<uniform> shadow_camera: ${CAMERA_UNIFORM_STRUCT.wgsl_ref};
-@group(0) @binding(3) var texture_depth: texture_depth_2d;
-
-@group(1) @binding(0) var<storage, read_write> output: Ocupancy;
-
-
-
-
-
-fn getWorldFromUVDepth(uv:vec2<f32>,z:f32)->vec3<f32>{
-    return project_position_from_depth(uv, z, main_camera.view_projection_matrix_inverse);
-}
-
-${chunk_split_by_2}
-
-fn vt_compose_tile_address(lod: u32, x:u32, y:u32 ) -> u32 {
-let lod_resolution = 1u << lod;
-
-let lod_mask = lod_resolution - 1u;
-
-let index_offset = split_by_2(lod_mask);
-
-return index_offset + y * lod_resolution + x;
-}
-
-@compute @workgroup_size(${VSM_RESIDENCE_WORKGROUP_TILE_RESOLUTION}, ${VSM_RESIDENCE_WORKGROUP_TILE_RESOLUTION}, 1)
-fn main(@builtin(global_invocation_id) global_id : vec3<u32>){
-
-    let input_texture_size = textureDimensions(texture_depth);
-    
-    if(global_id.x >= input_texture_size.x || global_id.y >= input_texture_size.y){
-        // out of bounds
-        return;
-    }
-    
-    let pixel_coordinates = global_id.xy;
-        
-    let depth_sample = textureLoad(texture_depth, pixel_coordinates, 0);
-    // The renderer clears reverse-Z depth to zero: these pixels have no receiver surface.
-    if (depth_sample == 0.0) {
-        return;
-    }
-    
-    // - we need to compute world bounds of the pixel based on the depth sample    
-    
-    let view_pixel_center = vec2f(pixel_coordinates) + 0.5;
-    let world_center = getWorldFromUVDepth(view_pixel_center / vec2f(input_texture_size), depth_sample);
-    let shadow_clip = shadow_camera.view_projection_matrix * vec4f(world_center, 1.0);
-    if (shadow_clip.w <= 0.0 || shadow_clip.z < 0.0 || shadow_clip.z > shadow_clip.w) {
-        return;
-    }
-    
-    let world_00 =  getWorldFromUVDepth( ( view_pixel_center + vec2(-0.5, -0.5)) / vec2f(input_texture_size), depth_sample);
-    let world_10 =  getWorldFromUVDepth( ( view_pixel_center + vec2( 0.5, -0.5)) / vec2f(input_texture_size), depth_sample);
-    let world_01 =  getWorldFromUVDepth( ( view_pixel_center + vec2( -0.5, 0.5)) / vec2f(input_texture_size), depth_sample);
-    let world_11 =  getWorldFromUVDepth( ( view_pixel_center + vec2( 0.5, 0.5)) / vec2f(input_texture_size), depth_sample);
-    
-    // project world coordinates into light space
-    let shadow_p_00 = v3_matrix4_project(world_00,shadow_camera.view_projection_matrix);
-    let shadow_p_10 = v3_matrix4_project(world_10,shadow_camera.view_projection_matrix);
-    let shadow_p_01 = v3_matrix4_project(world_01,shadow_camera.view_projection_matrix);
-    let shadow_p_11 = v3_matrix4_project(world_11,shadow_camera.view_projection_matrix);
-    
-    
-    //
-    let shadow_texel_bounds_min = min(min(shadow_p_00.xy, shadow_p_10.xy), min(shadow_p_01.xy, shadow_p_11.xy));
-    let shadow_texel_bounds_max = max(max(shadow_p_00.xy, shadow_p_10.xy), max(shadow_p_01.xy, shadow_p_11.xy));
-    
-        
-    // - get UV coordinate of the pixel in shadowmap space
-    let shadow_uv = ndc_to_uv(shadow_clip.xy / shadow_clip.w);
-    
-    if(shadow_uv.x < 0.0 || shadow_uv.x >= 1.0 || shadow_uv.y < 0.0 || shadow_uv.y >= 1.0){
-        // outside of the shadow
-        return;
-    }
-    
-    let shadowmap_resolution = vsm.tile_resolution * vsm.resolution_in_tiles;
-        
-    let shadow_dxy = (shadow_texel_bounds_max - shadow_texel_bounds_min)*0.5;
-    
-    // An isotropic mip must account for the larger axis of a rectangular pixel footprint.
-    let shadow_pixel_size = max(1.0, f32(shadowmap_resolution) * max(shadow_dxy.x, shadow_dxy.y));
-    
-    // - figure out LOD based on the UV bounds
-    // A grid with 2^k tiles has k+1 levels, including both the root and the finest grid.
-    let shadow_lod_max = countTrailingZeros(vsm.resolution_in_tiles);
-    
-    let mip_level = clamp( u32( floor( log2(shadow_pixel_size) ))  , 0u, shadow_lod_max  ) ;
-    let shadow_lod = shadow_lod_max - mip_level;
-    
-    // figure out tile XY
-    let lod_resolution_in_tiles =  1u << shadow_lod;
-    
-    let tile_coordinates = vec2u( floor( vec2(f32(lod_resolution_in_tiles)) * shadow_uv ) );
-    
-    let tile_index = vt_compose_tile_address(shadow_lod, tile_coordinates.x, tile_coordinates.y);
-    
-    atomicAdd(&output.counts[tile_index],1u);
-}
-`,
-    [
-        chunk_project_position_from_depth,
-        chunk_v3_matrix4_project,
-        chunk_ndc_to_uv,
-    ]
-).compile().text;
-
-export const vsm_residence_pipeline = ComputePipelineDescriptor.simplified({
-    label: 'VSM compute residence',
-
-    code: shader_code,
-    bindings: [
-        [
-            BufferBindingLayout.uniform,
-            BufferBindingLayout.uniform,
-            BufferBindingLayout.uniform,
-            TextureBindingLayout.depth
-        ],
-        [
-            BufferBindingLayout.storage
-        ]
-    ]
-})
-
-export class VSMTileUsagePassData {
-    residence = -1
-    /**
-     * @type {VirtualShadowMap}
-     */
-    map
-
-    /**
-     * @type {RenderTarget}
-     */
-    viz
-
-    /**
-     * @type {GPUViewContext}
-     */
-    main_view
-}
-
-/**
- *
- * @param {VSMTileUsagePassData} data
- * @param {FramePassResources} resources
- * @param {GraphicsExecutionContext} context
- */
-export function vsm_time_usage_data(data, resources, context) {
-
-    const encoder = context.encoder;
-
-    /**
-     *
-     * @type {GPUBuffer}
-     */
-    const residence_buffer = resources.get(data.residence);
-
-    const map = data.map;
-    const viz = data.viz;
-    const main_view = data.main_view;
-
-    const depth_texture = viz.depth_attachment;
-
-    // Demand belongs to this frame, including when the graph reuses a buffer from its pool.
-    encoder.clearBuffer(residence_buffer);
-
-    const residence_pass = encoder.constructComputePass({
-        label: "VSM / Compute Residence",
-        pipeline: vsm_residence_pipeline,
-        bindings: [
-            [
-                { buffer: map.buffer_uniforms },
-                { buffer: main_view.camera.gpu_buffer },
-                { buffer: map.light_views[0].gpu_buffer },
-                depth_texture.obtainView()
-            ],
-            [
-                { buffer: residence_buffer }
-            ]
-        ]
-    });
-
-    const viewport_resolution_x = depth_texture.size[0];
-    const viewport_resolution_y = depth_texture.size[1];
-
-    const res_x = Math.ceil(viewport_resolution_x / VSM_RESIDENCE_WORKGROUP_TILE_RESOLUTION);
-    const res_y = Math.ceil(viewport_resolution_y / VSM_RESIDENCE_WORKGROUP_TILE_RESOLUTION);
-
-    residence_pass.dispatchWorkgroups(res_x, res_y);
-
-    residence_pass.end();
-
-
-    // const debug_buffer = this.#debug_buffer;
-    // encoder.copyBufferToBuffer(residence_buffer, 0, debug_buffer, 0, residence_buffer_size);
-
-    // const device = this.#graphics.device;
-
-    // debug_inspect_residence_buffer(debug_buffer, device, map, viewport_resolution_x, viewport_resolution_y);
-
-}
-

package/src/shade/renderer/shadow/vsm/vsm_tile_allocation_pipeline.d.ts

@@ -1,19 +0,0 @@
-/**
- *
- * @param {VSMTileAllocationPassData} data
- * @param {FramePassResources} resources
- * @param {GraphicsExecutionContext} context
- */
-export function vsm_allocate_tiles_pass(data: VSMTileAllocationPassData, resources: FramePassResources, context: GraphicsExecutionContext): void;
-export const vsm_tile_allocation_pipeline: ComputePipelineDescriptor;
-export class VSMTileAllocationPassData {
-    allocation_count: number;
-    residence: number;
-    lookup: number;
-    /**
-     * @type {VirtualShadowMap}
-     */
-    map: VirtualShadowMap;
-}
-import { ComputePipelineDescriptor } from "../../../descriptor/pipeline/compute/ComputePipelineDescriptor.js";
-//# sourceMappingURL=vsm_tile_allocation_pipeline.d.ts.map

package/src/shade/renderer/shadow/vsm/vsm_tile_allocation_pipeline.d.ts.map

@@ -1 +0,0 @@
-{"version":3,"file":"vsm_tile_allocation_pipeline.d.ts","sourceRoot":"","sources":["../../../../../../src/shade/renderer/shadow/vsm/vsm_tile_allocation_pipeline.js"],"names":[],"mappings":"AAmFA;;;;;GAKG;AACH,8CAJW,yBAAyB,0EAyCnC;AArED,qEAaG;AAEH;IACI,yBAAqB;IACrB,kBAAc;IACd,eAAW;IAEX;;OAEG;IACH,sBAAG;CACN;0CA/EyC,mEAAmE"}

package/src/shade/renderer/shadow/vsm/vsm_tile_allocation_pipeline.js

@@ -1,127 +0,0 @@
-import { UINT32_MAX } from "../../../../core/binary/UINT32_MAX.js";
-import { BufferBindingLayout } from "../../../descriptor/binding/BufferBindingLayout.js";
-import { ComputePipelineDescriptor } from "../../../descriptor/pipeline/compute/ComputePipelineDescriptor.js";
-import { chunk_split_by_2 } from "../../shader/chunk/binary/chunk_split_by_2.js";
-import { VSM_UNIFORMS_STRUCT } from "./VSM_UNIFORMS_STRUCT.js";
-
-//language=WGSL
-const code = `
-
-${(VSM_UNIFORMS_STRUCT.wgsl_declaration)}
-${chunk_split_by_2}
-
-@group(0) @binding(0) var<uniform> vsm: ${(VSM_UNIFORMS_STRUCT.wgsl_ref)};
-@group(0) @binding(1) var<storage, read> residence: array<u32>;
-
-@group(1) @binding(0) var<storage, read_write> counter: atomic<u32>;
-@group(1) @binding(1) var<storage, read_write> allocation: array<u32>;
-
-const INVALID_ALLOCATION: u32 = ${UINT32_MAX}u;
-
-@compute @workgroup_size(256,1,1)
-fn main(@builtin(global_invocation_id) global_id : vec3<u32>){
-
-    let max_mip_level = countTrailingZeros(vsm.resolution_in_tiles); // fast log2 alternative
-    
-    let total_tiles = split_by_2((1u << (max_mip_level + 1u)) - 1);
-    
-    let tile_index = global_id.x;
-    
-    if(tile_index >= total_tiles){
-        return;
-    }
-    
-    let usage_count = residence[tile_index];
-    
-    if(usage_count == 0u){
-        allocation[tile_index] = INVALID_ALLOCATION;
-        return;
-    }
-    
-    // Only successful reservations advance the count. An unconditional atomicAdd would return
-    // pages outside the atlas, and clamping the count afterwards could grant duplicate pages.
-    var page = atomicLoad(&counter);
-    loop {
-        if (page >= vsm.physical_tile_count) {
-            allocation[tile_index] = INVALID_ALLOCATION;
-            return;
-        }
-        let reservation = atomicCompareExchangeWeak(&counter, page, page + 1u);
-        if (reservation.exchanged) {
-            allocation[tile_index] = page;
-            return;
-        }
-        page = reservation.old_value;
-    }
-}
-`;
-export const vsm_tile_allocation_pipeline = ComputePipelineDescriptor.simplified({
-    label: 'VSM / Tile Allocation',
-    code: code,
-    bindings: [
-        [
-            BufferBindingLayout.uniform,
-            BufferBindingLayout.readOnlyStorage,
-        ],
-        [
-            BufferBindingLayout.storage,
-            BufferBindingLayout.storage,
-        ]
-    ]
-});
-
-export class VSMTileAllocationPassData{
-    allocation_count = -1
-    residence = -1
-    lookup = -1
-
-    /**
-     * @type {VirtualShadowMap}
-     */
-    map
-}
-
-/**
- *
- * @param {VSMTileAllocationPassData} data
- * @param {FramePassResources} resources
- * @param {GraphicsExecutionContext} context
- */
-export function vsm_allocate_tiles_pass(data, resources, context) {
-    const encoder = context.encoder;
-
-    const allocation_count = resources.get(data.allocation_count);
-    const residence_buffer = resources.get(data.residence);
-    const lookup_buffer = resources.get(data.lookup);
-
-    const map = data.map;
-
-    encoder.clearBuffer(allocation_count);
-
-    const allocation_pass = encoder.constructComputePass({
-        pipeline: vsm_tile_allocation_pipeline,
-        bindings: [
-            [
-                { buffer: map.buffer_uniforms },
-                { buffer: residence_buffer }
-            ],
-            [
-                { buffer: allocation_count },
-                { buffer: lookup_buffer }
-            ]
-        ]
-    });
-
-    allocation_pass.dispatchWorkgroups(Math.ceil(map.total_tile_count / 256));
-
-    allocation_pass.end();
-
-    // inspect_gpu_buffer_typed({
-    //     buffer: allocation_count,
-    //     device:graphics.device,
-    //     type: WEBGPU_WGSL_PRIMITIVE_TYPES.u32,
-    //     count: 1
-    // }).then(arr => {
-    //     console.log(`Allocated tiles: ${arr[0]}`)
-    // });
-}