mirror of
https://github.com/Athemis/PyDSF.git
synced 2025-04-05 06:36:04 +00:00
78 lines
2.8 KiB
Python
78 lines
2.8 KiB
Python
from PyQt5 import QtWidgets
|
|
from PyQt5.QtCore import QCoreApplication
|
|
from matplotlib.backends.backend_qt5agg import (FigureCanvasQTAgg as
|
|
FigureCanvas)
|
|
from matplotlib.backends.backend_qt5agg import (NavigationToolbar2QT as
|
|
NavigationToolbar)
|
|
from matplotlib.figure import Figure
|
|
_translate = QCoreApplication.translate
|
|
|
|
|
|
class MplCanvas(FigureCanvas):
|
|
|
|
def __init__(self, parent=None, width=4, height=5, dpi=100):
|
|
self.fig = Figure(figsize=(width, height), dpi=dpi)
|
|
self.ax = self.fig.add_subplot(111)
|
|
FigureCanvas.__init__(self, self.fig)
|
|
self.setParent(parent)
|
|
FigureCanvas.setSizePolicy(self, QtWidgets.QSizePolicy.Expanding,
|
|
QtWidgets.QSizePolicy.Expanding)
|
|
FigureCanvas.updateGeometry(self)
|
|
|
|
def clear(self):
|
|
self.ax.clear()
|
|
self.fig.clear()
|
|
|
|
def save(self, filename):
|
|
try:
|
|
self.fig.savefig(filename, dpi=300)
|
|
except IOError:
|
|
QtWidgets.QMessageBox.critical(
|
|
self, _translate("MainWindow", "Error"),
|
|
_translate("MainWindow", "Error saving figure! Please check "
|
|
"permissions/free space of target path!"),
|
|
QtWidgets.QMessageBox.Close, QtWidgets.QMessageBox.Close)
|
|
|
|
|
|
class MplCanvasNoMouse(MplCanvas):
|
|
|
|
# override mouseMoveEvent with non-functional dummy
|
|
# this will prevent the gui thread to hang while moving the mouse
|
|
# while a large number of plots is shown simultaniously
|
|
|
|
def mouseMoveEvent(self, event):
|
|
pass
|
|
|
|
|
|
class CustomNavigationToolbar(NavigationToolbar):
|
|
|
|
toolitems = (
|
|
(_translate("CustomNavigationToolbar", "Save"),
|
|
_translate("CustomNavigationToolbar",
|
|
"Save the figure"), "filesave",
|
|
"save_figure"),
|
|
(_translate("CustomNavigationToolbar", "Subplots"),
|
|
_translate("CustomNavigationToolbar",
|
|
"Configure subplots"), "subplots",
|
|
"configure_subplots"),
|
|
(None, None, None, None), )
|
|
|
|
def __init__(self, canvas, parent, coordinates=True):
|
|
NavigationToolbar.__init__(self, canvas, parent,
|
|
coordinates=coordinates)
|
|
|
|
|
|
class MplWidget(QtWidgets.QGraphicsView):
|
|
|
|
def __init__(self, parent=None, mouse_event=False):
|
|
QtWidgets.QGraphicsView.__init__(self, parent)
|
|
if mouse_event:
|
|
self.canvas = MplCanvas()
|
|
else:
|
|
self.canvas = MplCanvasNoMouse()
|
|
self.ntb = CustomNavigationToolbar(self.canvas, self,
|
|
coordinates=False)
|
|
self.vbl = QtWidgets.QVBoxLayout()
|
|
self.vbl.addWidget(self.ntb)
|
|
self.vbl.addWidget(self.canvas)
|
|
self.setLayout(self.vbl)
|