Implemented Allocation Pool
This commit is contained in:
parent
cdd3d11496
commit
485752de82
50
src/AllocationPool.h
Normal file
50
src/AllocationPool.h
Normal file
@ -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;
|
||||||
|
}
|
Loading…
Reference in New Issue
Block a user