Quaternion.h 31.4 KB
Newer Older
LM's avatar
LM committed
1 2 3 4 5 6
// This file is part of Eigen, a lightweight C++ template library
// for linear algebra.
//
// Copyright (C) 2008-2010 Gael Guennebaud <gael.guennebaud@inria.fr>
// Copyright (C) 2009 Mathieu Gautier <mathieu.gautier@cea.fr>
//
Don Gagne's avatar
Don Gagne committed
7 8 9
// This Source Code Form is subject to the terms of the Mozilla
// Public License v. 2.0. If a copy of the MPL was not distributed
// with this file, You can obtain one at http://mozilla.org/MPL/2.0/.
LM's avatar
LM committed
10 11 12

#ifndef EIGEN_QUATERNION_H
#define EIGEN_QUATERNION_H
Don Gagne's avatar
Don Gagne committed
13 14
namespace Eigen { 

LM's avatar
LM committed
15 16 17 18 19 20 21 22 23 24 25 26 27

/***************************************************************************
* Definition of QuaternionBase<Derived>
* The implementation is at the end of the file
***************************************************************************/

namespace internal {
template<typename Other,
         int OtherRows=Other::RowsAtCompileTime,
         int OtherCols=Other::ColsAtCompileTime>
struct quaternionbase_assign_impl;
}

Don Gagne's avatar
Don Gagne committed
28 29 30 31 32 33
/** \geometry_module \ingroup Geometry_Module
  * \class QuaternionBase
  * \brief Base class for quaternion expressions
  * \tparam Derived derived type (CRTP)
  * \sa class Quaternion
  */
LM's avatar
LM committed
34 35 36
template<class Derived>
class QuaternionBase : public RotationBase<Derived, 3>
{
37
 public:
LM's avatar
LM committed
38
  typedef RotationBase<Derived, 3> Base;
39

LM's avatar
LM committed
40 41 42 43 44 45
  using Base::operator*;
  using Base::derived;

  typedef typename internal::traits<Derived>::Scalar Scalar;
  typedef typename NumTraits<Scalar>::Real RealScalar;
  typedef typename internal::traits<Derived>::Coefficients Coefficients;
46 47 48 49 50
  typedef typename Coefficients::CoeffReturnType CoeffReturnType;
  typedef typename internal::conditional<bool(internal::traits<Derived>::Flags&LvalueBit),
                                        Scalar&, CoeffReturnType>::type NonConstCoeffReturnType;


LM's avatar
LM committed
51 52 53 54 55 56 57 58 59 60 61 62 63 64 65
  enum {
    Flags = Eigen::internal::traits<Derived>::Flags
  };

 // typedef typename Matrix<Scalar,4,1> Coefficients;
  /** the type of a 3D vector */
  typedef Matrix<Scalar,3,1> Vector3;
  /** the equivalent rotation matrix type */
  typedef Matrix<Scalar,3,3> Matrix3;
  /** the equivalent angle-axis type */
  typedef AngleAxis<Scalar> AngleAxisType;



  /** \returns the \c x coefficient */
66
  EIGEN_DEVICE_FUNC inline CoeffReturnType x() const { return this->derived().coeffs().coeff(0); }
LM's avatar
LM committed
67
  /** \returns the \c y coefficient */
68
  EIGEN_DEVICE_FUNC inline CoeffReturnType y() const { return this->derived().coeffs().coeff(1); }
LM's avatar
LM committed
69
  /** \returns the \c z coefficient */
70
  EIGEN_DEVICE_FUNC inline CoeffReturnType z() const { return this->derived().coeffs().coeff(2); }
LM's avatar
LM committed
71
  /** \returns the \c w coefficient */
72
  EIGEN_DEVICE_FUNC inline CoeffReturnType w() const { return this->derived().coeffs().coeff(3); }
LM's avatar
LM committed
73

74 75 76 77 78 79 80 81
  /** \returns a reference to the \c x coefficient (if Derived is a non-const lvalue) */
  EIGEN_DEVICE_FUNC inline NonConstCoeffReturnType x() { return this->derived().coeffs().x(); }
  /** \returns a reference to the \c y coefficient (if Derived is a non-const lvalue) */
  EIGEN_DEVICE_FUNC inline NonConstCoeffReturnType y() { return this->derived().coeffs().y(); }
  /** \returns a reference to the \c z coefficient (if Derived is a non-const lvalue) */
  EIGEN_DEVICE_FUNC inline NonConstCoeffReturnType z() { return this->derived().coeffs().z(); }
  /** \returns a reference to the \c w coefficient (if Derived is a non-const lvalue) */
  EIGEN_DEVICE_FUNC inline NonConstCoeffReturnType w() { return this->derived().coeffs().w(); }
LM's avatar
LM committed
82 83

  /** \returns a read-only vector expression of the imaginary part (x,y,z) */
84
  EIGEN_DEVICE_FUNC inline const VectorBlock<const Coefficients,3> vec() const { return coeffs().template head<3>(); }
LM's avatar
LM committed
85 86

  /** \returns a vector expression of the imaginary part (x,y,z) */
87
  EIGEN_DEVICE_FUNC inline VectorBlock<Coefficients,3> vec() { return coeffs().template head<3>(); }
LM's avatar
LM committed
88 89

  /** \returns a read-only vector expression of the coefficients (x,y,z,w) */
90
  EIGEN_DEVICE_FUNC inline const typename internal::traits<Derived>::Coefficients& coeffs() const { return derived().coeffs(); }
LM's avatar
LM committed
91 92

  /** \returns a vector expression of the coefficients (x,y,z,w) */
93
  EIGEN_DEVICE_FUNC inline typename internal::traits<Derived>::Coefficients& coeffs() { return derived().coeffs(); }
LM's avatar
LM committed
94

95 96
  EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE QuaternionBase<Derived>& operator=(const QuaternionBase<Derived>& other);
  template<class OtherDerived> EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE Derived& operator=(const QuaternionBase<OtherDerived>& other);
LM's avatar
LM committed
97 98 99 100 101 102 103 104

// disabled this copy operator as it is giving very strange compilation errors when compiling
// test_stdvector with GCC 4.4.2. This looks like a GCC bug though, so feel free to re-enable it if it's
// useful; however notice that we already have the templated operator= above and e.g. in MatrixBase
// we didn't have to add, in addition to templated operator=, such a non-templated copy operator.
//  Derived& operator=(const QuaternionBase& other)
//  { return operator=<Derived>(other); }

105 106
  EIGEN_DEVICE_FUNC Derived& operator=(const AngleAxisType& aa);
  template<class OtherDerived> EIGEN_DEVICE_FUNC Derived& operator=(const MatrixBase<OtherDerived>& m);
LM's avatar
LM committed
107 108 109 110

  /** \returns a quaternion representing an identity rotation
    * \sa MatrixBase::Identity()
    */
111
  EIGEN_DEVICE_FUNC static inline Quaternion<Scalar> Identity() { return Quaternion<Scalar>(Scalar(1), Scalar(0), Scalar(0), Scalar(0)); }
LM's avatar
LM committed
112 113 114

  /** \sa QuaternionBase::Identity(), MatrixBase::setIdentity()
    */
115
  EIGEN_DEVICE_FUNC inline QuaternionBase& setIdentity() { coeffs() << Scalar(0), Scalar(0), Scalar(0), Scalar(1); return *this; }
LM's avatar
LM committed
116 117 118 119

  /** \returns the squared norm of the quaternion's coefficients
    * \sa QuaternionBase::norm(), MatrixBase::squaredNorm()
    */
120
  EIGEN_DEVICE_FUNC inline Scalar squaredNorm() const { return coeffs().squaredNorm(); }
LM's avatar
LM committed
121 122 123 124

  /** \returns the norm of the quaternion's coefficients
    * \sa QuaternionBase::squaredNorm(), MatrixBase::norm()
    */
125
  EIGEN_DEVICE_FUNC inline Scalar norm() const { return coeffs().norm(); }
LM's avatar
LM committed
126 127 128

  /** Normalizes the quaternion \c *this
    * \sa normalized(), MatrixBase::normalize() */
129
  EIGEN_DEVICE_FUNC inline void normalize() { coeffs().normalize(); }
LM's avatar
LM committed
130 131
  /** \returns a normalized copy of \c *this
    * \sa normalize(), MatrixBase::normalized() */
132
  EIGEN_DEVICE_FUNC inline Quaternion<Scalar> normalized() const { return Quaternion<Scalar>(coeffs().normalized()); }
LM's avatar
LM committed
133 134 135 136 137 138

    /** \returns the dot product of \c *this and \a other
    * Geometrically speaking, the dot product of two unit quaternions
    * corresponds to the cosine of half the angle between the two rotations.
    * \sa angularDistance()
    */
139
  template<class OtherDerived> EIGEN_DEVICE_FUNC inline Scalar dot(const QuaternionBase<OtherDerived>& other) const { return coeffs().dot(other.coeffs()); }
LM's avatar
LM committed
140

141
  template<class OtherDerived> EIGEN_DEVICE_FUNC Scalar angularDistance(const QuaternionBase<OtherDerived>& other) const;
LM's avatar
LM committed
142 143

  /** \returns an equivalent 3x3 rotation matrix */
144
  EIGEN_DEVICE_FUNC Matrix3 toRotationMatrix() const;
LM's avatar
LM committed
145 146 147

  /** \returns the quaternion which transform \a a into \a b through a rotation */
  template<typename Derived1, typename Derived2>
148
  EIGEN_DEVICE_FUNC Derived& setFromTwoVectors(const MatrixBase<Derived1>& a, const MatrixBase<Derived2>& b);
LM's avatar
LM committed
149

150 151
  template<class OtherDerived> EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE Quaternion<Scalar> operator* (const QuaternionBase<OtherDerived>& q) const;
  template<class OtherDerived> EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE Derived& operator*= (const QuaternionBase<OtherDerived>& q);
LM's avatar
LM committed
152 153

  /** \returns the quaternion describing the inverse rotation */
154
  EIGEN_DEVICE_FUNC Quaternion<Scalar> inverse() const;
LM's avatar
LM committed
155 156

  /** \returns the conjugated quaternion */
157
  EIGEN_DEVICE_FUNC Quaternion<Scalar> conjugate() const;
LM's avatar
LM committed
158

159
  template<class OtherDerived> EIGEN_DEVICE_FUNC Quaternion<Scalar> slerp(const Scalar& t, const QuaternionBase<OtherDerived>& other) const;
LM's avatar
LM committed
160 161 162 163 164 165

  /** \returns \c true if \c *this is approximately equal to \a other, within the precision
    * determined by \a prec.
    *
    * \sa MatrixBase::isApprox() */
  template<class OtherDerived>
166
  EIGEN_DEVICE_FUNC bool isApprox(const QuaternionBase<OtherDerived>& other, const RealScalar& prec = NumTraits<Scalar>::dummy_precision()) const
LM's avatar
LM committed
167 168
  { return coeffs().isApprox(other.coeffs(), prec); }

169 170
  /** return the result vector of \a v through the rotation*/
  EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE Vector3 _transformVector(const Vector3& v) const;
LM's avatar
LM committed
171 172 173 174 175 176 177

  /** \returns \c *this with scalar type casted to \a NewScalarType
    *
    * Note that if \a NewScalarType is equal to the current scalar type of \c *this
    * then this function smartly returns a const reference to \c *this.
    */
  template<typename NewScalarType>
178
  EIGEN_DEVICE_FUNC inline typename internal::cast_return_type<Derived,Quaternion<NewScalarType> >::type cast() const
LM's avatar
LM committed
179
  {
Don Gagne's avatar
Don Gagne committed
180
    return typename internal::cast_return_type<Derived,Quaternion<NewScalarType> >::type(derived());
LM's avatar
LM committed
181
  }
Don Gagne's avatar
Don Gagne committed
182

LM's avatar
LM committed
183 184 185 186 187 188 189 190 191 192 193 194 195 196 197
#ifdef EIGEN_QUATERNIONBASE_PLUGIN
# include EIGEN_QUATERNIONBASE_PLUGIN
#endif
};

/***************************************************************************
* Definition/implementation of Quaternion<Scalar>
***************************************************************************/

/** \geometry_module \ingroup Geometry_Module
  *
  * \class Quaternion
  *
  * \brief The quaternion class used to represent 3D orientations and rotations
  *
Don Gagne's avatar
Don Gagne committed
198
  * \tparam _Scalar the scalar type, i.e., the type of the coefficients
199
  * \tparam _Options controls the memory alignment of the coefficients. Can be \# AutoAlign or \# DontAlign. Default is AutoAlign.
LM's avatar
LM committed
200 201 202
  *
  * This class represents a quaternion \f$ w+xi+yj+zk \f$ that is a convenient representation of
  * orientations and rotations of objects in three dimensions. Compared to other representations
203
  * like Euler angles or 3x3 matrices, quaternions offer the following advantages:
LM's avatar
LM committed
204 205 206 207 208 209 210 211
  * \li \b compact storage (4 scalars)
  * \li \b efficient to compose (28 flops),
  * \li \b stable spherical interpolation
  *
  * The following two typedefs are provided for convenience:
  * \li \c Quaternionf for \c float
  * \li \c Quaterniond for \c double
  *
212 213
  * \warning Operations interpreting the quaternion as rotation have undefined behavior if the quaternion is not normalized.
  *
LM's avatar
LM committed
214 215 216 217 218 219 220 221 222 223 224
  * \sa  class AngleAxis, class Transform
  */

namespace internal {
template<typename _Scalar,int _Options>
struct traits<Quaternion<_Scalar,_Options> >
{
  typedef Quaternion<_Scalar,_Options> PlainObject;
  typedef _Scalar Scalar;
  typedef Matrix<_Scalar,4,1,_Options> Coefficients;
  enum{
225 226
    Alignment = internal::traits<Coefficients>::Alignment,
    Flags = LvalueBit
LM's avatar
LM committed
227 228 229 230 231
  };
};
}

template<typename _Scalar, int _Options>
Don Gagne's avatar
Don Gagne committed
232 233
class Quaternion : public QuaternionBase<Quaternion<_Scalar,_Options> >
{
234
public:
LM's avatar
LM committed
235
  typedef QuaternionBase<Quaternion<_Scalar,_Options> > Base;
236
  enum { NeedsAlignment = internal::traits<Quaternion>::Alignment>0 };
Don Gagne's avatar
Don Gagne committed
237

LM's avatar
LM committed
238 239
  typedef _Scalar Scalar;

240
  EIGEN_INHERIT_ASSIGNMENT_OPERATORS(Quaternion)
LM's avatar
LM committed
241 242
  using Base::operator*=;

Don Gagne's avatar
Don Gagne committed
243
  typedef typename internal::traits<Quaternion>::Coefficients Coefficients;
LM's avatar
LM committed
244 245 246
  typedef typename Base::AngleAxisType AngleAxisType;

  /** Default constructor leaving the quaternion uninitialized. */
247
  EIGEN_DEVICE_FUNC inline Quaternion() {}
LM's avatar
LM committed
248 249 250 251 252 253 254 255

  /** Constructs and initializes the quaternion \f$ w+xi+yj+zk \f$ from
    * its four coefficients \a w, \a x, \a y and \a z.
    *
    * \warning Note the order of the arguments: the real \a w coefficient first,
    * while internally the coefficients are stored in the following order:
    * [\c x, \c y, \c z, \c w]
    */
256
  EIGEN_DEVICE_FUNC inline Quaternion(const Scalar& w, const Scalar& x, const Scalar& y, const Scalar& z) : m_coeffs(x, y, z, w){}
LM's avatar
LM committed
257 258

  /** Constructs and initialize a quaternion from the array data */
259
  EIGEN_DEVICE_FUNC explicit inline Quaternion(const Scalar* data) : m_coeffs(data) {}
LM's avatar
LM committed
260 261

  /** Copy constructor */
262
  template<class Derived> EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE Quaternion(const QuaternionBase<Derived>& other) { this->Base::operator=(other); }
LM's avatar
LM committed
263 264

  /** Constructs and initializes a quaternion from the angle-axis \a aa */
265
  EIGEN_DEVICE_FUNC explicit inline Quaternion(const AngleAxisType& aa) { *this = aa; }
LM's avatar
LM committed
266 267 268 269 270 271

  /** Constructs and initializes a quaternion from either:
    *  - a rotation matrix expression,
    *  - a 4D vector expression representing quaternion coefficients.
    */
  template<typename Derived>
272
  EIGEN_DEVICE_FUNC explicit inline Quaternion(const MatrixBase<Derived>& other) { *this = other; }
LM's avatar
LM committed
273

Don Gagne's avatar
Don Gagne committed
274 275
  /** Explicit copy constructor with scalar conversion */
  template<typename OtherScalar, int OtherOptions>
276
  EIGEN_DEVICE_FUNC explicit inline Quaternion(const Quaternion<OtherScalar, OtherOptions>& other)
Don Gagne's avatar
Don Gagne committed
277 278
  { m_coeffs = other.coeffs().template cast<Scalar>(); }

279 280
  EIGEN_DEVICE_FUNC static Quaternion UnitRandom();

Don Gagne's avatar
Don Gagne committed
281
  template<typename Derived1, typename Derived2>
282
  EIGEN_DEVICE_FUNC static Quaternion FromTwoVectors(const MatrixBase<Derived1>& a, const MatrixBase<Derived2>& b);
Don Gagne's avatar
Don Gagne committed
283

284 285
  EIGEN_DEVICE_FUNC inline Coefficients& coeffs() { return m_coeffs;}
  EIGEN_DEVICE_FUNC inline const Coefficients& coeffs() const { return m_coeffs;}
LM's avatar
LM committed
286

287 288 289 290 291
  EIGEN_MAKE_ALIGNED_OPERATOR_NEW_IF(bool(NeedsAlignment))
  
#ifdef EIGEN_QUATERNION_PLUGIN
# include EIGEN_QUATERNION_PLUGIN
#endif
Don Gagne's avatar
Don Gagne committed
292

LM's avatar
LM committed
293 294 295 296
protected:
  Coefficients m_coeffs;
  
#ifndef EIGEN_PARSED_BY_DOXYGEN
Don Gagne's avatar
Don Gagne committed
297
    static EIGEN_STRONG_INLINE void _check_template_params()
LM's avatar
LM committed
298 299 300 301 302 303 304 305 306 307 308 309 310 311 312 313 314 315 316 317
    {
      EIGEN_STATIC_ASSERT( (_Options & DontAlign) == _Options,
        INVALID_MATRIX_TEMPLATE_PARAMETERS)
    }
#endif
};

/** \ingroup Geometry_Module
  * single precision quaternion type */
typedef Quaternion<float> Quaternionf;
/** \ingroup Geometry_Module
  * double precision quaternion type */
typedef Quaternion<double> Quaterniond;

/***************************************************************************
* Specialization of Map<Quaternion<Scalar>>
***************************************************************************/

namespace internal {
  template<typename _Scalar, int _Options>
Don Gagne's avatar
Don Gagne committed
318
  struct traits<Map<Quaternion<_Scalar>, _Options> > : traits<Quaternion<_Scalar, (int(_Options)&Aligned)==Aligned ? AutoAlign : DontAlign> >
LM's avatar
LM committed
319 320 321 322 323 324 325
  {
    typedef Map<Matrix<_Scalar,4,1>, _Options> Coefficients;
  };
}

namespace internal {
  template<typename _Scalar, int _Options>
Don Gagne's avatar
Don Gagne committed
326
  struct traits<Map<const Quaternion<_Scalar>, _Options> > : traits<Quaternion<_Scalar, (int(_Options)&Aligned)==Aligned ? AutoAlign : DontAlign> >
LM's avatar
LM committed
327 328
  {
    typedef Map<const Matrix<_Scalar,4,1>, _Options> Coefficients;
Don Gagne's avatar
Don Gagne committed
329
    typedef traits<Quaternion<_Scalar, (int(_Options)&Aligned)==Aligned ? AutoAlign : DontAlign> > TraitsBase;
LM's avatar
LM committed
330 331 332 333 334 335
    enum {
      Flags = TraitsBase::Flags & ~LvalueBit
    };
  };
}

Don Gagne's avatar
Don Gagne committed
336 337
/** \ingroup Geometry_Module
  * \brief Quaternion expression mapping a constant memory buffer
LM's avatar
LM committed
338
  *
Don Gagne's avatar
Don Gagne committed
339 340
  * \tparam _Scalar the type of the Quaternion coefficients
  * \tparam _Options see class Map
LM's avatar
LM committed
341 342 343 344 345 346 347 348 349 350
  *
  * This is a specialization of class Map for Quaternion. This class allows to view
  * a 4 scalar memory buffer as an Eigen's Quaternion object.
  *
  * \sa class Map, class Quaternion, class QuaternionBase
  */
template<typename _Scalar, int _Options>
class Map<const Quaternion<_Scalar>, _Options >
  : public QuaternionBase<Map<const Quaternion<_Scalar>, _Options> >
{
351
  public:
LM's avatar
LM committed
352 353 354 355
    typedef QuaternionBase<Map<const Quaternion<_Scalar>, _Options> > Base;

    typedef _Scalar Scalar;
    typedef typename internal::traits<Map>::Coefficients Coefficients;
356
    EIGEN_INHERIT_ASSIGNMENT_OPERATORS(Map)
LM's avatar
LM committed
357 358 359 360
    using Base::operator*=;

    /** Constructs a Mapped Quaternion object from the pointer \a coeffs
      *
361
      * The pointer \a coeffs must reference the four coefficients of Quaternion in the following order:
LM's avatar
LM committed
362 363 364
      * \code *coeffs == {x, y, z, w} \endcode
      *
      * If the template parameter _Options is set to #Aligned, then the pointer coeffs must be aligned. */
365
    EIGEN_DEVICE_FUNC explicit EIGEN_STRONG_INLINE Map(const Scalar* coeffs) : m_coeffs(coeffs) {}
LM's avatar
LM committed
366

367
    EIGEN_DEVICE_FUNC inline const Coefficients& coeffs() const { return m_coeffs;}
LM's avatar
LM committed
368 369 370 371 372

  protected:
    const Coefficients m_coeffs;
};

Don Gagne's avatar
Don Gagne committed
373 374
/** \ingroup Geometry_Module
  * \brief Expression of a quaternion from a memory buffer
LM's avatar
LM committed
375
  *
Don Gagne's avatar
Don Gagne committed
376 377
  * \tparam _Scalar the type of the Quaternion coefficients
  * \tparam _Options see class Map
LM's avatar
LM committed
378 379 380 381 382 383 384 385 386 387
  *
  * This is a specialization of class Map for Quaternion. This class allows to view
  * a 4 scalar memory buffer as an Eigen's  Quaternion object.
  *
  * \sa class Map, class Quaternion, class QuaternionBase
  */
template<typename _Scalar, int _Options>
class Map<Quaternion<_Scalar>, _Options >
  : public QuaternionBase<Map<Quaternion<_Scalar>, _Options> >
{
388
  public:
LM's avatar
LM committed
389 390 391 392
    typedef QuaternionBase<Map<Quaternion<_Scalar>, _Options> > Base;

    typedef _Scalar Scalar;
    typedef typename internal::traits<Map>::Coefficients Coefficients;
393
    EIGEN_INHERIT_ASSIGNMENT_OPERATORS(Map)
LM's avatar
LM committed
394 395 396 397
    using Base::operator*=;

    /** Constructs a Mapped Quaternion object from the pointer \a coeffs
      *
398
      * The pointer \a coeffs must reference the four coefficients of Quaternion in the following order:
LM's avatar
LM committed
399 400 401
      * \code *coeffs == {x, y, z, w} \endcode
      *
      * If the template parameter _Options is set to #Aligned, then the pointer coeffs must be aligned. */
402
    EIGEN_DEVICE_FUNC explicit EIGEN_STRONG_INLINE Map(Scalar* coeffs) : m_coeffs(coeffs) {}
LM's avatar
LM committed
403

404 405
    EIGEN_DEVICE_FUNC inline Coefficients& coeffs() { return m_coeffs; }
    EIGEN_DEVICE_FUNC inline const Coefficients& coeffs() const { return m_coeffs; }
LM's avatar
LM committed
406 407 408 409 410 411

  protected:
    Coefficients m_coeffs;
};

/** \ingroup Geometry_Module
412
  * Map an unaligned array of single precision scalars as a quaternion */
LM's avatar
LM committed
413 414
typedef Map<Quaternion<float>, 0>         QuaternionMapf;
/** \ingroup Geometry_Module
415
  * Map an unaligned array of double precision scalars as a quaternion */
LM's avatar
LM committed
416 417
typedef Map<Quaternion<double>, 0>        QuaternionMapd;
/** \ingroup Geometry_Module
418
  * Map a 16-byte aligned array of single precision scalars as a quaternion */
LM's avatar
LM committed
419 420
typedef Map<Quaternion<float>, Aligned>   QuaternionMapAlignedf;
/** \ingroup Geometry_Module
421
  * Map a 16-byte aligned array of double precision scalars as a quaternion */
LM's avatar
LM committed
422 423 424 425 426 427 428 429 430
typedef Map<Quaternion<double>, Aligned>  QuaternionMapAlignedd;

/***************************************************************************
* Implementation of QuaternionBase methods
***************************************************************************/

// Generic Quaternion * Quaternion product
// This product can be specialized for a given architecture via the Arch template argument.
namespace internal {
431
template<int Arch, class Derived1, class Derived2, typename Scalar> struct quat_product
LM's avatar
LM committed
432
{
433
  EIGEN_DEVICE_FUNC static EIGEN_STRONG_INLINE Quaternion<Scalar> run(const QuaternionBase<Derived1>& a, const QuaternionBase<Derived2>& b){
LM's avatar
LM committed
434 435 436 437 438 439 440 441 442 443 444 445 446 447
    return Quaternion<Scalar>
    (
      a.w() * b.w() - a.x() * b.x() - a.y() * b.y() - a.z() * b.z(),
      a.w() * b.x() + a.x() * b.w() + a.y() * b.z() - a.z() * b.y(),
      a.w() * b.y() + a.y() * b.w() + a.z() * b.x() - a.x() * b.z(),
      a.w() * b.z() + a.z() * b.w() + a.x() * b.y() - a.y() * b.x()
    );
  }
};
}

/** \returns the concatenation of two rotations as a quaternion-quaternion product */
template <class Derived>
template <class OtherDerived>
448
EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE Quaternion<typename internal::traits<Derived>::Scalar>
LM's avatar
LM committed
449 450 451 452 453
QuaternionBase<Derived>::operator* (const QuaternionBase<OtherDerived>& other) const
{
  EIGEN_STATIC_ASSERT((internal::is_same<typename Derived::Scalar, typename OtherDerived::Scalar>::value),
   YOU_MIXED_DIFFERENT_NUMERIC_TYPES__YOU_NEED_TO_USE_THE_CAST_METHOD_OF_MATRIXBASE_TO_CAST_NUMERIC_TYPES_EXPLICITLY)
  return internal::quat_product<Architecture::Target, Derived, OtherDerived,
454
                         typename internal::traits<Derived>::Scalar>::run(*this, other);
LM's avatar
LM committed
455 456 457 458 459
}

/** \sa operator*(Quaternion) */
template <class Derived>
template <class OtherDerived>
460
EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE Derived& QuaternionBase<Derived>::operator*= (const QuaternionBase<OtherDerived>& other)
LM's avatar
LM committed
461 462 463 464 465 466 467 468 469 470 471 472 473
{
  derived() = derived() * other.derived();
  return derived();
}

/** Rotation of a vector by a quaternion.
  * \remarks If the quaternion is used to rotate several points (>1)
  * then it is much more efficient to first convert it to a 3x3 Matrix.
  * Comparison of the operation cost for n transformations:
  *   - Quaternion2:    30n
  *   - Via a Matrix3: 24 + 15n
  */
template <class Derived>
474
EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE typename QuaternionBase<Derived>::Vector3
475
QuaternionBase<Derived>::_transformVector(const Vector3& v) const
LM's avatar
LM committed
476 477 478 479
{
    // Note that this algorithm comes from the optimization by hand
    // of the conversion to a Matrix followed by a Matrix/Vector product.
    // It appears to be much faster than the common algorithm found
480
    // in the literature (30 versus 39 flops). It also requires two
LM's avatar
LM committed
481 482 483 484 485 486 487
    // Vector3 as temporaries.
    Vector3 uv = this->vec().cross(v);
    uv += uv;
    return v + this->w() * uv + this->vec().cross(uv);
}

template<class Derived>
488
EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE QuaternionBase<Derived>& QuaternionBase<Derived>::operator=(const QuaternionBase<Derived>& other)
LM's avatar
LM committed
489 490 491 492 493 494 495
{
  coeffs() = other.coeffs();
  return derived();
}

template<class Derived>
template<class OtherDerived>
496
EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE Derived& QuaternionBase<Derived>::operator=(const QuaternionBase<OtherDerived>& other)
LM's avatar
LM committed
497 498 499 500 501 502 503 504
{
  coeffs() = other.coeffs();
  return derived();
}

/** Set \c *this from an angle-axis \a aa and returns a reference to \c *this
  */
template<class Derived>
505
EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE Derived& QuaternionBase<Derived>::operator=(const AngleAxisType& aa)
LM's avatar
LM committed
506
{
507 508
  EIGEN_USING_STD_MATH(cos)
  EIGEN_USING_STD_MATH(sin)
LM's avatar
LM committed
509
  Scalar ha = Scalar(0.5)*aa.angle(); // Scalar(0.5) to suppress precision loss warnings
Don Gagne's avatar
Don Gagne committed
510 511
  this->w() = cos(ha);
  this->vec() = sin(ha) * aa.axis();
LM's avatar
LM committed
512 513 514 515 516 517 518 519 520 521 522
  return derived();
}

/** Set \c *this from the expression \a xpr:
  *   - if \a xpr is a 4x1 vector, then \a xpr is assumed to be a quaternion
  *   - if \a xpr is a 3x3 matrix, then \a xpr is assumed to be rotation matrix
  *     and \a xpr is converted to a quaternion
  */

template<class Derived>
template<class MatrixDerived>
523
EIGEN_DEVICE_FUNC inline Derived& QuaternionBase<Derived>::operator=(const MatrixBase<MatrixDerived>& xpr)
LM's avatar
LM committed
524 525 526 527 528 529 530 531 532 533 534
{
  EIGEN_STATIC_ASSERT((internal::is_same<typename Derived::Scalar, typename MatrixDerived::Scalar>::value),
   YOU_MIXED_DIFFERENT_NUMERIC_TYPES__YOU_NEED_TO_USE_THE_CAST_METHOD_OF_MATRIXBASE_TO_CAST_NUMERIC_TYPES_EXPLICITLY)
  internal::quaternionbase_assign_impl<MatrixDerived>::run(*this, xpr.derived());
  return derived();
}

/** Convert the quaternion to a 3x3 rotation matrix. The quaternion is required to
  * be normalized, otherwise the result is undefined.
  */
template<class Derived>
535
EIGEN_DEVICE_FUNC inline typename QuaternionBase<Derived>::Matrix3
LM's avatar
LM committed
536 537 538 539 540 541 542 543
QuaternionBase<Derived>::toRotationMatrix(void) const
{
  // NOTE if inlined, then gcc 4.2 and 4.4 get rid of the temporary (not gcc 4.3 !!)
  // if not inlined then the cost of the return by value is huge ~ +35%,
  // however, not inlining this function is an order of magnitude slower, so
  // it has to be inlined, and so the return by value is not an issue
  Matrix3 res;

Don Gagne's avatar
Don Gagne committed
544 545 546
  const Scalar tx  = Scalar(2)*this->x();
  const Scalar ty  = Scalar(2)*this->y();
  const Scalar tz  = Scalar(2)*this->z();
LM's avatar
LM committed
547 548 549 550 551 552 553 554 555 556
  const Scalar twx = tx*this->w();
  const Scalar twy = ty*this->w();
  const Scalar twz = tz*this->w();
  const Scalar txx = tx*this->x();
  const Scalar txy = ty*this->x();
  const Scalar txz = tz*this->x();
  const Scalar tyy = ty*this->y();
  const Scalar tyz = tz*this->y();
  const Scalar tzz = tz*this->z();

Don Gagne's avatar
Don Gagne committed
557
  res.coeffRef(0,0) = Scalar(1)-(tyy+tzz);
LM's avatar
LM committed
558 559 560
  res.coeffRef(0,1) = txy-twz;
  res.coeffRef(0,2) = txz+twy;
  res.coeffRef(1,0) = txy+twz;
Don Gagne's avatar
Don Gagne committed
561
  res.coeffRef(1,1) = Scalar(1)-(txx+tzz);
LM's avatar
LM committed
562 563 564
  res.coeffRef(1,2) = tyz-twx;
  res.coeffRef(2,0) = txz-twy;
  res.coeffRef(2,1) = tyz+twx;
Don Gagne's avatar
Don Gagne committed
565
  res.coeffRef(2,2) = Scalar(1)-(txx+tyy);
LM's avatar
LM committed
566 567 568 569 570 571 572 573 574 575 576 577 578 579 580 581

  return res;
}

/** Sets \c *this to be a quaternion representing a rotation between
  * the two arbitrary vectors \a a and \a b. In other words, the built
  * rotation represent a rotation sending the line of direction \a a
  * to the line of direction \a b, both lines passing through the origin.
  *
  * \returns a reference to \c *this.
  *
  * Note that the two input vectors do \b not have to be normalized, and
  * do not need to have the same norm.
  */
template<class Derived>
template<typename Derived1, typename Derived2>
582
EIGEN_DEVICE_FUNC inline Derived& QuaternionBase<Derived>::setFromTwoVectors(const MatrixBase<Derived1>& a, const MatrixBase<Derived2>& b)
LM's avatar
LM committed
583
{
584
  EIGEN_USING_STD_MATH(sqrt)
LM's avatar
LM committed
585 586 587 588 589
  Vector3 v0 = a.normalized();
  Vector3 v1 = b.normalized();
  Scalar c = v1.dot(v0);

  // if dot == -1, vectors are nearly opposites
590
  // => accurately compute the rotation axis by computing the
LM's avatar
LM committed
591 592 593 594 595 596 597 598
  //    intersection of the two planes. This is done by solving:
  //       x^T v0 = 0
  //       x^T v1 = 0
  //    under the constraint:
  //       ||x|| = 1
  //    which yields a singular value problem
  if (c < Scalar(-1)+NumTraits<Scalar>::dummy_precision())
  {
599
    c = numext::maxi(c,Scalar(-1));
LM's avatar
LM committed
600 601 602 603 604
    Matrix<Scalar,2,3> m; m << v0.transpose(), v1.transpose();
    JacobiSVD<Matrix<Scalar,2,3> > svd(m, ComputeFullV);
    Vector3 axis = svd.matrixV().col(2);

    Scalar w2 = (Scalar(1)+c)*Scalar(0.5);
Don Gagne's avatar
Don Gagne committed
605 606
    this->w() = sqrt(w2);
    this->vec() = axis * sqrt(Scalar(1) - w2);
LM's avatar
LM committed
607 608 609
    return derived();
  }
  Vector3 axis = v0.cross(v1);
Don Gagne's avatar
Don Gagne committed
610
  Scalar s = sqrt((Scalar(1)+c)*Scalar(2));
LM's avatar
LM committed
611 612 613 614 615 616 617
  Scalar invs = Scalar(1)/s;
  this->vec() = axis * invs;
  this->w() = s * Scalar(0.5);

  return derived();
}

618 619 620 621 622 623 624 625 626 627 628 629 630 631 632 633 634 635
/** \returns a random unit quaternion following a uniform distribution law on SO(3)
  *
  * \note The implementation is based on http://planning.cs.uiuc.edu/node198.html
  */
template<typename Scalar, int Options>
EIGEN_DEVICE_FUNC Quaternion<Scalar,Options> Quaternion<Scalar,Options>::UnitRandom()
{
  EIGEN_USING_STD_MATH(sqrt)
  EIGEN_USING_STD_MATH(sin)
  EIGEN_USING_STD_MATH(cos)
  const Scalar u1 = internal::random<Scalar>(0, 1),
               u2 = internal::random<Scalar>(0, 2*EIGEN_PI),
               u3 = internal::random<Scalar>(0, 2*EIGEN_PI);
  const Scalar a = sqrt(1 - u1),
               b = sqrt(u1);
  return Quaternion (a * sin(u2), a * cos(u2), b * sin(u3), b * cos(u3));
}

Don Gagne's avatar
Don Gagne committed
636 637 638 639 640 641 642 643 644 645 646 647 648

/** Returns a quaternion representing a rotation between
  * the two arbitrary vectors \a a and \a b. In other words, the built
  * rotation represent a rotation sending the line of direction \a a
  * to the line of direction \a b, both lines passing through the origin.
  *
  * \returns resulting quaternion
  *
  * Note that the two input vectors do \b not have to be normalized, and
  * do not need to have the same norm.
  */
template<typename Scalar, int Options>
template<typename Derived1, typename Derived2>
649
EIGEN_DEVICE_FUNC Quaternion<Scalar,Options> Quaternion<Scalar,Options>::FromTwoVectors(const MatrixBase<Derived1>& a, const MatrixBase<Derived2>& b)
Don Gagne's avatar
Don Gagne committed
650 651 652 653 654 655 656
{
    Quaternion quat;
    quat.setFromTwoVectors(a, b);
    return quat;
}


LM's avatar
LM committed
657 658 659 660 661 662 663
/** \returns the multiplicative inverse of \c *this
  * Note that in most cases, i.e., if you simply want the opposite rotation,
  * and/or the quaternion is normalized, then it is enough to use the conjugate.
  *
  * \sa QuaternionBase::conjugate()
  */
template <class Derived>
664
EIGEN_DEVICE_FUNC inline Quaternion<typename internal::traits<Derived>::Scalar> QuaternionBase<Derived>::inverse() const
LM's avatar
LM committed
665 666 667
{
  // FIXME should this function be called multiplicativeInverse and conjugate() be called inverse() or opposite()  ??
  Scalar n2 = this->squaredNorm();
668
  if (n2 > Scalar(0))
LM's avatar
LM committed
669 670 671 672 673 674 675 676
    return Quaternion<Scalar>(conjugate().coeffs() / n2);
  else
  {
    // return an invalid result to flag the error
    return Quaternion<Scalar>(Coefficients::Zero());
  }
}

677 678 679 680 681 682 683 684 685 686
// Generic conjugate of a Quaternion
namespace internal {
template<int Arch, class Derived, typename Scalar> struct quat_conj
{
  EIGEN_DEVICE_FUNC static EIGEN_STRONG_INLINE Quaternion<Scalar> run(const QuaternionBase<Derived>& q){
    return Quaternion<Scalar>(q.w(),-q.x(),-q.y(),-q.z());
  }
};
}
                         
LM's avatar
LM committed
687 688 689 690 691 692 693
/** \returns the conjugate of the \c *this which is equal to the multiplicative inverse
  * if the quaternion is normalized.
  * The conjugate of a quaternion represents the opposite rotation.
  *
  * \sa Quaternion2::inverse()
  */
template <class Derived>
694
EIGEN_DEVICE_FUNC inline Quaternion<typename internal::traits<Derived>::Scalar>
LM's avatar
LM committed
695 696
QuaternionBase<Derived>::conjugate() const
{
697 698 699
  return internal::quat_conj<Architecture::Target, Derived,
                         typename internal::traits<Derived>::Scalar>::run(*this);
                         
LM's avatar
LM committed
700 701 702 703 704 705 706
}

/** \returns the angle (in radian) between two rotations
  * \sa dot()
  */
template <class Derived>
template <class OtherDerived>
707
EIGEN_DEVICE_FUNC inline typename internal::traits<Derived>::Scalar
LM's avatar
LM committed
708 709
QuaternionBase<Derived>::angularDistance(const QuaternionBase<OtherDerived>& other) const
{
710
  EIGEN_USING_STD_MATH(atan2)
711
  Quaternion<Scalar> d = (*this) * other.conjugate();
712
  return Scalar(2) * atan2( d.vec().norm(), numext::abs(d.w()) );
LM's avatar
LM committed
713 714
}

715 716
 
    
LM's avatar
LM committed
717
/** \returns the spherical linear interpolation between the two quaternions
718 719 720 721
  * \c *this and \a other at the parameter \a t in [0;1].
  * 
  * This represents an interpolation for a constant motion between \c *this and \a other,
  * see also http://en.wikipedia.org/wiki/Slerp.
LM's avatar
LM committed
722 723 724
  */
template <class Derived>
template <class OtherDerived>
725
EIGEN_DEVICE_FUNC Quaternion<typename internal::traits<Derived>::Scalar>
Don Gagne's avatar
Don Gagne committed
726
QuaternionBase<Derived>::slerp(const Scalar& t, const QuaternionBase<OtherDerived>& other) const
LM's avatar
LM committed
727
{
728 729 730
  EIGEN_USING_STD_MATH(acos)
  EIGEN_USING_STD_MATH(sin)
  const Scalar one = Scalar(1) - NumTraits<Scalar>::epsilon();
LM's avatar
LM committed
731
  Scalar d = this->dot(other);
732
  Scalar absD = numext::abs(d);
LM's avatar
LM committed
733 734 735 736

  Scalar scale0;
  Scalar scale1;

Don Gagne's avatar
Don Gagne committed
737
  if(absD>=one)
LM's avatar
LM committed
738 739 740 741 742 743 744 745
  {
    scale0 = Scalar(1) - t;
    scale1 = t;
  }
  else
  {
    // theta is the angle between the 2 quaternions
    Scalar theta = acos(absD);
Don Gagne's avatar
Don Gagne committed
746
    Scalar sinTheta = sin(theta);
LM's avatar
LM committed
747

Don Gagne's avatar
Don Gagne committed
748 749
    scale0 = sin( ( Scalar(1) - t ) * theta) / sinTheta;
    scale1 = sin( ( t * theta) ) / sinTheta;
LM's avatar
LM committed
750
  }
751
  if(d<Scalar(0)) scale1 = -scale1;
LM's avatar
LM committed
752 753 754 755 756 757 758 759 760 761 762

  return Quaternion<Scalar>(scale0 * coeffs() + scale1 * other.coeffs());
}

namespace internal {

// set from a rotation matrix
template<typename Other>
struct quaternionbase_assign_impl<Other,3,3>
{
  typedef typename Other::Scalar Scalar;
763
  template<class Derived> EIGEN_DEVICE_FUNC static inline void run(QuaternionBase<Derived>& q, const Other& a_mat)
LM's avatar
LM committed
764
  {
765 766
    const typename internal::nested_eval<Other,2>::type mat(a_mat);
    EIGEN_USING_STD_MATH(sqrt)
LM's avatar
LM committed
767 768 769 770 771 772 773 774 775 776 777 778 779 780
    // This algorithm comes from  "Quaternion Calculus and Fast Animation",
    // Ken Shoemake, 1987 SIGGRAPH course notes
    Scalar t = mat.trace();
    if (t > Scalar(0))
    {
      t = sqrt(t + Scalar(1.0));
      q.w() = Scalar(0.5)*t;
      t = Scalar(0.5)/t;
      q.x() = (mat.coeff(2,1) - mat.coeff(1,2)) * t;
      q.y() = (mat.coeff(0,2) - mat.coeff(2,0)) * t;
      q.z() = (mat.coeff(1,0) - mat.coeff(0,1)) * t;
    }
    else
    {
781
      Index i = 0;
LM's avatar
LM committed
782 783 784 785
      if (mat.coeff(1,1) > mat.coeff(0,0))
        i = 1;
      if (mat.coeff(2,2) > mat.coeff(i,i))
        i = 2;
786 787
      Index j = (i+1)%3;
      Index k = (j+1)%3;
LM's avatar
LM committed
788 789 790 791 792 793 794 795 796 797 798 799 800 801 802 803

      t = sqrt(mat.coeff(i,i)-mat.coeff(j,j)-mat.coeff(k,k) + Scalar(1.0));
      q.coeffs().coeffRef(i) = Scalar(0.5) * t;
      t = Scalar(0.5)/t;
      q.w() = (mat.coeff(k,j)-mat.coeff(j,k))*t;
      q.coeffs().coeffRef(j) = (mat.coeff(j,i)+mat.coeff(i,j))*t;
      q.coeffs().coeffRef(k) = (mat.coeff(k,i)+mat.coeff(i,k))*t;
    }
  }
};

// set from a vector of coefficients assumed to be a quaternion
template<typename Other>
struct quaternionbase_assign_impl<Other,4,1>
{
  typedef typename Other::Scalar Scalar;
804
  template<class Derived> EIGEN_DEVICE_FUNC static inline void run(QuaternionBase<Derived>& q, const Other& vec)
LM's avatar
LM committed
805 806 807 808 809 810 811
  {
    q.coeffs() = vec;
  }
};

} // end namespace internal

Don Gagne's avatar
Don Gagne committed
812 813
} // end namespace Eigen

LM's avatar
LM committed
814
#endif // EIGEN_QUATERNION_H