videopipe_rk3568.cpp

//
// Created by dizi on 2024/8/2.
//

#include "videopipe_rk3568.h"


JNIEXPORT jint JNICALL JNI_OnLoad(JavaVM *vm, void *reserved) {
    global_jvm = vm;
    JNIEnv *env;
    if (vm->GetEnv(reinterpret_cast<void**>(&env), JNI_VERSION_1_6) != JNI_OK) {
        return JNI_ERR;
    }
    java_class_video_pipe_manager = (jclass)env->NewGlobalRef(env->FindClass("com/wmdigit/feature/ai/videopipe/VideoPipeRepository"));

    return JNI_VERSION_1_6;
}

/**
 * 算法初始化
 */
extern "C"
JNIEXPORT jint JNICALL
Java_com_wmdigit_feature_ai_videopipe_VideoPipeRepository_init(JNIEnv *env, jobject thiz,
                                                         jstring modelBackBonePath,
                                                         jstring modelC3dPath) {
    int size_back_bone, size_c3d;
    unsigned char* data_back_bone = read_file_byte(env, modelBackBonePath, size_back_bone);
    unsigned char* data_c3d = read_file_byte(env, modelC3dPath, size_c3d);
    init_info.lat3d_backbone_model_data = data_back_bone;
    init_info.lat3d_backbone_model_len = size_back_bone;
    init_info.lat3d_c3d_model_data = data_c3d;
    init_info.lat3d_c3d_model_len = size_c3d;
    init_info.lat3d_enable = true;
    init_info.gpu_enable = true;

    motion = new libvideopipe::MotionVideoPipe();
    // 初始化
    int ret_init = motion->Init(init_info);
    // 设置回调
    if (ret_init == 0){
        libvideopipe::VideoPipeCallback callback = trace_result_callback;
        motion->SetResultCallback(callback, &user);
    }
    return ret_init;
}

/**
 * 配置空盘图
 */
extern "C"
JNIEXPORT jint JNICALL
Java_com_wmdigit_feature_ai_videopipe_VideoPipeRepository_setMarkMat(JNIEnv *env, jobject thiz,
                                                               jobject bitmap) {
    // Bitmap转mat
    cv::Mat mat = convert_bitmap_to_mat(env, bitmap);
    // RGBA转RGB
    cv::Mat mat_rgb = cv::Mat();
    cv::cvtColor(mat, mat_rgb, CV_RGBA2RGB);
    // 设置空盘图
    return motion->SetMarkMat(mat_rgb.clone());
}

extern "C"
JNIEXPORT void JNICALL
Java_com_wmdigit_feature_ai_videopipe_VideoPipeRepository_feedFrame(JNIEnv *env, jobject thiz,
                                                              jobject bitmap) {
    // Bitmap转mat
    cv::Mat mat = convert_bitmap_to_mat(env, bitmap);
    // rgba转rgb
    cv::Mat mat_rgb = cv::Mat();
    cv::cvtColor(mat, mat_rgb, CV_RGBA2RGB);
    // 组装入参
    libvideopipe::MOTION_VIDEOPIPE_INPUT input;
    input.img = mat_rgb.clone();
    input.pts = get_timestamp();
    input.debug_mode = false;

    motion->FeedFrame(input);
}

/**
 * FeedFrame的回调
 * @param data
 * @param len
 * @param user
 */
void trace_result_callback(const char* data, int len, void* user)
{
    auto* output = (libvideopipe::MOTION_VIDEOPIPE_OUTPUT*) data;
    if (output->status == libvideopipe::MOTION_STATUS_TYPES::MOTION_IN)
    {
        LOGD("进入");
        feed_frame_callback(2);
    }
    else if (output->status == libvideopipe::MOTION_STATUS_TYPES::MOTION_OUT)
    {
        LOGD("离开");
        feed_frame_callback(3);
    }
    else if (output->status == libvideopipe::MOTION_STATUS_TYPES::MOTION_STATIC_EMPTY)
    {
    }
    else if (output->status == libvideopipe::MOTION_STATUS_TYPES::MOTION_STATIC_FULL)
    {
    }
    else{
    }
}

void feed_frame_callback(int state){
    JNIEnv *env;
    global_jvm->GetEnv((void**)&env, JNI_VERSION_1_6);
    jmethodID j_method_instance = env->GetStaticMethodID(
            java_class_video_pipe_manager,
            "getInstance",
            "()Lcom/wmdigit/feature/ai/videopipe/VideoPipeRepository;"
    );
    jobject instance=env->CallStaticObjectMethod(java_class_video_pipe_manager, j_method_instance);
    jmethodID j_method_in_out;
    if (state == 2){
        j_method_in_out = env->GetMethodID(java_class_video_pipe_manager, "objectIn", "()V");
    }
    else{
        j_method_in_out = env->GetMethodID(java_class_video_pipe_manager, "objectOut", "()V");
    }
    env->CallVoidMethod(instance, j_method_in_out);
    env->DeleteLocalRef(instance);
}


/**
 * 读本地文件，用于加载模型
 * @param env
 * @param file_path
 * @return
 */
unsigned char* read_file_byte(JNIEnv *env, jstring file_path, int& size){
    const char *jpath = env->GetStringUTFChars(file_path, nullptr);
    FILE *pFile;
    pFile = fopen(jpath, "rw");
    if (pFile == nullptr) {
        env->ReleaseStringUTFChars(file_path, jpath);
        return nullptr;
    }
    unsigned char *pBuf;  //定义文件指针
    fseek(pFile, 0, SEEK_END); //把指针移动到文件的结尾 ，获取文件长度
    size = (int) ftell(pFile); //获取文件长度
    LOGD("size:%d", size);
    pBuf = new unsigned char[size + 1]; //定义数组长度
    rewind(pFile); //把指针移动到文件开头 因为我们一开始把指针移动到结尾，如果不移动回来 会出错
    fread(pBuf, 1, (size_t) size, pFile); //读文件
    pBuf[size] = 0; //把读到的文件最后一位 写为0 要不然系统会一直寻找到0后才结束
    fclose(pFile); // 关闭文件
    env->ReleaseStringUTFChars(file_path, jpath);
    return pBuf;
}

/**
 * 获取时间戳
 * @return
 */
long long get_timestamp()
{
    long long tmp;
    struct timeval tv;
    gettimeofday(&tv, nullptr);
    tmp = tv.tv_sec;
    tmp = tmp * 1000;
    tmp = tmp + (tv.tv_usec / 1000);
    return tmp;
}