I'm currently exploring realtime data plotting with pyqtgraph. I've
successfully managed to plot 3. Two of them are Analog-Digital Converter
data coming from an UDP port and last one is amplitude spectrum of one of
the datas. I also want to plot spectrogram data of this. Since I've plotted
one dimensional plots on same page, I'm having problems with plotting a
spectrogram because it's two dimensional image. How can I plot a
spectrogram on the same window with others?
#!/usr/bin/env python
from __future__ import division
from pyqtgraph.Qt import QtGui, QtCore
import pyqtgraph as pg
import numpy as np
import socket
import threading
UDP_IP = "192.168.180.64"
UDP_PORT = [5013, 5012]
data = ["", ""]
sock1 = socket.socket(socket.AF_INET, socket.SOCK_DGRAM)
sock2 = socket.socket(socket.AF_INET, socket.SOCK_DGRAM)
sock1.bind((UDP_IP, UDP_PORT[0]))
sock2.bind((UDP_IP, UDP_PORT[1]))
t = 0
time_step = 1 / 1024
class RingBuffer():
def __init__(self, length):
self.data = np.zeros(length, dtype='f')
self.index = 0
def extend(self, x):
x_index = (self.index + np.arange(x.size)) % self.data.size
self.data[x_index] = x
self.index = x_index[-1] + 1
def get(self):
idx = (self.index + np.arange(self.data.size)) % self.data.size
return self.data[idx]
class DynamicPlotter(pg.GraphicsLayoutWidget):
app = QtGui.QApplication([])
def __init__(self, sampleinterval, timewindow, size=(1024, 350)):
super(DynamicPlotter, self).__init__()
# Data
self._interval = int(sampleinterval * 1000)
self._bufsize = int(timewindow / sampleinterval)
self.databuffer1 = RingBuffer(self._bufsize)
self.databuffer2 = RingBuffer(self._bufsize)
self.databuffer3 = RingBuffer(1024)
for i in range(0, 1024):
self.databuffer3.extend(self.getdata(0))
self.fourier = np.fft.fft(self.databuffer3.get())
self.freqs = np.fft.fftfreq(len(self.databuffer3.get()),
float(time_step))
self.idx = np.argsort(self.freqs)
self.x = np.linspace(-timewindow, 0.0, self._bufsize)
self.y1 = np.zeros(self._bufsize, dtype=np.float)
self.y2 = np.zeros(self._bufsize, dtype=np.float)
self.x3 = self.freqs[self.idx]
self.y3 = np.abs(self.fourier)[self.idx]
# PyQtGraph Init
self.win = QtGui.QMainWindow()
self.cwin = pg.GraphicsLayoutWidget()
self.win.show()
self.win.resize(1600, 300)
self.win.setCentralWidget(self.cwin)
self.win.setWindowTitle('ADC Data Plots')
pg.setConfigOptions(antialias=True)
self.plt1 = self.cwin.addPlot(title='ADC Kanal 0')
self.plt1.resize(*size)
self.plt1.showGrid(x=True, y=True)
self.plt1.setLabel('left', 'Genlik', 'V')
self.plt1.setLabel('bottom', 'Zaman')
self.curve1 = self.plt1.plot(self.x, self.y1, pen=(0, 255, 0))
self.plt2 = self.cwin.addPlot(title='ADC Kanal 1')
self.plt2.resize(*size)
self.plt2.showGrid(x=True, y=True)
self.plt2.setLabel('left', 'Genlik', 'V')
self.plt2.setLabel('bottom', 'Zaman')
self.curve2 = self.plt2.plot(self.x, self.y1, pen=(0, 0, 255))
self.plt3 = self.cwin.addPlot(title='Genlik Spektrumu')
self.plt3.showGrid(x=True, y=True)
self.plt3.setLabel('left', 'Genlik')
self.plt3.setLabel('bottom', 'Frekans', 'Hz')
self.plt3.resize(*size)
self.curve3 = self.plt3.plot(self.x3, self.y3, pen=(0, 255, 255))
self.app.processEvents()
self.updatebuffers_thread = threading.Thread(target=self.updatebuffers)
self.updatebuffers_thread.start()
# QTimer
self.timer = QtCore.QTimer()
self.timer.timeout.connect(self.updateplot)
self.timer.start(self._interval)
def getdata(self, i):
data[0], addr = sock1.recvfrom(1024)
data[1], addr = sock2.recvfrom(1024)
return np.array([data[i]], dtype=np.float)
def updatebuffers(self):
global t
while True: # or while self.plt1.isVisible():
if t == 102:
self.fourier = np.fft.fft(self.databuffer3.get())
self.freqs = np.fft.fftfreq(len(self.databuffer3.get()),
float(time_step))
self.idx = np.argsort(self.freqs)
t = 0
self.databuffer1.extend(self.getdata(0))
self.databuffer2.extend(self.getdata(1))
self.databuffer3.extend(self.getdata(0))
self.y1[:] = self.databuffer1.get()
self.y2[:] = self.databuffer2.get()
self.x3[:] = self.freqs[self.idx]
self.y3[:] = np.abs(self.fourier)[self.idx]
self.y3[512] = 0
t += 1
def updateplot(self):
self.curve1.setData(self.x, self.y1)
self.curve2.setData(self.x, self.y2)
self.curve3.setData(self.x3[512:1024], self.y3[512:1024])
self.app.processEvents()
def run(self):
self.app.exec_()
if __name__ == '__main__':
m = DynamicPlotter(sampleinterval=0.01, timewindow=5)
m.run()
--
You received this message because you are subscribed to the Google Groups
"pyqtgraph" group.
To unsubscribe from this group and stop receiving emails from it, send an email
to [email protected].
To view this discussion on the web visit
https://groups.google.com/d/msgid/pyqtgraph/121b8c74-d83d-48d8-ae46-3ccb7db0a08d%40googlegroups.com.
For more options, visit https://groups.google.com/d/optout.
