From f946a62d3921e86b44ff8e2973138304b9cd53cd Mon Sep 17 00:00:00 2001
From: chenshijun <csj_sky@126.com>
Date: 星期二, 16 四月 2019 16:36:32 +0800
Subject: [PATCH] 解决修改参数和推流画面跳跃问题

---
 QiaoJiaSystem/VideoToImageMulth/RtspCaptureElement.cpp |   58 ++++++++++++++++++++++++++++++++++++++++++++++++++--------
 1 files changed, 50 insertions(+), 8 deletions(-)

diff --git a/QiaoJiaSystem/VideoToImageMulth/RtspCaptureElement.cpp b/QiaoJiaSystem/VideoToImageMulth/RtspCaptureElement.cpp
index b8d983d..7316f37 100644
--- a/QiaoJiaSystem/VideoToImageMulth/RtspCaptureElement.cpp
+++ b/QiaoJiaSystem/VideoToImageMulth/RtspCaptureElement.cpp
@@ -9,6 +9,7 @@
 #include "RtspAnalysManager.h"
 //#include <basic/pipe_element/ffmpeg/cap_ffmpeg_impl.hpp>
 #include "../../../BasicPlatForm/basic/pipe_element/ffmpeg/FfmpegElement.cpp"
+#include <basic/util/app/AppConfig.h>
 
 RtspCaptureElement::RtspCaptureElement(const std::string &path, const std::string &camId, int fps, int reopenTime,
                                        int gpuIndex, RtspAnalysManager *manager) :
@@ -38,25 +39,58 @@
         }
     }
 
-    m_picCount++;
+    u_char *data;
+    int width = 0, height = 0, step = 0, cn = 0;
+    cv::Mat copyMat;
+
+    //#publish Video
+    if (m_publishVideoRet) {
+        if (videoPublishElement == nullptr) {
+            string path = appConfig.getStringProperty("srsAddr") + "cam" + m_camId + ".flv";
+            cv::Size size_(appConfig.getIntProperty("pulish.width"), appConfig.getIntProperty("pulish.height"));
+            int gupIdx = appPref.getIntData("gpu.index");
+            videoPublishElement = new ffmpeg::PipeVideoPublishElement(path, size_, "flv", 25, gupIdx);
+            videoPublishElement->start();
+        } else {
+//            DBG("videoPublishElement->setImage()");
+            m_capture->retrieveFrame(0, &data, &step, &width, &height, &cn);
+            cv::Mat img(height, width, CV_8UC3, data, step);
+            img.copyTo(copyMat);
+
+            if (!copyMat.empty()) {
+                videoPublishElement->setImage(copyMat);
+                videoPublishElement->submit();
+            } else {
+                DBG("copyMat.empty()");
+            }
+        }
+    } else {
+        if (videoPublishElement != nullptr) {
+            DBG("videoPublishElement->stop()");
+            videoPublishElement->stop();
+            videoPublishElement->wait();
+            delete videoPublishElement;
+            videoPublishElement = nullptr;
+        }
+    }
+
     //鍑犲紶閫変竴寮犳斁鍏edis
+    m_picCount++;
     if (m_picCount % m_nPicsPickOne != 0) {
         return;
     } else {
         m_picCount.store(0);
     }
 
-    {
-        u_char *data;
-        int width = 0, height = 0, step = 0, cn = 0;
+    if (copyMat.empty()) {
+//        ERR("copyMat.empty()");
         m_capture->retrieveFrame(0, &data, &step, &width, &height, &cn);
-
         cv::Mat img(height, width, CV_8UC3, data, step);
-        cv::Mat copyMat;
         img.copyTo(copyMat);
-        std::string imageName = m_capture->GetImageName();
-        m_pManager->SaveImageToRedis(m_camId, imageName, copyMat);
     }
+    std::string imageName = m_capture->GetImageName();
+    m_pManager->SaveImageToRedis(m_camId, imageName, copyMat);
+
     fireConnectors();
 }
 
@@ -114,3 +148,11 @@
     m_capture->SetMinMaxVideoSeconds(minSeconds, maxSeconds);
 }
 
+void RtspCaptureElement::startPublishVideo() {
+    m_publishVideoRet = true;
+}
+
+void RtspCaptureElement::stopPublishVideo() {
+    m_publishVideoRet = false;
+}
+

--
Gitblit v1.8.0