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
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
|
from matplotlib.backends.backend_qtagg import FigureCanvasQTAgg as FigureCanvas
from matplotlib.backends.backend_qtagg import NavigationToolbar2QT as NavigationToolbar
from matplotlib.figure import Figure
from ..Qt import QtWidgets
import typing
__all__ = ['MatplotlibWidget']
class MatplotlibWidget(QtWidgets.QWidget):
"""
Implements a Matplotlib figure inside a QWidget.
Use getFigure() and redraw() to interact with matplotlib.
Example::
mw = MatplotlibWidget()
subplot = mw.getFigure().add_subplot(111)
subplot.plot(x,y)
mw.draw()
"""
parent_default = None
figsize_default = (5.0, 4.0)
dpi_default = 100
@typing.overload
def __init__(self, figsize=(5.0, 4.0), dpi=100, parent=None):
pass
@typing.overload
def __init__(self, parent=None, figsize=(5.0, 4.0), dpi=100):
pass
def __init__(self, *args, **kwargs):
if (args and not isinstance(args[0], QtWidgets.QWidget)):
figsize = args[0] if len(args) > 0 \
else kwargs.get("figsize", MatplotlibWidget.figsize_default)
dpi = args[1] if len(args) > 1 \
else kwargs.get("dpi", MatplotlibWidget.dpi_default)
parent = args[2] if len(args) > 2 \
else kwargs.get("parent", MatplotlibWidget.parent_default)
else:
parent = args[0] if len(args) > 0 \
else kwargs.get("parent", MatplotlibWidget.parent_default)
figsize = args[1] if len(args) > 1 \
else kwargs.get("figsize", MatplotlibWidget.figsize_default)
dpi = args[2] if len(args) > 2 \
else kwargs.get("dpi", MatplotlibWidget.dpi_default)
super().__init__(parent)
self.fig = Figure(figsize, dpi=dpi)
self.canvas = FigureCanvas(self.fig)
self.canvas.setParent(self)
self.toolbar = NavigationToolbar(self.canvas, self)
self.vbox = QtWidgets.QVBoxLayout()
self.vbox.addWidget(self.toolbar)
self.vbox.addWidget(self.canvas)
self.setLayout(self.vbox)
def getFigure(self):
return self.fig
def draw(self):
self.canvas.draw()
|
