298 lines
10 KiB
Python
298 lines
10 KiB
Python
# 제품 등록 페이지 — 기준 이미지 캡처
|
|
import cv2
|
|
import numpy as np
|
|
from PyQt5.QtCore import Qt, QSize
|
|
from PyQt5.QtGui import QImage, QPixmap
|
|
from PyQt5.QtWidgets import (
|
|
QWidget, QHBoxLayout, QVBoxLayout, QGroupBox,
|
|
QPushButton, QListWidget, QListWidgetItem, QLabel,
|
|
QMessageBox, QScrollArea, QFrame,
|
|
)
|
|
|
|
|
|
_GRP_STYLE = (
|
|
"QGroupBox {"
|
|
" background:#222222; border:1px solid #333333; border-radius:6px;"
|
|
" margin-top:14px; padding:14px 12px 12px 12px;"
|
|
"}"
|
|
"QGroupBox::title { color:#aaaaaa; subcontrol-origin:margin; left:10px; padding:0 4px; }"
|
|
)
|
|
|
|
|
|
class RegisterPage(QWidget):
|
|
def __init__(self, insight_cam, matcher=None, db_client=None, parent=None):
|
|
super().__init__(parent)
|
|
self._insight = insight_cam
|
|
self._db_client = db_client
|
|
self._db_items = []
|
|
self._selected = None
|
|
self._captured_img = None
|
|
|
|
self._build_ui()
|
|
|
|
# ================================================================== #
|
|
# UI 구성
|
|
# ================================================================== #
|
|
|
|
def _build_ui(self):
|
|
root = QHBoxLayout(self)
|
|
root.setContentsMargins(0, 0, 0, 0)
|
|
root.setSpacing(0)
|
|
root.addWidget(self._build_left_panel(), stretch=2)
|
|
root.addWidget(self._build_right_panel(), stretch=3)
|
|
|
|
def _build_left_panel(self) -> QWidget:
|
|
w = QWidget()
|
|
w.setStyleSheet("background:#1a1a1a;")
|
|
layout = QVBoxLayout(w)
|
|
layout.setContentsMargins(16, 16, 8, 16)
|
|
layout.setSpacing(10)
|
|
|
|
self._btn_mes = QPushButton("MES 불러오기")
|
|
self._btn_mes.setEnabled(False)
|
|
self._btn_mes.setFixedHeight(56)
|
|
self._btn_mes.setToolTip("DB 연결 후 사용 가능")
|
|
self._btn_mes.setStyleSheet(
|
|
"background:#444444; color:#777777; border:none; border-radius:4px; font-size:15px;"
|
|
)
|
|
self._btn_mes.clicked.connect(self._on_load_from_db)
|
|
layout.addWidget(self._btn_mes)
|
|
|
|
lbl = QLabel("제품 목록")
|
|
lbl.setStyleSheet("font-size:13px; color:#777777;")
|
|
layout.addWidget(lbl)
|
|
|
|
self._list = QListWidget()
|
|
self._list.setStyleSheet("""
|
|
QListWidget {
|
|
background:#222222; border:1px solid #333333;
|
|
border-radius:4px; outline:none; font-size:15px;
|
|
}
|
|
QListWidget::item {
|
|
padding:0px 14px; border-bottom:1px solid #2a2a2a; color:#dddddd;
|
|
}
|
|
QListWidget::item:selected { background:#185FA5; color:#ffffff; }
|
|
QListWidget::item:hover:!selected { background:#2d2d2d; }
|
|
""")
|
|
self._list.currentRowChanged.connect(self._on_select)
|
|
layout.addWidget(self._list, stretch=1)
|
|
return w
|
|
|
|
def _build_right_panel(self) -> QScrollArea:
|
|
scroll = QScrollArea()
|
|
scroll.setWidgetResizable(True)
|
|
scroll.setFrameShape(QScrollArea.NoFrame)
|
|
scroll.setStyleSheet("background:#1a1a1a;")
|
|
|
|
inner = QWidget()
|
|
inner.setStyleSheet("background:#1a1a1a;")
|
|
layout = QVBoxLayout(inner)
|
|
layout.setContentsMargins(8, 16, 16, 16)
|
|
layout.setSpacing(0)
|
|
|
|
layout.addWidget(self._build_detail_section())
|
|
layout.addWidget(_divider())
|
|
layout.addWidget(self._build_capture_section())
|
|
layout.addStretch()
|
|
|
|
scroll.setWidget(inner)
|
|
return scroll
|
|
|
|
# ── 섹션 1: 제품 상세 ──────────────────────────────────────────────
|
|
|
|
def _build_detail_section(self) -> QGroupBox:
|
|
g = QGroupBox("제품 상세 정보")
|
|
g.setStyleSheet(_GRP_STYLE)
|
|
layout = QVBoxLayout(g)
|
|
layout.setSpacing(8)
|
|
|
|
self._lbl_name = _info_value("—")
|
|
self._lbl_model = _info_value("—")
|
|
self._lbl_type = _info_value("—")
|
|
layout.addLayout(_info_row("카테고리", self._lbl_name))
|
|
layout.addLayout(_info_row("모델명", self._lbl_model))
|
|
layout.addLayout(_info_row("Type", self._lbl_type))
|
|
|
|
self._arrow_lbl = QLabel("")
|
|
self._arrow_lbl.setAlignment(Qt.AlignCenter)
|
|
self._arrow_lbl.setFixedHeight(80)
|
|
self._arrow_lbl.setStyleSheet("font-size:60px; background:transparent;")
|
|
layout.addWidget(self._arrow_lbl)
|
|
return g
|
|
|
|
# ── 섹션 2: 캡처 ──────────────────────────────────────────────────
|
|
|
|
def _build_capture_section(self) -> QGroupBox:
|
|
g = QGroupBox("기준 이미지 캡처")
|
|
g.setStyleSheet(_GRP_STYLE)
|
|
layout = QVBoxLayout(g)
|
|
layout.setSpacing(12)
|
|
|
|
btn = QPushButton("캡처 (In-Sight 트리거)")
|
|
btn.setFixedHeight(56)
|
|
btn.setStyleSheet(
|
|
"background:#1a3a5c; color:#ffffff; border:none; border-radius:4px;"
|
|
"font-size:15px; font-weight:bold;"
|
|
)
|
|
btn.clicked.connect(self._on_capture)
|
|
|
|
self._preview = QLabel()
|
|
self._preview.setAlignment(Qt.AlignCenter)
|
|
self._preview.setFixedSize(400, 300)
|
|
self._preview.setStyleSheet(
|
|
"background:#111111; color:#555555; border:1px solid #333333;"
|
|
"border-radius:4px; font-size:14px;"
|
|
)
|
|
self._preview.setText("캡처 이미지 없음")
|
|
|
|
layout.addWidget(btn)
|
|
layout.addWidget(self._preview, alignment=Qt.AlignHCenter)
|
|
return g
|
|
|
|
# ================================================================== #
|
|
# 슬롯
|
|
# ================================================================== #
|
|
|
|
def _on_select(self, row: int):
|
|
if row < 0:
|
|
return
|
|
item = self._list.item(row)
|
|
if item is None:
|
|
return
|
|
|
|
article_id = item.data(Qt.UserRole)
|
|
if article_id is None:
|
|
return
|
|
|
|
db_item = next(
|
|
(x for x in self._db_items if x["article_id"] == article_id), None
|
|
)
|
|
r = {
|
|
"id": article_id,
|
|
"name": item.text(),
|
|
"model": db_item.get("buyer_article_no", "") if db_item else "",
|
|
"type": "",
|
|
}
|
|
|
|
self._selected = r
|
|
self._captured_img = None
|
|
|
|
self._lbl_name.setText(r["name"])
|
|
self._lbl_model.setText(r["model"])
|
|
t = r.get("type", "")
|
|
self._lbl_type.setText(t if t else "—")
|
|
|
|
if t == "RH":
|
|
self._arrow_lbl.setText("→")
|
|
self._arrow_lbl.setStyleSheet("font-size:60px; color:#4488ff; background:transparent;")
|
|
elif t == "LH":
|
|
self._arrow_lbl.setText("←")
|
|
self._arrow_lbl.setStyleSheet("font-size:60px; color:#ff8844; background:transparent;")
|
|
else:
|
|
self._arrow_lbl.setText("")
|
|
|
|
self._reset_preview()
|
|
|
|
def _on_capture(self):
|
|
if self._selected is None:
|
|
QMessageBox.warning(self, "경고", "먼저 제품을 선택하세요.")
|
|
return
|
|
if not self._insight.is_connected():
|
|
QMessageBox.critical(self, "오류", "Cognex 카메라가 연결되어 있지 않습니다.")
|
|
return
|
|
|
|
raw = self._insight.trigger_and_get_image()
|
|
if not raw:
|
|
QMessageBox.critical(self, "캡처 실패",
|
|
"이미지를 수신하지 못했습니다.\n카메라 연결 상태를 확인하세요.")
|
|
return
|
|
|
|
arr = np.frombuffer(raw, dtype=np.uint8)
|
|
img = cv2.imdecode(arr, cv2.IMREAD_UNCHANGED)
|
|
if img is None:
|
|
QMessageBox.critical(self, "캡처 실패", "이미지 디코딩에 실패했습니다.")
|
|
return
|
|
|
|
self._captured_img = img
|
|
self._show_ndarray(img)
|
|
|
|
def update_db(self, db_client):
|
|
"""MainWindow에서 DB 연결/해제 시 호출."""
|
|
self._db_client = db_client
|
|
enabled = db_client is not None and db_client.is_connected()
|
|
self._btn_mes.setEnabled(enabled)
|
|
self._btn_mes.setStyleSheet(
|
|
"background:#1a3a5c; color:#ffffff; border:none; border-radius:4px; font-size:15px;"
|
|
if enabled else
|
|
"background:#444444; color:#777777; border:none; border-radius:4px; font-size:15px;"
|
|
)
|
|
|
|
def _on_load_from_db(self):
|
|
if not self._db_client or not self._db_client.is_connected():
|
|
QMessageBox.warning(self, "경고", "DB를 먼저 연결해주세요.")
|
|
return
|
|
|
|
items = self._db_client.get_reflector_list()
|
|
if not items:
|
|
QMessageBox.warning(self, "경고", "조회된 제품이 없습니다.")
|
|
return
|
|
|
|
self._list.clear()
|
|
self._db_items = items
|
|
for item in items:
|
|
li = QListWidgetItem(item['article'])
|
|
li.setSizeHint(QSize(0, 52))
|
|
li.setData(Qt.UserRole, item["article_id"])
|
|
self._list.addItem(li)
|
|
|
|
QMessageBox.information(
|
|
self, "완료", f"{len(items)}개 제품을 불러왔습니다."
|
|
)
|
|
|
|
# ================================================================== #
|
|
# 헬퍼
|
|
# ================================================================== #
|
|
|
|
def _show_ndarray(self, img: np.ndarray):
|
|
h_img, w_img = img.shape[:2]
|
|
if img.ndim == 2:
|
|
qimg = QImage(img.data, w_img, h_img, w_img, QImage.Format_Grayscale8)
|
|
else:
|
|
rgb = np.ascontiguousarray(cv2.cvtColor(img, cv2.COLOR_BGR2RGB))
|
|
qimg = QImage(rgb.data, w_img, h_img, w_img * 3, QImage.Format_RGB888)
|
|
pix = QPixmap.fromImage(qimg).scaled(400, 300, Qt.KeepAspectRatio, Qt.SmoothTransformation)
|
|
self._preview.setPixmap(pix)
|
|
self._preview.setText("")
|
|
|
|
def _reset_preview(self):
|
|
self._preview.clear()
|
|
self._preview.setText("캡처 이미지 없음")
|
|
|
|
|
|
# ================================================================== #
|
|
# 모듈 수준 유틸리티
|
|
# ================================================================== #
|
|
|
|
def _divider() -> QFrame:
|
|
f = QFrame()
|
|
f.setFrameShape(QFrame.HLine)
|
|
f.setFixedHeight(1)
|
|
f.setStyleSheet("background:#333333; border:none;")
|
|
return f
|
|
|
|
|
|
def _info_value(text: str) -> QLabel:
|
|
lbl = QLabel(text)
|
|
lbl.setStyleSheet("color:#ffffff; font-size:16px; font-weight:bold;")
|
|
return lbl
|
|
|
|
|
|
def _info_row(label: str, value_widget: QLabel) -> QHBoxLayout:
|
|
row = QHBoxLayout()
|
|
key = QLabel(label + ":")
|
|
key.setFixedWidth(80)
|
|
key.setStyleSheet("color:#888888; font-size:14px;")
|
|
row.addWidget(key)
|
|
row.addWidget(value_widget, stretch=1)
|
|
return row
|