1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
|
#include "stdafx.h"
#include "MemoryPool.h"
namespace util {
MemoryPool::MemoryPool(nat size) : data(new byte[size]), dataSize(size), firstFree(0), allocCount(0) {}
MemoryPool::~MemoryPool() {
dbg_assert(allocCount == 0, L"Outstanding allocations!");
delete []data;
}
void *MemoryPool::alloc(nat size) {
if (dataSize - firstFree < size - 1) {
//Not enough free space. We're filled.
firstFree = dataSize;
return null;
}
nat allocAt = firstFree + sizeof(void *);
void **ownerPtr = (void **)&data[firstFree];
*ownerPtr = this;
firstFree += sizeof(void *) + size;
allocCount++;
return (void *)&data[allocAt];
}
void MemoryPool::free(void *memory) {
UNUSED(memory);
assert(allocCount > 0);
if (--allocCount == 0) {
firstFree = 0;
}
}
MemoryPool *MemoryPool::getPoolFromAlloc(void *mem) {
MemoryPool **ptr = (MemoryPool **)mem;
return ptr[-1];
}
}
|
