b3Matrix3x3.h 44 KB

1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556575859606162636465666768697071727374757677787980818283848586878889909192939495969798991001011021031041051061071081091101111121131141151161171181191201211221231241251261271281291301311321331341351361371381391401411421431441451461471481491501511521531541551561571581591601611621631641651661671681691701711721731741751761771781791801811821831841851861871881891901911921931941951961971981992002012022032042052062072082092102112122132142152162172182192202212222232242252262272282292302312322332342352362372382392402412422432442452462472482492502512522532542552562572582592602612622632642652662672682692702712722732742752762772782792802812822832842852862872882892902912922932942952962972982993003013023033043053063073083093103113123133143153163173183193203213223233243253263273283293303313323333343353363373383393403413423433443453463473483493503513523533543553563573583593603613623633643653663673683693703713723733743753763773783793803813823833843853863873883893903913923933943953963973983994004014024034044054064074084094104114124134144154164174184194204214224234244254264274284294304314324334344354364374384394404414424434444454464474484494504514524534544554564574584594604614624634644654664674684694704714724734744754764774784794804814824834844854864874884894904914924934944954964974984995005015025035045055065075085095105115125135145155165175185195205215225235245255265275285295305315325335345355365375385395405415425435445455465475485495505515525535545555565575585595605615625635645655665675685695705715725735745755765775785795805815825835845855865875885895905915925935945955965975985996006016026036046056066076086096106116126136146156166176186196206216226236246256266276286296306316326336346356366376386396406416426436446456466476486496506516526536546556566576586596606616626636646656666676686696706716726736746756766776786796806816826836846856866876886896906916926936946956966976986997007017027037047057067077087097107117127137147157167177187197207217227237247257267277287297307317327337347357367377387397407417427437447457467477487497507517527537547557567577587597607617627637647657667677687697707717727737747757767777787797807817827837847857867877887897907917927937947957967977987998008018028038048058068078088098108118128138148158168178188198208218228238248258268278288298308318328338348358368378388398408418428438448458468478488498508518528538548558568578588598608618628638648658668678688698708718728738748758768778788798808818828838848858868878888898908918928938948958968978988999009019029039049059069079089099109119129139149159169179189199209219229239249259269279289299309319329339349359369379389399409419429439449459469479489499509519529539549559569579589599609619629639649659669679689699709719729739749759769779789799809819829839849859869879889899909919929939949959969979989991000100110021003100410051006100710081009101010111012101310141015101610171018101910201021102210231024102510261027102810291030103110321033103410351036103710381039104010411042104310441045104610471048104910501051105210531054105510561057105810591060106110621063106410651066106710681069107010711072107310741075107610771078107910801081108210831084108510861087108810891090109110921093109410951096109710981099110011011102110311041105110611071108110911101111111211131114111511161117111811191120112111221123112411251126112711281129113011311132113311341135113611371138113911401141114211431144114511461147114811491150115111521153115411551156115711581159116011611162116311641165116611671168116911701171117211731174117511761177117811791180118111821183118411851186118711881189119011911192119311941195119611971198119912001201120212031204120512061207120812091210121112121213121412151216121712181219122012211222122312241225122612271228122912301231123212331234123512361237123812391240124112421243124412451246124712481249125012511252125312541255125612571258125912601261126212631264126512661267126812691270127112721273127412751276127712781279128012811282128312841285128612871288128912901291129212931294129512961297129812991300130113021303130413051306130713081309131013111312131313141315131613171318131913201321132213231324132513261327132813291330133113321333133413351336133713381339134013411342134313441345134613471348134913501351135213531354135513561357135813591360136113621363
  1. /*
  2. Copyright (c) 2003-2013 Gino van den Bergen / Erwin Coumans http://bulletphysics.org
  3. This software is provided 'as-is', without any express or implied warranty.
  4. In no event will the authors be held liable for any damages arising from the use of this software.
  5. Permission is granted to anyone to use this software for any purpose,
  6. including commercial applications, and to alter it and redistribute it freely,
  7. subject to the following restrictions:
  8. 1. The origin of this software must not be misrepresented; you must not claim that you wrote the original software. If you use this software in a product, an acknowledgment in the product documentation would be appreciated but is not required.
  9. 2. Altered source versions must be plainly marked as such, and must not be misrepresented as being the original software.
  10. 3. This notice may not be removed or altered from any source distribution.
  11. */
  12. #ifndef B3_MATRIX3x3_H
  13. #define B3_MATRIX3x3_H
  14. #include "b3Vector3.h"
  15. #include "b3Quaternion.h"
  16. #include <stdio.h>
  17. #ifdef B3_USE_SSE
  18. //const __m128 B3_ATTRIBUTE_ALIGNED16(b3v2220) = {2.0f, 2.0f, 2.0f, 0.0f};
  19. const __m128 B3_ATTRIBUTE_ALIGNED16(b3vMPPP) = {-0.0f, +0.0f, +0.0f, +0.0f};
  20. #endif
  21. #if defined(B3_USE_SSE) || defined(B3_USE_NEON)
  22. const b3SimdFloat4 B3_ATTRIBUTE_ALIGNED16(b3v1000) = {1.0f, 0.0f, 0.0f, 0.0f};
  23. const b3SimdFloat4 B3_ATTRIBUTE_ALIGNED16(b3v0100) = {0.0f, 1.0f, 0.0f, 0.0f};
  24. const b3SimdFloat4 B3_ATTRIBUTE_ALIGNED16(b3v0010) = {0.0f, 0.0f, 1.0f, 0.0f};
  25. #endif
  26. #ifdef B3_USE_DOUBLE_PRECISION
  27. #define b3Matrix3x3Data b3Matrix3x3DoubleData
  28. #else
  29. #define b3Matrix3x3Data b3Matrix3x3FloatData
  30. #endif //B3_USE_DOUBLE_PRECISION
  31. /**@brief The b3Matrix3x3 class implements a 3x3 rotation matrix, to perform linear algebra in combination with b3Quaternion, b3Transform and b3Vector3.
  32. * Make sure to only include a pure orthogonal matrix without scaling. */
  33. B3_ATTRIBUTE_ALIGNED16(class) b3Matrix3x3 {
  34. ///Data storage for the matrix, each vector is a row of the matrix
  35. b3Vector3 m_el[3];
  36. public:
  37. /** @brief No initializaion constructor */
  38. b3Matrix3x3 () {}
  39. // explicit b3Matrix3x3(const b3Scalar *m) { setFromOpenGLSubMatrix(m); }
  40. /**@brief Constructor from Quaternion */
  41. explicit b3Matrix3x3(const b3Quaternion& q) { setRotation(q); }
  42. /*
  43. template <typename b3Scalar>
  44. Matrix3x3(const b3Scalar& yaw, const b3Scalar& pitch, const b3Scalar& roll)
  45. {
  46. setEulerYPR(yaw, pitch, roll);
  47. }
  48. */
  49. /** @brief Constructor with row major formatting */
  50. b3Matrix3x3(const b3Scalar& xx, const b3Scalar& xy, const b3Scalar& xz,
  51. const b3Scalar& yx, const b3Scalar& yy, const b3Scalar& yz,
  52. const b3Scalar& zx, const b3Scalar& zy, const b3Scalar& zz)
  53. {
  54. setValue(xx, xy, xz,
  55. yx, yy, yz,
  56. zx, zy, zz);
  57. }
  58. #if (defined (B3_USE_SSE_IN_API) && defined (B3_USE_SSE))|| defined (B3_USE_NEON)
  59. B3_FORCE_INLINE b3Matrix3x3 (const b3SimdFloat4 v0, const b3SimdFloat4 v1, const b3SimdFloat4 v2 )
  60. {
  61. m_el[0].mVec128 = v0;
  62. m_el[1].mVec128 = v1;
  63. m_el[2].mVec128 = v2;
  64. }
  65. B3_FORCE_INLINE b3Matrix3x3 (const b3Vector3& v0, const b3Vector3& v1, const b3Vector3& v2 )
  66. {
  67. m_el[0] = v0;
  68. m_el[1] = v1;
  69. m_el[2] = v2;
  70. }
  71. // Copy constructor
  72. B3_FORCE_INLINE b3Matrix3x3(const b3Matrix3x3& rhs)
  73. {
  74. m_el[0].mVec128 = rhs.m_el[0].mVec128;
  75. m_el[1].mVec128 = rhs.m_el[1].mVec128;
  76. m_el[2].mVec128 = rhs.m_el[2].mVec128;
  77. }
  78. // Assignment Operator
  79. B3_FORCE_INLINE b3Matrix3x3& operator=(const b3Matrix3x3& m)
  80. {
  81. m_el[0].mVec128 = m.m_el[0].mVec128;
  82. m_el[1].mVec128 = m.m_el[1].mVec128;
  83. m_el[2].mVec128 = m.m_el[2].mVec128;
  84. return *this;
  85. }
  86. #else
  87. /** @brief Copy constructor */
  88. B3_FORCE_INLINE b3Matrix3x3 (const b3Matrix3x3& other)
  89. {
  90. m_el[0] = other.m_el[0];
  91. m_el[1] = other.m_el[1];
  92. m_el[2] = other.m_el[2];
  93. }
  94. /** @brief Assignment Operator */
  95. B3_FORCE_INLINE b3Matrix3x3& operator=(const b3Matrix3x3& other)
  96. {
  97. m_el[0] = other.m_el[0];
  98. m_el[1] = other.m_el[1];
  99. m_el[2] = other.m_el[2];
  100. return *this;
  101. }
  102. #endif
  103. /** @brief Get a column of the matrix as a vector
  104. * @param i Column number 0 indexed */
  105. B3_FORCE_INLINE b3Vector3 getColumn(int i) const
  106. {
  107. return b3MakeVector3(m_el[0][i],m_el[1][i],m_el[2][i]);
  108. }
  109. /** @brief Get a row of the matrix as a vector
  110. * @param i Row number 0 indexed */
  111. B3_FORCE_INLINE const b3Vector3& getRow(int i) const
  112. {
  113. b3FullAssert(0 <= i && i < 3);
  114. return m_el[i];
  115. }
  116. /** @brief Get a mutable reference to a row of the matrix as a vector
  117. * @param i Row number 0 indexed */
  118. B3_FORCE_INLINE b3Vector3& operator[](int i)
  119. {
  120. b3FullAssert(0 <= i && i < 3);
  121. return m_el[i];
  122. }
  123. /** @brief Get a const reference to a row of the matrix as a vector
  124. * @param i Row number 0 indexed */
  125. B3_FORCE_INLINE const b3Vector3& operator[](int i) const
  126. {
  127. b3FullAssert(0 <= i && i < 3);
  128. return m_el[i];
  129. }
  130. /** @brief Multiply by the target matrix on the right
  131. * @param m Rotation matrix to be applied
  132. * Equivilant to this = this * m */
  133. b3Matrix3x3& operator*=(const b3Matrix3x3& m);
  134. /** @brief Adds by the target matrix on the right
  135. * @param m matrix to be applied
  136. * Equivilant to this = this + m */
  137. b3Matrix3x3& operator+=(const b3Matrix3x3& m);
  138. /** @brief Substractss by the target matrix on the right
  139. * @param m matrix to be applied
  140. * Equivilant to this = this - m */
  141. b3Matrix3x3& operator-=(const b3Matrix3x3& m);
  142. /** @brief Set from the rotational part of a 4x4 OpenGL matrix
  143. * @param m A pointer to the beginning of the array of scalars*/
  144. void setFromOpenGLSubMatrix(const b3Scalar *m)
  145. {
  146. m_el[0].setValue(m[0],m[4],m[8]);
  147. m_el[1].setValue(m[1],m[5],m[9]);
  148. m_el[2].setValue(m[2],m[6],m[10]);
  149. }
  150. /** @brief Set the values of the matrix explicitly (row major)
  151. * @param xx Top left
  152. * @param xy Top Middle
  153. * @param xz Top Right
  154. * @param yx Middle Left
  155. * @param yy Middle Middle
  156. * @param yz Middle Right
  157. * @param zx Bottom Left
  158. * @param zy Bottom Middle
  159. * @param zz Bottom Right*/
  160. void setValue(const b3Scalar& xx, const b3Scalar& xy, const b3Scalar& xz,
  161. const b3Scalar& yx, const b3Scalar& yy, const b3Scalar& yz,
  162. const b3Scalar& zx, const b3Scalar& zy, const b3Scalar& zz)
  163. {
  164. m_el[0].setValue(xx,xy,xz);
  165. m_el[1].setValue(yx,yy,yz);
  166. m_el[2].setValue(zx,zy,zz);
  167. }
  168. /** @brief Set the matrix from a quaternion
  169. * @param q The Quaternion to match */
  170. void setRotation(const b3Quaternion& q)
  171. {
  172. b3Scalar d = q.length2();
  173. b3FullAssert(d != b3Scalar(0.0));
  174. b3Scalar s = b3Scalar(2.0) / d;
  175. #if defined (B3_USE_SSE_IN_API) && defined (B3_USE_SSE)
  176. __m128 vs, Q = q.get128();
  177. __m128i Qi = b3CastfTo128i(Q);
  178. __m128 Y, Z;
  179. __m128 V1, V2, V3;
  180. __m128 V11, V21, V31;
  181. __m128 NQ = _mm_xor_ps(Q, b3vMzeroMask);
  182. __m128i NQi = b3CastfTo128i(NQ);
  183. V1 = b3CastiTo128f(_mm_shuffle_epi32 (Qi, B3_SHUFFLE(1,0,2,3))); // Y X Z W
  184. V2 = _mm_shuffle_ps(NQ, Q, B3_SHUFFLE(0,0,1,3)); // -X -X Y W
  185. V3 = b3CastiTo128f(_mm_shuffle_epi32 (Qi, B3_SHUFFLE(2,1,0,3))); // Z Y X W
  186. V1 = _mm_xor_ps(V1, b3vMPPP); // change the sign of the first element
  187. V11 = b3CastiTo128f(_mm_shuffle_epi32 (Qi, B3_SHUFFLE(1,1,0,3))); // Y Y X W
  188. V21 = _mm_unpackhi_ps(Q, Q); // Z Z W W
  189. V31 = _mm_shuffle_ps(Q, NQ, B3_SHUFFLE(0,2,0,3)); // X Z -X -W
  190. V2 = V2 * V1; //
  191. V1 = V1 * V11; //
  192. V3 = V3 * V31; //
  193. V11 = _mm_shuffle_ps(NQ, Q, B3_SHUFFLE(2,3,1,3)); // -Z -W Y W
  194. V11 = V11 * V21; //
  195. V21 = _mm_xor_ps(V21, b3vMPPP); // change the sign of the first element
  196. V31 = _mm_shuffle_ps(Q, NQ, B3_SHUFFLE(3,3,1,3)); // W W -Y -W
  197. V31 = _mm_xor_ps(V31, b3vMPPP); // change the sign of the first element
  198. Y = b3CastiTo128f(_mm_shuffle_epi32 (NQi, B3_SHUFFLE(3,2,0,3))); // -W -Z -X -W
  199. Z = b3CastiTo128f(_mm_shuffle_epi32 (Qi, B3_SHUFFLE(1,0,1,3))); // Y X Y W
  200. vs = _mm_load_ss(&s);
  201. V21 = V21 * Y;
  202. V31 = V31 * Z;
  203. V1 = V1 + V11;
  204. V2 = V2 + V21;
  205. V3 = V3 + V31;
  206. vs = b3_splat3_ps(vs, 0);
  207. // s ready
  208. V1 = V1 * vs;
  209. V2 = V2 * vs;
  210. V3 = V3 * vs;
  211. V1 = V1 + b3v1000;
  212. V2 = V2 + b3v0100;
  213. V3 = V3 + b3v0010;
  214. m_el[0] = b3MakeVector3(V1);
  215. m_el[1] = b3MakeVector3(V2);
  216. m_el[2] = b3MakeVector3(V3);
  217. #else
  218. b3Scalar xs = q.getX() * s, ys = q.getY() * s, zs = q.getZ() * s;
  219. b3Scalar wx = q.getW() * xs, wy = q.getW() * ys, wz = q.getW() * zs;
  220. b3Scalar xx = q.getX() * xs, xy = q.getX() * ys, xz = q.getX() * zs;
  221. b3Scalar yy = q.getY() * ys, yz = q.getY() * zs, zz = q.getZ() * zs;
  222. setValue(
  223. b3Scalar(1.0) - (yy + zz), xy - wz, xz + wy,
  224. xy + wz, b3Scalar(1.0) - (xx + zz), yz - wx,
  225. xz - wy, yz + wx, b3Scalar(1.0) - (xx + yy));
  226. #endif
  227. }
  228. /** @brief Set the matrix from euler angles using YPR around YXZ respectively
  229. * @param yaw Yaw about Y axis
  230. * @param pitch Pitch about X axis
  231. * @param roll Roll about Z axis
  232. */
  233. void setEulerYPR(const b3Scalar& yaw, const b3Scalar& pitch, const b3Scalar& roll)
  234. {
  235. setEulerZYX(roll, pitch, yaw);
  236. }
  237. /** @brief Set the matrix from euler angles YPR around ZYX axes
  238. * @param eulerX Roll about X axis
  239. * @param eulerY Pitch around Y axis
  240. * @param eulerZ Yaw aboud Z axis
  241. *
  242. * These angles are used to produce a rotation matrix. The euler
  243. * angles are applied in ZYX order. I.e a vector is first rotated
  244. * about X then Y and then Z
  245. **/
  246. void setEulerZYX(b3Scalar eulerX,b3Scalar eulerY,b3Scalar eulerZ) {
  247. ///@todo proposed to reverse this since it's labeled zyx but takes arguments xyz and it will match all other parts of the code
  248. b3Scalar ci ( b3Cos(eulerX));
  249. b3Scalar cj ( b3Cos(eulerY));
  250. b3Scalar ch ( b3Cos(eulerZ));
  251. b3Scalar si ( b3Sin(eulerX));
  252. b3Scalar sj ( b3Sin(eulerY));
  253. b3Scalar sh ( b3Sin(eulerZ));
  254. b3Scalar cc = ci * ch;
  255. b3Scalar cs = ci * sh;
  256. b3Scalar sc = si * ch;
  257. b3Scalar ss = si * sh;
  258. setValue(cj * ch, sj * sc - cs, sj * cc + ss,
  259. cj * sh, sj * ss + cc, sj * cs - sc,
  260. -sj, cj * si, cj * ci);
  261. }
  262. /**@brief Set the matrix to the identity */
  263. void setIdentity()
  264. {
  265. #if (defined(B3_USE_SSE_IN_API)&& defined (B3_USE_SSE)) || defined(B3_USE_NEON)
  266. m_el[0] = b3MakeVector3(b3v1000);
  267. m_el[1] = b3MakeVector3(b3v0100);
  268. m_el[2] = b3MakeVector3(b3v0010);
  269. #else
  270. setValue(b3Scalar(1.0), b3Scalar(0.0), b3Scalar(0.0),
  271. b3Scalar(0.0), b3Scalar(1.0), b3Scalar(0.0),
  272. b3Scalar(0.0), b3Scalar(0.0), b3Scalar(1.0));
  273. #endif
  274. }
  275. static const b3Matrix3x3& getIdentity()
  276. {
  277. #if (defined(B3_USE_SSE_IN_API)&& defined (B3_USE_SSE)) || defined(B3_USE_NEON)
  278. static const b3Matrix3x3
  279. identityMatrix(b3v1000, b3v0100, b3v0010);
  280. #else
  281. static const b3Matrix3x3
  282. identityMatrix(
  283. b3Scalar(1.0), b3Scalar(0.0), b3Scalar(0.0),
  284. b3Scalar(0.0), b3Scalar(1.0), b3Scalar(0.0),
  285. b3Scalar(0.0), b3Scalar(0.0), b3Scalar(1.0));
  286. #endif
  287. return identityMatrix;
  288. }
  289. /**@brief Fill the rotational part of an OpenGL matrix and clear the shear/perspective
  290. * @param m The array to be filled */
  291. void getOpenGLSubMatrix(b3Scalar *m) const
  292. {
  293. #if defined (B3_USE_SSE_IN_API) && defined (B3_USE_SSE)
  294. __m128 v0 = m_el[0].mVec128;
  295. __m128 v1 = m_el[1].mVec128;
  296. __m128 v2 = m_el[2].mVec128; // x2 y2 z2 w2
  297. __m128 *vm = (__m128 *)m;
  298. __m128 vT;
  299. v2 = _mm_and_ps(v2, b3vFFF0fMask); // x2 y2 z2 0
  300. vT = _mm_unpackhi_ps(v0, v1); // z0 z1 * *
  301. v0 = _mm_unpacklo_ps(v0, v1); // x0 x1 y0 y1
  302. v1 = _mm_shuffle_ps(v0, v2, B3_SHUFFLE(2, 3, 1, 3) ); // y0 y1 y2 0
  303. v0 = _mm_shuffle_ps(v0, v2, B3_SHUFFLE(0, 1, 0, 3) ); // x0 x1 x2 0
  304. v2 = b3CastdTo128f(_mm_move_sd(b3CastfTo128d(v2), b3CastfTo128d(vT))); // z0 z1 z2 0
  305. vm[0] = v0;
  306. vm[1] = v1;
  307. vm[2] = v2;
  308. #elif defined(B3_USE_NEON)
  309. // note: zeros the w channel. We can preserve it at the cost of two more vtrn instructions.
  310. static const uint32x2_t zMask = (const uint32x2_t) {-1, 0 };
  311. float32x4_t *vm = (float32x4_t *)m;
  312. float32x4x2_t top = vtrnq_f32( m_el[0].mVec128, m_el[1].mVec128 ); // {x0 x1 z0 z1}, {y0 y1 w0 w1}
  313. float32x2x2_t bl = vtrn_f32( vget_low_f32(m_el[2].mVec128), vdup_n_f32(0.0f) ); // {x2 0 }, {y2 0}
  314. float32x4_t v0 = vcombine_f32( vget_low_f32(top.val[0]), bl.val[0] );
  315. float32x4_t v1 = vcombine_f32( vget_low_f32(top.val[1]), bl.val[1] );
  316. float32x2_t q = (float32x2_t) vand_u32( (uint32x2_t) vget_high_f32( m_el[2].mVec128), zMask );
  317. float32x4_t v2 = vcombine_f32( vget_high_f32(top.val[0]), q ); // z0 z1 z2 0
  318. vm[0] = v0;
  319. vm[1] = v1;
  320. vm[2] = v2;
  321. #else
  322. m[0] = b3Scalar(m_el[0].getX());
  323. m[1] = b3Scalar(m_el[1].getX());
  324. m[2] = b3Scalar(m_el[2].getX());
  325. m[3] = b3Scalar(0.0);
  326. m[4] = b3Scalar(m_el[0].getY());
  327. m[5] = b3Scalar(m_el[1].getY());
  328. m[6] = b3Scalar(m_el[2].getY());
  329. m[7] = b3Scalar(0.0);
  330. m[8] = b3Scalar(m_el[0].getZ());
  331. m[9] = b3Scalar(m_el[1].getZ());
  332. m[10] = b3Scalar(m_el[2].getZ());
  333. m[11] = b3Scalar(0.0);
  334. #endif
  335. }
  336. /**@brief Get the matrix represented as a quaternion
  337. * @param q The quaternion which will be set */
  338. void getRotation(b3Quaternion& q) const
  339. {
  340. #if (defined (B3_USE_SSE_IN_API) && defined (B3_USE_SSE))|| defined (B3_USE_NEON)
  341. b3Scalar trace = m_el[0].getX() + m_el[1].getY() + m_el[2].getZ();
  342. b3Scalar s, x;
  343. union {
  344. b3SimdFloat4 vec;
  345. b3Scalar f[4];
  346. } temp;
  347. if (trace > b3Scalar(0.0))
  348. {
  349. x = trace + b3Scalar(1.0);
  350. temp.f[0]=m_el[2].getY() - m_el[1].getZ();
  351. temp.f[1]=m_el[0].getZ() - m_el[2].getX();
  352. temp.f[2]=m_el[1].getX() - m_el[0].getY();
  353. temp.f[3]=x;
  354. //temp.f[3]= s * b3Scalar(0.5);
  355. }
  356. else
  357. {
  358. int i, j, k;
  359. if(m_el[0].getX() < m_el[1].getY())
  360. {
  361. if( m_el[1].getY() < m_el[2].getZ() )
  362. { i = 2; j = 0; k = 1; }
  363. else
  364. { i = 1; j = 2; k = 0; }
  365. }
  366. else
  367. {
  368. if( m_el[0].getX() < m_el[2].getZ())
  369. { i = 2; j = 0; k = 1; }
  370. else
  371. { i = 0; j = 1; k = 2; }
  372. }
  373. x = m_el[i][i] - m_el[j][j] - m_el[k][k] + b3Scalar(1.0);
  374. temp.f[3] = (m_el[k][j] - m_el[j][k]);
  375. temp.f[j] = (m_el[j][i] + m_el[i][j]);
  376. temp.f[k] = (m_el[k][i] + m_el[i][k]);
  377. temp.f[i] = x;
  378. //temp.f[i] = s * b3Scalar(0.5);
  379. }
  380. s = b3Sqrt(x);
  381. q.set128(temp.vec);
  382. s = b3Scalar(0.5) / s;
  383. q *= s;
  384. #else
  385. b3Scalar trace = m_el[0].getX() + m_el[1].getY() + m_el[2].getZ();
  386. b3Scalar temp[4];
  387. if (trace > b3Scalar(0.0))
  388. {
  389. b3Scalar s = b3Sqrt(trace + b3Scalar(1.0));
  390. temp[3]=(s * b3Scalar(0.5));
  391. s = b3Scalar(0.5) / s;
  392. temp[0]=((m_el[2].getY() - m_el[1].getZ()) * s);
  393. temp[1]=((m_el[0].getZ() - m_el[2].getX()) * s);
  394. temp[2]=((m_el[1].getX() - m_el[0].getY()) * s);
  395. }
  396. else
  397. {
  398. int i = m_el[0].getX() < m_el[1].getY() ?
  399. (m_el[1].getY() < m_el[2].getZ() ? 2 : 1) :
  400. (m_el[0].getX() < m_el[2].getZ() ? 2 : 0);
  401. int j = (i + 1) % 3;
  402. int k = (i + 2) % 3;
  403. b3Scalar s = b3Sqrt(m_el[i][i] - m_el[j][j] - m_el[k][k] + b3Scalar(1.0));
  404. temp[i] = s * b3Scalar(0.5);
  405. s = b3Scalar(0.5) / s;
  406. temp[3] = (m_el[k][j] - m_el[j][k]) * s;
  407. temp[j] = (m_el[j][i] + m_el[i][j]) * s;
  408. temp[k] = (m_el[k][i] + m_el[i][k]) * s;
  409. }
  410. q.setValue(temp[0],temp[1],temp[2],temp[3]);
  411. #endif
  412. }
  413. /**@brief Get the matrix represented as euler angles around YXZ, roundtrip with setEulerYPR
  414. * @param yaw Yaw around Y axis
  415. * @param pitch Pitch around X axis
  416. * @param roll around Z axis */
  417. void getEulerYPR(b3Scalar& yaw, b3Scalar& pitch, b3Scalar& roll) const
  418. {
  419. // first use the normal calculus
  420. yaw = b3Scalar(b3Atan2(m_el[1].getX(), m_el[0].getX()));
  421. pitch = b3Scalar(b3Asin(-m_el[2].getX()));
  422. roll = b3Scalar(b3Atan2(m_el[2].getY(), m_el[2].getZ()));
  423. // on pitch = +/-HalfPI
  424. if (b3Fabs(pitch)==B3_HALF_PI)
  425. {
  426. if (yaw>0)
  427. yaw-=B3_PI;
  428. else
  429. yaw+=B3_PI;
  430. if (roll>0)
  431. roll-=B3_PI;
  432. else
  433. roll+=B3_PI;
  434. }
  435. };
  436. /**@brief Get the matrix represented as euler angles around ZYX
  437. * @param yaw Yaw around X axis
  438. * @param pitch Pitch around Y axis
  439. * @param roll around X axis
  440. * @param solution_number Which solution of two possible solutions ( 1 or 2) are possible values*/
  441. void getEulerZYX(b3Scalar& yaw, b3Scalar& pitch, b3Scalar& roll, unsigned int solution_number = 1) const
  442. {
  443. struct Euler
  444. {
  445. b3Scalar yaw;
  446. b3Scalar pitch;
  447. b3Scalar roll;
  448. };
  449. Euler euler_out;
  450. Euler euler_out2; //second solution
  451. //get the pointer to the raw data
  452. // Check that pitch is not at a singularity
  453. if (b3Fabs(m_el[2].getX()) >= 1)
  454. {
  455. euler_out.yaw = 0;
  456. euler_out2.yaw = 0;
  457. // From difference of angles formula
  458. b3Scalar delta = b3Atan2(m_el[0].getX(),m_el[0].getZ());
  459. if (m_el[2].getX() > 0) //gimbal locked up
  460. {
  461. euler_out.pitch = B3_PI / b3Scalar(2.0);
  462. euler_out2.pitch = B3_PI / b3Scalar(2.0);
  463. euler_out.roll = euler_out.pitch + delta;
  464. euler_out2.roll = euler_out.pitch + delta;
  465. }
  466. else // gimbal locked down
  467. {
  468. euler_out.pitch = -B3_PI / b3Scalar(2.0);
  469. euler_out2.pitch = -B3_PI / b3Scalar(2.0);
  470. euler_out.roll = -euler_out.pitch + delta;
  471. euler_out2.roll = -euler_out.pitch + delta;
  472. }
  473. }
  474. else
  475. {
  476. euler_out.pitch = - b3Asin(m_el[2].getX());
  477. euler_out2.pitch = B3_PI - euler_out.pitch;
  478. euler_out.roll = b3Atan2(m_el[2].getY()/b3Cos(euler_out.pitch),
  479. m_el[2].getZ()/b3Cos(euler_out.pitch));
  480. euler_out2.roll = b3Atan2(m_el[2].getY()/b3Cos(euler_out2.pitch),
  481. m_el[2].getZ()/b3Cos(euler_out2.pitch));
  482. euler_out.yaw = b3Atan2(m_el[1].getX()/b3Cos(euler_out.pitch),
  483. m_el[0].getX()/b3Cos(euler_out.pitch));
  484. euler_out2.yaw = b3Atan2(m_el[1].getX()/b3Cos(euler_out2.pitch),
  485. m_el[0].getX()/b3Cos(euler_out2.pitch));
  486. }
  487. if (solution_number == 1)
  488. {
  489. yaw = euler_out.yaw;
  490. pitch = euler_out.pitch;
  491. roll = euler_out.roll;
  492. }
  493. else
  494. {
  495. yaw = euler_out2.yaw;
  496. pitch = euler_out2.pitch;
  497. roll = euler_out2.roll;
  498. }
  499. }
  500. /**@brief Create a scaled copy of the matrix
  501. * @param s Scaling vector The elements of the vector will scale each column */
  502. b3Matrix3x3 scaled(const b3Vector3& s) const
  503. {
  504. #if (defined (B3_USE_SSE_IN_API) && defined (B3_USE_SSE))|| defined (B3_USE_NEON)
  505. return b3Matrix3x3(m_el[0] * s, m_el[1] * s, m_el[2] * s);
  506. #else
  507. return b3Matrix3x3(
  508. m_el[0].getX() * s.getX(), m_el[0].getY() * s.getY(), m_el[0].getZ() * s.getZ(),
  509. m_el[1].getX() * s.getX(), m_el[1].getY() * s.getY(), m_el[1].getZ() * s.getZ(),
  510. m_el[2].getX() * s.getX(), m_el[2].getY() * s.getY(), m_el[2].getZ() * s.getZ());
  511. #endif
  512. }
  513. /**@brief Return the determinant of the matrix */
  514. b3Scalar determinant() const;
  515. /**@brief Return the adjoint of the matrix */
  516. b3Matrix3x3 adjoint() const;
  517. /**@brief Return the matrix with all values non negative */
  518. b3Matrix3x3 absolute() const;
  519. /**@brief Return the transpose of the matrix */
  520. b3Matrix3x3 transpose() const;
  521. /**@brief Return the inverse of the matrix */
  522. b3Matrix3x3 inverse() const;
  523. b3Matrix3x3 transposeTimes(const b3Matrix3x3& m) const;
  524. b3Matrix3x3 timesTranspose(const b3Matrix3x3& m) const;
  525. B3_FORCE_INLINE b3Scalar tdotx(const b3Vector3& v) const
  526. {
  527. return m_el[0].getX() * v.getX() + m_el[1].getX() * v.getY() + m_el[2].getX() * v.getZ();
  528. }
  529. B3_FORCE_INLINE b3Scalar tdoty(const b3Vector3& v) const
  530. {
  531. return m_el[0].getY() * v.getX() + m_el[1].getY() * v.getY() + m_el[2].getY() * v.getZ();
  532. }
  533. B3_FORCE_INLINE b3Scalar tdotz(const b3Vector3& v) const
  534. {
  535. return m_el[0].getZ() * v.getX() + m_el[1].getZ() * v.getY() + m_el[2].getZ() * v.getZ();
  536. }
  537. /**@brief diagonalizes this matrix by the Jacobi method.
  538. * @param rot stores the rotation from the coordinate system in which the matrix is diagonal to the original
  539. * coordinate system, i.e., old_this = rot * new_this * rot^T.
  540. * @param threshold See iteration
  541. * @param iteration The iteration stops when all off-diagonal elements are less than the threshold multiplied
  542. * by the sum of the absolute values of the diagonal, or when maxSteps have been executed.
  543. *
  544. * Note that this matrix is assumed to be symmetric.
  545. */
  546. void diagonalize(b3Matrix3x3& rot, b3Scalar threshold, int maxSteps)
  547. {
  548. rot.setIdentity();
  549. for (int step = maxSteps; step > 0; step--)
  550. {
  551. // find off-diagonal element [p][q] with largest magnitude
  552. int p = 0;
  553. int q = 1;
  554. int r = 2;
  555. b3Scalar max = b3Fabs(m_el[0][1]);
  556. b3Scalar v = b3Fabs(m_el[0][2]);
  557. if (v > max)
  558. {
  559. q = 2;
  560. r = 1;
  561. max = v;
  562. }
  563. v = b3Fabs(m_el[1][2]);
  564. if (v > max)
  565. {
  566. p = 1;
  567. q = 2;
  568. r = 0;
  569. max = v;
  570. }
  571. b3Scalar t = threshold * (b3Fabs(m_el[0][0]) + b3Fabs(m_el[1][1]) + b3Fabs(m_el[2][2]));
  572. if (max <= t)
  573. {
  574. if (max <= B3_EPSILON * t)
  575. {
  576. return;
  577. }
  578. step = 1;
  579. }
  580. // compute Jacobi rotation J which leads to a zero for element [p][q]
  581. b3Scalar mpq = m_el[p][q];
  582. b3Scalar theta = (m_el[q][q] - m_el[p][p]) / (2 * mpq);
  583. b3Scalar theta2 = theta * theta;
  584. b3Scalar cos;
  585. b3Scalar sin;
  586. if (theta2 * theta2 < b3Scalar(10 / B3_EPSILON))
  587. {
  588. t = (theta >= 0) ? 1 / (theta + b3Sqrt(1 + theta2))
  589. : 1 / (theta - b3Sqrt(1 + theta2));
  590. cos = 1 / b3Sqrt(1 + t * t);
  591. sin = cos * t;
  592. }
  593. else
  594. {
  595. // approximation for large theta-value, i.e., a nearly diagonal matrix
  596. t = 1 / (theta * (2 + b3Scalar(0.5) / theta2));
  597. cos = 1 - b3Scalar(0.5) * t * t;
  598. sin = cos * t;
  599. }
  600. // apply rotation to matrix (this = J^T * this * J)
  601. m_el[p][q] = m_el[q][p] = 0;
  602. m_el[p][p] -= t * mpq;
  603. m_el[q][q] += t * mpq;
  604. b3Scalar mrp = m_el[r][p];
  605. b3Scalar mrq = m_el[r][q];
  606. m_el[r][p] = m_el[p][r] = cos * mrp - sin * mrq;
  607. m_el[r][q] = m_el[q][r] = cos * mrq + sin * mrp;
  608. // apply rotation to rot (rot = rot * J)
  609. for (int i = 0; i < 3; i++)
  610. {
  611. b3Vector3& row = rot[i];
  612. mrp = row[p];
  613. mrq = row[q];
  614. row[p] = cos * mrp - sin * mrq;
  615. row[q] = cos * mrq + sin * mrp;
  616. }
  617. }
  618. }
  619. /**@brief Calculate the matrix cofactor
  620. * @param r1 The first row to use for calculating the cofactor
  621. * @param c1 The first column to use for calculating the cofactor
  622. * @param r1 The second row to use for calculating the cofactor
  623. * @param c1 The second column to use for calculating the cofactor
  624. * See http://en.wikipedia.org/wiki/Cofactor_(linear_algebra) for more details
  625. */
  626. b3Scalar cofac(int r1, int c1, int r2, int c2) const
  627. {
  628. return m_el[r1][c1] * m_el[r2][c2] - m_el[r1][c2] * m_el[r2][c1];
  629. }
  630. void serialize(struct b3Matrix3x3Data& dataOut) const;
  631. void serializeFloat(struct b3Matrix3x3FloatData& dataOut) const;
  632. void deSerialize(const struct b3Matrix3x3Data& dataIn);
  633. void deSerializeFloat(const struct b3Matrix3x3FloatData& dataIn);
  634. void deSerializeDouble(const struct b3Matrix3x3DoubleData& dataIn);
  635. };
  636. B3_FORCE_INLINE b3Matrix3x3&
  637. b3Matrix3x3::operator*=(const b3Matrix3x3& m)
  638. {
  639. #if defined (B3_USE_SSE_IN_API) && defined (B3_USE_SSE)
  640. __m128 rv00, rv01, rv02;
  641. __m128 rv10, rv11, rv12;
  642. __m128 rv20, rv21, rv22;
  643. __m128 mv0, mv1, mv2;
  644. rv02 = m_el[0].mVec128;
  645. rv12 = m_el[1].mVec128;
  646. rv22 = m_el[2].mVec128;
  647. mv0 = _mm_and_ps(m[0].mVec128, b3vFFF0fMask);
  648. mv1 = _mm_and_ps(m[1].mVec128, b3vFFF0fMask);
  649. mv2 = _mm_and_ps(m[2].mVec128, b3vFFF0fMask);
  650. // rv0
  651. rv00 = b3_splat_ps(rv02, 0);
  652. rv01 = b3_splat_ps(rv02, 1);
  653. rv02 = b3_splat_ps(rv02, 2);
  654. rv00 = _mm_mul_ps(rv00, mv0);
  655. rv01 = _mm_mul_ps(rv01, mv1);
  656. rv02 = _mm_mul_ps(rv02, mv2);
  657. // rv1
  658. rv10 = b3_splat_ps(rv12, 0);
  659. rv11 = b3_splat_ps(rv12, 1);
  660. rv12 = b3_splat_ps(rv12, 2);
  661. rv10 = _mm_mul_ps(rv10, mv0);
  662. rv11 = _mm_mul_ps(rv11, mv1);
  663. rv12 = _mm_mul_ps(rv12, mv2);
  664. // rv2
  665. rv20 = b3_splat_ps(rv22, 0);
  666. rv21 = b3_splat_ps(rv22, 1);
  667. rv22 = b3_splat_ps(rv22, 2);
  668. rv20 = _mm_mul_ps(rv20, mv0);
  669. rv21 = _mm_mul_ps(rv21, mv1);
  670. rv22 = _mm_mul_ps(rv22, mv2);
  671. rv00 = _mm_add_ps(rv00, rv01);
  672. rv10 = _mm_add_ps(rv10, rv11);
  673. rv20 = _mm_add_ps(rv20, rv21);
  674. m_el[0].mVec128 = _mm_add_ps(rv00, rv02);
  675. m_el[1].mVec128 = _mm_add_ps(rv10, rv12);
  676. m_el[2].mVec128 = _mm_add_ps(rv20, rv22);
  677. #elif defined(B3_USE_NEON)
  678. float32x4_t rv0, rv1, rv2;
  679. float32x4_t v0, v1, v2;
  680. float32x4_t mv0, mv1, mv2;
  681. v0 = m_el[0].mVec128;
  682. v1 = m_el[1].mVec128;
  683. v2 = m_el[2].mVec128;
  684. mv0 = (float32x4_t) vandq_s32((int32x4_t)m[0].mVec128, b3vFFF0Mask);
  685. mv1 = (float32x4_t) vandq_s32((int32x4_t)m[1].mVec128, b3vFFF0Mask);
  686. mv2 = (float32x4_t) vandq_s32((int32x4_t)m[2].mVec128, b3vFFF0Mask);
  687. rv0 = vmulq_lane_f32(mv0, vget_low_f32(v0), 0);
  688. rv1 = vmulq_lane_f32(mv0, vget_low_f32(v1), 0);
  689. rv2 = vmulq_lane_f32(mv0, vget_low_f32(v2), 0);
  690. rv0 = vmlaq_lane_f32(rv0, mv1, vget_low_f32(v0), 1);
  691. rv1 = vmlaq_lane_f32(rv1, mv1, vget_low_f32(v1), 1);
  692. rv2 = vmlaq_lane_f32(rv2, mv1, vget_low_f32(v2), 1);
  693. rv0 = vmlaq_lane_f32(rv0, mv2, vget_high_f32(v0), 0);
  694. rv1 = vmlaq_lane_f32(rv1, mv2, vget_high_f32(v1), 0);
  695. rv2 = vmlaq_lane_f32(rv2, mv2, vget_high_f32(v2), 0);
  696. m_el[0].mVec128 = rv0;
  697. m_el[1].mVec128 = rv1;
  698. m_el[2].mVec128 = rv2;
  699. #else
  700. setValue(
  701. m.tdotx(m_el[0]), m.tdoty(m_el[0]), m.tdotz(m_el[0]),
  702. m.tdotx(m_el[1]), m.tdoty(m_el[1]), m.tdotz(m_el[1]),
  703. m.tdotx(m_el[2]), m.tdoty(m_el[2]), m.tdotz(m_el[2]));
  704. #endif
  705. return *this;
  706. }
  707. B3_FORCE_INLINE b3Matrix3x3&
  708. b3Matrix3x3::operator+=(const b3Matrix3x3& m)
  709. {
  710. #if (defined (B3_USE_SSE_IN_API) && defined (B3_USE_SSE))|| defined (B3_USE_NEON)
  711. m_el[0].mVec128 = m_el[0].mVec128 + m.m_el[0].mVec128;
  712. m_el[1].mVec128 = m_el[1].mVec128 + m.m_el[1].mVec128;
  713. m_el[2].mVec128 = m_el[2].mVec128 + m.m_el[2].mVec128;
  714. #else
  715. setValue(
  716. m_el[0][0]+m.m_el[0][0],
  717. m_el[0][1]+m.m_el[0][1],
  718. m_el[0][2]+m.m_el[0][2],
  719. m_el[1][0]+m.m_el[1][0],
  720. m_el[1][1]+m.m_el[1][1],
  721. m_el[1][2]+m.m_el[1][2],
  722. m_el[2][0]+m.m_el[2][0],
  723. m_el[2][1]+m.m_el[2][1],
  724. m_el[2][2]+m.m_el[2][2]);
  725. #endif
  726. return *this;
  727. }
  728. B3_FORCE_INLINE b3Matrix3x3
  729. operator*(const b3Matrix3x3& m, const b3Scalar & k)
  730. {
  731. #if (defined (B3_USE_SSE_IN_API) && defined (B3_USE_SSE))
  732. __m128 vk = b3_splat_ps(_mm_load_ss((float *)&k), 0x80);
  733. return b3Matrix3x3(
  734. _mm_mul_ps(m[0].mVec128, vk),
  735. _mm_mul_ps(m[1].mVec128, vk),
  736. _mm_mul_ps(m[2].mVec128, vk));
  737. #elif defined(B3_USE_NEON)
  738. return b3Matrix3x3(
  739. vmulq_n_f32(m[0].mVec128, k),
  740. vmulq_n_f32(m[1].mVec128, k),
  741. vmulq_n_f32(m[2].mVec128, k));
  742. #else
  743. return b3Matrix3x3(
  744. m[0].getX()*k,m[0].getY()*k,m[0].getZ()*k,
  745. m[1].getX()*k,m[1].getY()*k,m[1].getZ()*k,
  746. m[2].getX()*k,m[2].getY()*k,m[2].getZ()*k);
  747. #endif
  748. }
  749. B3_FORCE_INLINE b3Matrix3x3
  750. operator+(const b3Matrix3x3& m1, const b3Matrix3x3& m2)
  751. {
  752. #if (defined (B3_USE_SSE_IN_API) && defined (B3_USE_SSE))|| defined (B3_USE_NEON)
  753. return b3Matrix3x3(
  754. m1[0].mVec128 + m2[0].mVec128,
  755. m1[1].mVec128 + m2[1].mVec128,
  756. m1[2].mVec128 + m2[2].mVec128);
  757. #else
  758. return b3Matrix3x3(
  759. m1[0][0]+m2[0][0],
  760. m1[0][1]+m2[0][1],
  761. m1[0][2]+m2[0][2],
  762. m1[1][0]+m2[1][0],
  763. m1[1][1]+m2[1][1],
  764. m1[1][2]+m2[1][2],
  765. m1[2][0]+m2[2][0],
  766. m1[2][1]+m2[2][1],
  767. m1[2][2]+m2[2][2]);
  768. #endif
  769. }
  770. B3_FORCE_INLINE b3Matrix3x3
  771. operator-(const b3Matrix3x3& m1, const b3Matrix3x3& m2)
  772. {
  773. #if (defined (B3_USE_SSE_IN_API) && defined (B3_USE_SSE))|| defined (B3_USE_NEON)
  774. return b3Matrix3x3(
  775. m1[0].mVec128 - m2[0].mVec128,
  776. m1[1].mVec128 - m2[1].mVec128,
  777. m1[2].mVec128 - m2[2].mVec128);
  778. #else
  779. return b3Matrix3x3(
  780. m1[0][0]-m2[0][0],
  781. m1[0][1]-m2[0][1],
  782. m1[0][2]-m2[0][2],
  783. m1[1][0]-m2[1][0],
  784. m1[1][1]-m2[1][1],
  785. m1[1][2]-m2[1][2],
  786. m1[2][0]-m2[2][0],
  787. m1[2][1]-m2[2][1],
  788. m1[2][2]-m2[2][2]);
  789. #endif
  790. }
  791. B3_FORCE_INLINE b3Matrix3x3&
  792. b3Matrix3x3::operator-=(const b3Matrix3x3& m)
  793. {
  794. #if (defined (B3_USE_SSE_IN_API) && defined (B3_USE_SSE))|| defined (B3_USE_NEON)
  795. m_el[0].mVec128 = m_el[0].mVec128 - m.m_el[0].mVec128;
  796. m_el[1].mVec128 = m_el[1].mVec128 - m.m_el[1].mVec128;
  797. m_el[2].mVec128 = m_el[2].mVec128 - m.m_el[2].mVec128;
  798. #else
  799. setValue(
  800. m_el[0][0]-m.m_el[0][0],
  801. m_el[0][1]-m.m_el[0][1],
  802. m_el[0][2]-m.m_el[0][2],
  803. m_el[1][0]-m.m_el[1][0],
  804. m_el[1][1]-m.m_el[1][1],
  805. m_el[1][2]-m.m_el[1][2],
  806. m_el[2][0]-m.m_el[2][0],
  807. m_el[2][1]-m.m_el[2][1],
  808. m_el[2][2]-m.m_el[2][2]);
  809. #endif
  810. return *this;
  811. }
  812. B3_FORCE_INLINE b3Scalar
  813. b3Matrix3x3::determinant() const
  814. {
  815. return b3Triple((*this)[0], (*this)[1], (*this)[2]);
  816. }
  817. B3_FORCE_INLINE b3Matrix3x3
  818. b3Matrix3x3::absolute() const
  819. {
  820. #if (defined (B3_USE_SSE_IN_API) && defined (B3_USE_SSE))
  821. return b3Matrix3x3(
  822. _mm_and_ps(m_el[0].mVec128, b3vAbsfMask),
  823. _mm_and_ps(m_el[1].mVec128, b3vAbsfMask),
  824. _mm_and_ps(m_el[2].mVec128, b3vAbsfMask));
  825. #elif defined(B3_USE_NEON)
  826. return b3Matrix3x3(
  827. (float32x4_t)vandq_s32((int32x4_t)m_el[0].mVec128, b3v3AbsMask),
  828. (float32x4_t)vandq_s32((int32x4_t)m_el[1].mVec128, b3v3AbsMask),
  829. (float32x4_t)vandq_s32((int32x4_t)m_el[2].mVec128, b3v3AbsMask));
  830. #else
  831. return b3Matrix3x3(
  832. b3Fabs(m_el[0].getX()), b3Fabs(m_el[0].getY()), b3Fabs(m_el[0].getZ()),
  833. b3Fabs(m_el[1].getX()), b3Fabs(m_el[1].getY()), b3Fabs(m_el[1].getZ()),
  834. b3Fabs(m_el[2].getX()), b3Fabs(m_el[2].getY()), b3Fabs(m_el[2].getZ()));
  835. #endif
  836. }
  837. B3_FORCE_INLINE b3Matrix3x3
  838. b3Matrix3x3::transpose() const
  839. {
  840. #if (defined (B3_USE_SSE_IN_API) && defined (B3_USE_SSE))
  841. __m128 v0 = m_el[0].mVec128;
  842. __m128 v1 = m_el[1].mVec128;
  843. __m128 v2 = m_el[2].mVec128; // x2 y2 z2 w2
  844. __m128 vT;
  845. v2 = _mm_and_ps(v2, b3vFFF0fMask); // x2 y2 z2 0
  846. vT = _mm_unpackhi_ps(v0, v1); // z0 z1 * *
  847. v0 = _mm_unpacklo_ps(v0, v1); // x0 x1 y0 y1
  848. v1 = _mm_shuffle_ps(v0, v2, B3_SHUFFLE(2, 3, 1, 3) ); // y0 y1 y2 0
  849. v0 = _mm_shuffle_ps(v0, v2, B3_SHUFFLE(0, 1, 0, 3) ); // x0 x1 x2 0
  850. v2 = b3CastdTo128f(_mm_move_sd(b3CastfTo128d(v2), b3CastfTo128d(vT))); // z0 z1 z2 0
  851. return b3Matrix3x3( v0, v1, v2 );
  852. #elif defined(B3_USE_NEON)
  853. // note: zeros the w channel. We can preserve it at the cost of two more vtrn instructions.
  854. static const uint32x2_t zMask = (const uint32x2_t) {-1, 0 };
  855. float32x4x2_t top = vtrnq_f32( m_el[0].mVec128, m_el[1].mVec128 ); // {x0 x1 z0 z1}, {y0 y1 w0 w1}
  856. float32x2x2_t bl = vtrn_f32( vget_low_f32(m_el[2].mVec128), vdup_n_f32(0.0f) ); // {x2 0 }, {y2 0}
  857. float32x4_t v0 = vcombine_f32( vget_low_f32(top.val[0]), bl.val[0] );
  858. float32x4_t v1 = vcombine_f32( vget_low_f32(top.val[1]), bl.val[1] );
  859. float32x2_t q = (float32x2_t) vand_u32( (uint32x2_t) vget_high_f32( m_el[2].mVec128), zMask );
  860. float32x4_t v2 = vcombine_f32( vget_high_f32(top.val[0]), q ); // z0 z1 z2 0
  861. return b3Matrix3x3( v0, v1, v2 );
  862. #else
  863. return b3Matrix3x3( m_el[0].getX(), m_el[1].getX(), m_el[2].getX(),
  864. m_el[0].getY(), m_el[1].getY(), m_el[2].getY(),
  865. m_el[0].getZ(), m_el[1].getZ(), m_el[2].getZ());
  866. #endif
  867. }
  868. B3_FORCE_INLINE b3Matrix3x3
  869. b3Matrix3x3::adjoint() const
  870. {
  871. return b3Matrix3x3(cofac(1, 1, 2, 2), cofac(0, 2, 2, 1), cofac(0, 1, 1, 2),
  872. cofac(1, 2, 2, 0), cofac(0, 0, 2, 2), cofac(0, 2, 1, 0),
  873. cofac(1, 0, 2, 1), cofac(0, 1, 2, 0), cofac(0, 0, 1, 1));
  874. }
  875. B3_FORCE_INLINE b3Matrix3x3
  876. b3Matrix3x3::inverse() const
  877. {
  878. b3Vector3 co = b3MakeVector3(cofac(1, 1, 2, 2), cofac(1, 2, 2, 0), cofac(1, 0, 2, 1));
  879. b3Scalar det = (*this)[0].dot(co);
  880. b3FullAssert(det != b3Scalar(0.0));
  881. b3Scalar s = b3Scalar(1.0) / det;
  882. return b3Matrix3x3(co.getX() * s, cofac(0, 2, 2, 1) * s, cofac(0, 1, 1, 2) * s,
  883. co.getY() * s, cofac(0, 0, 2, 2) * s, cofac(0, 2, 1, 0) * s,
  884. co.getZ() * s, cofac(0, 1, 2, 0) * s, cofac(0, 0, 1, 1) * s);
  885. }
  886. B3_FORCE_INLINE b3Matrix3x3
  887. b3Matrix3x3::transposeTimes(const b3Matrix3x3& m) const
  888. {
  889. #if (defined (B3_USE_SSE_IN_API) && defined (B3_USE_SSE))
  890. // zeros w
  891. // static const __m128i xyzMask = (const __m128i){ -1ULL, 0xffffffffULL };
  892. __m128 row = m_el[0].mVec128;
  893. __m128 m0 = _mm_and_ps( m.getRow(0).mVec128, b3vFFF0fMask );
  894. __m128 m1 = _mm_and_ps( m.getRow(1).mVec128, b3vFFF0fMask);
  895. __m128 m2 = _mm_and_ps( m.getRow(2).mVec128, b3vFFF0fMask );
  896. __m128 r0 = _mm_mul_ps(m0, _mm_shuffle_ps(row, row, 0));
  897. __m128 r1 = _mm_mul_ps(m0, _mm_shuffle_ps(row, row, 0x55));
  898. __m128 r2 = _mm_mul_ps(m0, _mm_shuffle_ps(row, row, 0xaa));
  899. row = m_el[1].mVec128;
  900. r0 = _mm_add_ps( r0, _mm_mul_ps(m1, _mm_shuffle_ps(row, row, 0)));
  901. r1 = _mm_add_ps( r1, _mm_mul_ps(m1, _mm_shuffle_ps(row, row, 0x55)));
  902. r2 = _mm_add_ps( r2, _mm_mul_ps(m1, _mm_shuffle_ps(row, row, 0xaa)));
  903. row = m_el[2].mVec128;
  904. r0 = _mm_add_ps( r0, _mm_mul_ps(m2, _mm_shuffle_ps(row, row, 0)));
  905. r1 = _mm_add_ps( r1, _mm_mul_ps(m2, _mm_shuffle_ps(row, row, 0x55)));
  906. r2 = _mm_add_ps( r2, _mm_mul_ps(m2, _mm_shuffle_ps(row, row, 0xaa)));
  907. return b3Matrix3x3( r0, r1, r2 );
  908. #elif defined B3_USE_NEON
  909. // zeros w
  910. static const uint32x4_t xyzMask = (const uint32x4_t){ -1, -1, -1, 0 };
  911. float32x4_t m0 = (float32x4_t) vandq_u32( (uint32x4_t) m.getRow(0).mVec128, xyzMask );
  912. float32x4_t m1 = (float32x4_t) vandq_u32( (uint32x4_t) m.getRow(1).mVec128, xyzMask );
  913. float32x4_t m2 = (float32x4_t) vandq_u32( (uint32x4_t) m.getRow(2).mVec128, xyzMask );
  914. float32x4_t row = m_el[0].mVec128;
  915. float32x4_t r0 = vmulq_lane_f32( m0, vget_low_f32(row), 0);
  916. float32x4_t r1 = vmulq_lane_f32( m0, vget_low_f32(row), 1);
  917. float32x4_t r2 = vmulq_lane_f32( m0, vget_high_f32(row), 0);
  918. row = m_el[1].mVec128;
  919. r0 = vmlaq_lane_f32( r0, m1, vget_low_f32(row), 0);
  920. r1 = vmlaq_lane_f32( r1, m1, vget_low_f32(row), 1);
  921. r2 = vmlaq_lane_f32( r2, m1, vget_high_f32(row), 0);
  922. row = m_el[2].mVec128;
  923. r0 = vmlaq_lane_f32( r0, m2, vget_low_f32(row), 0);
  924. r1 = vmlaq_lane_f32( r1, m2, vget_low_f32(row), 1);
  925. r2 = vmlaq_lane_f32( r2, m2, vget_high_f32(row), 0);
  926. return b3Matrix3x3( r0, r1, r2 );
  927. #else
  928. return b3Matrix3x3(
  929. m_el[0].getX() * m[0].getX() + m_el[1].getX() * m[1].getX() + m_el[2].getX() * m[2].getX(),
  930. m_el[0].getX() * m[0].getY() + m_el[1].getX() * m[1].getY() + m_el[2].getX() * m[2].getY(),
  931. m_el[0].getX() * m[0].getZ() + m_el[1].getX() * m[1].getZ() + m_el[2].getX() * m[2].getZ(),
  932. m_el[0].getY() * m[0].getX() + m_el[1].getY() * m[1].getX() + m_el[2].getY() * m[2].getX(),
  933. m_el[0].getY() * m[0].getY() + m_el[1].getY() * m[1].getY() + m_el[2].getY() * m[2].getY(),
  934. m_el[0].getY() * m[0].getZ() + m_el[1].getY() * m[1].getZ() + m_el[2].getY() * m[2].getZ(),
  935. m_el[0].getZ() * m[0].getX() + m_el[1].getZ() * m[1].getX() + m_el[2].getZ() * m[2].getX(),
  936. m_el[0].getZ() * m[0].getY() + m_el[1].getZ() * m[1].getY() + m_el[2].getZ() * m[2].getY(),
  937. m_el[0].getZ() * m[0].getZ() + m_el[1].getZ() * m[1].getZ() + m_el[2].getZ() * m[2].getZ());
  938. #endif
  939. }
  940. B3_FORCE_INLINE b3Matrix3x3
  941. b3Matrix3x3::timesTranspose(const b3Matrix3x3& m) const
  942. {
  943. #if (defined (B3_USE_SSE_IN_API) && defined (B3_USE_SSE))
  944. __m128 a0 = m_el[0].mVec128;
  945. __m128 a1 = m_el[1].mVec128;
  946. __m128 a2 = m_el[2].mVec128;
  947. b3Matrix3x3 mT = m.transpose(); // we rely on transpose() zeroing w channel so that we don't have to do it here
  948. __m128 mx = mT[0].mVec128;
  949. __m128 my = mT[1].mVec128;
  950. __m128 mz = mT[2].mVec128;
  951. __m128 r0 = _mm_mul_ps(mx, _mm_shuffle_ps(a0, a0, 0x00));
  952. __m128 r1 = _mm_mul_ps(mx, _mm_shuffle_ps(a1, a1, 0x00));
  953. __m128 r2 = _mm_mul_ps(mx, _mm_shuffle_ps(a2, a2, 0x00));
  954. r0 = _mm_add_ps(r0, _mm_mul_ps(my, _mm_shuffle_ps(a0, a0, 0x55)));
  955. r1 = _mm_add_ps(r1, _mm_mul_ps(my, _mm_shuffle_ps(a1, a1, 0x55)));
  956. r2 = _mm_add_ps(r2, _mm_mul_ps(my, _mm_shuffle_ps(a2, a2, 0x55)));
  957. r0 = _mm_add_ps(r0, _mm_mul_ps(mz, _mm_shuffle_ps(a0, a0, 0xaa)));
  958. r1 = _mm_add_ps(r1, _mm_mul_ps(mz, _mm_shuffle_ps(a1, a1, 0xaa)));
  959. r2 = _mm_add_ps(r2, _mm_mul_ps(mz, _mm_shuffle_ps(a2, a2, 0xaa)));
  960. return b3Matrix3x3( r0, r1, r2);
  961. #elif defined B3_USE_NEON
  962. float32x4_t a0 = m_el[0].mVec128;
  963. float32x4_t a1 = m_el[1].mVec128;
  964. float32x4_t a2 = m_el[2].mVec128;
  965. b3Matrix3x3 mT = m.transpose(); // we rely on transpose() zeroing w channel so that we don't have to do it here
  966. float32x4_t mx = mT[0].mVec128;
  967. float32x4_t my = mT[1].mVec128;
  968. float32x4_t mz = mT[2].mVec128;
  969. float32x4_t r0 = vmulq_lane_f32( mx, vget_low_f32(a0), 0);
  970. float32x4_t r1 = vmulq_lane_f32( mx, vget_low_f32(a1), 0);
  971. float32x4_t r2 = vmulq_lane_f32( mx, vget_low_f32(a2), 0);
  972. r0 = vmlaq_lane_f32( r0, my, vget_low_f32(a0), 1);
  973. r1 = vmlaq_lane_f32( r1, my, vget_low_f32(a1), 1);
  974. r2 = vmlaq_lane_f32( r2, my, vget_low_f32(a2), 1);
  975. r0 = vmlaq_lane_f32( r0, mz, vget_high_f32(a0), 0);
  976. r1 = vmlaq_lane_f32( r1, mz, vget_high_f32(a1), 0);
  977. r2 = vmlaq_lane_f32( r2, mz, vget_high_f32(a2), 0);
  978. return b3Matrix3x3( r0, r1, r2 );
  979. #else
  980. return b3Matrix3x3(
  981. m_el[0].dot(m[0]), m_el[0].dot(m[1]), m_el[0].dot(m[2]),
  982. m_el[1].dot(m[0]), m_el[1].dot(m[1]), m_el[1].dot(m[2]),
  983. m_el[2].dot(m[0]), m_el[2].dot(m[1]), m_el[2].dot(m[2]));
  984. #endif
  985. }
  986. B3_FORCE_INLINE b3Vector3
  987. operator*(const b3Matrix3x3& m, const b3Vector3& v)
  988. {
  989. #if (defined (B3_USE_SSE_IN_API) && defined (B3_USE_SSE))|| defined (B3_USE_NEON)
  990. return v.dot3(m[0], m[1], m[2]);
  991. #else
  992. return b3MakeVector3(m[0].dot(v), m[1].dot(v), m[2].dot(v));
  993. #endif
  994. }
  995. B3_FORCE_INLINE b3Vector3
  996. operator*(const b3Vector3& v, const b3Matrix3x3& m)
  997. {
  998. #if (defined (B3_USE_SSE_IN_API) && defined (B3_USE_SSE))
  999. const __m128 vv = v.mVec128;
  1000. __m128 c0 = b3_splat_ps( vv, 0);
  1001. __m128 c1 = b3_splat_ps( vv, 1);
  1002. __m128 c2 = b3_splat_ps( vv, 2);
  1003. c0 = _mm_mul_ps(c0, _mm_and_ps(m[0].mVec128, b3vFFF0fMask) );
  1004. c1 = _mm_mul_ps(c1, _mm_and_ps(m[1].mVec128, b3vFFF0fMask) );
  1005. c0 = _mm_add_ps(c0, c1);
  1006. c2 = _mm_mul_ps(c2, _mm_and_ps(m[2].mVec128, b3vFFF0fMask) );
  1007. return b3MakeVector3(_mm_add_ps(c0, c2));
  1008. #elif defined(B3_USE_NEON)
  1009. const float32x4_t vv = v.mVec128;
  1010. const float32x2_t vlo = vget_low_f32(vv);
  1011. const float32x2_t vhi = vget_high_f32(vv);
  1012. float32x4_t c0, c1, c2;
  1013. c0 = (float32x4_t) vandq_s32((int32x4_t)m[0].mVec128, b3vFFF0Mask);
  1014. c1 = (float32x4_t) vandq_s32((int32x4_t)m[1].mVec128, b3vFFF0Mask);
  1015. c2 = (float32x4_t) vandq_s32((int32x4_t)m[2].mVec128, b3vFFF0Mask);
  1016. c0 = vmulq_lane_f32(c0, vlo, 0);
  1017. c1 = vmulq_lane_f32(c1, vlo, 1);
  1018. c2 = vmulq_lane_f32(c2, vhi, 0);
  1019. c0 = vaddq_f32(c0, c1);
  1020. c0 = vaddq_f32(c0, c2);
  1021. return b3MakeVector3(c0);
  1022. #else
  1023. return b3MakeVector3(m.tdotx(v), m.tdoty(v), m.tdotz(v));
  1024. #endif
  1025. }
  1026. B3_FORCE_INLINE b3Matrix3x3
  1027. operator*(const b3Matrix3x3& m1, const b3Matrix3x3& m2)
  1028. {
  1029. #if (defined (B3_USE_SSE_IN_API) && defined (B3_USE_SSE))
  1030. __m128 m10 = m1[0].mVec128;
  1031. __m128 m11 = m1[1].mVec128;
  1032. __m128 m12 = m1[2].mVec128;
  1033. __m128 m2v = _mm_and_ps(m2[0].mVec128, b3vFFF0fMask);
  1034. __m128 c0 = b3_splat_ps( m10, 0);
  1035. __m128 c1 = b3_splat_ps( m11, 0);
  1036. __m128 c2 = b3_splat_ps( m12, 0);
  1037. c0 = _mm_mul_ps(c0, m2v);
  1038. c1 = _mm_mul_ps(c1, m2v);
  1039. c2 = _mm_mul_ps(c2, m2v);
  1040. m2v = _mm_and_ps(m2[1].mVec128, b3vFFF0fMask);
  1041. __m128 c0_1 = b3_splat_ps( m10, 1);
  1042. __m128 c1_1 = b3_splat_ps( m11, 1);
  1043. __m128 c2_1 = b3_splat_ps( m12, 1);
  1044. c0_1 = _mm_mul_ps(c0_1, m2v);
  1045. c1_1 = _mm_mul_ps(c1_1, m2v);
  1046. c2_1 = _mm_mul_ps(c2_1, m2v);
  1047. m2v = _mm_and_ps(m2[2].mVec128, b3vFFF0fMask);
  1048. c0 = _mm_add_ps(c0, c0_1);
  1049. c1 = _mm_add_ps(c1, c1_1);
  1050. c2 = _mm_add_ps(c2, c2_1);
  1051. m10 = b3_splat_ps( m10, 2);
  1052. m11 = b3_splat_ps( m11, 2);
  1053. m12 = b3_splat_ps( m12, 2);
  1054. m10 = _mm_mul_ps(m10, m2v);
  1055. m11 = _mm_mul_ps(m11, m2v);
  1056. m12 = _mm_mul_ps(m12, m2v);
  1057. c0 = _mm_add_ps(c0, m10);
  1058. c1 = _mm_add_ps(c1, m11);
  1059. c2 = _mm_add_ps(c2, m12);
  1060. return b3Matrix3x3(c0, c1, c2);
  1061. #elif defined(B3_USE_NEON)
  1062. float32x4_t rv0, rv1, rv2;
  1063. float32x4_t v0, v1, v2;
  1064. float32x4_t mv0, mv1, mv2;
  1065. v0 = m1[0].mVec128;
  1066. v1 = m1[1].mVec128;
  1067. v2 = m1[2].mVec128;
  1068. mv0 = (float32x4_t) vandq_s32((int32x4_t)m2[0].mVec128, b3vFFF0Mask);
  1069. mv1 = (float32x4_t) vandq_s32((int32x4_t)m2[1].mVec128, b3vFFF0Mask);
  1070. mv2 = (float32x4_t) vandq_s32((int32x4_t)m2[2].mVec128, b3vFFF0Mask);
  1071. rv0 = vmulq_lane_f32(mv0, vget_low_f32(v0), 0);
  1072. rv1 = vmulq_lane_f32(mv0, vget_low_f32(v1), 0);
  1073. rv2 = vmulq_lane_f32(mv0, vget_low_f32(v2), 0);
  1074. rv0 = vmlaq_lane_f32(rv0, mv1, vget_low_f32(v0), 1);
  1075. rv1 = vmlaq_lane_f32(rv1, mv1, vget_low_f32(v1), 1);
  1076. rv2 = vmlaq_lane_f32(rv2, mv1, vget_low_f32(v2), 1);
  1077. rv0 = vmlaq_lane_f32(rv0, mv2, vget_high_f32(v0), 0);
  1078. rv1 = vmlaq_lane_f32(rv1, mv2, vget_high_f32(v1), 0);
  1079. rv2 = vmlaq_lane_f32(rv2, mv2, vget_high_f32(v2), 0);
  1080. return b3Matrix3x3(rv0, rv1, rv2);
  1081. #else
  1082. return b3Matrix3x3(
  1083. m2.tdotx( m1[0]), m2.tdoty( m1[0]), m2.tdotz( m1[0]),
  1084. m2.tdotx( m1[1]), m2.tdoty( m1[1]), m2.tdotz( m1[1]),
  1085. m2.tdotx( m1[2]), m2.tdoty( m1[2]), m2.tdotz( m1[2]));
  1086. #endif
  1087. }
  1088. /*
  1089. B3_FORCE_INLINE b3Matrix3x3 b3MultTransposeLeft(const b3Matrix3x3& m1, const b3Matrix3x3& m2) {
  1090. return b3Matrix3x3(
  1091. m1[0][0] * m2[0][0] + m1[1][0] * m2[1][0] + m1[2][0] * m2[2][0],
  1092. m1[0][0] * m2[0][1] + m1[1][0] * m2[1][1] + m1[2][0] * m2[2][1],
  1093. m1[0][0] * m2[0][2] + m1[1][0] * m2[1][2] + m1[2][0] * m2[2][2],
  1094. m1[0][1] * m2[0][0] + m1[1][1] * m2[1][0] + m1[2][1] * m2[2][0],
  1095. m1[0][1] * m2[0][1] + m1[1][1] * m2[1][1] + m1[2][1] * m2[2][1],
  1096. m1[0][1] * m2[0][2] + m1[1][1] * m2[1][2] + m1[2][1] * m2[2][2],
  1097. m1[0][2] * m2[0][0] + m1[1][2] * m2[1][0] + m1[2][2] * m2[2][0],
  1098. m1[0][2] * m2[0][1] + m1[1][2] * m2[1][1] + m1[2][2] * m2[2][1],
  1099. m1[0][2] * m2[0][2] + m1[1][2] * m2[1][2] + m1[2][2] * m2[2][2]);
  1100. }
  1101. */
  1102. /**@brief Equality operator between two matrices
  1103. * It will test all elements are equal. */
  1104. B3_FORCE_INLINE bool operator==(const b3Matrix3x3& m1, const b3Matrix3x3& m2)
  1105. {
  1106. #if (defined (B3_USE_SSE_IN_API) && defined (B3_USE_SSE))
  1107. __m128 c0, c1, c2;
  1108. c0 = _mm_cmpeq_ps(m1[0].mVec128, m2[0].mVec128);
  1109. c1 = _mm_cmpeq_ps(m1[1].mVec128, m2[1].mVec128);
  1110. c2 = _mm_cmpeq_ps(m1[2].mVec128, m2[2].mVec128);
  1111. c0 = _mm_and_ps(c0, c1);
  1112. c0 = _mm_and_ps(c0, c2);
  1113. return (0x7 == _mm_movemask_ps((__m128)c0));
  1114. #else
  1115. return
  1116. ( m1[0][0] == m2[0][0] && m1[1][0] == m2[1][0] && m1[2][0] == m2[2][0] &&
  1117. m1[0][1] == m2[0][1] && m1[1][1] == m2[1][1] && m1[2][1] == m2[2][1] &&
  1118. m1[0][2] == m2[0][2] && m1[1][2] == m2[1][2] && m1[2][2] == m2[2][2] );
  1119. #endif
  1120. }
  1121. ///for serialization
  1122. struct b3Matrix3x3FloatData
  1123. {
  1124. b3Vector3FloatData m_el[3];
  1125. };
  1126. ///for serialization
  1127. struct b3Matrix3x3DoubleData
  1128. {
  1129. b3Vector3DoubleData m_el[3];
  1130. };
  1131. B3_FORCE_INLINE void b3Matrix3x3::serialize(struct b3Matrix3x3Data& dataOut) const
  1132. {
  1133. for (int i=0;i<3;i++)
  1134. m_el[i].serialize(dataOut.m_el[i]);
  1135. }
  1136. B3_FORCE_INLINE void b3Matrix3x3::serializeFloat(struct b3Matrix3x3FloatData& dataOut) const
  1137. {
  1138. for (int i=0;i<3;i++)
  1139. m_el[i].serializeFloat(dataOut.m_el[i]);
  1140. }
  1141. B3_FORCE_INLINE void b3Matrix3x3::deSerialize(const struct b3Matrix3x3Data& dataIn)
  1142. {
  1143. for (int i=0;i<3;i++)
  1144. m_el[i].deSerialize(dataIn.m_el[i]);
  1145. }
  1146. B3_FORCE_INLINE void b3Matrix3x3::deSerializeFloat(const struct b3Matrix3x3FloatData& dataIn)
  1147. {
  1148. for (int i=0;i<3;i++)
  1149. m_el[i].deSerializeFloat(dataIn.m_el[i]);
  1150. }
  1151. B3_FORCE_INLINE void b3Matrix3x3::deSerializeDouble(const struct b3Matrix3x3DoubleData& dataIn)
  1152. {
  1153. for (int i=0;i<3;i++)
  1154. m_el[i].deSerializeDouble(dataIn.m_el[i]);
  1155. }
  1156. #endif //B3_MATRIX3x3_H