Add audio visualization as example for pyphastft

pyphastft/vis_qt.py

@@ -0,0 +1,99 @@
+from pyphastft import fft
+import numpy as np
+import pyaudio
+import pyqtgraph as pg
+from pyqtgraph.Qt import QtGui, QtWidgets, QtCore
+import sys
+
+
+class RealTimeAudioSpectrum(QtWidgets.QWidget):
+    def __init__(self, parent=None):
+        super(RealTimeAudioSpectrum, self).__init__(parent)
+        self.n_fft_bins = 32  # FFT size (number of bins)
+        self.sample_rate = 44100
+        self.smoothing_factor = 0.1  # Smaller value for more smoothing
+        self.ema_fft_data = np.zeros(self.n_fft_bins // 2)  # Initialize EMA array
+        self.init_ui()
+        self.init_audio_stream()
+
+    def init_ui(self):
+        self.layout = QtWidgets.QVBoxLayout(self)
+        self.plot_widget = pg.PlotWidget()
+        self.layout.addWidget(self.plot_widget)
+
+        # Customize plot aesthetics
+        self.plot_widget.setBackground('k')
+        self.plot_item = self.plot_widget.getPlotItem()
+        self.plot_item.setTitle("PyPhastFT RT Input Audio Spectrum Visualizer", color="w", size="16pt")
+
+        # self.plot_item.getAxis('left').setLabel("Magnitude", color="w", size="14pt")
+        # self.plot_item.getAxis('bottom').setLabel("Frequency (Hz)", color="w", size="14pt")
+
+        # Hide x and y axis lines, ticks, and numbers
+        # self.plot_item.getAxis('left').setStyle(showValues=False)
+        # self.plot_item.getAxis('left').setPen(pg.mkPen(None))  # Hides the axis line
+        # self.plot_item.getAxis('bottom').setStyle(showValues=False)
+        # self.plot_item.getAxis('bottom').setPen(pg.mkPen(None))  # Hides the axis line
+
+        # Hide axis labels
+        self.plot_item.getAxis('left').hide()
+        self.plot_item.getAxis('bottom').hide()
+
+        # Set fixed ranges for the x and y axes to prevent them from jumping
+        self.plot_item.setXRange(0, self.sample_rate / 2, padding=0)
+        self.plot_item.setYRange(0, 1, padding=0)  # You may want to adjust this based on your FFT data
+
+        self.bar_width = (self.sample_rate / 2) / (self.n_fft_bins // 2) * 0.90  # Width based on frequency spacing
+
+        # Calculate bar positions so they are centered with respect to their frequency values
+        self.freqs = np.linspace(0 + self.bar_width/2, self.sample_rate / 2 - self.bar_width/2, self.n_fft_bins // 2)
+
+        # Adjust x-axis range to account for the full width of the first and last bars
+        self.plot_item.setXRange(-self.bar_width, self.sample_rate / 2 + self.bar_width, padding=0)
+
+        self.bar_graph = pg.BarGraphItem(x=self.freqs, height=np.zeros(self.n_fft_bins // 2), width=self.bar_width, brush=pg.mkBrush('m'))
+        self.plot_item.addItem(self.bar_graph)
+
+        self.timer = QtCore.QTimer()
+        self.timer.timeout.connect(self.update)
+        self.timer.start(50)  # Update interval in milliseconds
+
+    def init_audio_stream(self):
+        self.p = pyaudio.PyAudio()
+        self.stream = self.p.open(
+            format=pyaudio.paFloat32,
+            channels=1,
+            rate=self.sample_rate,
+            input=True,
+            frames_per_buffer=self.n_fft_bins,
+            stream_callback=self.audio_callback,
+        )
+        self.stream.start_stream()
+
+    def audio_callback(self, in_data, frame_count, time_info, status):
+        audio_data = np.frombuffer(in_data, dtype=np.float32).astype(np.float64)
+        reals = audio_data
+        imaginaries = np.zeros(self.n_fft_bins, dtype=np.float64)
+        fft(reals, imaginaries, direction='f')
+        new_fft_data = np.sqrt(reals**2 + imaginaries**2)[:self.n_fft_bins // 2]
+
+        # Apply exponential moving average filter
+        self.ema_fft_data = self.ema_fft_data * self.smoothing_factor + \
+            new_fft_data * (1 - self.smoothing_factor)
+        return (in_data, pyaudio.paContinue)
+
+    def update(self):
+        self.bar_graph.setOpts(height=self.ema_fft_data, width=self.bar_width)
+
+    def closeEvent(self, event):
+        self.stream.stop_stream()
+        self.stream.close()
+        self.p.terminate()
+        event.accept()
+
+
+if __name__ == "__main__":
+    app = QtWidgets.QApplication(sys.argv)
+    window = RealTimeAudioSpectrum()
+    window.show()
+    sys.exit(app.exec_())