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 level,264 baseline level,main level,high 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结构体
// 通道ID在dvpp层面为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
|
}
|