//
// Created by ps on 19-4-9.
//

#ifndef QIAOJIASYSTEM_VPTDETECTWRAPPER_H
#define QIAOJIASYSTEM_VPTDETECTWRAPPER_H

#include <libs/Vpts/include/vpt_pic_header.h>
#include <libs/Vpts/include/vpt_pic.h>
#include <opencv2/opencv.hpp>
#include <CvMath.hpp>
#include "mainAssist.h"

//分类结果结构体
//struct ClassfyObjRes {
//    int res_index;      //分类结果
//    float res_prob;     //分类结构体
//};

struct PersonDetectRes {
//    std::string head;
//    std::string head_color;
//    std::string eye;
//    std::string mouth;
//    std::string up;
//    std::string up_color;
//    std::string clothing_text;
//    std::string down;
//    std::string down_color;
//    std::string bao;
//    std::string sex;
//    std::string age;
//    std::string viewpoint;
//    std::string dasan;
//    std::string child;
//    std::string personstate;
    classfy_boj_res res_objs[HP_FIR_INDEX_SIZE];          //分类结果
};

struct PerRideCarDetectRes {
//    std::string hcp_up;
//    std::string hcp_up_color;
//    std::string hcp_down;
//    std::string hcp_down_color;
//    std::string bao;
//    std::string hcp_bag_color;
//    std::string hcp_head;
//    std::string hcp_clothing_text;
//    std::string hcp_sex;
//    std::string hcp_figure;
//    std::string hcp_nationality;
//    std::string hcp_age;
//    std::string hcp_eye;
//    std::string hcp_mouth;
//    std::string hcp_weibo;
//    std::string hcp_carColor;
//    std::string hcp_orient;
//    std::string hcp_drivenum;
//    std::string hcp_dasan;
//    std::string hcp_take;

    classfy_boj_res res_objs[HCP_FIR_INDEX_SIZE];          //分类结果
};

struct CarDetectRes {
    //车型识别结果
    std::string vehicle_brand;              //车辆品牌
    std::string vehicle_subbrand;           //车辆子品牌
    std::string vehicle_issue_year;         //车辆年款
    std::string vehicle_type;               //车辆类型
    std::string freight_ton;                //货车吨级
    float vehicle_score;                    //车型识别置信度

    //车颜色识别结果
    ColorLabel colorLabel;                  //车颜色
    float colorLabelProb;                   //车颜色置信度

    //车牌检测结果
    BskRect carPlateRect;                   //车牌区域
    float carPlateRectScore;                //车牌区域置信度
    std::string carPlate;                   //车牌
    float carPlateScore;                    //车牌置信度
    int carPlatetype;
};

union VPTDetectRes {

    struct TestVPTDetectRes {
        int i;
    };

    VPTDetectRes() : testVPTDetectRes(TestVPTDetectRes()) {

    }

    ~VPTDetectRes() {

    }

    TestVPTDetectRes testVPTDetectRes;
    PersonDetectRes personDetectRes;
    PerRideCarDetectRes perRideCarDetectRes;
    CarDetectRes carDetectRes;
};

enum class SDKDetectType {
    test = -1,
    person = 0,     // PersonDetectRes
    bike,           // PerRideCarDetectRes
    motor,          // PerRideCarDetectRes
    tricycle,       //
    car,            // CarDetectRes
    bigbus,         // CarDetectRes
    lorry,          // CarDetectRes
    tractor,        //
    midibus         // CarDetectRes
};

struct VptDetectResults {
public:
    VptDetectResults() {}

    virtual ~VptDetectResults() {}

    VptDetectResults(const VptDetectResults &src) {

        this->sdkDetectType = src.sdkDetectType;
        this->bskRect = src.bskRect;

        switch (src.sdkDetectType) {
            case SDKDetectType::person :
                this->vptDetectRes.personDetectRes = src.vptDetectRes.personDetectRes;
                break;
            case SDKDetectType::bike:
            case SDKDetectType::motor :
                this->vptDetectRes.perRideCarDetectRes = src.vptDetectRes.perRideCarDetectRes;
                break;
            case SDKDetectType::car :
            case SDKDetectType::bigbus :
            case SDKDetectType::lorry :
            case SDKDetectType::midibus :
                this->vptDetectRes.carDetectRes = src.vptDetectRes.carDetectRes;
                break;
        }
    }

    SDKDetectType sdkDetectType;
    VPTDetectRes vptDetectRes;
    BskRect bskRect;
};

class VptDetectWrapper {
public:
    VptDetectWrapper();

    VptDetectWrapper(int _gpuIdx, string _dbFilePath);

    virtual ~VptDetectWrapper();


    void process_image(cv::Mat &_img);

private:

    void init();

    VptDetectResults changeAnalysisRes(int index, void *result);

private:
    list<VptDetectResults> m_results;

    void *m_handle;
    vptpic_param m_param;

    int m_gpuIdx{0};
    std::string m_dbFilePath{"CarBodyFeature.db"};
};


#endif //QIAOJIASYSTEM_VPTDETECTWRAPPER_H