com_cost.hpp

Go to the documentation of this file.

#ifndef ROBOTOC_COM_COST_HPP_
#define ROBOTOC_COM_COST_HPP_
 
#include "Eigen/Core"
 
#include "robotoc/robot/robot.hpp"
#include "robotoc/robot/contact_status.hpp"
#include "robotoc/robot/impact_status.hpp"
#include "robotoc/cost/cost_function_component_base.hpp"
#include "robotoc/cost/cost_function_data.hpp"
#include "robotoc/core/split_solution.hpp"
#include "robotoc/core/split_kkt_residual.hpp"
#include "robotoc/core/split_kkt_matrix.hpp"
#include "robotoc/core/split_solution.hpp"
#include "robotoc/cost/com_ref_base.hpp"
 
 
namespace robotoc {
 
class CoMCost final : public CostFunctionComponentBase {
public:
  CoMCost(const Robot& robot);
 
  CoMCost(const Robot& robot, const std::shared_ptr<CoMRefBase>& ref);
 
  CoMCost(const Robot& robot, const Eigen::Vector3d& const_ref);
 
  CoMCost();
 
  ~CoMCost();
 
  CoMCost(const CoMCost&) = default;
 
  CoMCost& operator=(const CoMCost&) = default;
 
  CoMCost(CoMCost&&) noexcept = default;
 
  CoMCost& operator=(CoMCost&&) noexcept = default;
 
  void set_ref(const std::shared_ptr<CoMRefBase>& ref);
 
  void set_const_ref(const Eigen::Vector3d& const_ref);
 
  void set_weight(const Eigen::Vector3d& weight);
 
  void set_weight_terminal(const Eigen::Vector3d& weight_terminal);
 
  void set_weight_impact(const Eigen::Vector3d& weight_impact);
 
  bool isCostActive(const GridInfo& grid_info) const {
    if (use_nonconst_ref_) {
      return ref_->isActive(grid_info);
    }
    else {
      return true;
    }
  }
 
  void evalDiff(const Robot& robot, CostFunctionData& data, 
                const GridInfo& grid_info) const {
    if (use_nonconst_ref_) {
      if (ref_->isActive(grid_info)) {
        ref_->updateRef(grid_info, data.x3d_ref);
        data.diff_3d = robot.CoM() - data.x3d_ref;
      }
    }
    else {
      data.diff_3d = robot.CoM() - const_ref_;
    }
  }
 
  double evalStageCost(Robot& robot, const ContactStatus& contact_status, 
                       CostFunctionData& data, const GridInfo& grid_info, 
                       const SplitSolution& s) const override;
 
  void evalStageCostDerivatives(Robot& robot, const ContactStatus& contact_status, 
                                CostFunctionData& data, const GridInfo& grid_info, 
                                const SplitSolution& s, 
                                SplitKKTResidual& kkt_residual) const override;
 
  void evalStageCostHessian(Robot& robot, const ContactStatus& contact_status, 
                            CostFunctionData& data, const GridInfo& grid_info, 
                            const SplitSolution& s, 
                            SplitKKTMatrix& kkt_matrix) const override;
 
  double evalTerminalCost(Robot& robot, CostFunctionData& data, 
                          const GridInfo& grid_info, const SplitSolution& s) const override;
 
  void evalTerminalCostDerivatives(Robot& robot, CostFunctionData& data, 
                                   const GridInfo& grid_info, const SplitSolution& s, 
                                   SplitKKTResidual& kkt_residual) const override;
 
  void evalTerminalCostHessian(Robot& robot, CostFunctionData& data, 
                               const GridInfo& grid_info, const SplitSolution& s, 
                               SplitKKTMatrix& kkt_matrix) const override;
 
  double evalImpactCost(Robot& robot, const ImpactStatus& impact_status, 
                         CostFunctionData& data, const GridInfo& grid_info, 
                         const SplitSolution& s) const override;
 
  void evalImpactCostDerivatives(Robot& robot, const ImpactStatus& impact_status, 
                                  CostFunctionData& data, const GridInfo& grid_info, 
                                  const SplitSolution& s, 
                                  SplitKKTResidual& kkt_residual) const override;
 
  void evalImpactCostHessian(Robot& robot, const ImpactStatus& impact_status, 
                              CostFunctionData& data, const GridInfo& grid_info, 
                              const SplitSolution& s, 
                              SplitKKTMatrix& kkt_matrix) const override;
 
  EIGEN_MAKE_ALIGNED_OPERATOR_NEW
 
private:
  Eigen::Vector3d const_ref_, weight_, weight_terminal_, weight_impact_;
  std::shared_ptr<CoMRefBase> ref_;
  bool use_nonconst_ref_, enable_cost_, enable_cost_terminal_, enable_cost_impact_;
};
 
} // namespace robotoc
 
#endif // ROBOTOC_COM_COST_HPP_