From e84510254860e916cbb48ec3de4557b14a99d76c Mon Sep 17 00:00:00 2001
From: zhangmeng <775834166@qq.com>
Date: 星期五, 26 七月 2019 10:55:22 +0800
Subject: [PATCH] add log

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

diff --git a/csrc/wrapper.hpp b/csrc/wrapper.hpp
index 1145044..db45ea7 100644
--- a/csrc/wrapper.hpp
+++ b/csrc/wrapper.hpp
@@ -1,6 +1,10 @@
 #ifndef _cffmpeg_wrapper_hpp_
 #define _cffmpeg_wrapper_hpp_
 
+extern "C"{
+#include <libavcodec/avcodec.h>
+}
+
 #include <stdint.h>
 
 #include <string>
@@ -51,13 +55,15 @@
 
         private: 
             std::unique_ptr<ffwrapper::FormatIn> init_reader(const char* input);
-            ffwrapper::FormatIn* init_reader_gb28181(const char* input);
+            // ffwrapper::FormatIn* init_reader_gb28181(const char* input);
             void init_worker(ffwrapper::FormatIn *in);
             int init_recorder(ffwrapper::FormatIn *in, std::string dir, const int mind, const int maxd);
             void run_worker(ffwrapper::FormatIn *in, buz::avpacket &pkt);
 
             void cache_rec_info(int &index, std::string &path);
             void cache_pic(std::shared_ptr<ffwrapper::FrameData> &frame);
+
+            void cacheAVPacket(const AVPacket &pkt);
         public: 
             int RunStream(const char* input);
         private: 
@@ -73,9 +79,12 @@
                                 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 GetPacket(unsigned char **pktData, int *size, int *key);
             //active api
             void ActiveDecoder(FUNC_DEC fn);
 
@@ -104,11 +113,16 @@
 
             std::list<pic_bgr24> list_pic_;
             std::mutex mutex_pic_;
+
+            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_;
         //////////////////test frame to bgr24
         public:
             uint8_t *decodeJPEG(const char *file, int *w, int *h);

--
Gitblit v1.8.0