feat(Segmentation): download RTSS from Labelmap(#3692)

Co-authored-by: Alireza <ar.sedghi@gmail.com>

extensions/cornerstone-dicom-seg/src/commandsModule.ts

@@ -2,9 +2,16 @@ import dcmjs from 'dcmjs';
 import { createReportDialogPrompt } from '@ohif/extension-default';
 import { ServicesManager, Types } from '@ohif/core';
 import { cache, metaData } from '@cornerstonejs/core';
-import { segmentation as cornerstoneToolsSegmentation } from '@cornerstonejs/tools';
-import { adaptersSEG, helpers } from '@cornerstonejs/adapters';
-import { DicomMetadataStore } from '@ohif/core';
+import {
+  segmentation as cornerstoneToolsSegmentation,
+  Enums as cornerstoneToolsEnums,
+} from '@cornerstonejs/tools';
+import { adaptersRT, helpers, adaptersSEG } from '@cornerstonejs/adapters';
+import { classes, DicomMetadataStore } from '@ohif/core';
+
+import vtkImageMarchingSquares from '@kitware/vtk.js/Filters/General/ImageMarchingSquares';
+import vtkDataArray from '@kitware/vtk.js/Common/Core/DataArray';
+import vtkImageData from '@kitware/vtk.js/Common/DataModel/ImageData';
 
 import {
   updateViewportsForSegmentationRendering,
@@ -12,12 +19,20 @@ import {
   getTargetViewport,
 } from './utils/hydrationUtils';
 
+const { datasetToBlob } = dcmjs.data;
+
 const {
   Cornerstone3D: {
     Segmentation: { generateLabelMaps2DFrom3D, generateSegmentation },
   },
 } = adaptersSEG;
 
+const {
+  Cornerstone3D: {
+    RTSS: { generateRTSSFromSegmentations },
+  },
+} = adaptersRT;
+
 const { downloadDICOMData } = helpers;
 
 const commandsModule = ({
@@ -348,6 +363,40 @@ const commandsModule = ({
 
       return naturalizedReport;
     },
+    /**
+     * Converts segmentations into RTSS for download.
+     * This sample function retrieves all segentations and passes to
+     * cornerstone tool adapter to convert to DICOM RTSS format. It then
+     * converts dataset to downloadable blob.
+     *
+     */
+    downloadRTSS: ({ segmentationId }) => {
+      const segmentations = segmentationService.getSegmentation(segmentationId);
+      const vtkUtils = {
+        vtkImageMarchingSquares,
+        vtkDataArray,
+        vtkImageData,
+      };
+
+      const RTSS = generateRTSSFromSegmentations(
+        segmentations,
+        classes.MetadataProvider,
+        DicomMetadataStore,
+        cache,
+        cornerstoneToolsEnums,
+        vtkUtils
+      );
+
+      try {
+        const reportBlob = datasetToBlob(RTSS);
+
+        //Create a URL for the binary.
+        const objectUrl = URL.createObjectURL(reportBlob);
+        window.location.assign(objectUrl);
+      } catch (e) {
+        console.warn(e);
+      }
+    },
   };
 
   const definitions = {
@@ -372,6 +421,9 @@ const commandsModule = ({
     storeSegmentation: {
       commandFn: actions.storeSegmentation,
     },
+    downloadRTSS: {
+      commandFn: actions.downloadRTSS,
+    },
   };
 
   return {

extensions/cornerstone-dicom-seg/src/panels/PanelSegmentation.tsx

@@ -1,7 +1,7 @@
 import { createReportAsync } from '@ohif/extension-default';
 import React, { useEffect, useState, useCallback } from 'react';
 import PropTypes from 'prop-types';
-import { SegmentationGroupTable } from '@ohif/ui';
+import { SegmentationGroupTable, LegacyButtonGroup, LegacyButton } from '@ohif/ui';
 
 import callInputDialog from './callInputDialog';
 import callColorPickerDialog from './colorPickerDialog';
@@ -215,6 +215,12 @@ export default function PanelSegmentation({
     }
   };
 
+  const onSegmentationDownloadRTSS = segmentationId => {
+    commandsManager.runCommand('downloadRTSS', {
+      segmentationId,
+    });
+  };
+
   return (
     <>
       <div className="ohif-scrollbar flex min-h-0 flex-auto select-none flex-col justify-between overflow-auto">
@@ -227,6 +233,7 @@ export default function PanelSegmentation({
           onSegmentationClick={onSegmentationClick}
           onSegmentationDelete={onSegmentationDelete}
           onSegmentationDownload={onSegmentationDownload}
+          onSegmentationDownloadRTSS={onSegmentationDownloadRTSS}
           storeSegmentation={storeSegmentation}
           onSegmentationEdit={onSegmentationEdit}
           onSegmentClick={onSegmentClick}

extensions/cornerstone-dicom-seg/src/panels/SegmentationToolbox.tsx

@@ -10,9 +10,9 @@ const TOOL_TYPES = {
   SPHERE_BRUSH: 'SphereBrush',
   CIRCULAR_ERASER: 'CircularEraser',
   SPHERE_ERASER: 'SphereEraser',
-  CIRCLE_SCISSOR: 'CircleScissor',
-  RECTANGLE_SCISSOR: 'RectangleScissor',
-  SPHERE_SCISSOR: 'SphereScissor',
+  CIRCLE_SHAPE: 'CircleScissor',
+  RECTANGLE_SHAPE: 'RectangleScissor',
+  SPHERE_SHAPE: 'SphereScissor',
   THRESHOLD_CIRCULAR_BRUSH: 'ThresholdCircularBrush',
   THRESHOLD_SPHERE_BRUSH: 'ThresholdSphereBrush',
 };
@@ -31,7 +31,7 @@ const initialState = {
     brushSize: 15,
     mode: 'CircularEraser', // Can be 'CircularEraser' or 'SphereEraser'
   },
-  Scissors: {
+  Shapes: {
     brushSize: 15,
     mode: 'CircleScissor', // E.g., 'CircleScissor', 'RectangleScissor', or 'SphereScissor'
   },
@@ -301,24 +301,24 @@ function SegmentationToolbox({ servicesManager, extensionManager }) {
           ],
         },
         {
-          name: 'Scissor',
-          icon: 'icon-tool-scissor',
+          name: 'Shapes',
+          icon: 'icon-tool-shape',
           disabled: !toolsEnabled,
           active:
-            state.activeTool === TOOL_TYPES.CIRCLE_SCISSOR ||
-            state.activeTool === TOOL_TYPES.RECTANGLE_SCISSOR ||
-            state.activeTool === TOOL_TYPES.SPHERE_SCISSOR,
-          onClick: () => setToolActive(TOOL_TYPES.CIRCLE_SCISSOR),
+            state.activeTool === TOOL_TYPES.CIRCLE_SHAPE ||
+            state.activeTool === TOOL_TYPES.RECTANGLE_SHAPE ||
+            state.activeTool === TOOL_TYPES.SPHERE_SHAPE,
+          onClick: () => setToolActive(TOOL_TYPES.CIRCLE_SHAPE),
           options: [
             {
               name: 'Mode',
               type: 'radio',
-              value: state.Scissors.mode,
-              id: 'scissor-mode',
+              value: state.Shapes.mode,
+              id: 'shape-mode',
               values: [
-                { value: TOOL_TYPES.CIRCLE_SCISSOR, label: 'Circle' },
-                { value: TOOL_TYPES.RECTANGLE_SCISSOR, label: 'Rectangle' },
-                { value: TOOL_TYPES.SPHERE_SCISSOR, label: 'Sphere' },
+                { value: TOOL_TYPES.CIRCLE_SHAPE, label: 'Circle' },
+                { value: TOOL_TYPES.RECTANGLE_SHAPE, label: 'Rectangle' },
+                { value: TOOL_TYPES.SPHERE_SHAPE, label: 'Sphere' },
               ],
               onChange: value => setToolActive(value),
             },