[add]: Custom memory managers and fix includes

1 files changed, 7 insertions, 5 deletions

diff --git a/src/Rendering/VertexArray.h b/src/Rendering/VertexArray.h
index 8ae61a7..10fff89 100644
--- a/src/Rendering/VertexArray.h
+++ b/src/Rendering/VertexArray.h
@@ -1,8 +1,10 @@
 #pragma once
 
+#include "Core/Memory.h"
+
 #include "VertexBuffer.h"
 #include "IndexBuffer.h"
-#include <memory>
+
 #include <vector>
 
 namespace Donut
@@ -15,11 +17,11 @@ namespace Donut
         virtual void Bind()   const = 0;
         virtual void Unbind() const = 0;
 
-        virtual void AddVertexBuffer(const std::shared_ptr<VertexBuffer>& vertexBuffer) = 0;
-        virtual void SetIndexBuffer(const std::shared_ptr<IndexBuffer>& indexBuffer)    = 0;
+        virtual void AddVertexBuffer(const Ref<VertexBuffer>& vertexBuffer) = 0;
+        virtual void SetIndexBuffer(const Ref<IndexBuffer>& indexBuffer)    = 0;
 
-        virtual const std::vector<std::shared_ptr<VertexBuffer>>& GetVertexBuffers() const = 0;
-        virtual const std::shared_ptr<IndexBuffer>& GetIndexBuffer()                 const = 0;
+        virtual const std::vector<Ref<VertexBuffer>>& GetVertexBuffers() const = 0;
+        virtual const Ref<IndexBuffer>& GetIndexBuffer()                 const = 0;
 
         static VertexArray* Create();
     };