@woosh/meep-engine 3.39.0 → 3.40.0

package/src/shade/renderer/postprocess/ssr/ssr_reproject_shader.js

@@ -1,105 +1,113 @@
-import { CAMERA_UNIFORM_STRUCT } from "../../camera/CAMERA_UNIFORM_STRUCT.js";
-import { chunk_screen_coordinate_to_world } from "../../shader/chunk/camera/chunk_screen_coordinate_to_world.js";
-import { chunk_get_view_space_depth } from "../../shader/chunk/depth/chunk_get_view_space_depth.js";
-import { chunk_decode_g_buffer_normal } from "../../shader/chunk/geometry/normal/chunk_decode_g_buffer_normal.js";
-import { chunk_decode_g_buffer_roughness } from "../../shader/chunk/material/chunk_decode_g_buffer_roughness.js";
-import { CodeChunk } from "../../shader/compiler/CodeChunk.js";
-import { ImageShader } from "../../shader/ImageShader.js";
-import { ShaderResourceSetDescriptor } from "../../shader/resource/ShaderResourceSetDescriptor.js";
-import { ShaderDescriptor } from "../../shader/ShaderDescriptor.js";
-import { WebGPUStruct } from "../../shader/type/WebGPUStruct.js";
-import { f32, u32, vec2f } from "../../shader/type/WEBGPU_WGSL_PRIMITIVE_TYPES.js";
-import { chunk_taa_velocity_confidence } from "../taa/shader_taa.js";
-import { chunk_depth_clip_offset_codec } from "../taa/chunk_depth_clip_offset_codec.js";
-import { chunk_ssr_reprojection } from "./reproject/chunk_ssr_reprojection.js";
-import { chunk_ssr_sample_history } from "./reproject/chunk_ssr_sample_history.js";
-import { chunk_ssr_temporal_accumulate } from "./reproject/chunk_ssr_temporal_accumulate.js";
-import { chunk_rgb_to_luminance } from "../../shader/chunk/color/chunk_rgb_to_luminance.js";
-
-const settings = WebGPUStruct.from({ history_validity: f32, reprojection_mode: u32, jitter_delta: vec2f });
-const resources = new ShaderResourceSetDescriptor();
-resources.setImmediate("settings", settings);
-resources.createGroup()
-    .addTexture("tInput")
-    .addTexture("tMetadata", "unfilterable-float")
-    .addTexture("tDepth")
-    .addTexture("tNormal", "uint")
-    .addTexture("tPBR")
-    .addTexture("tVelocity")
-    .addTexture("tOcclusion")
-    .addTexture("tHistory")
-    .addTexture("tHistoryMetadata", "unfilterable-float")
-    .addTexture("tHistoryGeometry")
-    .addTexture("tPreviousDepth")
-    .addUniform("camera_current", CAMERA_UNIFORM_STRUCT)
-    .addUniform("camera_previous", CAMERA_UNIFORM_STRUCT);
-
-const body = CodeChunk.from(/* language=WGSL */ `
-struct SSRTemporalOutput {
-    @location(0) radiance: vec4f,
-    @location(1) metadata: vec4f,
-    @location(2) geometry: vec4f,
-}
-
-@fragment
-fn main(@builtin(position) coord: vec4f) -> SSRTemporalOutput {
-    let st = vec2i(coord.xy);
-    let current = textureLoad(tInput, st, 0);
-    let metadata = textureLoad(tMetadata, st, 0).rg;
-    let normal = decode_g_buffer_normal(textureLoad(tNormal, st, 0).xy);
-    let roughness = decode_g_buffer_roughness(textureLoad(tPBR, st, 0));
-    var output: SSRTemporalOutput;
-    output.radiance = current;
-    output.metadata = vec4f(metadata, 1.0, rgb_to_luminance(current.rgb));
-    output.geometry = vec4f(normal, roughness);
-    let depth = textureLoad(tDepth, st, 0).r;
-    // A reset must never read history, even when stale textures contain non-finite values.
-    if (settings.history_validity <= 0.0 || settings.reprojection_mode == 2u || is_background(depth)) { return output; }
-
-    let position = screen_coordinate_to_world(vec2u(st), tDepth, camera_current);
-    let camera_position = camera_current.view_matrix_inverse[3].xyz;
-    let virtual_position = ssr_virtual_position(camera_position, position, metadata.x);
-    let plane_tolerance = max(1e-4, 0.002 * get_view_space_depth(depth, camera_current));
-    let size = vec2f(textureDimensions(tHistory));
-    var history: SSRHistorySample;
-    var confidence = 0.0;
-    var motion_confidence = 1.0;
-    var history_uv = vec2f(0.0);
-    let virtual_eligible = settings.reprojection_mode != 1u && metadata.y >= 0.5 && metadata.x > 0.0 && is_finite(metadata.x) && roughness <= 0.2;
-    if (virtual_eligible) {
-        let uv = ssr_history_uv(virtual_position, camera_previous.view_projection_matrix);
-        history_uv = uv;
-        history = ssr_sample_history(uv, position, normal, roughness, metadata, virtual_position, true, plane_tolerance);
-        confidence = history.confidence;
-    }
-    if (confidence < 0.5) {
-        let velocity = depth_clip_validated_velocity(tOcclusion, tVelocity, st);
-        motion_confidence = taa_velocity_confidence(velocity);
-        let surface_confidence = textureLoad(tOcclusion, st, 0).r * motion_confidence;
-        if (surface_confidence <= 0.001) { return output; }
-        let uv = ssr_surface_history_uv(coord.xy, velocity, settings.jitter_delta, size);
-        history_uv = uv;
-        history = ssr_sample_history(uv, position, normal, roughness, metadata, virtual_position, false, plane_tolerance);
-        confidence = history.confidence * surface_confidence;
-        confidence *= ssr_surface_parallax_confidence(camera_position,position,metadata.x,camera_previous.view_projection_matrix,size);
-        // Surface correspondence alone cannot justify a long history for a smooth finite reflection.
-        if (virtual_eligible) { confidence *= 0.25; }
-    }
-    if (confidence <= 0.001) { return output; }
-
-    let accumulated = ssr_temporal_accumulate(current, history.radiance, history.age,
-        history.short_luma, confidence, motion_confidence, st, history_uv);
-    output.radiance = accumulated.radiance;
-    output.metadata.z = accumulated.age;
-    output.metadata.w = accumulated.short_luma;
-    return output;
-}
-`, [chunk_ssr_reprojection, chunk_ssr_sample_history, chunk_screen_coordinate_to_world,
-    chunk_get_view_space_depth, chunk_decode_g_buffer_normal, chunk_decode_g_buffer_roughness,
-    chunk_depth_clip_offset_codec, chunk_taa_velocity_confidence,
-    chunk_ssr_temporal_accumulate, chunk_rgb_to_luminance]);
-
-export const ssr_reproject_shader = new ImageShader({
-    descriptor: ShaderDescriptor.from({ label: "SSR Reproject", body, resources }),
-    targets: [{ format: "rgba16float" }, { format: "rgba32float" }, { format: "rgba16float" }],
-});
+import { CAMERA_UNIFORM_STRUCT } from "../../camera/CAMERA_UNIFORM_STRUCT.js";
+import { chunk_screen_coordinate_to_world } from "../../shader/chunk/camera/chunk_screen_coordinate_to_world.js";
+import { chunk_get_view_space_depth } from "../../shader/chunk/depth/chunk_get_view_space_depth.js";
+import { chunk_decode_g_buffer_normal } from "../../shader/chunk/geometry/normal/chunk_decode_g_buffer_normal.js";
+import { chunk_decode_g_buffer_roughness } from "../../shader/chunk/material/chunk_decode_g_buffer_roughness.js";
+import { CodeChunk } from "../../shader/compiler/CodeChunk.js";
+import { ImageShader } from "../../shader/ImageShader.js";
+import { ShaderResourceSetDescriptor } from "../../shader/resource/ShaderResourceSetDescriptor.js";
+import { ShaderDescriptor } from "../../shader/ShaderDescriptor.js";
+import { WebGPUStruct } from "../../shader/type/WebGPUStruct.js";
+import { f32, u32, vec2f } from "../../shader/type/WEBGPU_WGSL_PRIMITIVE_TYPES.js";
+import { chunk_taa_velocity_confidence } from "../taa/shader_taa.js";
+import { chunk_depth_clip_offset_codec } from "../taa/chunk_depth_clip_offset_codec.js";
+import { chunk_ssr_reprojection } from "./reproject/chunk_ssr_reprojection.js";
+import { chunk_ssr_sample_history } from "./reproject/chunk_ssr_sample_history.js";
+import { chunk_ssr_temporal_accumulate } from "./reproject/chunk_ssr_temporal_accumulate.js";
+import { chunk_rgb_to_luminance } from "../../shader/chunk/color/chunk_rgb_to_luminance.js";
+import { SSR_DENOISE_TILE_SIZE } from "./SSR_DENOISE_TILE_SIZE.js";
+
+const settings = WebGPUStruct.from({ history_validity: f32, reprojection_mode: u32, jitter_delta: vec2f });
+function create_reproject_shader(tiled) {
+    const resources = new ShaderResourceSetDescriptor();
+    resources.setImmediate("settings", settings);
+    const group = resources.createGroup()
+        .addTexture("tInput")
+        .addTexture("tMetadata", "unfilterable-float")
+        .addTexture("tDepth")
+        .addTexture("tNormal", "uint")
+        .addTexture("tPBR")
+        .addTexture("tVelocity")
+        .addTexture("tOcclusion")
+        .addTexture("tHistory")
+        .addTexture("tHistoryMetadata", "unfilterable-float")
+        .addTexture("tHistoryGeometry")
+        .addTexture("tPreviousDepth")
+        .addUniform("camera_current", CAMERA_UNIFORM_STRUCT)
+        .addUniform("camera_previous", CAMERA_UNIFORM_STRUCT);
+    if (tiled) { group.addTexture("tDenoiseTiles", "uint"); }
+
+    const body = CodeChunk.from(/* language=WGSL */ `
+struct SSRTemporalOutput {
+    @location(0) radiance: vec4f,
+    @location(1) metadata: vec4f,
+    @location(2) geometry: vec4f,
+}
+
+@fragment
+fn main(@builtin(position) coord: vec4f) -> SSRTemporalOutput {
+    let st = vec2i(coord.xy);
+    let current = textureLoad(tInput, st, 0);
+    let metadata = textureLoad(tMetadata, st, 0).rg;
+    let normal = decode_g_buffer_normal(textureLoad(tNormal, st, 0).xy);
+    let roughness = decode_g_buffer_roughness(textureLoad(tPBR, st, 0));
+    var output: SSRTemporalOutput;
+    output.radiance = current;
+    output.metadata = vec4f(metadata, 1.0, rgb_to_luminance(current.rgb));
+    output.geometry = vec4f(normal, roughness);
+    let depth = textureLoad(tDepth, st, 0).r;
+    // A reset must never read history, even when stale textures contain non-finite values.
+    if (settings.history_validity <= 0.0 || settings.reprojection_mode == 2u || is_background(depth)) { return output; }
+
+    let position = screen_coordinate_to_world(vec2u(st), tDepth, camera_current);
+    let camera_position = camera_current.view_matrix_inverse[3].xyz;
+    let virtual_position = ssr_virtual_position(camera_position, position, metadata.x);
+    let plane_tolerance = max(1e-4, 0.002 * get_view_space_depth(depth, camera_current));
+    let size = vec2f(textureDimensions(tHistory));
+    var history: SSRHistorySample;
+    var confidence = 0.0;
+    var motion_confidence = 1.0;
+    var history_uv = vec2f(0.0);
+    let virtual_eligible = settings.reprojection_mode != 1u && metadata.y >= 0.5 && metadata.x > 0.0 && is_finite(metadata.x) && roughness <= 0.2;
+    if (virtual_eligible) {
+        let uv = ssr_history_uv(virtual_position, camera_previous.view_projection_matrix);
+        history_uv = uv;
+        history = ssr_sample_history(uv, position, normal, roughness, metadata, virtual_position, true, plane_tolerance);
+        confidence = history.confidence;
+    }
+    if (confidence < 0.5) {
+        let velocity = depth_clip_validated_velocity(tOcclusion, tVelocity, st);
+        motion_confidence = taa_velocity_confidence(velocity);
+        let surface_confidence = textureLoad(tOcclusion, st, 0).r * motion_confidence;
+        if (surface_confidence <= 0.001) { return output; }
+        let uv = ssr_surface_history_uv(coord.xy, velocity, settings.jitter_delta, size);
+        history_uv = uv;
+        history = ssr_sample_history(uv, position, normal, roughness, metadata, virtual_position, false, plane_tolerance);
+        confidence = history.confidence * surface_confidence;
+        confidence *= ssr_surface_parallax_confidence(camera_position,position,metadata.x,camera_previous.view_projection_matrix,size);
+        // Surface correspondence alone cannot justify a long history for a smooth finite reflection.
+        if (virtual_eligible) { confidence *= 0.25; }
+    }
+    if (confidence <= 0.001) { return output; }
+
+    let accumulated = ssr_temporal_accumulate(current, history.radiance, history.age,
+        history.short_luma, confidence, motion_confidence, st, history_uv,
+        ${tiled ? `textureLoad(tDenoiseTiles, st / ${SSR_DENOISE_TILE_SIZE}, 0).r == 0u` : "false"});
+    output.radiance = accumulated.radiance;
+    output.metadata.z = accumulated.age;
+    output.metadata.w = accumulated.short_luma;
+    return output;
+}
+    `, [chunk_ssr_reprojection, chunk_ssr_sample_history, chunk_screen_coordinate_to_world,
+        chunk_get_view_space_depth, chunk_decode_g_buffer_normal, chunk_decode_g_buffer_roughness,
+        chunk_depth_clip_offset_codec, chunk_taa_velocity_confidence,
+        chunk_ssr_temporal_accumulate, chunk_rgb_to_luminance]);
+
+    return new ImageShader({
+        descriptor: ShaderDescriptor.from({ label: "SSR Reproject" + (tiled ? " / tiles" : ""), body, resources }),
+        targets: [{ format: "rgba16float" }, { format: "rgba32float" }, { format: "rgba16float" }],
+    });
+}
+
+export const ssr_reproject_shader = create_reproject_shader(false);
+export const ssr_reproject_tile_shader = create_reproject_shader(true);

package/src/shade/renderer/postprocess/ssr/ssr_spatial_denoise_shader.d.ts

@@ -1,4 +1,6 @@
 export const ssr_spatial_denoise_shader: ImageShader;
 export const ssr_spatial_denoise_metadata_shader: ImageShader;
+export const ssr_spatial_denoise_tile_shader: ImageShader;
+export const ssr_spatial_denoise_metadata_tile_shader: ImageShader;
 import { ImageShader } from "../../shader/ImageShader.js";
 //# sourceMappingURL=ssr_spatial_denoise_shader.d.ts.map

package/src/shade/renderer/postprocess/ssr/ssr_spatial_denoise_shader.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"ssr_spatial_denoise_shader.d.ts","sourceRoot":"","sources":["../../../../../../src/shade/renderer/postprocess/ssr/ssr_spatial_denoise_shader.js"],"names":[],"mappings":"AAqKA,qDAAuE;AACvE,8DAA+E;4BA5JnD,6BAA6B"}
+{"version":3,"file":"ssr_spatial_denoise_shader.d.ts","sourceRoot":"","sources":["../../../../../../src/shade/renderer/postprocess/ssr/ssr_spatial_denoise_shader.js"],"names":[],"mappings":"AA6KA,qDAAuE;AACvE,8DAA+E;AAC/E,0DAAkF;AAClF,mEAA0F;4BArK9D,6BAA6B"}

package/src/shade/renderer/postprocess/ssr/ssr_spatial_denoise_shader.js

@@ -1,167 +1,177 @@
-import { chunk_ssr_spatial_variance } from "./chunk_ssr_spatial_variance.js";
-import { chunk_ssr_metadata } from "./chunk_ssr_metadata.js";
-import { chunk_rgb_to_luminance } from "../../shader/chunk/color/chunk_rgb_to_luminance.js";
-import { chunk_rgb_to_YCoCg } from "../../shader/chunk/color/YCoCg/chunk_rgb_to_YCoCg.js";
-import { chunk_decode_g_buffer_normal } from "../../shader/chunk/geometry/normal/chunk_decode_g_buffer_normal.js";
-import { chunk_safe_sqrt } from "../../shader/chunk/math/chunk_safe_sqrt.js";
-import {
-    chunk_texture_channel_gaussian_3x3
-} from "../../shader/chunk/texture/chunk_texture_channel_gaussian_3x3.js";
-import { CodeChunk } from "../../shader/compiler/CodeChunk.js";
-import { ImageShader } from "../../shader/ImageShader.js";
-import { ShaderResourceSetDescriptor } from "../../shader/resource/ShaderResourceSetDescriptor.js";
-import { ShaderDescriptor } from "../../shader/ShaderDescriptor.js";
-import { chunk_luma_edge_stopping_weight } from "../../shader/chunk/filter/chunk_compute_edge_stopping_weight.js";
-import { ATORUS_UNIFORMS_STRUCT } from "../denoise/fragment_shader_denoise_ao.js";
-import {
-    chunk_ffx_dnsr_reflections_get_edge_stopping_depth_weight
-} from "./prefilter/chunk_ffx_dnsr_reflections_get_edge_stopping_depth_weight.js";
-import {
-    chunk_ffx_dnsr_reflections_get_edge_stopping_normal_weight
-} from "./prefilter/chunk_ffx_dnsr_reflections_get_edge_stopping_normal_weight.js";
-
-function create_spatial_shader(with_metadata) {
-    const resources = new ShaderResourceSetDescriptor();
-
-    // per-pass, four bytes, read-only: immediate data rather than a uniform buffer
-    resources.setImmediate("settings", ATORUS_UNIFORMS_STRUCT);
-    const group = resources.createGroup()
-        .addTexture("tInput")
-        .addTexture("tDepth")
-        .addTexture("tNormal", "uint");
-    if (with_metadata) { group.addTexture("tMetadata", "unfilterable-float"); }
-
-    const body = CodeChunk.from(
-        //language=WGSL
-    `
-const phi_visibility = 10.0;
-const phi_normal = 32.0;
-const sigma_depth = 1.0;
- 
-@fragment
-fn main(
-    @builtin(position) coord: vec4<f32>,
-) -> ${with_metadata ? "SSRResolveOutput" : "@location(0) vec4f"}{
-    let ipos = vec2<i32>(coord.xy);
-    let screen_size = vec2<i32>(textureDimensions(tInput));
-    
-    const eps_variance      = 1e-10;
-    const kernel_weights = array<f32,3>( 1.0, 2.0 / 3.0, 1.0 / 6.0 );
-    
-    // constant samplers to prevent the compiler from generating code which
-    // fetches the sampler descriptor from memory for each texture access
-    let center_color = textureLoad(tInput, ipos, 0);
-    
-    let center_color_luma = rgb_to_luminance(center_color.rgb);
-
-    // variance for direct and indirect, filtered using 3x3 gaussin blur
-    let input_variance = max(0.0, texture_channel_gaussian_3x3(ipos, tInput, 3));
-${with_metadata ? `    let bootstrap_variance = ssr_spatial_variance(tInput,tDepth,tNormal,ipos);
-    let variance = max(input_variance,bootstrap_variance);
-    var sum_squared_weights = 1.0;` : `    let variance = input_variance;`}
-            
-    // explicitly store/accumulate center pixel with weight 1 to prevent issues
-    // with the edge-stopping functions
-    var sum_weight = 1.0;
-    var sum_color   = center_color;
-${with_metadata ? `    let center_metadata = textureLoad(tMetadata, ipos, 0).rg;
-    var distance_moments = ssr_distance_moments(center_metadata.x, center_metadata.y, 1.0);` : ""}
-    
-    
-    let current_normal = decode_g_buffer_normal( textureLoad(tNormal, ipos, 0).xy );
-    let center_depth   = textureLoad(tDepth, ipos, 0).r;
-    
-    const offsets = array< vec2<i32>,8>(
-      vec2( -1, -1 ),
-      vec2(  0, -1 ) ,
-      vec2(  1, -1 ), 
-      vec2( -1,  0 ), 
-      vec2(  1,  0 ), 
-      vec2( -1,  1 ),
-      vec2(  0,  1 ), 
-      vec2(  1,  1 ), 
-   );
-    
-    const num_samples = 8;
-    
-    let std_dev =  safe_sqrt(variance);
-    
-    for(var i = 0; i < num_samples; i++){
-        let offset: vec2i = offsets[i];
-                
-        let p      = ipos + offset * settings.step_size;
-        let inside = all(p >= vec2(0, 0)) && all(p < screen_size);
-        
-        if(!inside){
-            // off-screen
-            continue;
-        }
-        
-        let kernel = kernel_weights[ abs(offset.x) ] * kernel_weights[ abs(offset.y) ];
-        
-        let sample_color = textureLoad(tInput, p, 0);
-        let sample_color_luma = rgb_to_luminance(sample_color.rgb);
-
-        let sample_normal = decode_g_buffer_normal(textureLoad(tNormal, p, 0).xy);
-        let sample_depth  = textureLoad(tDepth, p, 0).r;
-
-        // compute the edge-stopping functions
-        var weight = 1.0;
-        
-        let phi_luma = max(0.001,std_dev * phi_visibility);
-
-        weight *= ffx_dnsr_reflections_get_edge_stopping_normal_weight(current_normal, sample_normal);
-        weight *= ffx_dnsr_reflections_get_edge_stopping_depth_weight(center_depth, sample_depth);
-        weight *= exp(-luma_edge_stopping_weight(center_color_luma,sample_color_luma,phi_luma));
-        
-        let w_visibility = weight * kernel;
-${with_metadata ? `        let metadata = textureLoad(tMetadata, p, 0).rg;
-        distance_moments += ssr_distance_moments(metadata.x, metadata.y, w_visibility);` : ""}
-
-        // alpha channel contains the variance, therefore the weights need to be squared, see paper for the formula
-        sum_weight += w_visibility;
-${with_metadata ? `        sum_squared_weights += w_visibility*w_visibility;` : ""}
-        sum_color += vec4(vec3(w_visibility), w_visibility * w_visibility) * sample_color;
-    }
-    
-    var out_visibility = sum_color / vec4(vec3(sum_weight), sum_weight * sum_weight);
-${with_metadata ? `    // Carry bootstrap uncertainty forward so the next passes do not freeze again.
-    out_visibility.w = max(out_visibility.w,bootstrap_variance*sum_squared_weights/(sum_weight*sum_weight));` : ""}
-    
-${with_metadata ? `    var output: SSRResolveOutput;
-    output.radiance = out_visibility;
-    output.metadata = ssr_resolve_metadata(distance_moments, 1.0);
-    return output;` : "    return out_visibility;"}
-}
-        `,
-        [
-            chunk_ssr_metadata,
-            ...(with_metadata ? [chunk_ssr_spatial_variance] : []),
-            ATORUS_UNIFORMS_STRUCT.declaration_chunk,
-            chunk_texture_channel_gaussian_3x3,
-            chunk_decode_g_buffer_normal,
-            chunk_rgb_to_luminance,
-            chunk_rgb_to_YCoCg,
-            chunk_ffx_dnsr_reflections_get_edge_stopping_normal_weight,
-            chunk_ffx_dnsr_reflections_get_edge_stopping_depth_weight,
-            chunk_luma_edge_stopping_weight,
-            chunk_safe_sqrt,
-        ]);
-
-    return new ImageShader({
-        descriptor: ShaderDescriptor.from({
-            label: with_metadata ? "SSR / spatial denoise with metadata" : "SSR / spatial denoise",
-            body,
-            resources
-        }),
-        targets: [
-            {
-                format: "rgba16float"
-            },
-            ...(with_metadata ? [{ format: "rg32float" }] : [])
-        ]
-    })
-}
-
-export const ssr_spatial_denoise_shader = create_spatial_shader(false);
-export const ssr_spatial_denoise_metadata_shader = create_spatial_shader(true);
+import { chunk_ssr_spatial_variance } from "./chunk_ssr_spatial_variance.js";
+import { chunk_ssr_metadata } from "./chunk_ssr_metadata.js";
+import { SSR_DENOISE_TILE_SIZE } from "./SSR_DENOISE_TILE_SIZE.js";
+import { chunk_rgb_to_luminance } from "../../shader/chunk/color/chunk_rgb_to_luminance.js";
+import { chunk_rgb_to_YCoCg } from "../../shader/chunk/color/YCoCg/chunk_rgb_to_YCoCg.js";
+import { chunk_decode_g_buffer_normal } from "../../shader/chunk/geometry/normal/chunk_decode_g_buffer_normal.js";
+import { chunk_safe_sqrt } from "../../shader/chunk/math/chunk_safe_sqrt.js";
+import {
+    chunk_texture_channel_gaussian_3x3
+} from "../../shader/chunk/texture/chunk_texture_channel_gaussian_3x3.js";
+import { CodeChunk } from "../../shader/compiler/CodeChunk.js";
+import { ImageShader } from "../../shader/ImageShader.js";
+import { ShaderResourceSetDescriptor } from "../../shader/resource/ShaderResourceSetDescriptor.js";
+import { ShaderDescriptor } from "../../shader/ShaderDescriptor.js";
+import { chunk_luma_edge_stopping_weight } from "../../shader/chunk/filter/chunk_compute_edge_stopping_weight.js";
+import { ATORUS_UNIFORMS_STRUCT } from "../denoise/fragment_shader_denoise_ao.js";
+import {
+    chunk_ffx_dnsr_reflections_get_edge_stopping_depth_weight
+} from "./prefilter/chunk_ffx_dnsr_reflections_get_edge_stopping_depth_weight.js";
+import {
+    chunk_ffx_dnsr_reflections_get_edge_stopping_normal_weight
+} from "./prefilter/chunk_ffx_dnsr_reflections_get_edge_stopping_normal_weight.js";
+
+function create_spatial_shader(with_metadata, tiled = false) {
+    const resources = new ShaderResourceSetDescriptor();
+
+    // per-pass, four bytes, read-only: immediate data rather than a uniform buffer
+    resources.setImmediate("settings", ATORUS_UNIFORMS_STRUCT);
+    const group = resources.createGroup()
+        .addTexture("tInput")
+        .addTexture("tDepth")
+        .addTexture("tNormal", "uint");
+    if (with_metadata) { group.addTexture("tMetadata", "unfilterable-float"); }
+    if (tiled) { group.addTexture("tDenoiseTiles", "uint"); }
+
+    const body = CodeChunk.from(
+        //language=WGSL
+    `
+const phi_visibility = 10.0;
+const phi_normal = 32.0;
+const sigma_depth = 1.0;
+ 
+@fragment
+fn main(
+    @builtin(position) coord: vec4<f32>,
+) -> ${with_metadata ? "SSRResolveOutput" : "@location(0) vec4f"}{
+    let ipos = vec2<i32>(coord.xy);
+    let screen_size = vec2<i32>(textureDimensions(tInput));
+${tiled ? `    if (textureLoad(tDenoiseTiles, ipos / ${SSR_DENOISE_TILE_SIZE}, 0).r == 0u) {
+${with_metadata ? `        var output: SSRResolveOutput;
+        output.radiance = textureLoad(tInput, ipos, 0);
+        output.metadata = textureLoad(tMetadata, ipos, 0).rg;
+        return output;` : `        return textureLoad(tInput, ipos, 0);`}
+    }` : ""}
+    
+    const eps_variance      = 1e-10;
+    const kernel_weights = array<f32,3>( 1.0, 2.0 / 3.0, 1.0 / 6.0 );
+    
+    // constant samplers to prevent the compiler from generating code which
+    // fetches the sampler descriptor from memory for each texture access
+    let center_color = textureLoad(tInput, ipos, 0);
+    
+    let center_color_luma = rgb_to_luminance(center_color.rgb);
+
+    // variance for direct and indirect, filtered using 3x3 gaussin blur
+    let input_variance = max(0.0, texture_channel_gaussian_3x3(ipos, tInput, 3));
+${with_metadata ? `    let bootstrap_variance = ssr_spatial_variance(tInput,tDepth,tNormal,ipos);
+    let variance = max(input_variance,bootstrap_variance);
+    var sum_squared_weights = 1.0;` : `    let variance = input_variance;`}
+            
+    // explicitly store/accumulate center pixel with weight 1 to prevent issues
+    // with the edge-stopping functions
+    var sum_weight = 1.0;
+    var sum_color   = center_color;
+${with_metadata ? `    let center_metadata = textureLoad(tMetadata, ipos, 0).rg;
+    var distance_moments = ssr_distance_moments(center_metadata.x, center_metadata.y, 1.0);` : ""}
+    
+    
+    let current_normal = decode_g_buffer_normal( textureLoad(tNormal, ipos, 0).xy );
+    let center_depth   = textureLoad(tDepth, ipos, 0).r;
+    
+    const offsets = array< vec2<i32>,8>(
+      vec2( -1, -1 ),
+      vec2(  0, -1 ) ,
+      vec2(  1, -1 ), 
+      vec2( -1,  0 ), 
+      vec2(  1,  0 ), 
+      vec2( -1,  1 ),
+      vec2(  0,  1 ), 
+      vec2(  1,  1 ), 
+   );
+    
+    const num_samples = 8;
+    
+    let std_dev =  safe_sqrt(variance);
+    
+    for(var i = 0; i < num_samples; i++){
+        let offset: vec2i = offsets[i];
+                
+        let p      = ipos + offset * settings.step_size;
+        let inside = all(p >= vec2(0, 0)) && all(p < screen_size);
+        
+        if(!inside){
+            // off-screen
+            continue;
+        }
+        
+        let kernel = kernel_weights[ abs(offset.x) ] * kernel_weights[ abs(offset.y) ];
+        
+        let sample_color = textureLoad(tInput, p, 0);
+        let sample_color_luma = rgb_to_luminance(sample_color.rgb);
+
+        let sample_normal = decode_g_buffer_normal(textureLoad(tNormal, p, 0).xy);
+        let sample_depth  = textureLoad(tDepth, p, 0).r;
+
+        // compute the edge-stopping functions
+        var weight = 1.0;
+        
+        let phi_luma = max(0.001,std_dev * phi_visibility);
+
+        weight *= ffx_dnsr_reflections_get_edge_stopping_normal_weight(current_normal, sample_normal);
+        weight *= ffx_dnsr_reflections_get_edge_stopping_depth_weight(center_depth, sample_depth);
+        weight *= exp(-luma_edge_stopping_weight(center_color_luma,sample_color_luma,phi_luma));
+        
+        let w_visibility = weight * kernel;
+${with_metadata ? `        let metadata = textureLoad(tMetadata, p, 0).rg;
+        distance_moments += ssr_distance_moments(metadata.x, metadata.y, w_visibility);` : ""}
+
+        // alpha channel contains the variance, therefore the weights need to be squared, see paper for the formula
+        sum_weight += w_visibility;
+${with_metadata ? `        sum_squared_weights += w_visibility*w_visibility;` : ""}
+        sum_color += vec4(vec3(w_visibility), w_visibility * w_visibility) * sample_color;
+    }
+    
+    var out_visibility = sum_color / vec4(vec3(sum_weight), sum_weight * sum_weight);
+${with_metadata ? `    // Carry bootstrap uncertainty forward so the next passes do not freeze again.
+    out_visibility.w = max(out_visibility.w,bootstrap_variance*sum_squared_weights/(sum_weight*sum_weight));` : ""}
+    
+${with_metadata ? `    var output: SSRResolveOutput;
+    output.radiance = out_visibility;
+    output.metadata = ssr_resolve_metadata(distance_moments, 1.0);
+    return output;` : "    return out_visibility;"}
+}
+        `,
+        [
+            chunk_ssr_metadata,
+            ...(with_metadata ? [chunk_ssr_spatial_variance] : []),
+            ATORUS_UNIFORMS_STRUCT.declaration_chunk,
+            chunk_texture_channel_gaussian_3x3,
+            chunk_decode_g_buffer_normal,
+            chunk_rgb_to_luminance,
+            chunk_rgb_to_YCoCg,
+            chunk_ffx_dnsr_reflections_get_edge_stopping_normal_weight,
+            chunk_ffx_dnsr_reflections_get_edge_stopping_depth_weight,
+            chunk_luma_edge_stopping_weight,
+            chunk_safe_sqrt,
+        ]);
+
+    return new ImageShader({
+        descriptor: ShaderDescriptor.from({
+            label: (with_metadata ? "SSR / spatial denoise with metadata" : "SSR / spatial denoise") + (tiled ? " / tiles" : ""),
+            body,
+            resources
+        }),
+        targets: [
+            {
+                format: "rgba16float"
+            },
+            ...(with_metadata ? [{ format: "rg32float" }] : [])
+        ]
+    })
+}
+
+export const ssr_spatial_denoise_shader = create_spatial_shader(false);
+export const ssr_spatial_denoise_metadata_shader = create_spatial_shader(true);
+export const ssr_spatial_denoise_tile_shader = create_spatial_shader(false, true);
+export const ssr_spatial_denoise_metadata_tile_shader = create_spatial_shader(true, true);

package/src/shade/renderer/rasterize/DEPTH_STENCIL_STATE_SHADOW.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"DEPTH_STENCIL_STATE_SHADOW.d.ts","sourceRoot":"","sources":["../../../../../src/shade/renderer/rasterize/DEPTH_STENCIL_STATE_SHADOW.js"],"names":[],"mappings":"AAEA;;;GAGG;AACH,yCAFU,SAAS,iBAAiB,CAAC,CAiBjC;kCArB8B,uDAAuD"}
+{"version":3,"file":"DEPTH_STENCIL_STATE_SHADOW.d.ts","sourceRoot":"","sources":["../../../../../src/shade/renderer/rasterize/DEPTH_STENCIL_STATE_SHADOW.js"],"names":[],"mappings":"AAEA;;;GAGG;AACH,yCAFU,SAAS,iBAAiB,CAAC,CA2BjC;kCA/B8B,uDAAuD"}

package/src/shade/renderer/rasterize/DEPTH_STENCIL_STATE_SHADOW.js

@@ -17,6 +17,16 @@ export const DEPTH_STENCIL_STATE_SHADOW = Object.freeze(DepthStencilState.from({
     /**
      * Combat shadow acne.
      * negative, because of reversed Z.
+     *
+     * A point light's cube faces need all of it at concave corners. Where a wall stands farther from
+     * the light than the light is above the ground, the ground in front of it is steep to the side
+     * cube face that sees it, and this pushes that ground back far enough that the wall's base does not
+     * read it as an occluder; at -1, as at 0, the bottom centimetres of the wall shadow themselves in
+     * scallops. The same push leaves a point light's shadow a seam on the cone 45° from straight down,
+     * where the ground stops being square-on to the -Y face: an occluder a little above a receiver stops
+     * shadowing it as the receiver's direction crosses into a side face (at 128 texels, from about
+     * 14 cm above to 22 cm). Measured 2026-10-04 and kept: a `depthBiasClamp` limits it by distance,
+     * not by angle, and no bias tried closed the seam without opening the corners.
      */
     depthBiasSlopeScale: -2.0,
 }));