def __init__(self, parent=None):
AbstractInteractiveSegmentTubes.__init__(self)
self.widget = None
self.moduleName = 'Interactive Segment Tubes'
self.logic = InteractiveSegmentTubesLogic()
self.interactionNode = None
self.linkIcon = qt.QIcon(os.path.join(ICON_DIR, 'Link.png'))
self.brokenIcon = qt.QIcon(os.path.join(ICON_DIR, 'Broken.png'))
if not parent:
self.parent = slicer.qMRMLWidget()
self.parent.setLayout(qt.QVBoxLayout())
self.parent.setMRMLScene(slicer.mrmlScene)
else:
self.parent = parent
self.layout = self.parent.layout()
if not parent:
self.setup()
self.parent.show()
class InteractiveSegmentTubesWidget(AbstractInteractiveSegmentTubes):
def __init__(self, parent=None):
AbstractInteractiveSegmentTubes.__init__(self)
self.widget = None
self.moduleName = 'Interactive Segment Tubes'
self.logic = InteractiveSegmentTubesLogic()
self.interactionNode = None
self.linkIcon = qt.QIcon(os.path.join(ICON_DIR, 'Link.png'))
self.brokenIcon = qt.QIcon(os.path.join(ICON_DIR, 'Broken.png'))
if not parent:
self.parent = slicer.qMRMLWidget()
self.parent.setLayout(qt.QVBoxLayout())
self.parent.setMRMLScene(slicer.mrmlScene)
else:
self.parent = parent
self.layout = self.parent.layout()
if not parent:
self.setup()
self.parent.show()
def setup(self):
self.loadUi('InteractiveSegmentTubes.ui')
self.widget.setMRMLScene(slicer.mrmlScene)
self.populateParameters()
# Connect widget internal signals here:
self.get('SeedsSizeSliderWidget').connect('valueChanged(double)', self.updateMRMLFromWidget)
self.get('SeedsShowStatusCheckBox').connect('stateChanged(int)', self.updateMRMLFromWidget)
self.get('DropSeedsPushButton').setIcon(qt.QIcon(os.path.join(ICON_DIR, 'AnnotationPointWithArrow.png')))
self.get('DropSeedsPushButton').connect('toggled(bool)', self.updateMRMLFromWidget)
interactionNode = slicer.app.applicationLogic().GetInteractionNode()
self.addObserver(interactionNode, 'ModifiedEvent', self.updateWidgetFromMRML)
comboBox = self.get('LayoutComboBox')
comboBox.setCurrentIndex(comboBox.findData(slicer.vtkMRMLLayoutNode.SlicerLayoutFourUpView))
comboBox.connect('currentIndexChanged(int)', self.updateMRMLFromWidget)
# Connect widget signals to logic here:
self.get('OutputNodeComboBox').connect('nodeAddedByUser(vtkMRMLNode*)', self.logic.addDisplayNodes)
self.logic.setGUICallback(self.onSegmentTubesUpdated)
self.get('InputNodeComboBox').connect('currentNodeChanged(vtkMRMLNode*)', self.setInputNode)
self.get('OutputNodeComboBox').connect('currentNodeChanged(vtkMRMLNode*)', self.setOutputNode)
self.get('SeedScaleSliderWidget').connect('valueChanged(double)', self.logic.setScaleValue)
# Connect signals that needs to be treated by both logic and GUI here:
self.get('ApplyPushButton').connect('toggled(bool)', self.run)
self.get('SeedPointNodeComboBox').connect('currentNodeChanged(vtkMRMLNode*)', self.setSeedNode)
self.get('SizeAndScaleLinkedButton').connect('toggled(bool)', self.linkSizeAndScale)
self.linkSizeAndScale(self.get('SizeAndScaleLinkedButton').isChecked())
self.get('ParameterFileComboBox').connect('currentIndexChanged(int)', self.onParamertClicked)
self.get('ParameterFileComboBox').setCurrentIndex(0)
self.onParamertClicked(0)
def enter(self):
slicer.app.layoutManager().setLayout(slicer.vtkMRMLLayoutNode.SlicerLayoutFourUpView)
self.updateWidgetFromMRML()
self.get('SeedsShowStatusCheckBox').setChecked(False)
interactionNode = slicer.app.applicationLogic().GetInteractionNode()
interactionNode.SetPlaceModePersistence(1)
def exit(self):
pass
def run( self, run ):
# GUI stuff
self.get('DropSeedsPushButton').setChecked(run)
if run:
self.get('ApplyPushButton').setText('Stop')
else:
self.get('ApplyPushButton').setText('Start')
# Logic
self.logic.run(run)
def onSegmentTubesUpdated( self, cliNode, *unused ):
if not cliNode:
return
if not cliNode.IsBusy():
print 'Segment Tubes %s' % cliNode.GetStatusString()
self.get('ApplyPushButton').enabled = not self.logic.isInTransitionState()
self.get('CLIProgressBar').setCommandLineModuleNode(cliNode)
def setInputNode( self, inputNode ):
self.get('ApplyPushButton').setChecked(False)
self.logic.setInputNode(inputNode)
self.createOutputIfNeeded(inputNode, 'vessel', self.get('OutputNodeComboBox'))
self.createOutputIfNeeded(inputNode, 'seed', self.get('SeedPointNodeComboBox'))
def setOutputNode( self, outputNode ):
self.get('ApplyPushButton').setChecked(False)
self.logic.setOutputNode(outputNode)
def setSeedNode( self, seedNode ):
self.get('ApplyPushButton').setChecked(False)
# Observe seed node to update GUI
if not self.hasObserver(seedNode, 'ModifiedEvent', self.updateWidgetFromMRML):
self.removeObservers('ModifiedEvent', 'vtkMRMLMarkupsFiducialNode')
self.addObserver(seedNode, 'ModifiedEvent', self.updateWidgetFromMRML)
seedDisplayNode = seedNode.GetMarkupsDisplayNode() if seedNode else None
# Observe seed display node as well
if not self.hasObserver(seedDisplayNode, 'ModifiedEvent', self.updateWidgetFromMRML):
self.removeObservers('ModifiedEvent', 'vtkMRMLMarkupsDisplayNode')
self.addObserver(seedDisplayNode, 'ModifiedEvent', self.updateWidgetFromMRML)
self.logic.setSeedNode(seedNode)
self.updateWidgetFromMRML()
def updateMRMLFromWidget( self ):
seedNode = self.get('SeedPointNodeComboBox').currentNode()
if seedNode:
seedDisplayNode = seedNode.GetMarkupsDisplayNode() if seedNode else None
if seedDisplayNode:
disabledModify = seedDisplayNode.StartModify()
scale = self.get('SeedsSizeSliderWidget').value
seedDisplayNode.SetGlyphScale(scale)
textScale = scale if self.get('SeedsShowStatusCheckBox').isChecked() else 0.0
seedDisplayNode.SetTextScale(textScale)
seedDisplayNode.EndModify(disabledModify)
interactionNode = slicer.app.applicationLogic().GetInteractionNode()
if self.get('DropSeedsPushButton').isChecked():
interactionNode.SetCurrentInteractionMode(interactionNode.Place)
else:
interactionNode.SetCurrentInteractionMode(interactionNode.ViewTransform)
comboBox = self.get('LayoutComboBox')
layout = comboBox.itemData(comboBox.currentIndex)
slicer.app.layoutManager().setLayout(layout)
def updateWidgetFromMRML( self, *unused ):
seedNode = self.get('SeedPointNodeComboBox').currentNode()
seedDisplayNode = None
seedDisplayNode = seedNode.GetMarkupsDisplayNode() if seedNode else None
if seedDisplayNode:
self.get('SeedsSizeSliderWidget').value = seedDisplayNode.GetGlyphScale()
self.get('SeedsShowStatusCheckBox').setChecked(seedDisplayNode.GetTextScale() > 1e-6)
interactionNode = slicer.app.applicationLogic().GetInteractionNode()
self.get('DropSeedsPushButton').setChecked(interactionNode.GetCurrentInteractionMode() == interactionNode.Place)
comboBox = self.get('LayoutComboBox')
comboBox.setCurrentIndex(comboBox.findData(slicer.app.layoutManager().layout))
def createOutputIfNeeded( self, node, suffix, combobox ):
if node == None:
return
nodeName = '%s-%s' % (node.GetName(), suffix)
node = slicer.util.getFirstNodeByClassByName(nodeName, node.GetClassName())
if node == None:
newNode = combobox.addNode()
newNode.SetName(nodeName)
node = newNode
combobox.setCurrentNode(node)
def linkSizeAndScale( self, link ):
if link:
self.get('SizeAndScaleLinkedButton').setIcon(self.linkIcon)
self.get('SeedsSizeSliderWidget').connect('valueChanged(double)', self.get('SeedScaleSliderWidget').setValue)
self.get('SeedScaleSliderWidget').connect('valueChanged(double)', self.get('SeedsSizeSliderWidget').setValue)
else:
self.get('SizeAndScaleLinkedButton').setIcon(self.brokenIcon)
self.get('SeedsSizeSliderWidget').disconnect('valueChanged(double)', self.get('SeedScaleSliderWidget').setValue)
self.get('SeedScaleSliderWidget').disconnect('valueChanged(double)', self.get('SeedsSizeSliderWidget').setValue)
self.get('SeedScaleSliderWidget').setValue(self.get('SeedsSizeSliderWidget').value)
def populateParameters( self ):
parameterDir = qt.QDir(PARAM_FILES_DIR)
for path in parameterDir.entryList(parameterDir.Files):
fileInfo = qt.QFileInfo(path)
name = fileInfo.baseName().replace("_", " ")
self.get('ParameterFileComboBox').addItem(name, fileInfo.absolutePath() + '/' + path)
# Add load from file
self.get('ParameterFileComboBox').insertSeparator(
self.get('ParameterFileComboBox').count)
self.get('ParameterFileComboBox').addItem('Load from file...', None)
comboBox = self.get('LayoutComboBox')
comboBox.addItem('Quad View', slicer.vtkMRMLLayoutNode.SlicerLayoutFourUpView)
comboBox.addItem('Red Slice', slicer.vtkMRMLLayoutNode.SlicerLayoutOneUpRedSliceView)
comboBox.addItem('Yellow Slice', slicer.vtkMRMLLayoutNode.SlicerLayoutOneUpYellowSliceView)
comboBox.addItem('Green Slice', slicer.vtkMRMLLayoutNode.SlicerLayoutOneUpGreenSliceView)
comboBox.addItem('3D View', slicer.vtkMRMLLayoutNode.SlicerLayoutOneUp3DView)
comboBox.addItem('MIP', slicer.vtkMRMLLayoutNode.SlicerLayoutOneUpRedSliceView)
def onParamertClicked( self, index ):
data = self.get('ParameterFileComboBox').itemData(index)
if data == self.logic.getParameterFile():
return
if not data:
data = qt.QFileDialog.getOpenFileName(
self.widget, 'Load a parameter file', qt.QDir.home().absolutePath(), '*.mtp')
if not data:
self.get('ParameterFileComboBox').setCurrentIndex(0)
return
wasBlocking = self.get('ParameterFileComboBox').blockSignals(True)
self.get('ParameterFileComboBox').insertItem(0, qt.QFileInfo(data).baseName(), data)
self.get('ParameterFileComboBox').blockSignals(wasBlocking)
self.get('ParameterFileComboBox').setCurrentIndex(0)
else:
self.logic.setParameterFile(data)
#-----------------------------------------------------------------------------
# Utilities functions
#-----------------------------------------------------------------------------
def loadUi(self, uiFileName):
loader = qt.QUiLoader()
scriptedModulesPath = eval('slicer.modules.%s.path' % self.moduleName.replace(" ", "").lower())
scriptedModulesPath = os.path.dirname(scriptedModulesPath)
path = os.path.join(scriptedModulesPath, 'Resources', 'UI', uiFileName)
qfile = qt.QFile(path)
qfile.open(qt.QFile.ReadOnly)
uiWidget = loader.load(qfile)
uiWidget.setAutoFillBackground(False)
uiWidget.setPalette(slicer.app.palette())
self.layout.setContentsMargins(0,0,0,0)
self.layout.addWidget(uiWidget)
self.widget = uiWidget
def findWidget(self, widget, objectName):
if widget.objectName == objectName:
return widget
else:
children = []
for w in widget.children():
resulting_widget = self.findWidget(w, objectName)
if resulting_widget:
return resulting_widget
return None
def get(self, objectName):
return self.findWidget(self.widget, objectName)
