| Directory: | ./ |
|---|---|
| File: | include/pinocchio/multibody/liegroup/cartesian-product-variant.hpp |
| Date: | 2025-02-12 21:03:38 |
| Exec | Total | Coverage | |
|---|---|---|---|
| Lines: | 35 | 35 | 100.0% |
| Branches: | 14 | 28 | 50.0% |
| Line | Branch | Exec | Source |
|---|---|---|---|
| 1 | // | ||
| 2 | // Copyright (c) 2018 CNRS | ||
| 3 | // | ||
| 4 | |||
| 5 | #ifndef __pinocchio_cartesian_product_variant_hpp__ | ||
| 6 | #define __pinocchio_cartesian_product_variant_hpp__ | ||
| 7 | |||
| 8 | #include "pinocchio/multibody/liegroup/liegroup-base.hpp" | ||
| 9 | #include "pinocchio/multibody/liegroup/liegroup-collection.hpp" | ||
| 10 | |||
| 11 | #include "pinocchio/container/aligned-vector.hpp" | ||
| 12 | |||
| 13 | namespace pinocchio | ||
| 14 | { | ||
| 15 | |||
| 16 | template< | ||
| 17 | typename Scalar, | ||
| 18 | int Options = context::Options, | ||
| 19 | template<typename, int> class LieGroupCollectionTpl = LieGroupCollectionDefaultTpl> | ||
| 20 | struct CartesianProductOperationVariantTpl; | ||
| 21 | typedef CartesianProductOperationVariantTpl< | ||
| 22 | context::Scalar, | ||
| 23 | context::Options, | ||
| 24 | LieGroupCollectionDefaultTpl> | ||
| 25 | CartesianProductOperationVariant; | ||
| 26 | |||
| 27 | template<typename _Scalar, int _Options, template<typename, int> class LieGroupCollectionTpl> | ||
| 28 | struct traits<CartesianProductOperationVariantTpl<_Scalar, _Options, LieGroupCollectionTpl>> | ||
| 29 | { | ||
| 30 | typedef _Scalar Scalar; | ||
| 31 | enum | ||
| 32 | { | ||
| 33 | Options = _Options, | ||
| 34 | NQ = Eigen::Dynamic, | ||
| 35 | NV = Eigen::Dynamic | ||
| 36 | }; | ||
| 37 | }; | ||
| 38 | |||
| 39 | /// | ||
| 40 | /// \brief Dynamic Cartesian product composed of elementary Lie groups defined in LieGroupVariant | ||
| 41 | /// | ||
| 42 | template<typename _Scalar, int _Options, template<typename, int> class LieGroupCollectionTpl> | ||
| 43 | struct CartesianProductOperationVariantTpl | ||
| 44 | : public LieGroupBase< | ||
| 45 | CartesianProductOperationVariantTpl<_Scalar, _Options, LieGroupCollectionTpl>> | ||
| 46 | { | ||
| 47 | EIGEN_MAKE_ALIGNED_OPERATOR_NEW | ||
| 48 | |||
| 49 | PINOCCHIO_LIE_GROUP_TPL_PUBLIC_INTERFACE(CartesianProductOperationVariantTpl); | ||
| 50 | |||
| 51 | typedef LieGroupCollectionTpl<Scalar, Options> LieGroupCollection; | ||
| 52 | typedef typename LieGroupCollection::LieGroupVariant LieGroupVariant; | ||
| 53 | typedef LieGroupGenericTpl<LieGroupCollection> LieGroupGeneric; | ||
| 54 | |||
| 55 | /// \brief Default constructor | ||
| 56 | 13 | CartesianProductOperationVariantTpl() | |
| 57 | 13 | : m_nq(0) | |
| 58 | 13 | , m_nv(0) | |
| 59 |
1/2✓ Branch 2 taken 13 times.
✗ Branch 3 not taken.
|
13 | , lg_nqs(0) |
| 60 |
1/2✓ Branch 2 taken 13 times.
✗ Branch 3 not taken.
|
13 | , lg_nvs(0) |
| 61 |
1/2✓ Branch 4 taken 13 times.
✗ Branch 5 not taken.
|
26 | , m_neutral(0) {}; |
| 62 | |||
| 63 | /// | ||
| 64 | /// \brief Constructor with one single Lie group | ||
| 65 | /// | ||
| 66 | /// \param[in] lg Lie group variant to insert inside the Cartesian product | ||
| 67 | /// | ||
| 68 | 40 | explicit CartesianProductOperationVariantTpl(const LieGroupGeneric & lg) | |
| 69 | 40 | : m_nq(0) | |
| 70 | 40 | , m_nv(0) | |
| 71 |
1/2✓ Branch 2 taken 40 times.
✗ Branch 3 not taken.
|
40 | , lg_nqs(0) |
| 72 |
1/2✓ Branch 2 taken 40 times.
✗ Branch 3 not taken.
|
40 | , lg_nvs(0) |
| 73 |
1/2✓ Branch 4 taken 40 times.
✗ Branch 5 not taken.
|
80 | , m_neutral(0) |
| 74 | { | ||
| 75 |
1/2✓ Branch 1 taken 40 times.
✗ Branch 2 not taken.
|
40 | append(lg); |
| 76 | 40 | }; | |
| 77 | |||
| 78 | /// | ||
| 79 | /// \brief Constructor with two Lie groups | ||
| 80 | /// | ||
| 81 | /// \param[in] lg1 Lie group variant to insert inside the Cartesian product | ||
| 82 | /// \param[in] lg2 Lie group variant to insert inside the Cartesian product | ||
| 83 | /// | ||
| 84 | 1 | CartesianProductOperationVariantTpl(const LieGroupGeneric & lg1, const LieGroupGeneric & lg2) | |
| 85 | 1 | : m_nq(0) | |
| 86 | 1 | , m_nv(0) | |
| 87 |
1/2✓ Branch 2 taken 1 times.
✗ Branch 3 not taken.
|
1 | , lg_nqs(0) |
| 88 |
1/2✓ Branch 2 taken 1 times.
✗ Branch 3 not taken.
|
1 | , lg_nvs(0) |
| 89 |
1/2✓ Branch 4 taken 1 times.
✗ Branch 5 not taken.
|
2 | , m_neutral(0) |
| 90 | { | ||
| 91 |
1/2✓ Branch 1 taken 1 times.
✗ Branch 2 not taken.
|
1 | append(lg1); |
| 92 |
1/2✓ Branch 1 taken 1 times.
✗ Branch 2 not taken.
|
1 | append(lg2); |
| 93 | 1 | }; | |
| 94 | |||
| 95 | /// | ||
| 96 | /// \brief Append a Lie group to the Cartesian product | ||
| 97 | /// | ||
| 98 | /// \param[in] lg Lie group variant to insert inside the Cartesian product | ||
| 99 | /// | ||
| 100 | void append(const LieGroupGeneric & lg); | ||
| 101 | |||
| 102 | /// | ||
| 103 | /// \brief Append a Lie group to the Cartesian product | ||
| 104 | /// | ||
| 105 | /// \param[in] lg Lie group to insert inside the Cartesian product | ||
| 106 | /// | ||
| 107 | template<typename LieGroupDerived> | ||
| 108 | 16 | void append(const LieGroupBase<LieGroupDerived> & lg) | |
| 109 | { | ||
| 110 |
1/2✓ Branch 1 taken 8 times.
✗ Branch 2 not taken.
|
16 | LieGroupGeneric lgGeneric(lg); |
| 111 |
1/2✓ Branch 1 taken 8 times.
✗ Branch 2 not taken.
|
16 | append(lgGeneric); |
| 112 | 16 | } | |
| 113 | |||
| 114 | /// | ||
| 115 | /// \brief Cartesian product between *this and other. | ||
| 116 | /// | ||
| 117 | /// \param[in] other CartesianProductOperation to compose with this | ||
| 118 | /// | ||
| 119 | /// \returns A new Cartesian product betwenn *this and other. | ||
| 120 | /// | ||
| 121 | CartesianProductOperationVariantTpl | ||
| 122 | operator*(const CartesianProductOperationVariantTpl & other) const; | ||
| 123 | |||
| 124 | /// | ||
| 125 | /// \brief Append other to *this. | ||
| 126 | /// | ||
| 127 | /// \param[in] other CartesianProductOperation to append to *this. | ||
| 128 | /// | ||
| 129 | CartesianProductOperationVariantTpl & | ||
| 130 | operator*=(const CartesianProductOperationVariantTpl & other); | ||
| 131 | |||
| 132 | /// | ||
| 133 | /// \brief Append a Lie group to *this. | ||
| 134 | /// | ||
| 135 | /// \param[in] lg LieGroupGeneric to append to *this. | ||
| 136 | /// | ||
| 137 | inline CartesianProductOperationVariantTpl & operator*=(const LieGroupGeneric & lg) | ||
| 138 | { | ||
| 139 | append(lg); | ||
| 140 | return *this; | ||
| 141 | } | ||
| 142 | |||
| 143 | /// | ||
| 144 | /// \brief Append a Lie group to *this. | ||
| 145 | /// | ||
| 146 | /// \param[in] lg LieGroupGeneric to append to *this. | ||
| 147 | /// | ||
| 148 | template<typename LieGroupDerived> | ||
| 149 | inline CartesianProductOperationVariantTpl & | ||
| 150 | operator*=(const LieGroupBase<LieGroupDerived> & lg) | ||
| 151 | { | ||
| 152 | append<LieGroupDerived>(lg); | ||
| 153 | return *this; | ||
| 154 | } | ||
| 155 | |||
| 156 | 66 | int nq() const | |
| 157 | { | ||
| 158 | 66 | return m_nq; | |
| 159 | } | ||
| 160 | 360 | int nv() const | |
| 161 | { | ||
| 162 | 360 | return m_nv; | |
| 163 | } | ||
| 164 | |||
| 165 | 16 | std::string name() const | |
| 166 | { | ||
| 167 | 16 | return m_name; | |
| 168 | } | ||
| 169 | |||
| 170 | 9 | ConfigVector_t neutral() const | |
| 171 | { | ||
| 172 | 9 | return m_neutral; | |
| 173 | } | ||
| 174 | |||
| 175 | template<class ConfigL_t, class ConfigR_t, class Tangent_t> | ||
| 176 | void difference_impl( | ||
| 177 | const Eigen::MatrixBase<ConfigL_t> & q0, | ||
| 178 | const Eigen::MatrixBase<ConfigR_t> & q1, | ||
| 179 | const Eigen::MatrixBase<Tangent_t> & d) const; | ||
| 180 | |||
| 181 | template<ArgumentPosition arg, class ConfigL_t, class ConfigR_t, class JacobianOut_t> | ||
| 182 | void dDifference_impl( | ||
| 183 | const Eigen::MatrixBase<ConfigL_t> & q0, | ||
| 184 | const Eigen::MatrixBase<ConfigR_t> & q1, | ||
| 185 | const Eigen::MatrixBase<JacobianOut_t> & J) const; | ||
| 186 | |||
| 187 | template< | ||
| 188 | ArgumentPosition arg, | ||
| 189 | class ConfigL_t, | ||
| 190 | class ConfigR_t, | ||
| 191 | class JacobianIn_t, | ||
| 192 | class JacobianOut_t> | ||
| 193 | void dDifference_product_impl( | ||
| 194 | const ConfigL_t & q0, | ||
| 195 | const ConfigR_t & q1, | ||
| 196 | const JacobianIn_t & Jin, | ||
| 197 | JacobianOut_t & Jout, | ||
| 198 | bool dDifferenceOnTheLeft, | ||
| 199 | const AssignmentOperatorType op) const; | ||
| 200 | |||
| 201 | template<class ConfigIn_t, class Velocity_t, class ConfigOut_t> | ||
| 202 | void integrate_impl( | ||
| 203 | const Eigen::MatrixBase<ConfigIn_t> & q, | ||
| 204 | const Eigen::MatrixBase<Velocity_t> & v, | ||
| 205 | const Eigen::MatrixBase<ConfigOut_t> & qout) const; | ||
| 206 | |||
| 207 | template<class Config_t, class Jacobian_t> | ||
| 208 | void integrateCoeffWiseJacobian_impl( | ||
| 209 | const Eigen::MatrixBase<Config_t> & q, const Eigen::MatrixBase<Jacobian_t> & J) const; | ||
| 210 | |||
| 211 | template<class Config_t, class Tangent_t, class JacobianOut_t> | ||
| 212 | void dIntegrate_dq_impl( | ||
| 213 | const Eigen::MatrixBase<Config_t> & q, | ||
| 214 | const Eigen::MatrixBase<Tangent_t> & v, | ||
| 215 | const Eigen::MatrixBase<JacobianOut_t> & J, | ||
| 216 | const AssignmentOperatorType op = SETTO) const; | ||
| 217 | |||
| 218 | template<class Config_t, class Tangent_t, class JacobianOut_t> | ||
| 219 | void dIntegrate_dv_impl( | ||
| 220 | const Eigen::MatrixBase<Config_t> & q, | ||
| 221 | const Eigen::MatrixBase<Tangent_t> & v, | ||
| 222 | const Eigen::MatrixBase<JacobianOut_t> & J, | ||
| 223 | const AssignmentOperatorType op = SETTO) const; | ||
| 224 | |||
| 225 | template<class Config_t, class Tangent_t, class JacobianIn_t, class JacobianOut_t> | ||
| 226 | void dIntegrate_product_impl( | ||
| 227 | const Config_t & q, | ||
| 228 | const Tangent_t & v, | ||
| 229 | const JacobianIn_t & Jin, | ||
| 230 | JacobianOut_t & Jout, | ||
| 231 | bool dIntegrateOnTheLeft, | ||
| 232 | const ArgumentPosition arg, | ||
| 233 | const AssignmentOperatorType op) const; | ||
| 234 | |||
| 235 | template<class Config_t, class Tangent_t, class JacobianIn_t, class JacobianOut_t> | ||
| 236 | void dIntegrateTransport_dq_impl( | ||
| 237 | const Eigen::MatrixBase<Config_t> & q, | ||
| 238 | const Eigen::MatrixBase<Tangent_t> & v, | ||
| 239 | const Eigen::MatrixBase<JacobianIn_t> & J_in, | ||
| 240 | const Eigen::MatrixBase<JacobianOut_t> & J_out) const; | ||
| 241 | |||
| 242 | template<class Config_t, class Tangent_t, class JacobianIn_t, class JacobianOut_t> | ||
| 243 | void dIntegrateTransport_dv_impl( | ||
| 244 | const Eigen::MatrixBase<Config_t> & q, | ||
| 245 | const Eigen::MatrixBase<Tangent_t> & v, | ||
| 246 | const Eigen::MatrixBase<JacobianIn_t> & J_in, | ||
| 247 | const Eigen::MatrixBase<JacobianOut_t> & J_out) const; | ||
| 248 | |||
| 249 | template<class Config_t, class Tangent_t, class JacobianOut_t> | ||
| 250 | void dIntegrateTransport_dq_impl( | ||
| 251 | const Eigen::MatrixBase<Config_t> & q, | ||
| 252 | const Eigen::MatrixBase<Tangent_t> & v, | ||
| 253 | const Eigen::MatrixBase<JacobianOut_t> & J) const; | ||
| 254 | |||
| 255 | template<class Config_t, class Tangent_t, class JacobianOut_t> | ||
| 256 | void dIntegrateTransport_dv_impl( | ||
| 257 | const Eigen::MatrixBase<Config_t> & q, | ||
| 258 | const Eigen::MatrixBase<Tangent_t> & v, | ||
| 259 | const Eigen::MatrixBase<JacobianOut_t> & J) const; | ||
| 260 | |||
| 261 | template<class ConfigL_t, class ConfigR_t> | ||
| 262 | Scalar squaredDistance_impl( | ||
| 263 | const Eigen::MatrixBase<ConfigL_t> & q0, const Eigen::MatrixBase<ConfigR_t> & q1) const; | ||
| 264 | |||
| 265 | template<class Config_t> | ||
| 266 | void normalize_impl(const Eigen::MatrixBase<Config_t> & qout) const; | ||
| 267 | |||
| 268 | template<class Config_t> | ||
| 269 | bool isNormalized_impl(const Eigen::MatrixBase<Config_t> & qout, const Scalar & prec) const; | ||
| 270 | |||
| 271 | template<class Config_t> | ||
| 272 | void random_impl(const Eigen::MatrixBase<Config_t> & qout) const; | ||
| 273 | |||
| 274 | template<class ConfigL_t, class ConfigR_t, class ConfigOut_t> | ||
| 275 | void randomConfiguration_impl( | ||
| 276 | const Eigen::MatrixBase<ConfigL_t> & lower, | ||
| 277 | const Eigen::MatrixBase<ConfigR_t> & upper, | ||
| 278 | const Eigen::MatrixBase<ConfigOut_t> & qout) const; | ||
| 279 | |||
| 280 | template<class ConfigL_t, class ConfigR_t> | ||
| 281 | bool isSameConfiguration_impl( | ||
| 282 | const Eigen::MatrixBase<ConfigL_t> & q0, | ||
| 283 | const Eigen::MatrixBase<ConfigR_t> & q1, | ||
| 284 | const Scalar & prec) const; | ||
| 285 | |||
| 286 | bool isEqual_impl(const CartesianProductOperationVariantTpl & other) const; | ||
| 287 | |||
| 288 | template<typename LieGroup1, typename LieGroup2> | ||
| 289 | bool isEqual(const CartesianProductOperation<LieGroup1, LieGroup2> & other) const; | ||
| 290 | |||
| 291 | protected: | ||
| 292 | PINOCCHIO_ALIGNED_STD_VECTOR(LieGroupGeneric) liegroups; | ||
| 293 | Index m_nq, m_nv; | ||
| 294 | std::vector<Index> lg_nqs, lg_nvs; | ||
| 295 | std::string m_name; | ||
| 296 | |||
| 297 | ConfigVector_t m_neutral; | ||
| 298 | }; | ||
| 299 | |||
| 300 | } // namespace pinocchio | ||
| 301 | |||
| 302 | #include <pinocchio/multibody/liegroup/cartesian-product-variant.hxx> | ||
| 303 | |||
| 304 | #endif // ifndef __pinocchio_cartesian_product_variant_hpp__ | ||
| 305 |