PBR Renderer: added base color factor to primitive attribs

Author: TheMostDiligent

Hydrogent/src/HnRenderPass.cpp

@@ -675,6 +675,7 @@ HnRenderPass::EXECUTE_RESULT HnRenderPass::Execute(HnRenderPassState& RPState, c
             &Transform.PosBias,
             pSkinningData ? reinterpret_cast<const float4x4*>(pSkinningData->GeomBindXform.Data()) : nullptr,
             pSkinningData ? reinterpret_cast<const float4x4*>(pSkinningData->GeomBindXform.Data()) : nullptr,
+            &DisplayColor,
             &ListItem.MeshUID,        // CustomData
             sizeof(ListItem.MeshUID), // CustomDataSize
         };

PBR/interface/GLTF_PBR_Renderer.hpp

@@ -259,6 +259,7 @@ class GLTF_PBR_Renderer : public PBR_Renderer
         const float3*   PosBias              = nullptr;
         const float4x4* SkinPreTransform     = nullptr; // #if USE_JOINTS && USE_SKIN_PRE_TRANSFORM
         const float4x4* PrevSkinPreTransform = nullptr; // #if USE_JOINTS && USE_SKIN_PRE_TRANSFORM && COMPUTE_MOTION_VECTORS
+        const float4*   BaseColorFactor      = nullptr;
         const void*     CustomData           = nullptr;
         size_t          CustomDataSize       = 0;
     };

PBR/src/GLTF_PBR_Renderer.cpp

@@ -766,6 +766,7 @@ void* GLTF_PBR_Renderer::WritePBRPrimitiveShaderAttribs(void*
     //        float4x4 PrevSkinPreTransform; // #if USE_JOINTS && USE_SKIN_PRE_TRANSFORM && COMPUTE_MOTION_VECTORS
     //    } Transforms;
     //
+    //    float4      BaseColorFactor;
     //    UserDefined CustomData;
     //};
 
@@ -839,6 +840,12 @@ void* GLTF_PBR_Renderer::WritePBRPrimitiveShaderAttribs(void*
         }
     }
 
+    {
+        const float4& BaseColorFactor = AttribsData.BaseColorFactor != nullptr ? *AttribsData.BaseColorFactor : float4{1, 1, 1, 1};
+        memcpy(pDstPtr, &BaseColorFactor, sizeof(BaseColorFactor));
+        pDstPtr += sizeof(BaseColorFactor);
+    }
+
     {
         if (AttribsData.CustomData != nullptr)
         {

PBR/src/PBR_Renderer.cpp

@@ -2033,6 +2033,7 @@ Uint32 PBR_Renderer::GetPBRPrimitiveAttribsSize(PSO_FLAGS Flags, Uint32 CustomDa
     //        float4x4 PrevSkinPreTransform; // #if USE_JOINTS && USE_SKIN_PRE_TRANSFORM && COMPUTE_MOTION_VECTORS
     //    } Transforms;
     //
+    //    float4      BaseColorFactor;
     //    UserDefined CustomData;
     //};
 
@@ -2045,6 +2046,7 @@ Uint32 PBR_Renderer::GetPBRPrimitiveAttribsSize(PSO_FLAGS Flags, Uint32 CustomDa
             (UseSkinPreTransform ? sizeof(float4x4) : 0) +                       // Transforms.SkinPreTransform
             (UsePrevSkinPreTransform ? sizeof(float4x4) : 0) +                   // Transforms.PrevSkinPreTransform
 
+            sizeof(float4) + // BaseColorFactor
             CustomDataSize);
 }
 

Shaders/PBR/private/RenderPBR.psh

@@ -379,6 +379,7 @@ PSOutput main(in VSOutput VSOut,
               in bool     IsFrontFace : SV_IsFrontFace)
 {
     float4 BaseColor = GetBaseColor(VSOut, g_Material, g_Frame.Renderer.MipBias);
+    BaseColor *= PRIMITIVE.BaseColorFactor;
 
 #if USE_VERTEX_NORMALS
     float3 MeshNormal = VSOut.Normal;

Shaders/PBR/private/RenderPBR_Structures.fxh

@@ -31,6 +31,7 @@ struct PBRPrimitiveAttribs
 {
     GLTFNodeShaderTransforms Transforms;
 
+    float4 BaseColorFactor;
     float4 CustomData;
 };
 #ifdef CHECK_STRUCT_ALIGNMENT