本例演示如何打开一个媒体文件,把输出到另一个媒体文件中,并把一些包去掉。
定义全局变量
AVFormatContext *g_inputContext = NULL;
AVFormatContext * g_outputContext;
int64_t g_lastReadPacktTime = 0;
1、打开一个输入媒体文件
打开媒体文件,读取媒体文件的数据包,分析其中的流信息。分析其帧率
int OpenInputMediaFile(string inputUrl)
{
//创建一个AVFormatContext类型变量并初始化默认参数
g_inputContext = avformat_alloc_context();
// 取得时间,单位为microsecond,微秒
g_lastReadPacktTime = av_gettime();
// 填写检测是否中止阻塞功能的回调函数。
g_inputContext->interrupt_callback.callback = interrupt_cb;
// 强制以指定格式打开,为NULL时,自动检测其格式。
AVInputFormat *fmt = NULL;
// 强制以指定格式打开,为NULL时,自动检测其格式。
AVDictionary **options = NULL;
// 打开一个输入流(本地文件名或者是网络流地址URL),并解析其头部格式,并把信息填写到g_inputContext中。
int ret = avformat_open_input(&g_inputContext, inputUrl.c_str(), fmt, options);
if (ret < 0)
{
return ret;
}
// 读取媒体文件的数据包,分析其中的流信息。分析其帧率
ret = avformat_find_stream_info(g_inputContext, nullptr);
return ret;
}
2、打开输出媒体文件
int OpenOutputMediaFile(string outUrl)
{
// 创建一个用于输出的格式信息对像(AVFormatContext)
// 格式指定flv
int ret = avformat_alloc_output_context2(&g_outputContext, nullptr, "flv", outUrl.c_str());
if (ret < 0)
{
goto Error;
}
// 打开文件 并初始AVIOContext(g_outputContext->pb)
ret = avio_open2(&g_outputContext->pb, outUrl.c_str(), AVIO_FLAG_WRITE, nullptr, nullptr);
if (ret < 0)
{
goto Error;
}
// 找出源媒体中流的信息
for (int i = 0; i < g_inputContext->nb_streams; i++)
{
// 用输入流的编码信息来创建一个流对像,
AVStream * stream = avformat_new_stream(g_outputContext, g_inputContext->streams[i]->codec->codec);
// 把输入流的编码信息拷贝到个输出流对的编码信息中,
ret = avcodec_copy_context(stream->codec, g_inputContext->streams[i]->codec);
if (ret < 0)
{
goto Error;
}
}
// 分配流的私有信息,并把流的头部信息写到输出文件
ret = avformat_write_header(g_outputContext, nullptr);
if (ret < 0)
{
goto Error;
}
return ret;
Error:
// 出错后关闭输出环境变量
if (g_outputContext)
{
for (int i = 0; i < g_outputContext->nb_streams; i++)
{
avcodec_close(g_outputContext->streams[i]->codec);
}
avformat_close_input(&g_outputContext);
}
return ret;
}
3、读入一帧数据
shared_ptr<AVPacket> ReadPacketFromSource()
{
shared_ptr<AVPacket> packet(static_cast<AVPacket*>(av_malloc(sizeof(AVPacket))), [&](AVPacket *p)
{ av_packet_free(&p);
av_freep(&p); }
);
// 初始化packet
av_init_packet(packet.get());
// 取时间
g_lastReadPacktTime = av_gettime();
// 读入一帧数据到packet
int ret = av_read_frame(g_inputContext, packet.get());
if (ret >= 0)
{
return packet;
}
else
{
return nullptr;
}
}
4、把数据写到输出文件中
int WritePacket(shared_ptr<AVPacket> packet)
{
return av_interleaved_write_frame(g_outputContext, packet.get());
}
5、初始化库
// 初始化FFMpeg库
void InitFFMpeg()
{
av_register_all(); // 注册所有的编解码器
avfilter_register_all();// 注册所有的滤镜
// 初网络组件。如果输入输出文件,不涉及网络,可以不调用。
avformat_network_init();
}
6、关闭资源
void CloseInput()
{
if (g_inputContext != nullptr)
{
// 关闭输入环境,释放结构中的资源
avformat_close_input(&g_inputContext);
}
}
void CloseOutput()
{
if (g_outputContext != nullptr)
{
for (int i = 0; i < g_outputContext->nb_streams; i++)
{
AVCodecContext *codecContext = g_outputContext->streams[i]->codec;
// 关闭输出编码器,释放相关的数据。
avcodec_close(codecContext);
}
// 关闭输出环境,释放结构中的资源
avformat_close_input(&g_outputContext);
}
}
7、主控流程
int convert_to_new_file(int startPacketNum, int discardPacketNum, string sourceFileName, string destFileName)
{
int discardtPacketNum = discardPacketNum;
int packetCount = 0;
int64_t lastPacketPts = AV_NOPTS_VALUE;
int64_t lastPts = AV_NOPTS_VALUE;
InitFFMpeg();
int ret = OpenInputMediaFile(sourceFileName);
if (ret >= 0)
{
ret = OpenOutputMediaFile(destFileName);
}
if (ret <0)
goto Error;
int endPackNum = startPacketNum + discardtPacketNum;
while (true)
{
// 从输入流中读取一个数据包
auto packet = ReadPacketFromSource();
if (packet)
{
packetCount++;
// 不在指定范围的数据包,输出到文件中。
if (packetCount <= startPacketNum || packetCount >= endPackNum)
{
// 重新计算输出数据包中的时间
if (packetCount >= endPackNum)
{
if (packet->pts - lastPacketPts > 120) // 说明是跨区间【startPacketNum,endPackNum】的帧
{
lastPts = lastPacketPts + 30; // 加上一个帧率间隔,假定是30ms.
}
else
{
auto diff = packet->pts - lastPacketPts;// 计算与上一数据包的时间差
lastPts += diff;
}
}
lastPacketPts = packet->pts;// 记录输入数据包中的时间信息
if (lastPts != AV_NOPTS_VALUE)
{
packet->pts = packet->dts = lastPts;
}
ret = WritePacket(packet);// 输出到文件中。
}// 不在指定范围的输入数据,不写入。
}
else
{
break;
}
}
Error:
CloseInput(); // 关闭输入
CloseOutput();// 关闭输出
return 0;
}
8、下载完成代码
在Debug–x86编译通过。