route_tool.hpp

// Copyright (c) 2024 John Chrosniak
//
// 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 NAV2_RVIZ_PLUGINS__ROUTE_TOOL_HPP_
#define NAV2_RVIZ_PLUGINS__ROUTE_TOOL_HPP_
 
#include <ui_route_tool.h>
#include <limits>
#include <memory>
#include <string>
#include <vector>
#include "geometry_msgs/msg/point_stamped.hpp"
#include "nav_msgs/msg/occupancy_grid.hpp"
#include "nav2_route/graph_loader.hpp"
#include "nav2_route/graph_saver.hpp"
#include "nav2_route/types.hpp"
#include "nav2_route/utils.hpp"
#include "rclcpp/rclcpp.hpp"
#include "rviz_common/panel.hpp"
#include "std_msgs/msg/int16.hpp"
#include "std_msgs/msg/string.hpp"
 
 
namespace nav2_rviz_plugins
{
class RouteTool : public rviz_common::Panel
{
  Q_OBJECT
 
public:
  explicit RouteTool(QWidget * parent = nullptr);
 
  void onInitialize() override;
 
  virtual void save(rviz_common::Config config) const;
  virtual void load(const rviz_common::Config & config);
 
 
private Q_SLOTS:
  void on_load_button_clicked(void);
 
  void on_save_button_clicked(void);
 
  void on_create_button_clicked(void);
 
  void on_confirm_button_clicked(void);
 
  void on_delete_button_clicked(void);
 
  void on_add_node_button_toggled(void);
 
  void on_edit_node_button_toggled(void);
 
protected:
  // UI pointer
  std::unique_ptr<Ui::route_tool> ui_;
 
private:
  void update_route_graph(void);
  nav2::LifecycleNode::SharedPtr node_;
  std::shared_ptr<nav2_route::GraphLoader> graph_loader_;
  std::shared_ptr<nav2_route::GraphSaver> graph_saver_;
  std::shared_ptr<tf2_ros::Buffer> tf_;
  nav2_route::Graph graph_;
  nav2_route::GraphToIDMap graph_to_id_map_;
  nav2_route::GraphToIDMap edge_to_node_map_;
  nav2_route::GraphToIncomingEdgesMap graph_to_incoming_edges_map_;
 
  rclcpp::Publisher<visualization_msgs::msg::MarkerArray>::SharedPtr
    graph_vis_publisher_;
  nav2::Subscription<geometry_msgs::msg::PointStamped>::SharedPtr
    clicked_point_subscription_;
 
  unsigned int next_node_id_ = 0;
};
}  // namespace nav2_rviz_plugins
#endif  // NAV2_RVIZ_PLUGINS__ROUTE_TOOL_HPP_