From df7a118286e6190deefd84e3a441dc29ae516f6e Mon Sep 17 00:00:00 2001
From: Paul Elliott <paul.elliott@kitware.com>
Date: Mon, 13 May 2024 18:24:44 -0400
Subject: [PATCH] fix: apply reset views button to slices and volume views

fixes #591
---
 src/components/SliceViewer.vue                   |  5 ++++-
 src/components/VolumeRendering.vue               | 16 ++++++++--------
 src/components/VolumeViewer.vue                  |  3 +++
 src/composables/useWindowingConfigInitializer.ts | 12 +++++++++++-
 4 files changed, 26 insertions(+), 10 deletions(-)

diff --git a/src/components/SliceViewer.vue b/src/components/SliceViewer.vue
index 87086efe..6a20efb3 100644
--- a/src/components/SliceViewer.vue
+++ b/src/components/SliceViewer.vue
@@ -143,6 +143,7 @@
 
 <script setup lang="ts">
 import { ref, toRefs, computed } from 'vue';
+import { storeToRefs } from 'pinia';
 import { useCurrentImage } from '@/src/composables/useCurrentImage';
 import { LPSAxisDir } from '@/src/types/lps';
 import { getLPSAxisFromDir } from '@/src/utils/lps';
@@ -175,7 +176,7 @@ import VtkSliceViewSlicingManipulator from '@/src/components/vtk/VtkSliceViewSli
 import VtkMouseInteractionManipulator from '@/src/components/vtk/VtkMouseInteractionManipulator.vue';
 import vtkMouseCameraTrackballPanManipulator from '@kitware/vtk.js/Interaction/Manipulators/MouseCameraTrackballPanManipulator';
 import vtkMouseCameraTrackballZoomToMouseManipulator from '@kitware/vtk.js/Interaction/Manipulators/MouseCameraTrackballZoomToMouseManipulator';
-import { storeToRefs } from 'pinia';
+import { useResetViewsEvents } from '@/src/components/tools/ResetViews.vue';
 
 interface Props extends LayoutViewProps {
   viewDirection: LPSAxisDir;
@@ -196,6 +197,8 @@ function resetCamera() {
   vtkView.value.resetCamera();
 }
 
+useResetViewsEvents().onClick(resetCamera);
+
 useWebGLWatchdog(vtkView);
 useViewAnimationListener(vtkView, viewId, viewType);
 
diff --git a/src/components/VolumeRendering.vue b/src/components/VolumeRendering.vue
index 44d84465..6071f2a2 100644
--- a/src/components/VolumeRendering.vue
+++ b/src/components/VolumeRendering.vue
@@ -16,6 +16,7 @@ import vtkPiecewiseFunctionProxy from '@kitware/vtk.js/Proxy/Core/PiecewiseFunct
 import vtkColorTransferFunction from '@kitware/vtk.js/Rendering/Core/ColorTransferFunction';
 import { onVTKEvent } from '@/src/composables/onVTKEvent';
 import useViewAnimationStore from '@/src/store/view-animation';
+import { useResetViewsEvents } from '@/src/components/tools/ResetViews.vue';
 import { useResizeObserver } from '../composables/useResizeObserver';
 import { useCurrentImage } from '../composables/useCurrentImage';
 import useVolumeColoringStore from '../store/view-configs/volume-coloring';
@@ -284,15 +285,14 @@ export default defineComponent({
     onKeyDown('Control', () => pwfWidget.setShiftOpacityValues(true));
     onKeyUp('Control', () => pwfWidget.setShiftOpacityValues(false));
 
+    const reset = () => {
+      rangeShift.value = 0;
+      rangeWidth.value = fullMappingRangeWidth.value;
+    };
     // reset case
-    watch(
-      [selectedPreset, currentImageID],
-      () => {
-        rangeShift.value = 0;
-        rangeWidth.value = fullMappingRangeWidth.value;
-      },
-      { immediate: true }
-    );
+    watch([selectedPreset, currentImageID], reset, { immediate: true });
+
+    useResetViewsEvents().onClick(reset);
 
     watch([rangeShift, rangeWidth], ([shift, width]) => {
       const imageID = currentImageID.value;
diff --git a/src/components/VolumeViewer.vue b/src/components/VolumeViewer.vue
index 4357cba9..a3289eec 100644
--- a/src/components/VolumeViewer.vue
+++ b/src/components/VolumeViewer.vue
@@ -73,6 +73,7 @@ import { useWebGLWatchdog } from '@/src/composables/useWebGLWatchdog';
 import VtkOrientationMarker from '@/src/components/vtk/VtkOrientationMarker.vue';
 import ViewOverlayGrid from '@/src/components/ViewOverlayGrid.vue';
 import useVolumeColoringStore from '@/src/store/view-configs/volume-coloring';
+import { useResetViewsEvents } from '@/src/components/tools/ResetViews.vue';
 
 interface Props extends LayoutViewProps {
   viewDirection: LPSAxisDir;
@@ -91,6 +92,8 @@ function resetCamera() {
   vtkView.value.renderer.updateLightsGeometryToFollowCamera();
 }
 
+useResetViewsEvents().onClick(resetCamera);
+
 useWebGLWatchdog(vtkView);
 useViewAnimationListener(vtkView, viewId, viewType);
 
diff --git a/src/composables/useWindowingConfigInitializer.ts b/src/composables/useWindowingConfigInitializer.ts
index 32971db8..c2ef802a 100644
--- a/src/composables/useWindowingConfigInitializer.ts
+++ b/src/composables/useWindowingConfigInitializer.ts
@@ -1,3 +1,4 @@
+import { MaybeRef, computed, unref, watch } from 'vue';
 import type { TypedArray } from '@kitware/vtk.js/types';
 import vtkDataArray from '@kitware/vtk.js/Common/Core/DataArray';
 import { watchImmediate } from '@vueuse/core';
@@ -7,7 +8,7 @@ import { WLAutoRanges, WL_AUTO_DEFAULT, WL_HIST_BINS } from '@/src/constants';
 import { getWindowLevels, useDICOMStore } from '@/src/store/datasets-dicom';
 import useWindowingStore from '@/src/store/view-configs/windowing';
 import { Maybe } from '@/src/types';
-import { MaybeRef, computed, unref, watch } from 'vue';
+import { useResetViewsEvents } from '@/src/components/tools/ResetViews.vue';
 
 function useAutoRangeValues(imageID: MaybeRef<Maybe<string>>) {
   const { imageData } = useImage(imageID);
@@ -139,4 +140,13 @@ export function useWindowingConfigInitializer(
     });
     store.resetWindowLevel(viewIdVal, imageIdVal);
   });
+
+  useResetViewsEvents().onClick(() => {
+    const imageIdVal = unref(imageID);
+    const viewIdVal = unref(viewID);
+    if (imageIdVal == null || windowConfig.value == null) {
+      return;
+    }
+    store.resetWindowLevel(viewIdVal, imageIdVal);
+  });
 }