1 # -*- coding: utf-8 -*- 2 3 import sys 4 from PyQt5 import QtWidgets 5 6 import numpy as np 7 from matplotlib.backends.backend_qt4agg import FigureCanvasQTAgg as FigureCanvas 8 from matplotlib.figure import Figure 9 from matplotlib.animation import FuncAnimation 10 11 class MyMplCanvas(FigureCanvas): 12 """Ultimately, this is a QWidget (as well as a FigureCanvasAgg, etc.).""" 13 def __init__(self, parent=None, width=5, height=4, dpi=100): 14 fig = Figure(figsize=(width, height), dpi=dpi) 15 self.axes = fig.add_subplot(111) 16 # We want the axes cleared every time plot() is called 17 self.axes.hold(False) 18 19 self.compute_initial_figure() 20 21 # 22 FigureCanvas.__init__(self, fig) 23 self.setParent(parent) 24 25 def compute_initial_figure(self): 26 pass 27 28 class AnimationWidget(QtWidgets.QWidget): 29 def __init__(self): 30 QtWidgets.QWidget.__init__(self) 31 32 vbox = QtWidgets.QVBoxLayout() 33 self.canvas = MyMplCanvas(self, width=5, height=4, dpi=100) 34 vbox.addWidget(self.canvas) 35 36 hbox = QtWidgets.QHBoxLayout() 37 self.start_button = QtWidgets.QPushButton("start", self) 38 self.stop_button = QtWidgets.QPushButton("stop", self) 39 self.start_button.clicked.connect(self.on_start) 40 self.stop_button.clicked.connect(self.on_stop) 41 hbox.addWidget(self.start_button) 42 hbox.addWidget(self.stop_button) 43 vbox.addLayout(hbox) 44 self.setLayout(vbox) 45 46 self.x = np.linspace(0, 5*np.pi, 400) 47 self.p = 0.0 48 self.y = np.sin(self.x + self.p) 49 self.line, = self.canvas.axes.plot(self.x, self.y, animated=True, lw=2) 50 51 def update_line(self, i): 52 self.p += 0.1 53 y = np.sin(self.x + self.p) 54 self.line.set_ydata(y) 55 return [self.line] 56 57 def on_start(self): 58 self.ani = FuncAnimation(self.canvas.figure, self.update_line, 59 blit=True, interval=25) 60 61 def on_stop(self): 62 self.ani._stop() 63 64 if __name__ == "__main__": 65 qApp = QtWidgets.QApplication(sys.argv) 66 aw = AnimationWidget() 67 aw.show() 68 sys.exit(qApp.exec_())
