2017-04-06 122 views
1

我想給的ffmpeg解碼H264,但最後我發現瞭解碼功能只用一個CPU核心FFMpeg如何使用多線程?

system monitor

ENV:Ubuntu的14.04 FFmpeg的3.2.4 CPU i7-7500U


所以,我搜索ffmpeg多線程並決定使用所有cpu內核進行解碼。
我設置AVCodecContext就象這樣:

//Init works 
//codecId=AV_CODEC_ID_H264; 
avcodec_register_all(); 
pCodec = avcodec_find_decoder(codecId); 
if (!pCodec) 
{ 
    printf("Codec not found\n"); 
    return -1; 
} 
pCodecCtx = avcodec_alloc_context3(pCodec); 
if (!pCodecCtx) 
{ 
    printf("Could not allocate video codec context\n"); 
    return -1; 
} 

pCodecParserCtx=av_parser_init(codecId); 
if (!pCodecParserCtx) 
{ 
    printf("Could not allocate video parser context\n"); 
    return -1; 
} 
pCodecCtx->thread_count = 4; 
pCodecCtx->thread_type = FF_THREAD_FRAME; 

pCodec->capabilities &= CODEC_CAP_TRUNCATED; 
pCodecCtx->flags |= CODEC_FLAG_TRUNCATED; 

if (avcodec_open2(pCodecCtx, pCodec, NULL) < 0) 
{ 
    printf("Could not open codec\n"); 
    return -1; 
} 
av_log_set_level(AV_LOG_QUIET); 
av_init_packet(&packet); 

//parse and decode 
//after av_parser_parse2, the packet has a complete frame data 
//in decode function, I just call avcodec_decode_video2 and do some frame copy work 
while (cur_size>0) 
{ 
    int len = av_parser_parse2(
        pCodecParserCtx, pCodecCtx, 
        &packet.data, &packet.size, 
        cur_ptr, cur_size, 
        AV_NOPTS_VALUE, AV_NOPTS_VALUE, AV_NOPTS_VALUE); 

    cur_ptr += len; 
    cur_size -= len; 
    if(GetPacketSize()==0) 
     continue; 

    AVFrame *pFrame = av_frame_alloc(); 
    int ret = Decode(pFrame); 
    if (ret < 0) 
    { 
     continue; 
    } 
    if (ret) 
    { 
     //some works 
    } 
} 

但沒有與之前不同。
如何在FFMpeg中使用多線程?有任何建議嗎?

+0

您將需要顯示更多代碼。你如何衡量使用了多少核心?你如何解碼幀?什麼版本的FFmpeg?你是如何分配pCodecParserCtx的? –

+0

Recv rtp流首先使用boost asio,然後用ffmpeg解碼,用opengl顯示。我添加了一些解析和解碼代碼。我只是看系統監視器來測量核心的用法,如果我只是解碼視頻和沒有顯示器,只有一個核心工作。 –

回答

1

pCodec-> capabilities & = CODEC_CAP_TRUNCATED;

這就是你的錯誤。請刪除此行。 avcodec_find_decoder()的返回值應該適用於所有實際的意圖和目的。

具體而言,此語句從編解碼器的功能中刪除AV_CODEC_CAP_FRAME_THREADS標誌,從而有效地禁用其餘代碼中的幀多線程。

+0

你是對的,這是一個錯誤。 –