//有些大段的备注是机器翻译,知道大概意思就行。
//C++,用ffmpeg提取视频转码并另存 ,本例为验证程序,只有这个程序搞懂了,才可能用ffmpge做个播放器什么的。
#include <QCoreApplication>
#include <iostream>
using namespace std;
extern "C"{
#include <libavcodec/avcodec.h>
#include <libavformat/avformat.h>
}
int main(int argc, char *argv[])
{
QCoreApplication a(argc, argv);
//const char* url {"a.mkv"};
const char* url {"b.mp4"};
const char* out {"out.yuv"};
AVFormatContext *fmt_ctx {};
int re = avformat_open_input(
&fmt_ctx, //AVFormatContext
url,
0, // 0表示自动选择解封器
0 //参数设置,比如rtsp的延时时间
);
if(re<0)
{
cout<<"avformat_open_input err"<<endl;
exit(-1);
}
av_dump_format(fmt_ctx, 0, url, 0);//打印文件信息
re = avformat_find_stream_info(fmt_ctx, 0);//获取流信息
if(re<0)
{
cout<<"avformat_find_stream_info err"<<endl;
exit(-1);
}
re=av_find_best_stream(fmt_ctx,AVMEDIA_TYPE_VIDEO,-1,-1,NULL,0);//在文件中找到“最佳”流。最佳流是根据各种启发式来确定的,因为最有可能是用户所期望的。如果decoder参数非null, av_find_best_stream将为流的编解码器找到默认的解码器;无法找到解码器的流将被忽略。
if(re<0)
{
cout<<"av_find_best_stream err"<<endl;
exit(-1);
}
int stream_index=re;
AVStream *video_stream=fmt_ctx->streams[stream_index];
const AVCodec *av_dec=avcodec_find_decoder(video_stream->codecpar->codec_id);//查找具有匹配编解码器ID的注册解码器。
AVCodecContext *dec_ctx=avcodec_alloc_context3(av_dec);//分配一个AVCodecContext并设置它的字段为默认值。结果结构应该使用avcodec_free_context()释放。
re=avcodec_parameters_to_context(dec_ctx,video_stream->codecpar); //根据提供的编解码器参数的值填充编解码器上下文。在编解码器中,任何在“par”中具有相应字段的已分配字段将被释放并替换为par中相应字段的副本。编解码器中没有对应字段的字段不会被修改。
if(re<0)
{
cout<<"avcodec_parameters_to_context err"<<endl;
exit(-1);
}
re=avcodec_open2(dec_ctx,av_dec,nullptr);
if(re<0)
{
cout<<"avcodec_open2 err"<<endl;
exit(-1);
}
FILE *output_video_file=fopen(out,"wb");
if(!output_video_file)
{
cout<<"fopen(out,wb) err"<<endl;
exit(-1);
}
av_dump_format(fmt_ctx,0,url,0); //打印文件信息,和上面的已有所不同
video_stream=fmt_ctx->streams[stream_index];
cout<<"video_stream="<<video_stream<<endl;
AVPacket *pkt= av_packet_alloc();
AVFrame *frame= av_frame_alloc();
while(av_read_frame(fmt_ctx,pkt)>=0) //返回流的下一帧。这个函数返回文件中存储的内容,并且不验证解码器是否存在有效的帧。它将文件中存储的内容分割成帧,并为每个调用返回一个帧。它不会省略有效帧之间的无效数据,以便给解码器解码的最大可能信息。如果成功,返回的数据包是引用计数的(设置了pkt->buf)并且无限期有效。当不再需要该包时,必须使用av_packet_unref()释放它。对于视频,数据包只包含一帧。对于音频,如果每一帧都有一个已知的固定大小(例如PCM或ADPCM数据),它包含一个整数帧数。如果音频帧有一个可变的大小(例如MPEG音频),那么它包含一个帧。pkt->pts, pkt->dts和pkt->持续时间总是设置为AVStream中的正确值。Time_base单位(猜测格式是否不能提供它们)。pkt->pts可以是AV_NOPTS_VALUE如果视频格式有b帧,所以最好依赖于pkt->dts如果你不解压有效负载。
{
int re {0};
if(pkt->stream_index==stream_index)
{
re=avcodec_send_packet(dec_ctx,pkt);
if(re<0)
{
cout<<"avcodec_send_packet err"<<endl;
break;
}
while(1)
{
re=avcodec_receive_frame(dec_ctx,frame);//一次avcodec_send_packet可能需要多次avcodec_receive_frame
if(re<0)
{
cout<<"avcodec_receive_frame err"<<endl;
break;
}
//write_frame_to_yuv(frame);以下为写入文件函数
uint8_t **pBuf=frame->data;
int *pStride=frame->linesize;
for(size_t i=0;i<3;i++)
{
int32_t width=(i==0?frame->width:frame->width/2);
int32_t height=(i==0?frame->height:frame->height/2);
for(size_t j=0;j<height;j++)
{
fwrite(pBuf[i],1,width,output_video_file);
pBuf[i]+=pStride[i];
}
}
cout<<"frame="<<frame<<endl;
//write_frame_to_yuv(frame);以上为写入文件函数
}
}
av_packet_unref(pkt);
}
//以下为分配空间释放
av_frame_free(&frame);
av_packet_free(&pkt);
avcodec_free_context(&dec_ctx);
avformat_close_input(&fmt_ctx);
if (output_video_file != nullptr) {
fclose(output_video_file);
output_video_file = nullptr;
cout<<"output_video_file close"<<endl;
}
end:
return a.exec();
}
/*
生成的out.yuv文件用以下命令测试,要注意视频文件的分辨率,分辨率不同需要调整,例如 1920x800、640x320等
ffplay -f rawvideo -video_size 1920x800 out.yuv
*/