Mat4.h 18 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828829830831832833834835836837838839840841842843844845846847848849850851852853854855856857858859860861862863864865866867868869870871872873874875876
  1. #ifndef ANKI_MATH_MAT4_H
  2. #define ANKI_MATH_MAT4_H
  3. #include "anki/math/CommonIncludes.h"
  4. namespace anki {
  5. /// @addtogroup Math
  6. /// @{
  7. /// Template struct that gives the type of the TVec4 SIMD
  8. template<typename T>
  9. struct TMat4Simd
  10. {
  11. typedef Array<T, 16> Type;
  12. };
  13. #if ANKI_SIMD == ANKI_SIMD_SSE
  14. // Specialize for F32
  15. template<>
  16. struct TMat4Simd<F32>
  17. {
  18. typedef Array<__m128, 4> Type;
  19. };
  20. #endif
  21. /// 4x4 Matrix. Used mainly for transformations but not necessarily. Its
  22. /// row major. SSE optimized
  23. template<typename T>
  24. ANKI_ATTRIBUTE_ALIGNED(class, 16) TMat4
  25. {
  26. /// @name Friends
  27. /// @{
  28. template<typename Y>
  29. friend TMat4<Y> operator+(const Y f, const TMat4<Y>& m4);
  30. template<typename Y>
  31. friend TMat4<Y> operator-(const Y f, const TMat4<Y>& m4);
  32. template<typename Y>
  33. friend TMat4<Y> operator*(const Y f, const TMat4<Y>& m4);
  34. template<typename Y>
  35. friend TMat4<Y> operator/(const Y f, const TMat4<Y>& m4);
  36. /// @}
  37. public:
  38. typedef typename TMat4Simd<T>::Type Simd;
  39. /// @name Constructors
  40. /// @{
  41. explicit TMat4()
  42. {}
  43. explicit TMat4(const T f)
  44. {
  45. for(U i = 0; i < 16; i++)
  46. {
  47. arr1[i] = f;
  48. }
  49. }
  50. explicit TMat4(const T m00, const T m01, const T m02,
  51. const T m03, const T m10, const T m11,
  52. const T m12, const T m13, const T m20,
  53. const T m21, const T m22, const T m23,
  54. const T m30, const T m31, const T m32,
  55. const T m33)
  56. {
  57. TMat4& m = *this;
  58. m(0, 0) = m00;
  59. m(0, 1) = m01;
  60. m(0, 2) = m02;
  61. m(0, 3) = m03;
  62. m(1, 0) = m10;
  63. m(1, 1) = m11;
  64. m(1, 2) = m12;
  65. m(1, 3) = m13;
  66. m(2, 0) = m20;
  67. m(2, 1) = m21;
  68. m(2, 2) = m22;
  69. m(2, 3) = m23;
  70. m(3, 0) = m30;
  71. m(3, 1) = m31;
  72. m(3, 2) = m32;
  73. m(3, 3) = m33;
  74. }
  75. explicit TMat4(const T arr[])
  76. {
  77. for(U i = 0; i < 16; i++)
  78. {
  79. arr1[i] = arr[i];
  80. }
  81. }
  82. TMat4(const TMat4& b)
  83. {
  84. for(U i = 0; i < 16; i++)
  85. {
  86. arr1[i] = b.arr1[i];
  87. }
  88. }
  89. explicit TMat4(const TMat3<T>& m3)
  90. {
  91. TMat4& m = *this;
  92. m(0, 0) = m3(0, 0);
  93. m(0, 1) = m3(0, 1);
  94. m(0, 2) = m3(0, 2);
  95. m(0, 3) = 0.0;
  96. m(1, 0) = m3(1, 0);
  97. m(1, 1) = m3(1, 1);
  98. m(1, 2) = m3(1, 2);
  99. m(1, 3) = 0.0;
  100. m(2, 0) = m3(2, 0);
  101. m(2, 1) = m3(2, 1);
  102. m(2, 2) = m3(2, 2);
  103. m(2, 3) = 0.0;
  104. m(3, 0) = 0.0;
  105. m(3, 1) = 0.0;
  106. m(3, 2) = 0.0;
  107. m(3, 3) = 1.0;
  108. }
  109. explicit TMat4(const TVec3<T>& v)
  110. {
  111. TMat4& m = *this;
  112. m(0, 0) = 1.0;
  113. m(0, 1) = 0.0;
  114. m(0, 2) = 0.0;
  115. m(0, 3) = v.x();
  116. m(1, 0) = 0.0;
  117. m(1, 1) = 1.0;
  118. m(1, 2) = 0.0;
  119. m(1, 3) = v.y();
  120. m(2, 0) = 0.0;
  121. m(2, 1) = 0.0;
  122. m(2, 2) = 1.0;
  123. m(2, 3) = v.z();
  124. m(3, 0) = 0.0;
  125. m(3, 1) = 0.0;
  126. m(3, 2) = 0.0;
  127. m(3, 3) = 1.0;
  128. }
  129. explicit TMat4(const TVec4<T>& v)
  130. {
  131. TMat4& m = *this;
  132. m(0, 0) = 1.0;
  133. m(0, 1) = 0.0;
  134. m(0, 2) = 0.0;
  135. m(0, 3) = v.x();
  136. m(1, 0) = 0.0;
  137. m(1, 1) = 1.0;
  138. m(1, 2) = 0.0;
  139. m(1, 3) = v.y();
  140. m(2, 0) = 0.0;
  141. m(2, 1) = 0.0;
  142. m(2, 2) = 1.0;
  143. m(2, 3) = v.z();
  144. m(3, 0) = 0.0;
  145. m(3, 1) = 0.0;
  146. m(3, 2) = 0.0;
  147. m(3, 3) = v.w();
  148. }
  149. explicit TMat4(const TVec3<T>& transl, const TMat3<T>& rot)
  150. {
  151. setRotationPart(rot);
  152. setTranslationPart(transl);
  153. TMat4& m = *this;
  154. m(3, 0) = m(3, 1) = m(3, 2) = 0.0;
  155. m(3, 3) = 1.0;
  156. }
  157. explicit TMat4(const TVec3<T>& transl, const TMat3<T>& rot, const T scale)
  158. {
  159. if(isZero<T>(scale - 1.0))
  160. {
  161. setRotationPart(rot);
  162. }
  163. else
  164. {
  165. setRotationPart(rot * scale);
  166. }
  167. setTranslationPart(transl);
  168. TMat4& m = *this;
  169. m(3, 0) = m(3, 1) = m(3, 2) = 0.0;
  170. m(3, 3) = 1.0;
  171. }
  172. explicit TMat4(const TTransform<T>& t)
  173. {
  174. (*this) = TMat4(t.getOrigin(), t.getRotation(), t.getScale());
  175. }
  176. /// @}
  177. /// @name Accessors
  178. /// @{
  179. T& operator()(const U i, const U j)
  180. {
  181. return arr2[i][j];
  182. }
  183. const T& operator()(const U i, const U j) const
  184. {
  185. return arr2[i][j];
  186. }
  187. T& operator[](const U i)
  188. {
  189. return arr1[i];
  190. }
  191. const T& operator[](const U i) const
  192. {
  193. return arr1[i];
  194. }
  195. /// @}
  196. /// @name Operators with same type
  197. /// @{
  198. TMat4& operator=(const TMat4& b)
  199. {
  200. for(U i = 0; i < 16; i++)
  201. {
  202. arr1[i] = b.arr1[i];
  203. }
  204. return *this;
  205. }
  206. TMat4 operator+(const TMat4& b) const
  207. {
  208. TMat4<T> c;
  209. for(U i = 0; i < 16; i++)
  210. {
  211. c.arr1[i] = arr1[i] + b.arr1[i];
  212. }
  213. return c;
  214. }
  215. TMat4& operator+=(const TMat4& b)
  216. {
  217. for(U i = 0; i < 16; i++)
  218. {
  219. arr1[i] += b.arr1[i];
  220. }
  221. return (*this);
  222. }
  223. TMat4 operator-(const TMat4& b) const
  224. {
  225. TMat4<T> c;
  226. for(U i = 0; i < 16; i++)
  227. {
  228. c.arr1[i] = arr1[i] - b.arr1[i];
  229. }
  230. return c;
  231. }
  232. TMat4& operator-=(const TMat4& b)
  233. {
  234. for(U i = 0; i < 16; i++)
  235. {
  236. arr1[i] -= b.arr1[i];
  237. }
  238. return (*this);
  239. }
  240. /// @note 64 muls, 48 adds
  241. TMat4 operator*(const TMat4& b) const
  242. {
  243. TMat4<T> c;
  244. const TMat4& m = *this;
  245. for(U i = 0; i < 4; i++)
  246. {
  247. for(U j = 0; j < 4; j++)
  248. {
  249. c(i, j) = m(i, 0) * b(0, j) + m(i, 1) * b(1, j)
  250. + m(i, 2) * b(2, j) + m(i, 3) * b(3, j);
  251. }
  252. }
  253. return c;
  254. }
  255. TMat4& operator*=(const TMat4& b)
  256. {
  257. (*this) = (*this) * b;
  258. return (*this);
  259. }
  260. Bool operator==(const TMat4& b) const
  261. {
  262. for(U i = 0; i < 16; i++)
  263. {
  264. if(!isZero<T>(arr1[i] - b.arr1[i]))
  265. {
  266. return false;
  267. }
  268. }
  269. return true;
  270. }
  271. Bool operator!=(const TMat4& b) const
  272. {
  273. for(U i = 0; i < 16; i++)
  274. {
  275. if(!isZero(arr1[i] - b.arr1[i]))
  276. {
  277. return true;
  278. }
  279. }
  280. return false;
  281. }
  282. /// @}
  283. /// @name Operators with T
  284. /// @{
  285. TMat4 operator+(const T f) const
  286. {
  287. TMat4 c;
  288. for(U i = 0; i < 16; i++)
  289. {
  290. c.arr1[i] = arr1[i] + f;
  291. }
  292. return c;
  293. }
  294. TMat4& operator+=(const T f)
  295. {
  296. for(U i = 0; i < 16; i++)
  297. {
  298. arr1[i] += f;
  299. }
  300. }
  301. TMat4 operator-(const T f) const
  302. {
  303. TMat4 c;
  304. for(U i = 0; i < 16; i++)
  305. {
  306. c.arr1[i] = arr1[i] - f;
  307. }
  308. return c;
  309. }
  310. TMat4& operator-=(const T f)
  311. {
  312. for(U i = 0; i < 16; i++)
  313. {
  314. arr1[i] -= f;
  315. }
  316. }
  317. TMat4 operator*(const T f) const
  318. {
  319. TMat4 c;
  320. for(U i = 0; i < 16; i++)
  321. {
  322. c.arr1[i] = arr1[i] * f;
  323. }
  324. return c;
  325. }
  326. TMat4& operator*=(const T f)
  327. {
  328. for(U i = 0; i < 16; i++)
  329. {
  330. arr1[i] *= f;
  331. }
  332. return *this;
  333. }
  334. TMat4 operator/(const T f) const
  335. {
  336. TMat4 c;
  337. for(U i = 0; i < 16; i++)
  338. {
  339. c[i] = arr1[i] / f;
  340. }
  341. return c;
  342. }
  343. TMat4& operator/=(const T f)
  344. {
  345. for(U i = 0; i < 16; i++)
  346. {
  347. arr1[i] /= f;
  348. }
  349. }
  350. /// @}
  351. /// @name Operators with other types
  352. /// @{
  353. /// 16 muls, 12 adds
  354. TVec4<T> operator*(const TVec4<T>& v4) const
  355. {
  356. TVec4<T> out;
  357. const TMat4& m = *this;
  358. out.x() = m(0, 0) * v4.x() + m(0, 1) * v4.y()
  359. + m(0, 2) * v4.z() + m(0, 3) * v4.w();
  360. out.y() = m(1, 0) * v4.x() + m(1, 1) * v4.y()
  361. + m(1, 2) * v4.z() + m(1, 3) * v4.w();
  362. out.z() = m(2, 0) * v4.x() + m(2, 1) * v4.y()
  363. + m(2, 2) * v4.z() + m(2, 3) * v4.w();
  364. out.w() = m(3, 0) * v4.x() + m(3, 1) * v4.y()
  365. + m(3, 2) * v4.z() + m(3, 3) * v4.w();
  366. return out;
  367. }
  368. /// @}
  369. /// @name Other
  370. /// @{
  371. void setRow(const U i, const TVec4<T>& v)
  372. {
  373. TMat4& m = *this;
  374. m(i, 0) = v.x();
  375. m(i, 1) = v.y();
  376. m(i, 2) = v.z();
  377. m(i, 3) = v.w();
  378. }
  379. void setRows(const TVec4<T>& a, const TVec4<T>& b, const TVec4<T>& c,
  380. const TVec4<T>& d)
  381. {
  382. setRow(0, a);
  383. setRow(1, b);
  384. setRow(2, c);
  385. setRow(3, d);
  386. }
  387. TVec4<T> getRow(const U i) const
  388. {
  389. const TMat4& m = *this;
  390. return TVec4<T>(m(i, 0), m(i, 1), m(i, 2), m(i, 3));
  391. }
  392. void getRows(TVec4<T>& a, TVec4<T>& b, TVec4<T>& c, TVec4<T>& d) const
  393. {
  394. a = getRow(0);
  395. b = getRow(1);
  396. c = getRow(2);
  397. d = getRow(3);
  398. }
  399. void setColumn(const U i, const TVec4<T>& v)
  400. {
  401. TMat4& m = *this;
  402. m(0, i) = v.x();
  403. m(1, i) = v.y();
  404. m(2, i) = v.z();
  405. m(3, i) = v.w();
  406. }
  407. void setColumns(const TVec4<T>& a, const TVec4<T>& b, const TVec4<T>& c,
  408. const TVec4<T>& d)
  409. {
  410. setColumn(0, a);
  411. setColumn(1, b);
  412. setColumn(2, c);
  413. setColumn(3, d);
  414. }
  415. TVec4<T> getColumn(const U i) const
  416. {
  417. const TMat4& m = *this;
  418. return TVec4<T>(m(0, i), m(1, i), m(2, i), m(3, i));
  419. }
  420. void getColumns(TVec4<T>& a, TVec4<T>& b, TVec4<T>& c, TVec4<T>& d) const
  421. {
  422. a = getColumn(0);
  423. b = getColumn(1);
  424. c = getColumn(2);
  425. d = getColumn(3);
  426. }
  427. void setRotationPart(const TMat3<T>& m3)
  428. {
  429. TMat4& m = *this;
  430. m(0, 0) = m3(0, 0);
  431. m(0, 1) = m3(0, 1);
  432. m(0, 2) = m3(0, 2);
  433. m(1, 0) = m3(1, 0);
  434. m(1, 1) = m3(1, 1);
  435. m(1, 2) = m3(1, 2);
  436. m(2, 0) = m3(2, 0);
  437. m(2, 1) = m3(2, 1);
  438. m(2, 2) = m3(2, 2);
  439. }
  440. TMat3<T> getRotationPart() const
  441. {
  442. const TMat4& m = *this;
  443. TMat3<T> m3;
  444. m3(0, 0) = m(0, 0);
  445. m3(0, 1) = m(0, 1);
  446. m3(0, 2) = m(0, 2);
  447. m3(1, 0) = m(1, 0);
  448. m3(1, 1) = m(1, 1);
  449. m3(1, 2) = m(1, 2);
  450. m3(2, 0) = m(2, 0);
  451. m3(2, 1) = m(2, 1);
  452. m3(2, 2) = m(2, 2);
  453. return m3;
  454. }
  455. void setTranslationPart(const TVec4<T>& v)
  456. {
  457. TMat4& m = *this;
  458. m(0, 3) = v.x();
  459. m(1, 3) = v.y();
  460. m(2, 3) = v.z();
  461. m(3, 3) = v.w();
  462. }
  463. void setTranslationPart(const TVec3<T>& v)
  464. {
  465. TMat4& m = *this;
  466. m(0, 3) = v.x();
  467. m(1, 3) = v.y();
  468. m(2, 3) = v.z();
  469. }
  470. TVec3<T> getTranslationPart() const
  471. {
  472. const TMat4& m = *this;
  473. return TVec3<T>(m(0, 3), m(1, 3), m(2, 3));
  474. }
  475. void transpose()
  476. {
  477. TMat4& m = *this;
  478. T tmp = m(0, 1);
  479. m(0, 1) = m(1, 0);
  480. m(1, 0) = tmp;
  481. tmp = m(0, 2);
  482. m(0, 2) = m(2, 0);
  483. m(2, 0) = tmp;
  484. tmp = m(0, 3);
  485. m(0, 3) = m(3, 0);
  486. m(3, 0) = tmp;
  487. tmp = m(1, 2);
  488. m(1, 2) = m(2, 1);
  489. m(2, 1) = tmp;
  490. tmp = m(1, 3);
  491. m(1, 3) = m(3, 1);
  492. m(3, 1) = tmp;
  493. tmp = m(2, 3);
  494. m(2, 3) = m(3, 2);
  495. m(3, 2) = tmp;
  496. }
  497. TMat4 getTransposed() const
  498. {
  499. const TMat4& m = *this;
  500. TMat4 out;
  501. for(U i = 0; i < 4; i++)
  502. {
  503. for(U j = 0; j < 4; j++)
  504. {
  505. out(i, j) = m(j, i);
  506. }
  507. }
  508. return out;
  509. }
  510. T getDet() const
  511. {
  512. const TMat4& t = *this;
  513. return t(0, 3) * t(1, 2) * t(2, 1) * t(3, 0)
  514. - t(0, 2) * t(1, 3) * t(2, 1) * t(3, 0)
  515. - t(0, 3) * t(1, 1) * t(2, 2) * t(3, 0)
  516. + t(0, 1) * t(1, 3) * t(2, 2) * t(3, 0)
  517. + t(0, 2) * t(1, 1) * t(2, 3) * t(3, 0)
  518. - t(0, 1) * t(1, 2) * t(2, 3) * t(3, 0)
  519. - t(0, 3) * t(1, 2) * t(2, 0) * t(3, 1)
  520. + t(0, 2) * t(1, 3) * t(2, 0) * t(3, 1)
  521. + t(0, 3) * t(1, 0) * t(2, 2) * t(3, 1)
  522. - t(0, 0) * t(1, 3) * t(2, 2) * t(3, 1)
  523. - t(0, 2) * t(1, 0) * t(2, 3) * t(3, 1)
  524. + t(0, 0) * t(1, 2) * t(2, 3) * t(3, 1)
  525. + t(0, 3) * t(1, 1) * t(2, 0) * t(3, 2)
  526. - t(0, 1) * t(1, 3) * t(2, 0) * t(3, 2)
  527. - t(0, 3) * t(1, 0) * t(2, 1) * t(3, 2)
  528. + t(0, 0) * t(1, 3) * t(2, 1) * t(3, 2)
  529. + t(0, 1) * t(1, 0) * t(2, 3) * t(3, 2)
  530. - t(0, 0) * t(1, 1) * t(2, 3) * t(3, 2)
  531. - t(0, 2) * t(1, 1) * t(2, 0) * t(3, 3)
  532. + t(0, 1) * t(1, 2) * t(2, 0) * t(3, 3)
  533. + t(0, 2) * t(1, 0) * t(2, 1) * t(3, 3)
  534. - t(0, 0) * t(1, 2) * t(2, 1) * t(3, 3)
  535. - t(0, 1) * t(1, 0) * t(2, 2) * t(3, 3)
  536. + t(0, 0) * t(1, 1) * t(2, 2) * t(3, 3);
  537. }
  538. /// Invert using Cramer's rule
  539. TMat4 getInverse() const
  540. {
  541. Array<T, 12> tmp;
  542. const TMat4& in = (*this);
  543. TMat4 m4;
  544. tmp[0] = in(2, 2) * in(3, 3);
  545. tmp[1] = in(3, 2) * in(2, 3);
  546. tmp[2] = in(1, 2) * in(3, 3);
  547. tmp[3] = in(3, 2) * in(1, 3);
  548. tmp[4] = in(1, 2) * in(2, 3);
  549. tmp[5] = in(2, 2) * in(1, 3);
  550. tmp[6] = in(0, 2) * in(3, 3);
  551. tmp[7] = in(3, 2) * in(0, 3);
  552. tmp[8] = in(0, 2) * in(2, 3);
  553. tmp[9] = in(2, 2) * in(0, 3);
  554. tmp[10] = in(0, 2) * in(1, 3);
  555. tmp[11] = in(1, 2) * in(0, 3);
  556. m4(0, 0) = tmp[0] * in(1, 1) + tmp[3] * in(2, 1) + tmp[4] * in(3, 1);
  557. m4(0, 0) -= tmp[1] * in(1, 1) + tmp[2] * in(2, 1) + tmp[5] * in(3, 1);
  558. m4(0, 1) = tmp[1] * in(0, 1) + tmp[6] * in(2, 1) + tmp[9] * in(3, 1);
  559. m4(0, 1) -= tmp[0] * in(0, 1) + tmp[7] * in(2, 1) + tmp[8] * in(3, 1);
  560. m4(0, 2) = tmp[2] * in(0, 1) + tmp[7] * in(1, 1) + tmp[10] * in(3, 1);
  561. m4(0, 2) -= tmp[3] * in(0, 1) + tmp[6] * in(1, 1) + tmp[11] * in(3, 1);
  562. m4(0, 3) = tmp[5] * in(0, 1) + tmp[8] * in(1, 1) + tmp[11] * in(2, 1);
  563. m4(0, 3) -= tmp[4] * in(0, 1) + tmp[9] * in(1, 1) + tmp[10] * in(2, 1);
  564. m4(1, 0) = tmp[1] * in(1, 0) + tmp[2] * in(2, 0) + tmp[5] * in(3, 0);
  565. m4(1, 0) -= tmp[0] * in(1, 0) + tmp[3] * in(2, 0) + tmp[4] * in(3, 0);
  566. m4(1, 1) = tmp[0] * in(0, 0) + tmp[7] * in(2, 0) + tmp[8] * in(3, 0);
  567. m4(1, 1) -= tmp[1] * in(0, 0) + tmp[6] * in(2, 0) + tmp[9] * in(3, 0);
  568. m4(1, 2) = tmp[3] * in(0, 0) + tmp[6] * in(1, 0) + tmp[11] * in(3, 0);
  569. m4(1, 2) -= tmp[2] * in(0, 0) + tmp[7] * in(1, 0) + tmp[10] * in(3, 0);
  570. m4(1, 3) = tmp[4] * in(0, 0) + tmp[9] * in(1, 0) + tmp[10] * in(2, 0);
  571. m4(1, 3) -= tmp[5] * in(0, 0) + tmp[8] * in(1, 0) + tmp[11] * in(2, 0);
  572. tmp[0] = in(2, 0) * in(3, 1);
  573. tmp[1] = in(3, 0) * in(2, 1);
  574. tmp[2] = in(1, 0) * in(3, 1);
  575. tmp[3] = in(3, 0) * in(1, 1);
  576. tmp[4] = in(1, 0) * in(2, 1);
  577. tmp[5] = in(2, 0) * in(1, 1);
  578. tmp[6] = in(0, 0) * in(3, 1);
  579. tmp[7] = in(3, 0) * in(0, 1);
  580. tmp[8] = in(0, 0) * in(2, 1);
  581. tmp[9] = in(2, 0) * in(0, 1);
  582. tmp[10] = in(0, 0) * in(1, 1);
  583. tmp[11] = in(1, 0) * in(0, 1);
  584. m4(2, 0) = tmp[0] * in(1, 3) + tmp[3] * in(2, 3) + tmp[4] * in(3, 3);
  585. m4(2, 0) -= tmp[1] * in(1, 3) + tmp[2] * in(2, 3) + tmp[5] * in(3, 3);
  586. m4(2, 1) = tmp[1] * in(0, 3) + tmp[6] * in(2, 3) + tmp[9] * in(3, 3);
  587. m4(2, 1) -= tmp[0] * in(0, 3) + tmp[7] * in(2, 3) + tmp[8] * in(3, 3);
  588. m4(2, 2) = tmp[2] * in(0, 3) + tmp[7] * in(1, 3) + tmp[10] * in(3, 3);
  589. m4(2, 2) -= tmp[3] * in(0, 3) + tmp[6] * in(1, 3) + tmp[11] * in(3, 3);
  590. m4(2, 3) = tmp[5] * in(0, 3) + tmp[8] * in(1, 3) + tmp[11] * in(2, 3);
  591. m4(2, 3) -= tmp[4] * in(0, 3) + tmp[9] * in(1, 3) + tmp[10] * in(2, 3);
  592. m4(3, 0) = tmp[2] * in(2, 2) + tmp[5] * in(3, 2) + tmp[1] * in(1, 2);
  593. m4(3, 0) -= tmp[4] * in(3, 2) + tmp[0] * in(1, 2) + tmp[3] * in(2, 2);
  594. m4(3, 1) = tmp[8] * in(3, 2) + tmp[0] * in(0, 2) + tmp[7] * in(2, 2);
  595. m4(3, 1) -= tmp[6] * in(2, 2) + tmp[9] * in(3, 2) + tmp[1] * in(0, 2);
  596. m4(3, 2) = tmp[6] * in(1, 2) + tmp[11] * in(3, 2) + tmp[3] * in(0, 2);
  597. m4(3, 2) -= tmp[10] * in(3, 2) + tmp[2] * in(0, 2) + tmp[7] * in(1, 2);
  598. m4(3, 3) = tmp[10] * in(2, 2) + tmp[4] * in(0, 2) + tmp[9] * in(1, 2);
  599. m4(3, 3) -= tmp[8] * in(1, 2) + tmp[11] * in(2, 2) + tmp[5] * in(0, 2);
  600. T det = in(0, 0) * m4(0, 0) + in(1, 0) * m4(0, 1)
  601. + in(2, 0) * m4(0, 2) + in(3, 0) * m4(0, 3);
  602. ANKI_ASSERT(!isZero<T>(det)); // Cannot invert, det == 0
  603. det = 1.0 / det;
  604. m4 *= det;
  605. return m4;
  606. }
  607. /// See getInverse
  608. void invert()
  609. {
  610. (*this) = getInverse();
  611. }
  612. /// If we suppose this matrix represents a transformation, return the
  613. /// inverted transformation
  614. TMat4 getInverseTransformation() const
  615. {
  616. TMat3<T> invertedRot = getRotationPart().getTransposed();
  617. TVec3<T> invertedTsl = getTranslationPart();
  618. invertedTsl = -(invertedRot * invertedTsl);
  619. return TMat4(invertedTsl, invertedRot);
  620. }
  621. TMat4 lerp(const TMat4& b, T t) const
  622. {
  623. return ((*this) * (1.0 - t)) + (b * t);
  624. }
  625. void setIdentity()
  626. {
  627. (*this) = getIdentity();
  628. }
  629. static const TMat4& getIdentity()
  630. {
  631. static const TMat4 ident(1.0, 0.0, 0.0, 0.0, 0.0, 1.0, 0.0, 0.0, 0.0,
  632. 0.0, 1.0, 0.0, 0.0, 0.0, 0.0, 1.0);
  633. return ident;
  634. }
  635. static const TMat4& getZero()
  636. {
  637. static const TMat4 zero(0.0);
  638. return zero;
  639. }
  640. /// 12 muls, 27 adds. Something like m4 = m0 * m1 but without touching
  641. /// the 4rth row and allot faster
  642. static TMat4 combineTransformations(const TMat4& m0, const TMat4& m1)
  643. {
  644. // See the clean code in < r664
  645. // one of the 2 mat4 doesnt represent transformation
  646. ANKI_ASSERT(isZero<T>(m0(3, 0) + m0(3, 1) + m0(3, 2) + m0(3, 3) - 1.0)
  647. && isZero<T>(m1(3, 0) + m1(3, 1) + m1(3, 2) + m1(3, 3) - 1.0));
  648. TMat4 m4;
  649. m4(0, 0) =
  650. m0(0, 0) * m1(0, 0) + m0(0, 1) * m1(1, 0) + m0(0, 2) * m1(2, 0);
  651. m4(0, 1) =
  652. m0(0, 0) * m1(0, 1) + m0(0, 1) * m1(1, 1) + m0(0, 2) * m1(2, 1);
  653. m4(0, 2) =
  654. m0(0, 0) * m1(0, 2) + m0(0, 1) * m1(1, 2) + m0(0, 2) * m1(2, 2);
  655. m4(1, 0) =
  656. m0(1, 0) * m1(0, 0) + m0(1, 1) * m1(1, 0) + m0(1, 2) * m1(2, 0);
  657. m4(1, 1) =
  658. m0(1, 0) * m1(0, 1) + m0(1, 1) * m1(1, 1) + m0(1, 2) * m1(2, 1);
  659. m4(1, 2) =
  660. m0(1, 0) * m1(0, 2) + m0(1, 1) * m1(1, 2) + m0(1, 2) * m1(2, 2);
  661. m4(2, 0) =
  662. m0(2, 0) * m1(0, 0) + m0(2, 1) * m1(1, 0) + m0(2, 2) * m1(2, 0);
  663. m4(2, 1) =
  664. m0(2, 0) * m1(0, 1) + m0(2, 1) * m1(1, 1) + m0(2, 2) * m1(2, 1);
  665. m4(2, 2) =
  666. m0(2, 0) * m1(0, 2) + m0(2, 1) * m1(1, 2) + m0(2, 2) * m1(2, 2);
  667. m4(0, 3) = m0(0, 0) * m1(0, 3) + m0(0, 1) * m1(1, 3)
  668. + m0(0, 2) * m1(2, 3) + m0(0, 3);
  669. m4(1, 3) = m0(1, 0) * m1(0, 3) + m0(1, 1) * m1(1, 3)
  670. + m0(1, 2) * m1(2, 3) + m0(1, 3);
  671. m4(2, 3) = m0(2, 0) * m1(0, 3) + m0(2, 1) * m1(1, 3)
  672. + m0(2, 2) * m1(2, 3) + m0(2, 3);
  673. m4(3, 0) = m4(3, 1) = m4(3, 2) = 0.0;
  674. m4(3, 3) = 1.0;
  675. return m4;
  676. }
  677. std::string toString() const
  678. {
  679. const TMat4& m = *this;
  680. std::string s;
  681. for(U i = 0; i < 4; i++)
  682. {
  683. for(U j = 0; j < 4; j++)
  684. {
  685. s += std::to_string(m(i, j)) + " ";
  686. }
  687. s += "\n";
  688. }
  689. return s;
  690. }
  691. /// @}
  692. private:
  693. /// @name Data
  694. /// @{
  695. union
  696. {
  697. Array<T, 16> arr1;
  698. Array<Array<T, 4>, 4> arr2;
  699. T carr1[16]; ///< For easier debugging with gdb
  700. T carr2[4][4]; ///< For easier debugging with gdb
  701. Simd simd;
  702. };
  703. /// @}
  704. };
  705. #if ANKI_SIMD == ANKI_SIMD_SSE
  706. // Forward declare specializations
  707. template<>
  708. TMat4<F32>::TMat4(const TMat4<F32>& b);
  709. template<>
  710. TMat4<F32>::TMat4(const F32 f);
  711. template<>
  712. TMat4<F32>& TMat4<F32>::operator=(const TMat4<F32>& b);
  713. template<>
  714. TMat4<F32> TMat4<F32>::operator+(const TMat4<F32>& b) const;
  715. template<>
  716. TMat4<F32>& TMat4<F32>::operator+=(const TMat4<F32>& b);
  717. template<>
  718. TMat4<F32> TMat4<F32>::operator-(const TMat4<F32>& b) const;
  719. template<>
  720. TMat4<F32>& TMat4<F32>::operator-=(const TMat4<F32>& b);
  721. template<>
  722. TMat4<F32> TMat4<F32>::operator*(const TMat4<F32>& b) const;
  723. template<>
  724. TMat4<F32> TMat4<F32>::operator+(const F32 f) const;
  725. template<>
  726. TMat4<F32>& TMat4<F32>::operator+=(const F32 f);
  727. template<>
  728. TMat4<F32> TMat4<F32>::operator-(const F32 f) const;
  729. template<>
  730. TMat4<F32>& TMat4<F32>::operator-=(const F32 f);
  731. template<>
  732. TMat4<F32> TMat4<F32>::operator*(const F32 f) const;
  733. template<>
  734. TMat4<F32>& TMat4<F32>::operator*=(const F32 f);
  735. template<>
  736. TMat4<F32> TMat4<F32>::operator/(const F32 f) const;
  737. template<>
  738. TMat4<F32>& TMat4<F32>::operator/=(const F32 f);
  739. template<>
  740. TVec4<F32> TMat4<F32>::operator*(const TVec4<F32>& b) const;
  741. template<>
  742. void TMat4<F32>::setRows(const TVec4<F32>& a, const TVec4<F32>& b,
  743. const TVec4<F32>& c, const TVec4<F32>& d);
  744. template<>
  745. void TMat4<F32>::setRow(const U i, const TVec4<F32>& v);
  746. template<>
  747. void TMat4<F32>::transpose();
  748. template<>
  749. TMat4<F32> operator-(const F32 f, const TMat4<F32>& m4);
  750. template<>
  751. TMat4<F32> operator/(const F32 f, const TMat4<F32>& m4);
  752. #endif
  753. /// F32 4x4 matrix
  754. typedef TMat4<F32> Mat4;
  755. static_assert(sizeof(Mat4) == sizeof(F32) * 4 * 4, "Incorrect size");
  756. /// @}
  757. } // end namespace anki
  758. #include "anki/math/Mat4.inl.h"
  759. #endif