WIP

src/pymovements/dataset/dataset.py

@@ -137,6 +137,9 @@
         if self.definition.has_files['precomputed_events']:
             self.load_precomputed_events()
 
+        if self.definition.has_files['precomputed_aois']:
+            self.load_precomputed_aois()
+
         # Reading measures files precomuted by authors of the dataset
         if self.definition.has_files['precomputed_reading_measures']:
             self.load_precomputed_reading_measures()
@@ -213,6 +216,15 @@
         )
         return self
 
+    def load_precomputed_aois(self) -> None:
+        """Load precomputed aois."""
+        self._check_fileinfo()
+        self.precomputed_events = dataset_files.load_precomputed_event_files(
+            self.definition,
+            self.fileinfo['precomputed_aois'],
+            self.paths,
+        )
+
     def load_precomputed_events(self) -> None:
         """Load precomputed events."""
         self._check_fileinfo()

src/pymovements/datasets/sb_sat.py

@@ -132,6 +132,7 @@ class SBSAT(DatasetDefinition):
             'gaze': True,
             'precomputed_events': True,
             'precomputed_reading_measures': False,
+            'precomputed_aois': True,
         },
     )
     mirrors: dict[str, tuple[str, ...]] = field(
@@ -143,6 +144,9 @@ class SBSAT(DatasetDefinition):
                 'precomputed_events': (
                     'https://raw.githubusercontent.com/ahnchive/SB-SAT/master/fixation/',
                 ),
+                'precomputed_aois': (
+                    'https://raw.githubusercontent.com/ahnchive/SB-SAT/master/fixation/',
+                ),
             },
     )
     resources: dict[str, tuple[dict[str, str], ...]] = field(