1 files changed, 15 insertions, 9 deletions

diff --git a/engine/xe_pipeline.hpp b/engine/xe_pipeline.hpp
index 5e2e1fb..269a171 100755
--- a/engine/xe_pipeline.hpp
+++ b/engine/xe_pipeline.hpp
@@ -28,17 +28,20 @@ struct PipelineConfigInfo {
   uint32_t subpass = 0;
 };
 
-class XePipeline {
+class Pipeline {
   public:
-    XePipeline(
-      XeDevice &device, 
+    Pipeline(
+      Device &device, 
       const std::string& vertFilepath, 
       const std::string& fragFilepath, 
-      const PipelineConfigInfo& configInfo);
-    ~XePipeline();
+      const PipelineConfigInfo& configInfo,
+      std::vector<VkVertexInputAttributeDescription> &attributeDescptions,
+      uint32_t vertexSize  
+    );
+    ~Pipeline();
 
-    XePipeline(const XePipeline&) = delete;
-    XePipeline operator=(const XePipeline&) = delete;
+    Pipeline(const Pipeline&) = delete;
+    Pipeline operator=(const Pipeline&) = delete;
 
     void bind(VkCommandBuffer commandBuffer);
     static void defaultPipelineConfigInfo(PipelineConfigInfo& configInfo);
@@ -49,11 +52,14 @@ class XePipeline {
     void createGraphicsPipeline(
       const std::string& vertFilePath, 
       const std::string& fragFilepath, 
-      const PipelineConfigInfo& configInfo);
+      const PipelineConfigInfo& configInfo,
+      std::vector<VkVertexInputAttributeDescription> &attributeDescptions,
+      uint32_t vertexSize
+    );
 
     void createShaderModule(const std::vector<char>& code, VkShaderModule* shaderModule);
 
-    XeDevice& xeDevice;
+    Device& xeDevice;
     VkPipeline graphicsPipeline;
     VkShaderModule vertShaderModule;
     VkShaderModule fragShaderModule;