diff options
author | Théo de la Hogue | 2024-02-20 14:56:29 +0100 |
---|---|---|
committer | Théo de la Hogue | 2024-02-20 14:56:29 +0100 |
commit | 511f641239ee40f9fc142c9d2873298924e4bc41 (patch) | |
tree | 61fe1bd7b44ce62977789dd476c065702dd6841e /src | |
parent | b58c4bdae021213cd6cf93ce3f3643b32f54e5a0 (diff) | |
download | argaze-511f641239ee40f9fc142c9d2873298924e4bc41.zip argaze-511f641239ee40f9fc142c9d2873298924e4bc41.tar.gz argaze-511f641239ee40f9fc142c9d2873298924e4bc41.tar.bz2 argaze-511f641239ee40f9fc142c9d2873298924e4bc41.tar.xz |
Overwritting list clear method to correctly clear transition matrix and sequence when clearing aoi scan path.
Diffstat (limited to 'src')
-rw-r--r-- | src/argaze/GazeFeatures.py | 28 |
1 files changed, 17 insertions, 11 deletions
diff --git a/src/argaze/GazeFeatures.py b/src/argaze/GazeFeatures.py index 81a87c8..cfd7419 100644 --- a/src/argaze/GazeFeatures.py +++ b/src/argaze/GazeFeatures.py @@ -979,6 +979,20 @@ class AOIScanPath(list): self.__duration = 0 + def clear(self): + """Clear aoi scan steps list, letter sequence and transition matrix.""" + + super().clear() + + self.__movements = TimeStampedGazeMovements() + self.__current_aoi = '' + self.__index = ord('A') + self.__aoi_letter = {} + self.__letter_aoi = {} + + size = len(self.__expected_aoi) + self.__transition_matrix = pandas.DataFrame(numpy.zeros((size, size)), index=self.__expected_aoi, columns=self.__expected_aoi) + @property def duration(self) -> float: """Sum of all scan steps duration""" @@ -1043,20 +1057,12 @@ class AOIScanPath(list): !!! warning This will clear the AOIScanPath """ - self.clear() - + self.__expected_aoi = [OutsideAOI] self.__expected_aoi += expected_aoi - self.__movements = TimeStampedGazeMovements() - self.__current_aoi = '' - self.__index = ord('A') - self.__aoi_letter = {} - self.__letter_aoi = {} - - size = len(self.__expected_aoi) - self.__transition_matrix = pandas.DataFrame(numpy.zeros((size, size)), index=self.__expected_aoi, columns=self.__expected_aoi) - + self.clear() + @property def current_aoi(self): """AOI name of aoi scan step under construction""" |