multi device client with netclock and monitor.

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()