From d3b7bbe7102cd089680a828f5d8f6402c8cf6342 Mon Sep 17 00:00:00 2001
From: pansen <pansen626@sina.com>
Date: 星期四, 07 三月 2019 14:43:28 +0800
Subject: [PATCH] GB28181集成完成,集成推流模块

---
 QiaoJiaSystem/YoloServer/ImageDrawElement.h |   30 ++++++++++++++++++++----------
 1 files changed, 20 insertions(+), 10 deletions(-)

diff --git a/QiaoJiaSystem/YoloServer/ImageDrawElement.h b/QiaoJiaSystem/YoloServer/ImageDrawElement.h
index 03fd9df..a9c2a51 100644
--- a/QiaoJiaSystem/YoloServer/ImageDrawElement.h
+++ b/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

--
Gitblit v1.8.0