From 54ea1c13885725584a6a50d520f67e8a75f85b6f Mon Sep 17 00:00:00 2001
From: zhangmeng <775834166@qq.com>
Date: 星期一, 19 八月 2019 11:33:13 +0800
Subject: [PATCH] fix rec

---
 csrc/wrapper.hpp |   38 ++++++++++++++++++++++++--------------
 1 files changed, 24 insertions(+), 14 deletions(-)

diff --git a/csrc/wrapper.hpp b/csrc/wrapper.hpp
index 8bbb157..66f2aca 100644
--- a/csrc/wrapper.hpp
+++ b/csrc/wrapper.hpp
@@ -35,6 +35,8 @@
         unsigned char *data;
         int w;
         int h;
+
+        int64_t id;
     }pic_bgr24;
 
     namespace buz{
@@ -42,11 +44,14 @@
         struct avpacket;
     }
 
-    typedef std::function<std::shared_ptr<buz::Recorder>(ffwrapper::FormatIn*)> FN_REC;
+    // typedef std::function<std::shared_ptr<buz::Recorder>(ffwrapper::FormatIn*)> FN_REC;
 
     typedef struct _fn_rec{
-        FN_REC fn_init;
-        std::shared_ptr<buz::Recorder> rec;    
+        std::string rid;
+        std::string dir;
+        int min;
+        int max;
+        std::unique_ptr<buz::Recorder> rec;    
     }FnRec;
 
     class Wrapper{
@@ -58,10 +63,10 @@
             std::unique_ptr<ffwrapper::FormatIn> init_reader(const char* input);
             // ffwrapper::FormatIn* init_reader_gb28181(const char* input);
             void run_worker(ffwrapper::FormatIn *in, buz::avpacket &pkt);
-            std::shared_ptr<buz::Recorder> init_recorder(ffwrapper::FormatIn *in, std::string id,std::string dir, const int mind, const int maxd);
+            std::unique_ptr<buz::Recorder> init_recorder(ffwrapper::FormatIn *in, std::string id,std::string dir, const int mind, const int maxd);
 
             void cache_rec_info(std::string &id, int &index, std::string &path);
-            void cache_pic(std::shared_ptr<ffwrapper::FrameData> &frame);
+            void cache_pic(std::shared_ptr<ffwrapper::FrameData> &frame, int64_t &id);
 
             void cacheAVPacket(const AVPacket &pkt);
         public: 
@@ -74,19 +79,15 @@
             int FireRecorder(const char* sid,const int64_t &id);
             void GetInfoRecorder(int &index, std::string &path);
             std::string GetRecorderID(const std::string &path);
-            // active api
-            void ActiveRecorder(const char *dir, const int mind, const int maxd,
-                                FUNC_REC func);
 
             void ScalePicture(const int w, const int h, const int flags);
             void UseGB28181();
             void UseCPU();
         public: //decoder
             void BuildDecoder();
-            void GetPicDecoder(unsigned char **data, int *w, int *h);
+            void GetPicDecoder(unsigned char **data, int *w, int *h, int64_t *id);
             void GetPacket(unsigned char **pktData, int *size, int *key);
             //active api
-            void ActiveDecoder(FUNC_DEC fn);
 
         private:
 
@@ -104,6 +105,7 @@
             struct record_file_info{
                 int file_frame_index;
                 std::string file_path;
+                std::string rec_id;
             };
             std::list<struct record_file_info>  list_rec_;
             std::unordered_map<std::string, std::string> list_rec_map_;
@@ -116,12 +118,20 @@
             std::list<AVPacket> list_avpkt_;
             std::mutex mutex_avpkt_;
 
-            // active api
-            FUNC_REC func_rec_;
-            FUNC_DEC func_dec_;
-
             int scale_w_, scale_h_, scale_f_;
             int gb_, cpu_;
+
+            typedef struct recpkt{
+                std::shared_ptr<ffwrapper::CodedData> data;
+                int64_t id;
+            }recpkt;
+            std::list<recpkt> list_rec_pkt_;
+            std::mutex mtx_rec_pkt_;
+            int cache_rec_pkt(const buz::avpacket &pkt);
+            void maybe_dump_rec_pkt();
+            int     maxduration;
+            int     minduration;
+
         //////////////////test frame to bgr24
         public:
             uint8_t *decodeJPEG(const char *file, int *w, int *h);

--
Gitblit v1.8.0