ffmpeg 手动添加 sps pps(处理内存泄露问题)

3/8/2017来源:ASP.NET技巧人气:4240

分离某些封装格式中的H.264

分离某些封装格式(例如MP4/FLV/MKV等)中的H.264的时候,需要首先写入SPS和PPS,否则会导致分离出来的数据没有SPS、PPS而无法播放。H.264码流的SPS和PPS信息存储在AVCodecContext结构体的extradata中。需要使用ffmpeg中名称为“h264_mp4toannexb”的bitstream filter处理。有两种处理方式:

(1)使用bitstream filter处理每个AVPacket(简单)

把每个AVPacket中的数据(data字段)经过bitstream filter“过滤”一遍。关键函数是av_bitstream_filter_filter()。示例代码如下。

[cpp] view plain copy  在CODE上查看代码片派生到我的代码片 AVBitStreamFilterContext* h264bsfc =  av_bitstream_filter_init("h264_mp4toannexb");    while(av_read_frame(ifmt_ctx, &pkt)>=0){       if(pkt.stream_index==videoindex){           av_bitstream_filter_filter(h264bsfc, ifmt_ctx->streams[videoindex]->codec, NULL, &pkt.data, &pkt.size, pkt.data, pkt.size, 0);           fwrite(pkt.data,1,pkt.size,fp_video);           //...       }       av_free_packet(&pkt);   }   av_bitstream_filter_close(h264bsfc);    

上述代码中,把av_bitstream_filter_filter()的输入数据和输出数据(分别对应第4,5,6,7个参数)都设置成AVPacket的data字段就可以了。

需要注意的是bitstream filter需要初始化和销毁,分别通过函数av_bitstream_filter_init()和av_bitstream_filter_close()。

经过上述代码处理之后,AVPacket中的数据有如下变化:

*每个AVPacket的data添加了H.264的NALU的起始码{0,0,0,1}

*每个IDR帧数据前面添加了SPS和PPS

(2)手工添加SPS,PPS(稍微复杂)

将AVCodecContext的extradata数据经过bitstream filter处理之后得到SPS、PPS,拷贝至每个IDR帧之前。下面代码示例了写入SPS、PPS的过程。

[cpp] view plain copy  在CODE上查看代码片派生到我的代码片 FILE *fp=fopen("test.264","ab");   AVCodecContext *pCodecCtx=...     unsigned char *dummy=NULL;      int dummy_len;     AVBitStreamFilterContext* bsfc =  av_bitstream_filter_init("h264_mp4toannexb");       av_bitstream_filter_filter(bsfc, pCodecCtx, NULL, &dummy, &dummy_len, NULL, 0, 0);     fwrite(pCodecCtx->extradata,pCodecCtx-->extradata_size,1,fp);     av_bitstream_filter_close(bsfc);       free(dummy);     然后修改AVPacket的data。把前4个字节改为起始码。示例代码如下所示。

[cpp] view plain copy  在CODE上查看代码片派生到我的代码片 char nal_start[]={0,0,0,1};   memcpy(packet->data,nal_start,4);   经过上述两步也可以得到可以播放的H.264码流,相对于第一种方法来说复杂一些。