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
|
/* This file is part of the Spring engine (GPL v2 or later), see LICENSE.html */
#ifndef LUA_HASH_STRING_H
#define LUA_HASH_STRING_H
#include <string>
#include <cstdint>
#include "LuaInclude.h"
#include "System/StringHash.h"
struct LuaHashString {
public:
LuaHashString(const char* s): hash(lua_calchash(s, slen = strlen(s))) {
assert(slen <= sizeof(str));
memset(str, 0, sizeof(str));
memcpy(str, s, std::min(size_t(slen), sizeof(str) - 1));
}
LuaHashString(const LuaHashString& hs) { *this = hs; }
LuaHashString& operator = (const LuaHashString& hs) {
memcpy(str, hs.str, sizeof(str));
slen = hs.slen;
hash = hs.hash;
return *this;
}
inline uint32_t GetHash() const { return hash; }
inline const char* GetString() const { return str; }
public:
inline void Push(lua_State* L) const {
lua_pushhstring(L, hash, str, slen);
}
inline void GetGlobal(lua_State* L) const {
Push(L);
lua_rawget(L, LUA_GLOBALSINDEX);
}
inline bool GetGlobalFunc(lua_State* L) const {
GetGlobal(L);
if (lua_isfunction(L, -1))
return true;
lua_pop(L, 1);
return false;
}
inline void GetRegistry(lua_State* L) const {
Push(L);
lua_rawget(L, LUA_REGISTRYINDEX);
}
inline bool GetRegistryFunc(lua_State* L) const {
GetRegistry(L);
if (lua_isfunction(L, -1))
return true;
lua_pop(L, 1);
return false;
}
inline void PushBool(lua_State* L, bool value) const {
Push(L);
lua_pushboolean(L, value);
lua_rawset(L, -3);
}
inline void PushNumber(lua_State* L, lua_Number value) const {
Push(L);
lua_pushnumber(L, value);
lua_rawset(L, -3);
}
inline void PushString(lua_State* L, const char* value) const {
Push(L);
lua_pushstring(L, value);
lua_rawset(L, -3);
}
inline void PushString(lua_State* L, const std::string& value) const {
Push(L);
lua_pushsstring(L, value);
lua_rawset(L, -3);
}
inline void PushCFunc(lua_State* L, int (*func)(lua_State*)) const {
Push(L);
lua_pushcfunction(L, func);
lua_rawset(L, -3);
}
private:
char str[32];
uint32_t slen = 0;
uint32_t hash = 0;
};
// NOTE: scoped to avoid name conflicts
// NOTE: since all the following are static, if name can change (e.g. within a loop)
// peculiar things will happen. => Only use raw strings (and not variables) in name.
#define HSTR_PUSH(L, name) \
{ lua_pushhstring(L, COMPILE_TIME_HASH(name), name, sizeof(name) - 1); }
#define HSTR_PUSH_BOOL(L, name, val) \
{ HSTR_PUSH(L, name); lua_pushboolean(L, val); lua_rawset(L, -3); }
#define HSTR_PUSH_NUMBER(L, name, val) \
{ HSTR_PUSH(L, name); lua_pushnumber(L, val); lua_rawset(L, -3); }
#define HSTR_PUSH_STRING(L, name, val) \
{ HSTR_PUSH(L, name); lua_pushsstring(L, val); lua_rawset(L, -3); }
#define HSTR_PUSH_CSTRING(L, name, val) \
{ HSTR_PUSH(L, name); lua_pushhstring(L, COMPILE_TIME_HASH(val), val, sizeof(val) - 1); lua_rawset(L, -3); }
#define HSTR_PUSH_CFUNC(L, name, val) \
{ HSTR_PUSH(L, name); lua_pushcfunction(L, val); lua_rawset(L, -3); }
#endif // LUA_HASH_STRING_H
|
