Exemple #1
0
    def test_addTrialData_2(self):

        exp = data.ExperimentHandler(name='Andrew',
                                     version=1,
                                     extraInfo={'participant': 1},
                                     runtimeInfo=None,
                                     originPath=None,
                                     savePickle=True,
                                     saveWideText=True,
                                     dataFileName='Andrew')
        rect_stim1 = visual.Rect(window,
                                 width=0.5,
                                 height=0.5,
                                 lineColor=(1, 1, 1),
                                 fillColor='blue',
                                 fillColorSpace='rgb',
                                 pos=(0, 0))
        rect_stim2 = visual.Rect(window,
                                 width=0.5,
                                 height=0.5,
                                 lineColor=(1, 1, 1),
                                 fillColor='green',
                                 fillColorSpace='rgb',
                                 pos=(0, 0))

        shapes = [rect_stim1, rect_stim2]
        res = addTrialData(shapes, 1, 2, exp)
        self.assertEqual(res, None)
    def test_addTrialData_3(self):

        exp = data.ExperimentHandler(name='Andrew', version=1, extraInfo={'participant':1},
                                         runtimeInfo=None, originPath=None, savePickle=True, saveWideText=True,
                                         dataFileName='Andrew')
        rect_stim1 = visual.Rect(window, width=0.5, height=0.5, lineColor=(1, 1, 1), fillColor='blue',
                                 fillColorSpace='rgb', pos=(0, 0))
        shapes = [rect_stim1]
        res= addTrialData(shapes, 2, 1,exp)
        self.assertEqual(res,None)
Exemple #3
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()
Exemple #4
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()