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

feat(library/kernel_bindings): add load method to Lua API

Browse source 
Signed-off-by: Leonardo de Moura <leonardo@microsoft.com>
This commit is contained in:
Leonardo de Moura 2014-01-07 14:21:24 -08:00
parent 6077dc61b7
commit d5ddb186d9
1 changed files with 18 additions and 4 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

22
src/library/kernel_bindings.cpp
View file

@ -1141,24 +1141,37 @@ static int environment_is_opaque(lua_State * L) {
return 1; return 1;
} }
static int environment_import(lua_State * L) { template<typename F>
static int environment_import_core(lua_State * L, F && import) {
rw_shared_environment env(L, 1); rw_shared_environment env(L, 1);
int nargs = lua_gettop(L); int nargs = lua_gettop(L);
if (nargs == 3) { if (nargs == 3) {
env->import(luaL_checkstring(L, 2), to_io_state(L, 3)); import(env, luaL_checkstring(L, 2), to_io_state(L, 3));
} else { } else {
io_state * ios = get_io_state(L); io_state * ios = get_io_state(L);
if (ios) { if (ios) {
env->import(luaL_checkstring(L, 2), *ios); import(env, luaL_checkstring(L, 2), *ios);
} else { } else {
io_state ios(mk_simple_formatter()); io_state ios(mk_simple_formatter());
ios.set_options(get_global_options(L)); ios.set_options(get_global_options(L));
env->import(luaL_checkstring(L, 2), ios); import(env, luaL_checkstring(L, 2), ios);
} }
} }
return 0; return 0;
} }
static int environment_import(lua_State * L) {
return environment_import_core(L, [](rw_shared_environment & env, char const * fname, io_state const & ios) {
return env->import(fname, ios);
});
}
static int environment_load(lua_State * L) {
return environment_import_core(L, [](rw_shared_environment & env, char const * fname, io_state const & ios) {
return env->load(fname, ios);
});
}
static int environment_get_universe_distance(lua_State * L) { static int environment_get_universe_distance(lua_State * L) {
ro_shared_environment env(L, 1); ro_shared_environment env(L, 1);
auto r = env->get_universe_distance(to_name_ext(L, 2), to_name_ext(L, 3)); auto r = env->get_universe_distance(to_name_ext(L, 2), to_name_ext(L, 3));
@ -1194,6 +1207,7 @@ static const struct luaL_Reg environment_m[] = {
{"set_opaque", safe_function<environment_set_opaque>}, {"set_opaque", safe_function<environment_set_opaque>},
{"is_opaque", safe_function<environment_is_opaque>}, {"is_opaque", safe_function<environment_is_opaque>},
{"import", safe_function<environment_import>}, {"import", safe_function<environment_import>},
{"load", safe_function<environment_load>},
{0, 0} {0, 0}
}; };