Adding a display tobii gaze script

2 files changed, 81 insertions, 0 deletions

diff --git a/src/argaze/utils/README.md b/src/argaze/utils/README.md
index 63d060e..c4ed010 100644
--- a/src/argaze/utils/README.md
+++ b/src/argaze/utils/README.md
@@ -24,6 +24,12 @@ python ./src/argaze/utils/export_calibration_board.py 7 5 5 3 -o export
 python ./src/argaze/utils/tobii_camera_calibration.py 7 5 5 3 -t IP_ADDRESS -o export/tobii_camera.json
 ```
 
+- Display Tobii Glasses Pro 2 gaze and camera video stream (replace IP_ADDRESS)
+
+```
+python ./src/argaze/utils/display_tobii_gaze.py -t IP_ADDRESS
+```
+
 - Record a Tobii Glasses Pro 2 'Test' session for a participant '1' on Tobii interface's SD card (replace IP_ADDRESS).
 
 ```
diff --git a/src/argaze/utils/display_tobii_gaze.py b/src/argaze/utils/display_tobii_gaze.py
new file mode 100644
index 0000000..8a3bb33
--- /dev/null
+++ b/src/argaze/utils/display_tobii_gaze.py
@@ -0,0 +1,75 @@
+#!/usr/bin/env python
+
+import argparse
+import os, time
+
+from argaze.TobiiGlassesPro2 import *
+
+import cv2 as cv
+import numpy
+
+def main():
+    """
+    Capture video camera and display gaze point
+    """
+
+    # 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')
+
+    args = parser.parse_args()
+
+    # create tobii controller
+    tobii_controller = TobiiController.TobiiController(args.tobii_ip, 'ArGaze', 1)
+
+    # calibrate tobii glasses
+    #tobii_controller.calibrate()
+
+    # create tobii data thread
+    tobii_data_thread = TobiiData.TobiiDataThread(tobii_controller)
+    tobii_data_thread.start()
+
+    # create tobii video thread
+    tobii_video_thread = TobiiVideo.TobiiVideoThread(tobii_controller)
+    tobii_video_thread.start()
+
+    # start tobii glasses streaming
+    tobii_controller.start_streaming()
+
+    # display loop
+    try:
+        
+        # wait 1ms between each frame until 'Esc' key is pressed
+        while cv.waitKey(1) != 27:
+        
+            frame, frame_width, frame_height, frame_time, pts = tobii_video_thread.read()
+
+            # draw tobii gaze
+            # TODO : sync gaze data according frame pts
+            gp_data = tobii_data_thread.read_gaze_data(pts)
+            if 'TIMESTAMP' in gp_data:
+                pointer = (int(gp_data['X'] * frame_width), int(gp_data['Y'] * frame_height))
+                cv.circle(frame, pointer, 4, (0, 255, 255), -1)
+            else:
+                pointer = (0, 0)
+            
+            # display frame
+            cv.imshow('Tobii Camera Record', frame)
+
+    # exit on 'ctrl+C' interruption
+    except KeyboardInterrupt:
+        pass
+
+    # stop frame display
+    cv.destroyAllWindows()
+
+    # stop tobii objects
+    tobii_video_thread.stop()
+    tobii_data_thread.stop()
+
+    tobii_controller.stop_streaming()
+    tobii_controller.close()
+
+if __name__ == '__main__':
+
+    main()
\ No newline at end of file