diff options
Diffstat (limited to 'src/lua/math/luaopen_math.cpp')
| -rw-r--r-- | src/lua/math/luaopen_math.cpp | 80 |
1 files changed, 80 insertions, 0 deletions
diff --git a/src/lua/math/luaopen_math.cpp b/src/lua/math/luaopen_math.cpp new file mode 100644 index 0000000..ea169e3 --- /dev/null +++ b/src/lua/math/luaopen_math.cpp @@ -0,0 +1,80 @@ +#include "lua/luax.h" +#include "libjin/jin.h" + +namespace jin +{ + namespace lua + { + static int l_and(lua_State* L) + { + int a = luax_checkinteger(L, 1); + int b = luax_checkinteger(L, 2); + luax_pushinteger(L, a & b); + return 1; + } + + static int l_or(lua_State* L) + { + int a = luax_checkinteger(L, 1); + int b = luax_checkinteger(L, 2); + luax_pushinteger(L, a | b); + return 1; + } + + static int l_xor(lua_State* L) + { + int a = luax_checkinteger(L, 1); + int b = luax_checkinteger(L, 2); + luax_pushinteger(L, a ^ b); + return 1; + } + + static int l_not(lua_State* L) + { + int n = luax_checkinteger(L, 1); + luax_pushinteger(L, ~n); + return 1; + } + + static int l_lshift(lua_State* L) + { + int a = luax_checkinteger(L, 1); + int b = luax_checkinteger(L, 2); + luax_pushinteger(L, a << b); + return 1; + } + + static int l_rshift(lua_State* L) + { + int a = luax_checkinteger(L, 1); + int b = luax_checkinteger(L, 2); + luax_pushinteger(L, a >> b); + return 1; + } + + static int l_include(lua_State* L) + { + int a = luax_checkinteger(L, 1); + int b = luax_checkinteger(L, 2); + luax_pushboolean(L, (a & b) == b); + return 1; + } + + static const luaL_Reg f[] = { + { "and", l_and }, + { "or" , l_or }, + { "xor", l_xor }, + { "not", l_not }, + { "lshift", l_lshift }, + { "rshift", l_rshift }, + { "inc", l_include}, + { 0, 0 } + }; + + int luaopen_math(lua_State* L) + { + luax_newlib(L, f); + return 1; + } + } +}
\ No newline at end of file |