最近有點忙,上來寫的時間不多。
今天就把之前寫的一個選型的簡易程序,供大家參考。
代碼:
import sys
from PyQt5.QtWidgets import (QApplication, QMainWindow, QWidget, QVBoxLayout, QHBoxLayout,QLabel, QLineEdit, QPushButton, QGroupBox, QFormLayout,QComboBox, QFileDialog, QMessageBox)
from PyQt5.QtCore import Qtclass CameraLensCalculator(QMainWindow):def __init__(self):super().__init__()self.setWindowTitle("相機與鏡頭選型工具")self.setFixedSize(600, 650)# 主窗口布局central_widget = QWidget()self.setCentralWidget(central_widget)main_layout = QVBoxLayout(central_widget)# 輸入參數區域self.setup_input_group()main_layout.addWidget(self.input_group)# 計算按鈕self.calculate_btn = QPushButton("計算", clicked=self.calculate)main_layout.addWidget(self.calculate_btn, alignment=Qt.AlignCenter)# 結果顯示區域self.setup_result_group()main_layout.addWidget(self.result_group)# 保存按鈕self.save_btn = QPushButton("保存報告", clicked=self.save_report)self.save_btn.setEnabled(False)main_layout.addWidget(self.save_btn, alignment=Qt.AlignCenter)# 設置默認值self.set_default_values()def setup_input_group(self):"""初始化輸入參數區域"""self.input_group = QGroupBox("輸入參數")layout = QFormLayout()# 物體尺寸self.obj_width = QLineEdit()self.obj_height = QLineEdit()layout.addRow(QLabel("物體寬度 (mm):"), self.obj_width)layout.addRow(QLabel("物體高度 (mm):"), self.obj_height)# 邊緣余量self.margin = QLineEdit()layout.addRow(QLabel("邊緣余量 (%):"), self.margin)# 檢測精度self.accuracy = QLineEdit()layout.addRow(QLabel("檢測精度 (mm):"), self.accuracy)# 工作距離self.working_distance = QLineEdit()layout.addRow(QLabel("工作距離 (mm):"), self.working_distance)# 傳感器類型(下拉選擇)self.sensor_combo = QComboBox()self.sensor_combo.addItems(["1/2\" (6.4x4.8mm)","1/1.8\" (7.2x5.3mm)","2/3\" (8.8x6.6mm)","自定義"])self.sensor_combo.currentTextChanged.connect(self.handle_sensor_change)layout.addRow(QLabel("傳感器類型:"), self.sensor_combo)# 自定義傳感器尺寸(默認隱藏)self.sensor_custom = QLineEdit()self.sensor_custom.setPlaceholderText("輸入寬x高(如 10.0x8.0)")self.sensor_custom.setVisible(False)layout.addRow(QLabel("自定義尺寸:"), self.sensor_custom)# 光圈值(景深計算用)self.aperture = QLineEdit()self.aperture.setPlaceholderText("F值(如 2.8)")layout.addRow(QLabel("光圈 (F):"), self.aperture)self.input_group.setLayout(layout)def setup_result_group(self):"""初始化結果顯示區域"""self.result_group = QGroupBox("計算結果")self.result_label = QLabel("請填寫參數后點擊計算")self.result_label.setAlignment(Qt.AlignLeft)self.result_label.setStyleSheet("font-family: monospace;")layout = QVBoxLayout()layout.addWidget(self.result_label)self.result_group.setLayout(layout)def handle_sensor_change(self, text):"""切換傳感器類型時顯示/隱藏自定義輸入框"""self.sensor_custom.setVisible(text == "自定義")def set_default_values(self):"""設置默認參數值"""self.obj_width.setText("100")self.obj_height.setText("50")self.margin.setText("20")self.accuracy.setText("0.1")self.working_distance.setText("400")self.aperture.setText("4.0")def calculate(self):"""執行計算邏輯"""try:# 獲取輸入值obj_w = float(self.obj_width.text())obj_h = float(self.obj_height.text())margin = float(self.margin.text()) / 100 # 轉為小數accuracy = float(self.accuracy.text())wd = float(self.working_distance.text())aperture = float(self.aperture.text()) if self.aperture.text() else 4.0# 解析傳感器尺寸sensor_text = self.sensor_combo.currentText()if sensor_text == "自定義":sensor_w, sensor_h = map(float, self.sensor_custom.text().split('x'))else:sensor_str = sensor_text.split('(')[1].split(')')[0].replace('mm', '')sensor_w, sensor_h = map(float, sensor_str.split('x'))# 計算FOV(含余量)fov_w = obj_w * (1 + margin)fov_h = obj_h * (1 + margin)# 計算分辨率(安全系數1.5)res_w = int((fov_w / accuracy) * 1.5)res_h = int((fov_h / accuracy) * 1.5)mp = (res_w * res_h) / 1e4 # 萬像素# 計算焦距(公式:f = (WD * 傳感器尺寸) / FOV)focal_w = (wd * sensor_w) / fov_wfocal_h = (wd * sensor_h) / fov_hfocal_length = max(focal_w, focal_h)# 計算景深(簡化公式)coc = 0.005 # 容許模糊圓直徑(默認0.005mm)depth_of_field = (2 * coc * aperture * (wd ** 2)) / (focal_length ** 2)# 顯示結果result_text = f"""=== 計算結果 ===視場(FOV): {fov_w:.2f} mm × {fov_h:.2f} mm所需分辨率: {res_w} × {res_h} 像素推薦相機: ≥ {mp:.1f} 萬像素鏡頭焦距: ≈ {focal_length:.1f} mm景深(DoF): ≈ {depth_of_field:.2f} mm(基于傳感器: {sensor_w}x{sensor_h} mm, 光圈 F{aperture})"""self.result_label.setText(result_text.strip())self.save_btn.setEnabled(True)self.calculation_result = result_text # 保存結果用于導出except Exception as e:QMessageBox.warning(self, "錯誤", f"參數輸入無效!\n{str(e)}")def save_report(self):"""保存計算結果到文件"""try:file_path, _ = QFileDialog.getSaveFileName(self, "保存報告", "", "Text Files (*.txt);;All Files (*)")if file_path:with open(file_path, 'w') as f:f.write("=== 相機與鏡頭選型報告 ===\n")f.write(f"物體尺寸: {self.obj_width.text()}x{self.obj_height.text()} mm\n")f.write(f"邊緣余量: {self.margin.text()}%\n")f.write(f"檢測精度: {self.accuracy.text()} mm\n")f.write(f"工作距離: {self.working_distance.text()} mm\n")f.write(f"傳感器: {self.sensor_combo.currentText()}\n")f.write(f"光圈: F{self.aperture.text()}\n")f.write("\n" + self.calculation_result.strip())QMessageBox.information(self, "成功", "報告已保存!")except Exception as e:QMessageBox.critical(self, "錯誤", f"保存失敗!\n{str(e)}")if __name__ == "__main__":app = QApplication(sys.argv)window = CameraLensCalculator()window.show()sys.exit(app.exec_())
以上對選型提供初步的想法,具體可以根據個人需求進一步優化。