From e38a5ad3576ac37cbec7cc0fc4fbee3136bd9d29 Mon Sep 17 00:00:00 2001
From: Herman Chen <herman.chen@rock-chips.com>
Date: Tue, 22 May 2018 16:19:36 +0800
Subject: [PATCH] [test]: Add frame count limit to mpi_dec_test

Add max frame number parameter with "-n" option to mpi_dec_test.

Change-Id: I21c00baee4d94cab69a95060ac5a694cd9f28945
Signed-off-by: Herman Chen <herman.chen@rock-chips.com>
---
 test/mpi_dec_test.c | 30 ++++++++++++++++++++++++++++--
 1 file changed, 28 insertions(+), 2 deletions(-)

diff --git a/test/mpi_dec_test.c b/test/mpi_dec_test.c
index 0614def0..d389d3d5 100644
--- a/test/mpi_dec_test.c
+++ b/test/mpi_dec_test.c
@@ -54,7 +54,8 @@ typedef struct {
 
     FILE            *fp_input;
     FILE            *fp_output;
-    RK_U32          frame_count;
+    RK_S32          frame_count;
+    RK_S32          frame_num;
 } MpiDecLoopData;
 
 typedef struct {
@@ -70,6 +71,7 @@ typedef struct {
 
     RK_U32          simple;
     RK_S32          timeout;
+    RK_S32          frame_num;
 } MpiDecTestCmd;
 
 static OptionInfo mpi_dec_cmd[] = {
@@ -80,6 +82,7 @@ static OptionInfo mpi_dec_cmd[] = {
     {"t",               "type",                 "input stream coding type"},
     {"d",               "debug",                "debug flag"},
     {"x",               "timeout",              "output timeout interval"},
+    {"n",               "frame_number",         "max output frame number"},
 };
 
 static int decode_simple(MpiDecLoopData *data)
@@ -224,7 +227,8 @@ static int decode_simple(MpiDecLoopData *data)
                         mpp_log("decoder_get_frame get err info:%d discard:%d.\n",
                                 mpp_frame_get_errinfo(frame), mpp_frame_get_discard(frame));
                     }
-                    mpp_log("decode_get_frame get frame %d\n", data->frame_count++);
+                    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);
                 }
@@ -245,11 +249,22 @@ static int decode_simple(MpiDecLoopData *data)
                 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;
 
@@ -530,6 +545,7 @@ int mpi_dec_test_decode(MpiDecTestCmd *cmd)
     data.packet_size    = packet_size;
     data.frame          = frame;
     data.frame_count    = 0;
+    data.frame_num      = cmd->frame_num;
 
     if (cmd->simple) {
         while (!data.eos) {
@@ -709,6 +725,15 @@ static RK_S32 mpi_dec_test_parse_options(int argc, char **argv, MpiDecTestCmd* c
                     goto PARSE_OPINIONS_OUT;
                 }
                 break;
+            case 'n':
+                if (next) {
+                    cmd->frame_num = atoi(next);
+                }
+                if (!next || cmd->frame_num < 0) {
+                    mpp_err("invalid frame number\n");
+                    goto PARSE_OPINIONS_OUT;
+                }
+                break;
             default:
                 goto PARSE_OPINIONS_OUT;
                 break;
@@ -733,6 +758,7 @@ static void mpi_dec_test_show_options(MpiDecTestCmd* cmd)
     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);
 }
 
 int main(int argc, char **argv)