generated from zhangwei/Train_Identify
104 lines
3.6 KiB
C
104 lines
3.6 KiB
C
|
/**
|
|||
|
|
* 集装箱选取最优结果Engine
|
|||
|
|
* */
|
|||
|
|
|
|||
|
|
#ifndef SELECTBESTCONTAINERENGINE_H
|
|||
|
|
#define SELECTBESTCONTAINERENGINE_H
|
|||
|
|
|
|||
|
|
#include "AppCommon.h"
|
|||
|
|
#include "MyYaml.h"
|
|||
|
|
#include "EngineBase.h"
|
|||
|
|
#include "EngineFactory.h"
|
|||
|
|
|
|||
|
|
class SelectBestContainerEngine : public ai_matrix::EngineBase
|
|||
|
|
{
|
|||
|
|
public:
|
|||
|
|
SelectBestContainerEngine();
|
|||
|
|
~SelectBestContainerEngine();
|
|||
|
|
|
|||
|
|
APP_ERROR Init() override;
|
|||
|
|
APP_ERROR DeInit() override;
|
|||
|
|
APP_ERROR Process() override;
|
|||
|
|
|
|||
|
|
private:
|
|||
|
|
|
|||
|
|
//初始化参数信息
|
|||
|
|
void InitParam();
|
|||
|
|
//验证集装箱号是否满足规则(前10位依次与2^0 ~2^9相乘）/11 所得余数，即是校验位)
|
|||
|
|
bool VerifyContainerNo(std::string &strContainerNo);
|
|||
|
|
//构造最优集装箱信息
|
|||
|
|
void MakeBestContainer(Container &containerBest,
|
|||
|
|
TransSubData *pTransSubData,
|
|||
|
|
std::shared_ptr<ProcessData> pProcessData);
|
|||
|
|
|
|||
|
|
void DealFailContianer(TransSubData &transSubData, std::shared_ptr<ProcessData> pProcessData, int iIndex,
|
|||
|
|
std::map<int, TrainContainer> &mapTCFail);
|
|||
|
|
void DealSuccContianer(TransSubData &transSubData, std::shared_ptr<ProcessData> pProcessData, int iIndex,
|
|||
|
|
std::map<int, TrainContainer> &mapTCSucc);
|
|||
|
|
void SelectBestTrainContainer(std::shared_ptr<TrainContainer> pTrainContainer);
|
|||
|
|
void DealContainer(std::shared_ptr<ProcessData> pProcessData, TransSubData &transSubData,
|
|||
|
|
std::map<int, TrainContainer> &mapTCSucc, std::map<int, TrainContainer> &mapTCFail,
|
|||
|
|
std::map<int, Step1Location> &mapStep1Pre, std::map<int, bool> &mapSecondFlag);
|
|||
|
|
|
|||
|
|
std::string strPort0_;
|
|||
|
|
std::string strPort1_;
|
|||
|
|
std::map<int, bool> mapDataSourceIsEnd_; //[key-数据源id, value-数据是否结束]
|
|||
|
|
std::set<int> setTopContainerSid_; //[顶部集装箱数据源ID]
|
|||
|
|
|
|||
|
|
//正向信息
|
|||
|
|
std::map<int, TrainContainer> mapTCFail0_; //[key-数据源id, value-校验失败的集装箱信息]
|
|||
|
|
std::map<int, TrainContainer> mapTCSucc0_; //[key-数据源id, value-校验成功的集装箱信息]
|
|||
|
|
std::map<int, Step1Location> mapStep1Pre0_; //[key-数据源id, value-上一帧大框信息]
|
|||
|
|
std::map<int, bool> mapSecondFlag0_; //[key-数据源id, value-是否保存第2个箱号]
|
|||
|
|
|
|||
|
|
//反向信息
|
|||
|
|
std::map<int, TrainContainer> mapTCFail1_; //[key-数据源id, value-校验失败的集装箱信息]
|
|||
|
|
std::map<int, TrainContainer> mapTCSucc1_; //[key-数据源id, value-校验成功的集装箱信息]
|
|||
|
|
std::map<int, Step1Location> mapStep1Pre1_; //[key-数据源id, value-上一帧大框信息]
|
|||
|
|
std::map<int, bool> mapSecondFlag1_; //[key-数据源id, value-是否保存第2个箱号]
|
|||
|
|
|
|||
|
|
//集装箱号转换
|
|||
|
|
std::map<std::string, int> mapExchange_ = {
|
|||
|
|
{"A", 10},
|
|||
|
|
{"B", 12},
|
|||
|
|
{"C", 13},
|
|||
|
|
{"D", 14},
|
|||
|
|
{"E", 15},
|
|||
|
|
{"F", 16},
|
|||
|
|
{"G", 17},
|
|||
|
|
{"H", 18},
|
|||
|
|
{"I", 19},
|
|||
|
|
{"J", 20},
|
|||
|
|
{"K", 21},
|
|||
|
|
{"L", 23},
|
|||
|
|
{"M", 24},
|
|||
|
|
{"N", 25},
|
|||
|
|
{"O", 26},
|
|||
|
|
{"P", 27},
|
|||
|
|
{"Q", 28},
|
|||
|
|
{"R", 29},
|
|||
|
|
{"S", 30},
|
|||
|
|
{"T", 31},
|
|||
|
|
{"U", 32},
|
|||
|
|
{"V", 34},
|
|||
|
|
{"W", 35},
|
|||
|
|
{"X", 36},
|
|||
|
|
{"Y", 37},
|
|||
|
|
{"Z", 38},
|
|||
|
|
{"0", 0},
|
|||
|
|
{"1", 1},
|
|||
|
|
{"2", 2},
|
|||
|
|
{"3", 3},
|
|||
|
|
{"4", 4},
|
|||
|
|
{"5", 5},
|
|||
|
|
{"6", 6},
|
|||
|
|
{"7", 7},
|
|||
|
|
{"8", 8},
|
|||
|
|
{"9", 9},
|
|||
|
|
};
|
|||
|
|
};
|
|||
|
|
|
|||
|
|
ENGINE_REGIST(SelectBestContainerEngine)
|
|||
|
|
|
|||
|
|
#endif
|