#! /usr/bin/env python
# Time-stamp: <2021-12-08 12:33:56 christophe@pallier.org>
"""This is a simple lexical decision experiment.

At each trial, an item is displayed and the participant must press the key 'f' if it is a word,
'j' if it is a non-word.
"""

import sys
import random
import pandas
from expyriment import design, control, stimuli

WORD_RESPONSE_KEY = 'f'
NONW_RESPONSE_KEY = 'j'
MAX_RESPONSE_DELAY = 2000

if len(sys.argv) < 2:
    print("""Usage: [python] lexdec_v3.py CSVFILE

where CSVFILE is a comma-separated file with two columns:
    - `item` (containing a word or a pseudoword)
    - `category` ('P' or 'W')
    """)
    sys.exit(1)

stim_file = sys.argv[1]

exp = design.Experiment(name="Lexical Decision", text_size=40)
control.initialize(exp)

# prepare the stimuli
materials = pandas.read_csv(stim_file)
trials = [(row['category'], row['item'], stimuli.TextLine(row['item']))
          for _, row in materials.iterrows()]

random.shuffle(trials)

cue = stimuli.FixCross(size=(50, 50), line_width=4)
blankscreen = stimuli.BlankScreen()
instructions = stimuli.TextScreen("Instructions",
    f"""When you'll see a stimulus, your task to decide, as quickly as possible, whether it is a word or not.

    if it is a word, press '{WORD_RESPONSE_KEY.upper()}'

    if it is a non-word, press '{NONW_RESPONSE_KEY.upper()}'

    Press the space bar to start.""")


exp.add_data_variable_names(['cat', 'word', 'respkey', 'RT'])

control.start(skip_ready_screen=True)
instructions.present()
exp.keyboard.wait_char(" ")

for category, item, stim in trials:
    blankscreen.present()
    exp.clock.wait(1000)
    cue.present()
    exp.clock.wait(500)
    stim.present()
    key, rt = exp.keyboard.wait_char([WORD_RESPONSE_KEY, NONW_RESPONSE_KEY],
                                     duration=MAX_RESPONSE_DELAY)
    exp.data.add([category, item, key, rt])

control.end()