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
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
|
/* This file is part of the Spring engine (GPL v2 or later), see LICENSE.html */
#ifndef LUA_MEM_POOL_H_
#define LUA_MEM_POOL_H_
#include <cstddef>
#include <vector>
#include "System/UnorderedMap.hpp"
class CLuaHandle;
class LuaMemPool {
public:
LuaMemPool(size_t lmpIndex);
~LuaMemPool() { Clear(); }
LuaMemPool(const LuaMemPool& p) = delete;
LuaMemPool(LuaMemPool&& p) = delete;
LuaMemPool& operator = (const LuaMemPool& p) = delete;
LuaMemPool& operator = (LuaMemPool&& p) = delete;
public:
static size_t GetPoolCount();
static LuaMemPool* GetSharedPtr();
static LuaMemPool* AcquirePtr(bool shared, bool owned);
static void ReleasePtr(LuaMemPool* p, const CLuaHandle* o);
static void FreeShared();
static void InitStatic(bool enable);
static void KillStatic();
public:
void Clear() {
DeleteBlocks();
ClearStats(true);
ClearTables();
}
void Reserve(size_t size) {
freeChunksTable.reserve(size);
chunkCountTable.reserve(size);
#if 1
allocBlocks.reserve(size / 16);
#endif
}
void DeleteBlocks();
void* Alloc(size_t size);
void* Realloc(void* ptr, size_t nsize, size_t osize);
void Free(void* ptr, size_t size);
void LogStats(const char* handle, const char* lctype) const;
void ClearStats(bool b) {
allocStats[STAT_NIA] *= (1 - b);
allocStats[STAT_NEA] *= (1 - b);
allocStats[STAT_NRA] *= (1 - b);
allocStats[STAT_NCB] *= (1 - b);
allocStats[STAT_NBB] *= (1 - b);
}
void ClearTables() {
freeChunksTable.clear();
chunkCountTable.clear();
}
size_t GetGlobalIndex() const { return globalIndex; }
size_t GetSharedCount() const { return sharedCount; }
size_t& GetSharedCount() { return sharedCount; }
public:
static constexpr size_t MIN_ALLOC_SIZE = sizeof(void*);
static constexpr size_t MAX_ALLOC_SIZE = (1024 * 1024) - 1;
static bool enabled;
private:
spring::unsynced_map<size_t, void*> freeChunksTable;
spring::unsynced_map<size_t, size_t> chunkCountTable;
std::vector<void*> allocBlocks;
enum {
STAT_NIA = 0, // number of internal allocs
STAT_NEA = 1, // number of external allocs
STAT_NRA = 2, // number of recycled allocs
STAT_NCB = 3, // number of chunk bytes currently in use
STAT_NBB = 4, // number of block bytes alloced in total
};
size_t allocStats[5] = {0, 0, 0, 0, 0};
size_t globalIndex = 0;
size_t sharedCount = 0;
};
#endif
|
