Show the plots

analyze_data.py

@@ -1,6 +1,6 @@
 import EEGrunt
 
-# Required settings 
+# Required settings
 
 # Data source. Options:
 # 'openbci' for data recorded with OBCI GUI;
@@ -25,44 +25,45 @@
 # Possible values are 'show' and 'save'
 EEG.plot = 'show'
 
-
 # Load the EEG data
 EEG.load_data()
 
-
 for channel in EEG.channels:
-    
+
     EEG.load_channel(channel)
 
     print("Processing channel "+ str(EEG.channel))
-        
+
     # Removes OpenBCI DC offset
     EEG.remove_dc_offset()
-    
+
     # Notches 60hz noise (if you're in Europe, switch to 50Hz)
     EEG.notch_mains_interference()
-    
+
     # Make signal plot
     EEG.signalplot()
 
     # Crunches spectrum data and stores as EEGrunt attribute(s) for reuse
     EEG.get_spectrum_data()
-    
-    # Returns bandpassed data 
+
+    # Returns bandpassed data
     # (uses scipy.signal butterworth filter)
-    # EEG.data = EEG.bandpass(start,stop)
+    EEG.data = EEG.bandpass(start,stop)
 
-    # Make Spectrogram 
+    # Make Spectrogram
     EEG.spectrogram()
 
     # Line graph of amplitude over time for a given frequency range.
     # Arguments are start frequency, end frequency, and label
     EEG.plot_band_power(8,12,"Alpha")
-    
+
     # Power spectrum
     EEG.plot_spectrum_avg_fft()
-        
+
     # Plot coherence fft (not tested recently...)
     # s1 = bandpass(seginfo["data"][:,1-1], config['band'])
     # s2 = bandpass(seginfo["data"][:,8-1], config['band'])
     # plot_coherence_fft(s1,s2,"1","8")
+
+# When all's said and done, show the plots
+EEG.showplots()