MPP-解码示例
提取出编码的代码后,现在提取解码例程,供以后需要的时候使用。
完整的解码代码如下,做了一些改动,指令参数全部去除,输入H264数据,解码后的数据保存在本地,官方示例解码后的数据是YUV_420sp和YUV_422sp格式,这里将其改为了YUV_420p格式。
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 116 117 118 119 120 121 122 123 124 125 126 127 128 129 130 131 132 133 134 135 136 137 138 139 140 141 142 143 144 145 146 147 148 149 150 151 152 153 154 155 156 157 158 159 160 161 162 163 164 165 166 167 168 169 170 171 172 173 174 175 176 177 178 179 180 181 182 183 184 185 186 187 188 189 190 191 192 193 194 195 196 197 198 199 200 201 202 203 204 205 206 207 208 209 210 211 212 213 214 215 216 217 218 219 220 221 222 223 224 225 226 227 228 229 230 231 232 233 234 235 236 237 238 239 240 241 242 243 244 245 246 247 248 249 250 251 252 253 254 255 256 257 258 259 260 261 262 263 264 265 266 267 268 269 270 271 272 273 274 275 276 277 278 279 280 281 282 283 284 285 286 287 288 289 290 291 292 293 294 295 296 297 298 299 300 301 302 303 304 305 306 307 308 309 310 311 312 313 314 315 316 317 318 319 320 321 322 323 324 325 326 327 328 329 330 331 332 333 334 335 336 337 338 339 340 341 342 343 344 345 346 347 348 349 350 351 352 353 354 355 356 357 358 359 360 361 362 363 364 365 366 367 368 369 370 371 372 373 374 375 376 377 378 379 380 381 382 383 384 385 386 387 388 389 390 391 392 393 394 395 396 397 398 399 400 401 402 403 404 405 406 407 408 409 410 411 412 413 414 415 416 417 418 419 420 421 422 423 424 425 426 427 428 429 430 431 432 433 434 435 436 437 438 439 440 441 442 443 444 445 446 447 448 449 450 451 452 453 454 455 456 457 458 459 460 461 462 463 464 | #if defined(_WIN32) #include "vld.h" #endif #define MODULE_TAG "mpi_dec_test" #include <string.h> #include "utils.h" #include "rk_mpi.h" #include "mpp_log.h" #include "mpp_mem.h" #include "mpp_env.h" #include "mpp_time.h" #include "mpp_common.h" #include "mpp_frame.h" #include "mpp_buffer_impl.h" #include "mpp_frame_impl.h" #define MPI_DEC_STREAM_SIZE (SZ_4K) #define MPI_DEC_LOOP_COUNT 4 #define MAX_FILE_NAME_LENGTH 256 typedef struct { MppCtx ctx; MppApi *mpi; RK_U32 eos; char *buf; MppBufferGroup frm_grp; MppBufferGroup pkt_grp; MppPacket packet; size_t packet_size; MppFrame frame; FILE *fp_input; FILE *fp_output; RK_S32 frame_count; RK_S32 frame_num; size_t max_usage; } MpiDecLoopData; typedef struct { char file_input[MAX_FILE_NAME_LENGTH]; char file_output[MAX_FILE_NAME_LENGTH]; MppCodingType type; MppFrameFormat format; RK_U32 width; RK_U32 height; RK_U32 debug; RK_U32 have_input; RK_U32 have_output; RK_U32 simple; RK_S32 timeout; RK_S32 frame_num; size_t max_usage; } MpiDecTestCmd; size_t mpp_frame_get_buf_size( const MppFrame s) { check_is_mpp_frame((MppFrameImpl*)s); return ((MppFrameImpl*)s)->buf_size; } void dump_mpp_frame_to_file(MppFrame frame, FILE *fp) { RK_U32 width = 0; RK_U32 height = 0; RK_U32 h_stride = 0; RK_U32 v_stride = 0; MppBuffer buffer = NULL; RK_U8 *base = NULL; width = mpp_frame_get_width(frame); height = mpp_frame_get_height(frame); h_stride = mpp_frame_get_hor_stride(frame); v_stride = mpp_frame_get_ver_stride(frame); buffer = mpp_frame_get_buffer(frame); base = (RK_U8 *)mpp_buffer_get_ptr(buffer); RK_U32 buf_size = mpp_frame_get_buf_size(frame); size_t base_length = mpp_buffer_get_size(buffer); mpp_log( "base_length = %d\n" ,base_length); RK_U32 i; RK_U8 *base_y = base; RK_U8 *base_c = base + h_stride * v_stride; //保存为YUV420sp格式 /*for (i = 0; i < height; i++, base_y += h_stride) { fwrite(base_y, 1, width, fp); } for (i = 0; i < height / 2; i++, base_c += h_stride) { fwrite(base_c, 1, width, fp); }*/ //保存为YUV420p格式 for (i = 0; i < height; i++, base_y += h_stride) { fwrite (base_y, 1, width, fp); } for (i = 0; i < height * width / 2; i+=2) { fwrite ((base_c + i), 1, 1, fp); } for (i = 1; i < height * width / 2; i+=2) { fwrite ((base_c + i), 1, 1, fp); } } size_t mpp_buffer_group_usage(MppBufferGroup group) { if (NULL == group) { mpp_err_f( "input invalid group %p\n" , group); return MPP_BUFFER_MODE_BUTT; } MppBufferGroupImpl *p = (MppBufferGroupImpl *)group; return p->usage; } static int decode_simple(MpiDecLoopData *data) { RK_U32 pkt_done = 0; RK_U32 pkt_eos = 0; RK_U32 err_info = 0; MPP_RET ret = MPP_OK; MppCtx ctx = data->ctx; MppApi *mpi = data->mpi; char *buf = data->buf; MppPacket packet = data->packet; MppFrame frame = NULL; size_t read_size = fread (buf, 1, data->packet_size, data->fp_input); if (read_size != data->packet_size || feof (data->fp_input)) { mpp_log( "found last packet\n" ); data->eos = pkt_eos = 1; } mpp_packet_write(packet, 0, buf, read_size); mpp_packet_set_pos(packet, buf); mpp_packet_set_length(packet, read_size); if (pkt_eos) { mpp_packet_set_eos(packet); } do { if (!pkt_done) { ret = mpi->decode_put_packet(ctx, packet); if (MPP_OK == ret) { pkt_done = 1; } } do { RK_S32 get_frm = 0; RK_U32 frm_eos = 0; ret = mpi->decode_get_frame(ctx, &frame); 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( "decode_get_frame get info changed found\n" ); mpp_log( "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) { 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 ; } 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 { ret = mpp_buffer_group_clear(data->frm_grp); if (ret) { mpp_err( "clear buffer group failed ret %d\n" , ret); break ; } } 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 ; } 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 { err_info = mpp_frame_get_errinfo(frame) | mpp_frame_get_discard(frame); if (err_info) { mpp_log( "decoder_get_frame get err info:%d discard:%d.\n" , mpp_frame_get_errinfo(frame), mpp_frame_get_discard(frame)); } data->frame_count++; mpp_log( "decode_get_frame get frame %d\n" , data->frame_count); if (data->fp_output && !err_info) { dump_mpp_frame_to_file(frame, data->fp_output); } } frm_eos = mpp_frame_get_eos(frame); mpp_frame_deinit(&frame); frame = NULL; get_frm = 1; } if (data->frm_grp) { size_t usage = mpp_buffer_group_usage(data->frm_grp); if (usage > data->max_usage) { data->max_usage = usage; } } if (pkt_eos && pkt_done && !frm_eos) { msleep(10); continue ; } if (frm_eos) { mpp_log( "found last frame\n" ); break ; } if (data->frame_num && data->frame_count >= data->frame_num) { data->eos = 1; break ; } if (get_frm) { continue ; } break ; } while (1); if (data->frame_num && data->frame_count >= data->frame_num) { data->eos = 1; mpp_log( "reach max frame number %d\n" , data->frame_count); break ; } if (pkt_done) { break ; } } while (1); return ret; } int mpi_dec_test_decode(MpiDecTestCmd *cmd) { MPP_RET ret = MPP_OK; size_t file_size = 0; MppCtx ctx = NULL; MppApi *mpi = NULL; MppPacket packet = NULL; MppFrame frame = NULL; MpiCmd mpi_cmd = MPP_CMD_BASE; MppParam param = NULL; RK_U32 need_split = 1; RK_U32 width = cmd->width; RK_U32 height = cmd->height; MppCodingType type = cmd->type; char *buf = NULL; size_t packet_size = MPI_DEC_STREAM_SIZE; MppBuffer pkt_buf = NULL; MppBuffer frm_buf = NULL; MpiDecLoopData data; mpp_log( "mpi_dec_test start\n" ); memset (&data, 0, sizeof (data)); data.fp_input = fopen ( "test.h264" , "rb" ); if (NULL == data.fp_input) { mpp_err( "failed to open input file %s\n" , cmd->file_input); goto MPP_TEST_OUT; } fseek (data.fp_input, 0L, SEEK_END); file_size = ftell (data.fp_input); rewind (data.fp_input); mpp_log( "input file size %ld\n" , file_size); data.fp_output = fopen ( "output.yuv" , "w+b" ); if (cmd->simple) { buf = mpp_malloc( char , packet_size); ret = mpp_packet_init(&packet, buf, packet_size); } mpp_log( "mpi_dec_test decoder test start w %d h %d type %d\n" , width, height, type); ret = mpp_create(&ctx, &mpi); mpi_cmd = MPP_DEC_SET_PARSER_SPLIT_MODE; param = &need_split; ret = mpi->control(ctx, mpi_cmd, param); if (MPP_OK != ret) { mpp_err( "mpi->control failed\n" ); goto MPP_TEST_OUT; } ret = mpp_init(ctx, MPP_CTX_DEC, type); if (MPP_OK != ret) { mpp_err( "mpp_init failed\n" ); goto MPP_TEST_OUT; } mpp_log( "packet_size = %d\n" , packet_size); data.ctx = ctx; data.mpi = mpi; data.eos = 0; data.buf = buf; data.packet = packet; data.packet_size = packet_size; data.frame = frame; data.frame_count = 0; data.frame_num = cmd->frame_num; mpp_log( "data.packet_size = %d\n" , data.packet_size); if (cmd->simple) { while (!data.eos) { decode_simple(&data); mpp_log( "data.eos = %d\n" , data.eos); } } cmd->max_usage = data.max_usage; ret = mpi->reset(ctx); if (MPP_OK != ret) { mpp_err( "mpi->reset failed\n" ); goto MPP_TEST_OUT; } 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 (buf) { mpp_free(buf); buf = NULL; } } if (data.pkt_grp) { mpp_buffer_group_put(data.pkt_grp); data.pkt_grp = NULL; } if (data.frm_grp) { mpp_buffer_group_put(data.frm_grp); data.frm_grp = NULL; } if (data.fp_output) { fclose (data.fp_output); data.fp_output = NULL; } if (data.fp_input) { fclose (data.fp_input); data.fp_input = NULL; } return ret; } int main( int argc, char **argv) { RK_S32 ret = 0; MpiDecTestCmd cmd_ctx; MpiDecTestCmd* cmd = &cmd_ctx; cmd->simple = 1; cmd->type = 7; cmd->width = 640; cmd->height = 480; ret = mpi_dec_test_decode(cmd); return ret; } |
具体分析:
1、MPI接口的结构和使用,可以参考上篇内容,解码与编码基本一致。
2、解码器接口
decode_put_packet:
MPP_RET decode_put_packet(MppCtx ctx,MppPacket packet)
ctx:MPP解码器实例;
packet:待输入的码流数据;
输入码流的方式:分帧与不分帧。裸码流输入有两种,一种是按帧分段的数据,每一个输入给decode_put_packet函数的数据包都包含完整的一帧,不多也不少。在这种情况下,MPP可以直接按包处理码流。另一种是按长度读取数据,无法判断一个包的数据是否为完整的一帧,需要MPP内部进行分帧处理。在进行这种形式的输入时,需要在mpp_init前,通过control接口的MPP_DEC_SET_PARSER_SPLIT_MODE命令,打开need_split标志。分帧方式效率高,但需要在输入码流前进行解析与分帧,不分帧方式使用简单,效率会受影响。官方解码示例采用的是不分帧方式,因此上述代码也是不分帧方式。
decode_get_frame:
MPP_RET decode_get_frame(MppCtx ctx,MppFrame *frame)
ctx:MPP解码器实例;
frame:用于MppFrame实例的指针;
完整的解码过程是上面两个函数的结合。
关山难越,谁悲失路之人;
萍水相逢,尽是他乡之客。
【推荐】编程新体验,更懂你的AI,立即体验豆包MarsCode编程助手
【推荐】凌霞软件回馈社区,博客园 & 1Panel & Halo 联合会员上线
【推荐】抖音旗下AI助手豆包,你的智能百科全书,全免费不限次数
【推荐】博客园社区专享云产品让利特惠,阿里云新客6.5折上折
【推荐】轻量又高性能的 SSH 工具 IShell:AI 加持,快人一步