feat(segmentation): Add draw segmentation effect

examples/viewer_lib/ui/segmentation/segment_editor_ui.py

@@ -17,6 +17,7 @@
 
 from trame_slicer.segmentation import (
     SegmentationEffect,
+    SegmentationEffectDraw,
     SegmentationEffectErase,
     SegmentationEffectIslands,
     SegmentationEffectNoTool,
@@ -144,6 +145,12 @@ def build_effect_buttons(self, all: bool = True, **kwargs):
             SegmentationEffectScissors,
             **kwargs,
         )
+        self._create_effect_button(
+            "Draw",
+            "mdi-draw",
+            SegmentationEffectDraw,
+            **kwargs,
+        )
         if all:
             self._create_effect_button(
                 "Threshold",

trame_slicer/core/segmentation_editor.py

@@ -31,6 +31,7 @@
     Segmentation,
     SegmentationDisplay,
     SegmentationEffect,
+    SegmentationEffectDraw,
     SegmentationEffectErase,
     SegmentationEffectIslands,
     SegmentationEffectNoTool,
@@ -54,6 +55,7 @@ class SegmentationEditor(SignalContainer):
     """
 
     builtin_effects: ClassVar[list[type[SegmentationEffect]]] = [
+        SegmentationEffectDraw,
         SegmentationEffectErase,
         SegmentationEffectIslands,
         SegmentationEffectNoTool,

trame_slicer/segmentation/__init__.py

@@ -11,6 +11,7 @@
 from .segmentation import Segmentation
 from .segmentation_display import SegmentationDisplay, SegmentationOpacityEnum
 from .segmentation_effect import SegmentationEffect
+from .segmentation_effect_draw import SegmentationEffectDraw
 from .segmentation_effect_islands import (
     SegmentationEffectIslands,
     SegmentationIslandsMode,
@@ -61,6 +62,7 @@
     "Segmentation",
     "SegmentationDisplay",
     "SegmentationEffect",
+    "SegmentationEffectDraw",
     "SegmentationEffectErase",
     "SegmentationEffectIslands",
     "SegmentationEffectNoTool",

trame_slicer/segmentation/segmentation_effect_draw.py

@@ -0,0 +1,24 @@
+from slicer import vtkMRMLAbstractViewNode, vtkMRMLNode, vtkMRMLSliceNode
+
+from .scissors_effect_parameters import ScissorsEffectFillMode, ScissorsEffectRangeMode
+from .segmentation_effect_pipeline import SegmentationEffectPipeline
+from .segmentation_effect_scissors import SegmentationEffectScissors
+
+
+class SegmentationEffectDraw(SegmentationEffectScissors):
+    def __init__(self):
+        super().__init__()
+        self.set_symmetric_distance(0.0)
+        self.set_fill_mode(ScissorsEffectFillMode.FILL_INSIDE)
+        self.set_range_mode(ScissorsEffectRangeMode.SYMMETRIC)
+
+    def _create_pipeline(
+        self, _view_node: vtkMRMLAbstractViewNode, _parameter: vtkMRMLNode
+    ) -> SegmentationEffectPipeline | None:
+        if isinstance(_view_node, vtkMRMLSliceNode):
+            from .segmentation_effect_scissors_widget import (
+                SegmentationScissorsPipeline,
+            )
+
+            return SegmentationScissorsPipeline()
+        return None  # Interaction in 3D not supported