front: add power restriction column to new times & stops table

Add a power restriction column with a custom native select, wire up
persistence via the API for both unique trains and paced train
occurrences, and refactor cell update helpers to support the new field.

Signed-off-by: nncluzu <ngamenichaka@yahoo.fr>

Author: kmer2016

front/src/applications/operationalStudies/views/Scenario/components/SimulationResults/SimulationResults.tsx

@@ -373,6 +373,7 @@ const SimulationResults = ({
                 upsertTimetableItems={upsertTimetableItems}
                 isSimulationDataLoading={isSimulationDataLoading}
                 operationalPointsOnPath={simulationResults.pathProperties?.operationalPoints}
+                rollingStock={simulationResults.rollingStock}
                 {...(simulationResults?.isValid &&
                   simulationSummary?.isValid && {
                     isValid: true,

front/src/modules/timesStops/TimesStopsOutput.tsx

@@ -7,6 +7,7 @@ import type { PathPropertiesFormatted } from 'applications/operationalStudies/ty
 import type {
   CorePathfindingResultSuccess,
   ReceptionSignal,
+  RollingStock,
   SimulationResponseSuccess,
 } from 'common/api/osrdEditoastApi';
 import type { SimulationSummary, TimetableItemWithDetails } from 'modules/timetableItem/types';
@@ -35,6 +36,7 @@ type TimesStopsOutputProps = {
   simulatedPathItemRespect?: Extract<SimulationSummary, { isValid: true }>['pathItemRespect'];
   operationalPointsOnPath?: PathPropertiesFormatted['operationalPoints'];
   isSimulationDataLoading?: boolean;
+  rollingStock?: RollingStock;
 };
 
 const TimesStopsOutput = ({
@@ -48,6 +50,7 @@ const TimesStopsOutput = ({
   simulatedPathItemRespect,
   operationalPointsOnPath,
   isSimulationDataLoading = false,
+  rollingStock,
 }: TimesStopsOutputProps) => {
   const useNewTimesStopsTable = useSelector(getUseNewTimesStopsTable);
 
@@ -107,13 +110,23 @@ const TimesStopsOutput = ({
 
   const startTime = useMemo(() => new Date(selectedTrain.start_time), [selectedTrain.start_time]);
 
-  const { updateArrival, updateStopDuration, updateDeparture, updateReceptionSignal } =
-    useUpdateTimesStopsTable(
-      selectedTrain,
-      newRows,
-      timetableItemsWithDetails,
-      upsertTimetableItems
-    );
+  const availablePowerRestrictions = useMemo(
+    () => Object.keys(rollingStock?.power_restrictions ?? {}),
+    [rollingStock]
+  );
+
+  const {
+    updateArrival,
+    updateStopDuration,
+    updateDeparture,
+    updateReceptionSignal,
+    updatePowerRestrictions,
+  } = useUpdateTimesStopsTable(
+    selectedTrain,
+    newRows,
+    timetableItemsWithDetails,
+    upsertTimetableItems
+  );
 
   // True if we are still waiting for fresh simulation data after a user edit.
   // Both conditions must be false before we clear the loading state:
@@ -176,17 +189,24 @@ const TimesStopsOutput = ({
     );
   };
 
+  const handlePowerRestrictionChange = (row: TimesStopsRowNew, value: string | null) =>
+    commitEdit({ rowId: row.id, field: 'powerRestriction', value }, () =>
+      updatePowerRestrictions(row, value)
+    );
+
   if (useNewTimesStopsTable) {
     return (
       <TimesStopsTable
         rows={optimisticRows}
         startTime={startTime}
         isValid={stableIsValid}
         isComputedDataPending={isAwaitingSimulation}
+        availablePowerRestrictions={availablePowerRestrictions}
         onArrivalChange={handleArrivalChange}
         onStopDurationChange={handleStopDurationChange}
         onDepartureChange={handleDepartureChange}
         onReceptionSignalChange={handleReceptionSignalChange}
+        onPowerRestrictionChange={handlePowerRestrictionChange}
       />
     );
   }

front/src/modules/timesStops/TimesStopsTable.tsx

@@ -1,7 +1,7 @@
 import { useCallback, Fragment, useMemo, useRef } from 'react';
 
 import { Checkbox } from '@osrd-project/ui-core';
-import { Moon } from '@osrd-project/ui-icons';
+import { Moon, TriangleDown } from '@osrd-project/ui-icons';
 import {
   createColumnHelper,
   flexRender,
@@ -14,6 +14,7 @@ import cx from 'classnames';
 import { useTranslation } from 'react-i18next';
 
 import type { ReceptionSignal } from 'common/api/osrdEditoastApi';
+import { NO_POWER_RESTRICTION } from 'modules/powerRestriction/consts';
 import { formatLocalTime, useDateTimeLocale } from 'utils/date';
 import { calculateTimeDifferenceInDays } from 'utils/timeManipulation';
 
@@ -33,10 +34,12 @@ declare module '@tanstack/react-table' {
   interface TableMeta<TData extends RowData> {
     allRows: TimesStopsRowNew[];
     isComputedDataPending?: boolean;
+    availablePowerRestrictions: string[];
     onArrivalChange: (row: TimesStopsRowNew, arrival: Date | null) => void;
     onStopDurationChange: (row: TimesStopsRowNew, durationSeconds: number | null) => void;
     onDepartureChange: (row: TimesStopsRowNew, departure: Date | null) => void;
     onReceptionSignalChange: (row: TimesStopsRowNew, signal: ReceptionSignal | undefined) => void;
+    onPowerRestrictionChange: (row: TimesStopsRowNew, value: string | null) => void;
   }
 }
 
@@ -80,10 +83,12 @@ type TimesStopsTableProps = {
   startTime: Date;
   isValid: boolean;
   isComputedDataPending?: boolean;
+  availablePowerRestrictions: string[];
   onArrivalChange: (row: TimesStopsRowNew, arrival: Date | null) => void;
   onStopDurationChange: (row: TimesStopsRowNew, durationSeconds: number | null) => void;
   onDepartureChange: (row: TimesStopsRowNew, departure: Date | null) => void;
   onReceptionSignalChange: (row: TimesStopsRowNew, signal: ReceptionSignal | undefined) => void;
+  onPowerRestrictionChange: (row: TimesStopsRowNew, value: string | null) => void;
 };
 
 const columnHelper = createColumnHelper<TimesStopsRowNew>();
@@ -100,10 +105,12 @@ const TimesStopsTable = ({
   startTime,
   isValid,
   isComputedDataPending,
+  availablePowerRestrictions,
   onArrivalChange,
   onStopDurationChange,
   onDepartureChange,
   onReceptionSignalChange,
+  onPowerRestrictionChange,
 }: TimesStopsTableProps) => {
   const { t } = useTranslation('translation', { keyPrefix: 'timeStopTable' });
   const dateTimeLocale = useDateTimeLocale();
@@ -375,6 +382,34 @@ const TimesStopsTable = ({
       }),
       columnHelper.accessor('powerRestriction', {
         header: () => t('powerRestriction'),
+        cell: (info) => {
+          const {
+            availablePowerRestrictions: codes,
+            onPowerRestrictionChange: onRestrictionChange,
+          } = info.table.options.meta!;
+          const value = info.getValue();
+          const row = info.row.original;
+          return (
+            <div className="power-restriction-select-wrapper">
+              <select
+                value={value ?? ''}
+                onChange={(e) => {
+                  const v = e.target.value;
+                  onRestrictionChange(row, v === '' ? null : v);
+                }}
+              >
+                <option value=""> </option>
+                <option value={NO_POWER_RESTRICTION}>Ø</option>
+                {codes.map((c) => (
+                  <option key={c} value={c}>
+                    {c}
+                  </option>
+                ))}
+              </select>
+              <TriangleDown className="power-restriction-arrow" />
+            </div>
+          );
+        },
         meta: {
           className: 'col-power-restriction',
         },
@@ -427,10 +462,12 @@ const TimesStopsTable = ({
     meta: {
       allRows: rows,
       isComputedDataPending,
+      availablePowerRestrictions,
       onArrivalChange,
       onStopDurationChange,
       onDepartureChange,
       onReceptionSignalChange,
+      onPowerRestrictionChange,
     },
   });
 

front/src/modules/timesStops/helpers/cellUpdate.ts

@@ -1,6 +1,11 @@
 import { v4 as uuidV4 } from 'uuid';
 
-import type { TrainSchedule, PathItem, ScheduleItem } from 'common/api/osrdEditoastApi';
+import type {
+  TrainSchedule,
+  PathItem,
+  PowerRestrictionItem,
+  ScheduleItem,
+} from 'common/api/osrdEditoastApi';
 import type { Train } from 'reducers/osrdconf/types';
 import { addElementAtIndex } from 'utils/array';
 import { Duration } from 'utils/duration';
@@ -69,7 +74,7 @@ export type ScheduleState = {
  */
 export const applyScheduleEdit = (
   current: ScheduleState,
-  edit: OptimisticEdit
+  edit: Exclude<OptimisticEdit, { field: 'powerRestriction' }>
 ): ScheduleState & { departure: Date | null } => {
   const { arrival, stop } = current;
 
@@ -115,7 +120,7 @@ export const applyScheduleEdit = (
     }
 
     case 'receptionSignal': {
-      // receptionSignal doesn't affect arrival/stop/departure values
+      // TODO: receptionSignal doesn't belong in applyScheduleEdit (tech debt — to be addressed separately)
       return {
         arrival,
         stop,
@@ -172,8 +177,23 @@ export const computeOptimisticRow = (
   edit: OptimisticEdit
 ): Pick<
   TimesStopsRowNew,
-  'requestedArrival' | 'stopDuration' | 'requestedDeparture' | 'closedSignal' | 'shortSlipDistance'
+  | 'requestedArrival'
+  | 'stopDuration'
+  | 'requestedDeparture'
+  | 'closedSignal'
+  | 'shortSlipDistance'
+  | 'powerRestriction'
 > => {
+  if (edit.field === 'powerRestriction') {
+    return {
+      requestedArrival: row.requestedArrival,
+      stopDuration: row.stopDuration,
+      requestedDeparture: row.requestedDeparture,
+      closedSignal: row.closedSignal,
+      shortSlipDistance: row.shortSlipDistance,
+      powerRestriction: edit.value,
+    };
+  }
   const { arrival, stop, departure } = applyScheduleEdit(
     { arrival: row.requestedArrival, stop: row.stopDuration },
     edit
@@ -187,20 +207,57 @@ export const computeOptimisticRow = (
     stopDuration: stop,
     requestedDeparture: departure,
     ...checkboxes,
+    powerRestriction: row.powerRestriction,
   };
 };
 
+/**
+ * Rebuilds the power_restrictions API array from path step rows.
+ * Each non-null powerRestriction on a path step row marks the start of a range.
+ * The range extends until the next path step with an explicit value (or the last path step).
+ *
+ * @example
+ * // rows: [{ id: 'A', powerRestriction: 'C1' }, { id: 'B', powerRestriction: null }, { id: 'C', powerRestriction: '∅' }, { id: 'D', powerRestriction: null }]
+ * // → [{ from: 'A', to: 'C', value: 'C1' }, { from: 'C', to: 'D', value: '∅' }]
+ *
+ * // rows: [{ id: 'A', powerRestriction: null }, { id: 'B', powerRestriction: 'C2' }, { id: 'C', powerRestriction: null }]
+ * // → [{ from: 'B', to: 'C', value: 'C2' }]
+ */
+export const buildPowerRestrictionsFromRows = (
+  rows: TimesStopsRowNew[]
+): PowerRestrictionItem[] => {
+  const pathStepRows = rows.filter((r) => r.isPathStep);
+  const result: PowerRestrictionItem[] = [];
+
+  for (let i = 0; i < pathStepRows.length - 1; i++) {
+    const code = pathStepRows[i].powerRestriction;
+    if (!code) continue;
+
+    const from = pathStepRows[i].id;
+    let j = i + 1;
+    while (j < pathStepRows.length - 1 && !pathStepRows[j].powerRestriction) {
+      j++;
+    }
+    const to = pathStepRows[j].id;
+    result.push({ from, to, value: code });
+  }
+
+  return result;
+};
+
 /** Build a TrainSchedule object from a Train with updated path and schedule. */
 export const buildUpdatedOccurrence = ({
   selectedTrain,
   updatedPath,
   updatedSchedule,
   trainName,
+  powerRestrictions,
 }: {
   selectedTrain: Train;
   updatedPath: PathItem[];
   updatedSchedule: ScheduleItem[];
   trainName: string;
+  powerRestrictions?: PowerRestrictionItem[];
 }): TrainSchedule => ({
   category: selectedTrain.category,
   comfort: selectedTrain.comfort,
@@ -210,7 +267,7 @@ export const buildUpdatedOccurrence = ({
   margins: selectedTrain.margins,
   options: selectedTrain.options,
   path: updatedPath,
-  power_restrictions: selectedTrain.power_restrictions,
+  power_restrictions: powerRestrictions ?? selectedTrain.power_restrictions,
   rolling_stock_name: selectedTrain.rolling_stock_name,
   schedule: updatedSchedule,
   speed_limit_tag: selectedTrain.speed_limit_tag,