Now, it auto detect the available LSL Stream and connect to it whether it is BioAmpDataStream or ORIC

Author: PayalLakra

beetle.py

@@ -2,18 +2,31 @@
 import pylsl
 import numpy as np
 import time
-from pylsl import StreamInlet, resolve_stream
+from pylsl import StreamInlet, resolve_streams, resolve_byprop
 from scipy.signal import iirnotch, butter, lfilter
 import math
 
 # Initialize LSL stream
-streams = resolve_stream('name', 'BioAmpDataStream')
-if not streams:
-    print("No LSL stream found!")
-    exit()
+print("Searching for available LSL streams...")
+streams = resolve_streams()
+available_streams = [s.name() for s in streams]
+
+if not available_streams:
+    print("No LSL streams found!")
+
+for stream_name in available_streams:
+    print(f"Trying to connect to {stream_name}...")
+    resolved_streams = resolve_byprop('name', stream_name, timeout=2)
 
-inlet = StreamInlet(streams[0])
-print("LSL Stream Started")
+    if resolved_streams:
+        print(f"Successfully connected to {stream_name}!")
+        inlet = StreamInlet(resolved_streams[0])
+        break
+    else:
+        print(f"Failed to connect to {stream_name}.")
+
+if inlet is None:
+    print("Could not connect to any stream.")
 sampling_rate = int(inlet.info().nominal_srate())
 print(f"Sampling rate: {sampling_rate} Hz")
 
@@ -26,7 +39,7 @@
 
 # Beetle properties
 beetle_x, beetle_y = 380, 530
-focus_speed_upward = 15
+focus_speed_upward = 10
 focus_speed_downward = 5
 focus_timeout = 2
 focus_threshold = None
@@ -36,14 +49,19 @@
 screen = pygame.display.set_mode((800, 600))
 pygame.display.set_caption('Beetle Game')
 
-# Load beetle image
-beetle_image = pygame.image.load('media\\beetle.jpg')
-beetle_image = pygame.transform.scale(beetle_image, (80, 80))
+sprite_count = 10
+beetle_sprites = [pygame.image.load(f'media/beetle{i}.JPG') for i in range(1, sprite_count + 1)]
+beetle_sprites = [pygame.transform.scale(sprite, (140, 160)) for sprite in beetle_sprites]
+
+# Animation Variables
+current_sprite = 0
+animation_speed = 100
+sprite_timer = 0
 
 # Function to display a message on the screen
 def show_message(message, duration=3):
     start_time = time.time()
-    font = pygame.font.Font(None, 50)
+    font = pygame.font.SysFont("Arial", 50)
     text = font.render(message, True, (0, 0, 0))
     text_rect = text.get_rect(center=(400, 300))
 
@@ -72,7 +90,6 @@ def calculate_focus_level(eeg_data, sampling_rate=500):
     gamma_power = math.sqrt(np.sum((fft_data[(freqs >= 30) & (freqs <= 45)]) ** 2))
 
     power = (beta_power + gamma_power) / (delta_power + theta_power + alpha_power + beta_power + gamma_power)
-    print(power)
     return power
 
 # Calibration Phase
@@ -97,21 +114,23 @@ def calculate_focus_level(eeg_data, sampling_rate=500):
     std_focus = np.std(baseline_focus_levels)
 
     focus_threshold = mean_focus + 1.5 * std_focus  
-    print(f"Calibration Complete. Focus Threshold set at: {focus_threshold:.2f}")
+    print(f"Calibration Complete. Focus Threshold set at: {focus_threshold:.3f}")
 else:
     print("Calibration failed due to insufficient data.")
     exit()
 
 # Show Game Start Message
 show_message("Game Starting...", 1)
+game_start_time = time.time()
+game_duration = 45  # Game lasts 45 seconds
 
 # Update beetle position
 def update_beetle_position(focus_level, is_focus_stable):
     global beetle_y
     if is_focus_stable:
-        beetle_y = max(10 + beetle_image.get_height() // 2, beetle_y - focus_speed_upward)     
+        beetle_y = max(10 + beetle_sprites[0].get_height() // 2, beetle_y - focus_speed_upward)     
     else:
-        beetle_y = min(580 - beetle_image.get_height() // 2, beetle_y + focus_speed_downward)  
+        beetle_y = min(580 - beetle_sprites[0].get_height() // 2, beetle_y + focus_speed_downward)  
 
 print("STARTING GAME...")
 running = True
@@ -131,6 +150,8 @@ def update_beetle_position(focus_level, is_focus_stable):
             buffer.append(filtered_sample)
 
         current_time = time.time()
+        elapsed_time = int(current_time - game_start_time)
+
         if current_time - last_time >= 1:
             last_time = current_time
             buffer = buffer[int(len(buffer) * 0.2):]
@@ -140,6 +161,7 @@ def update_beetle_position(focus_level, is_focus_stable):
             buffer = []
 
             focus_level = calculate_focus_level(eeg_data)
+            print(focus_level)
 
             if focus_level > focus_threshold:
                 focus_timer = min(focus_timeout, focus_timer + (current_time - last_focus_time))
@@ -152,10 +174,32 @@ def update_beetle_position(focus_level, is_focus_stable):
 
             last_focus_time = current_time
 
+        # Update sprite animation
+        sprite_timer += (current_time - last_focus_time)
+        if sprite_timer >= animation_speed:
+            sprite_timer = 0
+            current_sprite = (current_sprite + 1) % len(beetle_sprites)
+
+        # Check win condition
+        if beetle_y <= 80:
+            show_message("You Win!", 1)
+            running = False
+
+        # Check game over condition
+        if elapsed_time >= game_duration:
+            show_message("Game Over! Try Again.", 1)
+            running = False
+
+        # Draw everything
         screen.fill("#FFFFFF")
         pygame.draw.rect(screen, (0, 0, 0), (10, 10, 780, 580), 5)  
-        screen.blit(beetle_image, (beetle_x - beetle_image.get_width() // 2, beetle_y - beetle_image.get_height() // 2))
-        
+        screen.blit(beetle_sprites[current_sprite], (beetle_x - 40, beetle_y - 40))
+
+        # Display Timer
+        font = pygame.font.SysFont("Arial", 30)
+        timer_text = font.render(f"Time: {game_duration - elapsed_time}s", True, (0, 0, 0))
+        screen.blit(timer_text, (650, 20))  
+
         pygame.display.update()
 
     except KeyboardInterrupt:

emgenvelope.py

@@ -39,11 +39,25 @@ def __init__(self):
         self.setCentralWidget(central_widget)
 
         # Set up LSL stream inlet
-        streams = pylsl.resolve_stream('name', 'BioAmpDataStream')
-        if not streams:
-            print("No LSL stream found!")
+        print("Searching for available LSL streams...")
+        available_streams = pylsl.resolve_streams()
+
+        if not available_streams:
+            print("No LSL streams found! Exiting...")
+            sys.exit(0)
+
+        self.inlet = None
+        for stream in available_streams:
+            try:
+                self.inlet = pylsl.StreamInlet(stream)
+                print(f"Connected to LSL stream: {stream.name()}")
+                break
+            except Exception as e:
+                print(f"Failed to connect to {stream.name()}: {e}")
+
+        if self.inlet is None:
+            print("Unable to connect to any LSL stream! Exiting...")
             sys.exit(0)
-        self.inlet = pylsl.StreamInlet(streams[0])
 
         # Sampling rate
         self.sampling_rate = int(self.inlet.info().nominal_srate())

eog.py

@@ -44,11 +44,25 @@ def __init__(self):
         layout.addWidget(self.blink_plot)
 
         # Set up LSL stream inlet
-        streams = pylsl.resolve_stream('name', 'BioAmpDataStream')
-        if not streams:
-            print("No LSL stream found!")
+        print("Searching for available LSL streams...")
+        available_streams = pylsl.resolve_streams()
+
+        if not available_streams:
+            print("No LSL streams found! Exiting...")
+            sys.exit(0)
+
+        self.inlet = None
+        for stream in available_streams:
+            try:
+                self.inlet = pylsl.StreamInlet(stream)
+                print(f"Connected to LSL stream: {stream.name()}")
+                break
+            except Exception as e:
+                print(f"Failed to connect to {stream.name()}: {e}")
+
+        if self.inlet is None:
+            print("Unable to connect to any LSL stream! Exiting...")
             sys.exit(0)
-        self.inlet = pylsl.StreamInlet(streams[0])
 
         self.sampling_rate = int(self.inlet.info().nominal_srate())
         print(f"Sampling rate: {self.sampling_rate} Hz")

ffteeg.py

@@ -63,11 +63,25 @@ def __init__(self):
         self.setCentralWidget(self.central_widget)
 
         # Set up LSL stream inlet
-        streams = pylsl.resolve_stream('name', 'BioAmpDataStream')
-        if not streams:
-            print("No LSL stream found!")
+        print("Searching for available LSL streams...")
+        available_streams = pylsl.resolve_streams()
+
+        if not available_streams:
+            print("No LSL streams found! Exiting...")
+            sys.exit(0)
+
+        self.inlet = None
+        for stream in available_streams:
+            try:
+                self.inlet = pylsl.StreamInlet(stream)
+                print(f"Connected to LSL stream: {stream.name()}")
+                break
+            except Exception as e:
+                print(f"Failed to connect to {stream.name()}: {e}")
+
+        if self.inlet is None:
+            print("Unable to connect to any LSL stream! Exiting...")
             sys.exit(0)
-        self.inlet = pylsl.StreamInlet(streams[0])
 
         # Sampling rate
         self.sampling_rate = int(self.inlet.info().nominal_srate())

game.py

@@ -2,7 +2,7 @@
 import sys
 import queue
 import threading
-from pylsl import StreamInlet, resolve_stream
+from pylsl import StreamInlet, resolve_streams, resolve_byprop
 import numpy as np
 from scipy.signal import welch
 from scipy.integrate import simpson
@@ -89,12 +89,28 @@ def bandpower(data, sf, band, window_sec=None, relative=False):
 
 def eeg_data_thread(eeg_queue):
     global powerData1, powerData2
-    streams = resolve_stream('name', 'BioAmpDataStream')
-    if not streams:
-        print("No LSL stream found!")
-        return
-
-    inlet = StreamInlet(streams[0])
+    print("Searching for available LSL streams...")
+    streams = resolve_streams()
+    available_streams = [s.name() for s in streams]
+
+    if not available_streams:
+        print("No LSL streams found!")
+        return None
+
+    for stream_name in available_streams:
+        print(f"Trying to connect to {stream_name}...")
+        resolved_streams = resolve_byprop('name', stream_name, timeout=2)
+
+        if resolved_streams:
+            print(f"Successfully connected to {stream_name}!")
+            inlet = StreamInlet(resolved_streams[0])
+            break
+        else:
+            print(f"Failed to connect to {stream_name}.")
+
+    if inlet is None:
+        print("Could not connect to any stream.")
+        return None
     channel_assignments = {0:'Player A', 1:'Player B'}
     sampling_frequency = 500
     bands = {'Alpha': [8, 13],'Beta': [13, 30]}

gui.py

@@ -1,5 +1,5 @@
 import sys
-from pylsl import StreamInlet, resolve_stream
+from pylsl import StreamInlet, resolve_streams, resolve_byprop
 import pyqtgraph as pg  # For real-time plotting
 from pyqtgraph.Qt import QtWidgets, QtCore  # PyQt components for GUI
 import numpy as np
@@ -28,16 +28,30 @@ def update_plots():
 
 def plot_lsl_data():
     global inlet, num_channels, data
-    print("Looking for LSL Stream.")
-    streams = resolve_stream('name', 'BioAmpDataStream')  
 
-    if not streams:
-        print("No LSL Stream found.")
-        return
-    
-    inlet = StreamInlet(streams[0])
+    print("Searching for available LSL streams...")
+    streams = resolve_streams()
+    available_streams = [s.name() for s in streams]
+
+    if not available_streams:
+        print("No LSL streams found!")
+        return None
+
+    for stream_name in available_streams:
+        print(f"Trying to connect to {stream_name}...")
+        resolved_streams = resolve_byprop('name', stream_name, timeout=2)
+
+        if resolved_streams:
+            print(f"Successfully connected to {stream_name}!")
+            inlet = StreamInlet(resolved_streams[0])
+            break
+        else:
+            print(f"Failed to connect to {stream_name}.")
+
+    if inlet is None:
+        print("Could not connect to any stream.")
+        return None
 
-    # Get the number of channels from the stream
     info = inlet.info()
     num_channels = info.channel_count()
     print(f"Detected {num_channels} channels.")
@@ -94,4 +108,5 @@ def init_gui():
 
 if __name__ == "__main__":
     plot_lsl_data()
-    sys.exit(app.exec_())  # Start the Qt application
+    if inlet:
+        sys.exit(app.exec_())  # Start the Qt application only if a stream was connected

heartbeat_ecg.py

@@ -33,11 +33,25 @@ def __init__(self):
         self.setCentralWidget(central_widget)
 
         # Set up LSL stream inlet
-        streams = pylsl.resolve_stream('name', 'BioAmpDataStream')
-        if not streams:
-            print("No LSL stream found!")
+        print("Searching for available LSL streams...")
+        available_streams = pylsl.resolve_streams()
+
+        if not available_streams:
+            print("No LSL streams found! Exiting...")
+            sys.exit(0)
+
+        self.inlet = None
+        for stream in available_streams:
+            try:
+                self.inlet = pylsl.StreamInlet(stream)
+                print(f"Connected to LSL stream: {stream.name()}")
+                break
+            except Exception as e:
+                print(f"Failed to connect to {stream.name()}: {e}")
+
+        if self.inlet is None:
+            print("Unable to connect to any LSL stream! Exiting...")
             sys.exit(0)
-        self.inlet = pylsl.StreamInlet(streams[0])
 
         # Sampling rate
         self.sampling_rate = int(self.inlet.info().nominal_srate())