Adding debug property to ArContext. Logging image function.

1 files changed, 39 insertions, 4 deletions

diff --git a/src/argaze/ArFeatures.py b/src/argaze/ArFeatures.py
index 91b63ac..56ea41d 100644
--- a/src/argaze/ArFeatures.py
+++ b/src/argaze/ArFeatures.py
@@ -843,14 +843,20 @@ class ArFrame(DataFeatures.SharedObject, DataFeatures.PipelineStepObject):
 		# Use frame lock feature
 		with self._lock:
 
+			logging.debug('ArFrame.__image %s', self.name)
+
 			# Draw background only
 			if background_weight is not None and (heatmap_weight is None or self.__heatmap is None):
 
+				logging.debug('\t> drawing background only')
+
 				image = self.__background.copy()
 
 			# Draw mix background and heatmap if required
 			elif background_weight is not None and heatmap_weight is not None and self.__heatmap:
 
+				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)
 				image = cv2.addWeighted(heatmap_image, heatmap_weight, background_image, background_weight, 0)
@@ -858,21 +864,29 @@ class ArFrame(DataFeatures.SharedObject, DataFeatures.PipelineStepObject):
 			# Draw heatmap only
 			elif background_weight is None and heatmap_weight is not None and self.__heatmap:
 
+				logging.debug('\t> drawing heatmap only')
+
 				image = cv2.resize(self.__heatmap.image, dsize=self.__size, interpolation=cv2.INTER_LINEAR)
 
 			# Draw black image
 			else:
 
+				logging.debug('\t> drawing black image')
+
 				image = numpy.full((self.__size[1], self.__size[0], 3), 0).astype(numpy.uint8)
 
 			# Draw gaze position calibrator
 			if draw_gaze_position_calibrator is not None:
 
+				logging.debug('\t> drawing gaze position calibrator')
+
 				self.__gaze_position_calibrator.draw(image, size=self.__size, **draw_gaze_position_calibrator)
 
 			# Draw scan path if required
 			if draw_scan_path is not None and self.__scan_path is not None:
 
+				logging.debug('\t> drawing scan path')
+
 				self.__scan_path.draw(image, **draw_scan_path)
 
 			# Draw current fixation if required
@@ -880,6 +894,8 @@ class ArFrame(DataFeatures.SharedObject, DataFeatures.PipelineStepObject):
 
 				if self.__gaze_movement_identifier.current_fixation():
 
+					logging.debug('\t> drawing current fixation')
+
 					self.__gaze_movement_identifier.current_fixation().draw(image, **draw_fixations)
 
 			# Draw current saccade if required
@@ -887,6 +903,8 @@ class ArFrame(DataFeatures.SharedObject, DataFeatures.PipelineStepObject):
 
 				if self.__gaze_movement_identifier.current_saccade():
 
+					logging.debug('\t> drawing current saccade')
+
 					self.__gaze_movement_identifier.current_saccade().draw(image, **draw_saccades)
 
 			# Draw layers if required
@@ -894,7 +912,9 @@ class ArFrame(DataFeatures.SharedObject, DataFeatures.PipelineStepObject):
 
 				for layer_name, draw_layer in draw_layers.items():
 
-					try :
+					try:
+
+						logging.debug('\t> drawing %s layer', layer_name)
 
 						self._layers[layer_name].draw(image, **draw_layer)
 
@@ -905,8 +925,12 @@ class ArFrame(DataFeatures.SharedObject, DataFeatures.PipelineStepObject):
 			# Draw current gaze position if required
 			if draw_gaze_positions is not None:
 
+				logging.debug('\t> drawing current gaze position')
+
 				self.__calibrated_gaze_position.draw(image, **draw_gaze_positions)
 
+		logging.debug('\t> returning image (%i x %i)', *image.shape[:2])
+
 		return DataFeatures.TimestampedImage(image, timestamp = self.__background.timestamp)
 
 	def image(self, **kwargs: dict) -> numpy.array:
@@ -1417,6 +1441,7 @@ class ArContext(DataFeatures.PipelineStepObject):
 
 		# Init private attributes
 		self.__pipeline = None
+		self.__debug = False
 		self.__exceptions = DataFeatures.TimestampedExceptions()
 
 		# Init gaze position processing assement
@@ -1440,6 +1465,16 @@ class ArContext(DataFeatures.PipelineStepObject):
 		self.__pipeline = pipeline
 
 	@property
+	def debug(self) -> bool:
+		"""Disable pipeline exception catching to make it crash instead."""
+		return self.__debug
+
+	@debug.setter
+	def debug(self, debug: bool):
+
+		self.__debug = debug
+
+	@property
 	def image_parameters(self) -> dict:
 		"""Default image method parameters dictionary."""
 		return self._image_parameters
@@ -1496,12 +1531,12 @@ class ArContext(DataFeatures.PipelineStepObject):
 				if x is None and y is None:
 
 					# Edit empty gaze position
-					self.__pipeline.look( GazeFeatures.GazePosition( timestamp = timestamp) )
+					self.__pipeline.look( GazeFeatures.GazePosition( timestamp = timestamp), debug = self.__debug )
 
 				else:
 
 					# Edit gaze position
-					self.__pipeline.look( GazeFeatures.GazePosition( (x, y), precision = precision, timestamp = timestamp))
+					self.__pipeline.look( GazeFeatures.GazePosition( (x, y), precision = precision, timestamp = timestamp), debug = self.__debug)
 
 			except DataFeatures.TimestampedException as e:
 
@@ -1530,7 +1565,7 @@ class ArContext(DataFeatures.PipelineStepObject):
 
 				logging.debug('\t> watch image (%i x %i)', width, height)
 
-				self.__pipeline.watch( DataFeatures.TimestampedImage(image, timestamp = timestamp))
+				self.__pipeline.watch( DataFeatures.TimestampedImage(image, timestamp = timestamp), debug = self.__debug )
 
 			except DataFeatures.TimestampedException as e: