| Directory: | ./ |
|---|---|
| File: | include/pinocchio/spatial/motion-tpl.hpp |
| Date: | 2025-02-12 21:03:38 |
| Exec | Total | Coverage | |
|---|---|---|---|
| Lines: | 68 | 70 | 97.1% |
| Branches: | 23 | 47 | 48.9% |
| Line | Branch | Exec | Source |
|---|---|---|---|
| 1 | // | ||
| 2 | // Copyright (c) 2015-2018 CNRS | ||
| 3 | // Copyright (c) 2015-2016 Wandercraft, 86 rue de Paris 91400 Orsay, France. | ||
| 4 | // | ||
| 5 | |||
| 6 | #ifndef __pinocchio_spatial_motion_tpl_hpp__ | ||
| 7 | #define __pinocchio_spatial_motion_tpl_hpp__ | ||
| 8 | |||
| 9 | namespace pinocchio | ||
| 10 | { | ||
| 11 | template<typename _Scalar, int _Options> | ||
| 12 | struct traits<MotionTpl<_Scalar, _Options>> | ||
| 13 | { | ||
| 14 | typedef _Scalar Scalar; | ||
| 15 | typedef Eigen::Matrix<Scalar, 3, 1, _Options> Vector3; | ||
| 16 | typedef Eigen::Matrix<Scalar, 6, 1, _Options> Vector6; | ||
| 17 | typedef Eigen::Matrix<Scalar, 4, 4, _Options> Matrix4; | ||
| 18 | typedef Eigen::Matrix<Scalar, 6, 6, _Options> Matrix6; | ||
| 19 | typedef Matrix6 ActionMatrixType; | ||
| 20 | typedef Matrix4 HomogeneousMatrixType; | ||
| 21 | typedef typename PINOCCHIO_EIGEN_REF_CONST_TYPE(Vector6) ToVectorConstReturnType; | ||
| 22 | typedef typename PINOCCHIO_EIGEN_REF_TYPE(Vector6) ToVectorReturnType; | ||
| 23 | typedef typename Vector6::template FixedSegmentReturnType<3>::Type LinearType; | ||
| 24 | typedef typename Vector6::template FixedSegmentReturnType<3>::Type AngularType; | ||
| 25 | typedef typename Vector6::template ConstFixedSegmentReturnType<3>::Type ConstLinearType; | ||
| 26 | typedef typename Vector6::template ConstFixedSegmentReturnType<3>::Type ConstAngularType; | ||
| 27 | typedef MotionTpl<Scalar, _Options> MotionPlain; | ||
| 28 | typedef const MotionPlain & PlainReturnType; | ||
| 29 | enum | ||
| 30 | { | ||
| 31 | LINEAR = 0, | ||
| 32 | ANGULAR = 3, | ||
| 33 | Options = _Options | ||
| 34 | }; | ||
| 35 | |||
| 36 | typedef MotionRef<Vector6> MotionRefType; | ||
| 37 | }; // traits MotionTpl | ||
| 38 | |||
| 39 | template<typename _Scalar, int _Options> | ||
| 40 | class MotionTpl : public MotionDense<MotionTpl<_Scalar, _Options>> | ||
| 41 | { | ||
| 42 | public: | ||
| 43 | EIGEN_MAKE_ALIGNED_OPERATOR_NEW | ||
| 44 | typedef MotionDense<MotionTpl> Base; | ||
| 45 | MOTION_TYPEDEF_TPL(MotionTpl); | ||
| 46 | enum | ||
| 47 | { | ||
| 48 | Options = _Options | ||
| 49 | }; | ||
| 50 | |||
| 51 | using Base::operator=; | ||
| 52 | using Base::angular; | ||
| 53 | using Base::linear; | ||
| 54 | |||
| 55 | using Base::__mequ__; | ||
| 56 | using Base::__minus__; | ||
| 57 | using Base::__mult__; | ||
| 58 | using Base::__opposite__; | ||
| 59 | using Base::__pequ__; | ||
| 60 | using Base::__plus__; | ||
| 61 | |||
| 62 | // Constructors | ||
| 63 | 404165 | MotionTpl() | |
| 64 | 404165 | { | |
| 65 | 404165 | } | |
| 66 | |||
| 67 | template<typename V1, typename V2> | ||
| 68 | 22281 | MotionTpl(const Eigen::MatrixBase<V1> & v, const Eigen::MatrixBase<V2> & w) | |
| 69 | 22281 | { | |
| 70 | EIGEN_STATIC_ASSERT_VECTOR_ONLY(V1); | ||
| 71 | EIGEN_STATIC_ASSERT_VECTOR_ONLY(V2); | ||
| 72 |
3/5✓ Branch 1 taken 158 times.
✓ Branch 2 taken 11137 times.
✗ Branch 3 not taken.
✓ Branch 4 taken 158 times.
✗ Branch 5 not taken.
|
22281 | linear() = v; |
| 73 |
3/5✓ Branch 1 taken 158 times.
✓ Branch 2 taken 11137 times.
✗ Branch 3 not taken.
✓ Branch 4 taken 158 times.
✗ Branch 5 not taken.
|
22281 | angular() = w; |
| 74 | 22281 | } | |
| 75 | |||
| 76 | template<typename V6> | ||
| 77 | 767887 | explicit MotionTpl(const Eigen::MatrixBase<V6> & v) | |
| 78 | 767887 | : m_data(v) | |
| 79 | { | ||
| 80 | EIGEN_STATIC_ASSERT_VECTOR_ONLY(V6); | ||
| 81 | 767887 | } | |
| 82 | |||
| 83 | 579104 | MotionTpl(const MotionTpl & other) | |
| 84 | 579104 | { | |
| 85 |
1/2✓ Branch 1 taken 4543 times.
✗ Branch 2 not taken.
|
579104 | *this = other; |
| 86 | 579104 | } | |
| 87 | |||
| 88 | template<typename S2, int O2> | ||
| 89 | 1 | explicit MotionTpl(const MotionTpl<S2, O2> & other) | |
| 90 | 1 | { | |
| 91 |
1/2✓ Branch 2 taken 1 times.
✗ Branch 3 not taken.
|
1 | *this = other.template cast<Scalar>(); |
| 92 | 1 | } | |
| 93 | |||
| 94 | template<int O2> | ||
| 95 | explicit MotionTpl(const MotionTpl<Scalar, O2> & clone) | ||
| 96 | : m_data(clone.toVector()) | ||
| 97 | { | ||
| 98 | } | ||
| 99 | |||
| 100 | // Same explanation as converting constructor from MotionBase | ||
| 101 | template< | ||
| 102 | typename M2, | ||
| 103 | typename std::enable_if<!std::is_convertible<MotionDense<M2>, MotionTpl>::value, bool>::type = | ||
| 104 | true> | ||
| 105 | explicit MotionTpl(const MotionDense<M2> & clone) | ||
| 106 | { | ||
| 107 | linear() = clone.linear(); | ||
| 108 | angular() = clone.angular(); | ||
| 109 | } | ||
| 110 | |||
| 111 | // MotionBase implement a conversion function to PlainReturnType. | ||
| 112 | // Usually, PlainReturnType is defined as MotionTpl. | ||
| 113 | // In this case, this converting constructor is redundant and | ||
| 114 | // create a warning with -Wconversion | ||
| 115 | template< | ||
| 116 | typename M2, | ||
| 117 | typename std::enable_if<!std::is_convertible<MotionBase<M2>, MotionTpl>::value, bool>::type = | ||
| 118 | true> | ||
| 119 | ✗ | explicit MotionTpl(const MotionBase<M2> & clone) | |
| 120 | { | ||
| 121 | ✗ | *this = clone; | |
| 122 | } | ||
| 123 | |||
| 124 | /// | ||
| 125 | /// \brief Copy assignment operator. | ||
| 126 | /// | ||
| 127 | /// \param[in] other MotionTpl to copy | ||
| 128 | /// | ||
| 129 | 790040 | MotionTpl & operator=(const MotionTpl & clone) // Copy assignment operator | |
| 130 | { | ||
| 131 | 790040 | m_data = clone.toVector(); | |
| 132 | 790040 | return *this; | |
| 133 | } | ||
| 134 | |||
| 135 | // initializers | ||
| 136 | 214594 | static MotionTpl Zero() | |
| 137 | { | ||
| 138 |
1/2✓ Branch 2 taken 214588 times.
✗ Branch 3 not taken.
|
214594 | return MotionTpl(Vector6::Zero()); |
| 139 | } | ||
| 140 | 114 | static MotionTpl Random() | |
| 141 | { | ||
| 142 |
1/2✓ Branch 2 taken 114 times.
✗ Branch 3 not taken.
|
114 | return MotionTpl(Vector6::Random()); |
| 143 | } | ||
| 144 | |||
| 145 | 30888 | inline PlainReturnType plain() const | |
| 146 | { | ||
| 147 | 30888 | return *this; | |
| 148 | } | ||
| 149 | |||
| 150 | 1245512 | ToVectorConstReturnType toVector_impl() const | |
| 151 | { | ||
| 152 | 1245512 | return m_data; | |
| 153 | } | ||
| 154 | 200252 | ToVectorReturnType toVector_impl() | |
| 155 | { | ||
| 156 | 200814 | return m_data; | |
| 157 | } | ||
| 158 | |||
| 159 | // Getters | ||
| 160 | 1265209 | ConstAngularType angular_impl() const | |
| 161 | { | ||
| 162 | 1265209 | return m_data.template segment<3>(ANGULAR); | |
| 163 | } | ||
| 164 | 775130 | ConstLinearType linear_impl() const | |
| 165 | { | ||
| 166 | 777745 | return m_data.template segment<3>(LINEAR); | |
| 167 | } | ||
| 168 | 812827 | AngularType angular_impl() | |
| 169 | { | ||
| 170 | 815258 | return m_data.template segment<3>(ANGULAR); | |
| 171 | } | ||
| 172 | 571835 | LinearType linear_impl() | |
| 173 | { | ||
| 174 | 573445 | return m_data.template segment<3>(LINEAR); | |
| 175 | } | ||
| 176 | |||
| 177 | template<typename V3> | ||
| 178 | 1 | void angular_impl(const Eigen::MatrixBase<V3> & w) | |
| 179 | { | ||
| 180 | EIGEN_STATIC_ASSERT_VECTOR_SPECIFIC_SIZE(V3, 3); | ||
| 181 |
1/2✓ Branch 2 taken 1 times.
✗ Branch 3 not taken.
|
1 | angular_impl() = w; |
| 182 | 1 | } | |
| 183 | template<typename V3> | ||
| 184 | 6 | void linear_impl(const Eigen::MatrixBase<V3> & v) | |
| 185 | { | ||
| 186 | EIGEN_STATIC_ASSERT_VECTOR_SPECIFIC_SIZE(V3, 3); | ||
| 187 |
1/2✓ Branch 2 taken 6 times.
✗ Branch 3 not taken.
|
6 | linear_impl() = v; |
| 188 | 6 | } | |
| 189 | |||
| 190 | // Specific operators for MotionTpl and MotionRef | ||
| 191 | template<int O2> | ||
| 192 | 102003 | MotionPlain __plus__(const MotionTpl<Scalar, O2> & v) const | |
| 193 | { | ||
| 194 |
1/2✓ Branch 3 taken 101949 times.
✗ Branch 4 not taken.
|
102003 | return MotionPlain(m_data + v.toVector()); |
| 195 | } | ||
| 196 | |||
| 197 | template<typename Vector6ArgType> | ||
| 198 | 52 | MotionPlain __plus__(const MotionRef<Vector6ArgType> & v) const | |
| 199 | { | ||
| 200 |
3/5✓ Branch 2 taken 51 times.
✓ Branch 3 taken 1 times.
✗ Branch 4 not taken.
✓ Branch 5 taken 51 times.
✗ Branch 6 not taken.
|
52 | return MotionPlain(m_data + v.toVector()); |
| 201 | } | ||
| 202 | |||
| 203 | template<int O2> | ||
| 204 | 47745 | MotionPlain __minus__(const MotionTpl<Scalar, O2> & v) const | |
| 205 | { | ||
| 206 |
1/2✓ Branch 3 taken 47745 times.
✗ Branch 4 not taken.
|
47745 | return MotionPlain(m_data - v.toVector()); |
| 207 | } | ||
| 208 | |||
| 209 | template<typename Vector6ArgType> | ||
| 210 | MotionPlain __minus__(const MotionRef<Vector6ArgType> & v) const | ||
| 211 | { | ||
| 212 | return MotionPlain(m_data - v.toVector()); | ||
| 213 | } | ||
| 214 | |||
| 215 | template<int O2> | ||
| 216 | 243677 | MotionTpl & __pequ__(const MotionTpl<Scalar, O2> & v) | |
| 217 | { | ||
| 218 | 243677 | m_data += v.toVector(); | |
| 219 | 243677 | return *this; | |
| 220 | } | ||
| 221 | |||
| 222 | template<typename Vector6ArgType> | ||
| 223 | 782 | MotionTpl & __pequ__(const MotionRef<Vector6ArgType> & v) | |
| 224 | { | ||
| 225 |
1/2✓ Branch 2 taken 775 times.
✗ Branch 3 not taken.
|
782 | m_data += v.toVector(); |
| 226 | 782 | return *this; | |
| 227 | } | ||
| 228 | |||
| 229 | template<int O2> | ||
| 230 | 373 | MotionTpl & __mequ__(const MotionTpl<Scalar, O2> & v) | |
| 231 | { | ||
| 232 | 373 | m_data -= v.toVector(); | |
| 233 | 373 | return *this; | |
| 234 | } | ||
| 235 | |||
| 236 | template<typename Vector6ArgType> | ||
| 237 | MotionTpl & __mequ__(const MotionRef<Vector6ArgType> & v) | ||
| 238 | { | ||
| 239 | m_data -= v.toVector(); | ||
| 240 | return *this; | ||
| 241 | } | ||
| 242 | |||
| 243 | template<typename OtherScalar> | ||
| 244 | 30294 | MotionPlain __mult__(const OtherScalar & alpha) const | |
| 245 | { | ||
| 246 |
1/2✓ Branch 2 taken 30294 times.
✗ Branch 3 not taken.
|
30294 | return MotionPlain(alpha * m_data); |
| 247 | } | ||
| 248 | |||
| 249 | 2 | MotionRef<Vector6> ref() | |
| 250 | { | ||
| 251 | 2 | return MotionRef<Vector6>(m_data); | |
| 252 | } | ||
| 253 | |||
| 254 | /// \returns An expression of *this with the Scalar type casted to NewScalar. | ||
| 255 | template<typename NewScalar> | ||
| 256 | 29 | MotionTpl<NewScalar, Options> cast() const | |
| 257 | { | ||
| 258 | typedef MotionTpl<NewScalar, Options> ReturnType; | ||
| 259 |
4/10✓ Branch 2 taken 16 times.
✗ Branch 3 not taken.
✗ Branch 4 not taken.
✓ Branch 5 taken 16 times.
✗ Branch 6 not taken.
✗ Branch 7 not taken.
✓ Branch 8 taken 16 times.
✗ Branch 9 not taken.
✓ Branch 11 taken 16 times.
✗ Branch 12 not taken.
|
29 | ReturnType res(linear().template cast<NewScalar>(), angular().template cast<NewScalar>()); |
| 260 | 29 | return res; | |
| 261 | } | ||
| 262 | |||
| 263 | protected: | ||
| 264 | Vector6 m_data; | ||
| 265 | |||
| 266 | }; // class MotionTpl | ||
| 267 | |||
| 268 | } // namespace pinocchio | ||
| 269 | |||
| 270 | #endif // ifndef __pinocchio_spatial_motion_tpl_hpp__ | ||
| 271 |