michael/lean2
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

refactor(lua/options): improve options bindings for Lua

Browse source 
Signed-off-by: Leonardo de Moura <leonardo@microsoft.com>
This commit is contained in:
Leonardo de Moura 2013-11-04 18:46:58 -08:00
parent 7b77863507
commit 3c475e890d
2 changed files with 95 additions and 50 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

120
src/bindings/lua/options.cpp
View file

@ -10,56 +10,10 @@ Author: Leonardo de Moura
#include "util/debug.h"
#include "util/name.h"
#include "util/sexpr/options.h"
#include "util/sexpr/option_declarations.h"
#include "bindings/lua/util.h"
namespace lean {
static int mk_option(lua_State * L);
static int options_gc(lua_State * L);
static int options_tostring(lua_State * L);
static int options_contains(lua_State * L);
static int options_size(lua_State * L);
static int options_empty(lua_State * L);
static int options_get_bool(lua_State * L);
static int options_get_int(lua_State * L);
static int options_get_unsigned(lua_State * L);
static int options_get_double(lua_State * L);
static int options_get_string(lua_State * L);
static int options_update_bool(lua_State * L);
static int options_update_int(lua_State * L);
static int options_update_unsigned(lua_State * L);
static int options_update_double(lua_State * L);
static int options_update_string(lua_State * L);
static const struct luaL_Reg options_m[] = {
{"__gc", options_gc}, // never throws
{"__tostring", safe_function<options_tostring>},
{"__len", safe_function<options_size> },
{"contains", safe_function<options_contains>},
{"size", safe_function<options_size>},
{"empty", safe_function<options_empty>},
{"get_bool", safe_function<options_get_bool>},
{"get_int", safe_function<options_get_int>},
{"get_unsigned", safe_function<options_get_unsigned>},
{"get_double", safe_function<options_get_double>},
{"get_string", safe_function<options_get_string>},
{"update_bool", safe_function<options_update_bool>},
{"update_int", safe_function<options_update_int>},
{"update_unsigned", safe_function<options_update_unsigned>},
{"update_double", safe_function<options_update_double>},
{"update_string", safe_function<options_update_string>},
{0, 0}
};
void open_options(lua_State * L) {
luaL_newmetatable(L, "options.mt");
lua_pushvalue(L, -1);
lua_setfield(L, -2, "__index");
setfuncs(L, options_m, 0);
lua_pushcfunction(L, safe_function<mk_option>);
lua_setglobal(L, "options");
}
static int push_options(lua_State * L, options const & o) {
void * mem = lua_newuserdata(L, sizeof(options));
new (mem) options(o);
@ -71,9 +25,6 @@ static int push_options(lua_State * L, options const & o) {
static int mk_option(lua_State * L) {
// int nargs = lua_gettop(L);
options r;
// TODO(Leo): optional table argument
return push_options(L, r);
}
@ -172,5 +123,74 @@ static int options_update_double(lua_State * L) {
static int options_update_string(lua_State * L) {
return push_options(L, to_options(L, 1).update(to_key(L, 2), lua_tostring(L, 3)));
}
static int options_get(lua_State * L) {
name k = to_key(L, 2);
auto it = get_option_declarations().find(k);
if (it == get_option_declarations().end()) {
return luaL_error(L, "unknown option '%s'", k.to_string().c_str());
} else {
option_declaration const & d = it->second;
switch (d.kind()) {
case BoolOption: return options_get_bool(L);
case IntOption: return options_get_int(L);
case UnsignedOption: return options_get_unsigned(L);
case DoubleOption: return options_get_double(L);
case StringOption: return options_get_string(L);
default: return luaL_error(L, "unsupported option kind for '%s'", k.to_string().c_str());
}
}
}
static int options_update(lua_State * L) {
name k = to_key(L, 2);
auto it = get_option_declarations().find(k);
if (it == get_option_declarations().end()) {
return luaL_error(L, "unknown option '%s'", k.to_string().c_str());
} else {
option_declaration const & d = it->second;
switch (d.kind()) {
case BoolOption: return options_update_bool(L);
case IntOption: return options_update_int(L);
case UnsignedOption: return options_update_unsigned(L);
case DoubleOption: return options_update_double(L);
case StringOption: return options_update_string(L);
default: return luaL_error(L, "unsupported option kind for '%s'", k.to_string().c_str());
}
}
}
static const struct luaL_Reg options_m[] = {
{"__gc", options_gc}, // never throws
{"__tostring", safe_function<options_tostring>},
{"__len", safe_function<options_size> },
{"contains", safe_function<options_contains>},
{"size", safe_function<options_size>},
{"empty", safe_function<options_empty>},
{"get", safe_function<options_get>},
{"update", safe_function<options_update>},
// low-level API
{"get_bool", safe_function<options_get_bool>},
{"get_int", safe_function<options_get_int>},
{"get_unsigned", safe_function<options_get_unsigned>},
{"get_double", safe_function<options_get_double>},
{"get_string", safe_function<options_get_string>},
{"update_bool", safe_function<options_update_bool>},
{"update_int", safe_function<options_update_int>},
{"update_unsigned", safe_function<options_update_unsigned>},
{"update_double", safe_function<options_update_double>},
{"update_string", safe_function<options_update_string>},
{0, 0}
};
void open_options(lua_State * L) {
luaL_newmetatable(L, "options.mt");
lua_pushvalue(L, -1);
lua_setfield(L, -2, "__index");
setfuncs(L, options_m, 0);
lua_pushcfunction(L, safe_function<mk_option>);
lua_setglobal(L, "options");
}
}
#endif

25
tests/lua/opt1.lua Normal file
View file

@ -0,0 +1,25 @@
-- Return true if x is an integer
function is_integer(x)
return math.floor(x) == x
end
-- Convert a table into a Lean options object
function to_options(t, prefix, opts)
if opts == nil then opts = options() end
for k, v in pairs(t) do
if type(v) == "table" then
opts = to_options(v, name(prefix, k), opts)
else
opts = opts:update(name(prefix, k), v)
end
end
return opts
end
opts = options()
opts = opts:update(name('pp', 'colors'), false)
opts = opts:update(name('pp', 'colors'), true)
print(opts)
opts = to_options{pp={colors=true, width=10}}
print(opts)