PikaMath-test.cpp 7.7 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289
  1. #include "test_common.h"
  2. #if PIKA_SYNTAX_LEVEL == PIKA_SYNTAX_LEVEL_MAXIMAL
  3. TEST(PikaMath, test1) {
  4. /* init */
  5. pikaMemInfo.heapUsedMax = 0;
  6. PikaObj* pikaMain = newRootObj("pikaMain", New_PikaMain);
  7. /* run */
  8. __platform_printf("BEGIN\r\n");
  9. pikaVM_runSingleFile(pikaMain, "test/python/PikaMath/Quaternion_test.py");
  10. /* collect */
  11. /* assert */
  12. /* deinit */
  13. obj_deinit(pikaMain);
  14. EXPECT_EQ(pikaMemNow(), 0);
  15. }
  16. TEST(PikaMath, quat_add) {
  17. /* init */
  18. pikaMemInfo.heapUsedMax = 0;
  19. PikaObj* pikaMain = newRootObj("pikaMain", New_PikaMain);
  20. /* run */
  21. obj_run(pikaMain,
  22. "import PikaMath\n"
  23. "a = PikaMath.Quaternion()\n"
  24. "b = PikaMath.Quaternion()\n"
  25. "a.add(b)\n"
  26. "res1=a.get(0)\n"
  27. "res2=a.get(1)\n"
  28. "res3=a.get(2)\n"
  29. "res4=a.get(3)\n");
  30. /* collect */
  31. float res1 = obj_getFloat(pikaMain, "res1");
  32. float res2 = obj_getFloat(pikaMain, "res2");
  33. float res3 = obj_getFloat(pikaMain, "res3");
  34. float res4 = obj_getFloat(pikaMain, "res4");
  35. /* assert */
  36. EXPECT_FLOAT_EQ(res1, 0.0);
  37. EXPECT_FLOAT_EQ(res2, 0.0);
  38. EXPECT_FLOAT_EQ(res3, 0.0);
  39. EXPECT_FLOAT_EQ(res4, 2.0);
  40. /* deinit */
  41. obj_deinit(pikaMain);
  42. EXPECT_EQ(pikaMemNow(), 0);
  43. }
  44. TEST(PikaMath, quat_sub) {
  45. /* init */
  46. pikaMemInfo.heapUsedMax = 0;
  47. PikaObj* pikaMain = newRootObj("pikaMain", New_PikaMain);
  48. /* run */
  49. obj_run(pikaMain,
  50. "import PikaMath\n"
  51. "a = PikaMath.Quaternion()\n"
  52. "b = PikaMath.Quaternion()\n"
  53. "a.sub(b)\n"
  54. "res1=a.get(0)\n"
  55. "res2=a.get(1)\n"
  56. "res3=a.get(2)\n"
  57. "res4=a.get(3)\n");
  58. /* collect */
  59. float res1 = obj_getFloat(pikaMain, "res1");
  60. float res2 = obj_getFloat(pikaMain, "res2");
  61. float res3 = obj_getFloat(pikaMain, "res3");
  62. float res4 = obj_getFloat(pikaMain, "res4");
  63. /* assert */
  64. EXPECT_FLOAT_EQ(res1, 0.0);
  65. EXPECT_FLOAT_EQ(res2, 0.0);
  66. EXPECT_FLOAT_EQ(res3, 0.0);
  67. EXPECT_FLOAT_EQ(res4, 0.0);
  68. /* deinit */
  69. obj_deinit(pikaMain);
  70. EXPECT_EQ(pikaMemNow(), 0);
  71. }
  72. TEST(PikaMath, quat_mul) {
  73. /* init */
  74. pikaMemInfo.heapUsedMax = 0;
  75. PikaObj* pikaMain = newRootObj("pikaMain", New_PikaMain);
  76. /* run */
  77. obj_run(pikaMain,
  78. "import PikaMath\n"
  79. "a = PikaMath.Quaternion()\n"
  80. "b = PikaMath.Quaternion()\n"
  81. "a.mul(b)\n"
  82. "res1=a.get(0)\n"
  83. "res2=a.get(1)\n"
  84. "res3=a.get(2)\n"
  85. "res4=a.get(3)\n");
  86. /* collect */
  87. float res1 = obj_getFloat(pikaMain, "res1");
  88. float res2 = obj_getFloat(pikaMain, "res2");
  89. float res3 = obj_getFloat(pikaMain, "res3");
  90. float res4 = obj_getFloat(pikaMain, "res4");
  91. /* assert */
  92. EXPECT_FLOAT_EQ(res1, 0.0);
  93. EXPECT_FLOAT_EQ(res2, 0.0);
  94. EXPECT_FLOAT_EQ(res3, 0.0);
  95. EXPECT_FLOAT_EQ(res4, 1.0);
  96. /* deinit */
  97. obj_deinit(pikaMain);
  98. EXPECT_EQ(pikaMemNow(), 0);
  99. }
  100. TEST(PikaMath, quat_set) {
  101. /* init */
  102. pikaMemInfo.heapUsedMax = 0;
  103. PikaObj* pikaMain = newRootObj("pikaMain", New_PikaMain);
  104. /* run */
  105. obj_run(pikaMain,
  106. "import PikaMath\n"
  107. "a = PikaMath.Quaternion()\n"
  108. "a.set(1,1,1,1)\n"
  109. "res1=a.get(0)\n"
  110. "res2=a.get(1)\n"
  111. "res3=a.get(2)\n"
  112. "res4=a.get(3)\n");
  113. /* collect */
  114. float res1 = obj_getFloat(pikaMain, "res1");
  115. float res2 = obj_getFloat(pikaMain, "res2");
  116. float res3 = obj_getFloat(pikaMain, "res3");
  117. float res4 = obj_getFloat(pikaMain, "res4");
  118. /* assert */
  119. EXPECT_FLOAT_EQ(res1, 1);
  120. EXPECT_FLOAT_EQ(res2, 1);
  121. EXPECT_FLOAT_EQ(res3, 1);
  122. EXPECT_FLOAT_EQ(res4, 1);
  123. /* deinit */
  124. obj_deinit(pikaMain);
  125. EXPECT_EQ(pikaMemNow(), 0);
  126. }
  127. TEST(PikaMath, quat_dot) {
  128. /* init */
  129. pikaMemInfo.heapUsedMax = 0;
  130. PikaObj* pikaMain = newRootObj("pikaMain", New_PikaMain);
  131. /* run */
  132. obj_run(pikaMain,
  133. "import PikaMath\n"
  134. "a = PikaMath.Quaternion()\n"
  135. "b = PikaMath.Quaternion()\n"
  136. "a.set(1,2,3,4)\n"
  137. "b.set(5,6,7,8)\n"
  138. "res1=a.dot(b)\n");
  139. /* collect */
  140. float res1 = obj_getFloat(pikaMain, "res1");
  141. /* assert */
  142. EXPECT_FLOAT_EQ(res1, 70.0);
  143. /* deinit */
  144. obj_deinit(pikaMain);
  145. EXPECT_EQ(pikaMemNow(), 0);
  146. }
  147. TEST(PikaMath, quat_inv) {
  148. /* init */
  149. pikaMemInfo.heapUsedMax = 0;
  150. PikaObj* pikaMain = newRootObj("pikaMain", New_PikaMain);
  151. /* run */
  152. obj_run(pikaMain,
  153. "import PikaMath\n"
  154. "a = PikaMath.Quaternion()\n"
  155. "a.inverse()\n"
  156. "res1=a.get(0)\n"
  157. "res2=a.get(1)\n"
  158. "res3=a.get(2)\n"
  159. "res4=a.get(3)\n");
  160. /* collect */
  161. float res1 = obj_getFloat(pikaMain, "res1");
  162. float res2 = obj_getFloat(pikaMain, "res2");
  163. float res3 = obj_getFloat(pikaMain, "res3");
  164. float res4 = obj_getFloat(pikaMain, "res4");
  165. /* assert */
  166. EXPECT_FLOAT_EQ(res1, 0.0);
  167. EXPECT_FLOAT_EQ(res2, 0.0);
  168. EXPECT_FLOAT_EQ(res3, 0.0);
  169. EXPECT_FLOAT_EQ(res4, 1.0);
  170. /* deinit */
  171. obj_deinit(pikaMain);
  172. EXPECT_EQ(pikaMemNow(), 0);
  173. }
  174. TEST(PikaMath, quat_cross) {
  175. /* init */
  176. pikaMemInfo.heapUsedMax = 0;
  177. PikaObj* pikaMain = newRootObj("pikaMain", New_PikaMain);
  178. /* run */
  179. obj_run(pikaMain,
  180. "import PikaMath\n"
  181. "a = PikaMath.Quaternion()\n"
  182. "b = PikaMath.Quaternion()\n"
  183. "a.set(2,3,4,1)\n"
  184. "b.set(6,7,8,5)\n"
  185. "a.crossproduct(b)\n"
  186. "res1=a.get(0)\n"
  187. "res2=a.get(1)\n"
  188. "res3=a.get(2)\n"
  189. "res4=a.get(3)\n");
  190. /* collect */
  191. float res1 = obj_getFloat(pikaMain, "res1");
  192. float res2 = obj_getFloat(pikaMain, "res2");
  193. float res3 = obj_getFloat(pikaMain, "res3");
  194. float res4 = obj_getFloat(pikaMain, "res4");
  195. /* assert */
  196. EXPECT_FLOAT_EQ(res1, -4.0);
  197. EXPECT_FLOAT_EQ(res2, 8.0);
  198. EXPECT_FLOAT_EQ(res3, -4.0);
  199. EXPECT_FLOAT_EQ(res4, 0.0);
  200. /* deinit */
  201. obj_deinit(pikaMain);
  202. EXPECT_EQ(pikaMemNow(), 0);
  203. }
  204. TEST(PikaMath, quat_toEuler) {
  205. /* init */
  206. pikaMemInfo.heapUsedMax = 0;
  207. PikaObj* pikaMain = newRootObj("pikaMain", New_PikaMain);
  208. /* run */
  209. obj_run(pikaMain,
  210. "import PikaMath\n"
  211. "a = PikaMath.Quaternion()\n"
  212. "a.set(0,0,1,0)\n"
  213. "res=a.toEuler()\n"
  214. "res1=res[0]\n"
  215. "res2=res[1]\n"
  216. "res3=res[2]\n");
  217. /* collect */
  218. float res1 = obj_getFloat(pikaMain, "res1");
  219. float res2 = obj_getFloat(pikaMain, "res2");
  220. float res3 = obj_getFloat(pikaMain, "res3");
  221. /* assert */
  222. EXPECT_FLOAT_EQ(res1, 0.0);
  223. EXPECT_FLOAT_EQ(res2, 0.0);
  224. EXPECT_FLOAT_EQ(res3, 3.1415927);
  225. /* deinit */
  226. obj_deinit(pikaMain);
  227. EXPECT_EQ(pikaMemNow(), 0);
  228. }
  229. TEST(PikaMath, quat_fromEuler) {
  230. /* init */
  231. pikaMemInfo.heapUsedMax = 0;
  232. PikaObj* pikaMain = newRootObj("pikaMain", New_PikaMain);
  233. /* run */
  234. obj_run(pikaMain,
  235. "import PikaMath\n"
  236. "a = PikaMath.Quaternion()\n"
  237. "a.fromEuler(0,0.0,0.0,3.1415927)\n"
  238. "res1=a.get(0)\n"
  239. "res2=a.get(1)\n"
  240. "res3=a.get(2)\n"
  241. "res4=a.get(3)\n");
  242. /* collect */
  243. float res1 = obj_getFloat(pikaMain, "res1");
  244. float res2 = obj_getFloat(pikaMain, "res2");
  245. float res3 = obj_getFloat(pikaMain, "res3");
  246. float res4 = obj_getFloat(pikaMain, "res4");
  247. /* assert */
  248. EXPECT_FLOAT_EQ(res1, 0.0);
  249. EXPECT_FLOAT_EQ(res2, 0.0);
  250. EXPECT_FLOAT_EQ(res3, 0.0);
  251. EXPECT_FLOAT_EQ(res4, 1.0);
  252. /* deinit */
  253. obj_deinit(pikaMain);
  254. EXPECT_EQ(pikaMemNow(), 0);
  255. }
  256. #endif