#ifdef __cplusplus
extern "C"{
#endif

#include <dlfcn.h>
#include <stdlib.h>
#include <stdio.h>

#include "libcffmpeg.h"

#ifdef __cplusplus
}
#endif

#define release_if_err(fn, lib) do{if(!fn){dlclose(lib); return NULL;}}while(0)

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_create2 = (lib_cffmpeg_create)dlsym(lib, "c_ffmpeg_create2");
        release_if_err(fn_create2, 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_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_decode = (lib_cffmpeg_decode)dlsym(lib, "c_ffmpeg_decode");
        release_if_err(fn_decode, 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);

        fn_create_conv = (lib_cffmpeg_create_conv)dlsym(lib, "c_ffmpeg_create_conv");
        release_if_err(fn_create_conv, lib);
        fn_destroy_conv = (lib_cffmpeg_destroy_conv)dlsym(lib, "c_ffmpeg_destroy_conv");
        release_if_err(fn_destroy_conv, lib);
        fn_conv = (lib_cffmpeg_conv)dlsym(lib, "c_ffmpeg_conv");
        release_if_err(fn_conv, lib);

        fn_gpu_conv = (lib_gpu_conv)dlsym(lib, "c_gpu_conv");
        release_if_err(fn_gpu_conv, lib);

    }else{
        printf("dlopen - %s\n", dlerror());  
    }
    return lib;
}

void release_libcffmpeg(libcffmpeg lib){
    if(lib){
        dlclose(lib);
    }
}

cffmpeg wrap_fn_create(){
    return fn_create();
}

cffmpeg wrap_fn_create2(const char *logfile){
    return fn_create2(logfile);
}

void wrap_fn_destroy(const cffmpeg h){
    fn_destroy(h);
}

void wrap_fn_run(const cffmpeg h, const char* input){
    fn_run(h, input);
}

void wrap_fn_run_gb28181(const cffmpeg h){
    fn_gb28181(h);
}

void wrap_fn_use_cpu(const cffmpeg h){
    fn_cpu(h);
}

void wrap_fn_recorder(const cffmpeg h, const char* id, const char* dir, int mind, int maxd, int audio){
    fn_recorder(h, id, dir, mind, maxd, audio);
}

void wrap_fn_fire_recorder(const cffmpeg h, const char* sid, const int64_t id){
    fn_fire_recorder(h, sid, id);
}

void wrap_fn_info_recorder(const cffmpeg h, int* index, char** recid, int* recidLen, char** fpath, int* pathLen){
    return fn_info_recorder(h, index, recid, recidLen, fpath, pathLen);
}

void wrap_fn_decoder(const cffmpeg h){
    fn_decoder(h);
}

void* wrap_fn_decoder_pic(const cffmpeg h, int *wid, int *hei, int *format, int *length, int64_t *id){
    return fn_decoder_pic(h, wid, hei, format, length, id);
}

void* wrap_fn_get_avpacket(const cffmpeg h, int* size, int* key){
    return fn_get_avpacket(h, size, key);
}

// return val: -1 open error; -2, find stream error; -3, converter create error
void* wrap_fn_decode(const char* file, const int gb, int* wid, int* hei){
    return fn_decode(file, gb, wid, hei);
}

void* wran_fn_gpu_conv(void *in, const int w, const int h, const int dst_w, const int dst_h, int *length){
    return fn_gpu_conv(in, w, h, dst_w, dst_h, length);
}

// 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);
}

void wrap_fn_destroy_encoder(const cencoder h){
    fn_destroy_encoder(h);
}

void* wrap_fn_encode(cencoder hdl, void *in, const int w, const int h, int *out_size, int *key){
    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;
}

// for conv
cconv wrap_fn_create_conv(const int srcW, const int srcH, const int srcFormat,
                          const int dstW, const int dstH, const int flag){
    return fn_create_conv(srcW, srcH, srcFormat, dstW, dstH, flag);
}

void wrap_fn_destroy_conv(const cconv h){
    fn_destroy_conv(h);
}
void* wrap_fn_conv(const cconv h, uint8_t *in){
    return fn_conv(h, in);
}