Merge pull request #60 from simulate-digital-rail/fix-is-switch

Fix is_switch by also checking the degree of the node in the graph
simulate-digital-rail · May 22, 2024 · 38c9859 · 38c9859
2 parents 8c55dfa + a4fcfc7
commit 38c9859
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/orm_importer/importer.py b/orm_importer/importer.py
@@ -149,7 +149,7 @@ def run(self, polygon):
         # Only nodes with max 1 edge or that are a switch can be top nodes
         for node_id in self.graph.nodes:
             node = self.node_data[node_id]
-            if is_end_node(node, self.graph) or is_switch(node):
+            if is_end_node(node, self.graph) or is_switch(node, self.graph):
                 self.top_nodes.append(node)
 
         for node in self.top_nodes:

diff --git a/orm_importer/utils.py b/orm_importer/utils.py
@@ -45,8 +45,8 @@ def is_signal(node):
     return "railway:signal:direction" in node.tags.keys()
 
 
-def is_switch(node):
-    return is_x(node, "switch")
+def is_switch(node, graph):
+    return is_x(node, "switch") and graph.degree(node.id) == 3
 
 
 def is_x(node, x: str):