| /* |
| ** mathlib.c |
| ** Mathematics library to LUA |
| */ |
| |
| char *rcs_mathlib="$Id: mathlib.c,v 1.8 1995/01/04 18:49:54 roberto Exp $"; |
| |
| #include <stdio.h> /* NULL */ |
| #include <math.h> |
| |
| #include "lualib.h" |
| #include "lua.h" |
| |
| #define PI 3.14159265358979323846 |
| #define TODEGREE(a) ((a)*180.0/PI) |
| #define TORAD(a) ((a)*PI/180.0) |
| |
| static void math_abs (void) |
| { |
| double d; |
| lua_Object o = lua_getparam (1); |
| if (o == LUA_NOOBJECT) |
| lua_error ("too few arguments to function `abs'"); |
| if (!lua_isnumber(o)) |
| lua_error ("incorrect arguments to function `abs'"); |
| d = lua_getnumber(o); |
| if (d < 0) d = -d; |
| lua_pushnumber (d); |
| } |
| |
| |
| static void math_sin (void) |
| { |
| double d; |
| lua_Object o = lua_getparam (1); |
| if (o == LUA_NOOBJECT) |
| lua_error ("too few arguments to function `sin'"); |
| if (!lua_isnumber(o)) |
| lua_error ("incorrect arguments to function `sin'"); |
| d = lua_getnumber(o); |
| lua_pushnumber (sin(TORAD(d))); |
| } |
| |
| |
| |
| static void math_cos (void) |
| { |
| double d; |
| lua_Object o = lua_getparam (1); |
| if (o == LUA_NOOBJECT) |
| lua_error ("too few arguments to function `cos'"); |
| if (!lua_isnumber(o)) |
| lua_error ("incorrect arguments to function `cos'"); |
| d = lua_getnumber(o); |
| lua_pushnumber (cos(TORAD(d))); |
| } |
| |
| |
| |
| static void math_tan (void) |
| { |
| double d; |
| lua_Object o = lua_getparam (1); |
| if (o == LUA_NOOBJECT) |
| lua_error ("too few arguments to function `tan'"); |
| if (!lua_isnumber(o)) |
| lua_error ("incorrect arguments to function `tan'"); |
| d = lua_getnumber(o); |
| lua_pushnumber (tan(TORAD(d))); |
| } |
| |
| |
| static void math_asin (void) |
| { |
| double d; |
| lua_Object o = lua_getparam (1); |
| if (o == LUA_NOOBJECT) |
| lua_error ("too few arguments to function `asin'"); |
| if (!lua_isnumber(o)) |
| lua_error ("incorrect arguments to function `asin'"); |
| d = lua_getnumber(o); |
| lua_pushnumber (TODEGREE(asin(d))); |
| } |
| |
| |
| static void math_acos (void) |
| { |
| double d; |
| lua_Object o = lua_getparam (1); |
| if (o == LUA_NOOBJECT) |
| lua_error ("too few arguments to function `acos'"); |
| if (!lua_isnumber(o)) |
| lua_error ("incorrect arguments to function `acos'"); |
| d = lua_getnumber(o); |
| lua_pushnumber (TODEGREE(acos(d))); |
| } |
| |
| |
| |
| static void math_atan (void) |
| { |
| double d; |
| lua_Object o = lua_getparam (1); |
| if (o == LUA_NOOBJECT) |
| lua_error ("too few arguments to function `atan'"); |
| if (!lua_isnumber(o)) |
| lua_error ("incorrect arguments to function `atan'"); |
| d = lua_getnumber(o); |
| lua_pushnumber (TODEGREE(atan(d))); |
| } |
| |
| |
| static void math_ceil (void) |
| { |
| double d; |
| lua_Object o = lua_getparam (1); |
| if (o == LUA_NOOBJECT) |
| lua_error ("too few arguments to function `ceil'"); |
| if (!lua_isnumber(o)) |
| lua_error ("incorrect arguments to function `ceil'"); |
| d = lua_getnumber(o); |
| lua_pushnumber (ceil(d)); |
| } |
| |
| |
| static void math_floor (void) |
| { |
| double d; |
| lua_Object o = lua_getparam (1); |
| if (o == LUA_NOOBJECT) |
| lua_error ("too few arguments to function `floor'"); |
| if (!lua_isnumber(o)) |
| lua_error ("incorrect arguments to function `floor'"); |
| d = lua_getnumber(o); |
| lua_pushnumber (floor(d)); |
| } |
| |
| static void math_mod (void) |
| { |
| int d1, d2; |
| lua_Object o1 = lua_getparam (1); |
| lua_Object o2 = lua_getparam (2); |
| if (!lua_isnumber(o1) || !lua_isnumber(o2)) |
| lua_error ("incorrect arguments to function `mod'"); |
| d1 = (int) lua_getnumber(o1); |
| d2 = (int) lua_getnumber(o2); |
| lua_pushnumber (d1%d2); |
| } |
| |
| |
| static void math_sqrt (void) |
| { |
| double d; |
| lua_Object o = lua_getparam (1); |
| if (o == LUA_NOOBJECT) |
| lua_error ("too few arguments to function `sqrt'"); |
| if (!lua_isnumber(o)) |
| lua_error ("incorrect arguments to function `sqrt'"); |
| d = lua_getnumber(o); |
| lua_pushnumber (sqrt(d)); |
| } |
| |
| static int old_pow; |
| |
| static void math_pow (void) |
| { |
| lua_Object o1 = lua_getparam (1); |
| lua_Object o2 = lua_getparam (2); |
| lua_Object op = lua_getparam(3); |
| if (!lua_isnumber(o1) || !lua_isnumber(o2) || *(lua_getstring(op)) != 'p') |
| { |
| lua_Object old = lua_getlocked(old_pow); |
| lua_pushobject(o1); |
| lua_pushobject(o2); |
| lua_pushobject(op); |
| if (lua_callfunction(old) != 0) |
| lua_error(NULL); |
| } |
| else |
| { |
| double d1 = lua_getnumber(o1); |
| double d2 = lua_getnumber(o2); |
| lua_pushnumber (pow(d1,d2)); |
| } |
| } |
| |
| static void math_min (void) |
| { |
| int i=1; |
| double d, dmin; |
| lua_Object o; |
| if ((o = lua_getparam(i++)) == LUA_NOOBJECT) |
| lua_error ("too few arguments to function `min'"); |
| if (!lua_isnumber(o)) |
| lua_error ("incorrect arguments to function `min'"); |
| dmin = lua_getnumber (o); |
| while ((o = lua_getparam(i++)) != LUA_NOOBJECT) |
| { |
| if (!lua_isnumber(o)) |
| lua_error ("incorrect arguments to function `min'"); |
| d = lua_getnumber (o); |
| if (d < dmin) dmin = d; |
| } |
| lua_pushnumber (dmin); |
| } |
| |
| |
| static void math_max (void) |
| { |
| int i=1; |
| double d, dmax; |
| lua_Object o; |
| if ((o = lua_getparam(i++)) == LUA_NOOBJECT) |
| lua_error ("too few arguments to function `max'"); |
| if (!lua_isnumber(o)) |
| lua_error ("incorrect arguments to function `max'"); |
| dmax = lua_getnumber (o); |
| while ((o = lua_getparam(i++)) != LUA_NOOBJECT) |
| { |
| if (!lua_isnumber(o)) |
| lua_error ("incorrect arguments to function `max'"); |
| d = lua_getnumber (o); |
| if (d > dmax) dmax = d; |
| } |
| lua_pushnumber (dmax); |
| } |
| |
| |
| static void math_log (void) |
| { |
| double d; |
| lua_Object o = lua_getparam (1); |
| if (o == LUA_NOOBJECT) |
| lua_error ("too few arguments to function `log'"); |
| if (!lua_isnumber(o)) |
| lua_error ("incorrect arguments to function `log'"); |
| d = lua_getnumber(o); |
| lua_pushnumber (log(d)); |
| } |
| |
| |
| static void math_log10 (void) |
| { |
| double d; |
| lua_Object o = lua_getparam (1); |
| if (o == LUA_NOOBJECT) |
| lua_error ("too few arguments to function `log10'"); |
| if (!lua_isnumber(o)) |
| lua_error ("incorrect arguments to function `log10'"); |
| d = lua_getnumber(o); |
| lua_pushnumber (log10(d)); |
| } |
| |
| |
| static void math_exp (void) |
| { |
| double d; |
| lua_Object o = lua_getparam (1); |
| if (o == LUA_NOOBJECT) |
| lua_error ("too few arguments to function `exp'"); |
| if (!lua_isnumber(o)) |
| lua_error ("incorrect arguments to function `exp'"); |
| d = lua_getnumber(o); |
| lua_pushnumber (exp(d)); |
| } |
| |
| static void math_deg (void) |
| { |
| float d; |
| lua_Object o = lua_getparam (1); |
| if (o == LUA_NOOBJECT) |
| lua_error ("too few arguments to function `deg'"); |
| if (!lua_isnumber(o)) |
| lua_error ("incorrect arguments to function `deg'"); |
| d = lua_getnumber(o); |
| lua_pushnumber (d*180./PI); |
| } |
| |
| static void math_rad (void) |
| { |
| float d; |
| lua_Object o = lua_getparam (1); |
| if (o == LUA_NOOBJECT) |
| lua_error ("too few arguments to function `rad'"); |
| if (!lua_isnumber(o)) |
| lua_error ("incorrect arguments to function `rad'"); |
| d = lua_getnumber(o); |
| lua_pushnumber (d/180.*PI); |
| } |
| |
| /* |
| ** Open math library |
| */ |
| void mathlib_open (void) |
| { |
| lua_register ("abs", math_abs); |
| lua_register ("sin", math_sin); |
| lua_register ("cos", math_cos); |
| lua_register ("tan", math_tan); |
| lua_register ("asin", math_asin); |
| lua_register ("acos", math_acos); |
| lua_register ("atan", math_atan); |
| lua_register ("ceil", math_ceil); |
| lua_register ("floor", math_floor); |
| lua_register ("mod", math_mod); |
| lua_register ("sqrt", math_sqrt); |
| lua_register ("min", math_min); |
| lua_register ("max", math_max); |
| lua_register ("log", math_log); |
| lua_register ("log10", math_log10); |
| lua_register ("exp", math_exp); |
| lua_register ("deg", math_deg); |
| lua_register ("rad", math_rad); |
| old_pow = lua_lockobject(lua_setfallback("arith", math_pow)); |
| } |