mirror of
https://github.com/blakeblackshear/frigate.git
synced 2024-12-23 19:11:14 +01:00
01b9d4d848
* Use memo for recordings timezone * Add audio encoding field and simplify stream creation * Update docs and tests * Fix bad logic
88 lines
3.0 KiB
Python
88 lines
3.0 KiB
Python
"""Controls go2rtc restream."""
|
|
|
|
|
|
import logging
|
|
import requests
|
|
|
|
from typing import Optional
|
|
|
|
from frigate.config import FrigateConfig, RestreamAudioCodecEnum, RestreamVideoCodecEnum
|
|
from frigate.const import BIRDSEYE_PIPE
|
|
from frigate.ffmpeg_presets import (
|
|
parse_preset_hardware_acceleration_encode,
|
|
parse_preset_hardware_acceleration_go2rtc_engine,
|
|
)
|
|
from frigate.util import escape_special_characters
|
|
|
|
logger = logging.getLogger(__name__)
|
|
|
|
|
|
def get_manual_go2rtc_stream(
|
|
camera_url: str,
|
|
aCodecs: list[RestreamAudioCodecEnum],
|
|
vCodec: RestreamVideoCodecEnum,
|
|
engine: Optional[str],
|
|
) -> str:
|
|
"""Get a manual stream for go2rtc."""
|
|
stream = f"ffmpeg:{camera_url}"
|
|
|
|
for aCodec in aCodecs:
|
|
stream += f"#audio={aCodec}"
|
|
|
|
if vCodec == RestreamVideoCodecEnum.copy:
|
|
stream += "#video=copy"
|
|
else:
|
|
stream += f"#video={vCodec}"
|
|
|
|
if engine:
|
|
stream += f"#hardware={engine}"
|
|
|
|
return stream
|
|
|
|
|
|
class RestreamApi:
|
|
"""Control go2rtc relay API."""
|
|
|
|
def __init__(self, config: FrigateConfig) -> None:
|
|
self.config: FrigateConfig = config
|
|
|
|
def add_cameras(self) -> None:
|
|
"""Add cameras to go2rtc."""
|
|
self.relays: dict[str, str] = {}
|
|
|
|
for cam_name, camera in self.config.cameras.items():
|
|
if not camera.restream.enabled:
|
|
continue
|
|
|
|
for input in camera.ffmpeg.inputs:
|
|
if "restream" in input.roles:
|
|
if (
|
|
input.path.startswith("rtsp")
|
|
and camera.restream.video_encoding
|
|
== RestreamVideoCodecEnum.copy
|
|
and camera.restream.audio_encoding
|
|
== [RestreamAudioCodecEnum.copy]
|
|
):
|
|
self.relays[
|
|
cam_name
|
|
] = f"{escape_special_characters(input.path)}#backchannel=0"
|
|
else:
|
|
# go2rtc only supports rtsp for direct relay, otherwise ffmpeg is used
|
|
self.relays[cam_name] = get_manual_go2rtc_stream(
|
|
escape_special_characters(input.path),
|
|
camera.restream.audio_encoding,
|
|
camera.restream.video_encoding,
|
|
parse_preset_hardware_acceleration_go2rtc_engine(
|
|
self.config.ffmpeg.hwaccel_args
|
|
),
|
|
)
|
|
|
|
if self.config.restream.birdseye:
|
|
self.relays[
|
|
"birdseye"
|
|
] = f"exec:{parse_preset_hardware_acceleration_encode(self.config.ffmpeg.hwaccel_args, f'-f rawvideo -pix_fmt yuv420p -video_size {self.config.birdseye.width}x{self.config.birdseye.height} -r 10 -i {BIRDSEYE_PIPE}', '-rtsp_transport tcp -f rtsp {output}')}"
|
|
|
|
for name, path in self.relays.items():
|
|
params = {"src": path, "name": name}
|
|
requests.put("http://127.0.0.1:1984/api/streams", params=params)
|