#ifndef _cffmpeg_wrapper_hpp_ #define _cffmpeg_wrapper_hpp_ extern "C"{ #include } #include #include #include #include #include #include #include #include #include "common/callback.hpp" namespace ffwrapper{ class FormatIn; class FormatOut; class VideoProp; class CodedData; class FrameData; class cvbridge; } namespace cffmpeg_wrap{ typedef struct _pic_bgr24{ unsigned char *data; int w; int h; }pic_bgr24; namespace buz{ class Recorder; struct avpacket; } typedef std::function(ffwrapper::FormatIn*)> FN_REC; typedef struct _fn_rec{ FN_REC fn_init; std::shared_ptr rec; }FnRec; class Wrapper{ public: Wrapper(); ~Wrapper (); private: std::unique_ptr 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 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 &frame); 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(std::string &sid, int &index, std::string &path); // active api void ActiveRecorder(const char *dir, const int mind, const int maxd, FUNC_REC func); 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); void GetPacket(unsigned char **pktData, int *size, int *key); //active api void ActiveDecoder(FUNC_DEC fn); private: std::string input_url_; std::unique_ptr thread_; std::atomic_bool stop_stream_; ffwrapper::cvbridge *bridge_; bool use_decoder_; std::unordered_map map_rec_; //passive api struct record_file_info{ std::string id; int file_frame_index; std::string file_path; }; std::list list_rec_; std::mutex mutex_rec_; std::list list_pic_; std::mutex mutex_pic_; std::list list_avpkt_; std::mutex mutex_avpkt_; // active api FUNC_REC func_rec_; FUNC_DEC func_dec_; int scale_w_, scale_h_, scale_f_; int gb_, cpu_; //////////////////test frame to bgr24 public: uint8_t *decodeJPEG(const char *file, 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); } #endif