new function 'ifloor'

1 files changed, 14 insertions, 1 deletions

diff --git a/lmathlib.c b/lmathlib.c
index 26e40b8d..629f7c36 100644
--- a/lmathlib.c
+++ b/lmathlib.c
@@ -1,5 +1,5 @@
 /*
-** $Id: lmathlib.c,v 1.88 2013/06/25 14:02:18 roberto Exp roberto $
+** $Id: lmathlib.c,v 1.89 2013/06/25 19:37:00 roberto Exp roberto $
 ** Standard mathematical library
 ** See Copyright Notice in lua.h
 */
@@ -88,6 +88,18 @@ static int math_floor (lua_State *L) {
   return 1;
 }
 
+static int math_ifloor (lua_State *L) {
+  int valid;
+  lua_Integer n = lua_tointegerx(L, 1, &valid);
+  if (valid)
+    lua_pushinteger(L, n);
+  else {
+    luaL_checktype(L, 1, LUA_TNUMBER);  /* error if not a number */
+    lua_pushnil(L);  /* number with invalid integer value */
+  }
+  return 1;
+}
+
 static int math_fmod (lua_State *L) {
   lua_pushnumber(L, l_mathop(fmod)(luaL_checknumber(L, 1),
                                luaL_checknumber(L, 2)));
@@ -258,6 +270,7 @@ static const luaL_Reg mathlib[] = {
   {"deg",   math_deg},
   {"exp",   math_exp},
   {"floor", math_floor},
+  {"ifloor", math_ifloor},
   {"fmod",   math_fmod},
   {"frexp", math_frexp},
   {"isfloat", math_isfloat},