chore: import string

cognitedata · Aug 15, 2023 · 36a8f53 · 36a8f53
1 parent e18d7f3
commit 36a8f53
Showing 1 changed file with 4 additions and 10 deletions.
diff --git a/react-components/src/components/RevealToolbar/FitModelsButton.tsx b/react-components/src/components/RevealToolbar/FitModelsButton.tsx
@@ -4,21 +4,15 @@
 
 import { type ReactElement, useCallback } from 'react';
 
-import { Box3 } from 'three';
-
-import { useReveal } from '../RevealContainer/RevealContext';
 import { Button } from '@cognite/cogs.js';
+import { useCameraNavigation } from '../../hooks/useCameraNavigation';
 
 export const FitModelsButton = (): ReactElement => {
-  const viewer = useReveal();
+  const cameraNavigation = useCameraNavigation();
 
   const updateCamera = useCallback(() => {
-    const box = new Box3();
-
-    viewer.models.forEach((model) => box.union(model.getModelBoundingBox()));
-
-    viewer.cameraManager.fitCameraToBoundingBox(box);
-  }, [viewer, ...viewer.models]);
+    cameraNavigation.fitCameraToAllModels();
+  }, []);
 
   return (
     <Button