Merge pull request #19 from PayalLakra/bio_amptool

Updation in EMG and EOG

Author: lorforlinux

Notebooks/ECG.ipynb

@@ -77113,8 +77113,8 @@
     "\n",
     "    # Define a window around the R-peak to capture the full PQRST complex\n",
     "    window_size = int(0.5 * 250)  # Half a second before and after the R-peak \n",
-    "    start_idx = max(0, highest_r_peak_idx - window_size)  \n",
-    "    end_idx = min(len(ecg_signal), highest_r_peak_idx + window_size) \n",
+    "    # start_idx = max(0, highest_r_peak_idx - window_size)  \n",
+    "    # end_idx = min(len(ecg_signal), highest_r_peak_idx + window_size) \n",
     "\n",
     "    # Extract the time and ECG data for the window\n",
     "    window_time = time[start_idx:end_idx]\n",
@@ -115915,7 +115915,7 @@
  ],
  "metadata": {
   "kernelspec": {
-   "display_name": "venv",
+   "display_name": "Python 3 (ipykernel)",
    "language": "python",
    "name": "python3"
   },
@@ -115929,7 +115929,7 @@
    "name": "python",
    "nbconvert_exporter": "python",
    "pygments_lexer": "ipython3",
-   "version": "3.10.10"
+   "version": "3.11.7"
   }
  },
  "nbformat": 4,

README.md

@@ -27,7 +27,7 @@ Chords Python script is designed to interface with an Arduino-based bioamplifier
 
 ## Installation
 
-1. Ensure you have Python 3.10 installed.
+1. Ensure you have latest version of Python installed.
 2. Create Virtual Environment
    ```bash
    python -m venv venv    
@@ -95,7 +95,7 @@ To use the script, run it from the command line with various options:
 ## Applications
 
 > [!IMPORTANT]
- Before using the below Applications make sure you are in Application folder.
+ Before using the below Applications make sure you are in application folder.
 
 ### GUI
 
@@ -123,9 +123,9 @@ To use the script, run it from the command line with various options:
 
 `check_win_condition()`: Determines if either player has won based on the ball's position.
 
-### Heart Rate
+### HEART RATE
 
-- `python heartbeat.ecg.py`:Enable a GUI with real-time ECG and heart rate
+- `python heartbeat.ecg.py`:Enable a GUI with real-time ECG and heart rate.
 
 #### Script Functions
 
@@ -141,6 +141,22 @@ To use the script, run it from the command line with various options:
 
 `update_heart_rate(self)`: Calculates and updates the heart rate based on detected R-peaks in the ECG signal.
 
+### EMG ENVELOPE
+
+- `python emgenvelope.py` :Enable a GUI with real-time EMG & its Envelope.
+
+#### Script Functions
+
+ `update_plot` : Updates the plot with latest Filtered EMG Data and its Envelope.
+
+### EOG
+
+- `python eog.py` :Enable a GUI with real-time EOG.
+
+#### Script Functions
+
+ `update_plot` : Updates the plot with latest Filtered EOG Data.
+
 ## Troubleshooting
 
 - **Arduino Not Detected:** Ensure the Arduino is properly connected and powered. Check the serial port and baud rate settings.

applications/emgenvelope.py

@@ -16,19 +16,21 @@ def __init__(self):
         # Create layout
         layout = QVBoxLayout()
 
-        # Create plot widgets for raw EMG and EMG envelope
-        self.raw_emg_plot = PlotWidget(self)
-        self.raw_emg_plot.setBackground('w')
-        self.raw_emg_plot.showGrid(x=True, y=True)
-        self.raw_emg_plot.setTitle("Raw EMG Signal")
+        # Create plot widgets for Filtered EMG and EMG envelope
+        self.emg_plot = PlotWidget(self)
+        self.emg_plot.setBackground('w')
+        self.emg_plot.showGrid(x=True, y=True)
+        self.emg_plot.setMouseEnabled(x=False, y=False)  # Disable zoom
+        self.emg_plot.setTitle("Filtered EMG Signal (High Pass:70 Hz)")
 
         self.envelope_plot = PlotWidget(self)
         self.envelope_plot.setBackground('w')
         self.envelope_plot.showGrid(x=True, y=True)
-        self.envelope_plot.setTitle("EMG Envelope")
+        self.envelope_plot.setMouseEnabled(x=False, y=False)  # Disable zoom
+        self.envelope_plot.setTitle("EMG Envelope (Samples Average:10%)")
 
         # Add plots to layout
-        layout.addWidget(self.raw_emg_plot)
+        layout.addWidget(self.emg_plot)
         layout.addWidget(self.envelope_plot)
 
         # Central widget
@@ -55,23 +57,23 @@ def __init__(self):
 
         self.b, self.a = butter(4, 70.0 / (0.5 * self.sampling_rate), btype='high')
 
-        # Moving RMS window size (50 for 250 sampling rate and 100 for 500 sampling rate)
+        # Moving RMS window size (25 for 250 sampling rate and 50 for 500 sampling rate)
         self.rms_window_size = int(0.1 * self.sampling_rate)
 
         # Set fixed axis ranges
-        self.raw_emg_plot.setXRange(0, 10, padding=0)
+        self.emg_plot.setXRange(0, 10, padding=0)
         self.envelope_plot.setXRange(0, 10, padding=0)
 
-        # Set y-axis limits based on sampling rate for raw EMG
+        # Set y-axis limits based on sampling rate for Filtered EMG
         if self.sampling_rate == 250:  
-            self.raw_emg_plot.setYRange(-((2**10)/2), ((2**10)/2), padding=0)  # for R3
+            self.emg_plot.setYRange(-((2**10)/2), ((2**10)/2), padding=0)  # for R3
             self.envelope_plot.setYRange(0, ((2**10)/2), padding=0)  # for R3
         elif self.sampling_rate == 500:  
-            self.raw_emg_plot.setYRange(-((2**14)/2), ((2**14)/2), padding=0)  # for R4
+            self.emg_plot.setYRange(-((2**14)/2), ((2**14)/2), padding=0)  # for R4
             self.envelope_plot.setYRange(0, ((2**14)/2), padding=0)  # for R4
 
         # Plot curves for EMG data and envelope
-        self.emg_curve = self.raw_emg_plot.plot(self.time_data, self.emg_data, pen=pg.mkPen('b', width=1))
+        self.emg_curve = self.emg_plot.plot(self.time_data, self.emg_data, pen=pg.mkPen('b', width=1))
         self.envelope_curve = self.envelope_plot.plot(self.time_data, self.emg_data, pen=pg.mkPen('r', width=2))
 
         # Timer for plot update

applications/eog.py

@@ -0,0 +1,86 @@
+import numpy as np
+from scipy.signal import butter, filtfilt
+from PyQt5.QtWidgets import QApplication, QVBoxLayout, QMainWindow, QWidget
+from pyqtgraph import PlotWidget
+import pyqtgraph as pg
+import pylsl
+import sys
+
+class EOGMonitor(QMainWindow):
+    def __init__(self): 
+        super().__init__()
+
+        self.setWindowTitle("Real-Time EOG Monitor - Eye Blink Detection")
+        self.setGeometry(100, 100, 800, 400)
+
+        # Create layout
+        layout = QVBoxLayout()
+
+        # Create plot widget for EOG
+        self.eog_plot = PlotWidget(self)
+        self.eog_plot.setBackground('w')
+        self.eog_plot.showGrid(x=True, y=True)
+        self.eog_plot.setTitle("Filtered EOG Signal (Low Pass: 10 Hz)")
+
+        # Add plot to layout
+        layout.addWidget(self.eog_plot)
+
+        # Central widget
+        central_widget = QWidget()
+        central_widget.setLayout(layout)
+        self.setCentralWidget(central_widget)
+
+        # Set up LSL stream inlet
+        streams = pylsl.resolve_stream('name', 'BioAmpDataStream')
+        if not streams:
+            print("No LSL stream found!")
+            sys.exit(0)
+        self.inlet = pylsl.StreamInlet(streams[0])
+
+        # Sampling rate
+        self.sampling_rate = int(self.inlet.info().nominal_srate())
+        print(f"Sampling rate: {self.sampling_rate} Hz")
+        
+        # Data and buffer
+        self.buffer_size = self.sampling_rate * 5  # 5 seconds buffer for recent data
+        self.eog_data = np.zeros(self.buffer_size)
+        self.time_data = np.linspace(0, 5, self.buffer_size)
+        self.current_index = 0
+
+        # Low-pass filter for EOG (10 Hz)
+        self.b, self.a = butter(4, 10.0 / (0.5 * self.sampling_rate), btype='low')
+
+        # Set fixed axis ranges
+        self.eog_plot.setXRange(0, 5, padding=0)
+        if self.sampling_rate == 250:
+            self.eog_plot.setYRange(-((2**10)/2), ((2**10)/2), padding=0)
+        elif self.sampling_rate == 500:
+            self.eog_plot.setYRange(-((2**14)/2), ((2**14)/2), padding=0)
+
+        # Plot curve for EOG data
+        self.eog_curve = self.eog_plot.plot(self.time_data, self.eog_data, pen=pg.mkPen('b', width=1))
+
+        # Timer for plot update
+        self.timer = pg.QtCore.QTimer()
+        self.timer.timeout.connect(self.update_plot)
+        self.timer.start(15)
+
+    def update_plot(self):
+        samples, _ = self.inlet.pull_chunk(timeout=0.0, max_samples=30)
+        if samples:
+            for sample in samples:
+                # Overwrite the oldest data point in the buffer
+                self.eog_data[self.current_index] = sample[0]
+                self.current_index = (self.current_index + 1) % self.buffer_size
+
+            # Apply only the low-pass filter to the EOG data
+            filtered_eog = filtfilt(self.b, self.a, self.eog_data)
+
+            # Update curve with the low-pass filtered EOG signal
+            self.eog_curve.setData(self.time_data, filtered_eog)
+
+if __name__ == "__main__":
+    app = QApplication(sys.argv)
+    window = EOGMonitor()
+    window.show()
+    sys.exit(app.exec_())

chords_requirements.txt

@@ -1,3 +1,3 @@
-numpy==1.26.4
+numpy==2.1.3
 pylsl==1.16.2
 pyserial==3.5