#ifndef Down_Angle_Detect_H_
#define Down_Angle_Detect_H_

#include "BaseInclude.h"
#include "AI_Moudel.hpp"
#include "BlobBase.h"

#include <opencv2/freetype.hpp>           //0928-add
#include <cmath>                          //1028-add

class Down_Angle_Detect: public ImgCheckBase
{
public:
    Down_Angle_Detect();
    ~Down_Angle_Detect();

    // 初始化参数 pconfig 参数指针 返回：0 成功 其他异常
    int RunStart(void *pconfig1 = NULL);

    // 阻塞式检测
    int CheckImg(std::shared_ptr<shareImage> p, std::shared_ptr<CheckResult> &pResult);

    int CheckRotateTemplete2Img2(std::shared_ptr<shareImage> p, std::shared_ptr<CheckResult> &pResult, std::shared_ptr<CheckResult> &pResultOut);  //1107-add 添加检查模板图像和旋转之后的图像的函数

    // 更新参数 pconfig 参数指针，nConfigType 需要更新的参数类型  返回：0 成功 其他异常
    int UpdateConfig(void *pconfig, int nConfigType);


private:
    // 检测
    int CheckRun();

    cv::Rect GetDetRoi(const cv::Mat &img);

    // 初始化数据
    int CheckImgInit();
    // 初始化模型
    int InitModel();

    int AI_Run(const cv::Mat &img, cv::Point s2StartP, double LabelP1P2distance, float &outAngle, cv::Point2f &CenterP, 
               cv::Point &LeftDownP, cv::Point &RightDownP, int &minRectWidth, int &minRectHeight, cv::Mat &resultChipMaskImg); // 1107-add
    
    // int Draw3(const cv::Mat &img, cv::Mat &resultimg, cv::Point s2StartP, cv::Point LabelCircleP, cv::Point startP, cv::Point endP, bool drawLineFlag); // 1029-add
    int Draw3(const cv::Mat &img, cv::Mat &resultimg, cv::Point s2StartP, cv::Point s2OutCenterP, cv::Point LabelCircleP, cv::Point startS2LabelP, cv::Point endS2LabelP, cv::Point startLeftB_P, cv::Point endRightB_P,bool drawLineFlag); // 1107-add
    
    // 原始 size 到  显示图
    int GetShowSrcImg(const cv::Mat &img, cv::Mat &sizeimg); // 1013-add
    int AddResultErrorImg(cv::Mat &src);                     // 1013-add

    float calculate_angle2(cv::Point Mar1Point, cv::Point Mar2Point); // 1028-add
    float calculate_angle2f(cv::Point2f Mar1Point, cv::Point2f Mar2Point); // 1107-add
    // double CalculateAngle(cv::Point Mar1Point, cv::Point Mar2Point); //1029-add

    double calculateSplitLineAngle(const cv::Mat& binaryImage);

private:
    int m_nErrorCode; // 错误代码
    std::shared_ptr<shareImage> DetImgInfo_shareP;
    std::shared_ptr<CheckResult> m_CheckResult_shareP;


    std::shared_ptr<AI_Model_ConfigList> m_AIModelConfigList;
    std::shared_ptr<CheckBaseConfig> m_CheckBaseConfig;
    RunInfoST m_pRunConfig;

    ERROR_DOTS_BLOBS m_blob;
    bool bwriteImg;

    bool m_Update_config;

    std::vector<BlobResult> m_resultList;
    cv::Rect m_AlignMaxRoi;
    cv::Mat m_resultImg;
    cv::Mat m_reconstructImg;
    std::string m_strCheckBaseConfigPath;


    // 检查缺陷对应参数中的位置。
    int m_QxInParamListIdx[CONFIG_QX_NAME_count] = {0};

    cv::Point2f m_LabelCenterP;//当前模板的中心点

};

#endif