summaryrefslogtreecommitdiff
path: root/clients
diff options
context:
space:
mode:
Diffstat (limited to 'clients')
-rwxr-xr-xclients/source/ingest.py293
1 files changed, 293 insertions, 0 deletions
diff --git a/clients/source/ingest.py b/clients/source/ingest.py
new file mode 100755
index 0000000..2ac2b7f
--- /dev/null
+++ b/clients/source/ingest.py
@@ -0,0 +1,293 @@
+#!/usr/bin/env python3
+"""
+Ryan Verner <ryan.verner@gmail.com>
+Voctomix ingest streams used for LCA2016, hacked up on the fly.
+
+PIPELINES:
+ * dv
+ * dvpulse
+ * hdvpulse
+ * hdmi2usb
+ * blackmagichdmi
+ * test
+
+Example intended uses (NOTE expected environment variables):
+ * lca-videomix-ingest.py dvpulse 0
+ * lca-videomix-ingest.py hdmi2usb 1
+"""
+
+import sys
+
+import gi
+import signal
+import os
+import socket
+
+import argparse
+
+gi.require_version('Gst', '1.0')
+from gi.repository import Gst, GstNet, GObject
+
+# init GObject & Co. before importing local classes
+GObject.threads_init()
+Gst.init([])
+
+sys.path.insert(0, '../..' )
+
+import voctogui.lib.connection as Connection
+# import lib.clock as ClockManager
+
+
+def mk_video_src(args, videocaps):
+ # make video soure part of pipeline
+
+ video_device = "device={}".format(args.video_dev) \
+ if args.video_dev else ""
+
+ monitor = """tee name=t ! queue !
+ videoconvert ! fpsdisplaysink sync=false
+ t. ! queue !""" \
+ if args.monitor else ""
+
+
+ if args.video_source == 'dv':
+ video_src = """
+ dv1394src name=videosrc {video_device}!
+ dvdemux name=demux !
+ queue !
+ dvdec !
+ {monitor}
+ deinterlace mode=1 !
+ videoconvert !
+ videorate !
+ videoscale !
+ """
+
+ elif args.video_source == 'hdv':
+ video_src = """
+ hdv1394src do-timestamp=true name=videosrc {video_device} !
+ tsdemux name=demux!
+ queue !
+ decodebin !
+ {monitor}
+ deinterlace mode=1 !
+ videorate !
+ videoscale !
+ videoconvert !
+ """
+
+ elif args.video_source == 'hdmi2usb':
+ video_src = """
+ v4l2src device=%s name=videosrc !
+ queue !
+ image/jpeg,width=1280,height=720 !
+ jpegdec !
+ {monitor}
+ videoconvert !
+ videorate !
+ """
+
+ elif args.video_source == 'blackmagichdmi':
+ video_src = """
+ decklinkvideosrc mode=17 connection=2 !
+ {monitor}
+ videoconvert !
+ videorate !
+ videoscale !
+ """
+
+ elif args.video_source == 'test':
+ video_src = """
+ videotestsrc name=videosrc
+ pattern=ball
+ foreground-color=0x00ff0000 background-color=0x00440000 !
+ {monitor}
+ """
+
+ video_src = video_src.format(
+ video_device=video_device,
+ monitor=monitor)
+
+ video_src += videocaps + "!\n"
+
+ return video_src
+
+def mk_audio_src(args, audiocaps):
+
+ audio_device = "device={}".format(args.audio_dev) \
+ if args.audio_dev else ""
+
+ if args.audio_source in [ 'dv', 'hdv' ]:
+ # this only works if video is from DV also.
+ # or some gst source that gets demux ed
+ audio_src = """
+ demux. !
+ audioconvert !
+ """
+
+ elif args.audio_source == 'pulse':
+ audio_src = """
+ pulsesrc {audio_device} name=audiosrc !
+ """.format(audio_device=audio_device)
+
+ elif args.audio_source == 'blackmagichdmi':
+ audio_src = """
+ decklinkaudiosrc !
+ """
+
+ elif args.audio_source == 'test':
+ audio_src = """
+ audiotestsrc name=audiosrc freq=330 !
+ """
+ audio_src += audiocaps + "!\n"
+
+ return audio_src
+
+def mk_mux(args):
+
+ mux = """
+ mux.
+ matroskamux name=mux !
+ """
+ return mux
+
+def mk_client(args):
+ core_ip = socket.gethostbyname(args.host)
+ client = """
+ tcpclientsink host={host} port={port}
+ """.format(host=core_ip, port=args.port)
+
+ return client
+
+
+def mk_pipeline(args, server_caps):
+
+ video_src = mk_video_src(args, server_caps['videocaps'])
+ audio_src = mk_audio_src(args, server_caps['audiocaps'])
+ mux = mk_mux(args)
+ client = mk_client(args)
+
+ pipeline = video_src + "mux. \n" + audio_src + mux + client
+
+ return pipeline
+
+def get_server_caps():
+
+
+ # fetch config from server
+ server_config = Connection.fetchServerConfig()
+ server_caps = {'videocaps': server_config['mix']['videocaps'],
+ 'audiocaps': server_config['mix']['audiocaps']}
+
+ return server_caps
+
+ # obtain network-clock
+ ClockManager.obtainClock(Connection.ip)
+
+
+def run_pipeline(pipeline, args):
+
+ core_ip = socket.gethostbyname(args.host)
+
+ clock = GstNet.NetClientClock.new('voctocore', core_ip, 9998, 0)
+ print('obtained NetClientClock from host', clock)
+
+ print('waiting for NetClientClock to sync…')
+ clock.wait_for_sync(Gst.CLOCK_TIME_NONE)
+
+ print('starting pipeline')
+ senderPipeline = Gst.parse_launch(pipeline)
+ senderPipeline.use_clock(clock)
+ src = senderPipeline.get_by_name('src')
+
+ def on_eos(self, bus, message):
+ print('Received EOS-Signal')
+ sys.exit(1)
+
+ def on_error(self, bus, message):
+ print('Received Error-Signal')
+ (error, debug) = message.parse_error()
+ print('Error-Details: #%u: %s' % (error.code, debug))
+ sys.exit(1)
+
+
+ # Binding End-of-Stream-Signal on Source-Pipeline
+ senderPipeline.bus.add_signal_watch()
+ senderPipeline.bus.connect("message::eos", on_eos)
+ senderPipeline.bus.connect("message::error", on_error)
+
+ print("playing")
+ senderPipeline.set_state(Gst.State.PLAYING)
+
+ mainloop = GObject.MainLoop()
+ try:
+ mainloop.run()
+ except KeyboardInterrupt:
+ print('Terminated via Ctrl-C')
+
+
+
+
+def get_args():
+
+ parser = argparse.ArgumentParser(description='Vocto-ingesb')
+
+ parser.add_argument('-v', '--verbose', action='count', default=0,
+ help="Also print INFO and DEBUG messages.")
+
+ parser.add_argument( '--video-source', action='store',
+ choices=['dv', 'hdv', 'hdmi2usb', 'blackmagichdmi', 'test'],
+ default='test',
+ help="Where to get video from")
+
+ parser.add_argument( '--video-dev', action='store',
+ help="video device")
+
+ parser.add_argument( '--audio-source', action='store',
+ choices=['dv', 'alsa', 'pulse', 'blackmagichdmi', 'test'],
+ default='test',
+ help="Where to get audio from")
+
+ parser.add_argument( '--audio-dev', action='store',
+ default='hw:CARD=CODEC',
+ help="for alsa/pulse, audio device")
+ # maybe hw:1,0
+
+ parser.add_argument( '--audio-delay', action='store',
+ default='10',
+ help="ms to delay audio")
+
+ parser.add_argument('-m', '--monitor', action='store_true',
+ help="fps display sink")
+
+ parser.add_argument( '--host', action='store',
+ default='localhost',
+ help="hostname of vocto core")
+
+ parser.add_argument( '--port', action='store',
+ default='10000',
+ help="port of vocto core")
+
+ args = parser.parse_args()
+
+ return args
+
+
+def main():
+
+ args = get_args()
+
+ core_ip = socket.gethostbyname(args.host)
+ # establish a synchronus connection to server
+ Connection.establish(core_ip)
+
+ server_caps = get_server_caps()
+
+ pipeline = mk_pipeline(args, server_caps)
+ print(pipeline)
+ run_pipeline(pipeline, args)
+
+
+
+if __name__ == '__main__':
+ main()