Applying Threshold = 0.7, Moving Average of last 10 values of powerData and Ball Speed reduced to 30

Author: PayalLakra

applications/game.py

@@ -39,7 +39,7 @@
 ball_radius = 20
 ball_color = WHITE
 ball_pos = [WIDTH // 2, HEIGHT // 2]
-ball_speed = 40
+ball_speed = 30
 
 # Player properties
 player_width = 10
@@ -56,6 +56,9 @@
 game_started = False
 first_attempt = True  # Keeps track if it's the first game or a restart
 
+powerData1 = []
+powerData2 = []
+
 def bandpower(data, sf, band, window_sec=None, relative=False):
     band = np.asarray(band)
     low, high = band
@@ -75,13 +78,13 @@ def bandpower(data, sf, band, window_sec=None, relative=False):
     return bp
 
 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])
-    channel_assignments = {0: 'Player A', 1: 'Player B'}
     sampling_frequency = 500
     bands = {'Alpha': [8, 13],'Beta': [13, 30]}
     buffer_length = sampling_frequency * 1
@@ -94,6 +97,9 @@ def eeg_data_thread(eeg_queue):
     baseline1 = baseline2 = 1  # Initialize baselines
 
     while running:
+        if paused:
+            time.sleep(0.1)  # Pause the thread when the game is paused
+            continue
         try:
             sample, timestamp = inlet.pull_sample()
             if len(sample) >= 6:
@@ -146,9 +152,11 @@ def eeg_data_thread(eeg_queue):
 eeg_thread.start()
 
 def reset_game():
-    global ball_pos, force_player1, force_player2, paused, game_started, win_text, win_handled, restart_clicked
+    global ball_pos, force_player1, force_player2, paused, game_started, win_text, win_handled, restart_clicked, powerData1, powerData2
     ball_pos = [WIDTH // 2, HEIGHT // 2]
     force_player1 = force_player2 = 0
+    powerData1.clear()  # Clear force data for player 1
+    powerData2.clear()  # Clear force data for player 2
     paused = False  # Ensure the game is not paused after reset
     game_started = True  # Ensure the game is marked as started
     win_text = None  # Reset win text
@@ -160,16 +168,24 @@ def reset_game():
         eeg_queue.get()
     print("Game Reset Successfully.")
 
-def update_ball_position(force_player1, force_player2):
-    global ball_pos
-    net_force = force_player1 - force_player2  # force direction
-    ball_pos[0] += net_force * ball_speed * 0.01
+def update_ball_position(force_player1, force_player2, threshold=0.7):
+    global ball_pos, powerData1, powerData2
+
+    # Apply moving average
+    average_force1 = np.mean(powerData1[-10:]) if len(powerData1) >= 10 else 0
+    average_force2 = np.mean(powerData2[-10:]) if len(powerData2) >= 10 else 0
+
+    net_force = average_force1 - average_force2
+
+    # Apply the threshold
+    if abs(net_force) > threshold:
+        ball_pos[0] += net_force * ball_speed * 0.01
     if ball_pos[0] < ball_radius:
         ball_pos[0] = ball_radius
     elif ball_pos[0] > WIDTH - ball_radius:
         ball_pos[0] = WIDTH - ball_radius
 
-    print(f"Force Player 1: {force_player1:.2f}, Force Player 2: {force_player2:.2f}, Net Force: {net_force:.2f}")  # Print the forces to the console
+    print(f"Force Player 1: {average_force1:.2f}, Force Player 2: {average_force2:.2f}, Net Force: {net_force:.2f}")
 
 def draw_buttons(paused, first_attempt):  # Button dimensions and positions
     button_width = 120
@@ -248,12 +264,13 @@ def main():
                     if pygame.Rect(WIDTH // 4 - 60, HEIGHT - 80, 120, 40).collidepoint(mouse_pos):
                         # Start or restart the game
                         reset_game()
+                        time.sleep(1)
                         game_started = True
                         first_attempt = False
                         win_text = None
                         win_handled = False  # Reset win handling
                         paused = False  # Ensure the game is unpaused
-                        print("Gam e Restarted.")
+                        print("Game Restarted.")
                     elif pygame.Rect(WIDTH // 2 - 60, HEIGHT - 80, 120, 40).collidepoint(mouse_pos):
                         # Pause/Resume the game
                         paused = not paused
@@ -292,4 +309,8 @@ def main():
         clock.tick(60)  # 60 frames per second
 
 if __name__ == "__main__":
-    main()
+    main()
+
+# Threshold = 0.7
+# Moving Average of last 10 values of powerData.
+# Ball Speed = 30