diff --git a/react-components/src/components/RevealToolbar/FitModelsButton.tsx b/react-components/src/components/RevealToolbar/FitModelsButton.tsx index 242b4e2c662..00d3fbf78f7 100644 --- 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 (