- #!/usr/bin/python3
- import logging
- from gi.repository import Gst
- # import library components
- from lib.config import Config
- from lib.avsource import AVSource
- from lib.avrawoutput import AVRawOutput
- from lib.avpreviewoutput import AVPreviewOutput
- from lib.videomix import VideoMix
- from lib.audiomix import AudioMix
- from lib.streamblanker import StreamBlanker
- class Pipeline(object):
- """mixing, streaming and encoding pipeline constuction and control"""
- def __init__(self):
- self.log = logging.getLogger('Pipeline')
- self.log.info('Video-Caps configured to: %s', Config.get('mix', 'videocaps'))
- self.log.info('Audio-Caps configured to: %s', Config.get('mix', 'audiocaps'))
- names = Config.getlist('mix', 'sources')
- if len(names) < 1:
- raise RuntimeException("At least one AVSource must be configured!")
- self.sources = []
- self.mirrors = []
- self.previews = []
- self.sbsources = []
- self.log.info('Creating %u Creating AVSources: %s', len(names), names)
- for idx, name in enumerate(names):
- port = 10000 + idx
- self.log.info('Creating AVSource %s at tcp-port %u', name, port)
- outputs = [name+'_mixer', name+'_mirror']
- if Config.getboolean('previews', 'enabled'):
- outputs.append(name+'_preview')
- source = AVSource(name, port, outputs=outputs)
- self.sources.append(source)
- port = 13000 + idx
- self.log.info('Creating Mirror-Output for AVSource %s at tcp-port %u', name, port)
- mirror = AVRawOutput('%s_mirror' % name, port)
- self.mirrors.append(mirror)
- if Config.getboolean('previews', 'enabled'):
- port = 14000 + idx
- self.log.info('Creating Preview-Output for AVSource %s at tcp-port %u', name, port)
- preview = AVPreviewOutput('%s_preview' % name, port)
- self.previews.append(preview)
- self.log.info('Creating Videmixer')
- self.vmix = VideoMix()
- self.log.info('Creating Audiomixer')
- self.amix = AudioMix()
- port = 16000
- self.log.info('Creating Mixer-Background VSource at tcp-port %u', port)
- self.bgsrc = AVSource('background', port, has_audio=False)
- port = 11000
- self.log.info('Creating Mixer-Output at tcp-port %u', port)
- self.mixout = AVRawOutput('mix_out', port)
- if Config.getboolean('previews', 'enabled'):
- port = 12000
- self.log.info('Creating Preview-Output for AVSource %s at tcp-port %u', name, port)
- self.mixpreview = AVPreviewOutput('mix_preview', port)
- if Config.getboolean('stream-blanker', 'enabled'):
- names = Config.getlist('stream-blanker', 'sources')
- if len(names) < 1:
- raise RuntimeException("At least one StreamBlanker-Source must be configured or the StreamBlanker disabled!")
- for idx, name in enumerate(names):
- port = 17000 + idx
- self.log.info('Creating StreamBlanker VSource %s at tcp-port %u', name, port)
- source = AVSource('%s_streamblanker' % name, port, has_audio=False)
- self.sbsources.append(source)
- port = 18000
- self.log.info('Creating StreamBlanker ASource at tcp-port %u', port)
- source = AVSource('streamblanker', port, has_video=False)
- self.sbsources.append(source)
- self.log.info('Creating StreamBlanker')
- self.streamblanker = StreamBlanker()
- port = 15000
- self.log.info('Creating StreamBlanker-Output at tcp-port %u', port)
- self.streamout = AVRawOutput('streamblanker_out', port)
|