zhangjixing

2019-01-15 abd9036f024a77d22b47a108bc88b27b9732e3c5

 QiaoJiaSystem/StructureApp/YoloRpcElement.h

@@ -8,31 +8,22 @@
#include <basic/util/fastdfs/FastFds.hpp>
#include <RapidStructureApp/TriggerElement.h>
#include "TrackingTrigger.h"

#include "DBStruct.h"
class QSharedMemory;

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

    ~YoloRpcElement();

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

     cv::Mat getImage() const;

    ::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;

@@ -41,12 +32,10 @@
    cv::Mat image;
    QSharedMemory *sharedMemory;
    ::YoloDetect::ObjInfos objs;
    TrackingTrigger *trackingTrigger;
    std::vector<cv::Mat> triggerMats;
    ::YoloDetect::stringData cocoData;
    FastFdsWithLock *fdfsClient;
    std::vector<ScoredRect> triggerScoredRects;
   TrackingTrigger* trackingTrigger;

    TriggerElement m_triggerElement;
};

#endif // YOLORPCELEMENT_H