Bound Box Renderer: added RenderTargetMask parameter

Author: TheMostDiligent

Components/interface/BoundBoxRenderer.hpp

@@ -53,6 +53,12 @@ class BoundBoxRenderer
         TEXTURE_FORMAT RTVFormats[DILIGENT_MAX_RENDER_TARGETS] = {TEX_FORMAT_RGBA8_UNORM_SRGB};
         TEXTURE_FORMAT DSVFormat                               = TEX_FORMAT_D32_FLOAT;
 
+        /// A bit mask that defines the render targets to render to.
+        ///
+        /// \remarks    If bit N is set, the N-th render target's color write mask will be set to
+        ///             COLOR_MASK_ALL. Otherwise, it will be set to COLOR_MASK_NONE.
+        Uint32 RenderTargetMask = 0x1u;
+
         const char* PSMainSource = nullptr;
 
         /// Whether shader matrices are laid out in row-major order in GPU memory.
@@ -140,6 +146,7 @@ class BoundBoxRenderer
     const std::vector<TEXTURE_FORMAT> m_RTVFormats;
     const TEXTURE_FORMAT              m_DSVFormat;
     const std::string                 m_PSMainSource;
+    const Uint32                      m_RenderTargetMask;
     const bool                        m_PackMatrixRowMajor;
     const bool                        m_AsyncShaders;
 

Components/src/BoundBoxRenderer.cpp

@@ -56,6 +56,7 @@ BoundBoxRenderer::BoundBoxRenderer(const CreateInfo& CI) :
     m_RTVFormats{CI.RTVFormats, CI.RTVFormats + CI.NumRenderTargets},
     m_DSVFormat{CI.DSVFormat},
     m_PSMainSource{CI.PSMainSource != nullptr ? CI.PSMainSource : ""},
+    m_RenderTargetMask{CI.RenderTargetMask},
     m_PackMatrixRowMajor{CI.PackMatrixRowMajor},
     m_AsyncShaders{CI.AsyncShaders}
 {
@@ -152,17 +153,26 @@ IPipelineState* BoundBoxRenderer::GetPSO(const PSOKey& Key)
         .AddShader(pPS)
         .SetPrimitiveTopology(PRIMITIVE_TOPOLOGY_LINE_LIST)
         .SetDepthFormat(m_DSVFormat);
-    for (auto RTVFormat : m_RTVFormats)
+    for (TEXTURE_FORMAT RTVFormat : m_RTVFormats)
         PsoCI.AddRenderTarget(RTVFormat);
 
     PsoCI.PSODesc.ResourceLayout.DefaultVariableMergeStages  = SHADER_TYPE_VS_PS;
     PsoCI.PSODesc.ResourceLayout.DefaultVariableType         = SHADER_RESOURCE_VARIABLE_TYPE_MUTABLE;
     PsoCI.GraphicsPipeline.DepthStencilDesc.DepthFunc        = (Key.Flags & OPTION_FLAG_USE_REVERSE_DEPTH) != 0 ? COMPARISON_FUNC_GREATER_EQUAL : COMPARISON_FUNC_LESS_EQUAL;
     PsoCI.GraphicsPipeline.DepthStencilDesc.DepthWriteEnable = false;
+    for (Uint32 i = 0; i < PsoCI.GraphicsPipeline.NumRenderTargets; ++i)
+    {
+        PsoCI.GraphicsPipeline.BlendDesc.RenderTargets[i].RenderTargetWriteMask = (m_RenderTargetMask & (1u << i)) != 0 ?
+            COLOR_MASK_ALL :
+            COLOR_MASK_NONE;
+    }
+
     if (m_AsyncShaders)
+    {
         PsoCI.Flags |= PSO_CREATE_FLAG_ASYNCHRONOUS;
+    }
 
-    auto PSO = Device.CreateGraphicsPipelineState(PsoCI);
+    RefCntAutoPtr<IPipelineState> PSO = Device.CreateGraphicsPipelineState(PsoCI);
     if (!PSO)
     {
         UNEXPECTED("Failed to create bound box PSO");

Hydrogent/src/Tasks/HnRenderBoundBoxTask.cpp

@@ -177,11 +177,15 @@ void HnRenderBoundBoxTask::Prepare(pxr::HdTaskContext* TaskCtx,
             BoundBoxRndrCI.pCameraAttribsCB   = pRenderDelegate->GetFrameAttribsCB();
             BoundBoxRndrCI.NumRenderTargets   = RenderPassState->GetNumRenderTargets();
             for (Uint32 rt = 0; rt < BoundBoxRndrCI.NumRenderTargets; ++rt)
+            {
                 BoundBoxRndrCI.RTVFormats[rt] = RenderPassState->GetRenderTargetFormat(rt);
+            }
             BoundBoxRndrCI.DSVFormat = RenderPassState->GetDepthStencilFormat();
 
-            const std::string PSMain = GetBoundBoxPSMain(BoundBoxRndrCI.pDevice->GetDeviceInfo().IsGLDevice() ||
-                                                         BoundBoxRndrCI.pDevice->GetDeviceInfo().IsWebGPUDevice());
+            BoundBoxRndrCI.RenderTargetMask = ((1u << HnFrameRenderTargets::GBUFFER_TARGET_SCENE_COLOR) |
+                                               (1u << HnFrameRenderTargets::GBUFFER_TARGET_MOTION_VECTOR));
+
+            const std::string PSMain = GetBoundBoxPSMain(BoundBoxRndrCI.pDevice->GetDeviceInfo().IsGLDevice());
 
             BoundBoxRndrCI.PSMainSource = PSMain.c_str();