117 lines
3.4 KiB
C++
117 lines
3.4 KiB
C++
/*
|
|
Copyright (c) 2013 Microsoft Corporation. All rights reserved.
|
|
Released under Apache 2.0 license as described in the file LICENSE.
|
|
|
|
Author: Leonardo de Moura
|
|
*/
|
|
#include <sstream>
|
|
#include <lua.hpp>
|
|
#include <cstring>
|
|
#include "util/debug.h"
|
|
#include "util/sstream.h"
|
|
#include "util/sexpr/options.h"
|
|
#include "util/sexpr/format.h"
|
|
#include "bindings/lua/util.h"
|
|
#include "bindings/lua/name.h"
|
|
#include "bindings/lua/numerics.h"
|
|
#include "bindings/lua/options.h"
|
|
|
|
namespace lean {
|
|
DECL_UDATA(format)
|
|
|
|
format to_format_elem(lua_State * L, int idx) {
|
|
if (is_format(L, idx))
|
|
return to_format(L, idx);
|
|
else if (lua_isnumber(L, idx))
|
|
return format(static_cast<int>(lua_tonumber(L, idx)));
|
|
else if (is_name(L, idx))
|
|
return format(to_name(L, idx));
|
|
else if (is_mpz(L, idx))
|
|
return format(to_mpz(L, idx));
|
|
else if (is_mpq(L, idx))
|
|
return format(to_mpq(L, idx));
|
|
else
|
|
return format(lua_tostring(L, idx));
|
|
}
|
|
|
|
static int format_tostring(lua_State * L) {
|
|
std::ostringstream out;
|
|
out << mk_pair(to_format(L, 1), get_global_options(L));
|
|
lua_pushstring(L, out.str().c_str());
|
|
return 1;
|
|
}
|
|
|
|
static int format_concat(lua_State * L) {
|
|
return push_format(L, compose(to_format_elem(L, 1), to_format_elem(L, 2)));
|
|
}
|
|
|
|
static int mk_format(lua_State * L) {
|
|
format r;
|
|
int nargs = lua_gettop(L);
|
|
if (nargs == 0) {
|
|
r = format();
|
|
} else {
|
|
int i = nargs;
|
|
r = to_format_elem(L, i);
|
|
i--;
|
|
for (; i >= 1; i--) {
|
|
r = compose(to_format_elem(L, i), r);
|
|
}
|
|
}
|
|
return push_format(L, r);
|
|
}
|
|
|
|
static int format_nest(lua_State * L) {
|
|
return push_format(L, nest(luaL_checkinteger(L, 2), to_format(L, 1)));
|
|
}
|
|
|
|
static int format_group(lua_State * L) {
|
|
return push_format(L, group(to_format(L, 1)));
|
|
}
|
|
|
|
static int format_highlight(lua_State * L) {
|
|
char const * color_str = luaL_checkstring(L, 2);
|
|
format::format_color color;
|
|
if (strcmp(color_str, "red") == 0) {
|
|
color = format::RED;
|
|
} else if (strcmp(color_str, "green") == 0) {
|
|
color = format::GREEN;
|
|
} else if (strcmp(color_str, "orange") == 0) {
|
|
color = format::ORANGE;
|
|
} else if (strcmp(color_str, "blue") == 0) {
|
|
color = format::BLUE;
|
|
} else if (strcmp(color_str, "cyan") == 0) {
|
|
color = format::CYAN;
|
|
} else if (strcmp(color_str, "grey") == 0) {
|
|
color = format::GREY;
|
|
} else {
|
|
throw exception(sstream() << "unknown color '" << color_str << "'");
|
|
}
|
|
return push_format(L, highlight(to_format(L, 1), color));
|
|
}
|
|
|
|
static int format_line(lua_State * L) { return push_format(L, line()); }
|
|
static int format_space(lua_State * L) { return push_format(L, space()); }
|
|
|
|
static const struct luaL_Reg format_m[] = {
|
|
{"__gc", format_gc}, // never throws
|
|
{"__tostring", safe_function<format_tostring>},
|
|
{"__concat", safe_function<format_concat>},
|
|
{"nest", safe_function<format_nest>},
|
|
{"group", safe_function<format_group>},
|
|
{"highlight", safe_function<format_highlight>},
|
|
{0, 0}
|
|
};
|
|
|
|
void open_format(lua_State * L) {
|
|
luaL_newmetatable(L, format_mt);
|
|
lua_pushvalue(L, -1);
|
|
lua_setfield(L, -2, "__index");
|
|
setfuncs(L, format_m, 0);
|
|
|
|
SET_GLOBAL_FUN(mk_format, "format");
|
|
SET_GLOBAL_FUN(format_line, "line");
|
|
SET_GLOBAL_FUN(format_space, "space");
|
|
SET_GLOBAL_FUN(format_pred, "is_format");
|
|
}
|
|
}
|