1 /* see copyright notice in squirrel.h */
\r
2 #include <squirrel.h>
\r
5 #include <sqstdmath.h>
\r
7 #define SINGLE_ARG_FUNC(_funcname) static SQInteger math_##_funcname(HSQUIRRELVM v){ \
\r
9 sq_getfloat(v,2,&f); \
\r
10 sq_pushfloat(v,(SQFloat)_funcname(f)); \
\r
14 #define TWO_ARGS_FUNC(_funcname) static SQInteger math_##_funcname(HSQUIRRELVM v){ \
\r
16 sq_getfloat(v,2,&p1); \
\r
17 sq_getfloat(v,3,&p2); \
\r
18 sq_pushfloat(v,(SQFloat)_funcname(p1,p2)); \
\r
22 static SQInteger math_srand(HSQUIRRELVM v)
\r
25 if(!sq_getinteger(v,2,&i))return sq_throwerror(v,_SC("invalid param"));
\r
26 srand((unsigned int)i);
\r
30 static SQInteger math_rand(HSQUIRRELVM v)
\r
32 sq_pushinteger(v,rand());
\r
36 static SQInteger math_abs(HSQUIRRELVM v)
\r
39 sq_getinteger(v,2,&n);
\r
40 sq_pushinteger(v,(SQInteger)abs((int)n));
\r
44 SINGLE_ARG_FUNC(sqrt)
\r
45 SINGLE_ARG_FUNC(fabs)
\r
46 SINGLE_ARG_FUNC(sin)
\r
47 SINGLE_ARG_FUNC(cos)
\r
48 SINGLE_ARG_FUNC(asin)
\r
49 SINGLE_ARG_FUNC(acos)
\r
50 SINGLE_ARG_FUNC(log)
\r
51 SINGLE_ARG_FUNC(log10)
\r
52 SINGLE_ARG_FUNC(tan)
\r
53 SINGLE_ARG_FUNC(atan)
\r
54 TWO_ARGS_FUNC(atan2)
\r
56 SINGLE_ARG_FUNC(floor)
\r
57 SINGLE_ARG_FUNC(ceil)
\r
58 SINGLE_ARG_FUNC(exp)
\r
60 #define _DECL_FUNC(name,nparams,tycheck) {_SC(#name),math_##name,nparams,tycheck}
\r
61 static SQRegFunction mathlib_funcs[] = {
\r
62 _DECL_FUNC(sqrt,2,_SC(".n")),
\r
63 _DECL_FUNC(sin,2,_SC(".n")),
\r
64 _DECL_FUNC(cos,2,_SC(".n")),
\r
65 _DECL_FUNC(asin,2,_SC(".n")),
\r
66 _DECL_FUNC(acos,2,_SC(".n")),
\r
67 _DECL_FUNC(log,2,_SC(".n")),
\r
68 _DECL_FUNC(log10,2,_SC(".n")),
\r
69 _DECL_FUNC(tan,2,_SC(".n")),
\r
70 _DECL_FUNC(atan,2,_SC(".n")),
\r
71 _DECL_FUNC(atan2,3,_SC(".nn")),
\r
72 _DECL_FUNC(pow,3,_SC(".nn")),
\r
73 _DECL_FUNC(floor,2,_SC(".n")),
\r
74 _DECL_FUNC(ceil,2,_SC(".n")),
\r
75 _DECL_FUNC(exp,2,_SC(".n")),
\r
76 _DECL_FUNC(srand,2,_SC(".n")),
\r
77 _DECL_FUNC(rand,1,NULL),
\r
78 _DECL_FUNC(fabs,2,_SC(".n")),
\r
79 _DECL_FUNC(abs,2,_SC(".n")),
\r
84 #define M_PI (3.14159265358979323846)
\r
87 SQRESULT sqstd_register_mathlib(HSQUIRRELVM v)
\r
90 while(mathlib_funcs[i].name!=0) {
\r
91 sq_pushstring(v,mathlib_funcs[i].name,-1);
\r
92 sq_newclosure(v,mathlib_funcs[i].f,0);
\r
93 sq_setparamscheck(v,mathlib_funcs[i].nparamscheck,mathlib_funcs[i].typemask);
\r
94 sq_setnativeclosurename(v,-1,mathlib_funcs[i].name);
\r
95 sq_createslot(v,-3);
\r
98 sq_pushstring(v,_SC("RAND_MAX"),-1);
\r
99 sq_pushinteger(v,RAND_MAX);
\r
100 sq_createslot(v,-3);
\r
101 sq_pushstring(v,_SC("PI"),-1);
\r
102 sq_pushfloat(v,(SQFloat)M_PI);
\r
103 sq_createslot(v,-3);
\r