front: fix pr comments

Signed-off-by: Egor Berezovskiy <egor@berezify.fr>

editoast/src/core/simulation.rs

@@ -4,7 +4,6 @@ use std::collections::HashMap;
 use editoast_schemas::rolling_stock::EffortCurves;
 use editoast_schemas::rolling_stock::Gamma;
 use editoast_schemas::rolling_stock::RollingResistance;
-use editoast_schemas::rolling_stock::RollingStock;
 use editoast_schemas::rolling_stock::TowedRollingStock;
 use editoast_schemas::train_schedule::Comfort;
 use editoast_schemas::train_schedule::Distribution;
@@ -18,6 +17,7 @@ use utoipa::ToSchema;
 use super::pathfinding::TrackRange;
 use crate::core::{AsCoreRequest, Json};
 use crate::error::InternalError;
+use crate::models::RollingStockModel;
 use crate::views::path::pathfinding::PathfindingFailure;
 use derivative::Derivative;
 use editoast_schemas::primitives::Identifier;
@@ -78,11 +78,16 @@ pub struct PhysicsConsistParameters {
     /// In m/s
     pub max_speed: Option<f64>,
     pub towed_rolling_stock: Option<TowedRollingStock>,
-    pub traction_engine: RollingStock,
+    pub traction_engine: RollingStockModel,
 }
 
 impl PhysicsConsistParameters {
-    pub fn from_traction_engine(traction_engine: RollingStock) -> Self {
+    pub fn set_traction_engine(mut self, traction_engine: RollingStockModel) -> Self {
+        self.traction_engine = traction_engine;
+        self
+    }
+
+    pub fn from_traction_engine(traction_engine: RollingStockModel) -> Self {
         PhysicsConsistParameters {
             max_speed: None,
             total_length: None,

editoast/src/models/fixtures.rs

@@ -245,8 +245,10 @@ pub fn create_towed_rolling_stock() -> TowedRollingStock {
     }
 }
 
-pub fn create_simple_rolling_stock() -> RollingStock {
-    RollingStock {
+pub fn create_simple_rolling_stock() -> RollingStockModel {
+    RollingStockModel {
+        id: 1,
+        version: 1,
         name: "SIMPLE_ROLLING_STOCK".to_string(),
         loading_gauge: LoadingGaugeType::G1,
         supported_signaling_systems: RollingStockSupportedSignalingSystems(vec![]),

editoast/src/views/path/pathfinding.rs

@@ -23,6 +23,7 @@ use crate::core::pathfinding::PathfindingInputError;
 use crate::core::pathfinding::PathfindingNotFound;
 use crate::core::pathfinding::PathfindingRequest;
 use crate::core::pathfinding::PathfindingResultSuccess;
+use crate::core::simulation::PhysicsConsistParameters;
 use crate::core::AsCoreRequest;
 use crate::core::CoreClient;
 use crate::error::InternalError;
@@ -340,7 +341,12 @@ pub async fn pathfinding_from_train(
         RollingStockModel::retrieve(conn, train_schedule.rolling_stock_name.clone())
             .await?
             .into_iter()
-            .map(|rs| (rs.name.clone(), rs))
+            .map(|rs| {
+                (
+                    rs.name.clone(),
+                    PhysicsConsistParameters::from_traction_engine(rs),
+                )
+            })
             .collect();
 
     Ok(pathfinding_from_train_batch(
@@ -363,7 +369,7 @@ pub async fn pathfinding_from_train_batch(
     core: Arc<CoreClient>,
     infra: &Infra,
     train_schedules: &[TrainSchedule],
-    rolling_stocks: &HashMap<String, RollingStockModel>,
+    consists: &HashMap<String, PhysicsConsistParameters>,
 ) -> Result<Vec<PathfindingResult>> {
     let mut results = vec![
         PathfindingResult::Failure(PathfindingFailure::PathfindingInputError(
@@ -376,14 +382,16 @@ pub async fn pathfinding_from_train_batch(
     for (index, train_schedule) in train_schedules.iter().enumerate() {
         // Retrieve rolling stock
         let rolling_stock_name = &train_schedule.rolling_stock_name;
-        let Some(rolling_stock) = rolling_stocks.get(rolling_stock_name).cloned() else {
+        let Some(rolling_stock) = consists.get(rolling_stock_name).cloned() else {
             let rolling_stock_name = rolling_stock_name.clone();
             results[index] = PathfindingResult::Failure(PathfindingFailure::PathfindingInputError(
                 PathfindingInputError::RollingStockNotFound { rolling_stock_name },
             ));
             continue;
         };
 
+        let rolling_stock = rolling_stock.traction_engine;
+
         // Create the path input
         let path_input = PathfindingInput {
             rolling_stock_loading_gauge: rolling_stock.loading_gauge,

editoast/src/views/timetable/stdcm.rs

@@ -204,14 +204,10 @@ async fn stdcm(
         core_client.clone(),
         &infra,
         &train_schedules,
-        vec![rolling_stock.clone()]
-            .into_iter()
-            .map(|rs| (rs.name.clone(), rs.clone()))
-            .collect(),
-        vec![physics_consist_parameters.clone()]
-            .into_iter()
-            .map(|consist| (consist.traction_engine.name.clone(), consist))
-            .collect(),
+        HashMap::from([(
+            physics_consist_parameters.traction_engine.name.clone(),
+            physics_consist_parameters.clone(),
+        )]),
         stdcm_request.electrical_profile_set_id,
     )
     .await?
@@ -434,14 +430,12 @@ impl VirtualTrainRun {
             core_client,
             infra,
             &[train_schedule.clone()],
-            vec![rolling_stock]
-                .into_iter()
-                .map(|rs| (rs.name.clone(), rs.clone()))
-                .collect(),
-            vec![consist_parameters.clone()]
-                .into_iter()
-                .map(|consist| (consist.traction_engine.name.clone(), consist))
-                .collect(),
+            HashMap::from([(
+                rolling_stock.name.clone(),
+                consist_parameters
+                    .clone()
+                    .set_traction_engine(rolling_stock.clone()),
+            )]),
             None,
         )
         .await?

editoast/src/views/train_schedule.rs

@@ -423,7 +423,6 @@ pub async fn train_simulation_batch(
         core.clone(),
         infra,
         train_schedules,
-        rolling_stocks,
         consists,
         electrical_profile_set_id,
     )
@@ -436,7 +435,6 @@ pub async fn consist_train_simulation_batch(
     core: Arc<CoreClient>,
     infra: &Infra,
     train_schedules: &[TrainSchedule],
-    rolling_stocks: HashMap<String, RollingStockModel>,
     consists: HashMap<String, PhysicsConsistParameters>,
     electrical_profile_set_id: Option<i64>,
 ) -> Result<Vec<(SimulationResponse, PathfindingResult)>> {
@@ -448,7 +446,7 @@ pub async fn consist_train_simulation_batch(
         core.clone(),
         infra,
         train_schedules,
-        &rolling_stocks,
+        &consists,
     )
     .await?;