aboutsummaryrefslogtreecommitdiff
path: root/src/argaze/utils/record_tobii_session.py
blob: ecea08adc7660d8bd37664e8e8b047cb0ec2eed2 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
#!/usr/bin/env python

import argparse
import time

from argaze.TobiiGlassesPro2 import TobiiController

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 glasses streaming
    tobii_controller.stop_streaming()

if __name__ == '__main__':

    main()