mMatrix.cpp 8.0 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222
  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. #include "core/strings/stringFunctions.h"
  23. #include "core/frameAllocator.h"
  24. #include "math/mMatrix.h"
  25. #include "console/console.h"
  26. #include "console/enginePrimitives.h"
  27. #include "console/engineTypes.h"
  28. #ifndef USE_TEMPLATE_MATRIX
  29. const MatrixF MatrixF::Identity( true );
  30. // idx(i,j) is index to element in column i, row j
  31. void MatrixF::transposeTo(F32 *matrix) const
  32. {
  33. matrix[idx(0,0)] = m[idx(0,0)];
  34. matrix[idx(0,1)] = m[idx(1,0)];
  35. matrix[idx(0,2)] = m[idx(2,0)];
  36. matrix[idx(0,3)] = m[idx(3,0)];
  37. matrix[idx(1,0)] = m[idx(0,1)];
  38. matrix[idx(1,1)] = m[idx(1,1)];
  39. matrix[idx(1,2)] = m[idx(2,1)];
  40. matrix[idx(1,3)] = m[idx(3,1)];
  41. matrix[idx(2,0)] = m[idx(0,2)];
  42. matrix[idx(2,1)] = m[idx(1,2)];
  43. matrix[idx(2,2)] = m[idx(2,2)];
  44. matrix[idx(2,3)] = m[idx(3,2)];
  45. matrix[idx(3,0)] = m[idx(0,3)];
  46. matrix[idx(3,1)] = m[idx(1,3)];
  47. matrix[idx(3,2)] = m[idx(2,3)];
  48. matrix[idx(3,3)] = m[idx(3,3)];
  49. }
  50. bool MatrixF::isAffine() const
  51. {
  52. // An affine transform is defined by the following structure
  53. //
  54. // [ X X X P ]
  55. // [ X X X P ]
  56. // [ X X X P ]
  57. // [ 0 0 0 1 ]
  58. //
  59. // Where X is an orthonormal 3x3 submatrix and P is an arbitrary translation
  60. // We'll check in the following order:
  61. // 1: [3][3] must be 1
  62. // 2: Shear portion must be zero
  63. // 3: Dot products of rows and columns must be zero
  64. // 4: Length of rows and columns must be 1
  65. //
  66. if (m[idx(3,3)] != 1.0f)
  67. return false;
  68. if (m[idx(0,3)] != 0.0f ||
  69. m[idx(1,3)] != 0.0f ||
  70. m[idx(2,3)] != 0.0f)
  71. return false;
  72. Point3F one, two, three;
  73. getColumn(0, &one);
  74. getColumn(1, &two);
  75. getColumn(2, &three);
  76. if (mDot(one, two) > 0.0001f ||
  77. mDot(one, three) > 0.0001f ||
  78. mDot(two, three) > 0.0001f)
  79. return false;
  80. if (mFabs(1.0f - one.lenSquared()) > 0.0001f ||
  81. mFabs(1.0f - two.lenSquared()) > 0.0001f ||
  82. mFabs(1.0f - three.lenSquared()) > 0.0001f)
  83. return false;
  84. getRow(0, &one);
  85. getRow(1, &two);
  86. getRow(2, &three);
  87. if (mDot(one, two) > 0.0001f ||
  88. mDot(one, three) > 0.0001f ||
  89. mDot(two, three) > 0.0001f)
  90. return false;
  91. if (mFabs(1.0f - one.lenSquared()) > 0.0001f ||
  92. mFabs(1.0f - two.lenSquared()) > 0.0001f ||
  93. mFabs(1.0f - three.lenSquared()) > 0.0001f)
  94. return false;
  95. // We're ok.
  96. return true;
  97. }
  98. // Perform inverse on full 4x4 matrix. Used in special cases only, so not at all optimized.
  99. bool MatrixF::fullInverse()
  100. {
  101. Point4F a,b,c,d;
  102. getRow(0,&a);
  103. getRow(1,&b);
  104. getRow(2,&c);
  105. getRow(3,&d);
  106. // det = a0*b1*c2*d3 - a0*b1*c3*d2 - a0*c1*b2*d3 + a0*c1*b3*d2 + a0*d1*b2*c3 - a0*d1*b3*c2 -
  107. // b0*a1*c2*d3 + b0*a1*c3*d2 + b0*c1*a2*d3 - b0*c1*a3*d2 - b0*d1*a2*c3 + b0*d1*a3*c2 +
  108. // c0*a1*b2*d3 - c0*a1*b3*d2 - c0*b1*a2*d3 + c0*b1*a3*d2 + c0*d1*a2*b3 - c0*d1*a3*b2 -
  109. // d0*a1*b2*c3 + d0*a1*b3*c2 + d0*b1*a2*c3 - d0*b1*a3*c2 - d0*c1*a2*b3 + d0*c1*a3*b2
  110. 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
  111. - 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
  112. + 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
  113. - 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;
  114. if (mFabs(det)<0.00001f)
  115. return false;
  116. Point4F aa,bb,cc,dd;
  117. 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;
  118. 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;
  119. 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;
  120. 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;
  121. 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;
  122. 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;
  123. 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;
  124. 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;
  125. 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;
  126. 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;
  127. 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;
  128. 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;
  129. 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;
  130. 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;
  131. 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;
  132. 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;
  133. setRow(0,aa);
  134. setRow(1,bb);
  135. setRow(2,cc);
  136. setRow(3,dd);
  137. mul(1.0f/det);
  138. return true;
  139. }
  140. void MatrixF::reverseProjection()
  141. {
  142. m[idx(0, 2)] = m[idx(0, 3)] - m[idx(0, 2)];
  143. m[idx(1, 2)] = m[idx(1, 3)] - m[idx(1, 2)];
  144. m[idx(2, 2)] = m[idx(2, 3)] - m[idx(2, 2)];
  145. m[idx(3, 2)] = m[idx(3, 3)] - m[idx(3, 2)];
  146. }
  147. EulerF MatrixF::toEuler() const
  148. {
  149. const F32 * mat = m;
  150. EulerF r;
  151. r.x = mAsin(mClampF(mat[MatrixF::idx(2,1)], -1.0, 1.0));
  152. if(mCos(r.x) != 0.f)
  153. {
  154. r.y = mAtan2(-mat[MatrixF::idx(2,0)], mat[MatrixF::idx(2,2)]);
  155. r.z = mAtan2(-mat[MatrixF::idx(0,1)], mat[MatrixF::idx(1,1)]);
  156. }
  157. else
  158. {
  159. r.y = 0.f;
  160. r.z = mAtan2(mat[MatrixF::idx(1,0)], mat[MatrixF::idx(0,0)]);
  161. }
  162. return r;
  163. }
  164. void MatrixF::dumpMatrix(const char *caption /* =NULL */) const
  165. {
  166. U32 size = (caption == NULL)? 0 : dStrlen(caption);
  167. FrameTemp<char> spacer(size+1);
  168. char *spacerRef = spacer;
  169. dMemset(spacerRef, ' ', size);
  170. spacerRef[size] = 0;
  171. 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)]);
  172. 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)]);
  173. 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)]);
  174. 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)]);
  175. }
  176. EngineFieldTable::Field MatrixFEngineExport::getMatrixField()
  177. {
  178. typedef MatrixF ThisType;
  179. return _FIELD_AS(F32, m, m, 16, "");
  180. }
  181. #else // !USE_TEMPLATE_MATRIX
  182. //------------------------------------
  183. // Templatized matrix class to replace MATRIXF above
  184. // due to templated class, all functions need to be inline
  185. //------------------------------------
  186. #endif // !USE_TEMPLATE_MATRIX