From 2e1a71f9cd9d60c6ac1f09537d9649f96f1df811 Mon Sep 17 00:00:00 2001
From: zhangmeng <775834166@qq.com>
Date: 星期一, 16 九月 2019 13:37:56 +0800
Subject: [PATCH] add log
---
csrc/wrapper.hpp | 96 +++++++++++++++++++-----------------------------
1 files changed, 38 insertions(+), 58 deletions(-)
diff --git a/csrc/wrapper.hpp b/csrc/wrapper.hpp
index 10f699f..d398ed4 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>
@@ -15,33 +19,17 @@
namespace ffwrapper{
class FormatIn;
-
+
class VideoProp;
class CodedData;
- class FrameData;
-
- class cvbridge;
}
-
-enum WORKER{
- WORKER_RECORDER = 0,
- WORKER_DECODER,
-};
namespace cffmpeg_wrap{
- typedef std::function<int(ffwrapper::FormatIn*)> FUNC_WORKER;
+ class stream;
+ class decoder;
+ class rec;
- typedef struct _pic_bgr24{
- unsigned char *data;
- int w;
- int h;
- }pic_bgr24;
-
- namespace buz{
- class Recorder;
- struct avpacket;
- }
class Wrapper{
public:
Wrapper();
@@ -49,62 +37,54 @@
private:
std::unique_ptr<ffwrapper::FormatIn> init_reader(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 init_stream();
+ void init_decoder(ffwrapper::FormatIn *in);
+ void init_rec(ffwrapper::FormatIn *in);
+
public:
int RunStream(const char* input);
private:
void run_stream_thread();
public: //recorder
- void BuildRecorder(const char *dir, const int mind, const int maxd);
- int FireRecorder(const int64_t &id);
- void GetInfoRecorder(int &index, std::string &path);
-
- // active api
- void ActiveRecorder(const char *dir, const int mind, const int maxd,
- FUNC_REC func);
+ void BuildRecorder(const char* id,const char *dir, const int mind, const int maxd);
+ int FireRecorder(const char* sid,const int64_t &id);
+ void GetInfoRecorder(std::string &recID, int &index, std::string &path);
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);
- //active api
- void ActiveDecoder(FUNC_DEC fn);
+ void GetPicDecoder(unsigned char **data, int *w, int *h, int64_t *id);
+ void GetPacket(unsigned char **pktData, int *size, int *key);
+
private:
-
+ // stream 鍙傛暟
std::string input_url_;
- buz::Recorder *recorder_;
+ int scale_w_, scale_h_, scale_f_;
+ int gb_, cpu_;
+ bool use_decoder_;
+ // decoder 鍙傛暟
std::unique_ptr<std::thread> thread_;
std::atomic_bool stop_stream_;
- ffwrapper::cvbridge *bridge_;
- std::unordered_map<int, FUNC_WORKER> map_workers_;
+ // 涓氬姟绫�
+ // 鎺ㄦ祦绫�
+ stream* stream_;
+ // 瑙ｇ爜绫�
+ decoder* decoder_;
+ // 褰曞儚绫�
+ rec* rec_;
- //passive api
- struct record_file_info{
- int file_frame_index;
- std::string file_path;
- };
- std::list<struct record_file_info> list_rec_;
- std::mutex mutex_rec_;
-
- std::list<pic_bgr24> list_pic_;
- std::mutex mutex_pic_;
- // active api
- FUNC_REC func_rec_;
- FUNC_DEC func_dec_;
-
- int scale_w_, scale_h_, scale_f_;
- //////////////////test frame to bgr24
- public:
- uint8_t *decodeJPEG(const char *file, int *w, int *h);
};
+
+ uint8_t *DecodeJPEG(const char *file, int *w, int *h);
+ void *CreateEncoder(const int w, const int h, const int fps, const int br, 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);
+
}
#endif
\ No newline at end of file
--
Gitblit v1.8.0