// // 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 <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(rowNumber, j)[2] = r; // matRgbImage->at(rowNumber, j)[1] = g; // matRgbImage->at(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(rowNumber, j)[2] << " " << (unsigned short)matRgbImage->at(rowNumber, j)[1] << " " << (unsigned short)matRgbImage->at(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(rowNumber, j)[2] << " " << (unsigned short)matRgbImage->at(rowNumber, j)[1] << " " << (unsigned short)matRgbImage->at(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(i, j) = *(datacube + m_matFocusGrayImage->cols*i + j); m_matFocusGrayImage->at(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(i, j) = *(datacube + m_matFocusGrayImage.cols*i + j); // //m_matFocusGrayImage.at(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])); } }*/ }