diff options
Diffstat (limited to 'src/argaze/utils/contexts/Random.py')
-rw-r--r-- | src/argaze/utils/contexts/Random.py | 77 |
1 files changed, 77 insertions, 0 deletions
diff --git a/src/argaze/utils/contexts/Random.py b/src/argaze/utils/contexts/Random.py new file mode 100644 index 0000000..192e477 --- /dev/null +++ b/src/argaze/utils/contexts/Random.py @@ -0,0 +1,77 @@ +"""Define eye tracking data file context + +This program is free software: you can redistribute it and/or modify it under +the terms of the GNU General Public License as published by the Free Software +Foundation, either version 3 of the License, or (at your option) any later +version. +This program is distributed in the hope that it will be useful, but WITHOUT +ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or FITNESS +FOR A PARTICULAR PURPOSE. See the GNU General Public License for more details. +You should have received a copy of the GNU General Public License along with +this program. If not, see <https://www.gnu.org/licenses/>. +""" + +__author__ = "Théo de la Hogue" +__credits__ = [] +__copyright__ = "Copyright 2023, Ecole Nationale de l'Aviation Civile (ENAC)" +__license__ = "GPLv3" + +import logging +import time +import random + +from argaze import ArFeatures, DataFeatures, GazeFeatures + +class GazePositionGenerator(ArFeatures.ArContext): + + @DataFeatures.PipelineStepInit + def __init__(self, **kwargs): + + # Init ArContext class + super().__init__() + + # Init private attribute + self.__range = (0, 0) + self.__x = 0 + self.__y = 0 + + @property + def range(self) -> tuple[int, int]: + """Maximal ranges for X and Y axis""" + return self.__range + + @range.setter + def range(self, range: tuple[int, int]): + self.__range = range + + @DataFeatures.PipelineStepEnter + def __enter__(self): + + logging.info('GazePositionGenerator context starts...') + + start_time = time.time() + self.__x = int(self.range[0] / 2) + self.__y = int(self.range[1] / 2) + + while True: + + # Edit millisecond timestamp + timestamp = int((time.time() - start_time) * 1e3) + + self.__x += random.randint(-10, 10) + self.__y += random.randint(-10, 10) + + logging.info('> timestamp=%i, x=%i, y=%i', timestamp, self.__x, self.__y) + + # Process timestamped random gaze position + self._process_gaze_position(timestamp = timestamp, x = self.__x, y = self.__y) + + # wait 10ms + time.sleep(0.01) + + return self + + @DataFeatures.PipelineStepExit + def __exit__(self, exception_type, exception_value, exception_traceback): + + logging.info('GazePositionGenerator context ends...')
\ No newline at end of file |