Half-Life-RTX
/
xash3d-fwgs
mirror of https://github.com/w23/xash3d-fwgs
1
0
Fork 0
Code Issues Projects Releases Wiki Activity

rtx: make denoiser kernel gaussian

This commit is contained in:
Ivan 'provod' Avdeev 2021-11-12 09:40:08 -08:00
parent 59a15a0580
commit fdad4fa016
1 changed files with 13 additions and 7 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

20
ref_vk/shaders/denoiser.comp
View File

@ -29,6 +29,9 @@ vec3 aces_tonemap(vec3 color){
return pow(clamp(m2 * (a / b), 0.0, 1.0), vec3(1.0 / 2.2)); return pow(clamp(m2 * (a / b), 0.0, 1.0), vec3(1.0 / 2.2));
} }
float normpdf2(in float x2, in float sigma) { return 0.39894*exp(-0.5*x2/(sigma*sigma))/sigma; }
float normpdf(in float x, in float sigma) { return normpdf2(x*x, sigma); }
void main() { void main() {
ivec2 res = ivec2(imageSize(src_base_color)); ivec2 res = ivec2(imageSize(src_base_color));
ivec2 pix = ivec2(gl_GlobalInvocationID); ivec2 pix = ivec2(gl_GlobalInvocationID);
@ -51,7 +54,7 @@ void main() {
// return; // return;
const int KERNEL_SIZE = 8; const int KERNEL_SIZE = 8;
float scale = 0.; float total_scale = 0.;
for (int x = -KERNEL_SIZE; x <= KERNEL_SIZE; ++x) for (int x = -KERNEL_SIZE; x <= KERNEL_SIZE; ++x)
for (int y = -KERNEL_SIZE; y <= KERNEL_SIZE; ++y) { for (int y = -KERNEL_SIZE; y <= KERNEL_SIZE; ++y) {
const ivec2 p = pix + ivec2(x, y); const ivec2 p = pix + ivec2(x, y);
@ -60,14 +63,17 @@ void main() {
} }
const vec4 c = imageLoad(src_diffuse_gi, p); const vec4 c = imageLoad(src_diffuse_gi, p);
if (c.a == material_index) { if (c.a != material_index)
colour += imageLoad(src_diffuse_gi, p).rgb; continue;
scale += 1.;
} const float sigma = KERNEL_SIZE / 2.;
const float scale = normpdf(x, sigma) * normpdf(y, sigma);
colour += scale * imageLoad(src_diffuse_gi, p).rgb;
total_scale += scale;
} }
if (scale > 0.) { if (total_scale > 0.) {
colour /= scale; colour /= total_scale;
colour *= base_color.rgb; colour *= base_color.rgb;
} }