From ab3d17cd44625a762fc10f0f1fd8f2c46b980aff Mon Sep 17 00:00:00 2001
From: zhangmeng <775834166@qq.com>
Date: 星期二, 26 四月 2022 11:42:19 +0800
Subject: [PATCH] capture use cpu and add arm version
---
libcffmpeg.c | 206 +++++++++++++++++++++++++++++++++++----------------
1 files changed, 142 insertions(+), 64 deletions(-)
diff --git a/libcffmpeg.c b/libcffmpeg.c
index e33a376..279b292 100644
--- a/libcffmpeg.c
+++ b/libcffmpeg.c
@@ -17,43 +17,9 @@
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_gb28181 = (lib_cffmpeg_gb28181)dlsym(lib, "c_ffmpeg_run_gb28181");
- release_if_err(fn_gb28181, lib);
- fn_cpu = (lib_cffmpeg_cpu)dlsym(lib, "c_ffmepg_use_cpu");
- release_if_err(fn_cpu, 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_get_avpacket = (lib_cffmpeg_avpacket)dlsym(lib, "c_ffmpeg_get_avpacket");
- release_if_err(fn_get_avpacket, 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);
-
- fn_create_encoder = (lib_cffmpeg_create_encoder)dlsym(lib, "c_ffmpeg_create_encoder");
- release_if_err(fn_create_encoder, lib);
- fn_destroy_encoder = (lib_cffmpeg_destroy_encoder)dlsym(lib, "c_ffmpeg_destroy_encoder");
- release_if_err(fn_destroy_encoder, lib);
- fn_encode = (lib_cffmpeg_encode)dlsym(lib, "c_ffmpeg_encode");
- release_if_err(fn_encode, lib);
}else{
printf("dlopen - %s\n", dlerror());
@@ -67,79 +33,191 @@
}
}
-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_run_gb28181(const cffmpeg h){
+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_use_cpu(const cffmpeg 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_recorder(const cffmpeg h, const char* dir, int mind, int maxd){
- fn_recorder(h, dir, mind, maxd);
+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_fire_recorder(const cffmpeg h, const int64_t id){
- fn_fire_recorder(h, id);
+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);
}
-char* wrap_fn_info_recorder(const cffmpeg h, int* index, int* length){
- return fn_info_recorder(h, index, length);
+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);
}
-void wrap_fn_decoder(const cffmpeg h){
+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_get_avpacket(const cffmpeg h, int* size, int* key){
+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){
+ 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_active_recorder(const cffmpeg h, const char* dir, int mind, int maxd, rec_func fn){
- fn_active_recorder(h, dir, mind, maxd, fn);
+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);
}
-void wrap_fn_active_decoder(const cffmpeg h, dec_func fn){
- fn_active_decoder(h, fn);
+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_decode_jpeg(const cffmpeg h, const char* file, int* wid, int* hei){
- return fn_dec_jpeg(h, file, wid, hei);
+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(const int w, const int h, const int fps, const int br, const int scale_flag, const int gi){
- return fn_create_encoder(w, h, fps, br, scale_flag, 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, pix_fmt, scale_flag, gi);
}
-void wrap_fn_destroy_encoder(const cencoder h){
+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(cencoder hdl, void *in, const int w, const int h, int *out_size, int *key){
+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) {
+ if (flag == 0 && out != NULL) {
return out;
}
*out_size = 0;
--
Gitblit v1.8.0