JSBTDieBondH/View/die-bonder-ui/CameraMaterialGroupWnd/MaterialWindow/Waffle.cpp

#include "Waffle.h"
#include <QPainter>
#include <QPaintEvent>

Waffle::Waffle(int flag, QWidget *parent) : QWidget(parent) 
{ 

}

void Waffle::UpdataGenerateTestData()
{
    // //根据不同窗口设置不同大小 0：小 1：中 2：大
   // if (flag == 0) {
   //     // 华夫台的参数
   //     gridSize = 5;  // 每个格子的大小
   //     gap = 3;        // 格子之间的间隙
   //     boxGap = 6;    // 华夫台之间的间隙
   // } else if (flag == 1) {
   //     // 华夫台的参数
   //     gridSize = 20;  // 每个格子的大小
   //     gap = 6;        // 格子之间的间隙
   //     boxGap = 12;    // 华夫台之间的间隙
   // } else if (flag == 2) {
   //     // 华夫台的参数
   //     gridSize = 50;  // 每个格子的大小
   //     gap = 6;        // 格子之间的间隙
   //     boxGap = 12;    // 华夫台之间的间隙
   // }
    rows = 6;
    cols = 18;
    centerX = 25;
    centerY = 25;
    // 随机数初始化
    std::srand(std::time(nullptr)); // 使用当前时间作为种子
    for (int row = 0; row < rows; ++row) {
        for (int col = 0; col < cols; ++col) {
            ns_mat::WAFFLE_MATRIX_POINT_STRUCT point;
            point.nPackMatrixId = 1;
            point.nPackRow = 6;
            point.nPackCol = 18;
            point.nDieRow = row;
            point.nDieCol = col;
            point.nDieMatrixId = 1;
            point.iDieIndex = 1;
            point.bDisable = false; // 默认为可用
            point.stPosition.x = col * 10.0;
            point.stPosition.y = row * 10.0;
            int randomStatus = std::rand() % 5; // 随机选取DIE_EXIST, PICK_ING, NO_EXIST, SKIP_DIE
            point.eStatus = static_cast<ns_mat::PICK_DIE_STATUS>(randomStatus);
            // 将点添加到waferData容器中
            m_vWaffleData.append(point);
        }
    }
}

void Waffle::UpdataVal(const std::vector<ns_mat::WAFFLE_MATRIX_POINT_STRUCT>& veWaffle)
{
    maxRow = 0;
    maxCol = 0;
    waffleMatrixRowNum = 0;
    waffleMatrixColNum = 0;
    matrixInfo mInfo;
    mInfo.maxCol = 0;
    mInfo.maxRow = 0;
    UINT dieMatrixId = veWaffle[0].nDieMatrixId;
    for (const auto a: veWaffle)
    {
        if (dieMatrixId != a.nDieMatrixId) {
            matrixInfoMap.insert(dieMatrixId, mInfo);
            dieMatrixId = a.nPackMatrixId;
            mInfo.maxCol = 0;
            mInfo.maxRow = 0;
        }
        else {
            if (mInfo.maxRow < a.nDieRow)
                mInfo.maxRow = a.nDieRow;
            if (mInfo.maxCol < a.nDieCol)
                mInfo.maxCol = a.nDieCol;

        }
        m_vWaffleData.append(a);
        if (a.nPackRow == 1 && a.nDieRow==1) {
            waffleMatrixRowNum++;
        }
        if (a.nPackCol == 1 && a.nDieCol == 1) {
            waffleMatrixColNum++;
        }
        if (maxRow < a.nPackRow)
            maxRow = a.nPackRow;
        if (maxCol < a.nPackCol)
            maxCol = a.nPackCol;
    }
    matrixInfoMap.insert(dieMatrixId, mInfo);
    qDebug() << maxRow << maxCol << waffleMatrixRowNum << waffleMatrixColNum;
}

void Waffle::paintEvent(QPaintEvent *event) {
    QPainter painter(this);

    // 设置抗锯齿，使绘图更平滑
    painter.setRenderHint(QPainter::Antialiasing);

    // 绘制背景
    painter.fillRect(this->rect(), Qt::white);


    // 计算每个华夫台的总宽度和高度
    int totalWidth = cols * gridSize + (cols - 1) * gap;
    int totalHeight = rows * gridSize + (rows - 1) * gap;

    // 计算华夫盒的总宽度和高度
    int boxTotalWidth = boxCols * totalWidth + (boxCols - 1) * boxGap;
    int boxTotalHeight = boxRows * totalHeight + (boxRows - 1) * boxGap;

    // 确定华夫盒的起始位置，使其居中
    int boxStartX = (width() - boxTotalWidth) / 2;
    int boxStartY = (height() - boxTotalHeight) / 2;

    // 绘制多个华夫台
    for (int i = 0; i < boxRows; ++i) {
        for (int j = 0; j < boxCols; ++j) {
            // 计算每个华夫台的起始位置
            int startX = boxStartX + j * (totalWidth + boxGap);
            int startY = boxStartY + i * (totalHeight + boxGap);

            // 绘制单个华夫台
            drawWaffle(painter, startX, startY, rows, cols, gridSize, gap);
        }
    }
}

void Waffle::drawWaffle(QPainter &painter, int startX, int startY, int rows, int cols, int gridSize, int gap) {
    // 绘制网格
    for (int i = 0; i < rows; ++i) {
        for (int j = 0; j < cols; ++j) {
            // 计算每个格子的坐标
            int x = startX + j * (gridSize + gap);
            int y = startY + i * (gridSize + gap);
            QRect cellRect(x, y, gridSize, gridSize);

            // 填充颜色
            painter.fillRect(cellRect, Qt::yellow);

            // 绘制边框
            painter.setPen(Qt::black);
            painter.drawRect(cellRect);
        }
    }
}
QColor Waffle::getColorByStatus(ns_mat::PICK_DIE_STATUS status) {
    switch (status) {
    case ns_mat::PICK_DIE_STATUS::DIE_EXIST: return QColor(0, 102, 255); // 蓝色
    case ns_mat::PICK_DIE_STATUS::NO_EXIST: return QColor(200, 200, 200); // 浅灰
    case ns_mat::PICK_DIE_STATUS::PICK_ING: return QColor(255, 255, 0); // 黄色
    case ns_mat::PICK_DIE_STATUS::SKIP_DIE: return QColor(128, 128, 128); // 深灰
    case ns_mat::PICK_DIE_STATUS::EDGE_DIE: return QColor(255, 165, 0); // 橙色
    default: return QColor(0, 0, 0); // 默认黑色
    }
}
void Waffle::paintInitFrom(QWidget *parent) {

    // 获取当前窗口的宽高
    int width = parent->width();
    int height = parent->height();

    // 根据行列数计算每个晶圆点的大小，选择小的边来决定
    int cellSize = qMin(width, height) / qMax(rows, cols); // 固定大小为正方形，按最小边计算

    // 计算左上角偏移量，居中显示
    int offsetX = (width - cellSize * cols) / 2;
    int offsetY = (height - cellSize * rows) / 2;

    // 创建一个 QPixmap 对象用于保存绘制的图像
    globalPixmap = QPixmap(width, height);
    globalPixmap.fill(Qt::white);  // 填充背景色为白色

    // 创建 QPainter 以绘制到 QPixmap 上
    QPainter painter(&globalPixmap);
    painter.setRenderHint(QPainter::Antialiasing); // 启用抗锯齿

    // 设置画笔为无边框
    painter.setPen(Qt::NoPen);

    // 绘制每个晶圆点
    for (int i = 0; i < m_vWaffleData.size(); ++i) {
        int x = offsetX + m_vWaffleData[i].nDieCol * cellSize;
        int y = offsetY + m_vWaffleData[i].nDieRow * cellSize;

        // 根据点的状态设置颜色
        painter.setBrush(getColorByStatus(m_vWaffleData[i].eStatus));

        // 绘制晶圆点（每个点是一个矩形）
        painter.drawRect(x, y, cellSize, cellSize);
    }


    // painter.setBrush(Qt::black);
    // painter.drawEllipse(centerX - 3, centerY - 3, 6, 6);
    painter.end(); // 结束绘制
}
void Waffle::initFrom(QWidget *parent) {
    scene = new QGraphicsScene(parent);
    view = new WaffleGraphicsView(scene);
    /*
    // 获取当前窗口的宽高
    int width = parent->width();
    int height = parent->height();

    // 根据行列数计算每个晶圆点的大小，选择小的边来决定
    int cellSize = qMin(width, height) / qMax(rows, cols); // 固定大小为正方形，按最小边计算

    for (int i = 0; i < m_vWaffleData.size(); ++i) {
        DieItem* die = new DieItem(m_vWaffleData[i].nDieRow, m_vWaffleData[i].nDieCol, m_vWaffleData[i].eStatus, cellSize);
        die->setPos(m_vWaffleData[i].nDieCol * cellSize, m_vWaffleData[i].nDieRow * cellSize);
        scene->addItem(die);
    }

    view->setSceneRect(scene->itemsBoundingRect());
    view->resize(width, height);
    */
}
QPixmap Waffle::getGlobalPixmap() const {
    return globalPixmap;
}