use an external failsrc for the failover shmsrc

3 files changed, 39 insertions, 4 deletions

diff --git a/voctocore/lib/failvideosrc.py b/voctocore/lib/failvideosrc.py
new file mode 100644
index 0000000..500603a
--- /dev/null
+++ b/voctocore/lib/failvideosrc.py
@@ -0,0 +1,34 @@
+#!/usr/bin/python3
+import time, logging
+from gi.repository import GLib, Gst
+
+from lib.config import Config
+
+class FailVideoSrc(Gst.Bin):
+	log = logging.getLogger('FailVideoSrc')
+	colors = [
+		0xffff0000, # red
+		0xff00ff00, # green
+		0xff0000ff, # blue
+		0xffffff00, # yellow
+		0xff00ffff, # cyan
+		0xffff00ff, # magenta
+		0xffffffff, # white
+	]
+
+	def __init__(self, idx, name):
+		super().__init__()
+
+		# Create elements
+		self.failsrc = Gst.ElementFactory.make('videotestsrc', None)
+
+		# Add elements to Bin
+		self.add(self.failsrc)
+
+		# Set properties
+		self.failsrc.set_property('foreground-color', self.colors[idx % len(self.colors)])
+
+		# Add Ghost Pads
+		self.add_pad(
+			Gst.GhostPad.new('src', self.failsrc.get_static_pad('src'))
+		)
diff --git a/voctocore/lib/pipeline.py b/voctocore/lib/pipeline.py
index 35ecf2a..529aeca 100644
--- a/voctocore/lib/pipeline.py
+++ b/voctocore/lib/pipeline.py
@@ -12,6 +12,7 @@ from lib.videomix import VideoMix
 # from lib.audiomix import AudioMix
 from lib.distributor import TimesTwoDistributor
 from lib.shmsrc import FailsafeShmSrc
+from lib.failvideosrc import FailVideoSrc
 
 class Pipeline(Gst.Pipeline):
 	"""mixing, streaming and encoding pipeline constuction and control"""
@@ -45,11 +46,11 @@ class Pipeline(Gst.Pipeline):
 		self.videonames = Config.getlist('sources', 'video')
 		self.audionames = Config.getlist('sources', 'video')
 
-		for name in self.videonames:
+		for idx, name in enumerate(self.videonames):
 			socket = os.path.join(socketpath, 'v-'+name)
 
 			self.log.info('Creating video-source "%s" at socket-path %s', name, socket)
-			sourcebin = FailsafeShmSrc(socket)
+			sourcebin = FailsafeShmSrc(socket, FailVideoSrc(idx, name))
 			self.add(sourcebin)
 
 			distributor = TimesTwoDistributor()
diff --git a/voctocore/lib/shmsrc.py b/voctocore/lib/shmsrc.py
index 64e3555..1158887 100644
--- a/voctocore/lib/shmsrc.py
+++ b/voctocore/lib/shmsrc.py
@@ -10,7 +10,7 @@ class FailsafeShmSrc(Gst.Bin):
 	last_restart_retry = 0
 	is_in_failstate = True
 
-	def __init__(self, socket):
+	def __init__(self, socket, failsrc):
 		super().__init__()
 
 		caps = Gst.Caps.from_string(Config.get('sources', 'videocaps'))
@@ -21,7 +21,7 @@ class FailsafeShmSrc(Gst.Bin):
 		self.capsfilter = Gst.ElementFactory.make('capsfilter', None)
 		self.failsrcsyncer = Gst.ElementFactory.make('identity', None)
 		self.switch = Gst.ElementFactory.make('input-selector', None)
-		self.failsrc = Gst.ElementFactory.make('videotestsrc', None)
+		self.failsrc = failsrc;
 
 		if not self.shmsrc or not self.capsfilter or not self.failsrcsyncer or not self.switch or not self.failsrc:
 			self.log.error('could not create elements')