1 files changed, 0 insertions, 333 deletions
diff --git a/src/argaze/TobiiGlassesPro2/_TobiiGazeData_temp.py b/src/argaze/TobiiGlassesPro2/_TobiiGazeData_temp.py
deleted file mode 100644
index 5579948..0000000
--- a/src/argaze/TobiiGlassesPro2/_TobiiGazeData_temp.py
+++ /dev/null
@@ -1,333 +0,0 @@
-#!/usr/bin/env python
-
-import math
-import os
-
-import numpy as np
-
-from sortedcontainers import SortedList, SortedDict
-
-from tobiiglasses.livedata import *
-
-import tobiiglasses.events
-
-def import_json_items_from_gzipfile(source_dir, filename, decode_JSON=None):
-    logging.info('Importing JSON items from %s in %s' % (filename, source_dir))
-    with gzip.open(os.path.join(source_dir, filename)) as f:
-        for item in f:
-            json.loads(item.decode('utf-8'), object_hook=decode_JSON)
-
-class GazeItem:
-
-    def __init__(self, label, dtype):
-        self.__label__ = label
-        self.__data__ = SortedDict({})
-        self.__dtype__ = dtype
-
-    def __getitem__(self, key):
-        return self.__data__[key]
-
-    def __setitem__(self, key, value):
-        self.__data__[key] = value
-
-    def getData(self):
-        return self.__data__
-
-    def getLabel(self):
-        return self.__label__
-
-    def getType(self):
-        return self.__dtype__
-
-    def keys(self):
-        return list(self.__data__.keys())
-
-    def pop(self, key):
-        self.__data__.pop(key)
-
-    def values(self):
-        return list(self.__data__.values())
-
-class GazeData:
-
-    Timestamp = "Timestamp"
-    Gidx = "Gaze Index"
-    LoggedEvents = "Logged Events"
-    JSONEvents = "JSON Events"
-    GazePositionX = "Gaze Position X"
-    GazePositionY = "Gaze Position Y"
-    Gaze3DPositionX = "Gaze 3D Position X"
-    Gaze3DPositionY = "Gaze 3D Position Y"
-    Gaze3DPositionZ = "Gaze 3D Position Z"
-    Depth = "Gaze Depth"
-    Tilt = "Gaze Tilt"
-    Vergence = "Gaze Vergence"
-    Version = "Gaze Version"
-    GazeDirectionX_Left = "Gaze Direction Left X"
-    GazeDirectionY_Left = "Gaze Direction Left Y"
-    GazeDirectionZ_Left = "Gaze Direction Left Z"
-    GazeDirectionX_Right = "Gaze Direction Right X"
-    GazeDirectionY_Right = "Gaze Direction Right Y"
-    GazeDirectionZ_Right = "Gaze Direction Right Z"
-    GazePixelX = "Gaze Pixel X"
-    GazePixelY = "Gaze Pixel Y"
-    PupilCenterX_Left = "Pupil Center Left X"
-    PupilCenterY_Left = "Pupil Center Left Y"
-    PupilCenterZ_Left = "Pupil Center Left Z"
-    PupilCenterX_Right = "Pupil Center Right X"
-    PupilCenterY_Right = "Pupil Center Right Y"
-    PupilCenterZ_Right = "Pupil Center Right Z"
-    PupilDiameter_Left = "Pupil Diameter Left"
-    PupilDiameter_Right = "Pupil Diameter Right"
-
-
-    def __init__(self, segment):
-
-        self.__gazedata__ = {}
-        self.__gd_right__ = {}
-        self.__gd_left__ = {}
-        self.__experiment_vars__ = {} # keys: var_name, values: {keys: ts,  values: var_value}
-        self.__experiment_vars_headers__ = []
-
-        self.__rec_id__ = None
-        self.__pa_name__ = None
-        self.__pr_name__ = None
-        self.__segment__ = segment
-
-        self.__FRAME_HEIGHT__ = segment.getFrameHeight()
-        self.__FRAME_WIDTH__ = segment.getFrameWidth()
-        self.__FRAME_FPS__ = segment.getFrameFPS()
-        self.__vts__ = SortedDict({})
-
-        self.__init_datatypes__()
-        self.__importGazeData__(segment)
-        self.__importExpVars__()
-
-    def __addItem__(self, livedatajson_item):
-        if livedatajson_item is None:
-            return
-        if livedatajson_item.s.getValue() != 0: # Discard invalid samples
-            return
-        ts = livedatajson_item.ts.getValue()/1000.0
-
-        if isinstance(livedatajson_item, VTS):
-            self.__vts__[ts] = livedatajson_item.vts.getValue()/1000.0
-            return
-        else:
-            self.__gazedata__[GazeData.Timestamp][ts] = ts # Conversion in milliseconds
-
-        if isinstance(livedatajson_item, APISynch):
-            tvalue = livedatajson_item.type.getValue()
-            tag = livedatajson_item.tag.getValue()
-            tag_value = livedatajson_item.value.getValue()
-            if tvalue != "JsonEvent":
-                if tvalue.startswith('#') and tvalue.endswith('#'):
-                    var_name = tvalue[1:-1]
-                    if not var_name in self.__experiment_vars__.keys():
-                        self.__experiment_vars__[var_name] = SortedDict({})
-                    self.__experiment_vars__[var_name][ts] = tag
-
-                elif tvalue.startswith('@') and tvalue.endswith('@'):
-                    vars_list = eval(tvalue[1:-1])
-                    values_list = eval(tag)
-                    for i in range(0, len(vars_list)):
-                        var_name = vars_list[i]
-                        value = values_list[i]
-                        if not var_name in self.__experiment_vars__.keys():
-                            self.__experiment_vars__[var_name] = SortedDict({})
-                        self.__experiment_vars__[var_name][ts] = value
-                else:
-                    self.__gazedata__[GazeData.LoggedEvents][ts] = tvalue
-
-            else:
-                #self.__gazedata__[GazeData.Timestamp].pop(ts)
-                self.__gazedata__[GazeData.JSONEvents][ts] = tag_value
-            return
-        else:
-            gidx = livedatajson_item.gidx.getValue()
-            self.__gazedata__[GazeData.Gidx][ts] = gidx
-        if isinstance(livedatajson_item, GazePosition):
-            self.__gazedata__[GazeData.GazePositionX][ts] = livedatajson_item.gp.getValue()[0]
-            self.__gazedata__[GazeData.GazePositionY][ts] = livedatajson_item.gp.getValue()[1]
-            self.__gazedata__[GazeData.GazePixelX][ts] = int(self.__FRAME_WIDTH__*self.__gazedata__[GazeData.GazePositionX][ts])
-            self.__gazedata__[GazeData.GazePixelY][ts] = int(self.__FRAME_HEIGHT__*self.__gazedata__[GazeData.GazePositionY][ts])
-            return
-        elif isinstance(livedatajson_item, GazePosition3d):
-            self.__gazedata__[GazeData.Gaze3DPositionX][ts] = livedatajson_item.gp3.getValue()[0]
-            self.__gazedata__[GazeData.Gaze3DPositionY][ts] = livedatajson_item.gp3.getValue()[1]
-            self.__gazedata__[GazeData.Gaze3DPositionZ][ts] = livedatajson_item.gp3.getValue()[2]
-            self.__gazedata__[GazeData.Depth][ts] = (self.__gazedata__[GazeData.Gaze3DPositionX][ts]**2 + self.__gazedata__[GazeData.Gaze3DPositionY][ts]**2 + self.__gazedata__[GazeData.Gaze3DPositionZ][ts]**2)**0.5
-            self.__gazedata__[GazeData.Tilt][ts] = math.atan(self.__gazedata__[GazeData.Gaze3DPositionY][ts]/self.__gazedata__[GazeData.Gaze3DPositionZ][ts])*180./math.pi
-            return
-        elif isinstance(livedatajson_item, GazeDirection):
-            if livedatajson_item.eye.getValue() == "left":
-                self.__gazedata__[GazeData.GazeDirectionX_Left][ts] = livedatajson_item.gd.getValue()[0]
-                self.__gazedata__[GazeData.GazeDirectionY_Left][ts] = livedatajson_item.gd.getValue()[1]
-                self.__gazedata__[GazeData.GazeDirectionZ_Left][ts] = livedatajson_item.gd.getValue()[2]
-                self.__gazedata__[GazeData.Version][ts] = math.acos(self.__gazedata__[GazeData.GazeDirectionZ_Left][ts])*180./math.pi
-                self.__gd_left__[gidx] = np.array([self.__gazedata__[GazeData.GazeDirectionX_Left][ts],self.__gazedata__[GazeData.GazeDirectionY_Left][ts], self.__gazedata__[GazeData.GazeDirectionZ_Left][ts]])
-                return
-            elif livedatajson_item.eye.getValue() == "right":
-                self.__gazedata__[GazeData.GazeDirectionX_Right][ts] = livedatajson_item.gd.getValue()[0]
-                self.__gazedata__[GazeData.GazeDirectionY_Right][ts] = livedatajson_item.gd.getValue()[1]
-                self.__gazedata__[GazeData.GazeDirectionZ_Right][ts] = livedatajson_item.gd.getValue()[2]
-                self.__gazedata__[GazeData.Version][ts] = math.acos(self.__gazedata__[GazeData.GazeDirectionZ_Right][ts])*180./math.pi
-                self.__gd_right__[gidx] = np.array([self.__gazedata__[GazeData.GazeDirectionX_Right][ts], self.__gazedata__[GazeData.GazeDirectionY_Right][ts], self.__gazedata__[GazeData.GazeDirectionZ_Right][ts]])
-            try:
-                self.__gazedata__[GazeData.Vergence][ts] = math.acos( np.dot(self.__gd_left__[gidx], self.__gd_right__[gidx])/(np.linalg.norm(self.__gd_left__[gidx])*np.linalg.norm(self.__gd_right__[gidx])) )*180./math.pi
-            except:
-                pass
-        elif isinstance(livedatajson_item, PupilCenter):
-            if livedatajson_item.eye.getValue() == "left":
-                self.__gazedata__[GazeData.PupilCenterX_Left][ts] = livedatajson_item.pc.getValue()[0]
-                self.__gazedata__[GazeData.PupilCenterY_Left][ts] = livedatajson_item.pc.getValue()[1]
-                self.__gazedata__[GazeData.PupilCenterZ_Left][ts] = livedatajson_item.pc.getValue()[2]
-                return
-            elif livedatajson_item.eye.getValue() == "right":
-                self.__gazedata__[GazeData.PupilCenterX_Right][ts] = livedatajson_item.pc.getValue()[0]
-                self.__gazedata__[GazeData.PupilCenterY_Right][ts] = livedatajson_item.pc.getValue()[1]
-                self.__gazedata__[GazeData.PupilCenterZ_Right][ts] = livedatajson_item.pc.getValue()[2]
-                return
-        elif isinstance(livedatajson_item, PupilDiameter):
-            if livedatajson_item.eye.getValue() == "left":
-                self.__gazedata__[GazeData.PupilDiameter_Left][ts] = livedatajson_item.pd.getValue()
-                return
-            elif livedatajson_item.eye.getValue() == "right":
-                self.__gazedata__[GazeData.PupilDiameter_Right][ts] = livedatajson_item.pd.getValue()
-                return
-
-    def __decodeJSON__(self, json_item):
-
-        if TobiiJSONProperties.PupilCenter.key in json_item:
-            item = PupilCenter(json_item)
-        elif TobiiJSONProperties.PupilDiameter.key in json_item:
-            item = PupilDiameter(json_item)
-        elif TobiiJSONProperties.GazeDirection.key in json_item:
-            item = GazeDirection(json_item)
-        elif TobiiJSONProperties.GazePosition.key in json_item:
-            item = GazePosition(json_item)
-        elif TobiiJSONProperties.GazePosition3d.key in json_item:
-            item = GazePosition3d(json_item)
-        elif TobiiJSONProperties.APISynch_ETS.key in json_item:
-            item = APISynch(json_item)
-        elif TobiiJSONProperties.VTS.key in json_item:
-            item = VTS(json_item)
-        else:
-            item = None
-            
-        self.__addItem__(item)
-
-    def __importExpVars__(self):
-        self.__experiment_vars_headers__ = []
-        for k,v in self.__experiment_vars__.items():
-            current_var = None
-            self.__gazedata__[k] = GazeItem(k, np.dtype(str))
-            self.__experiment_vars_headers__.append(k)
-            for ts in self.__gazedata__[GazeData.Timestamp].getData().values():
-                try:
-                    current_var = v[ts]
-                except:
-                    pass
-                self.__gazedata__[k][ts] = current_var
-
-    def __importGazeData__(self, segment):
-        filepath = segment.getSegmentPath()
-        logging.info('Importing segment %s in %s' % (segment.getId(), filepath))
-        tobiiglasses.utils.import_json_items_from_gzipfile(filepath, tobiiglasses.entities.TobiiSegment.livedata_filename, self.__decodeJSON__)
-
-    def __init_datatypes__(self):
-        self.__gazedata__[GazeData.Timestamp] = GazeItem(GazeData.Timestamp, np.dtype('float'))
-        self.__gazedata__[GazeData.Gidx] = GazeItem(GazeData.Gidx, np.dtype('u4'))
-        self.__gazedata__[GazeData.LoggedEvents] = GazeItem(GazeData.LoggedEvents, np.dtype(object))
-        self.__gazedata__[GazeData.JSONEvents] = GazeItem(GazeData.JSONEvents, np.dtype(object))
-        self.__gazedata__[GazeData.GazePositionX] = GazeItem(GazeData.GazePositionX, np.dtype('float'))
-        self.__gazedata__[GazeData.GazePositionY] = GazeItem(GazeData.GazePositionY, np.dtype('float'))
-        self.__gazedata__[GazeData.GazePixelX] = GazeItem(GazeData.GazePixelX, np.dtype('u4'))
-        self.__gazedata__[GazeData.GazePixelY] = GazeItem(GazeData.GazePixelY, np.dtype('u4'))
-        self.__gazedata__[GazeData.Gaze3DPositionX] = GazeItem(GazeData.Gaze3DPositionX, np.dtype('float'))
-        self.__gazedata__[GazeData.Gaze3DPositionY] = GazeItem(GazeData.Gaze3DPositionY, np.dtype('float'))
-        self.__gazedata__[GazeData.Gaze3DPositionZ] = GazeItem(GazeData.Gaze3DPositionZ, np.dtype('float'))
-        self.__gazedata__[GazeData.Depth] = GazeItem(GazeData.Depth, np.dtype('float'))
-        self.__gazedata__[GazeData.Vergence] = GazeItem(GazeData.Vergence, np.dtype('float'))
-        self.__gazedata__[GazeData.Version] = GazeItem(GazeData.Version, np.dtype('float'))
-        self.__gazedata__[GazeData.Tilt] = GazeItem(GazeData.Tilt, np.dtype('float'))
-        self.__gazedata__[GazeData.GazeDirectionX_Left] = GazeItem(GazeData.GazeDirectionX_Left, np.dtype('float'))
-        self.__gazedata__[GazeData.GazeDirectionY_Left] = GazeItem(GazeData.GazeDirectionY_Left, np.dtype('float'))
-        self.__gazedata__[GazeData.GazeDirectionZ_Left] = GazeItem(GazeData.GazeDirectionZ_Left, np.dtype('float'))
-        self.__gazedata__[GazeData.GazeDirectionX_Right] = GazeItem(GazeData.GazeDirectionX_Right, np.dtype('float'))
-        self.__gazedata__[GazeData.GazeDirectionY_Right] = GazeItem(GazeData.GazeDirectionY_Right, np.dtype('float'))
-        self.__gazedata__[GazeData.GazeDirectionZ_Right] = GazeItem(GazeData.GazeDirectionZ_Right, np.dtype('float'))
-        self.__gazedata__[GazeData.PupilCenterX_Left] = GazeItem(GazeData.PupilCenterX_Left, np.dtype('float'))
-        self.__gazedata__[GazeData.PupilCenterY_Left] = GazeItem(GazeData.PupilCenterY_Left, np.dtype('float'))
-        self.__gazedata__[GazeData.PupilCenterZ_Left] = GazeItem(GazeData.PupilCenterZ_Left, np.dtype('float'))
-        self.__gazedata__[GazeData.PupilCenterX_Right] = GazeItem(GazeData.PupilCenterX_Right, np.dtype('float'))
-        self.__gazedata__[GazeData.PupilCenterY_Right] = GazeItem(GazeData.PupilCenterY_Right, np.dtype('float'))
-        self.__gazedata__[GazeData.PupilCenterZ_Right] = GazeItem(GazeData.PupilCenterZ_Right, np.dtype('float'))
-        self.__gazedata__[GazeData.PupilDiameter_Left] = GazeItem(GazeData.PupilDiameter_Left, np.dtype('float'))
-        self.__gazedata__[GazeData.PupilDiameter_Right] = GazeItem(GazeData.PupilDiameter_Right, np.dtype('float'))
-
-    def getData(self):
-        return self.__gazedata__.values
-
-    def getExpVarsHeaders(self):
-        return self.__experiment_vars_headers__
-
-    def getFrameHeight(self):
-        return self.__FRAME_HEIGHT__
-
-    def getFrameWidth(self):
-        return self.__FRAME_WIDTH__
-
-    def getFrameFPS(self):
-        return self.__FRAME_FPS__
-
-    def getLoggedEvents(self):
-        return self.__gazedata__[GazeData.LoggedEvents].getData()
-
-    def getJSONEvents(self):
-        return self.__gazedata__[GazeData.JSONEvents].getData()
-
-    def getTimestamps(self):
-        return self.__gazedata__[GazeData.Timestamp].values()
-
-    def getVTS(self):
-        return self.__vts__
-
-    def importEvents(self, events):
-        self.__gazedata__[tobiiglasses.events.GazeEvents.GazeType] = GazeItem(tobiiglasses.events.GazeEvents.GazeType, np.dtype(object))
-        self.__gazedata__[tobiiglasses.events.GazeEvents.Fixation_X] = GazeItem(tobiiglasses.events.GazeEvents.Fixation_X, np.dtype('u4'))
-        self.__gazedata__[tobiiglasses.events.GazeEvents.Fixation_Y] = GazeItem(tobiiglasses.events.GazeEvents.Fixation_Y, np.dtype('u4'))
-        self.__gazedata__[tobiiglasses.events.GazeEvents.EventIndex] = GazeItem(tobiiglasses.events.GazeEvents.EventIndex, np.dtype('u4'))
-        self.__gazedata__[tobiiglasses.events.GazeEvents.EventDuration] = GazeItem(tobiiglasses.events.GazeEvents.EventDuration, np.dtype('u4'))
-        self.__gazedata__[tobiiglasses.events.GazeEvents.AOI_Mapped_Fixation_X] = GazeItem(tobiiglasses.events.GazeEvents.AOI_Mapped_Fixation_X, np.dtype('u4'))
-        self.__gazedata__[tobiiglasses.events.GazeEvents.AOI_Mapped_Fixation_Y] = GazeItem(tobiiglasses.events.GazeEvents.AOI_Mapped_Fixation_Y, np.dtype('u4'))
-        self.__gazedata__[tobiiglasses.events.GazeEvents.AOI] = GazeItem(tobiiglasses.events.GazeEvents.AOI, np.dtype(object))
-        self.__gazedata__[tobiiglasses.events.GazeEvents.AOI_Score] = GazeItem(tobiiglasses.events.GazeEvents.AOI_Score, np.dtype('f2'))
-
-        events = events.getEvents()
-        events_ts = events[GazeData.Timestamp].getData().values()
-
-        evt_idx = 0
-        evt_ts = 0
-        evt_duration = 0
-        for ts in self.__gazedata__[GazeData.Timestamp].getData().values():
-
-            if (events_ts[evt_idx] - ts) <= 10.0:
-                evt_ts = events_ts[evt_idx]
-                evt_duration = events[tobiiglasses.events.GazeEvents.EventDuration][evt_ts]
-                if events_ts[evt_idx] != events_ts[-1]:
-                    evt_idx+=1
-
-            if ts <= (evt_ts + evt_duration):
-                self.__gazedata__[tobiiglasses.events.GazeEvents.GazeType][ts] = events[tobiiglasses.events.GazeEvents.GazeType][evt_ts]
-                self.__gazedata__[tobiiglasses.events.GazeEvents.EventIndex][ts] = events[tobiiglasses.events.GazeEvents.EventIndex][evt_ts]
-                self.__gazedata__[tobiiglasses.events.GazeEvents.EventDuration][ts] = evt_duration
-                self.__gazedata__[tobiiglasses.events.GazeEvents.Fixation_X][ts] = events[tobiiglasses.events.GazeEvents.Fixation_X][evt_ts]
-                self.__gazedata__[tobiiglasses.events.GazeEvents.Fixation_Y][ts] = events[tobiiglasses.events.GazeEvents.Fixation_Y][evt_ts]
-                try:
-                    self.__gazedata__[tobiiglasses.events.GazeEvents.AOI][ts] = events[tobiiglasses.events.GazeEvents.AOI][evt_ts]
-                    self.__gazedata__[tobiiglasses.events.GazeEvents.AOI_Score][ts] = events[tobiiglasses.events.GazeEvents.AOI][evt_ts]
-                    self.__gazedata__[tobiiglasses.events.GazeEvents.AOI_Mapped_Fixation_X][ts] = events[tobiiglasses.events.GazeEvents.AOI_Mapped_Fixation_X][evt_ts]
-                    self.__gazedata__[tobiiglasses.events.GazeEvents.AOI_Mapped_Fixation_Y][ts] = events[tobiiglasses.events.GazeEvents.AOI_Mapped_Fixation_Y][evt_ts]
-                except:
-                    pass