GetYUV format is NV12

zhangmeng

2023-12-26 18a05d269516a5e33d8460291c2f93e73d95adce

 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,136 +8,96 @@
#include <thread>
#include <atomic>
#include <mutex>
#include <unordered_map>
#include <memory>
#include "common/callback.hpp"

class CPacket;

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{
        std::string rid;
        std::string dir;
        int min;
        int max;
        std::unique_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::unique_ptr<buz::Recorder> init_recorder(ffwrapper::FormatIn *in, std::string id,std::string dir, const int mind, const int maxd);
        void init_worker(ffwrapper::FormatIn *in);
        int run_worker(ffwrapper::FormatIn *in, const CPacket &pkt);
        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 int64_t &fid, const int mind, const int maxd, const bool audio);
        int FireRecorder(const char* sid,const int64_t &id);
        int GetInfoRecorder(std::string &recID, int &index, std::string &path);
    
    public:
        void gb28181(){gb_ = 1;}
        void CPUDec(){cpu_ = 1;}
        void AudioSwitch(const bool a);
        void SetRecMinCacheTime(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);
        int GetFPS(){return fps_;}
    public: //decoder
        void OpenDecoder();
        int GetPicDecoder(unsigned char **data, int *w, int *h, int *format, int *length, int64_t *id);
    public: // push stream
        void CloseStream();
        int GetPacket(unsigned char **pktData, int *size, int *key);
    public: // recorder
        void OpenRecorder();

            void cacheAVPacket(const AVPacket &pkt);
        public: 
            int RunStream(const char* input);
        private: 
            void run_stream_thread();
    public:
        // set device id
        int SetDevID(const int devid){ devid_ = devid; return 0;}

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

    private:
        // stream 参数
        std::string input_url_;
        bool audio_;
        int gb_, cpu_, devid_;
        bool run_dec_;
        bool run_stream_;
        bool run_rec_;
        bool work_start;
        // 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_;
        int fps_;
        // 为兼容文件解码做的补丁
        bool file_finished_;
        int min_duration_;
    };

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

    int GetGb28181Pic(const char *filename, char *retData, int *retDataLen, const int tt);
}

#endif