增加配置对应
This commit is contained in:
197
utils/device_scanner.py
Normal file
197
utils/device_scanner.py
Normal file
@@ -0,0 +1,197 @@
|
||||
"""
|
||||
设备扫描工具 - 通过唯一标识符而非索引来识别设备
|
||||
解决重启后设备序号混乱的问题
|
||||
"""
|
||||
import cv2
|
||||
import serial.tools.list_ports
|
||||
import warnings
|
||||
import sys
|
||||
import io
|
||||
from typing import List, Dict, Optional
|
||||
|
||||
def get_camera_info(cam_index: int) -> Optional[Dict]:
|
||||
"""
|
||||
获取采集卡的详细信息（包括设备名称）
|
||||
:param cam_index: 采集卡索引
|
||||
:return: 包含设备信息的字典，如果失败返回None
|
||||
"""
|
||||
old_stderr = sys.stderr
|
||||
suppressed_output = io.StringIO()
|
||||
|
||||
try:
|
||||
sys.stderr = suppressed_output
|
||||
|
||||
with warnings.catch_warnings():
|
||||
warnings.filterwarnings('ignore')
|
||||
|
||||
# 尝试多种后端
|
||||
backends = [
|
||||
(cam_index, cv2.CAP_DSHOW),
|
||||
(cam_index, cv2.CAP_ANY),
|
||||
(cam_index, None),
|
||||
]
|
||||
|
||||
cap = None
|
||||
for idx, backend in backends:
|
||||
try:
|
||||
if backend is not None:
|
||||
cap = cv2.VideoCapture(idx, backend)
|
||||
else:
|
||||
cap = cv2.VideoCapture(idx)
|
||||
|
||||
if cap.isOpened():
|
||||
ret, frame = cap.read()
|
||||
if ret and frame is not None:
|
||||
# 尝试获取设备名称（不同后端可能支持不同）
|
||||
device_name = None
|
||||
backend_name = None
|
||||
|
||||
if backend == cv2.CAP_DSHOW:
|
||||
backend_name = "DirectShow"
|
||||
# DirectShow可能支持获取设备名称
|
||||
try:
|
||||
# 某些情况下可以通过属性获取
|
||||
pass # OpenCV限制，无法直接获取设备名称
|
||||
except:
|
||||
pass
|
||||
elif backend == cv2.CAP_ANY:
|
||||
backend_name = "Any"
|
||||
else:
|
||||
backend_name = "Default"
|
||||
|
||||
# 获取分辨率信息作为辅助标识
|
||||
width = int(cap.get(cv2.CAP_PROP_FRAME_WIDTH))
|
||||
height = int(cap.get(cv2.CAP_PROP_FRAME_HEIGHT))
|
||||
|
||||
cap.release()
|
||||
|
||||
return {
|
||||
'index': cam_index,
|
||||
'backend': backend_name,
|
||||
'width': width,
|
||||
'height': height,
|
||||
'name': f"采集卡-{cam_index}",
|
||||
'available': True
|
||||
}
|
||||
else:
|
||||
if cap:
|
||||
cap.release()
|
||||
cap = None
|
||||
except Exception:
|
||||
if cap:
|
||||
try:
|
||||
cap.release()
|
||||
except:
|
||||
pass
|
||||
cap = None
|
||||
continue
|
||||
|
||||
return None
|
||||
finally:
|
||||
sys.stderr = old_stderr
|
||||
|
||||
def scan_cameras_with_info(max_index: int = 10) -> List[Dict]:
|
||||
"""
|
||||
扫描所有可用采集卡，返回详细信息列表
|
||||
:param max_index: 最大扫描索引
|
||||
:return: 采集卡信息列表
|
||||
"""
|
||||
cameras = []
|
||||
for i in range(max_index + 1):
|
||||
info = get_camera_info(i)
|
||||
if info:
|
||||
cameras.append(info)
|
||||
return cameras
|
||||
|
||||
def get_serial_port_info(port_name: str) -> Optional[Dict]:
|
||||
"""
|
||||
获取串口的详细信息（包括硬件ID和描述）
|
||||
:param port_name: 串口名称（如 "COM3"）
|
||||
:return: 包含串口信息的字典
|
||||
"""
|
||||
try:
|
||||
ports = serial.tools.list_ports.comports()
|
||||
for port in ports:
|
||||
if port.device == port_name:
|
||||
return {
|
||||
'device': port.device,
|
||||
'description': port.description,
|
||||
'hwid': port.hwid, # 硬件ID（唯一标识）
|
||||
'vid': port.vid if hasattr(port, 'vid') else None,
|
||||
'pid': port.pid if hasattr(port, 'pid') else None,
|
||||
'serial_number': port.serial_number if hasattr(port, 'serial_number') else None,
|
||||
'manufacturer': port.manufacturer if hasattr(port, 'manufacturer') else None,
|
||||
'name': port.description or port.device
|
||||
}
|
||||
except Exception as e:
|
||||
print(f"⚠️ 获取串口信息失败: {e}")
|
||||
|
||||
return None
|
||||
|
||||
def scan_serial_ports_with_info() -> List[Dict]:
|
||||
"""
|
||||
扫描所有可用串口，返回详细信息列表
|
||||
:return: 串口信息列表
|
||||
"""
|
||||
ports_info = []
|
||||
try:
|
||||
ports = serial.tools.list_ports.comports()
|
||||
for port in ports:
|
||||
info = {
|
||||
'device': port.device,
|
||||
'description': port.description,
|
||||
'hwid': port.hwid,
|
||||
'vid': port.vid if hasattr(port, 'vid') else None,
|
||||
'pid': port.pid if hasattr(port, 'pid') else None,
|
||||
'serial_number': port.serial_number if hasattr(port, 'serial_number') else None,
|
||||
'manufacturer': port.manufacturer if hasattr(port, 'manufacturer') else None,
|
||||
'name': port.description or port.device
|
||||
}
|
||||
ports_info.append(info)
|
||||
except Exception as e:
|
||||
print(f"⚠️ 扫描串口失败: {e}")
|
||||
|
||||
# 按设备名排序
|
||||
ports_info.sort(key=lambda x: int(x['device'].replace('COM', '')) if x['device'].replace('COM', '').isdigit() else 999)
|
||||
return ports_info
|
||||
|
||||
def find_camera_by_index(cameras: List[Dict], index: int) -> Optional[Dict]:
|
||||
"""通过索引查找采集卡"""
|
||||
for cam in cameras:
|
||||
if cam['index'] == index:
|
||||
return cam
|
||||
return None
|
||||
|
||||
def find_serial_by_hwid(ports: List[Dict], hwid: str) -> Optional[Dict]:
|
||||
"""通过硬件ID查找串口（最可靠）"""
|
||||
for port in ports:
|
||||
if port['hwid'] == hwid:
|
||||
return port
|
||||
return None
|
||||
|
||||
def find_serial_by_device(ports: List[Dict], device: str) -> Optional[Dict]:
|
||||
"""通过设备名查找串口（兼容旧配置）"""
|
||||
for port in ports:
|
||||
if port['device'] == device:
|
||||
return port
|
||||
return None
|
||||
|
||||
if __name__ == "__main__":
|
||||
# 测试代码
|
||||
print("=" * 60)
|
||||
print("采集卡扫描:")
|
||||
print("=" * 60)
|
||||
cameras = scan_cameras_with_info()
|
||||
for cam in cameras:
|
||||
print(f" 索引 {cam['index']}: {cam['name']} ({cam['width']}x{cam['height']})")
|
||||
|
||||
print("\n" + "=" * 60)
|
||||
print("串口扫描:")
|
||||
print("=" * 60)
|
||||
ports = scan_serial_ports_with_info()
|
||||
for port in ports:
|
||||
print(f" {port['device']}: {port['name']}")
|
||||
print(f" 硬件ID: {port['hwid']}")
|
||||
if port['vid'] and port['pid']:
|
||||
print(f" VID/PID: {port['vid']:04X}/{port['pid']:04X}")
|
||||
|
||||
@@ -10,18 +10,32 @@ mouse = None
|
||||
def init_mouse_keyboard(config_group):
|
||||
"""初始化鼠标和串口"""
|
||||
global serial, mouse
|
||||
from utils.logger import logger
|
||||
|
||||
# 初始化串口
|
||||
serial.ser = serial.Serial(
|
||||
config_group['serial_port'],
|
||||
config_group['serial_baudrate']
|
||||
)
|
||||
try:
|
||||
logger.info(f"🔧 正在打开串口: {config_group['serial_port']} @ {config_group['serial_baudrate']}")
|
||||
serial.ser = serial.Serial(
|
||||
config_group['serial_port'],
|
||||
config_group['serial_baudrate'],
|
||||
timeout=1
|
||||
)
|
||||
logger.info(f"✅ 串口已打开: {config_group['serial_port']} @ {config_group['serial_baudrate']}")
|
||||
except Exception as e:
|
||||
logger.error(f"❌ 串口打开失败: {e}")
|
||||
raise
|
||||
|
||||
# 初始化鼠标
|
||||
mouse = ch9329Comm.mouse.DataComm(
|
||||
config_group['camera_width'],
|
||||
config_group['camera_height']
|
||||
)
|
||||
print(f"✅ 串口已打开: {config_group['serial_port']} {config_group['serial_baudrate']}")
|
||||
print(f"✅ 鼠标已初始化: {config_group['camera_width']}x{config_group['camera_height']}")
|
||||
try:
|
||||
logger.info(f"🔧 正在初始化鼠标: {config_group['camera_width']}x{config_group['camera_height']}")
|
||||
mouse = ch9329Comm.mouse.DataComm(
|
||||
config_group['camera_width'],
|
||||
config_group['camera_height']
|
||||
)
|
||||
logger.info(f"✅ 鼠标已初始化: {config_group['camera_width']}x{config_group['camera_height']}")
|
||||
except Exception as e:
|
||||
logger.error(f"❌ 鼠标初始化失败: {e}")
|
||||
raise
|
||||
|
||||
def bezier_point(t, p0, p1, p2, p3):
|
||||
"""计算三次贝塞尔曲线上的点"""
|
||||
|
||||
Reference in New Issue
Block a user