Bläddra i källkod

Correcting calculateInputFunction error

Andrej Studen 5 år sedan
förälder
incheckning
5f8eb418f1
1 ändrade filer med 6 tillägg och 1 borttagningar
  1. 6 1
      cardiacSPECT/cardiacSPECT.py

+ 6 - 1
cardiacSPECT/cardiacSPECT.py

@@ -702,6 +702,10 @@ class cardiacSPECTLogic(ScriptedLoadableModuleLogic):
         node.__class__.__name__=="vtkMRMLGridTransformNode"):
           suffix=".h5"
       fileName=nodeName+suffix
+
+      if not os.path.isdir(localPath):
+         os.mkdir(localPath)
+        
       file=os.path.join(localPath,fileName)
       slicer.util.saveNode(node,file)
       print("Stored to: {}").format(file)
@@ -812,7 +816,8 @@ class cardiacSPECTLogic(ScriptedLoadableModuleLogic):
 
        juse=-1
        for j in range(0,segmentation.GetNumberOfSegments()):
-           segment=segNode.GetSegmentation().GetNthSegmentID(j)
+           segmentID=segNode.GetSegmentation().GetNthSegmentID(j)
+           segment=segNode.GetSegmentation().GetSegment(segmentID)
            if segment.GetName()=='Ventricle':
                juse=j
                break