Browse Source

Adding segmentList to configuration file

Andrej Studen 2 years ago
parent
commit
f493aec523
1 changed files with 10 additions and 2 deletions
  1. 10 2
      slicerModules/imageBrowser.py

+ 10 - 2
slicerModules/imageBrowser.py

@@ -678,7 +678,11 @@ class imageBrowserLogic(ScriptedLoadableModuleLogic):
         seg=segNode.GetSegmentation()
         segNames=[seg.GetNthSegmentID(i) for i in range(seg.GetNumberOfSegments())]
         print('Segments')
-        for s in self.segmentList:
+        try:
+            segmentList=self.isetup['segmentList']
+        except KeyError:
+            segmentList=self.segmentList
+        for s in segmentList:
             if s not in segNames:
                 seg.AddEmptySegment(s,s)
                 print(s)
@@ -761,7 +765,11 @@ class imageBrowserLogic(ScriptedLoadableModuleLogic):
         slicer.mrmlScene.AddNode(segNode)
         segNode.CreateDefaultDisplayNodes()
         segNode.SetReferenceImageGeometryParameterFromVolumeNode(self.volumeNode['PET'])
-        for s in self.segmentList:
+        try:
+            segmentList=self.isetup['segmentList']
+        except KeyError:
+            segmentList=self.segmentList
+        for s in segmentList:
             segNode.GetSegmentation().AddEmptySegment(s,s)