35 lines
1.0 KiB
Python
35 lines
1.0 KiB
Python
import ffmpeg
|
|
from procmanager import ProcManager
|
|
|
|
class Streamer:
|
|
|
|
def __init__(self, screendef: str, screensize: str, videocodec: str, audioindex: int, protocol: str, target: str) -> None:
|
|
self.screendef = screendef
|
|
self.screensize = screensize
|
|
self.videocodec = videocodec
|
|
self.audioindex = audioindex
|
|
self.protocol = protocol # should be 'tcp' or 'udp'
|
|
self.target = target
|
|
self.process_manager = ProcManager()
|
|
|
|
def __del__(self) -> None:
|
|
self.stop()
|
|
|
|
def start(self):
|
|
streams = []
|
|
self.input = ffmpeg.input(self.screendef, f='x11grab', r=60, s=self.screensize, hide_banner='-y', loglevel='error')
|
|
streams.append(self.input)
|
|
|
|
if self.audioindex != -1:
|
|
self.audio = ffmpeg.input(str(self.audioindex), f='pulse')
|
|
streams.append(self.audio)
|
|
|
|
self.output = ffmpeg.output(*streams, self.protocol + '://' + self.target, f='mpegts', vcodec=self.videocodec, tune='zerolatency')
|
|
proc = ffmpeg.run_async(self.output, pipe_stderr=True)
|
|
self.process_manager.start(proc)
|
|
|
|
def stop(self):
|
|
self.process_manager.stop()
|
|
|
|
|