def try_calib_box(self, direction, level):
self.send_tag(time.time(), 'start_1', {'level':float(level)/100, 'direction': direction})
#self.send_tag(time.time(), 'direction', direction)
self.calib_box_state = CalibBox('')
self.calib_box_state.direction = direction
self.calib_box_state.set_level(1,1, level-20, level+20)
self.screen.display_box(direction, level-20, level+20)
self.blink_calib_box()
#self.screen.play_sound('win')
self.clear_queue()
self.send_tag(time.time(), 'start', level)
t = time.time()
while time.time()-t<=7:
for event in pygame.event.get():
if event.type == QUIT or (event.type == KEYDOWN and event.key == K_ESCAPE):
pygame.quit()
sample = self.get_message()
if sample is not None:
if sample.key == direction:
self.calib_box_state.update(sample.value)
else:
self.calib_box_state.reset()
self.screen.update_block(sample.key, sample.value, self.calib_box_state.get_area_state())
if self.calib_box_state.is_move():
self.screen.clear(True)
self.send_tag(time.time(), 'finish', '1')
self.screen.play_sound('win')
return True
self.send_tag(time.time(), 'finish', '0')
self.screen.play_sound('fall')
return False
class Calibration2(logic_queue.LogicQueue):
"""
- perform calibration2 logic (display boxes in specific order)
- optinally perform 'final check' on 4 selected boxes
- store selected boxes (starting levels for 4 directions) for user id
"""
def __init__(self, user_id, boxes_levels, tag_name, tag_dir):
"""
- user_id - string
- boxes_levels - list of integers representing % of max sway;
for every this number a box should be generated for tests
"""
super(Calibration2, self).__init__()
self._level_selector = {'up': LevelSelector(boxes_levels),
'right': LevelSelector(boxes_levels),
'down': LevelSelector(boxes_levels),
'left': LevelSelector(boxes_levels)
}
self.screen = CalibrationScreen()
self.user_id = user_id
self.file_name = tag_name
self.tagger_init(tag_name, tag_dir)
def tagger_init(self, tag_name, tag_dir):
self.tagger = Tagger(tag_name, tag_dir, status='ON')
def send_tag(self, timestamp, tag_name, tage_value=''):
self.tagger.set_tag(timestamp, str(tag_name), str(tage_value))
def save_tags(self):
self.tagger.finish()
def try_calib_box(self, direction, level):
self.send_tag(time.time(), 'start_1', {'level':float(level)/100, 'direction': direction})
#self.send_tag(time.time(), 'direction', direction)
self.calib_box_state = CalibBox('')
self.calib_box_state.direction = direction
self.calib_box_state.set_level(1,1, level-20, level+20)
self.screen.display_box(direction, level-20, level+20)
self.blink_calib_box()
#self.screen.play_sound('win')
self.clear_queue()
self.send_tag(time.time(), 'start', level)
t = time.time()
while time.time()-t<=7:
for event in pygame.event.get():
if event.type == QUIT or (event.type == KEYDOWN and event.key == K_ESCAPE):
pygame.quit()
sample = self.get_message()
if sample is not None:
if sample.key == direction:
self.calib_box_state.update(sample.value)
else:
self.calib_box_state.reset()
self.screen.update_block(sample.key, sample.value, self.calib_box_state.get_area_state())
if self.calib_box_state.is_move():
self.screen.clear(True)
self.send_tag(time.time(), 'finish', '1')
self.screen.play_sound('win')
return True
self.send_tag(time.time(), 'finish', '0')
self.screen.play_sound('fall')
return False
def wait_calib_box(self):
self.clear_queue()
t = time.time()
while time.time()-t<=2:
for event in pygame.event.get():
if event.type == QUIT or (event.type == KEYDOWN and event.key == K_ESCAPE):
pygame.quit()
sample = self.get_message()
if sample is not None:
self.screen.update_level(sample.key, sample.value)
self.screen._display()
return
def blink_calib_box(self):
self.clear_queue()
t = time.time()
color = ['white', 'green_2']
for color in ['green_2', 'white', 'green_2']:
while time.time()-t<=0.1:
for event in pygame.event.get():
if event.type == QUIT or (event.type == KEYDOWN and event.key == K_ESCAPE):
pygame.quit()
sample = self.get_message()
if sample is not None:
self.screen.update_level(sample.key, sample.value)
self.screen.display_color_box(color)
if color == 'green_2':
self.screen.play_sound('start')
t = time.time()
return
def instruction(self):
screens = ['start', 'instruction']
screen_status = 0
self.screen.display_screen(screens[screen_status])
while screen_status<len(screens):
for event in pygame.event.get():
if event.type == KEYDOWN:
if event.key == K_SPACE:
screen_status+=1
if screen_status<len(screens):
self.screen.display_screen(screens[screen_status])
def run(self):
self.instruction()
self.clear_queue()
self.screen.display_calib_start()
done = False
selected_levels = {'up': None,
'right': None,
'down': None,
'left': None
}
while not done:
for d in ['up', 'right', 'down', 'left']:
if selected_levels[d] is None:
self.wait_calib_box()
level = self._level_selector[d].get_level()
passed = self.try_calib_box(d, level)
#passed = True #TODO - perform trial on 'level' value in order to determine if patient succeeded in selecting box on level value 'level'
res = self._level_selector[d].update_level_result(passed)
print("For direction: "+str(d)+" got level: "+str(level)+" and result: "+str(res))
if res == 0:
#just try again
pass
elif res == 1:
#bingo, level is user's best possible level
selected_levels[d] = level
elif res == 2:
#kind of error - user cant succeede even on easiest level, finish somehow, error or sth. next get_level call would end up badly...
pass #TODO
self.send_tag(time.time(), "FINISH", 'error')
print("ERROR 2 in ventures boxes calibration! User couldnt pass even the easiest box. Sth. is wrong. Abort with fire!")
self.save_tags()
sys.exit(1)
if not (None in selected_levels.values()):
#all levels selected, optionally check them if needed,
#store selected_levels for user_id
#and finish gracefully so that logic_experiment can finish gracefully
pass#TODO
done = True
print "Selected levels: "
print str(selected_levels)
self.send_tag(time.time(), "FINISH", '')
self.save_tags()
calibration_2_set(self.user_id, selected_levels['up'], selected_levels['right'], selected_levels['down'], selected_levels['left'], self.file_name)
