openfoam/src/OpenFOAM/primitives/quaternion/quaternion.H
Henry Weller b4ebcd770f quaternion: Added generalized construction from and conversion to Euler-angles
The particular rotation sequence is specified via the enumeration:

    //- Euler-angle rotation sequence
    enum rotationSequence
    {
        ZYX, ZYZ, ZXY, ZXZ, YXZ, YXY, YZX, YZY, XYZ, XYX, XZY, XZX
    };

and provided as an argument to the constructor from Euler-angles

    //- Construct a quaternion given the three Euler angles:
    inline quaternion
    (
        const rotationSequence rs,
        const vector& angles
    );

and conversion to Euler-angles:

    //- Return a vector of euler angles corresponding to the
    //  specified rotation sequence
    inline vector eulerAngles(const rotationSequence rs) const;
2016-03-14 08:07:42 +00:00

299 lines
8.7 KiB
C++

/*---------------------------------------------------------------------------*\
========= |
\\ / F ield | OpenFOAM: The Open Source CFD Toolbox
\\ / O peration |
\\ / A nd | Copyright (C) 2011-2016 OpenFOAM Foundation
\\/ M anipulation |
-------------------------------------------------------------------------------
License
This file is part of OpenFOAM.
OpenFOAM is free software: you can redistribute it and/or modify it
under the terms of the GNU General Public License as published by
the Free Software Foundation, either version 3 of the License, or
(at your option) any later version.
OpenFOAM is distributed in the hope that it will be useful, but WITHOUT
ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or
FITNESS FOR A PARTICULAR PURPOSE. See the GNU General Public License
for more details.
You should have received a copy of the GNU General Public License
along with OpenFOAM. If not, see <http://www.gnu.org/licenses/>.
Class
Foam::quaternion
Description
Quaternion class used to perform rotations in 3D space.
SourceFiles
quaternionI.H
quaternion.C
\*---------------------------------------------------------------------------*/
#ifndef quaternion_H
#define quaternion_H
#include "scalar.H"
#include "vector.H"
#include "tensor.H"
#include "word.H"
#include "contiguous.H"
// * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * //
namespace Foam
{
// Forward declaration of friend functions and operators
class quaternion;
Istream& operator>>(Istream& is, quaternion&);
Ostream& operator<<(Ostream& os, const quaternion& C);
/*---------------------------------------------------------------------------*\
Class quaternion Declaration
\*---------------------------------------------------------------------------*/
class quaternion
{
// private data
//- Scalar part of the quaternion ( = cos(theta/2) for rotation)
scalar w_;
//- Vector part of the quaternion ( = axis of rotation)
vector v_;
//- Multiply vector v by quaternion as if v is a pure quaternion
inline quaternion mulq0v(const vector& v) const;
//- Conversion of two-axis rotation components into Euler-angles
inline static vector twoAxes
(
const scalar t11,
const scalar t12,
const scalar c2,
const scalar t31,
const scalar t32
);
//- Conversion of three-axis rotation components into Euler-angles
inline static vector threeAxes
(
const scalar t11,
const scalar t12,
const scalar s2,
const scalar t31,
const scalar t32
);
public:
//- Component type
typedef scalar cmptType;
//- Euler-angle rotation sequence
enum rotationSequence
{
ZYX, ZYZ, ZXY, ZXZ, YXZ, YXY, YZX, YZY, XYZ, XYX, XZY, XZX
};
// Member constants
//- Rank of quaternion is 1
static const direction rank = 1;
// Static data members
static const char* const typeName;
static const quaternion zero;
static const quaternion I;
// Constructors
//- Construct null
inline quaternion();
//- Construct given scalar and vector parts
inline quaternion(const scalar w, const vector& v);
//- Construct a rotation quaternion given the direction d
// and angle theta
inline quaternion(const vector& d, const scalar theta);
//- Construct a rotation quaternion given the direction d
// and cosine angle cosTheta and a if d is normalized
inline quaternion
(
const vector& d,
const scalar cosTheta,
const bool normalized
);
//- Construct given scalar part, the vector part = vector::zero
inline explicit quaternion(const scalar w);
//- Construct a pure quaternion given the vector part, scalar part = 0
inline explicit quaternion(const vector& v);
//- Construct a quaternion given the three Euler angles
inline quaternion
(
const rotationSequence rs,
const vector& angles
);
//- Construct a quaternion from a rotation tensor
inline explicit quaternion(const tensor& rotationTensor);
//- Construct from Istream
quaternion(Istream&);
// Member functions
// Access
//- Scalar part of the quaternion ( = cos(theta/2) for rotation)
inline scalar w() const;
//- Vector part of the quaternion ( = axis of rotation)
inline const vector& v() const;
//- The rotation tensor corresponding the quaternion
inline tensor R() const;
//- Return a vector of euler angles corresponding to the
// specified rotation sequence
inline vector eulerAngles(const rotationSequence rs) const;
inline quaternion normalized() const;
// Edit
//- Scalar part of the quaternion ( = cos(theta/2) for rotation)
inline scalar& w();
//- Vector part of the quaternion ( = axis of rotation)
inline vector& v();
inline void normalize();
// Transform
//- Rotate the given vector
inline vector transform(const vector& v) const;
//- Rotate the given vector anti-clockwise
inline vector invTransform(const vector& v) const;
//- Rotate the given quaternion (and normalize)
inline quaternion transform(const quaternion& q) const;
//- Rotate the given quaternion anti-clockwise (and normalize)
inline quaternion invTransform(const quaternion& q) const;
// Member operators
inline void operator=(const quaternion&);
inline void operator+=(const quaternion&);
inline void operator-=(const quaternion&);
inline void operator*=(const quaternion&);
inline void operator/=(const quaternion&);
inline void operator=(const scalar);
inline void operator=(const vector&);
inline void operator*=(const scalar);
inline void operator/=(const scalar);
// IOstream operators
friend Istream& operator>>(Istream& is, quaternion&);
friend Ostream& operator<<(Ostream& os, const quaternion& C);
};
// * * * * * * * * * * * * * * * Global Functions * * * * * * * * * * * * * //
inline scalar magSqr(const quaternion& q);
inline scalar mag(const quaternion& q);
//- Return the conjugate of the given quaternion
inline quaternion conjugate(const quaternion& q);
//- Return the normalized (unit) quaternion of the given quaternion
inline quaternion normalize(const quaternion& q);
//- Return the inverse of the given quaternion
inline quaternion inv(const quaternion& q);
//- Return a string representation of a quaternion
word name(const quaternion&);
//- Spherical linear interpolation of quaternions
quaternion slerp
(
const quaternion& qa,
const quaternion& qb,
const scalar t
);
//- Exponent of a quaternion
quaternion exp(const quaternion& q);
//- Power of a quaternion
quaternion pow(const quaternion& q, const label power);
//- Power of a quaternion
quaternion pow(const quaternion& q, const scalar power);
//- Data associated with quaternion type are contiguous
template<>
inline bool contiguous<quaternion>() {return true;}
// * * * * * * * * * * * * * * * Global Operators * * * * * * * * * * * * * //
inline bool operator==(const quaternion& q1, const quaternion& q2);
inline bool operator!=(const quaternion& q1, const quaternion& q2);
inline quaternion operator+(const quaternion& q1, const quaternion& q2);
inline quaternion operator-(const quaternion& q);
inline quaternion operator-(const quaternion& q1, const quaternion& q2);
inline scalar operator&(const quaternion& q1, const quaternion& q2);
inline quaternion operator*(const quaternion& q1, const quaternion& q2);
inline quaternion operator/(const quaternion& q1, const quaternion& q2);
inline quaternion operator*(const scalar s, const quaternion& q);
inline quaternion operator*(const quaternion& q, const scalar s);
inline quaternion operator/(const quaternion& q, const scalar s);
// * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * //
} // End namespace Foam
// * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * //
#include "quaternionI.H"
// * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * //
#endif
// ************************************************************************* //