From 4eab3ca9bf020df2bb53fa2a36aa4f57d684225c Mon Sep 17 00:00:00 2001
From: zhangmeng <775834166@qq.com>
Date: 星期三, 07 八月 2019 17:25:03 +0800
Subject: [PATCH] add rec notify, try catch
---
csrc/wrapper.hpp | 66 +++++++++++++++++---------------
1 files changed, 35 insertions(+), 31 deletions(-)
diff --git a/csrc/wrapper.hpp b/csrc/wrapper.hpp
index db45ea7..b11c9ca 100644
--- a/csrc/wrapper.hpp
+++ b/csrc/wrapper.hpp
@@ -28,26 +28,29 @@
class cvbridge;
}
-enum WORKER{
- WORKER_RECORDER = 0,
- WORKER_DECODER,
- WORKER_ENCODER,
-};
-
namespace cffmpeg_wrap{
- typedef std::function<int(ffwrapper::FormatIn*)> FUNC_WORKER;
typedef struct _pic_bgr24{
unsigned char *data;
int w;
int h;
+
+ int64_t id;
}pic_bgr24;
namespace buz{
class Recorder;
struct avpacket;
}
+
+ 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;
+ }FnRec;
+
class Wrapper{
public:
Wrapper();
@@ -56,12 +59,11 @@
private:
std::unique_ptr<ffwrapper::FormatIn> init_reader(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);
+ std::shared_ptr<buz::Recorder> init_recorder(ffwrapper::FormatIn *in, std::string id,std::string dir, const int mind, const int maxd);
- void cache_rec_info(int &index, std::string &path);
- void cache_pic(std::shared_ptr<ffwrapper::FrameData> &frame);
+ void cache_rec_info(std::string &id, int &index, std::string &path);
+ void cache_pic(std::shared_ptr<ffwrapper::FrameData> &frame, int64_t &id);
void cacheAVPacket(const AVPacket &pkt);
public:
@@ -70,45 +72,41 @@
void run_stream_thread();
public: //recorder
- void BuildRecorder(const char *dir, const int mind, const int maxd);
- int FireRecorder(const int64_t &id);
+ 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(int &index, std::string &path);
-
- // active api
- void ActiveRecorder(const char *dir, const int mind, const int maxd,
- FUNC_REC func);
+ std::string GetRecorderID(const 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);
+ 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);
-
- public:
- void BuildEncoder(const char *file, const int w, const int h, const int fps, const int br, const int gi);
- int init_encoder(ffwrapper::FormatIn *in, const char *file, const int w, const int h, const int fps, const int br, const int gi);
private:
std::string input_url_;
- buz::Recorder *recorder_;
std::unique_ptr<std::thread> thread_;
std::atomic_bool stop_stream_;
ffwrapper::cvbridge *bridge_;
- std::unordered_map<int, FUNC_WORKER> map_workers_;
+
+ bool use_decoder_;
+ std::unordered_map<std::string, FnRec> map_rec_;
//passive api
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_;
+
std::mutex mutex_rec_;
std::list<pic_bgr24> list_pic_;
@@ -117,18 +115,24 @@
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);
- private:
- ffwrapper::FormatOut *encoder_;
};
void *CreateEncoder(const int w, const int h, const int fps, const int br, const int scale_flag, const int gi);
--
Gitblit v1.8.0