import os import cv2 def _parse_bool(value): if isinstance(value, bool): return value text = str(value).strip().lower() return text in {"1", "true", "yes", "on"} def _parse_int(value): return int(str(value).strip()) def _parse_float(value): return float(str(value).strip()) def _parse_str(value): return str(value) def _parse_backend(value): text = str(value).strip().lower() if text.lstrip("+-").isdigit(): return int(text) return { "any": cv2.CAP_ANY, "dshow": cv2.CAP_DSHOW, "msmf": cv2.CAP_MSMF, "v4l2": cv2.CAP_V4L2, }.get(text, cv2.CAP_DSHOW) def _parse_source(value): text = str(value).strip() if text and text.lstrip("+-").isdigit(): return int(text) return text def _apply_env_overrides(namespace, spec): for env_name, (key, parser) in spec.items(): if env_name not in os.environ: continue namespace[key] = parser(os.environ[env_name]) def apply_config_env_overrides(namespace): _apply_env_overrides( namespace, { "FPV_MODEL_PATH": ("MODEL_PATH", _parse_str), "FPV_DEVICE": ("DEVICE", _parse_int), "FPV_USE_HALF": ("USE_HALF", _parse_bool), "FPV_CONF": ("CONF", _parse_float), "FPV_IMG_SIZE_ROI": ("IMG_SIZE_ROI", _parse_int), "FPV_IMG_SIZE_FULL": ("IMG_SIZE_FULL", _parse_int), "FPV_MAX_DET": ("MAX_DET", _parse_int), "FPV_SOURCE": ("SOURCE", _parse_source), "FPV_CAP_BACKEND": ("CAP_BACKEND", _parse_backend), "FPV_CAMERA_WIDTH": ("CAMERA_WIDTH", _parse_int), "FPV_CAMERA_HEIGHT": ("CAMERA_HEIGHT", _parse_int), "FPV_CAMERA_FPS": ("CAMERA_FPS", _parse_int), "FPV_CAMERA_FOURCC": ("CAMERA_FOURCC", _parse_str), "FPV_VIDEO_REALTIME": ("VIDEO_REALTIME", _parse_bool), "FPV_SHOW_OUTPUT": ("SHOW_OUTPUT", _parse_bool), "FPV_SAVE_INFER_VIDEO": ("SAVE_INFER_VIDEO", _parse_bool), "FPV_OUT_VIDEO_PATH": ("OUT_VIDEO_PATH", _parse_str), "FPV_ARCHIVE_RECORD_MODE": ("ARCHIVE_RECORD_MODE", _parse_str), "FPV_DETECTION_CLIP_MAX_GAP_SEC": ("DETECTION_CLIP_MAX_GAP_SEC", _parse_float), "FPV_TARGET_OUT_FPS": ("TARGET_OUT_FPS", _parse_int), "FPV_FORCE_EFFECTIVE_PAL": ("FORCE_EFFECTIVE_PAL", _parse_bool), "FPV_EFFECTIVE_W": ("EFFECTIVE_W", _parse_int), "FPV_EFFECTIVE_H": ("EFFECTIVE_H", _parse_int), "FPV_UI_FRAME_EXPORT_ENABLE": ("UI_FRAME_EXPORT_ENABLE", _parse_bool), "FPV_UI_FRAME_EXPORT_PATH": ("UI_FRAME_EXPORT_PATH", _parse_str), "FPV_UI_FRAME_EXPORT_EVERY": ("UI_FRAME_EXPORT_EVERY", _parse_int), "FPV_UI_FRAME_EXPORT_JPEG_QUALITY": ("UI_FRAME_EXPORT_JPEG_QUALITY", _parse_int), "FPV_UI_FRAME_EXPORT_MAX_FPS": ("UI_FRAME_EXPORT_MAX_FPS", _parse_float), "FPV_REALTIME_SKIP_STALE_FRAMES": ("REALTIME_SKIP_STALE_FRAMES", _parse_bool), "FPV_REALTIME_MAX_SKIP_FRAMES": ("REALTIME_MAX_SKIP_FRAMES", _parse_int), "FPV_REALTIME_PREVIEW_SKIPPED_FRAMES": ("REALTIME_PREVIEW_SKIPPED_FRAMES", _parse_bool), "FPV_REALTIME_ANALYSIS_EVERY": ("REALTIME_ANALYSIS_EVERY", _parse_int), "FPV_GUIDANCE_EXPORT_ENABLE": ("GUIDANCE_EXPORT_ENABLE", _parse_bool), "FPV_GUIDANCE_EXPORT_PATH": ("GUIDANCE_EXPORT_PATH", _parse_str), "FPV_ERROR_OUTPUT_ENABLE": ("ERROR_OUTPUT_ENABLE", _parse_bool), "FPV_ERROR_OUTPUT_PROTOCOL": ("ERROR_OUTPUT_PROTOCOL", _parse_str), "FPV_ERROR_OUTPUT_HOST": ("ERROR_OUTPUT_HOST", _parse_str), "FPV_ERROR_OUTPUT_PORT": ("ERROR_OUTPUT_PORT", _parse_int), "FPV_ERROR_OUTPUT_OBJECT_ID": ("ERROR_OUTPUT_OBJECT_ID", _parse_int), "FPV_ERROR_OUTPUT_UNITS": ("ERROR_OUTPUT_UNITS", _parse_str), "FPV_ERROR_OUTPUT_EVERY": ("ERROR_OUTPUT_EVERY", _parse_int), "FPV_ERROR_OUTPUT_HFOV_DEG": ("ERROR_OUTPUT_HFOV_DEG", _parse_float), "FPV_ERROR_OUTPUT_VFOV_DEG": ("ERROR_OUTPUT_VFOV_DEG", _parse_float), "FPV_ERROR_OUTPUT_RANGE_M": ("ERROR_OUTPUT_RANGE_M", _parse_float), }, ) def apply_intercept_env_overrides(namespace): _apply_env_overrides( namespace, { "FPV_AUTOPILOT_ENABLE": ("AUTOPILOT_ENABLE", _parse_bool), "FPV_AUTOPILOT_BACKEND": ("AUTOPILOT_BACKEND", _parse_str), "FPV_AUTOPILOT_JSON_PATH": ("AUTOPILOT_JSON_PATH", _parse_str), "FPV_MAVLINK_CONNECTION": ("MAVLINK_CONNECTION", _parse_str), "FPV_MSP_PORT": ("MSP_PORT", _parse_str), "FPV_PROTO_UDP_ENABLE": ("PROTO_UDP_ENABLE", _parse_bool), "FPV_PROTO_UDP_HOST": ("PROTO_UDP_HOST", _parse_str), "FPV_PROTO_UDP_PORT": ("PROTO_UDP_PORT", _parse_int), "FPV_PROTO_UDP_DESCRIPTOR": ("PROTO_UDP_DESCRIPTOR", _parse_int), }, )