FFmpeg5.0.1 代码读取MP4视频推送RTMP流

当下代码在FfmpegTest类创建实例时调用

可根据实际情况在main()函数中调用videoPlay(pszFile, pszRTMPURL)。

源代码分享

#include "ffmpegtest.h"
#include <iostream>
#include <QDebug>

using namespace std;

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

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

void XError(int errNum)
{
	char buf[1024] = { 0 };
	av_strerror(errNum, buf, sizeof(buf));
	cout << buf << endl;
	getchar();
}
static double r2d(AVRational r)
{
	return r.num == 0 || r.den == 0 ? 0. : (double)r.num / (double)r.den;
}

FfmpegTest::FfmpegTest(QWidget *parent)
	: QMainWindow(parent)
{
	ui.setupUi(this);
    //主函数调用部分
	const char* pszFile = "D:/work/Ffmpeg/ffmpeg_code/guduyueqiu.mp4";
	const char* pszRTMPURL = "rtmp://192.168.3.10/live/livestream";
	videoPlay(pszFile, pszRTMPURL);
}
void FfmpegTest::videoPlay(const char* pszFile, const char* pszRTMPURL)
{
	int nVideoIndex = -1;
	avformat_network_init();

	AVFormatContext* pInputAVFormatContext = NULL;
	int nRet = avformat_open_input(&pInputAVFormatContext, pszFile, 0, NULL);
	if (nRet < 0)
	{
		XError(nRet);
	}
	nRet = avformat_find_stream_info(pInputAVFormatContext, 0);
	if (nRet != 0)
	{
		XError(nRet);
	}
	av_dump_format(pInputAVFormatContext, 0, pszFile, 0);
	AVFormatContext* pOutputAVFormatContext = NULL;
	nRet = avformat_alloc_output_context2(&pOutputAVFormatContext, NULL, "flv", pszRTMPURL);
	if (nRet < 0)
	{
		XError(nRet);
	}

	for (int i = 0; i < pInputAVFormatContext->nb_streams; i++)
	{
		AVStream* pInputAVStream = pInputAVFormatContext->streams[i];
		AVStream* pOutputAVStream = avformat_new_stream(pOutputAVFormatContext, 0);
		nRet = avcodec_parameters_copy(pOutputAVStream->codecpar, pInputAVStream->codecpar);
		if (nRet < 0)
		{
			XError(nRet);
		}
		pOutputAVStream->codecpar->codec_tag = 0;
	}
	for (int i = 0; i < pInputAVFormatContext->nb_streams; i++)
	{
		if (pInputAVFormatContext->streams[i]->codecpar->codec_type == AVMEDIA_TYPE_VIDEO)
		{
			nVideoIndex = i;
			break;
		}
	}
	av_dump_format(pOutputAVFormatContext, 0, pszRTMPURL, 1);
	nRet = avio_open(&pOutputAVFormatContext->pb, pszRTMPURL, AVIO_FLAG_WRITE);
	if (nRet < 0)
	{
		XError(nRet);
	}
	nRet = avformat_write_header(pOutputAVFormatContext, 0);
	if (nRet < 0)
	{
		XError(nRet);
	}
	AVPacket pkt;
	std::int64_t llStartTime = av_gettime();
	std::int64_t llFrameIndex = 0;
	while (true)
	{
		AVStream* pInputStream = NULL;
		AVStream* pOutputStream = NULL;
		nRet = av_read_frame(pInputAVFormatContext, &pkt);
		if (nRet < 0)
		{
			break;
		}
		if (pkt.pts == AV_NOPTS_VALUE)
		{
			AVRational timeBase = pInputAVFormatContext->streams[nVideoIndex]->time_base;
			std::int64_t llCalcDuration = (double)AV_TIME_BASE / av_q2d(pInputAVFormatContext->streams[nVideoIndex]->r_frame_rate);
			pkt.pts = (double)(llFrameIndex * llCalcDuration) / (double(av_q2d(timeBase)*AV_TIME_BASE));
			pkt.dts = pkt.pts;
			pkt.duration = (double)llCalcDuration / (double)(av_q2d(timeBase)*AV_TIME_BASE);
		}
		if (pkt.stream_index == nVideoIndex)
		{
			AVRational timeBase = pInputAVFormatContext->streams[nVideoIndex]->time_base;
			AVRational timeBaseQ = { 1, AV_TIME_BASE };
			std::int64_t pts_time = av_rescale_q(pkt.dts, timeBase, timeBaseQ);
			std::int64_t now_time = av_gettime() - llStartTime;
			AVRational avr = pInputAVFormatContext->streams[nVideoIndex]->time_base;
			if (pts_time > now_time)
			{
				av_usleep((unsigned int)(pts_time - now_time));
			}
		}
		pInputStream = pInputAVFormatContext->streams[pkt.stream_index];
		pOutputStream = pOutputAVFormatContext->streams[pkt.stream_index];
		pkt.pts = av_rescale_q_rnd(pkt.pts, pInputStream->time_base, pOutputStream->time_base, (AVRounding)(AV_ROUND_NEAR_INF | AV_ROUND_PASS_MINMAX));
		pkt.dts = av_rescale_q_rnd(pkt.dts, pInputStream->time_base, pOutputStream->time_base, (AVRounding)(AV_ROUND_NEAR_INF | AV_ROUND_PASS_MINMAX));
		pkt.duration = (int)av_rescale_q(pkt.duration, pInputStream->time_base, pOutputStream->time_base);
		pkt.pos = -1;
		if (pkt.stream_index == nVideoIndex)
		{
			llFrameIndex++;
		}
		nRet = av_interleaved_write_frame(pOutputAVFormatContext, &pkt);
		if (nRet < 0) {
			std::cout << "发送数据包出错" << std::endl;
			break;
		}
		av_packet_unref(&pkt);
	}
	av_write_trailer(pOutputAVFormatContext);
	if (!(pOutputAVFormatContext->oformat->flags & AVFMT_NOFILE))
	{
		avio_close(pOutputAVFormatContext->pb);
	}
	avformat_free_context(pOutputAVFormatContext);
	avformat_close_input(&pInputAVFormatContext);
}

效果展示