fron: lmr consist prefill mass length max speed

Signed-off-by: Egor Berezovskiy <[email protected]>

Author: Wadjetz

front/src/applications/stdcm/components/StdcmForm/StdcmConsist.tsx

@@ -52,6 +52,7 @@ const StdcmConsist = ({ disabled = false }: StdcmConfigCardProps) => {
     onTotalLengthChange,
     maxSpeed,
     onMaxSpeedChange,
+    prefillConsist,
   } = useStdcmConsist();
 
   const { filters, searchRollingStock, searchRollingStockById, filteredRollingStockList } =
@@ -91,6 +92,9 @@ const StdcmConsist = ({ disabled = false }: StdcmConfigCardProps) => {
   };
 
   const onSelectSuggestion = (option?: LightRollingStockWithLiveries) => {
+    if (option) {
+      prefillConsist(option);
+    }
     dispatch(updateRollingStockID(option?.id));
   };
 
@@ -144,6 +148,9 @@ const StdcmConsist = ({ disabled = false }: StdcmConfigCardProps) => {
           suggestions={filteredTowedRollingStockList}
           getSuggestionLabel={(suggestion: TowedRollingStock) => suggestion.name}
           onSelectSuggestion={(towed) => {
+            if (towed) {
+              prefillConsist(rollingStock, towed);
+            }
             dispatch(updateTowedRollingStockID(towed?.id));
           }}
         />

front/src/applications/stdcm/hooks/useStdcmConsist.ts

@@ -1,12 +1,21 @@
+import { useState } from 'react';
+
 import { useSelector } from 'react-redux';
 
+import type { LightRollingStockWithLiveries, TowedRollingStock } from 'common/api/osrdEditoastApi';
 import { useOsrdConfActions, useOsrdConfSelectors } from 'common/osrdContext';
 import { type StdcmConfSliceActions } from 'reducers/osrdconf/stdcmConf';
 import type { StdcmConfSelectors } from 'reducers/osrdconf/stdcmConf/selectors';
 import { useAppDispatch } from 'store';
+import { kgToT } from 'utils/physics';
 
 const useStdcmConsist = () => {
   const dispatch = useAppDispatch();
+
+  const [totalMassChanged, setTotalMassChanged] = useState(false);
+  const [totalLengthChanged, setTotalLengthChanged] = useState(false);
+  const [maxSpeedChanged, setMaxSpeedChanged] = useState(false);
+
   const { getTotalMass, getTotalLength, getMaxSpeed } =
     useOsrdConfSelectors() as StdcmConfSelectors;
   const { updateTotalMass, updateTotalLength, updateMaxSpeed } =
@@ -15,28 +24,51 @@ const useStdcmConsist = () => {
   const totalMass = useSelector(getTotalMass);
   const onTotalMassChange = (e: React.ChangeEvent<HTMLInputElement>) => {
     const totalMassValue = Number(e.target.value);
+    setTotalMassChanged(true);
     dispatch(updateTotalMass(totalMassValue === 0 ? undefined : totalMassValue));
   };
 
   const totalLength = useSelector(getTotalLength);
   const onTotalLengthChange = (e: React.ChangeEvent<HTMLInputElement>) => {
     const totalLengthValue = Number(e.target.value);
+    setTotalLengthChanged(true);
     dispatch(updateTotalLength(totalLengthValue === 0 ? undefined : totalLengthValue));
   };
 
   const maxSpeed = useSelector(getMaxSpeed);
   const onMaxSpeedChange = (e: React.ChangeEvent<HTMLInputElement>) => {
     const totalMaxSpeed = Number(e.target.value);
+    setMaxSpeedChanged(true);
     dispatch(updateMaxSpeed(totalMaxSpeed === 0 ? undefined : totalMaxSpeed));
   };
 
+  const prefillConsist = (
+    rollingStock?: LightRollingStockWithLiveries,
+    towed?: TowedRollingStock
+  ) => {
+    if (!totalMassChanged) {
+      dispatch(updateTotalMass(Math.floor(kgToT((rollingStock?.mass ?? 0) + (towed?.mass ?? 0)))));
+    }
+
+    if (!totalLengthChanged) {
+      dispatch(updateTotalLength(Math.floor((rollingStock?.length ?? 0) + (towed?.length ?? 0))));
+    }
+
+    if (!maxSpeedChanged) {
+      dispatch(
+        updateMaxSpeed(rollingStock?.max_speed ? Math.floor(rollingStock.max_speed) : undefined)
+      );
+    }
+  };
+
   return {
     totalMass,
     onTotalMassChange,
     totalLength,
     onTotalLengthChange,
     maxSpeed,
     onMaxSpeedChange,
+    prefillConsist,
   };
 };
 

front/src/utils/physics.ts

@@ -101,3 +101,11 @@ export function decimalToPercentage(value: number) {
 export function tToKg(value: number) {
   return value * 1000;
 }
+
+/**
+ * ex: converts 12000kg to 12t
+ * @param value in kg
+ */
+export function kgToT(value: number) {
+  return value / 1000;
+}