1 files changed, 11 insertions, 8 deletions
diff --git a/source/blender/draw/engines/eevee/shaders/ssr_lib.glsl b/source/blender/draw/engines/eevee/shaders/ssr_lib.glsl
index 5a09120916a..612e95832e4 100644
--- a/source/blender/draw/engines/eevee/shaders/ssr_lib.glsl
+++ b/source/blender/draw/engines/eevee/shaders/ssr_lib.glsl
@@ -9,23 +9,26 @@
 
 #define BTDF_BIAS 0.85
 
+uniform sampler2D refractColorBuffer;
+
+uniform float refractionDepth;
+
 vec4 screen_space_refraction(vec3 vP, vec3 N, vec3 V, float ior, float roughnessSquared, vec4 rand)
 {
-  float a2 = max(5e-6, roughnessSquared * roughnessSquared);
+  float alpha = max(0.002, roughnessSquared);
 
   /* Importance sampling bias */
   rand.x = mix(rand.x, 0.0, BTDF_BIAS);
 
   vec3 T, B;
-  float NH;
   make_orthonormal_basis(N, T, B);
-  vec3 H = sample_ggx(rand.xzw, a2, N, T, B, NH); /* Microfacet normal */
-  float pdf = pdf_ggx_reflect(NH, a2);
+  float pdf;
+  /* Microfacet normal */
+  vec3 H = sample_ggx(rand.xzw, alpha, V, N, T, B, pdf);
 
   /* If ray is bad (i.e. going below the plane) regenerate. */
   if (F_eta(ior, dot(H, V)) < 1.0) {
-    H = sample_ggx(rand.xzw * vec3(1.0, -1.0, -1.0), a2, N, T, B, NH); /* Microfacet normal */
-    pdf = pdf_ggx_reflect(NH, a2);
+    H = sample_ggx(rand.xzw * vec3(1.0, -1.0, -1.0), alpha, V, N, T, B, pdf);
   }
 
   vec3 vV = viewCameraVec(vP);
@@ -75,10 +78,10 @@ vec4 screen_space_refraction(vec3 vP, vec3 N, vec3 V, float ior, float roughness
     vec2 hit_uvs = project_point(ProjectionMatrix, hit_pos).xy * 0.5 + 0.5;
 
     /* Texel footprint */
-    vec2 texture_size = vec2(textureSize(colorBuffer, 0).xy);
+    vec2 texture_size = vec2(textureSize(refractColorBuffer, 0).xy) / hizUvScale.xy;
     float mip = clamp(log2(cone_footprint * max(texture_size.x, texture_size.y)), 0.0, 9.0);
 
-    vec3 spec = textureLod(colorBuffer, hit_uvs * hizUvScale.xy, mip).xyz;
+    vec3 spec = textureLod(refractColorBuffer, hit_uvs * hizUvScale.xy, mip).xyz;
     float mask = screen_border_mask(hit_uvs);
 
     return vec4(spec, mask);