From 58941331dce53f4177ad184baab2ea8e7a4271b1 Mon Sep 17 00:00:00 2001
From: Mathias Oppedal Heggelund <98742460+mheggelund@users.noreply.github.com>
Date: Wed, 19 Jun 2024 15:47:16 +0200
Subject: [PATCH] feat: Add mouthbar as option for compute object. (#299)

---
 .../CaseCardComponent/FormattedMethodNames.tsx  |  2 ++
 src/features/Compute/CaseGroup/CaseGroup.tsx    | 17 ++++++++++++-----
 .../Compute/CaseGroup/CaseRow/CaseRow.tsx       | 13 +++++++++++++
 .../CaseGroup/CaseRow/hooks/useSetSaved.tsx     |  3 +++
 .../Compute/ComputeObject/ComputeObject.tsx     | 17 +++++++++++++----
 5 files changed, 43 insertions(+), 9 deletions(-)

diff --git a/src/components/CaseCardComponent/FormattedMethodNames.tsx b/src/components/CaseCardComponent/FormattedMethodNames.tsx
index 140291ec..c974fabd 100644
--- a/src/components/CaseCardComponent/FormattedMethodNames.tsx
+++ b/src/components/CaseCardComponent/FormattedMethodNames.tsx
@@ -1,4 +1,6 @@
 export const titleMapping: Record<string, string> = {
+  Channel: 'Channel',
+  Mouthbar: 'Mouthbar',
   Indicator: 'Indicator',
   'Net-To-Gross': 'Net-to-gross',
   ContiniousParameter: 'Continious parameter',
diff --git a/src/features/Compute/CaseGroup/CaseGroup.tsx b/src/features/Compute/CaseGroup/CaseGroup.tsx
index 712d6c15..e66854c0 100644
--- a/src/features/Compute/CaseGroup/CaseGroup.tsx
+++ b/src/features/Compute/CaseGroup/CaseGroup.tsx
@@ -137,7 +137,10 @@ export const CaseGroup = ({
       const setListItem = (methodType: string, newCase: ComputeCaseDto) => {
         const filteredList = filerLocalList(methodType);
 
-        if (methodType === 'Channel' && filteredList.length < 1) {
+        if (
+          (methodType === 'Channel' || methodType === 'Mouthbar') &&
+          filteredList.length < 1
+        ) {
           setLocalList([...localList, newCase]);
         } else if (
           methodType !== 'Channel' &&
@@ -325,7 +328,7 @@ export const CaseGroup = ({
 
   return (
     <>
-      {methodName === 'Channel' && (
+      {(methodName === 'Channel' || methodName === 'Mouthbar') && (
         <Styled.ButtonDiv>
           <Styled.ButtonGroup>
             <Tooltip
@@ -335,7 +338,7 @@ export const CaseGroup = ({
             >
               <Button
                 variant="outlined"
-                onClick={() => addCase('Channel')}
+                onClick={() => addCase(methodName)}
                 disabled={localList.length >= 1}
               >
                 <Icon data={ADD} size={18}></Icon>
@@ -353,7 +356,7 @@ export const CaseGroup = ({
         localList={localList}
       >
         <Styled.CaseList>
-          {methodName === 'Channel' ? (
+          {methodName === 'Channel' || methodName === 'Mouthbar' ? (
             <>
               {caseList.concat(localList).map((c, index) => (
                 <CaseRow
@@ -362,7 +365,11 @@ export const CaseGroup = ({
                   id={c.computeCaseId}
                   allCasesList={caseList.concat(localList)}
                   caseList={caseList}
-                  caseType={methodName === 'Channel' ? 'Object' : 'Variogram'}
+                  caseType={
+                    methodName === 'Channel' || methodName === 'Mouthbar'
+                      ? 'Object'
+                      : 'Variogram'
+                  }
                   saveCase={saveCase}
                   deleteCase={deleteCase}
                   setAlertMessage={setAlertMessage}
diff --git a/src/features/Compute/CaseGroup/CaseRow/CaseRow.tsx b/src/features/Compute/CaseGroup/CaseRow/CaseRow.tsx
index c6ba7ff4..15284782 100644
--- a/src/features/Compute/CaseGroup/CaseRow/CaseRow.tsx
+++ b/src/features/Compute/CaseGroup/CaseRow/CaseRow.tsx
@@ -333,6 +333,19 @@ export const CaseRow = ({
           </Styled.AutocompleteWrapper>
         )}
 
+        {rowCase.computeMethod.name === 'Mouthbar' && (
+          <Styled.AutocompleteWrapper>
+            <ModelAreaSelect
+              disableSelect={saved}
+              modelAreas={areaList ? areaList : []}
+              selectedModelArea={selectedRowArea(rowCase.computeCaseId)}
+              setModelArea={setModelArea}
+              existingCases={caseList}
+              caseError={caseError}
+            />
+          </Styled.AutocompleteWrapper>
+        )}
+
         <CaseButtons
           id={id}
           caseType={caseType === 'Object' ? 'Object' : 'Variogram'}
diff --git a/src/features/Compute/CaseGroup/CaseRow/hooks/useSetSaved.tsx b/src/features/Compute/CaseGroup/CaseRow/hooks/useSetSaved.tsx
index 151f8fdc..a75e7780 100644
--- a/src/features/Compute/CaseGroup/CaseRow/hooks/useSetSaved.tsx
+++ b/src/features/Compute/CaseGroup/CaseRow/hooks/useSetSaved.tsx
@@ -13,6 +13,9 @@ export const useSetSaved = (
       if (r.computeCaseId === id && r.computeMethod.name === 'Channel') {
         setSaved(false);
       }
+      if (r.computeCaseId === id && r.computeMethod.name === 'Mouthbar') {
+        setSaved(false);
+      }
     }
 
     allCasesList
diff --git a/src/pages/ModelPages/Compute/ComputeObject/ComputeObject.tsx b/src/pages/ModelPages/Compute/ComputeObject/ComputeObject.tsx
index 528dcda7..e67a5413 100644
--- a/src/pages/ModelPages/Compute/ComputeObject/ComputeObject.tsx
+++ b/src/pages/ModelPages/Compute/ComputeObject/ComputeObject.tsx
@@ -58,9 +58,9 @@ export const ComputeObject = () => {
     (method) => method.computeMethod.name === 'Channel',
   );
 
-  // const mouthbar = data?.data.filter(
-  //   (method) => method.computeMethod.name === 'Mouthbar',
-  // )
+  const mouthbar = data?.data.filter(
+    (method) => method.computeMethod.name === 'Mouthbar',
+  );
 
   return (
     <>
@@ -68,7 +68,16 @@ export const ComputeObject = () => {
         <ComputeHeader caseInfo={ObjectCaseInfo} />
         <CaseGroup
           caseList={channel !== undefined && channel.length > 0 ? channel : []}
-          methodName={ObjectCaseInfo.type}
+          methodName="Channel"
+          setAlertMessage={setAlertMessage}
+          runCase={runComputeObject}
+        />
+
+        <CaseGroup
+          caseList={
+            mouthbar !== undefined && mouthbar.length > 0 ? mouthbar : []
+          }
+          methodName="Mouthbar"
           setAlertMessage={setAlertMessage}
           runCase={runComputeObject}
         />