mirror of
https://github.com/blakeblackshear/frigate.git
synced 2026-05-09 15:05:26 +03:00
Compare commits
5 Commits
2d002d0fc5
...
a50c345d4a
| Author | SHA1 | Date | |
|---|---|---|---|
|
|
a50c345d4a | ||
|
|
ba4a6a53d7 | ||
|
|
e90079ab2f | ||
|
|
b5a360be39 | ||
|
|
54a7c5015e |
2
.github/workflows/stale.yml
vendored
2
.github/workflows/stale.yml
vendored
@ -18,7 +18,7 @@ jobs:
|
|||||||
close-issue-message: ""
|
close-issue-message: ""
|
||||||
days-before-stale: 30
|
days-before-stale: 30
|
||||||
days-before-close: 3
|
days-before-close: 3
|
||||||
exempt-draft-pr: true
|
exempt-draft-pr: false
|
||||||
exempt-issue-labels: "planned,security"
|
exempt-issue-labels: "planned,security"
|
||||||
exempt-pr-labels: "planned,security,dependencies"
|
exempt-pr-labels: "planned,security,dependencies"
|
||||||
operations-per-run: 120
|
operations-per-run: 120
|
||||||
|
|||||||
@ -8,7 +8,6 @@ import os
|
|||||||
import queue
|
import queue
|
||||||
import subprocess as sp
|
import subprocess as sp
|
||||||
import threading
|
import threading
|
||||||
import time
|
|
||||||
import traceback
|
import traceback
|
||||||
from multiprocessing.synchronize import Event as MpEvent
|
from multiprocessing.synchronize import Event as MpEvent
|
||||||
from typing import Any, Optional
|
from typing import Any, Optional
|
||||||
@ -594,112 +593,92 @@ class BirdsEyeFrameManager:
|
|||||||
) -> Optional[list[list[Any]]]:
|
) -> Optional[list[list[Any]]]:
|
||||||
"""Calculate the optimal layout for 2+ cameras."""
|
"""Calculate the optimal layout for 2+ cameras."""
|
||||||
|
|
||||||
def map_layout(
|
def find_available_x(
|
||||||
camera_layout: list[list[Any]], row_height: int
|
current_x: int,
|
||||||
) -> tuple[int, int, Optional[list[list[Any]]]]:
|
width: int,
|
||||||
"""Map the calculated layout."""
|
reserved_ranges: list[tuple[int, int]],
|
||||||
candidate_layout = []
|
max_width: int,
|
||||||
starting_x = 0
|
) -> Optional[int]:
|
||||||
x = 0
|
"""Find the first horizontal slot that does not collide with reservations."""
|
||||||
max_width = 0
|
x = current_x
|
||||||
y = 0
|
|
||||||
|
|
||||||
for row in camera_layout:
|
for reserved_start, reserved_end in sorted(reserved_ranges):
|
||||||
final_row = []
|
if x >= reserved_end:
|
||||||
max_width = max(max_width, x)
|
continue
|
||||||
x = starting_x
|
|
||||||
for cameras in row:
|
|
||||||
camera_dims = self.cameras[cameras[0]]["dimensions"].copy()
|
|
||||||
camera_aspect = cameras[1]
|
|
||||||
|
|
||||||
if camera_dims[1] > camera_dims[0]:
|
if x + width <= reserved_start:
|
||||||
scaled_height = int(row_height * 2)
|
return x
|
||||||
scaled_width = int(scaled_height * camera_aspect)
|
|
||||||
starting_x = scaled_width
|
|
||||||
else:
|
|
||||||
scaled_height = row_height
|
|
||||||
scaled_width = int(scaled_height * camera_aspect)
|
|
||||||
|
|
||||||
# layout is too large
|
x = max(x, reserved_end)
|
||||||
if (
|
|
||||||
x + scaled_width > self.canvas.width
|
|
||||||
or y + scaled_height > self.canvas.height
|
|
||||||
):
|
|
||||||
return x + scaled_width, y + scaled_height, None
|
|
||||||
|
|
||||||
final_row.append((cameras[0], (x, y, scaled_width, scaled_height)))
|
if x + width <= max_width:
|
||||||
x += scaled_width
|
return x
|
||||||
|
|
||||||
y += row_height
|
|
||||||
candidate_layout.append(final_row)
|
|
||||||
|
|
||||||
if max_width == 0:
|
|
||||||
max_width = x
|
|
||||||
|
|
||||||
return max_width, y, candidate_layout
|
|
||||||
|
|
||||||
canvas_aspect_x, canvas_aspect_y = self.canvas.get_aspect(coefficient)
|
|
||||||
camera_layout: list[list[Any]] = []
|
|
||||||
camera_layout.append([])
|
|
||||||
starting_x = 0
|
|
||||||
x = starting_x
|
|
||||||
y = 0
|
|
||||||
y_i = 0
|
|
||||||
max_y = 0
|
|
||||||
for camera in cameras_to_add:
|
|
||||||
camera_dims = self.cameras[camera]["dimensions"].copy()
|
|
||||||
camera_aspect_x, camera_aspect_y = self.canvas.get_camera_aspect(
|
|
||||||
camera, camera_dims[0], camera_dims[1]
|
|
||||||
)
|
|
||||||
|
|
||||||
if camera_dims[1] > camera_dims[0]:
|
|
||||||
portrait = True
|
|
||||||
else:
|
|
||||||
portrait = False
|
|
||||||
|
|
||||||
if (x + camera_aspect_x) <= canvas_aspect_x:
|
|
||||||
# insert if camera can fit on current row
|
|
||||||
camera_layout[y_i].append(
|
|
||||||
(
|
|
||||||
camera,
|
|
||||||
camera_aspect_x / camera_aspect_y,
|
|
||||||
)
|
|
||||||
)
|
|
||||||
|
|
||||||
if portrait:
|
|
||||||
starting_x = camera_aspect_x
|
|
||||||
else:
|
|
||||||
max_y = max(
|
|
||||||
max_y,
|
|
||||||
camera_aspect_y,
|
|
||||||
)
|
|
||||||
|
|
||||||
x += camera_aspect_x
|
|
||||||
else:
|
|
||||||
# move on to the next row and insert
|
|
||||||
y += max_y
|
|
||||||
y_i += 1
|
|
||||||
camera_layout.append([])
|
|
||||||
x = starting_x
|
|
||||||
|
|
||||||
if x + camera_aspect_x > canvas_aspect_x:
|
|
||||||
return None
|
|
||||||
|
|
||||||
camera_layout[y_i].append(
|
|
||||||
(
|
|
||||||
camera,
|
|
||||||
camera_aspect_x / camera_aspect_y,
|
|
||||||
)
|
|
||||||
)
|
|
||||||
x += camera_aspect_x
|
|
||||||
|
|
||||||
if y + max_y > canvas_aspect_y:
|
|
||||||
return None
|
return None
|
||||||
|
|
||||||
row_height = int(self.canvas.height / coefficient)
|
def map_layout(row_height: int) -> tuple[int, int, Optional[list[list[Any]]]]:
|
||||||
total_width, total_height, standard_candidate_layout = map_layout(
|
"""Lay out cameras row by row while reserving portrait spans for the next row."""
|
||||||
camera_layout, row_height
|
candidate_layout: list[list[Any]] = []
|
||||||
)
|
reserved_ranges: dict[int, list[tuple[int, int]]] = {}
|
||||||
|
current_row: list[Any] = []
|
||||||
|
row_index = 0
|
||||||
|
row_y = 0
|
||||||
|
row_x = 0
|
||||||
|
max_width = 0
|
||||||
|
max_height = 0
|
||||||
|
|
||||||
|
for camera in cameras_to_add:
|
||||||
|
camera_dims = self.cameras[camera]["dimensions"].copy()
|
||||||
|
camera_aspect_x, camera_aspect_y = self.canvas.get_camera_aspect(
|
||||||
|
camera, camera_dims[0], camera_dims[1]
|
||||||
|
)
|
||||||
|
portrait = camera_dims[1] > camera_dims[0]
|
||||||
|
scaled_height = row_height * 2 if portrait else row_height
|
||||||
|
scaled_width = int(scaled_height * (camera_aspect_x / camera_aspect_y))
|
||||||
|
|
||||||
|
while True:
|
||||||
|
x = find_available_x(
|
||||||
|
row_x,
|
||||||
|
scaled_width,
|
||||||
|
reserved_ranges.get(row_index, []),
|
||||||
|
self.canvas.width,
|
||||||
|
)
|
||||||
|
|
||||||
|
if x is not None and row_y + scaled_height <= self.canvas.height:
|
||||||
|
current_row.append(
|
||||||
|
(camera, (x, row_y, scaled_width, scaled_height))
|
||||||
|
)
|
||||||
|
row_x = x + scaled_width
|
||||||
|
max_width = max(max_width, row_x)
|
||||||
|
max_height = max(max_height, row_y + scaled_height)
|
||||||
|
|
||||||
|
if portrait:
|
||||||
|
reserved_ranges.setdefault(row_index + 1, []).append(
|
||||||
|
(x, row_x)
|
||||||
|
)
|
||||||
|
|
||||||
|
break
|
||||||
|
|
||||||
|
if current_row:
|
||||||
|
candidate_layout.append(current_row)
|
||||||
|
current_row = []
|
||||||
|
|
||||||
|
row_index += 1
|
||||||
|
row_y = row_index * row_height
|
||||||
|
row_x = 0
|
||||||
|
|
||||||
|
if row_y + scaled_height > self.canvas.height:
|
||||||
|
overflow_width = max(max_width, scaled_width)
|
||||||
|
overflow_height = row_y + scaled_height
|
||||||
|
return overflow_width, overflow_height, None
|
||||||
|
|
||||||
|
if current_row:
|
||||||
|
candidate_layout.append(current_row)
|
||||||
|
|
||||||
|
return max_width, max_height, candidate_layout
|
||||||
|
|
||||||
|
row_height = max(1, int(self.canvas.height / coefficient))
|
||||||
|
total_width, total_height, standard_candidate_layout = map_layout(row_height)
|
||||||
|
|
||||||
if not standard_candidate_layout:
|
if not standard_candidate_layout:
|
||||||
# if standard layout didn't work
|
# if standard layout didn't work
|
||||||
@ -708,9 +687,9 @@ class BirdsEyeFrameManager:
|
|||||||
total_width / self.canvas.width,
|
total_width / self.canvas.width,
|
||||||
total_height / self.canvas.height,
|
total_height / self.canvas.height,
|
||||||
)
|
)
|
||||||
row_height = int(row_height / scale_down_percent)
|
row_height = max(1, int(row_height / scale_down_percent))
|
||||||
total_width, total_height, standard_candidate_layout = map_layout(
|
total_width, total_height, standard_candidate_layout = map_layout(
|
||||||
camera_layout, row_height
|
row_height
|
||||||
)
|
)
|
||||||
|
|
||||||
if not standard_candidate_layout:
|
if not standard_candidate_layout:
|
||||||
@ -724,8 +703,8 @@ class BirdsEyeFrameManager:
|
|||||||
1 / (total_width / self.canvas.width),
|
1 / (total_width / self.canvas.width),
|
||||||
1 / (total_height / self.canvas.height),
|
1 / (total_height / self.canvas.height),
|
||||||
)
|
)
|
||||||
row_height = int(row_height * scale_up_percent)
|
row_height = max(1, int(row_height * scale_up_percent))
|
||||||
_, _, scaled_layout = map_layout(camera_layout, row_height)
|
_, _, scaled_layout = map_layout(row_height)
|
||||||
|
|
||||||
if scaled_layout:
|
if scaled_layout:
|
||||||
return scaled_layout
|
return scaled_layout
|
||||||
|
|||||||
@ -28,7 +28,7 @@ from frigate.ffmpeg_presets import (
|
|||||||
EncodeTypeEnum,
|
EncodeTypeEnum,
|
||||||
parse_preset_hardware_acceleration_encode,
|
parse_preset_hardware_acceleration_encode,
|
||||||
)
|
)
|
||||||
from frigate.models import Export, Previews, Recordings
|
from frigate.models import Export, Previews, Recordings, ReviewSegment
|
||||||
from frigate.util.time import is_current_hour
|
from frigate.util.time import is_current_hour
|
||||||
|
|
||||||
logger = logging.getLogger(__name__)
|
logger = logging.getLogger(__name__)
|
||||||
@ -347,6 +347,122 @@ class RecordingExporter(threading.Thread):
|
|||||||
# return in iso format
|
# return in iso format
|
||||||
return datetime.datetime.fromtimestamp(timestamp).strftime("%Y-%m-%d %H:%M:%S")
|
return datetime.datetime.fromtimestamp(timestamp).strftime("%Y-%m-%d %H:%M:%S")
|
||||||
|
|
||||||
|
def _chapter_metadata_path(self) -> str:
|
||||||
|
return os.path.join(CACHE_DIR, f"export_chapters_{self.export_id}.txt")
|
||||||
|
|
||||||
|
def _build_chapter_metadata_file(self, recordings: list) -> Optional[str]:
|
||||||
|
"""Write an FFmpeg metadata file with chapters for review items in range.
|
||||||
|
|
||||||
|
Chapter offsets are computed in *output time*: the VOD endpoint
|
||||||
|
concatenates recording clips back-to-back, so wall-clock gaps
|
||||||
|
between recordings collapse in the produced video. We walk the
|
||||||
|
same recording rows that feed the playlist and convert each
|
||||||
|
review item's wall-clock boundaries into output-time offsets.
|
||||||
|
Returns ``None`` when there are no recordings, no review items,
|
||||||
|
or any chapter would have zero output duration.
|
||||||
|
"""
|
||||||
|
if not recordings:
|
||||||
|
return None
|
||||||
|
|
||||||
|
windows: list[tuple[float, float, float]] = []
|
||||||
|
output_offset = 0.0
|
||||||
|
for rec in recordings:
|
||||||
|
clipped_start = max(float(rec.start_time), float(self.start_time))
|
||||||
|
clipped_end = min(float(rec.end_time), float(self.end_time))
|
||||||
|
if clipped_end <= clipped_start:
|
||||||
|
continue
|
||||||
|
windows.append((clipped_start, clipped_end, output_offset))
|
||||||
|
output_offset += clipped_end - clipped_start
|
||||||
|
|
||||||
|
if not windows:
|
||||||
|
return None
|
||||||
|
|
||||||
|
try:
|
||||||
|
review_rows = list(
|
||||||
|
ReviewSegment.select(
|
||||||
|
ReviewSegment.start_time,
|
||||||
|
ReviewSegment.end_time,
|
||||||
|
ReviewSegment.severity,
|
||||||
|
ReviewSegment.data,
|
||||||
|
)
|
||||||
|
.where(
|
||||||
|
ReviewSegment.start_time.between(self.start_time, self.end_time)
|
||||||
|
| ReviewSegment.end_time.between(self.start_time, self.end_time)
|
||||||
|
| (
|
||||||
|
(self.start_time > ReviewSegment.start_time)
|
||||||
|
& (self.end_time < ReviewSegment.end_time)
|
||||||
|
)
|
||||||
|
)
|
||||||
|
.where(ReviewSegment.camera == self.camera)
|
||||||
|
.order_by(ReviewSegment.start_time.asc())
|
||||||
|
.iterator()
|
||||||
|
)
|
||||||
|
except Exception:
|
||||||
|
logger.exception(
|
||||||
|
"Failed to query review segments for export %s", self.export_id
|
||||||
|
)
|
||||||
|
return None
|
||||||
|
|
||||||
|
if not review_rows:
|
||||||
|
return None
|
||||||
|
|
||||||
|
total_output = windows[-1][2] + (windows[-1][1] - windows[-1][0])
|
||||||
|
|
||||||
|
def wall_to_output(t: float) -> float:
|
||||||
|
t = max(float(self.start_time), min(float(self.end_time), t))
|
||||||
|
for w_start, w_end, w_offset in windows:
|
||||||
|
if t < w_start:
|
||||||
|
return w_offset
|
||||||
|
if t <= w_end:
|
||||||
|
return w_offset + (t - w_start)
|
||||||
|
return total_output
|
||||||
|
|
||||||
|
chapter_blocks: list[str] = []
|
||||||
|
for review in review_rows:
|
||||||
|
start_out = wall_to_output(float(review.start_time))
|
||||||
|
end_out = wall_to_output(float(review.end_time))
|
||||||
|
|
||||||
|
# Drop chapters that fall entirely in a recording gap, or are
|
||||||
|
# too short to be navigable in a player.
|
||||||
|
if end_out - start_out < 1.0:
|
||||||
|
continue
|
||||||
|
|
||||||
|
data = review.data or {}
|
||||||
|
labels: list[str] = []
|
||||||
|
for obj in data.get("objects") or []:
|
||||||
|
label = str(obj).split("-")[0]
|
||||||
|
if label and label not in labels:
|
||||||
|
labels.append(label)
|
||||||
|
|
||||||
|
title = str(review.severity).capitalize()
|
||||||
|
if labels:
|
||||||
|
title = f"{title}: {', '.join(labels)}"
|
||||||
|
|
||||||
|
chapter_blocks.append(
|
||||||
|
"[CHAPTER]\n"
|
||||||
|
"TIMEBASE=1/1000\n"
|
||||||
|
f"START={int(start_out * 1000)}\n"
|
||||||
|
f"END={int(end_out * 1000)}\n"
|
||||||
|
f"title={title}"
|
||||||
|
)
|
||||||
|
|
||||||
|
if not chapter_blocks:
|
||||||
|
return None
|
||||||
|
|
||||||
|
meta_path = self._chapter_metadata_path()
|
||||||
|
try:
|
||||||
|
with open(meta_path, "w", encoding="utf-8") as f:
|
||||||
|
f.write(";FFMETADATA1\n")
|
||||||
|
f.write("\n".join(chapter_blocks))
|
||||||
|
f.write("\n")
|
||||||
|
except OSError:
|
||||||
|
logger.exception(
|
||||||
|
"Failed to write chapter metadata file for export %s", self.export_id
|
||||||
|
)
|
||||||
|
return None
|
||||||
|
|
||||||
|
return meta_path
|
||||||
|
|
||||||
def save_thumbnail(self, id: str) -> str:
|
def save_thumbnail(self, id: str) -> str:
|
||||||
thumb_path = os.path.join(CLIPS_DIR, f"export/{id}.webp")
|
thumb_path = os.path.join(CLIPS_DIR, f"export/{id}.webp")
|
||||||
|
|
||||||
@ -451,6 +567,24 @@ class RecordingExporter(threading.Thread):
|
|||||||
if type(internal_port) is str:
|
if type(internal_port) is str:
|
||||||
internal_port = int(internal_port.split(":")[-1])
|
internal_port = int(internal_port.split(":")[-1])
|
||||||
|
|
||||||
|
recordings = list(
|
||||||
|
Recordings.select(
|
||||||
|
Recordings.start_time,
|
||||||
|
Recordings.end_time,
|
||||||
|
)
|
||||||
|
.where(
|
||||||
|
Recordings.start_time.between(self.start_time, self.end_time)
|
||||||
|
| Recordings.end_time.between(self.start_time, self.end_time)
|
||||||
|
| (
|
||||||
|
(self.start_time > Recordings.start_time)
|
||||||
|
& (self.end_time < Recordings.end_time)
|
||||||
|
)
|
||||||
|
)
|
||||||
|
.where(Recordings.camera == self.camera)
|
||||||
|
.order_by(Recordings.start_time.asc())
|
||||||
|
.iterator()
|
||||||
|
)
|
||||||
|
|
||||||
playlist_lines: list[str] = []
|
playlist_lines: list[str] = []
|
||||||
if (self.end_time - self.start_time) <= MAX_PLAYLIST_SECONDS:
|
if (self.end_time - self.start_time) <= MAX_PLAYLIST_SECONDS:
|
||||||
playlist_url = f"http://127.0.0.1:{internal_port}/vod/{self.camera}/start/{self.start_time}/end/{self.end_time}/index.m3u8"
|
playlist_url = f"http://127.0.0.1:{internal_port}/vod/{self.camera}/start/{self.start_time}/end/{self.end_time}/index.m3u8"
|
||||||
@ -458,32 +592,13 @@ class RecordingExporter(threading.Thread):
|
|||||||
f"-y -protocol_whitelist pipe,file,http,tcp -i {playlist_url}"
|
f"-y -protocol_whitelist pipe,file,http,tcp -i {playlist_url}"
|
||||||
)
|
)
|
||||||
else:
|
else:
|
||||||
# get full set of recordings
|
# Chunk the recording rows into pages so each playlist line
|
||||||
export_recordings = (
|
# references a bounded sub-range rather than the full export.
|
||||||
Recordings.select(
|
|
||||||
Recordings.start_time,
|
|
||||||
Recordings.end_time,
|
|
||||||
)
|
|
||||||
.where(
|
|
||||||
Recordings.start_time.between(self.start_time, self.end_time)
|
|
||||||
| Recordings.end_time.between(self.start_time, self.end_time)
|
|
||||||
| (
|
|
||||||
(self.start_time > Recordings.start_time)
|
|
||||||
& (self.end_time < Recordings.end_time)
|
|
||||||
)
|
|
||||||
)
|
|
||||||
.where(Recordings.camera == self.camera)
|
|
||||||
.order_by(Recordings.start_time.asc())
|
|
||||||
)
|
|
||||||
|
|
||||||
# Use pagination to process records in chunks
|
|
||||||
page_size = 1000
|
page_size = 1000
|
||||||
num_pages = (export_recordings.count() + page_size - 1) // page_size
|
for i in range(0, len(recordings), page_size):
|
||||||
|
chunk = recordings[i : i + page_size]
|
||||||
for page in range(1, num_pages + 1):
|
|
||||||
playlist = export_recordings.paginate(page, page_size)
|
|
||||||
playlist_lines.append(
|
playlist_lines.append(
|
||||||
f"file 'http://127.0.0.1:{internal_port}/vod/{self.camera}/start/{float(playlist[0].start_time)}/end/{float(playlist[-1].end_time)}/index.m3u8'"
|
f"file 'http://127.0.0.1:{internal_port}/vod/{self.camera}/start/{float(chunk[0].start_time)}/end/{float(chunk[-1].end_time)}/index.m3u8'"
|
||||||
)
|
)
|
||||||
|
|
||||||
ffmpeg_input = "-y -protocol_whitelist pipe,file,http,tcp -f concat -safe 0 -i /dev/stdin"
|
ffmpeg_input = "-y -protocol_whitelist pipe,file,http,tcp -f concat -safe 0 -i /dev/stdin"
|
||||||
@ -504,8 +619,12 @@ class RecordingExporter(threading.Thread):
|
|||||||
)
|
)
|
||||||
).split(" ")
|
).split(" ")
|
||||||
else:
|
else:
|
||||||
|
chapters_path = self._build_chapter_metadata_file(recordings)
|
||||||
|
chapter_args = (
|
||||||
|
f" -i {chapters_path} -map 0 -map_metadata 1" if chapters_path else ""
|
||||||
|
)
|
||||||
ffmpeg_cmd = (
|
ffmpeg_cmd = (
|
||||||
f"{self.config.ffmpeg.ffmpeg_path} -hide_banner {ffmpeg_input} -c copy -movflags +faststart"
|
f"{self.config.ffmpeg.ffmpeg_path} -hide_banner {ffmpeg_input}{chapter_args} -c copy -movflags +faststart"
|
||||||
).split(" ")
|
).split(" ")
|
||||||
|
|
||||||
# add metadata
|
# add metadata
|
||||||
@ -691,6 +810,8 @@ class RecordingExporter(threading.Thread):
|
|||||||
ffmpeg_cmd, playlist_lines, step="encoding_retry"
|
ffmpeg_cmd, playlist_lines, step="encoding_retry"
|
||||||
)
|
)
|
||||||
|
|
||||||
|
Path(self._chapter_metadata_path()).unlink(missing_ok=True)
|
||||||
|
|
||||||
if returncode != 0:
|
if returncode != 0:
|
||||||
logger.error(
|
logger.error(
|
||||||
f"Failed to export {self.playback_source.value} for command {' '.join(ffmpeg_cmd)}"
|
f"Failed to export {self.playback_source.value} for command {' '.join(ffmpeg_cmd)}"
|
||||||
|
|||||||
@ -1,11 +1,64 @@
|
|||||||
"""Test camera user and password cleanup."""
|
"""Tests for Birdseye canvas sizing and layout behavior."""
|
||||||
|
|
||||||
import unittest
|
import unittest
|
||||||
|
from multiprocessing import Event
|
||||||
|
|
||||||
from frigate.output.birdseye import get_canvas_shape
|
from frigate.config import FrigateConfig
|
||||||
|
from frigate.output.birdseye import BirdsEyeFrameManager, get_canvas_shape
|
||||||
|
|
||||||
|
|
||||||
class TestBirdseye(unittest.TestCase):
|
class TestBirdseye(unittest.TestCase):
|
||||||
|
def _build_manager(
|
||||||
|
self, camera_dimensions: dict[str, tuple[int, int]]
|
||||||
|
) -> BirdsEyeFrameManager:
|
||||||
|
config = {
|
||||||
|
"mqtt": {"host": "mqtt"},
|
||||||
|
"birdseye": {"width": 1280, "height": 720},
|
||||||
|
"cameras": {},
|
||||||
|
}
|
||||||
|
|
||||||
|
for order, (camera, dimensions) in enumerate(
|
||||||
|
camera_dimensions.items(), start=1
|
||||||
|
):
|
||||||
|
config["cameras"][camera] = {
|
||||||
|
"ffmpeg": {
|
||||||
|
"inputs": [
|
||||||
|
{
|
||||||
|
"path": f"rtsp://10.0.0.1:554/{camera}",
|
||||||
|
"roles": ["detect"],
|
||||||
|
}
|
||||||
|
]
|
||||||
|
},
|
||||||
|
"detect": {
|
||||||
|
"width": dimensions[0],
|
||||||
|
"height": dimensions[1],
|
||||||
|
"fps": 5,
|
||||||
|
},
|
||||||
|
"birdseye": {"order": order},
|
||||||
|
}
|
||||||
|
|
||||||
|
return BirdsEyeFrameManager(FrigateConfig(**config), Event())
|
||||||
|
|
||||||
|
def _assert_no_overlaps(
|
||||||
|
self, layout: list[list[tuple[str, tuple[int, int, int, int]]]]
|
||||||
|
):
|
||||||
|
rectangles = [position for row in layout for _, position in row]
|
||||||
|
|
||||||
|
for index, rect in enumerate(rectangles):
|
||||||
|
x1, y1, width1, height1 = rect
|
||||||
|
for other in rectangles[index + 1 :]:
|
||||||
|
x2, y2, width2, height2 = other
|
||||||
|
overlap = (
|
||||||
|
x1 < x2 + width2
|
||||||
|
and x2 < x1 + width1
|
||||||
|
and y1 < y2 + height2
|
||||||
|
and y2 < y1 + height1
|
||||||
|
)
|
||||||
|
self.assertFalse(
|
||||||
|
overlap,
|
||||||
|
msg=f"Overlapping rectangles found: {rect} and {other}",
|
||||||
|
)
|
||||||
|
|
||||||
def test_16x9(self):
|
def test_16x9(self):
|
||||||
"""Test 16x9 aspect ratio works as expected for birdseye."""
|
"""Test 16x9 aspect ratio works as expected for birdseye."""
|
||||||
width = 1280
|
width = 1280
|
||||||
@ -45,3 +98,104 @@ class TestBirdseye(unittest.TestCase):
|
|||||||
canvas_width, canvas_height = get_canvas_shape(width, height)
|
canvas_width, canvas_height = get_canvas_shape(width, height)
|
||||||
assert canvas_width == width # width will be the same
|
assert canvas_width == width # width will be the same
|
||||||
assert canvas_height != height
|
assert canvas_height != height
|
||||||
|
|
||||||
|
def test_portrait_camera_does_not_overlap_next_row(self):
|
||||||
|
"""Portrait cameras should reserve their real horizontal position on the next row."""
|
||||||
|
manager = self._build_manager(
|
||||||
|
{
|
||||||
|
"cam_a": (1280, 720),
|
||||||
|
"cam_p": (360, 640),
|
||||||
|
"cam_b": (1280, 720),
|
||||||
|
"cam_c": (640, 480),
|
||||||
|
}
|
||||||
|
)
|
||||||
|
|
||||||
|
layout = manager.calculate_layout(["cam_a", "cam_p", "cam_b", "cam_c"], 3)
|
||||||
|
|
||||||
|
self.assertIsNotNone(layout)
|
||||||
|
assert layout is not None
|
||||||
|
self._assert_no_overlaps(layout)
|
||||||
|
|
||||||
|
cam_c = [
|
||||||
|
position for row in layout for camera, position in row if camera == "cam_c"
|
||||||
|
][0]
|
||||||
|
self.assertEqual(cam_c[0], 0)
|
||||||
|
|
||||||
|
def test_portrait_reservation_only_applies_to_next_row(self):
|
||||||
|
"""Portrait reservations should not push later rows after the span ends."""
|
||||||
|
manager = self._build_manager(
|
||||||
|
{
|
||||||
|
"cam_a": (1280, 720),
|
||||||
|
"cam_p": (360, 640),
|
||||||
|
"cam_b": (1280, 720),
|
||||||
|
"cam_c": (1280, 720),
|
||||||
|
"cam_d": (1280, 720),
|
||||||
|
"cam_e": (1280, 720),
|
||||||
|
}
|
||||||
|
)
|
||||||
|
|
||||||
|
layout = manager.calculate_layout(
|
||||||
|
["cam_a", "cam_p", "cam_b", "cam_c", "cam_d", "cam_e"],
|
||||||
|
3,
|
||||||
|
)
|
||||||
|
|
||||||
|
self.assertIsNotNone(layout)
|
||||||
|
assert layout is not None
|
||||||
|
self._assert_no_overlaps(layout)
|
||||||
|
|
||||||
|
cam_e = [
|
||||||
|
position for row in layout for camera, position in row if camera == "cam_e"
|
||||||
|
][0]
|
||||||
|
self.assertEqual(cam_e[0], 0)
|
||||||
|
|
||||||
|
def test_multiple_portraits_reserve_distinct_ranges(self):
|
||||||
|
"""Multiple portrait cameras in one row should reserve separate spans below them."""
|
||||||
|
manager = self._build_manager(
|
||||||
|
{
|
||||||
|
"cam_a": (640, 480),
|
||||||
|
"cam_p1": (360, 640),
|
||||||
|
"cam_p2": (360, 640),
|
||||||
|
"cam_b": (640, 480),
|
||||||
|
"cam_c": (1280, 720),
|
||||||
|
"cam_d": (640, 480),
|
||||||
|
}
|
||||||
|
)
|
||||||
|
|
||||||
|
layout = manager.calculate_layout(
|
||||||
|
["cam_a", "cam_p1", "cam_p2", "cam_b", "cam_c", "cam_d"],
|
||||||
|
4,
|
||||||
|
)
|
||||||
|
|
||||||
|
self.assertIsNotNone(layout)
|
||||||
|
assert layout is not None
|
||||||
|
self._assert_no_overlaps(layout)
|
||||||
|
|
||||||
|
def test_two_landscapes_then_portrait_then_two_landscapes(self):
|
||||||
|
"""A portrait after two landscapes should reserve only its own tail span."""
|
||||||
|
manager = self._build_manager(
|
||||||
|
{
|
||||||
|
"cam_a": (1280, 720),
|
||||||
|
"cam_b": (1280, 720),
|
||||||
|
"cam_p": (360, 640),
|
||||||
|
"cam_c": (1280, 720),
|
||||||
|
"cam_d": (1280, 720),
|
||||||
|
}
|
||||||
|
)
|
||||||
|
|
||||||
|
layout = manager.calculate_layout(
|
||||||
|
["cam_a", "cam_b", "cam_p", "cam_c", "cam_d"],
|
||||||
|
3,
|
||||||
|
)
|
||||||
|
|
||||||
|
self.assertIsNotNone(layout)
|
||||||
|
assert layout is not None
|
||||||
|
self._assert_no_overlaps(layout)
|
||||||
|
|
||||||
|
cam_c = [
|
||||||
|
position for row in layout for camera, position in row if camera == "cam_c"
|
||||||
|
][0]
|
||||||
|
cam_d = [
|
||||||
|
position for row in layout for camera, position in row if camera == "cam_d"
|
||||||
|
][0]
|
||||||
|
self.assertEqual(cam_c[0], 0)
|
||||||
|
self.assertEqual(cam_d[0], cam_c[0] + cam_c[2])
|
||||||
|
|||||||
@ -85,7 +85,7 @@ export default function ReviewCard({
|
|||||||
{ playback: "realtime" },
|
{ playback: "realtime" },
|
||||||
)
|
)
|
||||||
.then((response) => {
|
.then((response) => {
|
||||||
if (response.status == 200) {
|
if (response.status < 300) {
|
||||||
toast.success(t("export.toast.success"), {
|
toast.success(t("export.toast.success"), {
|
||||||
position: "top-center",
|
position: "top-center",
|
||||||
action: (
|
action: (
|
||||||
|
|||||||
@ -278,7 +278,7 @@ export default function EventView({
|
|||||||
{ playback: "realtime", image_path: review.thumb_path },
|
{ playback: "realtime", image_path: review.thumb_path },
|
||||||
)
|
)
|
||||||
.then((response) => {
|
.then((response) => {
|
||||||
if (response.status == 200) {
|
if (response.status < 300) {
|
||||||
toast.success(
|
toast.success(
|
||||||
t("export.toast.success", { ns: "components/dialog" }),
|
t("export.toast.success", { ns: "components/dialog" }),
|
||||||
{
|
{
|
||||||
|
|||||||
@ -357,7 +357,7 @@ export default function MotionSearchView({
|
|||||||
},
|
},
|
||||||
)
|
)
|
||||||
.then((response) => {
|
.then((response) => {
|
||||||
if (response.status == 200) {
|
if (response.status < 300) {
|
||||||
toast.success(
|
toast.success(
|
||||||
t("export.toast.success", { ns: "components/dialog" }),
|
t("export.toast.success", { ns: "components/dialog" }),
|
||||||
{
|
{
|
||||||
|
|||||||
Loading…
Reference in New Issue
Block a user