re add selected train id to chart synchronizer v1

OpenRailAssociation · May 31, 2024 · b8baf63 · b8baf63
1 parent e4e4230
commit b8baf63
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/front/src/modules/simulationResult/components/ChartHelpers/ChartSynchronizer.ts b/front/src/modules/simulationResult/components/ChartHelpers/ChartSynchronizer.ts
@@ -51,12 +51,12 @@ export class ChartSynchronizer {
     this.subscribers.delete(key);
   }
 
-  computePositionValues() {
+  computePositionValues(newTrainId?: number) {
     if (!this.reduxStore) {
       throw new Error('Redux store reference was not set in Chart synchronizer');
     }
     const osrdSimulation = this.reduxStore.getState().osrdsimulation;
-    const trainId = osrdSimulation.selectedTrainId;
+    const trainId = newTrainId || osrdSimulation.selectedTrainId;
     const currentTrainSimulation = osrdSimulation.consolidatedSimulation.find(
       (consolidatedSimulation: SimulationTrain) => consolidatedSimulation.id === trainId
     );

diff --git a/front/src/reducers/osrdsimulation/actions.ts b/front/src/reducers/osrdsimulation/actions.ts
@@ -68,7 +68,7 @@ export function updateSelectedProjection(
 
 export function updateSelectedTrainId(selectedTrainId: OsrdSimulationState['selectedTrainId']) {
   return (dispatch: Dispatch, _: GetState, chartSynchronizer: ChartSynchronizer) => {
-    chartSynchronizer.computePositionValues();
+    chartSynchronizer.computePositionValues(selectedTrainId);
     dispatch({
       type: UPDATE_SELECTED_TRAIN_ID,
       selectedTrainId,