114 lines
2.5 KiB
C++
114 lines
2.5 KiB
C++
//
|
||
// Created by tangchao on 2022/12/24.
|
||
//
|
||
|
||
#ifndef XIMEAAIRBORNESYSTEM_RGBIMAGE_H
|
||
#define XIMEAAIRBORNESYSTEM_RGBIMAGE_H
|
||
|
||
#include <iostream>
|
||
#include <string>
|
||
|
||
#include <QObject>
|
||
#include <QImage>
|
||
#include <opencv2/opencv.hpp>//包含了所有东西，编译很慢
|
||
#include "opencv2/imgproc/types_c.h"
|
||
|
||
extern "C"
|
||
{
|
||
#include <libavcodec/avcodec.h>
|
||
#include <libavutil/opt.h>
|
||
#include "libavutil/pixfmt.h"
|
||
#include "libswscale/swscale.h"
|
||
#include <libavutil/imgutils.h>
|
||
#include <libavutil/avutil.h>
|
||
#include "libavdevice/avdevice.h"
|
||
}
|
||
|
||
class Encode
|
||
{
|
||
public:
|
||
Encode();
|
||
void initffmpeg(int width, int height);
|
||
|
||
FILE *fp;
|
||
AVCodecContext *avcodeccontext;
|
||
void savedata(AVFrame *frame);
|
||
void encode(AVCodecContext *enc_ctx, AVFrame *frame, AVPacket *pkt, FILE *outfile);
|
||
AVPacket m_avpacket;
|
||
AVFrame *inpic;
|
||
AVFrame *outpic;
|
||
|
||
bool isinit;
|
||
int index1;
|
||
|
||
private:
|
||
};
|
||
|
||
|
||
|
||
using namespace cv;
|
||
class rgbImage :public QObject
|
||
{
|
||
|
||
Q_OBJECT
|
||
public:
|
||
rgbImage(QWidget* pParent = NULL);
|
||
~rgbImage();
|
||
|
||
void SetRgbImageWidthAndHeight(int BandCount, int Sample, int FrameNumber);
|
||
void FillRgbImage(unsigned short *datacube);
|
||
void FillFocusGrayImage(unsigned short *datacube);
|
||
void FillFocusGrayQImage(unsigned short * datacube);
|
||
|
||
void FillOnerowofRgbImage(cv::Mat * matRgbImage, int rowNumber, unsigned short *datacube);
|
||
QImage Mat2QImage(cv::Mat cvImg);//https://www.cnblogs.com/annt/p/ant003.html
|
||
|
||
QImage *m_QRgbImage;
|
||
cv::Mat *m_matRgbImage;
|
||
QImage m_Qphoto;
|
||
|
||
QImage *m_qimageFocusGrayImage;
|
||
cv::Mat *m_matFocusGrayImage;//用于调焦时，显示一帧的灰度图
|
||
//cv::Mat m_matFocusGrayImage;//用于调焦时，显示一帧的灰度图
|
||
|
||
|
||
CvVideoWriter *m_frame_writer;
|
||
VideoWriter m_VideoWriter;
|
||
// VideoWriter m_video("appsrc ! autovideoconvert ! filesink location=/media/nvme/delete/live.avi", CV_FOURCC('M', 'J', 'P', 'G'), 25.0, Size(640, 480));
|
||
// VideoWriter video("test.avi", CV_FOURCC('M', 'J', 'P', 'G'), 25.0, Size(640, 480));//
|
||
|
||
|
||
|
||
|
||
|
||
//控制该填充rgb图像第几帧（行）数据
|
||
//以下两种情况需要重置为0：1）调用函数SetRgbImageWidthAndHeight；2）每次开始填充数据前
|
||
int m_iFrameCounter;
|
||
|
||
|
||
int m_iFramerate;//
|
||
|
||
|
||
protected:
|
||
|
||
private:
|
||
int m_iSampleNumber;//
|
||
int m_iBandNumber;//
|
||
|
||
int m_iFrameNumber;//
|
||
|
||
void initffmpeg();
|
||
|
||
public slots:
|
||
|
||
signals :
|
||
void sendstr(QString str);
|
||
void sendstr1(QString str);
|
||
void refreslabelimg(QImage* img1);
|
||
|
||
|
||
};
|
||
|
||
|
||
#endif //XIMEAAIRBORNESYSTEM_RGBIMAGE_H
|