From f7c886878dc2f8f22c9f5c8cc34952d26f866a3a Mon Sep 17 00:00:00 2001
From: dxlin <dxlin@users.noreply.github.com>
Date: Tue, 24 Jan 2023 15:43:27 -0500
Subject: [PATCH] fix(Hotkeys): add firstImage and lastImage commands
 referenced by 'home' and 'end' hotkeys (#3127)

* -fix 'home' and 'end' hotkeys
  ->implement firstImage lastImage commands

* -extra comments

* -add check for compatible viewport in firstImage command
---
 extensions/cornerstone/src/commandsModule.ts | 56 ++++++++++++++++++++
 1 file changed, 56 insertions(+)

diff --git a/extensions/cornerstone/src/commandsModule.ts b/extensions/cornerstone/src/commandsModule.ts
index c7839706b..585f1bf74 100644
--- a/extensions/cornerstone/src/commandsModule.ts
+++ b/extensions/cornerstone/src/commandsModule.ts
@@ -1,6 +1,7 @@
 import {
   getEnabledElement,
   StackViewport,
+  VolumeViewport,
   utilities as csUtils,
 } from '@cornerstonejs/core';
 import {
@@ -313,6 +314,51 @@ const commandsModule = ({ servicesManager }) => {
         }
       }
     },
+    firstImage: () => {
+      // Get current active viewport (return if none active)
+      const enabledElement = _getActiveViewportEnabledElement();
+      if (!enabledElement) {
+        return;
+      }
+      const { viewport } = enabledElement;
+
+      // Check viewport is supported
+      if (
+        viewport! instanceof StackViewport &&
+        viewport! instanceof VolumeViewport
+      ) {
+        throw new Error('Unsupported viewport type');
+      }
+
+      // Set slice to first slice
+      const options = { imageIndex: 0 };
+      cstUtils.jumpToSlice(viewport.element, options);
+    },
+    lastImage: () => {
+      // Get current active viewport (return if none active)
+      const enabledElement = _getActiveViewportEnabledElement();
+      if (!enabledElement) {
+        return;
+      }
+      const { viewport } = enabledElement;
+
+      // Get number of slices
+      // -> Copied from cornerstone3D jumpToSlice\_getImageSliceData()
+      let numberOfSlices = 0;
+
+      if (viewport instanceof StackViewport) {
+        numberOfSlices = viewport.getImageIds().length;
+      } else if (viewport instanceof VolumeViewport) {
+        numberOfSlices = csUtils.getImageSliceDataForVolumeViewport(viewport)
+          .numberOfSlices;
+      } else {
+        throw new Error('Unsupported viewport type');
+      }
+
+      // Set slice to last slice
+      const options = { imageIndex: numberOfSlices - 1 };
+      cstUtils.jumpToSlice(viewport.element, options);
+    },
     scroll: ({ direction }) => {
       const enabledElement = _getActiveViewportEnabledElement();
 
@@ -475,6 +521,16 @@ const commandsModule = ({ servicesManager }) => {
       storeContexts: [],
       options: { direction: -1 },
     },
+    firstImage: {
+      commandFn: actions.firstImage,
+      storeContexts: [],
+      options: {},
+    },
+    lastImage: {
+      commandFn: actions.lastImage,
+      storeContexts: [],
+      options: {},
+    },
     showDownloadViewportModal: {
       commandFn: actions.showDownloadViewportModal,
       storeContexts: [],