From 8a78b5b125272af96fb369fb47c5d1498b2c6d2d Mon Sep 17 00:00:00 2001
From: Turanszki Janos <turanszkij@gmail.com>
Date: Thu, 22 Apr 2021 19:19:06 +0200
Subject: [PATCH] ssr and raytraced reflections denoise improvement

---
 WickedEngine/shaders/ssr_raytraceCS.hlsl | 5 +++--
 WickedEngine/shaders/ssr_temporalCS.hlsl | 7 +++++++
 WickedEngine/wiVersion.cpp               | 2 +-
 3 files changed, 11 insertions(+), 3 deletions(-)

diff --git a/WickedEngine/shaders/ssr_raytraceCS.hlsl b/WickedEngine/shaders/ssr_raytraceCS.hlsl
index 98a13bf10..2fb7318d5 100644
--- a/WickedEngine/shaders/ssr_raytraceCS.hlsl
+++ b/WickedEngine/shaders/ssr_raytraceCS.hlsl
@@ -249,6 +249,7 @@ void main(uint3 DTid : SV_DispatchThreadID)
     
 	float4 H;
 	float3 L;
+	float jitter;
 	if (roughness > 0.05f)
 	{
 		float3x3 tangentBasis = GetTangentBasis(N);
@@ -314,19 +315,19 @@ void main(uint3 DTid : SV_DispatchThreadID)
 #endif
         
 		L = reflect(-V, H.xyz);
+		jitter = InterleavedGradientNoise(DTid.xy, g_xFrame_FrameCount);
 	}
 	else
 	{
 		H = float4(N.xyz, 1.0f);
 		L = reflect(-V, H.xyz);
+		jitter = 0;
 	}
     
 	float2 hitPixel = float2(0.0f, 0.0f);
 	float3 hitPoint = float3(0.0f, 0.0f, 0.0f);
 	float iterations = 0.0f;
 
-	float jitter = InterleavedGradientNoise(DTid.xy, g_xFrame_FrameCount);
-
 	bool hit = ScreenSpaceRayTrace(P, L, jitter, roughness, hitPixel, hitPoint, iterations);
 
 	float hitDepth = texture_depth.SampleLevel(sampler_point_clamp, hitPixel, 0);
diff --git a/WickedEngine/shaders/ssr_temporalCS.hlsl b/WickedEngine/shaders/ssr_temporalCS.hlsl
index 499640049..b4525a378 100644
--- a/WickedEngine/shaders/ssr_temporalCS.hlsl
+++ b/WickedEngine/shaders/ssr_temporalCS.hlsl
@@ -96,6 +96,13 @@ void main(uint3 DTid : SV_DispatchThreadID, uint3 GTid : SV_GroupThreadID, uint3
 {
     const float2 uv = (DTid.xy + 0.5f) * xPPResolution_rcp;
 
+	const float roughness = texture_gbuffer1.SampleLevel(sampler_point_clamp, uv, 0).a;
+	if (roughness < 0.05)
+	{
+		output[DTid.xy] = resolve_current[DTid.xy];
+		return;
+	}
+
 	const float2 velocity = texture_gbuffer2.SampleLevel(sampler_point_clamp, uv, 0).xy;
 	const float2 prevUV = uv + velocity;
 	if (!is_saturated(prevUV))
diff --git a/WickedEngine/wiVersion.cpp b/WickedEngine/wiVersion.cpp
index 052cbc434..2cb5e9812 100644
--- a/WickedEngine/wiVersion.cpp
+++ b/WickedEngine/wiVersion.cpp
@@ -9,7 +9,7 @@ namespace wiVersion
 	// minor features, major updates, breaking compatibility changes
 	const int minor = 56;
 	// minor bug fixes, alterations, refactors, updates
-	const int revision = 0;
+	const int revision = 1;
 
 	const std::string version_string = std::to_string(major) + "." + std::to_string(minor) + "." + std::to_string(revision);