使用Python实现简单的Excel绘图工具

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):
        """选择图片并预览"""
        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,
            "设置 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:
            # 强制缩放宽度为 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))

            # 创建 Excel 工作簿
            workbook = openpyxl.Workbook()
            sheet = workbook.active

            # 设置单元格为正方形（宽度 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

            # 创建进度条
            progress = QProgressDialog("正在绘制图片到 Excel...", "取消", 0, new_height, self)
            progress.setWindowTitle("进度")
            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 默认 BGR 格式
                    color = "{:02x}{:02x}{:02x}".format(r, g, b)  # 转换为 RGB 格式的十六进制
                    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())