#ifndef PERSONELEMENT_H
#define PERSONELEMENT_H

#include <YoloServer.h>
#include <basic/pipe/PipeElement.h>
#include <basic/rpc/IceRpc.hpp>
#include <opencv2/opencv.hpp>
#include <basic/util/fastdfs/FastFds.hpp>
#include <RapidStructureApp/TriggerElement.h>
#include "TrackingTrigger.h"
#include "DBStruct.h"
class QSharedMemory;

class PersonElement : public basic::PipeElement {
public:
    PersonElement(){}
    PersonElement(const SdkRule& sdkRule);

    ~PersonElement();

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

    std::vector<cv::Mat> getTriggerMats();

    std::vector<ScoredRect> getLastScoreRects() const;

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

    bool getTriggerState() const;

    bool getTrigger() const;
    void setObjsResults(const std::vector<ScoredRect>& objs);

private:
    virtual void threadFunc() override;

private:
//    IceRpcClient<YoloDetect::YoloDetectServerPrx> rpcClient;
    cv::Mat image;
//    QSharedMemory *sharedMemory;
   // ::YoloDetect::ObjInfos objs;
    std::vector<ScoredRect> m_objs;
  //  TrackingTrigger *trackingTrigger;
    std::vector<cv::Mat> triggerMats;
    ::YoloDetect::stringData cocoData;
    FastFdsWithLock *fdfsClient;

    TriggerElement m_triggerElement;
    SdkRule m_sdkRule;
};

#endif // YOLORPCELEMENT_H