From 485752de82e6058473dfe8dccdf64ac04e983ae9 Mon Sep 17 00:00:00 2001
From: Tycho <work.tycho+git@gmail.com>
Date: Wed, 21 May 2014 20:59:04 +0100
Subject: Implemented Allocation Pool

---
 src/AllocationPool.h | 50 ++++++++++++++++++++++++++++++++++++++++++++++++++
 1 file changed, 50 insertions(+)
 create mode 100644 src/AllocationPool.h

(limited to 'src/AllocationPool.h')

diff --git a/src/AllocationPool.h b/src/AllocationPool.h
new file mode 100644
index 000000000..f03897f98
--- /dev/null
+++ b/src/AllocationPool.h
@@ -0,0 +1,50 @@
+
+#pragma once
+
+template<class T, size_t BufferSize, class StarvationCallbacks>
+class AllocationPool {
+	public:
+		
+		~AllocationPool()
+		{
+			while (!m_FreeList.empty())
+			{
+				delete m_FreeList.front();
+				m_FreeList.pop_front();
+			}
+		}	
+		
+		T* Allocate()
+		{
+			if (m_FreeList.Size() <= BufferSize)
+			{
+				try
+				{
+					return new T;
+				}
+				catch (std::bad_alloc& ex)
+				{
+					if (m_FreeList.size() == BufferSize)
+					{
+						StarvationCallbacks.OnStartingUsingBuffer();
+					}
+					else if (m_FreeList.empty())
+					{
+						StarvationCallbacks.OnBufferEmpty();
+						// Try again until the memory is avalable
+						return Allocate();
+					}
+				}
+			}
+			T* ret = m_FreeList.front();
+			m_FreeList.pop_front();
+			return ret;
+		}
+		void Free(T* ptr)
+		{
+			m_FreeList.push_front(ptr);
+		}
+		
+	private:
+		std::list<T*> m_FreeList;
+}
-- 
cgit v1.2.3