commit f047f165da45e92ca45860689e7d8c15e714b4e6
author Dmitry Kovalev <dkovalev@google.com> Wed Feb 20 17:35:00 2019 -0800
committer Dmitry Kovalev <dkovalev@google.com> Thu Feb 21 19:06:40 2019 +0000
tree 1f28523daa5de101c4f167bc45a29368d918110f
parent 06ca38494011408fb7e03265c808021e15e07a5f

Add parameter to disable SIGINT handling from run_pipeline().

Change-Id: I47ad946f19aad0d1bdf06695d2ffce1027865a2c

edgetpuvision/camera.py

diff --git a/edgetpuvision/camera.py b/edgetpuvision/camera.py
index a5de10f..590c4df 100644
--- a/edgetpuvision/camera.py
+++ b/edgetpuvision/camera.py
@@ -37,7 +37,7 @@
 
         self._thread = threading.Thread(target=gstreamer.run_pipeline,
                                         args=(pipeline, self._layout, render_overlay,
-                                              gstreamer.Display.NONE, signals))
+                                              gstreamer.Display.NONE, False, signals))
         self._thread.start()
 
     def stop_recording(self):

edgetpuvision/gstreamer.py

diff --git a/edgetpuvision/gstreamer.py b/edgetpuvision/gstreamer.py
index 7b18837..c424bbe 100644
--- a/edgetpuvision/gstreamer.py
+++ b/edgetpuvision/gstreamer.py
@@ -255,7 +255,7 @@
 def quit():
     Gtk.main_quit()
 
-def run_pipeline(pipeline, layout, render_overlay, display, signals=None):
+def run_pipeline(pipeline, layout, render_overlay, display, handle_sigint=True, signals=None):
     signals = signals or {}
 
     # Create pipeline
@@ -317,8 +317,11 @@
         bus.add_signal_watch()
         bus.connect('message', on_bus_message)
 
+        # Handle signals.
+        if handle_sigint:
+            GLib.unix_signal_add(GLib.PRIORITY_DEFAULT, signal.SIGINT, Gtk.main_quit)
+
         # Run pipeline.
-        GLib.unix_signal_add(GLib.PRIORITY_DEFAULT, signal.SIGINT, Gtk.main_quit)
         pipeline.set_state(Gst.State.PLAYING)
         try:
             Gtk.main()