308 lines
12 KiB
C++
308 lines
12 KiB
C++
//
|
||
// 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]));
|
||
|
||
|
||
}
|
||
}*/
|
||
|
||
}
|