Implemented contains() function for edge

Signed-off-by: tanjunkiat <[email protected]>

Author: TanJunKiat

rmf_traffic_editor/gui/edge.cpp

@@ -202,3 +202,8 @@ double Edge::get_width() const
     return -1.0;// shouldn't get here
   return it->second.value_double;
 }
+
+bool Edge::contains(const int idx) const
+{
+  return start_idx == idx || end_idx == idx;
+}

rmf_traffic_editor/gui/edge.h

@@ -71,6 +71,7 @@ class Edge
   int get_graph_idx() const;
 
   double get_width() const;
+  bool contains(const int idx) const;
 };
 
 #endif

rmf_traffic_editor/gui/level.cpp

@@ -508,8 +508,7 @@ bool Level::delete_used_entities(int selected_vertex_idx)
         edges.end(),
         [selected_vertex_idx](const Edge& edge)
         {
-          return edge.start_idx == selected_vertex_idx ||
-            edge.end_idx == selected_vertex_idx;
+          return edge.contains(selected_vertex_idx);
         }),
       edges.end());