video analysis2.0拆分,ffmpeg封装go接口库
zhangmeng
2019-05-13 b0914c9c39bd1226f1789cc3b53b05cb87797527
csrc/wrapper.cpp
@@ -33,9 +33,13 @@
    ,thread_(nullptr)
    ,stop_stream_(false)
    ,bridge_(NULL)
    ,scale_w_(0)
    ,scale_h_(0)
    ,scale_f_(SWS_POINT)
    {
        makeTheWorld();
    }
    Wrapper::~Wrapper()
    {
@@ -49,6 +53,12 @@
        if(recorder_){
            delete recorder_; recorder_ = NULL;
        }
    }
    void Wrapper::ScalePicture(const int w, const int h, const int flags){
        scale_w_ = w;
        scale_f_ = flags;
        scale_h_ = h;
    }
    std::unique_ptr<ffwrapper::FormatIn> Wrapper::init_reader(const char* input){
@@ -74,10 +84,12 @@
                auto dec_ctx = in->getCodecContext();
                if(bridge_)delete bridge_;
                scale_w_ = scale_w_ > dec_ctx->width ? dec_ctx->width : scale_w_;
                scale_h_ = scale_h_ > dec_ctx->height ? dec_ctx->height : scale_h_;
                AVPixelFormat pix_fmt = AV_PIX_FMT_BGR24;
                bridge_ = new cvbridge(
                                dec_ctx->width, dec_ctx->height, dec_ctx->pix_fmt,
                                dec_ctx->width, dec_ctx->height, pix_fmt, SWS_BICUBIC);
                        dec_ctx->width, dec_ctx->height, dec_ctx->pix_fmt,
                        scale_w_, scale_h_, pix_fmt, scale_f_);
    
            }
            if (!flag){
@@ -224,11 +236,12 @@
        pic_bgr24 pic;
        if(bridge_){
            AVFrame *frm = frame->getAVFrame();
            unsigned char *data = (unsigned char*)malloc(frm->width * frm->height * 3);
            pic.w = scale_w_;
            pic.h = scale_h_;
            unsigned char *data = (unsigned char*)malloc(pic.w * pic.h * 3);
            bridge_->copyPicture(data, frm);
            pic.data = data;
            pic.w = frm->width;
            pic.h = frm->height;
        }
        if(func_dec_){
            func_dec_(pic.data, pic.w, pic.h);