gx
37232ae1b9
cuda-grid-controller (Python 3.11+) — control plane между HA/MQTT/HTTP
и FFmpeg's vf_cuda_grid filter через ZMQ.
Modules (~700 LOC Python):
- config.py — Pydantic schema (broker, instances[], ha_discovery, http, log) + YAML loader
- layouts.py — registry известных layouts (sync с vf_cuda_grid.c Phase 2)
- ha_discovery.py — HA MQTT Discovery payloads (select.layout, sensor.current_layout,
binary_sensor.online per instance + global device entry)
- zmq_client.py — async ZMQ REQ socket к FFmpeg zmq filter
(target command args → reply parsing)
- state.py — in-memory ControllerState (active_layout per instance, asyncio.Lock)
- mqtt_loop.py — aiomqtt async loop: subscribe cuda_grid/cmd/<inst>/+/+,
publish cuda_grid/state/* (retained) + cuda_grid/event/*, LWT, HA status reconnect
- dispatch.py — CommandDispatcher: layout.set action → ZMQ send_command + state update + events
- http_api.py — FastAPI: /health, /layouts, /state, POST /layout/{inst}/set
- __main__.py — typer CLI, asyncio.gather(mqtt_loop, uvicorn.server)
Examples + Dockerfile:
- examples/controller.yaml — 2 instances (livingroom_tv, public_stream)
- Dockerfile — python:3.11-slim, ENTRYPOINT cuda-grid-controller
- README — overview, usage, FFmpeg side filter graph
End-to-end flow ready:
HA dashboard → MQTT → controller → ZMQ → FFmpeg process_command → layout switch
↓
state публикуется обратно в MQTT → HA UI обновляется
Phase 3 deliverable per gx/vf-cuda-grid#1. Phase 4 = overlays (rect/text/icon).
85 lines
3.1 KiB
Python
85 lines
3.1 KiB
Python
"""Home Assistant MQTT Discovery payloads.
|
|
|
|
Создаём per-instance entities:
|
|
select.cuda_grid_<instance>_layout — выбор активного layout
|
|
sensor.cuda_grid_<instance>_state — текущий layout (для UI)
|
|
binary_sensor.cuda_grid_controller_online — availability
|
|
|
|
См. https://www.home-assistant.io/integrations/mqtt/#mqtt-discovery
|
|
"""
|
|
|
|
from __future__ import annotations
|
|
|
|
import json
|
|
from typing import Any
|
|
|
|
from .config import HaDiscoveryCfg, InstanceCfg
|
|
from .layouts import PREDEFINED_LAYOUTS
|
|
|
|
|
|
def _device_dict(ha: HaDiscoveryCfg) -> dict[str, Any]:
|
|
return {
|
|
"identifiers": [ha.device_identifier],
|
|
"name": ha.device_name,
|
|
"manufacturer": "gx/vf-cuda-grid",
|
|
"model": "cuda-grid-controller",
|
|
"sw_version": "0.1.0",
|
|
}
|
|
|
|
|
|
def availability_topic(prefix_base: str = "cuda_grid") -> str:
|
|
"""Где controller публикует online/offline (LWT)."""
|
|
return f"{prefix_base}/state/online"
|
|
|
|
|
|
def discovery_payloads(ha: HaDiscoveryCfg, instances: list[InstanceCfg]) -> list[tuple[str, str]]:
|
|
"""Список (discovery_topic, payload_json) для publish при startup."""
|
|
out: list[tuple[str, str]] = []
|
|
avail = availability_topic()
|
|
|
|
# Per-instance entities
|
|
for inst in instances:
|
|
# select.layout
|
|
select_topic = f"{ha.prefix}/select/cuda_grid_{inst.name}/layout/config"
|
|
select_payload = {
|
|
"name": f"Layout ({inst.name})",
|
|
"unique_id": f"cuda_grid_{inst.name}_layout_select",
|
|
"command_topic": f"cuda_grid/cmd/{inst.name}/layout/set",
|
|
"state_topic": f"cuda_grid/state/{inst.name}/layout",
|
|
"options": PREDEFINED_LAYOUTS,
|
|
"availability_topic": avail,
|
|
"payload_available": "online",
|
|
"payload_not_available": "offline",
|
|
"device": _device_dict(ha),
|
|
}
|
|
out.append((select_topic, json.dumps(select_payload)))
|
|
|
|
# sensor.current_layout (text — duplicate state, удобно для automations)
|
|
sensor_topic = f"{ha.prefix}/sensor/cuda_grid_{inst.name}/current_layout/config"
|
|
sensor_payload = {
|
|
"name": f"Current layout ({inst.name})",
|
|
"unique_id": f"cuda_grid_{inst.name}_layout_sensor",
|
|
"state_topic": f"cuda_grid/state/{inst.name}/layout",
|
|
"availability_topic": avail,
|
|
"payload_available": "online",
|
|
"payload_not_available": "offline",
|
|
"device": _device_dict(ha),
|
|
"icon": "mdi:view-grid",
|
|
}
|
|
out.append((sensor_topic, json.dumps(sensor_payload)))
|
|
|
|
# Глобальный availability binary_sensor
|
|
online_topic = f"{ha.prefix}/binary_sensor/cuda_grid_controller/online/config"
|
|
online_payload = {
|
|
"name": "CUDA Grid Controller online",
|
|
"unique_id": "cuda_grid_controller_online",
|
|
"state_topic": avail,
|
|
"payload_on": "online",
|
|
"payload_off": "offline",
|
|
"device_class": "connectivity",
|
|
"device": _device_dict(ha),
|
|
}
|
|
out.append((online_topic, json.dumps(online_payload)))
|
|
|
|
return out
|