tom/RT_GPU
1
0
Fork 0
mirror of https://github.com/TheRedShip/RT_GPU.git synced 2025-09-28 11:08:36 +02:00
Code Issues Packages Projects Releases Wiki Activity

+ | Denoising on light texture

Browse Source
This commit is contained in:
TheRedShip
2025-02-16 02:02:13 +01:00
parent 0544bc3a0f
commit ca91acbf67
4 changed files with 56 additions and 28 deletions
Download Patch File Download Diff File Expand all files Collapse all files

29
shaders/denoising.glsl
View File

@ -1,18 +1,22 @@
layout(local_size_x = 16, local_size_y = 16) in;
layout(binding = 0, rgba32f) uniform image2D read_texture;
layout(binding = 0, rgba32f) uniform image2D output_texture;
layout(binding = 2, rgba32f) uniform image2D write_texture;
layout(binding = 3, rgba32f) uniform image2D normal_texture;
layout(binding = 4, rgba32f) uniform image2D position_texture;
layout(binding = 5, rgba32f) uniform image2D light_texture;
layout(binding = 7, rgba32f) uniform image2D color_texture;
uniform vec2 u_resolution;
uniform int u_pass_count;
uniform int u_pass;
uniform float u_c_phi;
uniform float u_p_phi;
uniform float u_n_phi;
void main()
{
ivec2 pixel_coords = ivec2(gl_GlobalInvocationID.xy);
@ -21,14 +25,14 @@ void main()
int holes = int(pow(2, u_pass));
vec4 color_center = imageLoad(read_texture, pixel_coords);
vec4 light_center = imageLoad(light_texture, pixel_coords);
vec3 position_center = imageLoad(position_texture, pixel_coords).xyz;
vec3 normal_center = imageLoad(normal_texture, pixel_coords).xyz;
float kernel[5] = float[5](1.0/16.0, 1.0/4.0, 3.0/8.0, 1.0/4.0, 1.0/16.0);
float totalWeight = 0.;
vec4 color = vec4(vec3(0.), 1.0);
vec4 light = vec4(vec3(0.), 1.0);
for (int x = -2; x <= 2; x++)
@ -40,15 +44,15 @@ void main()
continue ;
// coords = clamp(coords, ivec2(-1.0), u_resolution);
vec4 color_sample = imageLoad(read_texture, coords);
vec4 light_sample = imageLoad(light_texture, coords);
vec3 position_sample = imageLoad(position_texture, coords).xyz;
vec3 normal_sample = imageLoad(normal_texture, coords).xyz;
// Calculate edge-stopping weights
// Color weight
float colorDist = distance(color_center, color_sample);
float w_c = exp(-colorDist / u_c_phi);
// light weight
float lightDist = distance(light_center, light_sample);
float w_c = exp(-lightDist / u_c_phi);
// Position weight
float posDist = distance(position_center, position_sample);
@ -60,10 +64,17 @@ void main()
float weight = kernel[x+2] * kernel[y+2] * w_c * w_p * w_n;
color += color_sample * weight;
light += light_sample * weight;
totalWeight += weight;
}
}
light /= totalWeight;
if (u_pass == u_pass_count - 1)
{
vec4 color = light * imageLoad(color_texture, pixel_coords);
imageStore(output_texture, pixel_coords, color);
return;
}
imageStore(write_texture, pixel_coords, color / totalWeight);
imageStore(write_texture, pixel_coords, light);
}