PQCVMedia/src/video_similarity_comparison.cpp

//
// Created by 阳坤 on 2022/3/1.
//

#include "utils/video_similarity.h"
#include "utils/count_down_latch.h"
#include "ffmpeg/av_tools.h"

extern "C" {
#include "utils/md5.h"
}


typedef struct VsimBean {
    const char *filepath;
    const char *image_md5;
    std::vector<VideoSimilarityModel *> lists;
    pthread_t id;
} VsimBean;

static void run(const char *filepath, std::vector<VideoSimilarityModel *> *list) {
    long id = video_similarity_detection_init(filepath, nullptr);
    if (id > 0) {
        int ret = video_similarity_detection_start(id, 0, AUDIO,
                                                   list);
//        char md5_str[64] = {0};
//        if (get_file_md5(vsimBean->filepath, md5_str) > 0);
//        vsimBean->image_md5 = strdup(md5_str);
        video_similarity_detection_close(id);
    }
}

static void *run_thread(void *p) {
    auto *vsimBean = (VsimBean *) p;
    run(vsimBean->filepath, &vsimBean->lists);
    return 0;
}


typedef struct Test {
    const char *input;
    CountDownLatch *countDownLatch;
};


static void *run(void *p) {
    auto *test = (Test *) p;

    const char *test_3 = "{\"videoPath\":\"/Users/ehlxr/Workspaces/tzld/videoDNA/videoDNA-core/target/tmposs/2.m3u8\",\"clips\":[{\"startTimeMs\":1000,\"endTimeMs\":2000},{\"startTimeMs\":2000,\"endTimeMs\":3000},{\"startTimeMs\":3000,\"endTimeMs\":4000},{\"startTimeMs\":4000,\"endTimeMs\":5000},{\"startTimeMs\":5000,\"endTimeMs\":6000},{\"startTimeMs\":6000,\"endTimeMs\":7000},{\"startTimeMs\":7000,\"endTimeMs\":8000},{\"startTimeMs\":8000,\"endTimeMs\":9000}]}";
    const char *ret_json3 = get_video_similarity_list(test_3);
    printf("====> %s",ret_json3);

//    const char *ret_json = get_video_similarity_list(test_1);
//    const char *ret_json2 = get_video_similarity_list(test_2);
//    printf("ret_json=%s\n", ret_json);
//    printf("相似值为：%f \n", get_video_similarity_value(ret_json, ret_json2));
//    test->countDownLatch->countDown();
//    MD5_CTX md5c;
//    MD5Init(&md5c); //初始化
//    unsigned char decrypt[16] = {0};
//    unsigned char decrypt32[32] = {0};
//    size_t read_len = strlen(ret_json);
//    MD5Update(&md5c, (unsigned char *) ret_json, read_len);
//    MD5Final(&md5c, decrypt);
//    char temp[8]={0};
//    strcpy((char *) decrypt32, "");
//    for (int i = 0; i < 16; i++) {
//        sprintf(temp, "%02x", decrypt[i]);
//        strcat((char *) decrypt32, temp);
//    }
//    printf("md5:%s\n", decrypt32);
//
//    char * dejson = "35070e4651614cced5604e93c4a59ff9";

    free(test);
    pthread_exit(NULL);
}

/**
 * 1、判断比对文件的 MD5 是否相等
 *  1.1、如果相等，
 *   1.1.1、判断媒体数据 mate 信息是否一致，如果一致，判断前面 N 关键帧的指纹码
 *  1.2、如果不等
 *   1.2.1、判断媒体数据 mate 信息是否一致, 如果一致，判断前面 N 关键帧的指纹码
 * @param argc
 * @param argv
 * @return
 */
int main(int argc, char *argv[]) {

    const char *arr = "0000010001001111011101110010011100100011001001110100001101110111";
    const char *arr2 = "0000010001001111011101110010011100100011001001110100001101010111";


    int threads = 1;
    const char *inputs[1];
//    inputs[0] =  "{\"videoPath\":\"http://rescdn.yishihui.com/longvideo/transcode/video/b46fd76f98364b3abad8c1297a868f82-1526976612340.m3u8\",\"clips\":[{\"startTimeMs\":3000,\"endTimeMs\":4000},{\"startTimeMs\":6000,\"endTimeMs\":7000},{\"startTimeMs\":9000,\"endTimeMs\":10000},{\"startTimeMs\":12000,\"endTimeMs\":13000},{\"startTimeMs\":15000,\"endTimeMs\":16000},{\"startTimeMs\":18000,\"endTimeMs\":19000},{\"startTimeMs\":21000,\"endTimeMs\":22000},{\"startTimeMs\":24000,\"endTimeMs\":25000}]}";
//    inputs[0] = "{\"videoPath\":\"http://rescdn.yishihui.com/longvideo/transcode/video/6b2efb8d1c6f4095a90b0b276d4b3a291532088309923.m3u8\",\"clips\":[{\"startTimeMs\":500,\"endTimeMs\":1000},{\"startTimeMs\":1000,\"endTimeMs\":1500},{\"startTimeMs\":1500,\"endTimeMs\":2000},{\"startTimeMs\":2000,\"endTimeMs\":2500},{\"startTimeMs\":2500,\"endTimeMs\":3000},{\"startTimeMs\":3000,\"endTimeMs\":3500},{\"startTimeMs\":3500,\"endTimeMs\":4000},{\"startTimeMs\":4000,\"endTimeMs\":4500}]}";
    inputs[0] = "{\"videoPath\":\"http://rescdn.yishihui.com/longvideo/transcode/video/3dbf36e8ec27479581514b8b7b4c8034-1526924772832-safe1594892815.m3u8\",\"clips\":[{\"startTimeMs\":6000,\"endTimeMs\":7000},{\"startTimeMs\":12000,\"endTimeMs\":13000},{\"startTimeMs\":18000,\"endTimeMs\":19000},{\"startTimeMs\":24000,\"endTimeMs\":25000},{\"startTimeMs\":30000,\"endTimeMs\":31000},{\"startTimeMs\":36000,\"endTimeMs\":37000},{\"startTimeMs\":42000,\"endTimeMs\":43000},{\"startTimeMs\":48000,\"endTimeMs\":49000}]}";
//    inputs[3] = "{\"videoPath\":\"http://rescdn.yishihui.com/longvideo/transcode/video/3dbf36e8ec27479581514b8b7b4c8034-1526924772832-safe1594892815.m3u8\",\"clips\":[{\"startTimeMs\":6000,\"endTimeMs\":7000},{\"startTimeMs\":12000,\"endTimeMs\":13000},{\"startTimeMs\":18000,\"endTimeMs\":19000},{\"startTimeMs\":24000,\"endTimeMs\":25000},{\"startTimeMs\":30000,\"endTimeMs\":31000},{\"startTimeMs\":36000,\"endTimeMs\":37000},{\"startTimeMs\":42000,\"endTimeMs\":43000},{\"startTimeMs\":48000,\"endTimeMs\":49000}]}";

//    for (;;) {
    CountDownLatch *countDownLatch = new CountDownLatch(threads);
    for (int i = 0; i < threads; ++i) {
        Test *test = static_cast<Test *>(malloc(sizeof(Test)));
        test->countDownLatch = countDownLatch;
        test->input = inputs[i];
        pthread_t id;
        pthread_create(&id, 0, run, test);
    }
    countDownLatch->await();
    delete countDownLatch;
//    }
    return 1;

//    const char *out_json = "{\n"
//                           "    \"videoPath\": \"\\\"/root/data/xxx.mp4\\\"\",\n"
//                           "    \"clips\": [\n"
//                           "        {\n"
//                           "            \"startTimeMs\": 15000,\n"
//                           "            \"endTimeMs\": 25000,\n"
//                           "            \"fingerprintCode\": [\n"
//                           "                \"\\\"0001000100101010101010\\\"\",\n"
//                           "                \"\\\"0001000100101010101010\\\"\",\n"
//                           "                \"\\\"0001000100101010101010\\\"\",\n"
//                           "                \"\\\"0001000100101010101010\\\"\"\n"
//                           "            ]\n"
//                           "        }\n"
//                           "    ]\n"
//                           "}";

//    auto vs = json2VideoSimilarity(input_json);
//    auto json = videoSimilarity2json(vs);
//
//    int32_t size = 7;
//    const char *filepath[size];
//    //票圈视频iOS客户端上传
//    filepath[0] = "http://rescdn.yishihui.com/longvideo/video/vpc/20220309/17025689Igao51Q0IMBYw8aNKH";
//    //ios 原生上传
//    filepath[1] = "http://rescdn.yishihui.com/longvideo/video/vpc/20220309/17025689VvI5KURrrxzlZ0Kcdv";
//    //iOS h5上传
//    filepath[2] = "http://rescdn.yishihui.com/longvideo/video/vpc/20220309/17025689iBy9WFswYdBNblY9au";
//    //票圈视频安卓客户端上传
//    filepath[3] = "http://rescdn.yishihui.com/longvideo/video/vpc/20220309/17677590exDXG6q75ELejK7aMY";
//    //安卓h5上传
//    filepath[4] = "http://rescdn.yishihui.com/longvideo/video/vpc/20220309/176775906GoWtUCYoVZYDbO1XX";
//    //安卓原生上传
//    filepath[5] = "";
//    //PC 上传
////    filepath[5] = "http://rescdn.yishihui.com/longvideo/video/vpc/20220309/17025689nSyGzifJFr4Apl0zc8";
//    filepath[5] = "http://rescdn.yishihui.com/longvideo/video/vpc/20220310/701939000F0qpyCLetflI3COz";
//    http://rescdn.yishihui.com/longvideo/video/vpc/20220304/17938576TxEh0UJXpzcoYRJuHu
//    //本地 local 上传的视频
//    filepath[6] = "/Users/devyk/Data/Project/sample/github_code/OpenCVSample/temp/19581045gIiFKepxbbplF3XtsG.mp4";

//    run(vs->videoPath,&vs->clips);

//    const char * ret = videoSimilarity2json(vs);

//    const char *local_url = "/Users/devyk/Data/Project/sample/github_code/OpenCVSample/temp/19581045gIiFKepxbbplF3XtsG.mp4";
//    std::vector<VideoSimilarityModel *> local_list;
//    run(local_url, &local_list);
//
//    std::vector<VideoSimilarityModel *> lists[size];
//    int split_count = 10;
////    int total = 3960000;
//    int ii = 1000 * 60 * 6;
//
//    int start = ii;
//    for (int i = 0; i < size; ++i) {
//        auto start_time = getCurrentTimeMills();
//        start = ii;
//        for (int j = 0; j < split_count; ++j) {
//            if (j == 0 || j == split_count - 1) {
//                continue;
//            }
//            auto *item = new VideoSimilarityModel();
//            item->startTimeMs = start;
//            item->endTimeMs = item->startTimeMs + 1000;
//            lists[i].push_back(item);
//            start += ii;
//        }
//        LOGE(">>>filepath=%s \n", filepath[i]);
//        run(filepath[i], &lists[i]);
//        LOGE("<<<filepath=%s cost time=%lld \n", filepath[i], getCurrentTimeMills() - start_time);
//    }
//
//    //本地 local
//    std::vector<VideoSimilarityModel *> local = lists[size - 1];
//    //相似 hash 值
//    int test_vis[6][8] = {0};
//    //所有的 帧
//    int test_vis_total[6][8] = {0};
//    for (int j = 0; j < size - 1; ++j) {
//        for (int g = 0; g < lists[j].size(); ++g) {
//            int *cur_sim = list_sim_compare(local[g]->hashs, lists[j][g]->hashs);
//            int a = cur_sim[0];
//            int b = cur_sim[1];
//            test_vis[j][g] = a;
//            test_vis_total[j][g] = b;
//        }
//    }
//
//
//    for (int i = 0; i < size - 1; ++i) {
//        int a_t = 0, b_t = 0;
//        for (int j = 0; j < 8; ++j) {
//            a_t += test_vis[i][j];
//            b_t += test_vis_total[i][j];
//        }
//        LOGE("%s %s 相似度 = %f \n", filepath[6], filepath[i], a_t * 1.0 / b_t * 1.0);
//    }
//    printf("");

    LOGE("end \n");
    return 0;
}