irrAllocator.hpp

// Copyright (C) 2002-2012 Nikolaus Gebhardt
// This file is part of the "Irrlicht Engine" and the "irrXML" project.
// For conditions of distribution and use, see copyright notice in nirtcpp/nirtcpp.hpp and irrXML.h
 
#ifndef NIRT_ALLOCATOR_HPP_INCLUDED
#define NIRT_ALLOCATOR_HPP_INCLUDED
 
#include <nirtcpp/core/engine/nirt_types.hpp>
#include <new>
// necessary for older compilers
#include <memory.h>
 
namespace nirt
{
namespace core
{
 
#ifdef DEBUG_CLIENTBLOCK
#undef DEBUG_CLIENTBLOCK
#define DEBUG_CLIENTBLOCK new
#endif
 
template<typename T>
class irrAllocator
{
public:
 
    virtual ~irrAllocator() {}
 
    T* allocate(size_t cnt)
    {
        return (T*)internal_new(cnt* sizeof(T));
    }
 
    void deallocate(T* ptr)
    {
        internal_delete(ptr);
    }
 
    void construct(T* ptr, const T&e)
    {
        new ((void*)ptr) T(e);
    }
 
    void destruct(T* ptr)
    {
        ptr->~T();
    }
 
protected:
 
    virtual void* internal_new(size_t cnt)
    {
        return operator new(cnt);
    }
 
    virtual void internal_delete(void* ptr)
    {
        operator delete(ptr);
    }
 
};
 
 
 
template<typename T>
class irrAllocatorFast
{
public:
 
    T* allocate(size_t cnt)
    {
        return (T*)operator new(cnt* sizeof(T));
    }
 
    void deallocate(T* ptr)
    {
        operator delete(ptr);
    }
 
    void construct(T* ptr, const T&e)
    {
        new ((void*)ptr) T(e);
    }
 
    void destruct(T* ptr)
    {
        ptr->~T();
    }
};
 
 
 
#ifdef DEBUG_CLIENTBLOCK
#undef DEBUG_CLIENTBLOCK
#define DEBUG_CLIENTBLOCK new( _CLIENT_BLOCK, __FILE__, __LINE__)
#endif
 
enum eAllocStrategy
{
    ALLOC_STRATEGY_SAFE    = 0, // increase size by 1
    ALLOC_STRATEGY_DOUBLE  = 1, // double size when under 500 elements, beyond that increase by 1/4th size. Plus a small constant.
    ALLOC_STRATEGY_SQRT    = 2  // not implemented
};
 
 
} // end namespace core
} // end namespace nirt
 
#endif