lua/mathlib.c

/*
** mathlib.c
** Mathematics library to LUA
*/

char *rcs_mathlib="$Id: mathlib.c,v 1.11 1995/10/04 13:52:09 roberto Exp $";

#include <stdio.h>		/* NULL */
#include <math.h>

#include "lualib.h"
#include "lua.h"

#ifndef PI
#define PI          3.14159265358979323846
#endif
#define TODEGREE(a) ((a)*180.0/PI)
#define TORAD(a)    ((a)*PI/180.0)


static void str_error(char *funcname)
{
  char buff[250];
  sprintf(buff, "incorrect arguments to function `%s'", funcname);
  lua_error(buff);
}

static float get_and_check_float (int numArg, char *funcname)
{
  lua_Object o = lua_getparam(numArg);
  if (!lua_isnumber(o))
    str_error(funcname);
  return lua_getnumber(o);
}

static void math_abs (void)
{
 double d = get_and_check_float(1, "abs"); 
 if (d < 0) d = -d;
 lua_pushnumber (d);
}


static void math_sin (void)
{
 double d = get_and_check_float(1, "sin");
 lua_pushnumber (sin(TORAD(d)));
}



static void math_cos (void)
{
 double d = get_and_check_float(1, "cos");
 lua_pushnumber (cos(TORAD(d)));
}



static void math_tan (void)
{
 double d = get_and_check_float(1, "tan");
 lua_pushnumber (tan(TORAD(d)));
}


static void math_asin (void)
{
 double d = get_and_check_float(1, "asin");
 lua_pushnumber (TODEGREE(asin(d)));
}


static void math_acos (void)
{
 double d = get_and_check_float(1, "acos");
 lua_pushnumber (TODEGREE(acos(d)));
}


static void math_atan (void)
{
 double d = get_and_check_float(1, "atan");
 lua_pushnumber (TODEGREE(atan(d)));
}


static void math_atan2 (void)
{
 double d1 = get_and_check_float(1, "atan2");
 double d2 = get_and_check_float(2, "atan2");
 lua_pushnumber (TODEGREE(atan2(d1, d2)));
}


static void math_ceil (void)
{
 double d = get_and_check_float(1, "ceil");
 lua_pushnumber (ceil(d));
}


static void math_floor (void)
{
 double d = get_and_check_float(1, "floor");
 lua_pushnumber (floor(d));
}

static void math_mod (void)
{
 int d1 = (int)get_and_check_float(1, "mod");
 int d2 = (int)get_and_check_float(2, "mod");
 lua_pushnumber (d1%d2);
}


static void math_sqrt (void)
{
 double d = get_and_check_float(1, "sqrt");
 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 dmin = get_and_check_float(i, "min");
 while (lua_getparam(++i) != LUA_NOOBJECT)
 {
  double d = get_and_check_float(i, "min");
  if (d < dmin) dmin = d;
 }
 lua_pushnumber (dmin);
}

static void math_max (void)
{
 int i=1;
 double dmax = get_and_check_float(i, "max");
 while (lua_getparam(++i) != LUA_NOOBJECT)
 {
  double d = get_and_check_float(i, "max");
  if (d > dmax) dmax = d;
 }
 lua_pushnumber (dmax);
}

static void math_log (void)
{
 double d = get_and_check_float(1, "log");
 lua_pushnumber (log(d));
}


static void math_log10 (void)
{
 double d = get_and_check_float(1, "log10");
 lua_pushnumber (log10(d));
}


static void math_exp (void)
{
 double d = get_and_check_float(1, "exp");
 lua_pushnumber (exp(d));
}

static void math_deg (void)
{
 float d = get_and_check_float(1, "deg");
 lua_pushnumber (d*180./PI);
}

static void math_rad (void)
{
 float d = get_and_check_float(1, "rad");
 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 ("atan2",  math_atan2);
 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));
}