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.cpp | 45 +++++++++++++++++++++++++++++++++------------ 1 files changed, 33 insertions(+), 12 deletions(-) diff --git a/csrc/wrapper.cpp b/csrc/wrapper.cpp index e26c1f4..3bf1b57 100644 --- a/csrc/wrapper.cpp +++ b/csrc/wrapper.cpp @@ -48,14 +48,29 @@ Wrapper::~Wrapper() { - if(thread_){ - stop_stream_.store(true); - thread_->join(); + try + { + if(thread_){ + stop_stream_.store(true); + thread_->join(); + } + if(bridge_){ + delete bridge_; bridge_ = NULL; + } + + map_rec_.clear(); + list_rec_pkt_.clear(); + + for(auto &i : list_pic_){ + free(i.data); + } } - if(bridge_){ - delete bridge_; bridge_ = NULL; + catch(const std::exception& e) + { + logIt("WRAPPER EXCEPTION: ", e.what()); } - + + } void Wrapper::ScalePicture(const int w, const int h, const int flags){ @@ -199,7 +214,7 @@ avpacket p = {k.data, k.id}; i.second.rec->CachePacket(p); } - logIt("start rec %d frames", list_rec_pkt_.size()); + logIt("START REC %d FRAMES", list_rec_pkt_.size()); } }else if (i.second.rec){ i.second.rec->CachePacket(pkt); @@ -240,13 +255,13 @@ } //////////////recorder - std::shared_ptr<Recorder> Wrapper::init_recorder(FormatIn *in, std::string id,std::string dir, const int mind, const int maxd){ + std::shared_ptr<Recorder> Wrapper::init_recorder(FormatIn *in, std::string id, std::string dir, const int mind, const int maxd){ if(!in){ logIt("Init wrapper first"); return nullptr; } - auto rec = std::make_shared<Recorder>(in, id); + auto rec = std::make_shared<Recorder>(in, id.c_str()); rec->SetCallback([&](std::string &id, int &index, std::string &path){ cache_rec_info(id, index, path); @@ -267,6 +282,7 @@ void Wrapper::BuildRecorder(const char* id, const char *output, const int mindur, const int maxdur){ std::string rid(id); std::string dir(output); + auto fn = [=](FormatIn *in){ return init_recorder(in, rid, dir, mindur, maxdur); }; @@ -286,7 +302,6 @@ iter->second.rec->FireRecorder(id); } } - logIt("FIRE REC %s, FRAME ID: %d", sid, id); } void Wrapper::cache_rec_info(std::string &id, int &index, std::string &path){ @@ -300,10 +315,11 @@ struct record_file_info info; info.file_frame_index = index; info.file_path = path; + info.rec_id = id; list_rec_.emplace_back(info); list_rec_map_[path] = id; - logIt("list rec files count : %d", list_rec_.size()); - map_rec_.erase(id); + logIt("LIST REC FILES COUNT : %d", list_rec_.size()); + } void Wrapper::GetInfoRecorder(int &index, std::string &path){ @@ -317,6 +333,9 @@ index = info.file_frame_index; path = info.file_path; list_rec_.pop_front(); + + if (map_rec_.find(info.rec_id) != map_rec_.end()) + map_rec_.erase(info.rec_id); // logIt("go get info index: %d, file: %s\n", index, path.c_str()); } @@ -327,6 +346,8 @@ ret = iter->second; list_rec_map_.erase(iter); } + + return ret; } ////////decoder -- Gitblit v1.8.0