-
-
Save boylea/1a0b5442171f9afbf372 to your computer and use it in GitHub Desktop.
| """ | |
| Tested on Linux with python 3.7 | |
| Must have portaudio installed (e.g. dnf install portaudio-devel) | |
| pip install pyqtgraph pyaudio PyQt5 | |
| """ | |
| import numpy as np | |
| import pyqtgraph as pg | |
| import pyaudio | |
| from PyQt5 import QtCore, QtGui | |
| FS = 44100 #Hz | |
| CHUNKSZ = 1024 #samples | |
| class MicrophoneRecorder(): | |
| def __init__(self, signal): | |
| self.signal = signal | |
| self.p = pyaudio.PyAudio() | |
| self.stream = self.p.open(format=pyaudio.paInt16, | |
| channels=1, | |
| rate=FS, | |
| input=True, | |
| frames_per_buffer=CHUNKSZ) | |
| def read(self): | |
| data = self.stream.read(CHUNKSZ, exception_on_overflow=False) | |
| y = np.fromstring(data, 'int16') | |
| self.signal.emit(y) | |
| def close(self): | |
| self.stream.stop_stream() | |
| self.stream.close() | |
| self.p.terminate() | |
| class SpectrogramWidget(pg.PlotWidget): | |
| read_collected = QtCore.pyqtSignal(np.ndarray) | |
| def __init__(self): | |
| super(SpectrogramWidget, self).__init__() | |
| self.img = pg.ImageItem() | |
| self.addItem(self.img) | |
| self.img_array = np.zeros((1000, int(CHUNKSZ/2+1))) | |
| # bipolar colormap | |
| pos = np.array([0., 1., 0.5, 0.25, 0.75]) | |
| color = np.array([[0,255,255,255], [255,255,0,255], [0,0,0,255], (0, 0, 255, 255), (255, 0, 0, 255)], dtype=np.ubyte) | |
| cmap = pg.ColorMap(pos, color) | |
| lut = cmap.getLookupTable(0.0, 1.0, 256) | |
| # set colormap | |
| self.img.setLookupTable(lut) | |
| self.img.setLevels([-50,40]) | |
| # setup the correct scaling for y-axis | |
| freq = np.arange((CHUNKSZ/2)+1)/(float(CHUNKSZ)/FS) | |
| yscale = 1.0/(self.img_array.shape[1]/freq[-1]) | |
| self.img.scale((1./FS)*CHUNKSZ, yscale) | |
| self.setLabel('left', 'Frequency', units='Hz') | |
| # prepare window for later use | |
| self.win = np.hanning(CHUNKSZ) | |
| self.show() | |
| def update(self, chunk): | |
| # normalized, windowed frequencies in data chunk | |
| spec = np.fft.rfft(chunk*self.win) / CHUNKSZ | |
| # get magnitude | |
| psd = abs(spec) | |
| # convert to dB scale | |
| psd = 20 * np.log10(psd) | |
| # roll down one and replace leading edge with new data | |
| self.img_array = np.roll(self.img_array, -1, 0) | |
| self.img_array[-1:] = psd | |
| self.img.setImage(self.img_array, autoLevels=False) | |
| if __name__ == '__main__': | |
| app = QtGui.QApplication([]) | |
| w = SpectrogramWidget() | |
| w.read_collected.connect(w.update) | |
| mic = MicrophoneRecorder(w.read_collected) | |
| # time (seconds) between reads | |
| interval = FS/CHUNKSZ | |
| t = QtCore.QTimer() | |
| t.timeout.connect(mic.read) | |
| t.start(1000/interval) #QTimer takes ms | |
| app.exec_() | |
| mic.close() |
Can someone please add a pictures of the result?
I discovered today after upgrading to matplotlib 3.7.1 to get colormaps like Turbo that I needed to update line 1084 in /usr/lib/python3/dist-packages/pyqtgraph/functions.py with the following
levels = levels.astype(float)
instead of
levels = levels.astype(np.float)
as it has been depreciated
looks great and works again
The photo is live audio from Dallas Love Field showing realtime spectrographs of aircraft taking off and their distinct audio signatures.
Does this still work? I installed pyqt5 but I get this error when I try to run it:
AttributeError: module 'PyQt5.QtGui' has no attribute 'QApplication'. Did you mean: 'QGuiApplication'?
Does this still work? I installed pyqt5 but I get this error when I try to run it:
AttributeError: module 'PyQt5.QtGui' has no attribute 'QApplication'. Did you mean: 'QGuiApplication'?
Yes, using 'QGuiApplication' fixed that error. I had a separate issue, getting a message:
Unable to locate theme engine in module_path: "adwaita"
...which cleared after installing gnome-themes-standard
If you find Qt5 is out of date, pip installing pyside2 will fix that, but it is a large download.
I tried QGuiApplication, but it didn't work. What got it working for me was switching to:
app = QtWidgets.QApplication([])
I'm having trouble with the scaling. Is this a Pyqt version problem?
File "livespec.py", line 142, in __init__
self.img.setScale((1./FS)*CHUNKSZ, yscale)
TypeError: setScale(self, scale: float): too many arguments
I also had this problem. I had to switch to using:
self.img.setTransform(QtGui.QTransform.fromScale((1./samplerate)*samplesize, yscale))
I'm not sure if it does the exact same thing but it seems to work.
Awesome tool. I am wondering if there is a way to an interactive color bar. Any help would be appreciated. Thanks
I finally got around to firing this up again and updated the gist with the suggested fixes, thanks @pranav6670 and @conraddisc !
Also thanks for the comment about using matplotlib's pre-defined maps @AB9IL