From 4ba61fe0b429e9bd68ea3f86383ad6335aee23dc Mon Sep 17 00:00:00 2001
From: Théo de la Hogue
Date: Tue, 2 Apr 2024 18:02:00 +0200
Subject: Fixing image.

---
 src/argaze/ArFeatures.py                      | 2 +-
 src/argaze/AreaOfInterest/AOIFeatures.py      | 1 -
 src/argaze/utils/contexts/TobiiProGlasses2.py | 4 ++--
 3 files changed, 3 insertions(+), 4 deletions(-)

diff --git a/src/argaze/ArFeatures.py b/src/argaze/ArFeatures.py
index da39c0b..2c29d3d 100644
--- a/src/argaze/ArFeatures.py
+++ b/src/argaze/ArFeatures.py
@@ -833,7 +833,7 @@ class ArFrame(DataFeatures.SharedObject, DataFeatures.PipelineStepObject):
 				logging.debug('\t> drawing background and heatmap')
 
 				background_image = self.__background.copy()
-				heatmap_image = cv2.resize(self.__heatmap.image, dsize=self.__size, interpolation=cv2.INTER_LINEAR)
+				heatmap_image = cv2.resize(self.__heatmap.image(), dsize=self.__size, interpolation=cv2.INTER_LINEAR)
 				image = cv2.addWeighted(heatmap_image, heatmap_weight, background_image, background_weight, 0)
 
 			# Draw heatmap only
diff --git a/src/argaze/AreaOfInterest/AOIFeatures.py b/src/argaze/AreaOfInterest/AOIFeatures.py
index 6c5f589..130390c 100644
--- a/src/argaze/AreaOfInterest/AOIFeatures.py
+++ b/src/argaze/AreaOfInterest/AOIFeatures.py
@@ -639,7 +639,6 @@ class Heatmap(DataFeatures.PipelineStepObject):
     @DataFeatures.PipelineStepImage
     def image(self):
         """Get heatmap image."""
-
         try:
 
             return self.__image
diff --git a/src/argaze/utils/contexts/TobiiProGlasses2.py b/src/argaze/utils/contexts/TobiiProGlasses2.py
index fc2e3f0..a51d325 100644
--- a/src/argaze/utils/contexts/TobiiProGlasses2.py
+++ b/src/argaze/utils/contexts/TobiiProGlasses2.py
@@ -587,7 +587,7 @@ class LiveStream(ArFeatures.ArContext):
 		threading.Thread.join(self.__video_thread)
 
 	@DataFeatures.PipelineStepImage
-	def image(self, draw_something: bool = None) -> numpy.array:
+	def image(self, draw_something: bool = None, **kwargs: dict) -> numpy.array:
 		"""Get Tobbi visualisation.
 
 		Parameters:
@@ -596,7 +596,7 @@ class LiveStream(ArFeatures.ArContext):
 		"""
 
 		# Get context image
-		image = super().image()
+		image = super().image(**kwargs)
 
 		if draw_something:
 
-- 
cgit v1.1