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
121
122
123
124
125
126
|
#include "LuaBindRefTable.h"
#include "LuaBindState.h"
#include "LuaBindVM.h"
namespace LuaBind
{
RefTable::RefTable(LuaBind::VM* vm)
: mVM(vm)
{
}
RefTable::~RefTable()
{
}
void RefTable::Init( cc8* name, cc8* mode)
{
assert(mVM);
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;
}
State state = mVM->GetMainState();
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 RefTable::IsKeyWeak()
{
assert(mVM);
return mMode & WEAK_KEY;
}
bool RefTable::IsValueWeak()
{
assert(mVM);
return mMode & WEAK_VALUE;
}
int RefTable::Ref(State& state, int idx)
{
assert(mVM);
assert(mVM == state.GetVM());
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 RefTable::UnRef(int refID)
{
assert(mVM);
State state = mVM->GetMainState();
state.GetField(LUA_REGISTRYINDEX, mName); // ref table
luaL_unref(state, -1, refID);
state.Pop();
return;
}
void RefTable::PushRefTable(State& state)
{
assert(mVM);
assert(mVM == state.GetVM());
lua_getfield(state, LUA_REGISTRYINDEX, mName);
}
void RefTable::PushRef(State& state, int refID)
{
assert(mVM);
assert(mVM == state.GetVM());
lua_getfield(state, LUA_REGISTRYINDEX, mName);
lua_rawgeti(state, -1, refID);
lua_replace(state, -2);
}
void RefTable::Clear()
{
assert(mVM);
State state = mVM->GetMainState();
lua_newtable(state);
state.SetField(LUA_REGISTRYINDEX, mName);
}
}
|
