Fixed no results when LowPass exceeds Nyquist frequency

Paradeluxe · Dec 14, 2024 · afc438b · afc438b
1 parent d665832
commit afc438b
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 2 deletions.
diff --git a/core.py b/core.py
@@ -52,6 +52,7 @@ def runPraditor(params, audio_obj, which_set):
         fs=_audio_obj.frame_rate
     )
 
+    # warning or auto change?
     if which_set == "offset":
         _audio_arr_filtered = np.flip(_audio_arr_filtered)
 

diff --git a/tool.py b/tool.py
@@ -28,8 +28,12 @@ def bandpass_filter(data, lowcut, highcut, fs, order=4):
     nyquist = 0.5 * fs
     low = lowcut / nyquist
     high = highcut / nyquist
-    b, a = butter(order, [low, high], btype='bandpass', output="ba")
-    filtered_data = filtfilt(b, a, data)
+    try:
+        b, a = butter(order, [low, high], btype='bandpass', output="ba")
+        filtered_data = filtfilt(b, a, data)
+    except ValueError:  # 如果设置的最高频率大于了可接受的范围，则变为（实际意义上的）高通滤波
+        b, a = butter(order, [low, 1], btype='bandpass', output="ba")
+        filtered_data = filtfilt(b, a, data)
     return filtered_data