editoast: add level crossings to railjson_generator

Author: Wadjetz

python/osrd_schemas/osrd_schemas/infra.py

@@ -600,6 +600,9 @@ class RailJsonInfra(BaseModel):
     operational_points: List[OperationalPoint] = Field(
         description="List of operational points of the corresponding infra"
     )
+    level_crossings: List[LevelCrossing] = Field(
+        description="List of level crossings of the corresponding infra"
+    )
     routes: List[Route] = Field(description="Routes of the infra")
     extended_switch_types: List[SwitchType] = Field(
         default=[],

python/railjson_generator/README.md

@@ -27,6 +27,7 @@ uv run -m railjson_generator ../../tests/data/infras ../../tests/infra-scripts/*
 
 - `__init__(self) -> InfraBuilder`: Instantiates an infra builder.
 - `add_track_section(self, length, label="track.X", waypoints=[], signals=[], operational_points=[]) -> TrackSection`: Add a track section.
+- `add_level_crossing(self, id, name="lc.X", short_zone_length=1000) -> LevelCrossing`: Add a level crossing.
 - `add_point_switch(self, base, left, right, label="switch.X", delay=0) -> Switch`: Add a point.
 - `add_crossing(self, north, south, east, west, label="switch.X", delay=0) -> Switch`: Add a cross switch.
 - `add_double_slip_switch(self, north_1, north_2, south_1, south_2, label="switch.X", delay=0) -> Switch`: Add a double cross switch.
@@ -67,6 +68,10 @@ Route can either be manually created, or generated using `generate_routes`, and
 
 - `add_part(self, track, offset)`: Link an operational point to a position.
 
+### Level crossing
+
+- `add_part(self, track, position, pedal_upstream, pedal_downstream)`: Add a part to a level crossing.
+
 ### Infra
 
 - `save(self, path)`: Format to railjson and save at the given location.

python/railjson_generator/railjson_generator/infra_builder.py

@@ -5,6 +5,7 @@
 from .schema.infra.infra import BufferStop, Infra, Switch, Detector
 from .schema.infra.neutral_section import NeutralSection
 from .schema.infra.operational_point import OperationalPoint
+from .schema.infra.level_crossing import LevelCrossing
 from .schema.infra.route import Route
 from .schema.infra.speed_section import SpeedSection
 from .schema.infra.switch import (
@@ -86,6 +87,16 @@ def add_point_switch(
         self.infra.switches.append(switch)
         return switch
 
+    def add_level_crossing(
+        self, id: str, name: str, short_zone_length: int
+    ) -> LevelCrossing:
+        """Build a level crossing, add it to the infra, and return it."""
+        lc = LevelCrossing(
+            id=id, name=name, short_zone_length=short_zone_length, parts=[]
+        )
+        self.infra.level_crossings.append(lc)
+        return lc
+
     def add_crossing(
         self,
         north: TrackEndpoint,

python/railjson_generator/railjson_generator/schema/infra/infra.py

@@ -7,6 +7,7 @@
 from railjson_generator.schema.infra.electrification import Electrification
 from railjson_generator.schema.infra.neutral_section import NeutralSection
 from railjson_generator.schema.infra.operational_point import OperationalPoint
+from railjson_generator.schema.infra.level_crossing import LevelCrossing
 from railjson_generator.schema.infra.route import Route
 from railjson_generator.schema.infra.speed_section import SpeedSection
 from railjson_generator.schema.infra.switch import Switch
@@ -19,6 +20,7 @@ class Infra:
     track_sections: List[TrackSection] = field(default_factory=list)
     switches: List[Switch] = field(default_factory=list)
     operational_points: List[OperationalPoint] = field(default_factory=list)
+    level_crossings: List[LevelCrossing] = field(default_factory=list)
     routes: List[Route] = field(default_factory=list)
     speed_sections: List[SpeedSection] = field(default_factory=list)
     electrifications: List[Electrification] = field(default_factory=list)
@@ -40,6 +42,7 @@ def to_rjs(self) -> infra.RailJsonInfra:
             buffer_stops=list(self.make_rjs_buffer_stops()),
             detectors=list(self.make_rjs_detectors()),
             operational_points=self.make_rjs_operational_points(),
+            level_crossings=[lc.to_rjs() for lc in self.level_crossings],
             extended_switch_types=[],
             speed_sections=[
                 speed_section.to_rjs() for speed_section in self.speed_sections

python/railjson_generator/railjson_generator/schema/infra/level_crossing.py

@@ -0,0 +1,47 @@
+from dataclasses import dataclass
+from typing import List
+
+from osrd_schemas import infra
+
+
+@dataclass
+class LevelCrossingPart:
+    pedal_upstream: int
+    pedal_downstream: int
+    position: float
+    track: str
+
+    def to_rjs(self):
+        return infra.LevelCrossingPart(
+            track=self.track,
+            position=self.position,
+            pedal_upstream=self.pedal_upstream,
+            pedal_downstream=self.pedal_downstream,
+        )
+
+
+@dataclass
+class LevelCrossing:
+    id: str
+    name: str
+    short_zone_length: int
+    parts: List[LevelCrossingPart]
+
+    def add_part(
+        self, track: str, position: float, pedal_upstream: int, pedal_downstream: int
+    ):
+        part = LevelCrossingPart(
+            track=track,
+            position=position,
+            pedal_upstream=pedal_upstream,
+            pedal_downstream=pedal_downstream,
+        )
+        self.parts.append(part)
+
+    def to_rjs(self):
+        return infra.LevelCrossing(
+            id=self.id,
+            name=self.name,
+            short_zone_length=self.short_zone_length,
+            parts=[part.to_rjs() for part in self.parts],
+        )