editoast: add simulation summary and simulation endpoints for paced t…

…rain Signed-off-by: Youness CHRIFI ALAOUI <[email protected]>
OpenRailAssociation · Mar 6, 2025 · 13b13bc · 13b13bc
1 parent 164c697
commit 13b13bc
Show file tree

Hide file tree

Showing 10 changed files with 361 additions and 102 deletions.
diff --git a/editoast/openapi.yaml b/editoast/openapi.yaml
@@ -5140,9 +5140,10 @@ components:
       - $ref: '#/components/schemas/EditoastOperationErrorInvalidPatch'
       - $ref: '#/components/schemas/EditoastOperationErrorModifyId'
       - $ref: '#/components/schemas/EditoastOperationErrorObjectNotFound'
-      - $ref: '#/components/schemas/EditoastPacedTrainErrorBatchPacedTrainNotFound'
+      - $ref: '#/components/schemas/EditoastPacedTrainErrorBatchNotFound'
       - $ref: '#/components/schemas/EditoastPacedTrainErrorDatabase'
-      - $ref: '#/components/schemas/EditoastPacedTrainErrorPacedTrainNotFound'
+      - $ref: '#/components/schemas/EditoastPacedTrainErrorInfraNotFound'
+      - $ref: '#/components/schemas/EditoastPacedTrainErrorNotFound'
       - $ref: '#/components/schemas/EditoastPaginationErrorInvalidPage'
       - $ref: '#/components/schemas/EditoastPaginationErrorInvalidPageSize'
       - $ref: '#/components/schemas/EditoastPathfindingErrorInfraNotFound'
@@ -5637,7 +5638,7 @@ components:
           type: string
           enum:
           - editoast:operation:ObjectNotFound
-    EditoastPacedTrainErrorBatchPacedTrainNotFound:
+    EditoastPacedTrainErrorBatchNotFound:
       type: object
       required:
       - type
@@ -5660,7 +5661,7 @@ components:
         type:
           type: string
           enum:
-          - editoast:paced_train:BatchPacedTrainNotFound
+          - editoast:paced_train:BatchNotFound
     EditoastPacedTrainErrorDatabase:
       type: object
       required:
@@ -5680,7 +5681,31 @@ components:
           type: string
           enum:
           - editoast:paced_train:Database
-    EditoastPacedTrainErrorPacedTrainNotFound:
+    EditoastPacedTrainErrorInfraNotFound:
+      type: object
+      required:
+      - type
+      - status
+      - message
+      properties:
+        context:
+          type: object
+          required:
+          - infra_id
+          properties:
+            infra_id:
+              type: integer
+        message:
+          type: string
+        status:
+          type: integer
+          enum:
+          - 404
+        type:
+          type: string
+          enum:
+          - editoast:paced_train:InfraNotFound
+    EditoastPacedTrainErrorNotFound:
       type: object
       required:
       - type
@@ -5703,7 +5728,7 @@ components:
         type:
           type: string
           enum:
-          - editoast:paced_train:PacedTrainNotFound
+          - editoast:paced_train:NotFound
     EditoastPaginationErrorInvalidPage:
       type: object
       required:
@@ -8570,7 +8595,7 @@ components:
           timetable_id:
             type: integer
             format: int64
-            description: Timetable attached to the train schedule
+            description: Timetable attached to the paced train
             nullable: true
     PacedTrainResult:
       allOf:

diff --git a/editoast/src/models/paced_train.rs b/editoast/src/models/paced_train.rs
@@ -1,3 +1,4 @@
+use crate::models::train_schedule::TrainSchedule;
 use chrono::DateTime;
 use chrono::Duration as ChronoDuration;
 use chrono::Utc;
@@ -101,3 +102,25 @@ impl From<PacedTrain> for PacedTrainBase {
         }
     }
 }
+
+impl From<PacedTrain> for TrainSchedule {
+    fn from(paced_train: PacedTrain) -> Self {
+        Self {
+            id: paced_train.id,
+            train_name: paced_train.train_name,
+            labels: paced_train.labels.into(),
+            rolling_stock_name: paced_train.rolling_stock_name,
+            timetable_id: paced_train.timetable_id,
+            path: paced_train.path,
+            start_time: paced_train.start_time,
+            schedule: paced_train.schedule,
+            margins: paced_train.margins,
+            initial_speed: paced_train.initial_speed,
+            comfort: paced_train.comfort,
+            constraint_distribution: paced_train.constraint_distribution,
+            speed_limit_tag: paced_train.speed_limit_tag,
+            power_restrictions: paced_train.power_restrictions,
+            options: paced_train.options,
+        }
+    }
+}
diff --git a/editoast/src/views/mod.rs b/editoast/src/views/mod.rs
@@ -170,9 +170,9 @@ pub struct InfraIdQueryParam {
     infra_id: i64,
 }
 
-#[derive(Debug, Serialize, ToSchema)]
+#[derive(Debug, PartialEq, Serialize, Deserialize, ToSchema)]
 #[serde(tag = "status", rename_all = "snake_case")]
-enum SimulationSummaryResult {
+pub enum SimulationSummaryResult {
     /// Minimal information on a simulation's result
     Success {
         /// Length of a path in mm