98 lines
2.4 KiB
C++
98 lines
2.4 KiB
C++
// Copyright (C) 2014 Jérôme Leclercq
|
|
// This file is part of the "Nazara Engine - Core module"
|
|
// For conditions of distribution and use, see copyright notice in Config.hpp
|
|
|
|
#include <new>
|
|
#include <Nazara/Core/Debug.hpp>
|
|
|
|
template<unsigned int typeSize, unsigned int count, bool canGrow>
|
|
NzMemoryPool<typeSize, count, canGrow>::NzMemoryPool() :
|
|
m_previous(nullptr),
|
|
m_freeCount(count)
|
|
{
|
|
m_pool.reset(new nzUInt8[typeSize * count]);
|
|
m_freeList.reset(new void*[count]);
|
|
|
|
// Remplissage de la free list
|
|
for (unsigned int i = 0; i < count; ++i)
|
|
m_freeList[i] = &m_pool[typeSize * (count-i-1)];
|
|
}
|
|
|
|
template<unsigned int typeSize, unsigned int count, bool canGrow>
|
|
NzMemoryPool<typeSize, count, canGrow>::NzMemoryPool(NzMemoryPool* pool) :
|
|
NzMemoryPool()
|
|
{
|
|
m_previous = pool;
|
|
}
|
|
|
|
template<unsigned int typeSize, unsigned int count, bool canGrow>
|
|
template<typename T>
|
|
T* NzMemoryPool<typeSize, count, canGrow>::Allocate()
|
|
{
|
|
static_assert(sizeof(T) <= typeSize, "This type is too large for this memory pool");
|
|
|
|
return static_cast<T*>(Allocate(sizeof(T)));
|
|
}
|
|
|
|
template<unsigned int typeSize, unsigned int count, bool canGrow>
|
|
void* NzMemoryPool<typeSize, count, canGrow>::Allocate(unsigned int size)
|
|
{
|
|
if (size > typeSize)
|
|
{
|
|
throw std::bad_alloc();
|
|
return nullptr;
|
|
}
|
|
|
|
if (m_freeCount > 0)
|
|
return m_freeList[--m_freeCount];
|
|
else if (canGrow)
|
|
{
|
|
if (!m_next)
|
|
m_next.reset(new NzMemoryPool(this));
|
|
|
|
return m_next->Allocate(size);
|
|
}
|
|
|
|
throw std::bad_alloc();
|
|
return nullptr;
|
|
}
|
|
|
|
template<unsigned int typeSize, unsigned int count, bool canGrow>
|
|
void NzMemoryPool<typeSize, count, canGrow>::Free(void* ptr)
|
|
{
|
|
if (ptr)
|
|
{
|
|
// Le pointer nous appartient-il ?
|
|
nzUInt8* freePtr = static_cast<nzUInt8*>(ptr);
|
|
nzUInt8* poolPtr = m_pool.get();
|
|
if (freePtr >= poolPtr && freePtr < poolPtr + typeSize*count)
|
|
{
|
|
#if NAZARA_CORE_SAFE
|
|
if ((freePtr - poolPtr) % typeSize != 0)
|
|
{
|
|
throw std::runtime_error("Pointer does not belong to memory pool");
|
|
return;
|
|
}
|
|
#endif
|
|
|
|
m_freeList[m_freeCount++] = ptr;
|
|
|
|
// Si nous sommes vide et l'extension d'un autre pool, on se suicide
|
|
if (m_freeCount == count && m_previous && !m_next)
|
|
{
|
|
m_previous->m_next.release();
|
|
delete this; // Suicide
|
|
}
|
|
}
|
|
else
|
|
{
|
|
if (m_next)
|
|
m_next->Free(ptr);
|
|
else
|
|
throw std::runtime_error("Pointer does not belong to memory pool");
|
|
}
|
|
}
|
|
}
|
|
|
|
#include <Nazara/Core/DebugOff.hpp>
|