feat(engine): add element count & resource shape to shader type reflection

1 files changed, 42 insertions, 0 deletions

diff --git a/engine/src/rendering/shader.rs b/engine/src/rendering/shader.rs
index 6b68e1e..4e7fddc 100644
--- a/engine/src/rendering/shader.rs
+++ b/engine/src/rendering/shader.rs
@@ -340,6 +340,13 @@ impl<'a> TypeLayout<'a>
         ))
     }
 
+    pub fn resource_shape(&self) -> Option<ResourceShape>
+    {
+        Some(ResourceShape::from_bits_retain(
+            self.inner.resource_shape()? as u32,
+        ))
+    }
+
     pub fn get_field_by_name(&self, name: &str) -> Option<VariableLayout<'a>>
     {
         let index = self.inner.find_field_index_by_name(name);
@@ -400,6 +407,11 @@ impl<'a> TypeLayout<'a>
         self.inner.field_count()
     }
 
+    pub fn element_cnt(&self) -> Option<usize>
+    {
+        self.inner.element_count()
+    }
+
     pub fn element_type_layout(&self) -> Option<TypeLayout<'a>>
     {
         self.inner
@@ -582,6 +594,36 @@ impl ScalarType
     }
 }
 
+bitflags! {
+#[derive(Debug, Copy, Clone)]
+pub struct ResourceShape: u32 {
+    const BASE = shader_slang::ResourceShape::SlangResourceBaseShapeMask as u32;
+    const NONE = shader_slang::ResourceShape::SlangResourceNone as u32;
+    const TEXTURE_1D = shader_slang::ResourceShape::SlangTexture1d as u32;
+    const TEXTURE_2D = shader_slang::ResourceShape::SlangTexture2d as u32;
+    const TEXTURE_3D = shader_slang::ResourceShape::SlangTexture3d as u32;
+    const TEXTURE_CUBE = shader_slang::ResourceShape::SlangTextureCube as u32;
+    const TEXTURE_BUFFER = shader_slang::ResourceShape::SlangTextureBuffer as u32;
+    const STRUCTURED_BUFFER = shader_slang::ResourceShape::SlangStructuredBuffer as u32;
+    const BYTE_ADDRESS_BUFFER = shader_slang::ResourceShape::SlangByteAddressBuffer as u32;
+    const UNKNOWN = shader_slang::ResourceShape::SlangResourceUnknown as u32;
+    const ACCELERATION_STRUCTURE = shader_slang::ResourceShape::SlangAccelerationStructure as u32;
+    const TEXTURE_SUBPASS = shader_slang::ResourceShape::SlangTextureSubpass as u32;
+    const EXT_SHAPE = shader_slang::ResourceShape::SlangResourceExtShapeMask as u32;
+    const TEXTURE_FEEDBACK_FLAG = shader_slang::ResourceShape::SlangTextureFeedbackFlag as u32;
+    const TEXTURE_SHADOW_FLAG = shader_slang::ResourceShape::SlangTextureShadowFlag as u32;
+    const TEXTURE_ARRAY_FLAG = shader_slang::ResourceShape::SlangTextureArrayFlag as u32;
+    const TEXTURE_MULTISAMPLE_FLAG = shader_slang::ResourceShape::SlangTextureMultisampleFlag as u32;
+    const TEXTURE_COMBINED_FLAG = shader_slang::ResourceShape::SlangTextureCombinedFlag as u32;
+    const TEXTURE_1D_ARRAY = shader_slang::ResourceShape::SlangTexture1dArray as u32;
+    const TEXTURE_2D_ARRAY = shader_slang::ResourceShape::SlangTexture2dArray as u32;
+    const TEXTURE_CUBE_ARRAY = shader_slang::ResourceShape::SlangTextureCubeArray as u32;
+    const TEXTURE_2D_MULTISAMPLE = shader_slang::ResourceShape::SlangTexture2dMultisample as u32;
+    const TEXTURE_2D_MULTISAMPLE_ARRAY = shader_slang::ResourceShape::SlangTexture2dMultisampleArray as u32;
+    const TEXTURE_SUBPASS_MULTISAMPLE = shader_slang::ResourceShape::SlangTextureSubpassMultisample as u32;
+}
+}
+
 #[derive(Debug, Copy, Clone, Hash, PartialEq, Eq)]
 pub enum ParameterCategory
 {