feat(segmentation): add undo command grouping

Author: Jo-Byr

trame_slicer/segmentation/segment_modifier.py

@@ -3,6 +3,7 @@
 import enum
 import logging
 import math
+from collections.abc import Generator
 from enum import auto
 
 from numpy.typing import NDArray
@@ -380,3 +381,6 @@ def is_source_volume_intensity_mask_enabled(self) -> bool:
         if not self.segment_editor_node:
             return False
         return self.segment_editor_node.GetSourceVolumeIntensityMask()
+
+    def group_undo_commands(self, text: str = "") -> Generator:
+        return self.segmentation.group_undo_commands(text)

trame_slicer/segmentation/segmentation.py

@@ -1,5 +1,7 @@
 from __future__ import annotations
 
+from collections.abc import Generator
+
 from numpy.typing import NDArray
 from slicer import (
     vtkMRMLSegmentationNode,
@@ -272,3 +274,6 @@ def set_segment_labelmap(self, segment_id, label_map: vtkImageData | NDArray):
 
     def get_display(self) -> SegmentationDisplay | None:
         return SegmentationDisplay(self._segmentation_node.GetDisplayNode()) if self._segmentation_node else None
+
+    def group_undo_commands(self, text: str = "") -> Generator:
+        return self.undo_stack.group_undo_commands(text)