Move files & implement output counting

3 files changed, 204 insertions, 0 deletions

diff --git a/voctocore/lib/video/mix.py b/voctocore/lib/video/mix.py
new file mode 100644
index 0000000..2d73d7b
--- /dev/null
+++ b/voctocore/lib/video/mix.py
@@ -0,0 +1,15 @@
+#!/usr/bin/python3
+import logging
+from gi.repository import Gst
+
+from lib.config import Config
+
+class VideoMix(object):
+	log = logging.getLogger('VideoMix')
+
+	def __init__(self):
+		caps = Config.get('mix', 'videocaps')
+
+		names = Config.getlist('sources', 'video')
+		self.log.info('Configuring Mixer for %u Sources', len(names))
+		#for idx, name in enumerate():
diff --git a/voctocore/lib/video/rawoutput.py b/voctocore/lib/video/rawoutput.py
new file mode 100644
index 0000000..ecdb1ba
--- /dev/null
+++ b/voctocore/lib/video/rawoutput.py
@@ -0,0 +1,84 @@
+#!/usr/bin/python3
+import logging, socket
+from gi.repository import GObject, Gst
+
+from lib.config import Config
+
+class VideoRawOutput(object):
+	log = logging.getLogger('VideoRawOutput')
+
+	name = None
+	port = None
+	caps = None
+
+	boundSocket = None
+
+	receiverPipelines = []
+	currentConnections = []
+
+	def __init__(self, channel, port, caps):
+		self.log = logging.getLogger('VideoRawOutput['+channel+']')
+
+		self.channel = channel
+		self.port = port
+		self.caps = caps
+
+		self.log.debug('Binding to Mirror-Socket on [::]:%u', port)
+		self.boundSocket = socket.socket(socket.AF_INET6)
+		self.boundSocket.setsockopt(socket.SOL_SOCKET, socket.SO_REUSEADDR, 1)
+		self.boundSocket.setsockopt(socket.IPPROTO_IPV6, socket.IPV6_V6ONLY, False)
+		self.boundSocket.bind(('::', port))
+		self.boundSocket.listen(1)
+
+		self.log.debug('Setting GObject io-watch on Socket')
+		GObject.io_add_watch(self.boundSocket, GObject.IO_IN, self.on_connect)
+
+	def on_connect(self, sock, *args):
+		conn, addr = sock.accept()
+		self.log.info("Incomming Connection from %s", addr)
+
+		pipeline = """
+			intervideosrc channel={channel} !
+			{caps} !
+			textoverlay text={channel} halignment=left valignment=top ypad=125 !
+			gdppay !
+			fdsink fd={fd}
+		""".format(
+			fd=conn.fileno(),
+			channel=self.channel,
+			caps=self.caps
+		)
+		self.log.debug('Launching Pipeline:\n%s', pipeline)
+		receiverPipeline = Gst.parse_launch(pipeline)
+
+		def on_eos(bus, message):
+			self.log.info('Received End-of-Stream-Signal on Source-Receiver-Pipeline')
+			self.disconnect(receiverPipeline, conn)
+
+		def on_error(bus, message):
+			self.log.info('Received Error-Signal on Source-Receiver-Pipeline')
+
+			(error, debug) = message.parse_error()
+			self.log.debug('Error-Message %s\n%s', error.message, debug)
+
+			self.disconnect(receiverPipeline, conn)
+
+		self.log.debug('Binding End-of-Stream-Signal on Pipeline')
+		receiverPipeline.bus.add_signal_watch()
+		receiverPipeline.bus.connect("message::eos", on_eos)
+		receiverPipeline.bus.connect("message::error", on_error)
+
+		receiverPipeline.set_state(Gst.State.PLAYING)
+
+		self.receiverPipelines.append(receiverPipeline)
+		self.currentConnections.append(conn)
+
+		self.log.info('Now %u Receiver connected', len(self.currentConnections))
+
+		return True
+
+	def disconnect(self, receiverPipeline, currentConnection):
+		receiverPipeline.set_state(Gst.State.NULL)
+		self.receiverPipelines.remove(receiverPipeline)
+		self.currentConnections.remove(currentConnection)
+		self.log.info('Disconnected Receiver, now %u Receiver connected', len(self.currentConnections))
diff --git a/voctocore/lib/video/src.py b/voctocore/lib/video/src.py
new file mode 100644
index 0000000..dbbe12d
--- /dev/null
+++ b/voctocore/lib/video/src.py
@@ -0,0 +1,105 @@
+#!/usr/bin/python3
+import logging, socket
+from gi.repository import GObject, Gst
+
+from lib.config import Config
+
+class VideoSrc(object):
+	log = logging.getLogger('VideoSrc')
+
+	name = None
+	port = None
+	caps = None
+
+	distributionPipeline = None
+	receiverPipeline = None
+
+	boundSocket = None
+	currentConnection = None
+
+	def __init__(self, name, port, caps):
+		self.log = logging.getLogger('VideoSrc['+name+']')
+
+		self.name = name
+		self.port = port
+		self.caps = caps
+
+		pipeline = """
+			intervideosrc channel=video_{name}_in !
+			{caps} !
+			timeoverlay halignment=left valignment=top !
+			textoverlay text=video_{name}_in halignment=left valignment=top ypad=75 !
+			queue !
+			tee name=tee
+
+			tee. ! queue ! intervideosink channel=video_{name}_mirror
+			tee. ! queue ! intervideosink channel=video_{name}_preview
+			tee. ! queue ! intervideosink channel=video_{name}_mixer
+		""".format(
+			name=self.name,
+			caps=self.caps
+		)
+
+		self.log.debug('Launching Source-Distribution-Pipeline:\n%s', pipeline)
+		self.distributionPipeline = Gst.parse_launch(pipeline)
+		self.distributionPipeline.set_state(Gst.State.PLAYING)
+
+		self.log.debug('Binding to Source-Socket on [::]:%u', port)
+		self.boundSocket = socket.socket(socket.AF_INET6)
+		self.boundSocket.setsockopt(socket.SOL_SOCKET, socket.SO_REUSEADDR, 1)
+		self.boundSocket.setsockopt(socket.IPPROTO_IPV6, socket.IPV6_V6ONLY, False)
+		self.boundSocket.bind(('::', port))
+		self.boundSocket.listen(1)
+
+		self.log.debug('Setting GObject io-watch on Socket')
+		GObject.io_add_watch(self.boundSocket, GObject.IO_IN, self.on_connect)
+
+	def on_connect(self, sock, *args):
+		conn, addr = sock.accept()
+		self.log.info("Incomming Connection from %s", addr)
+
+		if self.currentConnection is not None:
+			self.log.warn("Another Source is already connected")
+			return True
+
+		pipeline = """
+			fdsrc fd={fd} !
+			gdpdepay !
+			{caps} !
+			textoverlay text=video_{name}_fd halignment=left valignment=top ypad=175 !
+			intervideosink channel=video_{name}_in
+		""".format(
+			fd=conn.fileno(),
+			name=self.name,
+			caps=self.caps
+		)
+		self.log.debug('Launching Source-Receiver-Pipeline:\n%s', pipeline)
+		self.receiverPipeline = Gst.parse_launch(pipeline)
+
+		self.log.debug('Binding End-of-Stream-Signal on Source-Receiver-Pipeline')
+		self.receiverPipeline.bus.add_signal_watch()
+		self.receiverPipeline.bus.connect("message::eos", self.on_eos)
+		self.receiverPipeline.bus.connect("message::error", self.on_error)
+
+		self.receiverPipeline.set_state(Gst.State.PLAYING)
+
+		self.currentConnection = conn
+		return True
+
+	def on_eos(self, bus, message):
+		self.log.info('Received End-of-Stream-Signal on Source-Receiver-Pipeline')
+		if self.currentConnection is not None:
+			self.disconnect()
+
+	def on_error(self, bus, message):
+		self.log.info('Received Error-Signal on Source-Receiver-Pipeline')
+		(code, debug) = message.parse_error()
+		self.log.debug('Error-Details: #%u: %s', code, debug)
+
+		if self.currentConnection is not None:
+			self.disconnect()
+
+	def disconnect(self):
+		self.receiverPipeline.set_state(Gst.State.NULL)
+		self.receiverPipeline = None
+		self.currentConnection = None