91 lines
1.9 KiB
C++
91 lines
1.9 KiB
C++
#ifndef _BASE_ALLOCATOR_H_
|
|
#define _BASE_ALLOCATOR_H_
|
|
|
|
/************************************************************************
|
|
* 内存分配器的基类，其他分配器可继承其接口，实现不同的内存分配算法
|
|
* 这个基类实现的内存分配就是直接用crt分配内存
|
|
************************************************************************/
|
|
|
|
#include "os_def.h"
|
|
|
|
#include "memory_counter.hpp"
|
|
|
|
#ifdef _MEMORY_TRACE_
|
|
#define AllocBuffer(count_) _AllocBuffer(count_,__FILE__, __LINE__)
|
|
#define ReAllocBuffer(src,count_) _ReAllocBuffer(src,count_,__FILE__, __LINE__)
|
|
#define FreeBuffer(ptr) _FreeBuffer(ptr, __FILE__, __LINE__)
|
|
#endif
|
|
|
|
class BaseAllocator : public AllocatorCounterItem
|
|
{
|
|
public:
|
|
#ifndef _MEMORY_TRACE_
|
|
virtual void* AllocBuffer(size_t count)
|
|
#else
|
|
virtual void* _AllocBuffer(size_t count, const char*, int)
|
|
#endif
|
|
{
|
|
size_t* result = (size_t*)malloc(count + sizeof(size_t));
|
|
#ifdef _MEMORY_TRACE_
|
|
|
|
if (result != NULL)
|
|
{
|
|
alloc_total_ += count;
|
|
|
|
*result = count;
|
|
}
|
|
|
|
#endif
|
|
return (void*)(result + 1);
|
|
}
|
|
|
|
#ifndef _MEMORY_TRACE_
|
|
virtual void* ReAllocBuffer(void* src, size_t count)
|
|
#else
|
|
virtual void* _ReAllocBuffer(void* src, size_t count, const char*, int)
|
|
#endif
|
|
{
|
|
#ifdef _MEMORY_TRACE_
|
|
if (src)
|
|
{
|
|
size_t old = *((size_t*)src - 1);
|
|
alloc_total_ -= old;
|
|
src = (void*)((size_t*)src - 1);
|
|
}
|
|
alloc_total_ += count; // 注意是size_t类型
|
|
#endif
|
|
|
|
size_t* result = (size_t*)realloc(src, count + sizeof(size_t));
|
|
*result = count;
|
|
return (void*)(result + 1);
|
|
}
|
|
#ifndef _MEMORY_TRACE_
|
|
virtual void FreeBuffer(void* ptr)
|
|
#else
|
|
virtual void _FreeBuffer(void* ptr, const char*, int)
|
|
#endif
|
|
{
|
|
if (!ptr) return;
|
|
|
|
#ifdef _MEMORY_TRACE_
|
|
size_t old = *((size_t*)ptr - 1);
|
|
alloc_total_ -= old;
|
|
#endif
|
|
free((size_t*)ptr - 1);
|
|
}
|
|
BaseAllocator(const char* namestr) : AllocatorCounterItem(namestr)
|
|
{
|
|
}
|
|
virtual ~BaseAllocator()
|
|
{
|
|
}
|
|
|
|
void SetMultiThread(bool){}
|
|
void GC()
|
|
{
|
|
|
|
}
|
|
};
|
|
|
|
#endif //end _BASE_ALLOCATOR_H_
|