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 |  103 +++++++++++++++++++++++++++++----------------------
 1 files changed, 58 insertions(+), 45 deletions(-)

diff --git a/libcffmpeg.c b/libcffmpeg.c
index 7c30caf..279b292 100644
--- a/libcffmpeg.c
+++ b/libcffmpeg.c
@@ -44,7 +44,7 @@
 
 cffmpeg wrap_fn_create2(void *lib, const char *logfile){
     if (!fn_create2){
-        fn_create2 = (lib_cffmpeg_create)dlsym(lib, "c_ffmpeg_create2");
+        fn_create2 = (lib_cffmpeg_create2)dlsym(lib, "c_ffmpeg_create2");
         release_if_err(fn_create2, lib);
     }
     return fn_create2(logfile);
@@ -79,12 +79,31 @@
     fn_gb28181(h);
 }
 
+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);
+}
+
 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_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){
@@ -103,18 +122,18 @@
     fn_rec_duration(h, min, max);
 }
 
-void wrap_fn_fire_recorder(void *lib, const cffmpeg h, const char* sid, const int64_t id){
+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;
+        if (!fn_fire_recorder) return -1;
     }
-    fn_fire_recorder(h, sid, id);
+    return fn_fire_recorder(h, sid, id);
 }
 
-void wrap_fn_info_recorder(void *lib, const cffmpeg h, int* index, char** recid, int* recidLen, char** fpath, int* pathLen){
+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;
+        if (!fn_info_recorder) return -1;
     }
     return fn_info_recorder(h, index, recid, recidLen, fpath, pathLen);
 }
@@ -127,12 +146,20 @@
     fn_decoder(h);
 }
 
-void* wrap_fn_decoder_pic(void *lib, const cffmpeg h, int *wid, int *hei, int *format, int *length, int64_t *id){
+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");
-        release_if_err(fn_decoder_pic, lib);
+        if (!fn_decoder_pic) return -1;
     }
-    return fn_decoder_pic(h, wid, hei, format, length, id);
+    return fn_decoder_pic(h, data, wid, hei, format, length, id);
+}
+
+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_get_avpacket(void *lib, const cffmpeg h, int* size, int* key){
@@ -143,22 +170,35 @@
     return fn_get_avpacket(h, size, key);
 }
 
-// return val: -1 open error; -2, find stream error; -3, converter create error
-void* wrap_fn_decode(void *lib, const char* file, const int gb, int* wid, int* hei){
-    if (!fn_decode){
-        fn_decode = (lib_cffmpeg_decode)dlsym(lib, "c_ffmpeg_decode");
-        release_if_err(fn_decode, lib);
+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_decode(file, gb, wid, hei);
+    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 scale_flag, const int gi){
+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, scale_flag, gi);
+    return fn_create_encoder(w, h, fps, br, pix_fmt, scale_flag, gi);
 }
 
 void wrap_fn_destroy_encoder(void *lib, const cencoder h){
@@ -177,37 +217,10 @@
     
     uint8_t *out = NULL;
     const int flag = fn_encode(hdl, (uint8_t*)in, w, h, &out, out_size, key);
-    if (flag > 0 && out != NULL) {
+    if (flag == 0 && out != NULL) {
         return out;
     }
     *out_size = 0;
     *key = 0;
     return NULL;
-}
-
-// for conv
-cconv wrap_fn_create_conv(void *lib, const int srcW, const int srcH, const int srcFormat,
-                          const int dstW, const int dstH, const int dstFormat, const int flag){
-    if (!fn_create_conv){
-        fn_create_conv = (lib_cffmpeg_create_conv)dlsym(lib, "c_ffmpeg_create_conv");
-        release_if_err(fn_create_conv, lib);
-    }
-    return fn_create_conv(srcW, srcH, srcFormat, dstW, dstH, dstFormat, flag);
-}
-
-void wrap_fn_destroy_conv(void *lib, const cconv h){
-    if (!fn_destroy_conv){
-        fn_destroy_conv = (lib_cffmpeg_destroy_conv)dlsym(lib, "c_ffmpeg_destroy_conv");
-        if(!fn_destroy_conv) return;
-    }
-    fn_destroy_conv(h);
-}
-
-void* wrap_fn_conv(void *lib, const cconv h, uint8_t *in){
-    if (!fn_conv){
-        fn_conv = (lib_cffmpeg_conv)dlsym(lib, "c_ffmpeg_conv");
-        release_if_err(fn_conv, lib);
-    }
-
-    return fn_conv(h, in);
 }

--
Gitblit v1.8.0