lua_math.cpp 26 KB

12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455565758596061626364656667686970717273747576777879808182838485868788899091929394959697989910010110210310410510610710810911011111211311411511611711811912012112212312412512612712812913013113213313413513613713813914014114214314414514614714814915015115215315415515615715815916016116216316416516616716816917017117217317417517617717817918018118218318418518618718818919019119219319419519619719819920020120220320420520620720820921021121221321421521621721821922022122222322422522622722822923023123223323423523623723823924024124224324424524624724824925025125225325425525625725825926026126226326426526626726826927027127227327427527627727827928028128228328428528628728828929029129229329429529629729829930030130230330430530630730830931031131231331431531631731831932032132232332432532632732832933033133233333433533633733833934034134234334434534634734834935035135235335435535635735835936036136236336436536636736836937037137237337437537637737837938038138238338438538638738838939039139239339439539639739839940040140240340440540640740840941041141241341441541641741841942042142242342442542642742842943043143243343443543643743843944044144244344444544644744844945045145245345445545645745845946046146246346446546646746846947047147247347447547647747847948048148248348448548648748848949049149249349449549649749849950050150250350450550650750850951051151251351451551651751851952052152252352452552652752852953053153253353453553653753853954054154254354454554654754854955055155255355455555655755855956056156256356456556656756856957057157257357457557657757857958058158258358458558658758858959059159259359459559659759859960060160260360460560660760860961061161261361461561661761861962062162262362462562662762862963063163263363463563663763863964064164264364464564664764864965065165265365465565665765865966066166266366466566666766866967067167267367467567667767867968068168268368468568668768868969069169269369469569669769869970070170270370470570670770870971071171271371471571671771871972072172272372472572672772872973073173273373473573673773873974074174274374474574674774874975075175275375475575675775875976076176276376476576676776876977077177277377477577677777877978078178278378478578678778878979079179279379479579679779879980080180280380480580680780880981081181281381481581681781881982082182282382482582682782882983083183283383483583683783883984084184284384484584684784884985085185285385485585685785885986086186286386486586686786886987087187287387487587687787887988088188288388488588688788888989089189289389489589689789889990090190290390490590690790890991091191291391491591691791891992092192292392492592692792892993093193293393493593693793893994094194294394494594694794894995095195295395495595695795895996096196296396496596696796896997097197297397497597697797897998098198298398498598698798898999099199299399499599699799899910001001100210031004100510061007100810091010101110121013101410151016101710181019102010211022102310241025102610271028102910301031103210331034103510361037103810391040104110421043
  1. /*
  2. * Copyright (c) 2012-2015 Daniele Bartolini and individual contributors.
  3. * License: https://github.com/taylor001/crown/blob/master/LICENSE
  4. */
  5. #include "math_utils.h"
  6. #include "math_types.h"
  7. #include "vector3.h"
  8. #include "matrix4x4.h"
  9. #include "quaternion.h"
  10. #include "color4.h"
  11. #include "intersection.h"
  12. #include "lua_stack.h"
  13. #include "lua_environment.h"
  14. namespace crown
  15. {
  16. static int math_to_rad(lua_State* L)
  17. {
  18. LuaStack stack(L);
  19. stack.push_float(to_rad(stack.get_float(1)));
  20. return 1;
  21. }
  22. static int math_to_deg(lua_State* L)
  23. {
  24. LuaStack stack(L);
  25. stack.push_float(to_deg(stack.get_float(1)));
  26. return 1;
  27. }
  28. static int math_next_pow_2(lua_State* L)
  29. {
  30. LuaStack stack(L);
  31. stack.push_uint32(next_pow_2(stack.get_int(1)));
  32. return 1;
  33. }
  34. static int math_is_pow_2(lua_State* L)
  35. {
  36. LuaStack stack(L);
  37. stack.push_bool(is_pow_2(stack.get_int(1)));
  38. return 1;
  39. }
  40. static int math_ceil(lua_State* L)
  41. {
  42. LuaStack stack(L);
  43. stack.push_float(ceil(stack.get_float(1)));
  44. return 1;
  45. }
  46. static int math_floor(lua_State* L)
  47. {
  48. LuaStack stack(L);
  49. stack.push_float(floor(stack.get_float(1)));
  50. return 1;
  51. }
  52. static int math_sqrt(lua_State* L)
  53. {
  54. LuaStack stack(L);
  55. stack.push_float(sqrt(stack.get_float(1)));
  56. return 1;
  57. }
  58. static int math_inv_sqrt(lua_State* L)
  59. {
  60. LuaStack stack(L);
  61. stack.push_float(inv_sqrt(stack.get_float(1)));
  62. return 1;
  63. }
  64. static int math_sin(lua_State* L)
  65. {
  66. LuaStack stack(L);
  67. stack.push_float(sin(stack.get_float(1)));
  68. return 1;
  69. }
  70. static int math_cos(lua_State* L)
  71. {
  72. LuaStack stack(L);
  73. stack.push_float(cos(stack.get_float(1)));
  74. return 1;
  75. }
  76. static int math_asin(lua_State* L)
  77. {
  78. LuaStack stack(L);
  79. stack.push_float(asin(stack.get_float(1)));
  80. return 1;
  81. }
  82. static int math_acos(lua_State* L)
  83. {
  84. LuaStack stack(L);
  85. stack.push_float(acos(stack.get_float(1)));
  86. return 1;
  87. }
  88. static int math_tan(lua_State* L)
  89. {
  90. LuaStack stack(L);
  91. stack.push_float(tan(stack.get_float(1)));
  92. return 1;
  93. }
  94. static int math_atan2(lua_State* L)
  95. {
  96. LuaStack stack(L);
  97. stack.push_float(atan2(stack.get_float(1), stack.get_float(2)));
  98. return 1;
  99. }
  100. static int math_abs(lua_State* L)
  101. {
  102. LuaStack stack(L);
  103. stack.push_float(abs(stack.get_float(1)));
  104. return 1;
  105. }
  106. static int math_fmod(lua_State* L)
  107. {
  108. LuaStack stack(L);
  109. stack.push_float(fmod(stack.get_float(1), stack.get_float(2)));
  110. return 1;
  111. }
  112. static int math_ray_oobb_intersection(lua_State* L)
  113. {
  114. LuaStack stack(L);
  115. OBB oobb;
  116. oobb.tm = stack.get_matrix4x4(3);
  117. oobb.aabb.min = stack.get_vector3(4) * -0.5;
  118. oobb.aabb.max = stack.get_vector3(4) * 0.5;
  119. stack.push_float(ray_oobb_intersection(stack.get_vector3(1), stack.get_vector3(2), oobb));
  120. return 1;
  121. }
  122. static int vector3_new(lua_State* L)
  123. {
  124. LuaStack stack(L);
  125. stack.push_vector3(vector3(stack.get_float(1), stack.get_float(2), stack.get_float(3)));
  126. return 1;
  127. }
  128. static int vector3_ctor(lua_State* L)
  129. {
  130. LuaStack stack(L);
  131. stack.remove(1); // Remove table
  132. return vector3_new(L);
  133. }
  134. static int vector3_x(lua_State* L)
  135. {
  136. LuaStack stack(L);
  137. stack.push_float(stack.get_vector3(1).x);
  138. return 1;
  139. }
  140. static int vector3_y(lua_State* L)
  141. {
  142. LuaStack stack(L);
  143. stack.push_float(stack.get_vector3(1).y);
  144. return 1;
  145. }
  146. static int vector3_z(lua_State* L)
  147. {
  148. LuaStack stack(L);
  149. stack.push_float(stack.get_vector3(1).z);
  150. return 1;
  151. }
  152. static int vector3_set_x(lua_State* L)
  153. {
  154. LuaStack stack(L);
  155. stack.get_vector3(1).x = stack.get_float(2);
  156. return 0;
  157. }
  158. static int vector3_set_y(lua_State* L)
  159. {
  160. LuaStack stack(L);
  161. stack.get_vector3(1).y = stack.get_float(2);
  162. return 0;
  163. }
  164. static int vector3_set_z(lua_State* L)
  165. {
  166. LuaStack stack(L);
  167. stack.get_vector3(1).z = stack.get_float(2);
  168. return 0;
  169. }
  170. static int vector3_values(lua_State* L)
  171. {
  172. LuaStack stack(L);
  173. Vector3& a = stack.get_vector3(1);
  174. stack.push_float(a.x);
  175. stack.push_float(a.y);
  176. stack.push_float(a.z);
  177. return 3;
  178. }
  179. static int vector3_add(lua_State* L)
  180. {
  181. LuaStack stack(L);
  182. stack.push_vector3(stack.get_vector3(1) + stack.get_vector3(2));
  183. return 1;
  184. }
  185. static int vector3_subtract(lua_State* L)
  186. {
  187. LuaStack stack(L);
  188. stack.push_vector3(stack.get_vector3(1) - stack.get_vector3(2));
  189. return 1;
  190. }
  191. static int vector3_multiply(lua_State* L)
  192. {
  193. LuaStack stack(L);
  194. stack.push_vector3(stack.get_vector3(1) * stack.get_float(2));
  195. return 1;
  196. }
  197. static int vector3_divide(lua_State* L)
  198. {
  199. LuaStack stack(L);
  200. stack.push_vector3(stack.get_vector3(1) / stack.get_float(2));
  201. return 1;
  202. }
  203. static int vector3_dot(lua_State* L)
  204. {
  205. LuaStack stack(L);
  206. stack.push_float(dot(stack.get_vector3(1), stack.get_vector3(2)));
  207. return 1;
  208. }
  209. static int vector3_cross(lua_State* L)
  210. {
  211. LuaStack stack(L);
  212. stack.push_vector3(cross(stack.get_vector3(1), stack.get_vector3(2)));
  213. return 1;
  214. }
  215. static int vector3_equal(lua_State* L)
  216. {
  217. LuaStack stack(L);
  218. stack.push_bool(stack.get_vector3(1) == stack.get_vector3(2));
  219. return 1;
  220. }
  221. static int vector3_length(lua_State* L)
  222. {
  223. LuaStack stack(L);
  224. stack.push_float(length(stack.get_vector3(1)));
  225. return 1;
  226. }
  227. static int vector3_squared_length(lua_State* L)
  228. {
  229. LuaStack stack(L);
  230. stack.push_float(squared_length(stack.get_vector3(1)));
  231. return 1;
  232. }
  233. static int vector3_set_length(lua_State* L)
  234. {
  235. LuaStack stack(L);
  236. set_length(stack.get_vector3(1), stack.get_float(2));
  237. return 0;
  238. }
  239. static int vector3_normalize(lua_State* L)
  240. {
  241. LuaStack stack(L);
  242. stack.push_vector3(normalize(stack.get_vector3(1)));
  243. return 1;
  244. }
  245. static int vector3_distance(lua_State* L)
  246. {
  247. LuaStack stack(L);
  248. stack.push_float(distance(stack.get_vector3(1), stack.get_vector3(2)));
  249. return 1;
  250. }
  251. static int vector3_angle(lua_State* L)
  252. {
  253. LuaStack stack(L);
  254. stack.push_float(angle(stack.get_vector3(1), stack.get_vector3(2)));
  255. return 1;
  256. }
  257. static int vector3_forward(lua_State* L)
  258. {
  259. LuaStack stack(L);
  260. stack.push_vector3(VECTOR3_FORWARD);
  261. return 1;
  262. }
  263. static int vector3_backward(lua_State* L)
  264. {
  265. LuaStack stack(L);
  266. stack.push_vector3(VECTOR3_BACKWARD);
  267. return 1;
  268. }
  269. static int vector3_left(lua_State* L)
  270. {
  271. LuaStack stack(L);
  272. stack.push_vector3(VECTOR3_LEFT);
  273. return 1;
  274. }
  275. static int vector3_right(lua_State* L)
  276. {
  277. LuaStack stack(L);
  278. stack.push_vector3(VECTOR3_RIGHT);
  279. return 1;
  280. }
  281. static int vector3_up(lua_State* L)
  282. {
  283. LuaStack stack(L);
  284. stack.push_vector3(VECTOR3_UP);
  285. return 1;
  286. }
  287. static int vector3_down(lua_State* L)
  288. {
  289. LuaStack stack(L);
  290. stack.push_vector3(VECTOR3_DOWN);
  291. return 1;
  292. }
  293. static int vector3_zero(lua_State* L)
  294. {
  295. LuaStack stack(L);
  296. stack.push_vector3(VECTOR3_ZERO);
  297. return 1;
  298. }
  299. static int vector2_new(lua_State* L)
  300. {
  301. LuaStack stack(L);
  302. stack.push_vector2(vector2(stack.get_float(1), stack.get_float(2)));
  303. return 1;
  304. }
  305. static int vector2_ctor(lua_State* L)
  306. {
  307. LuaStack stack(L);
  308. stack.remove(1); // Remove table
  309. return vector2_new(L);
  310. }
  311. static int vector3box_new(lua_State* L)
  312. {
  313. LuaStack stack(L);
  314. if (stack.num_args() == 0)
  315. stack.push_vector3box(Vector3());
  316. else if (stack.num_args() == 1)
  317. stack.push_vector3box(stack.get_vector3(1));
  318. else
  319. stack.push_vector3box(vector3(stack.get_float(1)
  320. , stack.get_float(2)
  321. , stack.get_float(3)));
  322. return 1;
  323. }
  324. static int vector3box_ctor(lua_State* L)
  325. {
  326. LuaStack stack(L);
  327. stack.remove(1); // Remove table
  328. return vector3box_new(L);
  329. }
  330. static int vector3box_store(lua_State* L)
  331. {
  332. LuaStack stack(L);
  333. Vector3& v = stack.get_vector3box(1);
  334. if (stack.num_args() == 2)
  335. v = stack.get_vector3(2);
  336. else
  337. v = vector3(stack.get_float(2)
  338. , stack.get_float(3)
  339. , stack.get_float(4));
  340. return 0;
  341. }
  342. static int vector3box_unbox(lua_State* L)
  343. {
  344. LuaStack stack(L);
  345. stack.push_vector3(stack.get_vector3box(1));
  346. return 1;
  347. }
  348. static int vector3box_get_value(lua_State* L)
  349. {
  350. LuaStack stack(L);
  351. Vector3& v = stack.get_vector3box(1);
  352. const char* s = stack.get_string(2);
  353. switch (s[0])
  354. {
  355. case 'x': stack.push_float(v.x); return 1;
  356. case 'y': stack.push_float(v.y); return 1;
  357. case 'z': stack.push_float(v.z); return 1;
  358. default: LUA_ASSERT(false, stack, "Bad index: '%c'", s[0]); break;
  359. }
  360. return 0;
  361. }
  362. static int vector3box_set_value(lua_State* L)
  363. {
  364. LuaStack stack(L);
  365. Vector3& v = stack.get_vector3box(1);
  366. const char* s = stack.get_string(2);
  367. const float value = stack.get_float(3);
  368. switch (s[0])
  369. {
  370. case 'x': v.x = value; break;
  371. case 'y': v.y = value; break;
  372. case 'z': v.z = value; break;
  373. default: LUA_ASSERT(false, stack, "Bad index: '%c'", s[0]); break;
  374. }
  375. return 0;
  376. }
  377. static int vector3box_tostring(lua_State* L)
  378. {
  379. LuaStack stack(L);
  380. Vector3& v = stack.get_vector3box(1);
  381. stack.push_fstring("Vector3Box (%p)", &v);
  382. return 1;
  383. }
  384. static int matrix4x4_new(lua_State* L)
  385. {
  386. LuaStack stack(L);
  387. stack.push_matrix4x4(matrix4x4(stack.get_float(1)
  388. , stack.get_float(2)
  389. , stack.get_float(3)
  390. , stack.get_float(4)
  391. , stack.get_float(5)
  392. , stack.get_float(6)
  393. , stack.get_float(7)
  394. , stack.get_float(8)
  395. , stack.get_float(9)
  396. , stack.get_float(10)
  397. , stack.get_float(11)
  398. , stack.get_float(12)
  399. , stack.get_float(13)
  400. , stack.get_float(14)
  401. , stack.get_float(15)
  402. , stack.get_float(16)));
  403. return 1;
  404. }
  405. static int matrix4x4_ctor(lua_State* L)
  406. {
  407. LuaStack stack(L);
  408. stack.remove(1); // Remove table
  409. return matrix4x4_new(L);
  410. }
  411. static int matrix4x4_from_quaternion(lua_State* L)
  412. {
  413. LuaStack stack(L);
  414. stack.push_matrix4x4(matrix4x4(stack.get_quaternion(1), vector3(0, 0, 0)));
  415. return 1;
  416. }
  417. static int matrix4x4_from_translation(lua_State* L)
  418. {
  419. LuaStack stack(L);
  420. stack.push_matrix4x4(matrix4x4(QUATERNION_IDENTITY, stack.get_vector3(1)));
  421. return 1;
  422. }
  423. static int matrix4x4_from_quaternion_translation(lua_State* L)
  424. {
  425. LuaStack stack(L);
  426. stack.push_matrix4x4(matrix4x4(stack.get_quaternion(1), stack.get_vector3(2)));
  427. return 1;
  428. }
  429. static int matrix4x4_from_axes(lua_State* L)
  430. {
  431. LuaStack stack(L);
  432. stack.push_matrix4x4(matrix4x4(stack.get_vector3(1), stack.get_vector3(2), stack.get_vector3(3), stack.get_vector3(4)));
  433. return 1;
  434. }
  435. static int matrix4x4_add(lua_State* L)
  436. {
  437. LuaStack stack(L);
  438. stack.push_matrix4x4(stack.get_matrix4x4(1) + stack.get_matrix4x4(2));
  439. return 1;
  440. }
  441. static int matrix4x4_subtract(lua_State* L)
  442. {
  443. LuaStack stack(L);
  444. stack.push_matrix4x4(stack.get_matrix4x4(1) - stack.get_matrix4x4(2));
  445. return 1;
  446. }
  447. static int matrix4x4_multiply(lua_State* L)
  448. {
  449. LuaStack stack(L);
  450. stack.push_matrix4x4(stack.get_matrix4x4(1) * stack.get_matrix4x4(2));
  451. return 1;
  452. }
  453. static int matrix4x4_transpose(lua_State* L)
  454. {
  455. LuaStack stack(L);
  456. stack.push_matrix4x4(transpose(stack.get_matrix4x4(1)));
  457. return 1;
  458. }
  459. static int matrix4x4_determinant(lua_State* L)
  460. {
  461. LuaStack stack(L);
  462. stack.push_float(determinant(stack.get_matrix4x4(1)));
  463. return 1;
  464. }
  465. static int matrix4x4_invert(lua_State* L)
  466. {
  467. LuaStack stack(L);
  468. stack.push_matrix4x4(invert(stack.get_matrix4x4(1)));
  469. return 1;
  470. }
  471. static int matrix4x4_x(lua_State* L)
  472. {
  473. LuaStack stack(L);
  474. stack.push_vector3(x(stack.get_matrix4x4(1)));
  475. return 1;
  476. }
  477. static int matrix4x4_y(lua_State* L)
  478. {
  479. LuaStack stack(L);
  480. stack.push_vector3(y(stack.get_matrix4x4(1)));
  481. return 1;
  482. }
  483. static int matrix4x4_z(lua_State* L)
  484. {
  485. LuaStack stack(L);
  486. stack.push_vector3(z(stack.get_matrix4x4(1)));
  487. return 1;
  488. }
  489. static int matrix4x4_set_x(lua_State* L)
  490. {
  491. LuaStack stack(L);
  492. set_x(stack.get_matrix4x4(1), stack.get_vector3(2));
  493. return 0;
  494. }
  495. static int matrix4x4_set_y(lua_State* L)
  496. {
  497. LuaStack stack(L);
  498. set_y(stack.get_matrix4x4(1), stack.get_vector3(2));
  499. return 0;
  500. }
  501. static int matrix4x4_set_z(lua_State* L)
  502. {
  503. LuaStack stack(L);
  504. set_z(stack.get_matrix4x4(1), stack.get_vector3(2));
  505. return 0;
  506. }
  507. static int matrix4x4_translation(lua_State* L)
  508. {
  509. LuaStack stack(L);
  510. stack.push_vector3(translation(stack.get_matrix4x4(1)));
  511. return 1;
  512. }
  513. static int matrix4x4_set_translation(lua_State* L)
  514. {
  515. LuaStack stack(L);
  516. set_translation(stack.get_matrix4x4(1), stack.get_vector3(2));
  517. return 0;
  518. }
  519. static int matrix4x4_rotation(lua_State* L)
  520. {
  521. LuaStack stack(L);
  522. stack.push_quaternion(rotation(stack.get_matrix4x4(1)));
  523. return 1;
  524. }
  525. static int matrix4x4_set_rotation(lua_State* L)
  526. {
  527. LuaStack stack(L);
  528. set_rotation(stack.get_matrix4x4(1), stack.get_quaternion(2));
  529. return 0;
  530. }
  531. static int matrix4x4_identity(lua_State* L)
  532. {
  533. LuaStack stack(L);
  534. stack.push_matrix4x4(MATRIX4X4_IDENTITY);
  535. return 1;
  536. }
  537. static int matrix4x4_to_string(lua_State* L)
  538. {
  539. LuaStack stack(L);
  540. Matrix4x4& a = stack.get_matrix4x4(1);
  541. stack.push_fstring("%.1f, %.1f, %.1f, %.1f\n%.1f, %.1f, %.1f, %.1f\n%.1f, %.1f, %.1f, %.1f\n%.1f, %.1f, %.1f, %.1f\n",
  542. a.x.x, a.x.y, a.x.z, a.y.w, a.y.x, a.y.y, a.y.z, a.y.w, a.z.x, a.z.y, a.z.z, a.z.w, a.t.x, a.t.y, a.t.z, a.t.w);
  543. return 1;
  544. }
  545. static int matrix4x4box_new(lua_State* L)
  546. {
  547. LuaStack stack(L);
  548. stack.push_matrix4x4box(stack.get_matrix4x4(1));
  549. return 1;
  550. }
  551. static int matrix4x4box_ctor(lua_State* L)
  552. {
  553. LuaStack stack(L);
  554. stack.remove(1); // Remove table
  555. return matrix4x4box_new(L);
  556. }
  557. static int matrix4x4box_store(lua_State* L)
  558. {
  559. LuaStack stack(L);
  560. stack.get_matrix4x4box(1) = stack.get_matrix4x4(2);
  561. return 0;
  562. }
  563. static int matrix4x4box_unbox(lua_State* L)
  564. {
  565. LuaStack stack(L);
  566. stack.push_matrix4x4(stack.get_matrix4x4box(1));
  567. return 1;
  568. }
  569. static int matrix4x4box_tostring(lua_State* L)
  570. {
  571. LuaStack stack(L);
  572. Matrix4x4& m = stack.get_matrix4x4box(1);
  573. stack.push_fstring("Matrix4x4Box (%p)", &m);
  574. return 1;
  575. }
  576. static int quaternion_new(lua_State* L)
  577. {
  578. LuaStack stack(L);
  579. stack.push_quaternion(quaternion(stack.get_vector3(1), stack.get_float(2)));
  580. return 1;
  581. }
  582. static int quaternion_ctor(lua_State* L)
  583. {
  584. LuaStack stack(L);
  585. stack.remove(1); // Remove table
  586. return quaternion_new(L);
  587. }
  588. static int quaternion_negate(lua_State* L)
  589. {
  590. LuaStack stack(L);
  591. stack.push_quaternion(-stack.get_quaternion(1));
  592. return 1;
  593. }
  594. static int quaternion_identity(lua_State* L)
  595. {
  596. LuaStack stack(L);
  597. stack.push_quaternion(QUATERNION_IDENTITY);
  598. return 1;
  599. }
  600. static int quaternion_length(lua_State* L)
  601. {
  602. LuaStack stack(L);
  603. stack.push_float(length(stack.get_quaternion(1)));
  604. return 1;
  605. }
  606. static int quaternion_normalize(lua_State* L)
  607. {
  608. LuaStack stack(L);
  609. stack.push_quaternion(normalize(stack.get_quaternion(1)));
  610. return 1;
  611. }
  612. static int quaternion_conjugate(lua_State* L)
  613. {
  614. LuaStack stack(L);
  615. stack.push_quaternion(conjugate(stack.get_quaternion(1)));
  616. return 1;
  617. }
  618. static int quaternion_inverse(lua_State* L)
  619. {
  620. LuaStack stack(L);
  621. stack.push_quaternion(inverse(stack.get_quaternion(1)));
  622. return 1;
  623. }
  624. static int quaternion_multiply(lua_State* L)
  625. {
  626. LuaStack stack(L);
  627. stack.push_quaternion(stack.get_quaternion(1) * stack.get_quaternion(2));
  628. return 1;
  629. }
  630. static int quaternion_multiply_by_scalar(lua_State* L)
  631. {
  632. LuaStack stack(L);
  633. stack.push_quaternion(stack.get_quaternion(1) * stack.get_float(2));
  634. return 1;
  635. }
  636. static int quaternion_power(lua_State* L)
  637. {
  638. LuaStack stack(L);
  639. stack.push_quaternion(power(stack.get_quaternion(1), stack.get_float(2)));
  640. return 1;
  641. }
  642. static int quaternion_elements(lua_State* L)
  643. {
  644. LuaStack stack(L);
  645. const Quaternion& q = stack.get_quaternion(1);
  646. stack.push_float(q.x);
  647. stack.push_float(q.y);
  648. stack.push_float(q.z);
  649. stack.push_float(q.w);
  650. return 4;
  651. }
  652. static int quaternionbox_new(lua_State* L)
  653. {
  654. LuaStack stack(L);
  655. if (stack.num_args() == 1)
  656. stack.push_quaternionbox(stack.get_quaternion(1));
  657. else
  658. stack.push_quaternionbox(quaternion(stack.get_float(1)
  659. , stack.get_float(2)
  660. , stack.get_float(3)
  661. , stack.get_float(4)));
  662. return 1;
  663. }
  664. static int quaternionbox_ctor(lua_State* L)
  665. {
  666. LuaStack stack(L);
  667. stack.remove(1); // Remove table
  668. return quaternionbox_new(L);
  669. }
  670. static int quaternionbox_store(lua_State* L)
  671. {
  672. LuaStack stack(L);
  673. Quaternion& q = stack.get_quaternionbox(1);
  674. if (stack.num_args() == 2)
  675. q = stack.get_quaternion(2);
  676. else
  677. q = quaternion(stack.get_float(2)
  678. , stack.get_float(3)
  679. , stack.get_float(4)
  680. , stack.get_float(5));
  681. return 0;
  682. }
  683. static int quaternionbox_unbox(lua_State* L)
  684. {
  685. LuaStack stack(L);
  686. Quaternion& q = stack.get_quaternionbox(1);
  687. stack.push_quaternion(q);
  688. return 1;
  689. }
  690. static int quaternionbox_tostring(lua_State* L)
  691. {
  692. LuaStack stack(L);
  693. Quaternion& q = stack.get_quaternionbox(1);
  694. stack.push_fstring("QuaternionBox (%p)", &q);
  695. return 1;
  696. }
  697. static int color4_new(lua_State* L)
  698. {
  699. LuaStack stack(L);
  700. stack.push_quaternion(quaternion(stack.get_float(1)
  701. , stack.get_float(2)
  702. , stack.get_float(3)
  703. , stack.get_float(4)));
  704. return 1;
  705. }
  706. static int color4_ctor(lua_State* L)
  707. {
  708. LuaStack stack(L);
  709. stack.remove(1); // Remove table
  710. return color4_new(L);
  711. }
  712. static int lightuserdata_add(lua_State* L)
  713. {
  714. LuaStack stack(L);
  715. const Vector3& a = stack.get_vector3(1);
  716. const Vector3& b = stack.get_vector3(2);
  717. stack.push_vector3(a + b);
  718. return 1;
  719. }
  720. static int lightuserdata_sub(lua_State* L)
  721. {
  722. LuaStack stack(L);
  723. const Vector3& a = stack.get_vector3(1);
  724. const Vector3& b = stack.get_vector3(2);
  725. stack.push_vector3(a - b);
  726. return 1;
  727. }
  728. static int lightuserdata_mul(lua_State* L)
  729. {
  730. LuaStack stack(L);
  731. const Vector3& a = stack.get_vector3(1);
  732. const float b = stack.get_float(2);
  733. stack.push_vector3(a * b);
  734. return 1;
  735. }
  736. static int lightuserdata_div(lua_State* L)
  737. {
  738. LuaStack stack(L);
  739. const Vector3& a = stack.get_vector3(1);
  740. const float b = stack.get_float(2);
  741. stack.push_vector3(a / b);
  742. return 1;
  743. }
  744. static int lightuserdata_unm(lua_State* L)
  745. {
  746. LuaStack stack(L);
  747. stack.push_vector3(-stack.get_vector3(1));
  748. return 1;
  749. }
  750. static int lightuserdata_index(lua_State* L)
  751. {
  752. LuaStack stack(L);
  753. Vector3& v = stack.get_vector3(1);
  754. const char* s = stack.get_string(2);
  755. switch (s[0])
  756. {
  757. case 'x': stack.push_float(v.x); return 1;
  758. case 'y': stack.push_float(v.y); return 1;
  759. case 'z': stack.push_float(v.z); return 1;
  760. default: LUA_ASSERT(false, stack, "Bad index: '%c'", s[0]); break;
  761. }
  762. return 0;
  763. }
  764. static int lightuserdata_newindex(lua_State* L)
  765. {
  766. LuaStack stack(L);
  767. Vector3& v = stack.get_vector3(1);
  768. const char* s = stack.get_string(2);
  769. const float value = stack.get_float(3);
  770. switch (s[0])
  771. {
  772. case 'x': v.x = value; break;
  773. case 'y': v.y = value; break;
  774. case 'z': v.z = value; break;
  775. default: LUA_ASSERT(false, stack, "Bad index: '%c'", s[0]); break;
  776. }
  777. return 0;
  778. }
  779. void load_math(LuaEnvironment& env)
  780. {
  781. env.load_module_function("Math", "to_rad", math_to_rad);
  782. env.load_module_function("Math", "to_deg", math_to_deg);
  783. env.load_module_function("Math", "next_pow_2", math_next_pow_2);
  784. env.load_module_function("Math", "is_pow_2", math_is_pow_2);
  785. env.load_module_function("Math", "ceil", math_ceil);
  786. env.load_module_function("Math", "floor", math_floor);
  787. env.load_module_function("Math", "sqrt", math_sqrt);
  788. env.load_module_function("Math", "inv_sqrt", math_inv_sqrt);
  789. env.load_module_function("Math", "sin", math_sin);
  790. env.load_module_function("Math", "cos", math_cos);
  791. env.load_module_function("Math", "asin", math_asin);
  792. env.load_module_function("Math", "acos", math_acos);
  793. env.load_module_function("Math", "tan", math_tan);
  794. env.load_module_function("Math", "atan2", math_atan2);
  795. env.load_module_function("Math", "abs", math_abs);
  796. env.load_module_function("Math", "fmod", math_fmod);
  797. env.load_module_function("Math", "ray_oobb_intersection", math_ray_oobb_intersection);
  798. env.load_module_function("Vector3", "new", vector3_new);
  799. env.load_module_function("Vector3", "x", vector3_x);
  800. env.load_module_function("Vector3", "y", vector3_y);
  801. env.load_module_function("Vector3", "z", vector3_z);
  802. env.load_module_function("Vector3", "set_x", vector3_set_x);
  803. env.load_module_function("Vector3", "set_y", vector3_set_y);
  804. env.load_module_function("Vector3", "set_z", vector3_set_z);
  805. env.load_module_function("Vector3", "values", vector3_values);
  806. env.load_module_function("Vector3", "add", vector3_add);
  807. env.load_module_function("Vector3", "subtract", vector3_subtract);
  808. env.load_module_function("Vector3", "multiply", vector3_multiply);
  809. env.load_module_function("Vector3", "divide", vector3_divide);
  810. env.load_module_function("Vector3", "dot", vector3_dot);
  811. env.load_module_function("Vector3", "cross", vector3_cross);
  812. env.load_module_function("Vector3", "equal", vector3_equal);
  813. env.load_module_function("Vector3", "length", vector3_length);
  814. env.load_module_function("Vector3", "squared_length", vector3_squared_length);
  815. env.load_module_function("Vector3", "set_length", vector3_set_length);
  816. env.load_module_function("Vector3", "normalize", vector3_normalize);
  817. env.load_module_function("Vector3", "distance", vector3_distance);
  818. env.load_module_function("Vector3", "angle", vector3_angle);
  819. env.load_module_function("Vector3", "forward", vector3_forward);
  820. env.load_module_function("Vector3", "backward", vector3_backward);
  821. env.load_module_function("Vector3", "left", vector3_left);
  822. env.load_module_function("Vector3", "right", vector3_right);
  823. env.load_module_function("Vector3", "up", vector3_up);
  824. env.load_module_function("Vector3", "down", vector3_down);
  825. env.load_module_function("Vector3", "zero", vector3_zero);
  826. env.load_module_constructor("Vector3", vector3_ctor);
  827. env.load_module_function("Vector2", "new", vector2_new);
  828. env.load_module_constructor("Vector2", vector2_ctor);
  829. env.load_module_function("Vector3Box", "new", vector3box_new);
  830. env.load_module_function("Vector3Box", "store", vector3box_store);
  831. env.load_module_function("Vector3Box", "unbox", vector3box_unbox);
  832. env.load_module_function("Vector3Box", "__index", vector3box_get_value);
  833. env.load_module_function("Vector3Box", "__newindex", vector3box_set_value);
  834. env.load_module_function("Vector3Box", "__tostring", vector3box_tostring);
  835. env.load_module_constructor("Vector3Box", vector3box_ctor);
  836. env.load_module_function("Matrix4x4", "new", matrix4x4_new);
  837. env.load_module_function("Matrix4x4", "from_quaternion", matrix4x4_from_quaternion);
  838. env.load_module_function("Matrix4x4", "from_translation", matrix4x4_from_translation);
  839. env.load_module_function("Matrix4x4", "from_quaternion_translation", matrix4x4_from_quaternion_translation);
  840. env.load_module_function("Matrix4x4", "from_axes", matrix4x4_from_axes);
  841. env.load_module_function("Matrix4x4", "add", matrix4x4_add);
  842. env.load_module_function("Matrix4x4", "subtract", matrix4x4_subtract);
  843. env.load_module_function("Matrix4x4", "multiply", matrix4x4_multiply);
  844. env.load_module_function("Matrix4x4", "transpose", matrix4x4_transpose);
  845. env.load_module_function("Matrix4x4", "determinant", matrix4x4_determinant);
  846. env.load_module_function("Matrix4x4", "invert", matrix4x4_invert);
  847. env.load_module_function("Matrix4x4", "x", matrix4x4_x);
  848. env.load_module_function("Matrix4x4", "y", matrix4x4_y);
  849. env.load_module_function("Matrix4x4", "z", matrix4x4_z);
  850. env.load_module_function("Matrix4x4", "set_x", matrix4x4_set_x);
  851. env.load_module_function("Matrix4x4", "set_y", matrix4x4_set_y);
  852. env.load_module_function("Matrix4x4", "set_z", matrix4x4_set_z);
  853. env.load_module_function("Matrix4x4", "translation", matrix4x4_translation);
  854. env.load_module_function("Matrix4x4", "set_translation", matrix4x4_set_translation);
  855. env.load_module_function("Matrix4x4", "rotation", matrix4x4_rotation);
  856. env.load_module_function("Matrix4x4", "set_rotation", matrix4x4_set_rotation);
  857. env.load_module_function("Matrix4x4", "identity", matrix4x4_identity);
  858. env.load_module_function("Matrix4x4", "to_string", matrix4x4_to_string);
  859. env.load_module_constructor("Matrix4x4", matrix4x4_ctor);
  860. env.load_module_function("Matrix4x4Box", "new", matrix4x4box_new);
  861. env.load_module_function("Matrix4x4Box", "store", matrix4x4box_store);
  862. env.load_module_function("Matrix4x4Box", "unbox", matrix4x4box_unbox);
  863. env.load_module_function("Matrix4x4Box", "__tostring", matrix4x4box_tostring);
  864. env.load_module_constructor("Matrix4x4Box", matrix4x4box_ctor);
  865. env.load_module_function("Quaternion", "new", quaternion_new);
  866. env.load_module_function("Quaternion", "negate", quaternion_negate);
  867. env.load_module_function("Quaternion", "identity", quaternion_identity);
  868. env.load_module_function("Quaternion", "multiply", quaternion_multiply);
  869. env.load_module_function("Quaternion", "multiply_by_scalar", quaternion_multiply_by_scalar);
  870. env.load_module_function("Quaternion", "length", quaternion_length);
  871. env.load_module_function("Quaternion", "normalize", quaternion_normalize);
  872. env.load_module_function("Quaternion", "conjugate", quaternion_conjugate);
  873. env.load_module_function("Quaternion", "inverse", quaternion_inverse);
  874. env.load_module_function("Quaternion", "power", quaternion_power);
  875. env.load_module_function("Quaternion", "elements", quaternion_elements);
  876. env.load_module_constructor("Quaternion", quaternion_ctor);
  877. env.load_module_function("QuaternionBox", "new", quaternionbox_new);
  878. env.load_module_function("QuaternionBox", "store", quaternionbox_store);
  879. env.load_module_function("QuaternionBox", "unbox", quaternionbox_unbox);
  880. env.load_module_function("QuaternionBox", "__tostring", quaternionbox_tostring);
  881. env.load_module_constructor("QuaternionBox", quaternionbox_ctor);
  882. env.load_module_function("Color4", "new", color4_new);
  883. env.load_module_constructor("Color4", color4_ctor);
  884. env.load_module_function("Lightuserdata_mt", "__add", lightuserdata_add);
  885. env.load_module_function("Lightuserdata_mt", "__sub", lightuserdata_sub);
  886. env.load_module_function("Lightuserdata_mt", "__mul", lightuserdata_mul);
  887. env.load_module_function("Lightuserdata_mt", "__div", lightuserdata_div);
  888. env.load_module_function("Lightuserdata_mt", "__unm", lightuserdata_unm);
  889. env.load_module_function("Lightuserdata_mt", "__index", lightuserdata_index);
  890. env.load_module_function("Lightuserdata_mt", "__newindex", lightuserdata_newindex);
  891. }
  892. } // namespace crown