From 68a19a73681301c6712e10d55bc64324716dbd24 Mon Sep 17 00:00:00 2001
From: zhangmeng <775834166@qq.com>
Date: 星期三, 09 十月 2019 15:38:47 +0800
Subject: [PATCH] split scale
---
csrc/wrapper.hpp | 169 +++++++++++++++++++-------------------------------------
1 files changed, 58 insertions(+), 111 deletions(-)
diff --git a/csrc/wrapper.hpp b/csrc/wrapper.hpp
index b11c9ca..13d5f7d 100644
--- a/csrc/wrapper.hpp
+++ b/csrc/wrapper.hpp
@@ -1,10 +1,6 @@
#ifndef _cffmpeg_wrapper_hpp_
#define _cffmpeg_wrapper_hpp_
-extern "C"{
-#include <libavcodec/avcodec.h>
-}
-
#include <stdint.h>
#include <string>
@@ -12,133 +8,84 @@
#include <thread>
#include <atomic>
#include <mutex>
-#include <unordered_map>
#include <memory>
#include "common/callback.hpp"
+
namespace ffwrapper{
class FormatIn;
- class FormatOut;
-
- class VideoProp;
class CodedData;
- class FrameData;
-
- class cvbridge;
+ class VideoProp;
}
namespace cffmpeg_wrap{
-
- typedef struct _pic_bgr24{
- unsigned char *data;
- int w;
- int h;
-
- int64_t id;
- }pic_bgr24;
-
- namespace buz{
- class Recorder;
- struct avpacket;
- }
-
- 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;
- }FnRec;
+ class stream;
+ class decoder;
+ class rec;
class Wrapper{
- public:
- Wrapper();
- ~Wrapper ();
+ public:
+ Wrapper();
+ explicit Wrapper(const char *logfile);
+ ~Wrapper ();
+ private:
+ std::unique_ptr<ffwrapper::FormatIn> init_reader(const char* input);
- private:
- 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);
-
- void cache_rec_info(std::string &id, int &index, std::string &path);
- void cache_pic(std::shared_ptr<ffwrapper::FrameData> &frame, int64_t &id);
-
- void cacheAVPacket(const AVPacket &pkt);
- 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);
- int FireRecorder(const char* sid,const int64_t &id);
- void GetInfoRecorder(int &index, std::string &path);
- std::string GetRecorderID(const std::string &path);
-
- 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, int64_t *id);
- void GetPacket(unsigned char **pktData, int *size, int *key);
- //active api
-
- private:
-
- std::string input_url_;
-
- std::unique_ptr<std::thread> thread_;
- std::atomic_bool stop_stream_;
-
- ffwrapper::cvbridge *bridge_;
-
- bool use_decoder_;
- std::unordered_map<std::string, FnRec> map_rec_;
-
- //passive api
- 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_;
-
- std::mutex mutex_rec_;
-
- std::list<pic_bgr24> list_pic_;
- std::mutex mutex_pic_;
-
- std::list<AVPacket> list_avpkt_;
- std::mutex mutex_avpkt_;
-
- 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);
-
+ void init_worker(ffwrapper::FormatIn *in);
+ void run_worker(ffwrapper::FormatIn *in, std::shared_ptr<ffwrapper::CodedData> data, int64_t &id);
+ 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);
+ int FireRecorder(const char* sid,const int64_t &id);
+ void GetInfoRecorder(std::string &recID, int &index, std::string &path);
+
+ void GB28181(){gb_ = 1;}
+ void CPUDec(){cpu_ = 1;}
+ void AudioSwitch(const bool a);
+ public: //decoder
+ void BuildDecoder();
+ void GetPicDecoder(unsigned char **data, int *w, int *h, int *format, int *length, int64_t *id);
+ public: // push stream
+ void GetPacket(unsigned char **pktData, int *size, int *key);
+ private:
+ // stream 鍙傛暟
+ std::string input_url_;
+ bool audio_;
+ int gb_, cpu_;
+ bool run_dec_;
+ // decoder 鍙傛暟
+ std::unique_ptr<std::thread> thread_;
+ std::atomic_bool stop_stream_;
+ // 涓氬姟绫�
+ // 鎺ㄦ祦绫�
+ stream* stream_;
+ // 瑙ｇ爜绫�
+ decoder* decoder_;
+ // 褰曞儚绫�,涓�鐩村瓨鍦�
+ rec* rec_;
+ // 褰曞儚璇锋眰缂撳瓨,绛夊緟runstream鍚庢坊鍔�
+ std::function<void()> fn_rec_lazy_;
+ bool logit_;
};
+
+ 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 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 flag);
+ uint8_t *Convert(void *h, uint8_t *src);
+ void DestoryConvertor(void *h);
+
+ uint8_t* ConvertYUV2BGR(uint8_t *src, const int w, const int h, const int dst_w, const int dst_h, int *length);
}
#endif
\ No newline at end of file
--
Gitblit v1.8.0