262 lines
8.7 KiB
Python
262 lines
8.7 KiB
Python
|
#!/usr/bin/env python3
|
|||
|
|
# -*- coding: utf-8 -*-
|
|||
|
|
# 文件名: camera_scanner.py
|
|||
|
|
|
|||
|
|
import json
|
|||
|
|
import logging
|
|||
|
|
import sys
|
|||
|
|
import time
|
|||
|
|
from pathlib import Path
|
|||
|
|
|
|||
|
|
import cv2
|
|||
|
|
|
|||
|
|
# 导入ConfigManager类
|
|||
|
|
from src.utils.config_manager import ConfigManager
|
|||
|
|
|
|||
|
|
# 添加项目根目录到系统路径，以便导入src中的模块
|
|||
|
|
project_root = Path(__file__).parent.parent
|
|||
|
|
sys.path.append(str(project_root))
|
|||
|
|
|
|||
|
|
# 设置日志记录
|
|||
|
|
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}")
|
|||
|
|
|
|||
|
|
# 读取一帧以确保摄像头正常工作
|
|||
|
|
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}")
|