|
@@ -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)
|
|
|
|
|
|
|