From 0560a8fce4c9d005655fce5025bc0fda8c58bff8 Mon Sep 17 00:00:00 2001 From: zhangmeng <775834166@qq.com> Date: 星期一, 02 十二月 2019 16:15:47 +0800 Subject: [PATCH] update --- csrc/wrapper.hpp | 32 ++++++++++++++++++-------------- 1 files changed, 18 insertions(+), 14 deletions(-) diff --git a/csrc/wrapper.hpp b/csrc/wrapper.hpp index a6ebe55..60e1ac8 100644 --- a/csrc/wrapper.hpp +++ b/csrc/wrapper.hpp @@ -11,7 +11,7 @@ #include <memory> #include "common/callback.hpp" - +class CPacket; namespace ffwrapper{ class FormatIn; @@ -28,40 +28,43 @@ class Wrapper{ public: Wrapper(); + explicit Wrapper(const char *logfile); ~Wrapper (); private: std::unique_ptr<ffwrapper::FormatIn> init_reader(const char* input); void init_worker(ffwrapper::FormatIn *in); - void run_worker(ffwrapper::FormatIn *in, std::shared_ptr<ffwrapper::CodedData> data, int64_t &id); + int run_worker(ffwrapper::FormatIn *in, const CPacket &pkt); void deinit_worker(); public: int RunStream(const char* input); private: void run_stream_thread(); public: //recorder - void BuildRecorder(const char* id,const char *dir, const int mind, const int maxd, const bool audio); + void BuildRecorder(const char* id, const char *dir, const int64_t &fid, const int mind, const int maxd, const bool audio); 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){ - scale_w_ = w; - scale_h_ = h; - scale_f_ = flags; - } + + public: void GB28181(){gb_ = 1;} void CPUDec(){cpu_ = 1;} void AudioSwitch(const bool a); + void SetRecMinCacheTime(const int mind); + + int GetFPS(){return fps_;} public: //decoder void BuildDecoder(); - void GetPicDecoder(unsigned char **data, int *w, int *h, int64_t *id); + void GetPicDecoder(unsigned char **data, int *w, int *h, int *format, int *length, int64_t *id); public: // push stream + void CloseStream(); void GetPacket(unsigned char **pktData, int *size, int *key); private: // stream 鍙傛暟 std::string input_url_; - int scale_w_, scale_h_, scale_f_; bool audio_; int gb_, cpu_; + bool run_dec_; + bool run_stream_; // decoder 鍙傛暟 std::unique_ptr<std::thread> thread_; std::atomic_bool stop_stream_; @@ -73,14 +76,15 @@ // 褰曞儚绫�,涓�鐩村瓨鍦� rec* rec_; // 褰曞儚璇锋眰缂撳瓨,绛夊緟runstream鍚庢坊鍔� - std::function<void(ffwrapper::FormatIn*)> fn_rec_lazy_; + std::function<void()> fn_rec_lazy_; + bool logit_; + int fps_; }; - uint8_t* Decode(const char *file, const int gb, 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 *CreateEncoder(const int w, const int h, const int fps, const int br, + const int pix_fmt, 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