ffmpeg 代码实现rtmp推流到服务器

前面2篇已经讲解了如何搭建直播服务器 ffmpeg + nginx配置智播推流服务器和ffmpeg + crtmpserver实现直播推流，并且通过ffmpeg命令实现了推流，下面讲解下如何用代码实现推流，并且通过设置dts来实现推流的速度。

之前也写过，将摄像头的数据保存到本地为mp4：ffmpeg 获取电脑摄像头数据，并且保存为MP4视频 ，这里其实差不多，只是将输出设备从本地文件mp4 改为了 rtmp://192.168.32.129/live（自己搭建的rtmp服务器）

这里主要是介绍下，解封装后，如何创建输出流，并且控制推流速度，让播放器端正常播放。

创建输出流

• 创建输出上下文

 

//创建输出流上下文
	AVFormatContext *octx = NULL;
	re = avformat_alloc_output_context2(&octx, 0, "flv", outUrl);
	if (!octx)
	{
		return XError(re);
	}
	cout << "octx create success!" << endl;

• 创建输出流

for (int i = 0; i < ictx->nb_streams; i++)
	{
		//创建输出流
		AVCodec *codec = avcodec_find_decoder(ictx->streams[i]->codecpar->codec_id);
		AVStream *out = avformat_new_stream(octx, codec);
		
		if (!out)
		{
			return XError(0);
		}
		//复制配置信息,同于MP4，
		//re = avcodec_copy_context(out->codec, ictx->streams[i]->codec);
		re = avcodec_parameters_copy(out->codecpar, ictx->streams[i]->codecpar);
		//out->codec->codec_tag = 0;
	}
	av_dump_format(octx, 0, outUrl, 1);

这里因为avStream结构体中codec已经弃用，但用新版本推送mp4视频的时候会出现不支持，但ffmpeg还是会封装推送，在服务器端支持，并且解析出来，但为了不出现这个错误有2种方案：

1、将推流的mp4文件用 ffmpeg转封装为flv，然后用新版本来实现推流

ffmpeg -i 1080.mp4 test.flv

2、切换到旧版本

这里我用的新版本，推送flv视频

推流

推流主要是要通过pts 或者dts来进行控制速度。

原理： 在推流开始时候几下时间start（1970至今的时间，单位为微妙 微秒：百万分之一秒），然后解封转后每一帧，得到pts 和 dts，然后将packet 的pts 和 dts转化为outAvformatContext 的，并且如果dts 转化为与start一样时间单位时候，如果大于 now - start ，就休眠 dts-（now - start）

这里用到了2个ffmpeg的函数：

• av_strerror 输出错误信息

int XError(int errNum)
{
	char buf[1024] = { 0 };
	av_strerror(errNum, buf, sizeof(buf));
	cout << buf << endl;
	getchar();
	return -1;
}

• av_gettime：1970至今的时间，单位为微妙 微秒：百万分之一秒
• av_rescale_q_rnd 将输入的pts格式转化为输出上下文的格式

int64_t av_rescale_rnd(int64_t a, int64_t b, int64_t c, enum AVRounding rnd)

最后一个参数转换的方式：

AV_ROUND_ZERO     = 0, // Round toward zero.      趋近于0  
AV_ROUND_INF      = 1, // Round away from zero.   趋远于0  
AV_ROUND_DOWN     = 2, // Round toward -infinity. 趋于更小的整数  
AV_ROUND_UP       = 3, // Round toward +infinity. 趋于更大的整数  
AV_ROUND_NEAR_INF = 5, // Round to nearest and halfway cases away from zero.  
                       //                         四舍五入,小于0.5取值趋向0,大于0.5取值趋远于0  

完整的代码如下：（没有考虑控件释放，和 优化）

extern "C"
{
#include "libavformat/avformat.h"
#include "libavutil/time.h"
}
#include <iostream>
using namespace std;
#pragma comment(lib,"avformat.lib")
#pragma comment(lib,"avutil.lib")
#pragma comment(lib,"avcodec.lib")

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

int main(int argc, char *argv[])
{

	const char *inUrl = "test.flv";
	const char *outUrl = "rtmp://192.168.32.129/live";

	//初始化所有封装和解封装 flv mp4 mov mp3
	av_register_all();

	//初始化网络库
	avformat_network_init();

	//
	//输入流 1 打开文件，解封装
	//输入封装上下文
	AVFormatContext *ictx = NULL;

	//打开文件，解封文件头
	int re = avformat_open_input(&ictx, inUrl, 0, 0);
	if (re != 0)
	{
		return XError(re);
	}
	cout << "open file " << inUrl << " Success." << endl;

	//获取音频视频流信息 ,h264 flv
	re = avformat_find_stream_info(ictx, 0);
	if (re != 0)
	{
		return XError(re);
	}
	av_dump_format(ictx, 0, inUrl, 0);
	//


	//
	//输出流 

	//创建输出流上下文
	AVFormatContext *octx = NULL;
	re = avformat_alloc_output_context2(&octx, 0, "flv", outUrl);
	if (!octx)
	{
		return XError(re);
	}
	cout << "octx create success!" << endl;

	//配置输出流
	//遍历输入的AVStream
	for (int i = 0; i < ictx->nb_streams; i++)
	{
		//创建输出流
		AVCodec *codec = avcodec_find_decoder(ictx->streams[i]->codecpar->codec_id);
		AVStream *out = avformat_new_stream(octx, codec);
		
		//AVStream *out = avformat_new_stream(octx, );
		if (!out)
		{
			return XError(0);
		}
		//复制配置信息,同于MP4
		//re = avcodec_copy_context(out->codec, ictx->streams[i]->codec);
		re = avcodec_parameters_copy(out->codecpar, ictx->streams[i]->codecpar);
		//out->codec->codec_tag = 0;
	}
	av_dump_format(octx, 0, outUrl, 1);
	//


	//rtmp推流

	//打开io
	re = avio_open(&octx->pb, outUrl, AVIO_FLAG_WRITE);
	if (!octx->pb)
	{
		return XError(re);
	}

	//写入头信息
	re = avformat_write_header(octx, 0);
	if (re < 0)
	{
		return XError(re);
	}
	cout << "avformat_write_header " << re << endl;
	AVPacket pkt;
	long long startTime = av_gettime();
	for (;;)
	{
		re = av_read_frame(ictx, &pkt);
		if (re != 0)
		{
			break;
		}
		cout << pkt.pts << " " << flush;
		//计算转换pts dts
		AVRational itime = ictx->streams[pkt.stream_index]->time_base;
		AVRational otime = octx->streams[pkt.stream_index]->time_base;
		pkt.pts = av_rescale_q_rnd(pkt.pts, itime, otime, (AVRounding)(AV_ROUND_NEAR_INF | AV_ROUND_NEAR_INF));
		pkt.dts = av_rescale_q_rnd(pkt.pts, itime, otime, (AVRounding)(AV_ROUND_NEAR_INF | AV_ROUND_NEAR_INF));
		pkt.duration = av_rescale_q_rnd(pkt.duration, itime, otime, (AVRounding)(AV_ROUND_NEAR_INF | AV_ROUND_NEAR_INF));
		pkt.pos = -1;

		//视频帧推送速度
		if (ictx->streams[pkt.stream_index]->codecpar->codec_type == AVMEDIA_TYPE_VIDEO)
		{
			AVRational tb = ictx->streams[pkt.stream_index]->time_base;
			//已经过去的时间
			long long now = av_gettime() - startTime;
			long long dts = 0;
			dts = pkt.dts * (1000 * 1000 * r2d(tb));
			if (dts > now)
				av_usleep(dts - now);
		}

		re = av_interleaved_write_frame(octx, &pkt);
		if (re < 0)
		{
			return XError(re);
		}
	}

	cout << "file to rtmp test" << endl;
	getchar();
	return 0;
}