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
|
#ifndef SYSTEMS_LUA_HPP_
#define SYSTEMS_LUA_HPP_
#include <string>
#include <lua.hpp>
#include <tuple>
#include <vector>
using LuaVariable = std::tuple<std::string, float&>;
using LuaList = std::vector<LuaVariable>;
using LuaRetList = std::vector<double>;
class LuaScript {
private:
lua_State* state;
std::string script;
void setGlobal(const LuaVariable&) const;
void getReturns(LuaRetList& rets) const;
static void replace(std::string& s, const std::string& rid, const std::string& put) {
for (unsigned int i = 0; i < s.size(); i++) {
if (s.substr(i, rid.size()) == rid) {
s.replace(i, rid.size(), put);
i += put.size() - 1;
}
}
}
public:
LuaScript(const std::string& sc = "")
: script(sc) {
state = luaL_newstate();
luaL_openlibs(state);
replace(script, "<", "<");
replace(script, ">", ">");
luaL_loadstring(state, script.c_str());
lua_pcall(state, 0, 0, 0);
}
inline lua_State* getState(void)
{ return state; }
inline void addFunction(const std::string& name, lua_CFunction func) {
lua_pushcclosure(state, func, 0);
lua_setglobal(state, name.c_str());
}
void operator()(const std::string& func = "update") const;
void operator()(const std::string& func, LuaList vars) const;
void operator()(const std::string& func, LuaRetList& rets, LuaList vars) const;
void operator()(LuaList vars) const;
void operator()(LuaRetList& rets, LuaList vars) const;
};
class LuaSystem {
private:
class LuaInterpreter {
private:
lua_State* state;
public:
LuaInterpreter(void) {
state = luaL_newstate();
luaL_openlibs(state);
}
void registerFunc(const std::string& name, lua_CFunction func) {
lua_pushcclosure(state, func, 0);
lua_setglobal(state, name.c_str());
}
void loadFile(const std::string& name) {
luaL_dofile(state, name.c_str());
lua_pcall(state, 0, 0, 0); // 'prime' the file
}
void runFunc(const std::string& name) {
lua_getglobal(state, name.c_str());
lua_pcall(state, 0, 0, 0);
}
};
public:
inline static LuaScript makeScript(const std::string& s) {
return LuaScript(s);
}
};
#endif // SYSTEMS_LUA_HPP_
|
