13
0
livetrax/tools/luadevel/devel.cc

589 lines
15 KiB
C++
Raw Normal View History

2016-01-30 11:26:14 -05:00
#include <stdint.h>
#include <cstdio>
#include <iostream>
#include <string>
#include <list>
#include <vector>
#define LIBPBD_API
#include "../../libs/pbd/pbd/reallocpool.h"
#include "../../libs/pbd/reallocpool.cc"
#include <readline/readline.h>
#include <readline/history.h>
#include "lua/luastate.h"
#include "LuaBridge/LuaBridge.h"
static void my_lua_print (std::string s) {
std::cout << s << "\n";
}
2016-07-07 09:33:45 -04:00
luabridge::LuaRef::Proxy&
luabridge::LuaRef::Proxy::clone_instance (const void* classkey, void* p) {
lua_rawgeti (m_L, LUA_REGISTRYINDEX, m_tableRef);
lua_rawgeti (m_L, LUA_REGISTRYINDEX, m_keyRef);
luabridge::UserdataPtr::push_raw (m_L, p, classkey);
lua_rawset (m_L, -3);
lua_pop (m_L, 1);
return *this;
}
2016-01-30 11:26:14 -05:00
class A {
public:
2016-07-07 09:33:45 -04:00
A() { printf ("CTOR %p\n", this); _int = 4; for (int i = 0; i < 256; ++i) {arr[i] = i; ar2[i] = i/256.0; ar3[i] = i;} }
2016-01-30 11:26:14 -05:00
~A() { printf ("DTOR\n"); }
void set_int (int a) { _int = a; }
int get_int () const { return _int; }
int& get_ref () { return _int; }
int get_arg (int &a) { printf ("a = %d\n", a); a = _int; printf ("a = %d\n", a); return 1; }
void get_arg2 (int &a, int& b) { a = _int; b = 100; }
void get_args (std::string &a) { a = "hello"; }
void set_ref (int const &a) { _int = a; }
float * get_arr () { return arr; }
float * get_ar2 () { return ar2; }
int * get_ar3 () { return ar3; }
void set_list (std::list<std::string> sl) { _sl = sl; }
std::list<std::string>& get_list () { return _sl; }
uint32_t minone() { return -1; }
2016-07-07 09:33:45 -04:00
void pointer (float*f) const { printf ("PTR %p", f); }
2016-01-30 11:26:14 -05:00
enum EN {
RV1 = 1, RV2, RV3
};
enum EN ret_enum () { return _en;}
void set_enum (enum EN en) { _en = en; }
private:
std::list<std::string> _sl;
int _int;
enum EN _en;
float arr[256];
float ar2[256];
int ar3[256];
};
2016-07-07 09:33:45 -04:00
class LuaTableRef {
public:
LuaTableRef () {}
~LuaTableRef () {}
int get (lua_State* L) {
luabridge::LuaRef rv (luabridge::newTable (L));
for (std::vector<LuaTableEntry>::const_iterator i = _data.begin (); i != _data.end (); ++i) {
switch ((*i).keytype) {
case LUA_TSTRING:
assign(&rv, i->k_s, *i);
break;
case LUA_TNUMBER:
assign(&rv, i->k_n, *i);
break;
}
}
luabridge::push (L, rv);
return 1;
}
int set (lua_State* L) {
if (!lua_istable (L, -1)) { return luaL_error (L, "argument is not a table"); }
_data.clear ();
lua_pushvalue (L, -1);
lua_pushnil (L);
while (lua_next (L, -2)) {
lua_pushvalue (L, -2);
LuaTableEntry s (lua_type(L, -1), lua_type(L, -2));
switch (lua_type(L, -1)) {
case LUA_TSTRING:
s.k_s = luabridge::Stack<std::string>::get (L, -1);
break;
;
case LUA_TNUMBER:
s.k_n = luabridge::Stack<unsigned int>::get (L, -1);
break;
default:
// invalid key
lua_pop (L, 2);
continue;
}
switch(lua_type(L, -2)) {
case LUA_TSTRING:
s.s = luabridge::Stack<std::string>::get (L, -2);
break;
case LUA_TBOOLEAN:
s.b = lua_toboolean (L, -2);
break;
case LUA_TNUMBER:
s.n = lua_tonumber (L, -2);
break;
case LUA_TUSERDATA:
{
bool ok = false;
lua_getmetatable (L, -2);
lua_rawgetp (L, -1, luabridge::getIdentityKey ());
if (lua_isboolean (L, -1)) {
lua_pop (L, 1);
const void* key = lua_topointer (L, -1);
lua_pop (L, 1);
void const* classkey = findclasskey (L, key);
if (classkey) {
ok = true;
s.c = classkey;
s.p = luabridge::Userdata::get_ptr (L, -2);
}
} else {
lua_pop (L, 2);
}
if (ok) {
break;
}
// invalid userdata -- fall through
}
/* fall through */
2016-07-07 09:33:45 -04:00
case LUA_TFUNCTION: // no support -- we could... string.format("%q", string.dump(value, true))
case LUA_TTABLE: // no nested tables, sorry.
case LUA_TNIL: // fallthrough
default:
// invalid value
lua_pop (L, 2);
continue;
}
_data.push_back(s);
lua_pop (L, 2);
}
return 0;
}
static void* findclasskey (lua_State *L, const void* key)
{
lua_pushvalue(L, LUA_REGISTRYINDEX);
lua_pushnil (L);
while (lua_next (L, -2)) {
lua_pushvalue (L, -2);
if (lua_topointer(L, -2) == key) {
void* rv = lua_touserdata (L, -1);
lua_pop (L, 4);
return rv;
}
lua_pop (L, 2);
}
lua_pop (L, 1);
return NULL;
}
private:
struct LuaTableEntry {
LuaTableEntry (int kt, int vt)
: keytype (kt)
, valuetype (vt)
{ }
int keytype;
std::string k_s;
unsigned int k_n;
int valuetype;
// LUA_TUSERDATA
const void* c;
void* p;
// LUA_TBOOLEAN
bool b;
// LUA_TSTRING:
std::string s;
// LUA_TNUMBER:
double n;
};
template<typename T>
static void assign (luabridge::LuaRef* rv, T key, const LuaTableEntry& s)
{
switch (s.valuetype) {
case LUA_TSTRING:
(*rv)[key] = s.s;
break;
case LUA_TBOOLEAN:
(*rv)[key] = s.b;
break;
case LUA_TNUMBER:
(*rv)[key] = s.n;
break;
case LUA_TUSERDATA:
(*rv)[key].clone_instance (s.c, s.p);
break;
default:
assert (0);
break;
}
}
std::vector<LuaTableEntry> _data;
};
#if 0
static void* findclasskey (lua_State *L, const void* key)
{
lua_pushvalue(L, LUA_REGISTRYINDEX);
lua_pushnil (L);
while (lua_next (L, -2)) {
lua_pushvalue (L, -2);
if (lua_topointer(L, -2) == key) {
void* rv = lua_touserdata (L, -1);
lua_pop (L, 4);
return rv;
}
lua_pop (L, 2);
}
lua_pop (L, 1);
return NULL;
}
static int tableSerialize (lua_State *L)
{
if (!lua_istable (L, -1)) { return luaL_error (L, "argument is not a table"); }
luabridge::LuaRef rv (luabridge::newTable (L));
std::cout << "CLASS A KEY: " << luabridge::ClassInfo <A>::getClassKey () << "\n";
lua_rawgetp (L, LUA_REGISTRYINDEX, luabridge::ClassInfo <A>::getClassKey ());
std::cout << " CLASS A TABLE PTR=" << lua_topointer (L, -1) << "\n";
lua_pop (L, 1);
// for k,v in pairs (debug.getregistry ()) do print (k,v) end
lua_pushvalue (L, -1);
lua_pushnil (L);
while (lua_next (L, -2)) {
lua_pushvalue (L, -2);
unsigned int const i = luabridge::Stack<unsigned int>::get (L, -1);
int t = lua_type(L, -2);
switch(t) {
case LUA_TSTRING:
std::cout << " " << i << ": '" << lua_tostring(L, -2) << "'\n";
rv[i] = lua_tostring(L, -2);
break;
case LUA_TBOOLEAN:
std::cout << " " << i << ": " <<
(lua_toboolean(L, -2) ? "true" : "false") << "\n";
rv[i] = lua_toboolean(L, -2);
break;
case LUA_TNUMBER:
std::cout << " " << i << ": " << lua_tonumber(L, -2) << "\n";
rv[i] = lua_tonumber(L, -2);
break;
case LUA_TUSERDATA:
{
lua_getmetatable (L, -2);
lua_rawgetp (L, -1, luabridge::getIdentityKey ());
if (lua_isboolean (L, -1)) {
lua_pop (L, 1);
const void* key = lua_topointer (L, -1);
lua_pop (L, 1);
void const* classkey = findclasskey (L, key);
if (classkey) {
void* p = luabridge::Userdata::get_ptr (L, -2);
rv[i].clone_instance (classkey, p);
}
} else {
lua_pop (L, 2);
}
}
break;
case LUA_TNIL:
case LUA_TTABLE:
case LUA_TFUNCTION:
case LUA_TLIGHTUSERDATA:
default:
std::cout << " " << i << ": TYPE=" << t << ": " << lua_topointer(L, -2)<< "\n";
break;
}
lua_pop (L, 2);
}
lua_pop (L, 1);
lua_pop (L, 2);
luabridge::push (L, rv);
return 1;
}
#endif
LuaTableRef globalref;
int runone (LuaState& lua)
2016-01-30 11:26:14 -05:00
{
#if 0
2016-07-07 09:33:45 -04:00
LuaState lua (*new LuaState);
#elif 0
2016-01-30 11:26:14 -05:00
PBD::ReallocPool _mempool ("Devel", 1048576);
LuaState lua (lua_newstate (&PBD::ReallocPool::lalloc, &_mempool));
#endif
lua.Print.connect (&my_lua_print);
lua_State* L = lua.getState();
#if 1
luabridge::getGlobalNamespace (L)
.beginNamespace ("Test")
.beginStdList <std::string> ("StringList")
.endClass ()
.endNamespace ();
luabridge::getGlobalNamespace (L)
.beginNamespace ("Test")
.beginStdVector <std::string> ("StringVector")
.endClass ()
.endNamespace ();
luabridge::getGlobalNamespace (L)
.beginNamespace ("Test")
.beginStdMap <std::string,std::string> ("StringStringMap")
.endClass ()
.endNamespace ();
luabridge::getGlobalNamespace (L)
.beginNamespace ("Test")
.beginStdSet <std::string> ("StringSet")
.endClass ()
.endNamespace ();
luabridge::getGlobalNamespace (L)
.beginNamespace ("Test")
.registerArray <float> ("FloatArray")
.registerArray <int> ("IntArray")
.beginClass <A> ("A")
.addConstructor <void (*) ()> ()
.addFunction ("set_int", &A::set_int)
.addFunction ("get_int", &A::get_int)
.addRefFunction ("get_arg", &A::get_arg)
.addRefFunction ("get_arg2", &A::get_arg2)
.addRefFunction ("get_args", &A::get_args)
.addFunction ("set_ref", &A::set_ref)
.addFunction ("get_list", &A::get_list)
.addFunction ("set_list", &A::set_list)
.addFunction ("ret_enum", &A::ret_enum)
.addFunction ("set_enum", &A::set_enum)
.addFunction ("get_arr", &A::get_arr)
.addFunction ("get_ar2", &A::get_ar2)
.addFunction ("get_ar3", &A::get_ar3)
.endClass ()
.endNamespace ();
luabridge::getGlobalNamespace (L)
.beginNamespace ("Test")
.beginClass <A> ("A")
2016-07-07 09:33:45 -04:00
.addFunction ("pointer", &A::pointer)
2016-01-30 11:26:14 -05:00
.addFunction ("minone", &A::minone)
.addConst ("cologne", 4711)
.endClass ()
.addConst ("koln", 4711)
.endNamespace ();
#endif
2016-07-07 09:33:45 -04:00
luabridge::getGlobalNamespace (L)
.beginNamespace ("Dump")
.beginClass <LuaTableRef> ("TableRef")
.addCFunction ("get", &LuaTableRef::get)
.addCFunction ("set", &LuaTableRef::set)
.endClass ()
//.addCFunction ("dump", tableSerialize)
.endNamespace ();
luabridge::push <LuaTableRef *> (L, &globalref);
lua_setglobal (L, "ref");
2016-01-30 11:26:14 -05:00
#if 0 // session script test
lua.do_command (
"function ArdourSession ()"
" local self = { scripts = {}, instances = {} }"
""
" local foreach = function (fn)"
" for n, s in pairs (self.scripts) do"
" fn (n, s)"
" end"
" end"
""
" local run = function ()"
" for n, s in pairs (self.instances) do"
" local status, err = pcall (s)"
" if not status then print ('fn \"'.. n .. '\": ', err) end"
" end"
" collectgarbage()"
" end"
""
" local add = function (n, f, a)"
" assert(type(n) == 'string', 'function-name must be string')"
" assert(type(f) == 'function', 'Given script is a not a function')"
" assert(type(a) == 'table' or type(a) == 'nil', 'Given argument is invalid')"
" assert(self.scripts[n] == nil, 'Callback \"'.. n ..'\" already exists.')"
" self.scripts[n] = { ['f'] = f, ['a'] = a }"
" local env = { print = print, Session = Session, tostring = tostring, assert = assert, ipairs = ipairs, error = error, string = string, type = type, tonumber = tonumber, collectgarbage = collectgarbage, pairs = pairs, math = math, table = table, pcall = pcall }"
" self.instances[n] = load (string.dump(f), nil, nil, env)(a)"
" end"
""
" local remove = function (n)"
" self.scripts[n] = nil"
" end"
""
" local list = function ()"
" local rv = {}"
" foreach (function (n) rv[n] = true end)"
" return rv"
" end"
""
" local function basic_serialize (o)"
" if type(o) == \"number\" then"
" return tostring(o)"
" else"
" return string.format(\"%q\", o)"
" end"
" end"
""
" local function serialize (name, value)"
" local rv = name .. ' = '"
" if type(value) == \"number\" or type(value) == \"string\" or type(value) == \"nil\" then"
" return rv .. basic_serialize(value) .. ' '"
" elseif type(value) == \"table\" then"
" rv = rv .. '{} '"
" for k,v in pairs(value) do"
" local fieldname = string.format(\"%s[%s]\", name, basic_serialize(k))"
" rv = rv .. serialize(fieldname, v) .. ' '"
" end"
" return rv;"
" elseif type(value) == \"function\" then"
" return rv .. string.format(\"%q\", string.dump(value))"
" else"
" error('cannot save a ' .. type(value))"
" end"
" end"
""
""
" local save = function ()"
" return (serialize('scripts', self.scripts))"
" end"
""
" local restore = function (state)"
" self.scripts = {}"
" load (state)()"
" print (scripts)"
" for n, s in pairs (scripts) do"
" add (n, load(s['f']), s['a'])"
" end"
" end"
""
" return { run = run, add = add, remove = remove,"
" list = list, foreach = foreach,"
" restore = restore, save = save}"
" end"
" "
" sess = ArdourSession ()"
" ArdourSession = nil"
);
luabridge::LuaRef *lua_run;
luabridge::LuaRef *lua_add;
luabridge::LuaRef *lua_del;
luabridge::LuaRef *lua_save;
luabridge::LuaRef *lua_load;
{
luabridge::LuaRef lua_sess = luabridge::getGlobal (L, "sess");
lua.do_command ("sess = nil"); // hide it.
lua.do_command ("collectgarbage()");
lua_run = new luabridge::LuaRef(lua_sess["run"]);
lua_add = new luabridge::LuaRef(lua_sess["add"]);
lua_del = new luabridge::LuaRef(lua_sess["remove"]);
lua_save = new luabridge::LuaRef(lua_sess["save"]);
lua_load = new luabridge::LuaRef(lua_sess["restore"]);
}
lua.do_command ("collectgarbage()");
#if 1
lua.do_command ("function factory (t) return function () local p = t or { } local a = t[1] or 'Nibor' print ('Hello ' .. a) end end");
luabridge::LuaRef lua_fact = luabridge::getGlobal (L, "factory");
luabridge::LuaRef tbl_arg (luabridge::newTable(L));
//tbl_arg[1] = "Robin";
(*lua_add)("t2", lua_fact, tbl_arg);
#else
lua.do_command ("function factory (t) return function () print ('Boo') end end");
luabridge::LuaRef lua_fact = luabridge::getGlobal (L, "factory");
(*lua_add)("t2", lua_fact());
#endif
lua.do_command ("function factory (t) return function () print ('Ahoy') end end");
luabridge::LuaRef lua_fact2 = luabridge::getGlobal (L, "factory");
(*lua_add)("t1", lua_fact2);
luabridge::LuaRef savedstate ((*lua_save)());
std::string saved = savedstate.cast<std::string>();
(*lua_del)("t2");
try {
(*lua_run)();
} catch (luabridge::LuaException const& e) { printf ("LuaException: %s\n", e.what ()); }
(*lua_load)(saved);
for (int i = 0; i < 2; ++i) {
lua.do_command ("collectgarbage()");
lua.collect_garbage ();
try {
(*lua_run)();
} catch (luabridge::LuaException const& e) { printf ("LuaException: %s\n", e.what ()); }
}
#endif
add_history("a = Test:A() b = 2 c = 3 d = 'a'");
add_history("x = a:get_arg(b) y = a:get_arg2(b, c) z = a:get_args(d) ");
add_history("for i,n in ipairs(y) do print (i, n); end");
2016-07-07 09:33:45 -04:00
add_history("t = {} t[2] = 7; t[3] = Test:A() t[4] = Test:A() ref:set (t); f = ref:get()");
2016-01-30 11:26:14 -05:00
/////////////////////////////////////////////////////////////////////////////
char *line;
while ((line = readline ("> "))) {
if (!strcmp (line, "quit")) {
break;
}
if (strlen(line) == 0) {
//lua.do_command("collectgarbage();");
continue;
}
if (!lua.do_command (line)) {
add_history(line); // OK
} else {
add_history(line); // :)
}
}
printf("\n");
return 0;
}
2016-07-07 09:33:45 -04:00
int main (int argc, char **argv)
{
LuaState lua1;
LuaState lua2;
runone (lua1);
printf ("=====\n");
runone (lua2);
}