*update

1 files changed, 84 insertions, 0 deletions

diff --git a/src/lua/modules/bit/luaopen_bit.cpp b/src/lua/modules/bit/luaopen_bit.cpp
new file mode 100644
index 0000000..5b18125
--- /dev/null
+++ b/src/lua/modules/bit/luaopen_bit.cpp
@@ -0,0 +1,84 @@
+#include "lua/modules/luax.h"
+#include "libjin/jin.h"
+#include <cstdlib>
+
+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[] = {
+        { "bAnd", l_and     },
+        { "bOr" , l_or      },
+        { "bXor", l_xor     },
+        { "bNot", l_not     },
+        { "bLs",  l_lshift  },
+        { "bRs",  l_rshift  },
+        { "bInc", l_include },
+        { 0,      0         }
+    };
+
+    int luaopen_bit(lua_State* L)
+    {
+        luax_newlib(L, f);
+
+        return 1;
+    }
+
+}
+}
\ No newline at end of file