mMatrix.h 57 KB

1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556575859606162636465666768697071727374757677787980818283848586878889909192939495969798991001011021031041051061071081091101111121131141151161171181191201211221231241251261271281291301311321331341351361371381391401411421431441451461471481491501511521531541551561571581591601611621631641651661671681691701711721731741751761771781791801811821831841851861871881891901911921931941951961971981992002012022032042052062072082092102112122132142152162172182192202212222232242252262272282292302312322332342352362372382392402412422432442452462472482492502512522532542552562572582592602612622632642652662672682692702712722732742752762772782792802812822832842852862872882892902912922932942952962972982993003013023033043053063073083093103113123133143153163173183193203213223233243253263273283293303313323333343353363373383393403413423433443453463473483493503513523533543553563573583593603613623633643653663673683693703713723733743753763773783793803813823833843853863873883893903913923933943953963973983994004014024034044054064074084094104114124134144154164174184194204214224234244254264274284294304314324334344354364374384394404414424434444454464474484494504514524534544554564574584594604614624634644654664674684694704714724734744754764774784794804814824834844854864874884894904914924934944954964974984995005015025035045055065075085095105115125135145155165175185195205215225235245255265275285295305315325335345355365375385395405415425435445455465475485495505515525535545555565575585595605615625635645655665675685695705715725735745755765775785795805815825835845855865875885895905915925935945955965975985996006016026036046056066076086096106116126136146156166176186196206216226236246256266276286296306316326336346356366376386396406416426436446456466476486496506516526536546556566576586596606616626636646656666676686696706716726736746756766776786796806816826836846856866876886896906916926936946956966976986997007017027037047057067077087097107117127137147157167177187197207217227237247257267277287297307317327337347357367377387397407417427437447457467477487497507517527537547557567577587597607617627637647657667677687697707717727737747757767777787797807817827837847857867877887897907917927937947957967977987998008018028038048058068078088098108118128138148158168178188198208218228238248258268278288298308318328338348358368378388398408418428438448458468478488498508518528538548558568578588598608618628638648658668678688698708718728738748758768778788798808818828838848858868878888898908918928938948958968978988999009019029039049059069079089099109119129139149159169179189199209219229239249259269279289299309319329339349359369379389399409419429439449459469479489499509519529539549559569579589599609619629639649659669679689699709719729739749759769779789799809819829839849859869879889899909919929939949959969979989991000100110021003100410051006100710081009101010111012101310141015101610171018101910201021102210231024102510261027102810291030103110321033103410351036103710381039104010411042104310441045104610471048104910501051105210531054105510561057105810591060106110621063106410651066106710681069107010711072107310741075107610771078107910801081108210831084108510861087108810891090109110921093109410951096109710981099110011011102110311041105110611071108110911101111111211131114111511161117111811191120112111221123112411251126112711281129113011311132113311341135113611371138113911401141114211431144114511461147114811491150115111521153115411551156115711581159116011611162116311641165116611671168116911701171117211731174117511761177117811791180118111821183118411851186118711881189119011911192119311941195119611971198119912001201120212031204120512061207120812091210121112121213121412151216121712181219122012211222122312241225122612271228122912301231123212331234123512361237123812391240124112421243124412451246124712481249125012511252125312541255125612571258125912601261126212631264126512661267126812691270127112721273127412751276127712781279128012811282128312841285128612871288128912901291129212931294129512961297129812991300130113021303130413051306130713081309131013111312131313141315131613171318131913201321132213231324132513261327132813291330133113321333133413351336133713381339134013411342134313441345134613471348134913501351135213531354135513561357135813591360136113621363136413651366136713681369137013711372137313741375137613771378137913801381138213831384138513861387138813891390139113921393139413951396139713981399140014011402140314041405140614071408140914101411141214131414141514161417141814191420142114221423142414251426142714281429143014311432143314341435143614371438143914401441144214431444144514461447144814491450145114521453145414551456145714581459146014611462146314641465146614671468146914701471147214731474147514761477147814791480148114821483148414851486148714881489149014911492149314941495149614971498149915001501150215031504150515061507150815091510151115121513151415151516151715181519152015211522152315241525152615271528152915301531153215331534153515361537153815391540154115421543154415451546154715481549155015511552155315541555155615571558155915601561156215631564156515661567156815691570157115721573157415751576157715781579158015811582158315841585158615871588158915901591159215931594159515961597159815991600160116021603160416051606160716081609161016111612161316141615161616171618161916201621162216231624162516261627162816291630163116321633163416351636163716381639164016411642164316441645164616471648164916501651165216531654165516561657165816591660166116621663166416651666166716681669167016711672167316741675167616771678167916801681168216831684168516861687168816891690169116921693169416951696169716981699170017011702170317041705170617071708170917101711171217131714171517161717171817191720172117221723172417251726172717281729173017311732173317341735173617371738173917401741174217431744174517461747174817491750175117521753175417551756175717581759176017611762176317641765176617671768176917701771177217731774177517761777177817791780178117821783178417851786178717881789179017911792179317941795179617971798179918001801180218031804180518061807180818091810181118121813181418151816181718181819182018211822182318241825182618271828182918301831183218331834183518361837183818391840184118421843184418451846184718481849185018511852185318541855185618571858185918601861186218631864186518661867186818691870187118721873187418751876187718781879188018811882188318841885188618871888188918901891189218931894189518961897189818991900190119021903190419051906190719081909191019111912191319141915191619171918191919201921192219231924192519261927192819291930193119321933193419351936193719381939194019411942194319441945194619471948194919501951195219531954195519561957195819591960196119621963196419651966196719681969197019711972197319741975197619771978197919801981198219831984198519861987198819891990199119921993199419951996199719981999200020012002200320042005200620072008200920102011201220132014201520162017201820192020202120222023202420252026202720282029203020312032203320342035203620372038203920402041204220432044204520462047204820492050205120522053
  1. //-----------------------------------------------------------------------------
  2. // Copyright (c) 2012 GarageGames, LLC
  3. //
  4. // Permission is hereby granted, free of charge, to any person obtaining a copy
  5. // of this software and associated documentation files (the "Software"), to
  6. // deal in the Software without restriction, including without limitation the
  7. // rights to use, copy, modify, merge, publish, distribute, sublicense, and/or
  8. // sell copies of the Software, and to permit persons to whom the Software is
  9. // furnished to do so, subject to the following conditions:
  10. //
  11. // The above copyright notice and this permission notice shall be included in
  12. // all copies or substantial portions of the Software.
  13. //
  14. // THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
  15. // IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
  16. // FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
  17. // AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
  18. // LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING
  19. // FROM, OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS
  20. // IN THE SOFTWARE.
  21. //-----------------------------------------------------------------------------
  22. #ifndef _MMATRIX_H_
  23. #define _MMATRIX_H_
  24. #include <algorithm>
  25. #ifndef _MPLANE_H_
  26. #include "math/mPlane.h"
  27. #endif
  28. #ifndef _MBOX_H_
  29. #include "math/mBox.h"
  30. #endif
  31. #ifndef _MPOINT4_H_
  32. #include "math/mPoint4.h"
  33. #endif
  34. #ifndef _ENGINETYPEINFO_H_
  35. #include "console/engineTypeInfo.h"
  36. #endif
  37. #ifndef _FRAMEALLOCATOR_H_
  38. #include "core/frameAllocator.h"
  39. #endif
  40. #ifndef _STRINGFUNCTIONS_H_
  41. #include "core/strings/stringFunctions.h"
  42. #endif
  43. #ifndef _CONSOLE_H_
  44. #include "console/console.h"
  45. #endif
  46. #ifndef USE_TEMPLATE_MATRIX
  47. /// 4x4 Matrix Class
  48. ///
  49. /// This runs at F32 precision.
  50. class MatrixF
  51. {
  52. friend class MatrixFEngineExport;
  53. private:
  54. F32 m[16]; ///< Note: Torque uses row-major matrices
  55. public:
  56. /// Create an uninitialized matrix.
  57. ///
  58. /// @param identity If true, initialize to the identity matrix.
  59. explicit MatrixF(bool identity=false);
  60. /// Create a matrix to rotate about origin by e.
  61. /// @see set
  62. explicit MatrixF( const EulerF &e);
  63. /// Create a matrix to rotate about p by e.
  64. /// @see set
  65. MatrixF( const EulerF &e, const Point3F& p);
  66. /// Get the index in m to element in column i, row j
  67. ///
  68. /// This is necessary as we have m as a one dimensional array.
  69. ///
  70. /// @param i Column desired.
  71. /// @param j Row desired.
  72. static U32 idx(U32 i, U32 j) { return (i + j*4); }
  73. /// Initialize matrix to rotate about origin by e.
  74. MatrixF& set( const EulerF &e);
  75. /// Initialize matrix to rotate about p by e.
  76. MatrixF& set( const EulerF &e, const Point3F& p);
  77. /// Initialize matrix with a cross product of p.
  78. MatrixF& setCrossProduct( const Point3F &p);
  79. /// Initialize matrix with a tensor product of p.
  80. MatrixF& setTensorProduct( const Point3F &p, const Point3F& q);
  81. operator F32*() { return (m); } ///< Allow people to get at m.
  82. operator const F32*() const { return (F32*)(m); } ///< Allow people to get at m.
  83. bool isAffine() const; ///< Check to see if this is an affine matrix.
  84. bool isIdentity() const; ///< Checks for identity matrix.
  85. /// Make this an identity matrix.
  86. MatrixF& identity();
  87. /// Invert m.
  88. MatrixF& inverse();
  89. /// Copy the inversion of this into out matrix.
  90. void invertTo( MatrixF *out );
  91. /// Take inverse of matrix assuming it is affine (rotation,
  92. /// scale, sheer, translation only).
  93. MatrixF& affineInverse();
  94. /// Swap rows and columns.
  95. MatrixF& transpose();
  96. /// M * Matrix(p) -> M
  97. MatrixF& scale( const Point3F &s );
  98. MatrixF& scale( F32 s ) { return scale( Point3F( s, s, s ) ); }
  99. /// Return scale assuming scale was applied via mat.scale(s).
  100. Point3F getScale() const;
  101. EulerF toEuler() const;
  102. F32 determinant() const {
  103. return m_matF_determinant(*this);
  104. }
  105. /// Compute the inverse of the matrix.
  106. ///
  107. /// Computes inverse of full 4x4 matrix. Returns false and performs no inverse if
  108. /// the determinant is 0.
  109. ///
  110. /// Note: In most cases you want to use the normal inverse function. This method should
  111. /// be used if the matrix has something other than (0,0,0,1) in the bottom row.
  112. bool fullInverse();
  113. /// Reverse depth for projection matrix
  114. /// Simplifies reversal matrix mult to 4 subtractions
  115. void reverseProjection();
  116. /// Swaps rows and columns into matrix.
  117. void transposeTo(F32 *matrix) const;
  118. /// Normalize the matrix.
  119. void normalize();
  120. /// Copy the requested column into a Point4F.
  121. void getColumn(S32 col, Point4F *cptr) const;
  122. Point4F getColumn4F(S32 col) const { Point4F ret; getColumn(col,&ret); return ret; }
  123. /// Copy the requested column into a Point3F.
  124. ///
  125. /// This drops the bottom-most row.
  126. void getColumn(S32 col, Point3F *cptr) const;
  127. Point3F getColumn3F(S32 col) const { Point3F ret; getColumn(col,&ret); return ret; }
  128. /// Set the specified column from a Point4F.
  129. void setColumn(S32 col, const Point4F& cptr);
  130. /// Set the specified column from a Point3F.
  131. ///
  132. /// The bottom-most row is not set.
  133. void setColumn(S32 col, const Point3F& cptr);
  134. /// Copy the specified row into a Point4F.
  135. void getRow(S32 row, Point4F *cptr) const;
  136. Point4F getRow4F(S32 row) const { Point4F ret; getRow(row,&ret); return ret; }
  137. /// Copy the specified row into a Point3F.
  138. ///
  139. /// Right-most item is dropped.
  140. void getRow(S32 row, Point3F *cptr) const;
  141. Point3F getRow3F(S32 row) const { Point3F ret; getRow(row,&ret); return ret; }
  142. /// Set the specified row from a Point4F.
  143. void setRow(S32 row, const Point4F& cptr);
  144. /// Set the specified row from a Point3F.
  145. ///
  146. /// The right-most item is not set.
  147. void setRow(S32 row, const Point3F& cptr);
  148. /// Get the position of the matrix.
  149. ///
  150. /// This is the 4th column of the matrix.
  151. Point3F getPosition() const;
  152. /// Set the position of the matrix.
  153. ///
  154. /// This is the 4th column of the matrix.
  155. void setPosition( const Point3F &pos ) { setColumn( 3, pos ); }
  156. /// Add the passed delta to the matrix position.
  157. void displace( const Point3F &delta );
  158. /// Get the x axis of the matrix.
  159. ///
  160. /// This is the 1st column of the matrix and is
  161. /// normally considered the right vector.
  162. VectorF getRightVector() const;
  163. /// Get the y axis of the matrix.
  164. ///
  165. /// This is the 2nd column of the matrix and is
  166. /// normally considered the forward vector.
  167. VectorF getForwardVector() const;
  168. /// Get the z axis of the matrix.
  169. ///
  170. /// This is the 3rd column of the matrix and is
  171. /// normally considered the up vector.
  172. VectorF getUpVector() const;
  173. MatrixF& mul(const MatrixF &a); ///< M * a -> M
  174. MatrixF& mulL(const MatrixF &a); ///< a * M -> M
  175. MatrixF& mul(const MatrixF &a, const MatrixF &b); ///< a * b -> M
  176. // Scalar multiplies
  177. MatrixF& mul(const F32 a); ///< M * a -> M
  178. MatrixF& mul(const MatrixF &a, const F32 b); ///< a * b -> M
  179. void mul( Point4F& p ) const; ///< M * p -> p (full [4x4] * [1x4])
  180. void mulP( Point3F& p ) const; ///< M * p -> p (assume w = 1.0f)
  181. void mulP( const Point3F &p, Point3F *d) const; ///< M * p -> d (assume w = 1.0f)
  182. void mulV( VectorF& p ) const; ///< M * v -> v (assume w = 0.0f)
  183. void mulV( const VectorF &p, Point3F *d) const; ///< M * v -> d (assume w = 0.0f)
  184. void mul(Box3F& b) const; ///< Axial box -> Axial Box
  185. MatrixF& add( const MatrixF& m );
  186. /// <summary>
  187. /// Turns this matrix into a view matrix that looks at target.
  188. /// </summary>
  189. /// <param name="eye">The eye position.</param>
  190. /// <param name="target">The target position/direction.</param>
  191. /// <param name="up">The up direction.</param>
  192. void LookAt(const VectorF& eye, const VectorF& target, const VectorF& up);
  193. /// Convenience function to allow people to treat this like an array.
  194. F32& operator ()(S32 row, S32 col) { return m[idx(col,row)]; }
  195. F32 operator ()(S32 row, S32 col) const { return m[idx(col,row)]; }
  196. void dumpMatrix(const char *caption=NULL) const;
  197. // Math operator overloads
  198. //------------------------------------
  199. friend MatrixF operator * ( const MatrixF &m1, const MatrixF &m2 );
  200. MatrixF& operator *= ( const MatrixF &m );
  201. MatrixF &operator = (const MatrixF &m);
  202. bool isNaN();
  203. // Static identity matrix
  204. const static MatrixF Identity;
  205. };
  206. class MatrixFEngineExport
  207. {
  208. public:
  209. static EngineFieldTable::Field getMatrixField();
  210. };
  211. //--------------------------------------
  212. // Inline Functions
  213. inline MatrixF::MatrixF(bool _identity)
  214. {
  215. if (_identity)
  216. identity();
  217. else
  218. std::fill_n(m, 16, 0);
  219. }
  220. inline MatrixF::MatrixF( const EulerF &e )
  221. {
  222. set(e);
  223. }
  224. inline MatrixF::MatrixF( const EulerF &e, const Point3F& p )
  225. {
  226. set(e,p);
  227. }
  228. inline MatrixF& MatrixF::set( const EulerF &e)
  229. {
  230. m_matF_set_euler( e, *this );
  231. return (*this);
  232. }
  233. inline MatrixF& MatrixF::set( const EulerF &e, const Point3F& p)
  234. {
  235. m_matF_set_euler_point( e, p, *this );
  236. return (*this);
  237. }
  238. inline MatrixF& MatrixF::setCrossProduct( const Point3F &p)
  239. {
  240. m[1] = -(m[4] = p.z);
  241. m[8] = -(m[2] = p.y);
  242. m[6] = -(m[9] = p.x);
  243. m[0] = m[3] = m[5] = m[7] = m[10] = m[11] =
  244. m[12] = m[13] = m[14] = 0.0f;
  245. m[15] = 1;
  246. return (*this);
  247. }
  248. inline MatrixF& MatrixF::setTensorProduct( const Point3F &p, const Point3F &q)
  249. {
  250. m[0] = p.x * q.x;
  251. m[1] = p.x * q.y;
  252. m[2] = p.x * q.z;
  253. m[4] = p.y * q.x;
  254. m[5] = p.y * q.y;
  255. m[6] = p.y * q.z;
  256. m[8] = p.z * q.x;
  257. m[9] = p.z * q.y;
  258. m[10] = p.z * q.z;
  259. m[3] = m[7] = m[11] = m[12] = m[13] = m[14] = 0.0f;
  260. m[15] = 1.0f;
  261. return (*this);
  262. }
  263. inline bool MatrixF::isIdentity() const
  264. {
  265. return
  266. m[0] == 1.0f &&
  267. m[1] == 0.0f &&
  268. m[2] == 0.0f &&
  269. m[3] == 0.0f &&
  270. m[4] == 0.0f &&
  271. m[5] == 1.0f &&
  272. m[6] == 0.0f &&
  273. m[7] == 0.0f &&
  274. m[8] == 0.0f &&
  275. m[9] == 0.0f &&
  276. m[10] == 1.0f &&
  277. m[11] == 0.0f &&
  278. m[12] == 0.0f &&
  279. m[13] == 0.0f &&
  280. m[14] == 0.0f &&
  281. m[15] == 1.0f;
  282. }
  283. inline MatrixF& MatrixF::identity()
  284. {
  285. m[0] = 1.0f;
  286. m[1] = 0.0f;
  287. m[2] = 0.0f;
  288. m[3] = 0.0f;
  289. m[4] = 0.0f;
  290. m[5] = 1.0f;
  291. m[6] = 0.0f;
  292. m[7] = 0.0f;
  293. m[8] = 0.0f;
  294. m[9] = 0.0f;
  295. m[10] = 1.0f;
  296. m[11] = 0.0f;
  297. m[12] = 0.0f;
  298. m[13] = 0.0f;
  299. m[14] = 0.0f;
  300. m[15] = 1.0f;
  301. return (*this);
  302. }
  303. inline MatrixF& MatrixF::inverse()
  304. {
  305. m_matF_inverse(m);
  306. return (*this);
  307. }
  308. inline void MatrixF::invertTo( MatrixF *out )
  309. {
  310. m_matF_invert_to(m,*out);
  311. }
  312. inline MatrixF& MatrixF::affineInverse()
  313. {
  314. // AssertFatal(isAffine() == true, "Error, this matrix is not an affine transform");
  315. m_matF_affineInverse(m);
  316. return (*this);
  317. }
  318. inline MatrixF& MatrixF::transpose()
  319. {
  320. m_matF_transpose(m);
  321. return (*this);
  322. }
  323. inline MatrixF& MatrixF::scale(const Point3F& p)
  324. {
  325. m_matF_scale(m,p);
  326. return *this;
  327. }
  328. inline Point3F MatrixF::getScale() const
  329. {
  330. Point3F scale;
  331. scale.x = mSqrt(m[0]*m[0] + m[4] * m[4] + m[8] * m[8]);
  332. scale.y = mSqrt(m[1]*m[1] + m[5] * m[5] + m[9] * m[9]);
  333. scale.z = mSqrt(m[2]*m[2] + m[6] * m[6] + m[10] * m[10]);
  334. return scale;
  335. }
  336. inline void MatrixF::normalize()
  337. {
  338. m_matF_normalize(m);
  339. }
  340. inline MatrixF& MatrixF::mul( const MatrixF &a )
  341. { // M * a -> M
  342. AssertFatal(&a != this, "MatrixF::mul - a.mul(a) is invalid!");
  343. MatrixF tempThis(*this);
  344. m_matF_x_matF(tempThis, a, *this);
  345. return (*this);
  346. }
  347. inline MatrixF& MatrixF::mulL( const MatrixF &a )
  348. { // a * M -> M
  349. AssertFatal(&a != this, "MatrixF::mulL - a.mul(a) is invalid!");
  350. MatrixF tempThis(*this);
  351. m_matF_x_matF(a, tempThis, *this);
  352. return (*this);
  353. }
  354. inline MatrixF& MatrixF::mul( const MatrixF &a, const MatrixF &b )
  355. { // a * b -> M
  356. AssertFatal((&a != this) && (&b != this), "MatrixF::mul - a.mul(a, b) a.mul(b, a) a.mul(a, a) is invalid!");
  357. m_matF_x_matF(a, b, *this);
  358. return (*this);
  359. }
  360. inline MatrixF& MatrixF::mul(const F32 a)
  361. {
  362. for (U32 i = 0; i < 16; i++)
  363. m[i] *= a;
  364. return *this;
  365. }
  366. inline MatrixF& MatrixF::mul(const MatrixF &a, const F32 b)
  367. {
  368. *this = a;
  369. mul(b);
  370. return *this;
  371. }
  372. inline void MatrixF::mul( Point4F& p ) const
  373. {
  374. Point4F temp;
  375. m_matF_x_point4F(*this, &p.x, &temp.x);
  376. p = temp;
  377. }
  378. inline void MatrixF::mulP( Point3F& p) const
  379. {
  380. // M * p -> d
  381. Point3F d;
  382. m_matF_x_point3F(*this, &p.x, &d.x);
  383. p = d;
  384. }
  385. inline void MatrixF::mulP( const Point3F &p, Point3F *d) const
  386. {
  387. // M * p -> d
  388. m_matF_x_point3F(*this, &p.x, &d->x);
  389. }
  390. inline void MatrixF::mulV( VectorF& v) const
  391. {
  392. // M * v -> v
  393. VectorF temp;
  394. m_matF_x_vectorF(*this, &v.x, &temp.x);
  395. v = temp;
  396. }
  397. inline void MatrixF::mulV( const VectorF &v, Point3F *d) const
  398. {
  399. // M * v -> d
  400. m_matF_x_vectorF(*this, &v.x, &d->x);
  401. }
  402. inline void MatrixF::mul(Box3F& b) const
  403. {
  404. m_matF_x_box3F(*this, &b.minExtents.x, &b.maxExtents.x);
  405. }
  406. inline MatrixF& MatrixF::add( const MatrixF& a )
  407. {
  408. for( U32 i = 0; i < 16; ++ i )
  409. m[ i ] += a.m[ i ];
  410. return *this;
  411. }
  412. inline void MatrixF::LookAt(const VectorF& eye, const VectorF& target, const VectorF& up)
  413. {
  414. // Calculate the forward vector (camera direction).
  415. VectorF zAxis = target; // Camera looks towards the target
  416. zAxis.normalize();
  417. // Calculate the right vector.
  418. VectorF xAxis = mCross(up, zAxis);
  419. xAxis.normalize();
  420. // Recalculate the up vector.
  421. VectorF yAxis = mCross(zAxis, xAxis);
  422. // Set the rotation part of the matrix (camera axes).
  423. setColumn(0, xAxis); // Right
  424. setColumn(1, zAxis); // Forward
  425. setColumn(2, yAxis); // Up
  426. // Set the translation part (camera position).
  427. setPosition(eye);
  428. }
  429. inline void MatrixF::getColumn(S32 col, Point4F *cptr) const
  430. {
  431. cptr->x = m[col];
  432. cptr->y = m[col+4];
  433. cptr->z = m[col+8];
  434. cptr->w = m[col+12];
  435. }
  436. inline void MatrixF::getColumn(S32 col, Point3F *cptr) const
  437. {
  438. cptr->x = m[col];
  439. cptr->y = m[col+4];
  440. cptr->z = m[col+8];
  441. }
  442. inline void MatrixF::setColumn(S32 col, const Point4F &cptr)
  443. {
  444. m[col] = cptr.x;
  445. m[col+4] = cptr.y;
  446. m[col+8] = cptr.z;
  447. m[col+12]= cptr.w;
  448. }
  449. inline void MatrixF::setColumn(S32 col, const Point3F &cptr)
  450. {
  451. m[col] = cptr.x;
  452. m[col+4] = cptr.y;
  453. m[col+8] = cptr.z;
  454. }
  455. inline void MatrixF::getRow(S32 col, Point4F *cptr) const
  456. {
  457. col *= 4;
  458. cptr->x = m[col++];
  459. cptr->y = m[col++];
  460. cptr->z = m[col++];
  461. cptr->w = m[col];
  462. }
  463. inline void MatrixF::getRow(S32 col, Point3F *cptr) const
  464. {
  465. col *= 4;
  466. cptr->x = m[col++];
  467. cptr->y = m[col++];
  468. cptr->z = m[col];
  469. }
  470. inline void MatrixF::setRow(S32 col, const Point4F &cptr)
  471. {
  472. col *= 4;
  473. m[col++] = cptr.x;
  474. m[col++] = cptr.y;
  475. m[col++] = cptr.z;
  476. m[col] = cptr.w;
  477. }
  478. inline void MatrixF::setRow(S32 col, const Point3F &cptr)
  479. {
  480. col *= 4;
  481. m[col++] = cptr.x;
  482. m[col++] = cptr.y;
  483. m[col] = cptr.z;
  484. }
  485. inline Point3F MatrixF::getPosition() const
  486. {
  487. return Point3F( m[3], m[3+4], m[3+8] );
  488. }
  489. inline void MatrixF::displace( const Point3F &delta )
  490. {
  491. m[3] += delta.x;
  492. m[3+4] += delta.y;
  493. m[3+8] += delta.z;
  494. }
  495. inline VectorF MatrixF::getForwardVector() const
  496. {
  497. VectorF vec;
  498. getColumn( 1, &vec );
  499. return vec;
  500. }
  501. inline VectorF MatrixF::getRightVector() const
  502. {
  503. VectorF vec;
  504. getColumn( 0, &vec );
  505. return vec;
  506. }
  507. inline VectorF MatrixF::getUpVector() const
  508. {
  509. VectorF vec;
  510. getColumn( 2, &vec );
  511. return vec;
  512. }
  513. //------------------------------------
  514. // Math operator overloads
  515. //------------------------------------
  516. inline MatrixF operator * ( const MatrixF &m1, const MatrixF &m2 )
  517. {
  518. // temp = m1 * m2
  519. MatrixF temp;
  520. m_matF_x_matF(m1, m2, temp);
  521. return temp;
  522. }
  523. inline MatrixF& MatrixF::operator *= ( const MatrixF &m1 )
  524. {
  525. MatrixF tempThis(*this);
  526. m_matF_x_matF(tempThis, m1, *this);
  527. return (*this);
  528. }
  529. inline MatrixF &MatrixF::operator = (const MatrixF &m1)
  530. {
  531. for (U32 i=0;i<16;i++)
  532. this->m[i] = m1.m[i];
  533. return (*this);
  534. }
  535. inline bool MatrixF::isNaN()
  536. {
  537. bool isaNaN = false;
  538. for (U32 i = 0; i < 16; i++)
  539. if (mIsNaN_F(m[i]))
  540. isaNaN = true;
  541. return isaNaN;
  542. }
  543. //------------------------------------
  544. // Non-member methods
  545. //------------------------------------
  546. inline void mTransformPlane(const MatrixF& mat, const Point3F& scale, const PlaneF& plane, PlaneF * result)
  547. {
  548. m_matF_x_scale_x_planeF(mat, &scale.x, &plane.x, &result->x);
  549. }
  550. #else // !USE_TEMPLATE_MATRIX
  551. //------------------------------------
  552. // Templatized matrix class to replace MATRIXF above
  553. //------------------------------------
  554. template<typename DATA_TYPE, U32 rows, U32 cols>
  555. class Matrix {
  556. friend class MatrixTemplateExport;
  557. private:
  558. DATA_TYPE data[rows * cols];
  559. public:
  560. static_assert(rows >= 2 && cols >= 2, "Matrix must have at least 2 rows and 2 cols.");
  561. // ------ Setters and initializers ------
  562. explicit Matrix(bool identity = false) {
  563. std::fill(data, data + (rows * cols), DATA_TYPE(0));
  564. if (identity) {
  565. for (U32 i = 0; i < rows; i++) {
  566. for (U32 j = 0; j < cols; j++) {
  567. // others already get filled with 0
  568. if (j == i)
  569. (*this)(i, j) = static_cast<DATA_TYPE>(1);
  570. }
  571. }
  572. }
  573. }
  574. explicit Matrix(const EulerF& e) {
  575. set(e);
  576. }
  577. ~Matrix() = default;
  578. /// Make this an identity matrix.
  579. Matrix<DATA_TYPE, rows, cols>& identity();
  580. void reverseProjection();
  581. void normalize();
  582. Matrix<DATA_TYPE, rows, cols>& set(const EulerF& e);
  583. Matrix(const EulerF& e, const Point3F p);
  584. Matrix<DATA_TYPE, rows, cols>& set(const EulerF& e, const Point3F p);
  585. Matrix<DATA_TYPE, rows, cols>& inverse();
  586. Matrix<DATA_TYPE, rows, cols>& transpose();
  587. void invert();
  588. Matrix<DATA_TYPE, rows, cols>& setCrossProduct(const Point3F& p);
  589. Matrix<DATA_TYPE, rows, cols>& setTensorProduct(const Point3F& p, const Point3F& q);
  590. /// M * Matrix(p) -> M
  591. Matrix<DATA_TYPE, rows, cols>& scale(const Point3F& s);
  592. Matrix<DATA_TYPE, rows, cols>& scale(DATA_TYPE s) { return scale(Point3F(s, s, s)); }
  593. void setColumn(S32 col, const Point4F& cptr);
  594. void setColumn(S32 col, const Point3F& cptr);
  595. void setRow(S32 row, const Point4F& cptr);
  596. void setRow(S32 row, const Point3F& cptr);
  597. void displace(const Point3F& delta);
  598. bool fullInverse();
  599. void setPosition(const Point3F& pos) { setColumn(3, pos); }
  600. DATA_TYPE determinant() const {
  601. AssertFatal(rows == cols, "Determinant is only defined for square matrices.");
  602. // For simplicity, only implement for 3x3 matrices
  603. AssertFatal(rows >= 3 && cols >= 3, "Determinant only for 3x3 or more"); // Ensure the matrix is 3x3
  604. return (*this)(0, 0) * ((*this)(1, 1) * (*this)(2, 2) - (*this)(1, 2) * (*this)(2, 1)) +
  605. (*this)(1, 0) * ((*this)(0, 2) * (*this)(2, 1) - (*this)(0, 1) * (*this)(2, 2)) +
  606. (*this)(2, 0) * ((*this)(0, 1) * (*this)(1, 2) - (*this)(0, 2) * (*this)(1, 1));
  607. }
  608. ///< M * a -> M
  609. Matrix<DATA_TYPE, rows, cols>& mul(const Matrix<DATA_TYPE, rows, cols>& a)
  610. { return *this = *this * a; }
  611. ///< a * M -> M
  612. Matrix<DATA_TYPE, rows, cols>& mulL(const Matrix<DATA_TYPE, rows, cols>& a)
  613. { return *this = a * *this; }
  614. ///< a * b -> M
  615. Matrix<DATA_TYPE, rows, cols>& mul(const Matrix<DATA_TYPE, rows, cols>& a, const Matrix<DATA_TYPE, rows, cols>& b)
  616. { return *this = a * b; }
  617. ///< M * a -> M
  618. Matrix<DATA_TYPE, rows, cols>& mul(const F32 a)
  619. { return *this = *this * a; }
  620. ///< a * b -> M
  621. Matrix<DATA_TYPE, rows, cols>& mul(const Matrix<DATA_TYPE, rows, cols>& a, const F32 b)
  622. { return *this = a * b; }
  623. Matrix<DATA_TYPE, rows, cols>& add(const Matrix<DATA_TYPE, rows, cols>& a)
  624. {
  625. return *this = *this += a;
  626. }
  627. ///< M * p -> p (full [4x4] * [1x4])
  628. void mul(Point4F& p) const { p = *this * p; }
  629. ///< M * p -> p (assume w = 1.0f)
  630. void mulP(Point3F& p) const {
  631. Point3F result;
  632. result.x = (*this)(0, 0) * p.x + (*this)(0, 1) * p.y + (*this)(0, 2) * p.z + (*this)(0, 3);
  633. result.y = (*this)(1, 0) * p.x + (*this)(1, 1) * p.y + (*this)(1, 2) * p.z + (*this)(1, 3);
  634. result.z = (*this)(2, 0) * p.x + (*this)(2, 1) * p.y + (*this)(2, 2) * p.z + (*this)(2, 3);
  635. p = result;
  636. }
  637. ///< M * p -> d (assume w = 1.0f)
  638. void mulP(const Point3F& p, Point3F* d) const { *d = *this * p; }
  639. ///< M * v -> v (assume w = 0.0f)
  640. void mulV(VectorF& v) const
  641. {
  642. AssertFatal(rows == 4 && cols == 4, "Multiplying VectorF with matrix requires 4x4");
  643. VectorF result(
  644. (*this)(0, 0) * v.x + (*this)(0, 1) * v.y + (*this)(0, 2) * v.z,
  645. (*this)(1, 0) * v.x + (*this)(1, 1) * v.y + (*this)(1, 2) * v.z,
  646. (*this)(2, 0) * v.x + (*this)(2, 1) * v.y + (*this)(2, 2) * v.z
  647. );
  648. v = result;
  649. }
  650. ///< M * v -> d (assume w = 0.0f)
  651. void mulV(const VectorF& v, Point3F* d) const
  652. {
  653. AssertFatal(rows == 4 && cols == 4, "Multiplying VectorF with matrix requires 4x4");
  654. VectorF result(
  655. (*this)(0, 0) * v.x + (*this)(0, 1) * v.y + (*this)(0, 2) * v.z,
  656. (*this)(1, 0) * v.x + (*this)(1, 1) * v.y + (*this)(1, 2) * v.z,
  657. (*this)(2, 0) * v.x + (*this)(2, 1) * v.y + (*this)(2, 2) * v.z
  658. );
  659. d->x = result.x;
  660. d->y = result.y;
  661. d->z = result.z;
  662. }
  663. ///< Axial box -> Axial Box (too big a function to be inline)
  664. void mul(Box3F& box) const;
  665. // ------ Getters ------
  666. bool isNaN() {
  667. for (U32 i = 0; i < rows; i++) {
  668. for (U32 j = 0; j < cols; j++) {
  669. if (mIsNaN_F((*this)(i, j)))
  670. return true;
  671. }
  672. }
  673. return false;
  674. }
  675. // row + col * cols
  676. static U32 idx(U32 i, U32 j) { return (i + j * cols); }
  677. bool isAffine() const;
  678. bool isIdentity() const;
  679. /// Take inverse of matrix assuming it is affine (rotation,
  680. /// scale, sheer, translation only).
  681. Matrix<DATA_TYPE, rows, cols>& affineInverse();
  682. Point3F getScale() const;
  683. EulerF toEuler() const;
  684. Point3F getPosition() const;
  685. void getColumn(S32 col, Point4F* cptr) const;
  686. Point4F getColumn4F(S32 col) const { Point4F ret; getColumn(col, &ret); return ret; }
  687. void getColumn(S32 col, Point3F* cptr) const;
  688. Point3F getColumn3F(S32 col) const { Point3F ret; getColumn(col, &ret); return ret; }
  689. void getRow(S32 row, Point4F* cptr) const;
  690. Point4F getRow4F(S32 row) const { Point4F ret; getRow(row, &ret); return ret; }
  691. void getRow(S32 row, Point3F* cptr) const;
  692. Point3F getRow3F(S32 row) const { Point3F ret; getRow(row, &ret); return ret; }
  693. VectorF getRightVector() const;
  694. VectorF getForwardVector() const;
  695. VectorF getUpVector() const;
  696. DATA_TYPE* getData() {
  697. return data;
  698. }
  699. const DATA_TYPE* getData() const {
  700. return data;
  701. }
  702. void transposeTo(Matrix<DATA_TYPE, cols, rows>& matrix) const {
  703. for (U32 i = 0; i < rows; ++i) {
  704. for (U32 j = 0; j < cols; ++j) {
  705. matrix(j, i) = (*this)(i, j);
  706. }
  707. }
  708. }
  709. void swap(DATA_TYPE& a, DATA_TYPE& b) {
  710. DATA_TYPE temp = a;
  711. a = b;
  712. b = temp;
  713. }
  714. void invertTo(Matrix<DATA_TYPE, cols, rows>* matrix) const;
  715. void dumpMatrix(const char* caption = NULL) const;
  716. // Static identity matrix
  717. static const Matrix Identity;
  718. // ------ Operators ------
  719. friend Matrix<DATA_TYPE, rows, cols> operator*(const Matrix<DATA_TYPE, rows, cols>& m1, const Matrix<DATA_TYPE, rows, cols>& m2) {
  720. Matrix<DATA_TYPE, rows, cols> result;
  721. for (U32 i = 0; i < rows; ++i) {
  722. for (U32 j = 0; j < cols; ++j)
  723. {
  724. result(i, j) = static_cast<DATA_TYPE>(0);
  725. for (U32 k = 0; k < cols; ++k)
  726. {
  727. result(i, j) += m1(i, k) * m2(k, j);
  728. }
  729. }
  730. }
  731. return result;
  732. }
  733. Matrix<DATA_TYPE, rows, cols> operator *= (const Matrix<DATA_TYPE, rows, cols>& other) {
  734. *this = *this * other;
  735. return *this;
  736. }
  737. Matrix<DATA_TYPE, rows, cols> operator+(const Matrix<DATA_TYPE, rows, cols>& m2) {
  738. Matrix<DATA_TYPE, rows, cols> result;
  739. for (U32 i = 0; i < rows; ++i)
  740. {
  741. for (U32 j = 0; j < cols; ++j)
  742. {
  743. result(i, j) = 0; // Initialize result element to 0
  744. result(i, j) = (*this)(i, j) + m2(i, j);
  745. }
  746. }
  747. return result;
  748. }
  749. Matrix<DATA_TYPE, rows, cols> operator+=(const Matrix<DATA_TYPE, rows, cols>& m2) {
  750. for (U32 i = 0; i < rows; ++i)
  751. {
  752. for (U32 j = 0; j < cols; ++j)
  753. {
  754. (*this)(i, j) += m2(i, j);
  755. }
  756. }
  757. return (*this);
  758. }
  759. Matrix<DATA_TYPE, rows, cols> operator * (const DATA_TYPE scalar) const {
  760. Matrix<DATA_TYPE, rows, cols> result;
  761. for (U32 i = 0; i < rows; i++)
  762. {
  763. for (U32 j = 0; j < cols; j++)
  764. {
  765. result(i, j) = (*this)(i, j) * scalar;
  766. }
  767. }
  768. return result;
  769. }
  770. Matrix<DATA_TYPE, rows, cols>& operator *= (const DATA_TYPE scalar) {
  771. for (U32 i = 0; i < rows; i++)
  772. {
  773. for (U32 j = 0; j < cols; j++)
  774. {
  775. (*this)(i, j) *= scalar;
  776. }
  777. }
  778. return *this;
  779. }
  780. Point3F operator*(const Point3F& point) const {
  781. AssertFatal(rows == 4 && cols == 4, "Multiplying point3 with matrix requires 4x4");
  782. Point3F result;
  783. result.x = (*this)(0, 0) * point.x + (*this)(0, 1) * point.y + (*this)(0, 2) * point.z + (*this)(0, 3);
  784. result.y = (*this)(1, 0) * point.x + (*this)(1, 1) * point.y + (*this)(1, 2) * point.z + (*this)(1, 3);
  785. result.z = (*this)(2, 0) * point.x + (*this)(2, 1) * point.y + (*this)(2, 2) * point.z + (*this)(2, 3);
  786. return result;
  787. }
  788. Point4F operator*(const Point4F& point) const {
  789. AssertFatal(rows == 4 && cols == 4, "Multiplying point4 with matrix requires 4x4");
  790. return Point4F(
  791. (*this)(0, 0) * point.x + (*this)(0, 1) * point.y + (*this)(0, 2) * point.z + (*this)(0, 3) * point.w,
  792. (*this)(1, 0) * point.x + (*this)(1, 1) * point.y + (*this)(1, 2) * point.z + (*this)(1, 3) * point.w,
  793. (*this)(2, 0) * point.x + (*this)(2, 1) * point.y + (*this)(2, 2) * point.z + (*this)(2, 3) * point.w,
  794. (*this)(3, 0) * point.x + (*this)(3, 1) * point.y + (*this)(3, 2) * point.z + (*this)(3, 3) * point.w
  795. );
  796. }
  797. Matrix<DATA_TYPE, rows, cols>& operator = (const Matrix<DATA_TYPE, rows, cols>& other) {
  798. if (this != &other) {
  799. std::copy(other.data, other.data + rows * cols, this->data);
  800. }
  801. return *this;
  802. }
  803. bool operator == (const Matrix<DATA_TYPE, rows, cols>& other) const {
  804. for (U32 i = 0; i < rows; i++)
  805. {
  806. for (U32 j = 0; j < cols; j++)
  807. {
  808. if ((*this)(i, j) != other(i, j))
  809. return false;
  810. }
  811. }
  812. return true;
  813. }
  814. bool operator != (const Matrix<DATA_TYPE, rows, cols>& other) const {
  815. return !(*this == other);
  816. }
  817. operator DATA_TYPE* () { return (data); }
  818. operator const DATA_TYPE* () const { return (DATA_TYPE*)(data); }
  819. DATA_TYPE& operator () (U32 row, U32 col) {
  820. if (row >= rows || col >= cols)
  821. AssertFatal(false, "Matrix indices out of range");
  822. return data[idx(col,row)];
  823. }
  824. DATA_TYPE operator () (U32 row, U32 col) const {
  825. if (row >= rows || col >= cols)
  826. AssertFatal(false, "Matrix indices out of range");
  827. return data[idx(col, row)];
  828. }
  829. };
  830. //--------------------------------------------
  831. // INLINE FUNCTIONS
  832. //--------------------------------------------
  833. template<typename DATA_TYPE, U32 rows, U32 cols>
  834. inline Matrix<DATA_TYPE, rows, cols>& Matrix<DATA_TYPE, rows, cols>::transpose()
  835. {
  836. AssertFatal(rows == cols, "Transpose can only be performed on square matrices.");
  837. swap((*this)(0, 1), (*this)(1, 0));
  838. swap((*this)(0, 2), (*this)(2, 0));
  839. swap((*this)(0, 3), (*this)(3, 0));
  840. swap((*this)(1, 2), (*this)(2, 1));
  841. swap((*this)(1, 3), (*this)(3, 1));
  842. swap((*this)(2, 3), (*this)(3, 2));
  843. return (*this);
  844. }
  845. template<typename DATA_TYPE, U32 rows, U32 cols>
  846. inline Matrix<DATA_TYPE, rows, cols>& Matrix<DATA_TYPE, rows, cols>::identity()
  847. {
  848. for (U32 i = 0; i < rows; i++)
  849. {
  850. for (U32 j = 0; j < cols; j++)
  851. {
  852. if (j == i)
  853. (*this)(i, j) = static_cast<DATA_TYPE>(1);
  854. else
  855. (*this)(i, j) = static_cast<DATA_TYPE>(0);
  856. }
  857. }
  858. return (*this);
  859. }
  860. template<typename DATA_TYPE, U32 rows, U32 cols>
  861. inline void Matrix<DATA_TYPE, rows, cols>::normalize()
  862. {
  863. AssertFatal(rows >= 3 && cols >= 3, "Normalize can only be applied 3x3 or more");
  864. Point3F col0, col1, col2;
  865. getColumn(0, &col0);
  866. getColumn(1, &col1);
  867. mCross(col0, col1, &col2);
  868. mCross(col2, col0, &col1);
  869. col0.normalize();
  870. col1.normalize();
  871. col2.normalize();
  872. setColumn(0, col0);
  873. setColumn(1, col1);
  874. setColumn(2, col2);
  875. }
  876. template<typename DATA_TYPE, U32 rows, U32 cols>
  877. inline Matrix<DATA_TYPE, rows, cols>& Matrix<DATA_TYPE, rows, cols>::scale(const Point3F& s)
  878. {
  879. // torques scale applies directly, does not create another matrix to multiply with the translation matrix.
  880. AssertFatal(rows >= 4 && cols >= 4, "Scale can only be applied 4x4 or more");
  881. (*this)(0, 0) *= s.x; (*this)(0, 1) *= s.y; (*this)(0, 2) *= s.z;
  882. (*this)(1, 0) *= s.x; (*this)(1, 1) *= s.y; (*this)(1, 2) *= s.z;
  883. (*this)(2, 0) *= s.x; (*this)(2, 1) *= s.y; (*this)(2, 2) *= s.z;
  884. (*this)(3, 0) *= s.x; (*this)(3, 1) *= s.y; (*this)(3, 2) *= s.z;
  885. return (*this);
  886. }
  887. template<typename DATA_TYPE, U32 rows, U32 cols>
  888. inline bool Matrix<DATA_TYPE, rows, cols>::isIdentity() const {
  889. for (U32 i = 0; i < rows; i++)
  890. {
  891. for (U32 j = 0; j < cols; j++)
  892. {
  893. if (j == i)
  894. {
  895. if((*this)(i, j) != static_cast<DATA_TYPE>(1))
  896. {
  897. return false;
  898. }
  899. }
  900. else
  901. {
  902. if((*this)(i, j) != static_cast<DATA_TYPE>(0))
  903. {
  904. return false;
  905. }
  906. }
  907. }
  908. }
  909. return true;
  910. }
  911. template<typename DATA_TYPE, U32 rows, U32 cols>
  912. inline Point3F Matrix<DATA_TYPE, rows, cols>::getScale() const
  913. {
  914. // this function assumes the matrix has scale applied through the scale(const Point3F& s) function.
  915. // for now assume float since we have point3F.
  916. AssertFatal(rows >= 4 && cols >= 4, "Scale can only be applied 4x4 or more");
  917. Point3F scale;
  918. scale.x = mSqrt((*this)(0, 0) * (*this)(0, 0) + (*this)(1, 0) * (*this)(1, 0) + (*this)(2, 0) * (*this)(2, 0));
  919. scale.y = mSqrt((*this)(0, 1) * (*this)(0, 1) + (*this)(1, 1) * (*this)(1, 1) + (*this)(2, 1) * (*this)(2, 1));
  920. scale.z = mSqrt((*this)(0, 2) * (*this)(0, 2) + (*this)(1, 2) * (*this)(1, 2) + (*this)(2, 2) * (*this)(2, 2));
  921. return scale;
  922. }
  923. template<typename DATA_TYPE, U32 rows, U32 cols>
  924. inline Point3F Matrix<DATA_TYPE, rows, cols>::getPosition() const
  925. {
  926. Point3F pos;
  927. getColumn(3, &pos);
  928. return pos;
  929. }
  930. template<typename DATA_TYPE, U32 rows, U32 cols>
  931. inline void Matrix<DATA_TYPE, rows, cols>::getColumn(S32 col, Point4F* cptr) const
  932. {
  933. if (rows >= 2)
  934. {
  935. cptr->x = (*this)(0, col);
  936. cptr->y = (*this)(1, col);
  937. }
  938. if (rows >= 3)
  939. cptr->z = (*this)(2, col);
  940. else
  941. cptr->z = 0.0f;
  942. if (rows >= 4)
  943. cptr->w = (*this)(3, col);
  944. else
  945. cptr->w = 0.0f;
  946. }
  947. template<typename DATA_TYPE, U32 rows, U32 cols>
  948. inline void Matrix<DATA_TYPE, rows, cols>::getColumn(S32 col, Point3F* cptr) const
  949. {
  950. if (rows >= 2)
  951. {
  952. cptr->x = (*this)(0, col);
  953. cptr->y = (*this)(1, col);
  954. }
  955. if (rows >= 3)
  956. cptr->z = (*this)(2, col);
  957. else
  958. cptr->z = 0.0f;
  959. }
  960. template<typename DATA_TYPE, U32 rows, U32 cols>
  961. inline void Matrix<DATA_TYPE, rows, cols>::setColumn(S32 col, const Point4F &cptr) {
  962. if(rows >= 2)
  963. {
  964. (*this)(0, col) = cptr.x;
  965. (*this)(1, col) = cptr.y;
  966. }
  967. if(rows >= 3)
  968. (*this)(2, col) = cptr.z;
  969. if(rows >= 4)
  970. (*this)(3, col) = cptr.w;
  971. }
  972. template<typename DATA_TYPE, U32 rows, U32 cols>
  973. inline void Matrix<DATA_TYPE, rows, cols>::setColumn(S32 col, const Point3F &cptr) {
  974. if(rows >= 2)
  975. {
  976. (*this)(0, col) = cptr.x;
  977. (*this)(1, col) = cptr.y;
  978. }
  979. if(rows >= 3)
  980. (*this)(2, col) = cptr.z;
  981. }
  982. template<typename DATA_TYPE, U32 rows, U32 cols>
  983. inline void Matrix<DATA_TYPE, rows, cols>::getRow(S32 row, Point4F* cptr) const
  984. {
  985. if (cols >= 2)
  986. {
  987. cptr->x = (*this)(row, 0);
  988. cptr->y = (*this)(row, 1);
  989. }
  990. if (cols >= 3)
  991. cptr->z = (*this)(row, 2);
  992. else
  993. cptr->z = 0.0f;
  994. if (cols >= 4)
  995. cptr->w = (*this)(row, 3);
  996. else
  997. cptr->w = 0.0f;
  998. }
  999. template<typename DATA_TYPE, U32 rows, U32 cols>
  1000. inline void Matrix<DATA_TYPE, rows, cols>::getRow(S32 row, Point3F* cptr) const
  1001. {
  1002. if (cols >= 2)
  1003. {
  1004. cptr->x = (*this)(row, 0);
  1005. cptr->y = (*this)(row, 1);
  1006. }
  1007. if (cols >= 3)
  1008. cptr->z = (*this)(row, 2);
  1009. else
  1010. cptr->z = 0.0f;
  1011. }
  1012. template<typename DATA_TYPE, U32 rows, U32 cols>
  1013. inline VectorF Matrix<DATA_TYPE, rows, cols>::getRightVector() const
  1014. {
  1015. VectorF vec;
  1016. getColumn(0, &vec);
  1017. return vec;
  1018. }
  1019. template<typename DATA_TYPE, U32 rows, U32 cols>
  1020. inline VectorF Matrix<DATA_TYPE, rows, cols>::getForwardVector() const
  1021. {
  1022. VectorF vec;
  1023. getColumn(1, &vec);
  1024. return vec;
  1025. }
  1026. template<typename DATA_TYPE, U32 rows, U32 cols>
  1027. inline VectorF Matrix<DATA_TYPE, rows, cols>::getUpVector() const
  1028. {
  1029. VectorF vec;
  1030. getColumn(2, &vec);
  1031. return vec;
  1032. }
  1033. template<typename DATA_TYPE, U32 rows, U32 cols>
  1034. inline void Matrix<DATA_TYPE, rows, cols>::invertTo(Matrix<DATA_TYPE, cols, rows>* matrix) const
  1035. {
  1036. Matrix<DATA_TYPE, rows, cols> invMatrix;
  1037. for (U32 i = 0; i < rows; ++i)
  1038. {
  1039. for (U32 j = 0; j < cols; ++j)
  1040. {
  1041. invMatrix(i, j) = (*this)(i, j);
  1042. }
  1043. }
  1044. invMatrix.inverse();
  1045. for (U32 i = 0; i < rows; ++i)
  1046. {
  1047. for (U32 j = 0; j < cols; ++j)
  1048. {
  1049. (*matrix)(i, j) = invMatrix(i, j);
  1050. }
  1051. }
  1052. }
  1053. template<typename DATA_TYPE, U32 rows, U32 cols>
  1054. inline void Matrix<DATA_TYPE, rows, cols>::setRow(S32 row, const Point4F& cptr) {
  1055. if(cols >= 2)
  1056. {
  1057. (*this)(row, 0) = cptr.x;
  1058. (*this)(row, 1) = cptr.y;
  1059. }
  1060. if(cols >= 3)
  1061. (*this)(row, 2) = cptr.z;
  1062. if(cols >= 4)
  1063. (*this)(row, 3) = cptr.w;
  1064. }
  1065. template<typename DATA_TYPE, U32 rows, U32 cols>
  1066. inline void Matrix<DATA_TYPE, rows, cols>::setRow(S32 row, const Point3F& cptr) {
  1067. if(cols >= 2)
  1068. {
  1069. (*this)(row, 0) = cptr.x;
  1070. (*this)(row, 1) = cptr.y;
  1071. }
  1072. if(cols >= 3)
  1073. (*this)(row, 2) = cptr.z;
  1074. }
  1075. template<typename DATA_TYPE, U32 rows, U32 cols>
  1076. inline void Matrix<DATA_TYPE, rows, cols>::displace(const Point3F& delta)
  1077. {
  1078. (*this)(0, 3) += delta.x;
  1079. (*this)(1, 3) += delta.y;
  1080. (*this)(2, 3) += delta.z;
  1081. }
  1082. template<typename DATA_TYPE, U32 rows, U32 cols>
  1083. inline void Matrix<DATA_TYPE, rows, cols>::reverseProjection()
  1084. {
  1085. AssertFatal(rows == 4 && cols == 4, "reverseProjection requires a 4x4 matrix.");
  1086. (*this)(2, 0) = (*this)(3, 0) - (*this)(2, 0);
  1087. (*this)(2, 1) = (*this)(3, 1) - (*this)(2, 1);
  1088. (*this)(2, 2) = (*this)(3, 2) - (*this)(2, 2);
  1089. (*this)(2, 3) = (*this)(3, 3) - (*this)(2, 3);
  1090. }
  1091. template<typename DATA_TYPE, U32 rows, U32 cols>
  1092. const Matrix<DATA_TYPE, rows, cols> Matrix<DATA_TYPE, rows, cols>::Identity = []() {
  1093. Matrix<DATA_TYPE, rows, cols> identity(true);
  1094. return identity;
  1095. }();
  1096. template<typename DATA_TYPE, U32 rows, U32 cols>
  1097. inline Matrix<DATA_TYPE, rows, cols>& Matrix<DATA_TYPE, rows, cols>::set(const EulerF& e)
  1098. {
  1099. // when the template refactor is done, euler will be able to be setup in different ways
  1100. AssertFatal(rows >= 3 && cols >= 3, "EulerF can only initialize 3x3 or more");
  1101. static_assert(std::is_same<DATA_TYPE, float>::value, "Can only initialize eulers with floats for now");
  1102. F32 cosPitch, sinPitch;
  1103. mSinCos(e.x, sinPitch, cosPitch);
  1104. F32 cosYaw, sinYaw;
  1105. mSinCos(e.y, sinYaw, cosYaw);
  1106. F32 cosRoll, sinRoll;
  1107. mSinCos(e.z, sinRoll, cosRoll);
  1108. enum {
  1109. AXIS_X = (1 << 0),
  1110. AXIS_Y = (1 << 1),
  1111. AXIS_Z = (1 << 2)
  1112. };
  1113. U32 axis = 0;
  1114. if (e.x != 0.0f) axis |= AXIS_X;
  1115. if (e.y != 0.0f) axis |= AXIS_Y;
  1116. if (e.z != 0.0f) axis |= AXIS_Z;
  1117. switch (axis) {
  1118. case 0:
  1119. (*this) = Matrix<DATA_TYPE, rows, cols>(true);
  1120. break;
  1121. case AXIS_X:
  1122. (*this)(0, 0) = 1.0f; (*this)(0, 1) = 0.0f; (*this)(0, 2) = 0.0f;
  1123. (*this)(1, 0) = 0.0f; (*this)(1, 1) = cosPitch; (*this)(1, 2) = sinPitch;
  1124. (*this)(2, 0) = 0.0f; (*this)(2, 1) = -sinPitch; (*this)(2, 2) = cosPitch;
  1125. break;
  1126. case AXIS_Y:
  1127. (*this)(0, 0) = cosYaw; (*this)(1, 0) = 0.0f; (*this)(2, 0) = sinYaw;
  1128. (*this)(0, 1) = 0.0f; (*this)(1, 1) = 1.0f; (*this)(2, 1) = 0.0f;
  1129. (*this)(0, 2) = -sinYaw; (*this)(1, 2) = 0.0f; (*this)(2, 2) = cosYaw;
  1130. break;
  1131. case AXIS_Z:
  1132. (*this)(0, 0) = cosRoll; (*this)(0, 1) = sinRoll; (*this)(0, 2) = 0.0f;
  1133. (*this)(1, 0) = -sinRoll; (*this)(1, 1) = cosRoll; (*this)(1, 2) = 0.0f;
  1134. (*this)(2, 0) = 0.0f; (*this)(2, 1) = 0.0f; (*this)(2, 2) = 1.0f;
  1135. break;
  1136. default:
  1137. F32 r1 = cosYaw * cosRoll;
  1138. F32 r2 = cosYaw * sinRoll;
  1139. F32 r3 = sinYaw * cosRoll;
  1140. F32 r4 = sinYaw * sinRoll;
  1141. // the matrix looks like this:
  1142. // r1 - (r4 * sin(x)) r2 + (r3 * sin(x)) -cos(x) * sin(y)
  1143. // -cos(x) * sin(z) cos(x) * cos(z) sin(x)
  1144. // r3 + (r2 * sin(x)) r4 - (r1 * sin(x)) cos(x) * cos(y)
  1145. //
  1146. // where:
  1147. // r1 = cos(y) * cos(z)
  1148. // r2 = cos(y) * sin(z)
  1149. // r3 = sin(y) * cos(z)
  1150. // r4 = sin(y) * sin(z)
  1151. // init the euler 3x3 rotation matrix.
  1152. (*this)(0, 0) = r1 - (r4 * sinPitch); (*this)(0, 1) = r2 + (r3 * sinPitch); (*this)(0, 2) = -cosPitch * sinYaw;
  1153. (*this)(1, 0) = -cosPitch * sinRoll; (*this)(1, 1) = cosPitch * cosRoll; (*this)(1, 2) = sinPitch;
  1154. (*this)(2, 0) = r3 + (r2 * sinPitch); (*this)(2, 1) = r4 - (r1 * sinPitch); (*this)(2, 2) = cosPitch * cosYaw;
  1155. break;
  1156. }
  1157. if (rows == 4)
  1158. {
  1159. (*this)(3, 0) = 0.0f;
  1160. (*this)(3, 1) = 0.0f;
  1161. (*this)(3, 2) = 0.0f;
  1162. }
  1163. if (cols == 4)
  1164. {
  1165. (*this)(0, 3) = 0.0f;
  1166. (*this)(1, 3) = 0.0f;
  1167. (*this)(2, 3) = 0.0f;
  1168. }
  1169. if (rows == 4 && cols == 4)
  1170. {
  1171. (*this)(3, 3) = 1.0f;
  1172. }
  1173. return(*this);
  1174. }
  1175. template<typename DATA_TYPE, U32 rows, U32 cols>
  1176. Matrix<DATA_TYPE, rows, cols>::Matrix(const EulerF& e, const Point3F p)
  1177. {
  1178. set(e, p);
  1179. }
  1180. template<typename DATA_TYPE, U32 rows, U32 cols>
  1181. inline Matrix<DATA_TYPE, rows, cols>& Matrix<DATA_TYPE, rows, cols>::set(const EulerF& e, const Point3F p)
  1182. {
  1183. AssertFatal(rows >= 3 && cols >= 4, "Euler and Point can only initialize 3x4 or more");
  1184. // call set euler, this already sets the last row if it exists.
  1185. set(e);
  1186. // does this need to multiply with the result of the euler? or are we just setting position.
  1187. (*this)(0, 3) = p.x;
  1188. (*this)(1, 3) = p.y;
  1189. (*this)(2, 3) = p.z;
  1190. return (*this);
  1191. }
  1192. template<typename DATA_TYPE, U32 rows, U32 cols>
  1193. inline Matrix<DATA_TYPE, rows, cols>& Matrix<DATA_TYPE, rows, cols>::inverse()
  1194. {
  1195. #if 1
  1196. // NOTE: Gauss-Jordan elimination is yielding unpredictable results due to precission handling and
  1197. // numbers near 0.0
  1198. //
  1199. AssertFatal(rows == cols, "Can only perform inverse on square matrices.");
  1200. const U32 size = rows - 1;
  1201. const DATA_TYPE pivot_eps = static_cast<DATA_TYPE>(1e-20); // Smaller epsilon to handle numerical precision
  1202. // Create augmented matrix [this | I]
  1203. Matrix<DATA_TYPE, size, rows + size> augmentedMatrix;
  1204. for (U32 i = 0; i < size; i++)
  1205. {
  1206. for (U32 j = 0; j < size; j++)
  1207. {
  1208. augmentedMatrix(i, j) = (*this)(i, j);
  1209. augmentedMatrix(i, j + size) = (i == j) ? static_cast<DATA_TYPE>(1) : static_cast<DATA_TYPE>(0);
  1210. }
  1211. }
  1212. // Apply gauss-joran elimination
  1213. for (U32 i = 0; i < size; i++)
  1214. {
  1215. U32 pivotRow = i;
  1216. DATA_TYPE pivotValue = std::abs(augmentedMatrix(i, i));
  1217. for (U32 k = i + 1; k < size; k++)
  1218. {
  1219. DATA_TYPE curValue = std::abs(augmentedMatrix(k, i));
  1220. if (curValue > pivotValue) {
  1221. pivotRow = k;
  1222. pivotValue = curValue;
  1223. }
  1224. }
  1225. // Swap if needed.
  1226. if (i != pivotRow)
  1227. {
  1228. for (U32 j = 0; j < 2 * size; j++)
  1229. {
  1230. DATA_TYPE temp = augmentedMatrix(i, j);
  1231. augmentedMatrix(i, j) = augmentedMatrix(pivotRow, j);
  1232. augmentedMatrix(pivotRow, j) = temp;
  1233. }
  1234. }
  1235. // Early out if pivot is 0, return identity matrix.
  1236. if (std::abs(augmentedMatrix(i, i)) < pivot_eps)
  1237. {
  1238. return *this;
  1239. }
  1240. DATA_TYPE pivotVal = static_cast<DATA_TYPE>(1.0) / augmentedMatrix(i, i);
  1241. // scale the pivot
  1242. for (U32 j = 0; j < 2 * size; j++)
  1243. {
  1244. augmentedMatrix(i, j) *= pivotVal;
  1245. }
  1246. // Eliminate the current column in all other rows
  1247. for (U32 k = 0; k < size; k++)
  1248. {
  1249. if (k != i)
  1250. {
  1251. DATA_TYPE factor = augmentedMatrix(k, i);
  1252. for (U32 j = 0; j < 2 * size; j++)
  1253. {
  1254. augmentedMatrix(k, j) -= factor * augmentedMatrix(i, j);
  1255. }
  1256. }
  1257. }
  1258. }
  1259. for (U32 i = 0; i < size; i++)
  1260. {
  1261. for (U32 j = 0; j < size; j++)
  1262. {
  1263. (*this)(i, j) = augmentedMatrix(i, j + size);
  1264. }
  1265. }
  1266. #else
  1267. AssertFatal(rows == cols, "Can only perform inverse on square matrices.");
  1268. AssertFatal(rows >= 3 && cols >= 3, "Must be at least a 3x3 matrix");
  1269. DATA_TYPE det = determinant();
  1270. // Check if the determinant is non-zero
  1271. if (std::abs(det) < static_cast<DATA_TYPE>(1e-10)) {
  1272. this->identity(); // Return the identity matrix if the determinant is zero
  1273. return *this;
  1274. }
  1275. DATA_TYPE invDet = DATA_TYPE(1) / det;
  1276. Matrix<DATA_TYPE, rows, cols> temp;
  1277. // Calculate the inverse of the 3x3 upper-left submatrix using Cramer's rule
  1278. temp(0, 0) = ((*this)(1, 1) * (*this)(2, 2) - (*this)(1, 2) * (*this)(2, 1)) * invDet;
  1279. temp(0, 1) = ((*this)(2, 1) * (*this)(0, 2) - (*this)(2, 2) * (*this)(0, 1)) * invDet;
  1280. temp(0, 2) = ((*this)(0, 1) * (*this)(1, 2) - (*this)(0, 2) * (*this)(1, 1)) * invDet;
  1281. temp(1, 0) = ((*this)(1, 2) * (*this)(2, 0) - (*this)(1, 0) * (*this)(2, 2)) * invDet;
  1282. temp(1, 1) = ((*this)(2, 2) * (*this)(0, 0) - (*this)(2, 0) * (*this)(0, 2)) * invDet;
  1283. temp(1, 2) = ((*this)(0, 2) * (*this)(1, 0) - (*this)(0, 0) * (*this)(1, 2)) * invDet;
  1284. temp(2, 0) = ((*this)(1, 0) * (*this)(2, 1) - (*this)(1, 1) * (*this)(2, 0)) * invDet;
  1285. temp(2, 1) = ((*this)(2, 0) * (*this)(0, 1) - (*this)(2, 1) * (*this)(0, 0)) * invDet;
  1286. temp(2, 2) = ((*this)(0, 0) * (*this)(1, 1) - (*this)(0, 1) * (*this)(1, 0)) * invDet;
  1287. // Copy the 3x3 inverse back into this matrix
  1288. for (U32 i = 0; i < 3; ++i)
  1289. {
  1290. for (U32 j = 0; j < 3; ++j)
  1291. {
  1292. (*this)(i, j) = temp(i, j);
  1293. }
  1294. }
  1295. #endif
  1296. Point3F pos = -this->getPosition();
  1297. mulV(pos);
  1298. this->setPosition(pos);
  1299. return (*this);
  1300. }
  1301. template<typename DATA_TYPE, U32 rows, U32 cols>
  1302. inline bool Matrix<DATA_TYPE, rows, cols>::fullInverse()
  1303. {
  1304. #if 1
  1305. // NOTE: Gauss-Jordan elimination is yielding unpredictable results due to precission handling and
  1306. // numbers near 0.0
  1307. AssertFatal(rows == cols, "Can only perform inverse on square matrices.");
  1308. const U32 size = rows;
  1309. const DATA_TYPE pivot_eps = static_cast<DATA_TYPE>(1e-20); // Smaller epsilon to handle numerical precision
  1310. // Create augmented matrix [this | I]
  1311. Matrix<DATA_TYPE, size, rows + size> augmentedMatrix;
  1312. for (U32 i = 0; i < size; i++)
  1313. {
  1314. for (U32 j = 0; j < size; j++)
  1315. {
  1316. augmentedMatrix(i, j) = (*this)(i, j);
  1317. augmentedMatrix(i, j + size) = (i == j) ? static_cast<DATA_TYPE>(1) : static_cast<DATA_TYPE>(0);
  1318. }
  1319. }
  1320. // Apply gauss-joran elimination
  1321. for (U32 i = 0; i < size; i++)
  1322. {
  1323. U32 pivotRow = i;
  1324. DATA_TYPE pivotValue = std::abs(augmentedMatrix(i, i));
  1325. for (U32 k = i + 1; k < size; k++)
  1326. {
  1327. DATA_TYPE curValue = std::abs(augmentedMatrix(k, i));
  1328. if (curValue > pivotValue) {
  1329. pivotRow = k;
  1330. pivotValue = curValue;
  1331. }
  1332. }
  1333. // Swap if needed.
  1334. if (i != pivotRow)
  1335. {
  1336. for (U32 j = 0; j < 2 * size; j++)
  1337. {
  1338. DATA_TYPE temp = augmentedMatrix(i, j);
  1339. augmentedMatrix(i, j) = augmentedMatrix(pivotRow, j);
  1340. augmentedMatrix(pivotRow, j) = temp;
  1341. }
  1342. }
  1343. // Early out if pivot is 0, return identity matrix.
  1344. if (std::abs(augmentedMatrix(i, i)) < pivot_eps)
  1345. {
  1346. return false;
  1347. }
  1348. DATA_TYPE pivotVal = static_cast<DATA_TYPE>(1.0) / augmentedMatrix(i, i);
  1349. // scale the pivot
  1350. for (U32 j = 0; j < 2 * size; j++)
  1351. {
  1352. augmentedMatrix(i, j) *= pivotVal;
  1353. }
  1354. // Eliminate the current column in all other rows
  1355. for (U32 k = 0; k < size; k++)
  1356. {
  1357. if (k != i)
  1358. {
  1359. DATA_TYPE factor = augmentedMatrix(k, i);
  1360. for (U32 j = 0; j < 2 * size; j++)
  1361. {
  1362. augmentedMatrix(k, j) -= factor * augmentedMatrix(i, j);
  1363. }
  1364. }
  1365. }
  1366. }
  1367. for (U32 i = 0; i < size; i++)
  1368. {
  1369. for (U32 j = 0; j < size; j++)
  1370. {
  1371. (*this)(i, j) = augmentedMatrix(i, j + size);
  1372. }
  1373. }
  1374. #else
  1375. AssertFatal(rows == cols, "Can only perform inverse on square matrices.");
  1376. AssertFatal(rows >= 4 && cols >= 4, "Can only perform fullInverse on minimum 4x4 matrix");
  1377. Point4F a, b, c, d;
  1378. getRow(0, &a);
  1379. getRow(1, &b);
  1380. getRow(2, &c);
  1381. getRow(3, &d);
  1382. F32 det = a.x * b.y * c.z * d.w - a.x * b.y * c.w * d.z - a.x * c.y * b.z * d.w + a.x * c.y * b.w * d.z + a.x * d.y * b.z * c.w - a.x * d.y * b.w * c.z
  1383. - b.x * a.y * c.z * d.w + b.x * a.y * c.w * d.z + b.x * c.y * a.z * d.w - b.x * c.y * a.w * d.z - b.x * d.y * a.z * c.w + b.x * d.y * a.w * c.z
  1384. + c.x * a.y * b.z * d.w - c.x * a.y * b.w * d.z - c.x * b.y * a.z * d.w + c.x * b.y * a.w * d.z + c.x * d.y * a.z * b.w - c.x * d.y * a.w * b.z
  1385. - d.x * a.y * b.z * c.w + d.x * a.y * b.w * c.z + d.x * b.y * a.z * c.w - d.x * b.y * a.w * c.z - d.x * c.y * a.z * b.w + d.x * c.y * a.w * b.z;
  1386. if (mFabs(det) < 0.00001f)
  1387. return false;
  1388. Point4F aa, bb, cc, dd;
  1389. aa.x = b.y * c.z * d.w - b.y * c.w * d.z - c.y * b.z * d.w + c.y * b.w * d.z + d.y * b.z * c.w - d.y * b.w * c.z;
  1390. aa.y = -a.y * c.z * d.w + a.y * c.w * d.z + c.y * a.z * d.w - c.y * a.w * d.z - d.y * a.z * c.w + d.y * a.w * c.z;
  1391. aa.z = a.y * b.z * d.w - a.y * b.w * d.z - b.y * a.z * d.w + b.y * a.w * d.z + d.y * a.z * b.w - d.y * a.w * b.z;
  1392. aa.w = -a.y * b.z * c.w + a.y * b.w * c.z + b.y * a.z * c.w - b.y * a.w * c.z - c.y * a.z * b.w + c.y * a.w * b.z;
  1393. bb.x = -b.x * c.z * d.w + b.x * c.w * d.z + c.x * b.z * d.w - c.x * b.w * d.z - d.x * b.z * c.w + d.x * b.w * c.z;
  1394. bb.y = a.x * c.z * d.w - a.x * c.w * d.z - c.x * a.z * d.w + c.x * a.w * d.z + d.x * a.z * c.w - d.x * a.w * c.z;
  1395. bb.z = -a.x * b.z * d.w + a.x * b.w * d.z + b.x * a.z * d.w - b.x * a.w * d.z - d.x * a.z * b.w + d.x * a.w * b.z;
  1396. bb.w = a.x * b.z * c.w - a.x * b.w * c.z - b.x * a.z * c.w + b.x * a.w * c.z + c.x * a.z * b.w - c.x * a.w * b.z;
  1397. cc.x = b.x * c.y * d.w - b.x * c.w * d.y - c.x * b.y * d.w + c.x * b.w * d.y + d.x * b.y * c.w - d.x * b.w * c.y;
  1398. cc.y = -a.x * c.y * d.w + a.x * c.w * d.y + c.x * a.y * d.w - c.x * a.w * d.y - d.x * a.y * c.w + d.x * a.w * c.y;
  1399. cc.z = a.x * b.y * d.w - a.x * b.w * d.y - b.x * a.y * d.w + b.x * a.w * d.y + d.x * a.y * b.w - d.x * a.w * b.y;
  1400. cc.w = -a.x * b.y * c.w + a.x * b.w * c.y + b.x * a.y * c.w - b.x * a.w * c.y - c.x * a.y * b.w + c.x * a.w * b.y;
  1401. dd.x = -b.x * c.y * d.z + b.x * c.z * d.y + c.x * b.y * d.z - c.x * b.z * d.y - d.x * b.y * c.z + d.x * b.z * c.y;
  1402. dd.y = a.x * c.y * d.z - a.x * c.z * d.y - c.x * a.y * d.z + c.x * a.z * d.y + d.x * a.y * c.z - d.x * a.z * c.y;
  1403. dd.z = -a.x * b.y * d.z + a.x * b.z * d.y + b.x * a.y * d.z - b.x * a.z * d.y - d.x * a.y * b.z + d.x * a.z * b.y;
  1404. dd.w = a.x * b.y * c.z - a.x * b.z * c.y - b.x * a.y * c.z + b.x * a.z * c.y + c.x * a.y * b.z - c.x * a.z * b.y;
  1405. setRow(0, aa);
  1406. setRow(1, bb);
  1407. setRow(2, cc);
  1408. setRow(3, dd);
  1409. mul(1.0f / det);
  1410. #endif
  1411. return true;
  1412. }
  1413. template<typename DATA_TYPE, U32 rows, U32 cols>
  1414. inline void Matrix<DATA_TYPE, rows, cols>::invert()
  1415. {
  1416. (*this) = inverse();
  1417. }
  1418. template<typename DATA_TYPE, U32 rows, U32 cols>
  1419. inline Matrix<DATA_TYPE, rows, cols>& Matrix<DATA_TYPE, rows, cols>::setCrossProduct(const Point3F& p)
  1420. {
  1421. AssertFatal(rows == 4 && cols == 4, "Cross product only supported on 4x4 for now");
  1422. (*this)(0, 0) = 0;
  1423. (*this)(0, 1) = -p.z;
  1424. (*this)(0, 2) = p.y;
  1425. (*this)(0, 3) = 0;
  1426. (*this)(1, 0) = p.z;
  1427. (*this)(1, 1) = 0;
  1428. (*this)(1, 2) = -p.x;
  1429. (*this)(1, 3) = 0;
  1430. (*this)(2, 0) = -p.y;
  1431. (*this)(2, 1) = p.x;
  1432. (*this)(2, 2) = 0;
  1433. (*this)(2, 3) = 0;
  1434. (*this)(3, 0) = 0;
  1435. (*this)(3, 1) = 0;
  1436. (*this)(3, 2) = 0;
  1437. (*this)(3, 3) = 1;
  1438. return (*this);
  1439. }
  1440. template<typename DATA_TYPE, U32 rows, U32 cols>
  1441. inline Matrix<DATA_TYPE, rows, cols>& Matrix<DATA_TYPE, rows, cols>::setTensorProduct(const Point3F& p, const Point3F& q)
  1442. {
  1443. AssertFatal(rows == 4 && cols == 4, "Tensor product only supported on 4x4 for now");
  1444. (*this)(0, 0) = p.x * q.x;
  1445. (*this)(0, 1) = p.x * q.y;
  1446. (*this)(0, 2) = p.x * q.z;
  1447. (*this)(0, 3) = 0;
  1448. (*this)(1, 0) = p.y * q.x;
  1449. (*this)(1, 1) = p.y * q.y;
  1450. (*this)(1, 2) = p.y * q.z;
  1451. (*this)(1, 3) = 0;
  1452. (*this)(2, 0) = p.z * q.x;
  1453. (*this)(2, 1) = p.z * q.y;
  1454. (*this)(2, 2) = p.z * q.z;
  1455. (*this)(2, 3) = 0;
  1456. (*this)(3, 0) = 0;
  1457. (*this)(3, 1) = 0;
  1458. (*this)(3, 2) = 0;
  1459. (*this)(3, 3) = 1;
  1460. return (*this);
  1461. }
  1462. template<typename DATA_TYPE, U32 rows, U32 cols>
  1463. inline void Matrix<DATA_TYPE, rows, cols>::mul(Box3F& box) const
  1464. {
  1465. AssertFatal(rows == 4 && cols == 4, "Multiplying Box3F with matrix requires 4x4");
  1466. // Extract the min and max extents
  1467. const Point3F& originalMin = box.minExtents;
  1468. const Point3F& originalMax = box.maxExtents;
  1469. // Array to store transformed corners
  1470. Point3F transformedCorners[8];
  1471. // Compute all 8 corners of the box
  1472. Point3F corners[8] = {
  1473. {originalMin.x, originalMin.y, originalMin.z},
  1474. {originalMax.x, originalMin.y, originalMin.z},
  1475. {originalMin.x, originalMax.y, originalMin.z},
  1476. {originalMax.x, originalMax.y, originalMin.z},
  1477. {originalMin.x, originalMin.y, originalMax.z},
  1478. {originalMax.x, originalMin.y, originalMax.z},
  1479. {originalMin.x, originalMax.y, originalMax.z},
  1480. {originalMax.x, originalMax.y, originalMax.z}
  1481. };
  1482. // Transform each corner
  1483. for (U32 i = 0; i < 8; ++i)
  1484. {
  1485. const Point3F& corner = corners[i];
  1486. transformedCorners[i].x = (*this)(0, 0) * corner.x + (*this)(0, 1) * corner.y + (*this)(0, 2) * corner.z + (*this)(0, 3);
  1487. transformedCorners[i].y = (*this)(1, 0) * corner.x + (*this)(1, 1) * corner.y + (*this)(1, 2) * corner.z + (*this)(1, 3);
  1488. transformedCorners[i].z = (*this)(2, 0) * corner.x + (*this)(2, 1) * corner.y + (*this)(2, 2) * corner.z + (*this)(2, 3);
  1489. }
  1490. // Initialize min and max extents to the transformed values
  1491. Point3F newMin = transformedCorners[0];
  1492. Point3F newMax = transformedCorners[0];
  1493. // Compute the new min and max extents from the transformed corners
  1494. for (U32 i = 1; i < 8; ++i)
  1495. {
  1496. const Point3F& corner = transformedCorners[i];
  1497. if (corner.x < newMin.x) newMin.x = corner.x;
  1498. if (corner.y < newMin.y) newMin.y = corner.y;
  1499. if (corner.z < newMin.z) newMin.z = corner.z;
  1500. if (corner.x > newMax.x) newMax.x = corner.x;
  1501. if (corner.y > newMax.y) newMax.y = corner.y;
  1502. if (corner.z > newMax.z) newMax.z = corner.z;
  1503. }
  1504. // Update the box with the new min and max extents
  1505. box.minExtents = newMin;
  1506. box.maxExtents = newMax;
  1507. }
  1508. template<typename DATA_TYPE, U32 rows, U32 cols>
  1509. inline bool Matrix<DATA_TYPE, rows, cols>::isAffine() const
  1510. {
  1511. if ((*this)(3, 3) != 1.0f)
  1512. {
  1513. return false;
  1514. }
  1515. for (U32 col = 0; col < cols - 1; ++col)
  1516. {
  1517. if ((*this)(3, col) != 0.0f)
  1518. {
  1519. return false;
  1520. }
  1521. }
  1522. Point3F one, two, three;
  1523. getColumn(0, &one);
  1524. getColumn(1, &two);
  1525. getColumn(2, &three);
  1526. // check columns
  1527. {
  1528. if (mDot(one, two) > 0.0001f ||
  1529. mDot(one, three) > 0.0001f ||
  1530. mDot(two, three) > 0.0001f)
  1531. return false;
  1532. if (mFabs(1.0f - one.lenSquared()) > 0.0001f ||
  1533. mFabs(1.0f - two.lenSquared()) > 0.0001f ||
  1534. mFabs(1.0f - three.lenSquared()) > 0.0001f)
  1535. return false;
  1536. }
  1537. getRow(0, &one);
  1538. getRow(1, &two);
  1539. getRow(2, &three);
  1540. // check rows
  1541. {
  1542. if (mDot(one, two) > 0.0001f ||
  1543. mDot(one, three) > 0.0001f ||
  1544. mDot(two, three) > 0.0001f)
  1545. return false;
  1546. if (mFabs(1.0f - one.lenSquared()) > 0.0001f ||
  1547. mFabs(1.0f - two.lenSquared()) > 0.0001f ||
  1548. mFabs(1.0f - three.lenSquared()) > 0.0001f)
  1549. return false;
  1550. }
  1551. return true;
  1552. }
  1553. template<typename DATA_TYPE, U32 rows, U32 cols>
  1554. inline Matrix<DATA_TYPE, rows, cols>& Matrix<DATA_TYPE, rows, cols>::affineInverse()
  1555. {
  1556. AssertFatal(rows >= 4 && cols >= 4, "affineInverse requires at least 4x4");
  1557. Matrix<DATA_TYPE, rows, cols> temp = *this;
  1558. // Transpose rotation part
  1559. (*this)(0, 1) = temp(1, 0);
  1560. (*this)(0, 2) = temp(2, 0);
  1561. (*this)(1, 0) = temp(0, 1);
  1562. (*this)(1, 2) = temp(2, 1);
  1563. (*this)(2, 0) = temp(0, 2);
  1564. (*this)(2, 1) = temp(1, 2);
  1565. // Adjust translation part
  1566. (*this)(0, 3) = -(temp(0, 0) * temp(0, 3) + temp(1, 0) * temp(1, 3) + temp(2, 0) * temp(2, 3));
  1567. (*this)(1, 3) = -(temp(0, 1) * temp(0, 3) + temp(1, 1) * temp(1, 3) + temp(2, 1) * temp(2, 3));
  1568. (*this)(2, 3) = -(temp(0, 2) * temp(0, 3) + temp(1, 2) * temp(1, 3) + temp(2, 2) * temp(2, 3));
  1569. return *this;
  1570. }
  1571. template<typename DATA_TYPE, U32 rows, U32 cols>
  1572. inline EulerF Matrix<DATA_TYPE, rows, cols>::toEuler() const
  1573. {
  1574. AssertFatal(rows >= 3 && cols >= 3, "Euler rotations require at least a 3x3 matrix.");
  1575. // like all others assume float for now.
  1576. EulerF r;
  1577. r.x = mAsin(mClampF((*this)(1,2), -1.0, 1.0));
  1578. if (mCos(r.x) != 0.0f)
  1579. {
  1580. r.y = mAtan2(-(*this)(0, 2), (*this)(2, 2)); // yaw
  1581. r.z = mAtan2(-(*this)(1, 0), (*this)(1, 1)); // roll
  1582. }
  1583. else
  1584. {
  1585. r.y = 0.0f;
  1586. r.z = mAtan2((*this)(0, 1), (*this)(0, 0)); // this rolls when pitch is +90 degrees
  1587. }
  1588. return r;
  1589. }
  1590. template<typename DATA_TYPE, U32 rows, U32 cols>
  1591. inline void Matrix<DATA_TYPE, rows, cols>::dumpMatrix(const char* caption) const
  1592. {
  1593. U32 size = (caption == NULL) ? 0 : dStrlen(caption);
  1594. FrameTemp<char> spacer(size + 1);
  1595. char* spacerRef = spacer;
  1596. // is_floating_point should return true for floats and doubles.
  1597. const char* formatSpec = std::is_floating_point_v<DATA_TYPE> ? " %-8.4f" : " %d";
  1598. dMemset(spacerRef, ' ', size);
  1599. // null terminate.
  1600. spacerRef[size] = '\0';
  1601. /*Con::printf("%s = | %-8.4f %-8.4f %-8.4f %-8.4f |", caption, m[idx(0, 0)], m[idx(0, 1)], m[idx(0, 2)], m[idx(0, 3)]);
  1602. Con::printf("%s | %-8.4f %-8.4f %-8.4f %-8.4f |", spacerRef, m[idx(1, 0)], m[idx(1, 1)], m[idx(1, 2)], m[idx(1, 3)]);
  1603. Con::printf("%s | %-8.4f %-8.4f %-8.4f %-8.4f |", spacerRef, m[idx(2, 0)], m[idx(2, 1)], m[idx(2, 2)], m[idx(2, 3)]);
  1604. Con::printf("%s | %-8.4f %-8.4f %-8.4f %-8.4f |", spacerRef, m[idx(3, 0)], m[idx(3, 1)], m[idx(3, 2)], m[idx(3, 3)]);*/
  1605. StringBuilder str;
  1606. str.format("%s = |", caption);
  1607. for (U32 i = 0; i < rows; i++)
  1608. {
  1609. if (i > 0)
  1610. {
  1611. str.append(spacerRef);
  1612. }
  1613. for (U32 j = 0; j < cols; j++)
  1614. {
  1615. str.format(formatSpec, (*this)(i, j));
  1616. }
  1617. str.append(" |\n");
  1618. }
  1619. Con::printf("%s", str.end().c_str());
  1620. }
  1621. //------------------------------------
  1622. // Non-member methods
  1623. //------------------------------------
  1624. inline void mTransformPlane(
  1625. const MatrixF& mat,
  1626. const Point3F& scale,
  1627. const PlaneF& plane,
  1628. PlaneF* result
  1629. ) {
  1630. // Create the inverse scale matrix
  1631. MatrixF invScale(true);
  1632. invScale(0, 0) = 1.0f / scale.x;
  1633. invScale(1, 1) = 1.0f / scale.y;
  1634. invScale(2, 2) = 1.0f / scale.z;
  1635. const Point3F shear(mat(0, 3), mat(1, 3), mat(2, 3));
  1636. const Point3F row0 = mat.getRow3F(0);
  1637. const Point3F row1 = mat.getRow3F(1);
  1638. const Point3F row2 = mat.getRow3F(2);
  1639. const F32 A = -mDot(row0, shear);
  1640. const F32 B = -mDot(row1, shear);
  1641. const F32 C = -mDot(row2, shear);
  1642. // Compute the inverse transpose of the matrix
  1643. MatrixF invTrMatrix(true);
  1644. invTrMatrix(0, 0) = mat(0, 0);
  1645. invTrMatrix(0, 1) = mat(0, 1);
  1646. invTrMatrix(0, 2) = mat(0, 2);
  1647. invTrMatrix(1, 0) = mat(1, 0);
  1648. invTrMatrix(1, 1) = mat(1, 1);
  1649. invTrMatrix(1, 2) = mat(1, 2);
  1650. invTrMatrix(2, 0) = mat(2, 0);
  1651. invTrMatrix(2, 1) = mat(2, 1);
  1652. invTrMatrix(2, 2) = mat(2, 2);
  1653. invTrMatrix(3, 0) = A;
  1654. invTrMatrix(3, 1) = B;
  1655. invTrMatrix(3, 2) = C;
  1656. invTrMatrix.mul(invScale);
  1657. // Transform the plane normal
  1658. Point3F norm(plane.x, plane.y, plane.z);
  1659. invTrMatrix.mulP(norm);
  1660. norm.normalize();
  1661. // Transform the plane point
  1662. Point3F point = norm * -plane.d;
  1663. MatrixF temp = mat;
  1664. point.x *= scale.x;
  1665. point.y *= scale.y;
  1666. point.z *= scale.z;
  1667. temp.mulP(point);
  1668. // Recompute the plane distance
  1669. PlaneF resultPlane(point, norm);
  1670. result->x = resultPlane.x;
  1671. result->y = resultPlane.y;
  1672. result->z = resultPlane.z;
  1673. result->d = resultPlane.d;
  1674. }
  1675. //--------------------------------------------
  1676. // INLINE FUNCTIONS END
  1677. //--------------------------------------------
  1678. typedef Matrix<F32, 4, 4> MatrixF;
  1679. class MatrixTemplateExport
  1680. {
  1681. public:
  1682. template <typename T, U32 rows, U32 cols>
  1683. static EngineFieldTable::Field getMatrixField();
  1684. };
  1685. template<typename T, U32 rows, U32 cols>
  1686. inline EngineFieldTable::Field MatrixTemplateExport::getMatrixField()
  1687. {
  1688. typedef Matrix<T, rows, cols> ThisType;
  1689. return _FIELD_AS(T, data, data, rows * cols, "");
  1690. }
  1691. #endif // !USE_TEMPLATE_MATRIX
  1692. #endif //_MMATRIX_H_