evlli/hass-core
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

Add get_image method to Stream (#61918)

Browse source 
* Add get_image method to Stream

* Add KeyFrameConverter class
This commit is contained in:
uvjustin 2021-12-23 00:24:53 +08:00 • committed by GitHub
parent eda9291ca1
commit 6e13605cad
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
8 changed files with 153 additions and 6 deletions
Download patch file Download diff file Expand all files Collapse all files

15
homeassistant/components/stream/__init__.py
View file

@ -50,7 +50,7 @@ from .const import (
STREAM_RESTART_RESET_TIME,
TARGET_SEGMENT_DURATION_NON_LL_HLS,
)
from .core import PROVIDERS, IdleTimer, StreamOutput, StreamSettings
from .core import PROVIDERS, IdleTimer, KeyFrameConverter, StreamOutput, StreamSettings
from .hls import HlsStreamOutput, async_setup_hls
_LOGGER = logging.getLogger(__name__)
@ -137,6 +137,8 @@ def filter_libav_logging() -> None:
# Set log level to error for libav.mp4
logging.getLogger("libav.mp4").setLevel(logging.ERROR)
# Suppress "deprecated pixel format" WARNING
logging.getLogger("libav.swscaler").setLevel(logging.ERROR)
async def async_setup(hass: HomeAssistant, config: ConfigType) -> bool:
@ -214,6 +216,7 @@ class Stream:
self._thread_quit = threading.Event()
self._outputs: dict[str, StreamOutput] = {}
self._fast_restart_once = False
self._keyframe_converter = KeyFrameConverter(hass)
self._available: bool = True
self._update_callback: Callable[[], None] | None = None
self._logger = (
@ -327,6 +330,7 @@ class Stream:
self.source,
self.options,
stream_state,
self._keyframe_converter,
self._thread_quit,
)
except StreamWorkerError as err:
@ -419,3 +423,12 @@ class Stream:
# Wait for latest segment, then add the lookback
await hls.recv()
recorder.prepend(list(hls.get_segments())[-num_segments:])
async def get_image(
self,
width: int | None = None,
height: int | None = None,
) -> bytes | None:
"""Wrap get_image from KeyFrameConverter."""
return await self._keyframe_converter.get_image(width=width, height=height)