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
|
#include "OptionParser.h"
#include <cstdlib>
#include <iostream>
namespace LibUtil
{
using std::cout;
using std::cerr;
using std::endl;
OptionParser::OptionInfo::OptionInfo(
const String& var_name_,
bool has_arg_,
const String& arg_name_,
bool has_default_arg_value_,
const String& default_arg_value_,
const String& description_
)
: m_var_name_(var_name_),
m_has_arg_(has_arg_),
m_arg_name_(arg_name_),
m_has_default_arg_value_(has_default_arg_value_),
m_default_arg_value_(default_arg_value_),
m_description_(description_)
{}
OptionParser::OptionInfo::~OptionInfo()
{
}
OptionParser::OptionParser()
{}
OptionParser::~OptionParser()
{
clearPtrMap(&m_option_infos_);
}
void OptionParser::addOption(
const String& option_name_,
const String& var_name_,
bool has_arg_,
const String& arg_name_,
bool has_default_arg_value_,
const String& default_arg_value_,
const String& description_)
{
OptionInfo* option_info = new OptionInfo(var_name_, has_arg_, arg_name_,
has_default_arg_value_, default_arg_value_, description_);
ASSERT(!m_option_infos_.keyExist(option_name_), "Option exists: " + option_name_);
// Add the option name to an array for sorting
m_option_names_.push_back(option_name_);
// Add option info
m_option_infos_.set(option_name_, option_info);
// Set the default argument value
if(has_default_arg_value_)
{
set(var_name_, default_arg_value_);
}
return;
}
void OptionParser::parseArguments(int argc_, char** argv_)
{
bool is_print_options = false;
int arg_idx = 0;
while(arg_idx < argc_)
{
String option_name = String(argv_[arg_idx]);
// Print the options page if -help is specified
if(option_name == "-help")
{
is_print_options = true;
break;
}
else if(m_option_infos_.keyExist(option_name))
{
const OptionInfo* option_info = m_option_infos_.get(option_name);
const String& var_name = option_info->getVarName();
if(option_info->hasArg())
{
if((arg_idx + 1) >= argc_)
{
cerr << "[Error] Missing argument for option: '" << option_name << "'" << endl;
is_print_options = true;
break;
}
String option_arg = String(argv_[arg_idx + 1]);
set(var_name, option_arg);
arg_idx += 2;
}
else
{
// If the option does not require an argument
// then set it to true
set(var_name, "true");
arg_idx += 1;
}
}
else
{
cerr << "[Error] Unknown option: '" << option_name << "'" << endl;
is_print_options = true;
break;
}
}
// Check if all required options are set (the ones without default values)
vector<String>::const_iterator it;
for(it = m_option_names_.begin(); it != m_option_names_.end(); ++it)
{
const String& option_name = *it;
const OptionInfo* option_info = m_option_infos_.get(option_name);
if(!option_info->hasDefaultArgValue())
{
const String& var_name = option_info->getVarName();
if(!keyExist(var_name))
{
cerr << "[Error] Missing required option: '" << option_name << "'" << endl;
is_print_options = true;
}
}
}
if(is_print_options)
{
printOptions();
exit(0);
}
return;
}
void OptionParser::printOptions() const
{
cout << endl;
cout << "Available options:" << endl;
cout << "==================" << endl << endl;
vector<String>::const_iterator it;
for(it = m_option_names_.begin(); it != m_option_names_.end(); ++it)
{
const String& option_name = *it;
const OptionInfo* option_info = m_option_infos_.get(option_name);
cout << option_name;
if(option_info->hasArg())
{
cout << " <" << option_info->getArgName() << ">";
}
cout << endl;
cout << " " << option_info->getDescription() << endl;
if(option_info->hasArg() && option_info->hasDefaultArgValue())
{
cout << " " << "Default: " << option_info->getDefaultArgValue() << endl;
}
cout << endl;
}
cout << "-help" << endl;
cout << " " << "Print this page" << endl;
cout << endl;
return;
}
} // namespace LibUtil
|
