def __init__(self):
'''
Constructor
'''
"""
path=slicer.modules.usguidedprocedure.path
modulePath=os.path.dirname(path)
loadedDataGUIfile=os.path.join(modulePath,"USGuidedWizard/loadedData.ui")
f = qt.QFile(loadedDataGUIfile)
#f = qt.QFile('C:/Users/Usuario/devel/slicelets/USGuidedProcedure/USGuidedWizard/fiducials.ui')
f.open(qt.QFile.ReadOnly)
loader = qt.QUiLoader()
self.loadedDataWidget = loader.load(f)
f.close()
"""
self.listWidget = qt.QListWidget()
print("Constructor of VolumeRenderingViewer executed")
self.listWidget.show()
self.currentItem=None
self.listWidget.itemDoubleClicked.connect(self.onItemDoubleClicked)
self.propertiesMenu = PropertiesMenu()
#qt.QObject.connect(self.propertiesMenu,qt.SIGNAL("colorButtonClicked"),self.onColorButtonClicked)
#TODO: should be a way to emit a signal from the PropertiesMenu class but I do not how
self.propertiesMenu.colourButton.connect("clicked()",self.onColorButtonClicked)
self.propertiesMenu.meshOpacitySlider.connect('valueChanged(int)', self.onSliderMoved)
self.propertiesMenu.checkBoxVisible3D.connect("stateChanged(int)",self.onVisible3DChanged)
def __init__(self):
'''
Constructor
'''
"""
path=slicer.modules.usguidedprocedure.path
modulePath=os.path.dirname(path)
loadedDataGUIfile=os.path.join(modulePath,"USGuidedWizard/loadedData.ui")
f = qt.QFile(loadedDataGUIfile)
#f = qt.QFile('C:/Users/Usuario/devel/slicelets/USGuidedProcedure/USGuidedWizard/fiducials.ui')
f.open(qt.QFile.ReadOnly)
loader = qt.QUiLoader()
self.loadedDataWidget = loader.load(f)
f.close()
"""
self.modelsFrame= qt.QFrame()
self.modelsFrame.setLayout(qt.QHBoxLayout())
self.modelsFrameLayout=self.modelsFrame.layout()
self.listWidget = qt.QListWidget()
self.modelsLabel=qt.QLabel("Models:")
print("Constructor of ModelsViewer executed")
#self.listWidget.show()
self.currentItem=None
self.modelsButtonsFrame=qt.QFrame()
self.modelsButtonsFrame.setLayout(qt.QVBoxLayout())
self.modelsButtonsLayout=self.modelsButtonsFrame.layout()
self.showPropertiesButton = qt.QPushButton("Show properties")
self.showPropertiesButton.toolTip = "Show the model properties"
self.showPropertiesButton.setEnabled(False)
self.showPropertiesButton.connect('clicked(bool)', self.onPropertiesClicked)
self.removeModelButton = qt.QPushButton("Remove model")
self.removeModelButton.toolTip = "Remove the model"
self.removeModelButton.setEnabled(False)
self.removeModelButton.connect('clicked(bool)', self.onRemoveActionTriggered)
self.modelsButtonsLayout.addWidget(self.modelsLabel)
self.modelsButtonsLayout.addWidget(self.showPropertiesButton)
self.modelsButtonsLayout.addWidget(self.removeModelButton)
self.modelsFrameLayout.addWidget(self.listWidget)
self.modelsFrameLayout.addWidget(self.modelsButtonsFrame)
#self.listWidget.itemDoubleClicked.connect(self.onItemDoubleClicked)
self.propertiesMenu = PropertiesMenu()
#qt.QObject.connect(self.propertiesMenu,qt.SIGNAL("colorButtonClicked"),self.onColorButtonClicked)
#TODO: should be a way to emit a signal from the PropertiesMenu class but I do not how
self.propertiesMenu.colourButton.connect("clicked()",self.onColorButtonClicked)
self.propertiesMenu.meshOpacitySlider.connect('valueChanged(int)', self.onSliderMoved)
self.propertiesMenu.checkBoxVisible3D.connect("stateChanged(int)",self.onVisible3DChanged)
self.propertiesMenu.checkBoxIntersectionWithUSImage.connect("stateChanged(int)",self.onIntersectionWithUSImageChanged)
class VolumeRenderingViewer():
'''
classdocs
'''
def __init__(self):
'''
Constructor
'''
"""
path=slicer.modules.usguidedprocedure.path
modulePath=os.path.dirname(path)
loadedDataGUIfile=os.path.join(modulePath,"USGuidedWizard/loadedData.ui")
f = qt.QFile(loadedDataGUIfile)
#f = qt.QFile('C:/Users/Usuario/devel/slicelets/USGuidedProcedure/USGuidedWizard/fiducials.ui')
f.open(qt.QFile.ReadOnly)
loader = qt.QUiLoader()
self.loadedDataWidget = loader.load(f)
f.close()
"""
self.listWidget = qt.QListWidget()
print("Constructor of VolumeRenderingViewer executed")
self.listWidget.show()
self.currentItem=None
self.listWidget.itemDoubleClicked.connect(self.onItemDoubleClicked)
self.propertiesMenu = PropertiesMenu()
#qt.QObject.connect(self.propertiesMenu,qt.SIGNAL("colorButtonClicked"),self.onColorButtonClicked)
#TODO: should be a way to emit a signal from the PropertiesMenu class but I do not how
self.propertiesMenu.colourButton.connect("clicked()",self.onColorButtonClicked)
self.propertiesMenu.meshOpacitySlider.connect('valueChanged(int)', self.onSliderMoved)
self.propertiesMenu.checkBoxVisible3D.connect("stateChanged(int)",self.onVisible3DChanged)
def getListWidget(self):
return self.listWidget
def setModuleLogic(self,logic):
self.logic=logic
def listenToScene(self):
vl=slicer.modules.volumes
vl=vl.logic()
vl.SetMRMLScene(slicer.mrmlScene)
self.sceneObserver = vl.AddObserver('ModifiedEvent', self.onVolumeAdded)
#self.sceneObserver = slicer.mrmlScene.AddObserver('ModifiedEvent', self.onVolumeAdded)
def onVolumeAdded(self, caller, event):
print("Volumes module was modified")
numVolumes=slicer.mrmlScene.GetNumberOfNodesByClass("vtkMRMLScalarVolumeNode")
print "Number of volumes: " + str(numVolumes)
#print('A model was added to the scene !')
if numVolumes>0:
for i in xrange(0,numVolumes):
node = slicer.mrmlScene.GetNthNodeByClass(i, "vtkMRMLScalarVolumeNode")
if not (node==None):
if (not (node.GetName() == "Image_Reference")):
print(node.GetName())
isAlreadyInList=False
j=0
while ((j < self.listWidget.count) and (not isAlreadyInList)):
isAlreadyInList=node.GetName()==self.listWidget.item(j).text()
j=j+1
if not isAlreadyInList:
print('A volume was added to the scene !')
self.listWidget.addItem(node.GetName())
node=slicer.util.getNode(node.GetName())
node=slicer.vtkMRMLScalarVolumeNode.SafeDownCast(node)
self.logic.onVolumeAdded(node)
#print("Info of added node:")
def addItem(self,item):
self.listWidget.addItem(item)
def onItemDoubleClicked(self,item):
vrNode=slicer.util.getNode("CPURayCastVolumeRendering")
slicer.mrmlScene.AddNode(vrNode)
self.currentItem=item;
# Get the current display node
node=slicer.util.getNode(self.currentItem.text())
node=slicer.vtkMRMLScalarVolumeNode.SafeDownCast(node)
self.currentDisplayNode=node.GetDisplayNode()
currentOpacity= self.currentDisplayNode.GetOpacity()
self.propertiesMenu.meshOpacitySlider.setValue(currentOpacity*100)
isVisible=self.currentDisplayNode.GetVisibility()
self.propertiesMenu.checkBoxVisible3D.setCheckState(isVisible)
#print item.text()
self.propertiesMenu.show()
def onColorButtonClicked(self):
print("Color button clicked SIGNAL captured!!")
# Set the current color to the dialog
colorDialog=qt.QColorDialog()
currentColor=self.currentDisplayNode.GetColor()
qtCurrentColor=qt.QColor(currentColor[0]*255,currentColor[1]*255,currentColor[2]*255)
color=colorDialog.getColor(qtCurrentColor)
if color.isValid():
b=float(color.blue())/255
r=float(color.red())/255
g=float(color.green())/255
self.currentDisplayNode.SetColor(r,g,b)
def onSliderMoved(self,opacity):
# Get the current display node
print("Slider movement captured!")
#print opacity
self.currentDisplayNode.SetOpacity(float(opacity)/100)
def onVisible3DChanged(self,isVisible):
#print isVisible
self.currentDisplayNode.SetVisibility(isVisible)
class ModelsViewer():
'''
classdocs
'''
def __init__(self):
'''
Constructor
'''
"""
path=slicer.modules.usguidedprocedure.path
modulePath=os.path.dirname(path)
loadedDataGUIfile=os.path.join(modulePath,"USGuidedWizard/loadedData.ui")
f = qt.QFile(loadedDataGUIfile)
#f = qt.QFile('C:/Users/Usuario/devel/slicelets/USGuidedProcedure/USGuidedWizard/fiducials.ui')
f.open(qt.QFile.ReadOnly)
loader = qt.QUiLoader()
self.loadedDataWidget = loader.load(f)
f.close()
"""
self.modelsFrame= qt.QFrame()
self.modelsFrame.setLayout(qt.QHBoxLayout())
self.modelsFrameLayout=self.modelsFrame.layout()
self.listWidget = qt.QListWidget()
self.modelsLabel=qt.QLabel("Models:")
print("Constructor of ModelsViewer executed")
#self.listWidget.show()
self.currentItem=None
self.modelsButtonsFrame=qt.QFrame()
self.modelsButtonsFrame.setLayout(qt.QVBoxLayout())
self.modelsButtonsLayout=self.modelsButtonsFrame.layout()
self.showPropertiesButton = qt.QPushButton("Show properties")
self.showPropertiesButton.toolTip = "Show the model properties"
self.showPropertiesButton.setEnabled(False)
self.showPropertiesButton.connect('clicked(bool)', self.onPropertiesClicked)
self.removeModelButton = qt.QPushButton("Remove model")
self.removeModelButton.toolTip = "Remove the model"
self.removeModelButton.setEnabled(False)
self.removeModelButton.connect('clicked(bool)', self.onRemoveActionTriggered)
self.modelsButtonsLayout.addWidget(self.modelsLabel)
self.modelsButtonsLayout.addWidget(self.showPropertiesButton)
self.modelsButtonsLayout.addWidget(self.removeModelButton)
self.modelsFrameLayout.addWidget(self.listWidget)
self.modelsFrameLayout.addWidget(self.modelsButtonsFrame)
#self.listWidget.itemDoubleClicked.connect(self.onItemDoubleClicked)
self.propertiesMenu = PropertiesMenu()
#qt.QObject.connect(self.propertiesMenu,qt.SIGNAL("colorButtonClicked"),self.onColorButtonClicked)
#TODO: should be a way to emit a signal from the PropertiesMenu class but I do not how
self.propertiesMenu.colourButton.connect("clicked()",self.onColorButtonClicked)
self.propertiesMenu.meshOpacitySlider.connect('valueChanged(int)', self.onSliderMoved)
self.propertiesMenu.checkBoxVisible3D.connect("stateChanged(int)",self.onVisible3DChanged)
self.propertiesMenu.checkBoxIntersectionWithUSImage.connect("stateChanged(int)",self.onIntersectionWithUSImageChanged)
# #Actions
# self.listWidget.setContextMenuPolicy(qt.QActionEvent.ContextMenu)
# removeAction = qt.QAction("Remove", self.listWidget)
# removeAction.triggered.connect(self.onRemoveActionTriggered)
# viewPropertiesAction = qt.QAction("Properties", self.listWidget)
# viewPropertiesAction.triggered.connect(self.onPropertiesClicked)
# self.listWidget.addAction(viewPropertiesAction)
# self.listWidget.addAction(removeAction)
def getModelsViewerWidget(self):
return self.modelsFrame
def setModuleLogic(self,logic):
self.logic=logic
def listenToScene(self):
self.sceneObserver = slicer.mrmlScene.AddObserver('ModifiedEvent', self.onModelAdded)
def onModelAdded(self, caller, event):
numModels=slicer.mrmlScene.GetNumberOfNodesByClass("vtkMRMLModelNode")
#print('A model was added to the scene !')
if numModels>0:
for i in xrange(1,numModels):
node = slicer.mrmlScene.GetNthNodeByClass(i, "vtkMRMLModelNode")
if ((node is not None) and (not slicer.vtkMRMLSliceLogic.IsSliceModelNode(node)) and node.GetClassName()=="vtkMRMLModelNode" ):
#print(node.GetName())
self.showPropertiesButton.setEnabled(True)
self.removeModelButton.setEnabled(True)
isAlreadyInList=False
j=0
while ((j < self.listWidget.count) and (not isAlreadyInList)):
isAlreadyInList=node.GetName()==self.listWidget.item(j).text()
j=j+1
if not isAlreadyInList:
print('A model was added to the scene !')
self.listWidget.addItem(node.GetName())
node=slicer.util.getNode(node.GetName())
node=slicer.vtkMRMLModelNode.SafeDownCast(node)
displayNode=node.GetDisplayNode()
#print("Info of added node:")
#print displayNode.GetName()
#print displayNode.GetColor()
#print displayNode.GetOpacity()
def addItem(self,item):
self.listWidget.addItem(item)
def onPropertiesClicked(self):
self.currentItem = self.listWidget.currentItem()
if self.currentItem==None:
ret=qt.QMessageBox.warning(self.listWidget, 'Models List', 'You must select a model in the list to see the properties.', qt.QMessageBox.Ok , qt.QMessageBox.Ok )
return
# Get the current display node
node=slicer.util.getNode(self.currentItem.text())
node=slicer.vtkMRMLModelNode.SafeDownCast(node)
self.currentDisplayNode=node.GetDisplayNode()
currentOpacity= self.currentDisplayNode.GetOpacity()
self.propertiesMenu.meshOpacitySlider.setValue(currentOpacity*100)
isVisible=self.currentDisplayNode.GetVisibility()
isIntersectionShown=self.currentDisplayNode.GetSliceIntersectionVisibility()
self.propertiesMenu.checkBoxIntersectionWithUSImage.setCheckState(isIntersectionShown)
self.propertiesMenu.checkBoxVisible3D.setCheckState(isVisible)
#print item.text()
self.propertiesMenu.show()
def onColorButtonClicked(self):
print("Color button clicked SIGNAL captured!!")
# Set the current color to the dialog
colorDialog=qt.QColorDialog()
currentColor=self.currentDisplayNode.GetColor()
qtCurrentColor=qt.QColor(currentColor[0]*255,currentColor[1]*255,currentColor[2]*255)
color=colorDialog.getColor(qtCurrentColor)
if color.isValid():
b=float(color.blue())/255
r=float(color.red())/255
g=float(color.green())/255
self.currentDisplayNode.SetColor(r,g,b)
def onSliderMoved(self,opacity):
# Get the current display node
print("Slider movement captured!")
#print opacity
self.currentDisplayNode.SetOpacity(float(opacity)/100)
def onVisible3DChanged(self,isVisible):
#print isVisible
self.currentDisplayNode.SetVisibility(isVisible)
def onIntersectionWithUSImageChanged(self, isIntersectionShown):
self.currentDisplayNode.SetSliceIntersectionVisibility(isIntersectionShown)
def onRemoveActionTriggered(self):
item = self.listWidget.currentItem()
if self.currentItem==None:
ret=qt.QMessageBox.warning(self.listWidget, 'Models List', 'You must select a model to remove.', qt.QMessageBox.Ok , qt.QMessageBox.Ok )
return
node=slicer.util.getNode(item.text())
node=slicer.vtkMRMLModelNode.SafeDownCast(node)
currentDisplayNode=node.GetDisplayNode()
slicer.mrmlScene.RemoveNode(currentDisplayNode)
slicer.mrmlScene.RemoveNode(node)
# Delete the item
self.listWidget.takeItem(self.listWidget.row(item))
