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)
{


    maxPoint = QPointF(veWaffle[0].stPosition.x, veWaffle[0].stPosition.y);
    minPoint = QPointF(veWaffle[0].stPosition.x, veWaffle[0].stPosition.y);
    for (const auto a: veWaffle)
    {
        m_vWaffleData.append(a);
        //遍历找到最小和最大边界点
        if (a.stPosition.x > maxPoint.x())
            maxPoint.setX(a.stPosition.x);
        if (a.stPosition.x < minPoint.x())
            minPoint.setX(a.stPosition.x);
        if (a.stPosition.y > maxPoint.y())
            maxPoint.setY(a.stPosition.y);
        if (a.stPosition.y < minPoint.y())
            minPoint.setY(a.stPosition.y);


    }

}

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);
        }
    }
}

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);
    view->setCViewInterface(m_pCViewInterface);

    int width = parent->width();
    int height = parent->height();
    double maxWidth = maxPoint.x() - minPoint.x();
    double maxHeight = maxPoint.y() - minPoint.y();
    double ratio;

    if (maxHeight > maxWidth) {
        ratio = maxHeight / (width - 100);
    }
    else {
        ratio = maxWidth / (width - 100);
    }

    //目前没提供固晶点大小，以下用测试大小，后续可替换(实际大小，接口替换处)
    double dieWidth = 20000;
    double dieHeight = 20000;
    for (int i = 0; i < m_vWaffleData.size(); ++i) {

        // 绘制固晶点

        //转换大小
        double dieWidthr = dieWidth / ratio;
        double dieHeightr = dieHeight / ratio;
        double diex, diey;
        diex = (m_vWaffleData[i].stPosition.x - minPoint.x() + dieWidth) / ratio;
        diey = (m_vWaffleData[i].stPosition.y - minPoint.y() + dieHeight) / ratio;


        // 计算固晶点位置
        QRectF dieRect(
            diex,
            diey,
            dieWidthr,
            dieHeightr
        );
        WaffleItem* dieItem = new WaffleItem(m_vWaffleData[i]);
        dieItem->setRect(dieRect);
        dieItem->setZValue(1.0);
        scene->addItem(dieItem);
        /*
        // 添加固晶点编号
        QGraphicsTextItem* dieText = new QGraphicsTextItem(dieItem);
        dieText->setPlainText("PT" + QString::number(m_vWaffleData[i].stIndex.iIndex));
        double dieFontSize = qMin(dieWidthr, dieHeightr) * 0.2;
        QFont font;
        font.setPointSizeF(dieFontSize);  // 使用浮点大小
        dieText->setFont(font);
        dieText->setPos(dieRect.center() - QPointF(dieText->boundingRect().width() / 2,
            dieText->boundingRect().height() / 2));
            */
    }
  
}
QPixmap Waffle::getGlobalPixmap() const {
    return globalPixmap;
}