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

#include "Wafer.h"
#include <QPainter>
#include <QPaintEvent>
#include <cmath>
#include <QDebug>
#include <QPushButton>
#include <QGridLayout>

Wafer::Wafer(int flag, QWidget *parent) : QWidget(parent) {
   
}

void Wafer::UpdataGenerateTestData()
{
    /*
    Flag = 0;
    rows = 51;
    cols = 51;
    centerX = 25;
    centerY = 25;
    radius = 25;

    // 随机数初始化
    std::srand(std::time(nullptr)); // 使用当前时间作为种子

    // 遍历矩阵，按行列判断是否在圆形区域内
    for (int row = 0; row < rows; ++row) {
        for (int col = 0; col < cols; ++col) {
            // 计算当前点到中心点的距离
            double dx = col - centerX;
            double dy = row - centerY;
            double distance = std::sqrt(dx * dx + dy * dy);

            // 如果点在圆形区域外部，跳过
            if (distance > radius) {
                continue;
            }

            // 创建一个新的WAFER_MATRIX_POINT_INFO_STRUCT
            ns_mat::WAFER_MATRIX_POINT_INFO_STRUCT point;
            point.nDieMatrixId = 1; // 假设所有点属于同一晶圆矩阵
            point.nDieRow = row;
            point.nDieCol = col;
            point.iDieIndex = row * cols + col; // 假设ID为行列号的线性映射
            point.bDisable = false; // 默认为可用
            point.stPosition.x = col * 10.0;
            point.stPosition.y = row * 10.0;

            // 判断该点是否在圆形区域的边缘
            if (distance > radius - 1 && distance < radius + 1) {
                point.eStatus = ns_mat::PICK_DIE_STATUS::EDGE_DIE; // 如果在边缘区域，状态设置为EDGE_DIE
            }
            else {
                // 如果在圆形区域内，随机设置其他状态
                int randomStatus = std::rand() % 4; // 随机选取DIE_EXIST, PICK_ING, NO_EXIST, SKIP_DIE
                point.eStatus = static_cast<ns_mat::PICK_DIE_STATUS>(randomStatus);
            }

            // 将点添加到waferData容器中
            waferData.append(point);
        }
    }
    */
}

void Wafer::UpdataVal(const std::vector<ns_mat::WAFER_MATRIX_POINT_INFO_STRUCT>& veWafer)
{
    int currentDieMatrixId = veWafer[0].nDieMatrixId;
    int rowMax = veWafer[0].nDieRow;
    int colsMax = veWafer[0].nDieCol;
    double minx = veWafer[0].stPosition.x;
    double miny = veWafer[0].stPosition.y;
    double maxx = veWafer[0].stPosition.x;
    double maxy = veWafer[0].stPosition.y;
    for (const auto a : veWafer)
    {
        waferData.append(a);
        qDebug() << a.stPosition.x << "     " << a.stPosition.y;
        if (a.nDieMatrixId != currentDieMatrixId) {
            maxRow_Colmap.insert(currentDieMatrixId, { rowMax ,colsMax,minx,miny,maxx,maxy });
            currentDieMatrixId = a.nDieMatrixId;
            rowMax = a.nDieRow;
            colsMax = a.nDieCol;
            minx = a.stPosition.x;
            maxx = a.stPosition.x;
            miny = a.stPosition.y;
            maxy = a.stPosition.y;
        }
        else {
            if (a.nDieRow >= rowMax) rowMax = a.nDieRow;
            if (a.nDieCol >= colsMax) colsMax = a.nDieCol;
            if (a.stPosition.x >= maxx) maxx = a.stPosition.x;
            if (a.stPosition.y <= maxx) maxx = a.stPosition.y;
            if (a.stPosition.x >= minx) minx = a.stPosition.x;
            if (a.stPosition.x <= miny) miny = a.stPosition.y;
        }
    }
    maxRow_Colmap.insert(currentDieMatrixId, { rowMax ,colsMax,minx,miny,maxx,maxy });


}

QColor Wafer::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 Wafer::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 < waferData.size(); ++i) {
        int x = offsetX + waferData[i].nDieCol * cellSize;
        int y = offsetY + waferData[i].nDieRow * cellSize;

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

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

    // 根据角度绘制黑点
    int centerX, centerY;
    int angle = 0;
    switch (angle) {
    case 0: {
        centerX = offsetX + (cols / 2) * cellSize;
        centerY = offsetY + (rows - 1) * cellSize;
        break;
    }
    case 90: {
        centerX = offsetX;
        centerY = offsetY + (rows / 2) * cellSize;
        break;
    }
    case 180: {
        centerX = offsetX + (cols / 2) * cellSize;
        centerY = offsetY;
        break;
    }
    case 270: {
        centerX = offsetX + (cols - 1) * cellSize;
        centerY = offsetY + (rows / 2) * cellSize;
        break;
    }
    default:
        return;
    }

    painter.setBrush(Qt::black);
    painter.drawEllipse(centerX - 3, centerY - 3, 6, 6);
    painter.end(); // 结束绘制
    */
}

void Wafer::initFrom(QWidget* parent) {
    //scene = new QGraphicsScene(parent);
    //view = new WaferGraphicsView(scene);

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

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

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

    //view->setSceneRect(scene->itemsBoundingRect());
    //view->resize(width, height);

    // 创建视图
    scene = new QGraphicsScene(parent);
    view = new WaferGraphicsView(scene);
    view->resize(width, height);
    QPointF center(width / 2.0, height / 2.0);
    QPointF pointf(m_centerX, m_centerY);
    double radius = width / 2 - 10;
    //比例
    double radio = m_radius / radius;
    int dieLong = m_dieLong / radio;
    int dieWide = m_dieWide / radio;
    // 初始化晶圆参数
    view->initWafer(center, // 中心坐标
        radius,              // 半径
        dieLong, dieWide);          // 固晶点显示尺寸

// 添加固晶点
    int currentDieMatrixId = -1;
    double referPointX = 0.0;
    double referPointY = 0.0;
    MaxRow_Col maxRow_Col;
    for (int i = 0; i < waferData.size(); ++i) {
        //判断是否属于同一个矩阵
        if (waferData[i].nDieMatrixId != currentDieMatrixId) {
            //重新画矩阵框然后配置新的参考点
            currentDieMatrixId = waferData[i].nDieMatrixId;
            referPointX = center.x() + (waferData[i].stPosition.x - m_centerX) / radio;
            referPointY = center.y() - (waferData[i].stPosition.y - m_centerY) / radio;
            maxRow_Col = maxRow_Colmap.value(currentDieMatrixId);
            view->drawDieMatrix(QPointF(referPointX - dieLong, referPointY - dieWide), (maxRow_Col.maxRow+2) * dieLong, (maxRow_Col.maxCol+2) * dieWide);

        }
        referPointX = center.x() + (waferData[i].stPosition.x - m_centerX) / radio;
        referPointY = center.y() - (waferData[i].stPosition.y - m_centerY) / radio;
        view->addDiePoint(QPointF(referPointX, referPointY),waferData[i]);
    }
    scene->setSceneRect(0, 0, view->width(), view->height());
    
    
}

QPixmap Wafer::getGlobalPixmap() const {
    return globalPixmap;
}

void Wafer::setWaferInfo(ns_module::CViewInterface* CViewInterface) {
    m_pCViewInterface = CViewInterface;
    long result = m_pCViewInterface->GetViewMatrix()->GetDieSize(m_dieLong,m_dieWide);

   m_pCViewInterface->GetViewMatrix()->GetWaferSize(m_centerX, m_centerY,m_radius);

    //这里到时候更新接口时替换真实圆心点的数据
    /*m_dieLong = 10;
    m_dieWide = 10;
    m_radius = 200;*/
    m_centerX = 0;
    m_centerY = 0;
}