6 // Toss a runtime error for any NaN or Inf values produced. Note that
7 // this assumes an IEEE 754 format.
8 #define VALIDATE(r) (valid(r.num) ? (r) : die(c, __FUNCTION__+2))
10 static int valid(double d)
12 union { double d; unsigned long long ull; } u;
14 return ((u.ull >> 52) & 0x7ff) != 0x7ff;
17 static naRef die(naContext c, const char* fn)
19 naRuntimeError(c, "floating point error in math.%s()", fn);
23 static naRef f_sin(naContext c, naRef me, int argc, naRef* args)
25 naRef a = naNumValue(argc > 0 ? args[0] : naNil());
27 naRuntimeError(c, "non numeric argument to sin()");
32 static naRef f_cos(naContext c, naRef me, int argc, naRef* args)
34 naRef a = naNumValue(argc > 0 ? args[0] : naNil());
36 naRuntimeError(c, "non numeric argument to cos()");
41 static naRef f_exp(naContext c, naRef me, int argc, naRef* args)
43 naRef a = naNumValue(argc > 0 ? args[0] : naNil());
45 naRuntimeError(c, "non numeric argument to exp()");
50 static naRef f_ln(naContext c, naRef me, int argc, naRef* args)
52 naRef a = naNumValue(argc > 0 ? args[0] : naNil());
54 naRuntimeError(c, "non numeric argument to ln()");
59 static naRef f_sqrt(naContext c, naRef me, int argc, naRef* args)
61 naRef a = naNumValue(argc > 0 ? args[0] : naNil());
63 naRuntimeError(c, "non numeric argument to sqrt()");
68 static naRef f_pow(naContext c, naRef me, int argc, naRef* args)
70 naRef a = naNumValue(argc > 0 ? args[0] : naNil());
71 naRef b = naNumValue(argc > 1 ? args[1] : naNil());
72 if(naIsNil(a) || naIsNil(b))
73 naRuntimeError(c, "non numeric argument to pow()");
74 a.num = pow(a.num, b.num);
78 static naRef f_atan2(naContext c, naRef me, int argc, naRef* args)
80 naRef a = naNumValue(argc > 0 ? args[0] : naNil());
81 naRef b = naNumValue(argc > 1 ? args[1] : naNil());
82 if(naIsNil(a) || naIsNil(b))
83 naRuntimeError(c, "non numeric argument to atan2()");
84 a.num = atan2(a.num, b.num);
88 static naRef f_floor(naContext c, naRef me, int argc, naRef* args)
90 naRef a = naNumValue(argc > 0 ? args[0] : naNil());
92 naRuntimeError(c, "non numeric argument to floor()");
97 static naRef f_ceil(naContext c, naRef me, int argc, naRef* args)
99 naRef a = naNumValue(argc > 0 ? args[0] : naNil());
101 naRuntimeError(c, "non numeric argument to ceil()");
106 static naRef f_fmod(naContext c, naRef me, int argc, naRef* args)
108 naRef a = naNumValue(argc > 0 ? args[0] : naNil());
109 naRef b = naNumValue(argc > 1 ? args[1] : naNil());
110 if(naIsNil(a) || naIsNil(b))
111 naRuntimeError(c, "non numeric arguments to fmod()");
113 a.num = fmod(a.num, b.num);
117 static naRef f_clamp(naContext c, naRef me, int argc, naRef* args)
119 naRef a = naNumValue(argc > 0 ? args[0] : naNil());
120 naRef min = naNumValue(argc > 1 ? args[1] : naNil());
121 naRef max = naNumValue(argc > 2 ? args[2] : naNil());
123 if(naIsNil(a) || naIsNil(min) || naIsNil(max))
124 naRuntimeError(c, "non numeric arguments to clamp()");
126 a.num = a.num < min.num ? min.num : ( a.num > max.num ? max.num : a.num );
130 static naRef f_periodic(naContext c, naRef me, int argc, naRef* args)
133 naRef a = naNumValue(argc > 0 ? args[0] : naNil());
134 naRef b = naNumValue(argc > 1 ? args[1] : naNil());
135 naRef x = naNumValue(argc > 2 ? args[2] : naNil());
137 if(naIsNil(a) || naIsNil(b) || naIsNil(x))
138 naRuntimeError(c, "non numeric arguments to periodic()");
140 range = b.num - a.num;
141 x.num = x.num - range*floor((x.num - a.num)/range);
142 // two security checks that can only happen due to roundoff
149 // x.num = SGMiscd::normalizePeriodic(a, b, x);
153 static naRef f_round(naContext c, naRef me, int argc, naRef* args)
155 naRef a = naNumValue(argc > 0 ? args[0] : naNil());
156 naRef b = naNumValue(argc > 1 ? args[1] : naNil());
161 naRuntimeError(c, "non numeric arguments to round()");
165 #ifdef _MSC_VER // MSVC is not C99-compatible, no round() in math.h
169 double x = round(a.num / b.num);
177 static naRef f_tan(naContext c, naRef me, int argc, naRef* args)
179 naRef a = naNumValue(argc > 0 ? args[0] : naNil());
181 naRuntimeError(c, "non numeric arguments to tan()");
187 static naRef f_asin(naContext c, naRef me, int argc, naRef* args)
189 naRef a = naNumValue(argc > 0 ? args[0] : naNil());
191 naRuntimeError(c, "non numeric argument to asin()");
196 static naRef f_acos(naContext c, naRef me, int argc, naRef* args)
198 naRef a = naNumValue(argc > 0 ? args[0] : naNil());
200 naRuntimeError(c, "non numeric argument to acos()");
205 static naCFuncItem funcs[] = {
212 { "atan2", f_atan2 },
213 { "floor", f_floor },
216 { "clamp", f_clamp },
217 { "periodic", f_periodic },
218 { "round", f_round },
227 naRef naInit_math(naContext c)
229 naRef ns = naGenLib(c, funcs);
230 naAddSym(c, ns, "pi", naNum(3.14159265358979323846));
231 naAddSym(c, ns, "e", naNum(2.7182818284590452354));