#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