video analysis2.0拆分,ffmpeg封装go接口库
zhangmeng
2022-07-21 f346559ee07016172cab40b7fca5b2c5e09df899
csrc/wrapper.cpp
@@ -110,14 +110,16 @@
        VideoProp prop;
        prop.url_ = input;
        prop.rtsp_tcp_ = true;
        prop.gpu_acc_ = !cpu_;
        prop.gpu_acc_ = false;
        prop.gpu_index_ = devid_;
       std::unique_ptr<FormatIn> in(new FormatIn(prop, prop.gpuAccl()));
       int flag = -1;
#ifdef GB28181
        if (gb_){
            flag = in->openGb28181(input, NULL);
        }else{
#endif
           AVDictionary *avdic = prop.optsFormat();
            if(avdic){
                flag = in->open(input, &avdic);
@@ -125,7 +127,9 @@
           }else{
                flag = in->open(input, NULL);
            }
#ifdef GB28181
        }
#endif
       
        if(flag == 0){
            if(!in->findStreamInfo(NULL)){
@@ -200,11 +204,22 @@
        int64_t file_rebootime = 0;
        int64_t file_frame = 0;
        using namespace std;
        const string gb_suffix[] = {"/StreamType=2", "/StreamType=3", ""};
        const size_t gb_size = sizeof(gb_suffix) / sizeof(string);
        int gs_idx = 0;
        string url = input_url_;
        while(!stop_stream_.load()){
            auto in = init_reader(input_url_.c_str());
            if (gb_){
                if (input_url_.find("/StreamType=") == string::npos)
                    url = input_url_ + gb_suffix[gs_idx];
                logIt("======>>input real url %s\n", url.c_str());
            }
            auto in = init_reader(url.c_str());
            
            if (!in) {
                logIt("ERROR: init_reader! url: %s\n", input_url_.c_str());
                logIt("ERROR: init_reader! url: %s\n", url.c_str());
                sleep(2);
                continue;
            }
@@ -454,6 +469,7 @@
        return flag;
    }
#ifdef GB28181
    int GetGb28181Pic(const char *rtspUrl, char *retData, int *retDataLen, const int tt){
        int ret = 0;
        std::atomic_bool add_camera{false};
@@ -496,5 +512,6 @@
        t.detach();
        return -1;
    }
#endif
}