// // 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 lists; pthread_t id; } VsimBean; static void run(const char *filepath, std::vector *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(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 local_list; // run(local_url, &local_list); // // std::vector 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("<< 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; }