第一章 mpi_dec_multi_test.c 源码解析
int main(int argc, char **argv)
{
RK_S32 ret = 0;
MpiDecTestCmd cmd_ctx;
MpiDecTestCmd* cmd = &cmd_ctx; //命令行参数结构体
MpiDecMultiCtxInfo *ctxs = NULL; //多线程解码的上下文
RK_S32 i = 0; //循环计数器
float total_rate = 0.0; //所有线程解码帧率的总和
memset((void*)cmd, 0, sizeof(*cmd));
cmd->nthreads = 1;
// parse the cmd option
ret = mpi_dec_test_cmd_init(cmd, argc, argv); //解析命令行参数并进行初始化,如果出现错误则跳转到标签 RET 处进行释放资源操作
if (ret)
goto RET;
mpi_dec_test_cmd_options(cmd); //打印命令行参数信息
cmd->simple = (cmd->type != MPP_VIDEO_CodingMJPEG) ? (1) : (0); //根据视频编码类型设置 simple 变量的值
ctxs = mpp_calloc(MpiDecMultiCtxInfo, cmd->nthreads); //为多线程解码的上下文分配内存,如果分配失败则打印错误信息并返回 -1。
if (NULL == ctxs) {
mpp_err("failed to alloc context for instances\n");
return -1;
}
//使用循环为每个线程分配解码上下文并创建线程,如果创建线程失败则打印错误信息并返回。
for (i = 0; i < cmd->nthreads; i++) {
ctxs[i].cmd = cmd;
ret = pthread_create(&ctxs[i].thd, NULL, multi_dec_decode, &ctxs[i]);
if (ret) {
mpp_log("failed to create thread %d\n", i);
return ret;
}
}
//如果命令行参数中的帧数小于 0,则等待用户输入,之后设置所有线程的 loop_end 标志为 1,表示结束解码循环。
if (cmd->frame_num < 0) {
// wait for input then quit decoding
mpp_log("*******************************************\n");
mpp_log("**** Press Enter to stop loop decoding ****\n");
mpp_log("*******************************************\n");
getc(stdin);
for (i = 0; i < cmd->nthreads; i++)
ctxs[i].ctx.loop_end = 1;
}
//使用循环等待每个线程执行完成并回收线程资源。
for (i = 0; i < cmd->nthreads; i++)
pthread_join(ctxs[i].thd, NULL);
//在循环结束后,使用循环遍历每个线程的解码结果,打印出每个线程解码的帧数、耗时、延迟、帧率等信息,并计算出所有线程解码的平均帧率
for (i = 0; i < cmd->nthreads; i++) {
MpiDecMultiCtxRet *dec_ret = &ctxs[i].ret;
mpp_log("chn %2d decode %d frames time %lld ms delay %3d ms fps %3.2f\n", i,
dec_ret->frame_count, (RK_S64)(dec_ret->elapsed_time / 1000),
(RK_S32)(dec_ret->delay / 1000), dec_ret->frame_rate);
total_rate += dec_ret->frame_rate;
}
//最后使用 mpp_free 函数释放多线程解码的上下文内存,之后调用 mpi_dec_test_cmd_deinit 函数释放命令行参数结构体占用的资源。
mpp_free(ctxs);
ctxs = NULL;
total_rate /= cmd->nthreads;
mpp_log("average frame rate %.2f\n", total_rate);
RET:
mpi_dec_test_cmd_deinit(cmd);
return (int)total_rate;
}
void* multi_dec_decode(void *cmd_ctx)
{
MpiDecMultiCtxInfo *info = (MpiDecMultiCtxInfo *)cmd_ctx; //将cmd_ctx强制转换为MpiDecMultiCtxInfo类型
MpiDecMultiCtx *dec_ctx = &info->ctx; //解码器的上下文信息
MpiDecMultiCtxRet *rets = &info->ret; //返回的解码结果
MpiDecTestCmd *cmd = info->cmd;
MPP_RET ret = MPP_OK;
//接下来定义了一些变量和资源,包括视频帧的宽、高、编码类型等参数,
//还有用于输入和输出的MppPacket和MppFrame类型的变量,
//以及用于运行时配置的MppDecCfg类型的变量和是否需要分割输入帧的标志need_split。
// base flow context
MppCtx ctx = NULL;
MppApi *mpi = NULL;
// input / output
MppPacket packet = NULL;
MppFrame frame = NULL;
// config for runtime mode
MppDecCfg cfg = NULL;
RK_U32 need_split = 1;
// paramter for resource malloc
RK_U32 width = cmd->width;
RK_U32 height = cmd->height;
MppCodingType type = cmd->type;
// resources
MppBuffer frm_buf = NULL;
//如果需要输出解码后的视频帧,则打开文件并将文件指针保存在解码器上下文中。
if (cmd->have_output) {
dec_ctx->fp_output = fopen(cmd->file_output, "w+b");
if (NULL == dec_ctx->fp_output) {
mpp_err("failed to open output file %s\n", cmd->file_output);
goto MPP_TEST_OUT;
}
}
//根据参数simple的值,分别进行简单解码或高级解码
if (cmd->simple) {
ret = mpp_packet_init(&packet, NULL, 0); //初始化一个空的MppPacket
if (ret) {
mpp_err("mpp_packet_init failed\n");
goto MPP_TEST_OUT;
}
} else {
//需要先初始化MppFrame,然后创建一个解码器上下文,设置解码器参数,将解码器的相关信息保存在解码器上下文中
RK_U32 hor_stride = MPP_ALIGN(width, 16);
RK_U32 ver_stride = MPP_ALIGN(height, 16);
ret = mpp_buffer_group_get_internal(&dec_ctx->frm_grp, MPP_BUFFER_TYPE_ION);
if (ret) {
mpp_err("failed to get buffer group for input frame ret %d\n", ret);
goto MPP_TEST_OUT;
}
ret = mpp_frame_init(&frame); /* output frame */
if (ret) {
mpp_err("mpp_frame_init failed\n");
goto MPP_TEST_OUT;
}
/*
* NOTE: For jpeg could have YUV420 and YUV422 the buffer should be
* larger for output. And the buffer dimension should align to 16.
* YUV420 buffer is 3/2 times of w*h.
* YUV422 buffer is 2 times of w*h.
* So create larger buffer with 2 times w*h.
*/
ret = mpp_buffer_get(dec_ctx->frm_grp, &frm_buf, hor_stride * ver_stride * 2);
if (ret) {
mpp_err("failed to get buffer for input frame ret %d\n", ret);
goto MPP_TEST_OUT;
}
mpp_frame_set_buffer(frame, frm_buf);
}
// decoder demo
ret = mpp_create(&ctx, &mpi);
if (ret) {
mpp_err("mpp_create failed\n");
goto MPP_TEST_OUT;
}
mpp_log("%p mpi_dec_test decoder test start w %d h %d type %d\n",
ctx, width, height, type);
ret = mpp_init(ctx, MPP_CTX_DEC, type);
if (ret) {
mpp_err("mpp_init failed\n");
goto MPP_TEST_OUT;
}
mpp_dec_cfg_init(&cfg);
/* get default config from decoder context */
ret = mpi->control(ctx, MPP_DEC_GET_CFG, cfg);
if (ret) {
mpp_err("%p failed to get decoder cfg ret %d\n", ctx, ret);
goto MPP_TEST_OUT;
}
/*
* split_parse is to enable mpp internal frame spliter when the input
* packet is not aplited into frames.
*/
ret = mpp_dec_cfg_set_u32(cfg, "base:split_parse", need_split);
if (ret) {
mpp_err("%p failed to set split_parse ret %d\n", ctx, ret);
goto MPP_TEST_OUT;
}
ret = mpi->control(ctx, MPP_DEC_SET_CFG, cfg);
if (ret) {
mpp_err("%p failed to set cfg %p ret %d\n", ctx, cfg, ret);
goto MPP_TEST_OUT;
}
dec_ctx->cmd = cmd;
dec_ctx->ctx = ctx;
dec_ctx->mpi = mpi;
dec_ctx->packet = packet;
dec_ctx->frame = frame;
dec_ctx->packet_count = 0;
dec_ctx->frame_count = 0;
dec_ctx->frame_num = cmd->frame_num;
dec_ctx->quiet = cmd->quiet;
RK_S64 t_s, t_e;
t_s = mpp_time();
if (cmd->simple) {
while (!dec_ctx->loop_end) //通过一个循环读取输入文件中的数据并解码,直到loop_end标志被设置为true。
multi_dec_simple(dec_ctx);
} else {
while (!dec_ctx->loop_end)
multi_dec_advanced(dec_ctx); //读取输入文件中的数据并解码,直到loop_end标志被设置为true
}
//记录解码时间、解码帧数等解码结果,并在解码完成后进行资源的释放和重置。
t_e = mpp_time();
ret = mpi->reset(ctx);
if (ret) {
mpp_err("mpi->reset failed\n");
goto MPP_TEST_OUT;
}
rets->elapsed_time = t_e - t_s;
rets->frame_count = dec_ctx->frame_count;
rets->frame_rate = (float)dec_ctx->frame_count * 1000000 / rets->elapsed_time;
rets->delay = dec_ctx->first_frm - dec_ctx->first_pkt;
MPP_TEST_OUT:
if (packet) {
mpp_packet_deinit(&packet);
packet = NULL;
}
if (frame) {
mpp_frame_deinit(&frame);
frame = NULL;
}
if (ctx) {
mpp_destroy(ctx);
ctx = NULL;
}
if (!cmd->simple) {
if (frm_buf) {
mpp_buffer_put(frm_buf);
frm_buf = NULL;
}
}
if (dec_ctx->frm_grp) {
mpp_buffer_group_put(dec_ctx->frm_grp);
dec_ctx->frm_grp = NULL;
}
if (dec_ctx->fp_output) {
fclose(dec_ctx->fp_output);
dec_ctx->fp_output = NULL;
}
if (cfg) {
mpp_dec_cfg_deinit(cfg);
cfg = NULL;
}
return NULL;
}
//这个函数是 toybrick mpp 中的一个测试函数,用于对多路视频解码进行测试。具体分析如下:
static int multi_dec_advanced(MpiDecMultiCtx *data)
{
MPP_RET ret = MPP_OK;
MpiDecTestCmd *cmd = data->cmd;
MppCtx ctx = data->ctx;
MppApi *mpi = data->mpi;
MppPacket packet = NULL;
MppFrame frame = data->frame;
MppTask task = NULL;
RK_U32 quiet = data->quiet;
FileBufSlot *slot = NULL;
//调用 reader_index_read 函数从输入数据队列中读取一帧数据,该函数将读取的数据保存到 slot 变量中。
ret = reader_index_read(cmd->reader, 0, &slot);
mpp_assert(ret == MPP_OK);
mpp_assert(slot);
//将读取到的数据封装为 MppPacket 类型的 packet。
mpp_packet_init_with_buffer(&packet, slot->buf);
// setup eos flag
//如果读取到的数据是 end-of-stream 标记,则使用 mpp_packet_set_eos 函数将 packet 中的 end-of-stream 标记设置为 true。
if (slot->eos)
mpp_packet_set_eos(packet);
//使用 mpi->poll 函数等待解码器输入端口有空闲队列
ret = mpi->poll(ctx, MPP_PORT_INPUT, MPP_POLL_BLOCK);
if (ret) {
mpp_err("mpp input poll failed\n");
return ret;
}
//然后使用 mpi->dequeue 函数从解码器输入队列中取出一个输入任务 task。
ret = mpi->dequeue(ctx, MPP_PORT_INPUT, &task); /* input queue */
if (ret) {
mpp_err("mpp task input dequeue failed\n");
return ret;
}
mpp_assert(task);
//将 packet 和 frame 分别设置为 task 的输入和输出,并使用 mpi->enqueue 函数将任务 task 推送到解码器的输入队列中。
mpp_task_meta_set_packet(task, KEY_INPUT_PACKET, packet);
mpp_task_meta_set_frame (task, KEY_OUTPUT_FRAME, frame);
ret = mpi->enqueue(ctx, MPP_PORT_INPUT, task); /* input queue */
if (ret) {
mpp_err("mpp task input enqueue failed\n");
return ret;
}
if (!data->first_pkt)
data->first_pkt = mpp_time();
/* poll and wait here */
//使用 mpi->poll 函数等待解码器输出端口有解码后的数据输出,然后使用 mpi->dequeue 函数从解码器输出队列中取出一个输出任务 task。
ret = mpi->poll(ctx, MPP_PORT_OUTPUT, MPP_POLL_BLOCK);
if (ret) {
mpp_err("mpp output poll failed\n");
return ret;
}
ret = mpi->dequeue(ctx, MPP_PORT_OUTPUT, &task); /* output queue */ //然后使用 mpi->dequeue 函数从解码器输出队列中取出一个输出任务 task。
if (ret) {
mpp_err("mpp task output dequeue failed\n");
return ret;
}
mpp_assert(task);
if (task) {
MppFrame frame_out = NULL;
//将 task 中的输出帧 frame_out 赋值给变量 frame。
mpp_task_meta_get_frame(task, KEY_OUTPUT_FRAME, &frame_out);
//如果 frame 不为 NULL,说明解码器输出了一帧数据,计算帧率并将帧写入输出文件。
if (frame) {
if (!data->first_frm)
data->first_frm = mpp_time();///计算帧率
if (data->fp_output)
dump_mpp_frame_to_file(frame, data->fp_output);//将帧写入输出文件
mpp_log_q(quiet, "%p decoded frame %d\n", ctx, data->frame_count);
data->frame_count++;
if (mpp_frame_get_eos(frame_out)) { //如果帧中包含 end-of-stream 标记,则设置 loop_end 标记为 1。
mpp_log_q(quiet, "%p found eos frame\n", ctx);
}
fps_calc_inc(cmd->fps);
}
if (data->frame_num > 0) {
if (data->frame_count >= data->frame_num)
data->loop_end = 1;
} else if (data->frame_num == 0) {
if (slot->eos)
data->loop_end = 1;
}
/* output queue */
ret = mpi->enqueue(ctx, MPP_PORT_OUTPUT, task);//使用 mpi->enqueue 函数将任务 task 推送到解码器的输出队列中。
if (ret)
mpp_err("mpp task output enqueue failed\n");
}
/*
* The following input port task dequeue and enqueue is to make sure that
* the input packet can be released. We can directly deinit the input packet
* after frame output in most cases.
*/
//如果设置了宏 0,则直接释放 packet,否则从解码器输入队列中取出一个输入任务 task,将其中的输入 packet 释放,并将任务 task 推送回解码器输入队列。
if (0) {
mpp_packet_deinit(&packet);
} else {
ret = mpi->dequeue(ctx, MPP_PORT_INPUT, &task); /* input queue */ //从解码器输入队列中取出一个输入任务 task
if (ret) {
mpp_err("%p mpp task input dequeue failed\n", ctx);
return ret;
}
mpp_assert(task);
if (task) {
MppPacket packet_out = NULL;
mpp_task_meta_get_packet(task, KEY_INPUT_PACKET, &packet_out);
if (!packet_out || packet_out != packet)
mpp_err_f("mismatch packet %p -> %p\n", packet, packet_out);
mpp_packet_deinit(&packet_out);//将其中的输入 packet 释放
/* input empty task back to mpp to maintain task status */
ret = mpi->enqueue(ctx, MPP_PORT_INPUT, task);//并将任务 task 推送回解码器输入队列,
if (ret)
mpp_err("%p mpp task input enqueue failed\n", ctx);
}
}
//函数执行完成后,返回 ret 变量的值,表示执行过程中是否出现错误。
return ret;
}
这段代码是一个简单的多线程视频解码器的主要解码函数multi_dec_simple,其目的是从输入文件中读取视频数据包,将数据包送入解码器进行解码,并且获取解码器输出的视频帧。以下是对这段代码的详细分析:
函数的输入参数为MpiDecMultiCtx *data,包含了解码器上下文、解码器API等必要的信息。
在函数的第一步中,函数从输入结构体data中获取数据,包括cmd、pkt_done、pkt_eos、ctx、mpi、packet、reader、slot、quiet等参数。其中,cmd是命令结构体,pkt_done和pkt_eos是数据包的状态标记,ctx是解码器上下文,mpi是解码器API,packet是解码器使用的数据包,reader是读取输入文件的读取器,slot是文件数据读取的缓存区,quiet是日志输出的控制变量。
然后,使用读取器reader从输入文件中读取视频数据包,将读取到的数据包存储到packet中。同时,函数检查数据包是否已经全部读取完毕,如果读取完毕则根据是否满足帧数目标来判断是否要重新读取。如果没有读取完毕,则将pkt_eos标记设为0,表示还有数据包需要解码。
在循环中,首先检查pkt_done的状态,如果数据包没有被送入解码器进行解码,则将packet送入解码器进行解码。如果成功送入,将pkt_done标记设为1,记录首次读取到数据包的时间。
接下来,循环中使用decode_get_frame函数获取解码器解码后的视频帧。由于此函数是阻塞的,因此需要等待解码器解码出可用的视频帧。在获取到可用的视频帧后,函数进行错误处理,检查是否存在错误信息,并且记录首次获取到视频帧的时间。
如果解码器输出的视频帧带有信息变更标志,比如视频分辨率、帧率等发生了变化,函数需要重新设置解码器的输出缓存区,以便解码器可以正确地存储解码后的视频帧。函数首先通过buffer_group_get_internal函数获取解码器输出缓存区的内存空间,然后使用MPP_DEC_SET_EXT_BUF_GROUP命令设置解码器输出缓存区。之后,使用mpp_buffer_group_limit_config函数来限制解码器输出缓存区的大小,保证其能够存储24帧大小的视频帧。
如果解码器输出的视频帧不带有信息变更标志,则说明解码器已经正确设置了输出缓存区,并且解码器已经正确解码了视频帧。函数通过mpp_frame_get_errinfo和mpp_frame_get_discard函数来获取解码后视频帧中的错误信息和丢弃信息,并将其记录在日志中
static int multi_dec_simple(MpiDecMultiCtx *data)
{
//函数从输入结构体data中获取数据,包括 cmd、pkt_done、pkt_eos、ctx、mpi、packet、reader、slot、quiet等参数
MpiDecTestCmd *cmd = data->cmd; //cmd是命令结构体
RK_U32 pkt_done = 0; //pkt_done和pkt_eos是数据包的状态标记
RK_U32 pkt_eos = 0;
MppCtx ctx = data->ctx; //ctx是解码器上下文
MppApi *mpi = data->mpi; //mpi是解码器API
MppPacket packet = data->packet; //packet是解码器使用的数据包
FileReader reader = cmd->reader; //reader是读取输入文件的读取器
FileBufSlot *slot = NULL; //slot是文件数据读取的缓存区
RK_U32 quiet = data->quiet; //quiet是日志输出的控制变量。
//使用读取器reader从输入文件中读取视频数据包,,将读取到的数据包存储到packet中
MPP_RET ret = reader_index_read(reader, data->packet_count++, &slot);
mpp_assert(ret == MPP_OK);
mpp_assert(slot);
pkt_eos = slot->eos;
//函数检查数据包是否已经全部读取完毕
if (pkt_eos) {
//如果读取完毕则根据是否满足帧数目标来判断是否要重新读取
if (data->frame_num < 0 || data->frame_num > data->frame_count) {
mpp_log_q(quiet, "%p loop again\n", ctx);
data->packet_count = 0;//如果没有读取完毕,则将pkt_eos标记设为0,表示还有数据包需要解码。
pkt_eos = 0;
} else {
mpp_log_q(quiet, "%p found last packet\n", ctx);
data->loop_end = 1;
}
}
mpp_packet_set_data(packet, slot->data);
mpp_packet_set_size(packet, slot->size);
mpp_packet_set_pos(packet, slot->data);
mpp_packet_set_length(packet, slot->size);
// setup eos flag
if (pkt_eos)
mpp_packet_set_eos(packet);
do {
RK_U32 frm_eos = 0;
RK_S32 times = 5;
// send the packet first if packet is not done
//首先检查pkt_done的状态,如果数据包没有被送入解码器进行解码,则将packet送入解码器进行解码。如果成功送入,将pkt_done标记设为1,记录首次读取到数据包的时间。
if (!pkt_done) {
ret = mpi->decode_put_packet(ctx, packet);//将packet送入解码器进行解码
if (MPP_OK == ret) {
pkt_done = 1;//如果成功送入,将pkt_done标记设为1
if (!data->first_pkt)
data->first_pkt = mpp_time();//记录首次读取到数据包的时间
}
}
// then get all available frame and release
do {
RK_S32 get_frm = 0;
MppFrame frame = NULL;
try_again:
//使用decode_get_frame函数获取解码器解码后的视频帧
//由于此函数是阻塞的,因此需要等待解码器解码出可用的视频帧
ret = mpi->decode_get_frame(ctx, &frame);
if (MPP_ERR_TIMEOUT == ret) {
if (times > 0) {
times--;
msleep(2);
goto try_again;
}
mpp_err("decode_get_frame failed too much time\n");
}
if (ret) {
mpp_err("decode_get_frame failed ret %d\n", ret);
break;
}
if (frame) {
//如果解码器输出的视频帧带有信息变更标志,比如视频分辨率、帧率等发生了变化,函数需要重新设置解码器的输出缓存区,以便解码器可以正确地存储解码后的视频帧
if (mpp_frame_get_info_change(frame)) {
RK_U32 width = mpp_frame_get_width(frame);
RK_U32 height = mpp_frame_get_height(frame);
RK_U32 hor_stride = mpp_frame_get_hor_stride(frame);
RK_U32 ver_stride = mpp_frame_get_ver_stride(frame);
RK_U32 buf_size = mpp_frame_get_buf_size(frame);
mpp_log_q(quiet, "decode_get_frame get info changed found\n");
mpp_log_q(quiet, "decoder require buffer w:h [%d:%d] stride [%d:%d] buf_size %d",
width, height, hor_stride, ver_stride, buf_size);
if (NULL == data->frm_grp) {
/* If buffer group is not set create one and limit it */
//通过buffer_group_get_internal函数获取解码器输出缓存区的内存空间
ret = mpp_buffer_group_get_internal(&data->frm_grp, MPP_BUFFER_TYPE_ION);
if (ret) {
mpp_err("get mpp buffer group failed ret %d\n", ret);
break;
}
/* Set buffer to mpp decoder */
//使用MPP_DEC_SET_EXT_BUF_GROUP命令设置解码器输出缓存区
ret = mpi->control(ctx, MPP_DEC_SET_EXT_BUF_GROUP, data->frm_grp);
if (ret) {
mpp_err("set buffer group failed ret %d\n", ret);
break;
}
} else {
/* If old buffer group exist clear it */
ret = mpp_buffer_group_clear(data->frm_grp);
if (ret) {
mpp_err("clear buffer group failed ret %d\n", ret);
break;
}
}
/* Use limit config to limit buffer count to 24 with buf_size */
//使用mpp_buffer_group_limit_config函数来限制解码器输出缓存区的大小,保证其能够存储24帧大小的视频帧。
ret = mpp_buffer_group_limit_config(data->frm_grp, buf_size, 24);
if (ret) {
mpp_err("limit buffer group failed ret %d\n", ret);
break;
}
/*
* All buffer group config done. Set info change ready to let
* decoder continue decoding
*/
ret = mpi->control(ctx, MPP_DEC_SET_INFO_CHANGE_READY, NULL);
if (ret) {
mpp_err("info change ready failed ret %d\n", ret);
break;
}
} else {
//如果解码器输出的视频帧不带有信息变更标志,则说明解码器已经正确设置了输出缓存区,并且解码器已经正确解码了视频帧。
//函数通过mpp_frame_get_errinfo和mpp_frame_get_discard函数来获取解码后视频帧中的错误信息和丢弃信息,并将其记录在日志中
RK_U32 err_info = 0;
//在获取到可用的视频帧后,函数进行错误处理,检查是否存在错误信息,并且记录首次获取到视频帧的时间。
if (!data->first_frm)
data->first_frm = mpp_time();
err_info = mpp_frame_get_errinfo(frame) |
mpp_frame_get_discard(frame);
if (err_info) {
mpp_log_q(quiet, "decoder_get_frame get err info:%d discard:%d.\n",
mpp_frame_get_errinfo(frame),
mpp_frame_get_discard(frame));
}
mpp_log_q(quiet, "decode_get_frame get frame %d\n",
data->frame_count);
data->frame_count++;
if (data->fp_output && !err_info)
dump_mpp_frame_to_file(frame, data->fp_output);
fps_calc_inc(cmd->fps);
}
frm_eos = mpp_frame_get_eos(frame);
mpp_frame_deinit(&frame);
get_frm = 1;
}
// if last packet is send but last frame is not found continue
if (pkt_eos && pkt_done && !frm_eos) {
msleep(1);
continue;
}
if ((data->frame_num > 0 && (data->frame_count >= data->frame_num)) ||
((data->frame_num == 0) && frm_eos))
break;
if (get_frm)
continue;
break;
} while (1);
if ((data->frame_num > 0 && (data->frame_count >= data->frame_num)) ||
((data->frame_num == 0) && frm_eos)) {
data->loop_end = 1;
break;
}
if (pkt_done)
break;
/*
* why sleep here:
* mpi->decode_put_packet will failed when packet in internal queue is
* full,waiting the package is consumed .Usually hardware decode one
* frame which resolution is 1080p needs 2 ms,so here we sleep 3ms
* * is enough.
*/
msleep(1);
} while (1);
return ret;
}
MpiDecMultiCtx 结构体定义了以下字段:
cmd:指向全局命令行信息的指针。
ctx:mpp 上下文。
mpi:mpp 库的 API。
loop_end:循环结束标志。
frm_grp:帧组的缓冲区。
packet:码流数据包。
frame:解码后的帧数据。
fp_output:输出文件指针。
packet_count:码流数据包计数器。
frame_count:解码后的帧计数器。
frame_num:需要解码的总帧数。
first_pkt:第一个码流数据包的时间戳。
first_frm:第一帧的时间戳。
quiet:是否安静模式。
MpiDecMultiCtxRet 结构体定义了以下字段:
frame_rate:解码帧率。
elapsed_time:总解码时间。
frame_count:解码帧数。
delay:解码延迟。
MpiDecMultiCtxInfo 结构体定义了以下字段:
cmd:指向全局命令行信息的指针。
thd:每个实例的线程。
ctx:解码器的上下文。
ret:解码器的返回值。
multi_dec_simple 函数是使用 mpp 库进行多线程解码的入口函数,其参数为命令行信息和需要解码的文件列表。其主要逻辑如下:
定义变量 instance_count 为文件列表长度,并分配对应数量的 MpiDecMultiCtxInfo 结构体实例。
为每个实例分别创建一个解码器线程,并将其保存在 thd 字段中。
等待所有线程的结束,统计解码结果。
/* For each instance thread setup */
typedef struct {
MpiDecTestCmd *cmd;
MppCtx ctx;
MppApi *mpi;
/* end of stream flag when set quit the loop */
RK_U32 loop_end;
/* input and output */
MppBufferGroup frm_grp;
MppPacket packet;
MppFrame frame;
FILE *fp_output;
RK_S32 packet_count;
RK_S32 frame_count;
RK_S32 frame_num;
RK_S64 first_pkt;
RK_S64 first_frm;
/* runtime flag */
RK_U32 quiet;
} MpiDecMultiCtx;
/* For each instance thread return value */
typedef struct {
float frame_rate;
RK_S64 elapsed_time;
RK_S32 frame_count;
RK_S64 delay;
} MpiDecMultiCtxRet;
typedef struct {
MpiDecTestCmd *cmd; // pointer to global command line info
pthread_t thd; // thread for for each instance
MpiDecMultiCtx ctx; // context of decoder
MpiDecMultiCtxRet ret; // return of decoder
} MpiDecMultiCtxInfo;