mirror of
https://github.com/nyanmisaka/mpp.git
synced 2025-10-05 09:06:50 +08:00
841022f529
Change-Id: Iea28d4377212e4c18a91f30762ae8ec135702c16 Signed-off-by: Herman Chen <herman.chen@rock-chips.com>
366 lines
11 KiB
C
366 lines
11 KiB
C
/*
|
|
* Copyright 2020 Rockchip Electronics Co. LTD
|
|
*
|
|
* Licensed under the Apache License, Version 2.0 (the "License");
|
|
* you may not use this file except in compliance with the License.
|
|
* You may obtain a copy of the License at
|
|
*
|
|
* http://www.apache.org/licenses/LICENSE-2.0
|
|
*
|
|
* Unless required by applicable law or agreed to in writing, software
|
|
* distributed under the License is distributed on an "AS IS" BASIS,
|
|
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
|
* See the License for the specific language governing permissions and
|
|
* limitations under the License.
|
|
*/
|
|
|
|
#define MODULE_TAG "mpi_dec_utils"
|
|
|
|
#include <string.h>
|
|
|
|
#include "mpp_mem.h"
|
|
#include "mpp_log.h"
|
|
#include "rk_mpi.h"
|
|
#include "utils.h"
|
|
#include "mpp_common.h"
|
|
#include "mpi_dec_utils.h"
|
|
|
|
#define IVF_HEADER_LENGTH 32
|
|
#define IVF_FRAME_HEADER_LENGTH 12
|
|
|
|
typedef enum {
|
|
FILE_NORMAL_TYPE,
|
|
FILE_IVF_TYPE,
|
|
FILE_BUTT,
|
|
} FileType;
|
|
|
|
typedef struct FileReader_t {
|
|
char *buf;
|
|
FILE *fp_input;
|
|
size_t buf_size;
|
|
size_t read_size;
|
|
FileType file_type;
|
|
} FileReaderImpl;
|
|
|
|
OptionInfo mpi_dec_cmd[] = {
|
|
{"i", "input_file", "input bitstream file"},
|
|
{"o", "output_file", "output bitstream file, "},
|
|
{"c", "ops_file", "input operation config file"},
|
|
{"w", "width", "the width of input bitstream"},
|
|
{"h", "height", "the height of input bitstream"},
|
|
{"t", "type", "input stream coding type"},
|
|
{"f", "format", "output frame format type"},
|
|
{"d", "debug", "debug flag"},
|
|
{"x", "timeout", "output timeout interval"},
|
|
{"n", "frame_number", "max output frame number"},
|
|
{"s", "instance_nb", "number of instances"},
|
|
{NULL},
|
|
};
|
|
|
|
static RK_U32 read_ivf_file(FileReader data)
|
|
{
|
|
FileReaderImpl *reader = (FileReaderImpl*)data;
|
|
char *buf = reader->buf;
|
|
size_t data_size = 0;
|
|
size_t buf_size = reader->buf_size;
|
|
RK_U32 eos = 0;
|
|
RK_U8 ivf_data[IVF_FRAME_HEADER_LENGTH] = {0};
|
|
|
|
if (fread(ivf_data, 1, IVF_FRAME_HEADER_LENGTH, reader->fp_input) != IVF_FRAME_HEADER_LENGTH)
|
|
return -1;
|
|
data_size = ivf_data[0] |
|
|
(ivf_data[1] << 8) |
|
|
(ivf_data[2] << 16) |
|
|
(ivf_data[3] << 24);
|
|
/* buffer size not enough, realloc */
|
|
if (data_size > buf_size) {
|
|
size_t realloc_size = MPP_ALIGN(data_size, SZ_4K);
|
|
mpp_log("buf size not enough realloc size %d\n", realloc_size);
|
|
buf = mpp_realloc(buf, char, realloc_size);
|
|
if (NULL == buf) {
|
|
mpp_err("mpi_dec_test realloc input stream buffer failed\n");
|
|
}
|
|
reader->buf = buf;
|
|
reader->buf_size = realloc_size;
|
|
}
|
|
|
|
reader->read_size = fread(buf, 1, data_size, reader->fp_input);
|
|
/* check reach eos whether or not */
|
|
if (!data_size || reader->read_size != data_size || feof(reader->fp_input)) {
|
|
eos = 1;
|
|
}
|
|
return eos;
|
|
}
|
|
|
|
static RK_U32 read_normal_file(FileReader data)
|
|
{
|
|
FileReaderImpl *reader = (FileReaderImpl*)data;
|
|
char *buf = reader->buf;
|
|
size_t buf_size = reader->buf_size;
|
|
RK_U32 eos = 0;
|
|
|
|
reader->read_size = fread(buf, 1, buf_size, reader->fp_input);
|
|
/* check reach eos whether or not */
|
|
if (!buf_size || reader->read_size != buf_size || feof(reader->fp_input))
|
|
eos = 1;
|
|
|
|
return eos;
|
|
}
|
|
|
|
static void check_file_type(FileReader data, char *file_in)
|
|
{
|
|
FileReaderImpl *reader = (FileReaderImpl*)data;
|
|
|
|
if (strstr(file_in, ".ivf")) {
|
|
reader->file_type = FILE_IVF_TYPE;
|
|
reader->buf_size = SZ_2M;
|
|
} else {
|
|
reader->file_type = FILE_NORMAL_TYPE;
|
|
reader->buf_size = SZ_4K;
|
|
}
|
|
}
|
|
|
|
RK_U32 reader_read(FileReader file_reader, char** buf, size_t *size)
|
|
{
|
|
FileReaderImpl *reader = file_reader;
|
|
FileType type = reader->file_type;
|
|
RK_U32 eos = 0;
|
|
|
|
switch (type) {
|
|
case FILE_NORMAL_TYPE : {
|
|
eos = read_normal_file(reader);
|
|
} break;
|
|
case FILE_IVF_TYPE : {
|
|
eos = read_ivf_file(reader);
|
|
} break;
|
|
default : {
|
|
} break;
|
|
}
|
|
|
|
*buf = reader->buf;
|
|
*size = reader->read_size;
|
|
return eos;
|
|
}
|
|
|
|
void reader_rewind(FileReader file_reader)
|
|
{
|
|
FileReaderImpl *reader = (FileReaderImpl*)file_reader;
|
|
FileType type = reader->file_type;
|
|
|
|
clearerr(reader->fp_input);
|
|
|
|
switch (type) {
|
|
case FILE_NORMAL_TYPE : {
|
|
rewind(reader->fp_input);
|
|
} break;
|
|
case FILE_IVF_TYPE : {
|
|
fseek(reader->fp_input, IVF_HEADER_LENGTH, SEEK_SET);
|
|
} break;
|
|
default : {
|
|
} break;
|
|
}
|
|
}
|
|
|
|
void reader_init(FileReader* file_reader, char* file_in, FILE* fp_in)
|
|
{
|
|
FileReaderImpl *reader = mpp_malloc(FileReaderImpl, 1);
|
|
|
|
check_file_type(reader, file_in);
|
|
|
|
reader->buf = mpp_malloc(char, reader->buf_size);
|
|
reader->fp_input = fp_in;
|
|
|
|
if (reader->file_type == FILE_IVF_TYPE)
|
|
fseek(reader->fp_input, IVF_HEADER_LENGTH, SEEK_SET);
|
|
|
|
*file_reader = reader;
|
|
}
|
|
|
|
void reader_deinit(FileReader *file_reader)
|
|
{
|
|
FileReaderImpl *reader = (FileReaderImpl*)(*file_reader);
|
|
|
|
MPP_FREE(reader->buf);
|
|
MPP_FREE(reader);
|
|
*file_reader = NULL;
|
|
}
|
|
|
|
void mpi_dec_test_help()
|
|
{
|
|
mpp_log("usage: mpi_dec_test [options]\n");
|
|
show_options(mpi_dec_cmd);
|
|
mpp_show_support_format();
|
|
}
|
|
|
|
RK_S32 mpi_dec_test_parse_options(int argc, char **argv, MpiDecTestCmd* cmd)
|
|
{
|
|
const char *opt;
|
|
const char *next;
|
|
RK_S32 optindex = 1;
|
|
RK_S32 handleoptions = 1;
|
|
RK_S32 err = MPP_NOK;
|
|
|
|
if ((argc < 2) || (cmd == NULL)) {
|
|
err = 1;
|
|
return err;
|
|
}
|
|
|
|
/* parse options */
|
|
while (optindex < argc) {
|
|
opt = (const char*)argv[optindex++];
|
|
next = (const char*)argv[optindex];
|
|
|
|
if (handleoptions && opt[0] == '-' && opt[1] != '\0') {
|
|
if (opt[1] == '-') {
|
|
if (opt[2] != '\0') {
|
|
opt++;
|
|
} else {
|
|
handleoptions = 0;
|
|
continue;
|
|
}
|
|
}
|
|
|
|
opt++;
|
|
|
|
switch (*opt) {
|
|
case 'i' : {
|
|
if (next) {
|
|
strncpy(cmd->file_input, next, MAX_FILE_NAME_LENGTH - 1);
|
|
cmd->have_input = 1;
|
|
name_to_coding_type(cmd->file_input, &cmd->type);
|
|
} else {
|
|
mpp_err("input file is invalid\n");
|
|
goto PARSE_OPINIONS_OUT;
|
|
}
|
|
} break;
|
|
case 'o' : {
|
|
if (next) {
|
|
strncpy(cmd->file_output, next, MAX_FILE_NAME_LENGTH - 1);
|
|
cmd->have_output = 1;
|
|
} else {
|
|
mpp_log("output file is invalid\n");
|
|
goto PARSE_OPINIONS_OUT;
|
|
}
|
|
} break;
|
|
case 'c' : {
|
|
if (next) {
|
|
strncpy(cmd->file_config, next, MAX_FILE_NAME_LENGTH - 1);
|
|
cmd->have_config = 1;
|
|
|
|
// enlarge packet buffer size for large input stream case
|
|
cmd->pkt_size = SZ_1M;
|
|
} else {
|
|
mpp_log("output file is invalid\n");
|
|
goto PARSE_OPINIONS_OUT;
|
|
}
|
|
} break;
|
|
case 'd' : {
|
|
if (next) {
|
|
cmd->debug = atoi(next);;
|
|
} else {
|
|
mpp_err("invalid debug flag\n");
|
|
goto PARSE_OPINIONS_OUT;
|
|
}
|
|
} break;
|
|
case 'w' : {
|
|
if (next) {
|
|
cmd->width = atoi(next);
|
|
} else {
|
|
mpp_err("invalid input width\n");
|
|
goto PARSE_OPINIONS_OUT;
|
|
}
|
|
} break;
|
|
case 'h' : {
|
|
if ((*(opt + 1) != '\0') && !strncmp(opt, "help", 4)) {
|
|
mpi_dec_test_help();
|
|
err = 1;
|
|
goto PARSE_OPINIONS_OUT;
|
|
} else if (next) {
|
|
cmd->height = atoi(next);
|
|
} else {
|
|
mpp_log("input height is invalid\n");
|
|
goto PARSE_OPINIONS_OUT;
|
|
}
|
|
} break;
|
|
case 't' : {
|
|
if (next) {
|
|
cmd->type = (MppCodingType)atoi(next);
|
|
err = mpp_check_support_format(MPP_CTX_DEC, cmd->type);
|
|
}
|
|
|
|
if (!next || err) {
|
|
mpp_err("invalid input coding type\n");
|
|
goto PARSE_OPINIONS_OUT;
|
|
}
|
|
} break;
|
|
case 'f' : {
|
|
if (next) {
|
|
cmd->format = (MppFrameFormat)atoi(next);
|
|
}
|
|
|
|
if (!next || err) {
|
|
mpp_err("invalid input coding type\n");
|
|
goto PARSE_OPINIONS_OUT;
|
|
}
|
|
} break;
|
|
case 'x' : {
|
|
if (next) {
|
|
cmd->timeout = atoi(next);
|
|
}
|
|
|
|
if (!next || cmd->timeout < 0) {
|
|
mpp_err("invalid output timeout interval\n");
|
|
goto PARSE_OPINIONS_OUT;
|
|
}
|
|
} break;
|
|
case 'n' : {
|
|
if (next) {
|
|
cmd->frame_num = atoi(next);
|
|
if (cmd->frame_num < 0)
|
|
mpp_log("infinite loop decoding mode\n");
|
|
} else {
|
|
mpp_err("invalid frame number\n");
|
|
goto PARSE_OPINIONS_OUT;
|
|
}
|
|
} break;
|
|
case 's' : {
|
|
if (next) {
|
|
cmd->nthreads = atoi(next);
|
|
}
|
|
if (!next || cmd->nthreads < 0) {
|
|
mpp_err("invalid nthreads\n");
|
|
goto PARSE_OPINIONS_OUT;
|
|
}
|
|
} break;
|
|
case 'q' : {
|
|
cmd->quiet = 1;
|
|
optindex--;
|
|
} break;
|
|
default : {
|
|
mpp_err("skip invalid opt %c\n", *opt);
|
|
} break;
|
|
}
|
|
|
|
optindex++;
|
|
}
|
|
}
|
|
|
|
err = 0;
|
|
|
|
PARSE_OPINIONS_OUT:
|
|
return err;
|
|
}
|
|
|
|
void mpi_dec_test_show_options(MpiDecTestCmd* cmd)
|
|
{
|
|
mpp_log("cmd parse result:\n");
|
|
mpp_log("input file name: %s\n", cmd->file_input);
|
|
mpp_log("output file name: %s\n", cmd->file_output);
|
|
mpp_log("config file name: %s\n", cmd->file_config);
|
|
mpp_log("width : %4d\n", cmd->width);
|
|
mpp_log("height : %4d\n", cmd->height);
|
|
mpp_log("type : %d\n", cmd->type);
|
|
mpp_log("debug flag : %x\n", cmd->debug);
|
|
mpp_log("max frames : %d\n", cmd->frame_num);
|
|
}
|