feat: Keyboard shortcut to delete current image and clear scene

Author: Laurent Chauvin

src/composables/actions.ts

@@ -8,6 +8,7 @@ import { Action } from '../constants';
 import { useKeyboardShortcutsStore } from '../store/keyboard-shortcuts';
 import { useCurrentImage } from './useCurrentImage';
 import { useSliceConfig } from './useSliceConfig';
+import { useDatasetStore } from '../store/datasets';
 
 const applyLabelOffset = (offset: number) => () => {
   const toolToStore = {
@@ -47,6 +48,19 @@ const changeSlice = (offset: number) => () => {
   currentSlice.value += offset;
 };
 
+const clearScene = () => () => {
+  const datasetStore = useDatasetStore();
+  datasetStore.removeAll();
+};
+
+const deleteCurrentImage = () => () => {
+  const datasetStore = useDatasetStore();
+  datasetStore.remove(datasetStore.primaryImageID!);
+
+  // Automatically select next image
+  datasetStore.setPrimarySelection(datasetStore.idsAsSelections[0]);
+};
+
 export const ACTION_TO_FUNC = {
   windowLevel: setTool(Tools.WindowLevel),
   pan: setTool(Tools.Pan),
@@ -65,6 +79,9 @@ export const ACTION_TO_FUNC = {
   decrementLabel: applyLabelOffset(-1),
   incrementLabel: applyLabelOffset(1),
 
+  deleteCurrentImage: deleteCurrentImage(),
+  clearScene: clearScene(),
+
   mergeNewPolygon: () => {}, // acts as a modifier key rather than immediate effect, so no-op
 
   showKeyboardShortcuts,

src/config.ts

@@ -287,6 +287,9 @@ export const ACTION_TO_KEY = {
   decrementLabel: 'q',
   incrementLabel: 'w',
 
+  deleteCurrentImage: 'ctrl+d',
+  clearScene: 'ctrl+w',
+
   showKeyboardShortcuts: '?',
 } satisfies Record<Action, string>;
 

src/constants.ts

@@ -67,6 +67,14 @@ export const ACTIONS = {
     readable: 'Activate next Label',
   },
 
+  deleteCurrentImage: {
+    readable: 'Remove current active image',
+  },
+
+  clearScene: {
+    readable: 'Clear scene',
+  },
+
   mergeNewPolygon: {
     readable: 'Hold to merge new polygons with overlapping polygons',
   },

src/store/datasets.ts

@@ -12,6 +12,7 @@ import { useFileStore } from './datasets-files';
 import { StateFile } from '../io/state-file/schema';
 import { useErrorMessage } from '../composables/useErrorMessage';
 import { useLayersStore } from './datasets-layers';
+import { useModelStore } from './datasets-models';
 
 export const DataType = {
   Image: 'Image',
@@ -23,6 +24,7 @@ export const useDatasetStore = defineStore('dataset', () => {
   const dicomStore = useDICOMStore();
   const fileStore = useFileStore();
   const layersStore = useLayersStore();
+  const modelStore = useModelStore();
 
   // --- state --- //
 
@@ -81,6 +83,19 @@ export const useDatasetStore = defineStore('dataset', () => {
     layersStore.remove(id);
   };
 
+  const removeAll = () => {
+    // Create a copy to avoid iteration issue while removing data
+    const imageIdCopy = [...imageStore.idList];
+    imageIdCopy.forEach((id) => {
+      remove(id);
+    });
+
+    const modelIdCopy = [...modelStore.idList];
+    modelIdCopy.forEach((id) => {
+      remove(id);
+    });
+  };
+
   return {
     primaryImageID,
     primarySelection,
@@ -89,5 +104,6 @@ export const useDatasetStore = defineStore('dataset', () => {
     setPrimarySelection,
     serialize,
     remove,
+    removeAll,
   };
 });