首页 > 学院 > 开发设计 > 正文

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

2019-11-06 06:12:34
字体:
来源:转载
供稿:网友

分离某些封装格式中的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%20filter需要初始化和销毁,分别通过函数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%20filter处理之后得到SPS、PPS,拷贝至每个IDR帧之前。下面代码示例了写入SPS、PPS的过程。

[cpp] view%20plain copy 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%20plain copy char nal_start[]={0,0,0,1};  memcpy(packet->data,nal_start,4);  经过上述两步也可以得到可以播放的H.264码流,相对于第一种方法来说复杂一些。
发表评论 共有条评论
用户名: 密码:
验证码: 匿名发表