@cornerstonejs/tools 5.0.0-beta.1 → 5.0.1

package/dist/esm/eventListeners/touch/getTouchEventPoints.js

@@ -1,13 +1,20 @@
 import { getEnabledElement } from '@cornerstonejs/core';
 export default function getTouchEventPoints(evt, element) {
     const elementToUse = element || evt.currentTarget;
+    const { viewport } = getEnabledElement(elementToUse) || {};
+    if (!viewport) {
+        return [];
+    }
     const touches = evt.type === 'touchend' ? evt.changedTouches : evt.touches;
-    return Object.keys(touches).map((i) => {
+    return Object.keys(touches)
+        .map((i) => {
         const clientPoint = _clientToPoint(touches[i]);
         const pagePoint = _pageToPoint(touches[i]);
         const canvasPoint = _pagePointsToCanvasPoints(elementToUse, pagePoint);
-        const { viewport } = getEnabledElement(elementToUse);
-        const worldPoint = viewport.canvasToWorld(canvasPoint);
+        const worldPoint = getWorldPoint(viewport, canvasPoint);
+        if (!worldPoint) {
+            return;
+        }
         return {
             page: pagePoint,
             client: clientPoint,
@@ -21,7 +28,8 @@ export default function getTouchEventPoints(evt, element) {
                 rotationAngle: touches[i].rotationAngle,
             },
         };
-    });
+    })
+        .filter(Boolean);
 }
 function _pagePointsToCanvasPoints(element, pagePoint) {
     const rect = element.getBoundingClientRect();
@@ -36,3 +44,18 @@ function _pageToPoint(touch) {
 function _clientToPoint(touch) {
     return [touch.clientX, touch.clientY];
 }
+function getWorldPoint(viewport, canvasPoint) {
+    try {
+        return viewport.canvasToWorld(canvasPoint);
+    }
+    catch (error) {
+        if (isNoMountedDataError(error)) {
+            return;
+        }
+        throw error;
+    }
+}
+function isNoMountedDataError(error) {
+    return (error instanceof Error &&
+        error.message.includes('because no data is mounted'));
+}

package/dist/esm/eventListeners/touch/touchStartListener.js

@@ -1,4 +1,4 @@
-import { getEnabledElement, triggerEvent } from '@cornerstonejs/core';
+import { getEnabledElement, triggerEvent, utilities, } from '@cornerstonejs/core';
 import Events from '../../enums/Events';
 import { Swipe } from '../../enums/Touch';
 import getTouchEventPoints from './getTouchEventPoints';
@@ -59,8 +59,8 @@ const defaultTapState = {
     tapMaxDistance: 24,
     tapToleranceMs: 300,
 };
-let state = JSON.parse(JSON.stringify(defaultState));
-let tapState = JSON.parse(JSON.stringify(defaultTapState));
+let state = utilities.deepClone(defaultState);
+let tapState = utilities.deepClone(defaultTapState);
 function triggerEventCallback(ele, name, eventDetail) {
     return triggerEvent(ele, name, eventDetail);
 }
@@ -194,7 +194,7 @@ function _onTouchEnd(evt) {
     };
     triggerEventCallback(eventDetail.element, TOUCH_END, eventDetail);
     _checkTouchTap(evt);
-    state = JSON.parse(JSON.stringify(defaultState));
+    state = utilities.deepClone(defaultState);
     document.removeEventListener('touchmove', _onTouchDrag);
     document.removeEventListener('touchend', _onTouchEnd);
 }
@@ -236,7 +236,7 @@ function _checkTouchTap(evt) {
             taps: tapState.taps,
         };
         triggerEventCallback(eventDetail.element, TOUCH_TAP, eventDetail);
-        tapState = JSON.parse(JSON.stringify(defaultTapState));
+        tapState = utilities.deepClone(defaultTapState);
     }, tapState.tapToleranceMs);
 }
 function _checkTouchSwipe(evt, deltaPoints) {
@@ -267,9 +267,22 @@ function _checkTouchSwipe(evt, deltaPoints) {
     }
 }
 function _updateTouchEventsLastPoints(element, lastPoints) {
-    const { viewport } = getEnabledElement(element);
-    return lastPoints.map((lp) => {
-        const world = viewport.canvasToWorld(lp.canvas);
+    const { viewport } = getEnabledElement(element) || {};
+    if (!viewport) {
+        return lastPoints;
+    }
+    return lastPoints
+        .map((lp) => {
+        let world;
+        try {
+            world = viewport.canvasToWorld(lp.canvas);
+        }
+        catch (error) {
+            if (isNoMountedDataError(error)) {
+                return;
+            }
+            throw error;
+        }
         return {
             page: lp.page,
             client: lp.client,
@@ -277,6 +290,11 @@ function _updateTouchEventsLastPoints(element, lastPoints) {
             world,
             touch: lp.touch,
         };
-    });
+    })
+        .filter(Boolean);
+}
+function isNoMountedDataError(error) {
+    return (error instanceof Error &&
+        error.message.includes('because no data is mounted'));
 }
 export default touchStartListener;

package/dist/esm/eventListeners/wheel/wheelListener.js

@@ -9,6 +9,10 @@ function wheelListener(evt) {
     if (evt.deltaY > -1 && evt.deltaY < 1) {
         return;
     }
+    const points = getMouseEventPoints(evt);
+    if (!points) {
+        return;
+    }
     evt.preventDefault();
     const { spinX, spinY, pixelX, pixelY } = normalizeWheel(evt);
     const direction = spinY < 0 ? -1 : 1;
@@ -27,7 +31,7 @@ function wheelListener(evt) {
             pixelY,
             direction,
         },
-        points: getMouseEventPoints(evt),
+        points,
     };
     triggerEvent(element, Events.MOUSE_WHEEL, eventDetail);
 }

package/dist/esm/index.d.ts

@@ -12,7 +12,7 @@ import * as Types from './types';
 import * as annotation from './stateManagement/annotation';
 import * as segmentation from './stateManagement/segmentation';
 import * as splines from './tools/annotation/splines';
-import { BaseTool, AnnotationTool, AnnotationDisplayTool, PanTool, TrackballRotateTool, VolumeCroppingTool, VolumeCroppingControlTool, DragProbeTool, WindowLevelTool, ZoomTool, StackScrollTool, SegmentBidirectionalTool, PlanarRotateTool, MIPJumpToClickTool, LabelTool, LengthTool, HeightTool, ProbeTool, RectangleROITool, EllipticalROITool, CircleROITool, ETDRSGridTool, SplineROITool, SplineContourSegmentationTool, BidirectionalTool, PlanarFreehandROITool, PlanarFreehandContourSegmentationTool, LivewireContourTool, LivewireContourSegmentationTool, ArrowAnnotateTool, KeyImageTool, CrosshairsTool, ReferenceLinesTool, RectangleScissorsTool, CircleScissorsTool, SphereScissorsTool, RectangleROIThresholdTool, RectangleROIStartEndThresholdTool, CircleROIStartEndThresholdTool, BrushTool, AngleTool, CobbAngleTool, UltrasoundDirectionalTool, UltrasoundPleuraBLineTool, MagnifyTool, AdvancedMagnifyTool, ReferenceCursors, PaintFillTool, ScaleOverlayTool, OrientationMarkerTool, OverlayGridTool, SegmentationIntersectionTool, EraserTool, SculptorTool, SegmentSelectTool, WindowLevelRegionTool, VolumeRotateTool, RegionSegmentPlusTool, RegionSegmentTool, WholeBodySegmentTool, LabelmapBaseTool, SegmentLabelTool, LabelMapEditWithContourTool } from './tools';
+import { BaseTool, AnnotationTool, AnnotationDisplayTool, PanTool, TrackballRotateTool, VolumeCroppingTool, VolumeCroppingControlTool, DragProbeTool, WindowLevelTool, ZoomTool, StackScrollTool, SegmentBidirectionalTool, PlanarRotateTool, MIPJumpToClickTool, LabelTool, LengthTool, HeightTool, ProbeTool, RectangleROITool, EllipticalROITool, CircleROITool, ETDRSGridTool, SplineROITool, SplineContourSegmentationTool, BidirectionalTool, PlanarFreehandROITool, PlanarFreehandContourSegmentationTool, LivewireContourTool, LivewireContourSegmentationTool, ArrowAnnotateTool, KeyImageTool, CrosshairsTool, ReferenceLinesTool, RectangleScissorsTool, CircleScissorsTool, SphereScissorsTool, RectangleROIThresholdTool, RectangleROIStartEndThresholdTool, CircleROIStartEndThresholdTool, BrushTool, AngleTool, CobbAngleTool, UltrasoundDirectionalTool, UltrasoundPleuraBLineTool, MagnifyTool, AdvancedMagnifyTool, ReferenceCursors, PaintFillTool, ScaleOverlayTool, OrientationMarkerTool, OrientationControllerTool, OverlayGridTool, SegmentationIntersectionTool, EraserTool, SculptorTool, SegmentSelectTool, WindowLevelRegionTool, VolumeRotateTool, RegionSegmentPlusTool, RegionSegmentTool, WholeBodySegmentTool, LabelmapBaseTool, SegmentLabelTool, LabelMapEditWithContourTool } from './tools';
 import VideoRedactionTool from './tools/annotation/VideoRedactionTool';
 import * as Enums from './enums';
-export { VideoRedactionTool, init, destroy, addTool, removeTool, cancelActiveManipulations, BaseTool, AnnotationTool, AnnotationDisplayTool, PanTool, SegmentBidirectionalTool, TrackballRotateTool, VolumeCroppingTool, VolumeCroppingControlTool, DragProbeTool, WindowLevelTool, WindowLevelRegionTool, ZoomTool, StackScrollTool, PlanarRotateTool, MIPJumpToClickTool, LabelTool, LengthTool, HeightTool, CrosshairsTool, ReferenceLinesTool, OverlayGridTool, SegmentationIntersectionTool, ProbeTool, RectangleROITool, EllipticalROITool, CircleROITool, ETDRSGridTool, SplineROITool, SplineContourSegmentationTool, BidirectionalTool, PlanarFreehandROITool, PlanarFreehandContourSegmentationTool, LivewireContourTool, LivewireContourSegmentationTool, ArrowAnnotateTool, AngleTool, CobbAngleTool, UltrasoundDirectionalTool, UltrasoundPleuraBLineTool, KeyImageTool, MagnifyTool, AdvancedMagnifyTool, ReferenceCursors, ScaleOverlayTool, SculptorTool, EraserTool, RectangleScissorsTool, CircleScissorsTool, SphereScissorsTool, RectangleROIThresholdTool, RectangleROIStartEndThresholdTool, CircleROIStartEndThresholdTool, BrushTool, OrientationMarkerTool, SegmentSelectTool, SegmentLabelTool, synchronizers, Synchronizer, SynchronizerManager, PaintFillTool, Types, state, ToolGroupManager, store, Enums, CONSTANTS, drawing, annotation, segmentation, utilities, cursors, VolumeRotateTool, RegionSegmentPlusTool, RegionSegmentTool, WholeBodySegmentTool, LabelmapBaseTool, LabelMapEditWithContourTool, splines, version, };
+export { VideoRedactionTool, init, destroy, addTool, removeTool, cancelActiveManipulations, BaseTool, AnnotationTool, AnnotationDisplayTool, PanTool, SegmentBidirectionalTool, TrackballRotateTool, VolumeCroppingTool, VolumeCroppingControlTool, DragProbeTool, WindowLevelTool, WindowLevelRegionTool, ZoomTool, StackScrollTool, PlanarRotateTool, MIPJumpToClickTool, LabelTool, LengthTool, HeightTool, CrosshairsTool, ReferenceLinesTool, OverlayGridTool, SegmentationIntersectionTool, ProbeTool, RectangleROITool, EllipticalROITool, CircleROITool, ETDRSGridTool, SplineROITool, SplineContourSegmentationTool, BidirectionalTool, PlanarFreehandROITool, PlanarFreehandContourSegmentationTool, LivewireContourTool, LivewireContourSegmentationTool, ArrowAnnotateTool, AngleTool, CobbAngleTool, UltrasoundDirectionalTool, UltrasoundPleuraBLineTool, KeyImageTool, MagnifyTool, AdvancedMagnifyTool, ReferenceCursors, ScaleOverlayTool, SculptorTool, EraserTool, RectangleScissorsTool, CircleScissorsTool, SphereScissorsTool, RectangleROIThresholdTool, RectangleROIStartEndThresholdTool, CircleROIStartEndThresholdTool, BrushTool, OrientationMarkerTool, OrientationControllerTool, SegmentSelectTool, SegmentLabelTool, synchronizers, Synchronizer, SynchronizerManager, PaintFillTool, Types, state, ToolGroupManager, store, Enums, CONSTANTS, drawing, annotation, segmentation, utilities, cursors, VolumeRotateTool, RegionSegmentPlusTool, RegionSegmentTool, WholeBodySegmentTool, LabelmapBaseTool, LabelMapEditWithContourTool, splines, version, };

package/dist/esm/index.js

@@ -12,7 +12,7 @@ import * as Types from './types';
 import * as annotation from './stateManagement/annotation';
 import * as segmentation from './stateManagement/segmentation';
 import * as splines from './tools/annotation/splines';
-import { BaseTool, AnnotationTool, AnnotationDisplayTool, PanTool, TrackballRotateTool, VolumeCroppingTool, VolumeCroppingControlTool, DragProbeTool, WindowLevelTool, ZoomTool, StackScrollTool, SegmentBidirectionalTool, PlanarRotateTool, MIPJumpToClickTool, LabelTool, LengthTool, HeightTool, ProbeTool, RectangleROITool, EllipticalROITool, CircleROITool, ETDRSGridTool, SplineROITool, SplineContourSegmentationTool, BidirectionalTool, PlanarFreehandROITool, PlanarFreehandContourSegmentationTool, LivewireContourTool, LivewireContourSegmentationTool, ArrowAnnotateTool, KeyImageTool, CrosshairsTool, ReferenceLinesTool, RectangleScissorsTool, CircleScissorsTool, SphereScissorsTool, RectangleROIThresholdTool, RectangleROIStartEndThresholdTool, CircleROIStartEndThresholdTool, BrushTool, AngleTool, CobbAngleTool, UltrasoundDirectionalTool, UltrasoundPleuraBLineTool, MagnifyTool, AdvancedMagnifyTool, ReferenceCursors, PaintFillTool, ScaleOverlayTool, OrientationMarkerTool, OverlayGridTool, SegmentationIntersectionTool, EraserTool, SculptorTool, SegmentSelectTool, WindowLevelRegionTool, VolumeRotateTool, RegionSegmentPlusTool, RegionSegmentTool, WholeBodySegmentTool, LabelmapBaseTool, SegmentLabelTool, LabelMapEditWithContourTool, } from './tools';
+import { BaseTool, AnnotationTool, AnnotationDisplayTool, PanTool, TrackballRotateTool, VolumeCroppingTool, VolumeCroppingControlTool, DragProbeTool, WindowLevelTool, ZoomTool, StackScrollTool, SegmentBidirectionalTool, PlanarRotateTool, MIPJumpToClickTool, LabelTool, LengthTool, HeightTool, ProbeTool, RectangleROITool, EllipticalROITool, CircleROITool, ETDRSGridTool, SplineROITool, SplineContourSegmentationTool, BidirectionalTool, PlanarFreehandROITool, PlanarFreehandContourSegmentationTool, LivewireContourTool, LivewireContourSegmentationTool, ArrowAnnotateTool, KeyImageTool, CrosshairsTool, ReferenceLinesTool, RectangleScissorsTool, CircleScissorsTool, SphereScissorsTool, RectangleROIThresholdTool, RectangleROIStartEndThresholdTool, CircleROIStartEndThresholdTool, BrushTool, AngleTool, CobbAngleTool, UltrasoundDirectionalTool, UltrasoundPleuraBLineTool, MagnifyTool, AdvancedMagnifyTool, ReferenceCursors, PaintFillTool, ScaleOverlayTool, OrientationMarkerTool, OrientationControllerTool, OverlayGridTool, SegmentationIntersectionTool, EraserTool, SculptorTool, SegmentSelectTool, WindowLevelRegionTool, VolumeRotateTool, RegionSegmentPlusTool, RegionSegmentTool, WholeBodySegmentTool, LabelmapBaseTool, SegmentLabelTool, LabelMapEditWithContourTool, } from './tools';
 import VideoRedactionTool from './tools/annotation/VideoRedactionTool';
 import * as Enums from './enums';
-export { VideoRedactionTool, init, destroy, addTool, removeTool, cancelActiveManipulations, BaseTool, AnnotationTool, AnnotationDisplayTool, PanTool, SegmentBidirectionalTool, TrackballRotateTool, VolumeCroppingTool, VolumeCroppingControlTool, DragProbeTool, WindowLevelTool, WindowLevelRegionTool, ZoomTool, StackScrollTool, PlanarRotateTool, MIPJumpToClickTool, LabelTool, LengthTool, HeightTool, CrosshairsTool, ReferenceLinesTool, OverlayGridTool, SegmentationIntersectionTool, ProbeTool, RectangleROITool, EllipticalROITool, CircleROITool, ETDRSGridTool, SplineROITool, SplineContourSegmentationTool, BidirectionalTool, PlanarFreehandROITool, PlanarFreehandContourSegmentationTool, LivewireContourTool, LivewireContourSegmentationTool, ArrowAnnotateTool, AngleTool, CobbAngleTool, UltrasoundDirectionalTool, UltrasoundPleuraBLineTool, KeyImageTool, MagnifyTool, AdvancedMagnifyTool, ReferenceCursors, ScaleOverlayTool, SculptorTool, EraserTool, RectangleScissorsTool, CircleScissorsTool, SphereScissorsTool, RectangleROIThresholdTool, RectangleROIStartEndThresholdTool, CircleROIStartEndThresholdTool, BrushTool, OrientationMarkerTool, SegmentSelectTool, SegmentLabelTool, synchronizers, Synchronizer, SynchronizerManager, PaintFillTool, Types, state, ToolGroupManager, store, Enums, CONSTANTS, drawing, annotation, segmentation, utilities, cursors, VolumeRotateTool, RegionSegmentPlusTool, RegionSegmentTool, WholeBodySegmentTool, LabelmapBaseTool, LabelMapEditWithContourTool, splines, version, };
+export { VideoRedactionTool, init, destroy, addTool, removeTool, cancelActiveManipulations, BaseTool, AnnotationTool, AnnotationDisplayTool, PanTool, SegmentBidirectionalTool, TrackballRotateTool, VolumeCroppingTool, VolumeCroppingControlTool, DragProbeTool, WindowLevelTool, WindowLevelRegionTool, ZoomTool, StackScrollTool, PlanarRotateTool, MIPJumpToClickTool, LabelTool, LengthTool, HeightTool, CrosshairsTool, ReferenceLinesTool, OverlayGridTool, SegmentationIntersectionTool, ProbeTool, RectangleROITool, EllipticalROITool, CircleROITool, ETDRSGridTool, SplineROITool, SplineContourSegmentationTool, BidirectionalTool, PlanarFreehandROITool, PlanarFreehandContourSegmentationTool, LivewireContourTool, LivewireContourSegmentationTool, ArrowAnnotateTool, AngleTool, CobbAngleTool, UltrasoundDirectionalTool, UltrasoundPleuraBLineTool, KeyImageTool, MagnifyTool, AdvancedMagnifyTool, ReferenceCursors, ScaleOverlayTool, SculptorTool, EraserTool, RectangleScissorsTool, CircleScissorsTool, SphereScissorsTool, RectangleROIThresholdTool, RectangleROIStartEndThresholdTool, CircleROIStartEndThresholdTool, BrushTool, OrientationMarkerTool, OrientationControllerTool, SegmentSelectTool, SegmentLabelTool, synchronizers, Synchronizer, SynchronizerManager, PaintFillTool, Types, state, ToolGroupManager, store, Enums, CONSTANTS, drawing, annotation, segmentation, utilities, cursors, VolumeRotateTool, RegionSegmentPlusTool, RegionSegmentTool, WholeBodySegmentTool, LabelmapBaseTool, LabelMapEditWithContourTool, splines, version, };

package/dist/esm/init.js

@@ -10,11 +10,13 @@ import { defaultSegmentationStateManager } from './stateManagement/segmentation/
 import segmentationRepresentationModifiedListener from './eventListeners/segmentation/segmentationRepresentationModifiedListener';
 import { setConfig } from './config';
 import segmentationRemovedListener from './eventListeners/segmentation/segmentationRemovedEventListener';
+import { registerBuiltInSegmentationRepresentationDisplays } from './tools/displayTools/registerBuiltInSegmentationRepresentationDisplays';
 let csToolsInitialized = false;
 export function init(defaultConfiguration = {}) {
     if (csToolsInitialized) {
         return;
     }
+    registerBuiltInSegmentationRepresentationDisplays();
     setConfig(defaultConfiguration);
     _addCornerstoneEventListeners();
     _addCornerstoneToolsEventListeners();

package/dist/esm/stateManagement/annotation/FrameOfReferenceSpecificAnnotationManager.js

@@ -1,4 +1,7 @@
 import { Enums, eventTarget, getEnabledElement, utilities, } from '@cornerstonejs/core';
+function isUnsafeKey(key) {
+    return key === '__proto__' || key === 'constructor' || key === 'prototype';
+}
 class FrameOfReferenceSpecificAnnotationManager {
     constructor(uid) {
         this.getGroupKey = (annotationGroupSelector) => {
@@ -77,6 +80,9 @@ class FrameOfReferenceSpecificAnnotationManager {
             const { metadata } = annotation;
             const { FrameOfReferenceUID, toolName } = metadata;
             groupKey = groupKey || FrameOfReferenceUID;
+            if (isUnsafeKey(groupKey) || isUnsafeKey(toolName)) {
+                return;
+            }
             const annotations = this.annotations;
             let frameOfReferenceSpecificAnnotations = annotations[groupKey];
             if (!frameOfReferenceSpecificAnnotations) {
@@ -146,13 +152,13 @@ class FrameOfReferenceSpecificAnnotationManager {
                     return;
                 }
                 const toolSpecificAnnotations = frameOfReferenceSpecificAnnotations[toolName];
-                return structuredClone(toolSpecificAnnotations);
+                return utilities.deepClone(toolSpecificAnnotations);
             }
             else if (groupKey) {
                 const frameOfReferenceSpecificAnnotations = annotations[groupKey];
-                return structuredClone(frameOfReferenceSpecificAnnotations);
+                return utilities.deepClone(frameOfReferenceSpecificAnnotations);
             }
-            return structuredClone(annotations);
+            return utilities.deepClone(annotations);
         };
         this.restoreAnnotations = (state, groupKey, toolName) => {
             const annotations = this.annotations;
@@ -168,7 +174,7 @@ class FrameOfReferenceSpecificAnnotationManager {
                 annotations[groupKey] = state;
             }
             else {
-                this.annotations = structuredClone(state);
+                this.annotations = utilities.deepClone(state);
             }
         };
         this.getAllAnnotations = () => {

package/dist/esm/stateManagement/segmentation/SegmentationRenderingEngine.js

@@ -1,21 +1,13 @@
 import { triggerEvent, eventTarget, Enums, getRenderingEngines, getEnabledElementByViewportId, } from '@cornerstonejs/core';
 import { SegmentationRepresentations, Events as csToolsEvents, } from '../../enums';
-import Representations from '../../enums/SegmentationRepresentations';
 import { getSegmentation } from './getSegmentation';
 import { getSegmentationRepresentations } from './getSegmentationRepresentation';
-import surfaceDisplay from '../../tools/displayTools/Surface/surfaceDisplay';
-import contourDisplay from '../../tools/displayTools/Contour/contourDisplay';
-import labelmapDisplay from '../../tools/displayTools/Labelmap/labelmapDisplay';
 import { addTool } from '../../store/addTool';
 import { state } from '../../store/state';
 import PlanarFreehandContourSegmentationTool from '../../tools/annotation/PlanarFreehandContourSegmentationTool';
 import { getToolGroupForViewport } from '../../store/ToolGroupManager';
 import { addDefaultSegmentationListener } from './segmentationEventManager';
-const renderers = {
-    [Representations.Labelmap]: labelmapDisplay,
-    [Representations.Contour]: contourDisplay,
-    [Representations.Surface]: surfaceDisplay,
-};
+import { getSegmentationRepresentationDisplay } from './SegmentationRepresentationDisplayRegistry';
 const planarContourToolName = PlanarFreehandContourSegmentationTool.toolName;
 class SegmentationRenderingEngine {
     constructor() {
@@ -108,22 +100,33 @@ class SegmentationRenderingEngine {
             if (representation.type === SegmentationRepresentations.Contour) {
                 this._addPlanarFreeHandToolIfAbsent(viewport);
             }
-            const display = renderers[representation.type];
+            const display = getSegmentationRepresentationDisplay(representation.type);
             const segmentation = getSegmentation(representation.segmentationId);
             const existingRepresentation = segmentation.representationData[representation.type] !== undefined;
-            try {
-                display.render(viewport, representation).then(() => {
-                    if (!existingRepresentation) {
-                        addDefaultSegmentationListener(viewport, representation.segmentationId, representation.type);
-                    }
+            if (!display) {
+                console.warn(`No display registered for segmentation representation type ${representation.type}.`);
+                return Promise.resolve({
+                    segmentationId: representation.segmentationId,
+                    type: representation.type,
                 });
             }
-            catch (error) {
+            return display
+                .render(viewport, representation)
+                .then(() => {
+                if (!existingRepresentation) {
+                    addDefaultSegmentationListener(viewport, representation.segmentationId, representation.type);
+                }
+                return {
+                    segmentationId: representation.segmentationId,
+                    type: representation.type,
+                };
+            })
+                .catch((error) => {
                 console.error(error);
-            }
-            return Promise.resolve({
-                segmentationId: representation.segmentationId,
-                type: representation.type,
+                return {
+                    segmentationId: representation.segmentationId,
+                    type: representation.type,
+                };
             });
         });
         Promise.allSettled(segmentationRenderList).then((results) => {

package/dist/esm/stateManagement/segmentation/SegmentationRepresentationDisplayRegistry.d.ts

@@ -0,0 +1,12 @@
+import type { Types } from '@cornerstonejs/core';
+import type SegmentationRepresentations from '../../enums/SegmentationRepresentations';
+import type { SegmentationRepresentation } from '../../types/SegmentationStateTypes';
+type SegmentationRepresentationUpdateFunction = (segmentationId: string) => Promise<void>;
+export type SegmentationRepresentationDisplay = {
+    getUpdateFunction: (viewport: Types.IVolumeViewport | Types.IStackViewport) => SegmentationRepresentationUpdateFunction | void | null;
+    render: (viewport: Types.IVolumeViewport | Types.IStackViewport, representation: SegmentationRepresentation) => Promise<void>;
+    removeRepresentation: (viewportId: string, segmentationId: string, renderImmediate?: boolean) => void;
+};
+export declare function registerSegmentationRepresentationDisplay(representationType: SegmentationRepresentations, display: SegmentationRepresentationDisplay): void;
+export declare function getSegmentationRepresentationDisplay(representationType: SegmentationRepresentations): SegmentationRepresentationDisplay | undefined;
+export {};

package/dist/esm/stateManagement/segmentation/SegmentationRepresentationDisplayRegistry.js

@@ -0,0 +1,7 @@
+const segmentationRepresentationDisplays = new Map();
+export function registerSegmentationRepresentationDisplay(representationType, display) {
+    segmentationRepresentationDisplays.set(representationType, display);
+}
+export function getSegmentationRepresentationDisplay(representationType) {
+    return segmentationRepresentationDisplays.get(representationType);
+}

package/dist/esm/stateManagement/segmentation/SegmentationStateManager.d.ts

@@ -4,8 +4,7 @@ import type { RenderingConfig, RepresentationsData, Segmentation, SegmentationRe
 export default class SegmentationStateManager {
     private state;
     readonly uid: string;
-    private _stackLabelmapImageIdReferenceMap;
-    private _labelmapImageIdReferenceMap;
+    private readonly labelmapImageReferenceResolver;
     constructor(uid?: string);
     getState(): Readonly<SegmentationState>;
     private updateState;
@@ -20,9 +19,7 @@ export default class SegmentationStateManager {
     private addDefaultSegmentationRepresentation;
     addLabelmapRepresentation(state: SegmentationState, viewportId: string, segmentationId: string, renderingConfig?: RenderingConfig): void;
     processLabelmapRepresentationAddition(viewportId: string, segmentationId: string): Promise<void>;
-    _updateLabelmapSegmentationReferences(segmentationId: any, viewport: any, labelmapImageIds: any, updateCallback: any): string | undefined;
     updateLabelmapSegmentationImageReferences(viewportId: any, segmentationId: any): string;
-    _updateAllLabelmapSegmentationImageReferences(viewportId: any, segmentationId: any): void;
     getLabelmapImageIds(representationData: RepresentationsData): any;
     getLabelmapImageIdsForImageId(imageId: string, segmentationId: string): string[];
     getCurrentLabelmapImageIdsForViewport(viewportId: string, segmentationId: string): string[] | undefined;
@@ -43,11 +40,6 @@ export default class SegmentationStateManager {
         segmentationId: string;
         type: SegmentationRepresentations;
     }>;
-    _updateLabelmapImageIdReferenceMap({ segmentationId, referenceImageId, labelmapImageId, }: {
-        segmentationId: any;
-        referenceImageId: any;
-        labelmapImageId: any;
-    }): void;
     _setActiveSegmentation(state: SegmentationState, viewportId: string, segmentationId: string): void;
     setActiveSegmentation(viewportId: string, segmentationId: string): void;
     getActiveSegmentation(viewportId: string): Segmentation | undefined;
@@ -78,7 +70,6 @@ export default class SegmentationStateManager {
         viewportId: string;
         representations: SegmentationRepresentation[];
     }[];
-    private _generateMapKey;
 }
 declare function internalComputeVolumeLabelmapFromStack({ imageIds, options, }: {
     imageIds: string[];