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)
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)
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()
def __init__(self, experiment_data):
self.logger = FileLogger(SUBJECT_NAME)
self.experiment_data = experiment_data
SCREEN_RES = (int(experiment_data["screen_res"]["x"]),
int(experiment_data["screen_res"]["y"]))
self.screen = pygame.display.get_surface()
self.clock = pygame.time.Clock()
self.event_handler = EventHandler.EventHandler()
self.keyboard_handler = Event.Event()
self.sprites_group = pygame.sprite.LayeredDirty()
# self.background = pygame.sprite.DirtySprite()
self.background = ImageMessage("background.png", BACKGROUND_PROFILE)
self.background.scale(SCREEN_RES[0], SCREEN_RES[1])
self.background.show()
self.sprites_group.add(self.background, layer=BACKGR_lyr)
self.msgs = {}
self.msgs["welcome"] = ImageMessage("welcome.png", BACKGROUND_PROFILE)
self.msgs["the_end"] = ImageMessage("theEnd.png", BACKGROUND_PROFILE)
# self.msgs["background"] = ImageMessage("background.png")
self.msgs["op"] = ImageMessage("operator.png", BACKGROUND_PROFILE)
self.msgs["feed"] = ImageMessage("feedback.png", BACKGROUND_PROFILE)
self.msgs["subj"] = ImageMessage("subject.png", BACKGROUND_PROFILE)
self.msgs["end_block"] = ImageMessage("endBlock.png",
BACKGROUND_PROFILE)
for v in self.msgs.itervalues():
v.scale(SCREEN_RES[0], SCREEN_RES[1])
self.sprites_group.add(v, layer=MSG_lyr)
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.state = State.State()
# st = self.state
# self.trial = Trial(self.logger, self.show_messages,
# None,
# {"ok": self.feedback_ok.show, "no": self.feedback_no.show,
# "off": (lambda: [self.feedback_ok.hide(), self.feedback_no.hide()])},
# self)
# self.add_elements(experiment_data["trials"]["2"][1], 0)
# self.add_elements(experiment_data["trials"]["2"][2], 2)
self.btns = {}
self.btns["left"] = ImageButton(0)
self.btns["right"] = ImageButton(1)
for v in self.btns.itervalues():
self.sprites_group.add(v, layer=ANSW_BTN_lyr)
# self.btns["left"].set_callback(lambda: self.exp_runner.check_answer([
# c for i in xrange(len(self.palettes)) for c in self.palettes[i].get_selected() ]))
self.trial = Trial(self.logger, self.change_to_active_mode)
self.trial.feedback[1] = self.feedback_ok
self.trial.feedback[0] = self.feedback_no
self.block = Block(self.trial, self.msgs)
# self.exp_runner = ExpRunner(self.experiment_data, self.add_elements, self.add_screenshot)
self.exp_runner = ExpRunner(self.experiment_data, self.block,
self.msgs, self.set_images)
self.exp_runner.end_test = self.end_game
self.btns["left"].set_callback(self.trial.usr_answer)
self.btns["right"].set_callback(self.trial.usr_answer)
# self.exp_runner.instruction = self.instruction
# self.exp_runner.next()
self.change_to_active_mode(False)
# self.msgs["welcome"].show()
self.clicked_sprite = None
class Stroop():
def __init__(self, experiment_data):
self.logger = FileLogger(SUBJECT_NAME)
self.experiment_data = experiment_data
SCREEN_RES = (int(experiment_data["screen_res"]["x"]),
int(experiment_data["screen_res"]["y"]))
self.screen = pygame.display.get_surface()
self.clock = pygame.time.Clock()
self.event_handler = EventHandler.EventHandler()
self.keyboard_handler = Event.Event()
self.sprites_group = pygame.sprite.LayeredDirty()
# self.background = pygame.sprite.DirtySprite()
self.background = ImageMessage("background.png", BACKGROUND_PROFILE)
self.background.scale(SCREEN_RES[0], SCREEN_RES[1])
self.background.show()
self.sprites_group.add(self.background, layer=BACKGR_lyr)
self.msgs = {}
self.msgs["welcome"] = ImageMessage("welcome.png", BACKGROUND_PROFILE)
self.msgs["the_end"] = ImageMessage("theEnd.png", BACKGROUND_PROFILE)
# self.msgs["background"] = ImageMessage("background.png")
self.msgs["op"] = ImageMessage("operator.png", BACKGROUND_PROFILE)
self.msgs["feed"] = ImageMessage("feedback.png", BACKGROUND_PROFILE)
self.msgs["subj"] = ImageMessage("subject.png", BACKGROUND_PROFILE)
self.msgs["end_block"] = ImageMessage("endBlock.png",
BACKGROUND_PROFILE)
for v in self.msgs.itervalues():
v.scale(SCREEN_RES[0], SCREEN_RES[1])
self.sprites_group.add(v, layer=MSG_lyr)
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.state = State.State()
# st = self.state
# self.trial = Trial(self.logger, self.show_messages,
# None,
# {"ok": self.feedback_ok.show, "no": self.feedback_no.show,
# "off": (lambda: [self.feedback_ok.hide(), self.feedback_no.hide()])},
# self)
# self.add_elements(experiment_data["trials"]["2"][1], 0)
# self.add_elements(experiment_data["trials"]["2"][2], 2)
self.btns = {}
self.btns["left"] = ImageButton(0)
self.btns["right"] = ImageButton(1)
for v in self.btns.itervalues():
self.sprites_group.add(v, layer=ANSW_BTN_lyr)
# self.btns["left"].set_callback(lambda: self.exp_runner.check_answer([
# c for i in xrange(len(self.palettes)) for c in self.palettes[i].get_selected() ]))
self.trial = Trial(self.logger, self.change_to_active_mode)
self.trial.feedback[1] = self.feedback_ok
self.trial.feedback[0] = self.feedback_no
self.block = Block(self.trial, self.msgs)
# self.exp_runner = ExpRunner(self.experiment_data, self.add_elements, self.add_screenshot)
self.exp_runner = ExpRunner(self.experiment_data, self.block,
self.msgs, self.set_images)
self.exp_runner.end_test = self.end_game
self.btns["left"].set_callback(self.trial.usr_answer)
self.btns["right"].set_callback(self.trial.usr_answer)
# self.exp_runner.instruction = self.instruction
# self.exp_runner.next()
self.change_to_active_mode(False)
# self.msgs["welcome"].show()
self.clicked_sprite = None
def change_to_active_mode(self, toActive=True):
# self.change_background(toActive)
if toActive:
self.state = INTERACTIVE
self.btns["left"].show()
self.btns["right"].show()
else:
self.state = PASSIVE
self.btns["left"].hide()
self.btns["right"].hide()
def set_images(self, block_num):
# print "Set images"
self.sprites_group.empty()
self.stimulus = Stimulus(STIM_PROFILE)
self.feed = Feedback()
self.sprites_group.add(self.feedback_ok, layer=FEED_lyr)
self.sprites_group.add(self.feedback_no, layer=FEED_lyr)
# self.msgs["wait"] = ImageMessageBar("endTrial.png", self.feed)
for v in self.msgs.itervalues():
self.sprites_group.add(v, layer=MSG_lyr)
for v in self.btns.itervalues():
self.sprites_group.add(v, layer=ANSW_BTN_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.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.feed)
def remove_elements(self):
self.sprites_group.remove_sprites_of_layer(DIAGRAM_lyr)
self.sprites_group.remove_sprites_of_layer(SCREENSHOTS_lyr)
self.background.dirty = 1
self.msgs["done"].hide()
def add_elements(self, diags):
self.remove_elements()
self.msgs["done"].show()
self.diagrams = []
self.palettes = []
num_whites = 0
for row, diags in diags.iteritems():
for (n, i) in enumerate(diags):
d = Diagram(layout=i[0],
colors=i[1:],
position=(n, row, len(diags)))
num_whites += d.get_whites()
self.diagrams.append(d)
self.sprites_group.add(d, layer=DIAGRAM_lyr)
for i in xrange(num_whites):
p = ColorPalette(num=i + 1, num_palettes=num_whites)
self.palettes.append(p)
self.sprites_group.add(p, layer=DIAGRAM_lyr)
def add_screenshot(self, filename):
self.remove_elements()
self.sprites_group.add(Screenshot(filename), layer=DIAGRAM_lyr)
# self.show_messages(False)
def show_messages(self, toActive=True, info_or_feedback=2):
# self.change_background(toActive)
# print "To PASSIVE MODE"
for i in [GOAL_lyr, DISKS_lyr, CTRL_BTN_lyr, FEED_lyr, INST_lyr]:
for sp in self.sprites_group.get_sprites_from_layer(i):
sp.hide()
if toActive:
# print "To INTERACTIVE MODE"
# self.msgs["done"].show()
self.moving_state = INTERACTIVE
# self.instruction.hide()
# self.goal.visible = True
for i in [GOAL_lyr, DISKS_lyr, CTRL_BTN_lyr]:
for sp in self.sprites_group.get_sprites_from_layer(i):
sp.show()
else:
if info_or_feedback == 2:
self.moving_state = PASSIVE
self.instruction.show()
else:
self.moving_state = FEEDBACK
if info_or_feedback == 1:
self.feedback_ok.show()
elif info_or_feedback == 0:
self.feedback_no.show()
def set_events(self):
suscribir_tecla = self.keyboard_handler.suscribe
self.event_handler.suscribe(pygame.QUIT, self.end_game)
# self.event_handler.suscribe(pygame.MOUSEBUTTONDOWN, self.clicked)
self.event_handler.suscribe(pygame.MOUSEMOTION, self.move_mouse)
# self.event_handler.suscribe(pygame.MOUSEBUTTONUP, self.unclicked)
self.event_handler.suscribe(
pygame.KEYDOWN,
lambda ev: self.keyboard_handler.dispatch(ev.key, ev))
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())
self.event_handler.suscribe(pygame.MOUSEBUTTONDOWN, self.click)
self.event_handler.suscribe(pygame.MOUSEBUTTONUP, self.release_click)
suscribir_tecla(pygame.K_ESCAPE, self.end_game)
suscribir_tecla(pygame.K_LEFT, self.prev)
def click(self, res):
(x, y) = pygame.mouse.get_pos()
if self.state == INTERACTIVE:
# print (x,y, len(self.sprites_group.get_sprites_from_layer(ANSW_BTN_lyr)))
for i in self.sprites_group.get_sprites_from_layer(ANSW_BTN_lyr):
if (i.rect.collidepoint(x, y)):
self.clicked_sprite = i
# print "Click box ", i.box_name
i.click()
def release_click(self, res):
(x, y) = pygame.mouse.get_pos()
if self.state == INTERACTIVE and self.clicked_sprite is not None:
for i in self.sprites_group.get_sprites_from_layer(ANSW_BTN_lyr):
if (i.rect.collidepoint(x, y)):
# print "Click button "
i.release_click()
self.clicked_sprite = None
# def clicked(self, res):
# (x, y) = pygame.mouse.get_pos()
# if self.exp_runner.clicked():
# for i in self.sprites_group.get_sprites_from_layer(DIAGRAM_lyr) + self.sprites_group.get_sprites_from_layer(CTRL_BTN_lyr):
# if (i.rect.collidepoint(x, y)):
# # print "Clicked", (x, y)
# i.click()
# # else:
# # self.exp_runner.next()
def move_mouse(self, res):
(x, y) = pygame.mouse.get_pos()
if self.clicked_sprite is not None:
if not self.clicked_sprite.rect.collidepoint(x, y):
self.clicked_sprite.un_click()
self.clicked_sprite = None
# def unclicked(self, res):
# (x, y) = pygame.mouse.get_pos()
# if self.exp_runner.clicked():
# for i in self.sprites_group.get_sprites_from_layer(CTRL_BTN_lyr):
# if (i.rect.collidepoint(x, y)):
# i.release_click()
# else:
# self.exp_runner.next()
# if self.clicked_sprite is not None:
# qq = self.clicked_sprite
# for i in self.sprites_group.get_sprites_from_layer(STICK_lyr):
# if (i.rect.colliderect(self.clicked_sprite)):
# # print "Left on Stick ", i.stick.num
# if i.stick.not_full() and i.stick.num!=self.state.get_stick_of_disk(self.clicked_sprite.disk.num).num:
# # print "There is room enough. Number of moves: ", self.moves_num
# self.state.moveSP(self.clicked_sprite, i)
# self.clicked_sprite = None
# self.moves_num += 1
# self.sequence_boards.append(self.state.get_board_number())
# self.refresh_indicators()
# # if self.state.get_board_number() == self.goal_num:
# # print "GANASTE"
# already_placed = True
# self.logger.log_release_disk(self.exp_runner.trial.trial_name,
# self.trial.current_trial["source"],
# self.trial.current_trial["target"],
# self.state.get_board_number(),
# self.trial.tol.moves_num,
# qq.disk.num,
# True
# )
# return
# # if not already_placed:
# self.clicked_sprite.set_stick_pos()
# self.clicked_sprite = None
# self.logger.log_release_disk(self.exp_runner.trial.trial_name,
# self.trial.current_trial["source"],
# self.trial.current_trial["target"],
# self.state.get_board_number(),
# self.trial.tol.moves_num,
# qq.disk.num,
# False
# )
# if self.moving_state == INTERACTIVE:
# (x, y) = pygame.mouse.get_pos()
# for i in self.sprites_group.get_sprites_from_layer(CTRL_BTN_lyr):
# if (i.rect.collidepoint(x, y)):
# i.release_click()
# elif self.moving_state == PASSIVE:
# self.show_messages()
# elif self.moving_state == FEEDBACK:
# self.show_messages(False)
def run(self):
#~ import pdb; pdb.set_trace()
self.running = True
self.set_events()
self.mainLoop()
def prev(self, ev=None):
self.exp_runner.prev()
# self.logger.close()
# self.running = False
def end_game(self, ev=None):
self.logger.log_message("end game")
self.logger.close()
self.running = False
def mainLoop(self):
# print "MainLoop"
pygame.display.flip()
# pygame.image.save(out,fname+".tga")
while self.running:
self.event_handler.handle()
dt = self.clock.tick(10)
self.trial.tic()
self.block.tic()
self.sprites_group.draw(self.screen)
pygame.display.flip()
def __init__(self, experiment):
self.logger = FileLogger()
self.experiment = experiment
self.mode = EXPERIMENT_MODE
SCREEN_RES = (int(experiment["screen_res"]["x"]), int(experiment["screen_res"]["y"]))
self.properties = Properties.Properties(SCREEN_RES)
self.screen = pygame.display.get_surface()
self.clock = pygame.time.Clock()
self.event_handler = EventHandler.EventHandler()
self.keyboard_handler = Event.Event()
self.sprites_group = pygame.sprite.LayeredDirty()
# self.background = pygame.sprite.DirtySprite()
self.background = ImageMessage("scale_fondo{}.png".format(BACKGROUND_PROFILE))
self.background.scale(SCREEN_RES[0], SCREEN_RES[1])
self.background.show()
# self.background.image = pygame.surface.Surface(Properties.SCREEN_RES)
# self.background.image.fill([240,240,240])
# self.background.rect = self.background.image.get_rect()
self.sprites_group.add(self.background, layer=BACKGR_lyr)
self.floor = pygame.sprite.DirtySprite()
self.floor.image = pygame.surface.Surface((self.properties.SCREEN_RES[0], self.properties.floor_height))
self.floor.image.fill([0,200,0])
self.floor.rect = self.floor.image.get_rect()
self.floor.rect.bottom = self.properties.SCREEN_RES[1]
self.sprites_group.add(self.floor, layer=FLOOR_lyr)
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.state = State.State()
st = self.state
self.trial = Trial(self.logger, self.show_messages,
None,
{"ok": self.feedback_ok.show, "no": self.feedback_no.show,
"off": (lambda: [self.feedback_ok.hide(), self.feedback_no.hide()])},
self)
self.sprites_group.add(StickSprite(st.sticks[1], self.properties), layer=STICK_lyr)
self.sprites_group.add(StickSprite(st.sticks[2], self.properties), layer=STICK_lyr)
self.sprites_group.add(StickSprite(st.sticks[3], self.properties), layer=STICK_lyr)
self.sprites_group.add(DiskSprite(st.get_disk(State.B),st.get_disk_position(State.B), mode=DISKS_PROFILE, prop=self.properties), layer=DISKS_lyr)
self.sprites_group.add(DiskSprite(st.get_disk(State.G),st.get_disk_position(State.G), mode=DISKS_PROFILE, prop=self.properties), layer=DISKS_lyr)
self.sprites_group.add(DiskSprite(st.get_disk(State.R),st.get_disk_position(State.R), mode=DISKS_PROFILE, prop=self.properties), layer=DISKS_lyr)
self.goal = Goal(DISKS_PROFILE, prop=self.properties)
self.sprites_group.add(self.goal, layer=GOAL_lyr)
self.instruction = Instruction(prop=self.properties)
self.instruction.hide()
self.sprites_group.add(self.instruction, layer=INST_lyr)
self.msgs = {}
self.msgs["done"] = ImageDone(prop=self.properties)
self.msgs["done"].set_callback(self.trial.check_answer)
for v in self.msgs.itervalues():
self.sprites_group.add(v, layer=CTRL_BTN_lyr)
self.exp_runner = ExpRunner(self.experiment,
self.trial)
self.exp_runner.end_test = self.end_game
self.exp_runner.instruction = self.instruction
self.exp_runner.next()
self.clicked_sprite = None
if self.mode == FREE_MODE:
for i in range(1,8+1):
self.sprites_group.add(TextLevel(i,self.set_board_distance), layer=LEVEL_SEL_lyr)
self.show_messages(False)
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()