Blame view

src/decoder/dvpp/DvppDecoder.cpp 25.2 KB
09c2d08c   Hu Chunming   arm交付版
1
2
3
4
  #include "DvppDecoder.h"
  #include "DvppSourceManager.h"
  
  
25a4a277   Hu Chunming   简化解码器线程,解决500小站上因...
5
6
7
8
9
10
11
12
13
14
15
  #define CHECK_AND_RETURN(ret, message)    \
              if(ret != 0) {LOG_ERROR("[{}]- {}", m_dec_name, message); return ret;}
  #define CHECK_NOT_RETURN(ret, message)    \
              if(ret != 0) {LOG_ERROR("[{}]- {}", m_dec_name, message);}
  #define CHECK_AND_RETURN_NOVALUE(ret, message)    \
              if(ret != 0) {LOG_ERROR("[{}]- {}", m_dec_name, message); return;}
  #define CHECK_AND_BREAK(ret, message)    \
              if(ret != 0) {LOG_ERROR("[{}]- {}", m_dec_name, message); break;}
  
  
  
bc01d0b6   Hu Chunming   代码优化;主要是修改了截图逻辑,避...
16
  
09c2d08c   Hu Chunming   arm交付版
17
18
  struct Vdec_CallBack_UserData {
      uint64_t frameId;
746db74c   Hu Chunming   实现recode
19
      unsigned long long frame_nb;
09c2d08c   Hu Chunming   arm交付版
20
21
      long startTime;
      long sendTime;
09c2d08c   Hu Chunming   arm交付版
22
  	DvppDecoder* self;
bc01d0b6   Hu Chunming   代码优化;主要是修改了截图逻辑,避...
23
  
09c2d08c   Hu Chunming   arm交付版
24
25
      Vdec_CallBack_UserData() {
          frameId = 0;
746db74c   Hu Chunming   实现recode
26
          frame_nb = 0;
09c2d08c   Hu Chunming   arm交付版
27
28
29
      }
  };
  
25a4a277   Hu Chunming   简化解码器线程,解决500小站上因...
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
  
  static long get_cur_time_ms() {
      chrono::time_point<chrono::system_clock, chrono::milliseconds> tpMicro
          = chrono::time_point_cast<chrono::milliseconds>(chrono::system_clock::now());
      return tpMicro.time_since_epoch().count();
  }
  
  static void *ReportThd(void *arg)
  {
      DvppDecoder *self = (DvppDecoder *)arg;
  	if(nullptr != self){
  		self->doProcessReport();
  	}
      return (void *)0;
  }
  
  static void VdecCallback(acldvppStreamDesc *input, acldvppPicDesc *output, void *pUserData)
  {
  	Vdec_CallBack_UserData *userData = (Vdec_CallBack_UserData *) pUserData;
      if(nullptr != userData){
          DvppDecoder* self = userData->self;
          if(self != nullptr){
              self->doVdppVdecCallBack(input, output, userData->frame_nb);
          }
          delete userData;
  	    userData = nullptr;
      }
  }
  
  
09c2d08c   Hu Chunming   arm交付版
60
  DvppDecoder::DvppDecoder(){
bc01d0b6   Hu Chunming   代码优化;主要是修改了截图逻辑,避...
61
      m_read_thread = nullptr;
09c2d08c   Hu Chunming   arm交付版
62
63
64
65
  
      fmt_ctx = nullptr;
  	m_bRunning = false;
  
09c2d08c   Hu Chunming   arm交付版
66
67
68
69
70
71
72
73
74
75
      video_index = -1;
      pix_fmt = AV_PIX_FMT_NONE;
      m_dec_name = "";
  
  	m_bPause = false;
  	m_bReal = true;
  
  	m_bFinished = false;
  	m_dec_keyframe = false;
  	m_fps = 0.0;
09c2d08c   Hu Chunming   arm交付版
76
77
78
  }
  
  DvppDecoder::~DvppDecoder(){
bc01d0b6   Hu Chunming   代码优化;主要是修改了截图逻辑,避...
79
80
81
      close();
  
      LOG_DEBUG("[{}]- ~DvppDecoder() in_count:{}  out_count:{}", m_dec_name, m_in_count, m_out_count);
09c2d08c   Hu Chunming   arm交付版
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
  }
  
  bool DvppDecoder::init(FFDecConfig cfg){
  
      m_dec_name = cfg.dec_name;
  
      AVCodecContext* avctx = init_FFmpeg(cfg);
      if(avctx == nullptr){
          return false;
      }
  
      bool bRet = init_vdpp(cfg, avctx);
      if(!bRet){
          return false;
      }
  
      m_cfg = cfg;
  
      decode_finished_cbk = cfg.decode_finished_cbk;
  
      m_bFinished = false;
  
      return true;
  }
  
  AVCodecContext* DvppDecoder::init_FFmpeg(FFDecConfig config){
  
bc01d0b6   Hu Chunming   代码优化;主要是修改了截图逻辑,避...
109
110
111
112
113
114
115
116
117
      const char* input_file = config.uri.c_str();
      FILE* infile = fopen(input_file, "r");
      if(nullptr != infile) {
          m_bReal = false;
          fclose(infile);
          infile = nullptr;
      } else {
          m_bReal = true;
      }
09c2d08c   Hu Chunming   arm交付版
118
119
120
121
122
123
  
  	// 打开输入视频文件
  	AVDictionary *options = nullptr;
  	av_dict_set( &options, "bufsize", "655360", 0 );
  	av_dict_set( &options, "rtsp_transport", config.force_tcp ? "tcp" : "udp", 0 );
  	av_dict_set( &options, "stimeout", "30000000", 0 ); // 单位为 百万分之一秒
09c2d08c   Hu Chunming   arm交付版
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
  
      do{
          fmt_ctx = avformat_alloc_context();
          if (avformat_open_input(&fmt_ctx, input_file, nullptr, &options) != 0) {
              LOG_ERROR("[{}]- Cannot open input file: {}", m_dec_name, input_file);
              break;
          }
          av_dump_format(fmt_ctx, 0, input_file, 0);
  
          // 查找流信息
          if (avformat_find_stream_info(fmt_ctx, nullptr) < 0) {
              LOG_ERROR("[{}]- Cannot find input stream information!", m_dec_name);
              break;
          }
  
          // 查找视频流信息
          AVCodec *decoder = nullptr;
          video_index = av_find_best_stream(fmt_ctx, AVMEDIA_TYPE_VIDEO, -1, -1, &decoder, 0);
          if (video_index < 0) {
              LOG_ERROR("[{}]- Cannot find a video stream in the input file!", m_dec_name);
              break;
          }
09c2d08c   Hu Chunming   arm交付版
146
  
bc01d0b6   Hu Chunming   代码优化;主要是修改了截图逻辑,避...
147
          avctx = avcodec_alloc_context3(decoder);
09c2d08c   Hu Chunming   arm交付版
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
          if(avctx == nullptr){
              LOG_ERROR("[{}]- alloc AVCodecContext failed!", m_dec_name);
              break;
          }
  	
  		// 得到视频流对象
  		AVStream* stream = fmt_ctx->streams[video_index];
  		AVCodecParameters *codecpar = stream->codecpar;
  		if (avcodec_parameters_to_context(avctx, codecpar) < 0)
  			break;
  
  		const AVBitStreamFilter * filter = nullptr;
  		if(codecpar->codec_id == AV_CODEC_ID_H264){
  			filter = av_bsf_get_by_name("h264_mp4toannexb");
  		}else if(codecpar->codec_id == AV_CODEC_ID_HEVC){
  			filter = av_bsf_get_by_name("hevc_mp4toannexb");
  		}else {
              LOG_ERROR("[{}]- codec_id is not supported!", m_dec_name);
  			break;
  		}
  
25a4a277   Hu Chunming   简化解码器线程,解决500小站上因...
169
170
171
172
173
174
          int enType = getVdecType(codecpar->codec_id, codecpar->profile);
          if(-1 == enType) {
              break;
          }
          m_enType = static_cast<acldvppStreamFormat>(enType);
  
09c2d08c   Hu Chunming   arm交付版
175
176
177
178
179
180
181
182
183
184
185
  		int ret = av_bsf_alloc(filter, &h264bsfc);
  		if (ret < 0){
  			break;
  		}
  		
  		avcodec_parameters_copy(h264bsfc->par_in, codecpar);
  		av_bsf_init(h264bsfc);
  
  		frame_width = codecpar->width;
  		frame_height = codecpar->height;
  		pix_fmt = (AVPixelFormat)codecpar->format;
25a4a277   Hu Chunming   简化解码器线程,解决500小站上因...
186
187
  
          if (stream->avg_frame_rate.den) {
25a4a277   Hu Chunming   简化解码器线程,解决500小站上因...
188
189
190
191
              m_fps = av_q2d(stream ->avg_frame_rate);
          } else {
              m_fps = 0.0;
          }
09c2d08c   Hu Chunming   arm交付版
192
  
5a84488e   Hu Chunming   添加农村事件开关
193
  #ifdef USE_VILLAGE
1b57a1c5   Hu Chunming   代码优化,避免可能的崩溃
194
195
196
197
          bool bRet = m_recoderManager.init(stream, avctx);
          if (!bRet){
              LOG_ERROR("[{}]- m_recoderManager 初始化失败!", m_dec_name);
          }  
5a84488e   Hu Chunming   添加农村事件开关
198
  #endif
746db74c   Hu Chunming   实现recode
199
  
09c2d08c   Hu Chunming   arm交付版
200
201
202
203
204
205
206
207
208
209
210
211
          LOG_INFO("[{}]- init ffmpeg success! input:{} frame_width:{} frame_height:{} fps:{} ", m_dec_name, input_file, frame_width, frame_height, m_fps);
  
  		return avctx;
  	}while(0);
  
      release_ffmpeg();
  
      LOG_ERROR("[{}]- init ffmpeg failed ! input:{} ", m_dec_name, input_file);
  
      return nullptr;
  }
  
25a4a277   Hu Chunming   简化解码器线程,解决500小站上因...
212
213
214
215
216
217
218
219
220
221
222
223
224
225
226
227
228
229
230
231
232
233
234
235
236
237
238
239
240
241
242
243
244
245
246
247
248
249
250
251
  int DvppDecoder::getVdecType(int videoType, int profile)
  {
      int streamFormat = H264_MAIN_LEVEL;
  
      // VDEC only support H265 main level264 baseline levelmain levelhigh level
      if (videoType == AV_CODEC_ID_HEVC) {
          streamFormat = H265_MAIN_LEVEL;
      } else if (videoType == AV_CODEC_ID_H264) {
          switch (profile) {
              case FF_PROFILE_H264_BASELINE:
                  streamFormat = H264_BASELINE_LEVEL;
                  break;
              case FF_PROFILE_H264_MAIN:
                  streamFormat = H264_MAIN_LEVEL;
                  break;
              case FF_PROFILE_H264_HIGH:
              case FF_PROFILE_H264_HIGH_10:
              case FF_PROFILE_H264_HIGH_10_INTRA:
              case FF_PROFILE_H264_MULTIVIEW_HIGH:
              case FF_PROFILE_H264_HIGH_422:
              case FF_PROFILE_H264_HIGH_422_INTRA:
              case FF_PROFILE_H264_STEREO_HIGH:
              case FF_PROFILE_H264_HIGH_444:
              case FF_PROFILE_H264_HIGH_444_PREDICTIVE:
              case FF_PROFILE_H264_HIGH_444_INTRA:
                  streamFormat = H264_HIGH_LEVEL;
                  break;
              default:
                  LOG_INFO("Not support h264 profile {}, use as mp", profile);
                  streamFormat = H264_MAIN_LEVEL;
                  break;
          }
      } else {
          streamFormat = -1;
          LOG_ERROR("Not support stream, type {},  profile {}", videoType, profile);
      }
  
      return streamFormat;
  }
  
09c2d08c   Hu Chunming   arm交付版
252
253
254
255
256
   bool DvppDecoder::init_vdpp(FFDecConfig cfg, AVCodecContext* avctx) {
  
      LOG_INFO("[{}]- Init device start...", m_dec_name);
  
      m_dvpp_deviceId = atoi(cfg.gpuid.c_str());
09c2d08c   Hu Chunming   arm交付版
257
258
259
260
261
262
263
264
265
266
267
268
269
270
271
272
273
274
275
276
277
278
279
  
      post_decoded_cbk = cfg.post_decoded_cbk;
  
      do{
          aclError ret = aclrtSetDevice(m_dvpp_deviceId);
          if(ret != ACL_ERROR_NONE){
              LOG_ERROR("[{}]-aclrtSetDevice failed !", m_dec_name);
              return false;
          }
  
          ret = aclrtCreateContext(&m_context, m_dvpp_deviceId);
          if (ret != ACL_ERROR_NONE) {
              LOG_ERROR("[{}]-aclrtCreateContext failed !", m_dec_name);
              return false;
          }
  
          // DvppSourceManager 创建时包含 aclInit,析构时包含 aclFinalize
          DvppSourceManager* pSrcMgr = DvppSourceManager::getInstance();
          m_dvpp_channel = pSrcMgr->getChannel(m_dvpp_deviceId);
          if(m_dvpp_channel < 0){
              LOG_ERROR("[{}]-该设备channel已经用完了!", m_dec_name);
              return false;
          }
25a4a277   Hu Chunming   简化解码器线程,解决500小站上因...
280
          m_vdec_out_size = frame_width * frame_height * 3 / 2;
09c2d08c   Hu Chunming   arm交付版
281
282
283
284
285
286
287
288
289
290
291
292
293
294
295
296
297
  
          LOG_INFO("[{}]- init vdpp success! device:{} channel:{}", m_dec_name, m_dvpp_deviceId, m_dvpp_channel);
          return true;
      }while(0);
  
      release_dvpp();
  
      return false;
  }
  
  bool DvppDecoder::isSurport(FFDecConfig& cfg){
      return true;
  }
  
  bool DvppDecoder::start(){
      m_bRunning = true;
  
bc01d0b6   Hu Chunming   代码优化;主要是修改了截图逻辑,避...
298
      m_read_thread = new std::thread([](void* arg)
09c2d08c   Hu Chunming   arm交付版
299
300
301
302
          {
              DvppDecoder* a=(DvppDecoder*)arg;
              a->read_thread();
              return (void*)0;
bc01d0b6   Hu Chunming   代码优化;主要是修改了截图逻辑,避...
303
          }, this);
09c2d08c   Hu Chunming   arm交付版
304
305
306
307
308
309
310
  
  	return true;
  }
  
  void DvppDecoder::close(){
      m_bRunning=false;
  
bc01d0b6   Hu Chunming   代码优化;主要是修改了截图逻辑,避...
311
312
313
314
  	if(m_read_thread != nullptr){
          m_read_thread->join();
          delete m_read_thread;
          m_read_thread = nullptr;
09c2d08c   Hu Chunming   arm交付版
315
  	}
09c2d08c   Hu Chunming   arm交付版
316
317
318
319
320
321
322
323
324
325
326
327
328
329
330
331
332
333
334
335
336
337
338
339
340
341
342
343
344
345
346
347
348
349
350
351
352
353
354
355
356
357
358
359
  }
  
  void DvppDecoder::setPostDecArg(const void* postDecArg){
      m_postDecArg = postDecArg;
  }
  
  void DvppDecoder::setFinishedDecArg(const void* finishedDecArg){
      m_finishedDecArg = finishedDecArg;
  }
  
  void DvppDecoder::pause(){
      m_bPause = true;
  }
  
  void DvppDecoder::resume(){
      m_bPause = false;
  }
  
  void DvppDecoder::setDecKeyframe(bool bKeyframe){
      m_dec_keyframe = bKeyframe;
  }
  
  bool DvppDecoder::isRunning(){
      return m_bRunning;
  }
  
  bool DvppDecoder::isFinished(){
      return m_bFinished;
  }
  
  bool DvppDecoder::isPausing(){
      return m_bPause;
  }
  
  bool DvppDecoder::getResolution(int &width, int &height){
      width = frame_width;
  	height = frame_height;
  	return true;
  }
  
  float DvppDecoder::fps(){
      return m_fps;
  }
  
bc01d0b6   Hu Chunming   代码优化;主要是修改了截图逻辑,避...
360
361
  static int snap_count = 0;
  
09c2d08c   Hu Chunming   arm交付版
362
  DeviceMemory* DvppDecoder::snapshot(){
09c2d08c   Hu Chunming   arm交付版
363
  
bc01d0b6   Hu Chunming   代码优化;主要是修改了截图逻辑,避...
364
365
366
367
368
      int ret = aclrtSetCurrentContext(m_context);
      if(ret != ACL_ERROR_NONE){
          LOG_ERROR("[{}]- aclrtSetCurrentContext failed", m_dec_name);
          return nullptr;
      }
09c2d08c   Hu Chunming   arm交付版
369
  
bc01d0b6   Hu Chunming   代码优化;主要是修改了截图逻辑,避...
370
371
372
373
374
375
376
377
378
379
380
381
382
383
384
385
386
387
388
      // 注意有锁
      DeviceMemory* snapshot_mem = nullptr;
      int loop_times = 0;
      while(m_bRunning) {
          m_decoded_data_queue_mtx.lock();
          if(m_decoded_data_queue.size() <= 0) {
              m_decoded_data_queue_mtx.unlock();
              loop_times++;
              if(loop_times > 100) {
                  // 1s都没截取到图,退出
                  break;
              }
              std::this_thread::sleep_for(std::chrono::milliseconds(10));
              continue;
          }
  
          DvppDataMemory* mem = m_decoded_data_queue.front();
          snapshot_mem = new DvppDataMemory(mem);
          m_decoded_data_queue_mtx.unlock();
09c2d08c   Hu Chunming   arm交付版
389
  
bc01d0b6   Hu Chunming   代码优化;主要是修改了截图逻辑,避...
390
391
392
393
394
395
          snap_count++;
          LOG_INFO("[{}]- snap_count:{} ", m_dec_name, snap_count);
          break;
      }
  
      return snapshot_mem;
09c2d08c   Hu Chunming   arm交付版
396
397
398
399
400
401
402
403
404
405
406
407
  }
  
  int DvppDecoder::getCachedQueueLength(){
      return 0;
  }
  
  void DvppDecoder::release_ffmpeg() {
  	m_dec_keyframe = false;
  	if(h264bsfc){
  		av_bsf_free(&h264bsfc);
  		h264bsfc = nullptr;
  	}
09c2d08c   Hu Chunming   arm交付版
408
409
410
411
      if(avctx){
          avcodec_free_context(&avctx);
          avctx = nullptr;
      }
bc01d0b6   Hu Chunming   代码优化;主要是修改了截图逻辑,避...
412
413
414
415
416
417
      if (fmt_ctx){
  		avformat_close_input(&fmt_ctx);
  		fmt_ctx = nullptr;
  	}
  
      LOG_DEBUG("[{}]- release_ffmpeg", m_dec_name);
09c2d08c   Hu Chunming   arm交付版
418
419
420
421
  }
  
  void DvppDecoder::read_thread() {
  
09c2d08c   Hu Chunming   arm交付版
422
423
  	int ret = -1;
  
25a4a277   Hu Chunming   简化解码器线程,解决500小站上因...
424
425
426
427
428
429
430
431
432
      m_bExitReportThd = false;
  	pthread_t report_thread;
  	ret = pthread_create(&report_thread, nullptr, ReportThd, (void *)this);
  	if(ret != 0){
          LOG_ERROR("[{}]- pthread_create failed", m_dec_name);
  		return;
  	}
  
      m_bExitDisplayThd = false;
bc01d0b6   Hu Chunming   代码优化;主要是修改了截图逻辑,避...
433
      std::thread display_thread = std::thread(
09c2d08c   Hu Chunming   arm交付版
434
435
436
          [](void* arg)
          {
              DvppDecoder* a=(DvppDecoder*)arg;
25a4a277   Hu Chunming   简化解码器线程,解决500小站上因...
437
              a->display_thread();
09c2d08c   Hu Chunming   arm交付版
438
              return (void*)0;
bc01d0b6   Hu Chunming   代码优化;主要是修改了截图逻辑,避...
439
440
441
          },
          this
      );
09c2d08c   Hu Chunming   arm交付版
442
  
25a4a277   Hu Chunming   简化解码器线程,解决500小站上因...
443
444
445
446
447
448
449
450
451
452
453
454
455
      aclrtContext ctx = nullptr;
      aclrtStream stream = nullptr;
      aclvdecChannelDesc *vdecChannelDesc = nullptr;
  
      do {
          CHECK_AND_BREAK(aclrtSetDevice(m_dvpp_deviceId), "aclrtSetDevice failed");
          CHECK_AND_BREAK(aclrtCreateContext(&ctx, m_dvpp_deviceId), "aclrtCreateContext failed");
          CHECK_AND_BREAK(aclrtCreateStream(&stream), "aclrtCreateStream failed");
  
          vdecChannelDesc = aclvdecCreateChannelDesc();
          if (vdecChannelDesc == nullptr) { 
              LOG_ERROR("[{}]- aclvdecCreateChannelDesc failed", m_dec_name);
              break;
09c2d08c   Hu Chunming   arm交付版
456
          }
09c2d08c   Hu Chunming   arm交付版
457
  
25a4a277   Hu Chunming   简化解码器线程,解决500小站上因...
458
459
460
461
462
463
464
465
          // 创建 channel dec结构体
          // 通道IDdvpp层面为0~31
          CHECK_AND_BREAK(aclvdecSetChannelDescChannelId(vdecChannelDesc, m_dvpp_channel), "aclvdecSetChannelDescChannelId failed");
          CHECK_AND_BREAK(aclvdecSetChannelDescThreadId(vdecChannelDesc, report_thread), "aclvdecSetChannelDescThreadId failed");
          CHECK_AND_BREAK(aclvdecSetChannelDescCallback(vdecChannelDesc, VdecCallback), "aclvdecSetChannelDescCallback failed");
          CHECK_AND_BREAK(aclvdecSetChannelDescEnType(vdecChannelDesc, m_enType), "aclvdecSetChannelDescEnType failed");
          CHECK_AND_BREAK(aclvdecSetChannelDescOutPicFormat(vdecChannelDesc, PIXEL_FORMAT_YUV_SEMIPLANAR_420), "aclvdecSetChannelDescOutPicFormat failed");
          CHECK_AND_BREAK(aclvdecCreateChannel(vdecChannelDesc), "aclvdecCreateChannel failed");
09c2d08c   Hu Chunming   arm交付版
466
  
bc01d0b6   Hu Chunming   代码优化;主要是修改了截图逻辑,避...
467
468
          AVPacket* pkt = av_packet_alloc();
          av_init_packet( pkt );
25a4a277   Hu Chunming   简化解码器线程,解决500小站上因...
469
470
          unsigned long long frame_nb = 0;
          while (m_bRunning){
09c2d08c   Hu Chunming   arm交付版
471
  
25a4a277   Hu Chunming   简化解码器线程,解决500小站上因...
472
473
474
475
476
              if (!m_bReal){
                  if (m_bPause){
                      std::this_thread::sleep_for(std::chrono::milliseconds(10));
                      continue;
                  }
c19334fe   Hu Chunming   Merge branch 'dev...
477
                  
25a4a277   Hu Chunming   简化解码器线程,解决500小站上因...
478
479
480
481
482
483
484
485
486
                  // 非实时流,即为文件情形,因为不存在花屏问题,为保证不丢帧,这里做个循环等待
                  m_decoded_data_queue_mtx.lock();
                  if(m_decoded_data_queue.size() > 5){
                      m_decoded_data_queue_mtx.unlock();
                      std::this_thread::sleep_for(std::chrono::milliseconds(5));
                      continue;
                  }
                  m_decoded_data_queue_mtx.unlock();
              }
09c2d08c   Hu Chunming   arm交付版
487
  
25a4a277   Hu Chunming   简化解码器线程,解决500小站上因...
488
489
              int result = av_read_frame(fmt_ctx, pkt);
              if (result == AVERROR_EOF || result < 0){
bc01d0b6   Hu Chunming   代码优化;主要是修改了截图逻辑,避...
490
                  av_packet_unref(pkt);
25a4a277   Hu Chunming   简化解码器线程,解决500小站上因...
491
492
493
                  LOG_WARN("[{}]- Failed to read frame!", m_dec_name);
                  break;
              }
09c2d08c   Hu Chunming   arm交付版
494
  
25a4a277   Hu Chunming   简化解码器线程,解决500小站上因...
495
              if (m_dec_keyframe && !(pkt->flags & AV_PKT_FLAG_KEY)) {
bc01d0b6   Hu Chunming   代码优化;主要是修改了截图逻辑,避...
496
                  av_packet_unref(pkt);
09c2d08c   Hu Chunming   arm交付版
497
498
499
                  continue;
              }
  
25a4a277   Hu Chunming   简化解码器线程,解决500小站上因...
500
              if (video_index == pkt->stream_index){
09c2d08c   Hu Chunming   arm交付版
501
  
25a4a277   Hu Chunming   简化解码器线程,解决500小站上因...
502
503
504
                  ret = av_bsf_send_packet(h264bsfc, pkt);
                  if(ret < 0) {
                      LOG_ERROR("[{}]- av_bsf_send_packet error!", m_dec_name);
bc01d0b6   Hu Chunming   代码优化;主要是修改了截图逻辑,避...
505
                      av_packet_unref(pkt);
25a4a277   Hu Chunming   简化解码器线程,解决500小站上因...
506
507
                      continue;
                  }
09c2d08c   Hu Chunming   arm交付版
508
  
25a4a277   Hu Chunming   简化解码器线程,解决500小站上因...
509
510
                  int nSended = -1;
                  while ((ret = av_bsf_receive_packet(h264bsfc, pkt)) == 0) {
bf661eb0   Hu Chunming   录像文件保存优化
511
  
25a4a277   Hu Chunming   简化解码器线程,解决500小站上因...
512
513
514
                      if(!m_bRunning){
                          break;
                      }
d15d6736   Hu Chunming   优化代码;避免可能的泄漏
515
  
25a4a277   Hu Chunming   简化解码器线程,解决500小站上因...
516
517
518
519
520
521
                      frame_nb++;
      #ifdef USE_VILLAGE
                      m_recoderManager.cache_pkt(pkt, frame_nb);
      #endif
                      nSended = sendPkt(vdecChannelDesc, pkt, frame_nb);
                  }
09c2d08c   Hu Chunming   arm交付版
522
  
25a4a277   Hu Chunming   简化解码器线程,解决500小站上因...
523
524
525
526
527
528
                  if(nSended < 0) {
                      break;
                  }
              } 
              
              // 音频等其他分量的情形
bc01d0b6   Hu Chunming   代码优化;主要是修改了截图逻辑,避...
529
530
531
532
533
534
535
536
              av_packet_unref(pkt);
          }
  
          av_packet_free(&pkt);
          pkt = nullptr;
  
          if (vdecChannelDesc) {
              sendVdecEos(vdecChannelDesc);
25a4a277   Hu Chunming   简化解码器线程,解决500小站上因...
537
          }
09c2d08c   Hu Chunming   arm交付版
538
  
25a4a277   Hu Chunming   简化解码器线程,解决500小站上因...
539
540
541
          while(m_bRunning && m_decoded_data_queue.size() > 0) {
              std::this_thread::sleep_for(std::chrono::milliseconds(5));
          }
09c2d08c   Hu Chunming   arm交付版
542
  
25a4a277   Hu Chunming   简化解码器线程,解决500小站上因...
543
544
      } while (0);
  
bc01d0b6   Hu Chunming   代码优化;主要是修改了截图逻辑,避...
545
      LOG_INFO("[{}]- release.", m_dec_name);
09c2d08c   Hu Chunming   arm交付版
546
  
bc01d0b6   Hu Chunming   代码优化;主要是修改了截图逻辑,避...
547
      if (vdecChannelDesc) {
25a4a277   Hu Chunming   简化解码器线程,解决500小站上因...
548
549
550
          CHECK_NOT_RETURN(aclvdecDestroyChannel(vdecChannelDesc), "aclvdecDestroyChannel failed");
          CHECK_NOT_RETURN(aclvdecDestroyChannelDesc(vdecChannelDesc), "aclvdecDestroyChannelDesc failed");
          vdecChannelDesc = nullptr;
09c2d08c   Hu Chunming   arm交付版
551
      }
09c2d08c   Hu Chunming   arm交付版
552
  
294ef5bc   Hu Chunming   修复解码器内因线程退出顺序错误导致...
553
554
      m_bRunning=false;
  
25a4a277   Hu Chunming   简化解码器线程,解决500小站上因...
555
556
557
558
      m_bExitReportThd = true;
  	CHECK_NOT_RETURN(pthread_join(report_thread, nullptr), "report_thread join failed");
  
      m_bExitDisplayThd = true;
bc01d0b6   Hu Chunming   代码优化;主要是修改了截图逻辑,避...
559
      display_thread.join();
09c2d08c   Hu Chunming   arm交付版
560
  
bc01d0b6   Hu Chunming   代码优化;主要是修改了截图逻辑,避...
561
       if(stream){
25a4a277   Hu Chunming   简化解码器线程,解决500小站上因...
562
          CHECK_NOT_RETURN(aclrtDestroyStream(stream), "aclrtDestroyStream failed");
09c2d08c   Hu Chunming   arm交付版
563
564
      }
  
25a4a277   Hu Chunming   简化解码器线程,解决500小站上因...
565
566
      if (ctx){
          CHECK_NOT_RETURN(aclrtDestroyContext(ctx), "aclrtDestroyContext failed");
09c2d08c   Hu Chunming   arm交付版
567
      }
09c2d08c   Hu Chunming   arm交付版
568
  
e01a0397   Hu Chunming   代码优化;
569
      m_recoderManager.close();
25a4a277   Hu Chunming   简化解码器线程,解决500小站上因...
570
571
572
573
574
      
      release_ffmpeg();
      release_dvpp();
  
      m_bFinished = true;
e01a0397   Hu Chunming   代码优化;
575
  
09c2d08c   Hu Chunming   arm交付版
576
      LOG_INFO("[{}]- read thread exit.", m_dec_name);
25a4a277   Hu Chunming   简化解码器线程,解决500小站上因...
577
578
579
580
  
      if(decode_finished_cbk) {
          decode_finished_cbk(m_finishedDecArg);
      }
09c2d08c   Hu Chunming   arm交付版
581
582
  }
  
25a4a277   Hu Chunming   简化解码器线程,解决500小站上因...
583
584
585
586
587
588
589
  int DvppDecoder::sendPkt(aclvdecChannelDesc *vdecChannelDesc, AVPacket* pkt, unsigned long long frame_nb){
  
      void *vdecInputbuf = nullptr;
      void *vdecOutputBuf = nullptr;
      acldvppStreamDesc *input_stream_desc = nullptr;
      acldvppPicDesc *output_pic_desc = nullptr;
      do{
bc01d0b6   Hu Chunming   代码优化;主要是修改了截图逻辑,避...
590
          int ret = acldvppMalloc((void **)&vdecInputbuf, pkt->size);
25a4a277   Hu Chunming   简化解码器线程,解决500小站上因...
591
592
593
594
595
596
597
598
599
600
601
602
603
604
605
606
607
608
609
610
611
612
613
614
615
616
617
618
619
620
621
622
623
624
625
626
627
628
629
630
          if(ACL_ERROR_NONE != ret){
              LOG_ERROR("[{}]- acldvppMalloc failed!, ret:{}", m_dec_name, ret);
              break;
          }
  
           ret = aclrtMemcpy(vdecInputbuf, pkt->size, pkt->data, pkt->size, ACL_MEMCPY_HOST_TO_DEVICE);
          if(ACL_ERROR_NONE != ret){
              LOG_ERROR("[{}]- aclrtMemcpy failed", m_dec_name);
              break;
          }
  
          ret = acldvppMalloc((void **)&vdecOutputBuf, m_vdec_out_size);
          if(ret != ACL_ERROR_NONE){
              LOG_ERROR("[{}]- acldvppMalloc failed", m_dec_name);
              break;
          }
  
          input_stream_desc = acldvppCreateStreamDesc();
          if (input_stream_desc == nullptr) { 
              LOG_ERROR("[{}]- acldvppCreateStreamDesc failed", m_dec_name);
              break;
          }
          output_pic_desc = acldvppCreatePicDesc();
          if (output_pic_desc == nullptr) { 
              LOG_ERROR("[{}]- acldvppCreatePicDesc failed", m_dec_name);
              break;
          }
          CHECK_AND_BREAK(acldvppSetStreamDescData(input_stream_desc, vdecInputbuf), "acldvppSetStreamDescData failed");
          CHECK_AND_BREAK(acldvppSetStreamDescSize(input_stream_desc, pkt->size), "acldvppSetStreamDescSize failed");
          CHECK_AND_BREAK(acldvppSetPicDescData(output_pic_desc, vdecOutputBuf), "acldvppSetPicDescData failed");
          CHECK_AND_BREAK(acldvppSetPicDescSize(output_pic_desc, m_vdec_out_size), "acldvppSetPicDescSize failed");
          
          Vdec_CallBack_UserData *user_data = NULL;
          user_data = new Vdec_CallBack_UserData;
          user_data->frameId = frame_nb;
          user_data->frame_nb = frame_nb;
          // user_data->startTime = startTime;
          user_data->sendTime = UtilTools::get_cur_time_ms();
          user_data->self = this;
  
bc01d0b6   Hu Chunming   代码优化;主要是修改了截图逻辑,避...
631
632
          m_in_count++;
  
25a4a277   Hu Chunming   简化解码器线程,解决500小站上因...
633
634
635
636
637
638
639
640
641
642
643
644
645
646
647
648
649
650
651
652
653
654
655
656
657
658
659
660
661
662
663
          ret = aclvdecSendFrame(vdecChannelDesc, input_stream_desc, output_pic_desc, nullptr, reinterpret_cast<void *>(user_data));
          if(ret != ACL_ERROR_NONE){
              LOG_ERROR("[{}]- aclvdecSendFrame failed", m_dec_name);
              delete user_data;
              user_data = nullptr;
              return 1;
          }
  
          return 0;
      }while (0);
  
      if (vdecInputbuf){
          acldvppFree(vdecInputbuf);
          vdecInputbuf = nullptr;
      }
  
      // 报错情形
      if(input_stream_desc){
          CHECK_NOT_RETURN(acldvppDestroyStreamDesc(input_stream_desc), "acldvppDestroyStreamDesc failed");
      }
  
      if (vdecOutputBuf){
          acldvppFree(vdecOutputBuf);
  	    vdecOutputBuf = nullptr;
      }
  
      if(output_pic_desc){
          CHECK_NOT_RETURN(acldvppDestroyPicDesc(output_pic_desc), "acldvppDestroyPicDesc failed");
      }
  
      return -1;
09c2d08c   Hu Chunming   arm交付版
664
665
666
667
668
669
670
671
672
673
674
675
676
677
678
679
680
681
682
  }
  
  void DvppDecoder::doProcessReport(){
  
      aclError ret = aclrtSetDevice(m_dvpp_deviceId);
      if(ret != ACL_ERROR_NONE){
          // cout << "aclrtSetDevice failed" << endl;
          LOG_ERROR("aclrtSetDevice failed !");
          return ;
      }
  
      aclrtContext ctx;
      ret = aclrtCreateContext(&ctx, m_dvpp_deviceId);
      if (ret != ACL_ERROR_NONE) {
          // cout << "aclrtCreateContext failed " << endl;
          LOG_ERROR("aclrtCreateContext failed !");
          return ;
      }
  
09c2d08c   Hu Chunming   arm交付版
683
684
685
686
687
688
689
690
691
692
693
      while (!m_bExitReportThd) {
          aclrtProcessReport(1000);
      }
  
      ret = aclrtDestroyContext(ctx);
      if(ret != ACL_ERROR_NONE){
          LOG_ERROR("aclrtDestroyContext failed !");
      }
      LOG_INFO("doProcessReport exit.");
  }
  
746db74c   Hu Chunming   实现recode
694
  void DvppDecoder::doVdppVdecCallBack(acldvppStreamDesc *input, acldvppPicDesc *output, unsigned long long frame_nb){
09c2d08c   Hu Chunming   arm交付版
695
  
bc01d0b6   Hu Chunming   代码优化;主要是修改了截图逻辑,避...
696
      m_out_count++;
09c2d08c   Hu Chunming   arm交付版
697
  
bc01d0b6   Hu Chunming   代码优化;主要是修改了截图逻辑,避...
698
      CHECK_AND_RETURN_NOVALUE(aclrtSetCurrentContext(m_context), "aclrtSetCurrentContext failed");
09c2d08c   Hu Chunming   arm交付版
699
  
25a4a277   Hu Chunming   简化解码器线程,解决500小站上因...
700
701
702
      void *inputDataDev = acldvppGetStreamDescData(input);
      acldvppFree(inputDataDev);
      inputDataDev = nullptr;
09c2d08c   Hu Chunming   arm交付版
703
704
705
706
707
708
709
710
711
712
713
714
715
716
717
718
719
  
      void *outputDataDev = acldvppGetPicDescData(output);
      uint32_t outputSize = acldvppGetPicDescSize(output);
      uint32_t width = acldvppGetPicDescWidth(output);
      uint32_t width_stride = acldvppGetPicDescWidthStride(output);
      uint32_t height = acldvppGetPicDescHeight(output);
      uint32_t height_stride = acldvppGetPicDescHeightStride(output);
  
      do{
          int ret = acldvppGetPicDescRetCode(output);
          if(ret != ACL_ERROR_NONE){
              LOG_ERROR("[{}]- decode result error, retCode:{} ", m_dec_name, ret);
              acldvppFree(outputDataDev);
              outputDataDev = nullptr;
              break;
          }
  
c5fce6ce   Zhao Shuaihua   增加二轮车玩手机505、加装雨棚5...
720
          bool bCached = false;
c5fce6ce   Zhao Shuaihua   增加二轮车玩手机505、加装雨棚5...
721
          if(width > 0 && height > 0 && outputSize > 0){
bc01d0b6   Hu Chunming   代码优化;主要是修改了截图逻辑,避...
722
  
cf231d0a   Hu Chunming   优化弃帧逻辑
723
              if (!m_bReal) {
bc01d0b6   Hu Chunming   代码优化;主要是修改了截图逻辑,避...
724
                  while(!m_bExitReportThd) {
cf231d0a   Hu Chunming   优化弃帧逻辑
725
726
                      // 非实时流,即为文件情形,因为不存在花屏问题,为保证不丢帧,这里做个循环等待
                      m_decoded_data_queue_mtx.lock();
bc01d0b6   Hu Chunming   代码优化;主要是修改了截图逻辑,避...
727
                      if(m_decoded_data_queue.size() >= 5){
cf231d0a   Hu Chunming   优化弃帧逻辑
728
729
730
731
732
733
734
735
                          m_decoded_data_queue_mtx.unlock();
                          std::this_thread::sleep_for(std::chrono::milliseconds(5));
                          continue;
                      }
                      m_decoded_data_queue_mtx.unlock();
                      break;
                  }
              }
09c2d08c   Hu Chunming   arm交付版
736
              
c5fce6ce   Zhao Shuaihua   增加二轮车玩手机505、加装雨棚5...
737
738
739
740
741
              // cout << m_dec_name << " 解码时间间隔: " << get_cur_time_ms() - last_ts << endl;
              // last_ts = get_cur_time_ms();
  
              // 换成解码后数据, 这里这样做的是为了保证解码一直持续进行,避免后续操作阻碍文件读取和解码从而导致花屏
              m_decoded_data_queue_mtx.lock();
bc01d0b6   Hu Chunming   代码优化;主要是修改了截图逻辑,避...
742
              if(m_decoded_data_queue.size() <= 5) {
c5fce6ce   Zhao Shuaihua   增加二轮车玩手机505、加装雨棚5...
743
744
745
746
747
748
749
750
                  DvppDataMemory* mem = new DvppDataMemory(width, width_stride, height, height_stride, outputSize, m_dec_name, to_string(m_dvpp_deviceId), false, frame_nb, (unsigned char *)outputDataDev);
                  if(mem){
                      m_decoded_data_queue.push(mem);
                      bCached = true;
                  }
              }
              m_decoded_data_queue_mtx.unlock();
  
c5fce6ce   Zhao Shuaihua   增加二轮车玩手机505、加装雨棚5...
751
752
753
          } 
          
          if(!bCached) {
7c962865   Zhao Shuaihua   增加不减速(517)功能, mas...
754
              LOG_WARN("[{}]- decode result warning, width:{} width_stride:{} height:{} height_stride:{} size:{}", m_dec_name, width, width_stride, height, height_stride, outputSize);
09c2d08c   Hu Chunming   arm交付版
755
756
757
              acldvppFree(outputDataDev);
              outputDataDev = nullptr;
          }
09c2d08c   Hu Chunming   arm交付版
758
759
760
761
762
763
      }while(0);
  
  	CHECK_AND_RETURN_NOVALUE(acldvppDestroyStreamDesc(input), "acldvppDestroyStreamDesc failed");
  	CHECK_AND_RETURN_NOVALUE(acldvppDestroyPicDesc(output), "acldvppDestroyPicDesc failed");
  }
  
09c2d08c   Hu Chunming   arm交付版
764
765
766
767
768
769
770
771
772
773
774
775
776
777
778
779
780
781
782
783
784
785
786
787
788
789
  bool DvppDecoder::sendVdecEos(aclvdecChannelDesc *vdecChannelDesc) {
      // create stream desc
      acldvppStreamDesc *streamInputDesc = acldvppCreateStreamDesc();
      if (streamInputDesc == nullptr) {
          LOG_ERROR("[{}]- fail to create input stream desc", m_dec_name);
          return false;
      }
      aclError ret = acldvppSetStreamDescEos(streamInputDesc, 1);
      if (ret != ACL_SUCCESS) {
          LOG_ERROR("[{}]- fail to set eos for stream desc, errorCode = {}", m_dec_name, static_cast<int32_t>(ret));
          (void)acldvppDestroyStreamDesc(streamInputDesc);
          return false;
      }
  
      // send vdec eos frame. when all vdec callback are completed, aclvdecSendFrame can be returned.
      LOG_INFO("[{}]- send eos", m_dec_name);
      ret = aclvdecSendFrame(vdecChannelDesc, streamInputDesc, nullptr, nullptr, nullptr);
      (void)acldvppDestroyStreamDesc(streamInputDesc);
      if (ret != ACL_SUCCESS) {
          LOG_ERROR("[{}]- fail to send eos frame, ret={}", m_dec_name, ret);
          return false;
      }
  
      return true;
  }
  
c5fce6ce   Zhao Shuaihua   增加二轮车玩手机505、加装雨棚5...
790
791
  void DvppDecoder::display_thread(){
      LOG_INFO("[{}]- display_thread start...", m_dec_name);
294ef5bc   Hu Chunming   修复解码器内因线程退出顺序错误导致...
792
      while(!m_bExitDisplayThd) {
c5fce6ce   Zhao Shuaihua   增加二轮车玩手机505、加装雨棚5...
793
794
795
796
797
798
          m_decoded_data_queue_mtx.lock();
          if(m_decoded_data_queue.size() <= 0) {
              m_decoded_data_queue_mtx.unlock();
              std::this_thread::sleep_for(std::chrono::milliseconds(5));
              continue;
          }
bc01d0b6   Hu Chunming   代码优化;主要是修改了截图逻辑,避...
799
  
c5fce6ce   Zhao Shuaihua   增加二轮车玩手机505、加装雨棚5...
800
801
802
803
804
805
806
807
808
809
810
811
          DvppDataMemory* mem = m_decoded_data_queue.front();
          m_decoded_data_queue.pop();
          m_decoded_data_queue_mtx.unlock();
  
          if(post_decoded_cbk) {
              post_decoded_cbk(m_postDecArg, mem);
          } else {
              delete mem;
              mem = nullptr;
          }
      }
  
bc01d0b6   Hu Chunming   代码优化;主要是修改了截图逻辑,避...
812
      m_decoded_data_queue_mtx.lock();
c5fce6ce   Zhao Shuaihua   增加二轮车玩手机505、加装雨棚5...
813
814
815
816
817
818
      while (m_decoded_data_queue.size() > 0){
          DvppDataMemory* mem = m_decoded_data_queue.front();
          m_decoded_data_queue.pop();
          delete mem;
          mem = nullptr;
      }
bc01d0b6   Hu Chunming   代码优化;主要是修改了截图逻辑,避...
819
      m_decoded_data_queue_mtx.unlock();
c5fce6ce   Zhao Shuaihua   增加二轮车玩手机505、加装雨棚5...
820
      
294ef5bc   Hu Chunming   修复解码器内因线程退出顺序错误导致...
821
      LOG_INFO("[{}]- display_thread exit.", m_dec_name);
c5fce6ce   Zhao Shuaihua   增加二轮车玩手机505、加装雨棚5...
822
823
  }
  
09c2d08c   Hu Chunming   arm交付版
824
825
826
827
828
829
  void DvppDecoder::release_dvpp(){
      if(m_context){
          aclError ret = aclrtDestroyContext(m_context);
          if(ret != ACL_ERROR_NONE){
              LOG_ERROR("[{}]- aclrtDestroyContext failed !", m_dec_name);
          }
bc01d0b6   Hu Chunming   代码优化;主要是修改了截图逻辑,避...
830
          m_context = nullptr;
09c2d08c   Hu Chunming   arm交付版
831
832
      }
      
bc01d0b6   Hu Chunming   代码优化;主要是修改了截图逻辑,避...
833
834
835
836
837
      if(m_dvpp_channel >= 0){
          DvppSourceManager* pSrcMgr = DvppSourceManager::getInstance();
  	    pSrcMgr->releaseChannel(m_dvpp_deviceId, m_dvpp_channel);
          m_dvpp_channel = -1;
      }
746db74c   Hu Chunming   实现recode
838
839
840
  }
  
  void DvppDecoder::doRecode(RecoderInfo& recoderInfo) {
d15d6736   Hu Chunming   优化代码;避免可能的泄漏
841
      m_recoderManager.create_recode_task(recoderInfo);
d9fc3e82   Hu Chunming   recode添加colse功能和mq功能
842
843
844
845
  }
  
  void DvppDecoder::set_mq_callback(mq_callback_t cb) {
      m_recoderManager.set_mq_callback(cb);
09c2d08c   Hu Chunming   arm交付版
846
  }