Andrej Studen 5 роки тому
батько
коміт
4c677260c4
1 змінених файлів з 9 додано та 2 видалено
  1. 9 2
      dataExplorer/CTRegistration.py

+ 9 - 2
dataExplorer/CTRegistration.py

@@ -70,7 +70,9 @@ class CTRegistrationWidget(ScriptedLoadableModuleWidget):
       self.exportButton.clicked.connect(self.onExportButtonClicked)
       datasetFormLayout.addRow("Export:",self.exportButton)
 
-
+      self.debugCheckBox=qt.QCheckBox()
+      self.debugCheckBox.setChecked(True)
+      datasetFormLayout.addRow("Debug:",self.debugCheckBox)
 
   def onLoadDataButtonClicked(self):
       self.logic.loadData(self.onkoNet,self.patientId.text)
@@ -186,7 +188,12 @@ class CTRegistrationLogic(slicer.ScriptedLoadableModule.ScriptedLoadableModuleLo
          if node==None:
             segNode.SetAndObserveTransformNodeID(tNode.GetID())
             binaryRep={'node':segNode,
-                      'mask':segNode.GetBinaryLabelmapRepresentation(segID)}
+                      'mask':segNode.GetBinaryLabelmapRepresentation(segId),
+                      'segId':segId}
+            if binaryRep['mask']==None:
+                segNode.CreateBinaryLabelmapRepresentation()
+                binaryRep['mask']=segNode.GetBinaryLabelmapRepresentation(segId)
+
             self.resampler.rebinSegment(dmrNode,binaryRep)
             node=slicer.util.getFirstNodeByName(nodeName)