jedi-academy/code/qcommon/MiniHeap.h
2013-04-24 01:24:17 +10:00

67 lines
991 B
C++

#if !defined(MINIHEAP_H_INC)
#define MINIHEAP_H_INC
class CMiniHeap
{
char *mHeap;
char *mCurrentHeap;
int mSize;
#if _DEBUG
int mMaxAlloc;
#endif
public:
// reset the heap back to the start
void ResetHeap()
{
#if _DEBUG
if ((size_t)mCurrentHeap - (size_t)mHeap>mMaxAlloc)
{
mMaxAlloc=(size_t)mCurrentHeap - (size_t)mHeap;
}
#endif
mCurrentHeap = mHeap;
}
// initialise the heap
CMiniHeap(int size)
{
mHeap = (char *)Z_Malloc(size, TAG_GHOUL2, qtrue);
mSize = size;
#if _DEBUG
mMaxAlloc=0;
#endif
if (mHeap)
{
ResetHeap();
}
}
// free up the heap
~CMiniHeap()
{
if (mHeap)
{
// the quake heap will be long gone, no need to free it Z_Free(mHeap);
}
}
// give me some space from the heap please
char *MiniHeapAlloc(int size)
{
if (size < (mSize - ((size_t)mCurrentHeap - (size_t)mHeap)))
{
char *tempAddress = mCurrentHeap;
mCurrentHeap += size;
return tempAddress;
}
return NULL;
}
};
extern CMiniHeap *G2VertSpaceServer;
#endif //MINIHEAP_H_INC