add log

zhangmeng

2019-09-16 2e1a71f9cd9d60c6ac1f09537d9649f96f1df811

 csrc/wrapper.hpp

@@ -1,6 +1,10 @@
#ifndef _cffmpeg_wrapper_hpp_
#define _cffmpeg_wrapper_hpp_

extern "C"{
#include <libavcodec/avcodec.h>
}

#include <stdint.h>

#include <string>
@@ -15,35 +19,17 @@

namespace ffwrapper{
    class FormatIn;
    class FormatOut;
    
    class VideoProp;
    class CodedData;
    class FrameData;

    class cvbridge;
}

enum WORKER{
    WORKER_RECORDER = 0,
    WORKER_DECODER,
    WORKER_ENCODER,
};

namespace cffmpeg_wrap{

    typedef std::function<int(ffwrapper::FormatIn*)> FUNC_WORKER;
    class stream;
    class decoder;
    class rec;

    typedef struct _pic_bgr24{
        unsigned char *data;
        int w;
        int h;
    }pic_bgr24;

    namespace buz{
        class Recorder;
        struct avpacket;
    }
    class Wrapper{
        public:
            Wrapper();
@@ -51,75 +37,50 @@

        private: 
            std::unique_ptr<ffwrapper::FormatIn> init_reader(const char* input);
            // ffwrapper::FormatIn* init_reader_gb28181(const char* input);
            void init_worker(ffwrapper::FormatIn *in);
            int init_recorder(ffwrapper::FormatIn *in, std::string dir, const int mind, const int maxd);
            void run_worker(ffwrapper::FormatIn *in, buz::avpacket &pkt);

            void cache_rec_info(int &index, std::string &path);
            void cache_pic(std::shared_ptr<ffwrapper::FrameData> &frame);
            void init_stream();
            void init_decoder(ffwrapper::FormatIn *in);
            void init_rec(ffwrapper::FormatIn *in);
            
        public: 
            int RunStream(const char* input);
        private: 
            void run_stream_thread();

        public: //recorder
            void BuildRecorder(const char *dir, const int mind, const int maxd);
            int FireRecorder(const int64_t &id);
            void GetInfoRecorder(int &index, std::string &path);

            // active api
            void ActiveRecorder(const char *dir, const int mind, const int maxd,
                                FUNC_REC func);
            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 &recID, int &index, 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);
            //active api
            void ActiveDecoder(FUNC_DEC fn);

        public:
            void BuildEncoder(const char *file, const int w, const int h, const int fps, const int br, const int gi);
            int init_encoder(ffwrapper::FormatIn *in, const char *file, const int w, const int h, const int fps, const int br, const int gi);
            void GetPicDecoder(unsigned char **data, int *w, int *h, int64_t *id);
            void GetPacket(unsigned char **pktData, int *size, int *key);

        private:

            // stream 参数
            std::string input_url_;
            buz::Recorder  *recorder_;
            int scale_w_, scale_h_, scale_f_;
            int gb_, cpu_;
            bool use_decoder_;

            // decoder 参数
            std::unique_ptr<std::thread> thread_;
            std::atomic_bool    stop_stream_;

            ffwrapper::cvbridge *bridge_;
            std::unordered_map<int, FUNC_WORKER> map_workers_;
            // 业务类
            // 推流类
            stream* stream_;
            // 解码类
            decoder* decoder_;
            // 录像类
            rec* rec_;

            //passive api
            struct record_file_info{
                int file_frame_index;
                std::string file_path;
            };
            std::list<struct record_file_info>  list_rec_;
            std::mutex mutex_rec_;

            std::list<pic_bgr24> list_pic_;
            std::mutex mutex_pic_;
            // 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);

        private:
            ffwrapper::FormatOut *encoder_;
    };

    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);