处理场景:三台线阵相机或者多台其他传感器与工控机连着,通过调用相机的SDK控制相机,例如同时开关,同时从相机中获取数据。然后采集软件接收到相机数据后,将数据保存到硬盘。
处理的问题:如果三台相机发送数据频率很快,并且数据量很大的话,如何快速将数据写入到硬盘里或者对数据的其他处理,从而保证内存不增长?
解决方案简述:每收到一坨相机的数据,就将其从尾部压入队列;同时有另外一个线程在等着处理这个队列,只要这个队列有数据,就从头部弹出一个数据,然后对这个数据进行相应的处理。这里存在一个问题:如果相机频率太快,也就是说队列会涨得非常快,涨数据的速度大于处理数据的速度,内存就会上涨,直至崩溃。处理数据,例如将相机数据保存到硬盘,如果硬盘写数据的速度很慢,就会出现内存涨爆的情况,这个时候可以考虑用多线程进行处理,从队列弹出一个数据后,马上被一个线程接上,这样可以显著提升CPU的利用率,加快数据处理速度。
简要代码如下:
需特别注意代码中的注释。
//头文件代码
#include
#include
#include
#include
#include
#include "SThreadPool.h"
struct CameraData
{
int height;
int width;
int datasize;
std::unique_ptr data;
UINT64 timestamps;
};
class Sensor
{
public:
Sensor();
~Sensor();
//**获取相机数据的回调,并在回调中将相机数据压入容器中**//
friend int WINAPI s_ManagaGetImageCallBack(void* _context, void* _system, void* _data);
//从队列头部弹出数据并进行处理//
void t_ProcessDataQueue(void* para);
void SaveImage(std::shared_ptr data);
friend void t_SaveCameraImage(void* para, std::shared_ptr data);
private:
//用来存储三台相机数据的容器//
std::mutex m_DataMtx_;
std::queue> m_QueCamData_;
std::condition_variable m_DataVar_;
bool m_bQueExit_ = false;
std::shared_ptr m_Pool_;
std::future m_fuProcessDataQue_;
};
实现文件中,重点关注函数t_ProcessDataQueue(void* para)
;同时这段代码还包括一种有效退出类内线程的方式,请看析构函数部分。
//实现文件
Sensor::Sensor()
{
m_fuProcessDataQue_ = std::async(std::launch::async, &Sensor::t_ProcessDataQueue, this, nullptr);
unsigned long hardwares = std::thread::hardware_concurrency();
m_Pool_ = std::make_shared(hardwares);
}
Sensor::~Sensor()
{
{
std::unique_lock locker(m_DataMtx_);
m_bQueExit_ = true;
}
m_DataVar_.notify_one();
m_fuProcessDataQue_.get();
}
int WINAPI s_ManagaGetImageCallBack(void* _context, void* _system, void* _data)
{
Sensor* pThis = (Sensor*)_context;
std::shared_ptr data = std::make_shared();
//将从相机返回出来的数据组装成CameraData.....
{
std::unique_lock lock(pThis->m_DataMtx_);
pThis->m_QueCamData_.push(data);
}
pThis->m_DataVar_.notify_one();//通知另外一个线程进行数据的处理
return 0;
}
void Sensor::t_ProcessDataQueue(void* para)
{
std::shared_ptr tempdata;
std::unique_lock locker(m_DataMtx_);
while (true)
{
try
{
if (!m_bQueExit_) m_DataVar_.wait(locker);
locker.unlock();
while (true)
{
locker.lock();
if (m_QueCamData_.empty())
{
std::cout << "camera set queue is empty.\n";
break;
}
else
{
tempdata = m_QueCamData_.front();
m_QueCamData_.pop();
}
locker.unlock();
//process camera data//
//01 单线程方式存数据
SaveImage(tempdata);
//02 多线程的方式存数据//
m_Pool_->enqueue(t_SaveCameraImage, this, tempdata);
}
}
catch (const std::exception& e)
{
std::cout << e.what() << "\n";
}
if (m_bQueExit_)
{
break;
}
}
}
void Sensor::SaveImage(std::shared_ptr data)
{
//保存相机
//do want you want
}
void t_SaveCameraImage(void* para, std::shared_ptr data)
{
//保存相机数据
//do want you want
}
作为数据采集程序,这段代码非常适用。任何传感器的数据我都能用一个容器先接着,然后另外一个线程处理从这个容器中弹出来的数据。如果采用回调的方式获取传感器数据,这种方式还不堵塞回调。既然作为一种通用的方式,这段代码当然可以用模板类进行改进。笔者尝试着写了一个模板类,目前测试有效,下一篇文章公布模板代码。
欢迎大牛们与我进行交流探讨,如果对上述代码有疑问或者缺少某个头文件(如SThreadPool,这是一个线程池模板类),请扫码加我微信,备注“C++”。