#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;

    int init_plate_id_detector(const int width, const int height, char *soPath) {
        int fMemSize = 0x8000;
        int pMemSize = 200 * 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.nMaxImageWidth = width;
        config.nMaxImageHeight = height;

        config.bVertCompress = 0;
        config.bIsFieldImage = 0;
        config.bOutputSingleFrame = 1;

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

        config.bUTF8 = 1;

        // 动态模式
        config.bMovingImage = 1;
        config.nImageFormat = ImageFormatBGR;

        // 静态模式
        // config.bMovingImage = 0;

        // 识别车型
        config.bCarModel = 1;

        // 设置Licence Manager运行路径
        int ret = TH_SetSoPath(soPath);

        // 初始化 sdk
	    ret = TH_InitPlateIDSDK(&config);

        // 开启新能源车牌识别
        ret = TH_SetEnabledPlateFormat(PARAM_NEWENERGY_ON, &config);

        ret = TH_SetImageFormat(ImageFormatBGR, false, false, &config);

        // 设置识别阈值
        TH_SetRecogThreshold(5, 2, &config);

        return ret;
    }

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

        cPlateIDResult *ppos = NULL;
        int nResultNum = 1;

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

        // 设置识别区域
        ::TH_RECT rcDetect;
        rcDetect.top = 0;
        rcDetect.right = img->width;
        rcDetect.bottom = img->height;
        rcDetect.left = 0;        

        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 && 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