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