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 | 25 ++++++++++++++-----------
1 files changed, 14 insertions(+), 11 deletions(-)
diff --git a/csrc/wrapper.hpp b/csrc/wrapper.hpp
index 3e68b46..43e3f61 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;
@@ -34,7 +34,7 @@
std::unique_ptr<ffwrapper::FormatIn> init_reader(const char* input);
void init_worker(ffwrapper::FormatIn *in);
- int 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);
@@ -50,17 +50,24 @@
void CPUDec(){cpu_ = 1;}
void AudioSwitch(const bool a);
void SetRecMinCacheTime(const int mind);
+
+ 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_;
bool audio_;
int gb_, cpu_;
bool run_dec_;
+ bool run_stream_;
+ bool run_rec_;
// decoder 鍙傛暟
std::unique_ptr<std::thread> thread_;
std::atomic_bool stop_stream_;
@@ -74,18 +81,14 @@
// 褰曞儚璇锋眰缂撳瓨,绛夊緟runstream鍚庢坊鍔�
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);
-
- void *CreateConvertor(const int srcW, const int srcH, const int srcFormat,
- const int dstW, const int dstH, const int dstFormat, const int flag);
- uint8_t *Convert(void *h, uint8_t *src);
- void DestoryConvertor(void *h);
+ int GetGb28181Pic(const char *filename, char *retData, int *retDataLen);
}
#endif
\ No newline at end of file
--
Gitblit v1.8.0