feat(OperationClass): Add areaInHectare prop

treely · Nov 23, 2023 · 2e43ddc · 2e43ddc
1 parent 2ce0e35
commit 2e43ddc
Show file tree

Hide file tree

Showing 3 changed files with 40 additions and 4 deletions.
diff --git a/src/components/OperationClass/OperationClass.stories.tsx b/src/components/OperationClass/OperationClass.stories.tsx
@@ -36,3 +36,23 @@ export const Default: Story = {
     rotationPeriod: feasibilityStudyConfigMock.operationClasses[0].rotationPeriod,
   },
 };
+
+export const WithAreaInHectarePerSpecies: Story = {
+  args: {
+    listOfSpecies: feasibilityStudyConfigMock.operationClasses[0].listOfSpecies.map((species) => ({
+      title: species.title,
+      siteProductivity: species.siteProductivity,
+      yieldTable: species.yieldTable,
+      areaInHectare: 200 * species.areaInPercent,
+    })),
+    rotationPeriod: feasibilityStudyConfigMock.operationClasses[0].rotationPeriod,
+  },
+};
+
+export const WithTotalAreaInHectare: Story = {
+  args: {
+    listOfSpecies: feasibilityStudyConfigMock.operationClasses[0].listOfSpecies,
+    rotationPeriod: feasibilityStudyConfigMock.operationClasses[0].rotationPeriod,
+    areaInHectare: 200,
+  },
+};
diff --git a/src/components/OperationClass/OperationClass.tsx b/src/components/OperationClass/OperationClass.tsx
@@ -37,16 +37,31 @@ interface SpeciesWithAreaInHectare extends Species, AreaInHectare {}
 
 export interface OperationClassProps {
   rotationPeriod: number;
+  areaInHectare?: number;
   listOfSpecies: SpeciesWithPercentageArea[] | SpeciesWithAreaInHectare[];
 }
 
 export const OperationClass: React.FC<OperationClassProps> = ({
   rotationPeriod,
+  areaInHectare = 0,
   listOfSpecies,
 }: OperationClassProps) => {
   const { formatNumber, formatMessage } = useContext(IntlContext);
 
   const areasInPercentPerSpecies = useMemo(() => getAreaInPercent(listOfSpecies), [listOfSpecies]);
+  const areaInHectarePerSpecies: AreaInHectare[] = useMemo(() => {
+    if (listOfSpecies[0]?.hasOwnProperty('areaInHectare')) {
+      return listOfSpecies.map((species) => ({
+        areaInHectare: (species as SpeciesWithAreaInHectare).areaInHectare,
+      }));
+    }
+    if (areaInHectare && listOfSpecies[0]?.hasOwnProperty('areaInPercent')) {
+      return listOfSpecies.map((species) => ({
+        areaInHectare: (species as SpeciesWithPercentageArea).areaInPercent * areaInHectare,
+      }));
+    }
+    return [];
+  }, [listOfSpecies, areaInHectare]);
 
   const normalStocks = useMemo(
     () =>
@@ -57,8 +72,8 @@ export const OperationClass: React.FC<OperationClassProps> = ({
   );
 
   const hasAreaInHectare: boolean = useMemo(
-    () => listOfSpecies[0]?.hasOwnProperty('areaInHectare'),
-    [listOfSpecies]
+    () => areaInHectarePerSpecies.length > 0,
+    [areaInHectarePerSpecies]
   );
 
   return (
@@ -104,7 +119,7 @@ export const OperationClass: React.FC<OperationClassProps> = ({
                           id: 'units.ha',
                         },
                         {
-                          value: formatNumber((species as SpeciesWithAreaInHectare).areaInHectare, {
+                          value: formatNumber(areaInHectarePerSpecies[index].areaInHectare, {
                             minimumFractionDigits: 1,
                             maximumFractionDigits: 1,
                           }),
@@ -151,7 +166,7 @@ export const OperationClass: React.FC<OperationClassProps> = ({
                       { id: 'units.ha' },
                       {
                         value: formatNumber(
-                          (listOfSpecies as SpeciesWithAreaInHectare[]).reduce(
+                          areaInHectarePerSpecies.reduce(
                             (sum, value) => sum + value.areaInHectare,
                             0
                           ),

diff --git a/src/components/OperationClasses/OperationClasses.tsx b/src/components/OperationClasses/OperationClasses.tsx
@@ -16,6 +16,7 @@ export const OperationClasses = ({ config }: OperationClassesProps): JSX.Element
             <Heading>{operationClass.name}</Heading>
             <OperationClass
               rotationPeriod={operationClass.rotationPeriod}
+              areaInHectare={operationClass.areaInHectare}
               listOfSpecies={operationClass.listOfSpecies}
             />
             <Spacer height="4" />