如何更新实时绘图并使用按钮在pyqtgraph中进行交互?

时间:2016-01-28 08:45:53

标签: python pyqt4 pyqtgraph

我使用了"滚动图"示例来自pyqtgraph.examples作为模板创建绘图,显示连续的正弦波。现在,我想用按钮来改变正弦波的幅度。这就是我使用其他结构的原因(参见下面的代码)。这不起作用,它只绘制静态正弦波。 如何使用我的更新功能绘制连续的正弦波?如何使用按钮来调整振幅?我已经检查thisthis但没有成功。

import sys
from PyQt4 import QtGui, QtCore
import numpy as np
import pyqtgraph as pg


class sinus_wave(QtGui.QWidget):

    def __init__(self):
        super(sinus_wave, self).__init__()

        self.initUI()


    def initPlot(self, plots):
        a = 10
        ptr1 = 30

        data1 = a*np.sin(np.linspace(0,30,121))
        plots.plot(data1)

        timer = pg.QtCore.QTimer()
        timer.timeout.connect(lambda: self.update(self,p1 = plots,data1= data1, ptr1 = ptr1))
        timer.start(50)

    def initUI(self):


        IncreaseButton = QtGui.QPushButton("Increase Amplitude")
        DecreaseButton = QtGui.QPushButton("Decrease Amplitude")
        p1 = pg.PlotWidget()

        hbox = QtGui.QVBoxLayout()  
        hbox.addWidget(p1)
        hbox.addWidget(IncreaseButton)
        hbox.addWidget(DecreaseButton)  

        self.initPlot(p1)
        self.setLayout(hbox)

        self.setGeometry(10, 10, 1000, 600)
        self.setWindowTitle('Sinuswave')    
        self.show()


    def update(self, p1, data1, ptr1):

        data1[:-1] = data1[1:]  
        data1[-1] = np.sin(ptr1/4)
        p1.plot(data1)
        ptr1 += 1
        p1.show()

    def IncreaseButtonClick(self):
        print ("Amplitude increased")

    def DecreaseButtonClick(self):
        print ("Amplitude decreased")


def main():

    app = QtGui.QApplication(sys.argv)
    app.setApplicationName('Sinuswave')
    ex = sinus_wave()

    sys.exit(app.exec_())

if __name__ == '__main__':

    main()

1 个答案:

答案 0 :(得分:2)

所以这不是一个pyqtgraph问题。大多数情况下,您需要阅读python和pyqt基础知识中的类。但我会尝试为你的问题提供一些快速解决方案,希望你能在途中学到一些东西。

要使按钮工作,您需要将它们连接到方法。尝试在这里查看答案:https://stackoverflow.com/a/8763339/4328381

你需要这样的东西

def qt_connections(self):
    self.increasebutton.clicked.connect(self.on_increasebutton_clicked)
    self.decreasebutton.clicked.connect(self.on_decreasebutton_clicked)

然后让按钮实际上做一些他们需要改变幅度的东西。首先将幅度存储为实例的属性。还存储一个t属性以便以后移动。

你的a和ptr1只是方法中的变量,一旦方法完成就会被清除。通过使用self.,它们成为可以从类中的其他方法使用的实例属性。

def __init__(self):
    ...
    self.amplitude = 10
    self.t = 0
    ...

然后您可以在连接到按钮的方法中更改振幅。

def on_increasebutton_clicked(self):
    print ("Amplitude increased")
    self.amplitude += 1
    self.updateplot()

然后,为了使它连续,你首先需要确保计时器工作。尝试在其中添加print("test"),您会发现它没有。您需要保留它的引用,否则它将被清理。

    self.timer = pg.QtCore.QTimer()
    self.timer.timeout.connect(self.moveplot)
    self.timer.start(50)

要使正弦连续移动,您需要在连接到计时器的方法中移动它,您可以简单地创建一个moveplot方法。

def moveplot(self):
    self.t+=1
    self.updateplot()

然后将使用之前创建的属性的create和updateplot方法放在一起。

def updateplot(self):
    print "Update"
    data1 = self.amplitude*np.sin(np.linspace(0,30,121)+self.t)
    self.plotcurve.setData(data1)

最后你会得到类似的东西

import sys
from PyQt4 import QtGui, QtCore
import numpy as np
import pyqtgraph as pg


class sinus_wave(QtGui.QWidget):
    def __init__(self):
        super(sinus_wave, self).__init__()
        self.init_ui()
        self.qt_connections()
        self.plotcurve = pg.PlotCurveItem()
        self.plotwidget.addItem(self.plotcurve)
        self.amplitude = 10
        self.t = 0
        self.updateplot()

        self.timer = pg.QtCore.QTimer()
        self.timer.timeout.connect(self.moveplot)
        self.timer.start(500)

    def init_ui(self):
        self.setWindowTitle('Sinuswave')
        hbox = QtGui.QVBoxLayout()
        self.setLayout(hbox)

        self.plotwidget = pg.PlotWidget()
        hbox.addWidget(self.plotwidget)

        self.increasebutton = QtGui.QPushButton("Increase Amplitude")
        self.decreasebutton = QtGui.QPushButton("Decrease Amplitude")

        hbox.addWidget(self.increasebutton)
        hbox.addWidget(self.decreasebutton)

        self.setGeometry(10, 10, 1000, 600)
        self.show()

    def qt_connections(self):
        self.increasebutton.clicked.connect(self.on_increasebutton_clicked)
        self.decreasebutton.clicked.connect(self.on_decreasebutton_clicked)

    def moveplot(self):
        self.t+=1
        self.updateplot()

    def updateplot(self):
        print "Update"
        data1 = self.amplitude*np.sin(np.linspace(0,30,121)+self.t)
        self.plotcurve.setData(data1)

    def on_increasebutton_clicked(self):
        print ("Amplitude increased")
        self.amplitude += 1
        self.updateplot()

    def on_decreasebutton_clicked(self):
        print ("Amplitude decreased")
        self.amplitude -= 1
        self.updateplot()

def main():
    app = QtGui.QApplication(sys.argv)
    app.setApplicationName('Sinuswave')
    ex = sinus_wave()
    sys.exit(app.exec_())

if __name__ == '__main__':
    main()