|
|
|
|
@ -1,4 +1,4 @@
|
|
|
|
|
#include "workthread.h"
|
|
|
|
|
#include "SyncWorkThread.h"
|
|
|
|
|
#include "alg_jd.h"
|
|
|
|
|
#include "common.h"
|
|
|
|
|
#include "balluffcamera.h"
|
|
|
|
|
@ -15,6 +15,7 @@ extern DisplayLabelConf g_display_label_conf[NumberOfSupportedCameras];
|
|
|
|
|
|
|
|
|
|
extern int rotationAngle[NumberOfSupportedCameras]; //图片旋转角度
|
|
|
|
|
extern bool isNeedRotate[NumberOfSupportedCameras];
|
|
|
|
|
extern int work_camera_nums;
|
|
|
|
|
|
|
|
|
|
extern SyncQueue<std::pair<std::string, cv::Mat> >* g_save_queue; //图片保存队列
|
|
|
|
|
extern SyncQueue<std::pair<int, cv::Mat>>* g_image_queue[NumberOfSupportedCameras]; //int表示一个目标拍了几张
|
|
|
|
|
@ -44,8 +45,11 @@ void SyncWorkThread::init(SyncQueue<std::vector<std::pair<int, cv::Mat>>>* image
|
|
|
|
|
local_g_image_sync_queue = image_ptr;
|
|
|
|
|
local_g_result_queue = result_ptr;
|
|
|
|
|
b_quit = false;
|
|
|
|
|
frame_total = 0;
|
|
|
|
|
for (int i = 0; i < NumberOfSupportedCameras; i++) {
|
|
|
|
|
frame_total[i] = 0;
|
|
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
void SyncWorkThread::start_work()
|
|
|
|
|
{
|
|
|
|
|
start(HighestPriority);
|
|
|
|
|
@ -63,15 +67,16 @@ void SyncWorkThread::run()
|
|
|
|
|
QDateTime now_ts = QDateTime::currentDateTime();
|
|
|
|
|
std::vector<std::pair<int, cv::Mat>> element_vec;
|
|
|
|
|
local_g_image_sync_queue->take(element_vec);
|
|
|
|
|
|
|
|
|
|
bool IsNGForAll = false;
|
|
|
|
|
int merge_index;
|
|
|
|
|
cv::Mat merge_image = cv::Mat::zeros(512 * work_camera_nums, 640 * g_sys_conf.shoot[0], CV_8UC3);
|
|
|
|
|
for (int i = 0; i < element_vec.size(); i++)//每个相机的图轮流遍历
|
|
|
|
|
{
|
|
|
|
|
local_camera_number = i;
|
|
|
|
|
std::pair<int, cv::Mat> element;
|
|
|
|
|
element = element_vec[i];
|
|
|
|
|
int unit_count = element.first;
|
|
|
|
|
cv::Mat image = element.second;
|
|
|
|
|
|
|
|
|
|
{//不要删掉这个括号，用来定义锁的作用域
|
|
|
|
|
std::lock_guard<std::mutex> locker(g_sys_conf.lock);
|
|
|
|
|
local_SysConf.save = g_sys_conf.save;
|
|
|
|
|
@ -111,9 +116,11 @@ void SyncWorkThread::run()
|
|
|
|
|
}
|
|
|
|
|
if (local_SysConf.shoot[local_camera_number] == unit_count)
|
|
|
|
|
{
|
|
|
|
|
cv::Rect roi;
|
|
|
|
|
std::vector<cv::Mat> vec_in;
|
|
|
|
|
int w = image.cols;
|
|
|
|
|
int h = image.rows / unit_count;
|
|
|
|
|
|
|
|
|
|
for (int index = 0; index < unit_count; index++) {
|
|
|
|
|
cv::Rect temp_Rect(0, h * index, w, h);
|
|
|
|
|
cv::Mat temp_image = image(temp_Rect).clone();
|
|
|
|
|
@ -136,12 +143,15 @@ void SyncWorkThread::run()
|
|
|
|
|
alg_jd[local_camera_number].detect(imagein, imageout, results);
|
|
|
|
|
vec_out.push_back(imageout.clone());
|
|
|
|
|
vec_results.push_back(results);
|
|
|
|
|
}else{
|
|
|
|
|
}
|
|
|
|
|
else {
|
|
|
|
|
alg_jd[local_camera_number].detect_batch(vec_in, vec_out, vec_results);
|
|
|
|
|
}
|
|
|
|
|
QDateTime ts_jd = QDateTime::currentDateTime();
|
|
|
|
|
int time_process = ts_start.msecsTo(ts_jd);
|
|
|
|
|
#ifdef SYNC_CAMERA
|
|
|
|
|
emit display_timecost(local_camera_number, time_process);
|
|
|
|
|
#endif
|
|
|
|
|
#ifdef __UDPSend
|
|
|
|
|
UDPSendInfo.timecost = QString::number(time_process);
|
|
|
|
|
#endif
|
|
|
|
|
@ -153,7 +163,9 @@ void SyncWorkThread::run()
|
|
|
|
|
jd_no += QString::number(vec_results[index].size()) + ",";
|
|
|
|
|
}
|
|
|
|
|
jd_no.chop(1);
|
|
|
|
|
#ifdef SYNC_CAMERA
|
|
|
|
|
emit display_jd_no(local_camera_number, jd_no);
|
|
|
|
|
#endif
|
|
|
|
|
#ifdef __UDPSend
|
|
|
|
|
UDPSendInfo.JD = jd_no;
|
|
|
|
|
#endif
|
|
|
|
|
@ -164,8 +176,7 @@ void SyncWorkThread::run()
|
|
|
|
|
ng_reason_maps[1] = "less_than_setting";
|
|
|
|
|
ng_reason_maps[2] = "too_diff_from_model";
|
|
|
|
|
ng_reason_maps[3] = "out_of_setting_range";
|
|
|
|
|
for(int index=0;index<unit_count;index++)
|
|
|
|
|
{
|
|
|
|
|
for (int index = 0; index < unit_count; index++){
|
|
|
|
|
if (vec_results[index].size() < local_SysConf.no[local_camera_number][index])
|
|
|
|
|
{
|
|
|
|
|
IsNG |= true;
|
|
|
|
|
@ -179,20 +190,26 @@ void SyncWorkThread::run()
|
|
|
|
|
}
|
|
|
|
|
if (local_SysConf.save == 2)//三张照片分别存储
|
|
|
|
|
{
|
|
|
|
|
/// ºÏ³Éelement_vec.size() * unit_count ¹¬¸ñͼÏñ
|
|
|
|
|
cv::Mat m = vec_in[index].clone();
|
|
|
|
|
QString file_name = g_conf_path.save_pics_path + "/ALL/" +
|
|
|
|
|
now_ts.toString("yyyy-MM-dd") + "/"
|
|
|
|
|
+ QString::number(local_camera_number + 1) + "/" + QString::number(index + 1) + "/" +
|
|
|
|
|
now_ts.toString("yyyy-MM-dd_HH-mm-ss_zzz_") + QString::number(local_camera_number + 1) +
|
|
|
|
|
"#" + "_" + QString::number(index + 1) + ".jpg";
|
|
|
|
|
std::string filename = file_name.toLocal8Bit().constData();
|
|
|
|
|
g_save_queue->put(std::make_pair(filename, m));
|
|
|
|
|
QString file_name;
|
|
|
|
|
merge_index = i * unit_count + index + 1;
|
|
|
|
|
|
|
|
|
|
roi = cv::Rect(index * m.cols, i * m.rows, m.cols, m.rows);
|
|
|
|
|
m.copyTo(merge_image(roi));
|
|
|
|
|
if (merge_index == work_camera_nums * unit_count) {
|
|
|
|
|
file_name = g_conf_path.save_pics_path + "/"
|
|
|
|
|
+ now_ts.toString("yyyy-MM-dd") + "/"
|
|
|
|
|
+ now_ts.toString("yyyy-MM-dd_HH-mm-ss_zzz_") + ".jpg";
|
|
|
|
|
g_save_queue->put(std::make_pair(file_name.toLocal8Bit().constData(), merge_image));
|
|
|
|
|
|
|
|
|
|
#ifdef __TCPSend
|
|
|
|
|
TCPSendInfo.pics_name = filename;
|
|
|
|
|
TCPSendInfo.pics_name = file_name.toLocal8Bit().constData();
|
|
|
|
|
TCP_Info_queue->put(TCPSendInfo);
|
|
|
|
|
#endif
|
|
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
if (unit_count >= 2) {
|
|
|
|
|
image1 = vec_out[(result_index[i]) % 2].clone();
|
|
|
|
|
@ -202,7 +219,8 @@ void SyncWorkThread::run()
|
|
|
|
|
ngReason = 2;
|
|
|
|
|
}
|
|
|
|
|
#endif
|
|
|
|
|
}else {
|
|
|
|
|
}
|
|
|
|
|
else {
|
|
|
|
|
image1 = vec_out[0].clone();
|
|
|
|
|
#ifdef DRAW_RECT
|
|
|
|
|
IsNG |= CheckSelectRects(image1, vec_results, 0, local_DisplayLabelConf, 0);
|
|
|
|
|
@ -231,7 +249,9 @@ void SyncWorkThread::run()
|
|
|
|
|
{
|
|
|
|
|
if (!g_debug_mode)
|
|
|
|
|
{
|
|
|
|
|
#ifdef SYNC_CAMERA
|
|
|
|
|
emit event_ok(local_camera_number);
|
|
|
|
|
#endif
|
|
|
|
|
local_g_result_queue->put(true);
|
|
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
@ -239,7 +259,9 @@ void SyncWorkThread::run()
|
|
|
|
|
{
|
|
|
|
|
if (!g_debug_mode)
|
|
|
|
|
{
|
|
|
|
|
#ifdef SYNC_CAMERA
|
|
|
|
|
emit event_ng(local_camera_number);
|
|
|
|
|
#endif
|
|
|
|
|
local_g_result_queue->put(false);
|
|
|
|
|
IsNGForAll = TRUE;
|
|
|
|
|
}
|
|
|
|
|
@ -259,11 +281,8 @@ void SyncWorkThread::run()
|
|
|
|
|
now_ts.toString("yyyy-MM-dd_HH-mm-ss_zzz_") + QString::number(local_camera_number + 1) +
|
|
|
|
|
"#" + "_" + QString::number(index + 1) + "_" + ng_reason_maps[ngReason] +
|
|
|
|
|
".jpg";
|
|
|
|
|
g_save_queue->put(std::make_pair(file_name.toLocal8Bit().constData(), m));
|
|
|
|
|
#ifdef __TCPSend
|
|
|
|
|
TCPSendInfo.pics_name = file_name.toLocal8Bit().constData();
|
|
|
|
|
TCP_Info_queue->put(TCPSendInfo);
|
|
|
|
|
#endif
|
|
|
|
|
//g_save_queue->put(std::make_pair(file_name.toLocal8Bit().constData(), m));
|
|
|
|
|
|
|
|
|
|
m = vec_out[index].clone();
|
|
|
|
|
file_name = g_conf_path.save_pics_path + "/ng_result/" +
|
|
|
|
|
now_ts.toString("yyyy-MM-dd") + "/"
|
|
|
|
|
@ -276,28 +295,24 @@ void SyncWorkThread::run()
|
|
|
|
|
"#" + "_" + QString::number(index + 1) + "_" + ng_reason_maps[ngReason] +
|
|
|
|
|
".jpg";
|
|
|
|
|
//g_save_queue->put(std::make_pair(file_name.toStdString(), m));
|
|
|
|
|
g_save_queue->put(std::make_pair(file_name.toLocal8Bit().constData(), m));
|
|
|
|
|
|
|
|
|
|
#ifdef __TCPSend
|
|
|
|
|
TCPSendInfo.pics_name = file_name.toLocal8Bit().constData();
|
|
|
|
|
TCP_Info_queue->put(TCPSendInfo);
|
|
|
|
|
#endif
|
|
|
|
|
//g_save_queue->put(std::make_pair(file_name.toLocal8Bit().constData(), m));
|
|
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
#ifdef SYNC_CAMERA
|
|
|
|
|
if (!g_debug_mode)
|
|
|
|
|
{
|
|
|
|
|
emit display_check_total(local_camera_number, ++frame_total);
|
|
|
|
|
emit display_check_total(local_camera_number, ++(frame_total[local_camera_number]));
|
|
|
|
|
//exportDataInfo.cameraTotal = frame_total;
|
|
|
|
|
|
|
|
|
|
emit notify(local_camera_number, 0, image1);
|
|
|
|
|
if (unit_count >= 3)
|
|
|
|
|
emit notify(local_camera_number, 1, image2);
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
#endif
|
|
|
|
|
}
|
|
|
|
|
else
|
|
|
|
|
{
|
|
|
|
|
#ifdef SYNC_CAMERA
|
|
|
|
|
//保证不错位
|
|
|
|
|
if (!g_debug_mode)
|
|
|
|
|
{
|
|
|
|
|
@ -305,9 +320,10 @@ void SyncWorkThread::run()
|
|
|
|
|
emit event_ng(local_camera_number);
|
|
|
|
|
else if (local_SysConf.MisMatchAct == 0)//as ok
|
|
|
|
|
emit event_ok(local_camera_number);
|
|
|
|
|
emit display_check_total(local_camera_number, ++frame_total);
|
|
|
|
|
emit display_check_total(local_camera_number, ++(frame_total[local_camera_number]));
|
|
|
|
|
qDebug() << local_camera_number << "#camera# " << now_ts.toString("yyyy-MM-dd_HH-mm-ss_zzz_") << "xxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxx " << unit_count;
|
|
|
|
|
}
|
|
|
|
|
#endif
|
|
|
|
|
}
|
|
|
|
|
#ifdef __UDPSend
|
|
|
|
|
UDP_Info_queue->put(UDPSendInfo);
|
|
|
|
|
@ -315,7 +331,7 @@ void SyncWorkThread::run()
|
|
|
|
|
}
|
|
|
|
|
if (IsNGForAll)
|
|
|
|
|
{
|
|
|
|
|
/*to do somthing*/
|
|
|
|
|
/*to do something*/
|
|
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
|
seiyu
10 months ago