//
// Created by Scheaven on 2020/4/26.
//
#include "h_interface.h"
#include <iostream>
#include <fstream>
#include <json/json.h>
using namespace std;

string m_staticStruct::model_path = "path";
string m_staticStruct::model_cfg = "0"; // 初始化结构体静态变量值
string m_staticStruct::model_wts = "0"; // 初始化结构体静态变量值
int m_staticStruct::type = 1;
bool m_staticStruct::isTrack= true;
int m_staticStruct::max_cam_num = 0;
int m_staticStruct::wander_time = 0;
float m_staticStruct::mv_velocity = 0;
float m_staticStruct::fall_rate = 0;


API void* create(const char *conf, int *max_chan)
{
    //CLog::Initialize("/opt/vasystem/bin/models/baseDetector/log.properties");
    CLog::Initialize("../config/log4cplus.properties");
    ReadJsonFromFile(conf);
    AriManager *handle = new AriManager();
    *max_chan = m_staticStruct::max_cam_num;
    for (int i = 1; i <= m_staticStruct::max_cam_num; ++i)
    {
        handle->add_cam(i);
    }
    Timer::getInstance()->reset();
    return handle;
}

API void* get_result(void *handle, const void *img, const int chan)
{
    AriManager *h = (AriManager*)handle;
    TResult *t_result = (TResult*)malloc(sizeof(TResult));
    init_TResult(t_result);

    Timer::getInstance()->out("eveTime before yolo");
    // h->single_SDK(chan, img, *t_result);
    Timer::getInstance()->out("eveTime runing yolo");
    return t_result;
}


API void* get_result2(void *handle, const void *img, const int chan, const char* timestamp, const char* mode)
{
    DEBUG("-----------------------begin------------ ");
    AriManager *h = (AriManager*)handle;
    TResult *t_result = (TResult*)malloc(sizeof(TResult));
    init_TResult(t_result);

    if(!m_staticStruct::isTrack)
    {
        Timer::getInstance()->out("get_result2 before yolo");
        h->single_SDK(chan, img, t_result);
        Timer::getInstance()->out("get_result2 runing yolo");
        return t_result;
    }

    h->single_SDK(chan, img, t_result, const_cast<char*>(timestamp), mode);
    DEBUG("--cam id:" + to_string(chan) + "  image human_count:" + to_string(t_result->count));
    for (int i = 0; i < t_result->count; ++i)
    {
        if(mode == "video"){
            std::cout<<"=====================================exc==================================================="<<std::endl;
            DEBUG("--human_id:" + to_string(t_result->targets[i].id)+"  human_confidence:" + to_string(t_result->targets[i].confidence) + "  human_attribute:" + std::string(t_result->targets[i].attribute) +  "  human_top:" + to_string(t_result->targets[i].rect.top)+" human_left:" + to_string(t_result->targets[i].rect.left) + "  human_right:" + to_string(t_result->targets[i].rect.right)+ "  human_bottom:" + to_string(t_result->targets[i].rect.bottom));
        }
        else{
            DEBUG("human_confidence:" + to_string(t_result->targets[i].confidence) + "  human_top:" + to_string(t_result->targets[i].rect.top)+" human_left:" + to_string(t_result->targets[i].rect.left) + "  human_right:" + to_string(t_result->targets[i].rect.right)+ "  human_bottom:" + to_string(t_result->targets[i].rect.bottom));
        }
    }
    DEBUG("---------------------end over------------------------\n\n");
    return t_result;
}

API void release_result(void *res)
{
    if (!res) return;
    TResult *cres = (TResult*)res;
//    delete cres;
    for (int i = 0; i < cres->count; i++){
        Target t = cres->targets[i];
        if (t.feature) free(t.feature);
        if (t.attribute) free(t.attribute);
    }

    free(cres->targets);
    free(cres);
}

API void release(void* handle)
{
    AriManager *h = (AriManager*)handle;
    delete h;
    h = NULL;
}

void ReadJsonFromFile(const char* filename)
{
    Json::Reader reader;
    Json::Value root;
    int max_cam_num;
    int wander_time;
    float mv_velocity;
    float fall_rate;
    string tmp = filename;

    ifstream in(filename, ios::binary);
    if( !in.is_open() )
    {
        cout << "Error opening json config file\n";
        return;
    }
    if(reader.parse(in,root))
    {
        std::string model_path = root["param"]["model_path"].asString();
        std::string model_cfg= root["param"]["model_cfg"].asString();
        std::string model_wts = root["param"]["model_wts"].asString();
        int type = root["param"]["type"].asInt();
        bool isTrack = root["param"]["isTrack"].asBool();
        int max_cam_num = root["param"]["max_cam_num"].asInt();
        int wander_time = root["param"]["wander_time"].asInt();
        int mv_velocity = root["param"]["mv_velocity"].asFloat();
        int fall_rate = root["param"]["fall_rate"].asFloat();
        m_staticStruct::model_path  = model_path;
        m_staticStruct::model_cfg  = model_cfg;
        m_staticStruct::model_wts  = model_wts;
        m_staticStruct::type  = type;
        m_staticStruct::isTrack = isTrack;
        m_staticStruct::max_cam_num  = max_cam_num;
        m_staticStruct::wander_time = wander_time;
        m_staticStruct::mv_velocity = mv_velocity;
        m_staticStruct::fall_rate = fall_rate;

    }
    in.close();
}

void init_TResult(TResult *t)
{
    t->count = 0;
    t->targets = nullptr;
}