lua_math.cpp 23 KB

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