Adding AOIStep letter managment.

1 files changed, 13 insertions, 1 deletions

diff --git a/src/argaze/GazeFeatures.py b/src/argaze/GazeFeatures.py
index a26b1f1..31290b5 100644
--- a/src/argaze/GazeFeatures.py
+++ b/src/argaze/GazeFeatures.py
@@ -558,6 +558,7 @@ class AOIScanPath(list):
         self.__current_aoi = ''
         self.__index = ord('A')
         self.__aoi_letter = {}
+        self.__letter_aoi = {}
 
     def __repr__(self):
         """String representation."""
@@ -577,6 +578,11 @@ class AOIScanPath(list):
             self.__index += 1
             return letter
 
+    def get_letter_aoi(self, letter):
+        """Get which aoi is related to an unique letter."""
+
+        return self.__letter_aoi[letter]
+
     def __str__(self) -> str:
         """Convert aoi scan path into a string with unique letter per aoi step."""
 
@@ -611,8 +617,14 @@ class AOIScanPath(list):
 
             try: 
 
+                # Edit unique letter per aoi
+                letter = self.__get_aoi_letter(self.__current_aoi)
+
+                # Remember which letter identify which aoi
+                self.__letter_aoi[letter] = self.__current_aoi
+
                 # Edit new step
-                new_step = AOIScanStep(self.__movements, self.__current_aoi, self.__get_aoi_letter(self.__current_aoi))
+                new_step = AOIScanStep(self.__movements, self.__current_aoi, letter)
 
                 # Append new step
                 super().append(new_step)