261 lines
10 KiB
C++
261 lines
10 KiB
C++
//
|
||
// Created by matrixai on 10/26/24.
|
||
//
|
||
|
||
#include "ControlEngine.h"
|
||
|
||
ControlEngine::ControlEngine() {}
|
||
|
||
ControlEngine::~ControlEngine() {}
|
||
|
||
APP_ERROR ControlEngine::Init()
|
||
{
|
||
// 传给web socket 引擎
|
||
strPort0_ = engineName_ + "_" + std::to_string(engineId_) + "_0";
|
||
// 发给箱号推理引擎
|
||
strPort1_ = engineName_ + "_" + std::to_string(engineId_) + "_1";
|
||
// 发送箱角推理引擎
|
||
strPort2_ = engineName_ + "_" + std::to_string(engineId_) + "_2";
|
||
// 发给存储文本引擎
|
||
strPort3_ = engineName_ + "_" + std::to_string(engineId_) + "_3";
|
||
// 发给存图引擎
|
||
strPort4_ = engineName_ + "_" + std::to_string(engineId_) + "_4";
|
||
|
||
this->baseConfig_ = Config::getins()->getBaseConfig();
|
||
this->identifyConfig_ = Config::getins()->getIdentifyConfig();
|
||
this->vecDataSourceConfig_ = Config::getins()->getAllDataSourceConfig();
|
||
|
||
this->initParam();
|
||
LogInfo << "ControlEngine Init ok";
|
||
return APP_ERR_OK;
|
||
}
|
||
|
||
APP_ERROR ControlEngine::DeInit()
|
||
{
|
||
LogInfo << "ControlEngine DeInit ok";
|
||
return APP_ERR_OK;
|
||
}
|
||
|
||
/**
|
||
* 参数初始化(识别结束时需调用)
|
||
* inParam : N/A
|
||
* outParam: N/A
|
||
* return : N/A
|
||
*/
|
||
void ControlEngine::initParam()
|
||
{
|
||
this->strDetectDate_ = "";
|
||
for (int i = 0; i < this->vecDataSourceConfig_.size(); ++i)
|
||
{
|
||
this->mapDetectNO_[i] = 1;
|
||
}
|
||
}
|
||
|
||
void ControlEngine::endIdentify(int iDataSource)
|
||
{
|
||
std::shared_ptr<VDetectInfo> pVDetectInfo = std::make_shared<VDetectInfo>();
|
||
pVDetectInfo->bIsEnd = true;
|
||
pVDetectInfo->iDataSource = iDataSource;
|
||
pVDetectInfo->strDetectDate = this->strDetectDate_;
|
||
pVDetectInfo->strDetectTime = this->strDetectTime_;
|
||
// 传给推理引擎
|
||
outputQueMap_[strPort1_]->push(std::static_pointer_cast<void>(pVDetectInfo), true);
|
||
outputQueMap_[strPort2_]->push(std::static_pointer_cast<void>(pVDetectInfo), true);
|
||
|
||
this->mapDetectNO_[iDataSource] = 1;
|
||
|
||
bool bAllEnd = true;
|
||
for (const auto & dataSource_it : this->mapDetectNO_)
|
||
{
|
||
if (dataSource_it.second != 1) bAllEnd = false;
|
||
}
|
||
if (bAllEnd)
|
||
{
|
||
g_identify_type = IDENTIFY_INIT;
|
||
this->strDetectDate_ = "";
|
||
}
|
||
}
|
||
|
||
void ControlEngine::sendWSEngine(std::string msg)
|
||
{
|
||
msg = "{\"status\":\"normal\",\"msg\":" + msg +"}";
|
||
// 传给websocket引擎
|
||
outputQueMap_[strPort0_]->push(std::static_pointer_cast<void>(std::make_shared<std::string>(msg)));
|
||
}
|
||
|
||
void ControlEngine::detectControl(std::shared_ptr<std::string> pWSServerOrder)
|
||
{
|
||
Json::CharReaderBuilder readerBuilder;
|
||
std::istringstream iss(*pWSServerOrder);
|
||
Json::Value jvOrder;
|
||
std::string errs;
|
||
bool parsingSuccessful = Json::parseFromStream(readerBuilder, iss, &jvOrder, &errs);
|
||
if (!parsingSuccessful)
|
||
{
|
||
std::string msg = "Web Socket 接口调用参数异常，所传数据非json：" + *pWSServerOrder;
|
||
LogError << msg;
|
||
this->sendWSEngine(msg);
|
||
return;
|
||
}
|
||
|
||
if (!jvOrder.isObject() || !jvOrder.isMember("id") || !jvOrder.isMember("commandType"))
|
||
{
|
||
std::string msg = "Web Socket 接口调用参数异常，所传数据缺少核心字段：" + *pWSServerOrder;
|
||
LogError << msg;
|
||
this->sendWSEngine(msg);
|
||
return;
|
||
}
|
||
|
||
// if (jvOrder["commandType"].asInt() != 1 && jvOrder["commandType"].asInt() != 2)
|
||
// {
|
||
// std::string msg = "Web Socket 接口调用参数异常，所传状态不正确：" + *pWSServerOrder;
|
||
// LogError << msg;
|
||
// this->sendWSEngine(msg);
|
||
// return;
|
||
// }
|
||
|
||
switch (jvOrder["commandType"].asInt()) {
|
||
case IDENTIFY_START:
|
||
if (g_identify_type == IDENTIFY_START)
|
||
{
|
||
std::string msg = "当前正在识别，无需重复发送识别信号";
|
||
LogWarn << msg;
|
||
this->sendWSEngine(msg);
|
||
break;
|
||
}
|
||
g_identify_type = IDENTIFY_START;
|
||
break;
|
||
case IDENTIFY_STOP:
|
||
if (!g_identify_type)
|
||
{
|
||
std::string msg = "当前已停止识别，无需重复发送结束信号";
|
||
LogWarn << msg;
|
||
this->sendWSEngine(msg);
|
||
break;
|
||
}
|
||
g_identify_type = IDENTIFY_INIT;
|
||
break;
|
||
case IDENTIFY_RECORD:
|
||
if (!jvOrder.isMember("containerNo"))
|
||
{
|
||
std::string msg = "Web Socket 接口调用参数异常，所传数据缺少核心字段：" + *pWSServerOrder;
|
||
LogError << msg;
|
||
this->sendWSEngine(msg);
|
||
break;
|
||
}
|
||
{
|
||
std::string msg = "{\"status\":\"normal\",\"containerNo\":" + jvOrder["containerNo"].asString() +"}";
|
||
// 传给websocket引擎
|
||
outputQueMap_[strPort0_]->push(std::static_pointer_cast<void>(std::make_shared<std::string>(msg)));
|
||
}
|
||
break;
|
||
default:
|
||
std::string msg = "Web Socket 接口调用参数异常，所传状态不正确：" + *pWSServerOrder;
|
||
LogError << msg;
|
||
this->sendWSEngine(msg);
|
||
break;
|
||
}
|
||
}
|
||
|
||
APP_ERROR ControlEngine::Process()
|
||
{
|
||
while (!isStop_)
|
||
{
|
||
//pop端口0，接收命令
|
||
std::shared_ptr<void> pVoidData0 = nullptr;
|
||
inputQueMap_[strPort0_]->pop(pVoidData0);
|
||
if (nullptr != pVoidData0)
|
||
{
|
||
std::shared_ptr<std::string> pWSServerOrder = std::static_pointer_cast<std::string>(pVoidData0);
|
||
|
||
this->detectControl(pWSServerOrder);
|
||
}
|
||
|
||
//pop端口1，解码后数据
|
||
std::shared_ptr<void> pVoidData1 = nullptr;
|
||
inputQueMap_[strPort1_]->pop(pVoidData1);
|
||
|
||
if (nullptr != pVoidData1)
|
||
{
|
||
std::shared_ptr<ProcessData> pProcessData = std::static_pointer_cast<ProcessData>(pVoidData1);
|
||
|
||
if (pProcessData->bIsEnd)
|
||
{
|
||
// 仅读是视频模式下会进行
|
||
if (this->mapDetectNO_[pProcessData->iDataSource] == 1) continue;
|
||
this->endIdentify(pProcessData->iDataSource);
|
||
LogInfo << "数据源:" << pProcessData->iDataSource << " 视频画面播放结束：停止识别！";
|
||
continue;
|
||
}
|
||
|
||
if (!g_identify_type)
|
||
{
|
||
if (this->mapDetectNO_[pProcessData->iDataSource] != 1)
|
||
{
|
||
this->endIdentify(pProcessData->iDataSource);
|
||
}
|
||
continue;
|
||
}
|
||
|
||
if (this->mapDetectNO_[pProcessData->iDataSource] > this->identifyConfig_.iMaxIdentifyFrame)
|
||
{
|
||
this->endIdentify(pProcessData->iDataSource);
|
||
LogInfo << "数据源:" << pProcessData->iDataSource << " 超过最大允许识别帧数：" << this->identifyConfig_.iMaxIdentifyFrame << " 停止识别！";
|
||
continue;
|
||
}
|
||
|
||
cv::Mat image(pProcessData->dataSourceInfo.iHeight,
|
||
pProcessData->dataSourceInfo.iWidth,
|
||
CV_8UC3,
|
||
static_cast<uint8_t*>(pProcessData->sourceFrameData.pData.get()));
|
||
|
||
if (this->strDetectDate_.empty())
|
||
{
|
||
this->strDetectDate_ = ai_matrix::TimeUtil::getins()->getDate();
|
||
this->strDetectTime_ = ai_matrix::TimeUtil::getins()->getTime();
|
||
//创建该该列车的数据存储目录(防止后续多线程创建报错)
|
||
std::string strTrainPath = this->baseConfig_.strDebugResultPath + "/"
|
||
+ strDetectDate_ + "/"
|
||
+ ai_matrix::StringUtil::getins()->replace_all_distinct(strDetectTime_, ":", "-") + "/";
|
||
ai_matrix::FileUtil::getins()->createDirPath(strTrainPath);
|
||
std::string strBestImgPath = baseConfig_.strResultPath + "/"
|
||
+ strDetectDate_ + "/"
|
||
+ ai_matrix::StringUtil::getins()->replace_all_distinct(strDetectTime_, ":", "-") + "/";
|
||
ai_matrix::FileUtil::getins()->createDirPath(strBestImgPath);
|
||
}
|
||
|
||
std::string strFilePath = baseConfig_.strDebugResultPath + "/"
|
||
+ strDetectDate_ + "/"
|
||
+ StringUtil::getins()->replace_all_distinct(strDetectTime_, ":", "-")
|
||
+ "/";
|
||
|
||
// 识别 + 存文本
|
||
std::shared_ptr<VDetectInfo> pVDetectInfo = std::make_shared<VDetectInfo>();
|
||
pVDetectInfo->iDataSource = pProcessData->iDataSource;
|
||
pVDetectInfo->strFilePath = strFilePath;
|
||
pVDetectInfo->strFileName = std::to_string(this->mapDetectNO_[pProcessData->iDataSource]) + "_" + std::to_string(pProcessData->iDataSource) + ".json";
|
||
pVDetectInfo->iFrameId = this->mapDetectNO_[pProcessData->iDataSource]; //当前帧号
|
||
pVDetectInfo->i64TimeStamp = pProcessData->sourceFrameData.i64TimeStamp;
|
||
pVDetectInfo->bIsEnd = pProcessData->bIsEnd;
|
||
pVDetectInfo->strDetectDate = strDetectDate_;
|
||
pVDetectInfo->strDetectTime = strDetectTime_;
|
||
pVDetectInfo->cvImage = image;
|
||
outputQueMap_[strPort1_]->push(std::static_pointer_cast<void>(pVDetectInfo), true);
|
||
outputQueMap_[strPort2_]->push(std::static_pointer_cast<void>(pVDetectInfo), true);
|
||
outputQueMap_[strPort3_]->push(std::static_pointer_cast<void>(pVDetectInfo), true);
|
||
|
||
// 存图
|
||
std::shared_ptr<SaveImgData> pSaveImgData = std::make_shared<SaveImgData>();
|
||
pSaveImgData->strFilePath = strFilePath;
|
||
pSaveImgData->strFileName = std::to_string(this->mapDetectNO_[pProcessData->iDataSource]) + "_" + std::to_string(pProcessData->iDataSource) + ".jpg";
|
||
pSaveImgData->cvImage = image;
|
||
pSaveImgData->bIsEnd = pProcessData->bIsEnd;
|
||
outputQueMap_[strPort4_]->push(std::static_pointer_cast<void>(pSaveImgData), true);
|
||
|
||
this->mapDetectNO_[pProcessData->iDataSource]++;
|
||
}
|
||
|
||
}
|
||
return APP_ERR_OK;
|
||
}
|