#ifndef PERIMETERELEMENT
#define PERIMETERELEMENT

#include <basic/pipe/PipeElement.h>
//#include <basic/event/EventHandler.hpp>
#include <opencv2/opencv.hpp>
#include <Ice/Ice.h>
#include <YoloServer.h>
#include <QtGui/qpolygon.h>
#include "TrackingTrigger.h"
#include <QJsonArray>
#include <RapidStructureApp/TriggerElement.h>
#include <basic/util/fastdfs/FastFds.hpp>

class PerimeterElement : public basic::PipeElement {
public:
    PerimeterElement(int alarm_people_num = 1);

public:
    ~PerimeterElement();

    //是否有矩形
    bool hasRects();

    //设置ROI区域
    void setMask(std::string mask);

    void setImage(const cv::Mat &value);

    void setYoloObjects(std::vector<ScoredRect> value);

    //将检测结果放到此单元中
    //void setVptResults(const vector<VPT_ObjInfo> &value);
    void setObjsResults(const ::YoloDetect::ObjInfos &value);

    //设置置信度
    void setSensitivity(float value);

    //获取矩形框
    std::vector<cv::Rect2f> getRects() const;

    //获取YOLO检测的结果
    std::vector<ScoredRect> getLastScoreRects() const;

    int getRealNum() const;

    bool getTriggerState() const;

    void setFdfs(FastFdsWithLock *p_fdfsClient) {
        fdfsClient = p_fdfsClient;
    }


private:


    virtual void threadInitial() override;

    virtual void threadFunc() override;

    QJsonArray getJsonArrayFromQString(const QString &strJson);

    std::string uploadImgToFdfs(cv::Mat &image);

    bool saveInfoToEs(const std::string &imgUrl, const ::YoloDetect::ObjInfo &obj);


private:
    cv::Mat image;
    cv::Mat mask;
    //vector<VPT_ObjInfo> vptResults;
    ::YoloDetect::ObjInfos m_objs;
    float sensitivity;
    QPolygon mPolygon;
    TrackingTrigger *trackingTrigger;
    ::YoloDetect::stringData cocoData;
    int mRealNum;
    int m_nAlarmPeopleNum;
    std::vector<ScoredRect> mObjs;
    TriggerElement m_triggerElement;
    FastFdsWithLock *fdfsClient;
};

#endif // PAELEMENT_H