diff options
Diffstat (limited to 'kvmd/streamer.py')
| -rw-r--r-- | kvmd/streamer.py | 15 |
1 files changed, 13 insertions, 2 deletions
diff --git a/kvmd/streamer.py b/kvmd/streamer.py index f4c49bf2..e9cbf0e6 100644 --- a/kvmd/streamer.py +++ b/kvmd/streamer.py @@ -21,6 +21,7 @@ class Streamer: # pylint: disable=too-many-instance-attributes init_delay: float, init_restart_after: float, quality: int, + soft_fps: int, cmd: List[str], loop: asyncio.AbstractEventLoop, ) -> None: @@ -31,17 +32,23 @@ class Streamer: # pylint: disable=too-many-instance-attributes self.__init_delay = init_delay self.__init_restart_after = init_restart_after self.__quality = quality + self.__soft_fps = soft_fps self.__cmd = cmd self.__loop = loop self.__proc_task: Optional[asyncio.Task] = None - async def start(self, quality: int, no_init_restart: bool=False) -> None: + async def start(self, quality: int, soft_fps: int, no_init_restart: bool=False) -> None: logger = get_logger() logger.info("Starting streamer ...") + assert 1 <= quality <= 100 self.__quality = quality + + assert 1 <= soft_fps <= 30 + self.__soft_fps = soft_fps + await self.__inner_start() if self.__init_restart_after > 0.0 and not no_init_restart: logger.info("Stopping streamer to restart ...") @@ -59,10 +66,14 @@ class Streamer: # pylint: disable=too-many-instance-attributes def get_current_quality(self) -> int: return self.__quality + def get_current_soft_fps(self) -> int: + return self.__soft_fps + def get_state(self) -> Dict: return { "is_running": self.is_running(), "quality": self.__quality, + "soft_fps": self.__soft_fps, } def get_app(self) -> str: @@ -110,7 +121,7 @@ class Streamer: # pylint: disable=too-many-instance-attributes while True: # pylint: disable=too-many-nested-blocks proc: Optional[asyncio.subprocess.Process] = None # pylint: disable=no-member try: - cmd = [part.format(quality=self.__quality) for part in self.__cmd] + cmd = [part.format(quality=self.__quality, soft_fps=self.__soft_fps) for part in self.__cmd] proc = await asyncio.create_subprocess_exec( *cmd, stdout=asyncio.subprocess.PIPE, |