C++ 使用Intel Media SDK 实现H.264硬编码

---

前言

曾经做视频会议相关项目的时候，需要在客户端实现多路视频推流，对若干摄像头以及桌面的数据进行编码推流，当每路数据都是1080p 30fps时，采用软编对cpu的压力是比较大的而且在1080p的基础上达到30fps的帧率也是比较困难的。于是开始尝试使用硬编，发现Intel的硬编性能是非常不错的，对于多路1080p 30fps的编码实时流在相对较好的cpu中是很容易达到的。本文将说明Intel Media SDK的获取以及使用方法。

---

一、下载Media SDK

Media SDK是Intel官方提供的视频编解码库，我们可以在GitHub中下载源码：https://github.com/Intel-Media-SDK/MediaSDK
下载后在Windows上直接通过vs打开sln编译项目。

1.编译项目

在Windows上使用sln编译即可。

2.引用项目

按照如下步骤打开文件夹，即可找到项目。

二、官方示例

对于h264的编码，我们只需要找最简单的官方示例即可：

---

三、封装对象

为什么要将硬编的功能封装成一个对象：主要是为了简化调用，统一接口可替换编码器。而且其编码实现流程有多种，有同步和异步，一些参数可以简化或默认。
对象的接口设计如下（示例，未展示所有结构的定义）：

/// <summary>
/// Intel硬编码对象
/// </summary>
class IntelHardwareEncoder {
public:
	/// <summary>
	/// 编码数据到达事件参数
	/// </summary>
	class EncodedDataArrivedEventArgs
	{
	public:
		/// <summary>
		/// 数据
		/// </summary>
		unsigned char* Data;
		/// <summary>
		/// 数据长度
		/// </summary>
		int DataLength;
		/// <summary>
		/// 帧类型
		/// </summary>
		FrameType FrameType;
	};
	/// <summary>
	/// 编码数据到达事件
	/// </summary>
	std::function<void(void*,EncodedDataArrivedEventArgs*)> EncodedDataArrived;
	/// <summary>
	/// 构造方法
	/// </summary>
	/// <param name="">编码参数</param>
	IntelHardwareEncoder(EncodeParam&);
	/// <summary>
    /// 析构方法
    /// </summary>
    ~IntelHardwareEncoder();
	/// <summary>
	/// 写入数据
	/// </summary>
	/// <param name="">数据缓存</param>
	void Write(Buffer);
	/// <summary>
	/// 获取nv12缓存
	/// </summary>
	/// <returns>nv12缓存</returns>
	Buffer GetNV12Buffer();
	/// <summary>
	/// 获取编码参数
	/// </summary>
	/// <returns>编码参数</returns>
	const EncodeParam& GetParam();
};

---

四、完整代码

https://download.csdn.net/download/u013113678/77479485

五、使用示例

1.将nv12文件编码成h264文件

#include "IntelHardwareEncoder.h"
#include<stdio.h>
#include<exception>
int main(int argc, char** argv) {
	auto fin = fopen("screen.nv12", "rb+");
	auto fout = fopen("screen.h264", "wb+");
    int nFrame = 0;
	try {
		//设置参数
		AC::EncodeParam param;
		param.Bitrate = 1024;
		param.Framerate = 25;
		param.Width = 1920;
		param.Height = 1080;
		//初始化硬编对象
		AC::IntelHardwareEncoder he(param);
		//注册编码输出事件
		he.EncodedDataArrived = [&](auto s, auto e) {
			//将h264写入文件
			fwrite(e->Data, 1, e->DataLength, fout);
			++nFrame;
			printf("Frame number: %d\r", nFrame);
			fflush(stdout);
		};
		//开始编码
		while (1)
		{
			//获取缓存
			auto buf = he.GetNV12Buffer();
			//读取nv12文件数据到缓存，如果GetPitch与图像宽相等，也可以一整片数据读取。
			for (int i = 0; i < 1080; i++)
				//读取y数据
			{
				if (1920 != fread(buf.GetY() + i * buf.GetPitch(), 1, 1920, fin))
				{
					return 0;
				}
			}
			for (int i = 0; i < 540; i++)
				//读取uv数据，uv是交错排列的
			{
				if (1920 != fread(buf.GetUV() + i * buf.GetPitch(), 1, 1920, fin))
				{
					return 0;
				}
			}
			//缓存写入编码器
			he.Write(buf);
		}
	}
	catch (const std::exception& e) {
		printf(e.what());
	}
	catch (...) {
		printf("发生未知异常\n");
	}
	fclose(fin);
	fclose(fout);
	return 0;
}

2.实时流编码

对于实时流需要设置成baseline和best_speed，其他参考上述示例即可。

//设置参数
AC::EncodeParam param;
param.Bitrate = 1024;
param.Framerate = 25;
param.Width = 1920;
param.Height = 1080;
param.Profile = AC::PROFILE_AVC_BASELINE;
param.Usage = AC::TARGETUSAGE_BEST_SPEED;
//初始化硬编对象
AC::IntelHardwareEncoder he(param);

强制产生关键帧

//获取缓存
auto buf = he.GetNV12Buffer();
//写入数据 --略
//强制产生关键帧
buf.IsForceIdr = true;
//缓存写入编码器
he.Write(buf);

总结

总得来说intel的硬编h264的性能还是比较不错的，且官方样例实现也不算复杂，很容易就能看懂整个流程，当然考虑到在项目中的可移植性和可替换性还是需要做一定的封装，封装后的使用流程会简化很多，采用事件回调的方式获取编码数据，不用关心内部是同步还是异步实现。