zcy
/
multimedia
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity
multimedia/device/mpp/tools/vi_bayerdump.c

415 lines
10 KiB
C
Raw Blame History

#include <unistd.h>
#include <sys/types.h>
#include <sys/stat.h>
#include <sys/ioctl.h>
#include <sys/poll.h>
#include <sys/time.h>
#include <sys/mman.h>
#include <fcntl.h>
#include <errno.h>
#include <pthread.h>
#include <signal.h>
#include <stdlib.h>
#include <stdio.h>
#include <string.h>
#include "hi_common.h"
#include "hi_comm_video.h"
#include "hi_comm_sys.h"
#include "hi_comm_vi.h"
#include "hi_comm_vo.h"
#include "mpi_vb.h"
#include "mpi_sys.h"
#include "mpi_vi.h"
#include "mpi_vo.h"
#include "hi_comm_isp.h"
#include "mpi_isp.h"
#include "hi_sns_ctrl.h"
#include "mpi_af.h"
#include "mpi_awb.h"
#include "mpi_ae.h"
#define MAX_FRM_CNT 256
#define MAX_FRM_WIDTH 4096 //如果这个值太小，图像很大的话存不了
#define ALIGN_BACK(x, a) ((a) * (((x) / (a))))
#if 1
static HI_S32 s_s32MemDev = -1;
#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)
void usage(void)
{
printf(
"\n"
"*************************************************\n"
"Usage: ./vi_bayerdump [DataType] [nbit] [FrmCnt]\n"
"DataType: \n"
" -r(R) dump raw data\n"
" -i(I) dump ir data\n"
"nbit: \n"
" The bit num to be dump\n"
"FrmCnt: \n"
" the count of frame to be dump\n"
"e.g : ./vi_bayerdump -r 16 1 (dump one raw)\n"
"e.g : ./vi_bayerdump -i 16 1 (dump one ir)\n"
"*************************************************\n"
"\n");
exit(1);
}
HI_VOID* COMM_SYS_Mmap(HI_U32 u32PhyAddr, HI_U32 u32Size)
{
HI_U32 u32Diff;
HI_U32 u32PagePhy;
HI_U32 u32PageSize;
HI_U8* pPageAddr;
/* The mmap address should align with page */
u32PagePhy = u32PhyAddr & 0xfffff000;
u32Diff = u32PhyAddr - u32PagePhy;
/* The mmap size shuld be mutliples of 1024 */
u32PageSize = ((u32Size + u32Diff - 1) & 0xfffff000) + 0x1000;
pPageAddr = mmap ((void*)0, u32PageSize, PROT_READ | PROT_WRITE,
MAP_SHARED, s_s32MemDev, u32PagePhy);
if (MAP_FAILED == pPageAddr )
{
perror("mmap error");
return NULL;
}
return (HI_VOID*) (pPageAddr + u32Diff);
}
HI_S32 COMM_SYS_Munmap(HI_VOID* pVirAddr, HI_U32 u32Size)
{
HI_U32 u32PageAddr;
HI_U32 u32PageSize;
HI_U32 u32Diff;
u32PageAddr = (((HI_U32)pVirAddr) & 0xfffff000);
u32Diff = (HI_U32)pVirAddr - u32PageAddr;
u32PageSize = ((u32Size + u32Diff - 1) & 0xfffff000) + 0x1000;
return munmap((HI_VOID*)u32PageAddr, u32PageSize);
}
#endif
int sample_bayer_dump(VIDEO_FRAME_S* pVBuf, HI_U32 u32Nbit, FILE* pfd)
{
unsigned int w, h;
HI_U16* pVBufVirt_Y;
HI_U8 au8Data[MAX_FRM_WIDTH];
HI_U16 au16Data[MAX_FRM_WIDTH];
HI_U32 phy_addr, size;
HI_U8* pUserPageAddr[2];
size = (pVBuf->u32Stride[0]) * (pVBuf->u32Height) * 2;
phy_addr = pVBuf->u32PhyAddr[0];
MEM_DEV_OPEN();
pUserPageAddr[0] = (HI_U8*) COMM_SYS_Mmap(phy_addr, size);
if (NULL == pUserPageAddr[0])
{
return -1;
}
pVBufVirt_Y = (HI_U16*)pUserPageAddr[0];
/* save Y ----------------------------------------------------------------*/
fprintf(stderr, "saving......dump data......u32Stride[0]: %d, width: %d\n", pVBuf->u32Stride[0], pVBuf->u32Width);
fflush(stderr);
for (h = 0; h < pVBuf->u32Height; h++)
{
HI_U16 u16Data;
for (w = 0; w < pVBuf->u32Width; w++)
{
if (8 == u32Nbit)
{
au8Data[w] = (pVBufVirt_Y[h * pVBuf->u32Width + w] >> 8);
}
else if (10 == u32Nbit)
{
u16Data = (pVBufVirt_Y[h * pVBuf->u32Width + w] >> 6);
au16Data[w] = u16Data;
}
else if (12 == u32Nbit) //12bit 在高位
{
u16Data = (pVBufVirt_Y[h * pVBuf->u32Width + w] >> 4);
au16Data[w] = u16Data;
}
else if (14 == u32Nbit) //14bit 在高位
{
u16Data = (pVBufVirt_Y[h * pVBuf->u32Width + w] >> 2);
au16Data[w] = u16Data;
}
else if (16 == u32Nbit)
{
u16Data = pVBufVirt_Y[h * pVBuf->u32Width + w];
au16Data[w] = u16Data;
}
else
{
printf("Err! Bayer data can't support %d bits!eg: 8bits;10bits;12bits.\n", u32Nbit);
return -1;
}
}
if (8 == u32Nbit)
{
fwrite(au8Data, pVBuf->u32Width, 1, pfd);
}
else
{
fwrite(au16Data, pVBuf->u32Width, 2, pfd);
}
}
fflush(pfd);
fprintf(stderr, "done u32TimeRef: %d!\n", pVBuf->u32TimeRef);
fflush(stderr);
COMM_SYS_Munmap(pUserPageAddr[0], size);
MEM_DEV_CLOSE();
return 0;
}
HI_S32 VI_DumpBayer(VI_DEV ViDev, VI_DUMP_ATTR_S* pstViDumpAttr, HI_U32 u32Nbit, HI_U32 u32Cnt)
{
int i, j;
VI_FRAME_INFO_S stFrame;
VI_FRAME_INFO_S astFrame[MAX_FRM_CNT];
HI_CHAR szYuvName[128];
FILE* pfd;
HI_S32 s32MilliSec = 2000;
VI_CHN ViChn;
HI_U32 u32CapCnt;
VIU_GET_RAW_CHN(ViDev, ViChn);
if (HI_MPI_VI_SetFrameDepth(ViChn, 1))
{
printf("HI_MPI_VI_SetFrameDepth err, vi chn %d \n", ViChn);
return -1;
}
usleep(5000);
if (HI_MPI_VI_GetFrame(ViChn, &stFrame.stViFrmInfo, s32MilliSec))
{
printf("HI_MPI_VI_GetFrame err, vi chn %d \n", ViChn);
return -1;
}
/* get VI frame */
for (i = 0; i < u32Cnt; i++)
{
if (HI_MPI_VI_GetFrame(ViChn, &astFrame[i].stViFrmInfo, s32MilliSec) < 0)
{
printf("get vi chn %d frame err\n", ViChn);
printf("only get %d frame\n", i);
break;
}
}
u32CapCnt = i;
/* make file name */
if (VI_DUMP_TYPE_RAW == pstViDumpAttr->enDumpType)
{
sprintf(szYuvName, "./vi_dev_%d_%d_%d_%d_%dbits.raw", ViDev,
stFrame.stViFrmInfo.stVFrame.u32Width, stFrame.stViFrmInfo.stVFrame.u32Height,
u32CapCnt, u32Nbit);
printf("Dump raw frame of vi chn %d to file: \"%s\"\n", ViChn, szYuvName);
}
else if (VI_DUMP_TYPE_IR == pstViDumpAttr->enDumpType)
{
u32Nbit = 16;
sprintf(szYuvName, "./vi_dev_%d_%d_%d_%d_%dbits.ir", ViDev,
stFrame.stViFrmInfo.stVFrame.u32Width, stFrame.stViFrmInfo.stVFrame.u32Height,
u32CapCnt, u32Nbit);
printf("Dump ir frame of vi chn %d to file: \"%s\"\n", ViChn, szYuvName);
}
else
{
printf("Invalid dump type %d\n", pstViDumpAttr->enDumpType);
}
HI_MPI_VI_ReleaseFrame(ViChn, &stFrame.stViFrmInfo);
/* open file */
pfd = fopen(szYuvName, "wb");
if (NULL == pfd)
{
printf("open file failed:%s!\n", strerror(errno));
return -1;
}
for (j = 0; j < i; j++)
{
/* save VI frame to file */
sample_bayer_dump(&astFrame[j].stViFrmInfo.stVFrame, u32Nbit, pfd);
/* release frame after using */
HI_MPI_VI_ReleaseFrame(ViChn, &astFrame[j].stViFrmInfo);
}
fclose(pfd);
return 0;
}
HI_S32 main(int argc, char* argv[])
{
VI_DEV ViDev = 0;
ISP_DEV IspDev = 0;
HI_S32 s32Ret = 0;
HI_U32 u32Nbit = 8;
HI_U32 u32FrmCnt = 1;
HI_CHAR au8Type[20] = "-r";
VI_DUMP_ATTR_S stViDumpAttr = {0};
ISP_RGBIR_ATTR_S stRgbirAttr =
{
1,
ISP_IRPOS_TYPE_GR,
4028,
};
ISP_RGBIR_CTRL_S stRgbirCtrl =
{
1,
1,
1,
OP_TYPE_AUTO,
0x100,
{264, -1, 21, 8, 258, 9, -20, 26, 238, -261, -255, -229, 19, 12, 15}
};
ISP_RGBIR_ATTR_S stRgbirBkAttr;
ISP_RGBIR_CTRL_S stRgbirBkCtrl;
printf("\nNOTICE: This tool only can be used for TESTING !!!\n");
if (argc < 2)
{
printf("usage: ./vi_bayerdump [Type] [nbit] [frmcnt]. sample: ./vi_bayerdump -r 12 1\n");
printf("[nbit]----------Raw data:8bit;10bit;12bit;16bit\r\n");
printf("[frmcnt]----------the frame number \r\n\n\n");
return s32Ret;
}
if (argc > 1)
{
strcpy(au8Type, argv[1]);
}
if (argc > 2)
{
u32Nbit = atoi(argv[2]); /* nbit of Raw data:8bit;10bit;12bit;16bit */
}
if (argc > 3)
{
u32FrmCnt = atoi(argv[3]);/* the frame number */
}
if (!strcmp("-r", au8Type) || !strcmp("-R", au8Type))
{
stViDumpAttr.enDumpType = VI_DUMP_TYPE_RAW;
HI_MPI_VI_SetDevDumpAttr(ViDev, &stViDumpAttr);
}
else if (!strcmp("-i", au8Type) || !strcmp("-I", au8Type))
{
s32Ret = HI_MPI_ISP_GetRgbirAttr(IspDev, &stRgbirBkAttr);
if (s32Ret != HI_SUCCESS)
{
printf("Get RGBIR Attr failed, errcode: %#x\n", s32Ret);
return s32Ret;
}
s32Ret = HI_MPI_ISP_GetRgbirCtrl(IspDev, &stRgbirBkCtrl);
if (s32Ret != HI_SUCCESS)
{
printf("Get RGBIR Ctrl failed, errcode: %#x\n", s32Ret);
return s32Ret;
}
stViDumpAttr.enDumpType = VI_DUMP_TYPE_IR;
HI_MPI_ISP_SetRgbirAttr(IspDev, &stRgbirAttr);
HI_MPI_ISP_SetRgbirCtrl(IspDev, &stRgbirCtrl);
HI_MPI_VI_SetDevDumpAttr(ViDev, &stViDumpAttr);
}
else
{
usage();
}
s32Ret = HI_MPI_VI_EnableBayerDump(ViDev);
if (HI_SUCCESS != s32Ret)
{
printf("HI_MPI_VI_EnableBayerDump failed(0x%x)!\n", s32Ret);
return s32Ret;
}
printf("===press any key to continue.\n");
getchar();
s32Ret = VI_DumpBayer(ViDev, &stViDumpAttr, u32Nbit, u32FrmCnt);
if (HI_SUCCESS != s32Ret)
{
printf("VI_StartBayerData failed!\n");
goto EXIT;
}
EXIT:
s32Ret = HI_MPI_VI_DisableBayerDump(0);
if (HI_SUCCESS != s32Ret)
{
printf("HI_MPI_VI_DisableBayerDump failed!\n");
}
if (!strcmp("-i", au8Type) || !strcmp("-I", au8Type))
{
HI_MPI_ISP_SetRgbirAttr(IspDev, &stRgbirBkAttr);
HI_MPI_ISP_SetRgbirCtrl(IspDev, &stRgbirBkCtrl);
}
return HI_SUCCESS;
}
Reference in New Issue View Git Blame Copy Permalink