def test_randomizeBlocks_3(self):
check = randomizeBlocks(3.0, 1.0, 2.0, 3.0)
res = ((check == [1.0, 2.0, 3.0]) or (check == [2.0, 1.0, 3.0])
or (check == [3.0, 2.0, 1.0]) or (check == [3.0, 1.0, 2.0])
or (check == [2.0, 3.0, 1.0]) or (check == [1.0, 3.0, 2.0]))
self.assertEqual(res, 1)
def test_randomizeBlocks_1(self):
check =randomizeBlocks(1.0,1.0,2.0,3.0)
self.assertEqual(check, [1.0])
def test_randomizeBlocks_2(self):
check =randomizeBlocks(2.0,1.0,2.0,3.0)
res = ((check ==[1.0,2.0]) or (check == [2.0,1.0]))
self.assertEqual(res,1)
def test_randomizeBlocks_1(self):
check = randomizeBlocks(1.0, 1.0, 2.0, 3.0)
self.assertEqual(check, [1.0])
def run(self, *args):
test = False
if args[1] == 1:
test = True
#### READ CONFIGURATIONS (from config.txt) ####
with open(args[0], 'r') as f:
config_text = f.read()
config_text = config_text.replace(' ', '')
# Experiment Settings #
experiment_name = re.search('experiment_name.+?\n',
config_text).group(0)[16:-1]
participant_id = re.search('participant_id.+?\n',
config_text).group(0)[15:-1]
version = re.search('version.+?\n', config_text).group(0)[8:-1]
file_name = re.search('file_name.+?\n', config_text).group(0)[10:-1]
# Trial Settings #
random_modes = re.search('random_modes.+?\n',
config_text).group(0)[13:-1].lower() == "true"
num_random = int(
re.search('num_random.+?\n', config_text).group(0)[11:-1])
centered = re.search('centered.+?\n',
config_text).group(0)[9:-1].lower() == "true"
num_reps_motor = int(
re.search('num_reps_motor.+?\n', config_text).group(0)[15:-1])
num_blocks_motor = (re.search('num_blocks_motor.+?\n',
config_text).group(0)[17:-1]).split(',')
random_blocks_motor = re.search(
'random_blocks_motor.+?\n',
config_text).group(0)[20:-1].lower() == "true"
num_reps_keyboard = int(
re.search('num_reps_keyboard.+?\n', config_text).group(0)[18:-1])
num_blocks_keyboard = (re.search(
'num_blocks_keyboard.+?\n',
config_text).group(0)[20:-1]).split(',')
random_blocks_keyboard = re.search(
'random_blocks_keyboard.+?\n',
config_text).group(0)[23:-1].lower() == "true"
num_reps_speech = int(
re.search('num_reps_speech.+?\n', config_text).group(0)[16:-1])
num_blocks_speech = (re.search('num_blocks_speech.+?\n',
config_text).group(0)[18:-1]).split(',')
random_blocks_speech = re.search(
'random_blocks_speech.+?\n',
config_text).group(0)[21:-1].lower() == "true"
num_reps_eye = int(
re.search('num_reps_eye.+?\n', config_text).group(0)[13:-1])
num_blocks_eye = (re.search('num_blocks_eye.+?\n',
config_text).group(0)[15:-1]).split(',')
random_blocks_eye = re.search(
'random_blocks_eye.+?\n',
config_text).group(0)[18:-1].lower() == "true"
wait_time = int(
re.search('wait_time.+?\n', config_text).group(0)[10:-1])
warning_time = int(
re.search('warning_time.+?\n', config_text).group(0)[13:-1])
# Display Settings #
windowsizex = int(
re.search('windowsizex.+?\n', config_text).group(0)[12:-1])
windowsizey = int(
re.search('windowsizey.+?\n', config_text).group(0)[12:-1])
full_screen = re.search('full_screen.+?\n',
config_text).group(0)[12:-1].lower() == "true"
background_color = re.search('background_color.+?\n',
config_text).group(0)[17:-1]
text_color = re.search('text_color.+?\n', config_text).group(0)[11:-1]
shape1color = re.search('shape1color.+?\n',
config_text).group(0)[12:-1]
shape2color = re.search('shape2color.+?\n',
config_text).group(0)[12:-1]
shape3color = re.search('shape3color.+?$', config_text).group(0)[12:]
# Initialize Window #
window = visual.Window(size=(windowsizex, windowsizey),
units='norm',
color=background_color,
colorSpace='rgb',
fullscr=full_screen,
allowGUI=False)
# Device and Event Set Up #
clock = core.Clock()
display = self.devices.display
keys = self.devices.keyboard
mouse = evt.Mouse(win=window)
mouse.getPos()
# todo: implement serial output for sync; should it be 1-5 (or such) repeated to prevent overlap?
ser = serial.Serial('COM4', 9600)
# ser = -1
# Instructions and Button #
title_label = visual.TextStim(
window,
units='norm',
text=u'Remember the sequence of colored blocks.',
pos=[0, 0.2],
height=0.1,
color=text_color,
colorSpace='rgb',
alignHoriz='center',
alignVert='center')
continue_label = visual.TextStim(window,
units='norm',
text=u'Continue',
pos=[0, -0.6],
height=0.1,
color=text_color,
colorSpace='rgb',
alignHoriz='center',
alignVert='center')
continue_button = visual.Rect(window,
width=0.5,
height=0.2,
lineColor=(0, 0, 0),
lineWidth=2,
lineColorSpace='rgb',
pos=(0, -0.6))
# Text Cues #
next_label = visual.TextStim(window,
units='norm',
text=u'New Round',
pos=[0, 0],
height=0.1,
color=text_color,
colorSpace='rgb',
alignHoriz='center',
alignVert='center')
end_label = visual.TextStim(window,
units='norm',
text=u'Thank you!\nExiting...',
pos=[0, 0],
height=0.1,
color=text_color,
colorSpace='rgb',
alignHoriz='center',
alignVert='center')
# Color Blocks #
if centered:
rect_stim1 = visual.Rect(window,
width=0.5,
height=0.5,
lineColor=(1, 1, 1),
fillColor=shape1color,
fillColorSpace='rgb',
pos=(0, 0))
rect_stim2 = visual.Rect(window,
width=0.5,
height=0.5,
lineColor=(1, 1, 1),
fillColor=shape2color,
fillColorSpace='rgb',
pos=(0, 0))
rect_stim3 = visual.Rect(window,
width=0.5,
height=0.5,
lineColor=(1, 1, 1),
fillColor=shape3color,
fillColorSpace='rgb',
pos=(0, 0))
else:
rect_stim1 = visual.Rect(window,
width=0.5,
height=0.5,
lineColor=(1, 1, 1),
fillColor=shape1color,
fillColorSpace='rgb',
pos=(-0.5, 0.5))
rect_stim2 = visual.Rect(window,
width=0.5,
height=0.5,
lineColor=(1, 1, 1),
fillColor=shape2color,
fillColorSpace='rgb',
pos=(0.5, -0.5))
rect_stim3 = visual.Rect(window,
width=0.5,
height=0.5,
lineColor=(1, 1, .1),
fillColor=shape3color,
fillColorSpace='rgb',
pos=(0.5, 0.5))
# Clear all events from the global and device level ioHub Event Buffers.
self.hub.clearEvents('all')
# pyaud = pyaudio.PyAudio()
# t = np.linspace(0, 1, 48000, endpoint=False)
# audsig = signal.square(20*np.pi*5*t).astype(np.float32)
# stream = pyaud.open(format=pyaudio.paFloat32, channels=1, rate=48000, output=True)
# stream.write(0.5*audsig)
#### INITIAL STARTER SCREEN ####
# Initial Instruction Page #
if test == False:
title_label.draw()
continue_label.draw()
continue_button.draw()
window.flip()
self.hub.clearEvents('all')
FINISH_INSTR = False
while FINISH_INSTR is False:
buttons = mouse.getPressed()
if buttons[0]:
if mouse.isPressedIn(continue_button, buttons=[0]):
break
#### TRIALS OF BLOCK GAME ####
if test == True:
file_name = '%s/%s' % ('p' + (participant_id), file_name)
exp = data.ExperimentHandler(name=experiment_name,
version=version,
extraInfo={'participant': participant_id},
runtimeInfo=None,
originPath=None,
savePickle=True,
saveWideText=True,
dataFileName=file_name)
# Non-Randomized Mode #
if random_modes is False:
# if not randomized between modes, then go through each and run its respective number of trials
for trial_type in range(4):
if trial_type == 0:
name = "motor"
num_reps = num_reps_motor
temp_blocks = num_blocks_motor
random_blocks_var = random_blocks_motor
elif trial_type == 1:
name = "speech"
num_reps = num_reps_speech
temp_blocks = num_blocks_speech
random_blocks_var = random_blocks_speech
elif trial_type == 2:
name = "eye"
num_reps = num_reps_eye
temp_blocks = num_blocks_eye
random_blocks_var = random_blocks_eye
elif trial_type == 3:
name = "keyboard"
num_reps = num_reps_keyboard
temp_blocks = num_blocks_keyboard
random_blocks_var = random_blocks_keyboard
for num in range(num_reps):
# assign the random variables
if random_blocks_var:
num_blocks = random.randint(1, 3)
else:
if len(temp_blocks) == 1:
num_blocks = temp_blocks[0]
else:
num_blocks = random.sample(temp_blocks, 1)[0]
# begin trial set up
status = 1
trial_loop = data.TrialHandler(trialList=[],
nReps=1,
name=name)
exp.addLoop(trial_loop)
if status != -1:
# restart values and indicate new round
shapes = [rect_stim1, rect_stim2, rect_stim3]
helper.resetTrial(shapes, centered)
# randomize block order and begin new round
shapes = helper.randomizeBlocks(
num_blocks, rect_stim1, rect_stim2, rect_stim3)
if len(shapes) == 1:
shapes[0].setPos((0.0))
# data columns are standardized across modalities by adding the same data to every trial type
helper.addTrialData(shapes, trial_type, num_blocks,
exp)
# track types of trial
if trial_type == 0:
status = motor.trial(self, clock, window, shapes,
mouse, text_color, centered,
wait_time, warning_time, exp,
num, ser)
elif trial_type == 1:
status = speech.trial(self, clock, window, shapes,
mouse, keys, text_color,
wait_time, warning_time, exp,
num, ser)
elif trial_type == 2:
status = eye.trial(self, clock, window, shapes,
text_color, centered, wait_time,
warning_time, exp, num, ser)
elif trial_type == 3:
status = keyboard.trial(self, clock, window,
shapes, keys, text_color,
centered, wait_time,
warning_time, exp, num,
ser)
# always add shape colors since they will be relevant in every modality
exp.addData('correct', status)
exp.nextEntry()
# Randomized Mode #
else:
# randomize everything, except for total number of trials
# set the total counts of each trial type to 0
motor_count = 0
eye_count = 0
speech_count = 0
keyboard_count = 0
for num in range(num_random):
# check for disabled trials and choose a random trial type out of the enabled types
trial_array = []
if num_reps_motor != -1:
trial_array.append(0)
if num_reps_speech != -1:
trial_array.append(1)
if num_reps_eye != -1:
trial_array.append(2)
if num_reps_keyboard != -1:
trial_array.append(3)
if len(trial_array) == 0:
print "No trial modalities enabled."
break
trial_type = random.sample(trial_array, 1)[0]
num_blocks = random.randint(1, 3)
# or use: num_blocks = random.sample([1,2,3], 1)[0]
if trial_type == 0:
name = "motor"
elif trial_type == 1:
name = "speech"
elif trial_type == 2:
name = "eye"
elif trial_type == 3:
name = "keyboard"
status = 1
trial_loop = data.TrialHandler(trialList=[],
nReps=1,
name=name)
exp.addLoop(trial_loop)
if status != -1:
# restart values and indicate new round
shapes = [rect_stim1, rect_stim2, rect_stim3]
helper.resetTrial(shapes, centered)
# randomize block order and begin new round
shapes = helper.randomizeBlocks(num_blocks, rect_stim1,
rect_stim2, rect_stim3)
if len(shapes) == 1:
shapes[0].setPos((0.0))
# data columns are standardized across modalities
helper.addTrialData(shapes, trial_type, num_blocks, exp)
# track types of trial
if trial_type == 0:
status = motor.trial(self, clock, window, shapes,
mouse, text_color, centered,
wait_time, warning_time, exp,
motor_count, ser)
motor_count += 1
elif trial_type == 1:
status = speech.trial(self, clock, window, shapes,
mouse, keys, text_color,
wait_time, warning_time, exp,
speech_count, ser)
speech_count += 1
elif trial_type == 2:
status = eye.trial(self, clock, window, shapes,
text_color, centered, wait_time,
warning_time, exp, eye_count, ser)
eye_count += 1
elif trial_type == 3:
status = keyboard.trial(self, clock, window, shapes,
keys, text_color, centered,
wait_time, warning_time, exp,
keyboard_count, ser)
keyboard_count += 1
# always add shape colors since they will be relevant in every modality
exp.addData('correct', status)
exp.nextEntry()
#### END MESSAGE ####
# if trial loop is finished successfully, then end the program
for frameN in range(75):
end_label.draw()
window.flip()
def run(self, *args):
test = False
if args[1] == 1:
test = True
#### READ CONFIGURATIONS (from config.txt) ####
with open(args[0], 'r') as f:
config_text = f.read()
config_text = config_text.replace(' ', '')
# Experiment Settings #
experiment_name = re.search('experiment_name.+?\n', config_text).group(0)[16:-1]
participant_id = re.search('participant_id.+?\n', config_text).group(0)[15:-1]
version = re.search('version.+?\n', config_text).group(0)[8:-1]
file_name = re.search('file_name.+?\n', config_text).group(0)[10:-1]
# Trial Settings #
random_modes = re.search('random_modes.+?\n', config_text).group(0)[13:-1].lower() == "true"
num_random = int(re.search('num_random.+?\n', config_text).group(0)[11:-1])
centered = re.search('centered.+?\n', config_text).group(0)[9:-1].lower() == "true"
num_reps_motor = int(re.search('num_reps_motor.+?\n', config_text).group(0)[15:-1])
num_blocks_motor = (re.search('num_blocks_motor.+?\n', config_text).group(0)[17:-1]).split(',')
random_blocks_motor = re.search('random_blocks_motor.+?\n', config_text).group(0)[20:-1].lower() == "true"
num_reps_keyboard = int(re.search('num_reps_keyboard.+?\n', config_text).group(0)[18:-1])
num_blocks_keyboard = (re.search('num_blocks_keyboard.+?\n', config_text).group(0)[20:-1]).split(',')
random_blocks_keyboard = re.search('random_blocks_keyboard.+?\n', config_text).group(0)[23:-1].lower() == "true"
num_reps_speech = int(re.search('num_reps_speech.+?\n', config_text).group(0)[16:-1])
num_blocks_speech = (re.search('num_blocks_speech.+?\n', config_text).group(0)[18:-1]).split(',')
random_blocks_speech = re.search('random_blocks_speech.+?\n', config_text).group(0)[21:-1].lower() == "true"
num_reps_eye = int(re.search('num_reps_eye.+?\n', config_text).group(0)[13:-1])
num_blocks_eye = (re.search('num_blocks_eye.+?\n', config_text).group(0)[15:-1]).split(',')
random_blocks_eye = re.search('random_blocks_eye.+?\n', config_text).group(0)[18:-1].lower() == "true"
wait_time = int(re.search('wait_time.+?\n', config_text).group(0)[10:-1])
warning_time = int(re.search('warning_time.+?\n', config_text).group(0)[13:-1])
# Display Settings #
windowsizex = int(re.search('windowsizex.+?\n', config_text).group(0)[12:-1])
windowsizey = int(re.search('windowsizey.+?\n', config_text).group(0)[12:-1])
full_screen = re.search('full_screen.+?\n', config_text).group(0)[12:-1].lower() == "true"
background_color = re.search('background_color.+?\n', config_text).group(0)[17:-1]
text_color = re.search('text_color.+?\n', config_text).group(0)[11:-1]
shape1color = re.search('shape1color.+?\n', config_text).group(0)[12:-1]
shape2color = re.search('shape2color.+?\n', config_text).group(0)[12:-1]
shape3color = re.search('shape3color.+?$', config_text).group(0)[12:]
# Initialize Window #
window=visual.Window(size=(windowsizex,windowsizey), units='norm', color=background_color, colorSpace='rgb',
fullscr=full_screen, allowGUI=False)
# Device and Event Set Up #
clock = core.Clock()
display = self.devices.display
keys = self.devices.keyboard
mouse = evt.Mouse(win=window)
mouse.getPos()
# todo: implement serial output for sync; should it be 1-5 (or such) repeated to prevent overlap?
ser = serial.Serial('COM4', 9600)
# ser = -1
# Instructions and Button #
title_label = visual.TextStim(window, units='norm', text=u'Remember the sequence of colored blocks.',
pos=[0,0.2], height=0.1, color=text_color, colorSpace='rgb',alignHoriz='center',
alignVert='center')
continue_label = visual.TextStim(window, units='norm', text=u'Continue',
pos=[0,-0.6], height=0.1, color=text_color, colorSpace='rgb',alignHoriz='center',
alignVert='center')
continue_button = visual.Rect(window, width=0.5, height=0.2, lineColor=(0, 0, 0), lineWidth=2,
lineColorSpace='rgb', pos=(0, -0.6))
# Text Cues #
next_label = visual.TextStim(window, units='norm', text=u'New Round', pos=[0,0], height=0.1, color=text_color,
colorSpace='rgb',alignHoriz='center', alignVert='center')
end_label = visual.TextStim(window, units='norm', text=u'Thank you!\nExiting...', pos=[0,0], height=0.1,
color=text_color, colorSpace='rgb',alignHoriz='center', alignVert='center')
# Color Blocks #
if centered:
rect_stim1 = visual.Rect(window, width=0.5, height=0.5, lineColor=(1, 1, 1), fillColor=shape1color,
fillColorSpace='rgb', pos=(0, 0))
rect_stim2 = visual.Rect(window, width=0.5, height=0.5, lineColor=(1, 1, 1), fillColor=shape2color,
fillColorSpace='rgb', pos=(0, 0))
rect_stim3 = visual.Rect(window, width=0.5, height=0.5, lineColor=(1, 1, 1), fillColor=shape3color,
fillColorSpace='rgb', pos=(0, 0))
else:
rect_stim1 = visual.Rect(window, width=0.5, height=0.5, lineColor=(1, 1, 1), fillColor=shape1color,
fillColorSpace='rgb', pos=(-0.5, 0.5))
rect_stim2 = visual.Rect(window, width=0.5, height=0.5, lineColor=(1, 1, 1), fillColor=shape2color,
fillColorSpace='rgb', pos=(0.5, -0.5))
rect_stim3 = visual.Rect(window, width=0.5, height=0.5, lineColor=(1, 1,.1), fillColor=shape3color,
fillColorSpace='rgb', pos=(0.5, 0.5))
# Clear all events from the global and device level ioHub Event Buffers.
self.hub.clearEvents('all')
# pyaud = pyaudio.PyAudio()
# t = np.linspace(0, 1, 48000, endpoint=False)
# audsig = signal.square(20*np.pi*5*t).astype(np.float32)
# stream = pyaud.open(format=pyaudio.paFloat32, channels=1, rate=48000, output=True)
# stream.write(0.5*audsig)
#### INITIAL STARTER SCREEN ####
# Initial Instruction Page #
if test == False:
title_label.draw()
continue_label.draw()
continue_button.draw()
window.flip()
self.hub.clearEvents('all')
FINISH_INSTR = False
while FINISH_INSTR is False:
buttons = mouse.getPressed()
if buttons[0]:
if mouse.isPressedIn(continue_button, buttons=[0]):
break
#### TRIALS OF BLOCK GAME ####
if test == True:
file_name = '%s/%s' %('p'+(participant_id), file_name)
exp = data.ExperimentHandler(name=experiment_name, version=version, extraInfo={'participant':participant_id},
runtimeInfo=None, originPath=None, savePickle=True, saveWideText=True,
dataFileName=file_name)
# Non-Randomized Mode #
if random_modes is False:
# if not randomized between modes, then go through each and run its respective number of trials
for trial_type in range(4):
if trial_type == 0:
name = "motor"
num_reps = num_reps_motor
temp_blocks = num_blocks_motor
random_blocks_var = random_blocks_motor
elif trial_type == 1:
name = "speech"
num_reps = num_reps_speech
temp_blocks = num_blocks_speech
random_blocks_var = random_blocks_speech
elif trial_type == 2:
name = "eye"
num_reps = num_reps_eye
temp_blocks = num_blocks_eye
random_blocks_var = random_blocks_eye
elif trial_type == 3:
name = "keyboard"
num_reps = num_reps_keyboard
temp_blocks = num_blocks_keyboard
random_blocks_var = random_blocks_keyboard
for num in range(num_reps):
# assign the random variables
if random_blocks_var:
num_blocks = random.randint(1,3)
else:
if len(temp_blocks) == 1:
num_blocks = temp_blocks[0]
else:
num_blocks = random.sample(temp_blocks, 1)[0]
# begin trial set up
status = 1
trial_loop = data.TrialHandler(trialList=[], nReps=1, name=name)
exp.addLoop(trial_loop)
if status != -1:
# restart values and indicate new round
shapes = [rect_stim1, rect_stim2, rect_stim3]
helper.resetTrial(shapes, centered)
# randomize block order and begin new round
shapes = helper.randomizeBlocks(num_blocks, rect_stim1, rect_stim2, rect_stim3)
if len(shapes) == 1:
shapes[0].setPos((0.0))
# data columns are standardized across modalities by adding the same data to every trial type
helper.addTrialData(shapes, trial_type, num_blocks, exp)
# track types of trial
if trial_type == 0:
status = motor.trial(self, clock, window, shapes, mouse, text_color, centered, wait_time, warning_time, exp, num, ser)
elif trial_type == 1:
status = speech.trial(self, clock, window, shapes, mouse, keys, text_color, wait_time, warning_time, exp, num, ser)
elif trial_type == 2:
status = eye.trial(self, clock, window, shapes, text_color, centered, wait_time, warning_time, exp, num, ser)
elif trial_type == 3:
status = keyboard.trial(self, clock, window, shapes, keys, text_color, centered, wait_time, warning_time, exp, num, ser)
# always add shape colors since they will be relevant in every modality
exp.addData('correct', status)
exp.nextEntry()
# Randomized Mode #
else:
# randomize everything, except for total number of trials
# set the total counts of each trial type to 0
motor_count = 0
eye_count = 0
speech_count = 0
keyboard_count = 0
for num in range(num_random):
# check for disabled trials and choose a random trial type out of the enabled types
trial_array = []
if num_reps_motor != -1:
trial_array.append(0)
if num_reps_speech != -1:
trial_array.append(1)
if num_reps_eye != -1:
trial_array.append(2)
if num_reps_keyboard != -1:
trial_array.append(3)
if len(trial_array) == 0:
print "No trial modalities enabled."
break
trial_type = random.sample(trial_array, 1)[0]
num_blocks = random.randint(1,3)
# or use: num_blocks = random.sample([1,2,3], 1)[0]
if trial_type == 0:
name = "motor"
elif trial_type == 1:
name = "speech"
elif trial_type == 2:
name = "eye"
elif trial_type == 3:
name = "keyboard"
status = 1
trial_loop = data.TrialHandler(trialList=[], nReps=1, name=name)
exp.addLoop(trial_loop)
if status != -1:
# restart values and indicate new round
shapes = [rect_stim1, rect_stim2, rect_stim3]
helper.resetTrial(shapes, centered)
# randomize block order and begin new round
shapes = helper.randomizeBlocks(num_blocks, rect_stim1, rect_stim2, rect_stim3)
if len(shapes) == 1:
shapes[0].setPos((0.0))
# data columns are standardized across modalities
helper.addTrialData(shapes, trial_type, num_blocks, exp)
# track types of trial
if trial_type == 0:
status = motor.trial(self, clock, window, shapes, mouse, text_color, centered, wait_time, warning_time, exp, motor_count, ser)
motor_count += 1
elif trial_type == 1:
status = speech.trial(self, clock, window, shapes, mouse, keys, text_color, wait_time, warning_time, exp, speech_count, ser)
speech_count += 1
elif trial_type == 2:
status = eye.trial(self, clock, window, shapes, text_color, centered, wait_time, warning_time, exp, eye_count, ser)
eye_count += 1
elif trial_type == 3:
status = keyboard.trial(self, clock, window, shapes, keys, text_color, centered, wait_time, warning_time, exp, keyboard_count, ser)
keyboard_count += 1
# always add shape colors since they will be relevant in every modality
exp.addData('correct', status)
exp.nextEntry()
#### END MESSAGE ####
# if trial loop is finished successfully, then end the program
for frameN in range(75):
end_label.draw()
window.flip()
