lua/mathlib.c

229 lines
4.3 KiB
C
Raw Normal View History

1993-07-28 17:18:00 +04:00
/*
** mathlib.c
1993-12-17 21:41:19 +03:00
** Mathematics library to LUA
1993-07-28 17:18:00 +04:00
*/
char *rcs_mathlib="$Id: mathlib.c,v 1.11 1995/10/04 13:52:09 roberto Exp $";
1993-12-17 21:41:19 +03:00
1993-07-28 17:18:00 +04:00
#include <stdio.h> /* NULL */
#include <math.h>
1994-07-21 02:12:27 +04:00
#include "lualib.h"
1993-07-28 17:18:00 +04:00
#include "lua.h"
1995-10-02 20:03:33 +03:00
#ifndef PI
1995-02-06 22:36:43 +03:00
#define PI 3.14159265358979323846
1995-10-02 20:03:33 +03:00
#endif
1994-07-21 02:12:27 +04:00
#define TODEGREE(a) ((a)*180.0/PI)
#define TORAD(a) ((a)*PI/180.0)
1993-12-17 21:41:19 +03:00
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);
}
1993-07-28 17:18:00 +04:00
static void math_abs (void)
{
double d = get_and_check_float(1, "abs");
1993-07-28 17:18:00 +04:00
if (d < 0) d = -d;
lua_pushnumber (d);
}
static void math_sin (void)
{
double d = get_and_check_float(1, "sin");
1993-12-17 21:41:19 +03:00
lua_pushnumber (sin(TORAD(d)));
1993-07-28 17:18:00 +04:00
}
static void math_cos (void)
{
double d = get_and_check_float(1, "cos");
1993-12-17 21:41:19 +03:00
lua_pushnumber (cos(TORAD(d)));
1993-07-28 17:18:00 +04:00
}
static void math_tan (void)
{
double d = get_and_check_float(1, "tan");
1993-12-17 21:41:19 +03:00
lua_pushnumber (tan(TORAD(d)));
1993-07-28 17:18:00 +04:00
}
static void math_asin (void)
{
double d = get_and_check_float(1, "asin");
1993-12-17 21:41:19 +03:00
lua_pushnumber (TODEGREE(asin(d)));
1993-07-28 17:18:00 +04:00
}
static void math_acos (void)
{
double d = get_and_check_float(1, "acos");
1993-12-17 21:41:19 +03:00
lua_pushnumber (TODEGREE(acos(d)));
1993-07-28 17:18:00 +04:00
}
static void math_atan (void)
{
double d = get_and_check_float(1, "atan");
1993-12-17 21:41:19 +03:00
lua_pushnumber (TODEGREE(atan(d)));
1993-07-28 17:18:00 +04:00
}
1995-10-04 16:52:09 +03:00
static void math_atan2 (void)
{
double d1 = get_and_check_float(1, "atan2");
double d2 = get_and_check_float(2, "atan2");
1995-10-04 16:52:09 +03:00
lua_pushnumber (TODEGREE(atan2(d1, d2)));
}
1993-07-28 17:18:00 +04:00
static void math_ceil (void)
{
double d = get_and_check_float(1, "ceil");
1993-07-28 17:18:00 +04:00
lua_pushnumber (ceil(d));
}
static void math_floor (void)
{
double d = get_and_check_float(1, "floor");
1993-07-28 17:18:00 +04:00
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");
1993-07-28 17:18:00 +04:00
lua_pushnumber (d1%d2);
}
static void math_sqrt (void)
{
double d = get_and_check_float(1, "sqrt");
1993-07-28 17:18:00 +04:00
lua_pushnumber (sqrt(d));
}
1994-11-17 22:43:34 +03:00
static int old_pow;
1993-07-28 17:18:00 +04:00
static void math_pow (void)
{
lua_Object o1 = lua_getparam (1);
lua_Object o2 = lua_getparam (2);
1994-11-17 22:43:34 +03:00
lua_Object op = lua_getparam(3);
if (!lua_isnumber(o1) || !lua_isnumber(o2) || *(lua_getstring(op)) != 'p')
{
lua_Object old = lua_getlocked(old_pow);
1994-11-17 22:43:34 +03:00
lua_pushobject(o1);
lua_pushobject(o2);
lua_pushobject(op);
if (lua_callfunction(old) != 0)
1994-11-17 22:43:34 +03:00
lua_error(NULL);
}
else
{
double d1 = lua_getnumber(o1);
double d2 = lua_getnumber(o2);
lua_pushnumber (pow(d1,d2));
}
1993-07-28 17:18:00 +04:00
}
static void math_min (void)
{
int i=1;
double dmin = get_and_check_float(i, "min");
while (lua_getparam(++i) != LUA_NOOBJECT)
1993-07-28 17:18:00 +04:00
{
double d = get_and_check_float(i, "min");
1993-07-28 17:18:00 +04:00
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)
1993-07-28 17:18:00 +04:00
{
double d = get_and_check_float(i, "max");
1993-07-28 17:18:00 +04:00
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));
}
1993-07-28 17:18:00 +04:00
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);
}
1993-07-28 17:18:00 +04:00
/*
** 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);
1995-10-04 16:52:09 +03:00
lua_register ("atan2", math_atan2);
1993-07-28 17:18:00 +04:00
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));
1993-07-28 17:18:00 +04:00
}