306 lines
12 KiB
C
306 lines
12 KiB
C
|
/*
|
||
|
|
* Copyright (c) 2008-2016 Allwinner Technology Co. Ltd.
|
||
|
|
* All rights reserved.
|
||
|
|
*
|
||
|
|
* File : omx_venc.h
|
||
|
|
*
|
||
|
|
* Description : OpenMax Encoder Definition
|
||
|
|
* History :
|
||
|
|
*
|
||
|
|
*/
|
||
|
|
|
||
|
|
#ifndef __OMX_VENC_H__
|
||
|
|
#define __OMX_VENC_H__
|
||
|
|
|
||
|
|
#include <stdlib.h>
|
||
|
|
#include <stdio.h>
|
||
|
|
|
||
|
|
#include <pthread.h>
|
||
|
|
#include <semaphore.h>
|
||
|
|
|
||
|
|
#include "CdcUtil.h"
|
||
|
|
#include "OMX_Core.h"
|
||
|
|
#include "AWOMX_VideoIndexExtension.h"
|
||
|
|
#include "aw_omx_component.h"
|
||
|
|
#include "vencoder.h"
|
||
|
|
#include "omx_tsem.h"
|
||
|
|
#include "OMX_IndexExt.h"
|
||
|
|
#include "OMX_VideoExt.h"
|
||
|
|
|
||
|
|
extern "C"
|
||
|
|
{
|
||
|
|
OMX_API void* get_omx_component_factory_fn(void);
|
||
|
|
}
|
||
|
|
|
||
|
|
/*
|
||
|
|
* Enumeration for the commands processed by the component
|
||
|
|
*/
|
||
|
|
typedef enum ThrCmdType
|
||
|
|
{
|
||
|
|
SetState,
|
||
|
|
Flush,
|
||
|
|
StopPort,
|
||
|
|
RestartPort,
|
||
|
|
MarkBuf,
|
||
|
|
Stop,
|
||
|
|
FillBuf,
|
||
|
|
EmptyBuf
|
||
|
|
} ThrCmdType;
|
||
|
|
|
||
|
|
//////////////////////////////////////////////////////////////////////////////
|
||
|
|
// Module specific globals
|
||
|
|
//////////////////////////////////////////////////////////////////////////////
|
||
|
|
#define OMX_SPEC_VERSION 0x00000101
|
||
|
|
|
||
|
|
/*
|
||
|
|
* D E C L A R A T I O N S
|
||
|
|
*/
|
||
|
|
#define OMX_NOPORT 0xFFFFFFFE
|
||
|
|
#define NUM_IN_BUFFERS 2 // Input Buffers
|
||
|
|
#define NUM_MAX_IN_BUFFERS 16 // Input Buffers
|
||
|
|
#define NUM_OUT_BUFFERS 4 // Output Buffers
|
||
|
|
#define OMX_TIMEOUT 10 // Timeout value in milisecond
|
||
|
|
// Count of Maximum number of times the component can time out
|
||
|
|
#define OMX_MAX_TIMEOUTS 160
|
||
|
|
#define OMX_VIDEO_ENC_INPUT_BUFFER_SIZE (128*1024)
|
||
|
|
|
||
|
|
#define PRINTF_FRAME_SIZE 1
|
||
|
|
|
||
|
|
/*
|
||
|
|
* D E F I N I T I O N S
|
||
|
|
*/
|
||
|
|
|
||
|
|
typedef struct _BufferList BufferList;
|
||
|
|
|
||
|
|
|
||
|
|
/*
|
||
|
|
* The main structure for buffer management.
|
||
|
|
*
|
||
|
|
* pBufHdr - An array of pointers to buffer headers.
|
||
|
|
* The size of the array is set dynamically using the nBufferCountActual value
|
||
|
|
* send by the client.
|
||
|
|
* nListEnd - Marker to the boundary of the array. This points to the last index of the
|
||
|
|
* pBufHdr array.
|
||
|
|
* nSizeOfList - Count of valid data in the list.
|
||
|
|
* nAllocSize - Size of the allocated list. This is equal to (nListEnd + 1) in most of
|
||
|
|
* the times. When the list is freed this is decremented and at that
|
||
|
|
* time the value is not equal to (nListEnd + 1). This is because
|
||
|
|
* the list is not freed from the end and hence we cannot decrement
|
||
|
|
* nListEnd each time we free an element in the list. When nAllocSize is zero,
|
||
|
|
* the list is completely freed and the other paramaters of the list are
|
||
|
|
* initialized.
|
||
|
|
* If the client crashes before freeing up the buffers, this parameter is
|
||
|
|
* checked (for nonzero value) to see if there are still elements on the list.
|
||
|
|
* If yes, then the remaining elements are freed.
|
||
|
|
* nWritePos - The position where the next buffer would be written. The value is incremented
|
||
|
|
* after the write. It is wrapped around when it is greater than nListEnd.
|
||
|
|
* nReadPos - The position from where the next buffer would be read. The value is incremented
|
||
|
|
* after the read. It is wrapped around when it is greater than nListEnd.
|
||
|
|
* eDir - Type of BufferList.
|
||
|
|
* OMX_DirInput = Input Buffer List
|
||
|
|
* OMX_DirOutput = Output Buffer List
|
||
|
|
*/
|
||
|
|
struct _BufferList
|
||
|
|
{
|
||
|
|
OMX_BUFFERHEADERTYPE** pBufHdrList;
|
||
|
|
OMX_U32 nSizeOfList;
|
||
|
|
OMX_S32 nWritePos;
|
||
|
|
OMX_S32 nReadPos;
|
||
|
|
|
||
|
|
OMX_BUFFERHEADERTYPE* pBufArr;
|
||
|
|
OMX_S32 nAllocBySelfFlags;
|
||
|
|
OMX_S32 nBufArrSize;
|
||
|
|
OMX_U32 nAllocSize;
|
||
|
|
OMX_DIRTYPE eDir;
|
||
|
|
};
|
||
|
|
|
||
|
|
typedef struct VIDDEC_CUSTOM_PARAM
|
||
|
|
{
|
||
|
|
unsigned char cCustomParamName[128];
|
||
|
|
OMX_INDEXTYPE nCustomParamIndex;
|
||
|
|
} VIDDEC_CUSTOM_PARAM;
|
||
|
|
|
||
|
|
typedef struct VIDEO_PROFILE_LEVEL
|
||
|
|
{
|
||
|
|
OMX_S32 nProfile;
|
||
|
|
OMX_S32 nLevel;
|
||
|
|
} VIDEO_PROFILE_LEVEL_TYPE;
|
||
|
|
|
||
|
|
typedef struct OMXInputBufferInfoS
|
||
|
|
{
|
||
|
|
aw_ion_user_handle_t handle_ion;
|
||
|
|
unsigned long nBufPhyAddr;
|
||
|
|
char* pBufVirAddr;
|
||
|
|
int nBufFd;
|
||
|
|
int nShareFd;
|
||
|
|
long long nAwBufId;
|
||
|
|
}OMXInputBufferInfoT;
|
||
|
|
|
||
|
|
// OMX video decoder class
|
||
|
|
class omx_venc: public aw_omx_component
|
||
|
|
{
|
||
|
|
public:
|
||
|
|
omx_venc(); // constructor
|
||
|
|
virtual ~omx_venc(); // destructor
|
||
|
|
|
||
|
|
OMX_ERRORTYPE allocate_buffer(OMX_HANDLETYPE pHComp,
|
||
|
|
OMX_BUFFERHEADERTYPE** ppBufferHdr,
|
||
|
|
OMX_U32 uPort,
|
||
|
|
OMX_PTR pAppData,
|
||
|
|
OMX_U32 uBytes);
|
||
|
|
|
||
|
|
|
||
|
|
OMX_ERRORTYPE component_deinit(OMX_HANDLETYPE pHComp);
|
||
|
|
|
||
|
|
OMX_ERRORTYPE component_init(OMX_STRING pName);
|
||
|
|
|
||
|
|
OMX_ERRORTYPE component_role_enum(OMX_HANDLETYPE pHComp,
|
||
|
|
OMX_U8* pRole,
|
||
|
|
OMX_U32 uIndex);
|
||
|
|
|
||
|
|
OMX_ERRORTYPE component_tunnel_request(OMX_HANDLETYPE pHComp,
|
||
|
|
OMX_U32 uPort,
|
||
|
|
OMX_HANDLETYPE pPeerComponent,
|
||
|
|
OMX_U32 uPeerPort,
|
||
|
|
OMX_TUNNELSETUPTYPE* pTunnelSetup);
|
||
|
|
|
||
|
|
OMX_ERRORTYPE empty_this_buffer(OMX_HANDLETYPE pHComp,
|
||
|
|
OMX_BUFFERHEADERTYPE* pBuffer);
|
||
|
|
|
||
|
|
OMX_ERRORTYPE fill_this_buffer(OMX_HANDLETYPE pHComp,
|
||
|
|
OMX_BUFFERHEADERTYPE* pBuffer);
|
||
|
|
|
||
|
|
|
||
|
|
OMX_ERRORTYPE free_buffer(OMX_HANDLETYPE pHComp,
|
||
|
|
OMX_U32 uPort,
|
||
|
|
OMX_BUFFERHEADERTYPE* pBuffer);
|
||
|
|
|
||
|
|
OMX_ERRORTYPE get_component_version(OMX_HANDLETYPE pHComp,
|
||
|
|
OMX_STRING pComponentName,
|
||
|
|
OMX_VERSIONTYPE* pComponentVersion,
|
||
|
|
OMX_VERSIONTYPE* pSpecVersion,
|
||
|
|
OMX_UUIDTYPE* pComponentUUID);
|
||
|
|
|
||
|
|
OMX_ERRORTYPE get_config(OMX_HANDLETYPE pHComp,
|
||
|
|
OMX_INDEXTYPE eConfigIndex,
|
||
|
|
OMX_PTR pConfigData);
|
||
|
|
|
||
|
|
OMX_ERRORTYPE get_extension_index(OMX_HANDLETYPE pHComp,
|
||
|
|
OMX_STRING pParamName,
|
||
|
|
OMX_INDEXTYPE* pIndexType);
|
||
|
|
|
||
|
|
OMX_ERRORTYPE get_parameter(OMX_HANDLETYPE pHComp,
|
||
|
|
OMX_INDEXTYPE pParamIndex,
|
||
|
|
OMX_PTR pParamData);
|
||
|
|
|
||
|
|
OMX_ERRORTYPE get_state(OMX_HANDLETYPE pHComp,
|
||
|
|
OMX_STATETYPE *pState);
|
||
|
|
|
||
|
|
|
||
|
|
OMX_ERRORTYPE send_command(OMX_HANDLETYPE pHComp,
|
||
|
|
OMX_COMMANDTYPE eCmd,
|
||
|
|
OMX_U32 uParam1,
|
||
|
|
OMX_PTR pCmdData);
|
||
|
|
|
||
|
|
|
||
|
|
OMX_ERRORTYPE set_callbacks(OMX_HANDLETYPE pHComp,
|
||
|
|
OMX_CALLBACKTYPE* pCallbacks,
|
||
|
|
OMX_PTR pAppData);
|
||
|
|
|
||
|
|
OMX_ERRORTYPE set_config(OMX_HANDLETYPE pHComp,
|
||
|
|
OMX_INDEXTYPE eConfigIndex,
|
||
|
|
OMX_PTR pConfigData);
|
||
|
|
|
||
|
|
OMX_ERRORTYPE set_parameter(OMX_HANDLETYPE pHComp,
|
||
|
|
OMX_INDEXTYPE eCaramIndex,
|
||
|
|
OMX_PTR pParamData);
|
||
|
|
|
||
|
|
OMX_ERRORTYPE use_buffer(OMX_HANDLETYPE pHComp,
|
||
|
|
OMX_BUFFERHEADERTYPE** ppBufferHdr,
|
||
|
|
OMX_U32 uPort,
|
||
|
|
OMX_PTR pAppData,
|
||
|
|
OMX_U32 uBytes,
|
||
|
|
OMX_U8* pBuffer);
|
||
|
|
|
||
|
|
|
||
|
|
OMX_ERRORTYPE use_EGL_image(OMX_HANDLETYPE pHComp,
|
||
|
|
OMX_BUFFERHEADERTYPE** ppBufferHdr,
|
||
|
|
OMX_U32 uPort,
|
||
|
|
OMX_PTR pAppData,
|
||
|
|
void * pEglImage);
|
||
|
|
|
||
|
|
private:
|
||
|
|
OMX_ERRORTYPE post_event(ThrCmdType eCmd,
|
||
|
|
OMX_U32 param1,
|
||
|
|
OMX_PTR cmdData);
|
||
|
|
|
||
|
|
public:
|
||
|
|
OMX_STATETYPE m_state;
|
||
|
|
OMX_U8 m_cRole[OMX_MAX_STRINGNAME_SIZE];
|
||
|
|
OMX_U8 m_cName[OMX_MAX_STRINGNAME_SIZE];
|
||
|
|
OMX_VIDEO_CODINGTYPE m_eCompressionFormat;
|
||
|
|
OMX_CALLBACKTYPE m_Callbacks;
|
||
|
|
OMX_PTR m_pAppData;
|
||
|
|
OMX_PORT_PARAM_TYPE m_sPortParam;
|
||
|
|
OMX_PARAM_PORTDEFINITIONTYPE m_sInPortDefType;
|
||
|
|
OMX_PARAM_PORTDEFINITIONTYPE m_sOutPortDefType;
|
||
|
|
OMX_VIDEO_PARAM_PORTFORMATTYPE m_sInPortFormatType;
|
||
|
|
OMX_VIDEO_PARAM_PORTFORMATTYPE m_sOutPortFormatType;
|
||
|
|
OMX_PRIORITYMGMTTYPE m_sPriorityMgmtType;
|
||
|
|
OMX_PARAM_BUFFERSUPPLIERTYPE m_sInBufSupplierType;
|
||
|
|
OMX_PARAM_BUFFERSUPPLIERTYPE m_sOutBufSupplierType;
|
||
|
|
OMX_VIDEO_PARAM_BITRATETYPE m_sOutPutBitRateType;
|
||
|
|
OMX_VIDEO_PARAM_AVCTYPE m_sH264Type;
|
||
|
|
OMX_VIDEO_PARAM_HEVCTYPE m_sH265Type;
|
||
|
|
pthread_t m_thread_id;
|
||
|
|
pthread_t m_venc_thread_id;
|
||
|
|
int m_cmdpipe[2];
|
||
|
|
int m_cmddatapipe[2];
|
||
|
|
int m_venc_cmdpipe[2]; //pipe connect main_thread and vdrv_thread
|
||
|
|
BufferList m_sInBufList;
|
||
|
|
BufferList m_sOutBufList;
|
||
|
|
pthread_mutex_t m_inBufMutex;
|
||
|
|
pthread_mutex_t m_outBufMutex;
|
||
|
|
pthread_mutex_t m_pipeMutex;
|
||
|
|
|
||
|
|
VideoEncoder * m_encoder;
|
||
|
|
omx_sem_t m_msg_sem;
|
||
|
|
omx_sem_t m_input_sem;
|
||
|
|
OMX_BOOL m_useAndroidBuffer;
|
||
|
|
OMX_BOOL m_firstFrameFlag;
|
||
|
|
OMX_BOOL m_useAllocInputBuffer;
|
||
|
|
OMX_U32 m_framerate;
|
||
|
|
OMX_BOOL m_useMetaDataInBuffers;
|
||
|
|
OMX_BOOL m_prependSPSPPSToIDRFrames;
|
||
|
|
OMX_COLOR_FORMATTYPE m_inputcolorFormats[3];
|
||
|
|
VencH264Param m_vencH264Param;
|
||
|
|
VencH265Param m_vencH265Param;
|
||
|
|
VencAllocateBufferParam m_vencAllocBufferParam;
|
||
|
|
VencCyclicIntraRefresh m_vencCyclicIntraRefresh;
|
||
|
|
VencHeaderData m_headdata;
|
||
|
|
VENC_PIXEL_FMT m_vencColorFormat;
|
||
|
|
VENC_CODEC_TYPE m_vencCodecType;
|
||
|
|
char mCallingProcess[256];
|
||
|
|
OMX_BOOL mIsFromCts;
|
||
|
|
OMX_BOOL mIsFromVideoeditor;
|
||
|
|
OMX_BOOL mFirstInputFrame;
|
||
|
|
OMX_VIDEO_PARAMS_EXTENDED mVideoExtParams;
|
||
|
|
OMX_VIDEO_PARAMS_SUPER_FRAME mVideoSuperFrame;
|
||
|
|
OMX_VIDEO_PARAMS_SVC mVideoSVCSkip;
|
||
|
|
OMX_VIDEO_PARAMS_VBR mVideoVBR;
|
||
|
|
struct ScMemOpsS *memops;
|
||
|
|
OMX_BOOL m_usePSkip;
|
||
|
|
OMXInputBufferInfoT mInputBufInfo[NUM_MAX_IN_BUFFERS];
|
||
|
|
OMX_S32 mIonFd;
|
||
|
|
#if PRINTF_FRAME_SIZE
|
||
|
|
int mFrameCnt;
|
||
|
|
int mAllFrameSize;
|
||
|
|
uint64_t mTimeStart;
|
||
|
|
uint64_t mTimeOut;
|
||
|
|
int mEmptyBufCnt;
|
||
|
|
int mFillBufCnt;
|
||
|
|
#endif
|
||
|
|
};
|
||
|
|
|
||
|
|
#endif // __OMX_VENC_H__
|