Allow strings as track type

Author: arneboockmeyer

yaramo/topology.py

@@ -10,8 +10,8 @@
 from yaramo.node import Node
 from yaramo.route import Route
 from yaramo.signal import Signal
-from yaramo.vacancy_section import VacancySection
 from yaramo.track import Track
+from yaramo.vacancy_section import VacancySection
 
 
 class PlanningState(Enum):

yaramo/track.py

@@ -18,9 +18,13 @@ class TrackType(Enum):
 class Track(BaseElement):
     def __init__(self, track_type, **kwargs):
         super().__init__(**kwargs)
-        self.track_type: TrackType = track_type
         self.edge_sections: Dict[Edge, Tuple[float, float]] = {}
 
+        if isinstance(track_type, str):
+            self.track_type = TrackType.__members__.get(track_type, TrackType.sonstige)
+        elif isinstance(track_type, TrackType):
+            self.track_type = track_type
+
     def add_edge_section(self, edge: Edge, section_start: float, section_end: float):
         self.edge_sections[edge] = (section_start, section_end)