ximeaAirborneSystem/Source_Files/rgbImage.cpp

//
// Created by tangchao on 2022/12/24.
//

#include "../Header_Files/rgbImage.h"

rgbImage::rgbImage(QWidget* pParent)
{
    m_QRgbImage = nullptr;
    m_matRgbImage = nullptr;
    m_matFocusGrayImage = nullptr;
    m_qimageFocusGrayImage = nullptr;

    m_iRedBandNumber = 0;
    m_iGreenBandNumber = 0;
    m_iBlueBandNumber = 0;
}

rgbImage::~rgbImage()
{

}

void rgbImage::SetRgbBandNumber(int redBandNumber, int greenBandNumber, int blueBandNumber)
{
    m_iRedBandNumber = redBandNumber;
    m_iGreenBandNumber = greenBandNumber;
    m_iBlueBandNumber = blueBandNumber;

//    std::cout<<"rgbImage::SetRgbBandNumber，红波段的波段号："<< redBandNumber <<std::endl;
//    std::cout<<"rgbImage::SetRgbBandNumber，绿波段的波段号："<< greenBandNumber <<std::endl;
//    std::cout<<"rgbImage::SetRgbBandNumber，蓝波段的波段号："<< blueBandNumber <<std::endl;
}

void rgbImage::SetRgbImageWidthAndHeight(int BandCount, int Width, int height)
{
    using namespace cv;

    if (m_QRgbImage != nullptr)
    {
        delete m_QRgbImage;//有问题？？？？？？？？？？？？？？？？？？？？？？？？？？？？？？？？？？？？？？？？？？？？？？？？
    }
    //m_QRgbImage = new QImage(Width, height, QImage::Format_RGB888);


    if (m_matRgbImage != nullptr)
    {
        delete m_matRgbImage;
    }
    m_matRgbImage = new Mat(height, Width, CV_8UC3, Scalar(0, 0, 0));

//    int codec = VideoWriter::fourcc('H', '2', '6', '4');  // select desired codec (must be available at runtime)
//    double fps = 20.0;// framerate of the created video stream
//    std::string filename = "appsrc ! autovideoconvert ! filesink location=/media/nvme/live.mp4";//https://blog.csdn.net/ancientapesman/article/details/117324638
////    std::string filename = "/media/nvme/live.mp4";
//    auto ddddd=m_matRgbImage->size();
//    m_VideoWriter.open(filename, codec, fps, Size(20, 1368), true);

//    VideoWriter video("test.avi", CV_FOURCC('M', 'J', 'P', 'G'), 25.0, Size(640, 480));

//    m_frame_writer = cvCreateVideoWriter("/media/nvme/delete/live.avi",  cv::VideoWriter::fourcc('M', 'J', 'P', 'G'), 20.0, Size(688, 688), false);





    if (m_qimageFocusGrayImage == nullptr)
    {
        m_qimageFocusGrayImage = new QImage(Width, BandCount, QImage::Format_RGB32);
    }


    if (m_matFocusGrayImage == nullptr)
    {
        m_matFocusGrayImage = new Mat(BandCount, Width, CV_16U, Scalar(0));

        //cv::Mat matAdjustPreview = Mat::zeros(BandCount, Width, CV_16U);
    }

    //cv::Mat matAdjustPreview = Mat::zeros(BandCount, Width, CV_16U);
    //m_matFocusGrayImage = matAdjustPreview;

    std::cout << "高光谱rgb图像设置高度：" << height << std::endl;

    m_iFrameCounter = 0;//每次都重置为0
    m_iSampleNumber = Width;
    m_iBandNumber = BandCount;
    m_iFrameNumber = height;

    //std::cout << "rgb影像内存地址为：" << m_QRgbImage << std::endl;
}

void rgbImage::FillOnerowofRgbImage(cv::Mat * matRgbImage, int rowNumber, unsigned short *datacube)
{
    //方式1：逐像素修改
//    unsigned short r, g, b;
//    for (int j = 0; j < m_iSampleNumber; j++)
//    {
//        //取值：一帧影像中，从左到右的rgb像元值
//        r = *(datacube + m_iRedBandNumber * m_iSampleNumber + j)*255/4096;
//        g = *(datacube + m_iGreenBandNumber * m_iSampleNumber + j)*255/4096;
//        b = *(datacube + m_iBlueBandNumber * m_iSampleNumber + j)*255/4096;
//
////        r = *(datacube + m_iRedBandNumber * m_iSampleNumber + j);
////        g = *(datacube + m_iGreenBandNumber * m_iSampleNumber + j);
////        b = *(datacube + m_iBlueBandNumber * m_iSampleNumber + j);
//
//        //将像元值赋值到cv::Mat中，操作像元值：https://zhuanlan.zhihu.com/p/51842288
//        //int dataType = m_matRgbImage->type();//当数据类型为CV_16UC3时，返回18
//        //std::cout << "m_matRgbImage数据类型为：" << dataType << std::endl;
//        if (matRgbImage->type() == CV_8UC3)
//        {
////            std::cout << "操作像素值！" << std::endl;
//            matRgbImage->at<cv::Vec3b>(rowNumber, j)[2] = r;
//            matRgbImage->at<cv::Vec3b>(rowNumber, j)[1] = g;
//            matRgbImage->at<cv::Vec3b>(rowNumber, j)[0] = b;
//
////            QString savePath_cv = "/media/nvme/delete/" + QString::number(m_iFrameCounter) + "_cv.png";
////            cv::imwrite(savePath_cv.toStdString(), *matRgbImage);
//        }
//
//        int column = 800;
//        if(j == column)
//        {
//            std::cout << "行：" << rowNumber << "提取：第 " << column << " 列的 r g b 分别为 " << r <<  " " << g <<  " " << b << std::endl;
//            std::cout << "mat：第 " << column << " 列的 r g b 分别为 " << (unsigned short)matRgbImage->at<cv::Vec3b>(rowNumber, j)[2] <<  " " << (unsigned short)matRgbImage->at<cv::Vec3b>(rowNumber, j)[1] <<  " " << (unsigned short)matRgbImage->at<cv::Vec3b>(rowNumber, j)[0] << std::endl;
//        }
//    }

    //方式2：通过指针操作，更快
    unsigned short r, g, b;
    const int cols = matRgbImage->cols;
    const int step = matRgbImage->channels();

    unsigned char *p_row0_b = matRgbImage->ptr(rowNumber);
    unsigned char *p_row0_g = matRgbImage->ptr(rowNumber) + 1;
    unsigned char *p_row0_r = matRgbImage->ptr(rowNumber) + 2;

    for (int j = 0; j < m_iSampleNumber; j++)
    {
        //取值：一帧影像中，从左到右的rgb像元值，线性拉伸
        r = *(datacube + m_iRedBandNumber * m_iSampleNumber + j)*255/4096;
        g = *(datacube + m_iGreenBandNumber * m_iSampleNumber + j)*255/4096;
        b = *(datacube + m_iBlueBandNumber * m_iSampleNumber + j)*255/4096;

        *p_row0_b = b;
        *p_row0_g = g;
        *p_row0_r = r;

//        int column = 800;
//        if(j == column)
//        {
//            std::cout << "行：" << rowNumber << "提取：第 " << column << " 列的 r g b 分别为 " << r <<  " " << g <<  " " << b << std::endl;
////            std::cout << "修改后" << rowNumber << "提取：第 " << column << " 列的 r g b 分别为 " << (unsigned short)*p_row0_r <<  " " << (unsigned short)*p_row0_g <<  " " << (unsigned short)*p_row0_b << std::endl;
////            std::cout << "mat：第 " << column << " 列的 r g b 分别为 " << (unsigned short)matRgbImage->at<cv::Vec3b>(rowNumber, j)[2] <<  " " << (unsigned short)matRgbImage->at<cv::Vec3b>(rowNumber, j)[1] <<  " " << (unsigned short)matRgbImage->at<cv::Vec3b>(rowNumber, j)[0] << std::endl;
//        }

        p_row0_b += step;
        p_row0_g += step;
        p_row0_r += step;
    }

    //方式3：通过内存拷贝快速提取rgb
//    if (matRgbImage->isContinuous())// check mat is continuous or not
//        matRgbImage->reshape(1, matRgbImage->rows * matRgbImage->cols).col(0).setTo(Scalar(value));
//    else
//    {
//        for (int i = 0; i < matRgbImage->rows; i++)
//            matRgbImage->row(i).reshape(1, matRgbImage->cols).col(0).setTo(Scalar(value));
//    }
}

QImage rgbImage::Mat2QImage(cv::Mat cvImg)//https://www.cnblogs.com/annt/p/ant003.html
{
    QImage qImg;
    if (cvImg.channels() == 3)//3 channels color image
    {
        cv::cvtColor(cvImg, cvImg, CV_BGR2RGB);
        qImg = QImage((const unsigned char*)(cvImg.data),
                      cvImg.cols, cvImg.rows,
                      cvImg.cols*cvImg.channels(),
                      QImage::Format_RGB888);
    }
    else if (cvImg.channels() == 1)//grayscale image
    {
        qImg = QImage((const unsigned char*)(cvImg.data),
                      cvImg.cols, cvImg.rows,
                      cvImg.cols*cvImg.channels(),
                      QImage::Format_Indexed8);
    }
    else
    {
        qImg = QImage((const unsigned char*)(cvImg.data),
                      cvImg.cols, cvImg.rows,
                      cvImg.cols*cvImg.channels(),
                      QImage::Format_RGB888);
    }

    return qImg;
}

void rgbImage::FillRgbImage(unsigned short *datacube)
{
    //从第二行开始，向下移动一行，https://blog.csdn.net/u014686356/article/details/65937750
//    m_matRgbImage->rowRange(0, m_matRgbImage->rows - 1).copyTo(m_matRgbImage->rowRange(1, m_matRgbImage->rows));//经tc验证：此行代码工作异常，为啥不加.clone()就异常？？？？？？
//    m_matRgbImage->rowRange(0, m_matRgbImage->rows - 1).clone().copyTo(m_matRgbImage->rowRange(1, m_matRgbImage->rows));//此方式，ximea帧率130hz，1min左右就出现漏帧

//    cv::Mat upperPart = m_matRgbImage->rowRange(0, m_matRgbImage->rows - 1).clone();//此方式，ximea帧率130hz，1min左右就出现漏帧
//    upperPart.copyTo(m_matRgbImage->rowRange(1, m_matRgbImage->rows));

    for (int i = m_matRgbImage->rows - 2; i >= 0; --i)//此方式，ximea帧率130hz，4.5min左右出现漏帧 → 此方式效率最高
    {
        m_matRgbImage->row(i).copyTo(m_matRgbImage->row(i+1));
    }

    FillOnerowofRgbImage(m_matRgbImage, 0, datacube);

//    m_Qphoto = Mat2QImage(*m_matRgbImage);

    //保存rgb图片
//    if (m_iFrameCounter % m_iFramerate == 0 || m_iFrameCounter == m_iFrameNumber - 1)
//    {
////        QString savePath = "/media/nvme/delete/" + QString::number(m_iFrameCounter) + "_qt.jpg";
////        m_Qphoto.save(savePath);
//
//        QString savePath_cv = "/media/nvme/delete/" + QString::number(m_iFrameCounter) + "_cv.jpg";
//        cv::imwrite(savePath_cv.toStdString(), *m_matRgbImage);
//    }
//    m_VideoWriter.write(*m_matRgbImage);
//    std::string rgbFilePathNoStrech = "/media/nvme/delete/" + std::to_string(m_iFrameCounter) + "ctmp_image_no_strech.png";
//    cv::imwrite(rgbFilePathNoStrech, *m_matRgbImage);

    m_iFrameCounter++;
}

void rgbImage::FillFocusGrayImage(unsigned short * datacube)
{
    int rowCount = m_matFocusGrayImage->rows;
    int colCount = m_matFocusGrayImage->cols;
    for (unsigned short i = 0; i < m_matFocusGrayImage->rows; i++)
    {
        for (unsigned short j = 0; j < m_matFocusGrayImage->cols; j++)
        {
            //m_matFocusGrayImage->at<ushort>(i, j) = *(datacube + m_matFocusGrayImage->cols*i + j);
            m_matFocusGrayImage->at<ushort>(i, j) = datacube[m_matFocusGrayImage->cols*i + j];
        }
    }



    //int rowCount = m_matFocusGrayImage.rows;
    //int colCount = m_matFocusGrayImage.cols;
    ////memcpy(m_matFocusGrayImage.data, datacube, rowCount*colCount);
    //for (unsigned short i = 0; i < m_matFocusGrayImage.rows; i++)
    //{
    //	for (unsigned short j = 0; j < m_matFocusGrayImage.cols; j++)
    //	{
    //		m_matFocusGrayImage.at<ushort>(i, j) = *(datacube + m_matFocusGrayImage.cols*i + j);
    //		//m_matFocusGrayImage.at<ushort>(i, j) = datacube[colCount*i + j];
    //	}
    //}

    //将mat保存成文件
    //cv::imwrite("D:/delete/2222222222/test.bmp", m_matFocusGrayImage);
}

void rgbImage::FillFocusGrayQImage(unsigned short * datacube)
{
    float two_eight = pow(2.0, 8);
    float two_sixteen = pow(2.0, 12);

    int width = m_qimageFocusGrayImage->width();
    int height = m_qimageFocusGrayImage->height();
    for (unsigned short i = 0; i < height; i++)
    {
        for (unsigned short j = 0; j < width; j++)
        {
            //uint tmp = (two_eight* *(datacube + width * i + j)) / two_sixteen;
            uint tmp = (two_eight* datacube[width*i + j]) / two_sixteen;
            //uint tmp = datacube[width*i + j];


            //m_qimageFocusGrayImage->setPixel(j, i, tmp);
            m_qimageFocusGrayImage->setPixel(j, i, qRgb((unsigned char)tmp, (unsigned char)tmp, (unsigned char)tmp));
        }
    }

    m_qimageFocusGrayImage->save("D:/delete/2222222222/test.bmp");




    /*float two_eight = pow(2.0, 8);
    float two_sixteen = pow(2.0, 16);
    QImage *qi = new QImage(imwidth, imheight, QImage::Format_RGB32);
    for (int i = 0; i < imheight; i++)
    {
        for (int j = 0; j < imwidth; j++)
        {
            floatData[i*imwidth + j] = (two_eight* floatData[i*imwidth + j]) / two_sixteen;
            qi->setPixel(j, i, qRgb((unsigned char)floatData[i*imwidth + j], (unsigned char)floatData[i*imwidth + j], (unsigned char)floatData[i*imwidth + j]));


        }
    }*/

}