GLTF Viewer: added SSR debug views

Author: TheMostDiligent

Samples/GLTFViewer/assets/shaders/ApplyPostEffects.psh

@@ -21,9 +21,21 @@ float4 main(in FullScreenTriangleVSOutput VSOut) : SV_Target
     float4 SSR   = g_tex2DSSR.Load(int3(VSOut.f4PixelPos.xy, 0));
     float4 IBL   = g_tex2DIBL.Load(int3(VSOut.f4PixelPos.xy, 0));
 
-    float SSRScale = g_Frame.PrevCamera.f4ExtraData[0].x;
-    Color.rgb += (SSR.rgb - IBL.rgb) * SSR.w * Color.a * SSRScale;
- 
+    float SSRScale  = g_Frame.PrevCamera.f4ExtraData[0].x;
+    float DebugMode = g_Frame.PrevCamera.f4ExtraData[0].y;
+    if (DebugMode == 0.0)
+    {
+        Color.rgb += (SSR.rgb - IBL.rgb) * SSR.w * Color.a * SSRScale;
+    }
+    else if (DebugMode == 1.0)
+    {
+        Color.rgb = SSR.rgb;
+    }
+    else if (DebugMode == 2.0)
+    {
+        Color.rgb = float3(SSR.w, SSR.w, SSR.w);
+    }
+    
     {
         // Perform tone mapping
         ToneMappingAttribs TMAttribs;

Samples/GLTFViewer/src/GLTFViewer.cpp

@@ -1,5 +1,5 @@
 /*
- *  Copyright 2019-2023 Diligent Graphics LLC
+ *  Copyright 2019-2024 Diligent Graphics LLC
  *  Copyright 2015-2019 Egor Yusov
  *
  *  Licensed under the Apache License, Version 2.0 (the "License");
@@ -961,6 +961,11 @@ void GLTFViewer::UpdateUI()
 
             ImGui::SliderFloat("SSR scale", &m_ShaderAttribs.SSRScale, 0.f, 1.f);
 
+            ImGui::Combo("SSR Debug View", reinterpret_cast<int*>(&m_ShaderAttribs.SSRDebugMode),
+                         "None\0"
+                         "SSR\0"
+                         "Confidence\0\0");
+
             ImGui::TreePop();
         }
     }
@@ -1029,10 +1034,19 @@ void GLTFViewer::Render()
         FrameAttribs->Camera     = CurrCamAttribs;
         FrameAttribs->PrevCamera = PrevCamAttribs;
 
-        FrameAttribs->PrevCamera.f4ExtraData[0].x =
-            (m_RenderParams.DebugView == GLTF_PBR_Renderer::DebugViewType::None) ?
-            m_ShaderAttribs.SSRScale :
-            0;
+        if (m_RenderParams.DebugView == GLTF_PBR_Renderer::DebugViewType::None)
+        {
+            FrameAttribs->PrevCamera.f4ExtraData[0] = float4{
+                m_ShaderAttribs.SSRScale,
+                static_cast<float>(m_ShaderAttribs.SSRDebugMode),
+                0,
+                0,
+            };
+        }
+        else
+        {
+            FrameAttribs->PrevCamera.f4ExtraData[0] = float4{0};
+        }
 
         {
             if (m_BoundBoxMode != BoundBoxMode::None)

Samples/GLTFViewer/src/GLTFViewer.hpp

@@ -1,5 +1,5 @@
 /*
- *  Copyright 2019-2023 Diligent Graphics LLC
+ *  Copyright 2019-2024 Diligent Graphics LLC
  *  Copyright 2015-2019 Egor Yusov
  *
  *  Licensed under the Apache License, Version 2.0 (the "License");
@@ -87,6 +87,13 @@ class GLTFViewer final : public SampleBase
 
     GLTF_PBR_Renderer::RenderInfo m_RenderParams;
 
+    enum class SSRDebugViewMode : int
+    {
+        None,
+        SSR,
+        Confidence
+    };
+
     struct ShaderParams
     {
         float OcclusionStrength = 1;
@@ -99,7 +106,8 @@ class GLTFViewer final : public SampleBase
         float4 HighlightColor = float4{0, 0, 0, 0};
         float4 WireframeColor = float4{0.8f, 0.7f, 0.5f, 1.0f};
 
-        float SSRScale = 1;
+        float            SSRScale     = 1;
+        SSRDebugViewMode SSRDebugMode = SSRDebugViewMode::None;
     };
     ShaderParams m_ShaderAttribs;