]>
Commit | Line | Data |
---|---|---|
16a5fed6 | 1 | /*\r |
2 | ** $Id: lmathlib.c,v 1.83.1.1 2013/04/12 18:48:47 roberto Exp $\r | |
3 | ** Standard mathematical library\r | |
4 | ** See Copyright Notice in lua.h\r | |
5 | */\r | |
6 | \r | |
7 | \r | |
8 | #include <stdlib.h>\r | |
9 | #include <math.h>\r | |
10 | \r | |
11 | #define lmathlib_c\r | |
12 | #define LUA_LIB\r | |
13 | \r | |
14 | #include "lua.h"\r | |
15 | \r | |
16 | #include "lauxlib.h"\r | |
17 | #include "lualib.h"\r | |
18 | \r | |
19 | \r | |
20 | #undef PI\r | |
21 | #define PI ((lua_Number)(3.1415926535897932384626433832795))\r | |
22 | #define RADIANS_PER_DEGREE ((lua_Number)(PI/180.0))\r | |
23 | \r | |
24 | \r | |
25 | \r | |
26 | static int math_abs (lua_State *L) {\r | |
27 | lua_pushnumber(L, l_mathop(fabs)(luaL_checknumber(L, 1)));\r | |
28 | return 1;\r | |
29 | }\r | |
30 | \r | |
31 | static int math_sin (lua_State *L) {\r | |
32 | lua_pushnumber(L, l_mathop(sin)(luaL_checknumber(L, 1)));\r | |
33 | return 1;\r | |
34 | }\r | |
35 | \r | |
36 | static int math_sinh (lua_State *L) {\r | |
37 | lua_pushnumber(L, l_mathop(sinh)(luaL_checknumber(L, 1)));\r | |
38 | return 1;\r | |
39 | }\r | |
40 | \r | |
41 | static int math_cos (lua_State *L) {\r | |
42 | lua_pushnumber(L, l_mathop(cos)(luaL_checknumber(L, 1)));\r | |
43 | return 1;\r | |
44 | }\r | |
45 | \r | |
46 | static int math_cosh (lua_State *L) {\r | |
47 | lua_pushnumber(L, l_mathop(cosh)(luaL_checknumber(L, 1)));\r | |
48 | return 1;\r | |
49 | }\r | |
50 | \r | |
51 | static int math_tan (lua_State *L) {\r | |
52 | lua_pushnumber(L, l_mathop(tan)(luaL_checknumber(L, 1)));\r | |
53 | return 1;\r | |
54 | }\r | |
55 | \r | |
56 | static int math_tanh (lua_State *L) {\r | |
57 | lua_pushnumber(L, l_mathop(tanh)(luaL_checknumber(L, 1)));\r | |
58 | return 1;\r | |
59 | }\r | |
60 | \r | |
61 | static int math_asin (lua_State *L) {\r | |
62 | lua_pushnumber(L, l_mathop(asin)(luaL_checknumber(L, 1)));\r | |
63 | return 1;\r | |
64 | }\r | |
65 | \r | |
66 | static int math_acos (lua_State *L) {\r | |
67 | lua_pushnumber(L, l_mathop(acos)(luaL_checknumber(L, 1)));\r | |
68 | return 1;\r | |
69 | }\r | |
70 | \r | |
71 | static int math_atan (lua_State *L) {\r | |
72 | lua_pushnumber(L, l_mathop(atan)(luaL_checknumber(L, 1)));\r | |
73 | return 1;\r | |
74 | }\r | |
75 | \r | |
76 | static int math_atan2 (lua_State *L) {\r | |
77 | lua_pushnumber(L, l_mathop(atan2)(luaL_checknumber(L, 1),\r | |
78 | luaL_checknumber(L, 2)));\r | |
79 | return 1;\r | |
80 | }\r | |
81 | \r | |
82 | static int math_ceil (lua_State *L) {\r | |
83 | lua_pushnumber(L, l_mathop(ceil)(luaL_checknumber(L, 1)));\r | |
84 | return 1;\r | |
85 | }\r | |
86 | \r | |
87 | static int math_floor (lua_State *L) {\r | |
88 | lua_pushnumber(L, l_mathop(floor)(luaL_checknumber(L, 1)));\r | |
89 | return 1;\r | |
90 | }\r | |
91 | \r | |
92 | static int math_fmod (lua_State *L) {\r | |
93 | lua_pushnumber(L, l_mathop(fmod)(luaL_checknumber(L, 1),\r | |
94 | luaL_checknumber(L, 2)));\r | |
95 | return 1;\r | |
96 | }\r | |
97 | \r | |
98 | static int math_modf (lua_State *L) {\r | |
99 | lua_Number ip;\r | |
100 | lua_Number fp = l_mathop(modf)(luaL_checknumber(L, 1), &ip);\r | |
101 | lua_pushnumber(L, ip);\r | |
102 | lua_pushnumber(L, fp);\r | |
103 | return 2;\r | |
104 | }\r | |
105 | \r | |
106 | static int math_sqrt (lua_State *L) {\r | |
107 | lua_pushnumber(L, l_mathop(sqrt)(luaL_checknumber(L, 1)));\r | |
108 | return 1;\r | |
109 | }\r | |
110 | \r | |
111 | static int math_pow (lua_State *L) {\r | |
112 | lua_Number x = luaL_checknumber(L, 1);\r | |
113 | lua_Number y = luaL_checknumber(L, 2);\r | |
114 | lua_pushnumber(L, l_mathop(pow)(x, y));\r | |
115 | return 1;\r | |
116 | }\r | |
117 | \r | |
118 | static int math_log (lua_State *L) {\r | |
119 | lua_Number x = luaL_checknumber(L, 1);\r | |
120 | lua_Number res;\r | |
121 | if (lua_isnoneornil(L, 2))\r | |
122 | res = l_mathop(log)(x);\r | |
123 | else {\r | |
124 | lua_Number base = luaL_checknumber(L, 2);\r | |
125 | if (base == (lua_Number)10.0) res = l_mathop(log10)(x);\r | |
126 | else res = l_mathop(log)(x)/l_mathop(log)(base);\r | |
127 | }\r | |
128 | lua_pushnumber(L, res);\r | |
129 | return 1;\r | |
130 | }\r | |
131 | \r | |
132 | #if defined(LUA_COMPAT_LOG10)\r | |
133 | static int math_log10 (lua_State *L) {\r | |
134 | lua_pushnumber(L, l_mathop(log10)(luaL_checknumber(L, 1)));\r | |
135 | return 1;\r | |
136 | }\r | |
137 | #endif\r | |
138 | \r | |
139 | static int math_exp (lua_State *L) {\r | |
140 | lua_pushnumber(L, l_mathop(exp)(luaL_checknumber(L, 1)));\r | |
141 | return 1;\r | |
142 | }\r | |
143 | \r | |
144 | static int math_deg (lua_State *L) {\r | |
145 | lua_pushnumber(L, luaL_checknumber(L, 1)/RADIANS_PER_DEGREE);\r | |
146 | return 1;\r | |
147 | }\r | |
148 | \r | |
149 | static int math_rad (lua_State *L) {\r | |
150 | lua_pushnumber(L, luaL_checknumber(L, 1)*RADIANS_PER_DEGREE);\r | |
151 | return 1;\r | |
152 | }\r | |
153 | \r | |
154 | static int math_frexp (lua_State *L) {\r | |
155 | int e;\r | |
156 | lua_pushnumber(L, l_mathop(frexp)(luaL_checknumber(L, 1), &e));\r | |
157 | lua_pushinteger(L, e);\r | |
158 | return 2;\r | |
159 | }\r | |
160 | \r | |
161 | static int math_ldexp (lua_State *L) {\r | |
162 | lua_Number x = luaL_checknumber(L, 1);\r | |
163 | int ep = luaL_checkint(L, 2);\r | |
164 | lua_pushnumber(L, l_mathop(ldexp)(x, ep));\r | |
165 | return 1;\r | |
166 | }\r | |
167 | \r | |
168 | \r | |
169 | \r | |
170 | static int math_min (lua_State *L) {\r | |
171 | int n = lua_gettop(L); /* number of arguments */\r | |
172 | lua_Number dmin = luaL_checknumber(L, 1);\r | |
173 | int i;\r | |
174 | for (i=2; i<=n; i++) {\r | |
175 | lua_Number d = luaL_checknumber(L, i);\r | |
176 | if (d < dmin)\r | |
177 | dmin = d;\r | |
178 | }\r | |
179 | lua_pushnumber(L, dmin);\r | |
180 | return 1;\r | |
181 | }\r | |
182 | \r | |
183 | \r | |
184 | static int math_max (lua_State *L) {\r | |
185 | int n = lua_gettop(L); /* number of arguments */\r | |
186 | lua_Number dmax = luaL_checknumber(L, 1);\r | |
187 | int i;\r | |
188 | for (i=2; i<=n; i++) {\r | |
189 | lua_Number d = luaL_checknumber(L, i);\r | |
190 | if (d > dmax)\r | |
191 | dmax = d;\r | |
192 | }\r | |
193 | lua_pushnumber(L, dmax);\r | |
194 | return 1;\r | |
195 | }\r | |
196 | \r | |
197 | \r | |
198 | static int math_random (lua_State *L) {\r | |
199 | /* the `%' avoids the (rare) case of r==1, and is needed also because on\r | |
200 | some systems (SunOS!) `rand()' may return a value larger than RAND_MAX */\r | |
201 | lua_Number r = (lua_Number)(rand()%RAND_MAX) / (lua_Number)RAND_MAX;\r | |
202 | switch (lua_gettop(L)) { /* check number of arguments */\r | |
203 | case 0: { /* no arguments */\r | |
204 | lua_pushnumber(L, r); /* Number between 0 and 1 */\r | |
205 | break;\r | |
206 | }\r | |
207 | case 1: { /* only upper limit */\r | |
208 | lua_Number u = luaL_checknumber(L, 1);\r | |
209 | luaL_argcheck(L, (lua_Number)1.0 <= u, 1, "interval is empty");\r | |
210 | lua_pushnumber(L, l_mathop(floor)(r*u) + (lua_Number)(1.0)); /* [1, u] */\r | |
211 | break;\r | |
212 | }\r | |
213 | case 2: { /* lower and upper limits */\r | |
214 | lua_Number l = luaL_checknumber(L, 1);\r | |
215 | lua_Number u = luaL_checknumber(L, 2);\r | |
216 | luaL_argcheck(L, l <= u, 2, "interval is empty");\r | |
217 | lua_pushnumber(L, l_mathop(floor)(r*(u-l+1)) + l); /* [l, u] */\r | |
218 | break;\r | |
219 | }\r | |
220 | default: return luaL_error(L, "wrong number of arguments");\r | |
221 | }\r | |
222 | return 1;\r | |
223 | }\r | |
224 | \r | |
225 | \r | |
226 | static int math_randomseed (lua_State *L) {\r | |
227 | srand(luaL_checkunsigned(L, 1));\r | |
228 | (void)rand(); /* discard first value to avoid undesirable correlations */\r | |
229 | return 0;\r | |
230 | }\r | |
231 | \r | |
232 | \r | |
233 | static const luaL_Reg mathlib[] = {\r | |
234 | {"abs", math_abs},\r | |
235 | {"acos", math_acos},\r | |
236 | {"asin", math_asin},\r | |
237 | {"atan2", math_atan2},\r | |
238 | {"atan", math_atan},\r | |
239 | {"ceil", math_ceil},\r | |
240 | {"cosh", math_cosh},\r | |
241 | {"cos", math_cos},\r | |
242 | {"deg", math_deg},\r | |
243 | {"exp", math_exp},\r | |
244 | {"floor", math_floor},\r | |
245 | {"fmod", math_fmod},\r | |
246 | {"frexp", math_frexp},\r | |
247 | {"ldexp", math_ldexp},\r | |
248 | #if defined(LUA_COMPAT_LOG10)\r | |
249 | {"log10", math_log10},\r | |
250 | #endif\r | |
251 | {"log", math_log},\r | |
252 | {"max", math_max},\r | |
253 | {"min", math_min},\r | |
254 | {"modf", math_modf},\r | |
255 | {"pow", math_pow},\r | |
256 | {"rad", math_rad},\r | |
257 | {"random", math_random},\r | |
258 | {"randomseed", math_randomseed},\r | |
259 | {"sinh", math_sinh},\r | |
260 | {"sin", math_sin},\r | |
261 | {"sqrt", math_sqrt},\r | |
262 | {"tanh", math_tanh},\r | |
263 | {"tan", math_tan},\r | |
264 | {NULL, NULL}\r | |
265 | };\r | |
266 | \r | |
267 | \r | |
268 | /*\r | |
269 | ** Open math library\r | |
270 | */\r | |
271 | LUAMOD_API int luaopen_math (lua_State *L) {\r | |
272 | luaL_newlib(L, mathlib);\r | |
273 | lua_pushnumber(L, PI);\r | |
274 | lua_setfield(L, -2, "pi");\r | |
275 | lua_pushnumber(L, HUGE_VAL);\r | |
276 | lua_setfield(L, -2, "huge");\r | |
277 | return 1;\r | |
278 | }\r | |
279 | \r |