diff options
Diffstat (limited to 'assets/final/shaders/render')
| -rw-r--r-- | assets/final/shaders/render/scene_query.wgsl | 60 |
1 files changed, 44 insertions, 16 deletions
diff --git a/assets/final/shaders/render/scene_query.wgsl b/assets/final/shaders/render/scene_query.wgsl index b58f28c..e44b79c 100644 --- a/assets/final/shaders/render/scene_query.wgsl +++ b/assets/final/shaders/render/scene_query.wgsl @@ -1,4 +1,5 @@ #include "math/sdf_shapes" +#include "math/sdf_utils" // For aabb_sdf helper fn get_dist(p: vec3<f32>, obj_type: f32) -> f32 { if (obj_type == 1.0) { return length(p) - 1.0; } // Unit Sphere @@ -10,24 +11,51 @@ fn get_dist(p: vec3<f32>, obj_type: f32) -> f32 { fn map_scene(p: vec3<f32>, skip_idx: u32) -> f32 { var d = 1000.0; - let count = u32(globals.params.x); - - for (var i = 0u; i < count; i = i + 1u) { - if (i == skip_idx) { continue; } - let obj = object_data.objects[i]; - let obj_type = obj.params.x; - // Skip rasterized objects (like the floor) in the SDF map - if (obj_type <= 0.0) { continue; } + let use_bvh = (globals.params.y > 0.5); - let q = (obj.inv_model * vec4<f32>(p, 1.0)).xyz; + if (use_bvh) { + var stack: array<i32, 32>; + var stack_ptr = 0; - let scale_x = length(obj.model[0].xyz); - let scale_y = length(obj.model[1].xyz); - let scale_z = length(obj.model[2].xyz); - // Use conservative minimum scale to avoid overstepping the distance field - let s = min(scale_x, min(scale_y, scale_z)); - - d = min(d, get_dist(q, obj_type) * s); + if (arrayLength(&bvh_nodes) > 0u) { + stack[stack_ptr] = 0; + stack_ptr++; + } + + while (stack_ptr > 0) { + stack_ptr--; + let node_idx = stack[stack_ptr]; + let node = bvh_nodes[node_idx]; + + if (aabb_sdf(p, node.min, node.max) < d) { + if (node.left_idx < 0) { // Leaf + let obj_idx = u32(node.obj_idx_or_right); + if (obj_idx == skip_idx) { continue; } + let obj = object_data.objects[obj_idx]; + let q = (obj.inv_model * vec4<f32>(p, 1.0)).xyz; + let s = min(length(obj.model[0].xyz), min(length(obj.model[1].xyz), length(obj.model[2].xyz))); + d = min(d, get_dist(q, obj.params.x) * s); + } else { // Internal + if (stack_ptr < 31) { + stack[stack_ptr] = node.left_idx; + stack_ptr++; + stack[stack_ptr] = node.obj_idx_or_right; + stack_ptr++; + } + } + } + } + } else { // Fallback to linear scan + let count = u32(globals.params.x); + for (var i = 0u; i < count; i = i + 1u) { + if (i == skip_idx) { continue; } + let obj = object_data.objects[i]; + if (obj.params.x <= 0.0) { continue; } // Skip raster objects + let q = (obj.inv_model * vec4<f32>(p, 1.0)).xyz; + let s = min(length(obj.model[0].xyz), min(length(obj.model[1].xyz), length(obj.model[2].xyz))); + d = min(d, get_dist(q, obj.params.x) * s); + } } + return d; } |