From 703d444cbb8fce4988ca74b8a80c3a75ba65901b Mon Sep 17 00:00:00 2001
From: "HuYingzhuo(hugo/hyzboy)" <hyzboy@outlook.com>
Date: Sun, 19 Mar 2023 20:17:27 +0800
Subject: [PATCH] added GetInput at ShaderCreateInfoVertex

---
 inc/hgl/shadergen/ShaderCreateInfoVertex.h | 4 ++++
 src/ShaderGen/ShaderCreateInfoVertex.cpp   | 6 ++++++
 2 files changed, 10 insertions(+)

diff --git a/inc/hgl/shadergen/ShaderCreateInfoVertex.h b/inc/hgl/shadergen/ShaderCreateInfoVertex.h
index 8fd1d47a..82de6c70 100644
--- a/inc/hgl/shadergen/ShaderCreateInfoVertex.h
+++ b/inc/hgl/shadergen/ShaderCreateInfoVertex.h
@@ -1,6 +1,8 @@
 #pragma once
 
 #include<hgl/shadergen/ShaderCreateInfo.h>
+#include<hgl/graph/VKShaderStage.h>
+
 
 SHADERGEN_NAMESPACE_BEGIN
 class ShaderCreateInfoVertex:public ShaderCreateInfo
@@ -14,5 +16,7 @@ public:
 
     int AddInput(const graph::VAT &type,const AnsiString &name);
     int AddInput(const AnsiString &type,const AnsiString &name);
+
+    const hgl::graph::ShaderAttributeArray &GetInput()const;
 };
 SHADERGEN_NAMESPACE_END
diff --git a/src/ShaderGen/ShaderCreateInfoVertex.cpp b/src/ShaderGen/ShaderCreateInfoVertex.cpp
index 2c6499ec..66244f42 100644
--- a/src/ShaderGen/ShaderCreateInfoVertex.cpp
+++ b/src/ShaderGen/ShaderCreateInfoVertex.cpp
@@ -2,6 +2,7 @@
 #include<hgl/shadergen/ShaderDescriptorInfo.h>
 #include<hgl/graph/VertexAttrib.h>
 #include<hgl/graph/VKShaderStage.h>
+#include"GLSLCompiler.h"
 
 SHADERGEN_NAMESPACE_BEGIN
 
@@ -59,4 +60,9 @@ bool ShaderCreateInfoVertex::ProcInput(ShaderCreateInfo *)
 
     return(true);
 }
+
+const hgl::graph::ShaderAttributeArray &ShaderCreateInfoVertex::GetInput()const
+{
+    return spv_data->stage_io.input;
+}
 SHADERGEN_NAMESPACE_END