explicit-dynamics-direct-sum.hpp

Go to the documentation of this file.

 
#pragma once
 
#include "aligator/core/explicit-dynamics.hpp"
#include <proxsuite-nlp/modelling/spaces/cartesian-product.hpp>
 
namespace aligator {
 
template <typename _Scalar>
struct DirectSumExplicitDynamicsTpl : ExplicitDynamicsModelTpl<_Scalar> {
  using Scalar = _Scalar;
  ALIGATOR_DYNAMIC_TYPEDEFS(Scalar);
  using Base = ExplicitDynamicsModelTpl<Scalar>;
  using Manifold = ManifoldAbstractTpl<Scalar>;
  using CartesianProduct = proxsuite::nlp::CartesianProductTpl<Scalar>;
  using BaseData = ExplicitDynamicsDataTpl<Scalar>;
 
  struct Data;
 
  DirectSumExplicitDynamicsTpl(xyz::polymorphic<Base> f,
                               xyz::polymorphic<Base> g);
 
  void forward(const ConstVectorRef &x, const ConstVectorRef &u,
               BaseData &data) const override;
 
  void dForward(const ConstVectorRef &x, const ConstVectorRef &u,
                BaseData &data) const override;
 
  shared_ptr<DynamicsDataTpl<Scalar>> createData() const override {
    return std::make_shared<Data>(*this);
  }
 
  xyz::polymorphic<Base> f_, g_;
 
private:
  CartesianProduct product_space_;
};
 
template <typename Scalar>
struct DirectSumExplicitDynamicsTpl<Scalar>::Data : BaseData {
  shared_ptr<BaseData> data1_, data2_;
  Data(DirectSumExplicitDynamicsTpl const &model);
};
 
template <typename Scalar>
auto directSum(xyz::polymorphic<ExplicitDynamicsModelTpl<Scalar>> const &m1,
               xyz::polymorphic<ExplicitDynamicsModelTpl<Scalar>> const &m2) {
  return DirectSumExplicitDynamicsTpl<Scalar>(m1, m2);
}
 
} // namespace aligator
 
#ifdef ALIGATOR_ENABLE_TEMPLATE_INSTANTIATION
#include "./explicit-dynamics-direct-sum.txx"
#endif