From d0b8204e3939e1e2d841a90ea254bc4b91309ab5 Mon Sep 17 00:00:00 2001 From: zhangmeng <775834166@qq.com> Date: 星期日, 19 一月 2020 13:32:13 +0800 Subject: [PATCH] update ffmepg gpu balance --- csrc/wrapper.hpp | 6 +++++- 1 files changed, 5 insertions(+), 1 deletions(-) diff --git a/csrc/wrapper.hpp b/csrc/wrapper.hpp index 60e1ac8..43e3f61 100644 --- a/csrc/wrapper.hpp +++ b/csrc/wrapper.hpp @@ -53,11 +53,13 @@ int GetFPS(){return fps_;} public: //decoder - void BuildDecoder(); + void OpenDecoder(); 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); + public: // recorder + void OpenRecorder(); private: // stream 鍙傛暟 std::string input_url_; @@ -65,6 +67,7 @@ int gb_, cpu_; bool run_dec_; bool run_stream_; + bool run_rec_; // decoder 鍙傛暟 std::unique_ptr<std::thread> thread_; std::atomic_bool stop_stream_; @@ -85,6 +88,7 @@ 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); + int GetGb28181Pic(const char *filename, char *retData, int *retDataLen); } #endif \ No newline at end of file -- Gitblit v1.8.0