split scale

zhangmeng

2019-10-09 68a19a73681301c6712e10d55bc64324716dbd24

 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>
@@ -16,11 +12,11 @@
#include "common/callback.hpp"



namespace ffwrapper{
    class FormatIn;
    
    class VideoProp;
    class CodedData;
    class VideoProp;
}

namespace cffmpeg_wrap{
@@ -32,6 +28,7 @@
    class Wrapper{
    public:
        Wrapper();
        explicit Wrapper(const char *logfile);
        ~Wrapper ();
    private: 
        std::unique_ptr<ffwrapper::FormatIn> init_reader(const char* input);
@@ -47,25 +44,21 @@
        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 ScalePicture(const int w, const int h, const int flags){
            scale_w_ = w;
            scale_h_ = h;
            scale_f_ = flags;
        }
        
        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, int64_t *id);
        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_;
        int scale_w_, scale_h_, scale_f_;
        bool audio_;
        int gb_, cpu_;
        bool run_dec_;
        // decoder 参数
        std::unique_ptr<std::thread> thread_;
        std::atomic_bool    stop_stream_;
@@ -77,14 +70,22 @@
        // 录像类,一直存在
        rec* rec_;
        // 录像请求缓存,等待runstream后添加
        std::function<void(ffwrapper::FormatIn*)> fn_rec_lazy_;
        std::function<void()> fn_rec_lazy_;
        bool logit_;
    };

    uint8_t *DecodeJPEG(const char *file, int *w, int *h);
    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