Add GPUContext::SetBlendFactor

Source/Engine/Graphics/GPUContext.h

@@ -315,6 +315,12 @@ public:
     /// <param name="rts">The array with render targets to bind.</param>
     API_FUNCTION() virtual void SetRenderTarget(GPUTextureView* depthBuffer, const Span<GPUTextureView*>& rts) = 0;
 
+    /// <summary>
+    /// Sets the blend factor that modulate values for a pixel shader, render target, or both.
+    /// </summary>
+    /// <param name="value">Blend factors, one for each RGBA component.</param>
+    API_FUNCTION() virtual void SetBlendFactor(const Float4& value) = 0;
+
 public:
     /// <summary>
     /// Unbinds all shader resource slots and flushes the change with the driver (used to prevent driver detection of resource hazards, eg. when down-scaling the texture).

Source/Engine/Graphics/RenderTools.cpp

@@ -181,116 +181,75 @@ uint32 GetHash(const BlendingMode& key)
     return hash;
 }
 
+// @formatter:off
+
 BlendingMode BlendingMode::Opaque =
 {
-    false,
-    // AlphaToCoverageEnable
-    false,
-    // BlendEnable
-    Blend::One,
-    // SrcBlend
-    Blend::Zero,
-    // DestBlend
-    Operation::Add,
-    // BlendOp
-    Blend::One,
-    // SrcBlendAlpha
-    Blend::Zero,
-    // DestBlendAlpha
-    Operation::Add,
-    // BlendOpAlpha
-    ColorWrite::All,
-    // RenderTargetWriteMask
+    false, // AlphaToCoverageEnable
+    false, // BlendEnable
+    Blend::One, // SrcBlend
+    Blend::Zero, // DestBlend
+    Operation::Add, // BlendOp
+    Blend::One, // SrcBlendAlpha
+    Blend::Zero, // DestBlendAlpha
+    Operation::Add, // BlendOpAlpha
+    ColorWrite::All, // RenderTargetWriteMask
 };
 
 BlendingMode BlendingMode::Additive =
 {
-    false,
-    // AlphaToCoverageEnable
-    true,
-    // BlendEnable
-    Blend::SrcAlpha,
-    // SrcBlend
-    Blend::One,
-    // DestBlend
-    Operation::Add,
-    // BlendOp
-    Blend::SrcAlpha,
-    // SrcBlendAlpha
-    Blend::One,
-    // DestBlendAlpha
-    Operation::Add,
-    // BlendOpAlpha
-    ColorWrite::All,
-    // RenderTargetWriteMask
+    false, // AlphaToCoverageEnable
+    true, // BlendEnable
+    Blend::SrcAlpha, // SrcBlend
+    Blend::One, // DestBlend
+    Operation::Add, // BlendOp
+    Blend::SrcAlpha, // SrcBlendAlpha
+    Blend::One, // DestBlendAlpha
+    Operation::Add, // BlendOpAlpha
+    ColorWrite::All, // RenderTargetWriteMask
 };
 
 BlendingMode BlendingMode::AlphaBlend =
 {
-    false,
-    // AlphaToCoverageEnable
-    true,
-    // BlendEnable
-    Blend::SrcAlpha,
-    // SrcBlend
-    Blend::InvSrcAlpha,
-    // DestBlend
-    Operation::Add,
-    // BlendOp
-    Blend::One,
-    // SrcBlendAlpha
-    Blend::InvSrcAlpha,
-    // DestBlendAlpha
-    Operation::Add,
-    // BlendOpAlpha
-    ColorWrite::All,
-    // RenderTargetWriteMask
+    false, // AlphaToCoverageEnable
+    true, // BlendEnable
+    Blend::SrcAlpha, // SrcBlend
+    Blend::InvSrcAlpha, // DestBlend
+    Operation::Add, // BlendOp
+    Blend::One, // SrcBlendAlpha
+    Blend::InvSrcAlpha, // DestBlendAlpha
+    Operation::Add, // BlendOpAlpha
+    ColorWrite::All, // RenderTargetWriteMask
 };
 
 BlendingMode BlendingMode::Add =
 {
-    false,
-    // AlphaToCoverageEnable
-    true,
-    // BlendEnable
-    Blend::One,
-    // SrcBlend
-    Blend::One,
-    // DestBlend
-    Operation::Add,
-    // BlendOp
-    Blend::One,
-    // SrcBlendAlpha
-    Blend::One,
-    // DestBlendAlpha
-    Operation::Add,
-    // BlendOpAlpha
-    ColorWrite::All,
-    // RenderTargetWriteMask
+    false, // AlphaToCoverageEnable
+    true, // BlendEnable
+    Blend::One, // SrcBlend
+    Blend::One, // DestBlend
+    Operation::Add, // BlendOp
+    Blend::One, // SrcBlendAlpha
+    Blend::One, // DestBlendAlpha
+    Operation::Add, // BlendOpAlpha
+    ColorWrite::All, // RenderTargetWriteMask
 };
 
 BlendingMode BlendingMode::Multiply =
 {
-    false,
-    // AlphaToCoverageEnable
-    true,
-    // BlendEnable
-    Blend::Zero,
-    // SrcBlend
-    Blend::SrcColor,
-    // DestBlend
-    Operation::Add,
-    // BlendOp
-    Blend::Zero,
-    // SrcBlendAlpha
-    Blend::SrcAlpha,
-    // DestBlendAlpha
-    Operation::Add,
-    // BlendOpAlpha
-    ColorWrite::All,
-    // RenderTargetWriteMask
+    false, // AlphaToCoverageEnable
+    true, // BlendEnable
+    Blend::Zero, // SrcBlend
+    Blend::SrcColor, // DestBlend
+    Operation::Add, // BlendOp
+    Blend::Zero, // SrcBlendAlpha
+    Blend::SrcAlpha, // DestBlendAlpha
+    Operation::Add, // BlendOpAlpha
+    ColorWrite::All, // RenderTargetWriteMask
 };
 
+// @formatter:on
+
 FeatureLevel RenderTools::GetFeatureLevel(ShaderProfile profile)
 {
     switch (profile)

Source/Engine/GraphicsDevice/DirectX/DX11/GPUContextDX11.cpp

@@ -103,6 +103,7 @@ void GPUContextDX11::FrameBegin()
     CurrentPS = nullptr;
     CurrentCS = nullptr;
     CurrentPrimitiveTopology = D3D11_PRIMITIVE_TOPOLOGY_UNDEFINED;
+    CurrentBlendFactor = Float4::One;
 
     // Bind static samplers
     ID3D11SamplerState* samplers[] =
@@ -267,6 +268,13 @@ void GPUContextDX11::SetRenderTarget(GPUTextureView* depthBuffer, const Span<GPU
     }
 }
 
+void GPUContextDX11::SetBlendFactor(const Float4& value)
+{
+    CurrentBlendFactor = value;
+    if (CurrentBlendState)
+        _context->OMSetBlendState(CurrentBlendState, CurrentBlendFactor.Raw, D3D11_DEFAULT_SAMPLE_MASK);
+}
+
 void GPUContextDX11::ResetSR()
 {
     _srDirtyFlag = false;
@@ -560,8 +568,7 @@ void GPUContextDX11::SetState(GPUPipelineState* state)
         if (CurrentBlendState != blendState)
         {
             CurrentBlendState = blendState;
-            FLOAT blendFactor[4] = { 1, 1, 1, 1 };
-            _context->OMSetBlendState(blendState, blendFactor, D3D11_DEFAULT_SAMPLE_MASK);
+            _context->OMSetBlendState(blendState, CurrentBlendFactor.Raw, D3D11_DEFAULT_SAMPLE_MASK);
         }
         if (CurrentVS != vs)
         {

Source/Engine/GraphicsDevice/DirectX/DX11/GPUContextDX11.h

@@ -61,6 +61,7 @@ private:
     GPUShaderProgramPSDX11* CurrentPS;
     GPUShaderProgramCSDX11* CurrentCS;
     D3D11_PRIMITIVE_TOPOLOGY CurrentPrimitiveTopology;
+    Float4 CurrentBlendFactor;
 
 public:
 
@@ -115,6 +116,7 @@ public:
     void SetRenderTarget(GPUTextureView* rt) override;
     void SetRenderTarget(GPUTextureView* depthBuffer, GPUTextureView* rt) override;
     void SetRenderTarget(GPUTextureView* depthBuffer, const Span<GPUTextureView*>& rts) override;
+    void SetBlendFactor(const Float4& value) override;
     void ResetSR() override;
     void ResetUA() override;
     void ResetCB() override;

Source/Engine/GraphicsDevice/DirectX/DX12/GPUContextDX12.cpp

@@ -845,6 +845,11 @@ void GPUContextDX12::SetRenderTarget(GPUTextureView* depthBuffer, const Span<GPU
     }
 }
 
+void GPUContextDX12::SetBlendFactor(const Float4& value)
+{
+    _commandList->OMSetBlendFactor(value.Raw);
+}
+
 void GPUContextDX12::ResetSR()
 {
     for (int32 slot = 0; slot < GPU_MAX_SR_BINDED; slot++)

Source/Engine/GraphicsDevice/DirectX/DX12/GPUContextDX12.h

@@ -166,6 +166,7 @@ public:
     void SetRenderTarget(GPUTextureView* rt) override;
     void SetRenderTarget(GPUTextureView* depthBuffer, GPUTextureView* rt) override;
     void SetRenderTarget(GPUTextureView* depthBuffer, const Span<GPUTextureView*>& rts) override;
+    void SetBlendFactor(const Float4& value) override;
     void ResetSR() override;
     void ResetUA() override;
     void ResetCB() override;

Source/Engine/GraphicsDevice/Null/GPUContextNull.h

@@ -84,6 +84,10 @@ public:
     {
     }
 
+    void SetBlendFactor(const Float4& value) override
+    {
+    }
+
     void ResetSR() override
     {
     }

Source/Engine/GraphicsDevice/Vulkan/GPUContextVulkan.cpp

@@ -966,6 +966,12 @@ void GPUContextVulkan::SetRenderTarget(GPUTextureView* depthBuffer, const Span<G
     }
 }
 
+void GPUContextVulkan::SetBlendFactor(const Float4& value)
+{
+    const auto cmdBuffer = _cmdBufferManager->GetCmdBuffer();
+    vkCmdSetBlendConstants(cmdBuffer->GetHandle(), value.Raw);
+}
+
 void GPUContextVulkan::ResetSR()
 {
     Platform::MemoryClear(_srHandles, sizeof(_srHandles));

Source/Engine/GraphicsDevice/Vulkan/GPUContextVulkan.h

@@ -184,6 +184,7 @@ public:
     void SetRenderTarget(GPUTextureView* rt) override;
     void SetRenderTarget(GPUTextureView* depthBuffer, GPUTextureView* rt) override;
     void SetRenderTarget(GPUTextureView* depthBuffer, const Span<GPUTextureView*>& rts) override;
+    void SetBlendFactor(const Float4& value) override;
     void ResetSR() override;
     void ResetUA() override;
     void ResetCB() override;