mirror of
https://github.com/DarkflameUniverse/DarkflameServer.git
synced 2024-11-14 12:18:22 +00:00
41 lines
955 B
C++
41 lines
955 B
C++
#ifndef __BYTE_POOL_H
|
|
#define __BYTE_POOL_H
|
|
|
|
#include "RakMemoryOverride.h"
|
|
#include "DS_MemoryPool.h"
|
|
#include "Export.h"
|
|
#include "SimpleMutex.h"
|
|
#include <assert.h>
|
|
|
|
// #define _DISABLE_BYTE_POOL
|
|
// #define _THREADSAFE_BYTE_POOL
|
|
|
|
namespace DataStructures
|
|
{
|
|
// Allocate some number of bytes from pools. Uses the heap if necessary.
|
|
class RAK_DLL_EXPORT BytePool : public RakNet::RakMemoryOverride
|
|
{
|
|
public:
|
|
BytePool();
|
|
~BytePool();
|
|
// Should be at least 8 times bigger than 8192
|
|
void SetPageSize(int size);
|
|
unsigned char* Allocate(int bytesWanted);
|
|
void Release(unsigned char *data);
|
|
void Clear(void);
|
|
protected:
|
|
MemoryPool<unsigned char[128]> pool128;
|
|
MemoryPool<unsigned char[512]> pool512;
|
|
MemoryPool<unsigned char[2048]> pool2048;
|
|
MemoryPool<unsigned char[8192]> pool8192;
|
|
#ifdef _THREADSAFE_BYTE_POOL
|
|
SimpleMutex mutex128;
|
|
SimpleMutex mutex512;
|
|
SimpleMutex mutex2048;
|
|
SimpleMutex mutex8192;
|
|
#endif
|
|
};
|
|
}
|
|
|
|
#endif
|