GB28181集成完成,集成推流模块

pansen

2019-03-07 d3b7bbe7102cd089680a828f5d8f6402c8cf6342

 QiaoJiaSystem/YoloServer/ImageDrawElement.h

@@ -1,20 +1,30 @@
#ifndef IMAGEDRAWELEMENT_H
#define IMAGEDRAWELEMENT_H
#include <basic/pipe_element/ImageFactoryElement.h>
#include <mutex>

class ImageDrawElement : public ImageFactoryElement
{
#include <YoloServer.h>
//#include <FaceServer.h>
#include <basic/pipe_element/ImageFactoryElement.h>
#include <basic/core/DoubleBufferedData.h>
#include "../StructureApp/TrackingTrigger.h"

class ImageDrawElement : public ImageFactoryElement {
public:
    ImageDrawElement();
    void setRects(const std::vector<cv::Rect2f> &value);

    // ImageFactoryElement interface
//    void setFaces(std::vector<ScoredRect> value);

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

private:
    std::mutex rectsMtx;
    std::vector<cv::Rect2f> rects;
    std::vector<cv::Rect2f> rectsBuffer;
    bool rectsUpdated;
    virtual void processImage(cv::Mat& image);
    virtual void processImage(cv::Mat &) override;

    void darwProperty(cv::Mat &image, string key, string value, int x, int y);

private:
//    DoubleBufferedData<std::vector<ScoredRect>> faces;
    DoubleBufferedData<std::vector<ScoredRect>> yoloObjects;
};


#endif // IMAGEDRAWELEMENT_H