From d0b8204e3939e1e2d841a90ea254bc4b91309ab5 Mon Sep 17 00:00:00 2001
From: zhangmeng <775834166@qq.com>
Date: 星期日, 19 一月 2020 13:32:13 +0800
Subject: [PATCH] update ffmepg gpu balance

---
 csrc/wrapper.hpp |    6 +++++-
 1 files changed, 5 insertions(+), 1 deletions(-)

diff --git a/csrc/wrapper.hpp b/csrc/wrapper.hpp
index 60e1ac8..43e3f61 100644
--- a/csrc/wrapper.hpp
+++ b/csrc/wrapper.hpp
@@ -53,11 +53,13 @@
 
         int GetFPS(){return fps_;}
     public: //decoder
-        void BuildDecoder();
+        void OpenDecoder();
         void GetPicDecoder(unsigned char **data, int *w, int *h, int *format, int *length, int64_t *id);
     public: // push stream
         void CloseStream();
         void GetPacket(unsigned char **pktData, int *size, int *key);
+    public: // recorder
+        void OpenRecorder();
     private:
         // stream 鍙傛暟
         std::string input_url_;
@@ -65,6 +67,7 @@
         int gb_, cpu_;
         bool run_dec_;
         bool run_stream_;
+        bool run_rec_;
         // decoder 鍙傛暟
         std::unique_ptr<std::thread> thread_;
         std::atomic_bool    stop_stream_;
@@ -85,6 +88,7 @@
                         const int pix_fmt, const int scale_flag, const int gi);
     void DestroyEncoder(void *h);
     int Encode(void *hdl, uint8_t *in, const int w, const int h, uint8_t **out, int *size, int *key);
+    int GetGb28181Pic(const char *filename, char *retData, int *retDataLen);
 }
 
 #endif
\ No newline at end of file

--
Gitblit v1.8.0