docking_server.hpp

// Copyright (c) 2024 Open Navigation LLC
//
// Licensed under the Apache License, Version 2.0 (the "License");
// you may not use this file except in compliance with the License.
// You may obtain a copy of the License at
//
//     http://www.apache.org/licenses/LICENSE-2.0
//
// Unless required by applicable law or agreed to in writing, software
// distributed under the License is distributed on an "AS IS" BASIS,
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
// See the License for the specific language governing permissions and
// limitations under the License.
 
#ifndef OPENNAV_DOCKING__DOCKING_SERVER_HPP_
#define OPENNAV_DOCKING__DOCKING_SERVER_HPP_
 
#include <functional>
#include <memory>
#include <mutex>
#include <optional>
#include <string>
#include <vector>
 
#include "rclcpp/rclcpp.hpp"
#include "nav2_ros_common/lifecycle_node.hpp"
#include "nav2_ros_common/node_utils.hpp"
#include "nav2_ros_common/simple_action_server.hpp"
#include "nav2_util/twist_publisher.hpp"
#include "nav2_util/odometry_utils.hpp"
#include "opennav_docking/controller.hpp"
#include "opennav_docking/utils.hpp"
#include "opennav_docking/types.hpp"
#include "opennav_docking/dock_database.hpp"
#include "opennav_docking/navigator.hpp"
#include "opennav_docking/parameter_handler.hpp"
#include "opennav_docking_core/charging_dock.hpp"
#include "nav2_ros_common/tf2_factories.hpp"
 
namespace opennav_docking
{
class DockingServer : public nav2::LifecycleNode
{
public:
  using DockingActionServer = nav2::SimpleActionServer<DockRobot>;
  using UndockingActionServer = nav2::SimpleActionServer<UndockRobot>;
 
  explicit DockingServer(const rclcpp::NodeOptions & options = rclcpp::NodeOptions());
 
  ~DockingServer() = default;
 
  void stashDockData(bool use_dock_id, Dock * dock, bool successful);
 
  void publishDockingFeedback(uint16_t state);
 
  Dock * generateGoalDock(std::shared_ptr<const DockRobot::Goal> goal);
 
  void doInitialPerception(Dock * dock, geometry_msgs::msg::PoseStamped & dock_pose);
 
  bool approachDock(Dock * dock, geometry_msgs::msg::PoseStamped & dock_pose, bool backward);
 
  void rotateToDock(const geometry_msgs::msg::PoseStamped & dock_pose);
 
  bool waitForCharge(Dock * dock);
 
  bool resetApproach(const geometry_msgs::msg::PoseStamped & staging_pose, bool backward);
 
  bool getCommandToPose(
    geometry_msgs::msg::Twist & cmd, const geometry_msgs::msg::PoseStamped & pose,
    double linear_tolerance, double angular_tolerance, bool is_docking, bool backward);
 
  virtual geometry_msgs::msg::PoseStamped getRobotPoseInFrame(const std::string & frame);
 
  template<typename ActionT>
  void getPreemptedGoalIfRequested(
    typename std::shared_ptr<const typename ActionT::Goal> goal,
    const typename nav2::SimpleActionServer<ActionT>::SharedPtr & action_server);
 
  template<typename ActionT>
  bool checkAndWarnIfCancelled(
    typename nav2::SimpleActionServer<ActionT>::SharedPtr & action_server,
    const std::string & name);
 
  template<typename ActionT>
  bool checkAndWarnIfPreempted(
    typename nav2::SimpleActionServer<ActionT>::SharedPtr & action_server,
    const std::string & name);
 
  nav2::CallbackReturn on_configure(const rclcpp_lifecycle::State & state) override;
 
  nav2::CallbackReturn on_activate(const rclcpp_lifecycle::State & state) override;
 
  nav2::CallbackReturn on_deactivate(const rclcpp_lifecycle::State & state) override;
 
  nav2::CallbackReturn on_cleanup(const rclcpp_lifecycle::State & state) override;
 
  nav2::CallbackReturn on_shutdown(const rclcpp_lifecycle::State & state) override;
 
  void publishZeroVelocity();
 
protected:
  void dockRobot();
 
  void undockRobot();
 
  // Parameter handler
  std::unique_ptr<opennav_docking::ParameterHandler> param_handler_;
  Parameters * params_;
 
  // Maximum number of times the robot will return to staging pose and retry docking
  int num_retries_;
 
  // This is a class member so it can be accessed in publish feedback
  rclcpp::Time action_start_time_;
 
  std::unique_ptr<nav2_util::TwistPublisher> vel_publisher_;
  std::unique_ptr<nav2_util::OdomSmoother> odom_sub_;
  typename DockingActionServer::SharedPtr docking_action_server_;
  typename UndockingActionServer::SharedPtr undocking_action_server_;
 
  std::unique_ptr<DockDatabase> dock_db_;
  std::unique_ptr<Navigator> navigator_;
  std::unique_ptr<Controller> controller_;
  std::string curr_dock_type_;
 
  nav2::TransformBuffer::SharedPtr tf2_buffer_;
  nav2::TransformListener::SharedPtr tf2_listener_;
};
 
}  // namespace opennav_docking
 
#endif  // OPENNAV_DOCKING__DOCKING_SERVER_HPP_