kraken_engine/kraken
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Added ShaderValueType enum to shader reflection

Browse Source
This commit is contained in:
Kearwood Gilbert
2025-08-30 17:19:00 -07:00
parent 7ea376574c
commit 97da7eaa94
3 changed files with 44 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

29
kraken/KRPipeline.cpp
View File

@@ -444,6 +444,35 @@ void KRPipeline::initPushConstantStage(ShaderStage stage, const SpvReflectShader
if (stricmp(SHADER_VALUE_NAMES[iUniform], member.name) == 0) { if (stricmp(SHADER_VALUE_NAMES[iUniform], member.name) == 0) {
pushConstants.offset[iUniform] = member.offset; pushConstants.offset[iUniform] = member.offset;
pushConstants.size[iUniform] = member.size; pushConstants.size[iUniform] = member.size;
if (member.type_description->op == SpvOpTypeVector && member.numeric.scalar.width == 32) {
switch (member.numeric.vector.component_count) {
case 2:
pushConstants.type[iUniform] = ShaderValueType::type_vector2;
break;
case 3:
pushConstants.type[iUniform] = ShaderValueType::type_vector3;
break;
case 4:
pushConstants.type[iUniform] = ShaderValueType::type_vector4;
break;
}
}
if (member.type_description->op == SpvOpTypeFloat && member.numeric.scalar.width == 32) {
pushConstants.type[iUniform] = ShaderValueType::type_float32;
}
if (member.type_description->op == SpvOpTypeFloat && member.numeric.scalar.width == 64) {
pushConstants.type[iUniform] = ShaderValueType::type_float64;
}
if (member.type_description->op == SpvOpTypeInt && member.numeric.scalar.width == 32) {
pushConstants.type[iUniform] = ShaderValueType::type_int32;
}
if (member.type_description->op == SpvOpTypeInt && member.numeric.scalar.width == 64) {
pushConstants.type[iUniform] = ShaderValueType::type_int64;
}
if (member.type_description->op == SpvOpTypeMatrix && member.numeric.scalar.width == 32 && member.numeric.matrix.column_count == 4 && member.numeric.matrix.row_count == 4) {
pushConstants.type[iUniform] = ShaderValueType::type_matrix4;
}
// TODO: Support unsigned integer binding?
} }
} }
} }

1
kraken/KRPipeline.h
View File

@@ -246,6 +246,7 @@ private:
{ {
int offset[kPushConstantCount]; int offset[kPushConstantCount];
__uint8_t size[kPushConstantCount]; __uint8_t size[kPushConstantCount];
ShaderValueType type[kPushConstantCount];
uint8_t* buffer; uint8_t* buffer;
int bufferSize; int bufferSize;
VkPipelineLayout layout; VkPipelineLayout layout;

14
kraken/KRShaderReflection.h
View File

@@ -34,6 +34,20 @@
#include <map> #include <map>
enum class ShaderValueType : uint8_t
{
type_null = 0,
type_int32,
type_int64,
type_float32,
type_float64,
type_vector2,
type_vector3,
type_vector4,
type_matrix4,
NUM_SHADER_VALUE_TYPES
};
enum class ShaderValue : uint8_t enum class ShaderValue : uint8_t
{ {
material_ambient = 0, material_ambient = 0,