Merge pull request #196 from che85/last_minute_changes

BUG: slicer.mrmlScene.GetNodesByClass(..) causes leaks when quitting …
QIICR · Sep 18, 2017 · 0530006 · 0530006
2 parents 45dd506 + aec1e97
commit 0530006
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/DICOMPlugins/DICOMTID1500Plugin.py b/DICOMPlugins/DICOMTID1500Plugin.py
@@ -182,8 +182,8 @@ def load(self, loadable):
 
     table = self.metadata2vtkTableNode(outputFile)
     if table:
-      segmentationNodes = slicer.mrmlScene.GetNodesByClass("vtkMRMLSegmentationNode")
-      segmentationNodeID = segmentationNodes.GetItemAsObject(segmentationNodes.GetNumberOfItems()-1).GetID()
+      segmentationNodes = slicer.util.getNodesByClass('vtkMRMLSegmentationNode')
+      segmentationNodeID = segmentationNodes[-1].GetID()
       table.SetAttribute("ReferencedSegmentationNodeID", segmentationNodeID)
     self.cleanup()
     return table is not None