#include #include #include #include #include #include #include #include #include "hi_common.h" #include "hi_comm_video.h" #include "hi_comm_sys.h" #include "mpi_sys.h" #include "hi_comm_vb.h" #include "mpi_vb.h" #include "hi_comm_vpss.h" #include "mpi_vpss.h" #include "mpi_vgs.h" #define MAX_FRM_WIDTH 2048 #define VALUE_BETWEEN(x,min,max) (((x)>=(min)) && ((x) <= (max))) char pFrameData[307200]; typedef struct hiDUMP_MEMBUF_S { VB_BLK hBlock; VB_POOL hPool; HI_U32 u32PoolId; HI_U32 u32PhyAddr; HI_U8* pVirAddr; HI_S32 s32Mdev; } DUMP_MEMBUF_S; static HI_S32 s_s32MemDev = -1; static HI_U32 u32VpssDepthFlag = 0; static HI_U32 u32SignalFlag = 0; static VPSS_GRP VpssGrp = 0; static VPSS_CHN VpssChn = 0; static HI_U32 u32OrigDepth = 0; static VIDEO_FRAME_INFO_S stFrame; static VB_POOL hPool = VB_INVALID_POOLID; static DUMP_MEMBUF_S stMem = {0}; static VGS_HANDLE hHandle = -1; static HI_U32 u32BlkSize = 0; static HI_CHAR* pUserPageAddr[2] = {HI_NULL,HI_NULL}; static HI_U32 u32Size = 0; static FILE* pfd = HI_NULL; static int cw = 200; static int ch = 200; static int cXb = 0x00; static int cYb = 0x00; #define MEM_DEV_OPEN() \ do {\ if (s_s32MemDev <= 0)\ {\ s_s32MemDev = open("/dev/mem", O_CREAT|O_RDWR|O_SYNC);\ if (s_s32MemDev < 0)\ {\ perror("Open dev/mem error");\ return -1;\ }\ }\ }while(0) #define MEM_DEV_CLOSE() \ do {\ HI_S32 s32Ret;\ if (s_s32MemDev > 0)\ {\ s32Ret = close(s_s32MemDev);\ if(HI_SUCCESS != s32Ret)\ {\ perror("Close mem/dev Fail");\ return s32Ret;\ }\ s_s32MemDev = -1;\ }\ }while(0) /*When saving a file,sp420 will be denoted by p420 and sp422 will be denoted by p422 in the name of the file */ #define PIXEL_FORMAT_YUV_SEMIPLANAR_420 0 #define PIXEL_FORMAT_YUV_SEMIPLANAR_422 1 #define PIXEL_FORMAT_YUV_400 2 void sample_yuv_dump(VIDEO_FRAME_S* pVBuf, FILE* pfd) { unsigned int w, h; char* pVBufVirt_Y; char* pVBufVirt_C; char* pMemContent; unsigned char TmpBuff[MAX_FRM_WIDTH]; //If this value is too small and the image is big, this memory may not be enough HI_U32 phy_addr; PIXEL_FORMAT_E enPixelFormat = pVBuf->enPixelFormat; HI_U32 u32UvHeight;/*When the storage format is a planar format, this variable is used to keep the height of the UV component */ printf("enPixelFormat = %d\n",enPixelFormat); if (PIXEL_FORMAT_YUV_SEMIPLANAR_420 == enPixelFormat) { printf("enPixelFormat1 = %d\n",enPixelFormat); u32Size = (pVBuf->u32Stride[0]) * (pVBuf->u32Height) * 3 / 2; u32UvHeight = pVBuf->u32Height / 2; } else if(PIXEL_FORMAT_YUV_SEMIPLANAR_422 == enPixelFormat) { printf("enPixelFormat2 = %d\n",enPixelFormat); u32Size = (pVBuf->u32Stride[0]) * (pVBuf->u32Height) * 2; u32UvHeight = pVBuf->u32Height; } else if(PIXEL_FORMAT_YUV_400 == enPixelFormat) { printf("enPixelFormat2 = %d\n",enPixelFormat); u32Size = (pVBuf->u32Stride[0]) * (pVBuf->u32Height); u32UvHeight = pVBuf->u32Height; } phy_addr = pVBuf->u32PhyAddr[0]; //printf("phy_addr:%x, size:%d\n", phy_addr, size); pUserPageAddr[0] = (HI_CHAR*) HI_MPI_SYS_Mmap(phy_addr, u32Size); if (HI_NULL == pUserPageAddr[0]) { return; } //printf("stride: %d,%d\n",pVBuf->u32Stride[0],pVBuf->u32Stride[1] ); pVBufVirt_Y = pUserPageAddr[0]; pVBufVirt_C = pVBufVirt_Y + (pVBuf->u32Stride[0]) * (pVBuf->u32Height); /* save Y ----------------------------------------------------------------*/ fprintf(stderr, "saving......Y......"); fflush(stderr); for (h = 0; h < pVBuf->u32Height; h++) { pMemContent = pVBufVirt_Y + h * pVBuf->u32Stride[0]; fwrite(pMemContent, pVBuf->u32Width, 1, pfd); } if(PIXEL_FORMAT_YUV_400 != enPixelFormat) { fflush(pfd); /* save U ----------------------------------------------------------------*/ fprintf(stderr, "U......"); fflush(stderr); for (h = 0; h < u32UvHeight; h++) { pMemContent = pVBufVirt_C + h * pVBuf->u32Stride[1]; pMemContent += 1; for (w = 0; w < pVBuf->u32Width / 2; w++) { TmpBuff[w] = *pMemContent; pMemContent += 2; } fwrite(TmpBuff, pVBuf->u32Width / 2, 1, pfd); } fflush(pfd); /* save V ----------------------------------------------------------------*/ fprintf(stderr, "V......"); fflush(stderr); for (h = 0; h < u32UvHeight; h++) { pMemContent = pVBufVirt_C + h * pVBuf->u32Stride[1]; for (w = 0; w < pVBuf->u32Width / 2; w++) { TmpBuff[w] = *pMemContent; pMemContent += 2; } fwrite(TmpBuff, pVBuf->u32Width / 2, 1, pfd); } } fflush(pfd); fprintf(stderr, "done %d!\n", pVBuf->u32TimeRef); fflush(stderr); HI_MPI_SYS_Munmap(pUserPageAddr[0], u32Size); pUserPageAddr[0] = HI_NULL; } //定义每250ms可以拍多少张照片 #define N250 8 #define EYE_OPEN 1 #define EYE_CLOSE 2 #define EYE_NULL 3 unsigned char eyeStatus[11*N250]; unsigned char eyeStatus_bak[11*N250]; /* 眨眼状态,拍照状态 111 111 222 222 222 111 111 111.... 500ms - 750ms - 500ms 录像 1.5 - 2s 111 111 222 222 222 222 222 222 222 111 111 111.... 500ms - 1750ms - 500ms */ void PhotoCtl() { int count = 0; int i; for(i=0;i< 7*N250;i++) { if(eyeStatus[i] == EYE_NULL)count ++; } //如果有500ms以上的NULL 则为无效数据 if(count > 2*N250) return; //判断眨眼 } void AddCalcToList(unsigned char flag) { memcpy(eyeStatus,eyeStatus_bak,sizeof(eyeStatus_bak)); eyeStatus[0] = flag; memcpy(eyeStatus+1,eyeStatus_bak,sizeof(eyeStatus_bak)-1); } #define OPEN_Y_MIN 60 #define OPEN_Y_MAX 80 #define CLOSE_Y_MIN 40 #define CLOSE_Y_MIN 70 long get_timestamp() { struct timeval tv_date; /* gettimeofday() could return an error, and should be tested. However, the * only possible error, according to 'man', is EFAULT, which can not happen * here, since tv is a local variable. */ gettimeofday( &tv_date, NULL ); return( (long ) tv_date.tv_sec * 1000000 + (long ) tv_date.tv_usec ); } int nCalcCount = 0; long preTime = 0; /*When saving a file,sp420 will be denoted by p420 and sp422 will be denoted by p422 in the name of the file */ void calc_yuv_dump_y(VIDEO_FRAME_S* pVBuf) { unsigned int h; char* pVBufVirt_Y; char* pVBufVirt_C; char* pMemContent; unsigned char TmpBuff[MAX_FRM_WIDTH]; //If this value is too small and the image is big, this memory may not be enough HI_U32 phy_addr; PIXEL_FORMAT_E enPixelFormat = pVBuf->enPixelFormat; HI_U32 u32UvHeight;/*When the storage format is a planar format, this variable is used to keep the height of the UV component */ if (PIXEL_FORMAT_YUV_SEMIPLANAR_420 == enPixelFormat) { u32Size = (pVBuf->u32Stride[0]) * (pVBuf->u32Height) * 3 / 2; u32UvHeight = pVBuf->u32Height / 2; } else if(PIXEL_FORMAT_YUV_SEMIPLANAR_422 == enPixelFormat) { u32Size = (pVBuf->u32Stride[0]) * (pVBuf->u32Height) * 2; u32UvHeight = pVBuf->u32Height; } else if(PIXEL_FORMAT_YUV_400 == enPixelFormat) { u32Size = (pVBuf->u32Stride[0]) * (pVBuf->u32Height); u32UvHeight = pVBuf->u32Height; } phy_addr = pVBuf->u32PhyAddr[0]; //printf("phy_addr:%x, size:%d\n", phy_addr, size); pUserPageAddr[0] = (HI_CHAR*) HI_MPI_SYS_Mmap(phy_addr, u32Size); if (HI_NULL == pUserPageAddr[0]) { return; } //printf("stride: %d,%d\n",pVBuf->u32Stride[0],pVBuf->u32Stride[1] ); pVBufVirt_Y = pUserPageAddr[0]; pVBufVirt_C = pVBufVirt_Y + (pVBuf->u32Stride[0]) * (pVBuf->u32Height); //Get Y data. //配置是640*480的图像. //char chFrameTmp[640*480]; int ex = 0 ; if(pVBuf->u32Height == 480 && pVBuf->u32Width == 640) { //获取Y数据，一行一行的copy for (h = 0; h < pVBuf->u32Height; h++) { pMemContent = pVBufVirt_Y + h * pVBuf->u32Stride[0]; memcpy(pFrameData+ex, pMemContent,pVBuf->u32Width); ex += pVBuf->u32Width; } //以图像中心点计算w=cw h=ch方框的Y值 //然后取平均值. int xcb = cXb; int ycb = cYb; int xce = xcb + cw; int yce = ycb + ch; int x =0,y = 0; unsigned long avg = 0,index = 0; for(x = xcb;x <= xce;x++) for(y = ycb;y <= yce;y++) { avg += pFrameData[y*640+x]; index ++; } avg = (avg/index); #if 0 nCalcCount ++; if(nCalcCount%100 == 0) { nCalcCount = 0; long nowTime = get_timestamp(); printf("%d \n",nowTime - preTime); preTime = nowTime; printf("avg y :%d\n",avg); } #endif printf("avg y :%d\n",avg); //Calc Y avg.. } HI_MPI_SYS_Munmap(pUserPageAddr[0], u32Size); pUserPageAddr[0] = HI_NULL; } HI_S32 VPSS_Restore(VPSS_GRP VpssGrp, VPSS_CHN VpssChn) { if(VB_INVALID_POOLID != stFrame.u32PoolId) { HI_MPI_VPSS_ReleaseChnFrame(VpssGrp, VpssChn, &stFrame); stFrame.u32PoolId = VB_INVALID_POOLID; } if(-1 != hHandle) { HI_MPI_VGS_CancelJob(hHandle); hHandle = -1; } if(HI_NULL != stMem.pVirAddr) { HI_MPI_SYS_Munmap((HI_VOID*)stMem.pVirAddr, u32BlkSize ); stMem.u32PhyAddr = HI_NULL; } if(VB_INVALID_POOLID != stMem.hPool) { HI_MPI_VB_ReleaseBlock(stMem.hBlock); stMem.hPool = VB_INVALID_POOLID; } if (VB_INVALID_POOLID != hPool) { HI_MPI_VB_DestroyPool( hPool ); hPool = VB_INVALID_POOLID; } if(HI_NULL != pUserPageAddr[0]) { HI_MPI_SYS_Munmap(pUserPageAddr[0], u32Size); pUserPageAddr[0] = HI_NULL; } if(u32VpssDepthFlag) { if (HI_MPI_VPSS_SetDepth(VpssGrp, VpssChn, u32OrigDepth) != HI_SUCCESS) { printf("set depth error!!!\n"); } u32VpssDepthFlag = 0; } MEM_DEV_CLOSE(); return HI_SUCCESS; } void VPSS_Chn_Dump_HandleSig(HI_S32 signo) { if(u32SignalFlag) { exit(-1); } if (SIGINT == signo || SIGTERM == signo) { u32SignalFlag++; VPSS_Restore(VpssGrp, VpssChn); u32SignalFlag--; printf("\033[0;31mprogram termination abnormally!\033[0;39m\n"); } exit(-1); } HI_VOID* SAMPLE_MISC_VpssDump(VPSS_GRP Grp, VPSS_CHN Chn ) { //HI_CHAR szYuvName[128]; //HI_CHAR szPixFrm[10]; HI_U32 u32Depth = 2; HI_S32 s32MilliSec = 200; HI_S32 s32Times = 10; HI_BOOL bSendToVgs = HI_FALSE; VIDEO_FRAME_INFO_S stFrmInfo; VGS_TASK_ATTR_S stTask; HI_U32 u32LumaSize = 0; HI_U32 u32ChrmSize = 0; HI_U32 u32PicLStride = 0; HI_U32 u32PicCStride = 0; HI_U32 u32Width = 0; HI_U32 u32Height = 0; HI_S32 i = 0; HI_S32 s32Ret; if (HI_MPI_VPSS_GetDepth(Grp, Chn, &u32OrigDepth) != HI_SUCCESS) { printf("get depth error!!!\n"); return (HI_VOID*) - 1; } if (HI_MPI_VPSS_SetDepth(Grp, Chn, u32Depth) != HI_SUCCESS) { printf("set depth error!!!\n"); VPSS_Restore(Grp, Chn); return (HI_VOID*) - 1; } u32VpssDepthFlag = 1; /* get frame */ while (1) { if (HI_MPI_VPSS_GetChnFrame(Grp, Chn, &stFrame, s32MilliSec) != HI_SUCCESS) { printf("Get frame fail \n"); usleep(1000); continue; } bSendToVgs = ((stFrame.stVFrame.enCompressMode > 0) || (stFrame.stVFrame.enVideoFormat > 0)); if (bSendToVgs) { u32PicLStride = stFrame.stVFrame.u32Stride[0]; u32PicCStride = stFrame.stVFrame.u32Stride[0]; u32LumaSize = stFrame.stVFrame.u32Stride[0] * stFrame.stVFrame.u32Height; u32Width = stFrame.stVFrame.u32Width; u32Height = stFrame.stVFrame.u32Height; if(PIXEL_FORMAT_YUV_SEMIPLANAR_420 == stFrame.stVFrame.enPixelFormat) { u32BlkSize = stFrame.stVFrame.u32Stride[0] * stFrame.stVFrame.u32Height * 3 >> 1; u32ChrmSize = stFrame.stVFrame.u32Stride[0] * stFrame.stVFrame.u32Height >> 2; } else if(PIXEL_FORMAT_YUV_SEMIPLANAR_422 == stFrame.stVFrame.enPixelFormat) { u32BlkSize = stFrame.stVFrame.u32Stride[0] * stFrame.stVFrame.u32Height * 2; u32ChrmSize = stFrame.stVFrame.u32Stride[0] * stFrame.stVFrame.u32Height >> 1; } else if(PIXEL_FORMAT_YUV_400 == stFrame.stVFrame.enPixelFormat) { u32BlkSize = stFrame.stVFrame.u32Stride[0] * stFrame.stVFrame.u32Height; u32ChrmSize = 0; } else { printf("Unsupported pixelformat %d\n",stFrame.stVFrame.enPixelFormat); VPSS_Restore(Grp, Chn); return (HI_VOID*) - 1; } /*create comm vb pool*/ hPool = HI_MPI_VB_CreatePool( u32BlkSize, 1, HI_NULL); if (hPool == VB_INVALID_POOLID) { printf("HI_MPI_VB_CreatePool failed! \n"); VPSS_Restore(Grp, Chn); return (HI_VOID*) - 1; } stMem.hPool = hPool; while ((stMem.hBlock = HI_MPI_VB_GetBlock(stMem.hPool, u32BlkSize, HI_NULL)) == VB_INVALID_HANDLE) { ; } stMem.u32PhyAddr = HI_MPI_VB_Handle2PhysAddr(stMem.hBlock); stMem.pVirAddr = (HI_U8*) HI_MPI_SYS_Mmap( stMem.u32PhyAddr, u32BlkSize ); if (stMem.pVirAddr == HI_NULL) { printf("Mem dev may not open\n"); VPSS_Restore(Grp, Chn); return (HI_VOID*) - 1; } memset(&stFrmInfo.stVFrame, 0, sizeof(VIDEO_FRAME_S)); stFrmInfo.stVFrame.u32PhyAddr[0] = stMem.u32PhyAddr; stFrmInfo.stVFrame.u32PhyAddr[1] = stFrmInfo.stVFrame.u32PhyAddr[0] + u32LumaSize; stFrmInfo.stVFrame.u32PhyAddr[2] = stFrmInfo.stVFrame.u32PhyAddr[1] + u32ChrmSize; stFrmInfo.stVFrame.pVirAddr[0] = stMem.pVirAddr; stFrmInfo.stVFrame.pVirAddr[1] = (HI_U8*) stFrmInfo.stVFrame.pVirAddr[0] + u32LumaSize; stFrmInfo.stVFrame.pVirAddr[2] = (HI_U8*) stFrmInfo.stVFrame.pVirAddr[1] + u32ChrmSize; stFrmInfo.stVFrame.u32Width = u32Width; stFrmInfo.stVFrame.u32Height = u32Height; stFrmInfo.stVFrame.u32Stride[0] = u32PicLStride; stFrmInfo.stVFrame.u32Stride[1] = u32PicCStride; stFrmInfo.stVFrame.u32Stride[2] = u32PicCStride; stFrmInfo.stVFrame.enCompressMode = COMPRESS_MODE_NONE; stFrmInfo.stVFrame.enPixelFormat = stFrame.stVFrame.enPixelFormat; stFrmInfo.stVFrame.enVideoFormat = VIDEO_FORMAT_LINEAR; stFrmInfo.stVFrame.u64pts = (i * 40); stFrmInfo.stVFrame.u32TimeRef = (i * 2); stFrmInfo.u32PoolId = hPool; s32Ret = HI_MPI_VGS_BeginJob(&hHandle); if (s32Ret != HI_SUCCESS) { printf("HI_MPI_VGS_BeginJob failed\n"); hHandle = -1; VPSS_Restore(Grp, Chn); return (HI_VOID*) - 1; } memcpy(&stTask.stImgIn, &stFrame.stVFrame, sizeof(VIDEO_FRAME_INFO_S)); memcpy(&stTask.stImgOut , &stFrmInfo, sizeof(VIDEO_FRAME_INFO_S)); s32Ret = HI_MPI_VGS_AddScaleTask(hHandle, &stTask); if (s32Ret != HI_SUCCESS) { printf("HI_MPI_VGS_AddScaleTask failed\n"); VPSS_Restore(Grp, Chn); return (HI_VOID*) - 1; } s32Ret = HI_MPI_VGS_EndJob(hHandle); if (s32Ret != HI_SUCCESS) { printf("HI_MPI_VGS_EndJob failed\n"); VPSS_Restore(Grp, Chn); return (HI_VOID*) - 1; } hHandle = -1; /* save VO frame to file */ //sample_yuv_dump(&stFrmInfo.stVFrame, pfd); calc_yuv_dump_y(&stFrmInfo.stVFrame); HI_MPI_VB_ReleaseBlock(stMem.hBlock); stMem.hPool = VB_INVALID_POOLID; hHandle = -1; if(HI_NULL != stMem.pVirAddr) { HI_MPI_SYS_Munmap((HI_VOID*)stMem.pVirAddr, u32BlkSize ); stMem.u32PhyAddr = HI_NULL; } if (hPool != VB_INVALID_POOLID) { HI_MPI_VB_DestroyPool( hPool ); hPool = VB_INVALID_POOLID; } } else { //sample_yuv_dump(&stFrame.stVFrame, pfd); calc_yuv_dump_y(&stFrmInfo.stVFrame); } //printf("Get VpssGrp %d frame %d!!\n", Grp, u32Cnt); /* release frame after using */ HI_MPI_VPSS_ReleaseChnFrame(Grp, Chn, &stFrame); stFrame.u32PoolId = VB_INVALID_POOLID; } VPSS_Restore(Grp, Chn); return (HI_VOID*)0; } HI_S32 main(int argc, char* argv[]) { #if 0 if (argc == 3) { cw = atoi(argv[1]); ch = atoi(argv[2]); } if( cw < 0 || cw > 640) cw = 200; if( ch < 0 || ch > 480) ch = 200; printf("calc w = %d,h = %d\n",cw,ch); #endif if (argc == 3) { cXb = atoi(argv[1]); cXb = atoi(argv[2]); } if(argc == 5) { cXb = atoi(argv[1]); cXb = atoi(argv[2]); cw = atoi(argv[3]); ch = atoi(argv[4]); } printf("calc point(%d,%d),w = %d,h = %d\n",cXb,cYb,cw,ch); VpssGrp = 0; VpssChn = 1; u32VpssDepthFlag = 0; u32SignalFlag = 0; pUserPageAddr[0] = HI_NULL; stFrame.u32PoolId = VB_INVALID_POOLID; u32OrigDepth = 0; hPool = VB_INVALID_POOLID; hHandle = -1; u32BlkSize = 0; u32Size = 0; //pfd = HI_NULL; signal(SIGINT, VPSS_Chn_Dump_HandleSig); signal(SIGTERM, VPSS_Chn_Dump_HandleSig); MEM_DEV_OPEN(); SAMPLE_MISC_VpssDump(VpssGrp, VpssChn); MEM_DEV_CLOSE(); return HI_SUCCESS; }