api: add stops details to the report

Author: flomonster

api/openapi.yaml

@@ -843,6 +843,8 @@ components:
               time: { type: number, format: float }
               position: { type: number, format: float }
               duration: { type: number, format: float }
+              line_number: { type: number }
+              track_number: { type: number }
         route_aspects:
           type: array
           items:

api/osrd_infra/views/train_schedule/standalone_simulation_report.py

@@ -66,9 +66,7 @@ def convert_simulation_results(
     route_begin_occupancy, route_end_occupancy = convert_route_occupancies(
         simulation_result["route_occupancies"], projection_path_payload, departure_time
     )
-    route_aspects = project_signal_updates(
-        simulation_result["signal_updates"], projection_path_payload, departure_time
-    )
+    route_aspects = project_signal_updates(simulation_result["signal_updates"], projection_path_payload, departure_time)
 
     speeds = [{**speed, "time": speed["time"] + departure_time} for speed in simulation_result["speeds"]]
     stops = [{**stop, "time": stop["time"] + departure_time} for stop in simulation_result["stops"]]
@@ -161,14 +159,16 @@ def build_signal_updates(signal_updates, departure_time):
     results = []
 
     for update in signal_updates:
-        results.append({
-            "signal_id": update["signal_id"],
-            "time_start": update["time_start"] + departure_time,
-            "time_end": update["time_end"] + departure_time,
-            "color": update["color"],
-            "blinking": update["blinking"],
-            "aspect_label": update["aspect_label"]
-        })
+        results.append(
+            {
+                "signal_id": update["signal_id"],
+                "time_start": update["time_start"] + departure_time,
+                "time_end": update["time_end"] + departure_time,
+                "color": update["color"],
+                "blinking": update["blinking"],
+                "aspect_label": update["aspect_label"],
+            }
+        )
     return results
 
 
@@ -189,16 +189,18 @@ def project_signal_updates(signal_updates, projection_path_payload: PathPayload,
             end_pos += track_range.length()
 
         for update in updates_by_route_id[route_id]:
-            results.append({
-                "signal_id": update["signal_id"],
-                "route_id": route_id,
-                "time_start": update["time_start"] + departure_time,
-                "time_end": update["time_end"] + departure_time,
-                "position_start": start_pos,
-                "position_end": end_pos,
-                "color": update["color"],
-                "blinking": update["blinking"]
-            })
+            results.append(
+                {
+                    "signal_id": update["signal_id"],
+                    "route_id": route_id,
+                    "time_start": update["time_start"] + departure_time,
+                    "time_end": update["time_end"] + departure_time,
+                    "position_start": start_pos,
+                    "position_end": end_pos,
+                    "color": update["color"],
+                    "blinking": update["blinking"],
+                }
+            )
         start_pos = end_pos
     return results
 

core/src/main/java/fr/sncf/osrd/infra/api/tracks/undirected/TrackEdge.java

@@ -39,4 +39,5 @@ public sealed interface TrackEdge permits SwitchBranch, TrackSection {
 
     /** Returns a list of ranges, each having a set of blocked loading gauge type */
     ImmutableRangeMap<Double, LoadingGaugeConstraint> getLoadingGaugeConstraints();
+
 }

core/src/main/java/fr/sncf/osrd/infra/api/tracks/undirected/TrackSection.java

@@ -1,3 +1,10 @@
 package fr.sncf.osrd.infra.api.tracks.undirected;
 
-public non-sealed interface TrackSection extends TrackEdge {}
+public non-sealed interface TrackSection extends TrackEdge {
+
+    /** Returns line code */
+    int getLineCode();
+
+    /** Returns track number */
+    int getTrackNumber();
+}

core/src/main/java/fr/sncf/osrd/infra/implementation/tracks/undirected/TrackSectionImpl.java

@@ -20,6 +20,8 @@ public class TrackSectionImpl implements TrackSection {
     private final LineString geo;
     private final LineString sch;
     private final ImmutableRangeMap<Double, LoadingGaugeConstraint> loadingGaugeConstraints;
+    private int trackNumber = 0;
+    private int lineCode = 0;
 
     @Override
     @ExcludeFromGeneratedCodeCoverage
@@ -37,14 +39,18 @@ public TrackSectionImpl(
             ImmutableSet<OperationalPoint> operationalPoints,
             LineString geo,
             LineString sch,
-            ImmutableRangeMap<Double, LoadingGaugeConstraint> loadingGaugeConstraints
+            ImmutableRangeMap<Double, LoadingGaugeConstraint> loadingGaugeConstraints,
+            int trackNumber,
+            int lineCode
     ) {
         this.length = length;
         this.id = id;
         this.operationalPoints = operationalPoints;
         this.geo = geo;
         this.sch = sch;
         this.loadingGaugeConstraints = loadingGaugeConstraints;
+        this.trackNumber = trackNumber;
+        this.lineCode = lineCode;
     }
 
     /** Constructor with empty operational points and geometry */
@@ -103,6 +109,16 @@ public ImmutableRangeMap<Double, LoadingGaugeConstraint> getLoadingGaugeConstrai
         return loadingGaugeConstraints;
     }
 
+    @Override
+    public int getLineCode() {
+        return lineCode;
+    }
+
+    @Override
+    public int getTrackNumber() {
+        return trackNumber;
+    }
+
     @Override
     public LineString getGeo() {
         return geo;

core/src/main/java/fr/sncf/osrd/infra/implementation/tracks/undirected/UndirectedInfraBuilder.java

@@ -195,7 +195,9 @@ private TrackSectionImpl makeTrackSection(RJSTrackSection track) {
                 ImmutableSet.copyOf(operationalPointsPerTrack.get(track.id)),
                 track.geo,
                 track.sch,
-                buildLoadingGaugeLimits(track.loadingGaugeLimits)
+                buildLoadingGaugeLimits(track.loadingGaugeLimits),
+                track.trackNumber,
+                track.lineCode
         );
         builder.addEdge(begin, end, edge);
         edge.gradients = makeGradients(track);

core/src/main/java/fr/sncf/osrd/railjson/schema/infra/RJSTrackSection.java

@@ -1,5 +1,6 @@
 package fr.sncf.osrd.railjson.schema.infra;
 
+import com.squareup.moshi.Json;
 import edu.umd.cs.findbugs.annotations.SuppressFBWarnings;
 import fr.sncf.osrd.railjson.schema.common.ID;
 import fr.sncf.osrd.railjson.schema.common.Identified;
@@ -23,6 +24,12 @@ public class RJSTrackSection implements Identified {
     public LineString geo;
     public LineString sch;
 
+    @Json(name = "line_code")
+    public int lineCode;
+
+    @Json(name = "track_number")
+    public int trackNumber;
+
     public RJSTrackSection(String id, double length) {
         this.id = id;
         this.length = length;

core/src/main/java/fr/sncf/osrd/standalone_sim/ScheduleMetadataExtractor.java

@@ -50,7 +50,9 @@ public static ResultTrain run(
         var stops = new ArrayList<ResultStops>();
         for (var stop : schedule.stops) {
             var stopTime = ResultPosition.interpolateTime(stop.position, headPositions);
-            stops.add(new ResultStops(stopTime, stop.position, stop.duration));
+            var location = trainPath.findLocation(stop.position);
+            stops.add(new ResultStops(stopTime, stop.position, stop.duration,
+                    location.track().getLineCode(), location.track().getTrackNumber()));
         }
 
         // Compute events

core/src/main/java/fr/sncf/osrd/standalone_sim/result/ResultStops.java

@@ -1,7 +1,24 @@
 package fr.sncf.osrd.standalone_sim.result;
 
+import com.squareup.moshi.Json;
 import edu.umd.cs.findbugs.annotations.SuppressFBWarnings;
 
-@SuppressFBWarnings("URF_UNREAD_PUBLIC_OR_PROTECTED_FIELD")
-public record ResultStops(double time, double position, double duration) {
+@SuppressFBWarnings("URF_UNREAD_FIELD")
+public class ResultStops {
+    double time;
+    double position;
+    double duration;
+    @Json(name = "line_number")
+    int lineNumber;
+    @Json(name = "track_number")
+    int trackNumber;
+
+    /** RestultStops constructor */
+    public ResultStops(double time, double position, double duration, int lineNumber, int trackNumber) {
+        this.time = time;
+        this.position = position;
+        this.duration = duration;
+        this.lineNumber = lineNumber;
+        this.trackNumber = trackNumber;
+    }
 }