ENH: Add labels for export segments based on ROI points' labels

RuoyanMeng · Jan 21, 2025 · 745adc1 · 745adc1
1 parent e7a236e
commit 745adc1
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 1 deletion.
diff --git a/LiverVolumetry/LiverVolumetry.py b/LiverVolumetry/LiverVolumetry.py
@@ -452,6 +452,12 @@ def generateSegments(self, resectionNodes, ROIMarkersList, segmentsVolumeNode):
 
     seg = slicer.mrmlScene.AddNewNodeByClass("vtkMRMLSegmentationNode")
     slicer.modules.segmentations.logic().ImportLabelmapToSegmentationNode(generatedSegmentsNode, seg)
+
+    ##set segments label
+    seg.GetSegmentation().GetNthSegment(0).SetName("Remnant")
+    for i in range(ROIMarkersList.GetNumberOfControlPoints()):
+      seg.GetSegmentation().GetNthSegment(i+1).SetName(ROIMarkersList.GetNthFiducialLabel(i))
+
     slicer.mrmlScene.RemoveNode(generatedSegmentsNode)
 
 

diff --git a/LiverVolumetry/Logic/vtkLiverVolumetryLogic.cxx b/LiverVolumetry/Logic/vtkLiverVolumetryLogic.cxx
@@ -351,7 +351,7 @@ void vtkLiverVolumetryLogic::GetResectionsProjectionITKImage(vtkMRMLLabelMapVolu
       if(Res < 500){
         Res  =  500;
         }
-      BezierHR = GenerateBezierSurface(700, bezierSurfaceNode);
+      BezierHR = GenerateBezierSurface(Res, bezierSurfaceNode);
       if(i == 0){
         this->ProjectedTargetSegmentImage = vtkLabelMapHelper::VolumeNodeToItkImage(TargetSegmentLabelMapCopy, true, false);
         }