schihei/ascend-tools
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

clean code

Browse Source
This commit is contained in:
liuchengju
2020-09-01 11:51:28 +08:00
parent bea100c789
commit 7d654d22cb
44 changed files with 4486 additions and 724 deletions
Download Patch File Download Diff File Expand all files Collapse all files
msame/src/main.cpp
Executable → Regular
+45 -61
View File
@@ -10,9 +10,8 @@
*/
#include "sample_process.h"
#include "utils.h"
#include<getopt.h>
#include <getopt.h>
using namespace std;
bool f_isTXT = false;
bool g_isDevice = false;
@@ -25,54 +24,46 @@ string input_Ftype = ".bin";
string model_Ftype = ".om";
string check = "";
void InitAndCheckParams(int argc, char* argv[], map<char,string>& params, vector<string>& inputs)
//void InitAndCheckParams(int argc, char* argv[], vector<string>& params, vector<string>& inputs)
void InitAndCheckParams(int argc, char* argv[], map<char, string>& params, vector<string>& inputs)
{
const char *optstring="m::i::o::f::hd::p::l::y::e::g::";
int c,deb,index;
struct option opts[]={{"model",required_argument,NULL,'m'},
{"input",required_argument,NULL,'i'},
{"output",required_argument,NULL,'o'},
{"outfmt",required_argument,NULL,'f'},
{"help",no_argument,NULL,1},
{"dump",required_argument,NULL,'d'},
{"profiler",required_argument,NULL,'p'},
{"loop",required_argument,NULL,'l'},
{"dymBatch",required_argument,NULL,'y'},
{"device",required_argument,NULL,'e'},
{"debug",required_argument,NULL,'g'},
{0,0,0,0}};
while((c=getopt_long(argc,argv,optstring,opts,&index))!=-1)
{
switch(c)
{
const char* optstring = "m::i::o::f::hd::p::l::y::e::g::";
int c, deb, index;
struct option opts[] = { { "model", required_argument, NULL, 'm' },
{ "input", required_argument, NULL, 'i' },
{ "output", required_argument, NULL, 'o' },
{ "outfmt", required_argument, NULL, 'f' },
{ "help", no_argument, NULL, 1 },
{ "dump", required_argument, NULL, 'd' },
{ "profiler", required_argument, NULL, 'p' },
{ "loop", required_argument, NULL, 'l' },
{ "dymBatch", required_argument, NULL, 'y' },
{ "device", required_argument, NULL, 'e' },
{ "debug", required_argument, NULL, 'g' },
{ 0, 0, 0, 0 } };
while ((c = getopt_long(argc, argv, optstring, opts, &index)) != -1) {
switch (c) {
case 'm':
check = optarg;
if (check.find(model_Ftype) != string::npos){
// params.push_back(optarg);
check = optarg;
if (check.find(model_Ftype) != string::npos) {
params['m'] = optarg;
break;
}
else {
printf("input model file type is not .om , please check your model type!\n");
exit(0);
}
break;
} else {
printf("input model file type is not .om , please check your model type!\n");
exit(0);
}
case 'i':
check = optarg;
if (check.find(input_Ftype) == string::npos){
printf("input data file type is not .bin , please check your input file type!\n");
exit(0);
}
// params.push_back(optarg);
check = optarg;
if (check.find(input_Ftype) == string::npos) {
printf("input data file type is not .bin , please check your input file type!\n");
exit(0);
}
params['i'] = optarg;
Utils::SplitString(params['i'], inputs, ',');
break;
case 'o':
// params.push_back(optarg);
params['o'] = optarg;
break;
case 'f':
// params.push_back(optarg);
params['f'] = optarg;
break;
case '?':
@@ -81,34 +72,29 @@ void InitAndCheckParams(int argc, char* argv[], map<char,string>& params, vector
Utils::printHelpLetter();
exit(0);
case 'd':
// params.push_back(optarg);
params['d'] = optarg;
break;
case 'p':
// params.push_back(optarg);
params['p'] = optarg;
break;
case 'l':
loop = Utils::str2num(optarg);
cout << "loop:" << loop << endl;
if (loop > 100 || loop < 1)
{
printf("loop must in 1 to 100\n");
exit(0);
}
cout << "loop:" << loop << endl;
if (loop > 100 || loop < 1) {
printf("loop must in 1 to 100\n");
exit(0);
}
break;
case 'y':
// params.push_back(optarg);
params['y'] = optarg;
break;
case 'e':
device = Utils::str2num(optarg);
cout << "device:" << device << endl;
if (device > 255 || device < 0)
{
printf("device id must in 0 to 255\n");
exit(0);
}
if (device > 255 || device < 0) {
printf("device id must in 0 to 255\n");
exit(0);
}
break;
case 'g':
params['g'] = optarg;
@@ -127,13 +113,11 @@ void InitAndCheckParams(int argc, char* argv[], map<char,string>& params, vector
int main(int argc, char* argv[])
{
map<char,string>params;
// vector<string> params;
map<char, string> params;
vector<string> inputs;
InitAndCheckParams(argc, argv, params, inputs);
printf("******************************\n");
printf("Test Start!\n");
if (params.empty()) {
printf("Invalid params.\n");
@@ -147,20 +131,20 @@ int main(int argc, char* argv[])
}
if (params['p'].compare("true") == 0) {
is_profi = true;
}
}
if (params['g'].compare("true") == 0) {
is_debug = true;
}
if (is_profi && is_dump){
if (is_profi && is_dump) {
ERROR_LOG("dump and profiler can not both be true");
return FAILED;
}
Utils::ProfilerJson(is_profi, params);
Utils::DumpJson(is_dump, params);
SampleProcess processSample;
Result ret = processSample.InitResource();
if (ret != SUCCESS) {
ERROR_LOG("Sample init resource failed.");
@@ -176,6 +160,6 @@ int main(int argc, char* argv[])
INFO_LOG("Execute sample success.");
printf("Test Finish!\n");
printf("******************************\n");
return SUCCESS;
}