Add merge radius option for vertices

Signed-off-by: Michael X. Grey <[email protected]>

rmf_traffic_editor/gui/level.cpp

@@ -1184,6 +1184,7 @@ void Level::draw(
     v.draw(
       scene,
       vertex_radius / drawing_meters_per_pixel,
+      drawing_meters_per_pixel,
       vertex_name_font,
       coordinate_system);
 

rmf_traffic_editor/gui/vertex.cpp

@@ -21,6 +21,8 @@
 #include <QGraphicsSimpleTextItem>
 #include <QIcon>
 
+#include <iostream>
+
 #include "vertex.h"
 using std::string;
 using std::vector;
@@ -40,6 +42,7 @@ const vector<pair<string, Param::Type>> Vertex::allowed_params
   { "is_passthrough_point", Param::Type::BOOL },
   { "human_goal_set_name", Param::Type::STRING },
   { "mutex", Param::Type::STRING },
+  { "merge_radius", Param::Type::DOUBLE },
 };
 
 
@@ -132,6 +135,7 @@ YAML::Node Vertex::to_yaml(const CoordinateSystem& coordinate_system) const
 void Vertex::draw(
   QGraphicsScene* scene,
   const double radius,
+  const double drawing_meters_per_pixel,
   const QFont& font,
   const CoordinateSystem& coordinate_system) const
 {
@@ -164,6 +168,19 @@ void Vertex::draw(
   const double icon_ring_radius = radius * 2.5;
   const double icon_scale = 2.0 * radius / 128.0;
 
+  if (const auto r_merge_opt = merge_radius())
+  {
+    const double r_merge = *r_merge_opt / drawing_meters_per_pixel;
+    const QPen radius_pen(selected ? selected_color : Qt::black);
+    auto* radius_item = scene->addEllipse(
+      x - r_merge,
+      y - r_merge,
+      2 * r_merge,
+      2 * r_merge,
+      radius_pen);
+    radius_item->setZValue(19.0);
+  }
+
   if (is_holding_point())
   {
     const double icon_bearing = -135.0 * M_PI / 180.0;
@@ -343,6 +360,15 @@ bool Vertex::is_charger() const
   return it->second.value_bool;
 }
 
+std::optional<double> Vertex::merge_radius() const
+{
+  const auto it = params.find("merge_radius");
+  if (it == params.end())
+    return std::nullopt;
+
+  return it->second.value_double;
+}
+
 bool Vertex::is_cleaning_zone() const
 {
   const auto it = params.find("is_cleaning_zone");

rmf_traffic_editor/gui/vertex.h

@@ -22,6 +22,7 @@
 #include <map>
 #include <string>
 #include <vector>
+#include <optional>
 #include <yaml-cpp/yaml.h>
 
 #include <QColor>
@@ -58,13 +59,15 @@ class Vertex
   void draw(
     QGraphicsScene* scene,
     const double radius,
+    const double drawing_meters_per_pixel,
     const QFont& font,
     const CoordinateSystem& coordinate_system) const;
 
   bool is_parking_point() const;
   bool is_holding_point() const;
   bool is_cleaning_zone() const;
   bool is_charger() const;
+  std::optional<double> merge_radius() const;
 
   std::string dropoff_ingestor() const;
   std::string pickup_dispenser() const;