vraag11

main.py

@@ -103,30 +103,32 @@ def plot_emg_data(file_path):
     num_samples = int(6 * sampling_rate)
     num_samples = min(num_samples, data.shape[1])
     time_vector = np.linspace(0,6,num_samples)
+
+    
     channel_rectus_femoris = data[chosen_muscles['rectus_fomoris'], :num_samples]
     channel_tibialis_anterior = data[chosen_muscles['tibialis_anterior'], :num_samples]
     
+    signal_processor = SignalProcessing(data, sampling_rate)
+    filtered_rf = signal_processor.bandpass_filter(channel_rectus_femoris, 25, 300)
+    filtered_ta = signal_processor.bandpass_filter(channel_tibialis_anterior, 25, 300)
 
-    freq_rf, fft_rf = compute_fft(channel_rectus_femoris[:num_samples], sampling_rate)
-    freq_ta, fft_ta = compute_fft(channel_tibialis_anterior[:num_samples], sampling_rate)
+    freq_rf, fft_rf = compute_fft(filtered_rf, sampling_rate)
+    freq_ta, fft_ta = compute_fft(filtered_ta, sampling_rate)
 
     plt.figure(figsize=(13,8))
 
     plt.subplot(2, 2, 1)
-    plt.plot(time_vector, channel_rectus_femoris)
+    plt.plot(time_vector, filtered_rf)
     plt.title('EMG Data - Rectus femoris')
     plt.xlabel('Time [Seconds]')
     plt.ylabel('Amplitude')
 
     plt.subplot(2, 2, 2)
-    plt.plot(time_vector, channel_tibialis_anterior)
+    plt.plot(time_vector, filtered_ta)
     plt.title('EMG Data - Tibialis Anterior')
     plt.xlabel('Time [seconds]')
     plt.ylabel('Amplitude')
 
-
-    plt.tight_layout()
-
     plt.subplot(2, 2, 3)
     plt.plot(freq_rf, fft_rf)
     plt.title('Power Spectrum - Rectus Femoris')