#ifndef EIGEN_QUATERNION_H
#define EIGEN_QUATERNION_H
#include "./InternalHeaderCheck.h"
namespace Eigen {
namespace internal {
template <typename Other, int OtherRows = Other::RowsAtCompileTime, int OtherCols = Other::ColsAtCompileTime>
struct quaternionbase_assign_impl;
}
template <class Derived>
class QuaternionBase : public RotationBase<Derived, 3> {
public:
typedef RotationBase<Derived, 3> Base;
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;
typedef typename Coefficients::CoeffReturnType CoeffReturnType;
typedef std::conditional_t<bool(internal::traits<Derived>::Flags& LvalueBit), Scalar&, CoeffReturnType>
NonConstCoeffReturnType;
enum { Flags = Eigen::internal::traits<Derived>::Flags };
typedef Matrix<Scalar, 3, 1> Vector3;
typedef Matrix<Scalar, 3, 3> Matrix3;
typedef AngleAxis<Scalar> AngleAxisType;
EIGEN_DEVICE_FUNC constexpr CoeffReturnType x() const { return this->derived().coeffs().coeff(0); }
EIGEN_DEVICE_FUNC constexpr CoeffReturnType y() const { return this->derived().coeffs().coeff(1); }
EIGEN_DEVICE_FUNC constexpr CoeffReturnType z() const { return this->derived().coeffs().coeff(2); }
EIGEN_DEVICE_FUNC constexpr CoeffReturnType w() const { return this->derived().coeffs().coeff(3); }
EIGEN_DEVICE_FUNC constexpr NonConstCoeffReturnType x() { return this->derived().coeffs().x(); }
EIGEN_DEVICE_FUNC constexpr NonConstCoeffReturnType y() { return this->derived().coeffs().y(); }
EIGEN_DEVICE_FUNC constexpr NonConstCoeffReturnType z() { return this->derived().coeffs().z(); }
EIGEN_DEVICE_FUNC constexpr NonConstCoeffReturnType w() { return this->derived().coeffs().w(); }
EIGEN_DEVICE_FUNC inline const VectorBlock<const Coefficients, 3> vec() const { return coeffs().template head<3>(); }
EIGEN_DEVICE_FUNC inline VectorBlock<Coefficients, 3> vec() { return coeffs().template head<3>(); }
EIGEN_DEVICE_FUNC inline const typename internal::traits<Derived>::Coefficients& coeffs() const {
return derived().coeffs();
}
EIGEN_DEVICE_FUNC inline typename internal::traits<Derived>::Coefficients coeffsScalarFirst() const {
return derived().coeffsScalarFirst();
}
EIGEN_DEVICE_FUNC inline typename internal::traits<Derived>::Coefficients coeffsScalarLast() const {
return derived().coeffsScalarLast();
}
EIGEN_DEVICE_FUNC inline typename internal::traits<Derived>::Coefficients& coeffs() { return derived().coeffs(); }
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);
EIGEN_DEVICE_FUNC Derived& operator=(const AngleAxisType& aa);
template <class OtherDerived>
EIGEN_DEVICE_FUNC Derived& operator=(const MatrixBase<OtherDerived>& m);
EIGEN_DEVICE_FUNC static inline Quaternion<Scalar> Identity() {
return Quaternion<Scalar>(Scalar(1), Scalar(0), Scalar(0), Scalar(0));
}
EIGEN_DEVICE_FUNC inline QuaternionBase& setIdentity() {
coeffs() << Scalar(0), Scalar(0), Scalar(0), Scalar(1);
return *this;
}
EIGEN_DEVICE_FUNC inline Scalar squaredNorm() const { return coeffs().squaredNorm(); }
EIGEN_DEVICE_FUNC inline Scalar norm() const { return coeffs().norm(); }
EIGEN_DEVICE_FUNC inline void normalize() { coeffs().normalize(); }
EIGEN_DEVICE_FUNC inline Quaternion<Scalar> normalized() const { return Quaternion<Scalar>(coeffs().normalized()); }
template <class OtherDerived>
EIGEN_DEVICE_FUNC inline Scalar dot(const QuaternionBase<OtherDerived>& other) const {
return coeffs().dot(other.coeffs());
}
template <class OtherDerived>
EIGEN_DEVICE_FUNC Scalar angularDistance(const QuaternionBase<OtherDerived>& other) const;
EIGEN_DEVICE_FUNC inline Matrix3 toRotationMatrix() const;
template <typename Derived1, typename Derived2>
EIGEN_DEVICE_FUNC Derived& setFromTwoVectors(const MatrixBase<Derived1>& a, const MatrixBase<Derived2>& b);
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);
EIGEN_DEVICE_FUNC Quaternion<Scalar> inverse() const;
EIGEN_DEVICE_FUNC Quaternion<Scalar> conjugate() const;
template <class OtherDerived>
EIGEN_DEVICE_FUNC Quaternion<Scalar> slerp(const Scalar& t, const QuaternionBase<OtherDerived>& other) const;
template <class OtherDerived>
EIGEN_DEVICE_FUNC inline bool operator==(const QuaternionBase<OtherDerived>& other) const {
return coeffs() == other.coeffs();
}
template <class OtherDerived>
EIGEN_DEVICE_FUNC inline bool operator!=(const QuaternionBase<OtherDerived>& other) const {
return coeffs() != other.coeffs();
}
template <class OtherDerived>
EIGEN_DEVICE_FUNC bool isApprox(const QuaternionBase<OtherDerived>& other,
const RealScalar& prec = NumTraits<Scalar>::dummy_precision()) const {
return coeffs().isApprox(other.coeffs(), prec);
}
EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE Vector3 _transformVector(const Vector3& v) const;
#ifdef EIGEN_PARSED_BY_DOXYGEN
template <typename NewScalarType>
EIGEN_DEVICE_FUNC inline typename internal::cast_return_type<Derived, Quaternion<NewScalarType> >::type cast() const;
#else
template <typename NewScalarType>
EIGEN_DEVICE_FUNC inline std::enable_if_t<internal::is_same<Scalar, NewScalarType>::value, const Derived&> cast()
const {
return derived();
}
template <typename NewScalarType>
EIGEN_DEVICE_FUNC inline std::enable_if_t<!internal::is_same<Scalar, NewScalarType>::value,
Quaternion<NewScalarType> >
cast() const {
return Quaternion<NewScalarType>(coeffs().template cast<NewScalarType>());
}
#endif
#ifndef EIGEN_NO_IO
friend std::ostream& operator<<(std::ostream& s, const QuaternionBase<Derived>& q) {
s << q.x() << "i + " << q.y() << "j + " << q.z() << "k"
<< " + " << q.w();
return s;
}
#endif
#ifdef EIGEN_QUATERNIONBASE_PLUGIN
#include EIGEN_QUATERNIONBASE_PLUGIN
#endif
protected:
EIGEN_DEFAULT_COPY_CONSTRUCTOR(QuaternionBase)
EIGEN_DEFAULT_EMPTY_CONSTRUCTOR_AND_DESTRUCTOR(QuaternionBase)
};
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 { Alignment = internal::traits<Coefficients>::Alignment, Flags = LvalueBit };
};
}
template <typename Scalar_, int Options_>
class Quaternion : public QuaternionBase<Quaternion<Scalar_, Options_> > {
public:
typedef QuaternionBase<Quaternion<Scalar_, Options_> > Base;
enum { NeedsAlignment = internal::traits<Quaternion>::Alignment > 0 };
typedef Scalar_ Scalar;
EIGEN_INHERIT_ASSIGNMENT_OPERATORS(Quaternion)
using Base::operator*=;
typedef typename internal::traits<Quaternion>::Coefficients Coefficients;
typedef typename Base::AngleAxisType AngleAxisType;
EIGEN_DEVICE_FUNC inline Quaternion() {}
EIGEN_DEVICE_FUNC inline Quaternion(const Scalar& w, const Scalar& x, const Scalar& y, const Scalar& z)
: m_coeffs(x, y, z, w) {}
template <typename Derived>
EIGEN_DEVICE_FUNC inline Quaternion(const Scalar& w, const Eigen::MatrixBase<Derived>& vec)
: m_coeffs(vec.x(), vec.y(), vec.z(), w) {
EIGEN_STATIC_ASSERT_VECTOR_SPECIFIC_SIZE(Derived, 3);
}
EIGEN_DEVICE_FUNC explicit inline Quaternion(const Scalar* data) : m_coeffs(data) {}
template <class Derived>
EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE Quaternion(const QuaternionBase<Derived>& other) {
this->Base::operator=(other);
}
EIGEN_DEVICE_FUNC explicit inline Quaternion(const AngleAxisType& aa) { *this = aa; }
template <typename Derived>
EIGEN_DEVICE_FUNC explicit inline Quaternion(const MatrixBase<Derived>& other) {
*this = other;
}
template <typename OtherScalar, int OtherOptions>
EIGEN_DEVICE_FUNC explicit inline Quaternion(const Quaternion<OtherScalar, OtherOptions>& other) {
m_coeffs = other.coeffs().template cast<Scalar>();
}
EIGEN_DEVICE_FUNC inline Quaternion(Quaternion&& other) noexcept(std::is_nothrow_move_constructible<Scalar>::value)
: m_coeffs(std::move(other.coeffs())) {}
EIGEN_DEVICE_FUNC Quaternion& operator=(Quaternion&& other) noexcept(std::is_nothrow_move_assignable<Scalar>::value) {
m_coeffs = std::move(other.coeffs());
return *this;
}
EIGEN_DEVICE_FUNC static Quaternion UnitRandom();
EIGEN_DEVICE_FUNC static Quaternion FromCoeffsScalarLast(const Scalar& x, const Scalar& y, const Scalar& z,
const Scalar& w);
EIGEN_DEVICE_FUNC static Quaternion FromCoeffsScalarFirst(const Scalar& w, const Scalar& x, const Scalar& y,
const Scalar& z);
template <typename Derived1, typename Derived2>
EIGEN_DEVICE_FUNC static Quaternion FromTwoVectors(const MatrixBase<Derived1>& a, const MatrixBase<Derived2>& b);
EIGEN_DEVICE_FUNC inline Coefficients& coeffs() { return m_coeffs; }
EIGEN_DEVICE_FUNC inline const Coefficients& coeffs() const { return m_coeffs; }
EIGEN_DEVICE_FUNC inline Coefficients coeffsScalarLast() const { return m_coeffs; }
EIGEN_DEVICE_FUNC inline Coefficients coeffsScalarFirst() const {
return {m_coeffs.w(), m_coeffs.x(), m_coeffs.y(), m_coeffs.z()};
}
EIGEN_MAKE_ALIGNED_OPERATOR_NEW_IF(bool(NeedsAlignment))
#ifdef EIGEN_QUATERNION_PLUGIN
#include EIGEN_QUATERNION_PLUGIN
#endif
protected:
Coefficients m_coeffs;
#ifndef EIGEN_PARSED_BY_DOXYGEN
EIGEN_STATIC_ASSERT((Options_ & DontAlign) == Options_, INVALID_MATRIX_TEMPLATE_PARAMETERS)
#endif
};
typedef Quaternion<float> Quaternionf;
typedef Quaternion<double> Quaterniond;
namespace internal {
template <typename Scalar_, int Options_>
struct traits<Map<Quaternion<Scalar_>, Options_> >
: traits<Quaternion<Scalar_, (int(Options_) & Aligned) == Aligned ? AutoAlign : DontAlign> > {
typedef Map<Matrix<Scalar_, 4, 1>, Options_> Coefficients;
};
}
namespace internal {
template <typename Scalar_, int Options_>
struct traits<Map<const Quaternion<Scalar_>, Options_> >
: traits<Quaternion<Scalar_, (int(Options_) & Aligned) == Aligned ? AutoAlign : DontAlign> > {
typedef Map<const Matrix<Scalar_, 4, 1>, Options_> Coefficients;
typedef traits<Quaternion<Scalar_, (int(Options_) & Aligned) == Aligned ? AutoAlign : DontAlign> > TraitsBase;
enum { Flags = TraitsBase::Flags & ~LvalueBit };
};
}
template <typename Scalar_, int Options_>
class Map<const Quaternion<Scalar_>, Options_> : public QuaternionBase<Map<const Quaternion<Scalar_>, Options_> > {
public:
typedef QuaternionBase<Map<const Quaternion<Scalar_>, Options_> > Base;
typedef Scalar_ Scalar;
typedef typename internal::traits<Map>::Coefficients Coefficients;
EIGEN_INHERIT_ASSIGNMENT_OPERATORS(Map)
using Base::operator*=;
EIGEN_DEVICE_FUNC explicit EIGEN_STRONG_INLINE Map(const Scalar* coeffs) : m_coeffs(coeffs) {}
EIGEN_DEVICE_FUNC inline const Coefficients& coeffs() const { return m_coeffs; }
EIGEN_DEVICE_FUNC inline Coefficients coeffsScalarLast() const { return m_coeffs; }
EIGEN_DEVICE_FUNC inline Coefficients coeffsScalarFirst() const {
return {m_coeffs.w(), m_coeffs.x(), m_coeffs.y(), m_coeffs.z()};
}
protected:
const Coefficients m_coeffs;
};
template <typename Scalar_, int Options_>
class Map<Quaternion<Scalar_>, Options_> : public QuaternionBase<Map<Quaternion<Scalar_>, Options_> > {
public:
typedef QuaternionBase<Map<Quaternion<Scalar_>, Options_> > Base;
typedef Scalar_ Scalar;
typedef typename internal::traits<Map>::Coefficients Coefficients;
EIGEN_INHERIT_ASSIGNMENT_OPERATORS(Map)
using Base::operator*=;
EIGEN_DEVICE_FUNC explicit EIGEN_STRONG_INLINE Map(Scalar* coeffs) : m_coeffs(coeffs) {}
EIGEN_DEVICE_FUNC inline Coefficients& coeffs() { return m_coeffs; }
EIGEN_DEVICE_FUNC inline const Coefficients& coeffs() const { return m_coeffs; }
EIGEN_DEVICE_FUNC inline Coefficients coeffsScalarLast() const { return m_coeffs; }
EIGEN_DEVICE_FUNC inline Coefficients coeffsScalarFirst() const {
return {m_coeffs.w(), m_coeffs.x(), m_coeffs.y(), m_coeffs.z()};
}
protected:
Coefficients m_coeffs;
};
typedef Map<Quaternion<float>, 0> QuaternionMapf;
typedef Map<Quaternion<double>, 0> QuaternionMapd;
typedef Map<Quaternion<float>, Aligned> QuaternionMapAlignedf;
typedef Map<Quaternion<double>, Aligned> QuaternionMapAlignedd;
namespace internal {
template <int Arch, class Derived1, class Derived2, typename Scalar>
struct quat_product {
EIGEN_DEVICE_FUNC static EIGEN_STRONG_INLINE Quaternion<Scalar> run(const QuaternionBase<Derived1>& a,
const QuaternionBase<Derived2>& b) {
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());
}
};
}
template <class Derived>
template <class OtherDerived>
EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE Quaternion<typename internal::traits<Derived>::Scalar>
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,
typename internal::traits<Derived>::Scalar>::run(*this, other);
}
template <class Derived>
template <class OtherDerived>
EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE Derived& QuaternionBase<Derived>::operator*=(
const QuaternionBase<OtherDerived>& other) {
derived() = derived() * other.derived();
return derived();
}
template <class Derived>
EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE typename QuaternionBase<Derived>::Vector3
QuaternionBase<Derived>::_transformVector(const Vector3& v) const {
Vector3 uv = this->vec().cross(v);
uv += uv;
return v + this->w() * uv + this->vec().cross(uv);
}
template <class Derived>
EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE QuaternionBase<Derived>& QuaternionBase<Derived>::operator=(
const QuaternionBase<Derived>& other) {
coeffs() = other.coeffs();
return derived();
}
template <class Derived>
template <class OtherDerived>
EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE Derived& QuaternionBase<Derived>::operator=(
const QuaternionBase<OtherDerived>& other) {
coeffs() = other.coeffs();
return derived();
}
template <class Derived>
EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE Derived& QuaternionBase<Derived>::operator=(const AngleAxisType& aa) {
EIGEN_USING_STD(cos)
EIGEN_USING_STD(sin)
Scalar ha = Scalar(0.5) * aa.angle(); this->w() = cos(ha);
this->vec() = sin(ha) * aa.axis();
return derived();
}
template <class Derived>
template <class MatrixDerived>
EIGEN_DEVICE_FUNC inline Derived& QuaternionBase<Derived>::operator=(const MatrixBase<MatrixDerived>& xpr) {
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();
}
template <class Derived>
EIGEN_DEVICE_FUNC inline typename QuaternionBase<Derived>::Matrix3 QuaternionBase<Derived>::toRotationMatrix(
void) const {
Matrix3 res;
const Scalar tx = Scalar(2) * this->x();
const Scalar ty = Scalar(2) * this->y();
const Scalar tz = Scalar(2) * this->z();
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();
res.coeffRef(0, 0) = Scalar(1) - (tyy + tzz);
res.coeffRef(0, 1) = txy - twz;
res.coeffRef(0, 2) = txz + twy;
res.coeffRef(1, 0) = txy + twz;
res.coeffRef(1, 1) = Scalar(1) - (txx + tzz);
res.coeffRef(1, 2) = tyz - twx;
res.coeffRef(2, 0) = txz - twy;
res.coeffRef(2, 1) = tyz + twx;
res.coeffRef(2, 2) = Scalar(1) - (txx + tyy);
return res;
}
template <class Derived>
template <typename Derived1, typename Derived2>
EIGEN_DEVICE_FUNC inline Derived& QuaternionBase<Derived>::setFromTwoVectors(const MatrixBase<Derived1>& a,
const MatrixBase<Derived2>& b) {
EIGEN_USING_STD(sqrt)
Vector3 v0 = a.normalized();
Vector3 v1 = b.normalized();
Scalar c = v1.dot(v0);
if (c < Scalar(-1) + NumTraits<Scalar>::dummy_precision()) {
c = numext::maxi(c, Scalar(-1));
Matrix<Scalar, 2, 3> m;
m << v0.transpose(), v1.transpose();
JacobiSVD<Matrix<Scalar, 2, 3>, ComputeFullV> svd(m);
Vector3 axis = svd.matrixV().col(2);
Scalar w2 = (Scalar(1) + c) * Scalar(0.5);
this->w() = sqrt(w2);
this->vec() = axis * sqrt(Scalar(1) - w2);
return derived();
}
Vector3 axis = v0.cross(v1);
Scalar s = sqrt((Scalar(1) + c) * Scalar(2));
Scalar invs = Scalar(1) / s;
this->vec() = axis * invs;
this->w() = s * Scalar(0.5);
return derived();
}
template <typename Scalar, int Options>
EIGEN_DEVICE_FUNC Quaternion<Scalar, Options> Quaternion<Scalar, Options>::UnitRandom() {
EIGEN_USING_STD(sqrt)
EIGEN_USING_STD(sin)
EIGEN_USING_STD(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(Scalar(1) - u1), b = sqrt(u1);
return Quaternion(a * sin(u2), a * cos(u2), b * sin(u3), b * cos(u3));
}
template <typename Scalar, int Options>
EIGEN_DEVICE_FUNC Quaternion<Scalar, Options> Quaternion<Scalar, Options>::FromCoeffsScalarLast(const Scalar& x,
const Scalar& y,
const Scalar& z,
const Scalar& w) {
return Quaternion(w, x, y, z);
}
template <typename Scalar, int Options>
EIGEN_DEVICE_FUNC Quaternion<Scalar, Options> Quaternion<Scalar, Options>::FromCoeffsScalarFirst(const Scalar& w,
const Scalar& x,
const Scalar& y,
const Scalar& z) {
return Quaternion(w, x, y, z);
}
template <typename Scalar, int Options>
template <typename Derived1, typename Derived2>
EIGEN_DEVICE_FUNC Quaternion<Scalar, Options> Quaternion<Scalar, Options>::FromTwoVectors(
const MatrixBase<Derived1>& a, const MatrixBase<Derived2>& b) {
Quaternion quat;
quat.setFromTwoVectors(a, b);
return quat;
}
template <class Derived>
EIGEN_DEVICE_FUNC inline Quaternion<typename internal::traits<Derived>::Scalar> QuaternionBase<Derived>::inverse()
const {
Scalar n2 = this->squaredNorm();
if (n2 > Scalar(0))
return Quaternion<Scalar>(conjugate().coeffs() / n2);
else {
return Quaternion<Scalar>(Coefficients::Zero());
}
}
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());
}
};
}
template <class Derived>
EIGEN_DEVICE_FUNC inline Quaternion<typename internal::traits<Derived>::Scalar> QuaternionBase<Derived>::conjugate()
const {
return internal::quat_conj<Architecture::Target, Derived, typename internal::traits<Derived>::Scalar>::run(*this);
}
template <class Derived>
template <class OtherDerived>
EIGEN_DEVICE_FUNC inline typename internal::traits<Derived>::Scalar QuaternionBase<Derived>::angularDistance(
const QuaternionBase<OtherDerived>& other) const {
EIGEN_USING_STD(atan2)
Quaternion<Scalar> d = (*this) * other.conjugate();
return Scalar(2) * atan2(d.vec().norm(), numext::abs(d.w()));
}
template <class Derived>
template <class OtherDerived>
EIGEN_DEVICE_FUNC Quaternion<typename internal::traits<Derived>::Scalar> QuaternionBase<Derived>::slerp(
const Scalar& t, const QuaternionBase<OtherDerived>& other) const {
EIGEN_USING_STD(acos)
EIGEN_USING_STD(sin)
const Scalar one = Scalar(1) - NumTraits<Scalar>::epsilon();
Scalar d = this->dot(other);
Scalar absD = numext::abs(d);
Scalar scale0;
Scalar scale1;
if (absD >= one) {
scale0 = Scalar(1) - t;
scale1 = t;
} else {
Scalar theta = acos(absD);
Scalar sinTheta = numext::sqrt(Scalar(1) - absD * absD);
scale0 = sin((Scalar(1) - t) * theta) / sinTheta;
scale1 = sin((t * theta)) / sinTheta;
}
if (d < Scalar(0)) scale1 = -scale1;
return Quaternion<Scalar>(scale0 * coeffs() + scale1 * other.coeffs());
}
namespace internal {
template <typename Other>
struct quaternionbase_assign_impl<Other, 3, 3> {
typedef typename Other::Scalar Scalar;
template <class Derived>
EIGEN_DEVICE_FUNC static inline void run(QuaternionBase<Derived>& q, const Other& a_mat) {
const typename internal::nested_eval<Other, 2>::type mat(a_mat);
EIGEN_USING_STD(sqrt)
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 {
Index i = 0;
if (mat.coeff(1, 1) > mat.coeff(0, 0)) i = 1;
if (mat.coeff(2, 2) > mat.coeff(i, i)) i = 2;
Index j = (i + 1) % 3;
Index k = (j + 1) % 3;
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;
}
}
};
template <typename Other>
struct quaternionbase_assign_impl<Other, 4, 1> {
typedef typename Other::Scalar Scalar;
template <class Derived>
EIGEN_DEVICE_FUNC static inline void run(QuaternionBase<Derived>& q, const Other& vec) {
q.coeffs() = vec;
}
};
}
}
#endif