MemoryAllocator.h

Go to the documentation of this file.

#pragma once

#include <memory>

namespace ai {

class _DefaultAllocator {
private:
    _DefaultAllocator() {
    }
public:
    static inline void* allocate(size_t count) {
        void* ptr = new unsigned char[count];
        return ptr;
    }

    static inline void deallocate(void* ptr) {
        delete[] ((unsigned char*) ptr);
    }
};

template<class AllocatorClass>
class _MemObject {
public:
    explicit _MemObject() {
    }

    virtual ~_MemObject() {
    }

    inline void* operator new(size_t size) {
        return AllocatorClass::allocate(size);
    }

    inline void* operator new(size_t, void* ptr) {
        return ptr;
    }

    inline void* operator new[](size_t size) {
        return AllocatorClass::allocate(size);
    }

    inline void operator delete(void* ptr) {
        AllocatorClass::deallocate(ptr);
    }

    inline void operator delete(void* ptr, void*) {
        AllocatorClass::deallocate(ptr);
    }

    inline void operator delete[](void* ptr) {
        AllocatorClass::deallocate(ptr);
    }
};

#ifndef AI_ALLOCATOR_CLASS
#define AI_ALLOCATOR_CLASS _DefaultAllocator
#endif

typedef _MemObject<AI_ALLOCATOR_CLASS> MemObject;

}