123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538 |
- // SPDX-License-Identifier: GPL-2.0
- /*
- * Copyright (C) 2021 StarFive Technology Co., Ltd.
- */
- #include <stdio.h>
- #include <unistd.h>
- #include <signal.h>
- #include <getopt.h>
- #include <fcntl.h>
- #include <OMX_Core.h>
- #include <OMX_Component.h>
- #include <OMX_Video.h>
- #include <OMX_VideoExt.h>
- #include <OMX_IndexExt.h>
- #include <sys/time.h>
- #include <sys/types.h>
- #include <sys/ipc.h>
- #include <sys/msg.h>
- #include <errno.h>
- #include <string.h>
- #include <stdlib.h>
- #define OMX_INIT_STRUCTURE(a) \
- memset(&(a), 0, sizeof(a)); \
- (a).nSize = sizeof(a); \
- (a).nVersion.nVersion = 1; \
- (a).nVersion.s.nVersionMajor = 1; \
- (a).nVersion.s.nVersionMinor = 1; \
- (a).nVersion.s.nRevision = 1; \
- (a).nVersion.s.nStep = 1
- typedef struct Message
- {
- long msg_type;
- OMX_S32 msg_flag;
- OMX_BUFFERHEADERTYPE *pBuffer;
- } Message;
- typedef struct EncodeTestContext
- {
- OMX_HANDLETYPE hComponentEncoder;
- char sOutputFilePath[256];
- char sInputFilePath[256];
- FILE *pInputFile;
- char sInputFormat[64];
- char sOutputFormat[64];
- OMX_U32 nFrameBufferSize;
- OMX_U32 nBitrate;
- OMX_U32 nFrameRate;
- OMX_U32 nNumPFrame;
- OMX_STATETYPE comState;
- OMX_BUFFERHEADERTYPE *pInputBufferArray[64];
- OMX_BUFFERHEADERTYPE *pOutputBufferArray[64];
- int msgid;
- } EncodeTestContext;
- EncodeTestContext *encodeTestContext;
- static FILE *fb = NULL;
- static OMX_BOOL inputEndFlag = OMX_FALSE;
- static OMX_S32 FillInputBuffer(EncodeTestContext *encodeTestContext, OMX_BUFFERHEADERTYPE *pInputBuffer);
- static OMX_BOOL disableEVnt;
- static OMX_BOOL justQuit = OMX_FALSE;
- static OMX_ERRORTYPE event_handler(
- OMX_HANDLETYPE hComponent,
- OMX_PTR pAppData,
- OMX_EVENTTYPE eEvent,
- OMX_U32 nData1,
- OMX_U32 nData2,
- OMX_PTR pEventData)
- {
- EncodeTestContext *pEncodeTestContext = (EncodeTestContext *)pAppData;
- switch (eEvent)
- {
- case OMX_EventPortSettingsChanged:
- {
- OMX_PARAM_PORTDEFINITIONTYPE pOutputPortDefinition;
- OMX_INIT_STRUCTURE(pOutputPortDefinition);
- pOutputPortDefinition.nPortIndex = 1;
- OMX_GetParameter(pEncodeTestContext->hComponentEncoder, OMX_IndexParamPortDefinition, &pOutputPortDefinition);
- OMX_U32 nOutputBufferSize = pOutputPortDefinition.nBufferSize;
- OMX_U32 nOutputBufferCount = pOutputPortDefinition.nBufferCountMin;
- printf("enable output port and alloc buffer\n");
- OMX_SendCommand(pEncodeTestContext->hComponentEncoder, OMX_CommandPortEnable, 1, NULL);
- for (int i = 0; i < nOutputBufferCount; i++)
- {
- OMX_BUFFERHEADERTYPE *pBuffer = NULL;
- OMX_AllocateBuffer(hComponent, &pBuffer, 1, NULL, nOutputBufferSize);
- pEncodeTestContext->pOutputBufferArray[i] = pBuffer;
- OMX_FillThisBuffer(hComponent, pBuffer);
- }
- }
- break;
- case OMX_EventBufferFlag:
- {
- Message data;
- data.msg_type = 1;
- data.msg_flag = -1;
- if (msgsnd(pEncodeTestContext->msgid, (void *)&data, sizeof(data) - sizeof(data.msg_type), 0) == -1)
- {
- fprintf(stderr, "msgsnd failed\n");
- }
- }
- break;
- case OMX_EventCmdComplete:
- {
- switch ((OMX_COMMANDTYPE)(nData1))
- {
- case OMX_CommandStateSet:
- {
- pEncodeTestContext->comState = (OMX_STATETYPE)(nData2);
- }
- case OMX_CommandPortDisable:
- {
- if (nData2 == 1)
- disableEVnt = OMX_TRUE;
- }
- break;
- default:
- break;
- }
- }
- break;
- case OMX_EventError:
- {
- printf("receive err event %d %d\n", nData1, nData2);
- justQuit = OMX_TRUE;
- }
- break;
- default:
- break;
- }
- return OMX_ErrorNone;
- }
- static void help()
- {
- printf("video_enc_test - omx video hardware encode unit test case\r\n\r\n");
- printf("Usage:\r\n\r\n");
- printf("./video_enc_test -i <input file> input file \r\n");
- printf(" -o <output file> output file \r\n");
- printf(" -w <width> input width \r\n");
- printf(" -h <height> input hight \r\n");
- printf(" -s <Srcformat> stream format h264/h265 \r\n");
- printf(" -c <cformat> color format i420/nv12/nv21 \r\n");
- printf(" -b <bitrate> (optional) set bit rate \r\n");
- printf(" -v <frame rate> (optional) set frame rate \r\n");
- printf(" -g <num> (optional) Number of P frames between each I frame\r\n\r\n");
- printf("./video_enc_test --help: show this message\r\n");
- }
- static OMX_ERRORTYPE fill_output_buffer_done_handler(
- OMX_HANDLETYPE hComponent,
- OMX_PTR pAppData,
- OMX_BUFFERHEADERTYPE *pBuffer)
- {
- EncodeTestContext *pEncodeTestContext = (EncodeTestContext *)pAppData;
- Message data;
- data.msg_type = 1;
- if (pBuffer->nFlags & OMX_BUFFERFLAG_EOS)
- {
- data.msg_flag = -1;
- }
- else
- {
- data.msg_flag = 1;
- data.pBuffer = pBuffer;
- }
- if (msgsnd(pEncodeTestContext->msgid, (void *)&data, sizeof(data) - sizeof(data.msg_type), 0) == -1)
- {
- fprintf(stderr, "msgsnd failed\n");
- }
- return OMX_ErrorNone;
- }
- static OMX_ERRORTYPE empty_buffer_done_handler(
- OMX_HANDLETYPE hComponent,
- OMX_PTR pAppData,
- OMX_BUFFERHEADERTYPE *pBuffer)
- {
- EncodeTestContext *pEncodeTestContext = (EncodeTestContext *)pAppData;
- Message data;
- data.msg_type = 1;
- data.msg_flag = 0;
- data.pBuffer = pBuffer;
- if (msgsnd(pEncodeTestContext->msgid, (void *)&data, sizeof(data) - sizeof(data.msg_type), 0) == -1)
- {
- fprintf(stderr, "msgsnd failed\n");
- }
- return OMX_ErrorNone;
- }
- static void signal_handle(int sig)
- {
- printf("[%s,%d]: receive sig=%d \n", __FUNCTION__, __LINE__, sig);
- OMX_FreeHandle(encodeTestContext->hComponentEncoder);
- OMX_Deinit();
- if (fb)
- fclose(fb);
- if (encodeTestContext->pInputFile)
- fclose(encodeTestContext->pInputFile);
- exit(0);
- // Message data;
- // data.msg_type = 1;
- // data.msg_flag = -1;
- // if (msgsnd(encodeTestContext->msgid, (void *)&data, sizeof(data) - sizeof(data.msg_type), 0) == -1)
- // {
- // fprintf(stderr, "msgsnd failed\n");
- // }
- }
- static OMX_S32 FillInputBuffer(EncodeTestContext *encodeTestContext, OMX_BUFFERHEADERTYPE *pInputBuffer)
- {
- FILE *fp = encodeTestContext->pInputFile;
- OMX_U32 size = encodeTestContext->nFrameBufferSize;
- OMX_U32 count;
- if (inputEndFlag)
- {
- pInputBuffer->nFlags = 0;
- pInputBuffer->nFilledLen = 0;
- return 0;
- }
- count = fread(pInputBuffer->pBuffer, 1, size, fp);
- if (count < size)
- {
- pInputBuffer->nFlags = 0x1;
- pInputBuffer->nFilledLen = 0;
- count = 0;
- inputEndFlag = OMX_TRUE;
- }
- else
- {
- pInputBuffer->nFlags = 0x10;
- pInputBuffer->nFilledLen = size;
- }
- return count;
- }
- int main(int argc, char **argv)
- {
- printf("=============================\r\n");
- encodeTestContext = malloc(sizeof(EncodeTestContext));
- memset(encodeTestContext, 0, sizeof(EncodeTestContext));
- OMX_S32 msgid = -1;
- msgid = msgget(IPC_PRIVATE, 0666 | IPC_CREAT);
- if (msgid < 0)
- {
- perror("get ipc_id error");
- return -1;
- }
- encodeTestContext->msgid = msgid;
- struct option longOpt[] = {
- {"output", required_argument, NULL, 'o'},
- {"input", required_argument, NULL, 'f'},
- {"colorformat", required_argument, NULL, 'c'},
- {"streamformat", required_argument, NULL, 's'},
- {"bitrate", required_argument, NULL, 'b'},
- {"framerate", required_argument, NULL, 'v'},
- {"width", required_argument, NULL, 'w'},
- {"height", required_argument, NULL, 'h'},
- {"gop", required_argument, NULL, 'g'},
- {"help", no_argument, NULL, 0},
- {NULL, no_argument, NULL, 0},
- };
- char *shortOpt = "i:o:f:s:w:h:b:v:c:g:";
- OMX_U32 c;
- OMX_S32 l;
- if (argc == 0)
- {
- help();
- return 0;
- }
- OMX_U32 width = 0;
- OMX_U32 height = 0;
- while ((c = getopt_long(argc, argv, shortOpt, longOpt, (int *)&l)) != -1)
- {
- switch (c)
- {
- case 'i':
- printf("input: %s\r\n", optarg);
- if (access(optarg, R_OK) != -1)
- {
- memcpy(encodeTestContext->sInputFilePath, optarg, strlen(optarg));
- encodeTestContext->pInputFile = fopen(optarg, "r");
- }
- else
- {
- printf("input file not exist!\r\n");
- return -1;
- }
- break;
- case 'o':
- printf("output: %s\r\n", optarg);
- memcpy(encodeTestContext->sOutputFilePath, optarg, strlen(optarg));
- break;
- case 'c':
- printf("color format: %s\r\n", optarg);
- memcpy(encodeTestContext->sInputFormat, optarg, strlen(optarg));
- break;
- case 's':
- printf("stream format: %s\r\n", optarg);
- memcpy(encodeTestContext->sOutputFormat, optarg, strlen(optarg));
- break;
- case 'w':
- printf("width: %s\r\n", optarg);
- width = atoi(optarg);
- break;
- case 'h':
- printf("height: %s\r\n", optarg);
- height = atoi(optarg);
- break;
- case 'b':
- printf("bit rate: %s\r\n", optarg);
- encodeTestContext->nBitrate = atoi(optarg);
- break;
- case 'v':
- printf("frame rate: %s\r\n", optarg);
- encodeTestContext->nFrameRate = atoi(optarg);
- break;
- case 'g':
- printf("P frames: %s\r\n", optarg);
- encodeTestContext->nNumPFrame = atoi(optarg);
- break;
- case 0:
- default:
- help();
- return 0;
- }
- }
- encodeTestContext->nFrameBufferSize = width * height * 3 / 2;
- if (encodeTestContext == NULL ||
- encodeTestContext->nFrameBufferSize == 0 ||
- encodeTestContext->pInputFile == NULL)
- {
- help();
- return -1;
- }
- /*omx init*/
- OMX_HANDLETYPE hComponentEncoder = NULL;
- OMX_CALLBACKTYPE callbacks;
- int ret = OMX_ErrorNone;
- signal(SIGINT, signal_handle);
- printf("init omx\r\n");
- ret = OMX_Init();
- if (ret != OMX_ErrorNone)
- {
- printf("[%s,%d]: run OMX_Init failed. ret is %d \n", __FUNCTION__, __LINE__, ret);
- if (encodeTestContext->pInputFile)
- fclose(encodeTestContext->pInputFile);
- return 1;
- }
- callbacks.EventHandler = event_handler;
- callbacks.FillBufferDone = fill_output_buffer_done_handler;
- callbacks.EmptyBufferDone = empty_buffer_done_handler;
- printf("get handle %s\r\n", encodeTestContext->sOutputFormat);
- if (strstr(encodeTestContext->sOutputFormat, "h265") != NULL)
- {
- OMX_GetHandle(&hComponentEncoder, "OMX.sf.video_encoder.hevc", encodeTestContext, &callbacks);
- }
- if (hComponentEncoder == NULL)
- {
- printf("could not get handle\r\n");
- OMX_Deinit();
- if (encodeTestContext->pInputFile)
- fclose(encodeTestContext->pInputFile);
- return 0;
- }
- encodeTestContext->hComponentEncoder = hComponentEncoder;
- OMX_PARAM_PORTDEFINITIONTYPE pInputPortDefinition;
- OMX_INIT_STRUCTURE(pInputPortDefinition);
- pInputPortDefinition.nPortIndex = 0;
- OMX_GetParameter(hComponentEncoder, OMX_IndexParamPortDefinition, &pInputPortDefinition);
- pInputPortDefinition.format.video.nFrameWidth = width;
- pInputPortDefinition.format.video.nFrameHeight = height;
- if (strstr(encodeTestContext->sInputFormat, "i420") != NULL)
- {
- pInputPortDefinition.format.video.eColorFormat = OMX_COLOR_FormatYUV420Planar;
- }
- else if (strstr(encodeTestContext->sInputFormat, "nv12") != NULL)
- {
- pInputPortDefinition.format.video.eColorFormat = OMX_COLOR_FormatYUV420SemiPlanar;
- }
- else if (strstr(encodeTestContext->sInputFormat, "nv21") != NULL)
- {
- pInputPortDefinition.format.video.eColorFormat = OMX_COLOR_FormatYVU420SemiPlanar;
- }
- else
- {
- printf("unsupported color format: %s\r\n", encodeTestContext->sInputFormat);
- goto end;
- }
- if(encodeTestContext->nFrameRate){
- pInputPortDefinition.format.video.xFramerate = encodeTestContext->nFrameRate;
- }
- OMX_SetParameter(hComponentEncoder, OMX_IndexParamPortDefinition, &pInputPortDefinition);
- OMX_GetParameter(hComponentEncoder, OMX_IndexParamPortDefinition, &pInputPortDefinition);
- OMX_PARAM_PORTDEFINITIONTYPE pOutputPortDefinition;
- OMX_INIT_STRUCTURE(pOutputPortDefinition);
- pOutputPortDefinition.nPortIndex = 1;
- OMX_GetParameter(hComponentEncoder, OMX_IndexParamPortDefinition, &pOutputPortDefinition);
- pOutputPortDefinition.format.video.nFrameWidth = width;
- pOutputPortDefinition.format.video.nFrameHeight = height;
- pOutputPortDefinition.format.video.nBitrate = encodeTestContext->nBitrate;
- OMX_SetParameter(hComponentEncoder, OMX_IndexParamPortDefinition, &pOutputPortDefinition);
- if(encodeTestContext->nNumPFrame){
- if (strstr(encodeTestContext->sOutputFormat, "h264") != NULL)
- {
- OMX_VIDEO_PARAM_AVCTYPE avcType;
- OMX_INIT_STRUCTURE(avcType);
- avcType.nPortIndex = 1;
- OMX_GetParameter(hComponentEncoder, OMX_IndexParamVideoAvc, &avcType);
- avcType.nPFrames = encodeTestContext->nNumPFrame;
- OMX_SetParameter(hComponentEncoder, OMX_IndexParamVideoAvc, &avcType);
- }
- else if (strstr(encodeTestContext->sOutputFormat, "h265") != NULL)
- {
- OMX_VIDEO_PARAM_HEVCTYPE hevcType;
- OMX_INIT_STRUCTURE(hevcType);
- hevcType.nPortIndex = 1;
- OMX_GetParameter(hComponentEncoder, OMX_IndexParamVideoHevc, &hevcType);
- hevcType.nKeyFrameInterval = encodeTestContext->nNumPFrame;
- OMX_SetParameter(hComponentEncoder, OMX_IndexParamVideoHevc, &hevcType);
- }
- }
- /*Alloc input buffer*/
- OMX_U32 nInputBufferSize = pInputPortDefinition.nBufferSize;
- OMX_U32 nInputBufferCount = pInputPortDefinition.nBufferCountActual;
- disableEVnt = OMX_FALSE;
- OMX_SendCommand(hComponentEncoder, OMX_CommandPortDisable, 1, NULL);
- printf("wait for output port disable\r\n");
- while (!disableEVnt && !justQuit);
- if (justQuit)
- goto end;
- printf("output port disabled\r\n");
- OMX_SendCommand(hComponentEncoder, OMX_CommandStateSet, OMX_StateIdle, NULL);
- for (int i = 0; i < nInputBufferCount; i++)
- {
- OMX_BUFFERHEADERTYPE *pBuffer = NULL;
- OMX_AllocateBuffer(hComponentEncoder, &pBuffer, 0, NULL, nInputBufferSize);
- encodeTestContext->pInputBufferArray[i] = pBuffer;
- }
- printf("wait for Component idle\r\n");
- while (encodeTestContext->comState != OMX_StateIdle && !justQuit);
- if (justQuit)
- goto end;
- printf("Component in idle\r\n");
- for (int i = 0; i < nInputBufferCount; i++)
- {
- /*Fill Input Buffer*/
- FillInputBuffer(encodeTestContext, encodeTestContext->pInputBufferArray[i]);
- if (encodeTestContext->pInputBufferArray[i]->nFilledLen || encodeTestContext->pInputBufferArray[i]->nFlags)
- OMX_EmptyThisBuffer(hComponentEncoder, encodeTestContext->pInputBufferArray[i]);
- }
- fb = fopen(encodeTestContext->sOutputFilePath, "wb+");
- printf("start process\r\n");
- OMX_SendCommand(hComponentEncoder, OMX_CommandStateSet, OMX_StateExecuting, NULL);
- /*wait until decode finished*/
- Message data;
- while (OMX_TRUE)
- {
- if (msgrcv(msgid, (void *)&data, BUFSIZ, 0, 0) == -1)
- {
- fprintf(stderr, "msgrcv failed with errno: %d\n", errno);
- goto end;
- }
- switch (data.msg_flag)
- {
- case 0:
- {
- OMX_BUFFERHEADERTYPE *pBuffer = data.pBuffer;
- FillInputBuffer(encodeTestContext, pBuffer);
- if (pBuffer->nFilledLen || pBuffer->nFlags)
- OMX_EmptyThisBuffer(encodeTestContext->hComponentEncoder, pBuffer);
- }
- break;
- case 1:
- {
- OMX_BUFFERHEADERTYPE *pBuffer = data.pBuffer;
- fwrite(pBuffer->pBuffer, 1, pBuffer->nFilledLen, fb);
- if (pBuffer->nFlags & OMX_BUFFERFLAG_EOS)
- {
- goto end;
- }
- else
- {
- OMX_FillThisBuffer(encodeTestContext->hComponentEncoder, pBuffer);
- }
- }
- break;
- case 2:
- break;
- default:
- goto end;
- }
- }
- end:
- /*free resource*/
- if (encodeTestContext->comState == OMX_StateExecuting)
- {
- OMX_SendCommand(hComponentEncoder, OMX_CommandStateSet, OMX_StateIdle, NULL);
- printf("wait for Component idle\r\n");
- while (encodeTestContext->comState != OMX_StateIdle && !justQuit);
- printf("Component in idle\r\n");
- }
- OMX_FreeHandle(hComponentEncoder);
- OMX_Deinit();
- if (fb)
- fclose(fb);
- if (encodeTestContext->pInputFile)
- fclose(encodeTestContext->pInputFile);
- }
|