path: root/voctocore/lib/avsource.py

#!/usr/bin/python3
import logging
from gi.repository import Gst

from lib.config import Config
from lib.tcpsingleconnection import TCPSingleConnection

class AVSource(TCPSingleConnection):
    log = logging.getLogger('AVSource')

    name = None
    caps = None

    receiverPipeline = None

    def __init__(self, name, port):
        self.log = logging.getLogger('AVSource['+name+']')
        super().__init__(port)

        self.name = name

    def on_accepted(self, conn, addr):
        pipeline = """
            fdsrc fd={fd} !
            matroskademux name=demux

            demux. ! 
            {acaps} !
            queue !
            tee name=atee

            atee. ! queue ! interaudiosink channel=audio_{name}_mixer
            atee. ! queue ! interaudiosink channel=audio_{name}_mirror
        """.format(
            fd=conn.fileno(),
            name=self.name,
            acaps=Config.get('mix', 'audiocaps')
        )

        if Config.getboolean('previews', 'enabled'):
            pipeline += """
                atee. ! queue ! interaudiosink channel=audio_{name}_preview
            """.format(
                name=self.name
            )

        pipeline += """
            demux. ! 
            {vcaps} !
            textoverlay halignment=left valignment=top ypad=25 text=AVSource !
            timeoverlay halignment=left valignment=top ypad=25 xpad=400 !
            queue !
            tee name=vtee

            vtee. ! queue ! intervideosink channel=video_{name}_mixer
            vtee. ! queue ! intervideosink channel=video_{name}_mirror
        """.format(
            fd=conn.fileno(),
            name=self.name,
            vcaps=Config.get('mix', 'videocaps')
        )

        if Config.getboolean('previews', 'enabled'):
            pipeline += """
                vtee. ! queue ! intervideosink channel=video_{name}_preview
            """.format(
                name=self.name
            )

        self.log.debug('Launching Source-Pipeline:\n%s', pipeline)
        self.receiverPipeline = Gst.parse_launch(pipeline)

        self.log.debug('Binding End-of-Stream-Signal on Source-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)


    def on_eos(self, bus, message):
        self.log.debug('Received End-of-Stream-Signal on Source-Pipeline')
        if self.currentConnection is not None:
            self.disconnect()

    def on_error(self, bus, message):
        self.log.debug('Received Error-Signal on Source-Pipeline')
        (error, debug) = message.parse_error()
        self.log.debug('Error-Details: #%u: %s', error.code, debug)

        if self.currentConnection is not None:
            self.disconnect()

    def disconnect(self):
        self.receiverPipeline.set_state(Gst.State.NULL)
        self.receiverPipeline = None
        self.close_connection()