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

feat(lua/expr): add method for extracting semantic attachment data

Browse source 
Signed-off-by: Leonardo de Moura <leonardo@microsoft.com>
This commit is contained in:
Leonardo de Moura 2013-11-19 19:06:25 -08:00
parent f78cf6a415
commit 7b8bd97699
2 changed files with 25 additions and 2 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

17
src/bindings/lua/expr.cpp
View file

@ -265,7 +265,16 @@ static int expr_fields(lua_State * L) {
case expr_kind::Var: lua_pushinteger(L, var_idx(e)); return 1;
case expr_kind::Constant: return push_name(L, const_name(e));
case expr_kind::Type: return push_level(L, ty_level(e));
case expr_kind::Value: return 0;
case expr_kind::Value:
if (is_nat_value(e)) {
return push_mpz(L, nat_value_numeral(e));
} else if (is_int_value(e)) {
return push_mpz(L, int_value_numeral(e));
} else if (is_real_value(e)) {
return push_mpq(L, real_value_numeral(e));
} else {
return 0;
}
case expr_kind::App: lua_pushinteger(L, num_args(e)); expr_args(L); return 2;
case expr_kind::Eq: push_expr(L, eq_lhs(e)); push_expr(L, eq_rhs(e)); return 2;
case expr_kind::Lambda:
@ -389,6 +398,7 @@ static const struct luaL_Reg expr_m[] = {
{"is_value", safe_function<expr_is_value>},
{"is_metavar", safe_function<expr_is_metavar>},
{"fields", safe_function<expr_fields>},
{"data", safe_function<expr_fields>},
{"args", safe_function<expr_args>},
{"num_args", safe_function<expr_num_args>},
{"arg", safe_function<expr_arg>},
@ -408,7 +418,10 @@ static const struct luaL_Reg expr_m[] = {
};
static int mk_nat_value(lua_State * L) {
return push_expr(L, mk_nat_value(to_mpz_ext(L, 1)));
mpz v = to_mpz_ext(L, 1);
if (v < 0)
throw exception("arg #1 must be non-negative");
return push_expr(L, mk_nat_value(v));
}
static int mk_int_value(lua_State * L) {

10
tests/lua/num2.lua Normal file
View file

@ -0,0 +1,10 @@
assert(nVal(10):data() == mpz(10))
assert(not pcall(function() nVal(-10) end))
assert(iVal(10):data() == mpz(10))
assert(iVal(-10):data() == mpz(-10))
assert(rVal(10):data() == mpq(10))
assert(rVal(10):data() ~= mpz(10))
assert(rVal("10/3"):data() == mpq(10)/3)
assert(rVal("-10/3"):data() == mpq(-10)/3)
assert(Const("a"):data() == name("a"))