|
|
#ifndef ImgCheckBase_H_
|
|
|
#define ImgCheckBase_H_
|
|
|
#include <string>
|
|
|
#include <memory>
|
|
|
#define ALL_INTERFACE_VERSION 6
|
|
|
#define MAX_GPU_NUM 4
|
|
|
|
|
|
enum CHECK_THREAD_RUN_STATUS
|
|
|
{
|
|
|
CHECK_THREAD_STATUS_IDLE, // 空闲 0
|
|
|
CHECK_THREAD_STATUS_READY, // 准备好了 1
|
|
|
CHECK_THREAD_STATUS_BUSY, // 运行中 2
|
|
|
CHECK_THREAD_STATUS_COMPLETE, // 检测完成 3
|
|
|
CHECK_THREAD_STATUS_ERROR, // 运行错误 4
|
|
|
};
|
|
|
// 检测工位序号
|
|
|
enum DETECT_WROK_IDX
|
|
|
{
|
|
|
DETECT_WROK_0, // 0工位
|
|
|
DETECT_WROK_1, // 1工位
|
|
|
DETECT_WROK_2, // 2工位
|
|
|
DETECT_WROK_COUNT,
|
|
|
};
|
|
|
struct RunInfoST
|
|
|
{
|
|
|
int nThreadIdx; // 线程号id
|
|
|
int nWorkIdx; // 工位号
|
|
|
int nDeviceId; // GPU 设备 号 0 或 1
|
|
|
int UseGPUList[MAX_GPU_NUM]; // 使用的GPU设备号
|
|
|
int nCpu_start_Idx; // 绑定cpu 核号，
|
|
|
int nCpu_num; // 8 至少需要 8个
|
|
|
bool bSaveCheckImg; // 是否存储
|
|
|
bool bRetest; // 是否是复测标志，复测不需要加载模型
|
|
|
int flag1;
|
|
|
int flag2;
|
|
|
std::string str1;
|
|
|
std::string str2;
|
|
|
RunInfoST()
|
|
|
{
|
|
|
nThreadIdx = 0;
|
|
|
nDeviceId = 0;
|
|
|
nWorkIdx = DETECT_WROK_0;
|
|
|
nCpu_start_Idx = 0;
|
|
|
nCpu_num = 8;
|
|
|
flag1 = 0;
|
|
|
flag2 = 0;
|
|
|
bRetest = false;
|
|
|
str1 = "";
|
|
|
str2 = "";
|
|
|
bSaveCheckImg = false;
|
|
|
for (int i = 0; i < MAX_GPU_NUM; i++)
|
|
|
{
|
|
|
UseGPUList[i] = -1;
|
|
|
}
|
|
|
UseGPUList[0] = 0;
|
|
|
UseGPUList[1] = 1;
|
|
|
}
|
|
|
void copy(RunInfoST tem)
|
|
|
{
|
|
|
for (int i = 0; i < MAX_GPU_NUM; i++)
|
|
|
{
|
|
|
this->UseGPUList[i] = tem.UseGPUList[i];
|
|
|
}
|
|
|
this->nDeviceId = tem.nDeviceId;
|
|
|
this->nThreadIdx = tem.nThreadIdx;
|
|
|
this->nWorkIdx = tem.nWorkIdx;
|
|
|
this->nCpu_start_Idx = tem.nCpu_start_Idx;
|
|
|
this->nCpu_num = tem.nCpu_num;
|
|
|
this->bSaveCheckImg = tem.bSaveCheckImg;
|
|
|
this->flag1 = tem.flag1;
|
|
|
this->flag2 = tem.flag2;
|
|
|
this->bRetest = tem.bRetest;
|
|
|
this->str1 = tem.str1;
|
|
|
this->str2 = tem.str2;
|
|
|
}
|
|
|
};
|
|
|
/*******************一般调用流程*************************************************************************************************************************/
|
|
|
/*******************1、初始化 UpdateConfig*************************************************************************************************************************/
|
|
|
/*******************2、初始化 Init 并开启 *************************************************************************************************************************/
|
|
|
/*******************3、GetStatus 获取状态 如果 =CHECK_THREAD_STATUS_IDLE 可以设置检测数据并开启检测 SetDataRun***************************************************/
|
|
|
/*******************4、GetStatus 获取状态 如果 =CHECK_THREAD_STATUS_COMPLETE 检测完成，可以获取检测结果 GetCheckReuslt 拷贝检测结果 自动把状态设为 CHECK_THREAD_STATUS_IDLE*****************************/
|
|
|
struct shareImage;
|
|
|
struct CheckResult;
|
|
|
class ALLImgCheckBase
|
|
|
{
|
|
|
protected:
|
|
|
ALLImgCheckBase() {}
|
|
|
|
|
|
public:
|
|
|
// delete camera interface
|
|
|
~ALLImgCheckBase() {}
|
|
|
static ALLImgCheckBase *GetInstance();
|
|
|
// 初始化参数 pconfig 参数指针 返回：0 成功 其他异常
|
|
|
virtual int RunStart(void *pconfig1 = NULL) = 0;
|
|
|
|
|
|
// 设置检测数据，并开启检测 返回：0 成功 其他异常
|
|
|
virtual int SetDataRun_SharePtr(std::shared_ptr<shareImage> p) = 0;
|
|
|
|
|
|
// 获取结果信息 返回：0 成功 其他异常
|
|
|
virtual int GetCheckReuslt(std::shared_ptr<CheckResult> &pResult) = 0;
|
|
|
|
|
|
virtual int CheckImg(std::shared_ptr<shareImage> p, std::shared_ptr<CheckResult> &pResult) = 0;
|
|
|
|
|
|
// 获取检测库 状态信息 返回：CHECK_THREAD_RUN_STATUS
|
|
|
virtual int GetStatus() = 0;
|
|
|
|
|
|
// 更新参数 pconfig 参数指针，nConfigType 需要更新的参数类型 返回：0 成功 其他异常
|
|
|
virtual int UpdateConfig(void *pconfig, int nConfigType) = 0;
|
|
|
|
|
|
// 返回检测版本信息
|
|
|
virtual std::string GetVersion() = 0;
|
|
|
// 返回错误信息
|
|
|
virtual std::string GetErrorInfo() = 0;
|
|
|
|
|
|
static ALLImgCheckBase *instance;
|
|
|
};
|
|
|
#endif |
