从mp4、flv、ts文件中提取出AVC（h264）

分别从封装格式为mp4、flv、ts文件中提取出AVC(h264)并写入文件

/*****************************************************************//**
 * \file   WriteAVC.cpp
 * \brief  提取视频中的视频流，写入为AVC(h264)文件
 * 
 * \author 13648
 * \date   April 2024
 *********************************************************************/
#define _CRT_SECURE_NO_WARNINGS
#include "myLog.h"
#include <iostream>

extern "C"
{
#include <libavformat/avformat.h>
#include <libavcodec/avcodec.h>
#include <libavcodec/bsf.h>
}

#pragma comment(lib, "avformat.lib")
#pragma comment(lib, "avcodec.lib")

/**
 * 由于h264有两种格式，一种mp4格式（不带startcode，sps，pps信息）一种annexb格式（带startcode，sps、pps信息）
 * 但是大多数的编解码器只支持annexb格式pkt数据，
 * 因此在处理mp4或者flv格式的文件时需要使用h264_mp4toannexb比特流过滤器处理
 */
int main()
{
	AVFormatContext* ifmt_ctx = NULL;
	const std::string in_filename = "./MediaFile/input_5s.flv";		// input_5s.flv input_5s.ts

	int nRet = avformat_open_input(&ifmt_ctx, in_filename.c_str(), NULL, NULL);
	if (nRet != 0)
	{
		LOG_WARNING("avformat_open_input error\n");
		return -1;
	}

	nRet = avformat_find_stream_info(ifmt_ctx, NULL);
	if (nRet < 0)
	{
		LOG_WARNING("avformat_find_stream_info faild\n");
		avformat_close_input(&ifmt_ctx);
		return -2;
	}

	int video_idx = av_find_best_stream(ifmt_ctx, AVMEDIA_TYPE_VIDEO, -1, -1, NULL, 0);
	if (video_idx < 0)
	{
		LOG_WARNING("avformat_find_stream_info faild\n");
		avformat_close_input(&ifmt_ctx);
		return -3;
	}

	const std::string out_filename = "./MediaFile/input_5s.h264";
	FILE* out_fp = fopen(out_filename.c_str(), "wb");
	if (!out_fp)
	{
		LOG_WARNING("open out_filename faild\n");
		avformat_close_input(&ifmt_ctx);
		return -4;
	}

	// 获取比特流过滤器
	const AVBitStreamFilter* bsfilter = av_bsf_get_by_name("h264_mp4toannexb");
	AVBSFContext* bsf_ctx = NULL;
	// 初始化比特流过滤器
	av_bsf_alloc(bsfilter, &bsf_ctx);
	// 给过滤器上下文添加解码器属性
	avcodec_parameters_copy(bsf_ctx->par_in, ifmt_ctx->streams[video_idx]->codecpar);
	av_bsf_init(bsf_ctx);

	AVPacket* pkt = av_packet_alloc();
	pkt->data = NULL;
	pkt->size = 0;
	while (true)
	{
		nRet = av_read_frame(ifmt_ctx, pkt);
		if (nRet < 0)
		{
			LOG_WARNING("av_read_frame file finish\n");
			break;
		}
		if (nRet == 0 && pkt->stream_index == video_idx)
		{
#if 1
			if (av_bsf_send_packet(bsf_ctx, pkt) != 0)	// 发送包到比特流过滤器添加startcode、sps、pps信息
			{
				av_packet_unref(pkt);
				continue;
			}
			av_packet_unref(pkt);				// 清除，下面使用这个pkt接收bsf处理后的包
			while (av_bsf_receive_packet(bsf_ctx, pkt) == 0)	// 添加sps pps后可能不止一个包
			{
				size_t writeLen = fwrite(pkt->data, 1, pkt->size, out_fp);
				if (writeLen != pkt->size)
				{
					LOG_WARNING("write pkt error\n");
				}
				fflush(out_fp);
				av_packet_unref(pkt);
			}
#else
			// 如果是ts流(自带sps pps信息,可以直接写入)
			size_t writeLen = fwrite(pkt->data, 1, pkt->size, out_fp);
			if (writeLen != pkt->size)
			{
				LOG_WARNING("fwrite pkt error\n");
			}
			av_packet_unref(pkt);
#endif
		}
		else if(nRet == 0) 
		{
			av_packet_unref(pkt);
		}
	}

	if (pkt)
	{
		av_packet_free(&pkt);
	}
	if (out_fp)
	{
		fclose(out_fp);
	}
	if (bsf_ctx)
	{
		av_bsf_free(&bsf_ctx);
	}
	if (ifmt_ctx)
	{
		avformat_close_input(&ifmt_ctx);
	}

	return 0;
}