From 4606bf24e8dda01b626e510b5263db144dbcc74c Mon Sep 17 00:00:00 2001 From: Théo de la Hogue Date: Mon, 11 Apr 2022 12:12:49 +0200 Subject: Fixing export_tobii_segment_fixations.py output path management --- src/argaze/utils/export_tobii_segment_fixations.py | 25 +++++++++++----------- 1 file changed, 13 insertions(+), 12 deletions(-) diff --git a/src/argaze/utils/export_tobii_segment_fixations.py b/src/argaze/utils/export_tobii_segment_fixations.py index c20fe5d..7420d66 100644 --- a/src/argaze/utils/export_tobii_segment_fixations.py +++ b/src/argaze/utils/export_tobii_segment_fixations.py @@ -24,9 +24,18 @@ def main(): if args.segment_path != None: # manage destination path - if not os.path.exists(os.path.dirname(args.output)): - os.makedirs(os.path.dirname(args.output)) - print(f'{os.path.dirname(args.output)} folder created') + if args.output != None: + + if not os.path.exists(os.path.dirname(args.output)): + + os.makedirs(os.path.dirname(args.output)) + print(f'{os.path.dirname(args.output)} folder created') + + fixations_filepath = args.output + + else: + + fixations_filepath = f'{args.segment_path}/fixations.json' # Load a tobii segment tobii_segment = TobiiEntities.TobiiSegment(args.segment_path) @@ -44,10 +53,6 @@ def main(): print(f'{len(tobii_ts_gaze_positions)} gaze positions loaded') - # Access to video timestamp index - tobii_vts = tobii_segment_data.vts - print(f'{len(tobii_vts)} video timestamps loaded') - # Format tobii gaze data into generic gaze data and store them using millisecond unit timestamp generic_ts_gaze_positions = GazeFeatures.TimeStampedGazePositions() @@ -63,12 +68,8 @@ def main(): print(f'{len(fixation_analyser.fixations)} fixations found') # Export fixations analysis results - if args.output != None: - fixations_filepath = args.output - else: - fixations_filepath = f'{args.segment_path}/fixations.json' - fixation_analyser.fixations.export_as_json(fixations_filepath) + print(f'Fixations saved into {fixations_filepath}') if __name__ == '__main__': -- cgit v1.1