summaryrefslogtreecommitdiffstats
path: root/src/AllocationPool.h
diff options
context:
space:
mode:
Diffstat (limited to 'src/AllocationPool.h')
-rw-r--r--src/AllocationPool.h46
1 files changed, 28 insertions, 18 deletions
diff --git a/src/AllocationPool.h b/src/AllocationPool.h
index 1b27462ab..0a529a8e3 100644
--- a/src/AllocationPool.h
+++ b/src/AllocationPool.h
@@ -11,18 +11,18 @@ class cAllocationPool
public:
class cStarvationCallbacks
{
- public:
- virtual ~cStarvationCallbacks() {}
+ public:
+ virtual ~cStarvationCallbacks() {}
- /** Is called when the reserve buffer starts to be used */
- virtual void OnStartUsingReserve() = 0;
+ /** Is called when the reserve buffer starts to be used */
+ virtual void OnStartUsingReserve() = 0;
- /** Is called once the reserve buffer has returned to normal size */
- virtual void OnEndUsingReserve() = 0;
+ /** Is called once the reserve buffer has returned to normal size */
+ virtual void OnEndUsingReserve() = 0;
- /** Is called when the allocation pool is unable to allocate memory. Will be repeatedly
+ /** Is called when the allocation pool is unable to allocate memory. Will be repeatedly
called if it does not free sufficient memory */
- virtual void OnOutOfReserve() = 0;
+ virtual void OnOutOfReserve() = 0;
};
virtual ~cAllocationPool() {}
@@ -65,11 +65,12 @@ class cListAllocationPool:
{
public:
- cListAllocationPool(std::unique_ptr<typename cAllocationPool<T>::cStarvationCallbacks> a_Callbacks, size_t a_NumElementsInReserve):
- m_NumElementsInReserve(a_NumElementsInReserve),
+ cListAllocationPool(std::unique_ptr<typename cAllocationPool<T>::cStarvationCallbacks> a_Callbacks, size_t a_MinElementsInReserve, size_t a_MaxElementsInReserve) :
+ m_MinElementsInReserve(a_MinElementsInReserve),
+ m_MaxElementsInReserve(a_MaxElementsInReserve),
m_Callbacks(std::move(a_Callbacks))
{
- for (size_t i = 0; i < m_NumElementsInReserve; i++)
+ for (size_t i = 0; i < m_MinElementsInReserve; i++)
{
void * space = malloc(sizeof(T));
if (space == nullptr)
@@ -86,7 +87,7 @@ public:
{
while (!m_FreeList.empty())
{
- free (m_FreeList.front());
+ free(m_FreeList.front());
m_FreeList.pop_front();
}
}
@@ -94,7 +95,7 @@ public:
virtual T * Allocate() override
{
- if (m_FreeList.size() <= m_NumElementsInReserve)
+ if (m_FreeList.size() <= m_MinElementsInReserve)
{
void * space = malloc(sizeof(T));
if (space != nullptr)
@@ -113,7 +114,7 @@ public:
#pragma pop_macro("new")
#endif
}
- else if (m_FreeList.size() == m_NumElementsInReserve)
+ else if (m_FreeList.size() == m_MinElementsInReserve)
{
m_Callbacks->OnStartUsingReserve();
}
@@ -151,10 +152,17 @@ public:
{
return;
}
- // placement destruct.
- a_ptr->~T();
+
+ a_ptr->~T(); // placement destruct.
+
+ if (m_FreeList.size() >= m_MaxElementsInReserve)
+ {
+ free(a_ptr);
+ return;
+ }
+
m_FreeList.push_front(a_ptr);
- if (m_FreeList.size() == m_NumElementsInReserve)
+ if (m_FreeList.size() == m_MinElementsInReserve)
{
m_Callbacks->OnEndUsingReserve();
}
@@ -162,7 +170,9 @@ public:
private:
/** The minimum number of elements to keep in the free list before malloc fails */
- size_t m_NumElementsInReserve;
+ size_t m_MinElementsInReserve;
+ /** Maximum free list size before returning memory to the OS */
+ size_t m_MaxElementsInReserve;
std::list<void *> m_FreeList;
std::unique_ptr<typename cAllocationPool<T>::cStarvationCallbacks> m_Callbacks;