Catching AOISceneMissing exception.

1 files changed, 40 insertions, 25 deletions

diff --git a/src/argaze/GazeFeatures.py b/src/argaze/GazeFeatures.py
index acdbec8..4d7103b 100644
--- a/src/argaze/GazeFeatures.py
+++ b/src/argaze/GazeFeatures.py
@@ -27,10 +27,13 @@ class TimeStampedGazeAccuracies(DataStructures.TimeStampedBuffer):
     def __setitem__(self, key, value: GazeAccuracy):
         super().__setitem__(key, value)
 
-class InvalidGazeData(Exception):
-    """Exception to raise when gaze position or accuracy can't be processed."""
+class GazeDataMissing(Exception):
+    """Exception to raise when gaze position or accuracy is missing."""
     pass
 
+GazePositionMissingItem = GazePosition
+GazeAccuracyMissingItem = GazeAccuracy
+
 @dataclass
 class Movement():
     """Define abstract movement class."""
@@ -308,41 +311,53 @@ class PointerBasedVisualScan(VisualScanGenerator):
 
             (ts_current, aoi_scene_current) = self.__ts_aoi_scenes.pop_first()
 
-            try:
+            # is aoi scene a missing exception ?
+            try: raise aoi_scene_current
+
+            # when aoi scene is missing
+            except AOIFeatures.AOISceneMissing as e:
 
-                gaze_position = self.__ts_gaze_positions[ts_current]
+                #print(ts_current, e)
+                pass
 
-                for name, aoi in aoi_scene_current.items():
+            # when aoi scene is not missing
+            except:
 
-                    looked = aoi.looked(gaze_position)
+                try: 
 
-                    if looked:
+                    gaze_position = self.__ts_gaze_positions[ts_current]
 
-                        if not name in self.__step_dict.keys():
+                    for name, aoi in aoi_scene_current.items():
 
-                            # aoi starts to be looked
-                            self.__step_dict[name] = {
-                                'start': ts_current,
-                                'look_at': DataStructures.TimeStampedBuffer()
-                            }
+                        looked = aoi.looked(gaze_position)
 
-                        # store where the aoi is looked for 4 corners aoi
-                        if len(aoi) == 4:
-                            self.__step_dict[name]['look_at'][round(ts_current)] = aoi.look_at(gaze_position)
+                        if looked:
 
-                    elif name in self.__step_dict.keys():
+                            if not name in self.__step_dict.keys():
 
-                        ts_start = self.__step_dict[name]['start']
+                                # aoi starts to be looked
+                                self.__step_dict[name] = {
+                                    'start': ts_current,
+                                    'look_at': DataStructures.TimeStampedBuffer()
+                                }
 
-                        # aoi stops to be looked
-                        yield VisualScanStep(round(ts_start), round(ts_current - ts_start), name, self.__step_dict[name]['look_at'])
+                            # store where the aoi is looked for 4 corners aoi
+                            if len(aoi) == 4:
+                                self.__step_dict[name]['look_at'][round(ts_current)] = aoi.look_at(gaze_position)
 
-                        # forget the aoi
-                        del self.__step_dict[name]
+                        elif name in self.__step_dict.keys():
 
-            # ignore missing gaze position
-            except KeyError:
-                pass
+                            ts_start = self.__step_dict[name]['start']
+
+                            # aoi stops to be looked
+                            yield VisualScanStep(round(ts_start), round(ts_current - ts_start), name, self.__step_dict[name]['look_at'])
+
+                            # forget the aoi
+                            del self.__step_dict[name]
+
+                # ignore missing gaze position
+                except KeyError:
+                    pass
 
         # close started steps
         for name, step in self.__step_dict.items():