diff options
Diffstat (limited to 'src/argaze/utils/demo_gaze_features_run.py')
| -rw-r--r-- | src/argaze/utils/demo_gaze_features_run.py | 35 |
1 files changed, 10 insertions, 25 deletions
diff --git a/src/argaze/utils/demo_gaze_features_run.py b/src/argaze/utils/demo_gaze_features_run.py index a957ce0..6e7e01e 100644 --- a/src/argaze/utils/demo_gaze_features_run.py +++ b/src/argaze/utils/demo_gaze_features_run.py @@ -231,20 +231,10 @@ def main(): aoi_scene_projection.draw_circlecast(aoi_matrix, current_fixation.focus, current_fixation.deviation_max, base_color=(0, 0, 0), matching_color=(255, 255, 255)) # Draw current fixation - cv2.circle(aoi_matrix, (int(current_fixation.focus[0]), int(current_fixation.focus[1])), int(current_fixation.deviation_max), (255, 255, 255), len(current_fixation.positions)) - - # Draw current fixation gaze positions - gaze_positions = current_fixation.positions.copy() - while len(gaze_positions) >= 2: - - ts_start, start_gaze_position = gaze_positions.pop_first() - ts_next, next_gaze_position = gaze_positions.first - - # Draw start gaze - start_gaze_position.draw(aoi_matrix, draw_precision=False) + current_fixation.draw(aoi_matrix, color=(255, 255, 0)) - # Draw movement from start to next - cv2.line(aoi_matrix, start_gaze_position, next_gaze_position, (0, 55, 55), 1) + # Draw current fixation gaze positions + current_fixation.draw_positions(aoi_matrix) else: @@ -260,22 +250,17 @@ def main(): current_saccade = gaze_movement_identifier[identification_mode].current_saccade # Draw current saccade gaze positions - gaze_positions = current_saccade.positions.copy() - while len(gaze_positions) >= 2: - - ts_start, start_gaze_position = gaze_positions.pop_first() - ts_next, next_gaze_position = gaze_positions.first + current_saccade.draw_positions(aoi_matrix) - # Draw start gaze - start_gaze_position.draw(aoi_matrix, draw_precision=False) - - # Draw movement from start to next - cv2.line(aoi_matrix, start_gaze_position, next_gaze_position, (0, 0, 255), 1) + # Draw last 10 steps of raw scan path + raw_scan_path.draw(aoi_matrix, fixation_color=(255, 0, 255), deepness=10) # Write last 5 steps of aoi scan path path = '' for step in aoi_scan_path[-5:]: + path += f'> {step.aoi} ' + path += f'> {aoi_scan_path.current_aoi}' cv2.putText(aoi_matrix, path, (20, window_size[1]-40), cv2.FONT_HERSHEY_SIMPLEX, 1, (255, 255, 255), 1, cv2.LINE_AA) @@ -326,8 +311,8 @@ def main(): key_pressed = cv2.waitKey(10) - if key_pressed != -1: - print(key_pressed) + #if key_pressed != -1: + # print(key_pressed) # Switch identification mode with 'm' key if key_pressed == 109: |