DS_BytePool.h

Go to the documentation of this file.

#ifndef __BYTE_POOL_H
#define __BYTE_POOL_H

#include "RakMemoryOverride.h"
#include "DS_MemoryPool.h"
#include "Export.h"
#include "SimpleMutex.h"
#include "RakAssert.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:
                BytePool();
                ~BytePool();
                // Should be at least 8 times bigger than 8192
                void SetPageSize(int size);
                unsigned char* Allocate(int bytesWanted, const char *file, unsigned int line);
                void Release(unsigned char *data, const char *file, unsigned int line);
                void Clear(const char *file, unsigned int line);
        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