/*
 * @Author: your name
 * @Date: 2022-04-20 15:49:50
 * @LastEditTime: 2025-10-09 18:21:49
 * @LastEditors: xiewenji 527774126@qq.com
 * @Description: 打开koroFileHeader查看配置 进行设置: https://github.com/OBKoro1/koro1FileHeader/wiki/%E9%85%8D%E7%BD%AE
 * @FilePath: /ZCXD_MonitorPlatform/src/CoreLogicModule/include/CamDeal.h
 */
#ifndef ImageAllResult_H_
#define ImageAllResult_H_
#include <iostream>
#include <stdio.h>
#include <sys/time.h>
#include <opencv2/opencv.hpp>
#include <condition_variable>
#include <mutex>
#include <vector>
#include <thread>
#include <string>
#include <stdlib.h>
#include "Define_Product.hpp"
#include "CheckErrorCodeDefine.hpp"
#include "ImgCheckConfig.h"
#include "ImageDetConfig.h"
#include "DetLog.h"
#include "CheckResultJson.h"
using namespace std;
using namespace cv;

// 包含所有的图片检测结果，如 返回的结果，临时结果， 其他检测结果
class ImageAllResult
{
public:
    enum DetStep
    {
        DetStep_NotDet,
        DetStep_Deting,
        DetStep_Complet,
    };
    struct AI_Det_MaskImg
    {
        cv::Rect roi;
        cv::Mat AI_inImg;
        cv::Mat AI_mask;
    };
    struct Config_Update
    {
        bool bUpdate = false; 
        bool wait_DP= false; 
        bool wait_UP= false; 
    };

    // AI MaskImg 的相关使用的参数信息。
    struct AI_MaskImg_BlobParam
    {

        unsigned char *imgBlobHFlagData = NULL;                         // mask 图片在 Y方向是否有餐点，用以加速 blob 分析。
        std::queue<std::shared_ptr<AI_Det_MaskImg>> AIMaskImgBLobQueue; // AI 中间结果 队列
        std::mutex mtx_AIMaskImgBLobQueue;
        int ndatalen;
        AI_MaskImg_BlobParam()
        {
            Init();
        }
        void Init()
        {
            if (imgBlobHFlagData != NULL)
            {
                delete[] imgBlobHFlagData;
                imgBlobHFlagData = NULL;
            }
            ndatalen = 0;
        }
        void InitBLobHData(int ndataLen)
        {
            if (imgBlobHFlagData != NULL)
            {
                delete[] imgBlobHFlagData;
                imgBlobHFlagData = NULL;
            }
            ndatalen = ndataLen;
            imgBlobHFlagData = new unsigned char[ndataLen];
        }
        ~AI_MaskImg_BlobParam()
        {
            Init();
        }
    };

    // 图片的AI 检测相关结果。
    // 兼容多种AI模型输出结果的。
    struct Image_AI_Det_Result
    {

        bool bDet;                  // 是否有检测
        int Model_in_Image_width;   // AI 检测图片的宽度
        int Model_in_Image_height;  // AI 检测图片的高度
        int Model_out_Image_width;  // AI 推理后的图片的宽度
        int Model_out_Image_height; // AI 推理后的图片的宽度

        cv::Mat AI_MaskImg; // AI 推理的mask图片

        std::vector<std::shared_ptr<AI_Det_MaskImg>> AI_Qx_MaskList; // AI 推理 输出中间结果信息。
        std::shared_ptr<AI_MaskImg_BlobParam> blobDetParam;          // BLob 检测要用到的一些 参数 。
        Image_AI_Det_Result()
        {
            Init();
        }
        void Init()
        {
            bDet = false;
            Model_in_Image_width = 0;
            Model_in_Image_height = 0;
            Model_out_Image_width = 0;
            Model_out_Image_height = 0;
            if (AI_MaskImg.empty())
            {
                AI_MaskImg.release();
            }
            AI_Qx_MaskList.clear();
            blobDetParam = std::make_shared<AI_MaskImg_BlobParam>();
        }
        void InitBLobHData(int ndataLen)
        {
            blobDetParam->InitBLobHData(ndataLen);
        }
        ~Image_AI_Det_Result()
        {
            Init();
        }
    };
    struct WTB_Check_Result_
    {
        cv::Rect roi_src;
        cv::Rect roi_show;
        WTB_Check_Result_()
        {
            Init();
        }
        void Init()
        {
            roi_src = cv::Rect(0, 0, 0, 0);
            roi_show = cv::Rect(0, 0, 0, 0);
        }
    };

private:
    /* data */
public:
    ImageAllResult(/* args */);
    ~ImageAllResult();

    void AddLog(std::string str);
    // 添加图片信息
    int AddDetImage(std::shared_ptr<shareImage> p);
    DetStep getStep();
    void setStep(DetStep step);
    bool IsNotDet();
    bool IsChannel(std::string strChannelName);

public:
    // 裁切大小
    cv::Rect CropRoi = cv::Rect(0, 0, 0, 0);
    // 检测的原始图片
    cv::Mat detImg;
    // 专用于 AI 检测的图片，如果 AI的输入尺寸 大于原始图片的尺寸，则 AIDetImg 要进行合理的扩展。
    cv::Mat AI_detImg;
    // AI 推理的mask图片
    cv::Mat AIMaskImg;

    // AI 127Cell推理的mask图片
    cv::Mat AI_127CellMaskImg;

    std::shared_ptr<Image_AI_Det_Result> qx_DetAIResult;      // BLob 检测要用到的一些 参数 。
    std::shared_ptr<Image_AI_Det_Result> cell127_DetAIResult; // 127 cell 检测要用到的一些 参数 。

    // 结果图片
    cv::Mat resultImg;
    // 屏蔽图片  = 搜索边界的 屏蔽区域 +字符的屏蔽区 + 界面绘制的屏蔽区域
    cv::Mat shieldImg;
    // 专用于 AI 检测的屏蔽图片，如果 AI的输入尺寸 大于原始图片的尺寸，则 AI_shieldImg 要进行合理的扩展。
    cv::Mat AI_shieldImg;

    // 异显mask图片
    cv::Mat YX_MaskImg;
    // 缺失polmask图片
    cv::Mat LcakPol_MaskImg;

    float fscale_detToresult_x = 0;
    float fscale_detToresult_y = 0;

    int NG_num = 0;
    int YS_num = 0;
    int Ok_num = 0;
    int NoJudge_num = 0;

    int AI_Det_Img_Width = 0;  // AI 检测图片的宽度
    int AI_Det_Img_Height = 0; // AI 检测图片的宽度
    bool bChangeSize = false;  // 是否改变尺寸

    bool bWhiteOrBlackImg = false; // 是否是白底黑字图片

    bool bOnlyBlob = false;

    Config_Update config_update;


public:
    DetStep m_step = DetStep_NotDet;
    std::string strBaseInfo = "";
    std::string strChannel = "";
    std::vector<std::string> LogList;
    std::shared_ptr<DetLog> detlog;
    std::shared_ptr<CheckResult> result;                  // 返回检测结果
    std::shared_ptr<One_Image_CheckResult_> pDetResult;   // 检测结果  后续主要对pDetResult 继续分析
    std::shared_ptr<ProductBaseResult> productBaseResult; // 产品相关的基本结果信息
    std::shared_ptr<CameraBaseResult> cameraBaseResult;   // 相机相关的基本结果信息
    std::shared_ptr<WTB_Check_Result_> pWTB_Check_Result;

    std::vector<std::shared_ptr<AI_Det_MaskImg>> AI_Qx_MaskList;
    std::mutex mtx_Det; // 互斥量
private:
    PRINT_LOG_ m_PrintLog;

    CheckResultJson m_CheckResultJson;
};

#endif