/******************************************************************************
A simple program of Hisilicon Hi35xx video encode implementation.
Copyright (C), 2010-2011, Hisilicon Tech. Co., Ltd.
******************************************************************************
Modification: 2011-2 Created
******************************************************************************/
#ifdef __cplusplus
#if __cplusplus
extern "C" {
#endif
#endif /* End of #ifdef __cplusplus */
#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <unistd.h>
#include <pthread.h>
#include <signal.h>
#include "sample_comm.h"
VIDEO_NORM_E gs_enNorm = VIDEO_ENCODING_MODE_NTSC;
/******************************************************************************
* function : show usage
******************************************************************************/
void SAMPLE_VENC_Usage(char* sPrgNm)
{
#ifndef hi3516ev100
printf("Usage : %s <index> eg: %s 0 \n", sPrgNm, sPrgNm);
printf("index:\n");
printf("\t 0) NormalP:H.265@1080p@30fps+H.264@1080p@30fps\n");
printf("\t 1) SmartP: H.265@1080p@30fps+H.264@1080p@30fps\n");
printf("\t 2) DualP: H.265@1080p@30fps@BigSmallP+H.264@1080p@30fps@BigSmallP\n");
printf("\t 3) DualP2: H.265@1080p@30fps@SmallP+H.264@1080p@30fps@BigSmallP\n");
printf("\t 4) IntraRefresh: H.265@1080p@30fps@IntraRefresh+H.264@1080p@30fps@IntraRefresh\n");
printf("\t 5) Jpege:1*1080p mjpeg encode + 1*1080p jpeg\n");
printf("\t 6) OnlineLowDelay:H.264@1080p@30fps@OnlineLowDelay + H.265@1080p@30fps\n");
printf("\t 7) RoiBgFrameRate:H.264@1080p@30fps@RoiBgFrameRate + H.265@1080p@30fps@RoiBgFrameRate\n");
printf("\t 8) svc-t :H.264@1080p@30fps@svc-t\n");
printf("\t 9) EPTZ:H.264@1080p@30fps + H.265@1080p@30fps@EPTZ\n");
#else
printf("Usage : %s <index> eg: %s 0 \n", sPrgNm, sPrgNm);
printf("index:\n");
printf("\t 0) NormalP:H.265@1080p@30fps\n");
printf("\t 1) SmartP: H.265@1080p@30fps\n");
printf("\t 2) DualP: H.265@1080p@30fps@BigSmallP\n");
printf("\t 3) DualP2: H.265@1080p@30fps@SmallP\n");
printf("\t 4) IntraRefresh: H.265@1080p@30fps@IntraRefresh\n");
printf("\t 5) Jpege:1*1080p mjpeg encode + 1*1080p jpeg\n");
printf("\t 6) OnlineLowDelay:H.264@1080p@30fps@OnlineLowDelay\n");
printf("\t 7) RoiBgFrameRate:H.264@1080p@30fps@RoiBgFrameRate\n");
printf("\t 8) svc-t :H.264@1080p@30fps@svc-t\n");
#endif
return;
}
/******************************************************************************
* function : to process abnormal case
******************************************************************************/
#ifndef __HuaweiLite__
void SAMPLE_VENC_HandleSig(HI_S32 signo)
{
signal(SIGINT, SIG_IGN);
signal(SIGTERM, SIG_IGN);
if (SIGINT == signo || SIGTERM == signo)
{
SAMPLE_COMM_VENC_StopGetStream();
SAMPLE_COMM_VPSS_StopChnCropProc();
SAMPLE_COMM_ISP_Stop();
SAMPLE_COMM_SYS_Exit();
printf("\033[0;31mprogram termination abnormally!\033[0;39m\n");
}
exit(-1);
}
#endif
/******************************************************************************
* function : to process abnormal case - the case of stream venc
******************************************************************************/
void SAMPLE_VENC_StreamHandleSig(HI_S32 signo)
{
if (SIGINT == signo || SIGTERM == signo)
{
SAMPLE_COMM_SYS_Exit();
printf("\033[0;31mprogram exit abnormally!\033[0;39m\n");
}
exit(0);
}
/******************************************************************************
* function : H.264@1080p@30fps+H.264@VGA@30fps
******************************************************************************/
HI_S32 SAMPLE_VENC_NORMALP_CLASSIC(HI_VOID)
{
PAYLOAD_TYPE_E enPayLoad[2]= {PT_H265, PT_H264};
PIC_SIZE_E enSize[2] = {PIC_HD1080, PIC_HD1080};
HI_U32 u32Profile = 0;
VB_CONF_S stVbConf;
SAMPLE_VI_CONFIG_S stViConfig = {0};
VPSS_GRP VpssGrp;
VPSS_CHN VpssChn;
VPSS_GRP_ATTR_S stVpssGrpAttr = {0};
VPSS_CHN_ATTR_S stVpssChnAttr = {0};
VPSS_CHN_MODE_S stVpssChnMode;
VENC_CHN VencChn;
SAMPLE_RC_E enRcMode= SAMPLE_RC_CBR;
#ifndef hi3516ev100
HI_S32 s32ChnNum=2;
#else
HI_S32 s32ChnNum=1;
#endif
HI_S32 s32Ret = HI_SUCCESS;
HI_U32 u32BlkSize;
SIZE_S stSize;
char c;
/******************************************
step 1: init sys variable
******************************************/
memset(&stVbConf,0,sizeof(VB_CONF_S));
SAMPLE_COMM_VI_GetSizeBySensor(&enSize[0]);
stVbConf.u32MaxPoolCnt = 128;
//printf("s32ChnNum:%d,Sensor Size:%d\n",s32ChnNum,enSize[0]);
/*video buffer*/
if(s32ChnNum >= 1)
{
u32BlkSize = SAMPLE_COMM_SYS_CalcPicVbBlkSize(gs_enNorm,\
enSize[0], SAMPLE_PIXEL_FORMAT, SAMPLE_SYS_ALIGN_WIDTH);
stVbConf.astCommPool[0].u32BlkSize = u32BlkSize;
#ifndef hi3516ev100
stVbConf.astCommPool[0].u32BlkCnt = 10;
#else
stVbConf.astCommPool[0].u32BlkCnt = 5;
#endif
}
if(s32ChnNum >= 2)
{
u32BlkSize = SAMPLE_COMM_SYS_CalcPicVbBlkSize(gs_enNorm,\
enSize[1], SAMPLE_PIXEL_FORMAT, SAMPLE_SYS_ALIGN_WIDTH);
stVbConf.astCommPool[1].u32BlkSize = u32BlkSize;
stVbConf.astCommPool[1].u32BlkCnt =10;
}
if(s32ChnNum >= 3)
{
u32BlkSize = SAMPLE_COMM_SYS_CalcPicVbBlkSize(gs_enNorm,\
enSize[2], SAMPLE_PIXEL_FORMAT, SAMPLE_SYS_ALIGN_WIDTH);
stVbConf.astCommPool[2].u32BlkSize = u32BlkSize;
stVbConf.astCommPool[2].u32BlkCnt = 10;
}
/******************************************
step 2: mpp system init.
******************************************/
s32Ret = SAMPLE_COMM_SYS_Init(&stVbConf);
if (HI_SUCCESS != s32Ret)
{
SAMPLE_PRT("system init failed with %d!\n", s32Ret);
goto END_VENC_1080P_CLASSIC_0;
}
/******************************************
step 3: start vi dev & chn to capture
******************************************/
stViConfig.enViMode = SENSOR_TYPE;
stViConfig.enRotate = ROTATE_NONE;
stViConfig.enNorm = VIDEO_ENCODING_MODE_AUTO;
stViConfig.enViChnSet = VI_CHN_SET_NORMAL;
stViConfig.enWDRMode = WDR_MODE_NONE;
s32Ret = SAMPLE_COMM_VI_StartVi(&stViConfig);
if (HI_SUCCESS != s32Ret)
{
SAMPLE_PRT("start vi failed!\n");
goto END_VENC_1080P_CLASSIC_1;
}
/******************************************
step 4: start vpss and vi bind vpss
******************************************/
s32Ret = SAMPLE_COMM_SYS_GetPicSize(gs_enNorm, enSize[0], &stSize);
if (HI_SUCCESS != s32Ret)
{
SAMPLE_PRT("SAMPLE_COMM_SYS_GetPicSize failed!\n");
goto END_VENC_1080P_CLASSIC_1;
}
if(s32ChnNum >= 1)
{
VpssGrp = 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 = SAMPLE_PIXEL_FORMAT;
stVpssGrpAttr.bSharpenEn = HI_FALSE;
s32Ret = SAMPLE_COMM_VPSS_StartGroup(VpssGrp, &stVpssGrpAttr);
if (HI_SUCCESS != s32Ret)
{
SAMPLE_PRT("Start Vpss failed!\n");
goto END_VENC_1080P_CLASSIC_2;
}
s32Ret = SAMPLE_COMM_VI_BindVpss(stViConfig.enViMode);
if (HI_SUCCESS != s32Ret)
{
SAMPLE_PRT("Vi bind Vpss failed!\n");
goto END_VENC_1080P_CLASSIC_3;
}
VpssChn = 0;
stVpssChnMode.enChnMode = VPSS_CHN_MODE_USER;
stVpssChnMode.bDouble = HI_FALSE;
stVpssChnMode.enPixelFormat = SAMPLE_PIXEL_FORMAT;
stVpssChnMode.u32Width = stSize.u32Width;
stVpssChnMode.u32Height = stSize.u32Height;
stVpssChnMode.enCompressMode = COMPRESS_MODE_SEG;
memset(&stVpssChnAttr, 0, sizeof(stVpssChnAttr));
评论2