class MeshGeneratorWidget(QtGui.QWidget):
def __init__(self, model, parent=None):
super(MeshGeneratorWidget, self).__init__(parent)
self._ui = Ui_MeshGeneratorWidget()
self._ui.setupUi(self)
self._model = model
self._generator_model = model.getGeneratorModel()
self._annotation_model = model.getMeshAnnotationModel()
self._segmentation_data_model = model.getSegmentationDataModel()
self._ui.sceneviewer_widget.setContext(model.getContext())
self._ui.sceneviewer_widget.setGeneratorModel(
model.getGeneratorModel())
self._model.getGeneratorModel().registerCustomParametersCallback(
self._customParametersChange)
self._model.registerSceneChangeCallback(self._sceneChanged)
self._generator_model.registerTransformationChangeCallback(
self._transformationChanged)
self._doneCallback = None
# self._populateAnnotationTree()
self._refreshScaffoldTypeNames()
self._refreshParameterSetNames()
self._makeConnections()
def _graphicsInitialized(self):
"""
Callback for when SceneviewerWidget is initialised
Set custom scene from model
"""
sceneviewer = self._ui.sceneviewer_widget.getSceneviewer()
if sceneviewer is not None:
self._model.loadSettings()
self._refreshOptions()
scene = self._model.getScene()
self._ui.sceneviewer_widget.setScene(scene)
# self._ui.sceneviewer_widget.setSelectModeAll()
sceneviewer.setLookatParametersNonSkew([2.0, -2.0, 1.0],
[0.0, 0.0, 0.0],
[0.0, 0.0, 1.0])
sceneviewer.setTransparencyMode(sceneviewer.TRANSPARENCY_MODE_SLOW)
self._autoPerturbLines()
self._viewAll()
def _customParametersChange(self):
"""
Callback when scaffold options or mesh edits are made, so custom parameter set now in use.
"""
self._refreshParameterSetNames()
def _sceneChanged(self):
sceneviewer = self._ui.sceneviewer_widget.getSceneviewer()
if sceneviewer is not None:
scene = self._model.getScene()
self._ui.sceneviewer_widget.setScene(scene)
self._autoPerturbLines()
def _transformationChanged(self):
self._ui.rotation_lineEdit.setText(
self._generator_model.getRotationText())
self._ui.scale_lineEdit.setText(self._generator_model.getScaleText())
self._ui.translation_lineEdit.setText(
self._generator_model.getTranslationText())
def _autoPerturbLines(self):
"""
Enable scene viewer perturb lines iff solid surfaces are drawn with lines.
Call whenever lines, surfaces or translucency changes
"""
sceneviewer = self._ui.sceneviewer_widget.getSceneviewer()
if sceneviewer is not None:
sceneviewer.setPerturbLinesFlag(
self._generator_model.needPerturbLines())
def _makeConnections(self):
self._ui.sceneviewer_widget.graphicsInitialized.connect(
self._graphicsInitialized)
self._ui.done_button.clicked.connect(self._doneButtonClicked)
self._ui.viewAll_button.clicked.connect(self._viewAll)
self._ui.subscaffoldBack_pushButton.clicked.connect(
self._subscaffoldBackButtonPressed)
self._ui.meshType_comboBox.currentIndexChanged.connect(
self._scaffoldTypeChanged)
self._ui.parameterSet_comboBox.currentIndexChanged.connect(
self._parameterSetChanged)
self._ui.deleteElementsRanges_lineEdit.returnPressed.connect(
self._deleteElementRangesLineEditChanged)
self._ui.deleteElementsRanges_lineEdit.editingFinished.connect(
self._deleteElementRangesLineEditChanged)
self._ui.deleteElementsSelection_pushButton.clicked.connect(
self._deleteElementsSelectionButtonPressed)
self._ui.rotation_lineEdit.returnPressed.connect(
self._rotationLineEditChanged)
self._ui.rotation_lineEdit.editingFinished.connect(
self._rotationLineEditChanged)
self._ui.scale_lineEdit.returnPressed.connect(
self._scaleLineEditChanged)
self._ui.scale_lineEdit.editingFinished.connect(
self._scaleLineEditChanged)
self._ui.translation_lineEdit.returnPressed.connect(
self._translationLineEditChanged)
self._ui.translation_lineEdit.editingFinished.connect(
self._translationLineEditChanged)
self._ui.displayDataPoints_checkBox.clicked.connect(
self._displayDataPointsClicked)
self._ui.displayDataContours_checkBox.clicked.connect(
self._displayDataContoursClicked)
self._ui.displayDataRadius_checkBox.clicked.connect(
self._displayDataRadiusClicked)
self._ui.displayDataMarkerPoints_checkBox.clicked.connect(
self._displayDataMarkerPointsClicked)
self._ui.displayDataMarkerNames_checkBox.clicked.connect(
self._displayDataMarkerNamesClicked)
self._ui.displayMarkerPoints_checkBox.clicked.connect(
self._displayMarkerPointsClicked)
self._ui.displayAxes_checkBox.clicked.connect(self._displayAxesClicked)
self._ui.displayElementAxes_checkBox.clicked.connect(
self._displayElementAxesClicked)
self._ui.displayElementNumbers_checkBox.clicked.connect(
self._displayElementNumbersClicked)
self._ui.displayLines_checkBox.clicked.connect(
self._displayLinesClicked)
self._ui.displayLinesExterior_checkBox.clicked.connect(
self._displayLinesExteriorClicked)
self._ui.displayModelRadius_checkBox.clicked.connect(
self._displayModelRadiusClicked)
self._ui.displayNodeDerivativeLabelsD1_checkBox.clicked.connect(
self._displayNodeDerivativeLabelsD1Clicked)
self._ui.displayNodeDerivativeLabelsD2_checkBox.clicked.connect(
self._displayNodeDerivativeLabelsD2Clicked)
self._ui.displayNodeDerivativeLabelsD3_checkBox.clicked.connect(
self._displayNodeDerivativeLabelsD3Clicked)
self._ui.displayNodeDerivativeLabelsD12_checkBox.clicked.connect(
self._displayNodeDerivativeLabelsD12Clicked)
self._ui.displayNodeDerivativeLabelsD13_checkBox.clicked.connect(
self._displayNodeDerivativeLabelsD13Clicked)
self._ui.displayNodeDerivativeLabelsD23_checkBox.clicked.connect(
self._displayNodeDerivativeLabelsD23Clicked)
self._ui.displayNodeDerivativeLabelsD123_checkBox.clicked.connect(
self._displayNodeDerivativeLabelsD123Clicked)
self._ui.displayNodeDerivatives_checkBox.clicked.connect(
self._displayNodeDerivativesClicked)
self._ui.displayNodeNumbers_checkBox.clicked.connect(
self._displayNodeNumbersClicked)
self._ui.displayNodePoints_checkBox.clicked.connect(
self._displayNodePointsClicked)
self._ui.displaySurfaces_checkBox.clicked.connect(
self._displaySurfacesClicked)
self._ui.displaySurfacesExterior_checkBox.clicked.connect(
self._displaySurfacesExteriorClicked)
self._ui.displaySurfacesTranslucent_checkBox.clicked.connect(
self._displaySurfacesTranslucentClicked)
self._ui.displaySurfacesWireframe_checkBox.clicked.connect(
self._displaySurfacesWireframeClicked)
def keyPressEvent(self, event):
if (event.key() == QtCore.Qt.Key_S) and (event.isAutoRepeat()
== False):
self._ui.sceneviewer_widget._selectionKeyPressed = True
event.setAccepted(True)
else:
event.ignore()
def keyReleaseEvent(self, event):
if (event.key() == QtCore.Qt.Key_S) and (event.isAutoRepeat()
== False):
self._ui.sceneviewer_widget._selectionKeyPressed = False
event.setAccepted(True)
else:
event.ignore()
def _refreshComboBoxNames(self, comboBox, names, currentName):
comboBox.blockSignals(True)
comboBox.clear()
currentIndex = 0
index = 0
for name in names:
comboBox.addItem(name)
if name == currentName:
currentIndex = index
index += 1
comboBox.setCurrentIndex(currentIndex)
comboBox.blockSignals(False)
def _refreshScaffoldTypeNames(self):
self._refreshComboBoxNames(
self._ui.meshType_comboBox,
self._generator_model.getAvailableScaffoldTypeNames(),
self._generator_model.getEditScaffoldTypeName())
def _refreshParameterSetNames(self):
self._refreshComboBoxNames(
self._ui.parameterSet_comboBox,
self._generator_model.getAvailableParameterSetNames(),
self._generator_model.getParameterSetName())
def _createFMAItem(self, parent, text, fma_id):
item = QtGui.QTreeWidgetItem(parent)
item.setText(0, text)
item.setData(0, QtCore.Qt.UserRole + 1, fma_id)
item.setCheckState(0, QtCore.Qt.Unchecked)
item.setFlags(QtCore.Qt.ItemIsUserCheckable | QtCore.Qt.ItemIsEnabled
| QtCore.Qt.ItemIsTristate)
return item
def _populateAnnotationTree(self):
tree = self._ui.treeWidgetAnnotation
tree.clear()
rsh_item = self._createFMAItem(tree, 'right side of heart', 'FMA_7165')
self._createFMAItem(rsh_item, 'ventricle', 'FMA_7098')
self._createFMAItem(rsh_item, 'atrium', 'FMA_7096')
self._createFMAItem(rsh_item, 'auricle', 'FMA_7218')
lsh_item = self._createFMAItem(tree, 'left side of heart', 'FMA_7166')
self._createFMAItem(lsh_item, 'ventricle', 'FMA_7101')
self._createFMAItem(lsh_item, 'atrium', 'FMA_7097')
self._createFMAItem(lsh_item, 'auricle', 'FMA_7219')
apex_item = self._createFMAItem(tree, 'apex of heart', 'FMA_7164')
vortex_item = self._createFMAItem(tree, 'vortex of heart', 'FMA_84628')
self._ui.treeWidgetAnnotation.addTopLevelItem(rsh_item)
self._ui.treeWidgetAnnotation.addTopLevelItem(lsh_item)
self._ui.treeWidgetAnnotation.addTopLevelItem(apex_item)
self._ui.treeWidgetAnnotation.addTopLevelItem(vortex_item)
def getModel(self):
return self._model
def registerDoneExecution(self, doneCallback):
self._doneCallback = doneCallback
def _updateUi(self):
pass
def _doneButtonClicked(self):
self._ui.dockWidget.setFloating(False)
self._model.done()
self._model = None
self._doneCallback()
def _scaffoldTypeChanged(self, index):
scaffoldTypeName = self._ui.meshType_comboBox.itemText(index)
self._generator_model.setScaffoldTypeByName(scaffoldTypeName)
self._annotation_model.setScaffoldTypeByName(scaffoldTypeName)
self._refreshScaffoldOptions()
self._refreshParameterSetNames()
def _parameterSetChanged(self, index):
parameterSetName = self._ui.parameterSet_comboBox.itemText(index)
self._generator_model.setParameterSetName(parameterSetName)
self._refreshScaffoldOptions()
def _meshTypeOptionCheckBoxClicked(self, checkBox):
dependentChanges = self._generator_model.setScaffoldOption(
checkBox.objectName(), checkBox.isChecked())
if dependentChanges:
self._refreshScaffoldOptions()
def _subscaffoldBackButtonPressed(self):
self._generator_model.endEditScaffoldPackageOption()
self._refreshScaffoldTypeNames()
self._refreshParameterSetNames()
self._refreshScaffoldOptions()
def _meshTypeOptionScaffoldPackageButtonPressed(self, pushButton):
optionName = pushButton.objectName()
self._generator_model.editScaffoldPackageOption(optionName)
self._refreshScaffoldTypeNames()
self._refreshParameterSetNames()
self._refreshScaffoldOptions()
def _meshTypeOptionLineEditChanged(self, lineEdit):
dependentChanges = self._generator_model.setScaffoldOption(
lineEdit.objectName(), lineEdit.text())
if dependentChanges:
self._refreshScaffoldOptions()
else:
finalValue = self._generator_model.getEditScaffoldOption(
lineEdit.objectName())
lineEdit.setText(str(finalValue))
def _refreshScaffoldOptions(self):
layout = self._ui.meshTypeOptions_frame.layout()
# remove all current mesh type widgets
while layout.count():
child = layout.takeAt(0)
if child.widget():
child.widget().deleteLater()
optionNames = self._generator_model.getEditScaffoldOrderedOptionNames()
for key in optionNames:
value = self._generator_model.getEditScaffoldOption(key)
# print('key ', key, ' value ', value)
if type(value) is bool:
checkBox = QtGui.QCheckBox(self._ui.meshTypeOptions_frame)
checkBox.setObjectName(key)
checkBox.setText(key)
checkBox.setChecked(value)
callback = partial(self._meshTypeOptionCheckBoxClicked,
checkBox)
checkBox.clicked.connect(callback)
layout.addWidget(checkBox)
else:
label = QtGui.QLabel(self._ui.meshTypeOptions_frame)
label.setObjectName(key)
label.setText(key)
layout.addWidget(label)
if isinstance(value, ScaffoldPackage):
pushButton = QtGui.QPushButton()
pushButton.setObjectName(key)
pushButton.setText('Edit >>')
callback = partial(
self._meshTypeOptionScaffoldPackageButtonPressed,
pushButton)
pushButton.clicked.connect(callback)
layout.addWidget(pushButton)
else:
lineEdit = QtGui.QLineEdit(self._ui.meshTypeOptions_frame)
lineEdit.setObjectName(key)
lineEdit.setText(str(value))
callback = partial(self._meshTypeOptionLineEditChanged,
lineEdit)
#lineEdit.returnPressed.connect(callback)
lineEdit.editingFinished.connect(callback)
layout.addWidget(lineEdit)
# refresh or show/hide standard scaffold options for transformation and deleting element ranges
editingRootScaffold = self._generator_model.editingRootScaffoldPackage(
)
self._ui.done_button.setEnabled(editingRootScaffold)
self._ui.subscaffold_frame.setVisible(not editingRootScaffold)
if editingRootScaffold:
self._ui.deleteElementsRanges_lineEdit.setText(
self._generator_model.getDeleteElementsRangesText())
else:
self._ui.subscaffold_label.setText(
self._generator_model.getEditScaffoldOptionDisplayName())
self._ui.deleteElementsRanges_frame.setVisible(editingRootScaffold)
self._ui.rotation_lineEdit.setText(
self._generator_model.getRotationText())
self._ui.scale_lineEdit.setText(self._generator_model.getScaleText())
self._ui.translation_lineEdit.setText(
self._generator_model.getTranslationText())
def _refreshOptions(self):
self._ui.identifier_label.setText('Identifier: ' +
self._model.getIdentifier())
self._ui.displayDataPoints_checkBox.setChecked(
self._segmentation_data_model.isDisplayDataPoints())
self._ui.displayDataContours_checkBox.setChecked(
self._segmentation_data_model.isDisplayDataContours())
self._ui.displayDataRadius_checkBox.setChecked(
self._segmentation_data_model.isDisplayDataRadius())
self._ui.displayDataMarkerPoints_checkBox.setChecked(
self._segmentation_data_model.isDisplayDataMarkerPoints())
self._ui.displayDataMarkerNames_checkBox.setChecked(
self._segmentation_data_model.isDisplayDataMarkerNames())
self._ui.displayData_frame.setVisible(
self._segmentation_data_model.hasData())
self._ui.displayMarkerPoints_checkBox.setChecked(
self._generator_model.isDisplayMarkerPoints())
self._ui.displayAxes_checkBox.setChecked(
self._generator_model.isDisplayAxes())
self._ui.displayElementNumbers_checkBox.setChecked(
self._generator_model.isDisplayElementNumbers())
self._ui.displayElementAxes_checkBox.setChecked(
self._generator_model.isDisplayElementAxes())
self._ui.displayLines_checkBox.setChecked(
self._generator_model.isDisplayLines())
self._ui.displayLinesExterior_checkBox.setChecked(
self._generator_model.isDisplayLinesExterior())
self._ui.displayModelRadius_checkBox.setChecked(
self._generator_model.isDisplayModelRadius())
self._ui.displayNodeDerivativeLabelsD1_checkBox.setChecked(
self._generator_model.isDisplayNodeDerivativeLabels('D1'))
self._ui.displayNodeDerivativeLabelsD2_checkBox.setChecked(
self._generator_model.isDisplayNodeDerivativeLabels('D2'))
self._ui.displayNodeDerivativeLabelsD3_checkBox.setChecked(
self._generator_model.isDisplayNodeDerivativeLabels('D3'))
self._ui.displayNodeDerivativeLabelsD12_checkBox.setChecked(
self._generator_model.isDisplayNodeDerivativeLabels('D12'))
self._ui.displayNodeDerivativeLabelsD13_checkBox.setChecked(
self._generator_model.isDisplayNodeDerivativeLabels('D13'))
self._ui.displayNodeDerivativeLabelsD23_checkBox.setChecked(
self._generator_model.isDisplayNodeDerivativeLabels('D23'))
self._ui.displayNodeDerivativeLabelsD123_checkBox.setChecked(
self._generator_model.isDisplayNodeDerivativeLabels('D123'))
self._ui.displayNodeDerivatives_checkBox.setChecked(
self._generator_model.isDisplayNodeDerivatives())
self._ui.displayNodeNumbers_checkBox.setChecked(
self._generator_model.isDisplayNodeNumbers())
self._ui.displayNodePoints_checkBox.setChecked(
self._generator_model.isDisplayNodePoints())
self._ui.displaySurfaces_checkBox.setChecked(
self._generator_model.isDisplaySurfaces())
self._ui.displaySurfacesExterior_checkBox.setChecked(
self._generator_model.isDisplaySurfacesExterior())
self._ui.displaySurfacesTranslucent_checkBox.setChecked(
self._generator_model.isDisplaySurfacesTranslucent())
self._ui.displaySurfacesWireframe_checkBox.setChecked(
self._generator_model.isDisplaySurfacesWireframe())
index = self._ui.meshType_comboBox.findText(
self._generator_model.getEditScaffoldTypeName())
self._ui.meshType_comboBox.blockSignals(True)
self._ui.meshType_comboBox.setCurrentIndex(index)
self._ui.meshType_comboBox.blockSignals(False)
self._refreshParameterSetNames()
self._refreshScaffoldOptions()
self._ui.done_button.setEnabled(True)
self._ui.subscaffold_frame.setVisible(False)
def _deleteElementRangesLineEditChanged(self):
self._generator_model.setDeleteElementsRangesText(
self._ui.deleteElementsRanges_lineEdit.text())
self._ui.deleteElementsRanges_lineEdit.setText(
self._generator_model.getDeleteElementsRangesText())
def _deleteElementsSelectionButtonPressed(self):
self._generator_model.deleteElementsSelection()
self._ui.deleteElementsRanges_lineEdit.setText(
self._generator_model.getDeleteElementsRangesText())
def _rotationLineEditChanged(self):
self._generator_model.setRotationText(
self._ui.rotation_lineEdit.text())
self._ui.rotation_lineEdit.setText(
self._generator_model.getRotationText())
def _scaleLineEditChanged(self):
self._generator_model.setScaleText(self._ui.scale_lineEdit.text())
self._ui.scale_lineEdit.setText(self._generator_model.getScaleText())
def _translationLineEditChanged(self):
self._generator_model.setTranslationText(
self._ui.translation_lineEdit.text())
self._ui.translation_lineEdit.setText(
self._generator_model.getTranslationText())
def _displayDataPointsClicked(self):
self._segmentation_data_model.setDisplayDataPoints(
self._ui.displayDataPoints_checkBox.isChecked())
def _displayDataContoursClicked(self):
self._segmentation_data_model.setDisplayDataContours(
self._ui.displayDataContours_checkBox.isChecked())
def _displayDataRadiusClicked(self):
self._segmentation_data_model.setDisplayDataRadius(
self._ui.displayDataRadius_checkBox.isChecked())
def _displayDataMarkerPointsClicked(self):
self._segmentation_data_model.setDisplayDataMarkerPoints(
self._ui.displayDataMarkerPoints_checkBox.isChecked())
def _displayDataMarkerNamesClicked(self):
self._segmentation_data_model.setDisplayDataMarkerNames(
self._ui.displayDataMarkerNames_checkBox.isChecked())
def _displayMarkerPointsClicked(self):
self._generator_model.setDisplayMarkerPoints(
self._ui.displayMarkerPoints_checkBox.isChecked())
def _displayAxesClicked(self):
self._generator_model.setDisplayAxes(
self._ui.displayAxes_checkBox.isChecked())
def _displayElementAxesClicked(self):
self._generator_model.setDisplayElementAxes(
self._ui.displayElementAxes_checkBox.isChecked())
def _displayElementNumbersClicked(self):
self._generator_model.setDisplayElementNumbers(
self._ui.displayElementNumbers_checkBox.isChecked())
def _displayLinesClicked(self):
self._generator_model.setDisplayLines(
self._ui.displayLines_checkBox.isChecked())
self._autoPerturbLines()
def _displayLinesExteriorClicked(self):
self._generator_model.setDisplayLinesExterior(
self._ui.displayLinesExterior_checkBox.isChecked())
def _displayModelRadiusClicked(self):
self._generator_model.setDisplayModelRadius(
self._ui.displayModelRadius_checkBox.isChecked())
def _displayNodeDerivativesClicked(self):
self._generator_model.setDisplayNodeDerivatives(
self._ui.displayNodeDerivatives_checkBox.isChecked())
def _displayNodeDerivativeLabelsD1Clicked(self):
self._generator_model.setDisplayNodeDerivativeLabels(
'D1', self._ui.displayNodeDerivativeLabelsD1_checkBox.isChecked())
def _displayNodeDerivativeLabelsD2Clicked(self):
self._generator_model.setDisplayNodeDerivativeLabels(
'D2', self._ui.displayNodeDerivativeLabelsD2_checkBox.isChecked())
def _displayNodeDerivativeLabelsD3Clicked(self):
self._generator_model.setDisplayNodeDerivativeLabels(
'D3', self._ui.displayNodeDerivativeLabelsD3_checkBox.isChecked())
def _displayNodeDerivativeLabelsD12Clicked(self):
self._generator_model.setDisplayNodeDerivativeLabels(
'D12',
self._ui.displayNodeDerivativeLabelsD12_checkBox.isChecked())
def _displayNodeDerivativeLabelsD13Clicked(self):
self._generator_model.setDisplayNodeDerivativeLabels(
'D13',
self._ui.displayNodeDerivativeLabelsD13_checkBox.isChecked())
def _displayNodeDerivativeLabelsD23Clicked(self):
self._generator_model.setDisplayNodeDerivativeLabels(
'D23',
self._ui.displayNodeDerivativeLabelsD23_checkBox.isChecked())
def _displayNodeDerivativeLabelsD123Clicked(self):
self._generator_model.setDisplayNodeDerivativeLabels(
'D123',
self._ui.displayNodeDerivativeLabelsD123_checkBox.isChecked())
def _displayNodeNumbersClicked(self):
self._generator_model.setDisplayNodeNumbers(
self._ui.displayNodeNumbers_checkBox.isChecked())
def _displayNodePointsClicked(self):
self._generator_model.setDisplayNodePoints(
self._ui.displayNodePoints_checkBox.isChecked())
def _displaySurfacesClicked(self):
self._generator_model.setDisplaySurfaces(
self._ui.displaySurfaces_checkBox.isChecked())
self._autoPerturbLines()
def _displaySurfacesExteriorClicked(self):
self._generator_model.setDisplaySurfacesExterior(
self._ui.displaySurfacesExterior_checkBox.isChecked())
def _displaySurfacesTranslucentClicked(self):
self._generator_model.setDisplaySurfacesTranslucent(
self._ui.displaySurfacesTranslucent_checkBox.isChecked())
self._autoPerturbLines()
def _displaySurfacesWireframeClicked(self):
self._generator_model.setDisplaySurfacesWireframe(
self._ui.displaySurfacesWireframe_checkBox.isChecked())
def _annotationItemChanged(self, item):
print(item.text(0))
print(item.data(0, QtCore.Qt.UserRole + 1))
def _viewAll(self):
"""
Ask sceneviewer to show all of scene.
"""
if self._ui.sceneviewer_widget.getSceneviewer() is not None:
self._ui.sceneviewer_widget.viewAll()
class MeshGeneratorWidget(QtWidgets.QWidget):
def __init__(self, model, parent=None):
super(MeshGeneratorWidget, self).__init__(parent)
self._ui = Ui_MeshGeneratorWidget()
self._ui.setupUi(self)
self._model = model
self._generator_model = model.getGeneratorModel()
self._annotation_model = model.getMeshAnnotationModel()
self._segmentation_data_model = model.getSegmentationDataModel()
self._ui.sceneviewer_widget.setContext(model.getContext())
self._ui.sceneviewer_widget.setGeneratorModel(
model.getGeneratorModel())
self._model.getGeneratorModel().registerCustomParametersCallback(
self._customParametersChange)
self._model.registerSceneChangeCallback(self._sceneChanged)
self._generator_model.registerTransformationChangeCallback(
self._transformationChanged)
self._doneCallback = None
self._refreshScaffoldTypeNames()
self._refreshParameterSetNames()
self._refreshAnnotationGroups()
self._makeConnections()
def _graphicsInitialized(self):
"""
Callback for when SceneviewerWidget is initialised
Set custom scene from model
"""
sceneviewer = self._ui.sceneviewer_widget.getSceneviewer()
if sceneviewer is not None:
self._model.loadSettings()
self._refreshOptions()
scene = self._model.getScene()
self._ui.sceneviewer_widget.setScene(scene)
# self._ui.sceneviewer_widget.setSelectModeAll()
sceneviewer.setLookatParametersNonSkew([2.0, -2.0, 1.0],
[0.0, 0.0, 0.0],
[0.0, 0.0, 1.0])
sceneviewer.setTransparencyMode(sceneviewer.TRANSPARENCY_MODE_SLOW)
self._autoPerturbLines()
self._viewAllButtonClicked()
def _customParametersChange(self):
"""
Callback when scaffold options or mesh edits are made, so custom parameter set now in use.
"""
self._refreshParameterSetNames()
def _sceneChanged(self):
# new region for choosing coordinate field from
self._ui.displayModelCoordinates_fieldChooser.setRegion(
self._generator_model.getRegion())
self._ui.displayModelCoordinates_fieldChooser.setField(
self._generator_model.getModelCoordinatesField())
sceneviewer = self._ui.sceneviewer_widget.getSceneviewer()
if sceneviewer is not None:
scene = self._model.getScene()
self._ui.sceneviewer_widget.setScene(scene)
self._autoPerturbLines()
def _transformationChanged(self):
self._ui.rotation_lineEdit.setText(
self._generator_model.getRotationText())
self._ui.scale_lineEdit.setText(self._generator_model.getScaleText())
self._ui.translation_lineEdit.setText(
self._generator_model.getTranslationText())
def _autoPerturbLines(self):
"""
Enable scene viewer perturb lines iff solid surfaces are drawn with lines.
Call whenever lines, surfaces or translucency changes
"""
sceneviewer = self._ui.sceneviewer_widget.getSceneviewer()
if sceneviewer is not None:
sceneviewer.setPerturbLinesFlag(
self._generator_model.needPerturbLines())
def _makeConnections(self):
self._ui.sceneviewer_widget.graphicsInitialized.connect(
self._graphicsInitialized)
self._ui.done_pushButton.clicked.connect(self._doneButtonClicked)
self._ui.stdViews_pushButton.clicked.connect(
self._stdViewsButtonClicked)
self._ui.viewAll_pushButton.clicked.connect(self._viewAllButtonClicked)
self._ui.subscaffoldBack_pushButton.clicked.connect(
self._subscaffoldBackButtonPressed)
self._ui.meshType_comboBox.currentIndexChanged.connect(
self._scaffoldTypeChanged)
self._ui.parameterSet_comboBox.currentIndexChanged.connect(
self._parameterSetChanged)
self._ui.deleteElementsRanges_lineEdit.editingFinished.connect(
self._deleteElementRangesLineEditChanged)
self._ui.deleteElementsSelection_pushButton.clicked.connect(
self._deleteElementsSelectionButtonPressed)
self._ui.rotation_lineEdit.editingFinished.connect(
self._rotationLineEditChanged)
self._ui.scale_lineEdit.editingFinished.connect(
self._scaleLineEditChanged)
self._ui.translation_lineEdit.editingFinished.connect(
self._translationLineEditChanged)
self._ui.applyTransformation_pushButton.clicked.connect(
self._applyTransformationButtonPressed)
self._ui.displayDataPoints_checkBox.clicked.connect(
self._displayDataPointsClicked)
self._ui.displayDataContours_checkBox.clicked.connect(
self._displayDataContoursClicked)
self._ui.displayDataRadius_checkBox.clicked.connect(
self._displayDataRadiusClicked)
self._ui.displayDataMarkerPoints_checkBox.clicked.connect(
self._displayDataMarkerPointsClicked)
self._ui.displayDataMarkerNames_checkBox.clicked.connect(
self._displayDataMarkerNamesClicked)
self._ui.displayMarkerPoints_checkBox.clicked.connect(
self._displayMarkerPointsClicked)
self._ui.displayModelCoordinates_fieldChooser.setRegion(
self._generator_model.getRegion())
self._ui.displayModelCoordinates_fieldChooser.setConditional(
fieldIsManagedCoordinates)
self._ui.displayModelCoordinates_fieldChooser.currentIndexChanged.connect(
self._displayModelCoordinatesFieldChanged)
self._ui.displayAxes_checkBox.clicked.connect(self._displayAxesClicked)
self._ui.displayElementAxes_checkBox.clicked.connect(
self._displayElementAxesClicked)
self._ui.displayElementNumbers_checkBox.clicked.connect(
self._displayElementNumbersClicked)
self._ui.displayLines_checkBox.clicked.connect(
self._displayLinesClicked)
self._ui.displayLinesExterior_checkBox.clicked.connect(
self._displayLinesExteriorClicked)
self._ui.displayModelRadius_checkBox.clicked.connect(
self._displayModelRadiusClicked)
self._ui.displayNodeDerivativeLabelsD1_checkBox.clicked.connect(
self._displayNodeDerivativeLabelsD1Clicked)
self._ui.displayNodeDerivativeLabelsD2_checkBox.clicked.connect(
self._displayNodeDerivativeLabelsD2Clicked)
self._ui.displayNodeDerivativeLabelsD3_checkBox.clicked.connect(
self._displayNodeDerivativeLabelsD3Clicked)
self._ui.displayNodeDerivativeLabelsD12_checkBox.clicked.connect(
self._displayNodeDerivativeLabelsD12Clicked)
self._ui.displayNodeDerivativeLabelsD13_checkBox.clicked.connect(
self._displayNodeDerivativeLabelsD13Clicked)
self._ui.displayNodeDerivativeLabelsD23_checkBox.clicked.connect(
self._displayNodeDerivativeLabelsD23Clicked)
self._ui.displayNodeDerivativeLabelsD123_checkBox.clicked.connect(
self._displayNodeDerivativeLabelsD123Clicked)
self._ui.displayNodeDerivatives_checkBox.clicked.connect(
self._displayNodeDerivativesClicked)
self._ui.displayNodeNumbers_checkBox.clicked.connect(
self._displayNodeNumbersClicked)
self._ui.displayNodePoints_checkBox.clicked.connect(
self._displayNodePointsClicked)
self._ui.displaySurfaces_checkBox.clicked.connect(
self._displaySurfacesClicked)
self._ui.displaySurfacesExterior_checkBox.clicked.connect(
self._displaySurfacesExteriorClicked)
self._ui.displaySurfacesTranslucent_checkBox.clicked.connect(
self._displaySurfacesTranslucentClicked)
self._ui.displaySurfacesWireframe_checkBox.clicked.connect(
self._displaySurfacesWireframeClicked)
self._ui.annotationGroup_comboBox.currentIndexChanged.connect(
self._annotationGroupChanged)
self._ui.annotationGroupNew_pushButton.clicked.connect(
self._annotationGroupNewButtonClicked)
self._ui.annotationGroupRedefine_pushButton.clicked.connect(
self._annotationGroupRedefineButtonClicked)
self._ui.annotationGroupDelete_pushButton.clicked.connect(
self._annotationGroupDeleteButtonClicked)
self._ui.annotationGroupOntId_lineEdit.editingFinished.connect(
self._annotationGroupOntIdLineEditChanged)
def keyPressEvent(self, event):
if (event.key() == QtCore.Qt.Key_S) and (event.isAutoRepeat()
== False):
self._ui.sceneviewer_widget._selectionKeyPressed = True
event.setAccepted(True)
else:
event.ignore()
def keyReleaseEvent(self, event):
if (event.key() == QtCore.Qt.Key_S) and (event.isAutoRepeat()
== False):
self._ui.sceneviewer_widget._selectionKeyPressed = False
event.setAccepted(True)
else:
event.ignore()
def _refreshComboBoxNames(self, comboBox, names, currentName):
comboBox.blockSignals(True)
comboBox.clear()
currentIndex = 0
index = 0
for name in names:
comboBox.addItem(name)
if name == currentName:
currentIndex = index
index += 1
comboBox.setCurrentIndex(currentIndex)
comboBox.blockSignals(False)
def _refreshScaffoldTypeNames(self):
self._refreshComboBoxNames(
self._ui.meshType_comboBox,
self._generator_model.getAvailableScaffoldTypeNames(),
self._generator_model.getEditScaffoldTypeName())
def _refreshParameterSetNames(self):
self._refreshComboBoxNames(
self._ui.parameterSet_comboBox,
self._generator_model.getAvailableParameterSetNames(),
self._generator_model.getParameterSetName())
def _refreshAnnotationGroups(self):
annotationGroups = self._generator_model.getAnnotationGroups()
currentAnnotationGroup = self._generator_model.getCurrentAnnotationGroup(
)
self._refreshComboBoxNames(
self._ui.annotationGroup_comboBox, ['-'] + [
annotationGroup.getName()
for annotationGroup in annotationGroups
],
currentAnnotationGroup.getName()
if currentAnnotationGroup else '-')
def getModel(self):
return self._model
def registerDoneExecution(self, doneCallback):
self._doneCallback = doneCallback
def _doneButtonClicked(self):
self._ui.dockWidget.setFloating(False)
self._model.done()
self._model = None
self._doneCallback()
def _stdViewsButtonClicked(self):
sceneviewer = self._ui.sceneviewer_widget.getSceneviewer()
if sceneviewer is not None:
result, eyePosition, lookatPosition, upVector = sceneviewer.getLookatParameters(
)
upVector = normalize(upVector)
viewVector = sub(lookatPosition, eyePosition)
viewDistance = magnitude(viewVector)
viewVector = normalize(viewVector)
viewX = dot(viewVector, [1.0, 0.0, 0.0])
viewY = dot(viewVector, [0.0, 1.0, 0.0])
viewZ = dot(viewVector, [0.0, 0.0, 1.0])
upX = dot(upVector, [1.0, 0.0, 0.0])
upY = dot(upVector, [0.0, 1.0, 0.0])
upZ = dot(upVector, [0.0, 0.0, 1.0])
if (viewZ < -0.999) and (upY > 0.999):
# XY -> XZ
viewVector = [0.0, 1.0, 0.0]
upVector = [0.0, 0.0, 1.0]
elif (viewY > 0.999) and (upZ > 0.999):
# XZ -> YZ
viewVector = [-1.0, 0.0, 0.0]
upVector = [0.0, 0.0, 1.0]
else:
# XY
viewVector = [0.0, 0.0, -1.0]
upVector = [0.0, 1.0, 0.0]
eyePosition = sub(lookatPosition, mult(viewVector, viewDistance))
sceneviewer.setLookatParametersNonSkew(eyePosition, lookatPosition,
upVector)
def _viewAllButtonClicked(self):
if self._ui.sceneviewer_widget.getSceneviewer() is not None:
self._ui.sceneviewer_widget.viewAll()
def _annotationGroupChanged(self, index):
annotationGroupName = self._ui.annotationGroup_comboBox.itemText(index)
#print('_annotationGroupChanged', 'index', index, annotationGroupName)
self._generator_model.setCurrentAnnotationGroupByName(
annotationGroupName)
self._refreshCurrentAnnotationGroupSettings()
def _annotationGroupNewButtonClicked(self):
annotationGroup = self._generator_model.createUserAnnotationGroup()
self._refreshAnnotationGroups()
self._refreshCurrentAnnotationGroupSettings()
def _annotationGroupRedefineButtonClicked(self):
annotationGroup = self._generator_model.getCurrentAnnotationGroup()
if annotationGroup:
reply = QtWidgets.QMessageBox.question(
self, 'Confirm action', 'Redefine annotation group \'' +
annotationGroup.getName() + '\' from selection?',
QtWidgets.QMessageBox.Yes | QtWidgets.QMessageBox.No,
QtWidgets.QMessageBox.No)
if reply == QtWidgets.QMessageBox.Yes:
if self._generator_model.redefineCurrentAnnotationGroupFromSelection(
):
self._refreshCurrentAnnotationGroupSettings()
def _annotationGroupDeleteButtonClicked(self):
annotationGroup = self._generator_model.getCurrentAnnotationGroup()
if annotationGroup:
reply = QtWidgets.QMessageBox.question(
self, 'Confirm action', 'Delete annotation group \'' +
annotationGroup.getName() + '\'?',
QtWidgets.QMessageBox.Yes | QtWidgets.QMessageBox.No,
QtWidgets.QMessageBox.No)
if reply == QtWidgets.QMessageBox.Yes:
if self._generator_model.deleteAnnotationGroup(
annotationGroup):
self._refreshAnnotationGroups()
self._refreshCurrentAnnotationGroupSettings()
def _annotationGroupNameLineEditChanged(self):
newName = self._ui.annotationGroup_comboBox.currentText()
if not self._generator_model.setCurrentAnnotationGroupName(newName):
self._refreshAnnotationGroups()
def _annotationGroupOntIdLineEditChanged(self):
newOntId = self._ui.annotationGroupOntId_lineEdit.text()
if not self._generator_model.setCurrentAnnotationGroupOntId(newOntId):
self._refreshCurrentAnnotationGroupSettings()
def _refreshCurrentAnnotationGroupSettings(self):
'''
Display current annotation group settings.
'''
annotationGroup = self._generator_model.getCurrentAnnotationGroup()
isUser = self._generator_model.isUserAnnotationGroup(annotationGroup)
self._ui.annotationGroup_comboBox.setEditable(isUser)
if isUser:
self._ui.annotationGroup_comboBox.lineEdit(
).editingFinished.connect(self._annotationGroupNameLineEditChanged)
self._ui.annotationGroup_comboBox.setInsertPolicy(
QtWidgets.QComboBox.InsertAtCurrent)
self._ui.annotationGroupOntId_lineEdit.setText(
annotationGroup.getId() if annotationGroup else '-')
self._ui.annotationGroupOntId_lineEdit.setEnabled(isUser)
self._ui.annotationGroupDimension_spinBox.setValue(
annotationGroup.getDimension() if annotationGroup else 0)
self._ui.annotationGroupDimension_spinBox.setEnabled(False)
self._ui.annotationGroupRedefine_pushButton.setEnabled(isUser)
self._ui.annotationGroupDelete_pushButton.setEnabled(isUser)
def _scaffoldTypeChanged(self, index):
scaffoldTypeName = self._ui.meshType_comboBox.itemText(index)
self._generator_model.setScaffoldTypeByName(scaffoldTypeName)
self._annotation_model.setScaffoldTypeByName(scaffoldTypeName)
self._refreshParameterSetNames()
self._refreshScaffoldOptions()
self._refreshAnnotationGroups()
self._refreshCurrentAnnotationGroupSettings()
def _parameterSetChanged(self, index):
parameterSetName = self._ui.parameterSet_comboBox.itemText(index)
self._generator_model.setParameterSetName(parameterSetName)
self._refreshScaffoldOptions()
self._refreshAnnotationGroups()
self._refreshCurrentAnnotationGroupSettings()
def _meshTypeOptionCheckBoxClicked(self, checkBox):
dependentChanges = self._generator_model.setScaffoldOption(
checkBox.objectName(), checkBox.isChecked())
if dependentChanges:
self._refreshScaffoldOptions()
self._refreshAnnotationGroups()
self._refreshCurrentAnnotationGroupSettings()
def _subscaffoldBackButtonPressed(self):
self._generator_model.endEditScaffoldPackageOption()
self._refreshScaffoldTypeNames()
self._refreshParameterSetNames()
self._refreshScaffoldOptions()
self._refreshAnnotationGroups()
self._refreshCurrentAnnotationGroupSettings()
def _meshTypeOptionScaffoldPackageButtonPressed(self, pushButton):
optionName = pushButton.objectName()
self._generator_model.editScaffoldPackageOption(optionName)
self._refreshScaffoldTypeNames()
self._refreshParameterSetNames()
self._refreshScaffoldOptions()
self._refreshAnnotationGroups()
self._refreshCurrentAnnotationGroupSettings()
def _meshTypeInteractiveFunctionButtonPressed(self, pushButton):
functionName = pushButton.objectName()
functionOptions = self._generator_model.getInteractiveFunctionOptions(
functionName)
functionOptionsDialog = FunctionOptionsDialog(functionName,
functionOptions, self)
if functionOptionsDialog.exec():
optionsChanged = self._generator_model.performInteractiveFunction(
functionName, functionOptions)
if optionsChanged:
self._refreshScaffoldOptions()
def _meshTypeOptionLineEditChanged(self, lineEdit):
dependentChanges = self._generator_model.setScaffoldOption(
lineEdit.objectName(), lineEdit.text())
if dependentChanges:
self._refreshScaffoldOptions()
else:
lineEdit.setText(
self._generator_model.getEditScaffoldOptionStr(
lineEdit.objectName()))
self._refreshAnnotationGroups()
self._refreshCurrentAnnotationGroupSettings()
def _refreshScaffoldOptions(self):
layout = self._ui.meshTypeOptions_frame.layout()
# remove all current mesh type widgets
while layout.count():
child = layout.takeAt(0)
if child.widget():
child.widget().deleteLater()
optionNames = self._generator_model.getEditScaffoldOrderedOptionNames()
for key in optionNames:
value = self._generator_model.getEditScaffoldOption(key)
# print('key ', key, ' value ', value)
if type(value) is bool:
checkBox = QtWidgets.QCheckBox(self._ui.meshTypeOptions_frame)
checkBox.setObjectName(key)
checkBox.setText(key)
checkBox.setChecked(value)
callback = partial(self._meshTypeOptionCheckBoxClicked,
checkBox)
checkBox.clicked.connect(callback)
layout.addWidget(checkBox)
else:
label = QtWidgets.QLabel(self._ui.meshTypeOptions_frame)
label.setObjectName(key)
label.setText(key)
layout.addWidget(label)
if isinstance(value, ScaffoldPackage):
pushButton = QtWidgets.QPushButton()
pushButton.setObjectName(key)
pushButton.setText('Edit >>')
callback = partial(
self._meshTypeOptionScaffoldPackageButtonPressed,
pushButton)
pushButton.clicked.connect(callback)
layout.addWidget(pushButton)
else:
lineEdit = QtWidgets.QLineEdit(
self._ui.meshTypeOptions_frame)
lineEdit.setObjectName(key)
lineEdit.setText(
self._generator_model.getEditScaffoldOptionStr(key))
callback = partial(self._meshTypeOptionLineEditChanged,
lineEdit)
lineEdit.editingFinished.connect(callback)
layout.addWidget(lineEdit)
interativeFunctions = self._generator_model.getInteractiveFunctions()
for interactiveFunction in interativeFunctions:
pushButton = QtWidgets.QPushButton()
pushButton.setObjectName(interactiveFunction[0])
pushButton.setText(interactiveFunction[0])
callback = partial(self._meshTypeInteractiveFunctionButtonPressed,
pushButton)
pushButton.clicked.connect(callback)
layout.addWidget(pushButton)
# refresh or show/hide standard scaffold options for transformation and deleting element ranges
editingRootScaffold = self._generator_model.editingRootScaffoldPackage(
)
self._ui.done_pushButton.setEnabled(editingRootScaffold)
self._ui.subscaffold_frame.setVisible(not editingRootScaffold)
if editingRootScaffold:
self._ui.deleteElementsRanges_lineEdit.setText(
self._generator_model.getDeleteElementsRangesText())
else:
self._ui.subscaffold_label.setText(
self._generator_model.getEditScaffoldOptionDisplayName())
self._ui.deleteElementsRanges_frame.setVisible(editingRootScaffold)
self._ui.rotation_lineEdit.setText(
self._generator_model.getRotationText())
self._ui.scale_lineEdit.setText(self._generator_model.getScaleText())
self._ui.translation_lineEdit.setText(
self._generator_model.getTranslationText())
def _refreshOptions(self):
self._ui.identifier_label.setText('Identifier: ' +
self._model.getIdentifier())
self._ui.displayDataPoints_checkBox.setChecked(
self._segmentation_data_model.isDisplayDataPoints())
self._ui.displayDataContours_checkBox.setChecked(
self._segmentation_data_model.isDisplayDataContours())
self._ui.displayDataRadius_checkBox.setChecked(
self._segmentation_data_model.isDisplayDataRadius())
self._ui.displayDataMarkerPoints_checkBox.setChecked(
self._segmentation_data_model.isDisplayDataMarkerPoints())
self._ui.displayDataMarkerNames_checkBox.setChecked(
self._segmentation_data_model.isDisplayDataMarkerNames())
self._ui.displayData_frame.setVisible(
self._segmentation_data_model.hasData())
self._ui.displayMarkerPoints_checkBox.setChecked(
self._generator_model.isDisplayMarkerPoints())
self._ui.displayAxes_checkBox.setChecked(
self._generator_model.isDisplayAxes())
self._ui.displayElementNumbers_checkBox.setChecked(
self._generator_model.isDisplayElementNumbers())
self._ui.displayElementAxes_checkBox.setChecked(
self._generator_model.isDisplayElementAxes())
self._ui.displayLines_checkBox.setChecked(
self._generator_model.isDisplayLines())
self._ui.displayLinesExterior_checkBox.setChecked(
self._generator_model.isDisplayLinesExterior())
self._ui.displayModelRadius_checkBox.setChecked(
self._generator_model.isDisplayModelRadius())
self._ui.displayNodeDerivativeLabelsD1_checkBox.setChecked(
self._generator_model.isDisplayNodeDerivativeLabels('D1'))
self._ui.displayNodeDerivativeLabelsD2_checkBox.setChecked(
self._generator_model.isDisplayNodeDerivativeLabels('D2'))
self._ui.displayNodeDerivativeLabelsD3_checkBox.setChecked(
self._generator_model.isDisplayNodeDerivativeLabels('D3'))
self._ui.displayNodeDerivativeLabelsD12_checkBox.setChecked(
self._generator_model.isDisplayNodeDerivativeLabels('D12'))
self._ui.displayNodeDerivativeLabelsD13_checkBox.setChecked(
self._generator_model.isDisplayNodeDerivativeLabels('D13'))
self._ui.displayNodeDerivativeLabelsD23_checkBox.setChecked(
self._generator_model.isDisplayNodeDerivativeLabels('D23'))
self._ui.displayNodeDerivativeLabelsD123_checkBox.setChecked(
self._generator_model.isDisplayNodeDerivativeLabels('D123'))
displayNodeDerivatives = self._generator_model.getDisplayNodeDerivatives(
)
self._ui.displayNodeDerivatives_checkBox.setCheckState(
QtCore.Qt.Unchecked if not displayNodeDerivatives else QtCore.Qt.
PartiallyChecked if (
displayNodeDerivatives == 1) else QtCore.Qt.Checked)
self._ui.displayNodeNumbers_checkBox.setChecked(
self._generator_model.isDisplayNodeNumbers())
self._ui.displayNodePoints_checkBox.setChecked(
self._generator_model.isDisplayNodePoints())
self._ui.displaySurfaces_checkBox.setChecked(
self._generator_model.isDisplaySurfaces())
self._ui.displaySurfacesExterior_checkBox.setChecked(
self._generator_model.isDisplaySurfacesExterior())
self._ui.displaySurfacesTranslucent_checkBox.setChecked(
self._generator_model.isDisplaySurfacesTranslucent())
self._ui.displaySurfacesWireframe_checkBox.setChecked(
self._generator_model.isDisplaySurfacesWireframe())
index = self._ui.meshType_comboBox.findText(
self._generator_model.getEditScaffoldTypeName())
self._ui.meshType_comboBox.blockSignals(True)
self._ui.meshType_comboBox.setCurrentIndex(index)
self._ui.meshType_comboBox.blockSignals(False)
self._refreshParameterSetNames()
self._refreshScaffoldOptions()
self._refreshAnnotationGroups()
self._refreshCurrentAnnotationGroupSettings()
self._ui.done_pushButton.setEnabled(True)
self._ui.subscaffold_frame.setVisible(False)
def _deleteElementRangesLineEditChanged(self):
self._generator_model.setDeleteElementsRangesText(
self._ui.deleteElementsRanges_lineEdit.text())
self._ui.deleteElementsRanges_lineEdit.setText(
self._generator_model.getDeleteElementsRangesText())
def _deleteElementsSelectionButtonPressed(self):
self._generator_model.deleteElementsSelection()
self._ui.deleteElementsRanges_lineEdit.setText(
self._generator_model.getDeleteElementsRangesText())
def _rotationLineEditChanged(self):
self._generator_model.setRotationText(
self._ui.rotation_lineEdit.text())
self._ui.rotation_lineEdit.setText(
self._generator_model.getRotationText())
def _scaleLineEditChanged(self):
self._generator_model.setScaleText(self._ui.scale_lineEdit.text())
self._ui.scale_lineEdit.setText(self._generator_model.getScaleText())
def _translationLineEditChanged(self):
self._generator_model.setTranslationText(
self._ui.translation_lineEdit.text())
self._ui.translation_lineEdit.setText(
self._generator_model.getTranslationText())
def _applyTransformationButtonPressed(self):
self._generator_model.applyTransformation()
self._transformationChanged()
def _displayDataPointsClicked(self):
self._segmentation_data_model.setDisplayDataPoints(
self._ui.displayDataPoints_checkBox.isChecked())
def _displayDataContoursClicked(self):
self._segmentation_data_model.setDisplayDataContours(
self._ui.displayDataContours_checkBox.isChecked())
def _displayDataRadiusClicked(self):
self._segmentation_data_model.setDisplayDataRadius(
self._ui.displayDataRadius_checkBox.isChecked())
def _displayDataMarkerPointsClicked(self):
self._segmentation_data_model.setDisplayDataMarkerPoints(
self._ui.displayDataMarkerPoints_checkBox.isChecked())
def _displayDataMarkerNamesClicked(self):
self._segmentation_data_model.setDisplayDataMarkerNames(
self._ui.displayDataMarkerNames_checkBox.isChecked())
def _displayMarkerPointsClicked(self):
self._generator_model.setDisplayMarkerPoints(
self._ui.displayMarkerPoints_checkBox.isChecked())
def _displayModelCoordinatesFieldChanged(self, index):
"""
Callback for change in model coordinates field chooser widget.
"""
field = self._ui.displayModelCoordinates_fieldChooser.getField()
if field:
self._generator_model.setModelCoordinatesField(
field) # will re-create graphics
def _displayAxesClicked(self):
self._generator_model.setDisplayAxes(
self._ui.displayAxes_checkBox.isChecked())
def _displayElementAxesClicked(self):
self._generator_model.setDisplayElementAxes(
self._ui.displayElementAxes_checkBox.isChecked())
def _displayElementNumbersClicked(self):
self._generator_model.setDisplayElementNumbers(
self._ui.displayElementNumbers_checkBox.isChecked())
def _displayLinesClicked(self):
self._generator_model.setDisplayLines(
self._ui.displayLines_checkBox.isChecked())
self._autoPerturbLines()
def _displayLinesExteriorClicked(self):
self._generator_model.setDisplayLinesExterior(
self._ui.displayLinesExterior_checkBox.isChecked())
def _displayModelRadiusClicked(self):
self._generator_model.setDisplayModelRadius(
self._ui.displayModelRadius_checkBox.isChecked())
def _displayNodeDerivativesClicked(self):
checkState = self._ui.displayNodeDerivatives_checkBox.checkState()
triState = 0 if (checkState == QtCore.Qt.Unchecked) else 1 if (
checkState == QtCore.Qt.PartiallyChecked) else 2
self._generator_model.setDisplayNodeDerivatives(triState)
def _displayNodeDerivativeLabelsD1Clicked(self):
self._generator_model.setDisplayNodeDerivativeLabels(
'D1', self._ui.displayNodeDerivativeLabelsD1_checkBox.isChecked())
def _displayNodeDerivativeLabelsD2Clicked(self):
self._generator_model.setDisplayNodeDerivativeLabels(
'D2', self._ui.displayNodeDerivativeLabelsD2_checkBox.isChecked())
def _displayNodeDerivativeLabelsD3Clicked(self):
self._generator_model.setDisplayNodeDerivativeLabels(
'D3', self._ui.displayNodeDerivativeLabelsD3_checkBox.isChecked())
def _displayNodeDerivativeLabelsD12Clicked(self):
self._generator_model.setDisplayNodeDerivativeLabels(
'D12',
self._ui.displayNodeDerivativeLabelsD12_checkBox.isChecked())
def _displayNodeDerivativeLabelsD13Clicked(self):
self._generator_model.setDisplayNodeDerivativeLabels(
'D13',
self._ui.displayNodeDerivativeLabelsD13_checkBox.isChecked())
def _displayNodeDerivativeLabelsD23Clicked(self):
self._generator_model.setDisplayNodeDerivativeLabels(
'D23',
self._ui.displayNodeDerivativeLabelsD23_checkBox.isChecked())
def _displayNodeDerivativeLabelsD123Clicked(self):
self._generator_model.setDisplayNodeDerivativeLabels(
'D123',
self._ui.displayNodeDerivativeLabelsD123_checkBox.isChecked())
def _displayNodeNumbersClicked(self):
self._generator_model.setDisplayNodeNumbers(
self._ui.displayNodeNumbers_checkBox.isChecked())
def _displayNodePointsClicked(self):
self._generator_model.setDisplayNodePoints(
self._ui.displayNodePoints_checkBox.isChecked())
def _displaySurfacesClicked(self):
self._generator_model.setDisplaySurfaces(
self._ui.displaySurfaces_checkBox.isChecked())
self._autoPerturbLines()
def _displaySurfacesExteriorClicked(self):
self._generator_model.setDisplaySurfacesExterior(
self._ui.displaySurfacesExterior_checkBox.isChecked())
def _displaySurfacesTranslucentClicked(self):
self._generator_model.setDisplaySurfacesTranslucent(
self._ui.displaySurfacesTranslucent_checkBox.isChecked())
self._autoPerturbLines()
def _displaySurfacesWireframeClicked(self):
self._generator_model.setDisplaySurfacesWireframe(
self._ui.displaySurfacesWireframe_checkBox.isChecked())
class MeshGeneratorWidget(QtGui.QWidget):
'''
classdocs
'''
def __init__(self, model, parent=None):
'''
Constructor
'''
super(MeshGeneratorWidget, self).__init__(parent)
self._ui = Ui_MeshGeneratorWidget()
self._ui.setupUi(self)
self._model = model
self._ui.sceneviewer_widget.setContext(model.getContext())
self._ui.sceneviewer_widget.graphicsInitialized.connect(self._graphicsInitialized)
self._model.registerSceneChangeCallback(self._sceneChanged)
self._doneCallback = None
self._refreshOptions()
self._makeConnections()
def _graphicsInitialized(self):
'''
Callback for when SceneviewerWidget is initialised
Set custom scene from model
'''
sceneviewer = self._ui.sceneviewer_widget.getSceneviewer()
if sceneviewer is not None:
scene = self._model.getScene()
self._ui.sceneviewer_widget.setScene(scene)
#self._ui.sceneviewer_widget.setSelectModeAll()
sceneviewer.setLookatParametersNonSkew([2.0, -2.0, 1.0], [0.0, 0.0, 0.0], [0.0, 0.0, 1.0])
sceneviewer.setTransparencyMode(sceneviewer.TRANSPARENCY_MODE_SLOW)
self._viewAll()
def _sceneChanged(self):
sceneviewer = self._ui.sceneviewer_widget.getSceneviewer()
if sceneviewer is not None:
scene = self._model.getScene()
self._ui.sceneviewer_widget.setScene(scene)
def _makeConnections(self):
self._ui.done_button.clicked.connect(self._doneButtonClicked)
self._ui.viewAll_button.clicked.connect(self._viewAll)
meshTypeNames = self._model.getAllMeshTypeNames()
index = 0
for meshTypeName in meshTypeNames:
self._ui.meshType_comboBox.addItem(meshTypeName)
if meshTypeName == self._model.getMeshTypeName():
self._ui.meshType_comboBox.setCurrentIndex(index)
index = index + 1
self._ui.meshType_comboBox.currentIndexChanged.connect(self._meshTypeChanged)
self._ui.deleteElementsRanges_lineEdit.returnPressed.connect(self._deleteElementRangesLineEditChanged)
self._ui.deleteElementsRanges_lineEdit.editingFinished.connect(self._deleteElementRangesLineEditChanged)
self._ui.scale_lineEdit.returnPressed.connect(self._scaleLineEditChanged)
self._ui.scale_lineEdit.editingFinished.connect(self._scaleLineEditChanged)
self._ui.displayAxes_checkBox.clicked.connect(self._displayAxesClicked)
self._ui.displayElementNumbers_checkBox.clicked.connect(self._displayElementNumbersClicked)
self._ui.displayLines_checkBox.clicked.connect(self._displayLinesClicked)
self._ui.displayNodeDerivatives_checkBox.clicked.connect(self._displayNodeDerivativesClicked)
self._ui.displayNodeNumbers_checkBox.clicked.connect(self._displayNodeNumbersClicked)
self._ui.displaySurfaces_checkBox.clicked.connect(self._displaySurfacesClicked)
self._ui.displaySurfacesExterior_checkBox.clicked.connect(self._displaySurfacesExteriorClicked)
self._ui.displaySurfacesTranslucent_checkBox.clicked.connect(self._displaySurfacesTranslucentClicked)
self._ui.displaySurfacesWireframe_checkBox.clicked.connect(self._displaySurfacesWireframeClicked)
self._ui.displayXiAxes_checkBox.clicked.connect(self._displayXiAxesClicked)
def getModel(self):
return self._model
def registerDoneExecution(self, doneCallback):
self._doneCallback = doneCallback
def _doneButtonClicked(self):
self._ui.dockWidget.setFloating(False)
self._model.done()
self._model = None
self._doneCallback()
def _meshTypeChanged(self, index):
meshTypeName = self._ui.meshType_comboBox.itemText(index)
self._model.setMeshTypeByName(meshTypeName)
self._refreshMeshTypeOptions()
def _meshTypeOptionCheckBoxClicked(self, checkBox):
self._model.setMeshTypeOption(checkBox.objectName(), checkBox.isChecked())
def _meshTypeOptionLineEditChanged(self, lineEdit):
self._model.setMeshTypeOption(lineEdit.objectName(), lineEdit.text())
finalValue = self._model.getMeshTypeOption(lineEdit.objectName())
lineEdit.setText(str(finalValue))
def _refreshMeshTypeOptions(self):
layout = self._ui.meshTypeOptions_frame.layout()
# remove all current mesh type widgets
while layout.count():
child = layout.takeAt(0)
if child.widget():
child.widget().deleteLater()
optionNames = self._model.getMeshTypeOrderedOptionNames()
for key in optionNames:
value = self._model.getMeshTypeOption(key)
# print('key ', key, ' value ', value)
if type(value) is bool:
checkBox = QtGui.QCheckBox(self._ui.meshTypeOptions_frame)
checkBox.setObjectName(key)
checkBox.setText(key)
checkBox.setChecked(value)
callback = partial(self._meshTypeOptionCheckBoxClicked, checkBox)
checkBox.clicked.connect(callback)
layout.addWidget(checkBox)
else:
label = QtGui.QLabel(self._ui.meshTypeOptions_frame)
label.setObjectName(key)
label.setText(key)
layout.addWidget(label)
lineEdit = QtGui.QLineEdit(self._ui.meshTypeOptions_frame)
lineEdit.setObjectName(key)
lineEdit.setText(str(value))
callback = partial(self._meshTypeOptionLineEditChanged, lineEdit)
lineEdit.returnPressed.connect(callback)
lineEdit.editingFinished.connect(callback)
layout.addWidget(lineEdit)
def _refreshOptions(self):
self._ui.identifier_label.setText('Identifier: ' + self._model.getIdentifier())
self._ui.deleteElementsRanges_lineEdit.setText(self._model.getDeleteElementsRangesText())
self._ui.scale_lineEdit.setText(self._model.getScaleText())
self._refreshMeshTypeOptions()
self._ui.displayAxes_checkBox.setChecked(self._model.isDisplayAxes())
self._ui.displayElementNumbers_checkBox.setChecked(self._model.isDisplayElementNumbers())
self._ui.displayLines_checkBox.setChecked(self._model.isDisplayLines())
self._ui.displayNodeDerivatives_checkBox.setChecked(self._model.isDisplayNodeDerivatives())
self._ui.displayNodeNumbers_checkBox.setChecked(self._model.isDisplayNodeNumbers())
self._ui.displaySurfaces_checkBox.setChecked(self._model.isDisplaySurfaces())
self._ui.displaySurfacesExterior_checkBox.setChecked(self._model.isDisplaySurfacesExterior())
self._ui.displaySurfacesTranslucent_checkBox.setChecked(self._model.isDisplaySurfacesTranslucent())
self._ui.displaySurfacesWireframe_checkBox.setChecked(self._model.isDisplaySurfacesWireframe())
self._ui.displayXiAxes_checkBox.setChecked(self._model.isDisplayXiAxes())
def _deleteElementRangesLineEditChanged(self):
self._model.setDeleteElementsRangesText(self._ui.deleteElementsRanges_lineEdit.text())
self._ui.deleteElementsRanges_lineEdit.setText(self._model.getDeleteElementsRangesText())
def _scaleLineEditChanged(self):
self._model.setScaleText(self._ui.scale_lineEdit.text())
self._ui.scale_lineEdit.setText(self._model.getScaleText())
def _displayAxesClicked(self):
self._model.setDisplayAxes(self._ui.displayAxes_checkBox.isChecked())
def _displayElementNumbersClicked(self):
self._model.setDisplayElementNumbers(self._ui.displayElementNumbers_checkBox.isChecked())
def _displayLinesClicked(self):
self._model.setDisplayLines(self._ui.displayLines_checkBox.isChecked())
def _displayNodeDerivativesClicked(self):
self._model.setDisplayNodeDerivatives(self._ui.displayNodeDerivatives_checkBox.isChecked())
def _displayNodeNumbersClicked(self):
self._model.setDisplayNodeNumbers(self._ui.displayNodeNumbers_checkBox.isChecked())
def _displaySurfacesClicked(self):
self._model.setDisplaySurfaces(self._ui.displaySurfaces_checkBox.isChecked())
def _displaySurfacesExteriorClicked(self):
self._model.setDisplaySurfacesExterior(self._ui.displaySurfacesExterior_checkBox.isChecked())
def _displaySurfacesTranslucentClicked(self):
self._model.setDisplaySurfacesTranslucent(self._ui.displaySurfacesTranslucent_checkBox.isChecked())
def _displaySurfacesWireframeClicked(self):
self._model.setDisplaySurfacesWireframe(self._ui.displaySurfacesWireframe_checkBox.isChecked())
def _displayXiAxesClicked(self):
self._model.setDisplayXiAxes(self._ui.displayXiAxes_checkBox.isChecked())
def _viewAll(self):
'''
Ask sceneviewer to show all of scene.
'''
if self._ui.sceneviewer_widget.getSceneviewer() is not None:
self._ui.sceneviewer_widget.viewAll()
class MeshGeneratorWidget(QtGui.QWidget):
def __init__(self, model, parent=None):
super(MeshGeneratorWidget, self).__init__(parent)
self._ui = Ui_MeshGeneratorWidget()
self._ui.setupUi(self)
self._model = model
self._generator_model = model.getGeneratorModel()
self._annotation_model = model.getMeshAnnotationModel()
self._ui.sceneviewer_widget.setContext(model.getContext())
self._model.registerSceneChangeCallback(self._sceneChanged)
self._doneCallback = None
# self._populateAnnotationTree()
self._refreshMeshTypeNames()
self._refreshParameterSetNames()
self._makeConnections()
def _graphicsInitialized(self):
"""
Callback for when SceneviewerWidget is initialised
Set custom scene from model
"""
sceneviewer = self._ui.sceneviewer_widget.getSceneviewer()
if sceneviewer is not None:
self._model.loadSettings()
self._refreshOptions()
scene = self._model.getScene()
self._ui.sceneviewer_widget.setScene(scene)
# self._ui.sceneviewer_widget.setSelectModeAll()
sceneviewer.setLookatParametersNonSkew([2.0, -2.0, 1.0],
[0.0, 0.0, 0.0],
[0.0, 0.0, 1.0])
sceneviewer.setTransparencyMode(sceneviewer.TRANSPARENCY_MODE_SLOW)
self._autoPerturbLines()
self._viewAll()
def _sceneChanged(self):
sceneviewer = self._ui.sceneviewer_widget.getSceneviewer()
if sceneviewer is not None:
scene = self._model.getScene()
self._ui.sceneviewer_widget.setScene(scene)
self._autoPerturbLines()
def _autoPerturbLines(self):
"""
Enable scene viewer perturb lines iff solid surfaces are drawn with lines.
Call whenever lines, surfaces or translucency changes
"""
sceneviewer = self._ui.sceneviewer_widget.getSceneviewer()
if sceneviewer is not None:
sceneviewer.setPerturbLinesFlag(
self._generator_model.needPerturbLines())
def _makeConnections(self):
self._ui.sceneviewer_widget.graphicsInitialized.connect(
self._graphicsInitialized)
self._ui.done_button.clicked.connect(self._doneButtonClicked)
self._ui.viewAll_button.clicked.connect(self._viewAll)
self._ui.meshType_comboBox.currentIndexChanged.connect(
self._meshTypeChanged)
self._ui.parameterSet_comboBox.currentIndexChanged.connect(
self._parameterSetChanged)
self._ui.deleteElementsRanges_lineEdit.returnPressed.connect(
self._deleteElementRangesLineEditChanged)
self._ui.deleteElementsRanges_lineEdit.editingFinished.connect(
self._deleteElementRangesLineEditChanged)
self._ui.scale_lineEdit.returnPressed.connect(
self._scaleLineEditChanged)
self._ui.scale_lineEdit.editingFinished.connect(
self._scaleLineEditChanged)
self._ui.displayAnnotationPoints_checkBox.clicked.connect(
self._displayAnnotationPointsClicked)
self._ui.displayAxes_checkBox.clicked.connect(self._displayAxesClicked)
self._ui.displayElementNumbers_checkBox.clicked.connect(
self._displayElementNumbersClicked)
self._ui.displayLines_checkBox.clicked.connect(
self._displayLinesClicked)
self._ui.displayLinesExterior_checkBox.clicked.connect(
self._displayLinesExteriorClicked)
self._ui.displayNodeDerivatives_checkBox.clicked.connect(
self._displayNodeDerivativesClicked)
self._ui.displayNodeNumbers_checkBox.clicked.connect(
self._displayNodeNumbersClicked)
self._ui.displaySurfaces_checkBox.clicked.connect(
self._displaySurfacesClicked)
self._ui.displaySurfacesExterior_checkBox.clicked.connect(
self._displaySurfacesExteriorClicked)
self._ui.displaySurfacesTranslucent_checkBox.clicked.connect(
self._displaySurfacesTranslucentClicked)
self._ui.displaySurfacesWireframe_checkBox.clicked.connect(
self._displaySurfacesWireframeClicked)
self._ui.displayXiAxes_checkBox.clicked.connect(
self._displayXiAxesClicked)
# self._ui.treeWidgetAnnotation.itemSelectionChanged.connect(self._annotationSelectionChanged)
# self._ui.treeWidgetAnnotation.itemChanged.connect(self._annotationItemChanged)
def _refreshMeshTypeNames(self):
meshTypeNames = self._generator_model.getAllMeshTypeNames()
for meshTypeName in meshTypeNames:
self._ui.meshType_comboBox.addItem(meshTypeName)
def _refreshParameterSetNames(self):
self._ui.parameterSet_comboBox.blockSignals(True)
self._ui.parameterSet_comboBox.clear()
currentIndex = 0
currentParameterSetName = self._generator_model.getCurrentParameterSetName(
)
index = 0
for parameterSetName in self._generator_model.getMeshTypeParameterSetNames(
):
self._ui.parameterSet_comboBox.addItem(parameterSetName)
if parameterSetName == currentParameterSetName:
currentIndex = index
index += 1
self._ui.parameterSet_comboBox.setCurrentIndex(currentIndex)
self._ui.parameterSet_comboBox.blockSignals(False)
def _updateForCustomParameterSet(self):
'''
Call to update parameter set names in combobox if doesn't match number in model due to custom parameter set.
'''
if (self._generator_model.getCurrentParameterSetName() != self._ui.parameterSet_comboBox.currentText()) or \
(len(self._generator_model.getMeshTypeParameterSetNames()) != self._ui.parameterSet_comboBox.count()):
self._refreshParameterSetNames()
def _createFMAItem(self, parent, text, fma_id):
item = QtGui.QTreeWidgetItem(parent)
item.setText(0, text)
item.setData(0, QtCore.Qt.UserRole + 1, fma_id)
item.setCheckState(0, QtCore.Qt.Unchecked)
item.setFlags(QtCore.Qt.ItemIsUserCheckable | QtCore.Qt.ItemIsEnabled
| QtCore.Qt.ItemIsTristate)
return item
def _populateAnnotationTree(self):
tree = self._ui.treeWidgetAnnotation
tree.clear()
rsh_item = self._createFMAItem(tree, 'right side of heart', 'FMA_7165')
self._createFMAItem(rsh_item, 'ventricle', 'FMA_7098')
self._createFMAItem(rsh_item, 'atrium', 'FMA_7096')
self._createFMAItem(rsh_item, 'auricle', 'FMA_7218')
lsh_item = self._createFMAItem(tree, 'left side of heart', 'FMA_7166')
self._createFMAItem(lsh_item, 'ventricle', 'FMA_7101')
self._createFMAItem(lsh_item, 'atrium', 'FMA_7097')
self._createFMAItem(lsh_item, 'auricle', 'FMA_7219')
apex_item = self._createFMAItem(tree, 'apex of heart', 'FMA_7164')
vortex_item = self._createFMAItem(tree, 'vortex of heart', 'FMA_84628')
self._ui.treeWidgetAnnotation.addTopLevelItem(rsh_item)
self._ui.treeWidgetAnnotation.addTopLevelItem(lsh_item)
self._ui.treeWidgetAnnotation.addTopLevelItem(apex_item)
self._ui.treeWidgetAnnotation.addTopLevelItem(vortex_item)
def getModel(self):
return self._model
def registerDoneExecution(self, doneCallback):
self._doneCallback = doneCallback
def _updateUi(self):
pass
def _doneButtonClicked(self):
self._ui.dockWidget.setFloating(False)
self._model.done()
self._model = None
self._doneCallback()
def _meshTypeChanged(self, index):
meshTypeName = self._ui.meshType_comboBox.itemText(index)
self._generator_model.setMeshTypeByName(meshTypeName)
self._annotation_model.setMeshTypeByName(meshTypeName)
self._refreshParameterSetNames()
self._refreshMeshTypeOptions()
def _parameterSetChanged(self, index):
parameterSetName = self._ui.parameterSet_comboBox.itemText(index)
self._generator_model.setParameterSetName(parameterSetName)
self._refreshMeshTypeOptions()
def _meshTypeOptionCheckBoxClicked(self, checkBox):
dependentChanges = self._generator_model.setMeshTypeOption(
checkBox.objectName(), checkBox.isChecked())
if dependentChanges:
self._refreshMeshTypeOptions()
self._updateForCustomParameterSet()
def _meshTypeOptionLineEditChanged(self, lineEdit):
dependentChanges = self._generator_model.setMeshTypeOption(
lineEdit.objectName(), lineEdit.text())
if dependentChanges:
self._refreshMeshTypeOptions()
else:
finalValue = self._generator_model.getMeshTypeOption(
lineEdit.objectName())
lineEdit.setText(str(finalValue))
self._updateForCustomParameterSet()
def _refreshMeshTypeOptions(self):
layout = self._ui.meshTypeOptions_frame.layout()
# remove all current mesh type widgets
while layout.count():
child = layout.takeAt(0)
if child.widget():
child.widget().deleteLater()
optionNames = self._generator_model.getMeshTypeOrderedOptionNames()
for key in optionNames:
value = self._generator_model.getMeshTypeOption(key)
# print('key ', key, ' value ', value)
if type(value) is bool:
checkBox = QtGui.QCheckBox(self._ui.meshTypeOptions_frame)
checkBox.setObjectName(key)
checkBox.setText(key)
checkBox.setChecked(value)
callback = partial(self._meshTypeOptionCheckBoxClicked,
checkBox)
checkBox.clicked.connect(callback)
layout.addWidget(checkBox)
else:
label = QtGui.QLabel(self._ui.meshTypeOptions_frame)
label.setObjectName(key)
label.setText(key)
layout.addWidget(label)
lineEdit = QtGui.QLineEdit(self._ui.meshTypeOptions_frame)
lineEdit.setObjectName(key)
lineEdit.setText(str(value))
callback = partial(self._meshTypeOptionLineEditChanged,
lineEdit)
#lineEdit.returnPressed.connect(callback)
lineEdit.editingFinished.connect(callback)
layout.addWidget(lineEdit)
def _refreshOptions(self):
self._ui.identifier_label.setText('Identifier: ' +
self._model.getIdentifier())
self._ui.deleteElementsRanges_lineEdit.setText(
self._generator_model.getDeleteElementsRangesText())
self._ui.scale_lineEdit.setText(self._generator_model.getScaleText())
self._ui.displayAnnotationPoints_checkBox.setChecked(
self._generator_model.isDisplayAnnotationPoints())
self._ui.displayAxes_checkBox.setChecked(
self._generator_model.isDisplayAxes())
self._ui.displayElementNumbers_checkBox.setChecked(
self._generator_model.isDisplayElementNumbers())
self._ui.displayLines_checkBox.setChecked(
self._generator_model.isDisplayLines())
self._ui.displayLinesExterior_checkBox.setChecked(
self._generator_model.isDisplayLinesExterior())
self._ui.displayNodeDerivatives_checkBox.setChecked(
self._generator_model.isDisplayNodeDerivatives())
self._ui.displayNodeNumbers_checkBox.setChecked(
self._generator_model.isDisplayNodeNumbers())
self._ui.displaySurfaces_checkBox.setChecked(
self._generator_model.isDisplaySurfaces())
self._ui.displaySurfacesExterior_checkBox.setChecked(
self._generator_model.isDisplaySurfacesExterior())
self._ui.displaySurfacesTranslucent_checkBox.setChecked(
self._generator_model.isDisplaySurfacesTranslucent())
self._ui.displaySurfacesWireframe_checkBox.setChecked(
self._generator_model.isDisplaySurfacesWireframe())
self._ui.displayXiAxes_checkBox.setChecked(
self._generator_model.isDisplayXiAxes())
index = self._ui.meshType_comboBox.findText(
self._generator_model.getMeshTypeName())
self._ui.meshType_comboBox.blockSignals(True)
self._ui.meshType_comboBox.setCurrentIndex(index)
self._ui.meshType_comboBox.blockSignals(False)
self._refreshParameterSetNames()
self._refreshMeshTypeOptions()
def _deleteElementRangesLineEditChanged(self):
self._generator_model.setDeleteElementsRangesText(
self._ui.deleteElementsRanges_lineEdit.text())
self._ui.deleteElementsRanges_lineEdit.setText(
self._generator_model.getDeleteElementsRangesText())
def _scaleLineEditChanged(self):
self._generator_model.setScaleText(self._ui.scale_lineEdit.text())
self._ui.scale_lineEdit.setText(self._generator_model.getScaleText())
def _displayAnnotationPointsClicked(self):
self._generator_model.setDisplayAnnotationPoints(
self._ui.displayAnnotationPoints_checkBox.isChecked())
def _displayAxesClicked(self):
self._generator_model.setDisplayAxes(
self._ui.displayAxes_checkBox.isChecked())
def _displayElementNumbersClicked(self):
self._generator_model.setDisplayElementNumbers(
self._ui.displayElementNumbers_checkBox.isChecked())
def _displayLinesClicked(self):
self._generator_model.setDisplayLines(
self._ui.displayLines_checkBox.isChecked())
self._autoPerturbLines()
def _displayLinesExteriorClicked(self):
self._generator_model.setDisplayLinesExterior(
self._ui.displayLinesExterior_checkBox.isChecked())
def _displayNodeDerivativesClicked(self):
self._generator_model.setDisplayNodeDerivatives(
self._ui.displayNodeDerivatives_checkBox.isChecked())
def _displayNodeNumbersClicked(self):
self._generator_model.setDisplayNodeNumbers(
self._ui.displayNodeNumbers_checkBox.isChecked())
def _displaySurfacesClicked(self):
self._generator_model.setDisplaySurfaces(
self._ui.displaySurfaces_checkBox.isChecked())
self._autoPerturbLines()
def _displaySurfacesExteriorClicked(self):
self._generator_model.setDisplaySurfacesExterior(
self._ui.displaySurfacesExterior_checkBox.isChecked())
def _displaySurfacesTranslucentClicked(self):
self._generator_model.setDisplaySurfacesTranslucent(
self._ui.displaySurfacesTranslucent_checkBox.isChecked())
self._autoPerturbLines()
def _displaySurfacesWireframeClicked(self):
self._generator_model.setDisplaySurfacesWireframe(
self._ui.displaySurfacesWireframe_checkBox.isChecked())
def _displayXiAxesClicked(self):
self._generator_model.setDisplayXiAxes(
self._ui.displayXiAxes_checkBox.isChecked())
def _annotationItemChanged(self, item):
print(item.text(0))
print(item.data(0, QtCore.Qt.UserRole + 1))
def _viewAll(self):
"""
Ask sceneviewer to show all of scene.
"""
if self._ui.sceneviewer_widget.getSceneviewer() is not None:
self._ui.sceneviewer_widget.viewAll()
class MeshGeneratorWidget(QtGui.QWidget):
def __init__(self, model, parent=None):
super(MeshGeneratorWidget, self).__init__(parent)
self._ui = Ui_MeshGeneratorWidget()
self._ui.setupUi(self)
self._model = model
self._model.registerTimeValueUpdateCallback(self._updateTimeValue)
self._model.registerFrameIndexUpdateCallback(self._updateFrameIndex)
self._generator_model = model.getGeneratorModel()
self._plane_model = model.getPlaneModel()
self._fiducial_marker_model = model.getFiducialMarkerModel()
self._ui.sceneviewer_widget.setContext(model.getContext())
self._ui.sceneviewer_widget.setModel(self._plane_model)
self._model.registerSceneChangeCallback(self._sceneChanged)
self._doneCallback = None
self._populateFiducialMarkersComboBox()
self._marker_mode_active = False
self._have_images = False
# self._populateAnnotationTree()
meshTypeNames = self._generator_model.getAllMeshTypeNames()
for meshTypeName in meshTypeNames:
self._ui.meshType_comboBox.addItem(meshTypeName)
self._makeConnections()
def _graphicsInitialized(self):
"""
Callback for when SceneviewerWidget is initialised
Set custom scene from model
"""
sceneviewer = self._ui.sceneviewer_widget.getSceneviewer()
if sceneviewer is not None:
self._model.loadSettings()
self._refreshOptions()
scene = self._model.getScene()
self._ui.sceneviewer_widget.setScene(scene)
# self._ui.sceneviewer_widget.setSelectModeAll()
sceneviewer.setLookatParametersNonSkew([2.0, -2.0, 1.0],
[0.0, 0.0, 0.0],
[0.0, 0.0, 1.0])
sceneviewer.setTransparencyMode(sceneviewer.TRANSPARENCY_MODE_SLOW)
self._autoPerturbLines()
self._viewAll()
def _sceneChanged(self):
sceneviewer = self._ui.sceneviewer_widget.getSceneviewer()
if sceneviewer is not None:
if self._have_images:
self._plane_model.setSceneviewer(sceneviewer)
scene = self._model.getScene()
self._ui.sceneviewer_widget.setScene(scene)
self._autoPerturbLines()
def _autoPerturbLines(self):
"""
Enable scene viewer perturb lines iff solid surfaces are drawn with lines.
Call whenever lines, surfaces or translucency changes
"""
sceneviewer = self._ui.sceneviewer_widget.getSceneviewer()
if sceneviewer is not None:
sceneviewer.setPerturbLinesFlag(
self._generator_model.needPerturbLines())
def _makeConnections(self):
self._ui.sceneviewer_widget.graphicsInitialized.connect(
self._graphicsInitialized)
self._ui.done_button.clicked.connect(self._doneButtonClicked)
self._ui.viewAll_button.clicked.connect(self._viewAll)
self._ui.meshType_comboBox.currentIndexChanged.connect(
self._meshTypeChanged)
self._ui.deleteElementsRanges_lineEdit.returnPressed.connect(
self._deleteElementRangesLineEditChanged)
self._ui.deleteElementsRanges_lineEdit.editingFinished.connect(
self._deleteElementRangesLineEditChanged)
self._ui.scale_lineEdit.returnPressed.connect(
self._scaleLineEditChanged)
self._ui.scale_lineEdit.editingFinished.connect(
self._scaleLineEditChanged)
self._ui.displayAxes_checkBox.clicked.connect(self._displayAxesClicked)
self._ui.displayElementNumbers_checkBox.clicked.connect(
self._displayElementNumbersClicked)
self._ui.displayLines_checkBox.clicked.connect(
self._displayLinesClicked)
self._ui.displayNodeDerivatives_checkBox.clicked.connect(
self._displayNodeDerivativesClicked)
self._ui.displayNodeNumbers_checkBox.clicked.connect(
self._displayNodeNumbersClicked)
self._ui.displaySurfaces_checkBox.clicked.connect(
self._displaySurfacesClicked)
self._ui.displaySurfacesExterior_checkBox.clicked.connect(
self._displaySurfacesExteriorClicked)
self._ui.displaySurfacesTranslucent_checkBox.clicked.connect(
self._displaySurfacesTranslucentClicked)
self._ui.displaySurfacesWireframe_checkBox.clicked.connect(
self._displaySurfacesWireframeClicked)
self._ui.displayXiAxes_checkBox.clicked.connect(
self._displayXiAxesClicked)
self._ui.activeModel_comboBox.currentIndexChanged.connect(
self._activeModelChanged)
self._ui.toImage_pushButton.clicked.connect(self._imageButtonClicked)
self._ui.displayImagePlane_checkBox.clicked.connect(
self._displayImagePlaneClicked)
self._ui.fixImagePlane_checkBox.clicked.connect(
self._fixImagePlaneClicked)
self._ui.timeValue_doubleSpinBox.valueChanged.connect(
self._timeValueChanged)
self._ui.timePlayStop_pushButton.clicked.connect(
self._timePlayStopClicked)
self._ui.frameIndex_spinBox.valueChanged.connect(
self._frameIndexValueChanged)
self._ui.framesPerSecond_spinBox.valueChanged.connect(
self._framesPerSecondValueChanged)
self._ui.timeLoop_checkBox.clicked.connect(self._timeLoopClicked)
self._ui.displayFiducialMarkers_checkBox.clicked.connect(
self._displayFiducialMarkersClicked)
self._ui.fiducialMarker_comboBox.currentIndexChanged.connect(
self._fiducialMarkerChanged)
# self._ui.treeWidgetAnnotation.itemSelectionChanged.connect(self._annotationSelectionChanged)
# self._ui.treeWidgetAnnotation.itemChanged.connect(self._annotationItemChanged)
def _fiducialMarkerChanged(self):
self._fiducial_marker_model.setActiveMarker(
self._ui.fiducialMarker_comboBox.currentText())
def _displayFiducialMarkersClicked(self):
self._fiducial_marker_model.setDisplayFiducialMarkers(
self._ui.displayFiducialMarkers_checkBox.isChecked())
def _populateFiducialMarkersComboBox(self):
self._ui.fiducialMarker_comboBox.addItems(FIDUCIAL_MARKER_LABELS)
def _createFMAItem(self, parent, text, fma_id):
item = QtGui.QTreeWidgetItem(parent)
item.setText(0, text)
item.setData(0, QtCore.Qt.UserRole + 1, fma_id)
item.setCheckState(0, QtCore.Qt.Unchecked)
item.setFlags(QtCore.Qt.ItemIsUserCheckable | QtCore.Qt.ItemIsEnabled
| QtCore.Qt.ItemIsTristate)
return item
def _populateAnnotationTree(self):
tree = self._ui.treeWidgetAnnotation
tree.clear()
rsh_item = self._createFMAItem(tree, 'right side of heart', 'FMA_7165')
self._createFMAItem(rsh_item, 'ventricle', 'FMA_7098')
self._createFMAItem(rsh_item, 'atrium', 'FMA_7096')
self._createFMAItem(rsh_item, 'auricle', 'FMA_7218')
lsh_item = self._createFMAItem(tree, 'left side of heart', 'FMA_7166')
self._createFMAItem(lsh_item, 'ventricle', 'FMA_7101')
self._createFMAItem(lsh_item, 'atrium', 'FMA_7097')
self._createFMAItem(lsh_item, 'auricle', 'FMA_7219')
apex_item = self._createFMAItem(tree, 'apex of heart', 'FMA_7164')
vortex_item = self._createFMAItem(tree, 'vortex of heart', 'FMA_84628')
self._ui.treeWidgetAnnotation.addTopLevelItem(rsh_item)
self._ui.treeWidgetAnnotation.addTopLevelItem(lsh_item)
self._ui.treeWidgetAnnotation.addTopLevelItem(apex_item)
self._ui.treeWidgetAnnotation.addTopLevelItem(vortex_item)
def getModel(self):
return self._model
def registerDoneExecution(self, doneCallback):
self._doneCallback = doneCallback
def _updateUi(self):
if self._have_images:
frame_count = self._plane_model.getFrameCount()
self._ui.numFramesValue_label.setText("{0}".format(frame_count))
self._ui.frameIndex_spinBox.setMaximum(frame_count)
self._ui.timeValue_doubleSpinBox.setMaximum(
frame_count / self._model.getFramesPerSecond())
else:
self._generator_model.disableAlignment()
self._plane_model.disableAlignment()
self._ui.alignment_groupBox.setVisible(False)
self._ui.fiducialMarkers_groupBox.setVisible(False)
self._ui.video_groupBox.setVisible(False)
self._ui.displayImagePlane_checkBox.setVisible(False)
self._ui.displayFiducialMarkers_checkBox.setVisible(False)
def setImageInfo(self, image_info):
self._plane_model.setImageInfo(image_info)
self._have_images = image_info is not None
self._updateUi()
def _doneButtonClicked(self):
self._ui.dockWidget.setFloating(False)
self._model.done()
self._model = None
self._doneCallback()
def _imageButtonClicked(self):
sceneviewer = self._ui.sceneviewer_widget.getSceneviewer()
normal, up, offset = self._plane_model.getPlaneInfo()
_, current_lookat_pos = sceneviewer.getLookatPosition()
_, current_eye_pos = sceneviewer.getEyePosition()
view_distance = vectorops.magnitude(
vectorops.sub(current_eye_pos, current_lookat_pos))
eye_pos = vectorops.add(vectorops.mult(normal, view_distance), offset)
lookat_pos = offset
sceneviewer.setLookatParametersNonSkew(eye_pos, lookat_pos, up)
def _updateTimeValue(self, value):
self._ui.timeValue_doubleSpinBox.blockSignals(True)
frame_count = self._plane_model.getFrameCount()
max_time_value = frame_count / self._ui.framesPerSecond_spinBox.value()
if value > max_time_value:
self._ui.timeValue_doubleSpinBox.setValue(max_time_value)
self._timePlayStopClicked()
else:
self._ui.timeValue_doubleSpinBox.setValue(value)
self._ui.timeValue_doubleSpinBox.blockSignals(False)
def _updateFrameIndex(self, value):
self._ui.frameIndex_spinBox.blockSignals(True)
self._ui.frameIndex_spinBox.setValue(value)
self._ui.frameIndex_spinBox.blockSignals(False)
def _timeValueChanged(self, value):
self._model.setTimeValue(value)
def _timeDurationChanged(self, value):
self._model.setTimeDuration(value)
def _timePlayStopClicked(self):
play_text = 'Play'
stop_text = 'Stop'
current_text = self._ui.timePlayStop_pushButton.text()
if current_text == play_text:
self._ui.timePlayStop_pushButton.setText(stop_text)
self._model.play()
else:
self._ui.timePlayStop_pushButton.setText(play_text)
self._model.stop()
def _timeLoopClicked(self):
self._model.setTimeLoop(self._ui.timeLoop_checkBox.isChecked())
def _frameIndexValueChanged(self, value):
self._model.setFrameIndex(value)
def _framesPerSecondValueChanged(self, value):
self._model.setFramesPerSecond(value)
self._ui.timeValue_doubleSpinBox.setMaximum(
self._plane_model.getFrameCount() / value)
def _fixImagePlaneClicked(self):
self._plane_model.setImagePlaneFixed(
self._ui.fixImagePlane_checkBox.isChecked())
def _displayImagePlaneClicked(self):
self._plane_model.setImagePlaneVisible(
self._ui.displayImagePlane_checkBox.isChecked())
def _activeModelChanged(self, index):
if index == 0:
self._ui.sceneviewer_widget.setModel(self._plane_model)
else:
self._ui.sceneviewer_widget.setModel(self._generator_model)
def _meshTypeChanged(self, index):
meshTypeName = self._ui.meshType_comboBox.itemText(index)
self._generator_model.setMeshTypeByName(meshTypeName)
self._refreshMeshTypeOptions()
def _meshTypeOptionCheckBoxClicked(self, checkBox):
self._generator_model.setMeshTypeOption(checkBox.objectName(),
checkBox.isChecked())
def _meshTypeOptionLineEditChanged(self, lineEdit):
self._generator_model.setMeshTypeOption(lineEdit.objectName(),
lineEdit.text())
finalValue = self._generator_model.getMeshTypeOption(
lineEdit.objectName())
lineEdit.setText(str(finalValue))
def _refreshMeshTypeOptions(self):
layout = self._ui.meshTypeOptions_frame.layout()
# remove all current mesh type widgets
while layout.count():
child = layout.takeAt(0)
if child.widget():
child.widget().deleteLater()
optionNames = self._generator_model.getMeshTypeOrderedOptionNames()
for key in optionNames:
value = self._generator_model.getMeshTypeOption(key)
# print('key ', key, ' value ', value)
if type(value) is bool:
checkBox = QtGui.QCheckBox(self._ui.meshTypeOptions_frame)
checkBox.setObjectName(key)
checkBox.setText(key)
checkBox.setChecked(value)
callback = partial(self._meshTypeOptionCheckBoxClicked,
checkBox)
checkBox.clicked.connect(callback)
layout.addWidget(checkBox)
else:
label = QtGui.QLabel(self._ui.meshTypeOptions_frame)
label.setObjectName(key)
label.setText(key)
layout.addWidget(label)
lineEdit = QtGui.QLineEdit(self._ui.meshTypeOptions_frame)
lineEdit.setObjectName(key)
lineEdit.setText(str(value))
callback = partial(self._meshTypeOptionLineEditChanged,
lineEdit)
lineEdit.returnPressed.connect(callback)
lineEdit.editingFinished.connect(callback)
layout.addWidget(lineEdit)
def _refreshOptions(self):
self._ui.identifier_label.setText('Identifier: ' +
self._model.getIdentifier())
self._ui.deleteElementsRanges_lineEdit.setText(
self._generator_model.getDeleteElementsRangesText())
self._ui.scale_lineEdit.setText(self._generator_model.getScaleText())
self._ui.displayAxes_checkBox.setChecked(
self._generator_model.isDisplayAxes())
self._ui.displayElementNumbers_checkBox.setChecked(
self._generator_model.isDisplayElementNumbers())
self._ui.displayLines_checkBox.setChecked(
self._generator_model.isDisplayLines())
self._ui.displayNodeDerivatives_checkBox.setChecked(
self._generator_model.isDisplayNodeDerivatives())
self._ui.displayNodeNumbers_checkBox.setChecked(
self._generator_model.isDisplayNodeNumbers())
self._ui.displaySurfaces_checkBox.setChecked(
self._generator_model.isDisplaySurfaces())
self._ui.displaySurfacesExterior_checkBox.setChecked(
self._generator_model.isDisplaySurfacesExterior())
self._ui.displaySurfacesTranslucent_checkBox.setChecked(
self._generator_model.isDisplaySurfacesTranslucent())
self._ui.displaySurfacesWireframe_checkBox.setChecked(
self._generator_model.isDisplaySurfacesWireframe())
self._ui.displayXiAxes_checkBox.setChecked(
self._generator_model.isDisplayXiAxes())
self._ui.displayImagePlane_checkBox.setChecked(
self._plane_model.isDisplayImagePlane())
self._ui.displayFiducialMarkers_checkBox.setChecked(
self._fiducial_marker_model.isDisplayFiducialMarkers())
self._ui.fixImagePlane_checkBox.setChecked(
self._plane_model.isImagePlaneFixed())
self._ui.framesPerSecond_spinBox.setValue(
self._model.getFramesPerSecond())
self._ui.timeLoop_checkBox.setChecked(self._model.isTimeLoop())
index = self._ui.meshType_comboBox.findText(
self._generator_model.getMeshTypeName())
self._ui.meshType_comboBox.blockSignals(True)
self._ui.meshType_comboBox.setCurrentIndex(index)
self._ui.meshType_comboBox.blockSignals(False)
index = self._ui.fiducialMarker_comboBox.findText(
self._fiducial_marker_model.getActiveMarker())
self._ui.fiducialMarker_comboBox.blockSignals(True)
self._ui.fiducialMarker_comboBox.setCurrentIndex(0 if index ==
-1 else index)
self._ui.fiducialMarker_comboBox.blockSignals(False)
self._refreshMeshTypeOptions()
def _deleteElementRangesLineEditChanged(self):
self._generator_model.setDeleteElementsRangesText(
self._ui.deleteElementsRanges_lineEdit.text())
self._ui.deleteElementsRanges_lineEdit.setText(
self._generator_model.getDeleteElementsRangesText())
def _scaleLineEditChanged(self):
self._generator_model.setScaleText(self._ui.scale_lineEdit.text())
self._ui.scale_lineEdit.setText(self._generator_model.getScaleText())
def _displayAxesClicked(self):
self._generator_model.setDisplayAxes(
self._ui.displayAxes_checkBox.isChecked())
def _displayElementNumbersClicked(self):
self._generator_model.setDisplayElementNumbers(
self._ui.displayElementNumbers_checkBox.isChecked())
def _displayLinesClicked(self):
self._generator_model.setDisplayLines(
self._ui.displayLines_checkBox.isChecked())
self._autoPerturbLines()
def _displayNodeDerivativesClicked(self):
self._generator_model.setDisplayNodeDerivatives(
self._ui.displayNodeDerivatives_checkBox.isChecked())
def _displayNodeNumbersClicked(self):
self._generator_model.setDisplayNodeNumbers(
self._ui.displayNodeNumbers_checkBox.isChecked())
def _displaySurfacesClicked(self):
self._generator_model.setDisplaySurfaces(
self._ui.displaySurfaces_checkBox.isChecked())
self._autoPerturbLines()
def _displaySurfacesExteriorClicked(self):
self._generator_model.setDisplaySurfacesExterior(
self._ui.displaySurfacesExterior_checkBox.isChecked())
def _displaySurfacesTranslucentClicked(self):
self._generator_model.setDisplaySurfacesTranslucent(
self._ui.displaySurfacesTranslucent_checkBox.isChecked())
self._autoPerturbLines()
def _displaySurfacesWireframeClicked(self):
self._generator_model.setDisplaySurfacesWireframe(
self._ui.displaySurfacesWireframe_checkBox.isChecked())
def _displayXiAxesClicked(self):
self._generator_model.setDisplayXiAxes(
self._ui.displayXiAxes_checkBox.isChecked())
def _annotationItemChanged(self, item):
print(item.text(0))
print(item.data(0, QtCore.Qt.UserRole + 1))
def _viewAll(self):
"""
Ask sceneviewer to show all of scene.
"""
if self._ui.sceneviewer_widget.getSceneviewer() is not None:
self._ui.sceneviewer_widget.viewAll()
def keyPressEvent(self, event):
if event.modifiers(
) & QtCore.Qt.CTRL and QtGui.QApplication.mouseButtons(
) == QtCore.Qt.NoButton:
self._marker_mode_active = True
self._ui.sceneviewer_widget._model = self._fiducial_marker_model
self._original_mousePressEvent = self._ui.sceneviewer_widget.mousePressEvent
self._ui.sceneviewer_widget._calculatePointOnPlane = types.MethodType(
_calculatePointOnPlane, self._ui.sceneviewer_widget)
self._ui.sceneviewer_widget.mousePressEvent = types.MethodType(
mousePressEvent, self._ui.sceneviewer_widget)
self._model.printLog()
def keyReleaseEvent(self, event):
if self._marker_mode_active:
self._marker_mode_active = False
self._ui.sceneviewer_widget._model = self._plane_model
self._ui.sceneviewer_widget._calculatePointOnPlane = None
self._ui.sceneviewer_widget.mousePressEvent = self._original_mousePressEvent