gx
ac86534769
Frigate 0.17 schema отвергает objects.filters.<obj>.required_zones
("Extra inputs not permitted" → safe mode), поэтому фильтрацию по зонам
делаем на стороне composer'а. cfc_overlay_detbox_match_zones() сверяет
after.current_zones события с whitelist'ом overlay'я. end-события
вызываются безусловно (no-op если рамки нет).
CLI: --detection-cell key,camera,dx,dy,dw,dh,detect_w,detect_h[,zone1:zone2:...]
9-е поле опционально (colon-separated). Без него filter выключен.
Co-Authored-By: Claude Opus 4.7 <noreply@anthropic.com>
220 lines
12 KiB
C
220 lines
12 KiB
C
/* cuframes-composer — overlay'и (border, png-иконка, текст) поверх
|
||
* композиции.
|
||
*
|
||
* Архитектура:
|
||
* cfc_overlay_t — opaque объект одного overlay'я (border, png, text).
|
||
* composer ведёт список overlays + порядок (z-order); на каждом compose'е
|
||
* проходит по списку и рисует их поверх NV12 grid'а.
|
||
*
|
||
* Типы overlay'ев:
|
||
* BORDER — цветная рамка вокруг rect'а (Phase 3a). Реализован через 4
|
||
* fill_nv12 операции (top/bottom/left/right). Полезно для:
|
||
* - визуальной отбивки ячеек grid'а
|
||
* - подсветки активных/неактивных источников (красная при DEAD)
|
||
* PNG — RGBA-изображение (Phase 3b). Decode-раз через stb_image,
|
||
* upload в VRAM, на каждом кадре alpha-blit через
|
||
* cfc_kern_alpha_blit_rgba_*. Для статичных иконок (NO SIGNAL,
|
||
* offline, recording dot).
|
||
* TEXT — динамический рендер строки (Phase 3c). Требует FreeType +
|
||
* font atlas. Сделать позже когда FreeType добавлен как зависимость.
|
||
*
|
||
* Жизненный цикл:
|
||
* cfc_overlay_create_*() — выделяет ресурсы (для PNG/TEXT — VRAM-атлас).
|
||
* cfc_overlay_update_*() — обновляет параметры (положение, видимость, текст).
|
||
* cfc_overlay_draw() — вызывается composer'ом на каждом кадре.
|
||
* cfc_overlay_destroy() — освобождает VRAM.
|
||
*
|
||
* Phase 3a лимит: только BORDER через fill_nv12.
|
||
*
|
||
* Лицензия: LGPL-2.1+
|
||
*/
|
||
|
||
#ifndef CUFRAMES_COMPOSER_OVERLAY_H
|
||
#define CUFRAMES_COMPOSER_OVERLAY_H
|
||
|
||
#include <cuda.h>
|
||
#include <stdint.h>
|
||
|
||
#ifdef __cplusplus
|
||
extern "C" {
|
||
#endif
|
||
|
||
typedef enum cfc_overlay_type {
|
||
CFC_OVERLAY_BORDER = 0,
|
||
CFC_OVERLAY_PNG = 1, /* Phase 3b */
|
||
CFC_OVERLAY_TEXT = 2, /* Phase 3c */
|
||
CFC_OVERLAY_DETECTION_BOXES = 3, /* Phase 7 — Frigate-driven rectangles */
|
||
} cfc_overlay_type_t;
|
||
|
||
typedef struct cfc_overlay cfc_overlay_t;
|
||
|
||
/* Назначить overlay'ю короткий ID для lookup через control plane (Phase 4).
|
||
* ID копируется внутрь overlay'я (макс 31 символ). Без ID overlay тоже работает,
|
||
* но управлять им runtime'ом нельзя. Можно вызвать после create. */
|
||
int cfc_overlay_set_id(cfc_overlay_t *ov, const char *id);
|
||
|
||
/* Получить ID overlay'я (NULL если не задан). */
|
||
const char *cfc_overlay_get_id(const cfc_overlay_t *ov);
|
||
|
||
/* Получить тип overlay'я. */
|
||
cfc_overlay_type_t cfc_overlay_get_type(const cfc_overlay_t *ov);
|
||
|
||
/* Параметры BORDER overlay'я. */
|
||
typedef struct cfc_overlay_border_config {
|
||
int x, y, w, h; /* прямоугольник в full-res пикселях */
|
||
int thickness; /* толщина рамки в пикселях */
|
||
int color_y, color_u, color_v; /* цвет в BT.709 (Y:16-235, UV:16-240) */
|
||
int alpha; /* 0..255; 0 = invisible */
|
||
int visible; /* 0/1 — выводить ли вообще */
|
||
} cfc_overlay_border_config_t;
|
||
|
||
/* Создать BORDER overlay. */
|
||
int cfc_overlay_create_border(const cfc_overlay_border_config_t *cfg,
|
||
cfc_overlay_t **out);
|
||
|
||
/* Параметры PNG overlay'я. */
|
||
typedef struct cfc_overlay_png_config {
|
||
const char *path; /* путь к PNG-файлу (декод один раз) */
|
||
int x, y; /* позиция top-left на output буфере (чётные) */
|
||
int extra_alpha; /* 0..255 общий множитель прозрачности */
|
||
int visible; /* 0/1 — выводить ли */
|
||
} cfc_overlay_png_config_t;
|
||
|
||
/* Создать PNG overlay. Декодирует файл через libpng, аллоцирует RGBA-атлас
|
||
* в VRAM (cuMemAlloc), копирует туда декодированные пиксели один раз.
|
||
*
|
||
* Размер изображения берётся из PNG header'а — caller узнаёт width/height
|
||
* через cfc_overlay_png_size после create. */
|
||
int cfc_overlay_create_png(const cfc_overlay_png_config_t *cfg,
|
||
cfc_overlay_t **out);
|
||
|
||
/* Получить реальный размер декодированного PNG. После create — стабильны
|
||
* до destroy. */
|
||
int cfc_overlay_png_size(cfc_overlay_t *ov, int *width, int *height);
|
||
|
||
/* Обновить параметры PNG overlay'я (без re-decode). */
|
||
int cfc_overlay_update_png(cfc_overlay_t *ov,
|
||
const cfc_overlay_png_config_t *cfg);
|
||
|
||
/* Параметры TEXT overlay'я (Phase 3c). */
|
||
typedef struct cfc_overlay_text_config {
|
||
const char *font_path; /* путь к .ttf / .otf */
|
||
const char *text; /* UTF-8 строка для рендера */
|
||
int pixel_size; /* высота glyph'а в пикселях (10..200) */
|
||
int x, y; /* top-left на output буфере (чётные) */
|
||
int r, g, b; /* sRGB цвет 0..255 */
|
||
int extra_alpha; /* 0..255 общий множитель прозрачности */
|
||
int visible; /* 0/1 — выводить ли */
|
||
} cfc_overlay_text_config_t;
|
||
|
||
/* Создать TEXT overlay. Открывает font через FreeType, рендерит строку
|
||
* в RGBA-атлас на CPU (alpha-channel из glyph bitmap'ов anti-aliased),
|
||
* заливает в VRAM. */
|
||
int cfc_overlay_create_text(const cfc_overlay_text_config_t *cfg,
|
||
cfc_overlay_t **out);
|
||
|
||
/* Обновить TEXT overlay. Если text изменился — re-render atlas (VRAM
|
||
* перевыделяется). font_path и pixel_size менять нельзя — заведите новый
|
||
* overlay (face и связанные ресурсы пере-init'ить дорого). */
|
||
int cfc_overlay_update_text(cfc_overlay_t *ov,
|
||
const cfc_overlay_text_config_t *cfg);
|
||
|
||
/* Получить ширину/высоту текущего рендеренного текста (в пикселях). */
|
||
int cfc_overlay_text_size(cfc_overlay_t *ov, int *width, int *height);
|
||
|
||
/* ── DETECTION_BOXES (Phase 7) ─────────────────────────────────────────
|
||
* Managed-список border'ов для bbox'ов от Frigate. Один overlay соответствует
|
||
* одной композитор-ячейке (одной камере). MQTT-subscriber (отдельный thread)
|
||
* вызывает _upsert при каждом frigate/events update и _end при event завершён.
|
||
* draw_detection_boxes итерирует по active box'ам, проверяет TTL, рисует
|
||
* рамку с координатным mapping'ом detect → cell.
|
||
*
|
||
* Координаты Frigate отдаются в detect-разрешении камеры (640×480 для
|
||
* parking_overview), composer ячейка обычно 960×540 на 1080p output → нужен
|
||
* линейный scale. Mapping вынесен внутрь draw, не кэшируется — layout switch
|
||
* безопасен. */
|
||
typedef struct cfc_overlay_detbox_config {
|
||
/* Какая Frigate-камера → какая ячейка composer'а. */
|
||
const char *camera_key; /* "parking_overview" — для lookup'а */
|
||
int detect_w, detect_h; /* Frigate detect.{width,height} */
|
||
int cell_x, cell_y, cell_w, cell_h; /* куда мапится на output frame */
|
||
|
||
/* Стиль рамки. */
|
||
int thickness; /* px (4-8 рекомендуется) */
|
||
int color_y, color_u, color_v; /* BT.709 limited (Y=16-235, UV=16-240) */
|
||
int alpha; /* 0..255 */
|
||
|
||
/* TTL — независимая страховка от потери MQTT events.
|
||
* Если последний update был > stale_ms назад — рамка пропадает.
|
||
* Рекомендуется 5000-8000 мс (Frigate publish раз в минуту для stationary,
|
||
* меньше при движении). */
|
||
int stale_ms;
|
||
|
||
/* Список разрешённых zones — если задан, MQTT subscriber пропускает
|
||
* события у которых after.current_zones пусто либо не пересекается с
|
||
* этим списком. Это робастный фильтр когда Frigate 0.17 schema не даёт
|
||
* native objects.filters.<obj>.required_zones (см. commit 1726137).
|
||
* NULL или пустой массив → принимать все события. */
|
||
const char *const *required_zones; /* массив строк */
|
||
int required_zones_count;
|
||
} cfc_overlay_detbox_config_t;
|
||
|
||
int cfc_overlay_create_detection_boxes(
|
||
const cfc_overlay_detbox_config_t *cfg,
|
||
cfc_overlay_t **out
|
||
);
|
||
|
||
/* Получить camera_key из overlay'я — для MQTT subscriber'а чтобы найти
|
||
* правильный overlay по incoming event'у. */
|
||
const char *cfc_overlay_detbox_camera_key(cfc_overlay_t *ov);
|
||
|
||
/* Проверить пересечение current_zones события с required_zones overlay'я.
|
||
* - если required_zones пуст → всегда 1 (filter off)
|
||
* - если current_zones пуст → 0 (объект вне зон)
|
||
* - иначе 1 если хотя бы одна current ∈ required, 0 иначе
|
||
* Использует overlay subscriber для отсева street-флуда (см. commit 1726137). */
|
||
int cfc_overlay_detbox_match_zones(cfc_overlay_t *ov,
|
||
const char *const *current_zones,
|
||
int n);
|
||
|
||
/* Upsert одного active детекта.
|
||
* event_id — идентификатор Frigate event'а (для трекинга/end).
|
||
* label — "car", "person", и т.п. (для будущего цветового кодирования).
|
||
* x1/y1/x2/y2 — bbox в detect-разрешении (raw Frigate coords).
|
||
* frame_time_ms — Frigate frame_time для TTL.
|
||
* Thread-safe (mutex внутри). */
|
||
int cfc_overlay_detbox_upsert(
|
||
cfc_overlay_t *ov,
|
||
const char *event_id,
|
||
const char *label,
|
||
int x1, int y1, int x2, int y2,
|
||
int64_t frame_time_ms
|
||
);
|
||
|
||
/* End event'а — удалить рамку (вызывается при `type == "end"` в Frigate). */
|
||
int cfc_overlay_detbox_end(cfc_overlay_t *ov, const char *event_id);
|
||
|
||
/* Обновить параметры BORDER overlay'я (можно переключить visible,
|
||
* сменить цвет, изменить позицию). Thread-safe? Нет — caller должен сам
|
||
* заботиться о том, чтобы update не пересекался с draw. В рамках одного
|
||
* compose-thread'а это естественно. */
|
||
int cfc_overlay_update_border(cfc_overlay_t *ov,
|
||
const cfc_overlay_border_config_t *cfg);
|
||
|
||
/* Нарисовать overlay поверх NV12 кадра. Вызывается из cfc_composer_compose.
|
||
* Возвращает 0 при успехе. */
|
||
int cfc_overlay_draw(cfc_overlay_t *ov,
|
||
CUstream stream,
|
||
CUdeviceptr dst_y, int pitch_y,
|
||
CUdeviceptr dst_uv, int pitch_uv,
|
||
int frame_w, int frame_h);
|
||
|
||
/* Уничтожить overlay (освободить VRAM если был). */
|
||
int cfc_overlay_destroy(cfc_overlay_t *ov);
|
||
|
||
#ifdef __cplusplus
|
||
}
|
||
#endif
|
||
|
||
#endif /* CUFRAMES_COMPOSER_OVERLAY_H */
|