#ifndef __BYTE_POOL_H

#define __BYTE_POOL_H

#include "DS_MemoryPool.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 BytePool

{

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