crocoddyl  1.4.0
Contact RObot COntrol by Differential DYnamic programming Library (Crocoddyl)
multibody.hpp
1 // BSD 3-Clause License
3 //
4 // Copyright (C) 2018-2020, LAAS-CNRS, University of Edinburgh
5 // Copyright note valid unless otherwise stated in individual files.
6 // All rights reserved.
8 
9 #ifndef CROCODDYL_MULTIBODY_STATES_MULTIBODY_HPP_
10 #define CROCODDYL_MULTIBODY_STATES_MULTIBODY_HPP_
11 #include "crocoddyl/multibody/fwd.hpp"
12 #include "crocoddyl/core/state-base.hpp"
13 #include <pinocchio/multibody/model.hpp>
14 
15 namespace crocoddyl {
16 
17 template <typename _Scalar>
18 class StateMultibodyTpl : public StateAbstractTpl<_Scalar> {
19  public:
20  EIGEN_MAKE_ALIGNED_OPERATOR_NEW
21 
22  typedef _Scalar Scalar;
23  typedef MathBaseTpl<Scalar> MathBase;
24  typedef StateAbstractTpl<Scalar> Base;
25  typedef typename MathBase::VectorXs VectorXs;
26  typedef typename MathBase::MatrixXs MatrixXs;
27 
28  enum JointType { FreeFlyer = 0, Spherical, Simple };
29 
30  explicit StateMultibodyTpl(boost::shared_ptr<pinocchio::ModelTpl<Scalar> > model);
31  virtual ~StateMultibodyTpl();
32 
33  virtual VectorXs zero() const;
34  virtual VectorXs rand() const;
35  virtual void diff(const Eigen::Ref<const VectorXs>& x0, const Eigen::Ref<const VectorXs>& x1,
36  Eigen::Ref<VectorXs> dxout) const;
37  virtual void integrate(const Eigen::Ref<const VectorXs>& x, const Eigen::Ref<const VectorXs>& dx,
38  Eigen::Ref<VectorXs> xout) const;
39  virtual void Jdiff(const Eigen::Ref<const VectorXs>&, const Eigen::Ref<const VectorXs>&, Eigen::Ref<MatrixXs> Jfirst,
40  Eigen::Ref<MatrixXs> Jsecond, const Jcomponent firstsecond = both) const;
41 
42  virtual void Jintegrate(const Eigen::Ref<const VectorXs>& x, const Eigen::Ref<const VectorXs>& dx,
43  Eigen::Ref<MatrixXs> Jfirst, Eigen::Ref<MatrixXs> Jsecond,
44  const Jcomponent firstsecond = both, const AssignmentOp = setto) const;
45  virtual void JintegrateTransport(const Eigen::Ref<const VectorXs>& x, const Eigen::Ref<const VectorXs>& dx,
46  Eigen::Ref<MatrixXs> Jin, const Jcomponent firstsecond) const;
47 
48  const boost::shared_ptr<pinocchio::ModelTpl<Scalar> >& get_pinocchio() const;
49 
50  protected:
51  using Base::has_limits_;
52  using Base::lb_;
53  using Base::ndx_;
54  using Base::nq_;
55  using Base::nv_;
56  using Base::nx_;
57  using Base::ub_;
58 
59  private:
60  boost::shared_ptr<pinocchio::ModelTpl<Scalar> > pinocchio_;
61  VectorXs x0_;
62  JointType joint_type_;
63 };
64 
65 } // namespace crocoddyl
66 
67 /* --- Details -------------------------------------------------------------- */
68 /* --- Details -------------------------------------------------------------- */
69 /* --- Details -------------------------------------------------------------- */
70 #include "crocoddyl/multibody/states/multibody.hxx"
71 
72 #endif // CROCODDYL_MULTIBODY_STATES_MULTIBODY_HPP_
crocoddyl::StateAbstractTpl::lb_
VectorXs lb_
Lower state limits.
Definition: state-base.hpp:289
crocoddyl::StateAbstractTpl::has_limits_
bool has_limits_
Indicates whether any of the state limits is finite.
Definition: state-base.hpp:291
crocoddyl::StateAbstractTpl::ub_
VectorXs ub_
Upper state limits.
Definition: state-base.hpp:290
crocoddyl::StateAbstractTpl::nx_
std::size_t nx_
State dimension.
Definition: state-base.hpp:285
crocoddyl::StateAbstractTpl::nv_
std::size_t nv_
Velocity dimension.
Definition: state-base.hpp:288
crocoddyl::StateAbstractTpl::ndx_
std::size_t ndx_
State rate dimension.
Definition: state-base.hpp:286
crocoddyl::StateAbstractTpl::nq_
std::size_t nq_
Configuration dimension.
Definition: state-base.hpp:287