File: MemPool.cpp

package info (click to toggle)
spring 98.0%2Bdfsg-1
  • links: PTS, VCS
  • area: main
  • in suites: jessie, jessie-kfreebsd
  • size: 41,928 kB
  • ctags: 60,665
  • sloc: cpp: 356,167; ansic: 39,434; python: 12,228; java: 12,203; awk: 5,856; sh: 1,719; xml: 997; perl: 405; php: 253; objc: 194; makefile: 72; sed: 2
file content (60 lines) | stat: -rw-r--r-- 1,308 bytes parent folder | download
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
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
/* This file is part of the Spring engine (GPL v2 or later), see LICENSE.html */

#include "System/MemPool.h"
//#include "System/mmgr.h"

CMemPool mempool;

CMemPool::CMemPool()
{
	for (size_t a = 0; a < (MAX_MEM_SIZE + 1); a++) {
		nextFree[a] = NULL;
		poolSize[a] = 10;
	}
}

void* CMemPool::Alloc(size_t numBytes)
{
	if (UseExternalMemory(numBytes)) {
		return ::operator new(numBytes);
	} else {
		void* pnt = nextFree[numBytes];

		if (pnt) {
			nextFree[numBytes] = (*(void**)pnt);
		} else {
			void* newBlock = ::operator new(numBytes * poolSize[numBytes]);
			allocated.push_back(newBlock);
			for (int i = 0; i < (poolSize[numBytes] - 1); ++i) {
				*(void**)&((char*)newBlock)[(i) * numBytes] = (void*)&((char*)newBlock)[(i + 1) * numBytes];
			}

			*(void**)&((char*)newBlock)[(poolSize[numBytes] - 1) * numBytes] = 0;

			pnt = newBlock;
			nextFree[numBytes] = (*(void**)pnt);
			poolSize[numBytes] *= 2;
		}
		return pnt;
	}
}

void CMemPool::Free(void* pnt, size_t numBytes)
{
	if (pnt == NULL) {
		return;
	}

	if (UseExternalMemory(numBytes)) {
		::operator delete(pnt);
	} else {
		*(void**)pnt = nextFree[numBytes];
		nextFree[numBytes] = pnt;
	}
}

CMemPool::~CMemPool()
{
	for(std::vector<void *>::iterator i = allocated.begin(); i != allocated.end(); ++i)
		::operator delete(*i);
}