frontends/omx/av1: enable AV1 OMX Bellagio support
By adding name, role and profile to OMX decoder Signed-off-by: Leo Liu <leo.liu@amd.com> Reviewed-by: Boyuan Zhang <boyuan.zhang@amd.com> Part-of: <https://gitlab.freedesktop.org/mesa/mesa/-/merge_requests/7596>
This commit is contained in:
@@ -71,7 +71,7 @@ OMX_ERRORTYPE vid_dec_LoaderComponent(stLoaderComponentType *comp)
|
||||
comp->componentVersion.s.nVersionMinor = 0;
|
||||
comp->componentVersion.s.nRevision = 0;
|
||||
comp->componentVersion.s.nStep = 1;
|
||||
comp->name_specific_length = 3;
|
||||
comp->name_specific_length = 4;
|
||||
|
||||
comp->name = CALLOC(1, OMX_MAX_STRINGNAME_SIZE);
|
||||
if (comp->name == NULL)
|
||||
@@ -97,6 +97,10 @@ OMX_ERRORTYPE vid_dec_LoaderComponent(stLoaderComponentType *comp)
|
||||
if (comp->name_specific[2] == NULL)
|
||||
goto error_specific;
|
||||
|
||||
comp->name_specific[3] = CALLOC(1, OMX_MAX_STRINGNAME_SIZE);
|
||||
if (comp->name_specific[3] == NULL)
|
||||
goto error_specific;
|
||||
|
||||
comp->role_specific[0] = CALLOC(1, OMX_MAX_STRINGNAME_SIZE);
|
||||
if (comp->role_specific[0] == NULL)
|
||||
goto error_specific;
|
||||
@@ -109,23 +113,31 @@ OMX_ERRORTYPE vid_dec_LoaderComponent(stLoaderComponentType *comp)
|
||||
if (comp->role_specific[2] == NULL)
|
||||
goto error_specific;
|
||||
|
||||
comp->role_specific[3] = CALLOC(1, OMX_MAX_STRINGNAME_SIZE);
|
||||
if (comp->role_specific[3] == NULL)
|
||||
goto error_specific;
|
||||
|
||||
strcpy(comp->name, OMX_VID_DEC_BASE_NAME);
|
||||
strcpy(comp->name_specific[0], OMX_VID_DEC_MPEG2_NAME);
|
||||
strcpy(comp->name_specific[1], OMX_VID_DEC_AVC_NAME);
|
||||
strcpy(comp->name_specific[2], OMX_VID_DEC_HEVC_NAME);
|
||||
strcpy(comp->name_specific[3], OMX_VID_DEC_AV1_NAME);
|
||||
|
||||
strcpy(comp->role_specific[0], OMX_VID_DEC_MPEG2_ROLE);
|
||||
strcpy(comp->role_specific[1], OMX_VID_DEC_AVC_ROLE);
|
||||
strcpy(comp->role_specific[2], OMX_VID_DEC_HEVC_ROLE);
|
||||
strcpy(comp->role_specific[3], OMX_VID_DEC_AV1_ROLE);
|
||||
|
||||
comp->constructor = vid_dec_Constructor;
|
||||
|
||||
return OMX_ErrorNone;
|
||||
|
||||
error_specific:
|
||||
FREE(comp->role_specific[3]);
|
||||
FREE(comp->role_specific[2]);
|
||||
FREE(comp->role_specific[1]);
|
||||
FREE(comp->role_specific[0]);
|
||||
FREE(comp->name_specific[3]);
|
||||
FREE(comp->name_specific[2]);
|
||||
FREE(comp->name_specific[1]);
|
||||
FREE(comp->name_specific[0]);
|
||||
@@ -168,7 +180,13 @@ static OMX_ERRORTYPE vid_dec_Constructor(OMX_COMPONENTTYPE *comp, OMX_STRING nam
|
||||
if (!strcmp(name, OMX_VID_DEC_HEVC_NAME))
|
||||
priv->profile = PIPE_VIDEO_PROFILE_HEVC_MAIN;
|
||||
|
||||
priv->BufferMgmtCallback = vid_dec_FrameDecoded;
|
||||
if (!strcmp(name, OMX_VID_DEC_AV1_NAME))
|
||||
priv->profile = PIPE_VIDEO_PROFILE_AV1_MAIN;
|
||||
|
||||
if (priv->profile == PIPE_VIDEO_PROFILE_AV1_MAIN)
|
||||
priv->BufferMgmtCallback = vid_dec_av1_FrameDecoded;
|
||||
else
|
||||
priv->BufferMgmtCallback = vid_dec_FrameDecoded;
|
||||
priv->messageHandler = vid_dec_MessageHandler;
|
||||
priv->destructor = vid_dec_Destructor;
|
||||
|
||||
@@ -221,8 +239,12 @@ static OMX_ERRORTYPE vid_dec_Constructor(OMX_COMPONENTTYPE *comp, OMX_STRING nam
|
||||
port->sPortParam.format.video.eCompressionFormat = OMX_VIDEO_CodingMPEG2;
|
||||
port->sVideoParam.eCompressionFormat = OMX_VIDEO_CodingMPEG2;
|
||||
port->Port_SendBufferFunction = vid_dec_DecodeBuffer;
|
||||
port->Port_FreeBuffer = vid_dec_FreeDecBuffer;
|
||||
if (priv->profile == PIPE_VIDEO_PROFILE_AV1_MAIN) {
|
||||
port->Port_AllocateBuffer = vid_dec_av1_AllocateInBuffer;
|
||||
port->Port_UseBuffer = vid_dec_av1_UseInBuffer;
|
||||
}
|
||||
|
||||
port->Port_FreeBuffer = vid_dec_FreeDecBuffer;
|
||||
port = (omx_base_video_PortType *)priv->ports[OMX_BASE_FILTER_OUTPUTPORT_INDEX];
|
||||
port->sPortParam.nBufferCountActual = 8;
|
||||
port->sPortParam.nBufferCountMin = 4;
|
||||
@@ -239,6 +261,9 @@ static OMX_ERRORTYPE vid_dec_Destructor(OMX_COMPONENTTYPE *comp)
|
||||
vid_dec_PrivateType* priv = comp->pComponentPrivate;
|
||||
int i;
|
||||
|
||||
if (priv->profile == PIPE_VIDEO_PROFILE_AV1_MAIN)
|
||||
vid_dec_av1_ReleaseTasks(priv);
|
||||
|
||||
if (priv->ports) {
|
||||
for (i = 0; i < priv->sPortTypesParam[OMX_PortDomainVideo].nPorts; ++i) {
|
||||
if(priv->ports[i])
|
||||
@@ -309,6 +334,8 @@ static OMX_ERRORTYPE vid_dec_SetParameter(OMX_HANDLETYPE handle, OMX_INDEXTYPE i
|
||||
priv->profile = PIPE_VIDEO_PROFILE_MPEG4_AVC_HIGH;
|
||||
} else if (!strcmp((char *)role->cRole, OMX_VID_DEC_HEVC_ROLE)) {
|
||||
priv->profile = PIPE_VIDEO_PROFILE_HEVC_MAIN;
|
||||
} else if (!strcmp((char *)role->cRole, OMX_VID_DEC_AV1_ROLE)) {
|
||||
priv->profile = PIPE_VIDEO_PROFILE_AV1_MAIN;
|
||||
} else {
|
||||
return OMX_ErrorBadParameter;
|
||||
}
|
||||
@@ -359,6 +386,8 @@ static OMX_ERRORTYPE vid_dec_GetParameter(OMX_HANDLETYPE handle, OMX_INDEXTYPE i
|
||||
strcpy((char *)role->cRole, OMX_VID_DEC_AVC_ROLE);
|
||||
else if (priv->profile == PIPE_VIDEO_PROFILE_HEVC_MAIN)
|
||||
strcpy((char *)role->cRole, OMX_VID_DEC_HEVC_ROLE);
|
||||
else if (priv->profile == PIPE_VIDEO_PROFILE_AV1_MAIN)
|
||||
strcpy((char *)role->cRole, OMX_VID_DEC_AV1_ROLE);
|
||||
|
||||
break;
|
||||
}
|
||||
@@ -406,6 +435,9 @@ static OMX_ERRORTYPE vid_dec_MessageHandler(OMX_COMPONENTTYPE* comp, internalReq
|
||||
vid_dec_h264_Init(priv);
|
||||
else if (priv->profile == PIPE_VIDEO_PROFILE_HEVC_MAIN)
|
||||
vid_dec_h265_Init(priv);
|
||||
else if (priv->profile == PIPE_VIDEO_PROFILE_AV1_MAIN) {
|
||||
vid_dec_av1_Init(priv);
|
||||
}
|
||||
|
||||
} else if ((msg->messageParam == OMX_StateLoaded) && (priv->state == OMX_StateIdle)) {
|
||||
if (priv->shadow) {
|
||||
@@ -469,7 +501,10 @@ static OMX_ERRORTYPE vid_dec_DecodeBuffer(omx_base_PortType *port, OMX_BUFFERHEA
|
||||
priv->in_buffers[0]->nFilledLen = priv->in_buffers[0]->nAllocLen;
|
||||
r = base_port_SendBufferFunction(port, priv->in_buffers[0]);
|
||||
} else if (eos) {
|
||||
vid_dec_FreeInputPortPrivate(priv->in_buffers[0]);
|
||||
if (priv->profile == PIPE_VIDEO_PROFILE_AV1_MAIN)
|
||||
vid_dec_av1_FreeInputPortPrivate(priv, priv->in_buffers[0]);
|
||||
else
|
||||
vid_dec_FreeInputPortPrivate(priv->in_buffers[0]);
|
||||
priv->in_buffers[0]->nFilledLen = priv->in_buffers[0]->nAllocLen;
|
||||
r = base_port_SendBufferFunction(port, priv->in_buffers[0]);
|
||||
} else {
|
||||
@@ -495,7 +530,14 @@ static OMX_ERRORTYPE vid_dec_DecodeBuffer(omx_base_PortType *port, OMX_BUFFERHEA
|
||||
|
||||
static OMX_ERRORTYPE vid_dec_FreeDecBuffer(omx_base_PortType *port, OMX_U32 idx, OMX_BUFFERHEADERTYPE *buf)
|
||||
{
|
||||
vid_dec_FreeInputPortPrivate(buf);
|
||||
OMX_COMPONENTTYPE* comp = port->standCompContainer;
|
||||
vid_dec_PrivateType *priv = comp->pComponentPrivate;
|
||||
|
||||
if (priv->profile == PIPE_VIDEO_PROFILE_AV1_MAIN)
|
||||
vid_dec_av1_FreeInputPortPrivate(priv, buf);
|
||||
else
|
||||
vid_dec_FreeInputPortPrivate(buf);
|
||||
|
||||
return base_port_FreeBuffer(port, idx, buf);
|
||||
}
|
||||
|
||||
|
@@ -54,6 +54,9 @@
|
||||
#define OMX_VID_DEC_HEVC_NAME "OMX.mesa.video_decoder.hevc"
|
||||
#define OMX_VID_DEC_HEVC_ROLE "video_decoder.hevc"
|
||||
|
||||
#define OMX_VID_DEC_AV1_NAME "OMX.mesa.video_decoder.av1"
|
||||
#define OMX_VID_DEC_AV1_ROLE "video_decoder.av1"
|
||||
|
||||
#define OMX_VID_DEC_TIMESTAMP_INVALID ((OMX_TICKS) -1)
|
||||
|
||||
OMX_ERRORTYPE vid_dec_LoaderComponent(stLoaderComponentType *comp);
|
||||
|
Reference in New Issue
Block a user