Matrix3x3Tests.cpp 26 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592
  1. /*
  2. * Copyright (c) Contributors to the Open 3D Engine Project.
  3. * For complete copyright and license terms please see the LICENSE at the root of this distribution.
  4. *
  5. * SPDX-License-Identifier: Apache-2.0 OR MIT
  6. *
  7. */
  8. #include <AzCore/Math/Matrix3x3.h>
  9. #include <AzCore/Math/Transform.h>
  10. #include <AzCore/Math/Quaternion.h>
  11. #include <AzCore/UnitTest/TestTypes.h>
  12. #include <AZTestShared/Math/MathTestHelpers.h>
  13. using namespace AZ;
  14. namespace UnitTest
  15. {
  16. static constexpr int32_t testIndices3x3[] = { 0, 1, 2, 3 };
  17. float testFloats[] = { 1.0f, 2.0f, 3.0f, 4.0f, 5.0f, 6.0f, 7.0f, 8.0f, 9.0f };
  18. float testFloatsMtx[9];
  19. TEST(MATH_Matrix3x3, TestCreateIdentity)
  20. {
  21. Matrix3x3 m1 = Matrix3x3::CreateIdentity();
  22. EXPECT_THAT(m1.GetRow(0), IsCloseTolerance(Vector3(1.0f, 0.0f, 0.0f), 1e-6f));
  23. EXPECT_THAT(m1.GetRow(1), IsCloseTolerance(Vector3(0.0f, 1.0f, 0.0f), 1e-6f));
  24. EXPECT_THAT(m1.GetRow(2), IsCloseTolerance(Vector3(0.0f, 0.0f, 1.0f), 1e-6f));
  25. }
  26. TEST(MATH_Matrix3x3, TestCreateZero)
  27. {
  28. Matrix3x3 m1 = Matrix3x3::CreateZero();
  29. EXPECT_THAT(m1.GetRow(0), IsCloseTolerance(Vector3(0.0f), 1e-6f));
  30. EXPECT_THAT(m1.GetRow(1), IsCloseTolerance(Vector3(0.0f), 1e-6f));
  31. EXPECT_THAT(m1.GetRow(2), IsCloseTolerance(Vector3(0.0f), 1e-6f));
  32. }
  33. TEST(MATH_Matrix3x3, TestCreateFromValue)
  34. {
  35. Matrix3x3 m1 = Matrix3x3::CreateFromValue(2.0f);
  36. EXPECT_THAT(m1.GetRow(0), IsCloseTolerance(Vector3(2.0f), 1e-6f));
  37. EXPECT_THAT(m1.GetRow(1), IsCloseTolerance(Vector3(2.0f), 1e-6f));
  38. EXPECT_THAT(m1.GetRow(2), IsCloseTolerance(Vector3(2.0f), 1e-6f));
  39. }
  40. TEST(MATH_Matrix3x3, TestCreateFromRowMajorFloat9)
  41. {
  42. Matrix3x3 m1 = Matrix3x3::CreateFromRowMajorFloat9(testFloats);
  43. EXPECT_THAT(m1.GetRow(0), IsCloseTolerance(Vector3(1.0f, 2.0f, 3.0f), 1e-6f));
  44. EXPECT_THAT(m1.GetRow(1), IsCloseTolerance(Vector3(4.0f, 5.0f, 6.0f), 1e-6f));
  45. EXPECT_THAT(m1.GetRow(2), IsCloseTolerance(Vector3(7.0f, 8.0f, 9.0f), 1e-6f));
  46. m1.StoreToRowMajorFloat9(testFloatsMtx);
  47. EXPECT_EQ(memcmp(testFloatsMtx, testFloats, sizeof(testFloatsMtx)), 0);
  48. }
  49. TEST(MATH_Matrix3x3, TestCreateFromColumnMajorFloat9)
  50. {
  51. Matrix3x3 m1 = Matrix3x3::CreateFromColumnMajorFloat9(testFloats);
  52. EXPECT_THAT(m1.GetRow(0), IsCloseTolerance(Vector3(1.0f, 4.0f, 7.0f), 1e-6f));
  53. EXPECT_THAT(m1.GetRow(1), IsCloseTolerance(Vector3(2.0f, 5.0f, 8.0f), 1e-6f));
  54. EXPECT_THAT(m1.GetRow(2), IsCloseTolerance(Vector3(3.0f, 6.0f, 9.0f), 1e-6f));
  55. m1.StoreToColumnMajorFloat9(testFloatsMtx);
  56. EXPECT_EQ(memcmp(testFloatsMtx, testFloats, sizeof(testFloatsMtx)), 0);
  57. }
  58. TEST(MATH_Matrix3x3, TestStoreToRowMajorFloat11)
  59. {
  60. Matrix3x3 m1 = Matrix3x3::CreateFromRowMajorFloat9(testFloats);
  61. float output[12];
  62. output[11] = -1;
  63. // we expect the first two rows to be written with 4 floats, with the last value as garbage,
  64. // but the last row only with 3 floats, such that the final value remains untouched
  65. // clang-format off
  66. float expected[12] = {
  67. testFloats[0], testFloats[1], testFloats[2], 0.0f,
  68. testFloats[3], testFloats[4], testFloats[5], 0.0f,
  69. testFloats[6], testFloats[7], testFloats[8], -1.0f};
  70. // clang-format on
  71. m1.StoreToRowMajorFloat11(output);
  72. EXPECT_EQ(memcmp(&expected[0], &output[0], sizeof(float) * 3), 0);
  73. EXPECT_EQ(memcmp(&expected[3], &output[3], sizeof(float) * 3), 0);
  74. EXPECT_EQ(memcmp(&expected[6], &output[6], sizeof(float) * 3), 0);
  75. EXPECT_EQ(output[11], -1);
  76. }
  77. TEST(MATH_Matrix3x3, TestStoreToColumnMajorFloat11)
  78. {
  79. Matrix3x3 m1 = Matrix3x3::CreateFromRowMajorFloat9(testFloats);
  80. float output[12];
  81. output[11] = -1;
  82. // we expect the first two columns to be written with 4 floats, with the last value as garbage,
  83. // but the last column only with 3 floats, such that the final value remains untouched
  84. // clang-format off
  85. float expected[12] = {
  86. testFloats[0], testFloats[3], testFloats[6], 0.0f,
  87. testFloats[1], testFloats[4], testFloats[7], 0.0f,
  88. testFloats[2], testFloats[5], testFloats[8], -1.0f};
  89. // clang-format on
  90. m1.StoreToColumnMajorFloat11(output);
  91. EXPECT_EQ(memcmp(&expected[0], &output[0], sizeof(float) * 3), 0);
  92. EXPECT_EQ(memcmp(&expected[3], &output[3], sizeof(float) * 3), 0);
  93. EXPECT_EQ(memcmp(&expected[6], &output[6], sizeof(float) * 3), 0);
  94. EXPECT_EQ(output[11], -1);
  95. }
  96. TEST(MATH_Matrix3x3, TestCreateRotationX)
  97. {
  98. Matrix3x3 m1 = Matrix3x3::CreateRotationX(DegToRad(30.0f));
  99. EXPECT_THAT(m1.GetRow(0), IsClose(Vector3(1.0f, 0.0f, 0.0f)));
  100. EXPECT_THAT(m1.GetRow(1), IsClose(Vector3(0.0f, 0.866f, -0.5f)));
  101. EXPECT_THAT(m1.GetRow(2), IsClose(Vector3(0.0f, 0.5f, 0.866f)));
  102. }
  103. TEST(MATH_Matrix3x3, TestCreateRotationY)
  104. {
  105. Matrix3x3 m1 = Matrix3x3::CreateRotationY(DegToRad(30.0f));
  106. EXPECT_THAT(m1.GetRow(0), IsClose(Vector3(0.866f, 0.0f, 0.5f)));
  107. EXPECT_THAT(m1.GetRow(1), IsClose(Vector3(0.0f, 1.0f, 0.0f)));
  108. EXPECT_THAT(m1.GetRow(2), IsClose(Vector3(-0.5f, 0.0f, 0.866f)));
  109. }
  110. TEST(MATH_Matrix3x3, TestCreateRotationZ)
  111. {
  112. Matrix3x3 m1 = Matrix3x3::CreateRotationZ(DegToRad(30.0f));
  113. EXPECT_THAT(m1.GetRow(0), IsClose(Vector3(0.866f, -0.5f, 0.0f)));
  114. EXPECT_THAT(m1.GetRow(1), IsClose(Vector3(0.5f, 0.866f, 0.0f)));
  115. EXPECT_THAT(m1.GetRow(2), IsClose(Vector3(0.0f, 0.0f, 1.0f)));
  116. }
  117. TEST(MATH_Matrix3x3, TestCreateFromTransform)
  118. {
  119. Matrix3x3 m1 = Matrix3x3::CreateFromTransform(Transform::CreateRotationX(DegToRad(30.0f)));
  120. EXPECT_THAT(m1.GetRow(0), IsClose(Vector3(1.0f, 0.0f, 0.0f)));
  121. EXPECT_THAT(m1.GetRow(1), IsClose(Vector3(0.0f, 0.866f, -0.5f)));
  122. EXPECT_THAT(m1.GetRow(2), IsClose(Vector3(0.0f, 0.5f, 0.866f)));
  123. }
  124. TEST(MATH_Matrix3x3, TestCreateFromMatrix4x4)
  125. {
  126. Matrix3x3 m1 = Matrix3x3::CreateFromMatrix4x4(Matrix4x4::CreateRotationX(DegToRad(30.0f)));
  127. EXPECT_THAT(m1.GetRow(0), IsClose(Vector3(1.0f, 0.0f, 0.0f)));
  128. EXPECT_THAT(m1.GetRow(1), IsClose(Vector3(0.0f, 0.866f, -0.5f)));
  129. EXPECT_THAT(m1.GetRow(2), IsClose(Vector3(0.0f, 0.5f, 0.866f)));
  130. }
  131. TEST(MATH_Matrix3x3, TestCreateFromQuaternion)
  132. {
  133. Matrix3x3 m1 = Matrix3x3::CreateFromQuaternion(AZ::Quaternion::CreateRotationX(DegToRad(30.0f)));
  134. EXPECT_THAT(m1.GetRow(0), IsClose(Vector3(1.0f, 0.0f, 0.0f)));
  135. EXPECT_THAT(m1.GetRow(1), IsClose(Vector3(0.0f, 0.866f, -0.5f)));
  136. EXPECT_THAT(m1.GetRow(2), IsClose(Vector3(0.0f, 0.5f, 0.866f)));
  137. }
  138. TEST(MATH_Matrix3x3, TestCreateScale)
  139. {
  140. Matrix3x3 m1 = Matrix3x3::CreateScale(Vector3(1.0f, 2.0f, 3.0f));
  141. EXPECT_THAT(m1.GetRow(0), IsCloseTolerance(Vector3(1.0f, 0.0f, 0.0f), 1e-6f));
  142. EXPECT_THAT(m1.GetRow(1), IsCloseTolerance(Vector3(0.0f, 2.0f, 0.0f), 1e-6f));
  143. EXPECT_THAT(m1.GetRow(2), IsCloseTolerance(Vector3(0.0f, 0.0f, 3.0f), 1e-6f));
  144. }
  145. TEST(MATH_Matrix3x3, TestCreateDiagonal)
  146. {
  147. Matrix3x3 m1 = Matrix3x3::CreateDiagonal(Vector3(2.0f, 3.0f, 4.0f));
  148. EXPECT_THAT(m1.GetRow(0), IsCloseTolerance(Vector3(2.0f, 0.0f, 0.0f), 1e-6f));
  149. EXPECT_THAT(m1.GetRow(1), IsCloseTolerance(Vector3(0.0f, 3.0f, 0.0f), 1e-6f));
  150. EXPECT_THAT(m1.GetRow(2), IsCloseTolerance(Vector3(0.0f, 0.0f, 4.0f), 1e-6f));
  151. }
  152. TEST(MATH_Matrix3x3, TestCreateCrossProduct)
  153. {
  154. Matrix3x3 m1 = Matrix3x3::CreateCrossProduct(Vector3(1.0f, 2.0f, 3.0f));
  155. EXPECT_THAT(m1.GetRow(0), IsCloseTolerance(Vector3(0.0f, -3.0f, 2.0f), 1e-6f));
  156. EXPECT_THAT(m1.GetRow(1), IsCloseTolerance(Vector3(3.0f, 0.0f, -1.0f), 1e-6f));
  157. EXPECT_THAT(m1.GetRow(2), IsCloseTolerance(Vector3(-2.0f, 1.0f, 0.0f), 1e-6f));
  158. }
  159. TEST(MATH_Matrix3x3, TestElementAccess)
  160. {
  161. Matrix3x3 m1 = Matrix3x3::CreateRotationX(DegToRad(30.0f));
  162. EXPECT_NEAR(m1.GetElement(1, 2), -0.5f, 2e-3f);
  163. EXPECT_NEAR(m1.GetElement(2, 2), 0.866f, 2e-3f);
  164. m1.SetElement(2, 1, 5.0f);
  165. EXPECT_NEAR(m1.GetElement(2, 1), 5.0f, 1e-6f);
  166. EXPECT_NEAR(m1(1, 2), -0.5f, 2e-3f);
  167. EXPECT_NEAR(m1(2, 2), 0.866f, 2e-3f);
  168. m1.SetElement(2, 1, 15.0f);
  169. EXPECT_NEAR(m1(2, 1), 15.0f, 1e-6f);
  170. }
  171. TEST(MATH_Matrix3x3, TestRowAccess)
  172. {
  173. Matrix3x3 m1 = Matrix3x3::CreateRotationX(DegToRad(30.0f));
  174. EXPECT_THAT(m1.GetRow(2), IsClose(Vector3(0.0f, 0.5f, 0.866f)));
  175. m1.SetRow(0, 1.0f, 2.0f, 3.0f);
  176. EXPECT_THAT(m1.GetRow(0), IsClose(Vector3(1.0f, 2.0f, 3.0f)));
  177. m1.SetRow(1, Vector3(4.0f, 5.0f, 6.0f));
  178. EXPECT_THAT(m1.GetRow(1), IsClose(Vector3(4.0f, 5.0f, 6.0f)));
  179. m1.SetRow(2, Vector3(7.0f, 8.0f, 9.0f));
  180. EXPECT_THAT(m1.GetRow(2), IsClose(Vector3(7.0f, 8.0f, 9.0f)));
  181. // test GetRow with non-constant, we have different implementations for constants and variables
  182. EXPECT_THAT(m1.GetRow(testIndices3x3[0]), IsClose(Vector3(1.0f, 2.0f, 3.0f)));
  183. EXPECT_THAT(m1.GetRow(testIndices3x3[1]), IsClose(Vector3(4.0f, 5.0f, 6.0f)));
  184. EXPECT_THAT(m1.GetRow(testIndices3x3[2]), IsClose(Vector3(7.0f, 8.0f, 9.0f)));
  185. Vector3 row0, row1, row2;
  186. m1.GetRows(&row0, &row1, &row2);
  187. EXPECT_THAT(row0, IsClose(Vector3(1.0f, 2.0f, 3.0f)));
  188. EXPECT_THAT(row1, IsClose(Vector3(4.0f, 5.0f, 6.0f)));
  189. EXPECT_THAT(row2, IsClose(Vector3(7.0f, 8.0f, 9.0f)));
  190. m1.SetRows(Vector3(10.0f, 11.0f, 12.0f), Vector3(13.0f, 14.0f, 15.0f), Vector3(16.0f, 17.0f, 18.0f));
  191. EXPECT_THAT(m1.GetRow(0), IsClose(Vector3(10.0f, 11.0f, 12.0f)));
  192. EXPECT_THAT(m1.GetRow(1), IsClose(Vector3(13.0f, 14.0f, 15.0f)));
  193. EXPECT_THAT(m1.GetRow(2), IsClose(Vector3(16.0f, 17.0f, 18.0f)));
  194. }
  195. TEST(MATH_Matrix3x3, TestColumnAccess)
  196. {
  197. Matrix3x3 m1 = Matrix3x3::CreateRotationX(DegToRad(30.0f));
  198. EXPECT_THAT(m1.GetColumn(1), IsClose(Vector3(0.0f, 0.866f, 0.5f)));
  199. m1.SetColumn(2, 1.0f, 2.0f, 3.0f);
  200. EXPECT_THAT(m1.GetColumn(2), IsClose(Vector3(1.0f, 2.0f, 3.0f)));
  201. EXPECT_THAT(m1.GetRow(0), IsClose(Vector3(1.0f, 0.0f, 1.0f))); // checking all components in case others get messed up with the shuffling
  202. EXPECT_THAT(m1.GetRow(1), IsClose(Vector3(0.0f, 0.866f, 2.0f)));
  203. EXPECT_THAT(m1.GetRow(2), IsClose(Vector3(0.0f, 0.5f, 3.0f)));
  204. m1.SetColumn(0, Vector3(2.0f, 3.0f, 4.0f));
  205. EXPECT_THAT(m1.GetColumn(0), IsClose(Vector3(2.0f, 3.0f, 4.0f)));
  206. EXPECT_THAT(m1.GetRow(0), IsClose(Vector3(2.0f, 0.0f, 1.0f)));
  207. EXPECT_THAT(m1.GetRow(1), IsClose(Vector3(3.0f, 0.866f, 2.0f)));
  208. EXPECT_THAT(m1.GetRow(2), IsClose(Vector3(4.0f, 0.5f, 3.0f)));
  209. // test GetColumn with non-constant, we have different implementations for constants and variables
  210. EXPECT_THAT(m1.GetColumn(testIndices3x3[0]), IsClose(Vector3(2.0f, 3.0f, 4.0f)));
  211. EXPECT_THAT(m1.GetColumn(testIndices3x3[1]), IsClose(Vector3(0.0f, 0.866f, 0.5f)));
  212. EXPECT_THAT(m1.GetColumn(testIndices3x3[2]), IsClose(Vector3(1.0f, 2.0f, 3.0f)));
  213. Vector3 col0, col1, col2;
  214. m1.GetColumns(&col0, &col1, &col2);
  215. EXPECT_THAT(col0, IsClose(Vector3(2.0f, 3.0f, 4.0f)));
  216. EXPECT_THAT(col1, IsClose(Vector3(0.0f, 0.866f, 0.5f)));
  217. EXPECT_THAT(col2, IsClose(Vector3(1.0f, 2.0f, 3.0f)));
  218. m1.SetColumns(Vector3(10.0f, 11.0f, 12.0f), Vector3(13.0f, 14.0f, 15.0f), Vector3(16.0f, 17.0f, 18.0f));
  219. EXPECT_THAT(m1.GetColumn(0), IsClose(Vector3(10.0f, 11.0f, 12.0f)));
  220. EXPECT_THAT(m1.GetColumn(1), IsClose(Vector3(13.0f, 14.0f, 15.0f)));
  221. EXPECT_THAT(m1.GetColumn(2), IsClose(Vector3(16.0f, 17.0f, 18.0f)));
  222. }
  223. TEST(MATH_Matrix3x3, TestBasisAccess)
  224. {
  225. Matrix3x3 m1 = Matrix3x3::CreateRotationX(DegToRad(30.0f));
  226. EXPECT_THAT(m1.GetBasisX(), IsClose(Vector3(1.0f, 0.0f, 0.0f)));
  227. EXPECT_THAT(m1.GetBasisY(), IsClose(Vector3(0.0f, 0.866f, 0.5f)));
  228. EXPECT_THAT(m1.GetBasisZ(), IsClose(Vector3(0.0f, -0.5f, 0.866f)));
  229. m1.SetBasisX(1.0f, 2.0f, 3.0f);
  230. EXPECT_THAT(m1.GetBasisX(), IsClose(Vector3(1.0f, 2.0f, 3.0f)));
  231. m1.SetBasisY(4.0f, 5.0f, 6.0f);
  232. EXPECT_THAT(m1.GetBasisY(), IsClose(Vector3(4.0f, 5.0f, 6.0f)));
  233. m1.SetBasisZ(7.0f, 8.0f, 9.0f);
  234. EXPECT_THAT(m1.GetBasisZ(), IsClose(Vector3(7.0f, 8.0f, 9.0f)));
  235. m1.SetBasisX(Vector3(10.0f, 11.0f, 12.0f));
  236. EXPECT_THAT(m1.GetBasisX(), IsClose(Vector3(10.0f, 11.0f, 12.0f)));
  237. m1.SetBasisY(Vector3(13.0f, 14.0f, 15.0f));
  238. EXPECT_THAT(m1.GetBasisY(), IsClose(Vector3(13.0f, 14.0f, 15.0f)));
  239. m1.SetBasisZ(Vector3(16.0f, 17.0f, 18.0f));
  240. EXPECT_THAT(m1.GetBasisZ(), IsClose(Vector3(16.0f, 17.0f, 18.0f)));
  241. Vector3 basis0, basis1, basis2;
  242. m1.GetBasis(&basis0, &basis1, &basis2);
  243. EXPECT_THAT(basis0, IsClose(Vector3(10.0f, 11.0f, 12.0f)));
  244. EXPECT_THAT(basis1, IsClose(Vector3(13.0f, 14.0f, 15.0f)));
  245. EXPECT_THAT(basis2, IsClose(Vector3(16.0f, 17.0f, 18.0f)));
  246. m1.SetBasis(Vector3(1.0f, 2.0f, 3.0f), Vector3(4.0f, 5.0f, 6.0f), Vector3(7.0f, 8.0f, 9.0f));
  247. EXPECT_THAT(m1.GetBasisX(), IsClose(Vector3(1.0f, 2.0f, 3.0f)));
  248. EXPECT_THAT(m1.GetBasisY(), IsClose(Vector3(4.0f, 5.0f, 6.0f)));
  249. EXPECT_THAT(m1.GetBasisZ(), IsClose(Vector3(7.0f, 8.0f, 9.0f)));
  250. }
  251. TEST(MATH_Matrix3x3, TestMatrixMultiplication)
  252. {
  253. Matrix3x3 m1;
  254. m1.SetRow(0, 1.0f, 2.0f, 3.0f);
  255. m1.SetRow(1, 4.0f, 5.0f, 6.0f);
  256. m1.SetRow(2, 7.0f, 8.0f, 9.0f);
  257. Matrix3x3 m2;
  258. m2.SetRow(0, 7.0f, 8.0f, 9.0f);
  259. m2.SetRow(1, 10.0f, 11.0f, 12.0f);
  260. m2.SetRow(2, 13.0f, 14.0f, 15.0f);
  261. Matrix3x3 m3 = m1 * m2;
  262. EXPECT_THAT(m3.GetRow(0), IsClose(Vector3(66.0f, 72.0f, 78.0f)));
  263. EXPECT_THAT(m3.GetRow(1), IsClose(Vector3(156.0f, 171.0f, 186.0f)));
  264. EXPECT_THAT(m3.GetRow(2), IsClose(Vector3(246.0f, 270.0f, 294.0f)));
  265. Matrix3x3 m4 = m1;
  266. m4 *= m2;
  267. EXPECT_THAT(m4.GetRow(0), IsClose(Vector3(66.0f, 72.0f, 78.0f)));
  268. EXPECT_THAT(m4.GetRow(1), IsClose(Vector3(156.0f, 171.0f, 186.0f)));
  269. EXPECT_THAT(m4.GetRow(2), IsClose(Vector3(246.0f, 270.0f, 294.0f)));
  270. m3 = m1.TransposedMultiply(m2);
  271. EXPECT_THAT(m3.GetRow(0), IsClose(Vector3(138.0f, 150.0f, 162.0f)));
  272. EXPECT_THAT(m3.GetRow(1), IsClose(Vector3(168.0f, 183.0f, 198.0f)));
  273. EXPECT_THAT(m3.GetRow(2), IsClose(Vector3(198.0f, 216.0f, 234.0f)));
  274. }
  275. TEST(MATH_Matrix3x3, TestVectorMultiplication)
  276. {
  277. Matrix3x3 m1;
  278. m1.SetRow(0, 1.0f, 2.0f, 3.0f);
  279. m1.SetRow(1, 4.0f, 5.0f, 6.0f);
  280. m1.SetRow(2, 7.0f, 8.0f, 9.0f);
  281. Matrix3x3 m2;
  282. m2.SetRow(0, 7.0f, 8.0f, 9.0f);
  283. m2.SetRow(1, 10.0f, 11.0f, 12.0f);
  284. m2.SetRow(2, 13.0f, 14.0f, 15.0f);
  285. EXPECT_THAT((m1 * Vector3(1.0f, 2.0f, 3.0f)), IsClose(Vector3(14.0f, 32.0f, 50.0f)));
  286. Vector3 v1(1.0f, 2.0f, 3.0f);
  287. EXPECT_THAT((v1 * m1), IsClose(Vector3(30.0f, 36.0f, 42.0f)));
  288. v1 *= m1;
  289. EXPECT_THAT(v1, IsClose(Vector3(30.0f, 36.0f, 42.0f)));
  290. }
  291. TEST(MATH_Matrix3x3, TestSum)
  292. {
  293. Matrix3x3 m1;
  294. m1.SetRow(0, 1.0f, 2.0f, 3.0f);
  295. m1.SetRow(1, 4.0f, 5.0f, 6.0f);
  296. m1.SetRow(2, 7.0f, 8.0f, 9.0f);
  297. Matrix3x3 m2;
  298. m2.SetRow(0, 7.0f, 8.0f, 9.0f);
  299. m2.SetRow(1, 10.0f, 11.0f, 12.0f);
  300. m2.SetRow(2, 13.0f, 14.0f, 15.0f);
  301. Matrix3x3 m3 = m1 + m2;
  302. EXPECT_THAT(m3.GetRow(0), IsClose(Vector3(8.0f, 10.0f, 12.0f)));
  303. EXPECT_THAT(m3.GetRow(1), IsClose(Vector3(14.0f, 16.0f, 18.0f)));
  304. EXPECT_THAT(m3.GetRow(2), IsClose(Vector3(20.0f, 22.0f, 24.0f)));
  305. m3 = m1;
  306. m3 += m2;
  307. EXPECT_THAT(m3.GetRow(0), IsClose(Vector3(8.0f, 10.0f, 12.0f)));
  308. EXPECT_THAT(m3.GetRow(1), IsClose(Vector3(14.0f, 16.0f, 18.0f)));
  309. EXPECT_THAT(m3.GetRow(2), IsClose(Vector3(20.0f, 22.0f, 24.0f)));
  310. }
  311. TEST(MATH_Matrix3x3, TestDifference)
  312. {
  313. Matrix3x3 m1;
  314. m1.SetRow(0, 1.0f, 2.0f, 3.0f);
  315. m1.SetRow(1, 4.0f, 5.0f, 6.0f);
  316. m1.SetRow(2, 7.0f, 8.0f, 9.0f);
  317. Matrix3x3 m2;
  318. m2.SetRow(0, 7.0f, 8.0f, 9.0f);
  319. m2.SetRow(1, 10.0f, 11.0f, 12.0f);
  320. m2.SetRow(2, 13.0f, 14.0f, 15.0f);
  321. Matrix3x3 m3 = m1 - m2;
  322. EXPECT_THAT(m3.GetRow(0), IsClose(Vector3(-6.0f, -6.0f, -6.0f)));
  323. EXPECT_THAT(m3.GetRow(1), IsClose(Vector3(-6.0f, -6.0f, -6.0f)));
  324. EXPECT_THAT(m3.GetRow(2), IsClose(Vector3(-6.0f, -6.0f, -6.0f)));
  325. m3 = m1;
  326. m3 -= m2;
  327. EXPECT_THAT(m3.GetRow(0), IsClose(Vector3(-6.0f, -6.0f, -6.0f)));
  328. EXPECT_THAT(m3.GetRow(1), IsClose(Vector3(-6.0f, -6.0f, -6.0f)));
  329. EXPECT_THAT(m3.GetRow(2), IsClose(Vector3(-6.0f, -6.0f, -6.0f)));
  330. }
  331. TEST(MATH_Matrix3x3, TestScalarMultiplication)
  332. {
  333. Matrix3x3 m1;
  334. m1.SetRow(0, 1.0f, 2.0f, 3.0f);
  335. m1.SetRow(1, 4.0f, 5.0f, 6.0f);
  336. m1.SetRow(2, 7.0f, 8.0f, 9.0f);
  337. Matrix3x3 m2;
  338. m2.SetRow(0, 7.0f, 8.0f, 9.0f);
  339. m2.SetRow(1, 10.0f, 11.0f, 12.0f);
  340. m2.SetRow(2, 13.0f, 14.0f, 15.0f);
  341. Matrix3x3 m3 = m1 * 2.0f;
  342. EXPECT_THAT(m3.GetRow(0), IsClose(Vector3(2.0f, 4.0f, 6.0f)));
  343. EXPECT_THAT(m3.GetRow(1), IsClose(Vector3(8.0f, 10.0f, 12.0f)));
  344. EXPECT_THAT(m3.GetRow(2), IsClose(Vector3(14.0f, 16.0f, 18.0f)));
  345. m3 = m1;
  346. m3 *= 2.0f;
  347. EXPECT_THAT(m3.GetRow(0), IsClose(Vector3(2.0f, 4.0f, 6.0f)));
  348. EXPECT_THAT(m3.GetRow(1), IsClose(Vector3(8.0f, 10.0f, 12.0f)));
  349. EXPECT_THAT(m3.GetRow(2), IsClose(Vector3(14.0f, 16.0f, 18.0f)));
  350. m3 = 2.0f * m1;
  351. EXPECT_THAT(m3.GetRow(0), IsClose(Vector3(2.0f, 4.0f, 6.0f)));
  352. EXPECT_THAT(m3.GetRow(1), IsClose(Vector3(8.0f, 10.0f, 12.0f)));
  353. EXPECT_THAT(m3.GetRow(2), IsClose(Vector3(14.0f, 16.0f, 18.0f)));
  354. }
  355. TEST(MATH_Matrix3x3, TestScalarDivision)
  356. {
  357. Matrix3x3 m1;
  358. m1.SetRow(0, 1.0f, 2.0f, 3.0f);
  359. m1.SetRow(1, 4.0f, 5.0f, 6.0f);
  360. m1.SetRow(2, 7.0f, 8.0f, 9.0f);
  361. Matrix3x3 m2;
  362. m2.SetRow(0, 7.0f, 8.0f, 9.0f);
  363. m2.SetRow(1, 10.0f, 11.0f, 12.0f);
  364. m2.SetRow(2, 13.0f, 14.0f, 15.0f);
  365. Matrix3x3 m3 = m1 / 0.5f;
  366. EXPECT_THAT(m3.GetRow(0), IsClose(Vector3(2.0f, 4.0f, 6.0f)));
  367. EXPECT_THAT(m3.GetRow(1), IsClose(Vector3(8.0f, 10.0f, 12.0f)));
  368. EXPECT_THAT(m3.GetRow(2), IsClose(Vector3(14.0f, 16.0f, 18.0f)));
  369. m3 = m1;
  370. m3 /= 0.5f;
  371. EXPECT_THAT(m3.GetRow(0), IsClose(Vector3(2.0f, 4.0f, 6.0f)));
  372. EXPECT_THAT(m3.GetRow(1), IsClose(Vector3(8.0f, 10.0f, 12.0f)));
  373. EXPECT_THAT(m3.GetRow(2), IsClose(Vector3(14.0f, 16.0f, 18.0f)));
  374. }
  375. TEST(MATH_Matrix3x3, TestNegation)
  376. {
  377. Matrix3x3 m1;
  378. m1.SetRow(0, 1.0f, 2.0f, 3.0f);
  379. m1.SetRow(1, 4.0f, 5.0f, 6.0f);
  380. m1.SetRow(2, 7.0f, 8.0f, 9.0f);
  381. EXPECT_THAT(-(-m1), IsClose(m1));
  382. EXPECT_THAT(-Matrix3x3::CreateZero(), IsClose(Matrix3x3::CreateZero()));
  383. Matrix3x3 m2 = -m1;
  384. EXPECT_THAT(m2.GetRow(0), IsClose(Vector3(-1.0f, -2.0f, -3.0f)));
  385. EXPECT_THAT(m2.GetRow(1), IsClose(Vector3(-4.0f, -5.0f, -6.0f)));
  386. EXPECT_THAT(m2.GetRow(2), IsClose(Vector3(-7.0f, -8.0f, -9.0f)));
  387. Matrix3x3 m3 = m1 + (-m1);
  388. EXPECT_THAT(m3, IsClose(Matrix3x3::CreateZero()));
  389. }
  390. TEST(MATH_Matrix3x3, TestTranspose)
  391. {
  392. Matrix3x3 m1;
  393. m1.SetRow(0, 1.0f, 2.0f, 3.0f);
  394. m1.SetRow(1, 4.0f, 5.0f, 6.0f);
  395. m1.SetRow(2, 7.0f, 8.0f, 9.0f);
  396. Matrix3x3 m2 = m1.GetTranspose();
  397. EXPECT_THAT(m2.GetRow(0), IsClose(Vector3(1.0f, 4.0f, 7.0f)));
  398. EXPECT_THAT(m2.GetRow(1), IsClose(Vector3(2.0f, 5.0f, 8.0f)));
  399. EXPECT_THAT(m2.GetRow(2), IsClose(Vector3(3.0f, 6.0f, 9.0f)));
  400. m2 = m1;
  401. m2.Transpose();
  402. EXPECT_THAT(m2.GetRow(0), IsClose(Vector3(1.0f, 4.0f, 7.0f)));
  403. EXPECT_THAT(m2.GetRow(1), IsClose(Vector3(2.0f, 5.0f, 8.0f)));
  404. EXPECT_THAT(m2.GetRow(2), IsClose(Vector3(3.0f, 6.0f, 9.0f)));
  405. }
  406. TEST(MATH_Matrix3x3, TestFastInverse)
  407. {
  408. // orthogonal matrix only
  409. Matrix3x3 m1 = Matrix3x3::CreateRotationX(1.0f);
  410. EXPECT_THAT((m1 * m1.GetInverseFast()), IsCloseTolerance(Matrix3x3::CreateIdentity(), 0.02f));
  411. Matrix3x3 m2 = Matrix3x3::CreateRotationZ(2.0f) * Matrix3x3::CreateRotationX(1.0f);
  412. Matrix3x3 m3 = m2.GetInverseFast();
  413. // allow a little bigger threshold, because of the 2 rot matrices (sin,cos differences)
  414. EXPECT_THAT((m2 * m3), IsCloseTolerance(Matrix3x3::CreateIdentity(), 0.1f));
  415. EXPECT_THAT(m3.GetRow(0), IsCloseTolerance(Vector3(-0.420f, 0.909f, 0.0f), 0.06f));
  416. EXPECT_THAT(m3.GetRow(1), IsCloseTolerance(Vector3(-0.493f, -0.228f, 0.841f), 0.06f));
  417. EXPECT_THAT(m3.GetRow(2), IsCloseTolerance(Vector3(0.765f, 0.353f, 0.542f), 0.06f));
  418. }
  419. TEST(MATH_Matrix3x3, TestFullInverse)
  420. {
  421. Matrix3x3 m1;
  422. m1.SetRow(0, -1.0f, 2.0f, 3.0f);
  423. m1.SetRow(1, 4.0f, 5.0f, 6.0f);
  424. m1.SetRow(2, 7.0f, 8.0f, -9.0f);
  425. EXPECT_THAT((m1 * m1.GetInverseFull()), IsClose(Matrix3x3::CreateIdentity()));
  426. }
  427. TEST(MATH_Matrix3x3, TestScaleAccess)
  428. {
  429. Matrix3x3 m1 = Matrix3x3::CreateRotationX(DegToRad(40.0f)) * Matrix3x3::CreateScale(Vector3(2.0f, 3.0f, 4.0f));
  430. EXPECT_THAT(m1.RetrieveScale(), IsClose(Vector3(2.0f, 3.0f, 4.0f)));
  431. EXPECT_THAT(m1.ExtractScale(), IsClose(Vector3(2.0f, 3.0f, 4.0f)));
  432. EXPECT_THAT(m1.RetrieveScale(), IsClose(Vector3::CreateOne()));
  433. m1.MultiplyByScale(Vector3(3.0f, 4.0f, 5.0f));
  434. EXPECT_THAT(m1.RetrieveScale(), IsClose(Vector3(3.0f, 4.0f, 5.0f)));
  435. }
  436. TEST(MATH_Matrix3x3, TestScaleSqAccess)
  437. {
  438. Matrix3x3 m1 = Matrix3x3::CreateRotationX(DegToRad(40.0f)) * Matrix3x3::CreateScale(Vector3(2.0f, 3.0f, 4.0f));
  439. EXPECT_THAT(m1.RetrieveScaleSq(), IsClose(Vector3(4.0f, 9.0f, 16.0f)));
  440. m1.ExtractScale();
  441. EXPECT_THAT(m1.RetrieveScaleSq(), IsClose(Vector3::CreateOne()));
  442. m1.MultiplyByScale(Vector3(3.0f, 4.0f, 5.0f));
  443. EXPECT_THAT(m1.RetrieveScaleSq(), IsClose(Vector3(9.0f, 16.0f, 25.0f)));
  444. }
  445. TEST(MATH_Matrix3x3, TestReciprocalScaled)
  446. {
  447. Matrix3x3 orthogonalMatrix = Matrix3x3::CreateRotationX(DegToRad(40.0f));
  448. EXPECT_THAT(orthogonalMatrix.GetReciprocalScaled(), IsClose(orthogonalMatrix));
  449. const AZ::Vector3 scale(2.8f, 0.7f, 1.3f);
  450. AZ::Matrix3x3 scaledMatrix = orthogonalMatrix;
  451. scaledMatrix.MultiplyByScale(scale);
  452. AZ::Matrix3x3 reciprocalScaledMatrix = orthogonalMatrix;
  453. reciprocalScaledMatrix.MultiplyByScale(scale.GetReciprocal());
  454. EXPECT_THAT(scaledMatrix.GetReciprocalScaled(), IsClose(reciprocalScaledMatrix));
  455. }
  456. TEST(MATH_Matrix3x3, TestPolarDecomposition)
  457. {
  458. Matrix3x3 m1 = Matrix3x3::CreateRotationX(DegToRad(30.0f));
  459. Matrix3x3 m2 = Matrix3x3::CreateScale(Vector3(5.0f, 6.0f, 7.0f));
  460. Matrix3x3 m3 = m1 * m2;
  461. EXPECT_THAT(m3.GetPolarDecomposition(), IsClose(m1));
  462. Matrix3x3 m4, m5;
  463. m3.GetPolarDecomposition(&m4, &m5);
  464. EXPECT_THAT((m4 * m5), IsClose(m3));
  465. EXPECT_THAT(m4, IsClose(m1));
  466. EXPECT_THAT(m5, IsCloseTolerance(m2, 0.01f));
  467. }
  468. TEST(MATH_Matrix3x3, TestOrthogonalize)
  469. {
  470. Matrix3x3 m1 = Matrix3x3::CreateRotationX(AZ::DegToRad(30.0f)) * Matrix3x3::CreateScale(Vector3(2.0f, 3.0f, 4.0f));
  471. m1.SetElement(0, 1, 0.2f);
  472. Matrix3x3 m2 = m1.GetOrthogonalized();
  473. EXPECT_NEAR(m2.GetRow(0).GetLength(), 1.0f, AZ::Constants::Tolerance);
  474. EXPECT_NEAR(m2.GetRow(1).GetLength(), 1.0f, AZ::Constants::Tolerance);
  475. EXPECT_NEAR(m2.GetRow(2).GetLength(), 1.0f, AZ::Constants::Tolerance);
  476. EXPECT_NEAR(m2.GetRow(0).Dot(m2.GetRow(1)), 0.0f, AZ::Constants::Tolerance);
  477. EXPECT_NEAR(m2.GetRow(0).Dot(m2.GetRow(2)), 0.0f, AZ::Constants::Tolerance);
  478. EXPECT_NEAR(m2.GetRow(1).Dot(m2.GetRow(2)), 0.0f, AZ::Constants::Tolerance);
  479. EXPECT_THAT(m2.GetRow(0).Cross(m2.GetRow(1)), IsClose(m2.GetRow(2)));
  480. EXPECT_THAT(m2.GetRow(1).Cross(m2.GetRow(2)), IsClose(m2.GetRow(0)));
  481. EXPECT_THAT(m2.GetRow(2).Cross(m2.GetRow(0)), IsClose(m2.GetRow(1)));
  482. m1.Orthogonalize();
  483. EXPECT_THAT(m1, IsClose(m2));
  484. }
  485. TEST(MATH_Matrix3x3, TestOrthogonal)
  486. {
  487. Matrix3x3 m1 = Matrix3x3::CreateRotationX(AZ::DegToRad(30.0f));
  488. EXPECT_TRUE(m1.IsOrthogonal(0.05f));
  489. m1.SetRow(1, m1.GetRow(1) * 2.0f);
  490. EXPECT_FALSE(m1.IsOrthogonal(0.05f));
  491. m1 = Matrix3x3::CreateRotationX(AZ::DegToRad(30.0f));
  492. m1.SetRow(1, Vector3(0.0f, 1.0f, 0.0f));
  493. EXPECT_FALSE(m1.IsOrthogonal(0.05f));
  494. }
  495. TEST(MATH_Matrix3x3, TestIsClose)
  496. {
  497. Matrix3x3 m1 = Matrix3x3::CreateRotationX(DegToRad(30.0f));
  498. Matrix3x3 m2 = m1;
  499. EXPECT_TRUE(m1.IsClose(m2));
  500. m2.SetElement(0, 0, 2.0f);
  501. EXPECT_FALSE(m1.IsClose(m2));
  502. m2 = m1;
  503. m2.SetElement(0, 2, 2.0f);
  504. EXPECT_FALSE(m1.IsClose(m2));
  505. }
  506. TEST(MATH_Matrix3x3, TestGetDiagonal)
  507. {
  508. Matrix3x3 m1;
  509. m1.SetRow(0, 1.0f, 2.0f, 3.0f);
  510. m1.SetRow(1, 4.0f, 5.0f, 6.0f);
  511. m1.SetRow(2, 7.0f, 8.0f, 9.0f);
  512. EXPECT_THAT(m1.GetDiagonal(), IsCloseTolerance(Vector3(1.0f, 5.0f, 9.0f), 1e-6f));
  513. }
  514. TEST(MATH_Matrix3x3, TestDeterminant)
  515. {
  516. Matrix3x3 m1;
  517. m1.SetRow(0, -1.0f, 2.0f, 3.0f);
  518. m1.SetRow(1, 4.0f, 5.0f, 6.0f);
  519. m1.SetRow(2, 7.0f, 8.0f, -9.0f);
  520. EXPECT_NEAR(m1.GetDeterminant(), 240.0f, AZ::Constants::Tolerance);
  521. }
  522. TEST(MATH_Matrix3x3, TestAdjugate)
  523. {
  524. Matrix3x3 m1;
  525. m1.SetRow(0, 1.0f, 2.0f, 3.0f);
  526. m1.SetRow(1, 4.0f, 5.0f, 6.0f);
  527. m1.SetRow(2, 7.0f, 8.0f, 9.0f);
  528. Matrix3x3 m2 = m1.GetAdjugate();
  529. EXPECT_THAT(m2.GetRow(0), IsClose(Vector3(-3.0f, 6.0f, -3.0f)));
  530. EXPECT_THAT(m2.GetRow(1), IsClose(Vector3(6.0f, -12.0f, 6.0f)));
  531. EXPECT_THAT(m2.GetRow(2), IsClose(Vector3(-3.0f, 6.0f, -3.0f)));
  532. }
  533. }