aboutsummaryrefslogtreecommitdiff
path: root/example-scripts
diff options
context:
space:
mode:
Diffstat (limited to 'example-scripts')
-rwxr-xr-xexample-scripts/gstreamer/ingest.py310
1 files changed, 310 insertions, 0 deletions
diff --git a/example-scripts/gstreamer/ingest.py b/example-scripts/gstreamer/ingest.py
new file mode 100755
index 0000000..14ce043
--- /dev/null
+++ b/example-scripts/gstreamer/ingest.py
@@ -0,0 +1,310 @@
+#!/usr/bin/env python3
+
+# ingest.py
+"""
+Source client for Voctomix.
+Features:
+ Retrieves audio and video-caps config from core.
+ Uses core's clock.
+ Mix and match audio and video sources muxed into one streem.
+ Can display video locally, including frame count and fps.
+ Defaults to test audio and video sent to local core.
+
+"""
+
+import argparse
+import gi
+import os
+import signal
+import socket
+import sys
+
+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([])
+
+# this is to use the same code tha gui uses to get config from core
+sys.path.insert(0, '../..' )
+sys.path.insert(0, '.' )
+
+import voctogui.lib.connection as Connection
+
+
+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 {video_device} do-timestamp=true name=videosrc !
+ tsdemux name=demux!
+ queue !
+ decodebin !
+ {monitor}
+ deinterlace mode=1 !
+ videorate !
+ videoscale !
+ videoconvert !
+ """
+
+ elif args.video_source == 'hdmi2usb':
+ # https://hdmi2usb.tv
+ # Note: this code works with 720p
+ video_src = """
+ v4l2src {video_device} name=videosrc !
+ queue !
+ image/jpeg,width=1280,height=720 !
+ jpegdec !
+ {monitor}
+ videoconvert !
+ videorate !
+ """
+
+ elif args.video_source == 'ximage':
+ video_src = """
+ ximagesrc name=videosrc
+ use-damage=false !
+ {monitor}
+ videoconvert !
+ videorate !
+ videoscale !
+ """
+ # startx=0 starty=0 endx=1919 endy=1079 !
+
+ 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
+
+ # remove blank lines to make it more human readable
+ pipeline = pipeline.replace("\n\n","\n")
+
+ 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
+
+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')
+
+ print('Shutting down...')
+ senderPipeline.set_state(Gst.State.NULL)
+ print('Done.')
+
+ return
+
+def get_args():
+
+ parser = argparse.ArgumentParser(
+ description='''Vocto-ingest Client with Net-time support.
+ Gst caps are retrieved from the server.
+ Run without parameters: send test av to localhost:10000
+ ''')
+
+ 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',
+ 'ximage',
+ '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()