1 files changed, 4 insertions, 2 deletions
diff --git a/include/RenderBackend/OpenGL/VertexShader.hpp b/include/RenderBackend/OpenGL/VertexShader.hpp
index 43a5dd8..aaf8b24 100644
--- a/include/RenderBackend/OpenGL/VertexShader.hpp
+++ b/include/RenderBackend/OpenGL/VertexShader.hpp
@@ -3,6 +3,7 @@
 #include "../../DataView.hpp"
 #include "../../utils.hpp"
 #include "../../Result.hpp"
+#include "CommonShader.hpp"
 #include "ShaderVec.hpp"
 #include <string>
 #include <unordered_map>
@@ -51,7 +52,8 @@ namespace amalgine
     public:
         VertexShader();
         
-        const std::string& getInputAttributeName(i32 attributeIndex);
+        const std::string& getInputAttributeName(i32 attributeIndex) const;
+        AttributeType getInputAttributeType(i32 attributeIndex) const;
         ShaderInputVec2 defineInputVec2(const std::string &name);
         void defineMain(VertexShaderMainFunc mainFunc);
         Result<CompiledVertexShader*> compile();
@@ -71,7 +73,7 @@ namespace amalgine
         int locationCounter;
         i32 maxVertexAttribs; // Could make this static
         std::unordered_map<std::string, i32> vertexAttributes;
-        std::vector<std::string> vertexAttributeNames;
+        std::vector<ShaderAttribute> vertexAttributeNames;
         bool mainFuncDefined;
     };
 }