@@ -61,15 +61,25 @@ const char *error_message(scanbotsdk_error_code_t ec) {
6161}
6262
6363
64- char * get_flag (int argc , char * argv [], const char * flag ) {
65- for (int i = 3 ; i < argc ; i ++ ) {
66- if (strcmp (argv [i ], flag ) == 0 && i + 1 < argc ) {
67- return argv [i + 1 ];
64+ const char * get_flag (int argc , char * argv [], const char * flag ) {
65+ size_t len = strlen (flag );
66+
67+ for (int i = 1 ; i < argc ; i ++ ) {
68+
69+ // "--flag"
70+ if (strcmp (argv [i ], flag ) == 0 ) {
71+ if (i + 1 < argc && strncmp (argv [i + 1 ], "--" , 2 ) != 0 )
72+ return argv [i + 1 ];
73+
74+ return "" ;
6875 }
69- if (strncmp (argv [i ], flag , strlen (flag )) == 0 && argv [i ][strlen (flag )] == '=' ) {
70- return argv [i ] + strlen (flag ) + 1 ;
76+
77+ // "--flag=value"
78+ if (strncmp (argv [i ], flag , len ) == 0 && argv [i ][len ] == '=' ) {
79+ return argv [i ] + len + 1 ;
7180 }
7281 }
82+
7383 return NULL ;
7484}
7585
@@ -84,7 +94,7 @@ void print_usage(const char *prog) {
8494 printf ("or\n" );
8595 printf (" %s parse <command> --text \"<input>\" [--license <KEY>]\n\n" , prog );
8696 printf ("or\n" );
87- printf (" %s live <command> --file \"<input>\" [--license <KEY>]\n\n" , prog );
97+ printf (" %s live <command> --file \"<input>\" [--license <KEY>] [--use_tensorrt] \n\n" , prog );
8898
8999 printf ("Available scan commands:\n" );
90100 printf (" barcode | document | check | credit_card | document_extractor |\n" );
0 commit comments