海思的SDK里其实有H264编码的sample，但因为要匹配很多东西，代码有点复杂，让初学都感到有点混乱。我根据sample自己修改了一下代码，从最简单的情况(确定摄像头类型，只选一种尺寸的图片，只用一个通道)来说明海思HI3518是怎么编码为H264的。
先把源代码下载下来，再分析程序。
下载链接：https://download.csdn.net/download/zhanshenrui/10324766
首先从main函数开始。
//./myvenc
int main(int argc, char *argv[])//main()
{
HI_S32 s32Ret;
signal(SIGINT, SAMPLE_VENC_HandleSig);//ctrl+c,delete
signal(SIGTERM, SAMPLE_VENC_HandleSig);//shell命令kill缺省产生这个信号.
s32Ret = H264_Venc();
if(s32Ret==HI_SUCCESS)
printf("normally\n");
else
printf("unnormally\n");
return -1;
}
main函数里先是用signal定义了两个信号来中断程序的运行，在板子上输入./myvenc 执行程序，按ctrl+c来中断程序，然后再往下就是H264编码函数H264_Venc()。
在H264_Venc()函数里先去初始化MPI系统，然后根据摄像头设置参数。我这里是AR030摄像头，其拍摄图片大小为720P,编码为H264,为了简单只输出一路。在初始化MPI系统时要计算视频缓存池VB_CONF_S的大小，然后填充VB_CONF_S结构体。填充完VB_CONF_S结构体后再用HI_MPI_VB_SetConf设置，最后初始化视频视频缓存池HI_MPI_VB_Init。这里注意的是基本上每个函数调用后都会判断返回值并作出错出理，后面很多函数也是这样处理的，这样利于程序排错。
//为了简单点，编码类型就选择PT_H264，图片大小就选择PIC_HD720，通道也只用1路s32ChnNum=1
PAYLOAD_TYPE_E enPayLoad=PT_H264;//264编码
PIC_SIZE_E enSize=PIC_HD720;//摄像头拍摄图片的大小，这里只用720P
HI_S32 s32ChnNum=1;//支持一路摄像
HI_S32 s32Ret=HI_FAILURE;
/******************************************
mpp system init.
******************************************/
HI_MPI_SYS_Exit();
HI_MPI_VB_Exit();
VB_CONF_S stVbConf;//缓存池参数结构体
HI_U32 u32BlkSize;//一张图片占多少字节
memset(&stVbConf,0,sizeof(VB_CONF_S));
//根据制式，图片大小，图片格式及对齐方式确定图片缓存大小
//这里用NTSC,720P,YUV420,64字节对齐
u32BlkSize=SAMPLE_COMM_SYS_CalcPicVbBlkSize(gs_enNorm,enSize, PIXEL_FORMAT_YUV_SEMIPLANAR_420, SAMPLE_SYS_ALIGN_WIDTH);
printf("u32BlkSize=%d\n",u32BlkSize);
stVbConf.u32MaxPoolCnt = 128;//用默认值，3518默认是128
stVbConf.astCommPool[0].u32BlkSize = u32BlkSize;
stVbConf.astCommPool[0].u32BlkCnt = g_u32BlkCnt;
s32Ret = HI_MPI_VB_SetConf(&stVbConf);//设置 MPP 视频缓存池属性
if (HI_SUCCESS != s32Ret)
{
SAMPLE_PRT("HI_MPI_VB_SetConf failed!\n");
return HI_FAILURE;
}
s32Ret = HI_MPI_VB_Init();//初始化 MPP 视频缓存池
if (HI_SUCCESS != s32Ret)
{
SAMPLE_PRT("HI_MPI_VB_Init failed!\n");
return HI_FAILURE;
}
视频缓存池初始化后就是初始化MPI系统。海思模块化了各个功能，我们想实现哪一种产品就按这种产品的框架来实现各个模块功能，而实现模块化功能也很简单，基本套路就是定义一个结构体，给结构体赋值，然后设置就可以了，像前面缓存池模块就是先定一个VB_CONF_S结构体，然后对结体体赋值，然后调用函数设置进去，最后初始化。同理，初始化MPI系统也要先定义一个结构体再赋值然后设置最后初始化，这种思路在linux里用得很多。
//定义结构体，对结构体赋值，然后设置，最后初始化
MPP_SYS_CONF_S stSysConf = {0};
stSysConf.u32AlignWidth = SAMPLE_SYS_ALIGN_WIDTH;
s32Ret = HI_MPI_SYS_SetConf(&stSysConf);//配置系统控制参数
if (HI_SUCCESS != s32Ret)
{
SAMPLE_PRT("HI_MPI_SYS_SetConf failed\n");
return HI_FAILURE;
}
s32Ret = HI_MPI_SYS_Init();//初始化 MPP 系统
if (HI_SUCCESS != s32Ret)
{
SAMPLE_PRT("HI_MPI_SYS_Init failed!\n");
return HI_FAILURE;
}
系统初始化完后还要打开硬件，摄像头用的是MIPI，应用层只需要用open打开MIPI,用ioctl设置
HI_S32 fdmipi;
combo_dev_attr_t *pstcomboDevAttr = NULL;
fdmipi = open("/dev/hi_mipi", O_RDWR);
if (fdmipi < 0)
{
printf("warning: open hi_mipi dev failed\n");
return -1;
}
pstcomboDevAttr = &MIPI_CMOS3V3_ATTR;//AR0130模组属性
if (ioctl(fdmipi, HI_MIPI_SET_DEV_ATTR, pstcomboDevAttr))
{
printf("set mipi attr failed\n");
close(fdmipi);
return HI_FAILURE;
}
close(fdmipi);
根据海思SDK里的文档介绍，H264编码过程应该是：系统初始化--VI--VPSS-VENC.系统初始化前面已完成，现在要做的就是实现输入模块VI.VI模块有通道，所以还要设置通道参数。具体解释和代码如下,不再做过多解释：
HI_S32 i,ViChn=0;//只有一路输出，所以ViChn=0
//设置输入配置参数
SAMPLE_VI_CONFIG_S stViConfig = {0};
stViConfig.enViMode = APTINA_AR0130_DC_720P_30FPS;//摄像头是AR0130模组
stViConfig.enRotate = ROTATE_NONE;//不翻转
stViConfig.enNorm = VIDEO_ENCODING_MODE_AUTO;//编码模式自动
stViConfig.enViChnSet = VI_CHN_SET_NORMAL;//普通
stViConfig.enWDRMode = WDR_MODE_NONE;//不设置宽动态
VI_DEV_ATTR_S stViDevAttr;
VI_DEV ViDev=0;//只有一个摄像头设备，所以设备序号为0
ISP_DEV s32IspDev=0;//同样，ISP序号也为0
memset(&stViDevAttr,0,sizeof(stViDevAttr));
memcpy(&stViDevAttr,&DEV_ATTR_9M034_DC_720P_BASE,sizeof(stViDevAttr));
/******************************************
step 1: mipi configure
******************************************/
/*
s32Ret = SAMPLE_COMM_VI_StartMIPI(&stViConfig);
if (HI_SUCCESS != s32Ret)
{
SAMPLE_PRT("%s: MIPI init failed!\n", __FUNCTION__);
return HI_FAILURE;
}
*/
/******************************************
step 2: configure sensor and ISP (include WDR mode).
note: you can jump over this step, if you do not use Hi3516A interal isp.
//虽然说了不用，但还要需要，因为后面HI_MPI_ISP_GetWDRMode函数会调用ISP_CHECK_MEM_INIT
//检测内存，所以还是在这里使用
******************************************/
s32Ret = SAMPLE_COMM_ISP_Init(stViConfig.enWDRMode);
if (HI_SUCCESS != s32Ret)
{
SAMPLE_PRT("%s: Sensor init failed!\n", __FUNCTION__);
return HI_FAILURE;
}
/******************************************
step 3: run isp thread
note: you can jump over this step, if you do not use Hi3516A interal isp.
******************************************/
s32Ret = SAMPLE_COMM_ISP_Run();
if (HI_SUCCESS != s32Ret)
{
SAMPLE_PRT("%s: ISP init failed!\n", __FUNCTION__);
/* disable videv */
return HI_FAILURE;
}
s32Ret=HI_MPI_VI_SetDevAttr(ViDev, &stViDevAttr);//只有一个设备，所以设备号为0
if (s32Ret != HI_SUCCESS)
{
SAMPLE_PRT("HI_MPI_VI_SetDevAttr failed with %#x!\n", s32Ret);
return HI_FAILURE;
}
ISP_WDR_MODE_S stWdrMode;
s32Ret = HI_MPI_ISP_GetWDRMode(s32IspDev, &stWdrMode);
if (s32Ret != HI_SUCCESS)
{
SAMPLE_PRT("HI_MPI_ISP_GetWDRMode failed with %#x!\n", s32Ret);
return HI_FAILURE;
}
VI_WDR_ATTR_S stWdrAttr;
stWdrAttr.enWDRMode = stWdrMode.enWDRMode;
stWdrAttr.bCompress = HI_FALSE;
s32Ret = HI_MPI_VI_SetWDRAttr(ViDev, &stWdrAttr);
if (s32Ret)
{
SAMPLE_PRT("HI_MPI_VI_SetWDRAttr failed with %#x!\n", s32Ret);
return HI_FAILURE;
}
s32Ret = HI_MPI_VI_EnableDev(ViDev);
if (s32Ret != HI_SUCCESS)
{
SAMPLE_PRT("HI_MPI_VI_EnableDev failed with %#x!\n", s32Ret);
return HI_FAILURE;
}
RECT_S stCapRect;
SIZE_S stTargetSize;
stCapRect.s32X = 0;
stCapRect.s32Y = 0;
stCapRect.u32Width = 1280;
stCapRect.u32Height = 720;//AR030输出1280*720图像
stTargetSize.u32Width = stCapRect.u32Width;
stTargetSize.u32Height = stCapRect.u32Height;
//设置通道属性
VI_CHN_ATTR_S stChnAttr;
memcpy(&stChnAttr.stCapRect, &stCapRect, sizeof(RECT_S));
stChnAttr.enCapSel = VI_CAPSEL_BOTH;
stChnAttr.stDestSize.u32Width = stTargetSize.u32Width;
stChnAttr.stDestSize.u32Height = stTargetSize.u32Height;
stChnAttr.enPixFormat = PIXEL_FORMAT_YUV_SEMIPLANAR_420;
stChnAttr.bMirror = HI_FALSE;
stChnAttr.bFlip = HI_FALSE;
stChnAttr.s32SrcFrameRate = -1;
stChnAttr.s32DstFrameRate = -1;
stChnAttr.enCompressMode = COMPRESS_MODE_NONE;
s32Ret = HI_MPI_VI_SetChnAttr(ViChn, &stChnAttr);//设置通道属性
if (s32Ret != HI_SUCCESS)
{
SAMPLE_PRT("in HI_MPI_VI_SetChnAttr failed with %#x!\n", s32Ret);
SAMPLE_COMM_ISP_Stop();
return HI_FAILURE;
}
s32Ret = HI_MPI_VI_EnableChn(ViChn);//使能通道
if (s32Ret != HI_SUCCESS)
{
SAMPLE_PRT("HI_MPI_VI_EnableChn failed with %#x!\n", s32Ret);
SAMPLE_COMM_ISP_Stop();
return HI_FAILURE;
}
VI模块设置完以后就要设置VPSS模块，VPSS模块有组，通道概念，所以设置VPSS模块也要设置VPSS组、VPSS通道。
设置VPSS组过程如下：定义结构体，给结构体赋值，设置(创建的同时就设置了)，启动
SIZE_S stSize;
stSize.u32Width = 1280;
stSize.u32Height = 720;
VPSS_GRP_ATTR_S stVpssGrpAttr;
VPSS_NR_PARAM_U unNrParam = {{0}};
VPSS_GRP VpssGrp=0;//只有一个设备，序号为0
stVpssGrpAttr.u32MaxW = stSize.u32Width;
stVpssGrpAttr.u32MaxH = stSize.u32Height;
stVpssGrpAttr.bIeEn = HI_FALSE;
stVpssGrpAttr.bNrEn = HI_TRUE;
stVpssGrpAttr.bHistEn = HI_FALSE;
stVpssGrpAttr.bDciEn = HI_FALSE;
stVpssGrpAttr.enDieMode = VPSS_DIE_MODE_NODIE;
stVpssGrpAttr.enPixFmt = PIXEL_FORMAT_YUV_SEMIPLANAR_420;
s32Ret = HI_MPI_VPSS_CreateGrp(VpssGrp, &stVpssGrpAttr);//创建组的同时将组的属性设置进去
if (s32Ret != HI_SUCCESS)
{
SAMPLE_PRT("HI_MPI_VPSS_CreateGrp failed with %#x!\n", s32Ret);
return HI_FAILURE;
}
s32Ret = HI_MPI_VPSS_GetNRParam(VpssGrp, &unNrParam);
if (s32Ret != HI_SUCCESS)
{
SAMPLE_PRT("HI_MPI_VPSS_GetNRParam failed with %#x!\n", s32Ret);
return HI_FAILURE;
}
s32Ret = HI_MPI_VPSS_SetNRParam(VpssGrp, &unNrParam);
if (s32Ret != HI_SUCCESS)
{
SAMPLE_PRT("HI_MPI_VPSS_SetNRParam failed with %#x!\n", s32Ret);
return HI_FAILURE;
}
s32Ret = HI_MPI_VPSS_StartGrp(VpssGrp);//启动VPSS组
if (s32Ret != HI_SUCCESS)
{
SAMPLE_PRT("HI_MPI_VPSS_StartGrp failed with %#x\n", s32Ret);
return HI_FAILURE;
}
VPSS还要与VI绑定，以便VI的数据直接输入VPSS，不需要我们手动从VI复制数据到VPSS,绑定时将绑定源设置为VI的哪
一个通道，目的源设置为VPSS的一个通道
SAMPLE_VI_PARAM_S stViParam;
stViParam.s32ViDevCnt = 1;
stViParam.s32ViDevInterval = 1;
stViParam.s32ViChnCnt = 1;
stViParam.s32ViChnInterval = 1;
MPP_CHN_S stSrcChn;
MPP_CHN_S stDestChn;
stSrcChn.enModId = HI_ID_VIU;
stSrcChn.s32DevId = 0;
stSrcChn.s32ChnId = ViChn;
stDestChn.enModId = HI_ID_VPSS;
stDestChn.s32DevId = VpssGrp;
stDestChn.s32ChnId = 0;
s32Ret = HI_MPI_SYS_Bind(&stSrcChn, &stDestChn);//将VI的0通道与VPSS的组的0通道绑定起来
if (s32Ret != HI_SUCCESS)
{
SAMPLE_PRT("failed with %#x!\n", s32Ret);
return HI_FAILURE;
}
前面设置了VPSS组，将VI通道与VPSS组里的0通道绑定起来，但还未设置0通道，下面设置0通道.还是老套路，定义结构体，给结构体赋值，将结构体设置进去，开启通道。
//设置VPSS通道，只有一个通道，所以VpssChn=0
VPSS_CHN VpssChn=0;
VPSS_CHN_ATTR_S stVpssChnAttr;
VPSS_CHN_MODE_S stVpssChnMode;
stVpssChnMode.enChnMode = VPSS_CHN_MODE_USER;
stVpssChnMode.bDouble = HI_FALSE;
stVpssChnMode.enPixelFormat = PIXEL_FORMAT_YUV_SEMIPLANAR_420;
stVpssChnMode.u32Width = stSize.u32Width;
stVpssChnMode.u32Height = stSize.u32Height;
stVpssChnMode.enCompressMode = COMPRESS_MODE_SEG;
memset(&stVpssChnAttr, 0, sizeof(stVpssChnAttr));
stVpssChnAttr.s32SrcFrameRate = -1;
stVpssChnAttr.s32DstFrameRate = -1;
s32Ret = HI_MPI_VPSS_SetChnAttr(VpssGrp, VpssChn, &stVpssChnAttr);
if (s32Ret != HI_SUCCESS)
{
SAMPLE_PRT("HI_MPI_VPSS_SetChnAttr failed with %#x\n", s32Ret);
return HI_FAILURE;
}
s32Ret = HI_MPI_VPSS_SetChnMode(VpssGrp, VpssChn, &stVpssChnMode);
if (s32Ret != HI_SUCCESS)
{
SAMPLE_PRT("%s failed with %#x\n", __FUNCTION__, s32Ret);
return HI_FAILURE;
}
s32Ret = HI_MPI_VPSS_EnableChn(VpssGrp, VpssChn);
if (s32Ret != HI_SUCCESS)
{
SAMPLE_PRT("HI_MPI_VPSS_EnableChn failed with %#x\n", s32Ret);
return HI_FAILURE;
}
上面将VPSS设置完成，按照框架，下一个应该设置VENC模块，跟VPSS一样，老套路设置，然后绑定。
//定义结构体，给结构体赋值，创建通道同时将结构体设置进去
SAMPLE_RC_E enRcMode= SAMPLE_RC_CBR;
VENC_CHN VencChn=0;
VENC_CHN_ATTR_S stVencChnAttr;
VENC_ATTR_H264_S stH264Attr;
VENC_ATTR_H264_CBR_S stH264Cbr;
stVencChnAttr.stVeAttr.enType =enPayLoad;
stH264Attr.u32MaxPicWidth =stSize.u32Width;
stH264Attr.u32MaxPicHeight = stSize.u32Height;
stH264Attr.u32PicWidth = stSize.u32Width;/*the picture width*/
stH264Attr.u32PicHeight = stSize.u32Height;/*the picture height*/
stH264Attr.u32BufSize = stSize.u32Width * stSize.u32Height;/*stream buffer size*/
stH264Attr.u32Profile = 0;/*0: baseline; 1:MP; 2:HP; 3:svc_t */
stH264Attr.bByFrame = HI_TRUE;/*get stream mode is slice mode or frame mode?*/
stH264Attr.u32BFrameNum = 0;/* 0: not support B frame; >=1: number of B frames */
stH264Attr.u32RefNum = 1;/* 0: default; number of refrence frame*/
memcpy(&stVencChnAttr.stVeAttr.stAttrH264e, &stH264Attr, sizeof(VENC_ATTR_H264_S));
stVencChnAttr.stRcAttr.enRcMode = VENC_RC_MODE_H264CBR;
stH264Cbr.u32Gop = (VIDEO_ENCODING_MODE_PAL== gs_enNorm)?25:30;
stH264Cbr.u32StatTime = 1; /* stream rate statics time(s) */
stH264Cbr.u32SrcFrmRate = (VIDEO_ENCODING_MODE_PAL== gs_enNorm)?25:30;/* input (vi) frame rate */
stH264Cbr.fr32DstFrmRate = (VIDEO_ENCODING_MODE_PAL== gs_enNorm)?25:30;/* target frame rate */
stH264Cbr.u32BitRate = 1024*2;
stH264Cbr.u32FluctuateLevel = 0; /* average bit rate */
memcpy(&stVencChnAttr.stRcAttr.stAttrH264Cbr, &stH264Cbr, sizeof(VENC_ATTR_H264_CBR_S));
s32Ret = HI_MPI_VENC_CreateChn(VencChn, &stVencChnAttr);//创建通道同时将结构体设置进去
if (HI_SUCCESS != s32Ret)
{
SAMPLE_PRT("HI_MPI_VENC_CreateChn [%d] faild with %#x!\n",VencChn, s32Ret);
return s32Ret;
}
设置完以后，开启接收图片并将VPSS通道与VENC通道绑定起来。VPSS通道作为源通道，VENC通道作为目的通道
//将VPSS通道作为源通道与VENC通道作为目的通道绑定起来
stSrcChn.enModId = HI_ID_VPSS;
stSrcChn.s32DevId = VpssGrp;
stSrcChn.s32ChnId = VpssChn;
stDestChn.enModId = HI_ID_VENC;
stDestChn.s32DevId = 0;
stDestChn.s32ChnId = VencChn;
s32Ret=HI_MPI_SYS_Bind(&stSrcChn, &stDestChn);
if (s32Ret != HI_SUCCESS)
{
SAMPLE_PRT("failed with %#x!\n", s32Ret);
return HI_FAILURE;
}
这样，整个系统我们就设置好了，下一步要操作地是开始接收H264码流并将码流保存起来。这里用到多线程，用pthread_create创建个线程函数，在线程函数里用select多路IO复用来获取码流并保存。
gs_stPara.bThreadStart = HI_TRUE;
gs_stPara.s32Cnt = s32ChnNum;
pthread_create(&gs_VencPid, 0, VENC_GetVencStreamProc, (HI_VOID*)&gs_stPara);
printf("please press twice ENTER to exit this sample\n");
gs_stPara是自己定义的一个结构体，里面的成员变量bThreadStart为真时表示不断读取保存H264码流，当接连有两次按下键盘时或有ctrl+c按下时，这个成员变量会为假，从而会停止VENC_GetVencStreamProc线程函数。VENC_GetVencStreamProc线程函数可以说是获取H264码流的最关键的函数，我们稍会再说，先按正常流程说下后面代码的作用。后面的代码如下：
getchar();
getchar();
if (HI_TRUE == gs_stPara.bThreadStart)
{
gs_stPara.bThreadStart = HI_FALSE;
pthread_join(gs_VencPid, 0);
}
getchar（）函数的执行模式是阻塞式的，当需要接收字符流的时候，当前线程(main函数所在线程)就会被挂起，其后的所有代码均要等待用户输入回车表示输入完毕后，线程才会被调度执行余下的代码，即进入if让gs_stPara.bThreadStart 为假，同时调用pthread_join(gs_VencPid, 0)函数，main函数又会进入阻塞模式，等待线程函数VENC_GetVencStreamProc结束。线程函数VENC_GetVencStreamProc结束后，CPU又进入main函数执行pthread_join(gs_VencPid, 0)后的代码，可以猜测后面的代码就是收尾工作，应该与前面的步骤动作相反，前面绑定通道，后面就要解绑，前面创建开启通道，后面就要停止销毁通道，前面初始化， 后面就要去初始化，前面按顺序设置MPI_SYS,VI,VPSS,VENC,后面就要反过来按顺序设置VENC,VPSS,VI,MPI_SYS,看代码的确如此：
stSrcChn.enModId = HI_ID_VPSS;
stSrcChn.s32DevId = VpssGrp;
stSrcChn.s32ChnId = VpssChn;
stDestChn.enModId = HI_ID_VENC;
stDestChn.s32DevId = 0;
stDestChn.s32ChnId = VencChn;
s32Ret = HI_MPI_SYS_UnBind(&stSrcChn, &stDestChn);//先解绑VPSS和VENC的绑定，后面再解绑VPSS和VI的绑定
if (s32Ret != HI_SUCCESS)
{
SAMPLE_PRT("failed with %#x!\n", s32Ret);
return HI_FAILURE;
}
s32Ret = HI_MPI_VENC_StopRecvPic(VencChn);//前面开启了接收图片，这里就要停止
if (HI_SUCCESS != s32Ret)
{
SAMPLE_PRT("HI_MPI_VENC_StopRecvPic vechn[%d] failed with %#x!\n",VencChn, s32Ret);
return HI_FAILURE;
}
/******************************************
Distroy Venc Channel
******************************************/
s32Ret = HI_MPI_VENC_DestroyChn(VencChn);
//前面创建了VENC通道，这里就要销毁VENC通道，而且最先销毁，后面再销毁VI通道
if (HI_SUCCESS != s32Ret)
{
SAMPLE_PRT("HI_MPI_VENC_DestroyChn vechn[%d] failed with %#x!\n",VencChn, s32Ret);
return HI_FAILURE;
}
stSrcChn.enModId = HI_ID_VIU;
stSrcChn.s32DevId = ViDev;
stSrcChn.s32ChnId = ViChn;
stDestChn.enModId = HI_ID_VPSS;
stDestChn.s32DevId = VpssGrp;
stDestChn.s32ChnId = 0;
s32Ret = HI_MPI_SYS_UnBind(&stSrcChn, &stDestChn);
//后面再解绑VPSS和VI的绑定,在解绑VPSS和VENC的绑定之后，顺序与绑定顺序相反
if (s32Ret != HI_SUCCESS)
{
SAMPLE_PRT("failed with %#x!\n", s32Ret);
return HI_FAILURE;
}
s32Ret = HI_MPI_VPSS_DisableChn(VpssGrp, VpssChn);//失能VPSS组和VPSS通道
if (s32Ret != HI_SUCCESS)
{
SAMPLE_PRT("%s failed with %#x\n", __FUNCTION__, s32Ret);
return HI_FAILURE;
}
s32Ret = HI_MPI_VPSS_StopGrp(VpssGrp);//停止VPSS组
if (s32Ret != HI_SUCCESS)
{
SAMPLE_PRT("%s failed with %#x\n", __FUNCTION__, s32Ret);
return HI_FAILURE;
}
s32Ret = HI_MPI_VPSS_DestroyGrp(VpssGrp);//销毁组
if (s32Ret != HI_SUCCESS)
{
SAMPLE_PRT("%s failed with %#x\n", __FUNCTION__, s32Ret);
return HI_FAILURE;
}
s32Ret = HI_MPI_VI_DisableChn(ViChn);//失能通道
if (HI_SUCCESS != s32Ret)
{
SAMPLE_PRT("HI_MPI_VI_DisableChn failed with %#x\n",s32Ret);
return HI_FAILURE;
}
s32Ret = HI_MPI_VI_DisableDev(ViDev);//失能VI设备即摄像头
if (HI_SUCCESS != s32Ret)
{
SAMPLE_PRT("HI_MPI_VI_DisableDev failed with %#x\n", s32Ret);
return HI_FAILURE;
}
SAMPLE_COMM_ISP_Stop();//停止ISP
HI_MPI_SYS_Exit();//去初始化
HI_MPI_VB_Exit();
//由以上几个步骤可以看出，设置与销毁顺序相反，而且是成对出现
return HI_SUCCESS;
至此，H264_Venc函数分析完，现在我们来重点分析线程函数VENC_GetVencStreamProc是怎么获取码流并保存到一个文件里的。
HI_VOID* VENC_GetVencStreamProc(HI_VOID *p)
{
SAMPLE_VENC_GETSTREAM_PARA_S *pstPara;
HI_S32 s32ChnTotal;
VENC_CHN VencChn;
HI_S32 s32Ret;
PAYLOAD_TYPE_E enPayLoadType[VENC_MAX_CHN_NUM];
HI_CHAR aszFileName[VENC_MAX_CHN_NUM][64];
VENC_CHN_ATTR_S stVencChnAttr;
struct timeval TimeoutVal;
fd_set read_fds;
HI_S32 VencFd[VENC_MAX_CHN_NUM],maxfd;
VENC_STREAM_S stStream;
VENC_CHN_STAT_S stStat;
FILE *pFile[VENC_MAX_CHN_NUM];
HI_S32 i;
pstPara = (SAMPLE_VENC_GETSTREAM_PARA_S*)p;
s32ChnTotal = pstPara->s32Cnt;//pstPara->s32Cnt是由参数传进来的，为1
for (i = 0; i < s32ChnTotal; i++)
{
VencChn = i;
s32Ret = HI_MPI_VENC_GetChnAttr(VencChn, &stVencChnAttr);
//这里是为了取得是什么编码类型，以便确定保存文件的后缀名
//比如这里是H264编码，所以保存文件的后缀后就是.h264
if(s32Ret != HI_SUCCESS)
{
SAMPLE_PRT("HI_MPI_VENC_GetChnAttr chn[%d] failed with %#x!\n", \
VencChn, s32Ret);
return NULL;
}
enPayLoadType[i] = stVencChnAttr.stVeAttr.enType;
gettime(filename);//获取时间如20171101085639为文件名
sprintf(aszFileName[i], "%s_%d%s",filename, i, ".h264");//20171101085639.h264
pFile[i] = fopen(aszFileName[i], "wb");
if (!pFile[i])
{
SAMPLE_PRT("open file[%s] failed!\n", aszFileName[i]);
return NULL;
}
VencFd[i] = HI_MPI_VENC_GetFd(i);//获取文件句柄，以便后面能用select来IO复用
if (VencFd[i] < 0)
{
SAMPLE_PRT("HI_MPI_VENC_GetFd failed with %#x!\n", VencFd[i]);
return NULL;
}
if (maxfd <= VencFd[i])
{
maxfd = VencFd[i];
}
}
while (HI_TRUE == pstPara->bThreadStart)
//当main函数所在的线程接收到两个键盘字符或ctrl+c时，pstPara->bThreadStart会为假，跳出while循环
//然后往下执行关闭前面打开的文件，执行完这个VENC_GetVencStreamProc线程函数，线程结束
{
/*IO复用4步骤
1.清空文件集合FD_ZERO(&read_fds);
2.将文件加入文件集合FD_SET(VencFd[i], &read_fds);
3.设置超时时间并用select(maxfd + 1, &read_fds, NULL, NULL, &TimeoutVal)来等待文件状态有变化唤醒线程
或超时唤醒
4.FD_ISSET查询文件状态是否有变化，有变化则处理
*/
FD_ZERO(&read_fds);
for (i = 0; i < s32ChnTotal; i++)
{
FD_SET(VencFd[i], &read_fds);
}
TimeoutVal.tv_sec = 2;
TimeoutVal.tv_usec = 0;
s32Ret = select(maxfd + 1, &read_fds, NULL, NULL, &TimeoutVal);
if (s32Ret < 0)
{
SAMPLE_PRT("select failed!\n");
break;
}
else if (s32Ret == 0)
{
SAMPLE_PRT("get venc stream time out, exit thread\n");
continue;
}
else
{
for (i = 0; i < s32ChnTotal; i++)
{
if (FD_ISSET(VencFd[i], &read_fds))
{
memset(&stStream, 0, sizeof(stStream));
s32Ret = HI_MPI_VENC_Query(i, &stStat);//查询是否有码流，并将码流信息填充到stStat结构体中
if (HI_SUCCESS != s32Ret)
{
SAMPLE_PRT("HI_MPI_VENC_Query chn[%d] failed with %#x!\n", i, s32Ret);
break;
}
if(0 == stStat.u32CurPacks)
{
SAMPLE_PRT("NOTE: Current frame is NULL!\n");
continue;
}
stStream.pstPack = (VENC_PACK_S*)malloc(sizeof(VENC_PACK_S) * stStat.u32CurPacks);
//分配内存以便保存码流包数据
if (NULL == stStream.pstPack)
{
SAMPLE_PRT("malloc stream pack failed!\n");
break;
}
stStream.u32PackCount = stStat.u32CurPacks;
//printf("stStream.u32PackCount=%d\n",stStream.u32PackCount);
s32Ret = HI_MPI_VENC_GetStream(i, &stStream, HI_TRUE);//获取码流数据并保存到stStream结构体中
if (HI_SUCCESS != s32Ret)
{
free(stStream.pstPack);//获取失败则要释放前面分配的内存，否则会造成内存溢出
stStream.pstPack = NULL;
SAMPLE_PRT("HI_MPI_VENC_GetStream failed with %#x!\n", s32Ret);
break;
}
HI_S32 u32PackIndex;
for (u32PackIndex= 0;u32PackIndex < stStream.u32PackCount; u32PackIndex++)
{
fwrite( stStream.pstPack[u32PackIndex].pu8Addr+stStream.pstPack[u32PackIndex].u32Offset,\
stStream.pstPack[u32PackIndex].u32Len- stStream.pstPack[u32PackIndex].u32Offset, \
1, pFile[i]);
fflush(pFile[i]);
#if 1
printf("stStream.u32PackCount=%d,stStream.pstPack[%d].pu8Addr=0x%08x,\
stStream.pstPack[%d].u32Offset=%d,stStream.pstPack[%d].u32Len=%d\n",\
stStream.u32PackCount,u32PackIndex,stStream.pstPack[u32PackIndex].pu8Addr,\
u32PackIndex,stStream.pstPack[u32PackIndex].u32Offset,u32PackIndex,stStream.pstPack[u32PackIndex].u32Len);
//添加打印信息，查看保存码流内容的内存是怎么样的
#endif
}
s32Ret = HI_MPI_VENC_ReleaseStream(i, &stStream);//保存后要释放码流
if (HI_SUCCESS != s32Ret)
{
free(stStream.pstPack);//获取失败则要释放前面分配的内存，否则会造成内存溢出
stStream.pstPack = NULL;
break;
}
free(stStream.pstPack);//释放码流后，也要释放分配的内存，避免内存溢出
stStream.pstPack = NULL;
}
}
}
}
for (i = 0; i < s32ChnTotal; i++)
{
fclose(pFile[i]);
}
return NULL;
}
线程函数里先通过HI_MPI_VENC_GetFd取得文件句柄，然后在while循环里不断清空读文件集合FD_ZERO(&read_fds);将文件句柄加入读文件集合FD_SET(VencFd[i], &read_fds);用select来休眠线程，有数据或超时又唤醒线程select(maxfd + 1, &read_fds, NULL, NULL, &TimeoutVal);用FD_ISSET(VencFd[i], &read_fds)来查询是哪个文件句柄有数据以便处理，然后用HI_MPI_VENC_Query查询码流统计信息以便分配内存，HI_MPI_VENC_GetStream来取得码流包数据，fwrite将码流包数据保存到文件中，码流包处理后要HI_MPI_VENC_ReleaseStream释放掉码流，释放掉申请的内存。
综合整个程序，在创建线程函数前都是对系统设置，按照海思SDK里的文档步骤设置就行，基本上就是定义结构体，给结构体赋值，然后设置进系统。到后面创建线程后就有两个线程函数了，一个main函数所在线程，一个VENC_GetVencStreamProc函数所在线程。main函数所在线程创建VENC_GetVencStreamProc线程后就调用getchar();函数进入休眠状态，只有当键盘有输入时getchar()函数有返回值时才会唤醒main函数所在线程，所以CPU就切换到VENC_GetVencStreamProc线程，VENC_GetVencStreamProc线程在调用select函数时，如果读文件集合read_fds有要读的内容则会往下执行，如果没有VENC_GetVencStreamProc线程也会休眠，CPU会切换到别的线程，当read_fds有要读的内容内容时或超时VENC_GetVencStreamProc线程会唤醒，往下执行去进行读码流的处理过程。
程序写完后，在sample文件夹里创建个子文件夹如myvenc,然后将这个文件保存到myvenc里，从sample\venc里复制makefile到sample\myvenc里，这个makefile都不要我们改些什么东西，只要位置放对了位置(跟源代码文件在同一个目录里)就直接可以用，然后make clean,make编译，如果没问题的话会生成myvenc可执行文件，将这个可执行文件拷贝到HI3518开发板(可通过SD卡从PC拷贝到开发板，也可挂载根文件系统直接复制过去，方法多样)，然后执行./myvenc就会运行程序，过一会就会在myvenc同级目录下生成一个以时间为名的，后缀名为h264的文件，这个文件就是H264码流保存的文件，将这文件从Hi3518里复制到PC机上，用VLC播放器就可以播放出摄像头拍摄的内容。
编译运行程序后生成了h264文件，这个文件及程序运行时打印的输出内容有助于我们分析h264,下一篇我们将分析h264以便后期怎么将h264数据打包到rtp中进行rtsp传送数据。
这个程序比较简单，只是为了让大家了解海思芯片是怎么进行多媒体开发，所以很多功能没有写成函数，感觉内容很多，但避免了sample里复杂的函数调用，有助于将了解各个数据结构及海思SDK各个函数的用法。程序从前往后看，流程一目了然，各个数据结构、函数对应哪个模块很清楚，对入门很有好处。我将代码上传到github,欢迎朋友们下载，一起讨论学习。
最后看一下程序运行效果，用VLC打开H264文件，可以看到摄像头拍摄界面：
项目源代码及工具:Hi3518编码H264
---------------------
作者：zhanshenrui
来源：CSDN
原文：https://blog.csdn.net/zhanshenrui/article/details/79082157
版权声明：本文为博主原创文章，转载请附上博文链接！
扫一扫
8160
被折叠的 条评论
为什么被折叠?
到【灌水乐园】发言
