From 166e8de8c6d71e29f65c67ad93395ecd8f2ea6df Mon Sep 17 00:00:00 2001 From: Théo de la Hogue Date: Mon, 22 Apr 2024 12:01:24 +0200 Subject: Using pipeline step object enter method before private enter method. --- src/argaze/DataFeatures.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) (limited to 'src') diff --git a/src/argaze/DataFeatures.py b/src/argaze/DataFeatures.py index 08e0ef1..1bc43a8 100644 --- a/src/argaze/DataFeatures.py +++ b/src/argaze/DataFeatures.py @@ -678,9 +678,9 @@ def PipelineStepEnter(method): logging.debug('%s.__enter__', get_class_path(self)) - method(self) + PipelineStepObject.__enter__(self) - return PipelineStepObject.__enter__(self) + return method(self) return wrapper -- cgit v1.1 From c9f2536946d3fbef34cf4289db50c1dc5d260f18 Mon Sep 17 00:00:00 2001 From: Théo de la Hogue Date: Mon, 22 Apr 2024 12:01:55 +0200 Subject: Improving ArContext class documentation. --- src/argaze/ArFeatures.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) (limited to 'src') diff --git a/src/argaze/ArFeatures.py b/src/argaze/ArFeatures.py index a3ce1e3..58b23a0 100644 --- a/src/argaze/ArFeatures.py +++ b/src/argaze/ArFeatures.py @@ -1442,7 +1442,7 @@ DEFAULT_ARCONTEXT_IMAGE_PARAMETERS = { class ArContext(DataFeatures.PipelineStepObject): """ - Define class to ... + Defines abstract Python context manager to handle pipeline inputs. """ # noinspection PyMissingConstructor -- cgit v1.1 From 818156366467a8a48f419edb15eebeeef143535b Mon Sep 17 00:00:00 2001 From: Théo de la Hogue Date: Mon, 22 Apr 2024 12:02:46 +0200 Subject: Adding a new Random context demonstration. --- src/argaze/utils/contexts/Random.py | 77 +++++++++++++++++++++++++++++++ src/argaze/utils/demo/random_context.json | 12 +++++ 2 files changed, 89 insertions(+) create mode 100644 src/argaze/utils/contexts/Random.py create mode 100644 src/argaze/utils/demo/random_context.json (limited to 'src') 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 . +""" + +__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 diff --git a/src/argaze/utils/demo/random_context.json b/src/argaze/utils/demo/random_context.json new file mode 100644 index 0000000..bb18a41 --- /dev/null +++ b/src/argaze/utils/demo/random_context.json @@ -0,0 +1,12 @@ +{ + "argaze.utils.contexts.Random.GazePositionGenerator" : { + "name": "Random gaze position generator", + "range": [1920, 1149], + "pipeline": "gaze_analysis_pipeline.json", + "catch_exceptions": true, + "image_parameters": { + "draw_times": true, + "draw_exceptions": true + } + } +} \ No newline at end of file -- cgit v1.1 From 191da6950e7d4e9a7822b779ffb41dec3a1b7cb5 Mon Sep 17 00:00:00 2001 From: Théo de la Hogue Date: Mon, 22 Apr 2024 14:29:55 +0200 Subject: Moving position generation into a separated thread. --- src/argaze/utils/contexts/Random.py | 26 +++++++++++++++++++++----- 1 file changed, 21 insertions(+), 5 deletions(-) (limited to 'src') diff --git a/src/argaze/utils/contexts/Random.py b/src/argaze/utils/contexts/Random.py index 192e477..7fc4d91 100644 --- a/src/argaze/utils/contexts/Random.py +++ b/src/argaze/utils/contexts/Random.py @@ -18,6 +18,7 @@ __license__ = "GPLv3" import logging import time +import threading import random from argaze import ArFeatures, DataFeatures, GazeFeatures @@ -49,11 +50,23 @@ class GazePositionGenerator(ArFeatures.ArContext): logging.info('GazePositionGenerator context starts...') + # Create stop event + self.__stop_event = threading.Event() + + # Start gaze position generator thread + self.__gaze_thread = threading.Thread(target=self.__generate_gaze_position) + self.__gaze_thread.start() + + return self + + def __generate_gaze_position(self): + """Generate gaze position.""" + start_time = time.time() self.__x = int(self.range[0] / 2) self.__y = int(self.range[1] / 2) - while True: + while not self.__stop_event.is_set(): # Edit millisecond timestamp timestamp = int((time.time() - start_time) * 1e3) @@ -61,17 +74,20 @@ class GazePositionGenerator(ArFeatures.ArContext): 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) + logging.debug('> 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 + logging.info('GazePositionGenerator context ends...') + + # Stop gaze position generator thread + self.__stop_event.set() + threading.Thread.join(self.__gaze_thread) + \ No newline at end of file -- cgit v1.1