#include "detector.h"

#include <stdio.h>
#include <stdlib.h>
#include "TH_PlateID.h"

#include "csdk_struct.h"

namespace csdk_wrap
{
    static TH_PlateIDCfg config;
    static unsigned char *fmem;
    static unsigned char *pmem;

    void init_plate_id_config(const cPlateIDCfg *params);
    void set_enable_plate_format(const cPlateIDCfg *params);

    int init_plate_id_detector(const cPlateIDCfg *params, char *soPath) {
        // 初始化配置参数
        init_plate_id_config(params);

        // 设置Licence Manager运行路径
        int ret = TH_SetSoPath(soPath);
        printf("TH_SetSoPath ret %d\n", ret);
        // 初始化 sdk
	    ret = TH_InitPlateIDSDK(&config);
        printf("TH_InitPlateIDSDK ret %d\n", ret);

        // 设置车牌类型识别
        set_enable_plate_format(params);

        // 设置识别阈值
        TH_SetRecogThreshold(params->bLocateTh, params->bOCRTh, &config);

        return ret;
    }

    void init_plate_id_config(const cPlateIDCfg *params) {
        int fMemSize = params->nFastMemorySize * 1024;
        int pMemSize = params->nMemorySize * 1024 * 1024;

        fmem = (unsigned char*)malloc(fMemSize * sizeof(unsigned char));
        pmem = (unsigned char*)malloc(pMemSize * sizeof(unsigned char));

        // 默认参数
        config.nMinPlateWidth = 60;
        config.nMaxPlateWidth = 400;
        config.bVertCompress = 0;
        config.nImageFormat = ImageFormatBGR;
        config.bOutputSingleFrame = 1;
        config.bUTF8 = 1;

        config.pFastMemory = fmem;
        config.nFastMemorySize = fMemSize;
        config.pMemory = pmem;
        config.nMemorySize= pMemSize;

        if (params->nMinPlateWidth > 60 && params->nMinPlateWidth < 400) {
            config.nMinPlateWidth = params->nMinPlateWidth;
        }
        if (params->nMaxPlateWidth > 60 && params->nMaxPlateWidth < 400) {
            config.nMaxPlateWidth = params->nMaxPlateWidth;
        }
        
        config.nMaxImageWidth = params->nMaxImageWidth;
        config.nMaxImageHeight = params->nMaxImageHeight;
        config.bIsFieldImage = params->bIsFieldImage;
        config.bMovingImage = params->bMovingImage;
        config.nOrderOpt = params->nOrderOpt;
        config.bLeanCorrection = params->bLeanCorrection;
        config.nImproveSpeed = params->nImproveSpeed;
        config.bCarLogo = params->bCarLogo;
        config.bLotDetect = params->bLotDetect;
        config.bShadow = params->bShadow;
        config.bShieldRailing = params->bShieldRailing;
        config.bCarModel = params->bCarModel;
    }

    void set_enable_plate_format(const cPlateIDCfg *params) {
        if (params->bOnlyLocation == 1) {
            TH_SetEnabledPlateFormat(PARAM_ONLY_LOCATION_ON, &config);
            return;
        }
        if (params->bOnlyTwoRowYellow == 1) {
            TH_SetEnabledPlateFormat(PARAM_ONLY_TWOROWYELLOW_ON, &config);
            return;
        }
        if (params->bIndividual == 1) {
            TH_SetEnabledPlateFormat(PARAM_INDIVIDUAL_ON, &config);
        }
        if (params->bTwoRowYellow == 1) {
            TH_SetEnabledPlateFormat(PARAM_TWOROWYELLOW_ON, &config);
        }
        if (params->bArmPolice == 1) {
            TH_SetEnabledPlateFormat(PARAM_ARMPOLICE_ON, &config);
        }
        if (params->bArmPolice2 == 1) {
            TH_SetEnabledPlateFormat(PARAM_ARMPOLICE2_ON, &config);
        }
        if (params->bTwoRowArmy == 1) {
            TH_SetEnabledPlateFormat(PARAM_TWOROWARMY_ON, &config);
        }
        if (params->bTractor == 1) {
            TH_SetEnabledPlateFormat(PARAM_TRACTOR_ON, &config);
        }
        if (params->bEmbassy == 1) {
            TH_SetEnabledPlateFormat(PARAM_EMBASSY_ON, &config);
        }
        if (params->bChangNei == 1) {
            TH_SetEnabledPlateFormat(PARAM_CHANGNEI_ON, &config);
        }
        if (params->bMinHang == 1) {
            TH_SetEnabledPlateFormat(PARAM_MINHANG_ON, &config);
        }
        if (params->bConsulate == 1) {
            TH_SetEnabledPlateFormat(PARAM_CONSULATE_ON, &config);
        }
        if (params->bNewEnergy == 1) {
            TH_SetEnabledPlateFormat(PARAM_NEWENERGY_ON, &config);
        }
    }

    cPlateIDResult* plate_id_detect(int *plateIDCount, const cIMAGE *img, const cRECT *rect) {
        if (!img) {
            return NULL;
        }

        cPlateIDResult *ppos = NULL;
        int nResultNum = 1;

        ::TH_PlateIDResult result[6];        
        memset(&result[0], 0, sizeof(result));

        // 设置识别区域
        ::TH_RECT rcDetect = {rect->left, rect->top, rect->right, rect->bottom};

        int ret = TH_RecogImage((BYTE*)(img->data), img->width, img->height, result, &nResultNum, &rcDetect, &config);
        // printf("TH_RecogImage ret = %d\n", ret);
        if (ret == 0){
            ret = TH_EvaluateCarColor((BYTE*)(img->data), img->width, img->height, result, &nResultNum, &rcDetect, &config);
        }
        if (ret == 0 && nResultNum > 0){
            ppos = (cPlateIDResult*)malloc(nResultNum * sizeof(cPlateIDResult));
            *plateIDCount = nResultNum;
            memcpy(ppos, result, sizeof(TH_PlateIDResult) * nResultNum);
        }
        return ppos;
    }

    int uninit_plate_id_detector() {
        free(fmem);
        free(pmem);
        fmem = NULL;
        pmem = NULL;

        return TH_UninitPlateIDSDK(&config);
    }

} // csdk_wrap