aboutsummaryrefslogtreecommitdiff
path: root/src
diff options
context:
space:
mode:
Diffstat (limited to 'src')
-rw-r--r--src/argaze/TobiiGlassesPro2/TobiiController.py24
-rw-r--r--src/argaze/utils/record_tobii_session.py67
2 files changed, 85 insertions, 6 deletions
diff --git a/src/argaze/TobiiGlassesPro2/TobiiController.py b/src/argaze/TobiiGlassesPro2/TobiiController.py
index 10af5fe..0cc97f0 100644
--- a/src/argaze/TobiiGlassesPro2/TobiiController.py
+++ b/src/argaze/TobiiGlassesPro2/TobiiController.py
@@ -5,7 +5,7 @@ import tobiiglassesctrl
class TobiiController(tobiiglassesctrl.TobiiGlassesController):
"""As TobiiController inherits from TobiiGlassesPyController, here is its [code](https://github.com/ddetommaso/TobiiGlassesPyController/blob/master/tobiiglassesctrl/controller.py)."""
- project_id = None
+ project_name = None
"""Project identifier."""
participant_id = None
@@ -14,14 +14,24 @@ class TobiiController(tobiiglassesctrl.TobiiGlassesController):
calibration_id = None
"""Calibration identifier."""
+ recording_id = None
+ """Recording identifier."""
+
def __init__(self, ip_address, project_name, participant_id):
"""Create a project, a participant and start calibration."""
super().__init__(ip_address, video_scene = True)
- self.project_id = super().create_project(project_name)
- self.participant_id = super().create_participant(project_id, project_name)
+ self.project_name = project_name
+ self.project_id = super().create_project(self.project_name)
+ self.participant_id = super().create_participant(self.project_id, self.project_name)
+ def __del__(self):
+ pass
+
+ def calibrate(self):
+ """Start Tobii glasses calibration"""
+
input('Position Tobbi glasses calibration target then presse \'Enter\' to start calibration.')
self.calibration_id = super().create_calibration(self.project_id, self.participant_id)
super().start_calibration(self.calibration_id)
@@ -29,6 +39,8 @@ class TobiiController(tobiiglassesctrl.TobiiGlassesController):
if not super().wait_until_calibration_is_done(self.calibration_id):
raise Error('Tobii calibration failed')
- # destruction
- def __del__(self):
- pass
+ def record(self):
+ """Enable recording on the Tobii interface's SD Card"""
+
+ self.recording_id = super().create_recording(self.participant_id)
+ super().start_recording(self.recording_id)
diff --git a/src/argaze/utils/record_tobii_session.py b/src/argaze/utils/record_tobii_session.py
new file mode 100644
index 0000000..a92d818
--- /dev/null
+++ b/src/argaze/utils/record_tobii_session.py
@@ -0,0 +1,67 @@
+#!/usr/bin/env python
+
+import argparse
+import time
+
+from argaze.TobiiGlassesPro2 import *
+
+import cv2 as cv
+
+def main():
+ """
+ Record a Tobii Glasses Pro 2 session on Tobii interface's SD Card
+ """
+
+ # 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='Untitled project', help='project name')
+ parser.add_argument('-i', '--participant_id', metavar='PARTICIPANT_ID', type=int, default=1, help='participant identifier')
+ args = parser.parse_args()
+
+ # create tobii controller
+ tobii_controller = TobiiController.TobiiController(args.tobii_ip, args.project_name, args.participant_id)
+
+ # calibrate tobii glasses
+ tobii_controller.calibrate()
+
+ # start tobii glasses streaming
+ tobii_controller.start_streaming()
+
+ # start recording
+ tobii_controller.record()
+
+ # waiting until keyboard interruption
+ try:
+
+ last_battery_level = 0
+ time_count = 0
+ while True:
+
+ # print storage info each minutes
+ if time_count % 60 == 0:
+
+ print(tobii_controller.get_storage_info())
+
+ # print battery level each time it changes
+ battery_level = tobii_controller.get_battery_level()
+ if battery_level != last_battery_level:
+
+ print(tobii_controller.get_battery_info())
+ last_battery_level = battery_level
+
+ # sleep
+ time.sleep(1)
+ time_count += 1
+
+ # exit on keyboard interruption
+ except KeyboardInterrupt:
+ pass
+
+ # stop tobii controller
+ tobii_controller.stop_streaming()
+ tobii_controller.close()
+
+if __name__ == '__main__':
+
+ main() \ No newline at end of file