Files
luban-lite-t3e-pro/packages/artinchip/mpp/middle_media/openmax/demo/adec2ao.c
刘可亮 7bbc029dae v1.0.0
2023-08-30 16:21:18 +08:00

403 lines
9.9 KiB
C

/*
* Copyright (C) 2020-2023 ArtInChip Technology Co. Ltd
*
* author: <jun.ma@artinchip.com>
* Desc: OMX_AdecComponent Tunneld OMX_AudidoRendercComponent demo
*/
#include <string.h>
#include <malloc.h>
#include <stddef.h>
#include <stdlib.h>
#include <stdlib.h>
#include <stdio.h>
#include <fcntl.h>
#include <unistd.h>
#include <sys/types.h>
#include "OMX_Core.h"
#include "OMX_CoreExt1.h"
#include "mpp_dec_type.h"
#include "mpp_list.h"
#include "mpp_log.h"
#include "mpp_mem.h"
#include "aic_message.h"
#include "bit_stream_parser.h"
#define ADEC_EventCmdComplete 1
#define ADEC_FILL_BUFFER_DONE 2
#define ADEC_EVENT_ERROR 3
#define ADEC_EventBufferFlag 4
#define AUDIO_RENDER_EventBufferFlag 5
struct packet_node {
struct OMX_BUFFERHEADERTYPE sBuff;
struct mpp_list sList;
};
#define PKT_MAX_NUM 16
static struct aic_message_queue g_msg_que;
static OMX_ERRORTYPE adec_event_handler (
OMX_HANDLETYPE hComponent,
OMX_PTR pAppData,
OMX_EVENTTYPE eEvent,
OMX_U32 Data1,
OMX_U32 Data2,
OMX_PTR pEventData)
{
OMX_ERRORTYPE eError = OMX_ErrorNone;
struct aic_message msg;
if(eEvent == OMX_EventCmdComplete){
logw("adec state set to %u\n", Data2);
msg.message_id = ADEC_EventCmdComplete;
msg.data_size = 0;
aic_msg_put(&g_msg_que,&msg);
}else if(eEvent == OMX_EventBufferFlag){
logw("\nEOS ,0x%x\n",Data1);
msg.message_id = ADEC_EventBufferFlag;
msg.data_size = 0;
aic_msg_put(&g_msg_que,&msg);
}else if(eEvent == ADEC_EVENT_ERROR){
logw("\nADEC_EVENT_ERROR ,0x%x,0x%x\n",Data1,Data2);
}else{
logw("param1 = 0x%x param2 = %d\n", (int)Data1, (int)Data2);
}
return eError;
}
static OMX_ERRORTYPE adec_empty_buffer_done (
OMX_HANDLETYPE hComponent,
OMX_PTR pAppData,
OMX_BUFFERHEADERTYPE* pBuffer)
{
OMX_ERRORTYPE eError = OMX_ErrorNone;
return eError;
}
static OMX_ERRORTYPE adec_fill_buffer_done (
OMX_HANDLETYPE hComponent,
OMX_PTR pAppData,
OMX_BUFFERHEADERTYPE* pBuffer)
{
OMX_ERRORTYPE eError = OMX_ErrorNone;
return eError;
}
OMX_CALLBACKTYPE adec_callbacks = {
.EventHandler = adec_event_handler,
//.EmptyBufferDone = adec_empty_buffer_done,
.EmptyBufferDone = NULL,
//.FillBufferDone = adec_fill_buffer_done
.FillBufferDone = NULL
};
static OMX_ERRORTYPE audio_render_event_handler (
OMX_HANDLETYPE hComponent,
OMX_PTR pAppData,
OMX_EVENTTYPE eEvent,
OMX_U32 Data1,
OMX_U32 Data2,
OMX_PTR pEventData)
{
OMX_ERRORTYPE eError = OMX_ErrorNone;
struct aic_message msg;
if(eEvent == OMX_EventCmdComplete){
logw("adec state set to %u\n", Data2);
msg.message_id = ADEC_EventCmdComplete;
msg.data_size = 0;
aic_msg_put(&g_msg_que,&msg);
}else if(eEvent == OMX_EventBufferFlag){
logw("\nEOS ,0x%x\n",Data1);
msg.message_id = AUDIO_RENDER_EventBufferFlag;
msg.data_size = 0;
aic_msg_put(&g_msg_que,&msg);
}else if(eEvent == ADEC_EVENT_ERROR){
logw("\nADEC_EVENT_ERROR ,0x%x,0x%x\n",Data1,Data2);
}else{
logw("param1 = 0x%x param2 = %d\n", (int)Data1, (int)Data2);
}
return eError;
}
OMX_CALLBACKTYPE audio_render_callbacks = {
.EventHandler = audio_render_event_handler,
.EmptyBufferDone = NULL,
.FillBufferDone = NULL
};
int main(int argc,char **argv)
{
OMX_ERRORTYPE eError = OMX_ErrorNone;
OMX_BUFFERHEADERTYPE buff_head;
OMX_BUFFERHEADERTYPE *p_buff_head;
OMX_AUDIO_PARAM_PORTFORMATTYPE port_format;
OMX_HANDLETYPE adecoder;
OMX_HANDLETYPE audio_render;
struct packet_node *pkt_node;
struct mpp_packet pkt;
s32 pre_pkt_szie;
int i;
struct aic_message msg;
int file_fd;
int ret = 0;
int bs_eof_flag = 0;
struct mpp_frame *pFrameInfo;
OMX_STATETYPE state = OMX_StateInvalid;
int try_times;
int adec_stream_end;
int audio_render_stream_end;
int file_size = 0;
int read_len = 0;
int read_sum_len = 0;
int need_read = 1;
if(argc < 2){
loge("param error!!!\n");
return -1;
}
memset(&pkt,0x00,sizeof(struct mpp_packet));
pre_pkt_szie = 0;
if(aic_msg_create(&g_msg_que) != 0){
loge("aic_msg_create error!!!\n");
goto _EXIT0;
}
logi("OMX_init!!!\n");
eError = OMX_Init();
if(eError != OMX_ErrorNone){
loge("OMX_init error!!!\n");
goto _EXIT1;
}
// 1 get adecoder audio_render
logi("OMX_GetHandle!!!\n");
if (OMX_ErrorNone !=OMX_GetHandle(&adecoder, OMX_COMPONENT_ADEC_NAME,NULL, &adec_callbacks)){
loge("unable to get adecoder handle.\n");
goto _EXIT2;
}
if (OMX_ErrorNone !=OMX_GetHandle(&audio_render, OMX_COMPONENT_AUDIO_RENDER_NAME,NULL, &audio_render_callbacks)){
loge("unable to get audio_render handle.\n");
goto _EXIT3;
}
// 2 set up tunnel adecoder.out_port=1 ----> audio_render.in_port=0
if (OMX_ErrorNone !=OMX_SetupTunnel(adecoder, ADEC_PORT_OUT_INDEX, audio_render, AUDIO_RENDER_PORT_IN_AUDIO_INDEX)){
loge("unable to get audio_render handle.\n");
goto _EXIT4;
}
//3 start audio_render
logi(" audio_render OMX_SendCommand OMX_StateIdle!!!\n");
if (OMX_ErrorNone !=OMX_SendCommand(audio_render, OMX_CommandStateSet, OMX_StateIdle, NULL)){
loge("unable to set adec to OMX_StateIdle.\n");
goto _EXIT4;
}
do{
OMX_GetState(audio_render, &state);
usleep(1000);
}while(state != OMX_StateIdle);
logi("audio_render state: OMX_StateLoaded-->OMX_StateIdle!!!\n");
if (OMX_ErrorNone !=OMX_SendCommand(audio_render, OMX_CommandStateSet, OMX_StateExecuting, NULL)){
loge("unable to set adec to OMX_StateExecuting.\n");
goto _EXIT4;
}
do{
OMX_GetState(audio_render, &state);
usleep(1000);
}while(state != OMX_StateExecuting);
logi("audio_render state: OMX_StateIdle-->OMX_StateExecuting!!!\n");
// 4 start adecoder
logi("OMX_SetParameter!!!\n");
port_format.nPortIndex = 0;
port_format.eEncoding = OMX_AUDIO_CodingMP3;
if(OMX_ErrorNone != OMX_SetParameter(adecoder, OMX_IndexParamAudioPortFormat,&port_format)){
loge("OMX_SetParameter Error!!!!.\n");
goto _EXIT4;
}
logi("OMX_SendCommand OMX_StateIdle!!!\n");
if (OMX_ErrorNone !=OMX_SendCommand(adecoder, OMX_CommandStateSet, OMX_StateIdle, NULL)){
loge("unable to set adec to OMX_StateIdle.\n");
goto _EXIT4;
}
do{
OMX_GetState(adecoder, &state);
usleep(1000);
}while(state != OMX_StateIdle);
logi("adecdoer state: OMX_StateLoaded-->OMX_StateIdle!!!\n");
if (OMX_ErrorNone !=OMX_SendCommand(adecoder, OMX_CommandStateSet, OMX_StateExecuting, NULL)){
loge("unable to set adec to OMX_StateExecuting.\n");
goto _EXIT4;
}
do{
OMX_GetState(adecoder, &state);
usleep(1000);
}while(state != OMX_StateExecuting);
logi("adecdoer state: OMX_StateIdle-->OMX_StateExecuting!!!\n");
// 5 open stream
file_fd = open(argv[1], O_RDONLY);
if (file_fd < 0) {
loge("failed to open input file %s", argv[1]);
goto _EXIT4;
}
file_size = lseek(file_fd, 0, SEEK_END);
lseek(file_fd, 0, SEEK_SET);
// 6 read stream to adec and process msg from adecoder and audio_render
while(1){
if(aic_msg_get(&g_msg_que, &msg) == 0){
if(msg.message_id == ADEC_EventCmdComplete){
logi("receive ADEC_EventCmdComplete \n");
}else if(msg.message_id == ADEC_EventBufferFlag){ //end fo stream stop
logi("receive end of stream from decoder!!!!\n");
adec_stream_end = 1;
if(adec_stream_end==1 && audio_render_stream_end==1){
logi("receive end of stream from audio_render and adec!!!!\n");
break;
}
}else if(msg.message_id == AUDIO_RENDER_EventBufferFlag){
logi("receive end of stream from audio_render!!!!\n");
audio_render_stream_end = 1;
if(adec_stream_end==1 && audio_render_stream_end==1){
logi("receive end of stream from audio_render and adec!!!!\n");
break;
}
}else{
loge("unknown event");
}
}
#define PKT_SIZE (2*1024)
if(!bs_eof_flag){
if(pkt.size == 0 || pkt.data == NULL){
pkt.size = PKT_SIZE;
pkt.data = mpp_alloc(pkt.size);
if(pkt.data == NULL){
loge("mpp_alloc err!\n");
break;
}
}
if(need_read){
read_len = read(file_fd, pkt.data, pkt.size);
if(read_len > 0){
pkt.size = read_len;
read_sum_len += read_len;
if(read_sum_len >= file_size){
pkt.flag |= PACKET_FLAG_EOS;
}
}else if(read_len == 0){
pkt.size = read_len;
pkt.flag |= PACKET_FLAG_EOS;
logi("end of stream\n");
}else{
logd("read error!!!\n");
// need to do
break;
}
}
buff_head.nFilledLen = pkt.size;
buff_head.pBuffer = pkt.data;
buff_head.nFlags = pkt.flag;
buff_head.nTimeStamp = pkt.pts;
if(OMX_EmptyThisBuffer(adecoder,&buff_head) == OMX_ErrorNone){
need_read = 1;//ok,next need to read
if(pkt.flag & PACKET_FLAG_EOS){
bs_eof_flag = 1;
logi("end of stream\n");
}
}else{
need_read = 0;//fail,next do not need to read
}
}
}
// 7 stop audio render must first stop
OMX_SendCommand(audio_render, OMX_CommandStateSet, OMX_StateIdle, NULL);
do{
OMX_GetState(audio_render, &state);
usleep(1000);
}while(state != OMX_StateIdle);
logi("audio_render state: OMX_StateExecuting-->OMX_StateIdle!!!\n");
OMX_SendCommand(audio_render, OMX_CommandStateSet, OMX_StateLoaded, NULL);
do{
OMX_GetState(audio_render, &state);
usleep(1000);
}while(state != OMX_StateLoaded);
logi("audio_render state: OMX_StateIdle-->OMX_StateLoaded!!!\n");
//8 stop adecoder
OMX_SendCommand(adecoder, OMX_CommandStateSet, OMX_StateIdle, NULL);
do{
OMX_GetState(adecoder, &state);
usleep(1000);
}while(state != OMX_StateIdle);
logi("adecoder state: OMX_StateExecuting-->OMX_StateIdle!!!\n");
OMX_SendCommand(adecoder, OMX_CommandStateSet, OMX_StateLoaded, NULL);
do{
OMX_GetState(adecoder, &state);
usleep(1000);
}while(state != OMX_StateLoaded);
logi("adecoder state: OMX_StateIdle-->OMX_StateLoaded!!!\n");
OMX_SetupTunnel(adecoder,ADEC_PORT_OUT_INDEX,NULL,0);
OMX_SetupTunnel(NULL,0,audio_render,AUDIO_RENDER_PORT_IN_AUDIO_INDEX);
logi("adecoder and audio_render cancel setup OMX_SetupTunnel!!!\n");
close(file_fd);
_EXIT4:
if(audio_render){
OMX_FreeHandle(audio_render);
audio_render = NULL;
}
_EXIT3:
if(adecoder){
OMX_FreeHandle(adecoder);
adecoder = NULL;
}
_EXIT2:
OMX_Deinit();
_EXIT1:
aic_msg_destroy(&g_msg_que);
_EXIT0:
logi("adec2ao_test exitl!!!\n");
return (int )eError;
}