You're breathtaking!

Source/Engine/Graphics/GPUPipelineStatePermutations.h

@@ -0,0 +1,154 @@
+// Copyright (c) 2012-2020 Wojciech Figat. All rights reserved.
+
+#pragma once
+
+#include "GPUPipelineState.h"
+#include "Shaders/GPUShader.h"
+#include "GPUDevice.h"
+
+template<int Size>
+class GPUPipelineStatePermutations
+{
+public:
+
+    GPUPipelineState* States[Size];
+
+public:
+
+    GPUPipelineStatePermutations()
+    {
+        Platform::MemoryClear(States, sizeof(States));
+    }
+
+    ~GPUPipelineStatePermutations()
+    {
+        Delete();
+    }
+
+public:
+
+    bool IsValid() const
+    {
+        for (int i = 0; i < Size; i++)
+        {
+            if (States[i] == nullptr || !States[i]->IsValid())
+            {
+                return false;
+            }
+        }
+
+        return true;
+    }
+
+    FORCE_INLINE GPUPipelineState* Get(int index) const
+    {
+        return States[index];
+    }
+
+public:
+
+    void CreatePipelineStates()
+    {
+        for (int i = 0; i < Size; i++)
+        {
+            if (States[i] == nullptr)
+            {
+                States[i] = GPUDevice::Instance->CreatePipelineState();
+            }
+        }
+    }
+
+    void Release()
+    {
+        for (int i = 0; i < Size; i++)
+        {
+            if (States[i])
+            {
+                States[i]->ReleaseGPU();
+            }
+        }
+    }
+
+    void Delete()
+    {
+        for (int i = 0; i < Size; i++)
+        {
+            SAFE_DELETE_GPU_RESOURCE(States[i]);
+        }
+    }
+};
+
+template<int Size>
+class GPUPipelineStatePermutationsPs : public GPUPipelineStatePermutations<Size>
+{
+public:
+
+    typedef GPUPipelineStatePermutations<Size> Base;
+
+public:
+
+    GPUPipelineStatePermutationsPs()
+    {
+    }
+
+    ~GPUPipelineStatePermutationsPs()
+    {
+    }
+
+public:
+
+    bool Create(GPUPipelineState::Description& desc, GPUShader* shader, const StringAnsiView& psName)
+    {
+        for (int i = 0; i < Size; i++)
+        {
+            ASSERT(Base::States[i]);
+
+            desc.PS = shader->GetPS(psName, i);
+            if (Base::States[i]->Init(desc))
+                return true;
+        }
+
+        return false;
+    }
+};
+
+template<int Size>
+class ComputeShaderPermutation
+{
+public:
+
+    GPUShaderProgramCS* Shaders[Size];
+
+public:
+
+    ComputeShaderPermutation()
+    {
+        Platform::MemoryClear(Shaders, sizeof(Shaders));
+    }
+
+    ~ComputeShaderPermutation()
+    {
+    }
+
+public:
+
+    FORCE_INLINE GPUShaderProgramCS* Get(const int index) const
+    {
+        return Shaders[index];
+    }
+
+public:
+
+    void Clear()
+    {
+        Platform::MemoryClear(Shaders, sizeof(Shaders));
+    }
+
+    void Get(GPUShader* shader, const StringAnsiView& name)
+    {
+        for (int i = 0; i < Size; i++)
+        {
+            Shaders[i] = shader->GetCS(name, i);
+        }
+    }
+};