jedi-academy/code/qcommon/MiniHeap.h

68 lines
991 B
C
Raw Normal View History

2013-04-19 02:52:48 +00:00
#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
2013-04-20 06:19:40 +00:00
if ((size_t)mCurrentHeap - (size_t)mHeap>mMaxAlloc)
2013-04-19 02:52:48 +00:00
{
2013-04-20 06:19:40 +00:00
mMaxAlloc=(size_t)mCurrentHeap - (size_t)mHeap;
2013-04-19 02:52:48 +00:00
}
#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)
{
2013-04-20 06:19:40 +00:00
if (size < (mSize - ((size_t)mCurrentHeap - (size_t)mHeap)))
2013-04-19 02:52:48 +00:00
{
char *tempAddress = mCurrentHeap;
mCurrentHeap += size;
return tempAddress;
}
return NULL;
}
};
extern CMiniHeap *G2VertSpaceServer;
#endif //MINIHEAP_H_INC