point_point_max_constraint.hpp

// Copyright (c) TrajoptLib contributors
 
#pragma once
 
#include <cassert>
#include <utility>
 
#include <sleipnir/autodiff/variable.hpp>
#include <sleipnir/optimization/problem.hpp>
 
#include "trajopt/geometry/pose2.hpp"
#include "trajopt/geometry/translation2.hpp"
#include "trajopt/util/symbol_exports.hpp"
 
namespace trajopt {

class TRAJOPT_DLLEXPORT PointPointMaxConstraint {
 public:
  explicit PointPointMaxConstraint(Translation2d robot_point,
                                   Translation2d field_point,
                                   double max_distance)
      : m_robot_point{std::move(robot_point)},
        m_field_point{std::move(field_point)},
        m_max_distance{max_distance} {
    assert(max_distance >= 0.0);
  }

  void apply(
      slp::Problem<double>& problem, const Pose2v<double>& pose,
      [[maybe_unused]] const Translation2v<double>& linear_velocity,
      [[maybe_unused]] const slp::Variable<double>& angular_velocity,
      [[maybe_unused]] const Translation2v<double>& linear_acceleration,
      [[maybe_unused]] const slp::Variable<double>& angular_acceleration) {
    auto bumper_corner =
        pose.translation() + m_robot_point.rotate_by(pose.rotation());
    auto dx = m_field_point.x() - bumper_corner.x();
    auto dy = m_field_point.y() - bumper_corner.y();
    problem.subject_to(dx * dx + dy * dy <= m_max_distance * m_max_distance);
  }
 
 private:
  Translation2d m_robot_point;
  Translation2d m_field_point;
  double m_max_distance;
};
 
}  // namespace trajopt