#ifndef YOLORPCELEMENT_H
#define YOLORPCELEMENT_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 <TriggerElement.h>
//#include "../StructureApp/YoloRpcElement.h"
#include "../StructureApp/TrackingTrigger.h"

class QSharedMemory;

class YoloRpcElement : public basic::PipeElement {
public:
    YoloRpcElement(string);

    ~YoloRpcElement();

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

    ::YoloDetect::ObjInfos getObjects() const;

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

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

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

    bool getTriggerState() const;

    bool getTrigger() const;

private:
    virtual void threadFunc() override;

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

    TriggerElement m_triggerElement;
};

#endif // YOLORPCELEMENT_H