From 5ed6b8eccd7bb419d2b673113dad79a45edbf1fc Mon Sep 17 00:00:00 2001
From: kearwood <kearwood@kearwood.com>
Date: Fri, 9 Sep 2022 00:38:46 -0700
Subject: [PATCH] Push constants are now supported for all stages.

---
 kraken/KRPipeline.cpp | 8 +-------
 1 file changed, 1 insertion(+), 7 deletions(-)

diff --git a/kraken/KRPipeline.cpp b/kraken/KRPipeline.cpp
index 71cc735..bd6555d 100644
--- a/kraken/KRPipeline.cpp
+++ b/kraken/KRPipeline.cpp
@@ -205,14 +205,8 @@ KRPipeline::KRPipeline(KRContext& context, KRSurface& surface, const PipelineInf
           attribute_locations[KRMesh::KRENGINE_ATTRIB_BONEWEIGHTS] = input_var.location + 1;
         }
       }
-
-      initPushConstantStage(ShaderStage::vert, reflection);
-
-    } else if (stageInfo.stage == VK_SHADER_STAGE_FRAGMENT_BIT) {
-      initPushConstantStage(ShaderStage::frag, reflection);
-    } else {
-      // failed! TODO - Error handling
     }
+    initPushConstantStage(shader->getShaderStage(), reflection);
     stageInfo.module = shaderModule;
     stageInfo.pName = "main";
   }