Ejemplo n.º 1
0
 def __init_attributes(self):
     self._sound = pygame.mixer.Sound(path.join(datadir, 'sound.ogg'))
     self._palette = Palette()
     self._trial_types = ['Count', 'YesNo', 'Calibration', 'FreeSpelling',
                          'CopySpelling']
Ejemplo n.º 2
0
class Control(VisionEggFeedback, Config):
    def __init__(self, *args, **kwargs):
        pygame.mixer.init()
        self.__init_attributes()
        VisionEggFeedback.__init__(self, *args, **kwargs)

    def init(self):
        Config.init(self)
        self.update_parameters()

    def __init_attributes(self):
        self._sound = pygame.mixer.Sound(path.join(datadir, 'sound.ogg'))
        self._palette = Palette()
        self._trial_types = [
            'Count', 'YesNo', 'Calibration', 'FreeSpelling', 'CopySpelling'
        ]

    def _create_view(self):
        return View(self._palette)

    def update_parameters(self):
        VisionEggFeedback.update_parameters(self)
        self._palette.set(self.symbol_colors, self.color_groups)
        self.alphabet = ''.join(self.color_groups)
        self._sorted_alphabet = sorted(self.alphabet, key=lambda s: s.lower())
        self.eeg_alphabet = ''.join(
            filter(lambda c: c.isalpha(), self._sorted_alphabet) +
            [e[0] for e in self.nonalpha_trigger])
        self._trial_name = self._trial_types[self.trial_type - 1]
        self._setup_input_handler()
        self._setup_trial()
        self._setup_experiment()

    def _setup_input_handler(self):
        input_handler_type = self._trial_name + 'InputHandler'
        self._input_handler = eval(input_handler_type)(self)

    def _setup_trial(self):
        trial_type = self._trial_name + 'Trial'
        self._trial = eval(trial_type)(self._view, self._trigger, self._iter,
                                       self.stimulus_sequence, self)

    def _setup_experiment(self):
        experiment_type = self._trial_name + 'Experiment'
        self._experiment = eval(experiment_type)(self._view, self._trial,
                                                 self._input_handler,
                                                 self._flag, self._iter,
                                                 self._palette, self)

    def run(self):
        try:
            self._view.alphabet(self.eeg_alphabet)
            self._experiment.run()
        except RSVPSpellerException as e:
            self.logger.error(e)
        self.quit()

    def keyboard_input(self, event):
        if self._trial.asking:
            self._input_handler.keyboard(event)
        VisionEggFeedback.keyboard_input(self, event)

    def quit(self):
        self._view.answered()
        VisionEggFeedback.quit(self)

    def on_control_event(self, data):
        cls = data.get('cl_output', None)
        if cls is not None:
            self._input_handler.eeg_select(cls)
Ejemplo n.º 3
0
class Control(VisionEggFeedback, Config):
    def __init__(self, *args, **kwargs):
        pygame.mixer.init()
        self.__init_attributes()
        VisionEggFeedback.__init__(self, *args, **kwargs)

    def init(self):
        Config.init(self)
        self.update_parameters()

    def __init_attributes(self):
        self._sound = pygame.mixer.Sound(path.join(datadir, 'sound.ogg'))
        self._palette = Palette()
        self._trial_types = ['Count', 'YesNo', 'Calibration', 'FreeSpelling',
                             'CopySpelling']

    def _create_view(self):
        return View(self._palette)

    def update_parameters(self):
        VisionEggFeedback.update_parameters(self)
        self._palette.set(self.symbol_colors, self.color_groups)
        self.alphabet = ''.join(self.color_groups)
        self._sorted_alphabet = sorted(self.alphabet, key=lambda s: s.lower())
        self.eeg_alphabet = ''.join(filter(lambda c: c.isalpha(),
                                           self._sorted_alphabet)
                                    + [e[0] for e in self.nonalpha_trigger])
        self._trial_name = self._trial_types[self.trial_type - 1]
        self._setup_input_handler()
        self._setup_trial()
        self._setup_experiment()

    def _setup_input_handler(self):
        input_handler_type = self._trial_name + 'InputHandler'
        self._input_handler = eval(input_handler_type)(self)

    def _setup_trial(self):
        trial_type = self._trial_name + 'Trial'
        self._trial = eval(trial_type)(self._view, self._trigger, self._iter,
                                       self.stimulus_sequence, self)

    def _setup_experiment(self):
        experiment_type = self._trial_name + 'Experiment'
        self._experiment = eval(experiment_type)(self._view, self._trial,
                                                 self._input_handler,
                                                 self._flag, self._iter,
                                                 self._palette, self)

    def run(self):
        try:
            self._view.alphabet(self.eeg_alphabet)
            self._experiment.run()
        except RSVPSpellerException as e:
            self.logger.error(e)
        self.quit()

    def keyboard_input(self, event):
        if self._trial.asking:
            self._input_handler.keyboard(event)
        VisionEggFeedback.keyboard_input(self, event)

    def quit(self):
        self._view.answered()
        VisionEggFeedback.quit(self)

    def on_control_event(self, data):
        cls = data.get('cl_output', None)
        if cls is not None:
            self._input_handler.eeg_select(cls)