fly-high.hpp

Go to the documentation of this file.

#pragma once
 
#include "aligator/core/unary-function.hpp"
#include "./fwd.hpp"
#include <proxsuite-nlp/modelling/spaces/multibody.hpp>
 
namespace aligator {
 
template <typename _Scalar>
struct FlyHighResidualTpl : UnaryFunctionTpl<_Scalar>, frame_api {
  using Scalar = _Scalar;
  ALIGATOR_DYNAMIC_TYPEDEFS(Scalar);
  static constexpr int NR = 2;
 
  using Base = UnaryFunctionTpl<Scalar>;
  using BaseData = StageFunctionDataTpl<Scalar>;
  using Model = pinocchio::ModelTpl<Scalar>;
 
  struct Data;
 
  FlyHighResidualTpl(const int ndx, const Model &model,
                     const pinocchio::FrameIndex frame_id, Scalar slope,
                     int nu);
 
  void evaluate(const ConstVectorRef &x, BaseData &data) const;
  void computeJacobians(const ConstVectorRef &x, BaseData &data) const;
 
  shared_ptr<BaseData> createData() const {
    return std::make_shared<Data>(*this);
  }
 
  const auto &getModel() const { return pin_model_; }
 
  Scalar slope_;
 
private:
  Model pin_model_;
};
 
template <typename Scalar>
using FlyHighResidualDataTpl = typename FlyHighResidualTpl<Scalar>::Data;
 
template <typename Scalar>
struct FlyHighResidualTpl<Scalar>::Data : StageFunctionDataTpl<Scalar> {
  EIGEN_MAKE_ALIGNED_OPERATOR_NEW
  using BaseData::ndx1;
  using BaseData::nr;
  using BaseData::nu;
 
  Data(FlyHighResidualTpl const &model)
      : BaseData(model.ndx1, model.nu, model.nr), pdata_(model.pin_model_),
        d_dq(6, model.pin_model_.nv), d_dv(6, model.pin_model_.nv),
        l_dnu_dq(6, model.pin_model_.nv), l_dnu_dv(6, model.pin_model_.nv),
        o_dv_dq(3, model.pin_model_.nv), o_dv_dv(3, model.pin_model_.nv),
        vxJ(3, model.pin_model_.nv) {
    d_dq.setZero();
    d_dv.setZero();
    l_dnu_dq.setZero();
    l_dnu_dv.setZero();
    o_dv_dq.setZero();
    o_dv_dv.setZero();
    vxJ.setZero();
  }
 
  pinocchio::DataTpl<Scalar> pdata_;
  Matrix6Xs d_dq, d_dv;
  Matrix6Xs l_dnu_dq, l_dnu_dv;
  Matrix3Xs o_dv_dq, o_dv_dv, vxJ;
  Scalar ez;
};
 
} // namespace aligator
 
#ifdef ALIGATOR_ENABLE_TEMPLATE_INSTANTIATION
#include "aligator/modelling/multibody/fly-high.txx"
#endif