2
0

mMatrix.h 25 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828829830831832833834835836837838839840841842843844845846847848849850851852853854855856857858859860861862863864865866867868869870871872873874875876877878879880881882883884885886887888889890891892893894895896897898899900901902903904905906907908909910911912913914915916917918919920921922923924925926927928929930931932933934935936937938939940941942943944945946947948949950951952953954955956957958959960961962963964965966967968969970971972973974975976977978979980981982983984985986
  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. /// 4x4 Matrix Class
  38. ///
  39. /// This runs at F32 precision.
  40. class MatrixF
  41. {
  42. friend class MatrixFEngineExport;
  43. private:
  44. F32 m[16]; ///< Note: Torque uses row-major matrices
  45. public:
  46. /// Create an uninitialized matrix.
  47. ///
  48. /// @param identity If true, initialize to the identity matrix.
  49. explicit MatrixF(bool identity=false);
  50. /// Create a matrix to rotate about origin by e.
  51. /// @see set
  52. explicit MatrixF( const EulerF &e);
  53. /// Create a matrix to rotate about p by e.
  54. /// @see set
  55. MatrixF( const EulerF &e, const Point3F& p);
  56. /// Get the index in m to element in column i, row j
  57. ///
  58. /// This is necessary as we have m as a one dimensional array.
  59. ///
  60. /// @param i Column desired.
  61. /// @param j Row desired.
  62. static U32 idx(U32 i, U32 j) { return (i + j*4); }
  63. /// Initialize matrix to rotate about origin by e.
  64. MatrixF& set( const EulerF &e);
  65. /// Initialize matrix to rotate about p by e.
  66. MatrixF& set( const EulerF &e, const Point3F& p);
  67. /// Initialize matrix with a cross product of p.
  68. MatrixF& setCrossProduct( const Point3F &p);
  69. /// Initialize matrix with a tensor product of p.
  70. MatrixF& setTensorProduct( const Point3F &p, const Point3F& q);
  71. operator F32*() { return (m); } ///< Allow people to get at m.
  72. operator const F32*() const { return (F32*)(m); } ///< Allow people to get at m.
  73. bool isAffine() const; ///< Check to see if this is an affine matrix.
  74. bool isIdentity() const; ///< Checks for identity matrix.
  75. /// Make this an identity matrix.
  76. MatrixF& identity();
  77. /// Invert m.
  78. MatrixF& inverse();
  79. /// Copy the inversion of this into out matrix.
  80. void invertTo( MatrixF *out );
  81. /// Take inverse of matrix assuming it is affine (rotation,
  82. /// scale, sheer, translation only).
  83. MatrixF& affineInverse();
  84. /// Swap rows and columns.
  85. MatrixF& transpose();
  86. /// M * Matrix(p) -> M
  87. MatrixF& scale( const Point3F &s );
  88. MatrixF& scale( F32 s ) { return scale( Point3F( s, s, s ) ); }
  89. /// Return scale assuming scale was applied via mat.scale(s).
  90. Point3F getScale() const;
  91. EulerF toEuler() const;
  92. /// Compute the inverse of the matrix.
  93. ///
  94. /// Computes inverse of full 4x4 matrix. Returns false and performs no inverse if
  95. /// the determinant is 0.
  96. ///
  97. /// Note: In most cases you want to use the normal inverse function. This method should
  98. /// be used if the matrix has something other than (0,0,0,1) in the bottom row.
  99. bool fullInverse();
  100. /// Reverse depth for projection matrix
  101. /// Simplifies reversal matrix mult to 4 subtractions
  102. void reverseProjection();
  103. /// Swaps rows and columns into matrix.
  104. void transposeTo(F32 *matrix) const;
  105. /// Normalize the matrix.
  106. void normalize();
  107. /// Copy the requested column into a Point4F.
  108. void getColumn(S32 col, Point4F *cptr) const;
  109. Point4F getColumn4F(S32 col) const { Point4F ret; getColumn(col,&ret); return ret; }
  110. /// Copy the requested column into a Point3F.
  111. ///
  112. /// This drops the bottom-most row.
  113. void getColumn(S32 col, Point3F *cptr) const;
  114. Point3F getColumn3F(S32 col) const { Point3F ret; getColumn(col,&ret); return ret; }
  115. /// Set the specified column from a Point4F.
  116. void setColumn(S32 col, const Point4F& cptr);
  117. /// Set the specified column from a Point3F.
  118. ///
  119. /// The bottom-most row is not set.
  120. void setColumn(S32 col, const Point3F& cptr);
  121. /// Copy the specified row into a Point4F.
  122. void getRow(S32 row, Point4F *cptr) const;
  123. Point4F getRow4F(S32 row) const { Point4F ret; getRow(row,&ret); return ret; }
  124. /// Copy the specified row into a Point3F.
  125. ///
  126. /// Right-most item is dropped.
  127. void getRow(S32 row, Point3F *cptr) const;
  128. Point3F getRow3F(S32 row) const { Point3F ret; getRow(row,&ret); return ret; }
  129. /// Set the specified row from a Point4F.
  130. void setRow(S32 row, const Point4F& cptr);
  131. /// Set the specified row from a Point3F.
  132. ///
  133. /// The right-most item is not set.
  134. void setRow(S32 row, const Point3F& cptr);
  135. /// Get the position of the matrix.
  136. ///
  137. /// This is the 4th column of the matrix.
  138. Point3F getPosition() const;
  139. /// Set the position of the matrix.
  140. ///
  141. /// This is the 4th column of the matrix.
  142. void setPosition( const Point3F &pos ) { setColumn( 3, pos ); }
  143. /// Add the passed delta to the matrix position.
  144. void displace( const Point3F &delta );
  145. /// Get the x axis of the matrix.
  146. ///
  147. /// This is the 1st column of the matrix and is
  148. /// normally considered the right vector.
  149. VectorF getRightVector() const;
  150. /// Get the y axis of the matrix.
  151. ///
  152. /// This is the 2nd column of the matrix and is
  153. /// normally considered the forward vector.
  154. VectorF getForwardVector() const;
  155. /// Get the z axis of the matrix.
  156. ///
  157. /// This is the 3rd column of the matrix and is
  158. /// normally considered the up vector.
  159. VectorF getUpVector() const;
  160. MatrixF& mul(const MatrixF &a); ///< M * a -> M
  161. MatrixF& mulL(const MatrixF &a); ///< a * M -> M
  162. MatrixF& mul(const MatrixF &a, const MatrixF &b); ///< a * b -> M
  163. // Scalar multiplies
  164. MatrixF& mul(const F32 a); ///< M * a -> M
  165. MatrixF& mul(const MatrixF &a, const F32 b); ///< a * b -> M
  166. void mul( Point4F& p ) const; ///< M * p -> p (full [4x4] * [1x4])
  167. void mulP( Point3F& p ) const; ///< M * p -> p (assume w = 1.0f)
  168. void mulP( const Point3F &p, Point3F *d) const; ///< M * p -> d (assume w = 1.0f)
  169. void mulV( VectorF& p ) const; ///< M * v -> v (assume w = 0.0f)
  170. void mulV( const VectorF &p, Point3F *d) const; ///< M * v -> d (assume w = 0.0f)
  171. void mul(Box3F& b) const; ///< Axial box -> Axial Box
  172. MatrixF& add( const MatrixF& m );
  173. /// Convenience function to allow people to treat this like an array.
  174. F32& operator ()(S32 row, S32 col) { return m[idx(col,row)]; }
  175. F32 operator ()(S32 row, S32 col) const { return m[idx(col,row)]; }
  176. void dumpMatrix(const char *caption=NULL) const;
  177. // Math operator overloads
  178. //------------------------------------
  179. friend MatrixF operator * ( const MatrixF &m1, const MatrixF &m2 );
  180. MatrixF& operator *= ( const MatrixF &m );
  181. MatrixF &operator = (const MatrixF &m);
  182. bool isNaN();
  183. // Static identity matrix
  184. const static MatrixF Identity;
  185. };
  186. class MatrixFEngineExport
  187. {
  188. public:
  189. static EngineFieldTable::Field getMatrixField();
  190. };
  191. //--------------------------------------
  192. // Inline Functions
  193. inline MatrixF::MatrixF(bool _identity)
  194. {
  195. if (_identity)
  196. identity();
  197. else
  198. std::fill_n(m, 16, 0);
  199. }
  200. inline MatrixF::MatrixF( const EulerF &e )
  201. {
  202. set(e);
  203. }
  204. inline MatrixF::MatrixF( const EulerF &e, const Point3F& p )
  205. {
  206. set(e,p);
  207. }
  208. inline MatrixF& MatrixF::set( const EulerF &e)
  209. {
  210. m_matF_set_euler( e, *this );
  211. return (*this);
  212. }
  213. inline MatrixF& MatrixF::set( const EulerF &e, const Point3F& p)
  214. {
  215. m_matF_set_euler_point( e, p, *this );
  216. return (*this);
  217. }
  218. inline MatrixF& MatrixF::setCrossProduct( const Point3F &p)
  219. {
  220. m[1] = -(m[4] = p.z);
  221. m[8] = -(m[2] = p.y);
  222. m[6] = -(m[9] = p.x);
  223. m[0] = m[3] = m[5] = m[7] = m[10] = m[11] =
  224. m[12] = m[13] = m[14] = 0.0f;
  225. m[15] = 1;
  226. return (*this);
  227. }
  228. inline MatrixF& MatrixF::setTensorProduct( const Point3F &p, const Point3F &q)
  229. {
  230. m[0] = p.x * q.x;
  231. m[1] = p.x * q.y;
  232. m[2] = p.x * q.z;
  233. m[4] = p.y * q.x;
  234. m[5] = p.y * q.y;
  235. m[6] = p.y * q.z;
  236. m[8] = p.z * q.x;
  237. m[9] = p.z * q.y;
  238. m[10] = p.z * q.z;
  239. m[3] = m[7] = m[11] = m[12] = m[13] = m[14] = 0.0f;
  240. m[15] = 1.0f;
  241. return (*this);
  242. }
  243. inline bool MatrixF::isIdentity() const
  244. {
  245. return
  246. m[0] == 1.0f &&
  247. m[1] == 0.0f &&
  248. m[2] == 0.0f &&
  249. m[3] == 0.0f &&
  250. m[4] == 0.0f &&
  251. m[5] == 1.0f &&
  252. m[6] == 0.0f &&
  253. m[7] == 0.0f &&
  254. m[8] == 0.0f &&
  255. m[9] == 0.0f &&
  256. m[10] == 1.0f &&
  257. m[11] == 0.0f &&
  258. m[12] == 0.0f &&
  259. m[13] == 0.0f &&
  260. m[14] == 0.0f &&
  261. m[15] == 1.0f;
  262. }
  263. inline MatrixF& MatrixF::identity()
  264. {
  265. m[0] = 1.0f;
  266. m[1] = 0.0f;
  267. m[2] = 0.0f;
  268. m[3] = 0.0f;
  269. m[4] = 0.0f;
  270. m[5] = 1.0f;
  271. m[6] = 0.0f;
  272. m[7] = 0.0f;
  273. m[8] = 0.0f;
  274. m[9] = 0.0f;
  275. m[10] = 1.0f;
  276. m[11] = 0.0f;
  277. m[12] = 0.0f;
  278. m[13] = 0.0f;
  279. m[14] = 0.0f;
  280. m[15] = 1.0f;
  281. return (*this);
  282. }
  283. inline MatrixF& MatrixF::inverse()
  284. {
  285. m_matF_inverse(m);
  286. return (*this);
  287. }
  288. inline void MatrixF::invertTo( MatrixF *out )
  289. {
  290. m_matF_invert_to(m,*out);
  291. }
  292. inline MatrixF& MatrixF::affineInverse()
  293. {
  294. // AssertFatal(isAffine() == true, "Error, this matrix is not an affine transform");
  295. m_matF_affineInverse(m);
  296. return (*this);
  297. }
  298. inline MatrixF& MatrixF::transpose()
  299. {
  300. m_matF_transpose(m);
  301. return (*this);
  302. }
  303. inline MatrixF& MatrixF::scale(const Point3F& p)
  304. {
  305. m_matF_scale(m,p);
  306. return *this;
  307. }
  308. inline Point3F MatrixF::getScale() const
  309. {
  310. Point3F scale;
  311. scale.x = mSqrt(m[0]*m[0] + m[4] * m[4] + m[8] * m[8]);
  312. scale.y = mSqrt(m[1]*m[1] + m[5] * m[5] + m[9] * m[9]);
  313. scale.z = mSqrt(m[2]*m[2] + m[6] * m[6] + m[10] * m[10]);
  314. return scale;
  315. }
  316. inline void MatrixF::normalize()
  317. {
  318. m_matF_normalize(m);
  319. }
  320. inline MatrixF& MatrixF::mul( const MatrixF &a )
  321. { // M * a -> M
  322. AssertFatal(&a != this, "MatrixF::mul - a.mul(a) is invalid!");
  323. MatrixF tempThis(*this);
  324. m_matF_x_matF(tempThis, a, *this);
  325. return (*this);
  326. }
  327. inline MatrixF& MatrixF::mulL( const MatrixF &a )
  328. { // a * M -> M
  329. AssertFatal(&a != this, "MatrixF::mulL - a.mul(a) is invalid!");
  330. MatrixF tempThis(*this);
  331. m_matF_x_matF(a, tempThis, *this);
  332. return (*this);
  333. }
  334. inline MatrixF& MatrixF::mul( const MatrixF &a, const MatrixF &b )
  335. { // a * b -> M
  336. AssertFatal((&a != this) && (&b != this), "MatrixF::mul - a.mul(a, b) a.mul(b, a) a.mul(a, a) is invalid!");
  337. m_matF_x_matF(a, b, *this);
  338. return (*this);
  339. }
  340. inline MatrixF& MatrixF::mul(const F32 a)
  341. {
  342. for (U32 i = 0; i < 16; i++)
  343. m[i] *= a;
  344. return *this;
  345. }
  346. inline MatrixF& MatrixF::mul(const MatrixF &a, const F32 b)
  347. {
  348. *this = a;
  349. mul(b);
  350. return *this;
  351. }
  352. inline void MatrixF::mul( Point4F& p ) const
  353. {
  354. Point4F temp;
  355. m_matF_x_point4F(*this, &p.x, &temp.x);
  356. p = temp;
  357. }
  358. inline void MatrixF::mulP( Point3F& p) const
  359. {
  360. // M * p -> d
  361. Point3F d;
  362. m_matF_x_point3F(*this, &p.x, &d.x);
  363. p = d;
  364. }
  365. inline void MatrixF::mulP( const Point3F &p, Point3F *d) const
  366. {
  367. // M * p -> d
  368. m_matF_x_point3F(*this, &p.x, &d->x);
  369. }
  370. inline void MatrixF::mulV( VectorF& v) const
  371. {
  372. // M * v -> v
  373. VectorF temp;
  374. m_matF_x_vectorF(*this, &v.x, &temp.x);
  375. v = temp;
  376. }
  377. inline void MatrixF::mulV( const VectorF &v, Point3F *d) const
  378. {
  379. // M * v -> d
  380. m_matF_x_vectorF(*this, &v.x, &d->x);
  381. }
  382. inline void MatrixF::mul(Box3F& b) const
  383. {
  384. m_matF_x_box3F(*this, &b.minExtents.x, &b.maxExtents.x);
  385. }
  386. inline MatrixF& MatrixF::add( const MatrixF& a )
  387. {
  388. for( U32 i = 0; i < 16; ++ i )
  389. m[ i ] += a.m[ i ];
  390. return *this;
  391. }
  392. inline void MatrixF::getColumn(S32 col, Point4F *cptr) const
  393. {
  394. cptr->x = m[col];
  395. cptr->y = m[col+4];
  396. cptr->z = m[col+8];
  397. cptr->w = m[col+12];
  398. }
  399. inline void MatrixF::getColumn(S32 col, Point3F *cptr) const
  400. {
  401. cptr->x = m[col];
  402. cptr->y = m[col+4];
  403. cptr->z = m[col+8];
  404. }
  405. inline void MatrixF::setColumn(S32 col, const Point4F &cptr)
  406. {
  407. m[col] = cptr.x;
  408. m[col+4] = cptr.y;
  409. m[col+8] = cptr.z;
  410. m[col+12]= cptr.w;
  411. }
  412. inline void MatrixF::setColumn(S32 col, const Point3F &cptr)
  413. {
  414. m[col] = cptr.x;
  415. m[col+4] = cptr.y;
  416. m[col+8] = cptr.z;
  417. }
  418. inline void MatrixF::getRow(S32 col, Point4F *cptr) const
  419. {
  420. col *= 4;
  421. cptr->x = m[col++];
  422. cptr->y = m[col++];
  423. cptr->z = m[col++];
  424. cptr->w = m[col];
  425. }
  426. inline void MatrixF::getRow(S32 col, Point3F *cptr) const
  427. {
  428. col *= 4;
  429. cptr->x = m[col++];
  430. cptr->y = m[col++];
  431. cptr->z = m[col];
  432. }
  433. inline void MatrixF::setRow(S32 col, const Point4F &cptr)
  434. {
  435. col *= 4;
  436. m[col++] = cptr.x;
  437. m[col++] = cptr.y;
  438. m[col++] = cptr.z;
  439. m[col] = cptr.w;
  440. }
  441. inline void MatrixF::setRow(S32 col, const Point3F &cptr)
  442. {
  443. col *= 4;
  444. m[col++] = cptr.x;
  445. m[col++] = cptr.y;
  446. m[col] = cptr.z;
  447. }
  448. inline Point3F MatrixF::getPosition() const
  449. {
  450. return Point3F( m[3], m[3+4], m[3+8] );
  451. }
  452. inline void MatrixF::displace( const Point3F &delta )
  453. {
  454. m[3] += delta.x;
  455. m[3+4] += delta.y;
  456. m[3+8] += delta.z;
  457. }
  458. inline VectorF MatrixF::getForwardVector() const
  459. {
  460. VectorF vec;
  461. getColumn( 1, &vec );
  462. return vec;
  463. }
  464. inline VectorF MatrixF::getRightVector() const
  465. {
  466. VectorF vec;
  467. getColumn( 0, &vec );
  468. return vec;
  469. }
  470. inline VectorF MatrixF::getUpVector() const
  471. {
  472. VectorF vec;
  473. getColumn( 2, &vec );
  474. return vec;
  475. }
  476. //------------------------------------
  477. // Math operator overloads
  478. //------------------------------------
  479. inline MatrixF operator * ( const MatrixF &m1, const MatrixF &m2 )
  480. {
  481. // temp = m1 * m2
  482. MatrixF temp;
  483. m_matF_x_matF(m1, m2, temp);
  484. return temp;
  485. }
  486. inline MatrixF& MatrixF::operator *= ( const MatrixF &m1 )
  487. {
  488. MatrixF tempThis(*this);
  489. m_matF_x_matF(tempThis, m1, *this);
  490. return (*this);
  491. }
  492. inline MatrixF &MatrixF::operator = (const MatrixF &m1)
  493. {
  494. for (U32 i=0;i<16;i++)
  495. this->m[i] = m1.m[i];
  496. return (*this);
  497. }
  498. inline bool MatrixF::isNaN()
  499. {
  500. bool isaNaN = false;
  501. for (U32 i = 0; i < 16; i++)
  502. if (mIsNaN_F(m[i]))
  503. isaNaN = true;
  504. return isaNaN;
  505. }
  506. //------------------------------------
  507. // Non-member methods
  508. //------------------------------------
  509. inline void mTransformPlane(const MatrixF& mat, const Point3F& scale, const PlaneF& plane, PlaneF * result)
  510. {
  511. m_matF_x_scale_x_planeF(mat, &scale.x, &plane.x, &result->x);
  512. }
  513. //------------------------------------
  514. // Templatized matrix class to replace MATRIXF above
  515. // row-major for now, since torque says it uses that
  516. // but in future could cut down on transpose calls if
  517. // we switch to column major.
  518. //------------------------------------
  519. template<typename DATA_TYPE, U32 rows, U32 cols>
  520. class Matrix {
  521. friend class MatrixTemplateExport;
  522. private:
  523. DATA_TYPE data[rows * cols];
  524. public:
  525. static_assert(rows >= 2 && cols >= 2, "Matrix must have at least 2 rows and 2 cols.");
  526. // ------ Setters and initializers ------
  527. explicit Matrix(bool identity = false) {
  528. std::fill(data, data + (rows * cols), DATA_TYPE(0));
  529. if (identity) {
  530. for (U32 i = 0; i < rows; i++) {
  531. for (U32 j = 0; j < cols; j++) {
  532. // others already get filled with 0
  533. if (j == i)
  534. (*this)(i, j) = static_cast<DATA_TYPE>(1);
  535. }
  536. }
  537. }
  538. }
  539. explicit Matrix(const EulerF& e);
  540. /// Make this an identity matrix.
  541. Matrix<DATA_TYPE, rows, cols>& identity();
  542. Matrix<DATA_TYPE, rows, cols>& set(const EulerF& e);
  543. Matrix(const EulerF& e, const Point3F p);
  544. Matrix<DATA_TYPE, rows, cols>& set(const EulerF& e, const Point3F p);
  545. Matrix<DATA_TYPE, rows, cols>& inverse();
  546. Matrix<DATA_TYPE, rows, cols>& transpose();
  547. void invert();
  548. Matrix<DATA_TYPE, rows, cols>& setCrossProduct(const Point3F& p);
  549. Matrix<DATA_TYPE, rows, cols>& setTensorProduct(const Point3F& p, const Point3F& q);
  550. /// M * Matrix(p) -> M
  551. Matrix<DATA_TYPE, rows, cols>& scale(const Point3F& s);
  552. Matrix<DATA_TYPE, rows, cols>& scale(DATA_TYPE s) { return scale(Point3F(s, s, s)); }
  553. void setColumn(S32 col, const Point4F& cptr);
  554. void setColumn(S32 col, const Point3F& cptr);
  555. void setRow(S32 row, const Point4F& cptr);
  556. void setRow(S32 row, const Point3F& cptr);
  557. // ------ Getters ------
  558. bool isAffine() const;
  559. bool isIdentity() const;
  560. Point3F getScale() const;
  561. EulerF toEuler() const;
  562. Point3F getPosition() const;
  563. void getColumn(S32 col, Point4F* cptr) const;
  564. Point4F getColumn4F(S32 col) const { Point4F ret; getColumn(col, &ret); return ret; }
  565. void getColumn(S32 col, Point3F* cptr) const;
  566. Point3F getColumn3F(S32 col) const { Point3F ret; getColumn(col, &ret); return ret; }
  567. void getRow(S32 row, Point4F* cptr) const;
  568. Point4F getRow4F(S32 row) const { Point4F ret; getRow(row, &ret); return ret; }
  569. void getRow(S32 row, Point3F* cptr) const;
  570. Point3F getRow3F(S32 row) const { Point3F ret; getRow(row, &ret); return ret; }
  571. DATA_TYPE* getData() {
  572. return data;
  573. }
  574. const DATA_TYPE* getData() const {
  575. return data;
  576. }
  577. void dumpMatrix(const char* caption = NULL) const;
  578. // Static identity matrix
  579. static const Matrix Identity;
  580. // ------ Operators ------
  581. operator DATA_TYPE* () { return (data); }
  582. operator const DATA_TYPE* () const { return (DATA_TYPE*)(data); }
  583. DATA_TYPE& operator()(U32 row, U32 col) {
  584. if (row >= rows || col >= cols)
  585. AssertFatal(false, "Matrix indices out of range");
  586. return data[col * rows + row];
  587. }
  588. const DATA_TYPE& operator()(U32 row, U32 col) const {
  589. if (row >= rows || col >= cols)
  590. AssertFatal(false, "Matrix indices out of range");
  591. return data[col * rows + row];
  592. }
  593. };
  594. //--------------------------------------------
  595. // INLINE FUNCTIONS
  596. //--------------------------------------------
  597. template<typename DATA_TYPE, U32 rows, U32 cols>
  598. inline Matrix<DATA_TYPE, rows, cols>& Matrix<DATA_TYPE, rows, cols>::transpose()
  599. {
  600. // square matrices can just swap, non square requires a temp mat.
  601. if (rows == cols) {
  602. for (U32 i = 0; i < rows; i++) {
  603. for (U32 j = 0; j < cols; j++) {
  604. std::swap((*this)(j, i), (*this)(i, j));
  605. }
  606. }
  607. }
  608. else {
  609. Matrix<DATA_TYPE, rows, cols> result;
  610. for (U32 i = 0; i < rows; i++) {
  611. for (U32 j = 0; j < cols; j++) {
  612. result(j, i) = (*this)(i, j);
  613. }
  614. }
  615. std::copy(std::begin(result.data), std::end(result.data), std::begin(data));
  616. }
  617. return (*this);
  618. }
  619. template<typename DATA_TYPE, U32 rows, U32 cols>
  620. inline Matrix<DATA_TYPE, rows, cols>& Matrix<DATA_TYPE, rows, cols>::identity()
  621. {
  622. for (U32 i = 0; i < rows; i++) {
  623. for (U32 j = 0; j < cols; j++) {
  624. if (j == i)
  625. (*this)(i, j) = static_cast<DATA_TYPE>(1);
  626. else
  627. (*this)(i, j) = static_cast<DATA_TYPE>(0);
  628. }
  629. }
  630. return (*this);
  631. }
  632. template<typename DATA_TYPE, U32 rows, U32 cols>
  633. inline Matrix<DATA_TYPE, rows, cols>& Matrix<DATA_TYPE, rows, cols>::scale(const Point3F& s)
  634. {
  635. // torques scale applies directly, does not create another matrix to multiply with the translation matrix.
  636. AssertFatal(rows >= 3 && cols >= 3, "Scale can only be applied 3x3 or more");
  637. for (U32 i = 0; i < 3; i++) {
  638. for (U32 j = 0; j < 3; j++) {
  639. DATA_TYPE scale = (i == 0) ? s.x : (i == 1) ? s.y : s.z;
  640. (*this)(i, j) *= scale;
  641. }
  642. }
  643. return (*this);
  644. }
  645. template<typename DATA_TYPE, U32 rows, U32 cols>
  646. inline bool Matrix<DATA_TYPE, rows, cols>::isIdentity() const {
  647. for (U32 i = 0; i < rows; i++) {
  648. for (U32 j = 0; j < cols; j++) {
  649. if (j == i) {
  650. if((*this)(i, j) != static_cast<DATA_TYPE>(1)) {
  651. return false;
  652. }
  653. }
  654. else {
  655. if((*this)(i, j) != static_cast<DATA_TYPE>(0)) {
  656. return false;
  657. }
  658. }
  659. }
  660. }
  661. return true;
  662. }
  663. template<typename DATA_TYPE, U32 rows, U32 cols>
  664. inline Point3F Matrix<DATA_TYPE, rows, cols>::getScale() const
  665. {
  666. // this function assumes the matrix has scale applied through the scale(const Point3F& s) function.
  667. // for now assume float since we have point3F.
  668. AssertFatal(rows >= 3 && cols >= 3, "Scale can only be applied 3x3 or more");
  669. Point3F scale;
  670. scale.x = mSqrt((*this)(0, 0) * (*this)(0, 0) + (*this)(1, 0) * (*this)(1, 0) + (*this)(2, 0) * (*this)(2, 0));
  671. scale.y = mSqrt((*this)(0, 1) * (*this)(0, 1) + (*this)(1, 1) * (*this)(1, 1) + (*this)(2, 1) * (*this)(2, 1));
  672. scale.z = mSqrt((*this)(0, 2) * (*this)(0, 2) + (*this)(1, 2) * (*this)(1, 2) + (*this)(2, 2) * (*this)(2, 2));
  673. return scale;
  674. }
  675. template<typename DATA_TYPE, U32 rows, U32 cols>
  676. inline Point3F Matrix<DATA_TYPE, rows, cols>::getPosition() const
  677. {
  678. Point3F pos;
  679. getColumn(3, &pos);
  680. return pos;
  681. }
  682. template<typename DATA_TYPE, U32 rows, U32 cols>
  683. inline void Matrix<DATA_TYPE, rows, cols>::getColumn(S32 col, Point4F* cptr) const
  684. {
  685. if (rows >= 2)
  686. {
  687. cptr->x = (*this)(0, col);
  688. cptr->y = (*this)(1, col);
  689. }
  690. if (rows >= 3)
  691. cptr->z = (*this)(2, col);
  692. else
  693. cptr->z = 0.0f;
  694. if (rows >= 4)
  695. cptr->w = (*this)(3, col);
  696. else
  697. cptr->w = 0.0f;
  698. }
  699. template<typename DATA_TYPE, U32 rows, U32 cols>
  700. inline void Matrix<DATA_TYPE, rows, cols>::getColumn(S32 col, Point3F* cptr) const
  701. {
  702. if (rows >= 2)
  703. {
  704. cptr->x = (*this)(0, col);
  705. cptr->y = (*this)(1, col);
  706. }
  707. if (rows >= 3)
  708. cptr->z = (*this)(2, col);
  709. else
  710. cptr->z = 0.0f;
  711. }
  712. template<typename DATA_TYPE, U32 rows, U32 cols>
  713. inline void Matrix<DATA_TYPE, rows, cols>::setColumn(S32 col, const Point4F &cptr) {
  714. if(rows >= 2)
  715. {
  716. (*this)(0, col) = cptr.x;
  717. (*this)(1, col) = cptr.y;
  718. }
  719. if(rows >= 3)
  720. (*this)(2, col) = cptr.z;
  721. if(rows >= 4)
  722. (*this)(3, col) = cptr.w;
  723. }
  724. template<typename DATA_TYPE, U32 rows, U32 cols>
  725. inline void Matrix<DATA_TYPE, rows, cols>::setColumn(S32 col, const Point3F &cptr) {
  726. if(rows >= 2)
  727. {
  728. (*this)(0, col) = cptr.x;
  729. (*this)(1, col) = cptr.y;
  730. }
  731. if(rows >= 3)
  732. (*this)(2, col) = cptr.z;
  733. }
  734. template<typename DATA_TYPE, U32 rows, U32 cols>
  735. inline void Matrix<DATA_TYPE, rows, cols>::getRow(S32 row, Point4F* cptr) const
  736. {
  737. if (cols >= 2)
  738. {
  739. cptr->x = (*this)(row, 0);
  740. cptr->y = (*this)(row, 1);
  741. }
  742. if (cols >= 3)
  743. cptr->z = (*this)(row, 2);
  744. else
  745. cptr->z = 0.0f;
  746. if (cols >= 4)
  747. cptr->w = (*this)(row, 3);
  748. else
  749. cptr->w = 0.0f;
  750. }
  751. template<typename DATA_TYPE, U32 rows, U32 cols>
  752. inline void Matrix<DATA_TYPE, rows, cols>::getRow(S32 row, Point3F* cptr) const
  753. {
  754. if (cols >= 2)
  755. {
  756. cptr->x = (*this)(row, 0);
  757. cptr->y = (*this)(row, 1);
  758. }
  759. if (cols >= 3)
  760. cptr->z = (*this)(row, 2);
  761. else
  762. cptr->z = 0.0f;
  763. }
  764. template<typename DATA_TYPE, U32 rows, U32 cols>
  765. inline void Matrix<DATA_TYPE, rows, cols>::setRow(S32 row, const Point4F& cptr) {
  766. if(cols >= 2)
  767. {
  768. (*this)(row, 0) = cptr.x;
  769. (*this)(row, 1) = cptr.y;
  770. }
  771. if(cols >= 3)
  772. (*this)(row, 2) = cptr.z;
  773. if(cols >= 4)
  774. (*this)(row, 3) = cptr.w;
  775. }
  776. template<typename DATA_TYPE, U32 rows, U32 cols>
  777. inline void Matrix<DATA_TYPE, rows, cols>::setRow(S32 row, const Point3F& cptr) {
  778. if(cols >= 2)
  779. {
  780. (*this)(row, 0) = cptr.x;
  781. (*this)(row, 1) = cptr.y;
  782. }
  783. if(cols >= 3)
  784. (*this)(row, 2) = cptr.z;
  785. }
  786. //--------------------------------------------
  787. // INLINE FUNCTIONS END
  788. //--------------------------------------------
  789. typedef Matrix<F32, 4, 4> Matrix4F;
  790. class MatrixTemplateExport
  791. {
  792. public:
  793. template <typename T, U32 rows, U32 cols>
  794. static EngineFieldTable::Field getMatrixField();
  795. };
  796. template<typename T, U32 rows, U32 cols>
  797. inline EngineFieldTable::Field MatrixTemplateExport::getMatrixField()
  798. {
  799. typedef Matrix<T, rows, cols> ThisType;
  800. return _FIELD_AS(T, data, data, rows * cols, "");
  801. }
  802. #endif //_MMATRIX_H_