diff options
Diffstat (limited to 'src/argaze/utils/export_tobii_segment_movements.py')
| -rw-r--r-- | src/argaze/utils/export_tobii_segment_movements.py | 15 |
1 files changed, 7 insertions, 8 deletions
diff --git a/src/argaze/utils/export_tobii_segment_movements.py b/src/argaze/utils/export_tobii_segment_movements.py index 0b0d867..7222414 100644 --- a/src/argaze/utils/export_tobii_segment_movements.py +++ b/src/argaze/utils/export_tobii_segment_movements.py @@ -47,28 +47,27 @@ def main(): movements_filepath = f'{destination_path}/movements.csv' # Load a tobii segment - tobii_segment = TobiiEntities.TobiiSegment(args.segment_path, int(args.time_range[0] * 1000000), int(args.time_range[1] * 1000000) if args.time_range[1] != None else None) + tobii_segment = TobiiEntities.TobiiSegment(args.segment_path, int(args.time_range[0] * 1e6), int(args.time_range[1] * 1e6) if args.time_range[1] != None else None) # Load a tobii segment video tobii_segment_video = tobii_segment.load_video() - print(f'Video properties:\n\tduration: {tobii_segment_video.get_duration()/1000000} s\n\twidth: {tobii_segment_video.get_width()} px\n\theight: {tobii_segment_video.get_height()} px') + print(f'Video properties:\n\tduration: {tobii_segment_video.get_duration()/1e6} s\n\twidth: {tobii_segment_video.get_width()} px\n\theight: {tobii_segment_video.get_height()} px') # Load a tobii segment data tobii_segment_data = tobii_segment.load_data() - print(f'Data keys:') + print(f'Loaded data count:') for name in tobii_segment_data.keys(): - print(f'\t{name}') + print(f'\t{name}: {len(tobii_segment_data[name])} data') # Access to timestamped gaze position data buffer - tobii_ts_gaze_positions = tobii_segment_data.gidx_l_gp - print(f'{len(tobii_ts_gaze_positions)} gaze positions loaded') + tobii_ts_gaze_positions = tobii_segment_data['GazePosition'] # Format tobii gaze position in pixel and store them using millisecond unit timestamp ts_gaze_positions = GazeFeatures.TimeStampedGazePositions() - for ts, tobii_data in tobii_ts_gaze_positions.items(): - video_gaze_pixel = (int(tobii_data.gp[0] * tobii_segment_video.get_width()), int(tobii_data.gp[1] * tobii_segment_video.get_height())) + for ts, tobii_gaze_position in tobii_ts_gaze_positions.items(): + video_gaze_pixel = (int(tobii_gaze_position.value[0] * tobii_segment_video.get_width()), int(tobii_gaze_position.value[1] * tobii_segment_video.get_height())) ts_gaze_positions[ts/1000] = video_gaze_pixel print(f'Dispersion threshold: {args.dispersion_threshold}') |