add get_nodes_in_order

Author: BergJulian

yaramo/track.py

@@ -1,14 +1,15 @@
-from enum import Enum, auto
+from collections import Counter
+from enum import IntEnum, auto
 from typing import Dict, List, Tuple
 
 from yaramo.base_element import BaseElement
 from yaramo.edge import Edge
 from yaramo.node import Node
 
 
-class TrackType(Enum):
-    Hauptgleis = auto()
+class TrackType(IntEnum):
     Durchgehendes_Hauptgleis = auto()
+    Hauptgleis = auto()
     Streckengleis = auto()
     Anschlussgleis = auto()
     Nebengleis = auto()
@@ -82,3 +83,24 @@ def is_node_in_track(self, node: Node):
             if edge.is_node_connected(node):
                 return True
         return False
+
+    def get_nodes_in_order(self) -> List[Node]:
+        nodes = [node for edge in self.edges for node in (edge.node_a, edge.node_b)]
+        start_nodes = [node for node, count in Counter(nodes).items() if count == 1]
+
+        if len(start_nodes) != 2:
+            raise ValueError("Edges of Track separated")
+
+        previous_node = start_nodes[0]
+        start_edge = next(edge for edge in self.edges if previous_node in (edge.node_a, edge.node_b))
+
+        edges_in_order = self.get_edges_in_order()
+        if edges_in_order[0] != start_edge:
+            edges_in_order.reverse()
+
+        nodes_in_order: list[Node] = [previous_node]
+        for edge in edges_in_order:
+            previous_node = edge.get_other_node(previous_node)
+            nodes_in_order.append(previous_node)
+
+        return nodes_in_order