matthew
/
pteropus
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Add antialiasing

This commit is contained in:
Matthew Gordon 2025-01-06 11:18:43 -04:00
parent 53221856aa
commit 91a66e3bb4
1 changed files with 55 additions and 36 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

91
src/app/dem_renderer/dem_renderer.wgsl
View File

@ -38,44 +38,63 @@ fn fs_solid(vertex: VertexOutput) -> @location(0) vec4<f32> {
var ray: Ray;
ray.origin = vertex.world_space_position.xyz;
ray.direction = normalize(vertex.world_space_position.xyz - uniforms.camera_position.xyz);
// Spread rays into RGSS antialiasing pattern.
let ray_dx = dpdy(ray.direction);
let ray_dy = dpdy(ray.direction);
var ray_directions: array<vec3<f32>,4>;
ray_directions[0] = ray.direction + 0.125*ray_dx + 0.375*ray_dy;
ray_directions[1] = ray.direction - 0.125*ray_dx - 0.375*ray_dy;
ray_directions[2] = ray.direction + 0.375*ray_dx + 0.125*ray_dy;
ray_directions[3] = ray.direction - 0.375*ray_dx - 0.125*ray_dy;
var root_node: BoundingNode;
root_node.index = vec2<u32>(0);
root_node.level = textureNumLevels(dembvh_texture) - 1;
// Possibly these ray directions could be stored in a matrix and we could
// evaluate them all at once instead of looping.
var hit_index = vec2<u32>(0);
var hit_location: vec3<f32>;
var hit_normal: vec3<f32>;
if intersect_ray_with_node(dembvh_texture,
uniforms.dem_min_corner,
uniforms.dem_cell_size,
uniforms.dem_z_range,
ray,
root_node,
&hit_index,
&hit_location,
&hit_normal) {
// Calculate light
let sun_direction = (uniforms.camera_to_world_matrix
* vec4<f32>(normalize(vec3<f32>(1.0, 1.0, 1.0)), 0.0)).xyz;
let l = dot(hit_normal, sun_direction);
var shadow_ray :Ray;
shadow_ray.origin = hit_location + hit_normal * 0.1;
shadow_ray.direction = sun_direction;
var dummy0: vec3<f32>;
var dummy1: vec3<f32>;
let shadow_value =
select(1.0, 0.0, intersect_ray_with_node(dembvh_texture,
uniforms.dem_min_corner,
uniforms.dem_cell_size,
uniforms.dem_z_range,
shadow_ray,
root_node,
&hit_index,
&dummy0,
&dummy1));
return vec4<f32>(vec3<f32>(l)*shadow_value, 1.0);
} else {
var color_accumulator = vec4<f32>(0);
for(var i=0; i<4; i++) {
ray.direction = ray_directions[i];
var root_node: BoundingNode;
root_node.index = vec2<u32>(0);
root_node.level = textureNumLevels(dembvh_texture) - 1;
var hit_index = vec2<u32>(0);
var hit_location: vec3<f32>;
var hit_normal: vec3<f32>;
if intersect_ray_with_node(dembvh_texture,
uniforms.dem_min_corner,
uniforms.dem_cell_size,
uniforms.dem_z_range,
ray,
root_node,
&hit_index,
&hit_location,
&hit_normal) {
// Calculate light
let sun_direction =
(uniforms.camera_to_world_matrix
* vec4<f32>(normalize(vec3<f32>(1.0, 1.0, 1.0)), 0.0)).xyz;
let l = dot(hit_normal, sun_direction);
var shadow_ray :Ray;
shadow_ray.origin = hit_location + hit_normal * 0.1;
shadow_ray.direction = sun_direction;
var dummy0: vec3<f32>;
var dummy1: vec3<f32>;
let shadow_value =
select(1.0, 0.0, intersect_ray_with_node(dembvh_texture,
uniforms.dem_min_corner,
uniforms.dem_cell_size,
uniforms.dem_z_range,
shadow_ray,
root_node,
&hit_index,
&dummy0,
&dummy1));
color_accumulator += vec4<f32>(vec3<f32>(l)*shadow_value, 1.0);
}
}
if color_accumulator.w == 0.0 {
discard;
}
return color_accumulator * 0.25;
}