class Corsi():
def __init__(self, experiment):
self.logger = FileLogger()
self.screen = pygame.display.get_surface()
self.clock = pygame.time.Clock()
self.experiment = experiment
self.event_handler = EventHandler()
self.keyboard_handler = Event()
self.sprites_group = pygame.sprite.LayeredDirty()
self.background = {}
self.background["pasive"] = pygame.sprite.DirtySprite()
# self.background["pasive"].image = pygame.surface.Surface(Properties.SCREEN_RES)
self.background["pasive"].image = pygame.surface.Surface(
self.screen.get_size())
self.background["pasive"].image.fill([40, 40, 40])
self.background["pasive"].rect = self.background[
"pasive"].image.get_rect()
self.sprites_group.add(self.background["pasive"], layer=BACKGR_lyr)
# self.background["active"] = pygame.sprite.DirtySprite()
# self.background["active"].image = pygame.surface.Surface(Properties.SCREEN_RES)
# # self.background["active"].image = pygame.surface.Surface(self.screen.get_size())
# self.background["active"].image.fill([80,80,80])
# self.background["active"].rect = self.background["active"].image.get_rect()
# self.sprites_group.add(self.background["active"], layer=BACKGR_lyr)
# self.background["active"].visible = False
self.feedback_ok = Feedback()
self.feedback_no = Feedback(False)
self.sprites_group.add(self.feedback_ok, layer=FEED_lyr)
self.sprites_group.add(self.feedback_no, layer=FEED_lyr)
self.msgs = {}
self.boxes = {}
self.instruction = Instruction()
self.instruction.hide()
self.sprites_group.add(self.instruction, layer=INST_lyr)
self.msgs["done"] = ImageDone()
for v in self.msgs.itervalues():
self.sprites_group.add(v, layer=BTN_lyr)
for box_name, position in self.experiment["box_positions"].iteritems():
box = Box(box_name, json.loads(position),
self.experiment["properties"])
self.sprites_group.add(box, layer=BOXES_lyr)
self.boxes[box_name] = box
self.tic_group = []
self.trial = Trial(
self.logger, self.experiment["properties"],
self.change_to_active_mode, None, {
"ok": self.feedback_ok.show,
"no": self.feedback_no.show,
"off":
(lambda: [self.feedback_ok.hide(),
self.feedback_no.hide()])
})
self.exp_runner = ExpRunner(self.experiment, self.trial, self.boxes)
self.exp_runner.end_test = self.terminar_juego
self.exp_runner.instruction = self.instruction
self.trial.handle_end = self.exp_runner.next_trial
self.trial.set_boxes(self.boxes, self.suscribe_box_on,
self.unsuscribe_box)
self.boxes_on = []
self.msgs["done"].set_callback(self.trial.next_by_usr)
self.change_to_active_mode(False)
def change_to_active_mode(self, toActive=True):
# self.change_background(toActive)
if toActive:
self.state = INTERACTIVE
self.msgs["done"].show()
else:
self.state = PASSIVE
self.msgs["done"].hide()
def change_background(self, toActive=True):
if toActive:
self.background["active"].visible = True
self.background["pasive"].visible = False
else:
self.background["active"].visible = False
self.background["pasive"].visible = True
self.background["active"].dirty = True
self.background["pasive"].dirty = True
def set_events(self):
key_suscribe = self.keyboard_handler.suscribe
self.event_handler.suscribe(
pygame.KEYDOWN,
lambda ev: self.keyboard_handler.dispatch(ev.key, ev))
key_suscribe(
pygame.K_ESCAPE,
(lambda ev:
[self.logger.log_message("esc pressed"),
self.terminar_juego(ev)]))
self.event_handler.suscribe(
pygame.QUIT,
(lambda ev:
[self.logger.log_message("pygame quit"),
self.terminar_juego(ev)]))
self.event_handler.suscribe(pygame.MOUSEBUTTONDOWN, self.click)
# self.event_handler.suscribe(pygame.MOUSEBUTTONUP, self.click)
self.event_handler.suscribe(pygame.MOUSEBUTTONUP, self.release_click)
# def click(self, res):
# (x, y) = pygame.mouse.get_pos()
# if self.state == INTERACTIVE:
# for i in self.sprites_group.get_sprites_from_layer(BOXES_lyr):
# if (i.rect.collidepoint(x, y)):
# a = i.tic_off
# ev = pygame.USEREVENT + 1
# print "Suscribing event ", ev
# self.event_handler.suscribe(ev, a)
# i.click(lambda: self.event_handler.unsuscribe(ev, a))
# pygame.time.set_timer(ev, self.experiment["properties"]["tansweron"])
def click(self, res):
(x, y) = pygame.mouse.get_pos()
if self.state == INTERACTIVE:
click_in_box = False
for i in self.sprites_group.get_sprites_from_layer(BOXES_lyr):
if (i.rect.collidepoint(x, y)):
# print "Click box ", i.box_name
if not i.is_on:
self.suscribe_box_on(i)
i.click(self.experiment["properties"]["tansweron"],
self.unsuscribe_box)
self.trial.usr_answer(i.box_name, (x, y))
click_in_box = True
for i in self.sprites_group.get_sprites_from_layer(BTN_lyr):
if (i.rect.collidepoint(x, y)):
# print "Click button "
click_in_box = True
i.click()
if not click_in_box:
self.trial.usr_answer(None, (x, y))
elif self.state == PASSIVE:
if self.instruction.visible:
self.instruction.hide()
def release_click(self, res):
(x, y) = pygame.mouse.get_pos()
if self.state == INTERACTIVE:
for i in self.sprites_group.get_sprites_from_layer(BTN_lyr):
if (i.rect.collidepoint(x, y)):
# print "Click button "
click_in_box = True
i.release_click()
def suscribe_box_on(self, box):
self.boxes_on.append(box)
def unsuscribe_box(self, box):
# import pdb; pdb.set_trace();
self.boxes_on.remove(box)
def terminar_juego(self, ev=None):
print "TERMINANDO"
self.logger.log_message("end game")
self.logger.close()
self.running = False
def report_answ(self, ev=None):
if not self.trial.usr_answer():
self.logger.log_invalid_press()
def run(self):
#~ import pdb; pdb.set_trace()
self.running = True
self.set_events()
self.exp_runner.next()
self.mainLoop()
def trial_handle_next(self):
pass
def mainLoop(self):
pygame.display.flip()
self.playing = None
while self.running:
self.clock.tick(60)
self.event_handler.handle()
self.trial.tic()
for box in self.boxes_on:
box.tic()
self.sprites_group.draw(self.screen)
pygame.display.flip()
class Chrono():
def __init__(self, experiment):
self.logger = FileLogger()
self.screen = pygame.display.get_surface()
self.clock = pygame.time.Clock()
self.experiment = experiment
# self.level = Level()
self.event_handler = EventHandler()
self.keyboard_handler = Event()
self.sprites_group = pygame.sprite.LayeredDirty()
self.msgs = {}
self.msgs["welcome"] = ImageMessage(str(0), "welcome.jpg")
self.msgs["the_end"] = ImageMessage(str(0), "theEnd.jpg")
for v in self.msgs.itervalues():
self.sprites_group.add(v, layer=MSG_lyr)
self.trial = Trial(self.logger)
self.block = Block(self.trial, self.msgs)
self.exp_runner = ExpRunner(self.experiment, self.block, self.msgs,
self.set_images)
# self.log("INIT ", "INICIANDO JUEGO")
# self.state = State()
def set_images(self, block_num):
# print "Set images"
self.sprites_group.empty()
self.background = ImageMessage(block_num, "background.jpg")
self.background.show()
self.stimulus = Stimulus(block_num)
self.target_img = Stimulus(block_num, False)
self.feed = Feedback(block_num)
self.msgs["background"] = ImageMessage(block_num, "background.jpg")
self.msgs["op"] = ImageMessage(block_num, "operator.jpg")
self.msgs["feed"] = ImageMessage(block_num, "feedback.jpg")
self.msgs["subj"] = ImageMessage(block_num, "subject.jpg")
self.msgs["wait"] = ImageMessageBar(block_num, "endTrial.jpg",
self.feed)
self.msgs["end_block"] = ImageMessage(block_num, "endBlock.jpg")
for v in self.msgs.itervalues():
self.sprites_group.add(v, layer=MSG_lyr)
self.sprites_group.add(self.background, layer=BACKGR_lyr)
self.sprites_group.add(self.stimulus, layer=STIM_lyr)
self.sprites_group.add(self.target_img, layer=STIM_lyr)
self.sprites_group.add(self.feed.reference, layer=FEED_lyr)
self.sprites_group.add(self.feed.slow, layer=FEED_lyr)
self.sprites_group.add(self.feed.ok, layer=FEED_lyr)
self.sprites_group.add(self.feed.quick, layer=FEED_lyr)
self.trial.set_images(self.stimulus, self.target_img, self.feed)
def set_events(self):
key_suscribe = self.keyboard_handler.suscribe
# self.event_handler.suscribe(pygame.QUIT, self.terminar_juego)
# self.event_handler.suscribe(pygame.MOUSEBUTTONDOWN, self.sprites_clicked)
self.event_handler.suscribe(
pygame.KEYDOWN,
lambda ev: self.keyboard_handler.dispatch(ev.key, ev))
key_suscribe(pygame.K_ESCAPE, self.terminar_juego)
# key_suscribe(pygame.K_SPACE, lambda ev: self.restart_game())
# key_suscribe(pygame.K_q, self.start_stimulus)
key_suscribe(pygame.K_SPACE, self.report_answ)
self.event_handler.suscribe(pygame.MOUSEBUTTONDOWN,
lambda ev: self.block.next_by_usr())
self.event_handler.suscribe(pygame.MOUSEBUTTONDOWN,
lambda ev: self.exp_runner.next_block())
# key_suscribe(pygame.K_a, lambda ev: self.block.next_by_usr())
# key_suscribe(pygame.K_a, lambda ev: self.exp_runner.next_block())
def terminar_juego(self, ev=None):
self.logger.close()
self.running = False
def report_answ(self, ev=None):
if not self.trial.usr_answer():
self.logger.log_invalid_press()
def run(self):
#~ import pdb; pdb.set_trace()
self.running = True
self.set_events()
self.mainLoop()
self.current_block = 0
def trial_handle_next(self):
pass
def mainLoop(self):
# print "MainLoop"
pygame.display.flip()
self.playing = None
while self.running:
self.clock.tick(10)
self.event_handler.handle()
self.trial.tic()
self.block.tic()
self.sprites_group.draw(self.screen)
pygame.display.flip()