@woosh/meep-engine 3.53.0 → 3.54.0

package/src/shade/renderer/particles/shaders/shader_particle_avboit_draw.js

@@ -1,27 +1,30 @@
-import { RenderPipelineDescriptor } from "../../../descriptor/pipeline/render/RenderPipelineDescriptor.js";
 import { DepthStencilState } from "../../../descriptor/pipeline/render/DepthStencilState.js";
-import { WebGPUArray } from "../../shader/type/WebGPUArray.js";
+import { RenderPipelineDescriptor } from "../../../descriptor/pipeline/render/RenderPipelineDescriptor.js";
 import { CAMERA_UNIFORM_STRUCT } from "../../camera/CAMERA_UNIFORM_STRUCT.js";
 import { chunk_sample_volumetrics } from "../../deferred/chunk_sample_volumetrics.js";
 import { AVBOIT_EXTINCT_TRANSMITTANCE } from "../../rasterize/native/avboit/AVBOIT_CONSTANTS.js";
-import { AVBOIT_WARP_STRUCT } from "../../rasterize/native/avboit/AVBOIT_WARP_STRUCT.js";
 import { AVBOIT_DRAW_TARGETS } from "../../rasterize/native/avboit/avboit_draw_fragment_body.js";
-import { chunk_avboit_view_depth_to_virtual } from "../../rasterize/native/avboit/chunk_avboit_view_depth_to_virtual.js";
+import { AVBOIT_WARP_STRUCT } from "../../rasterize/native/avboit/AVBOIT_WARP_STRUCT.js";
 import { chunk_avboit_footprint_bias } from "../../rasterize/native/avboit/chunk_avboit_footprint_bias.js";
 import { chunk_avboit_sample_transmittance } from "../../rasterize/native/avboit/chunk_avboit_sample_transmittance.js";
-import { ShaderDescriptor } from "../../shader/ShaderDescriptor.js";
+import { chunk_avboit_view_depth_to_virtual } from "../../rasterize/native/avboit/chunk_avboit_view_depth_to_virtual.js";
 import { CodeChunk } from "../../shader/compiler/CodeChunk.js";
 import { ShaderResourceSetDescriptor } from "../../shader/resource/ShaderResourceSetDescriptor.js";
+import { ShaderDescriptor } from "../../shader/ShaderDescriptor.js";
+import { WebGPUArray } from "../../shader/type/WebGPUArray.js";
 import { SAMPLER_LINEAR_UPSCALE } from "../../texture/SAMPLER_LINEAR_UPSCALE.js";
-import { VIEW_UNIFORM_STRUCT } from "../../view/VIEW_UNIFORM_STRUCT.js";
 import { chunk_view_mip_bias } from "../../view/chunk_view_mip_bias.js";
+import { VIEW_UNIFORM_STRUCT } from "../../view/VIEW_UNIFORM_STRUCT.js";
+import {
+    irradiance_clipmap_add_resources
+} from "../../volumetrics/irradiance_clipmap/chunk_irradiance_clipmap_sample.js";
 import { VOLUMETRICS_METADATA_STRUCT } from "../../volumetrics/VOLUMETRICS_METADATA_STRUCT.js";
 import { EMITTER_FLAG } from "../data/PARTICLE_EMITTER_STRUCT.js";
 import { PARTICLE_SOFT_FADE_RANGE } from "../ParticleConstants.js";
 import { chunk_particle_avboit_contribution } from "./chunk_particle_avboit_contribution.js";
-import { chunk_particle_scene_view_depth } from "./chunk_particle_scene_view_depth.js";
 import { chunk_particle_billboard_vertex } from "./chunk_particle_billboard_vertex.js";
 import { chunk_particle_lighting } from "./chunk_particle_lighting.js";
+import { chunk_particle_scene_view_depth } from "./chunk_particle_scene_view_depth.js";
 
 /**
  * The particles' AVBOIT draw: the billboard quads at internal resolution, depth-tested against the
@@ -30,28 +33,13 @@ import { chunk_particle_lighting } from "./chunk_particle_lighting.js";
  *
  * Per fragment: the atlas sample times the particle colour, the soft-depth fade against the scene
  * depth, lighting for an emitter flagged `EMITTER_FLAG.LIGHTING` (`chunk_particle_lighting`: the
- * froxel irradiance volume: the lights and the indirect lighting), the volumetric fog composite exactly as the
- * mesh draw applies it, then the transmittance of everything in front from the integral texture —
- * and a fragment the volume already calls extinct is discarded unshaded. An emitter without the
- * flag is its own colour, which is what an additive one wants.
- *
- * What is written depends on how the particle blends:
- *
- * - **alpha / premultiplied** — a surface: `color += premultiplied × T`, `norm += coverage × T`,
- *   `extinction += -log(1 - coverage)`, the same three the meshes write, with an untinted
- *   `T_self = 1 - coverage`; the splat recorded exactly that.
- * - **additive** — light: `emission += premultiplied × T`, nothing else. It joins the resolve
- *   attenuated by what is in front of it and outside the coverage normalization, which is what an
- *   additive fragment means.
- *
- * The scene depth is bound twice: as the read-only depth attachment the pipeline tests against, and
- * as the texture the soft fade reads. Both are reads, which WebGPU allows within one pass; the pass
- * is recorded with the attachment read-only for that reason.
+ * world-space irradiance clipmap and cached far-field SH), the volumetric fog composite, then the
+ * transmittance of everything in front from the integral texture.
  */
 
 const resources = new ShaderResourceSetDescriptor();
 
-resources.createGroup()
+const group = resources.createGroup()
     .addUniform("camera", CAMERA_UNIFORM_STRUCT)
     .addUniform("view", VIEW_UNIFORM_STRUCT)
     .addStorageBuffer("warp", AVBOIT_WARP_STRUCT)
@@ -67,10 +55,10 @@ resources.createGroup()
     // volumetrics (composited inline)
     .addUniform("volumetrics_metadata", VOLUMETRICS_METADATA_STRUCT)
     .addTexture("tVolumetrics", "float", "3d")
-    .addTexture("tSTBN_vec3", "float", "3d")
-    // lighting (chunk_particle_lighting): the froxel irradiance volume, trilinear through sLinear
-    .addTexture("tIrradianceY", "float", "3d")
-    .addTexture("tIrradianceCoCg", "float", "3d");
+    .addTexture("tSTBN_vec3", "float", "3d");
+
+// lighting (chunk_particle_lighting): world-space irradiance clipmap + cached far-field SH fallback
+irradiance_clipmap_add_resources(group, { include_far_field: true });
 
 const body = CodeChunk.from(
     /* language=wgsl */ `
@@ -111,16 +99,11 @@ fn vs_general(@builtin(vertex_index) vi : u32, @builtin(instance_index) ii : u32
 
 @fragment
 fn fs_main(in : ParticleVertexOut) -> ParticleAVBOITDrawOutput {
-    let screen_uv = in.clip.xy / vec2<f32>(f32(view.width), f32(view.height));
-    let uv_bias = view_uv_derivative_bias(view.upscale_ratio);
-    let uv_ddx = dpdx(in.uv) * uv_bias.x;
+    let screen_uv = in.clip.xy / vec2<f32>(f32(view.width), f32(view.height));
+    let uv_bias = view_uv_derivative_bias(view.upscale_ratio);
+    let uv_ddx = dpdx(in.uv) * uv_bias.x;
     let uv_ddy = dpdy(in.uv) * uv_bias.y;
 
-    // a billboard, stretched or not, is built from the camera's right and up and lies in the view plane,
-    // so its footprint is zero; a horizontal, vertical or world-facing quad seen at an angle runs
-    // through depth across a voxel (avboit_footprint_bias). An additive particle writes nothing into
-    // the volume, so nothing of its own is there to occlude it, and takes none. The derivative is
-    // taken here, in uniform control flow, whatever the blend
     let footprint = select(
         avboit_footprint_bias(&warp, fwidth(avboit_view_depth_to_virtual(&warp, in.view_depth))),
         0.0,
@@ -170,7 +153,7 @@ fn fs_main(in : ParticleVertexOut) -> ParticleAVBOITDrawOutput {
     chunk_sample_volumetrics,
     chunk_avboit_view_depth_to_virtual,
     chunk_avboit_footprint_bias,
-    chunk_avboit_sample_transmittance,
+    chunk_avboit_sample_transmittance,
     chunk_view_mip_bias,
 ]);
 

package/src/shade/renderer/postprocess/nss/NSS.d.ts

@@ -53,7 +53,7 @@ export class NSS {
      * @returns {number}
      */
     static recommended_jitter_sequence_size(upscale_ratio: number): number;
-    static "__#201@#preprocess_tensor_descriptor"(render_w: any, render_h: any): TextureResourceDescriptor;
+    static "__#203@#preprocess_tensor_descriptor"(render_w: any, render_h: any): TextureResourceDescriptor;
     /**
      * @param {GraphicsContext} graphics
      */

package/src/shade/renderer/shader/chunk/depth/chunk_query_depth_from_screen_space_bb.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"chunk_query_depth_from_screen_space_bb.d.ts","sourceRoot":"","sources":["../../../../../../../src/shade/renderer/shader/chunk/depth/chunk_query_depth_from_screen_space_bb.js"],"names":[],"mappings":"AAMA;;;GAGG;AACH,qDAFU,SAAS,CAqDZ;0BA7DmB,6BAA6B"}
+{"version":3,"file":"chunk_query_depth_from_screen_space_bb.d.ts","sourceRoot":"","sources":["../../../../../../../src/shade/renderer/shader/chunk/depth/chunk_query_depth_from_screen_space_bb.js"],"names":[],"mappings":"AAKA;;;GAGG;AACH,qDAFU,SAAS,CAsDZ;0BA7DmB,6BAA6B"}

package/src/shade/renderer/shader/chunk/depth/chunk_query_depth_from_screen_space_bb.js

@@ -1,62 +1,62 @@
-import { CodeChunk } from "../../compiler/CodeChunk.js";
-import { AABB3_STRUCT } from "../geometry/aabb3/AABB3_STRUCT.js";
-import { chunk_min4 } from "../math/chunk_min4.js";
-import { chunk_ndc_to_uv } from "../texture/chunk_ndc_to_uv.js";
-import { chunk_uv_to_texel_coordinate } from "../texture/chunk_uv_to_texel_coordinate.js";
-
-/**
- * TODO check out this post https://x.com/anki3d/status/1713238150418518023
- * @type {CodeChunk}
- */
-export const chunk_query_depth_from_screen_space_bb = CodeChunk.from(
-    //language=WGSL
-    `
-fn query_depth_from_screen_space_bb( 
-    pb: ${AABB3_STRUCT.wgsl_ref},
-    hzb: texture_2d<f32>,
-) -> f32 {
-    var n_min = saturate( ndc_to_uv(pb.min.xy) );
-    var n_max = saturate( ndc_to_uv(pb.max.xy) );
-
-    let bb_dim = abs(n_max - n_min);
-
-    let hzb_size_u = textureDimensions(hzb);
-    let hzb_size_f = vec2<f32>(hzb_size_u);
-
-    let bb_dim_hz = bb_dim * hzb_size_f;
-    
-    let longest_edge = max(bb_dim_hz.x, bb_dim_hz.y);
-
-    let mip_level_count = textureNumLevels(hzb);
-
-    let mip_level = min( 
-        u32( ceil(log2( max(longest_edge, 1.0) ) ) ),
-        mip_level_count - 1
-    );
-    
-    let mip_resolution = hzb_size_u >> vec2(mip_level);
-
-    // sample 4 corners, we know that they fit in a 2x2 footprint at most
-    let p0 = vec2<u32>((uv_to_texel_coordinate( n_min, mip_resolution)));
-    let p3 = vec2<u32>((uv_to_texel_coordinate( n_max, mip_resolution)));
-    let p1 = vec2(p3.x, p0.y);
-    let p2 = vec2(p0.x, p3.y);
-
-    let d0 = textureLoad(hzb, p0, mip_level).x;
-    let d1 = textureLoad(hzb, p1, mip_level).x;
-    let d2 = textureLoad(hzb, p2, mip_level).x;
-    let d3 = textureLoad(hzb, p3, mip_level).x;
-
-    // value that's most far away from us
-    let min_depth = min4(d0, d1, d2, d3);
-
-    let depth_delta = pb.max.z - min_depth;
-
-    return depth_delta;
-}
-`, [
-        AABB3_STRUCT.declaration_chunk,
-        chunk_min4,
-        chunk_uv_to_texel_coordinate,
-        chunk_ndc_to_uv
-    ]);
+import { CodeChunk } from "../../compiler/CodeChunk.js";
+import { AABB3_STRUCT } from "../geometry/aabb3/AABB3_STRUCT.js";
+import { chunk_min4 } from "../math/chunk_min4.js";
+import { chunk_ndc_to_uv } from "../texture/chunk_ndc_to_uv.js";
+
+/**
+ * TODO check out this post https://x.com/anki3d/status/1713238150418518023
+ * @type {CodeChunk}
+ */
+export const chunk_query_depth_from_screen_space_bb = CodeChunk.from(
+    //language=WGSL
+    `
+fn query_depth_from_screen_space_bb( 
+    pb: ${AABB3_STRUCT.wgsl_ref},
+    hzb: texture_2d<f32>,
+) -> f32 {
+    var n_min = saturate( ndc_to_uv(pb.min.xy) );
+    var n_max = saturate( ndc_to_uv(pb.max.xy) );
+
+    let bb_dim = abs(n_max - n_min);
+
+    let hzb_size_u = textureDimensions(hzb);
+    let hzb_size_f = vec2<f32>(hzb_size_u);
+
+    let bb_dim_hz = bb_dim * hzb_size_f;
+    
+    let longest_edge = max(bb_dim_hz.x, bb_dim_hz.y);
+
+    let mip_level_count = textureNumLevels(hzb);
+
+    let mip_level = min( 
+        u32( ceil(log2( max(longest_edge, 1.0) ) ) ),
+        mip_level_count - 1
+    );
+    
+    let mip_resolution = textureDimensions(hzb, mip_level);
+    let mip_resolution_f = vec2<f32>(mip_resolution);
+    let max_coord = mip_resolution - vec2(1u);
+
+    // sample 4 corners, we know that they fit in a 2x2 footprint at most
+    let p0 = min(vec2<u32>(n_min * mip_resolution_f), max_coord);
+    let p3 = min(vec2<u32>(n_max * mip_resolution_f), max_coord);
+    let p1 = vec2(p3.x, p0.y);
+    let p2 = vec2(p0.x, p3.y);
+
+    let d0 = textureLoad(hzb, p0, mip_level).x;
+    let d1 = textureLoad(hzb, p1, mip_level).x;
+    let d2 = textureLoad(hzb, p2, mip_level).x;
+    let d3 = textureLoad(hzb, p3, mip_level).x;
+
+    // value that's most far away from us
+    let min_depth = min4(d0, d1, d2, d3);
+
+    let depth_delta = pb.max.z - min_depth;
+
+    return depth_delta;
+}
+`, [
+        AABB3_STRUCT.declaration_chunk,
+        chunk_min4,
+        chunk_ndc_to_uv
+    ]);

package/src/shade/renderer/shader/chunk/random/lowbias/chunk_hash_lowbias32_vec2u.d.ts

@@ -0,0 +1,6 @@
+/**
+ * @type {CodeChunk}
+ */
+export const chunk_hash_lowbias32_vec2u: CodeChunk;
+import { CodeChunk } from "../../../compiler/CodeChunk.js";
+//# sourceMappingURL=chunk_hash_lowbias32_vec2u.d.ts.map

package/src/shade/renderer/shader/chunk/random/lowbias/chunk_hash_lowbias32_vec2u.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"chunk_hash_lowbias32_vec2u.d.ts","sourceRoot":"","sources":["../../../../../../../../src/shade/renderer/shader/chunk/random/lowbias/chunk_hash_lowbias32_vec2u.js"],"names":[],"mappings":"AAGA;;GAEG;AACH,yCAFU,SAAS,CAclB;0BAlByB,gCAAgC"}

package/src/shade/renderer/shader/chunk/random/lowbias/chunk_hash_lowbias32_vec2u.js

@@ -0,0 +1,19 @@
+import { CodeChunk } from "../../../compiler/CodeChunk.js";
+import { chunk_hash_lowbias32 } from "../chunk_hash_lowbias32.js";
+
+/**
+ * @type {CodeChunk}
+ */
+export const chunk_hash_lowbias32_vec2u = CodeChunk.from(
+    //language=WGSL
+    `
+fn hash_lowbias32_vec2u(v:vec2<u32>) -> u32{
+    let combined = v.x * 0x85ebca6bu + 
+                   v.y * 0xc2b2ae35u;
+                   
+    return hash_lowbias32(combined);
+}
+    `,[
+        chunk_hash_lowbias32,
+    ]
+)

package/src/shade/renderer/shader/chunk/screen_space/chunk_ffx_sssr_get_mip_resolution.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"chunk_ffx_sssr_get_mip_resolution.d.ts","sourceRoot":"","sources":["../../../../../../../src/shade/renderer/shader/chunk/screen_space/chunk_ffx_sssr_get_mip_resolution.js"],"names":[],"mappings":"AAEA,0DAOE;0BATwB,6BAA6B"}
+{"version":3,"file":"chunk_ffx_sssr_get_mip_resolution.d.ts","sourceRoot":"","sources":["../../../../../../../src/shade/renderer/shader/chunk/screen_space/chunk_ffx_sssr_get_mip_resolution.js"],"names":[],"mappings":"AAGA,0DAUE;0BAbwB,6BAA6B"}

package/src/shade/renderer/shader/chunk/screen_space/chunk_ffx_sssr_get_mip_resolution.js

@@ -1,10 +1,16 @@
-import { CodeChunk } from "../../compiler/CodeChunk.js";
-
-export const chunk_ffx_sssr_get_mip_resolution = CodeChunk.from(
-    //language=WGSL
-    `
-fn ffx_sssr_get_mip_resolution(screen_dimensions:vec2<f32>, mip_level:i32) -> vec2<f32>{
-    return screen_dimensions * pow(0.5, f32(mip_level));
-}
-    `
-);
+import { CodeChunk } from "../../compiler/CodeChunk.js";
+import { SCREEN_HZB_RESOURCE } from "./SCREEN_SPACE_RESOURCES.js";
+
+export const chunk_ffx_sssr_get_mip_resolution = CodeChunk.from(
+    //language=WGSL
+    `
+fn ffx_sssr_get_mip_resolution(screen_dimensions:vec2<f32>, mip_level:i32) -> vec2<f32>{
+    if (mip_level <= 0) {
+        return screen_dimensions;
+    }
+    return vec2<f32>(textureDimensions(tHZB, mip_level - 1));
+}
+    `
+);
+
+chunk_ffx_sssr_get_mip_resolution.requireResource(SCREEN_HZB_RESOURCE);

package/src/shade/renderer/shader/chunk/screen_space/chunk_ffx_sssr_hierarchical_raymarch.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"chunk_ffx_sssr_hierarchical_raymarch.d.ts","sourceRoot":"","sources":["../../../../../../../src/shade/renderer/shader/chunk/screen_space/chunk_ffx_sssr_hierarchical_raymarch.js"],"names":[],"mappings":"AASA;;;;;;GAMG;AACH,mDAFU,SAAS,CAqGZ;0BAlHmB,6BAA6B"}
+{"version":3,"file":"chunk_ffx_sssr_hierarchical_raymarch.d.ts","sourceRoot":"","sources":["../../../../../../../src/shade/renderer/shader/chunk/screen_space/chunk_ffx_sssr_hierarchical_raymarch.js"],"names":[],"mappings":"AASA;;;;;;GAMG;AACH,mDAFU,SAAS,CAuGZ;0BApHmB,6BAA6B"}

package/src/shade/renderer/shader/chunk/screen_space/chunk_ffx_sssr_hierarchical_raymarch.js

@@ -1,117 +1,118 @@
-import { chunk_is_background } from "../depth/chunk_is_background.js";
-import { CodeChunk } from "../../compiler/CodeChunk.js";
-import { chunk_rcp_v2 } from "../geometry/vec2/chunk_rcp_v2.js";
-import { F32_MAX_DEFINITION } from "../math/F32_MAX_DEFINITION.js";
-import { chunk_ffx_sssr_advance_ray } from "./chunk_ffx_sssr_advance_ray.js";
-import { chunk_ffx_sssr_get_mip_resolution } from "./chunk_ffx_sssr_get_mip_resolution.js";
-import { chunk_ffx_sssr_initial_advance_ray } from "./chunk_ffx_sssr_initial_advance_ray.js";
-import { chunk_ffx_sssr_load_depth } from "./chunk_ffx_sssr_load_depth.js";
-
-/**
- *
- * Requires origin and direction of the ray to be in screen space [0, 1] x [0, 1]
- * NOTE: "low occupancy" exit was removed, as WebGPU doesn't have pre-requisite API for it
- * @see https://github.com/GPUOpen-LibrariesAndSDKs/FidelityFX-SDK/blob/1bbe66e7ea0d358e84af4471a90aa9c35b583de0/samples/hybridreflections/shaders/ffx_sssr_intersect.h#L110
- * @type {CodeChunk}
- */
-export const chunk_ffx_sssr_hierarchical_raymarch = CodeChunk.from(
-    //language=WGSL
-    `
-fn ffx_sssr_hierarchical_raymarch(
-    origin: vec3<f32>,
-    direction: vec3<f32>,
-    screen_size: vec2<f32>,
-    most_detailed_mip: i32,
-    max_traversal_intersections:u32,
-    valid_hit: ptr<function, bool>,
-    _num_iters: ptr<function, u32>,
-) -> vec3<f32> {
-
-    let inv_direction = select( vec3(F32_MAX), 1.0 / direction, abs(direction) > vec3(1.0e-12));
-
-    // Start on mip with highest detail.
-    var current_mip = most_detailed_mip;
-
-    // Could recompute these every iteration, but it's faster to hoist them out and update them.
-    var current_mip_resolution     = ffx_sssr_get_mip_resolution(screen_size, current_mip);
-    var current_mip_resolution_inv = 1.0 / (current_mip_resolution);
-
-    // Offset to the bounding boxes uv space to intersect the ray with the center of the next pixel.
-    // This means we ever so slightly over shoot into the next region.
-    var uv_offset = 0.005 * exp2(f32(most_detailed_mip)) / screen_size;
-    uv_offset        = select( uv_offset, -uv_offset, direction.xy < vec2(0.0) );
-
-    // Offset applied depending on current mip resolution to move the boundary to the left/right upper/lower border depending on ray direction.
-    let floor_offset = select(vec2(1.0), vec2(0.0), direction.xy < vec2(0.0)); // TODO can replace with 'step'
-
-    // Initially advance ray to avoid immediate self intersections.
-    var current_t:f32;
-    var position:vec3<f32>;
-    
-    ffx_sssr_initial_advance_ray(
-        origin, 
-        direction,
-        inv_direction,
-        current_mip_resolution,
-        current_mip_resolution_inv,
-        floor_offset,
-        uv_offset,
-        &position,
-        &current_t
-    );
-    
-    var i = 0u;
-    for (
-        ;
-        i < max_traversal_intersections && current_mip >= most_detailed_mip;
-        i++
-    ) {
-        
-        if (any(position.xy > vec2(1.0, 1.0)) || any(position.xy < vec2(0.0, 0.0))){
-            // outside of the viewport
-            break;
-        }
-
-        if (is_background(position.z)){
-            // hit sky/background, out of Z bounds basically
-            break;
-        }
-
-        let current_mip_position = current_mip_resolution * position.xy;
-        let surface_z            = ffx_sssr_load_depth(vec2<i32>(current_mip_position), current_mip);
-        
-        let skipped_tile = ffx_sssr_advance_ray(
-                origin,
-                direction,
-                inv_direction,
-                current_mip_position,
-                current_mip_resolution_inv,
-                floor_offset,
-                uv_offset,
-                surface_z,
-                &position,
-                &current_t
-        );
-        
-        current_mip += select(-1, 1, skipped_tile );
-        current_mip_resolution *= select(2.0, 0.5, skipped_tile);
-        current_mip_resolution_inv *=  select(0.5, 2.0, skipped_tile);
-        
-    }
-    
-    *_num_iters = i;
-    
-    *valid_hit = (current_mip <= most_detailed_mip && i > 0);
-
-    return position;
-}
-`, [
-        F32_MAX_DEFINITION,
-        chunk_rcp_v2,
-        chunk_ffx_sssr_get_mip_resolution,
-        chunk_ffx_sssr_initial_advance_ray,
-        chunk_ffx_sssr_advance_ray,
-        chunk_ffx_sssr_load_depth,
-        chunk_is_background
-    ]);
-
+import { chunk_is_background } from "../depth/chunk_is_background.js";
+import { CodeChunk } from "../../compiler/CodeChunk.js";
+import { chunk_rcp_v2 } from "../geometry/vec2/chunk_rcp_v2.js";
+import { F32_MAX_DEFINITION } from "../math/F32_MAX_DEFINITION.js";
+import { chunk_ffx_sssr_advance_ray } from "./chunk_ffx_sssr_advance_ray.js";
+import { chunk_ffx_sssr_get_mip_resolution } from "./chunk_ffx_sssr_get_mip_resolution.js";
+import { chunk_ffx_sssr_initial_advance_ray } from "./chunk_ffx_sssr_initial_advance_ray.js";
+import { chunk_ffx_sssr_load_depth } from "./chunk_ffx_sssr_load_depth.js";
+
+/**
+ *
+ * Requires origin and direction of the ray to be in screen space [0, 1] x [0, 1]
+ * NOTE: "low occupancy" exit was removed, as WebGPU doesn't have pre-requisite API for it
+ * @see https://github.com/GPUOpen-LibrariesAndSDKs/FidelityFX-SDK/blob/1bbe66e7ea0d358e84af4471a90aa9c35b583de0/samples/hybridreflections/shaders/ffx_sssr_intersect.h#L110
+ * @type {CodeChunk}
+ */
+export const chunk_ffx_sssr_hierarchical_raymarch = CodeChunk.from(
+    //language=WGSL
+    `
+fn ffx_sssr_hierarchical_raymarch(
+    origin: vec3<f32>,
+    direction: vec3<f32>,
+    screen_size: vec2<f32>,
+    most_detailed_mip: i32,
+    max_traversal_intersections:u32,
+    valid_hit: ptr<function, bool>,
+    _num_iters: ptr<function, u32>,
+) -> vec3<f32> {
+
+    let inv_direction = select( vec3(F32_MAX), 1.0 / direction, abs(direction) > vec3(1.0e-12));
+    let max_mip = i32(textureNumLevels(tHZB));
+
+    // Start on mip with highest detail.
+    var current_mip = min(most_detailed_mip, max_mip);
+
+    var current_mip_resolution     = ffx_sssr_get_mip_resolution(screen_size, current_mip);
+    var current_mip_resolution_inv = 1.0 / (current_mip_resolution);
+
+    // Offset to the bounding boxes uv space to intersect the ray with the center of the next pixel.
+    // This means we ever so slightly over shoot into the next region.
+    var uv_offset = 0.005 * exp2(f32(most_detailed_mip)) / screen_size;
+    uv_offset        = select( uv_offset, -uv_offset, direction.xy < vec2(0.0) );
+
+    // Offset applied depending on current mip resolution to move the boundary to the left/right upper/lower border depending on ray direction.
+    let floor_offset = select(vec2(1.0), vec2(0.0), direction.xy < vec2(0.0)); // TODO can replace with 'step'
+
+    // Initially advance ray to avoid immediate self intersections.
+    var current_t:f32;
+    var position:vec3<f32>;
+    
+    ffx_sssr_initial_advance_ray(
+        origin, 
+        direction,
+        inv_direction,
+        current_mip_resolution,
+        current_mip_resolution_inv,
+        floor_offset,
+        uv_offset,
+        &position,
+        &current_t
+    );
+    
+    var i = 0u;
+    for (
+        ;
+        i < max_traversal_intersections && current_mip >= most_detailed_mip;
+        i++
+    ) {
+        
+        if (any(position.xy > vec2(1.0, 1.0)) || any(position.xy < vec2(0.0, 0.0))){
+            // outside of the viewport
+            break;
+        }
+
+        if (is_background(position.z)){
+            // hit sky/background, out of Z bounds basically
+            break;
+        }
+
+        let current_mip_position = current_mip_resolution * position.xy;
+        let surface_z            = ffx_sssr_load_depth(vec2<i32>(current_mip_position), current_mip);
+        
+        let skipped_tile = ffx_sssr_advance_ray(
+                origin,
+                direction,
+                inv_direction,
+                current_mip_position,
+                current_mip_resolution_inv,
+                floor_offset,
+                uv_offset,
+                surface_z,
+                &position,
+                &current_t
+        );
+        
+        current_mip = min(current_mip + select(-1, 1, skipped_tile), max_mip);
+        if (current_mip >= most_detailed_mip) {
+            current_mip_resolution = ffx_sssr_get_mip_resolution(screen_size, current_mip);
+            current_mip_resolution_inv = 1.0 / current_mip_resolution;
+        }
+        
+    }
+    
+    *_num_iters = i;
+    
+    *valid_hit = (current_mip <= most_detailed_mip && i > 0);
+
+    return position;
+}
+`, [
+        F32_MAX_DEFINITION,
+        chunk_rcp_v2,
+        chunk_ffx_sssr_get_mip_resolution,
+        chunk_ffx_sssr_initial_advance_ray,
+        chunk_ffx_sssr_advance_ray,
+        chunk_ffx_sssr_load_depth,
+        chunk_is_background
+    ]);