#!/usr/bin/env python3 # -*- coding: utf-8 -*- # 文件名: camera_scanner.py import cv2 import json import logging import sys import time from pathlib import Path # 添加项目根目录到系统路径，以便导入src中的模块 project_root = Path(__file__).parent.parent sys.path.append(str(project_root)) # 导入ConfigManager类 from src.utils.config_manager import ConfigManager # 设置日志记录 logging.basicConfig( level=logging.INFO, format='%(asctime)s - %(name)s - %(levelname)s - %(message)s' ) logger = logging.getLogger("CameraScanner") def get_camera_capabilities(cam): """获取摄像头的参数和能力""" capabilities = {} # 获取可用的分辨率 standard_resolutions = [ (640, 480), # VGA (800, 600), # SVGA (1024, 768), # XGA (1280, 720), # HD (1280, 960), # 4:3 HD (1920, 1080), # Full HD (2560, 1440), # QHD (3840, 2160) # 4K UHD ] supported_resolutions = [] original_width = int(cam.get(cv2.CAP_PROP_FRAME_WIDTH)) original_height = int(cam.get(cv2.CAP_PROP_FRAME_HEIGHT)) # 记录原始分辨率 capabilities["default_resolution"] = (original_width, original_height) # 测试标准分辨率 for width, height in standard_resolutions: cam.set(cv2.CAP_PROP_FRAME_WIDTH, width) cam.set(cv2.CAP_PROP_FRAME_HEIGHT, height) actual_width = int(cam.get(cv2.CAP_PROP_FRAME_WIDTH)) actual_height = int(cam.get(cv2.CAP_PROP_FRAME_HEIGHT)) # 如果设置成功（实际分辨率与请求的相同） if actual_width == width and actual_height == height: supported_resolutions.append((width, height)) # 恢复原始分辨率 cam.set(cv2.CAP_PROP_FRAME_WIDTH, original_width) cam.set(cv2.CAP_PROP_FRAME_HEIGHT, original_height) capabilities["supported_resolutions"] = supported_resolutions # 获取帧率 fps = int(cam.get(cv2.CAP_PROP_FPS)) capabilities["fps"] = fps if fps > 0 else 30 # 默认为30fps # 获取后端名称 backend_name = cam.getBackendName() capabilities["backend"] = backend_name return capabilities def detect_cameras(): """检测并列出所有可用摄像头""" print("\n===== 摄像头设备检测 =====\n") # 获取ConfigManager实例 config_manager = ConfigManager.get_instance() # 获取当前相机配置 current_camera_config = config_manager.get_config("CAMERA", {}) logger.info(f"当前相机配置: {current_camera_config}") # 存储找到的设备 camera_devices = [] # 尝试打开多个摄像头索引 max_cameras_to_check = 10 # 最多检查10个摄像头索引 for i in range(max_cameras_to_check): try: # 尝试打开摄像头 cap = cv2.VideoCapture(i) if cap.isOpened(): # 获取摄像头信息 device_name = f"Camera {i}" try: # 在某些系统上可能可以获取设备名称 device_name = cap.getBackendName() + f" Camera {i}" except Exception as e: logger.warning(f"获取设备{i}名称失败: {e}") pass # 读取一帧以确保摄像头正常工作 ret, frame = cap.read() if not ret: print(f"设备 {i}: 打开成功但无法读取画面，跳过") cap.release() continue # 获取摄像头能力 capabilities = get_camera_capabilities(cap) # 打印设备信息 width, height = capabilities['default_resolution'] resolutions_str = ", ".join( [f"{w}x{h}" for w, h in capabilities['supported_resolutions']] ) print(f"设备 {i}: {device_name}") print(f" - 默认分辨率: {width}x{height}") print(f" - 支持分辨率: {resolutions_str}") print(f" - 帧率: {capabilities['fps']}") print(f" - 后端: {capabilities['backend']}") print("") # 添加到设备列表 camera_devices.append({ "index": i, "name": device_name, "capabilities": capabilities }) # 显示预览画面 print(f"正在显示设备 {i} 的预览画面，按 'q' 键继续...") preview_start = time.time() while time.time() - preview_start < 3: # 预览3秒 ret, frame = cap.read() if ret: cv2.imshow(f'Camera {i} Preview', frame) if cv2.waitKey(1) & 0xFF == ord('q'): break cv2.destroyAllWindows() cap.release() else: # 如果连续两个索引无法打开摄像头，则认为没有更多摄像头了 consecutive_failures = 0 for j in range(i, i + 2): temp_cap = cv2.VideoCapture(j) if not temp_cap.isOpened(): consecutive_failures += 1 temp_cap.release() if consecutive_failures >= 2 and i > 0: break except Exception as e: print(f"检测设备 {i} 时出错: {e}") # 总结找到的设备 print("\n===== 设备总结 =====\n") if not camera_devices: print("未找到可用的摄像头设备！") return None print(f"找到 {len(camera_devices)} 个摄像头设备:") for device in camera_devices: width, height = device['capabilities']['default_resolution'] print(f" - 设备 {device['index']}: {device['name']}") print(f" 分辨率: {width}x{height}") # 推荐最佳设备 print("\n===== 推荐设备 =====\n") # 首选高清摄像头，其次是分辨率最高的 recommended_camera = None highest_resolution = 0 for device in camera_devices: width, height = device['capabilities']['default_resolution'] resolution = width * height # 如果是HD或以上分辨率 if width >= 1280 and height >= 720: if resolution > highest_resolution: highest_resolution = resolution recommended_camera = device elif recommended_camera is None or resolution > highest_resolution: highest_resolution = resolution recommended_camera = device # 打印推荐设备 if recommended_camera: r_width, r_height = recommended_camera['capabilities']['default_resolution'] print(f"推荐摄像头: 设备 {recommended_camera['index']} " f"({recommended_camera['name']})") print(f" - 分辨率: {r_width}x{r_height}") print(f" - 帧率: {recommended_camera['capabilities']['fps']}") # 从现有配置中获取VL API信息 vl_url = current_camera_config.get( "Loacl_VL_url", "https://open.bigmodel.cn/api/paas/v4/" ) vl_api_key = current_camera_config.get("VLapi_key", "你自己的key") model = current_camera_config.get("models", "glm-4v-plus") # 生成配置文件示例 print("\n===== 配置文件示例 =====\n") new_camera_config = { "camera_index": recommended_camera['index'], "frame_width": r_width, "frame_height": r_height, "fps": recommended_camera['capabilities']['fps'], "Loacl_VL_url": vl_url, # 保留原有值 "VLapi_key": vl_api_key, # 保留原有值 "models": model # 保留原有值 } print(json.dumps(new_camera_config, indent=2, ensure_ascii=False)) # 询问是否更新配置文件 print("\n是否要更新配置文件中的摄像头配置？(y/n)") choice = input().strip().lower() if choice == 'y': try: # 使用ConfigManager更新配置 success = config_manager.update_config("CAMERA", new_camera_config) if success: print("\n摄像头配置已成功更新到config.json!") else: print("\n更新摄像头配置失败!") except Exception as e: logger.error(f"更新配置时出错: {e}") print(f"\n更新配置时出错: {e}") return camera_devices if __name__ == "__main__": try: cameras = detect_cameras() if cameras: print(f"\n检测到 {len(cameras)} 个摄像头设备！") else: print("\n未检测到可用的摄像头设备！") except Exception as e: logger.error(f"检测过程中出错: {e}") print(f"检测过程中出错: {e}")