#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 >= maxy) maxy = 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);
    //获取当前角度
    double angle;
    m_pCViewInterface->GetViewMatrix()->GetWaferTableAngle(angle);
    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;

    //test 实际使用时需要删除或注释
    angle = 0.00;
    // 初始化晶圆参数
    view->initWafer(center, // 中心坐标
        radius,              // 半径
        dieLong, dieWide,angle);          // 固晶点显示尺寸

// 添加固晶点
    int currentDieMatrixId = -1;
    double referPointX = 0.0;
    double referPointY = 0.0;
    MaxRow_Col maxRow_Col;
    double minx, miny,maxx,maxy, widthMatrix, heightMatrix;
    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);
            minx = center.x()+(maxRow_Col.minX - m_centerX) / radio - 20;
            miny = center.y() - (maxRow_Col.minY - m_centerY) / radio + 20;
            maxx = center.x() + (maxRow_Col.maxX - m_centerX) / radio + 20;
            maxy = center.y() - (maxRow_Col.maxY - m_centerY) / radio - 20;
            widthMatrix = maxx - minx;
            heightMatrix = miny - maxy;
            view->drawDieMatrix(QPointF(minx, maxy), widthMatrix, heightMatrix, currentDieMatrixId);

        }
        referPointX = center.x() + (waferData[i].stPosition.x - m_centerX) / radio;
        referPointY = center.y() - (waferData[i].stPosition.y - m_centerY) / radio;

        //test
        //referPointX = i * 5;
        //referPointY = i * 5;
        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);
   ImageInfo image;
   m_pCViewInterface->GetViewMatrix()->GetWaferRefImage(image);

   QPixmap test = convertToPixmap(image);

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

void Wafer::yuv422_to_rgb888(const unsigned char* src, unsigned char* dst, int width, int height)
{
    for (int i = 0; i < width * height; i += 2) {
        unsigned char y0 = src[0];
        unsigned char u = src[1];
        unsigned char y1 = src[2];
        unsigned char v = src[1];

        // 简单反色度插值，适用于 U/V 在相邻像素间共享的情况
        int r, g, b;

        // YUV to RGB 转换公式
#define CLIP(x) qBound(0, int(x), 255)

// Pixel 0
        r = CLIP(y0 + 1.402 * (v - 128));
        g = CLIP(y0 - 0.344 * (u - 128) - 0.714 * (v - 128));
        b = CLIP(y0 + 1.772 * (u - 128));
        *dst++ = r;
        *dst++ = g;
        *dst++ = b;

        // Pixel 1
        r = CLIP(y1 + 1.402 * (v - 128));
        g = CLIP(y1 - 0.344 * (u - 128) - 0.714 * (v - 128));
        b = CLIP(y1 + 1.772 * (u - 128));
        *dst++ = r;
        *dst++ = g;
        *dst++ = b;

        src += 4;
    }
}

QPixmap Wafer::convertToPixmap(const ImageInfo& imgData)
{
    QImage::Format qFormat = QImage::Format_Invalid;

    switch (imgData.format) {
    case ImageFormat::GRAY8:
        qFormat = QImage::Format_Grayscale8;
        break;
    case ImageFormat::RGB888:
        qFormat = QImage::Format_RGB888;
        break;
    case ImageFormat::ARGB32:
        qFormat = QImage::Format_ARGB32;
        break;
    case ImageFormat::RGB32:
        qFormat = QImage::Format_RGB32;
        break;
    case ImageFormat::YUV422: {
        // 需要先转换为 RGB888
        int byteCount = imgData.width * imgData.height * 3;
        unsigned char* rgbData = new unsigned char[byteCount];
        yuv422_to_rgb888(imgData.data, rgbData, imgData.width, imgData.height);

        QImage tmp(rgbData, imgData.width, imgData.height, QImage::Format_RGB888);
        QPixmap pixmap = QPixmap::fromImage(tmp);

        delete[] rgbData;
        return pixmap;
    }
    default:
        qDebug() << "Unsupported image format!";
        return QPixmap();
    }

    QImage qImg(imgData.data, imgData.width, imgData.height,
        imgData.width * imgData.channel, qFormat);

    return QPixmap::fromImage(qImg);
}