Allowing to set project name and participant name

1 files changed, 3 insertions, 2 deletions

diff --git a/src/argaze/utils/live_tobii_aruco_aoi_ivy_controller.py b/src/argaze/utils/live_tobii_aruco_aoi_ivy_controller.py
index 68de089..14888bb 100644
--- a/src/argaze/utils/live_tobii_aruco_aoi_ivy_controller.py
+++ b/src/argaze/utils/live_tobii_aruco_aoi_ivy_controller.py
@@ -8,7 +8,6 @@ from argaze import DataStructures, GazeFeatures
 from argaze.TobiiGlassesPro2 import *
 from argaze.ArUcoMarkers import ArUcoTracker, ArUcoCamera
 from argaze.AreaOfInterest import *
-from argaze.TobiiGlassesPro2 import *
 
 import cv2 as cv
 import numpy
@@ -25,6 +24,8 @@ def main():
     # Manage arguments
     parser = argparse.ArgumentParser(description=main.__doc__.split('-')[0])
     parser.add_argument('-t', '--tobii_ip', metavar='TOBII_IP', type=str, default='192.168.1.10', help='tobii glasses ip')
+    parser.add_argument('-p', '--project_name', metavar='PROJECT_NAME', type=str, default=TobiiController.DEFAULT_PROJECT_NAME, help='project name')
+    parser.add_argument('-u', '--participant_name', metavar='PARTICIPANT_NAME', type=str, default=TobiiController.DEFAULT_PARTICIPANT_NAME, help='participant name')
     parser.add_argument('-c', '--camera_calibration', metavar='CAM_CALIB', type=str, default='tobii_camera.json', help='json camera calibration filepath')
     parser.add_argument('-y', '--ivy_bus', metavar='IVY_BUS', type=str, default='0.0.0.0:2010', help='Ivy bus ip and port')
     parser.add_argument('-a', '--aoi_scene', metavar='AOI_SCENE', type=str, default='aoi3D_scene.obj', help='obj aoi scene filepath')
@@ -44,7 +45,7 @@ def main():
     IvyStart(args.ivy_bus)
 
     # Create tobii controller
-    tobii_controller = TobiiController.TobiiController(args.tobii_ip, 'myProject', 'mySelf')
+    tobii_controller = TobiiController.TobiiController(args.tobii_ip, args.project_name, args.participant_name)
 
     # Calibrate tobii glasses
     tobii_controller.calibrate()