From 8786f6611eb9410fd32dffa0aff40d1926793dec Mon Sep 17 00:00:00 2001
From: gobbo_a <alexandre.gobbo@psi.ch>
Date: Fri, 26 Aug 2022 13:57:20 +0200
Subject: [PATCH] Startup

---
 script/local.py               | 23 +++++++++++++++++++++++
 script/test/test_pipeline2.py |  3 +--
 2 files changed, 24 insertions(+), 2 deletions(-)

diff --git a/script/local.py b/script/local.py
index 0d5bba6..997d46a 100644
--- a/script/local.py
+++ b/script/local.py
@@ -38,6 +38,10 @@ from cam_server.start_camera_server import start_camera_server
 
 
 
+###################################################################################################
+#  Pipeline Utilities
+###################################################################################################
+
 camera_client = CamClient("http://" + App.getArgumentValue("cam_srv_url"))
 
 class Namespace(object):
@@ -108,6 +112,10 @@ def dont_exit(status):
 sys.exit = dont_exit
 
 
+###################################################################################################
+#  Current pipeline handling
+###################################################################################################
+
 PIPELINE_DEFAULT_PORT = 5005
 pipeline_stop_event = None
 pipeline_queue = None
@@ -162,6 +170,21 @@ def get_pipeline_msg():
             raise Exception("Received None message.")
         return data.data    
 
+
+###################################################################################################
+#  Image rendering
+###################################################################################################
+
+def start_viewer(host="localhost", port = PIPELINE_DEFAULT_PORT):
+    viewer.show_stream("tcp://" + str(host) + ":" + str(port))
+
+
+def stop_viewer():
+    viewer.stop()
+    
+###################################################################################################
+#  Callbacks
+###################################################################################################
 def on_system_restart():   
     print ("restart")
     stop_pipeline()     
diff --git a/script/test/test_pipeline2.py b/script/test/test_pipeline2.py
index f48f5de..10654f8 100644
--- a/script/test/test_pipeline2.py
+++ b/script/test/test_pipeline2.py
@@ -1,5 +1,4 @@
-
-viewer.show_stream("tcp://localhost:" + str(PIPELINE_DEFAULT_PORT))
+start_viewer()
 pipeline_config = PipelineConfig("test_pipeline", parameters={"camera_name": "simulation", "function":"transparent"})
 start_pipeline("test_pipeline", {"camera_name": "simulation", "function":"transparent"})