-
Notifications
You must be signed in to change notification settings - Fork 0
/
Copy pathmplwidget.py
38 lines (28 loc) · 1.3 KB
/
mplwidget.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
from PyQt5 import QtWidgets
from matplotlib.figure import Figure
from matplotlib.backends.backend_qt5agg import FigureCanvasQTAgg as Canvas
from matplotlib.backends.backend_qt5agg import NavigationToolbar2QT as NavigationToolbar
import matplotlib
matplotlib.use('QT5Agg')
class MplWidget(QtWidgets.QWidget):
def __init__(self, parent=None):
QtWidgets.QWidget.__init__(self, parent) # Inherit from QWidget
self.vbl = QtWidgets.QVBoxLayout()
def make_plot(self,plot):
try:
self.vbl.removeWidget(self.vbl.canvas)
self.vbl.removeItem(self.vbl.canvas)
self.vbl.removeWidget(self.vbl.toolbar)
self.vbl.removeItem(self.vbl.toolbar)
del(self.vbl.canvas,self.vbl.toolbar )
except AttributeError:
pass
self.canvas = Canvas(plot) # Create canvas object
Canvas.setSizePolicy(self, QtWidgets.QSizePolicy.Expanding,
QtWidgets.QSizePolicy.Expanding)
Canvas.updateGeometry(self)
self.toolbar = NavigationToolbar(self.canvas, self) # toolbar
# Set box for plotting
self.vbl.addWidget(self.toolbar)
self.vbl.addWidget(self.canvas)
self.setLayout(self.vbl)