Commit c76f3927edaf8e852e92d8cdd5f315fd2bcb7aff

Authored by Zhao Shuaihua
2 parents 29a1cd95 c4e50858

合并分支

.gitignore
... ... @@ -4,4 +4,7 @@ bin/logs/*
4 4 bin/res/*
5 5 bin/vpt_proj
6 6 bin/libvpt_ascend.so
  7 +<<<<<<< HEAD
7 8 data/*
  9 +=======
  10 +>>>>>>> c4e50858604d927a3467ae4d0c3b3c63b4a35afb
... ...
src/ai_platform/MultiSourceProcess.cpp
... ... @@ -1418,8 +1418,12 @@ void CMultiSourceProcess::manned_snapshot(vector&lt;string&gt;&amp; vpt_interest_task_id,
1418 1418 RecoderInfo recoderInfo;
1419 1419 recoderInfo.task_id = task_id;
1420 1420 recoderInfo.object_id = std::to_string(j);
  1421 +<<<<<<< HEAD
1421 1422 recoderInfo.recoderDir = video_folder;
1422 1423 // recoderInfo.recoderDir = video_file_name;
  1424 +=======
  1425 + recoderInfo.recoderPath = "./res/recode";
  1426 +>>>>>>> c4e50858604d927a3467ae4d0c3b3c63b4a35afb
1423 1427 recoderInfo.frame_nb = mem->getFrameNb();
1424 1428 recoderInfo.mq_info = json_str;
1425 1429  
... ...
src/ai_platform/common_header.h
... ... @@ -32,7 +32,7 @@ struct box_t {
32 32 };
33 33  
34 34 struct RecoderInfo {
35   - std::string recoderDir;
  35 + std::string recoderPath;
36 36 std::string task_id;
37 37 std::string object_id;
38 38 unsigned long long frame_nb;
... ...
src/decoder/dvpp/FFRecoderTaskManager.cpp
... ... @@ -300,7 +300,7 @@ void FFRecoderTaskManager::recode_thread(RecodeParam recodeParam){
300 300 RecoderInfo recoderInfo;
301 301 recoderInfo = recodeParam.recoderInfo;
302 302 std::string id = recoderInfo.task_id + "_" + recoderInfo.object_id + "_" + std::to_string(recoderInfo.frame_nb);
303   - string file_name = recoderInfo.recoderDir + "/recoder_" + id + "_" + std::to_string(get_cur_time()) + ".mp4";
  303 + string file_name = recoderInfo.recoderPath;
304 304 FFRecoder ffrecoder;
305 305 bool bInit = ffrecoder.init(m_inStream, recodeParam.avctx, file_name.c_str());
306 306 if (!bInit) {
... ... @@ -369,7 +369,7 @@ void FFRecoderTaskManager::recode_thread2() {
369 369 m_pkt_list_mtx.unlock();
370 370  
371 371 std::string id = recoderinfo.task_id + "_" + recoderinfo.object_id + "_" + std::to_string(recoderinfo.frame_nb);
372   - string file_name = recoderinfo.recoderDir + "/recoder_" + id + "_" + std::to_string(get_cur_time()) + ".mp4";
  372 + string file_name = recoderinfo.recoderPath;
373 373 FFRecoder ffrecoder;
374 374 bool bInit = ffrecoder.init(m_inStream, m_avctx, file_name.c_str());
375 375 if (!bInit) {
... ... @@ -377,7 +377,7 @@ void FFRecoderTaskManager::recode_thread2() {
377 377 ffrecoder.uninit();
378 378 continue;
379 379 }
380   - LOG_DEBUG("record start, pkt_list size: {} id: {}", m_pkt_list.size(), id);
  380 + LOG_DEBUG("record start, pkt_list size: {} task_id: {} object_id:{} frame_nb: {}", m_pkt_list.size(), recoderinfo.task_id, recoderinfo.object_id, recoderinfo.frame_nb);
381 381  
382 382 int count = 0;
383 383 auto it_save = it_data;
... ... @@ -448,7 +448,7 @@ void FFRecoderTaskManager::recode_thread3() {
448 448 m_frame_list_mtx.unlock();
449 449  
450 450 std::string id = recoderinfo.task_id + "_" + recoderinfo.object_id + "_" + std::to_string(recoderinfo.frame_nb);
451   - string file_name = recoderinfo.recoderDir + "/recoder_" + id + "_" + std::to_string(get_cur_time()) + ".mp4";
  451 + string file_name = recoderinfo.recoderPath;
452 452 FFRecoder ffrecoder;
453 453 bool bInit = ffrecoder.init(m_avctx->width, m_avctx->height, m_time_base, m_avctx, file_name.c_str());
454 454 if (!bInit) {
... ... @@ -523,7 +523,7 @@ void FFRecoderTaskManager::recode_thread4() {
523 523 m_pkt_list_mtx.unlock();
524 524  
525 525 std::string id = recoderinfo.task_id + "_" + recoderinfo.object_id + "_" + std::to_string(recoderinfo.frame_nb);
526   - string file_name = recoderinfo.recoderDir + "/recoder_" + id + "_" + std::to_string(get_cur_time()) + ".mp4";
  526 + string file_name = recoderinfo.recoderPath;
527 527 FFRecoder ffrecoder;
528 528 bool bInit = ffrecoder.init(m_inStream, m_avctx, file_name.c_str());
529 529 if (!bInit) {
... ... @@ -578,4 +578,4 @@ void FFRecoderTaskManager::close() {
578 578  
579 579 void FFRecoderTaskManager::set_mq_callback(mq_callback_t cb) {
580 580 mq_publish_func = cb;
581   -}
582 581 \ No newline at end of file
  582 +}
... ...
src/decoder/test_recoder.cpp
... ... @@ -179,7 +179,7 @@ void algorthim_face_detect(vector&lt;DeviceMemory*&gt; vec_gpuMem) {
179 179 RecoderInfo recoderInfo;
180 180 recoderInfo.task_id = task_id;
181 181 recoderInfo.object_id = std::to_string(obj_id);
182   - recoderInfo.recoderDir = "./res/recode";
  182 + recoderInfo.recoderPath = "./res/recode";
183 183 recoderInfo.frame_nb = mem->getFrameNb();
184 184  
185 185 m_recoderinfo_queue_mtx.lock();
... ... @@ -211,7 +211,7 @@ void test_recode_thread() {
211 211 RecoderInfo recoderInfo;
212 212 recoderInfo.task_id = gpuMem->getId();
213 213 recoderInfo.object_id = std::to_string(obj_id);
214   - recoderInfo.recoderDir = "./res/recode";
  214 + recoderInfo.recoderPath = "./res/recode";
215 215 recoderInfo.frame_nb = gpuMem->getFrameNb();
216 216  
217 217 DecoderManager* pDecManager = DecoderManager::getInstance();
... ...