rpgxef/code/game/lua_qmath.c

// qmath lib for lua

#include "g_lua.h"

#ifdef G_LUA
/***
Provides access to many mathematical functions.
@module qmath
*/

/***
Get the integer part of a number.
@function abs
@param number Number
@return Integer part of number.
*/
static int Qmath_Abs(lua_State * L)
{
	lua_pushnumber(L, fabs(luaL_checknumber(L, 1)));
	return 1;
}

/***
Get the sine for a number (degree).
@function sin
@param number Number
@return Sine of number.
*/
static int Qmath_Sin(lua_State * L)
{
	lua_pushnumber(L, sin(DEG2RAD(luaL_checknumber(L, 1))));
	return 1;
}

/***
Get the cosine for a number (degree).
@function cos
@param number Number
@return Consine of number.
*/
static int Qmath_Cos(lua_State * L)
{
	lua_pushnumber(L, cos(DEG2RAD(luaL_checknumber(L, 1))));
	return 1;
}

/***
Get the tangent for a number.
@function tan
@param number Number
@return Tangent of number.
*/
static int Qmath_Tan(lua_State * L)
{
	lua_pushnumber(L, tan(DEG2RAD(luaL_checknumber(L, 1))));
	return 1;
}

/***
Get the arcsine for a number.
@function asin
@param number Number
@return arcsine for number.
*/
static int Qmath_Asin(lua_State * L)
{
	lua_pushnumber(L, RAD2DEG(asin(luaL_checknumber(L, 1))));
	return 1;
}

/***
Get the arccosine for a number.
@function acos
@param number Number
@return arccosine of number.
*/
static int Qmath_Acos(lua_State * L)
{
	lua_pushnumber(L, RAD2DEG(acos(luaL_checknumber(L, 1))));
	return 1;
}

/***
Get the arctangent for a number.
@function atan
@param number Number
@return arctangent of number.
*/
static int Qmath_Atan(lua_State * L)
{
	lua_pushnumber(L, RAD2DEG(atan(luaL_checknumber(L, 1))));
	return 1;
}

/***
Get the arctangent for a number. Alternate to atan.
@function atan2
@param number Number
@return arctangent of number.
*/
static int Qmath_Atan2(lua_State * L)
{
	lua_pushnumber(L, RAD2DEG(atan2(luaL_checknumber(L, 1), luaL_checknumber(L, 2))));
	return 1;
}

/***
Get the ceiled value of a number.
@function ceil
@param number Number
@return Ceiled value of number.
*/
static int Qmath_Ceil(lua_State * L)
{
	lua_pushnumber(L, ceil(luaL_checknumber(L, 1)));
	return 1;
}

/***
Get the floored value of a number.
@function floor
@param numerb Number
@return Floored value of number.
*/
static int Qmath_Floor(lua_State * L)
{
	lua_pushnumber(L, floor(luaL_checknumber(L, 1)));
	return 1;
}

/***
Get the remainder of numerator/denominator.
@functiom fmod
@param n numerator
@param f denominator
@return Remainder of numerator/denominator.
*/
static int Qmath_Fmod(lua_State * L)
{
	lua_pushnumber(L, fmod(luaL_checknumber(L, 1), luaL_checknumber(L, 2)));
	return 1;
}

/***
BUGGED
Breaks a given number apart into its integer part and its fractional part. The fractional part is returned while the integer part is assigned to number.
@function modf
@param number Number
@return Array containing integer and fractional part.
*/
/* TODO return an array the current way won't work */
static int Qmath_Modf(lua_State * L)
{
	double          ip;
	double          fp = modf(luaL_checknumber(L, 1), &ip);

	lua_pushnumber(L, ip);
	lua_pushnumber(L, fp);
	return 2;
}

/***
Get the square root of a number.
@function sqrt
@param number Number
@return Square root of number.
*/
static int Qmath_Sqrt(lua_State * L)
{
	lua_pushnumber(L, sqrt(luaL_checknumber(L, 1)));
	return 1;
}

// What's this?? f^n e.g. 2^4 = 2 * 2 * 2 = 8
/***
Returns f raised to the power of n.
@function pow
@param f Number
@param n Number
@return f raised to the power of n.
*/
static int Qmath_Pow(lua_State * L)
{
	lua_pushnumber(L, pow(luaL_checknumber(L, 1), luaL_checknumber(L, 2)));
	return 1;
}

/***
Get the logarithm of a number.
@function log
@param number Number
@return Logarithm of number.
*/
static int Qmath_Log(lua_State * L)
{
	lua_pushnumber(L, log(luaL_checknumber(L, 1)));
	return 1;
}

/***
Get logarithm to the base of 10 of number.
@function log10
@param number Number
@return Logarithm to the base of 10 of number.
*/
static int Qmath_Log10(lua_State * L)
{
	lua_pushnumber(L, log10(luaL_checknumber(L, 1)));
	return 1;
}

/***
Get the base-e exponential function of a given number.
@function exp
@param number Number
@return Base-e exponential of number.
*/
static int Qmath_Exp(lua_State * L)
{
	lua_pushnumber(L, exp(luaL_checknumber(L, 1)));
	return 1;
}

/***
Convert radian to degree.
@function deg
@param number Number
@return number converted to degree.
*/
static int Qmath_Deg(lua_State * L)
{
	lua_pushnumber(L, RAD2DEG(luaL_checknumber(L, 1)));
	return 1;
}

/***
Convert degree to radian.
@function rad
@param number Number
@return number converted to radian.
*/
static int Qmath_Rad(lua_State * L)
{
	lua_pushnumber(L, DEG2RAD(luaL_checknumber(L, 1)));
	return 1;
}

/***
BUGGED
Breaks a number into its binary significant and an integral exponent for 2.
@function frexp
@param number
@return Array containing binary significant and integral exponent for 2.
*/
/* TODO return array */
static int Qmath_Frexp(lua_State * L)
{
	int             e;

	lua_pushnumber(L, frexp(luaL_checknumber(L, 1), &e));
	lua_pushnumber(L, e);
	return 2;
}

/***
Get the result from multiplying f by 2 raised to the power of n.
@function ldexp
@param f Number
@param n Number
@return The result from multiplying f by 2 raised to the power of n.
*/
static int Qmath_Ldexp(lua_State * L)
{
	lua_pushnumber(L, ldexp(luaL_checknumber(L, 1), luaL_checkint(L, 2)));
	return 1;
}


/***
Get the lowest value in array[].
@function min
@param array Array of numbers.
@return the lowest value in the array.
*/
static int Qmath_Min(lua_State * L)
{
	int             n = lua_gettop(L);	/* number of arguments */
	lua_Number      dmin = luaL_checknumber(L, 1);
	int             i;

	for(i = 2; i <= n; i++)
	{
		lua_Number      d = luaL_checknumber(L, i);

		if(d < dmin)
			dmin = d;
	}
	lua_pushnumber(L, dmin);
	return 1;
}

/***
Get the highest value in array[].
@function min
@param array Array of numbers.
@return the highest value in the array.
*/
static int Qmath_Max(lua_State * L)
{
	int             n = lua_gettop(L);	/* number of arguments */
	lua_Number      dmax = luaL_checknumber(L, 1);
	int             i;

	for(i = 2; i <= n; i++)
	{
		lua_Number      d = luaL_checknumber(L, i);

		if(d > dmax)
			dmax = d;
	}
	lua_pushnumber(L, dmax);
	return 1;
}

/***
Get a random integer.
@function rand
@return Random integer.
*/
static int Qmath_Rand(lua_State * L)
{
	lua_pushinteger(L, rand());
	return 1;
}

/***
Get a random floating point number.
@function random
@return Random floating point number.
*/
static int Qmath_Random(lua_State * L)
{
	lua_pushnumber(L, random());
	return 1;
}

/***
Get a random floating point number (using crazy random function).
@function crandom
@return A random floating point number (using crazy random function).
*/
static int Qmath_Crandom(lua_State * L)
{
	lua_pushnumber(L, crandom());
	return 1;
}

/***
Get a random integer from the range of integers defined by and including i and j.
@function irandom
@param i Number, lower limit.
@param j Number, upper limit.
@return A random integer from the range of integers defined by and including i and j.
*/
static int Qmath_Irandom(lua_State * L)
{
	lua_pushnumber(L, irandom(luaL_checkint(L, 1), luaL_checkint(L, 2)));
	return 1;
}

/***
Get a random float from the range of floats defined by and including i and j.
@function flrandom
@param i Number, lower limit.
@param j Number, upper limit.
@return A random float from the range of floats defined by and including i and j.
*/
static int Qmath_FLrandom(lua_State * L)
{
	lua_pushnumber(L, flrandom(luaL_checknumber(L, 1), luaL_checknumber(L, 2)));
	return 1;
}

static const luaL_Reg lib_qmath[] = {
	{"abs", Qmath_Abs},
	{"sin", Qmath_Sin},
	{"cos", Qmath_Cos},
	{"tan", Qmath_Tan},
	{"asin", Qmath_Asin},
	{"acos", Qmath_Acos},
	{"atan", Qmath_Atan},
	{"atan2", Qmath_Atan2},
	{"ceil", Qmath_Ceil},
	{"floor", Qmath_Floor},
	{"fmod", Qmath_Fmod},
	{"modf", Qmath_Modf},
	{"frexp", Qmath_Frexp},
	{"ldexp", Qmath_Ldexp},
	{"sqrt", Qmath_Sqrt},
	{"min", Qmath_Min},
	{"max", Qmath_Max},
	{"log", Qmath_Log},
	{"log10", Qmath_Log10},
	{"exp", Qmath_Exp},
	{"deg", Qmath_Deg},
	{"pow", Qmath_Pow},
	{"rad", Qmath_Rad},
	{"rand", Qmath_Rand},
	{"candom", Qmath_Crandom},
	{"random", Qmath_Random},
	{"irandom", Qmath_Irandom},
	{"flrandom", Qmath_FLrandom},
	{NULL, NULL}
};

int Luaopen_Qmath(lua_State * L)
{
	luaL_register(L, "qmath", lib_qmath);
	lua_pushnumber(L, M_PI);
	lua_setfield(L, -2, "pi");
	lua_pushnumber(L, HUGE_VAL);
	lua_setfield(L, -2, "huge");
	return 1;
}
#endif