From b6281576355ec8478f9696b1b769cb05d902ae1b Mon Sep 17 00:00:00 2001
From: Pramod S <pramod.s@cognite.com>
Date: Wed, 28 Aug 2024 09:02:53 +0200
Subject: [PATCH] fixed camera state when Home button is selected in active 360
 image mode

---
 .../src/components/RevealToolbar/ResetCameraButton.tsx     | 7 ++++++-
 .../src/components/RevealTopbar/RevealTopbar.tsx           | 2 ++
 2 files changed, 8 insertions(+), 1 deletion(-)

diff --git a/react-components/src/components/RevealToolbar/ResetCameraButton.tsx b/react-components/src/components/RevealToolbar/ResetCameraButton.tsx
index 1407e6636db..7458076938e 100644
--- a/react-components/src/components/RevealToolbar/ResetCameraButton.tsx
+++ b/react-components/src/components/RevealToolbar/ResetCameraButton.tsx
@@ -7,6 +7,7 @@ import { Button, Tooltip as CogsTooltip } from '@cognite/cogs.js';
 import { useTranslation } from '../i18n/I18n';
 import { useCameraNavigation } from '../../hooks/useCameraNavigation';
 import { useSceneDefaultCamera } from '../../hooks/useSceneDefaultCamera';
+import { useReveal } from '../RevealCanvas/ViewerContext';
 
 type ResetCameraButtonProps = {
   sceneExternalId?: string;
@@ -18,16 +19,20 @@ export const ResetCameraButton = ({
   sceneSpaceId
 }: ResetCameraButtonProps): ReactElement => {
   const { t } = useTranslation();
+  const viewer = useReveal();
   const cameraNavigation = useCameraNavigation();
   const resetToDefaultSceneCamera = useSceneDefaultCamera(sceneExternalId, sceneSpaceId);
 
   const resetCameraToHomePosition = useCallback(() => {
+    if (viewer.get360ImageCollections() !== undefined) {
+      viewer.exit360Image();
+    }
     if (sceneExternalId !== undefined && sceneSpaceId !== undefined) {
       resetToDefaultSceneCamera.fitCameraToSceneDefault();
       return;
     }
     cameraNavigation.fitCameraToAllModels();
-  }, [sceneExternalId, sceneSpaceId, cameraNavigation, resetToDefaultSceneCamera]);
+  }, [sceneExternalId, sceneSpaceId, cameraNavigation, resetToDefaultSceneCamera, viewer]);
 
   return (
     <CogsTooltip
diff --git a/react-components/src/components/RevealTopbar/RevealTopbar.tsx b/react-components/src/components/RevealTopbar/RevealTopbar.tsx
index 98483160649..92c1b5e2581 100644
--- a/react-components/src/components/RevealTopbar/RevealTopbar.tsx
+++ b/react-components/src/components/RevealTopbar/RevealTopbar.tsx
@@ -12,6 +12,7 @@ import { Divider } from '@cognite/cogs.js';
 import { SetOrbitOrFirstPersonControlsType } from '../RevealToolbar/SetFlexibleControlsType';
 import { RuleBasedOutputsButton } from '../RevealToolbar/RuleBasedOutputsButton';
 import { LayersButtonStrip } from '../RevealToolbar/LayersContainer/LayersButtonsStrip';
+import { ResetCameraButton } from '../RevealToolbar/ResetCameraButton';
 
 export type CustomTopbarContent = CustomToolbarContent & { topbarContent?: ReactNode };
 
@@ -36,6 +37,7 @@ const DefaultContentWrapper = (props: CustomTopbarContent): ReactElement => {
           highQualitySettings={props.highFidelitySettings}
         />
       </FlexSection>
+      <ResetCameraButton />
     </>
   );
 };