Compare commits
8 Commits
| Author | SHA1 | Date | |
|---|---|---|---|
| e8760dcfe5 | |||
| bd4d055129 | |||
| a91f5f5b04 | |||
| 2e4679aaef | |||
| 061e1f83bd | |||
| d0b67b47c1 | |||
| 7fa3b70d10 | |||
| 781b33f577 |
@ -23,6 +23,9 @@ find_package(OpenCV 4.2.0 REQUIRED)
|
||||
include_directories(/usr/local/include/opencv4/)
|
||||
link_directories(/usr/local/lib)
|
||||
|
||||
include_directories(/home/300tc/library/ffmpeg_build/include)
|
||||
link_directories(/home/300tc/library/ffmpeg_build/lib)
|
||||
|
||||
add_executable(${CMAKE_PROJECT_NAME}
|
||||
Source_Files/fileoperation.cpp
|
||||
Header_Files/fileoperation.h
|
||||
@ -53,4 +56,8 @@ target_link_libraries(${CMAKE_PROJECT_NAME}
|
||||
irisXimeaImager
|
||||
libconfig.so
|
||||
libconfig++.so
|
||||
${OpenCV_LIBS})
|
||||
${OpenCV_LIBS}
|
||||
avformat
|
||||
avcodec
|
||||
swscale
|
||||
avutil)
|
||||
|
||||
@ -42,6 +42,8 @@ public:
|
||||
bool getBufferPolicy(int &bufferPolicy);
|
||||
bool getAcqBufferSize(int &acqBufferSize);
|
||||
|
||||
bool getPushFlowParam(int &flowSwitch, int &rgbHeight, int &framerateVideo);
|
||||
|
||||
bool createConfigFile();
|
||||
bool updateConfigFile();
|
||||
|
||||
@ -68,8 +70,36 @@ public:
|
||||
// bool updateConfigFile();
|
||||
bool writeConfig2File();
|
||||
|
||||
private:
|
||||
string m_configfilePath;
|
||||
Config cfg;
|
||||
private:
|
||||
};
|
||||
|
||||
class StateOf300tcConfigfile:public ParameterConfigfile
|
||||
{
|
||||
public:
|
||||
StateOf300tcConfigfile();
|
||||
|
||||
bool getSwitchState(bool &switchState);
|
||||
bool setSwitchState(bool switchState);
|
||||
|
||||
bool getSbgState(int &sbgState);
|
||||
bool setSbgState(int sbgState);
|
||||
|
||||
bool getSbgSolutionMode(int &sbgSolutionMode);
|
||||
bool setSbgSolutionMode(int sbgSolutionMode);
|
||||
|
||||
bool getXimeaState(int &ximeaState);
|
||||
bool setXimeaState(int ximeaState);
|
||||
|
||||
bool getExposeMaxValueOfOneFrame(int &exposeMaxValueOfOneFrame);
|
||||
bool setExposeMaxValueOfOneFrame(int exposeMaxValueOfOneFrame);
|
||||
|
||||
bool createConfigFile();
|
||||
|
||||
private:
|
||||
bool getIntValue(int &value, string field);
|
||||
bool setIntValue(int value, string field);
|
||||
|
||||
};
|
||||
#endif //XIMEAIMAGERECORDER_CONFIGFILE_H
|
||||
|
||||
@ -11,6 +11,18 @@
|
||||
#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"
|
||||
}
|
||||
|
||||
using namespace cv;
|
||||
class rgbImage :public QObject
|
||||
@ -21,38 +33,33 @@ public:
|
||||
rgbImage(QWidget* pParent = NULL);
|
||||
~rgbImage();
|
||||
|
||||
void SetRgbImageWidthAndHeight(int BandCount, int Sample, int FrameNumber);
|
||||
void SetRgbImageWidthAndHeight(int BandCount, int Width, int height);
|
||||
void SetRgbBandNumber(int redBandNumber, int greenBandNumber, int blueBandNumber);
|
||||
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:
|
||||
@ -61,15 +68,17 @@ private:
|
||||
|
||||
int m_iFrameNumber;//
|
||||
|
||||
void initffmpeg();
|
||||
|
||||
int m_iRedBandNumber;
|
||||
int m_iGreenBandNumber;
|
||||
int m_iBlueBandNumber;
|
||||
|
||||
public slots:
|
||||
|
||||
signals :
|
||||
void sendstr(QString str);
|
||||
void sendstr1(QString str);
|
||||
void refreslabelimg(QImage* img1);
|
||||
|
||||
|
||||
};
|
||||
|
||||
|
||||
#endif //XIMEAAIRBORNESYSTEM_RGBIMAGE_H
|
||||
|
||||
@ -11,17 +11,43 @@
|
||||
#include <QObject>
|
||||
#include <QThread>
|
||||
#include <QDir>
|
||||
#include <QtSerialPort/QSerialPort>
|
||||
#include <QtSerialPort/QSerialPortInfo>
|
||||
|
||||
#include "ximeaimager.h"
|
||||
#include "sbgrecorder.h"
|
||||
#include "fileoperation.h"
|
||||
|
||||
#include "configfile.h"
|
||||
|
||||
extern "C"
|
||||
{
|
||||
// #include <sbgEComLib.h>
|
||||
#include <stdlib.h>
|
||||
}
|
||||
|
||||
class Record300TcTemperature : public QObject
|
||||
{
|
||||
Q_OBJECT
|
||||
|
||||
public:
|
||||
Record300TcTemperature();
|
||||
|
||||
void stopRecordTemperature();
|
||||
|
||||
private:
|
||||
bool m_bIsRecord;
|
||||
|
||||
QSerialPort * m_serial;
|
||||
|
||||
float parseMessage(QByteArray * data);
|
||||
public slots:
|
||||
void recordTemperature(QString filePath);
|
||||
|
||||
signals:
|
||||
|
||||
};
|
||||
|
||||
class UdpServer:public QObject
|
||||
{
|
||||
Q_OBJECT
|
||||
@ -48,6 +74,11 @@ class UdpServer:public QObject
|
||||
|
||||
void sender(int status);
|
||||
|
||||
QThread * m_recordTempThread;
|
||||
Record300TcTemperature * m_300tcTemperature;
|
||||
|
||||
StateOf300tcConfigfile stateOf300tc;
|
||||
|
||||
signals:
|
||||
void systemStart();
|
||||
void systemStop();
|
||||
@ -56,6 +87,7 @@ signals:
|
||||
void startDeleteFileSignal();
|
||||
|
||||
void recordXimeaOnlySignal(double,QString);
|
||||
void record300tcTemperatureSignal(QString);
|
||||
|
||||
public slots:
|
||||
void onRecordFinished();
|
||||
|
||||
@ -27,6 +27,8 @@
|
||||
#include <exception>
|
||||
#include <fcntl.h>
|
||||
#include <sys/mman.h>
|
||||
#include <cmath>
|
||||
#include <vector>
|
||||
|
||||
#include <QObject>
|
||||
#include <QDateTime>
|
||||
@ -42,9 +44,12 @@
|
||||
#include "MemoryPool.h"
|
||||
#include <queue>
|
||||
#include <QMutex>
|
||||
#include <QUdpSocket>
|
||||
|
||||
#include "rgbImage.h"
|
||||
|
||||
#define PUSH_FLOW_PORT 666
|
||||
|
||||
|
||||
|
||||
//#ifdef WIN32
|
||||
@ -122,6 +127,32 @@ public slots:
|
||||
signals:
|
||||
};
|
||||
|
||||
class PushFlow : public QObject
|
||||
{
|
||||
Q_OBJECT
|
||||
|
||||
public:
|
||||
PushFlow();
|
||||
void setParm(rgbImage * img, int width, int height, int framerateVideo);
|
||||
void exitPushFlow();
|
||||
void setVedioFilePath(QString path);
|
||||
|
||||
private:
|
||||
QString m_QVedioFilePath;
|
||||
bool isExitPushFlow;
|
||||
|
||||
rgbImage * m_rgbImage;
|
||||
|
||||
int m_iWidth;
|
||||
int m_iHeight;
|
||||
int m_iFramerateVideo;
|
||||
|
||||
public slots:
|
||||
void encodePushFlow();
|
||||
|
||||
signals:
|
||||
};
|
||||
|
||||
class XimeaImager : public QObject
|
||||
{
|
||||
Q_OBJECT
|
||||
@ -145,6 +176,9 @@ public:
|
||||
int getWindowEndBand();
|
||||
double geWavelengthAtBand(int x);
|
||||
|
||||
static int findClosestIndex(const std::vector<double>& numbers, double target);
|
||||
void getRgbBandNumber(int &redBandNumber, int &greenBandNumber, int &blueBandNumber);
|
||||
|
||||
void stopRecord();
|
||||
int getFrameCounter();
|
||||
void writeXiApiErrorCodes(QString filePath, int xiApiErrorCodes);
|
||||
@ -169,6 +203,10 @@ private:
|
||||
queue<int> * m_qFrameCounter;
|
||||
MemoryPool<DataBuffer> * m_pool;
|
||||
|
||||
QThread * m_pushFlowThread;
|
||||
PushFlow * m_pushFlow;
|
||||
int m_iFlowSwitch;
|
||||
|
||||
QString m_baseFileName;
|
||||
QString m_ximeaTemperatureCSVPath;
|
||||
|
||||
@ -207,6 +245,7 @@ signals:
|
||||
|
||||
void recordXimeaTemperatureSignal(QString);
|
||||
void startWriteDiskSignal();
|
||||
void startPushFlowSignal();
|
||||
|
||||
void autoExposeMaxValueOfOneFrame(int, double);
|
||||
void frameRateSignal(double);
|
||||
|
||||
@ -166,6 +166,64 @@ bool Configfile::getEffectiveWindowRoi(int &width, int &offsetx)
|
||||
return true;
|
||||
}
|
||||
|
||||
bool Configfile::getPushFlowParam(int &flowSwitch, int &rgbHeight, int &framerateVideo)
|
||||
{
|
||||
Setting& root = cfg.getRoot();
|
||||
|
||||
if (!root.exists("push_flow_param"))
|
||||
{
|
||||
// 配置项不存在，添加配置项
|
||||
Setting & push_flow_param = root.add("push_flow_param", Setting::TypeGroup);
|
||||
|
||||
push_flow_param.add("flow_switch", Setting::TypeInt) = 0;
|
||||
push_flow_param.add("rgb_height", Setting::TypeInt) = 720;
|
||||
push_flow_param.add("framerate_video", Setting::TypeInt) = 5;
|
||||
|
||||
// 保存修改后的配置到文件
|
||||
try
|
||||
{
|
||||
QList<QString> fileInfo = getFileInfo(QString::fromStdString(m_configfilePath));
|
||||
bool ret = createDir(fileInfo[0]);
|
||||
|
||||
cfg.writeFile(m_configfilePath.c_str());
|
||||
std::cout << "Config item 'push_flow_param' added." << std::endl;
|
||||
|
||||
flowSwitch = 0;
|
||||
rgbHeight = 720;
|
||||
framerateVideo = 5;
|
||||
|
||||
return true;
|
||||
}
|
||||
catch (const libconfig::FileIOException &fioex)
|
||||
{
|
||||
std::cerr << "I/O error while writing file." << std::endl;
|
||||
return false;
|
||||
}
|
||||
}
|
||||
else
|
||||
{
|
||||
try
|
||||
{
|
||||
const Setting &push_flow_param = root["push_flow_param"];
|
||||
|
||||
if(!(push_flow_param.lookupValue("rgb_height", rgbHeight)
|
||||
&& push_flow_param.lookupValue("framerate_video", framerateVideo)
|
||||
&& push_flow_param.lookupValue("flow_switch", flowSwitch)
|
||||
))
|
||||
{
|
||||
return false;
|
||||
}
|
||||
}
|
||||
catch(const SettingNotFoundException &nfex)
|
||||
{
|
||||
// Ignore.
|
||||
return false;
|
||||
}
|
||||
|
||||
return true;
|
||||
}
|
||||
}
|
||||
|
||||
bool Configfile::getWindowOffsety_HeightOfSpectral(int &offsety, int &height, string spectralBinString)
|
||||
{
|
||||
const Setting& root = cfg.getRoot();
|
||||
@ -554,7 +612,7 @@ bool ParameterConfigfile::getFrameRate(int &frameRate)
|
||||
const Setting &ximeaParam = root["ximeaParam"];
|
||||
frameRate = ximeaParam.lookup("frameRate");
|
||||
|
||||
std::cout << "----------------"<< frameRate<<std::endl;
|
||||
// std::cout << "ParameterConfigfile::getFrameRate："<< frameRate<<std::endl;
|
||||
return true;
|
||||
}
|
||||
catch(const SettingNotFoundException &nfex)
|
||||
@ -591,7 +649,7 @@ bool ParameterConfigfile::getExposeTime(float &exposeTime)
|
||||
const Setting &ximeaParam = root["ximeaParam"];
|
||||
exposeTime = ximeaParam.lookup("exposeTime");
|
||||
|
||||
std::cout << "----------------"<< exposeTime<<std::endl;
|
||||
// std::cout << "ParameterConfigfile::getExposeTime"<< exposeTime<<std::endl;
|
||||
return true;
|
||||
}
|
||||
catch(const SettingNotFoundException &nfex)
|
||||
@ -628,7 +686,7 @@ bool ParameterConfigfile::writeConfig2File()
|
||||
bool ret = createDir(fileInfo[0]);
|
||||
|
||||
cfg.writeFile(m_configfilePath.c_str());
|
||||
cerr << "New configuration successfully written to: " << m_configfilePath.c_str() << endl;
|
||||
// cerr << "New configuration successfully written to: " << m_configfilePath.c_str() << endl;
|
||||
}
|
||||
catch(const FileIOException &fioex)
|
||||
{
|
||||
@ -637,3 +695,126 @@ bool ParameterConfigfile::writeConfig2File()
|
||||
}
|
||||
return true;
|
||||
}
|
||||
|
||||
StateOf300tcConfigfile::StateOf300tcConfigfile()
|
||||
{
|
||||
}
|
||||
|
||||
bool StateOf300tcConfigfile::getIntValue(int &value, string field)
|
||||
{
|
||||
const Setting& root = cfg.getRoot();
|
||||
try
|
||||
{
|
||||
const Setting &ximeaParam = root["ximeaParam"];
|
||||
value = ximeaParam.lookup(field);
|
||||
|
||||
// std::cout << "StateOf300tcConfigfile::getIntValue"<< value<<std::endl;
|
||||
return true;
|
||||
}
|
||||
catch(const SettingNotFoundException &nfex)
|
||||
{
|
||||
cerr << "No setting in configuration file." << endl;
|
||||
return false;
|
||||
}
|
||||
}
|
||||
|
||||
bool StateOf300tcConfigfile::setIntValue(int value, string field)
|
||||
{
|
||||
const Setting& root = cfg.getRoot();
|
||||
try
|
||||
{
|
||||
Setting &frameRate2 = root["ximeaParam"][field];
|
||||
frameRate2 = value;
|
||||
|
||||
writeConfig2File();
|
||||
|
||||
return true;
|
||||
}
|
||||
catch(const SettingNotFoundException &nfex)
|
||||
{
|
||||
cerr << "No setting in configuration file." << endl;
|
||||
return false;
|
||||
}
|
||||
}
|
||||
|
||||
bool StateOf300tcConfigfile::createConfigFile()
|
||||
{
|
||||
using namespace std;
|
||||
using namespace libconfig;
|
||||
|
||||
Config cfg;
|
||||
|
||||
Setting &root = cfg.getRoot();
|
||||
|
||||
// Add some settings to the configuration.
|
||||
// Setting &SN = root.add("SN", Setting::TypeString) = "0098";
|
||||
// Setting &spatialBin = root.add("spatialBin", Setting::TypeInt) = 2;
|
||||
// Setting &SpectralBin = root.add("spectralBin", Setting::TypeInt) = 2;
|
||||
|
||||
Setting &ximeaParam = root.add("ximeaParam", Setting::TypeGroup);
|
||||
ximeaParam.add("sbgState", Setting::TypeInt) = 0;
|
||||
ximeaParam.add("sbgSolutionMode", Setting::TypeInt) = 0;
|
||||
ximeaParam.add("ximeaState", Setting::TypeInt) = 100;
|
||||
ximeaParam.add("frameRate", Setting::TypeInt) = 0;
|
||||
ximeaParam.add("exposeTime", Setting::TypeFloat) = 0.0;
|
||||
ximeaParam.add("exposeMaxValueOfOneFrame", Setting::TypeInt) = 0;
|
||||
|
||||
|
||||
// Write out the new configuration.
|
||||
try
|
||||
{
|
||||
QList<QString> fileInfo = getFileInfo(QString::fromStdString(m_configfilePath));
|
||||
bool ret = createDir(fileInfo[0]);
|
||||
|
||||
cfg.writeFile(m_configfilePath.c_str());
|
||||
cerr << "New configuration successfully written to: " << m_configfilePath.c_str() << endl;
|
||||
|
||||
}
|
||||
catch(const FileIOException &fioex)
|
||||
{
|
||||
cerr << "I/O error while writing configuration file: " << m_configfilePath.c_str() << endl;
|
||||
return true;
|
||||
}
|
||||
|
||||
return true;
|
||||
}
|
||||
|
||||
bool StateOf300tcConfigfile::getSbgState(int &sbgState)
|
||||
{
|
||||
return getIntValue(sbgState,"sbgState");
|
||||
}
|
||||
|
||||
bool StateOf300tcConfigfile::setSbgState(int sbgState)
|
||||
{
|
||||
return setIntValue(sbgState,"sbgState");
|
||||
}
|
||||
|
||||
bool StateOf300tcConfigfile::getSbgSolutionMode(int &sbgSolutionMode)
|
||||
{
|
||||
return getIntValue(sbgSolutionMode,"sbgSolutionMode");
|
||||
}
|
||||
|
||||
bool StateOf300tcConfigfile::setSbgSolutionMode(int sbgSolutionMode)
|
||||
{
|
||||
return setIntValue(sbgSolutionMode,"sbgSolutionMode");
|
||||
}
|
||||
|
||||
bool StateOf300tcConfigfile::getXimeaState(int &ximeaState)
|
||||
{
|
||||
return getIntValue(ximeaState,"ximeaState");
|
||||
}
|
||||
|
||||
bool StateOf300tcConfigfile::setXimeaState(int ximeaState)
|
||||
{
|
||||
return setIntValue(ximeaState,"ximeaState");
|
||||
}
|
||||
|
||||
bool StateOf300tcConfigfile::getExposeMaxValueOfOneFrame(int &exposeMaxValueOfOneFrame)
|
||||
{
|
||||
return getIntValue(exposeMaxValueOfOneFrame,"exposeMaxValueOfOneFrame");
|
||||
}
|
||||
|
||||
bool StateOf300tcConfigfile::setExposeMaxValueOfOneFrame(int exposeMaxValueOfOneFrame)
|
||||
{
|
||||
return setIntValue(exposeMaxValueOfOneFrame,"exposeMaxValueOfOneFrame");
|
||||
}
|
||||
|
||||
@ -3,7 +3,7 @@
|
||||
|
||||
int main(int argc, char *argv[])
|
||||
{
|
||||
std::cout<<"ximeaAirborneSystem 版本："<< "39." <<std::endl;
|
||||
std::cout<<"ximeaAirborneSystem 版本："<< "43." <<std::endl;
|
||||
QCoreApplication a(argc, argv);
|
||||
|
||||
//UdpServer* x=new UdpServer();
|
||||
|
||||
@ -4,7 +4,6 @@
|
||||
|
||||
#include "../Header_Files/rgbImage.h"
|
||||
|
||||
|
||||
rgbImage::rgbImage(QWidget* pParent)
|
||||
{
|
||||
m_QRgbImage = nullptr;
|
||||
@ -12,8 +11,9 @@ rgbImage::rgbImage(QWidget* pParent)
|
||||
m_matFocusGrayImage = nullptr;
|
||||
m_qimageFocusGrayImage = nullptr;
|
||||
|
||||
|
||||
|
||||
m_iRedBandNumber = 0;
|
||||
m_iGreenBandNumber = 0;
|
||||
m_iBlueBandNumber = 0;
|
||||
}
|
||||
|
||||
rgbImage::~rgbImage()
|
||||
@ -21,8 +21,18 @@ rgbImage::~rgbImage()
|
||||
|
||||
}
|
||||
|
||||
void rgbImage::SetRgbBandNumber(int redBandNumber, int greenBandNumber, int blueBandNumber)
|
||||
{
|
||||
m_iRedBandNumber = redBandNumber;
|
||||
m_iGreenBandNumber = greenBandNumber;
|
||||
m_iBlueBandNumber = blueBandNumber;
|
||||
|
||||
void rgbImage::SetRgbImageWidthAndHeight(int BandCount, int Sample, int FrameNumber)
|
||||
// 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;
|
||||
|
||||
@ -30,19 +40,21 @@ void rgbImage::SetRgbImageWidthAndHeight(int BandCount, int Sample, int FrameNum
|
||||
{
|
||||
delete m_QRgbImage;//有问题？？？？？？？？？？？？？？？？？？？？？？？？？？？？？？？？？？？？？？？？？？？？？？？？
|
||||
}
|
||||
//m_QRgbImage = new QImage(Sample, FrameNumber, QImage::Format_RGB888);
|
||||
//m_QRgbImage = new QImage(Width, height, QImage::Format_RGB888);
|
||||
|
||||
|
||||
if (m_matRgbImage != nullptr)
|
||||
{
|
||||
delete m_matRgbImage;
|
||||
}
|
||||
m_matRgbImage = new Mat(FrameNumber, Sample, CV_8UC3, Scalar(0, 0, 0));
|
||||
m_matRgbImage = new Mat(height, Width, CV_8UC3, Scalar(0, 0, 0));
|
||||
|
||||
int codec = VideoWriter::fourcc('M', 'P', '4', '2'); // 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/delete/live.avi";//https://blog.csdn.net/ancientapesman/article/details/117324638
|
||||
m_VideoWriter.open(filename, codec, fps, m_matRgbImage->size(), true);
|
||||
// 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));
|
||||
|
||||
@ -54,127 +66,169 @@ void rgbImage::SetRgbImageWidthAndHeight(int BandCount, int Sample, int FrameNum
|
||||
|
||||
if (m_qimageFocusGrayImage == nullptr)
|
||||
{
|
||||
m_qimageFocusGrayImage = new QImage(Sample, BandCount, QImage::Format_RGB32);
|
||||
m_qimageFocusGrayImage = new QImage(Width, BandCount, QImage::Format_RGB32);
|
||||
}
|
||||
|
||||
|
||||
if (m_matFocusGrayImage == nullptr)
|
||||
{
|
||||
m_matFocusGrayImage = new Mat(BandCount, Sample, CV_16U, Scalar(0));
|
||||
m_matFocusGrayImage = new Mat(BandCount, Width, CV_16U, Scalar(0));
|
||||
|
||||
//cv::Mat matAdjustPreview = Mat::zeros(BandCount, Sample, CV_16U);
|
||||
//cv::Mat matAdjustPreview = Mat::zeros(BandCount, Width, CV_16U);
|
||||
}
|
||||
|
||||
//cv::Mat matAdjustPreview = Mat::zeros(BandCount, Sample, CV_16U);
|
||||
//cv::Mat matAdjustPreview = Mat::zeros(BandCount, Width, CV_16U);
|
||||
//m_matFocusGrayImage = matAdjustPreview;
|
||||
|
||||
|
||||
|
||||
|
||||
|
||||
|
||||
|
||||
|
||||
std::cout << "设置帧数：" << FrameNumber << std::endl;
|
||||
std::cout << "高光谱rgb图像设置高度：" << height << std::endl;
|
||||
|
||||
m_iFrameCounter = 0;//每次都重置为0
|
||||
m_iSampleNumber = Sample;
|
||||
m_iSampleNumber = Width;
|
||||
m_iBandNumber = BandCount;
|
||||
m_iFrameNumber = FrameNumber;
|
||||
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 + 121 * m_iSampleNumber + j)*255/4096;
|
||||
g = *(datacube + 79 * m_iSampleNumber + j)*255/4096;
|
||||
b = *(datacube + 40 * m_iSampleNumber + j)*255/4096;
|
||||
//取值：一帧影像中，从左到右的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;
|
||||
|
||||
//将像元值赋值到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_16UC3)
|
||||
{
|
||||
//std::cout << "操作像素值！" << std::endl;
|
||||
matRgbImage->at<cv::Vec3w>(rowNumber, j)[2] = r;
|
||||
matRgbImage->at<cv::Vec3w>(rowNumber, j)[1] = g;
|
||||
matRgbImage->at<cv::Vec3w>(rowNumber, j)[0] = b;
|
||||
}
|
||||
*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)
|
||||
{
|
||||
unsigned short *r_row, *g_row, *b_row;
|
||||
//从第二行开始，向下移动一行，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左右就出现漏帧
|
||||
|
||||
if(m_iFrameCounter<m_iFrameNumber)
|
||||
// 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左右出现漏帧 → 此方式效率最高
|
||||
{
|
||||
FillOnerowofRgbImage(m_matRgbImage, m_iFrameCounter, datacube);
|
||||
// std::cout << "小于：" << m_iFrameNumber << std::endl;
|
||||
m_matRgbImage->row(i).copyTo(m_matRgbImage->row(i+1));
|
||||
}
|
||||
else
|
||||
{
|
||||
// std::cout << "大于：" << m_iFrameNumber << std::endl;
|
||||
|
||||
//通过行赋值将前m_iFrameNumber-1行向上移动一行，https://blog.csdn.net/u014686356/article/details/65937750
|
||||
// m_matRgbImage->rowRange(1, m_matRgbImage->rows).copyTo(m_matRgbImage->rowRange(0, m_matRgbImage->rows-1));
|
||||
for (int i = 1; i < m_matRgbImage->rows; ++i)
|
||||
{
|
||||
// std::cout << "大于：" << i << std::endl;
|
||||
m_matRgbImage->col(i).copyTo(m_matRgbImage->col(i-1));
|
||||
// std::cout << "--------------" << i << std::endl;
|
||||
}
|
||||
FillOnerowofRgbImage(m_matRgbImage, 0, datacube);
|
||||
|
||||
// m_Qphoto = Mat2QImage(*m_matRgbImage);
|
||||
|
||||
// std::cout << "1111111111111111111111111111"<< std::endl;
|
||||
//通过FillOnerowofRgbImage为m_iFrameNumber行赋值
|
||||
FillOnerowofRgbImage(m_matRgbImage, m_iFrameNumber-1, datacube);
|
||||
// std::cout << "22222222222222222"<< std::endl;
|
||||
|
||||
|
||||
|
||||
|
||||
|
||||
// //循环给每行像素赋值
|
||||
// r_row = datacube + 121 * m_iSampleNumber;
|
||||
// g_row = datacube + 79 * m_iSampleNumber;
|
||||
// b_row = datacube + 40 * m_iSampleNumber;
|
||||
// for (int j = 0; j < m_iFrameNumber; j++)
|
||||
// {
|
||||
// p = m_matRgbImage.ptr<uchar>(j);
|
||||
// for ( j = 0; j < nCols; ++j){
|
||||
// p[j] = table[p[j]];
|
||||
// }
|
||||
//保存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);
|
||||
//
|
||||
// }
|
||||
|
||||
|
||||
|
||||
//保存rgb图片
|
||||
if (m_iFrameCounter % m_iFramerate == 0 || m_iFrameCounter == m_iFrameNumber - 1)
|
||||
{
|
||||
////保存文件
|
||||
//FileOperation * fileOperation = new FileOperation();
|
||||
//string directory = fileOperation->getDirectoryOfExe();
|
||||
//string rgbFilePathStrech = “/media/nvme/300TC/config/” + "\\tmp_image_strech.png";//没有拉伸图片
|
||||
// std::string rgbFilePathNoStrech = "/media/nvme/300TC/config/" + std::to_string(m_iFrameCounter) + "ctmp_image_no_strech.png";
|
||||
|
||||
//m_QRgbImage->save(QString::fromStdString(rgbFilePathNoStrech), "PNG");
|
||||
|
||||
// cv::imwrite(rgbFilePathNoStrech, *m_matRgbImage);
|
||||
//cv::imwrite(rgbFilePathStrech, CStretch(*m_matRgbImage, 0.01));
|
||||
|
||||
}
|
||||
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);
|
||||
}
|
||||
|
||||
// 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++;
|
||||
}
|
||||
|
||||
@ -55,7 +55,18 @@ UdpServer::UdpServer()
|
||||
connect(m_imager, SIGNAL(frameRateSignal(double)),this, SLOT(sendXimeaImageFrameRate(double)));
|
||||
connect(m_copyFile, SIGNAL(copyFileStatus(int)),this, SLOT(sendCopyFileStatus(int)));
|
||||
|
||||
QString ximeaParamCfgFile = "/media/nvme/300TC/config/StateOf300tc.cfg";
|
||||
stateOf300tc.setConfigfilePath(ximeaParamCfgFile.toStdString());
|
||||
if(!stateOf300tc.isConfigfileExist())
|
||||
stateOf300tc.createConfigFile();
|
||||
stateOf300tc.parseConfigfile();
|
||||
|
||||
stateOf300tc.setSbgState(0);
|
||||
stateOf300tc.setSbgSolutionMode(0);
|
||||
stateOf300tc.setXimeaState(100);
|
||||
stateOf300tc.setFrameRate(0);
|
||||
stateOf300tc.setExposeTime(0);
|
||||
stateOf300tc.setExposeMaxValueOfOneFrame(0);
|
||||
|
||||
//当软件不正常关闭并且重启后，通知其他psdk程序
|
||||
m_clientIpAddress=QHostAddress(QHostAddress::LocalHost);
|
||||
@ -65,6 +76,16 @@ UdpServer::UdpServer()
|
||||
|
||||
system("sudo gpio write 10 0");
|
||||
|
||||
m_recordTempThread=new QThread();
|
||||
m_300tcTemperature = new Record300TcTemperature();
|
||||
m_300tcTemperature->moveToThread(m_recordTempThread);
|
||||
m_recordTempThread->start();
|
||||
connect(this, SIGNAL(record300tcTemperatureSignal(QString)),m_300tcTemperature, SLOT(recordTemperature(QString)));
|
||||
|
||||
QDateTime curDateTime = QDateTime::currentDateTime();
|
||||
QString currentTime = curDateTime.toString("yyyy_MM_dd_hh_mm_ss");
|
||||
QString Temperature300tcCSVPath = QDir::cleanPath(QString::fromStdString("/media/nvme/300TC/programRunLog/300tcTemperature") + QDir::separator() + "300tcTemperature_" + currentTime + ".csv");
|
||||
emit record300tcTemperatureSignal(Temperature300tcCSVPath);
|
||||
|
||||
std::cout<<"UdpServer::UdpServer--------:System ready!"<<std::endl;
|
||||
}
|
||||
@ -102,7 +123,7 @@ void UdpServer::processPendingDatagrams()
|
||||
case 2://关闭系统:关闭相机和sbg串口,关闭软件
|
||||
{
|
||||
std::cout<<"2代表关闭系统!"<<std::endl;
|
||||
|
||||
m_300tcTemperature->stopRecordTemperature();
|
||||
|
||||
if(m_sbgRecorder->getSbgState()>=1)
|
||||
{
|
||||
@ -167,8 +188,8 @@ void UdpServer::processPendingDatagrams()
|
||||
{
|
||||
if(m_imager->getImagerState()>=101 && m_imager->getImagerState()<=103)
|
||||
{
|
||||
float time = datagramList[1].toFloat();//ms
|
||||
m_imager->wrapSetExposureTime(time*1000);
|
||||
float time = datagramList[1].toFloat();//μs
|
||||
m_imager->wrapSetExposureTime(time);
|
||||
std::cout<<"7，手动设置曝光时间为：" << time <<std::endl;
|
||||
}
|
||||
|
||||
@ -197,8 +218,30 @@ void UdpServer::processPendingDatagrams()
|
||||
|
||||
break;
|
||||
}
|
||||
case 10:
|
||||
{
|
||||
std::cout<<"10代表!"<<std::endl;
|
||||
|
||||
int sbgState,sbgSolutionMode,ximeaState,frameRate,exposeMaxValueOfOneFrame;
|
||||
float exposeTime;
|
||||
|
||||
stateOf300tc.getSbgState(sbgState);
|
||||
stateOf300tc.getSbgSolutionMode(sbgSolutionMode);
|
||||
stateOf300tc.getXimeaState(ximeaState);
|
||||
stateOf300tc.getFrameRate(frameRate);
|
||||
stateOf300tc.getExposeTime(exposeTime);
|
||||
stateOf300tc.getExposeMaxValueOfOneFrame(exposeMaxValueOfOneFrame);
|
||||
|
||||
sendSerialPortStatus(sbgState);
|
||||
sendSbgSolutionModeState(sbgSolutionMode);
|
||||
sendXimeaImageStatus(ximeaState);
|
||||
sendXimeaAutoExposeMaxValueOfOneFrame(exposeMaxValueOfOneFrame, exposeTime);
|
||||
sendXimeaImageFrameRate(frameRate);
|
||||
|
||||
break;
|
||||
}
|
||||
default:
|
||||
std::cout<<">=9没有意义!"<<std::endl;
|
||||
std::cout<<">=11没有意义!"<<std::endl;
|
||||
|
||||
break;
|
||||
}
|
||||
@ -260,6 +303,8 @@ void UdpServer::sendSerialPortStatus(int serialPortStatus)
|
||||
|
||||
datagram2send.operator =(status.toStdString().c_str());
|
||||
m_udpSocket->writeDatagram(datagram2send.data(),datagram2send.size(),m_clientIpAddress, 45455);
|
||||
|
||||
stateOf300tc.setSbgState(serialPortStatus);
|
||||
}
|
||||
|
||||
void UdpServer::sendSbgSolutionModeState(int SolutionMode)
|
||||
@ -272,6 +317,8 @@ void UdpServer::sendSbgSolutionModeState(int SolutionMode)
|
||||
|
||||
datagram2send.operator =(status.toStdString().c_str());
|
||||
m_udpSocket->writeDatagram(datagram2send.data(),datagram2send.size(),m_clientIpAddress, 45455);
|
||||
|
||||
stateOf300tc.setSbgSolutionMode(SolutionMode);
|
||||
}
|
||||
|
||||
void UdpServer::sendSbgAccuracyState(int Accuracy,int SatelliteCounter)
|
||||
@ -296,11 +343,13 @@ void UdpServer::sendXimeaImageStatus(int ximeaImageStatus)
|
||||
|
||||
datagram2send.operator =(status.toStdString().c_str());
|
||||
m_udpSocket->writeDatagram(datagram2send.data(),datagram2send.size(),m_clientIpAddress, 45455);
|
||||
|
||||
stateOf300tc.setXimeaState(ximeaImageStatus);
|
||||
}
|
||||
|
||||
void UdpServer::sendXimeaAutoExposeMaxValueOfOneFrame(int autoExposeMaxValueOfOneFrame, double exposeTime)
|
||||
{
|
||||
// std::cout<<"UdpServer::sendXimeaAutoExposeMaxValueOfOneFrame---------------------:"<< ximeaImageStatus <<std::endl;
|
||||
// std::cout<<"UdpServer::sendXimeaAutoExposeMaxValueOfOneFrame---------------------:"<< exposeTime << " " << autoExposeMaxValueOfOneFrame <<std::endl;
|
||||
|
||||
QByteArray datagram2send;
|
||||
|
||||
@ -308,6 +357,9 @@ void UdpServer::sendXimeaAutoExposeMaxValueOfOneFrame(int autoExposeMaxValueOfOn
|
||||
|
||||
datagram2send.operator =(status.toStdString().c_str());
|
||||
m_udpSocket->writeDatagram(datagram2send.data(),datagram2send.size(),m_clientIpAddress, 45455);
|
||||
|
||||
stateOf300tc.setExposeTime(exposeTime);
|
||||
stateOf300tc.setExposeMaxValueOfOneFrame(autoExposeMaxValueOfOneFrame);
|
||||
}
|
||||
|
||||
void UdpServer::sendXimeaBinState(int spatialBin, int spectralBin)
|
||||
@ -332,6 +384,8 @@ void UdpServer::sendXimeaImageFrameRate(double frameRate)
|
||||
|
||||
datagram2send.operator =(status.toStdString().c_str());
|
||||
m_udpSocket->writeDatagram(datagram2send.data(),datagram2send.size(),m_clientIpAddress, 45455);
|
||||
|
||||
stateOf300tc.setFrameRate(frameRate);
|
||||
}
|
||||
|
||||
void UdpServer::sendCopyFileStatus(int fileStatus)
|
||||
@ -350,3 +404,199 @@ void UdpServer::onRecordFinished()
|
||||
{
|
||||
std::cout<<"UdpServer::onRecordFinished----------------:影像停止采集"<<std::endl;
|
||||
}
|
||||
|
||||
Record300TcTemperature::Record300TcTemperature()
|
||||
{
|
||||
m_bIsRecord = true;
|
||||
}
|
||||
|
||||
void Record300TcTemperature::stopRecordTemperature()
|
||||
{
|
||||
m_bIsRecord = false;
|
||||
}
|
||||
|
||||
void Record300TcTemperature::recordTemperature(QString filePath= nullptr)
|
||||
{
|
||||
if(filePath== nullptr)
|
||||
filePath="300tcTemperature.csv";
|
||||
QList<QString> fileInfo = getFileInfo(filePath);
|
||||
bool ret = createDir(fileInfo[0]);
|
||||
|
||||
ofstream temperatureFile300Tc(filePath.toStdString().c_str(),ios::app);
|
||||
int counter = 0;
|
||||
|
||||
m_serial=new QSerialPort();
|
||||
|
||||
if(m_serial->isOpen())//如果串口已经打开了 先给他关闭了
|
||||
{
|
||||
m_serial->clear();
|
||||
m_serial->close();
|
||||
}
|
||||
|
||||
QString portname = "ttyUSB2";
|
||||
|
||||
m_serial->setPortName(portname);
|
||||
m_serial->open(QIODevice::ReadWrite);
|
||||
|
||||
bool x=m_serial->setBaudRate(9600);
|
||||
|
||||
QByteArray requestData;
|
||||
while (m_bIsRecord)
|
||||
{
|
||||
counter++;
|
||||
if(m_serial->waitForReadyRead(30000))
|
||||
{
|
||||
requestData = m_serial->readAll();
|
||||
|
||||
float temp = parseMessage(&requestData);
|
||||
if (temp == -10000)
|
||||
{
|
||||
continue;
|
||||
}
|
||||
|
||||
QDateTime curDateTime = QDateTime::currentDateTime();
|
||||
QString currentTime = curDateTime.toString("yyyy/MM/dd hh:mm:ss");
|
||||
|
||||
temperatureFile300Tc << currentTime.toStdString() << "," << temp << "\n";
|
||||
|
||||
// if(temp > 80)
|
||||
// {
|
||||
// system("/home/300tc/projects/udpClient/udpClient 127.0.0.1 9,0");
|
||||
// }
|
||||
// if(temp > 90)
|
||||
// {
|
||||
// system("/home/300tc/projects/udpClient/udpClient 127.0.0.1 2");
|
||||
// }
|
||||
//
|
||||
if (counter % 60 == 0)
|
||||
{
|
||||
temperatureFile300Tc.flush();
|
||||
}
|
||||
sleep(1);
|
||||
}
|
||||
else
|
||||
{
|
||||
// std::cout<<"Record300TcTemperature::recordTemperature--Wait response timeout."<<std::endl;
|
||||
}
|
||||
}
|
||||
m_bIsRecord = true;
|
||||
|
||||
temperatureFile300Tc.close();
|
||||
}
|
||||
|
||||
float Record300TcTemperature::parseMessage(QByteArray * data)
|
||||
{
|
||||
float Lux;
|
||||
float T;
|
||||
float P;
|
||||
float Hum;
|
||||
float H;
|
||||
bool hasTemperature = false;
|
||||
|
||||
// qDebug() << "Received data(hex): " << data->toHex();
|
||||
|
||||
// for (int i = 0; i < data->length(); ++i)
|
||||
// {
|
||||
// qDebug() << i << ":" << (int)data->at(i);
|
||||
// }
|
||||
|
||||
QByteArray target = QByteArray::fromHex("5a5a"); // The pattern to search for
|
||||
|
||||
int offset_tmp = data->indexOf(target);
|
||||
QList<int> offsets;
|
||||
|
||||
while (offset_tmp != -1)
|
||||
{
|
||||
offsets.append(offset_tmp);
|
||||
offset_tmp = data->indexOf(target, offset_tmp + target.size());
|
||||
}
|
||||
|
||||
if (offsets.isEmpty())
|
||||
{
|
||||
// qDebug() << "5A5A not found in the data.";
|
||||
return -10000;
|
||||
}
|
||||
// qDebug() << "5A5A found at offsets:" << offsets;
|
||||
|
||||
for (int i = 0; i < offsets.length(); ++i)
|
||||
{
|
||||
int offset = offsets.at(i);
|
||||
|
||||
int messageClass = data->at(offset + 2);
|
||||
int messageCount = data->at(offset + 3) + 5;
|
||||
// qDebug() << "messageCount:" << messageCount;
|
||||
if (offset + messageCount - 1 > data->length() - 1)
|
||||
{
|
||||
qDebug() << i << ": Some data of this frame is missing! Discarding the frame.";
|
||||
continue;
|
||||
}
|
||||
|
||||
|
||||
int checksum = 0;
|
||||
QList<int> numbers;
|
||||
for (int j = 0; j < messageCount - 1; ++j)
|
||||
{
|
||||
if (i==offsets.length()-1 && j== messageCount-1)
|
||||
{
|
||||
int sdf=1;
|
||||
}
|
||||
|
||||
int index = offset + j;
|
||||
|
||||
if (index >= data->length())//Some data of this frame is missing! Discarding the frame.
|
||||
{
|
||||
checksum = -1;
|
||||
break;
|
||||
}
|
||||
int tmp = static_cast<int>(data->at(index));
|
||||
numbers.append(tmp);
|
||||
checksum += tmp;
|
||||
}
|
||||
// qDebug() << "jiaqilai:" << numbers;
|
||||
|
||||
if (checksum == -1)
|
||||
{
|
||||
qDebug() << "Some data of this frame is missing! Discarding the frame.";
|
||||
continue;
|
||||
}
|
||||
|
||||
if ((checksum & 0xFF) != data->at(offset + messageCount - 1))
|
||||
{
|
||||
qDebug() << "Checksum mismatch! Discarding the frame.";
|
||||
continue;
|
||||
}
|
||||
|
||||
//parse data
|
||||
switch (messageClass)
|
||||
{
|
||||
case 0x15:
|
||||
Lux = (float)((data->at(offset + 4)<<24)|(data->at(offset + 5)<<16)|(data->at(offset + 6)<<8)|data->at(offset + 7)) / 100;
|
||||
// qDebug() << "lux: " << Lux;
|
||||
break;
|
||||
case 0x45:
|
||||
T = (float)((data->at(offset + 4)<<8)|data->at(offset + 5)) / 100;
|
||||
P = (float)((data->at(offset + 6)<<24)|(data->at(offset + 7)<<16)|(data->at(offset + 8)<<8)|data->at(offset + 9)) / 100;
|
||||
Hum = (float)((data->at(offset + 10)<<8) | data->at(offset + 11)) / 100;
|
||||
H = (float)((data->at(offset + 12)<<8)|data->at(offset + 13));
|
||||
|
||||
hasTemperature = true;
|
||||
|
||||
// qDebug() << "T: " << T;
|
||||
// qDebug() << "P: " << P;
|
||||
// qDebug() << "Hum: " << Hum;
|
||||
// qDebug() << "H: " << H;
|
||||
break;
|
||||
default : // 可选的
|
||||
;
|
||||
|
||||
}
|
||||
}
|
||||
if (hasTemperature)
|
||||
{
|
||||
return T;
|
||||
}
|
||||
else
|
||||
{
|
||||
return -10000;
|
||||
}
|
||||
}
|
||||
|
||||
@ -47,6 +47,12 @@ XimeaImager::XimeaImager()
|
||||
writeData2DiskThread->start(QThread::HighestPriority);
|
||||
connect(this, SIGNAL(startWriteDiskSignal()), writeData2Disk, SLOT(write2Disk()));
|
||||
|
||||
m_pushFlowThread=new QThread();
|
||||
m_pushFlow = new PushFlow();
|
||||
m_pushFlow->moveToThread(m_pushFlowThread);
|
||||
m_pushFlowThread->start();
|
||||
connect(this, SIGNAL(startPushFlowSignal()), m_pushFlow, SLOT(encodePushFlow()));
|
||||
|
||||
m_pool = new MemoryPool<DataBuffer>;
|
||||
q = new queue<DataBuffer *>;
|
||||
m_qFrameCounter = new queue<int>;
|
||||
@ -100,8 +106,21 @@ void XimeaImager::openImger()
|
||||
ret = m_configfile.getEffectiveWindow(width, offsetx, height, offsety);
|
||||
if (ret)
|
||||
{
|
||||
int rgbHeight;
|
||||
int framerateVideo;
|
||||
m_configfile.getPushFlowParam(m_iFlowSwitch, rgbHeight, framerateVideo);
|
||||
std::cout <<"rgbHeight：" << rgbHeight << ", framerateVideo：" << framerateVideo << std::endl;
|
||||
|
||||
m_imager.setEffectiveWindow(offsetx, width, offsety, height);
|
||||
m_rgbImage->SetRgbImageWidthAndHeight(height, width, 20);
|
||||
m_rgbImage->SetRgbImageWidthAndHeight(height, width, rgbHeight);
|
||||
m_pushFlow->setParm(m_rgbImage,width,rgbHeight,framerateVideo);
|
||||
|
||||
int redBandNumber;
|
||||
int greenBandNumber;
|
||||
int blueBandNumber;
|
||||
getRgbBandNumber(redBandNumber, greenBandNumber, blueBandNumber);
|
||||
m_rgbImage->SetRgbBandNumber(redBandNumber, greenBandNumber, blueBandNumber);
|
||||
|
||||
std::cout<<"height："<< height <<std::endl;
|
||||
std::cout<<"width："<< width <<std::endl;
|
||||
std::cout<<"每帧字节数："<< width * height * 2 <<std::endl;
|
||||
@ -487,6 +506,65 @@ double XimeaImager::geWavelengthAtBand(int x)
|
||||
}
|
||||
}
|
||||
|
||||
void XimeaImager::getRgbBandNumber(int &redBandNumber, int &greenBandNumber, int &blueBandNumber)
|
||||
{
|
||||
vector<double> wavelengths;
|
||||
if (m_imager.getSpectralBin() == 1)
|
||||
{
|
||||
for (int i = getWindowStartBand(); i < getWindowEndBand(); i++)
|
||||
{
|
||||
wavelengths.push_back(geWavelengthAtBand(i));
|
||||
}
|
||||
}
|
||||
else if (m_imager.getSpectralBin() == 2)
|
||||
{
|
||||
for (int i = m_iOffsetyOfSpectralBin2; i < m_iOffsetyOfSpectralBin2 + m_iHeightOfSpectralBin2; i++)
|
||||
{
|
||||
if (i*2 + 1 > m_iOffsetyOfSpectralBin1 + m_iHeightOfSpectralBin1)
|
||||
{
|
||||
printf("XimeaImager::writeHdr 出现错误：窗口中，光谱 bin1 波段数小于 bin2 的 2 倍。\n");
|
||||
break;
|
||||
}
|
||||
wavelengths.push_back((geWavelengthAtBand(i*2) + geWavelengthAtBand(i*2 + 1)) / 2);
|
||||
}
|
||||
}
|
||||
|
||||
//envi打开文件时的红绿蓝波长（nm）
|
||||
int r_envi = 640;
|
||||
int g_envi = 550;
|
||||
int b_envi = 470;
|
||||
redBandNumber = findClosestIndex(wavelengths, r_envi);
|
||||
greenBandNumber = findClosestIndex(wavelengths, g_envi);
|
||||
blueBandNumber = findClosestIndex(wavelengths, b_envi);
|
||||
|
||||
// std::cout<<"红波段的波段号："<< redBandNumber <<std::endl;
|
||||
// std::cout<<"绿波段的波段号："<< greenBandNumber <<std::endl;
|
||||
// std::cout<<"蓝波段的波段号："<< blueBandNumber <<std::endl;
|
||||
}
|
||||
|
||||
int XimeaImager::findClosestIndex(const std::vector<double>& numbers, double target)
|
||||
{
|
||||
if (numbers.empty()) {
|
||||
// 处理空向量的情况
|
||||
return -1;
|
||||
}
|
||||
|
||||
double minDifference = std::abs(numbers[0] - target);
|
||||
int closestIndex = 0;
|
||||
|
||||
for (int i = 1; i < numbers.size(); ++i)
|
||||
{
|
||||
double currentDifference = std::abs(numbers[i] - target);
|
||||
if (currentDifference < minDifference)
|
||||
{
|
||||
minDifference = currentDifference;
|
||||
closestIndex = i;
|
||||
}
|
||||
}
|
||||
|
||||
return closestIndex;
|
||||
}
|
||||
|
||||
int XimeaImager::getMaxValueOfOneFrame(unsigned short * data, int numberOfPixel)
|
||||
{
|
||||
//排序
|
||||
@ -577,7 +655,7 @@ void XimeaImager::startRecord(double TimeDifferenceBetweensOSAndSbg,QString base
|
||||
|
||||
int number_WriteDisk = 100;
|
||||
writeData2Disk->setParm(q, m_qFrameCounter,m_baseFileName,m_iFrameSizeInByte, number_WriteDisk, m_pool, m_rgbImage);
|
||||
emit startWriteDiskSignal();
|
||||
// emit startWriteDiskSignal();
|
||||
|
||||
int indexofbuff;
|
||||
DataBuffer * buffer;
|
||||
@ -588,6 +666,17 @@ void XimeaImager::startRecord(double TimeDifferenceBetweensOSAndSbg,QString base
|
||||
double timeDifferenceBetweenSbgAndXimea;
|
||||
double * sbgTimeBuffer = new double[number_WriteDisk];
|
||||
|
||||
QString imageFileName=m_baseFileName+".bil";
|
||||
FILE *hFile=fopen(imageFileName.toStdString().c_str(),"w+b");
|
||||
double * imageBuffer = new double[number_WriteDisk];
|
||||
|
||||
QString vedioFileName=m_baseFileName+".h264";
|
||||
m_pushFlow->setVedioFilePath(vedioFileName);
|
||||
if(m_iFlowSwitch == 1)
|
||||
{
|
||||
emit startPushFlowSignal();
|
||||
}
|
||||
|
||||
m_imager.start();
|
||||
struct timeval timeStart, timeEnd;
|
||||
double runTime=0;
|
||||
@ -601,34 +690,26 @@ void XimeaImager::startRecord(double TimeDifferenceBetweensOSAndSbg,QString base
|
||||
{
|
||||
timeDifferenceBetweenSbgAndXimea = calculateTimeDifferenceBetweenSbgAndximea(&m_imager.m_image, TimeDifferenceBetweensOSAndSbg);
|
||||
}
|
||||
fwrite(m_imager.m_image.bp,1,m_iFrameSizeInByte, hFile);
|
||||
//构造rgb图像，用于推流到m300遥控器
|
||||
if(m_iFlowSwitch == 1)
|
||||
{
|
||||
m_rgbImage->FillRgbImage((unsigned short *)m_imager.m_image.bp);
|
||||
}
|
||||
|
||||
indexofbuff = m_iFrameCounter % number_WriteDisk;
|
||||
|
||||
if (indexofbuff == 1)
|
||||
{
|
||||
r_qtx.lock();
|
||||
buffer = m_pool->newElement();
|
||||
r_qtx.unlock();
|
||||
}
|
||||
|
||||
if (indexofbuff == 0)
|
||||
{
|
||||
memcpy((void *)buffer->data + (number_WriteDisk - 1) * m_iFrameSizeInByte,m_imager.m_image.bp,m_iFrameSizeInByte);
|
||||
sbgTimeBuffer[number_WriteDisk - 1] = getSbgTime(&m_imager.m_image, timeDifferenceBetweenSbgAndXimea);
|
||||
}
|
||||
else
|
||||
{
|
||||
memcpy((void *)buffer->data + (indexofbuff - 1) * m_iFrameSizeInByte,m_imager.m_image.bp,m_iFrameSizeInByte);
|
||||
sbgTimeBuffer[indexofbuff - 1] = getSbgTime(&m_imager.m_image, timeDifferenceBetweenSbgAndXimea);
|
||||
}
|
||||
|
||||
if (indexofbuff == 0)
|
||||
{
|
||||
r_qtx.lock();
|
||||
q->push(buffer);
|
||||
m_qFrameCounter->push(number_WriteDisk);
|
||||
r_qtx.unlock();
|
||||
|
||||
for (int i = 0; i < number_WriteDisk; ++i)
|
||||
{
|
||||
fprintf(hHimesFile,"%f\n",sbgTimeBuffer[i]);
|
||||
@ -637,11 +718,6 @@ void XimeaImager::startRecord(double TimeDifferenceBetweensOSAndSbg,QString base
|
||||
}
|
||||
if (indexofbuff != 0)
|
||||
{
|
||||
r_qtx.lock();
|
||||
q->push(buffer);
|
||||
m_qFrameCounter->push(indexofbuff);
|
||||
r_qtx.unlock();
|
||||
|
||||
for (int i = 0; i < indexofbuff; ++i)
|
||||
{
|
||||
fprintf(hHimesFile,"%f\n",sbgTimeBuffer[i]);
|
||||
@ -655,6 +731,8 @@ void XimeaImager::startRecord(double TimeDifferenceBetweensOSAndSbg,QString base
|
||||
writeData2Disk->exitWriteData2Disk();
|
||||
writeHdr();
|
||||
|
||||
m_pushFlow->exitPushFlow();
|
||||
|
||||
delete[] sbgTimeBuffer;
|
||||
|
||||
double frameInTheory=runTime * getFramerate();
|
||||
@ -672,6 +750,7 @@ void XimeaImager::startRecord(double TimeDifferenceBetweensOSAndSbg,QString base
|
||||
std::cout<<"丢帧率为: "<<frameLossRate * 100<< "%" <<std::endl;
|
||||
|
||||
fclose(hHimesFile);
|
||||
fclose(hFile);
|
||||
// timesFileHandle.close();
|
||||
// timesFile.close();
|
||||
|
||||
@ -944,17 +1023,36 @@ void RecordXimeaTemperature::recordTemperature(QString filePath= nullptr)
|
||||
float temp = m_imager->getTemperature();
|
||||
if(temp > 80 && m_ximeaImager->getImagerState() == 104)
|
||||
{
|
||||
ximeaTemperatureFile.flush();
|
||||
system("/home/300tc/projects/udpClient/udpClient 127.0.0.1 9,0");
|
||||
}
|
||||
if(temp > 90)
|
||||
{
|
||||
ximeaTemperatureFile.flush();
|
||||
system("/home/300tc/projects/udpClient/udpClient 127.0.0.1 2");
|
||||
}
|
||||
|
||||
// 获取剩余硬盘空间和剩余采集时间
|
||||
FILE *fp;
|
||||
char buffer[128];
|
||||
fp = popen("cat /sys/devices/virtual/thermal/thermal_zone1/temp", "r");
|
||||
if (fp == NULL) {
|
||||
perror("popen");
|
||||
}
|
||||
|
||||
// 读取输出并处理
|
||||
unsigned long long temper;
|
||||
if (fgets(buffer, sizeof(buffer), fp) != NULL)
|
||||
{
|
||||
temper = strtoull(buffer, NULL, 10);
|
||||
// printf("CPU温度: %.2f;\n", (float)temper/1000);
|
||||
}
|
||||
pclose(fp);
|
||||
|
||||
QDateTime curDateTime = QDateTime::currentDateTime();
|
||||
QString currentTime = curDateTime.toString("yyyy/MM/dd hh:mm:ss");
|
||||
|
||||
ximeaTemperatureFile << currentTime.toStdString() << "," << temp << "\n";
|
||||
ximeaTemperatureFile << currentTime.toStdString() << "," << temp << "," << (float)temper/1000 << "\n";
|
||||
|
||||
// std::cout<<"RecordXimeaTemperature::recordTemperature----------------:ximea Temperature is "<< temp <<std::endl;
|
||||
// std::cout<<"RecordXimeaTemperature::recordTemperature----------------:ximea state is "<< m_ximeaImager->getImagerState() <<std::endl;
|
||||
@ -1019,7 +1117,7 @@ void WriteData2Disk::write2Disk()
|
||||
r_qtx.unlock();
|
||||
|
||||
//构造rgb图像，用于推流到m300遥控器
|
||||
// m_rgbImage->FillRgbImage(dataBuffer);
|
||||
m_rgbImage->FillRgbImage(dataBuffer);
|
||||
|
||||
|
||||
// std::cout<<"WriteData2Disk::write2Disk-----------------------正在写磁盘!" << m_pool->max_size() <<std::endl;//
|
||||
@ -1055,3 +1153,213 @@ void WriteData2Disk::setParm(queue<DataBuffer *> * q, queue<int> * qFrameCounter
|
||||
|
||||
m_rgbImage = rgbImage;
|
||||
}
|
||||
|
||||
PushFlow::PushFlow()
|
||||
{
|
||||
m_iWidth = 1368;
|
||||
m_iHeight = 720;
|
||||
m_iFramerateVideo = 5;
|
||||
}
|
||||
|
||||
void PushFlow::setParm(rgbImage * img, int width, int height, int framerateVideo)
|
||||
{
|
||||
m_rgbImage = img;
|
||||
|
||||
m_iWidth = width;
|
||||
m_iHeight = height;
|
||||
m_iFramerateVideo = framerateVideo;
|
||||
}
|
||||
|
||||
void PushFlow::setVedioFilePath(QString path)
|
||||
{
|
||||
m_QVedioFilePath = path;
|
||||
}
|
||||
|
||||
void PushFlow::exitPushFlow()
|
||||
{
|
||||
isExitPushFlow = true;
|
||||
}
|
||||
|
||||
void PushFlow::encodePushFlow()
|
||||
{
|
||||
// 创建输出视频的AVFormatContext
|
||||
const char* outputVideoPath = "/media/nvme/delete/300tc.h264";
|
||||
|
||||
FILE *fp = fopen(m_QVedioFilePath.toStdString().c_str(),"w+b");
|
||||
|
||||
AVFormatContext* formatContext = nullptr;
|
||||
avformat_alloc_output_context2(&formatContext, nullptr, "mp4", outputVideoPath);
|
||||
if (!formatContext)
|
||||
{
|
||||
qDebug() << "Error: Failed to allocate output context";
|
||||
return;
|
||||
}
|
||||
|
||||
// 查找H.264编码器
|
||||
const AVCodec* codec = avcodec_find_encoder(AV_CODEC_ID_H264);
|
||||
if (!codec)
|
||||
{
|
||||
qDebug() << "Error: Codec not found";
|
||||
avformat_free_context(formatContext);
|
||||
return;
|
||||
}
|
||||
|
||||
// 创建输出视频流
|
||||
AVStream* videoStream = avformat_new_stream(formatContext, codec);
|
||||
if (!videoStream)
|
||||
{
|
||||
qDebug() << "Error: Failed to create video stream";
|
||||
avformat_free_context(formatContext);
|
||||
return;
|
||||
}
|
||||
// 设置视频流的参数，例如分辨率、帧率等
|
||||
videoStream->codecpar->width = m_iWidth;
|
||||
videoStream->codecpar->height = m_iHeight;
|
||||
videoStream->codecpar->codec_id = AV_CODEC_ID_H264; // 设置为H.264编解码器
|
||||
videoStream->codecpar->format = AV_PIX_FMT_YUV420P; // 设置为YUV420P像素格式
|
||||
|
||||
// 配置视频流的参数
|
||||
AVCodecContext* codecContext = avcodec_alloc_context3(codec);
|
||||
if (!codecContext)
|
||||
{
|
||||
qDebug() << "Error: Failed to allocate codec context";
|
||||
avformat_free_context(formatContext);
|
||||
return;
|
||||
}
|
||||
|
||||
// 设置视频流的参数，例如分辨率、帧率等
|
||||
codecContext->width = m_iWidth;
|
||||
codecContext->height = m_iHeight;
|
||||
codecContext->time_base = {1, m_iFramerateVideo};
|
||||
codecContext->pix_fmt = AV_PIX_FMT_YUV420P; // 设置为YUV420P格式
|
||||
codecContext->gop_size = 50;//多少帧产生一组关键帧
|
||||
codecContext->max_b_frames = 0;//b帧，参考帧：设置为1会导致视频回退的现象
|
||||
// codecContext->bit_rate = 1000000; // 设置比特率为 1000000
|
||||
|
||||
// 打开视频编码器
|
||||
if (avcodec_open2(codecContext, codec, nullptr) < 0)
|
||||
{
|
||||
qDebug() << "Error: Failed to open codec";
|
||||
avcodec_free_context(&codecContext);
|
||||
avformat_free_context(formatContext);
|
||||
return;
|
||||
}
|
||||
|
||||
// 打开输出文件
|
||||
// if (avio_open(&formatContext->pb, outputVideoPath, AVIO_FLAG_WRITE) < 0)
|
||||
// {
|
||||
// qDebug() << "Error: Failed to open output file";
|
||||
// avcodec_close(codecContext);
|
||||
// avcodec_free_context(&codecContext);
|
||||
// avformat_free_context(formatContext);
|
||||
// return;
|
||||
// }
|
||||
|
||||
// 写入文件头
|
||||
// avformat_write_header(formatContext, nullptr);
|
||||
|
||||
// 使用sws_scale进行颜色空间转换
|
||||
SwsContext* swsContext = sws_getContext(m_iWidth, m_iHeight, AV_PIX_FMT_BGR24,
|
||||
m_iWidth, m_iHeight, AV_PIX_FMT_YUV420P,
|
||||
SWS_BICUBIC, nullptr, nullptr, nullptr);
|
||||
if (!swsContext)
|
||||
{
|
||||
qDebug() << "Error: Failed to create sws context";
|
||||
avio_closep(&formatContext->pb);
|
||||
avcodec_close(codecContext);
|
||||
avcodec_free_context(&codecContext);
|
||||
avformat_free_context(formatContext);
|
||||
return;
|
||||
}
|
||||
|
||||
// 创建 AVFrame 作为目标图像
|
||||
AVFrame* dstFrame = av_frame_alloc();
|
||||
av_image_alloc(dstFrame->data, dstFrame->linesize, m_iWidth, m_iHeight, AV_PIX_FMT_YUV420P, 1);
|
||||
|
||||
// 设置目标图像参数
|
||||
dstFrame->width = m_iWidth;
|
||||
dstFrame->height = m_iHeight;
|
||||
dstFrame->format = AV_PIX_FMT_YUV420P;
|
||||
|
||||
AVFrame* frame = av_frame_alloc();
|
||||
av_image_alloc(frame->data, frame->linesize, m_iWidth, m_iHeight, AV_PIX_FMT_BGR24, 1);
|
||||
|
||||
QUdpSocket * m_udpSocket = new QUdpSocket();
|
||||
m_udpSocket->bind(PUSH_FLOW_PORT, QUdpSocket::ShareAddress);
|
||||
QHostAddress m_clientIpAddress=QHostAddress(QHostAddress::LocalHost);
|
||||
// QHostAddress m_clientIpAddress("192.168.1.30");
|
||||
// QHostAddress m_clientIpAddress("192.168.111.1");
|
||||
int udpSendCounter=0;
|
||||
int encodeCounter=0;
|
||||
|
||||
isExitPushFlow = false;
|
||||
|
||||
unsigned long sleepTime = 1/(float)m_iFramerateVideo * 1000;
|
||||
std::cout<< "推流帧率: " << m_iFramerateVideo << ", sleepTime：" << sleepTime << "ms." << std::endl;
|
||||
|
||||
while(true)
|
||||
{
|
||||
QThread::msleep(sleepTime);
|
||||
|
||||
memcpy(frame->data[0], m_rgbImage->m_matRgbImage->data, m_rgbImage->m_matRgbImage->rows * m_rgbImage->m_matRgbImage->step[0]);
|
||||
// memcpy(frame->data[0], m_rgbImage->m_Qphoto.bits(), m_rgbImage->m_Qphoto.byteCount());
|
||||
|
||||
// 使用sws_scale进行颜色空间转换
|
||||
sws_scale(swsContext, frame->data, frame->linesize, 0, m_iHeight,
|
||||
dstFrame->data, dstFrame->linesize);
|
||||
dstFrame->pts = encodeCounter;
|
||||
|
||||
// 将AVFrame编码为视频帧
|
||||
AVPacket pkt;
|
||||
av_init_packet(&pkt);
|
||||
pkt.data = nullptr;
|
||||
pkt.size = 0;
|
||||
|
||||
if (avcodec_send_frame(codecContext, dstFrame) == 0 &&
|
||||
avcodec_receive_packet(codecContext, &pkt) == 0)
|
||||
{
|
||||
fwrite(pkt.data, 1, pkt.size, fp);
|
||||
|
||||
m_udpSocket->writeDatagram((const char *)pkt.data,pkt.size,m_clientIpAddress, PUSH_FLOW_PORT);
|
||||
|
||||
// std::cout << "编码第 " << udpSendCounter << " 帧数据大小: " << pkt.size << std::endl;
|
||||
// std::cout<< "pkt.pts: " << pkt.pts << std::endl;
|
||||
// std::cout<< "pkt.dts: " << pkt.dts << std::endl << std::endl;
|
||||
udpSendCounter++;
|
||||
|
||||
// 将编码后的帧写入文件
|
||||
// pkt.stream_index = videoStream->index;
|
||||
// av_interleaved_write_frame(formatContext, &pkt);
|
||||
// av_write_frame(formatContext, &pkt);
|
||||
|
||||
av_packet_unref(&pkt);
|
||||
}
|
||||
encodeCounter++;
|
||||
|
||||
if(isExitPushFlow)
|
||||
{
|
||||
std::cout<<"PushFlow::encodePushFlow-----------------------推流线程将退出！"<<std::endl;
|
||||
break;
|
||||
}
|
||||
}
|
||||
|
||||
fclose(fp);
|
||||
// 写入文件尾
|
||||
// av_write_trailer(formatContext);
|
||||
|
||||
// 释放AVFrame和相关资源
|
||||
av_freep(&frame->data[0]);
|
||||
av_frame_free(&frame);
|
||||
|
||||
// 释放资源
|
||||
sws_freeContext(swsContext);
|
||||
av_freep(&dstFrame->data[0]);
|
||||
av_frame_free(&dstFrame);
|
||||
// av_packet_free(&pkt);
|
||||
avcodec_close(codecContext);
|
||||
avcodec_free_context(&codecContext);
|
||||
avio_closep(&formatContext->pb);
|
||||
avformat_free_context(formatContext);
|
||||
|
||||
std::cout<<"PushFlow::encodePushFlow-----------------------推流线程已经退出！" << std::endl;
|
||||
}
|
||||
|
||||
Reference in New Issue
Block a user