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

#include <basic/util/app/AppPreference.hpp>
#include "VptRpcElement.h"

VptRpcElement::VptRpcElement(std::string shareMemoryName) : m_shareMemoryName(shareMemoryName),
                                                            m_rpcClient(appPref.getStringData("vpt.proxy"),
                                                                        appPref.getStringData("vpt.ip"),
                                                                        appPref.getIntData("vpt.port"), "tcp"),
                                                            m_sharedMemory(nullptr), m_trackingTrigger(nullptr) {
    init();

}

VptRpcElement::~VptRpcElement() {
    if (m_sharedMemory) {
        delete m_sharedMemory;
    }
    if (m_trackingTrigger) {
        delete m_trackingTrigger;
        m_trackingTrigger = nullptr;
    }
}

void VptRpcElement::setImage(const cv::Mat &value) {

    if (value.size != m_image.size) {
        m_image = cv::Mat(value.rows, value.cols, CV_8UC3, m_sharedMemory->data());
    }
    value.copyTo(m_image);
}

cv::Mat VptRpcElement::getImage() const {
    return m_image;
}

VptDetect::ObjInfos VptRpcElement::getObjects() const {
    return m_objs;
}

std::vector<ScoredRect> VptRpcElement::getLastScoreRects() const {
    return m_trackingTrigger->getLastScoreRects();
}

void VptRpcElement::threadFunc() {
    //#todo 业务
    //  ClockTimer ct("YoloRpcElement::threadFunc");
    m_triggerScoredRects.clear();
    m_CarScoredRects.clear();
    m_HPScoredRects.clear();
    m_HCPScoredRects.clear();
    try {
        auto server = m_rpcClient.getServer();
        if (!server) {
            ERR("server is null");
            return;
        }
        m_objs = server->VptDetect(m_image.cols, m_image.rows, m_sharedMemory->key().toStdString());//TODO
    }
    catch (std::exception &e) {
        ERR(e.what())
    }

    for (auto &obj: m_objs) {
//        //#todo 人/人骑车/车
//        if (obj.sdkDetectType != 0)
//            continue;

        ScoredRect scoredRect;

        //#todo
        scoredRect.rect = cv::Rect(obj.bskRect.left, obj.bskRect.top, obj.bskRect.width, obj.bskRect.height);
        scoredRect.score = scoredRect.rect.area() > 0 ? obj.obj_score : 0;
        scoredRect.properties["time"] = getProperty("time");
        scoredRect.properties["imgKey"] = getProperty("imgKey");
        if (m_trackingTrigger->triggerOnce(scoredRect)) {
            m_trackingTrigger->getLastRect().properties["id"] = to_string(scoredRect.id);
            m_trackingTrigger->getLastRect().properties["type"] = m_TypeStr[obj.sdkDetectType];
            m_trackingTrigger->getLastRect().properties["time"] = getProperty("time");
            m_trackingTrigger->getLastRect().properties["imgKey"] = getProperty("imgKey");
            m_triggerScoredRects.push_back(m_trackingTrigger->getLastRect());

        }
    }
    m_trackingTrigger->triggerLine();
    auto res_ = m_trackingTrigger->getLastScoreRects();
    for (auto &obj: res_) {
        if(obj.properties["type"] == "person") {
            m_HPScoredRects.push_back(obj);
            DBG("person:" << obj.score);
        }else if((obj.properties["type"] == "bike") ||
                (obj.properties["type"] == "motor")){
            m_HCPScoredRects.push_back(obj);
            DBG("bike or moto:" << obj.score);
        }else{
            m_CarScoredRects.push_back(obj);
            DBG("car:" << obj.score);
        }
    }

    fireConnectors();
}

void VptRpcElement::init() {

    m_trackingTrigger = new TrackingTrigger(0.5);
    m_sharedMemory = new QSharedMemory(QString(m_shareMemoryName.c_str()));
//    DBG(shareMemoryName);
    //1520 x 2688   1080 x 1920 //2560 * 1440 * 4
    if (!m_sharedMemory->create(4608 * 2592 * 4)) {
        m_sharedMemory->attach();
    }

    try {
        auto server = m_rpcClient.getServer();
        m_HpResStr = server->getStr(0);      // 行人二次属性   case 0
        m_HcpResStr = server->getStr(1);      // 人骑车二次属性   case 1
        m_TypeStr = server->getStr(2)[0];        // VPT检测种类   case 2
        m_colorLabelStr = server->getStr(3)[0];    // 车牌颜色   case 3
    }
    catch (std::exception &e) {
        ERR(e.what())
    }
}

std::vector<ScoredRect> VptRpcElement::getLastHpRects() const {
    return m_HPScoredRects;
}

std::vector<ScoredRect> VptRpcElement::getLastHCPRects() const {
    return m_HCPScoredRects;
}

std::vector<ScoredRect> VptRpcElement::getLastCarRects() const {
    return m_CarScoredRects;
}