Adding fixation_count method.

1 files changed, 15 insertions, 0 deletions

diff --git a/src/argaze/GazeFeatures.py b/src/argaze/GazeFeatures.py
index 31290b5..a098dc4 100644
--- a/src/argaze/GazeFeatures.py
+++ b/src/argaze/GazeFeatures.py
@@ -652,6 +652,21 @@ class AOIScanPath(list):
 
             return None
 
+    def fixation_count(self):
+        """Get how many fixations are there in the scan path and how many fixation are there in each aoi."""
+
+        scan_fixation_count = 0
+        aoi_fixation_count = {aoi: 0 for aoi in self.__aoi_letter}
+
+        for aoi_scan_step in self:
+
+            step_fixation_count = len(aoi_scan_step.movements) - 1 # -1: to ignore last saccade
+
+            scan_fixation_count += step_fixation_count
+            aoi_fixation_count[aoi_scan_step.aoi] += step_fixation_count
+            
+        return scan_fixation_count, aoi_fixation_count
+
 class AOIScanPathAnalyzer():
     """Abstract class to define what should provide a aoi scan path analyzer."""