import sys
from PyQt5.QtWidgets import QApplication, QMainWindow, QFileDialog, QMessageBox, QProgressDialog
from PyQt5.QtGui import QImage, QPixmap
from excel_ui import Ui_MainWindow  # Qt Designer 生成的 UI 文件
from PyQt5.QtCore import Qt, QTimer
import cv2
import numpy as np
import openpyxl
from openpyxl.styles import PatternFill
from openpyxl.utils import get_column_letter

class MainWindow(QMainWindow, Ui_MainWindow):
    def __init__(self, parent=None):
        super().__init__(parent)
        self.setupUi(self)
        self.img_ori = None  # 存儲原始圖像

    def select_pic(self):
        """選擇圖片并預(yù)覽"""
        file_path, _ = QFileDialog.getOpenFileName(self, "選擇圖片", "", "Image Files(*.png *.jpg *.jpeg *.bmp)")
        if file_path:
            self.img_ori = cv2.imread(file_path)
            img = cv2.cvtColor(self.img_ori, cv2.COLOR_BGR2RGB)
            h, w, ch = img.shape
            bytes_per_line = ch * w
            q_image = QImage(img.data, w, h, bytes_per_line, QImage.Format.Format_RGB888)
            q_pixmap = QPixmap.fromImage(q_image)
            self.imageLabel.setPixmap(q_pixmap.scaled(
                self.imageLabel.width(), self.imageLabel.height(),
                Qt.AspectRatioMode.KeepAspectRatio, Qt.TransformationMode.SmoothTransformation
            ))

    def select_excel(self):
        """選擇 Excel 保存路徑"""
        file_path, _ = QFileDialog.getSaveFileName(
            self,
            "設(shè)置 Excel 路徑",
            "",
            "Excel Files (*.xlsx *.xls)"
        )
        if file_path:
            self.excelPathLineEdit.setText(file_path)
            return file_path
        else:
            return None

    def huizhi(self):
        """繪制圖片到 Excel"""
        if not hasattr(self, 'img_ori') or self.img_ori is None:
            QMessageBox.warning(self, "警告", "請先選擇圖片！")
            return
        excel_path = self.excelPathLineEdit.text()
        if not excel_path:
            QMessageBox.warning(self, "警告", "請先選擇 Excel 文件保存路徑！")
            return
        try:
            # 強(qiáng)制縮放寬度為 320，高度按比例計算
            min_width = 320
            image = self.img_ori
            height, width = image.shape[:2]
            scale_factor = min_width / width
            new_width = min_width
            new_height = int(height * scale_factor)
            image = cv2.resize(image, (new_width, new_height))

            # 創(chuàng)建 Excel 工作簿
            workbook = openpyxl.Workbook()
            sheet = workbook.active

            # 設(shè)置單元格為正方形（寬度 3，高度 17）
            for col in range(1, new_width + 1):
                sheet.column_dimensions[get_column_letter(col)].width = 3
            for row in range(1, new_height + 1):
                sheet.row_dimensions[row].height = 17

            # 創(chuàng)建進(jìn)度條
            progress = QProgressDialog("正在繪制圖片到 Excel...", "取消", 0, new_height, self)
            progress.setWindowTitle("進(jìn)度")
            progress.setWindowModality(Qt.WindowModal)
            progress.setValue(0)

            # 遍歷圖片像素并填充 Excel 單元格
            for y in range(new_height):
                if progress.wasCanceled():
                    break
                for x in range(new_width):
                    b, g, r = image[y, x]  # OpenCV 默認(rèn) BGR 格式
                    color = "{:02x}{:02x}{:02x}".format(r, g, b)  # 轉(zhuǎn)換為 RGB 格式的十六進(jìn)制
                    if color == "000000":
                        color = "010101"  # 純黑色替換為深灰色
                    fill = PatternFill(start_color=color, end_color=color, fill_type="solid")
                    sheet.cell(row=y + 1, column=x + 1).fill = fill
                progress.setValue(y + 1)
                QApplication.processEvents()  # 確保 UI 更新

            # 保存 Excel 文件
            if not progress.wasCanceled():
                workbook.save(excel_path)
                QMessageBox.information(self, "成功", "圖片已成功繪制到 Excel 文件！")
            progress.close()
        except Exception as e:
            QMessageBox.critical(self, "錯誤", f"繪制失敗: {str(e)}")

if __name__ == '__main__':
    app = QApplication(sys.argv)
    window = MainWindow()
    window.show()
    sys.exit(app.exec())