Skip to content

Instantly share code, notes, and snippets.

@mathigatti
Last active June 5, 2022 00:40
Show Gist options
  • Save mathigatti/3635a6414118e34fa90786fb67b6b7ea to your computer and use it in GitHub Desktop.
Save mathigatti/3635a6414118e34fa90786fb67b6b7ea to your computer and use it in GitHub Desktop.
Psychopy example
from psychopy import visual, core, event
import datetime
import pandas as pd
# Colours
gray = '#969696'
black = '#000000'
white = '#FFFFFF'
# Window parameters
resolution = [300, 300]
def loadInstructionsAndFlip(win):
background = visual.Rect(win, width=resolution[0]+10, height=resolution[1]+10, fillColor=black, fillColorSpace='hex')
msg1 = visual.TextStim(win, text="press [ q ] to exit", pos=(0.0,(-resolution[1]*0.10)), color=white, colorSpace='hex')
msg2 = visual.TextStim(win, text="press [ n ] to continue", color=white, colorSpace='hex',alignHoriz='center', alignVert='center')
background.draw()
msg1.draw()
msg2.draw()
# Elements are only displayed after the flip command is executed
win.flip()
def loadStartScreenAndFlip(win):
background = visual.Rect(win, width=resolution[0]+10, height=resolution[1]+10, fillColor=gray, fillColorSpace='hex')
msg1 = visual.TextStim(win, text="press any key to start", color=white, colorSpace='hex')
background.draw()
msg1.draw()
# Elements are only displayed after the flip command is executed
win.flip()
def startScreensAndRecordData(win):
clock = core.Clock()
win.clearBuffer()
data = []
loadStartScreenAndFlip(win)
event.waitKeys()
while True:
loadInstructionsAndFlip(win)
clock.reset()
keys = event.waitKeys(keyList=["n","q"])
for key in keys:
time = clock.getTime()
print("You pressed the {} key on {} seconds".format(key,round(time,3)))
data.append([key,time])
if key == "q":
return data
else:
loadStartScreenAndFlip(win)
event.waitKeys()
def window(resolution):
fullScreen = False
win = visual.Window(resolution,units="pix", color=gray, colorSpace='hex', fullscr=fullScreen, monitor = "testMonitor")
win.setMouseVisible(False)
return win
def main():
win = window(resolution)
data = startScreensAndRecordData(win)
pd.DataFrame(data,columns=["Key","Time"]).to_csv('experiment_' + str(datetime.date.today()) + '.csv')
print("Experiment saved as:",'experiment_' + str(datetime.date.today()) + '.csv')
if __name__ == "__main__":
main()
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment