added more frequency options

Author: Jonathan Zhao

audio_visualizer.py

@@ -135,14 +135,13 @@ def __init__(self, py_audio, data_format=pyaudio.paInt16,
         self.fft_size = int(self.fft_size)
 
         self.max_freq = min(max_frequency, int(self.sample_rate / 2))
-        self.bass_freq = bass_frequency
-
-        if self.max_freq < high_frequency:
-            high_frequency = self.max_freq
+        self.low_freq = max(0, low_frequency)
+        self.high_freq = min(self.max_freq, high_frequency)
+        self.bass_freq = max(0, bass_frequency)
 
         self.bass_index = int(self.bass_freq / self.max_freq * self.fft_size)
-        self.low_index = int(low_frequency / self.max_freq * self.fft_size)
-        self.high_index = int(high_frequency / self.max_freq * self.fft_size)
+        self.low_index = int(self.low_freq / self.max_freq * self.fft_size)
+        self.high_index = int(self.high_freq / self.max_freq * self.fft_size)
 
         # sets up QtPy application
         signal.signal(signal.SIGINT, signal.SIG_DFL)

custom_qt.py

@@ -87,6 +87,20 @@ class SettingsPanel(QtWidgets.QWidget):
     def assign_bass_freq(self, text):
         self.visualizer.bass_freq = int(text or '0')
         self.visualizer.bass_index = int(self.visualizer.bass_freq / self.visualizer.max_freq * self.visualizer.fft_size)
+    
+    def assign_low_freq(self, text):
+        self.visualizer.low_freq = int(text or '0')
+        self.visualizer.low_index = int(self.visualizer.low_freq / self.visualizer.max_freq * self.visualizer.fft_size)
+    
+    def assign_high_freq(self, text):
+        self.visualizer.high_freq = int(text or '1')
+        self.visualizer.high_index = int(self.visualizer.high_freq / self.visualizer.max_freq * self.visualizer.fft_size)
+    
+    def assign_max_freq(self, text):
+        self.visualizer.max_freq = int(text or '1')
+        self.visualizer.bass_index = int(self.visualizer.bass_freq / self.visualizer.max_freq * self.visualizer.fft_size)
+        self.visualizer.low_index = int(self.visualizer.low_freq / self.visualizer.max_freq * self.visualizer.fft_size)
+        self.visualizer.high_index = int(self.visualizer.high_freq / self.visualizer.max_freq * self.visualizer.fft_size)
 
     def assign_wav_decay(self, text):
         self.visualizer.wav_decay_speed = float(text or '0')
@@ -134,11 +148,29 @@ def __init__(self, button):
         layout = QtWidgets.QFormLayout()
 
         bass_freq_field = QtWidgets.QLineEdit()
-        bass_freq_field.setValidator(QtGui.QIntValidator(0, self.visualizer.max_freq))
+        bass_freq_field.setValidator(QtGui.QIntValidator(0, self.visualizer.sample_rate // 2))
         bass_freq_field.setText(str(self.visualizer.bass_freq))
         bass_freq_field.textChanged.connect(self.assign_bass_freq)
         layout.addRow("Bass Frequency", bass_freq_field)
 
+        low_freq_field = QtWidgets.QLineEdit()
+        low_freq_field.setValidator(QtGui.QIntValidator(0, self.visualizer.sample_rate // 2))
+        low_freq_field.setText(str(self.visualizer.low_freq))
+        low_freq_field.textChanged.connect(self.assign_low_freq)
+        layout.addRow("Low Frequency", low_freq_field)
+
+        high_freq_field = QtWidgets.QLineEdit()
+        high_freq_field.setValidator(QtGui.QIntValidator(1, self.visualizer.sample_rate // 2))
+        high_freq_field.setText(str(self.visualizer.high_freq))
+        high_freq_field.textChanged.connect(self.assign_high_freq)
+        layout.addRow("High Frequency", high_freq_field)
+        
+        max_freq_field = QtWidgets.QLineEdit()
+        max_freq_field.setValidator(QtGui.QIntValidator(1, self.visualizer.sample_rate // 2))
+        max_freq_field.setText(str(self.visualizer.max_freq))
+        max_freq_field.textChanged.connect(self.assign_max_freq)
+        layout.addRow("Max Frequency", max_freq_field)
+
         wav_decay_speed_field = QtWidgets.QLineEdit()
         wav_decay_speed_field.setValidator(QtGui.QDoubleValidator(0, 1, 2))
         wav_decay_speed_field.setText(str(self.visualizer.wav_decay_speed))