-
Notifications
You must be signed in to change notification settings - Fork 0
/
Copy pathutils.py
20 lines (18 loc) · 984 Bytes
/
utils.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
from psychopy import visual, core, event
import pandas as pd
def check_for_esc(win):
"""Check if Esc key is pressed and quit if detected."""
if "escape" in event.getKeys(keyList=["escape"]):
win.close() # Close the window
core.quit() # Exit the task immediately
def get_random_walk_path(subject_number, session_number):
if subject_number % 2 == 0 and session_number == 1:
return pd.read_csv("random_walk/experiment/random_walk1.csv",header=None)
elif subject_number % 2 == 0 and session_number == 2:
return pd.read_csv("random_walk/experiment/random_walk2.csv",header=None)
elif subject_number % 2 != 0 and session_number == 1:
return pd.read_csv("random_walk/experiment/random_walk2.csv",header=None)
elif subject_number % 2 != 0 and session_number == 2:
return pd.read_csv("random_walk/experiment/random_walk1.csv",header=None)
else:
raise ValueError("Invalid session_number or subject_number")