From 18a05d269516a5e33d8460291c2f93e73d95adce Mon Sep 17 00:00:00 2001
From: zhangmeng <775834166@qq.com>
Date: 星期二, 26 十二月 2023 10:45:31 +0800
Subject: [PATCH] GetYUV format is NV12

---
 libcffmpeg.c |  210 +++++++++++++++++++++++++++++++++++++++++----------
 1 files changed, 167 insertions(+), 43 deletions(-)

diff --git a/libcffmpeg.c b/libcffmpeg.c
index cb9edf7..279b292 100644
--- a/libcffmpeg.c
+++ b/libcffmpeg.c
@@ -17,30 +17,10 @@
 libcffmpeg init_libcffmpeg(const char *so_file){
     libcffmpeg lib = dlopen(so_file, RTLD_LAZY);
     if(lib){
-        fn_create = (lib_cffmpeg_create)dlsym(lib, "c_ffmpeg_create");
-        release_if_err(fn_create, lib);
+
         fn_destroy = (lib_cffmpeg_destroy)dlsym(lib, "c_ffmpeg_destroy");
         release_if_err(fn_destroy, lib);
-        fn_run = (lib_cffmpeg_run)dlsym(lib, "c_ffmpeg_run");
-        release_if_err(fn_run, lib);
-        fn_scale = (lib_cffmpeg_scale)dlsym(lib, "c_ffmpeg_scale");
-        release_if_err(fn_scale, lib);
-        fn_recorder = (lib_cffmpeg_recorder)dlsym(lib, "c_ffmpeg_build_recorder");
-        release_if_err(fn_recorder, lib);
-        fn_fire_recorder = (lib_cffmpeg_fire_recorder)dlsym(lib, "c_ffmpeg_fire_recorder");
-        release_if_err(fn_fire_recorder, lib);
-        fn_info_recorder = (lib_cffmpeg_info_recorder)dlsym(lib, "c_ffmpeg_get_info_recorder");
-        release_if_err(fn_info_recorder, lib);
-        fn_decoder = (lib_cffmpeg_decoder)dlsym(lib, "c_ffmpeg_build_decoder");
-        release_if_err(fn_decoder, lib);
-        fn_decoder_pic = (lib_cffmpeg_pic)dlsym(lib, "c_ffmpeg_get_pic_decoder");
-        release_if_err(fn_decoder_pic, lib);
-        fn_active_recorder = (lib_cffmpeg_active_recorder)dlsym(lib, "c_ffmpeg_active_recorder");
-        release_if_err(fn_active_recorder, lib);
-        fn_active_decoder = (lib_cffmpeg_active_decoder)dlsym(lib, "c_ffmpeg_active_decoder");
-        release_if_err(fn_active_decoder, lib);
-        fn_dec_jpeg = (lib_cffmpeg_decode_jpeg)dlsym(lib, "c_ffmpeg_decode_jpeg");
-        release_if_err(fn_dec_jpeg, lib);
+
     }else{
         printf("dlopen - %s\n", dlerror());  
     }
@@ -53,50 +33,194 @@
     }
 }
 
-cffmpeg wrap_fn_create(){
+cffmpeg wrap_fn_create(void *lib){
+    if (!fn_create){
+        fn_create = (lib_cffmpeg_create)dlsym(lib, "c_ffmpeg_create");
+        release_if_err(fn_create, lib);
+    }
+
     return fn_create();
 }
 
-void wrap_fn_destroy(const cffmpeg h){
+cffmpeg wrap_fn_create2(void *lib, const char *logfile){
+    if (!fn_create2){
+        fn_create2 = (lib_cffmpeg_create2)dlsym(lib, "c_ffmpeg_create2");
+        release_if_err(fn_create2, lib);
+    }
+    return fn_create2(logfile);
+}
+
+void wrap_fn_destroy(void *lib, const cffmpeg h){
     fn_destroy(h);
 }
 
-void wrap_fn_run(const cffmpeg h, const char* input){
+void wrap_fn_run(void *lib, const cffmpeg h, const char* input){
+    if (!fn_run){
+        fn_run = (lib_cffmpeg_run)dlsym(lib, "c_ffmpeg_run");
+        if (!fn_run) return;
+    }
+
     fn_run(h, input);
 }
 
-void wrap_fn_scale(const cffmpeg h, const int wid, const int hei, const int flags){
-    fn_scale(h, wid, hei, flags);
+int wrap_fn_fps(void *lib, const cffmpeg h){
+    if(!fn_fps){
+        fn_fps = (lib_cffmpeg_fps)dlsym(lib, "c_ffmpeg_get_fps");
+        if (!fn_fps) return 25;
+    }
+    return fn_fps(h);
 }
 
-void wrap_fn_recorder(const cffmpeg h, const char* dir, int mind, int maxd){
-    fn_recorder(h, dir, mind, maxd);
+void wrap_fn_run_gb28181(void *lib, const cffmpeg h){
+    if (!fn_gb28181){
+        fn_gb28181 = (lib_cffmpeg_gb28181)dlsym(lib, "c_ffmpeg_run_gb28181");
+        if (!fn_gb28181) return;
+    }
+    fn_gb28181(h);
 }
 
-void wrap_fn_fire_recorder(const cffmpeg h, const int64_t id){
-    fn_fire_recorder(h, id);
+char * wrap_fn_get_gb28181_pic(void *lib, const char *rtspUrl, int *retDataLen, const int tt){
+    if (!fn_get_gb28181_pic){
+        fn_get_gb28181_pic = (lib_cffmpeg_get_gb28181_pic)dlsym(lib, "c_ffmpeg_get_gb28181_pic");
+        if(!fn_get_gb28181_pic) {
+            *retDataLen = 0;
+            return NULL;
+        }
+    }
+    return fn_get_gb28181_pic(rtspUrl, retDataLen, tt);
 }
 
-char* wrap_fn_info_recorder(const cffmpeg h, int* index, int* length){
-    return fn_info_recorder(h, index, length);
+void wrap_fn_use_cpu(void *lib, const cffmpeg h){
+    if (!fn_cpu){
+        fn_cpu = (lib_cffmpeg_cpu)dlsym(lib, "c_ffmepg_use_cpu");
+        if (!fn_cpu) return;
+    }
+    fn_cpu(h);
 }
 
-void wrap_fn_decoder(const cffmpeg h){
+void wrap_fn_open_rec(void *lib, const cffmpeg h){
+    if (!fn_open_recorder){
+        fn_open_recorder = (lib_cffmpeg_open_recorder)dlsym(lib, "c_ffmpeg_open_recorder");
+        if (!fn_open_recorder) return;
+    }
+    fn_open_recorder(h);
+}
+
+void wrap_fn_recorder(void *lib, const cffmpeg h, const char* id, const char* dir, const int64_t fid, int mind, int maxd, int audio){
+    if (!fn_recorder){
+        fn_recorder = (lib_cffmpeg_recorder)dlsym(lib, "c_ffmpeg_build_recorder");
+        if (!fn_recorder) return;
+    }
+    fn_recorder(h, id, dir, fid, mind, maxd, audio);
+}
+
+void wrap_fn_rec_duration(void *lib, const cffmpeg h, const int min, const int max){
+    if (!fn_rec_duration){
+        fn_rec_duration = (lib_cffmpeg_rec_duration)dlsym(lib, "c_ffmpeg_set_record_duration");
+        if (!fn_rec_duration) return;
+    }
+    fn_rec_duration(h, min, max);
+}
+
+int wrap_fn_fire_recorder(void *lib, const cffmpeg h, const char* sid, const int64_t id){
+    if (!fn_fire_recorder){
+        fn_fire_recorder = (lib_cffmpeg_fire_recorder)dlsym(lib, "c_ffmpeg_fire_recorder");
+        if (!fn_fire_recorder) return -1;
+    }
+    return fn_fire_recorder(h, sid, id);
+}
+
+int wrap_fn_info_recorder(void *lib, const cffmpeg h, int* index, char** recid, int* recidLen, char** fpath, int* pathLen){
+    if (!fn_info_recorder){
+        fn_info_recorder = (lib_cffmpeg_info_recorder)dlsym(lib, "c_ffmpeg_get_info_recorder");
+        if (!fn_info_recorder) return -1;
+    }
+    return fn_info_recorder(h, index, recid, recidLen, fpath, pathLen);
+}
+
+void wrap_fn_decoder(void *lib, const cffmpeg h){
+    if (!fn_decoder){
+        fn_decoder = (lib_cffmpeg_decoder)dlsym(lib, "c_ffmpeg_build_decoder");
+        if (!fn_decoder) return;
+    }
     fn_decoder(h);
 }
 
-void* wrap_fn_decoder_pic(const cffmpeg h, int* wid, int* hei){
-    return fn_decoder_pic(h, wid, hei);
+int wrap_fn_decoder_pic(void *lib, const cffmpeg h, unsigned char** data, int *wid, int *hei, int *format, int *length, int64_t *id){
+    if (!fn_decoder_pic){
+        fn_decoder_pic = (lib_cffmpeg_pic)dlsym(lib, "c_ffmpeg_get_pic_decoder");
+        if (!fn_decoder_pic) return -1;
+    }
+    return fn_decoder_pic(h, data, wid, hei, format, length, id);
 }
 
-void wrap_fn_active_recorder(const cffmpeg h, const char* dir, int mind, int maxd, rec_func fn){
-    fn_active_recorder(h, dir, mind, maxd, fn);
+void wrap_fn_close_stream(void *lib, const cffmpeg h){
+    if (!fn_close_stream){
+        fn_close_stream = (lib_cffmpeg_close_stream)dlsym(lib, "c_ffmpeg_close_stream");
+        if (!fn_close_stream) return;
+    }
+    fn_close_stream(h);
 }
 
-void wrap_fn_active_decoder(const cffmpeg h, dec_func fn){
-    fn_active_decoder(h, fn);
+void* wrap_fn_get_avpacket(void *lib, const cffmpeg h, int* size, int* key){
+    if(!fn_get_avpacket){
+        fn_get_avpacket = (lib_cffmpeg_avpacket)dlsym(lib, "c_ffmpeg_get_avpacket");
+        release_if_err(fn_get_avpacket, lib);
+    }
+    return fn_get_avpacket(h, size, key);
 }
 
-void* wrap_fn_decode_jpeg(const cffmpeg h, const char* file, int* wid, int* hei){
-    return fn_dec_jpeg(h, file, wid, hei);
-}
\ No newline at end of file
+int wrap_fn_get_avpacket2(void *lib, const cffmpeg h, unsigned char **data, int* size, int* key){
+    if(!fn_get_avpacket2){
+        fn_get_avpacket2 = (lib_cffmpeg_avpacket2)dlsym(lib, "c_ffmpeg_get_avpacket2");
+        if (!fn_get_avpacket2) return -1;
+    }
+    return fn_get_avpacket2(h, data, size, key);
+}
+
+int wrap_fn_set_devid(void *lib, const cffmpeg h, const int devid){
+    if (!fn_set_devid){
+        fn_set_devid = (lib_cffmpeg_devid)dlsym(lib, "c_ffmpeg_set_devid");
+        if (!fn_set_devid) return -1;
+    }
+    return fn_set_devid(h, devid);
+}
+
+void wrap_fn_release_buf(void *lib, void *buf){
+    if (!fn_release_buf){
+        fn_release_buf = (lib_cffmpeg_release_buf)dlsym(lib, "c_ffmpeg_release_buf");
+    }
+    if (fn_release_buf) fn_release_buf(buf);
+}
+// for encoder
+cencoder wrap_fn_create_encoder(void *lib, const int w, const int h, const int fps, const int br, const int pix_fmt, const int scale_flag, const int gi){
+    if (!fn_create_encoder){
+        fn_create_encoder = (lib_cffmpeg_create_encoder)dlsym(lib, "c_ffmpeg_create_encoder");
+        release_if_err(fn_create_encoder, lib);
+    }
+    return fn_create_encoder(w, h, fps, br, pix_fmt, scale_flag, gi);
+}
+
+void wrap_fn_destroy_encoder(void *lib, const cencoder h){
+    if (!fn_destroy_encoder){
+        fn_destroy_encoder = (lib_cffmpeg_destroy_encoder)dlsym(lib, "c_ffmpeg_destroy_encoder");
+        if(!fn_destroy_encoder) return;
+    }
+    fn_destroy_encoder(h);
+}
+
+void* wrap_fn_encode(void *lib, cencoder hdl, void *in, const int w, const int h, int *out_size, int *key){
+    if (!fn_encode){
+        fn_encode = (lib_cffmpeg_encode)dlsym(lib, "c_ffmpeg_encode");
+        release_if_err(fn_encode, lib);
+    }
+    
+    uint8_t *out = NULL;
+    const int flag = fn_encode(hdl, (uint8_t*)in, w, h, &out, out_size, key);
+    if (flag == 0 && out != NULL) {
+        return out;
+    }
+    *out_size = 0;
+    *key = 0;
+    return NULL;
+}

--
Gitblit v1.8.0