math.tointeger, math.floor, math.ceil

2 files changed, 52 insertions, 8 deletions

diff --git a/src/lmathlib.js b/src/lmathlib.js
index d1daca5..9921cc9 100644
--- a/src/lmathlib.js
+++ b/src/lmathlib.js
@@ -52,14 +52,54 @@ const math_atan = function(L) {
     return 1;
 };
 
+const math_toint = function(L) {
+    let n = lapi.lua_tointegerx(L, 1)
+    if (n !== false)
+        lapi.lua_pushinteger(L, n);
+    else {
+        lauxlib.luaL_checkany(L, 1);
+        lapi.lua_pushnil(L);  /* value is not convertible to integer */
+    }
+    return 1;
+};
+
+const pushnumint = function(L, d) {
+    let n = luaconf.lua_numbertointeger(d);
+    if (n !== false)  /* does 'd' fit in an integer? */
+        lapi.lua_pushinteger(L, n);  /* result is integer */
+    else
+        lapi.lua_pushnumber(L, d);  /* result is float */
+};
+
+const math_floor = function(L) {
+    if (lapi.lua_isinteger(L, 1))
+        lapi.lua_settop(L, 1);
+    else
+        pushnumint(L, lauxlib.luaL_checknumber(L, 1));
+
+    return 1;
+};
+
+const math_ceil = function(L) {
+    if (lapi.lua_isinteger(L, 1))
+        lapi.lua_settop(L, 1);
+    else
+        pushnumint(L, Math.ceil(lauxlib.luaL_checknumber(L, 1)));
+
+    return 1;
+};
+
 const mathlib = {
-    "abs":  math_abs,
-    "sin":  math_sin,
-    "cos":  math_cos,
-    "tan":  math_tan,
-    "asin": math_asin,
-    "acos": math_acos,
-    "atan": math_atan
+    "abs":       math_abs,
+    "acos":      math_acos,
+    "asin":      math_asin,
+    "atan":      math_atan,
+    "ceil":      math_ceil,
+    "cos":       math_cos,
+    "floor":     math_floor,
+    "sin":       math_sin,
+    "tan":       math_tan,
+    "tointeger": math_toint
 };
 
 const luaopen_math = function(L) {
diff --git a/src/luaconf.js b/src/luaconf.js
index c415a65..7297b28 100644
--- a/src/luaconf.js
+++ b/src/luaconf.js
@@ -14,7 +14,11 @@ const LUAI_MAXSTACK = 1000000;
 @@ of a function in debug information.
 ** CHANGE it if you want a different size.
 */
-const LUA_IDSIZE = 60
+const LUA_IDSIZE = 60;
+
+const lua_numbertointeger = function(n) {
+    return n|0;
+};
 
 module.exports.LUAI_MAXSTACK = LUAI_MAXSTACK;
 module.exports.LUA_IDSIZE    = LUA_IDSIZE;
\ No newline at end of file