class OutputWidgetBase(QtWidgets.QWidget):
"""
Plugin responsible for triggering the creation of png/pdf/py files and live script window.
"""
write = QtCore.pyqtSignal(str)
def __init__(self):
super(OutputWidgetBase, self).__init__()
self._icon_size = QtCore.QSize(32, 32)
self.MainLayout = QtWidgets.QHBoxLayout()
self.MainLayout.setContentsMargins(0, 0, 0, 0)
self.MainLayout.setSpacing(5)
self.PythonButton = QtWidgets.QPushButton()
self.PDFButton = QtWidgets.QPushButton()
self.PNGButton = QtWidgets.QPushButton()
self.LiveScriptButton = QtWidgets.QPushButton()
self.LiveScript = TextSubWindow()
self.setLayout(self.MainLayout)
self.setSizePolicy(QtWidgets.QSizePolicy.Fixed, QtWidgets.QSizePolicy.Fixed)
self.MainLayout.addWidget(self.PythonButton)
self.MainLayout.addWidget(self.PDFButton)
self.MainLayout.addWidget(self.PNGButton)
self.MainLayout.addWidget(self.LiveScriptButton)
def updateLiveScriptText(self):
"""
Updates the live script view.
"""
if self.LiveScript.isVisible() and hasattr(self, "_plugin_manager"):
# don't reset the text if it is the same. This allows for easier select/copy
s = self._plugin_manager.repr()
if s != self.LiveScript.toPlainText():
self.LiveScript.setText(s)
def _setupPythonButton(self, qobject):
"""
Setup method for python script output button.
"""
qobject.clicked.connect(self._callbackPythonButton)
qobject.setIcon(WidgetUtils.createIcon('py.svg'))
qobject.setIconSize(self._icon_size)
qobject.setFixedSize(qobject.iconSize())
qobject.setToolTip("Create python script to reproduce this figure.")
qobject.setStyleSheet("QPushButton {border:none}")
def _callbackPythonButton(self):
"""
Open dialog and write script.
"""
dialog = QtWidgets.QFileDialog()
dialog.setWindowTitle('Write Python Script')
dialog.setNameFilter('Python Files (*.py)')
dialog.setFileMode(QtWidgets.QFileDialog.AnyFile)
dialog.setAcceptMode(QtWidgets.QFileDialog.AcceptSave)
dialog.setOption(QtWidgets.QFileDialog.DontUseNativeDialog)
if dialog.exec_() == QtWidgets.QDialog.Accepted:
filename = str(dialog.selectedFiles()[0])
self.write.emit(filename)
def _setupPDFButton(self, qobject):
"""
Setup method pdf image output.
"""
qobject.clicked.connect(self._callbackPDFButton)
qobject.setIcon(WidgetUtils.createIcon('pdf.svg'))
qobject.setIconSize(self._icon_size)
qobject.setFixedSize(qobject.iconSize())
qobject.setToolTip("Create a pdf file of the current figure.")
qobject.setStyleSheet("QPushButton {border:none}")
def _callbackPDFButton(self):
"""
Write a PDF file of figure.
"""
dialog = QtWidgets.QFileDialog()
dialog.setWindowTitle('Write *.pdf of figure')
dialog.setNameFilter('PDF files (*.pdf)')
dialog.setFileMode(QtWidgets.QFileDialog.AnyFile)
dialog.setAcceptMode(QtWidgets.QFileDialog.AcceptSave)
dialog.setOption(QtWidgets.QFileDialog.DontUseNativeDialog)
if dialog.exec_() == QtWidgets.QDialog.Accepted:
filename = str(dialog.selectedFiles()[0])
self.write.emit(filename)
def _setupPNGButton(self, qobject):
"""
Setup method png image output.
"""
qobject.clicked.connect(self._callbackPNGButton)
qobject.setIcon(WidgetUtils.createIcon('png.svg'))
qobject.setIconSize(self._icon_size)
qobject.setFixedSize(qobject.iconSize())
qobject.setToolTip("Create a png file of the current figure.")
qobject.setStyleSheet("QPushButton {border:none}")
def _callbackPNGButton(self):
"""
Write a png file of figure.
"""
dialog = QtWidgets.QFileDialog()
dialog.setWindowTitle('Write *.png of figure')
dialog.setNameFilter('PNG files (*.png)')
dialog.setFileMode(QtWidgets.QFileDialog.AnyFile)
dialog.setAcceptMode(QtWidgets.QFileDialog.AcceptSave)
dialog.setOption(QtWidgets.QFileDialog.DontUseNativeDialog)
dialog.setDefaultSuffix("png")
if dialog.exec_() == QtWidgets.QDialog.Accepted:
filename = str(dialog.selectedFiles()[0])
self.write.emit(filename)
def _setupLiveScriptButton(self, qobject):
"""
Setup method png image output.
"""
qobject.clicked.connect(self._callbackLiveScriptButton)
qobject.setIcon(WidgetUtils.createIcon('script.svg'))
qobject.setIconSize(self._icon_size)
qobject.setFixedSize(qobject.iconSize())
qobject.setToolTip("Show the current python script.")
qobject.setStyleSheet("QPushButton {border:none}")
def _callbackLiveScriptButton(self):
"""
Write a png file of figure.
"""
self.LiveScript.show()
self.updateLiveScriptText()
def _setupLiveScript(self, qobject):
"""
Setup for the script text window.
"""
self.setSizePolicy(QtWidgets.QSizePolicy.Preferred, QtWidgets.QSizePolicy.Preferred)
qobject.setReadOnly(True)
class OutputPlugin(QtWidgets.QGroupBox, ExodusPlugin):
"""
Plugin responsible for triggering the creation of png/py files and live script viewing.
"""
#: pyqtSignal: Write the chigger script portions from this plugin
write = QtCore.pyqtSignal(str)
#: pyqtSignal: Create an observer for the given event with the supplied method callback function
addObserver = QtCore.pyqtSignal(type(vtk.vtkCommand.RenderEvent),
types.MethodType)
def __init__(self):
super(OutputPlugin, self).__init__()
self.MainLayout = QtWidgets.QHBoxLayout(self)
self.LiveScriptWindow = TextSubWindow()
self.hide()
self.LiveScript = None
self.ExportPNG = None
self.ExportPython = None
self.setup()
self._observer = None
def addToMenu(self, menu):
"""
Adds output options to the 'Results' menu.
"""
self.LiveScript = menu.addAction('Show live script')
self.LiveScript.setCheckable(True)
self.LiveScript.setChecked(False)
self.LiveScript.setEnabled(False)
self.LiveScript.toggled.connect(self._callbackLiveScript)
export = menu.addMenu('Export')
self.ExportPNG = export.addAction('PNG')
self.ExportPNG.triggered.connect(self._callbackExportPNG)
self.ExportPNG.setEnabled(False)
self.ExportPython = export.addAction('Python')
self.ExportPython.triggered.connect(self._callbackExportPython)
self.ExportPython.setEnabled(False)
def updateLiveScriptWindow(self, *args):
"""
Updates the chigger script live view.
"""
if self.LiveScriptWindow.isVisible() and hasattr(
self, "_plugin_manager"):
# don't reset the text if it is the same. This allows for easier select/copy
s = self._plugin_manager.repr()
if s != self.LiveScriptWindow.toPlainText():
self.LiveScriptWindow.setText(s)
def onWindowReset(self):
"""
Disable the live view when there are no results.
"""
self.LiveScript.setEnabled(False)
self.ExportPNG.setEnabled(False)
self.ExportPython.setEnabled(False)
def onWindowResult(self, *args):
"""
Adds an observer for the live window update to the VTK renderer.
"""
self.addObserver.emit(vtk.vtkCommand.RenderEvent,
self.updateLiveScriptWindow)
self.LiveScript.setEnabled(True)
self.ExportPNG.setEnabled(True)
self.ExportPython.setEnabled(True)
def _setupLiveScriptWindow(self, qobject):
qobject.setReadOnly(True)
qobject.hide()
qobject.windowClosed.connect(self._callbackCloseLiveScriptWindow)
def _callbackLiveScript(self, *args):
if self.LiveScript.isChecked() and hasattr(self, "_plugin_manager"):
self.LiveScriptWindow.show()
self.updateLiveScriptWindow()
else:
self.LiveScriptWindow.hide()
def _callbackExportPNG(self):
"""
Write a png file of figure.
"""
dialog = QtWidgets.QFileDialog()
dialog.setWindowTitle('Write *.png of figure')
dialog.setNameFilter('PNG files (*.png)')
dialog.setFileMode(QtWidgets.QFileDialog.AnyFile)
dialog.setAcceptMode(QtWidgets.QFileDialog.AcceptSave)
dialog.setOption(QtWidgets.QFileDialog.DontUseNativeDialog)
dialog.setDefaultSuffix("png")
if dialog.exec_() == QtWidgets.QDialog.Accepted:
filename = str(dialog.selectedFiles()[0])
self.write.emit(filename)
def _callbackExportPython(self):
"""
Open dialog and write script.
"""
dialog = QtWidgets.QFileDialog()
dialog.setWindowTitle('Write Python Script')
dialog.setNameFilter('Python Files (*.py)')
dialog.setFileMode(QtWidgets.QFileDialog.AnyFile)
dialog.setAcceptMode(QtWidgets.QFileDialog.AcceptSave)
dialog.setOption(QtWidgets.QFileDialog.DontUseNativeDialog)
if dialog.exec_() == QtWidgets.QDialog.Accepted:
filename = str(dialog.selectedFiles()[0])
self.write.emit(filename)
def _callbackCloseLiveScriptWindow(self, *args):
self.LiveScript.setChecked(False)
class OutputWidgetBase(QtWidgets.QWidget):
"""
Plugin responsible for triggering the creation of png/pdf/py files and live script window.
"""
write = QtCore.pyqtSignal(str)
def __init__(self):
super(OutputWidgetBase, self).__init__()
self._icon_size = QtCore.QSize(32, 32)
self.MainLayout = QtWidgets.QHBoxLayout()
self.MainLayout.setContentsMargins(0, 0, 0, 0)
self.MainLayout.setSpacing(5)
self.PythonButton = QtWidgets.QPushButton()
self.PDFButton = QtWidgets.QPushButton()
self.PNGButton = QtWidgets.QPushButton()
self.LiveScriptButton = QtWidgets.QPushButton()
self.LiveScript = TextSubWindow()
self.setLayout(self.MainLayout)
self.setSizePolicy(QtWidgets.QSizePolicy.Fixed,
QtWidgets.QSizePolicy.Fixed)
self.MainLayout.addWidget(self.PythonButton)
self.MainLayout.addWidget(self.PDFButton)
self.MainLayout.addWidget(self.PNGButton)
self.MainLayout.addWidget(self.LiveScriptButton)
def updateLiveScriptText(self):
"""
Updates the live script view.
"""
if self.LiveScript.isVisible() and hasattr(self, "_plugin_manager"):
s = self._plugin_manager.repr()
self.LiveScript.setText(s)
def _setupPythonButton(self, qobject):
"""
Setup method for python script output button.
"""
qobject.clicked.connect(self._callbackPythonButton)
qobject.setIcon(WidgetUtils.createIcon('py.svg'))
qobject.setIconSize(self._icon_size)
qobject.setFixedSize(qobject.iconSize())
qobject.setToolTip("Create python script to reproduce this figure.")
qobject.setStyleSheet("QPushButton {border:none}")
def _callbackPythonButton(self):
"""
Open dialog and write script.
"""
dialog = QtWidgets.QFileDialog()
dialog.setWindowTitle('Write Python Script')
dialog.setNameFilter('Python Files (*.py)')
dialog.setFileMode(QtWidgets.QFileDialog.AnyFile)
dialog.setAcceptMode(QtWidgets.QFileDialog.AcceptSave)
dialog.setOption(QtWidgets.QFileDialog.DontUseNativeDialog)
if dialog.exec_() == QtWidgets.QDialog.Accepted:
filename = str(dialog.selectedFiles()[0])
self.write.emit(filename)
def _setupPDFButton(self, qobject):
"""
Setup method pdf image output.
"""
qobject.clicked.connect(self._callbackPDFButton)
qobject.setIcon(WidgetUtils.createIcon('pdf.svg'))
qobject.setIconSize(self._icon_size)
qobject.setFixedSize(qobject.iconSize())
qobject.setToolTip("Create a pdf file of the current figure.")
qobject.setStyleSheet("QPushButton {border:none}")
def _callbackPDFButton(self):
"""
Write a PDF file of figure.
"""
dialog = QtWidgets.QFileDialog()
dialog.setWindowTitle('Write *.pdf of figure')
dialog.setNameFilter('PDF files (*.pdf)')
dialog.setFileMode(QtWidgets.QFileDialog.AnyFile)
dialog.setAcceptMode(QtWidgets.QFileDialog.AcceptSave)
dialog.setOption(QtWidgets.QFileDialog.DontUseNativeDialog)
if dialog.exec_() == QtWidgets.QDialog.Accepted:
filename = str(dialog.selectedFiles()[0])
self.write.emit(filename)
def _setupPNGButton(self, qobject):
"""
Setup method png image output.
"""
qobject.clicked.connect(self._callbackPNGButton)
qobject.setIcon(WidgetUtils.createIcon('png.svg'))
qobject.setIconSize(self._icon_size)
qobject.setFixedSize(qobject.iconSize())
qobject.setToolTip("Create a png file of the current figure.")
qobject.setStyleSheet("QPushButton {border:none}")
def _callbackPNGButton(self):
"""
Write a png file of figure.
"""
dialog = QtWidgets.QFileDialog()
dialog.setWindowTitle('Write *.png of figure')
dialog.setNameFilter('PNG files (*.png)')
dialog.setFileMode(QtWidgets.QFileDialog.AnyFile)
dialog.setAcceptMode(QtWidgets.QFileDialog.AcceptSave)
dialog.setOption(QtWidgets.QFileDialog.DontUseNativeDialog)
dialog.setDefaultSuffix("png")
if dialog.exec_() == QtWidgets.QDialog.Accepted:
filename = str(dialog.selectedFiles()[0])
self.write.emit(filename)
def _setupLiveScriptButton(self, qobject):
"""
Setup method png image output.
"""
qobject.clicked.connect(self._callbackLiveScriptButton)
qobject.setIcon(WidgetUtils.createIcon('script.svg'))
qobject.setIconSize(self._icon_size)
qobject.setFixedSize(qobject.iconSize())
qobject.setToolTip("Show the current python script.")
qobject.setStyleSheet("QPushButton {border:none}")
def _callbackLiveScriptButton(self):
"""
Write a png file of figure.
"""
self.LiveScript.show()
self.updateLiveScriptText()
def _setupLiveScript(self, qobject):
"""
Setup for the script text window.
"""
self.setSizePolicy(QtWidgets.QSizePolicy.Preferred,
QtWidgets.QSizePolicy.Preferred)
qobject.setReadOnly(True)