Feature/graceful app shutdown

apps/conference-demos/rgb-depth-connections/main.py

@@ -1,9 +1,25 @@
+import logging
+import signal
+
 import depthai as dai
 
 from utils.host_bird_eye_view import BirdsEyeView
 from utils.host_rgb_conference_node import CombineOutputs
 from utils.arguments import initialize_argparser
 
+logger = logging.getLogger(__name__)
+shutdown_requested = False
+
+
+def _handle_shutdown_signal(_signum, _frame):
+    global shutdown_requested
+    logger.info("Application received a stop signal. Stopping the app...")
+    shutdown_requested = True
+
+
+signal.signal(signal.SIGINT, _handle_shutdown_signal)
+signal.signal(signal.SIGTERM, _handle_shutdown_signal)
+
 _, args = initialize_argparser()
 
 visualizer = dai.RemoteConnection(httpPort=8082)
@@ -96,6 +112,9 @@
     visualizer.registerPipeline(pipeline)
 
     while pipeline.isRunning():
+        if shutdown_requested:
+            pipeline.stop()
+            break
         key = visualizer.waitKey(1)
         if key == ord("q"):
             print("Got q key from the remote connection!")

apps/data-collection/backend/src/main.py

@@ -1,4 +1,5 @@
 import logging as log
+import signal
 
 import depthai as dai
 
@@ -12,6 +13,17 @@
 
 log.basicConfig(level=log.INFO)
 logger = log.getLogger(__name__)
+shutdown_requested = False
+
+
+def _handle_shutdown_signal(_signum, _frame):
+    global shutdown_requested
+    logger.info("Application received a stop signal. Stopping the app...")
+    shutdown_requested = True
+
+
+signal.signal(signal.SIGINT, _handle_shutdown_signal)
+signal.signal(signal.SIGTERM, _handle_shutdown_signal)
 
 
 def main():
@@ -101,6 +113,9 @@ def main():
         logger.info("Pipeline running!")
 
         while pipeline.isRunning():
+            if shutdown_requested:
+                pipeline.stop()
+                break
             key = visualizer.waitKey(1)
             pipeline.processTasks()
             if key == ord("q"):

apps/data-collection/oakapp.toml

@@ -1,6 +1,6 @@
 identifier = "com.example.apps.data-collection"
 entrypoint = ["bash", "-c", "/usr/bin/runsvdir -P /etc/service"]
-app_version = "0.9.3"
+app_version = "1.0.0"
 
 prepare_container = [
     { type = "COPY", source = "./backend/requirements.txt", target = "./backend/requirements.txt" },

apps/default-app/main.py

@@ -1,10 +1,26 @@
 #!/usr/bin/env python3
+import logging
+import signal
+from typing import Optional
+
 import cv2
 import depthai as dai
 from depthai_nodes.node import ApplyDepthColormap
-from typing import Optional
 from utils.arguments import initialize_argparser
 
+logger = logging.getLogger(__name__)
+shutdown_requested = False
+
+
+def _handle_shutdown_signal(_signum, _frame):
+    global shutdown_requested
+    logger.info("Application received a stop signal. Stopping the app...")
+    shutdown_requested = True
+
+
+signal.signal(signal.SIGINT, _handle_shutdown_signal)
+signal.signal(signal.SIGTERM, _handle_shutdown_signal)
+
 _, args = initialize_argparser()
 
 visualizer = dai.RemoteConnection(httpPort=8082)
@@ -97,6 +113,9 @@
     visualizer.registerPipeline(pipeline)
 
     while pipeline.isRunning():
+        if shutdown_requested:
+            pipeline.stop()
+            break
         key = visualizer.waitKey(1)
         if key == ord("q"):
             print("Got q key from the remote connection!")

apps/default-app/oakapp.toml

@@ -1,6 +1,6 @@
 identifier = "com.example.apps.default-app"
 entrypoint = ["bash", "-c", "/usr/bin/runsvdir -P /etc/service"]
-app_version = "1.0.0"
+app_version = "1.0.1"
 assign_frontend_port = true
 
 prepare_container = [

apps/dino-tracking/backend/src/main.py

@@ -1,3 +1,5 @@
+import logging
+import signal
 from pathlib import Path
 
 import depthai as dai
@@ -34,6 +36,19 @@
 )
 from object_selection.mask_selection_node import MaskSelection
 
+logger = logging.getLogger(__name__)
+shutdown_requested = False
+
+
+def _handle_shutdown_signal(_signum, _frame):
+    global shutdown_requested
+    logger.info("Application received a stop signal. Stopping the app...")
+    shutdown_requested = True
+
+
+signal.signal(signal.SIGINT, _handle_shutdown_signal)
+signal.signal(signal.SIGTERM, _handle_shutdown_signal)
+
 load_dotenv(override=True)
 
 
@@ -168,6 +183,9 @@ def main():
         visualizer.registerPipeline(pipeline)
 
         while pipeline.isRunning():
+            if shutdown_requested:
+                pipeline.stop()
+                break
             key = visualizer.waitKey(1)
             if key == ord("q"):
                 print("Received q. Exiting...")

apps/dino-tracking/oakapp.toml

@@ -1,6 +1,6 @@
 identifier = "com.example.apps.dino-tracking"
 entrypoint = ["bash", "-c", "/usr/bin/runsvdir -P /etc/service"]
-app_version = "0.9.1"
+app_version = "1.0.0"
 
 prepare_container = [
     { type = "COPY", source = "./backend/requirements.txt", target = "./backend/requirements.txt" },

apps/focused-vision/backend/src/main.py

@@ -1,4 +1,5 @@
 import logging
+import signal
 
 import depthai as dai
 
@@ -18,6 +19,17 @@
 from pathlib import Path
 
 logger = logging.getLogger(__name__)
+shutdown_requested = False
+
+
+def _handle_shutdown_signal(_signum, _frame):
+    global shutdown_requested
+    logger.info("Application received a stop signal. Stopping the app...")
+    shutdown_requested = True
+
+
+signal.signal(signal.SIGINT, _handle_shutdown_signal)
+signal.signal(signal.SIGTERM, _handle_shutdown_signal)
 
 _, args = initialize_argparser()
 
@@ -241,6 +253,9 @@
 
     counter = 0
     while pipeline.isRunning():
+        if shutdown_requested:
+            pipeline.stop()
+            break
         pipeline.processTasks()
         key = visualizer.waitKey(1)
         counter += 1

apps/focused-vision/oakapp.toml

@@ -1,6 +1,6 @@
 identifier = "com.example.apps.focused-vision"
 entrypoint = ["bash", "-c", "/usr/bin/runsvdir -P /etc/service"]
-app_version = "0.9.3"
+app_version = "1.0.0"
 
 prepare_container = [
     { type = "COPY", source = "./backend/requirements.txt", target = "./backend/requirements.txt" },

apps/object-volume-measurement-3d/backend/src/main.py

@@ -1,3 +1,6 @@
+import logging
+import signal
+
 import depthai as dai
 
 from depthai_nodes.node import ParsingNeuralNetwork, ImgDetectionsFilter
@@ -8,6 +11,19 @@
 from utils.annotation_node import AnnotationNode
 from utils.measurement_node import MeasurementNode
 
+logger = logging.getLogger(__name__)
+shutdown_requested = False
+
+
+def _handle_shutdown_signal(_signum, _frame):
+    global shutdown_requested
+    logger.info("Application received a stop signal. Stopping the app...")
+    shutdown_requested = True
+
+
+signal.signal(signal.SIGINT, _handle_shutdown_signal)
+signal.signal(signal.SIGTERM, _handle_shutdown_signal)
+
 _, args = initialize_argparser()
 
 IP = args.ip or "localhost"
@@ -229,6 +245,9 @@ def measurement_method_service(payload: dict):
     print("Press 'q' to stop")
 
     while pipeline.isRunning():
+        if shutdown_requested:
+            pipeline.stop()
+            break
         pipeline.processTasks()
         key = visualizer.waitKey(1)
         if key == ord("q"):

apps/object-volume-measurement-3d/oakapp.toml

@@ -1,6 +1,6 @@
 identifier = "com.example.apps.object-volume-measurement-3d"
 entrypoint = ["bash", "-c", "/usr/bin/runsvdir -P /etc/service"]
-app_version = "0.9.4"
+app_version = "1.0.0"
 
 storage_dir_name = "storage"
 prepare_container = [

apps/p2p-measurement/backend/src/main.py

@@ -1,3 +1,6 @@
+import logging
+import signal
+
 import depthai as dai
 from depthai_nodes.node import ApplyDepthColormap
 import numpy as np
@@ -6,6 +9,18 @@
 from utils.arguments import initialize_argparser
 from utils.point_tracker import PointTracker
 
+logger = logging.getLogger(__name__)
+shutdown_requested = False
+
+
+def _handle_shutdown_signal(_signum, _frame):
+    global shutdown_requested
+    logger.info("Application received a stop signal. Stopping the app...")
+    shutdown_requested = True
+
+
+signal.signal(signal.SIGINT, _handle_shutdown_signal)
+signal.signal(signal.SIGTERM, _handle_shutdown_signal)
 
 _, args = initialize_argparser()
 
@@ -138,6 +153,9 @@ def get_tracking_status_service(data: dict):
     visualizer.registerPipeline(pipeline)
 
     while pipeline.isRunning():
+        if shutdown_requested:
+            pipeline.stop()
+            break
         pipeline.processTasks()
         key = visualizer.waitKey(1)
         if key == ord("q"):

apps/p2p-measurement/oakapp.toml

@@ -1,6 +1,6 @@
 identifier = "com.example.apps.p2p-measurement"
 entrypoint = ["bash", "-c", "/usr/bin/runsvdir -P /etc/service"]
-app_version = "0.9.3"
+app_version = "1.0.0"
 
 storage_dir_name = "storage"
 prepare_container = [

apps/people-demographics-and-sentiment-analysis/backend/src/main.py

@@ -1,3 +1,6 @@
+import logging
+import signal
+
 import depthai as dai
 
 from config.arguments import initialize_argparser
@@ -18,6 +21,19 @@
 from visualization.monitor_node import MonitorFacesNode
 from visualization.annotation_node import AnnotationNode
 
+logger = logging.getLogger(__name__)
+shutdown_requested = False
+
+
+def _handle_shutdown_signal(_signum, _frame):
+    global shutdown_requested
+    logger.info("Application received a stop signal. Stopping the app...")
+    shutdown_requested = True
+
+
+signal.signal(signal.SIGINT, _handle_shutdown_signal)
+signal.signal(signal.SIGTERM, _handle_shutdown_signal)
+
 
 def main():
     _, args = initialize_argparser()
@@ -109,6 +125,9 @@ def main():
         visualizer.registerPipeline(pipeline)
 
         while pipeline.isRunning():
+            if shutdown_requested:
+                pipeline.stop()
+                break
             key = visualizer.waitKey(1)
             pipeline.processTasks()
             if key == ord("q"):

apps/people-demographics-and-sentiment-analysis/oakapp.toml

@@ -1,6 +1,6 @@
 identifier = "com.example.apps.people-demographic-sentiment-analysis"
 entrypoint = ["bash", "-c", "/usr/bin/runsvdir -P /etc/service"]
-app_version = "0.9.2"
+app_version = "1.0.0"
 
 prepare_container = [
     { type = "COPY", source = "./backend/requirements.txt", target = "./backend/requirements.txt" },

apps/qr-tiling/backend/src/main.py

@@ -1,5 +1,6 @@
 from pathlib import Path
 import logging
+import signal
 
 import depthai as dai
 from depthai_nodes.node import (
@@ -29,8 +30,19 @@
 
 logging.basicConfig(level=logging.INFO)
 logger = logging.getLogger(__name__)
+shutdown_requested = False
 MODEL_DIR = Path(__file__).resolve().parent / "depthai_models"
 
+
+def _handle_shutdown_signal(_signum, _frame):
+    global shutdown_requested
+    logger.info("Application received a stop signal. Stopping the app...")
+    shutdown_requested = True
+
+
+signal.signal(signal.SIGINT, _handle_shutdown_signal)
+signal.signal(signal.SIGTERM, _handle_shutdown_signal)
+
 visualizer = dai.RemoteConnection(httpPort=8082)
 device = dai.Device()
 
@@ -171,6 +183,9 @@
     logger.info("Pipeline running!")
 
     while pipeline.isRunning():
+        if shutdown_requested:
+            pipeline.stop()
+            break
         pipeline.processTasks()
         key = visualizer.waitKey(1)
         if key == ord("q"):

apps/qr-tiling/oakapp.toml

@@ -1,6 +1,6 @@
 identifier = "com.example.apps.qr-tiling"
 entrypoint = ["bash", "-c", "/usr/bin/runsvdir -P /etc/service"]
-app_version = "1.0.0"
+app_version = "1.0.1"
 
 prepare_container = [
     { type = "RUN", command = "apt-get update" },