From 1177d50eda69fb5be711b5ebfa6ee290500ae342 Mon Sep 17 00:00:00 2001
From: Jack Andersen <jackoalan@gmail.com>
Date: Sun, 11 Dec 2016 10:20:29 -1000
Subject: [PATCH] Metal pool buffers

---
 include/boo/graphicsdev/Metal.hpp |  1 +
 lib/graphicsdev/GL.cpp            |  2 +-
 lib/graphicsdev/Metal.mm          | 14 ++++++++++----
 3 files changed, 12 insertions(+), 5 deletions(-)

diff --git a/include/boo/graphicsdev/Metal.hpp b/include/boo/graphicsdev/Metal.hpp
index d1c957d..8af0d15 100644
--- a/include/boo/graphicsdev/Metal.hpp
+++ b/include/boo/graphicsdev/Metal.hpp
@@ -31,6 +31,7 @@ class MetalDataFactory : public IGraphicsDataFactory
     void destroyPool(IGraphicsBufferPool*);
     IGraphicsBufferD* newPoolBuffer(IGraphicsBufferPool* pool, BufferUse use,
                                     size_t stride, size_t count);
+    void deletePoolBuffer(IGraphicsBufferPool* p, IGraphicsBufferD* buf);
 public:
     MetalDataFactory(IGraphicsContext* parent, MetalContext* ctx, uint32_t sampleCount);
     ~MetalDataFactory() {}
diff --git a/lib/graphicsdev/GL.cpp b/lib/graphicsdev/GL.cpp
index 9795698..8ddec18 100644
--- a/lib/graphicsdev/GL.cpp
+++ b/lib/graphicsdev/GL.cpp
@@ -739,7 +739,7 @@ IGraphicsBufferD* GLDataFactory::newPoolBuffer(IGraphicsBufferPool* p, BufferUse
 {
     GLPool* pool = static_cast<GLPool*>(p);
     GLGraphicsBufferD* retval = new GLGraphicsBufferD(use, stride * count);
-    pool->m_DBufs.emplace(std::make_pair(retval, retval));
+    pool->m_DBufs.emplace(std::make_pair(retval, std::unique_ptr<GLGraphicsBufferD>(retval)));
     return retval;
 }
 
diff --git a/lib/graphicsdev/Metal.mm b/lib/graphicsdev/Metal.mm
index 49e845a..e5e27e2 100644
--- a/lib/graphicsdev/Metal.mm
+++ b/lib/graphicsdev/Metal.mm
@@ -33,7 +33,7 @@ struct MetalData : IGraphicsData
 
 struct MetalPool : IGraphicsBufferPool
 {
-    std::vector<std::unique_ptr<class MetalGraphicsBufferD>> m_DBufs;
+    std::unordered_map<class MetalGraphicsBufferD*, std::unique_ptr<class MetalGraphicsBufferD>> m_DBufs;
 };
 
 #define MTL_STATIC MTLResourceCPUCacheModeWriteCombined|MTLResourceStorageModeShared
@@ -888,8 +888,8 @@ struct MetalCommandQueue : IGraphicsCommandQueue
         }
         for (MetalPool* p : gfxF->m_committedPools)
         {
-            for (std::unique_ptr<MetalGraphicsBufferD>& b : p->m_DBufs)
-                b->update(m_fillBuf);
+            for (auto& b : p->m_DBufs)
+                b.second->update(m_fillBuf);
         }
         datalk.unlock();
 
@@ -1184,10 +1184,16 @@ IGraphicsBufferD* MetalDataFactory::newPoolBuffer(IGraphicsBufferPool* p, Buffer
     MetalPool* pool = static_cast<MetalPool*>(p);
     MetalCommandQueue* q = static_cast<MetalCommandQueue*>(m_parent->getCommandQueue());
     MetalGraphicsBufferD* retval = new MetalGraphicsBufferD(q, use, m_ctx, stride, count);
-    pool->m_DBufs.emplace_back(retval);
+    pool->m_DBufs.emplace(std::make_pair(retval, std::unique_ptr<MetalGraphicsBufferD>(retval)));
     return retval;
 }
 
+void MetalDataFactory::deletePoolBuffer(IGraphicsBufferPool* p, IGraphicsBufferD* buf)
+{
+    MetalPool* pool = static_cast<MetalPool*>(p);
+    pool->m_DBufs.erase(static_cast<MetalGraphicsBufferD*>(buf));
+}
+
 IGraphicsCommandQueue* _NewMetalCommandQueue(MetalContext* ctx, IWindow* parentWindow,
                                              IGraphicsContext* parent)
 {