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
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
|
#include "luax_reftable.h"
#include "luax_state.h"
namespace Luax
{
LuaxRefTable::LuaxRefTable()
: mState(nullptr)
{
}
LuaxRefTable::~LuaxRefTable()
{
}
void LuaxRefTable::Init(LuaxState& state, cc8* name, cc8* mode)
{
assert(!mState);
assert(name);
mName = name;
mMode = 0;
for (int i = 0; mode && mode[i]; ++i)
{
if (mode[i] == 'k') mMode |= WEAK_KEY;
else if (mode[i] == 'v') mMode |= WEAK_VALUE;
}
mState = state.GetHandle();
state.GetField(LUA_REGISTRYINDEX, name); // register[mName]
if (state.IsNil(-1))
{
state.Pop();
lua_newtable(state); // ref table
int ridx = state.AbsIndex(-1);
lua_newtable(state); // metatable of ref table
int idx = state.AbsIndex(-1);
// __mode
if (mode)
{
state.Push(mode);
state.SetField(idx, "__mode");
}
state.Settop(idx);
lua_setmetatable(state, ridx);
state.Settop(ridx);
state.SetField(LUA_REGISTRYINDEX, name);
}
else
{
state.Pop();
}
}
bool LuaxRefTable::IsKeyWeak()
{
assert(mState);
return mMode & WEAK_KEY;
}
bool LuaxRefTable::IsValueWeak()
{
assert(mState);
return mMode & WEAK_VALUE;
}
int LuaxRefTable::Ref(LuaxState& state, int idx)
{
assert(mState);
idx = state.AbsIndex(idx);
state.GetField(LUA_REGISTRYINDEX, mName); // ref table
lua_pushvalue(state, idx); // stuff
int refID = luaL_ref(state, -2);
assert(refID != LUA_NOREF);
state.Pop();
return refID;
}
void LuaxRefTable::Unref(LuaxState& state, int refID)
{
assert(mState);
state.GetField(LUA_REGISTRYINDEX, mName); // ref table
luaL_unref(state, -1, refID);
state.Pop();
return;
}
void LuaxRefTable::PushRefTable(LuaxState& state)
{
assert(mState);
lua_getfield(state, LUA_REGISTRYINDEX, mName);
}
void LuaxRefTable::PushRef(LuaxState& state, int refID)
{
assert(mState);
lua_getfield(state, LUA_REGISTRYINDEX, mName);
lua_rawgeti(state, -1, refID);
lua_replace(state, -2);
}
void LuaxRefTable::Clear(LuaxState& state)
{
assert(mState);
lua_newtable(state);
state.SetField(LUA_REGISTRYINDEX, mName);
}
}
|
