mirror of
https://github.com/Athemis/PyDSF.git
synced 2025-04-05 22:56:02 +00:00
56 lines
2.3 KiB
Python
56 lines
2.3 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
|
|
|
|
|
|
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)
|
|
|
|
# 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
|
|
|
|
def clear(self):
|
|
self.ax.clear()
|
|
self.fig.clear()
|
|
|
|
|
|
class CustomNavigationToolbar(NavigationToolbar):
|
|
toolitems = (
|
|
(QCoreApplication.translate("CustomNavigationToolbar", 'Save'),
|
|
QCoreApplication.translate("CustomNavigationToolbar",
|
|
'Save the figure'), 'filesave',
|
|
'save_figure'),
|
|
(QCoreApplication.translate("CustomNavigationToolbar", 'Subplots'),
|
|
QCoreApplication.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):
|
|
QtWidgets.QGraphicsView.__init__(self, parent)
|
|
self.canvas = MplCanvas()
|
|
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)
|