From 886fe82d67798522f4d5d4677fce17c25d822ac4 Mon Sep 17 00:00:00 2001
From: Théo de la Hogue
Date: Tue, 25 Jun 2024 17:44:52 +0200
Subject: Adding a wait parameter to PipelineStepImage wrapper in order to not
 raise an exception when a sharedObject is busy. Using this option into
 __main__ script.

---
 src/argaze/DataFeatures.py | 23 ++++++++++++++++++-----
 src/argaze/__main__.py     |  8 ++++----
 2 files changed, 22 insertions(+), 9 deletions(-)

diff --git a/src/argaze/DataFeatures.py b/src/argaze/DataFeatures.py
index fa9b52e..32dd0e1 100644
--- a/src/argaze/DataFeatures.py
+++ b/src/argaze/DataFeatures.py
@@ -940,16 +940,29 @@ def PipelineStepImage(method):
 	"""Define a decorator use into PipelineStepObject class to wrap pipeline step image method."""
 
 	@wraps(method)
-	def wrapper(self, **kwargs) -> numpy.array:
-		"""Wrap pipeline step image method."""
+	def wrapper(self, wait: bool = True, **kwargs) -> numpy.array:
+		"""Wrap pipeline step image method.
+
+		Parameters:
+			wait: for SharedObject, wait until the object is available else throw a SharedObjectBusy exception.
+		"""
 
 		# Check shared object instance
 		if issubclass(type(self), SharedObject):
 
-			# Busy shared object can't return image
-			if self.busy():
+			# Wait until the shared object is available if required
+			if wait:
+
+				while self.busy():
+
+					time.sleep(1e-6)
+
+			# Otherwise, busy shared object can't return image
+			else:
+
+				if self.busy():
 
-				raise SharedObjectBusy(f'Can\'t return image because {self.name} is busy.')
+					raise SharedObjectBusy(f'Can\'t return image because {self.name} is busy.')
 
 		if kwargs:
 
diff --git a/src/argaze/__main__.py b/src/argaze/__main__.py
index 2647198..cb8bd01 100644
--- a/src/argaze/__main__.py
+++ b/src/argaze/__main__.py
@@ -154,8 +154,8 @@ def load_context(args):
 
 					try:
 
-						# Display context
-						display(context.name, context.image(draw_pipeline=draw_pipeline), 0.75, draw_help=draw_help)
+						# Display context if the pipeline is available
+						display(context.name, context.image(wait = False, draw_pipeline=draw_pipeline), 0.75, draw_help=draw_help)
 
 					except SharedObjectBusy:
 
@@ -168,8 +168,8 @@ def load_context(args):
 
 							try:
 
-								# Display scene's frame
-								display(scene_frame.name, scene_frame.image(), 0.5)
+								# Display scene's frame if available
+								display(scene_frame.name, scene_frame.image(wait = False), 0.5)
 
 							except SharedObjectBusy:
 
-- 
cgit v1.1