/*
//实现对部分缺陷 需要进行 数量 和距离上分析的
 */
#ifndef AI_Edge_QX_Det_H_
#define AI_Edge_QX_Det_H_
#include <opencv2/opencv.hpp>
#include "CheckUtil.hpp"
#include "OtherDetect.h"
#include "OtherDetBaseDefine.h"
#include "CheckErrorCodeDefine.hpp"
#include "ImageDetConfig.h"
#include "CheckConfigDefine.h"
#include "ImageStorage.h"
#include "AI_Factory.h"

using namespace std;
using namespace cv;

// 边缘缺陷检测
class AI_Edge_QX_Det
{
public:
	/// @brief 检测过程的参数
	struct DetConfig
	{
		std::shared_ptr<std::vector<cv::Rect>> pEdgeDet_roiList;
		Function_Edge_Det *pdege_Det_config;
		std::string strCamChannel;
		cv::Rect CutRoi;
		int ninstruct;
		DetConfig()
		{
			Init();
		}
		void Init()
		{
			pdege_Det_config = NULL;
			strCamChannel = "";
			ninstruct = 0;
			CutRoi = cv::Rect(0, 0, 0, 0);
		}
	};

public:
	AI_Edge_QX_Det(/* args */);
	~AI_Edge_QX_Det();

	int Detect(const cv::Mat &img, cv::Mat &detmask, DetConfig *pDetConfig,std::shared_ptr<DetLog> pdetlog);

private:
	int SaveProcessImg(const cv::Mat &inImg, const cv::Mat &outImg, bool isALLzeros, DetConfig *pDetConfig, int idx = 0, int nruntype = 0);
	std::string GetImgSaveName(int flag = 0);

private:
	bool m_bInitSucc; // 是否初始化成功

	std::string m_strSaveImgRootPath;
	std::string m_strSaveImgRootPath_OK;
	std::string m_strSaveImgRootPath_Qx;
	ImageStorage *m_pImageStorage;
	DetConfig *m_pDetConfig;

	std::shared_ptr<AIFactory> AI_Factory;
	std::shared_ptr<DetLog> m_pdetlog;

	int m_nImge_num;
	std::string m_strCamChannel;

private:
	/* data */
};

#endif