298 lines
11 KiB
Python
298 lines
11 KiB
Python
import sys
|
||
import os
|
||
import logging
|
||
from ctypes import *
|
||
from camera.CameraParams_const import *
|
||
|
||
# 添加相机模块路径
|
||
sys.path.append(os.path.join(os.path.dirname(os.path.dirname(__file__)), "camera"))
|
||
|
||
# 导入相机相关模块
|
||
from camera.MvCameraControl_class import MvCamera
|
||
from camera.CamOperation_class import CameraOperation
|
||
from camera.MvErrorDefine_const import *
|
||
from camera.CameraParams_header import * # 导入MV_CC_DEVICE_INFO_LIST等类型
|
||
|
||
# 尝试从PySide6导入，如果失败则从PyQt5导入
|
||
try:
|
||
from PySide6.QtWidgets import QWidget, QVBoxLayout, QHBoxLayout, QLabel, QPushButton, QComboBox, QMessageBox, QFrame
|
||
from PySide6.QtCore import Signal
|
||
USE_PYSIDE6 = True
|
||
except ImportError:
|
||
from PyQt5.QtWidgets import QWidget, QVBoxLayout, QHBoxLayout, QLabel, QPushButton, QComboBox, QMessageBox, QFrame
|
||
from PyQt5.QtCore import pyqtSignal as Signal
|
||
USE_PYSIDE6 = False
|
||
|
||
|
||
class CameraWidget(QWidget):
|
||
"""海康威视工业相机控制组件"""
|
||
|
||
# 定义信号
|
||
camera_connected = Signal(bool)
|
||
camera_error = Signal(str)
|
||
|
||
def __init__(self, parent=None):
|
||
super().__init__(parent)
|
||
|
||
# 初始化相机变量
|
||
self.deviceList = None
|
||
self.cam = MvCamera()
|
||
self.nSelCamIndex = 0
|
||
self.obj_cam_operation = None
|
||
self.isOpen = False
|
||
self.isGrabbing = False
|
||
|
||
# 初始化UI
|
||
self.init_ui()
|
||
|
||
# 枚举相机设备
|
||
self.enum_devices()
|
||
|
||
def init_ui(self):
|
||
# 主布局
|
||
self.main_layout = QVBoxLayout(self)
|
||
self.main_layout.setContentsMargins(5, 5, 5, 5)
|
||
self.main_layout.setSpacing(5)
|
||
|
||
# 相机选择区域
|
||
self.device_layout = QHBoxLayout()
|
||
self.device_label = QLabel("相机:")
|
||
self.device_combo = QComboBox()
|
||
self.device_combo.setMinimumWidth(150)
|
||
self.refresh_button = QPushButton("刷新")
|
||
self.refresh_button.clicked.connect(self.enum_devices)
|
||
|
||
self.device_layout.addWidget(self.device_label)
|
||
self.device_layout.addWidget(self.device_combo)
|
||
self.device_layout.addWidget(self.refresh_button)
|
||
self.device_layout.addStretch()
|
||
|
||
# 相机控制区域
|
||
self.control_layout = QHBoxLayout()
|
||
self.open_button = QPushButton("打开相机")
|
||
self.open_button.clicked.connect(self.open_device)
|
||
self.close_button = QPushButton("关闭相机")
|
||
self.close_button.clicked.connect(self.close_device)
|
||
self.start_button = QPushButton("开始采集")
|
||
self.start_button.clicked.connect(self.start_grabbing)
|
||
self.stop_button = QPushButton("停止采集")
|
||
self.stop_button.clicked.connect(self.stop_grabbing)
|
||
|
||
self.control_layout.addWidget(self.open_button)
|
||
self.control_layout.addWidget(self.close_button)
|
||
self.control_layout.addWidget(self.start_button)
|
||
self.control_layout.addWidget(self.stop_button)
|
||
|
||
# 视频显示区域
|
||
self.display_frame = QFrame()
|
||
self.display_frame.setFrameShape(QFrame.Box)
|
||
self.display_frame.setLineWidth(1)
|
||
self.display_frame.setMinimumHeight(200)
|
||
self.display_frame.setStyleSheet("background-color: black;")
|
||
|
||
# 将所有组件添加到主布局
|
||
self.main_layout.addLayout(self.device_layout)
|
||
self.main_layout.addLayout(self.control_layout)
|
||
self.main_layout.addWidget(self.display_frame, 1) # 显示区域占据剩余空间
|
||
|
||
# 初始化按钮状态
|
||
self.update_controls()
|
||
|
||
def update_controls(self):
|
||
"""更新控件状态"""
|
||
has_devices = self.device_combo.count() > 0
|
||
|
||
self.open_button.setEnabled(has_devices and not self.isOpen)
|
||
self.close_button.setEnabled(self.isOpen)
|
||
self.start_button.setEnabled(self.isOpen and not self.isGrabbing)
|
||
self.stop_button.setEnabled(self.isOpen and self.isGrabbing)
|
||
self.device_combo.setEnabled(not self.isOpen)
|
||
self.refresh_button.setEnabled(not self.isOpen)
|
||
|
||
def enum_devices(self):
|
||
"""枚举相机设备"""
|
||
try:
|
||
# 初始化SDK
|
||
MvCamera.MV_CC_Initialize()
|
||
|
||
# 清空设备列表
|
||
self.device_combo.clear()
|
||
|
||
# 枚举设备
|
||
self.deviceList = MV_CC_DEVICE_INFO_LIST()
|
||
n_layer_type = (MV_GIGE_DEVICE | MV_USB_DEVICE | MV_GENTL_CAMERALINK_DEVICE
|
||
| MV_GENTL_CXP_DEVICE | MV_GENTL_XOF_DEVICE)
|
||
ret = MvCamera.MV_CC_EnumDevices(n_layer_type, self.deviceList)
|
||
if ret != 0:
|
||
error_msg = f"枚举设备失败! 错误码: 0x{ret:x}"
|
||
self.camera_error.emit(error_msg)
|
||
logging.error(error_msg)
|
||
return
|
||
|
||
if self.deviceList.nDeviceNum == 0:
|
||
logging.info("未找到相机设备")
|
||
return
|
||
|
||
logging.info(f"找到 {self.deviceList.nDeviceNum} 个相机设备")
|
||
|
||
# 将设备添加到下拉列表
|
||
for i in range(0, self.deviceList.nDeviceNum):
|
||
mvcc_dev_info = cast(self.deviceList.pDeviceInfo[i], POINTER(MV_CC_DEVICE_INFO)).contents
|
||
|
||
if mvcc_dev_info.nTLayerType == MV_GIGE_DEVICE:
|
||
# GigE相机
|
||
nip1 = ((mvcc_dev_info.SpecialInfo.stGigEInfo.nCurrentIp & 0xff000000) >> 24)
|
||
nip2 = ((mvcc_dev_info.SpecialInfo.stGigEInfo.nCurrentIp & 0x00ff0000) >> 16)
|
||
nip3 = ((mvcc_dev_info.SpecialInfo.stGigEInfo.nCurrentIp & 0x0000ff00) >> 8)
|
||
nip4 = (mvcc_dev_info.SpecialInfo.stGigEInfo.nCurrentIp & 0x000000ff)
|
||
ip = f"{nip1}.{nip2}.{nip3}.{nip4}"
|
||
|
||
device_info = f"[{i}]GigE: {ip}"
|
||
self.device_combo.addItem(device_info)
|
||
elif mvcc_dev_info.nTLayerType == MV_USB_DEVICE:
|
||
# USB相机
|
||
device_info = f"[{i}]USB"
|
||
self.device_combo.addItem(device_info)
|
||
else:
|
||
# 其他类型相机
|
||
device_info = f"[{i}]Other"
|
||
self.device_combo.addItem(device_info)
|
||
|
||
# 更新控件状态
|
||
self.update_controls()
|
||
except Exception as e:
|
||
error_msg = f"枚举设备时发生异常: {str(e)}"
|
||
self.camera_error.emit(error_msg)
|
||
logging.error(error_msg)
|
||
|
||
def get_selected_device_index(self):
|
||
"""获取选中的设备索引"""
|
||
if self.device_combo.count() == 0:
|
||
return -1
|
||
|
||
device_text = self.device_combo.currentText()
|
||
start_idx = device_text.find('[') + 1
|
||
end_idx = device_text.find(']')
|
||
if start_idx >= 0 and end_idx > start_idx:
|
||
return int(device_text[start_idx:end_idx])
|
||
return 0
|
||
|
||
def open_device(self):
|
||
"""打开相机设备"""
|
||
if self.isOpen:
|
||
QMessageBox.warning(self, "错误", "相机已经打开!")
|
||
return
|
||
|
||
try:
|
||
self.nSelCamIndex = self.get_selected_device_index()
|
||
if self.nSelCamIndex < 0:
|
||
QMessageBox.warning(self, "错误", "请选择一个相机!")
|
||
return
|
||
|
||
self.obj_cam_operation = CameraOperation(self.cam, self.deviceList, self.nSelCamIndex)
|
||
ret = self.obj_cam_operation.Open_device()
|
||
if ret != 0:
|
||
error_msg = f"打开相机失败! 错误码: 0x{ret:x}"
|
||
QMessageBox.warning(self, "错误", error_msg)
|
||
self.isOpen = False
|
||
logging.error(error_msg)
|
||
else:
|
||
# 设置连续模式
|
||
self.obj_cam_operation.Set_trigger_mode(False)
|
||
# 获取参数
|
||
self.obj_cam_operation.Get_parameter()
|
||
|
||
self.isOpen = True
|
||
self.camera_connected.emit(True)
|
||
logging.info("相机已打开")
|
||
|
||
self.update_controls()
|
||
except Exception as e:
|
||
error_msg = f"打开相机时发生异常: {str(e)}"
|
||
QMessageBox.warning(self, "错误", error_msg)
|
||
logging.error(error_msg)
|
||
self.isOpen = False
|
||
self.update_controls()
|
||
|
||
def close_device(self):
|
||
"""关闭相机设备"""
|
||
if not self.isOpen:
|
||
return
|
||
|
||
try:
|
||
if self.isGrabbing:
|
||
self.stop_grabbing()
|
||
|
||
ret = self.obj_cam_operation.Close_device()
|
||
if ret != 0:
|
||
error_msg = f"关闭相机失败! 错误码: 0x{ret:x}"
|
||
QMessageBox.warning(self, "错误", error_msg)
|
||
logging.error(error_msg)
|
||
else:
|
||
self.isOpen = False
|
||
self.camera_connected.emit(False)
|
||
logging.info("相机已关闭")
|
||
|
||
self.update_controls()
|
||
except Exception as e:
|
||
error_msg = f"关闭相机时发生异常: {str(e)}"
|
||
logging.error(error_msg)
|
||
self.isOpen = False
|
||
self.update_controls()
|
||
|
||
def start_grabbing(self):
|
||
"""开始取图"""
|
||
if not self.isOpen or self.isGrabbing:
|
||
return
|
||
|
||
try:
|
||
# 根据使用的框架不同，获取窗口句柄的方式不同
|
||
if USE_PYSIDE6:
|
||
window_id = int(self.display_frame.winId())
|
||
else:
|
||
window_id = self.display_frame.winId()
|
||
|
||
ret = self.obj_cam_operation.Start_grabbing(window_id)
|
||
if ret != 0:
|
||
error_msg = f"开始取图失败! 错误码: 0x{ret:x}"
|
||
QMessageBox.warning(self, "错误", error_msg)
|
||
logging.error(error_msg)
|
||
else:
|
||
self.isGrabbing = True
|
||
logging.info("开始图像采集")
|
||
|
||
self.update_controls()
|
||
except Exception as e:
|
||
error_msg = f"开始取图时发生异常: {str(e)}"
|
||
QMessageBox.warning(self, "错误", error_msg)
|
||
logging.error(error_msg)
|
||
self.isGrabbing = False
|
||
self.update_controls()
|
||
|
||
def stop_grabbing(self):
|
||
"""停止取图"""
|
||
if not self.isOpen or not self.isGrabbing:
|
||
return
|
||
|
||
try:
|
||
ret = self.obj_cam_operation.Stop_grabbing()
|
||
if ret != 0:
|
||
error_msg = f"停止取图失败! 错误码: 0x{ret:x}"
|
||
QMessageBox.warning(self, "错误", error_msg)
|
||
logging.error(error_msg)
|
||
else:
|
||
self.isGrabbing = False
|
||
logging.info("停止图像采集")
|
||
|
||
self.update_controls()
|
||
except Exception as e:
|
||
error_msg = f"停止取图时发生异常: {str(e)}"
|
||
logging.error(error_msg)
|
||
self.isGrabbing = False
|
||
self.update_controls()
|
||
|
||
def closeEvent(self, event):
|
||
"""窗口关闭事件"""
|
||
if self.isOpen:
|
||
self.close_device()
|
||
event.accept() |