1 | // Copyright Vladimir Prus 2002-2004. |
2 | // Distributed under the Boost Software License, Version 1.0. |
3 | // (See accompanying file LICENSE_1_0.txt |
4 | // or copy at http://www.boost.org/LICENSE_1_0.txt) |
5 | |
6 | |
7 | #define BOOST_PROGRAM_OPTIONS_SOURCE |
8 | #include <boost/program_options/config.hpp> |
9 | #include <boost/program_options/parsers.hpp> |
10 | #include <boost/program_options/options_description.hpp> |
11 | #include <boost/program_options/value_semantic.hpp> |
12 | #include <boost/program_options/variables_map.hpp> |
13 | |
14 | #include <cassert> |
15 | |
16 | namespace boost { namespace program_options { |
17 | |
18 | using namespace std; |
19 | |
20 | // First, performs semantic actions for 'oa'. |
21 | // Then, stores in 'm' all options that are defined in 'desc'. |
22 | BOOST_PROGRAM_OPTIONS_DECL |
23 | void store(const parsed_options& options, variables_map& xm, |
24 | bool utf8) |
25 | { |
26 | // TODO: what if we have different definition |
27 | // for the same option name during different calls |
28 | // 'store'. |
29 | assert(options.description); |
30 | const options_description& desc = *options.description; |
31 | |
32 | // We need to access map's operator[], not the overriden version |
33 | // variables_map. Ehmm.. messy. |
34 | std::map<std::string, variable_value>& m = xm; |
35 | |
36 | std::set<std::string> new_final; |
37 | |
38 | // Declared once, to please Intel in VC++ mode; |
39 | unsigned i; |
40 | |
41 | // Declared here so can be used to provide context for exceptions |
42 | string option_name; |
43 | string original_token; |
44 | |
45 | #ifndef BOOST_NO_EXCEPTIONS |
46 | try |
47 | #endif |
48 | { |
49 | |
50 | // First, convert/store all given options |
51 | for (i = 0; i < options.options.size(); ++i) { |
52 | |
53 | option_name = options.options[i].string_key; |
54 | // Skip positional options without name |
55 | if (option_name.empty()) |
56 | continue; |
57 | |
58 | // Ignore unregistered option. The 'unregistered' |
59 | // field can be true only if user has explicitly asked |
60 | // to allow unregistered options. We can't store them |
61 | // to variables map (lacking any information about paring), |
62 | // so just ignore them. |
63 | if (options.options[i].unregistered) |
64 | continue; |
65 | |
66 | // If option has final value, skip this assignment |
67 | if (xm.m_final.count(option_name)) |
68 | continue; |
69 | |
70 | original_token = options.options[i].original_tokens.size() ? |
71 | options.options[i].original_tokens[0] : "" ; |
72 | const option_description& d = desc.find(option_name, false, |
73 | false, false); |
74 | |
75 | variable_value& v = m[option_name]; |
76 | if (v.defaulted()) { |
77 | // Explicit assignment here erases defaulted value |
78 | v = variable_value(); |
79 | } |
80 | |
81 | d.semantic()->parse(v.value(), options.options[i].value, utf8); |
82 | |
83 | v.m_value_semantic = d.semantic(); |
84 | |
85 | // The option is not composing, and the value is explicitly |
86 | // provided. Ignore values of this option for subsequent |
87 | // calls to 'store'. We store this to a temporary set, |
88 | // so that several assignment inside *this* 'store' call |
89 | // are allowed. |
90 | if (!d.semantic()->is_composing()) |
91 | new_final.insert(option_name); |
92 | } |
93 | } |
94 | #ifndef BOOST_NO_EXCEPTIONS |
95 | catch(error_with_option_name& e) |
96 | { |
97 | // add context and rethrow |
98 | e.add_context(option_name, original_token, options.m_options_prefix); |
99 | throw; |
100 | } |
101 | #endif |
102 | xm.m_final.insert(new_final.begin(), new_final.end()); |
103 | |
104 | |
105 | |
106 | // Second, apply default values and store required options. |
107 | const vector<shared_ptr<option_description> >& all = desc.options(); |
108 | for(i = 0; i < all.size(); ++i) |
109 | { |
110 | const option_description& d = *all[i]; |
111 | string key = d.key("" ); |
112 | // FIXME: this logic relies on knowledge of option_description |
113 | // internals. |
114 | // The 'key' is empty if options description contains '*'. |
115 | // In that |
116 | // case, default value makes no sense at all. |
117 | if (key.empty()) |
118 | { |
119 | continue; |
120 | } |
121 | if (m.count(key) == 0) { |
122 | |
123 | boost::any def; |
124 | if (d.semantic()->apply_default(def)) { |
125 | m[key] = variable_value(def, true); |
126 | m[key].m_value_semantic = d.semantic(); |
127 | } |
128 | } |
129 | |
130 | // add empty value if this is an required option |
131 | if (d.semantic()->is_required()) { |
132 | |
133 | // For option names specified in multiple ways, e.g. on the command line, |
134 | // config file etc, the following precedence rules apply: |
135 | // "--" > ("-" or "/") > "" |
136 | // Precedence is set conveniently by a single call to length() |
137 | string canonical_name = d.canonical_display_name(options.m_options_prefix); |
138 | if (canonical_name.length() > xm.m_required[key].length()) |
139 | xm.m_required[key] = canonical_name; |
140 | } |
141 | } |
142 | } |
143 | |
144 | BOOST_PROGRAM_OPTIONS_DECL |
145 | void store(const wparsed_options& options, variables_map& m) |
146 | { |
147 | store(options.utf8_encoded_options, m, true); |
148 | } |
149 | |
150 | BOOST_PROGRAM_OPTIONS_DECL |
151 | void notify(variables_map& vm) |
152 | { |
153 | vm.notify(); |
154 | } |
155 | |
156 | abstract_variables_map::abstract_variables_map() |
157 | : m_next(0) |
158 | {} |
159 | |
160 | abstract_variables_map:: |
161 | abstract_variables_map(const abstract_variables_map* next) |
162 | : m_next(next) |
163 | {} |
164 | |
165 | const variable_value& |
166 | abstract_variables_map::operator[](const std::string& name) const |
167 | { |
168 | const variable_value& v = get(name); |
169 | if (v.empty() && m_next) |
170 | return (*m_next)[name]; |
171 | else if (v.defaulted() && m_next) { |
172 | const variable_value& v2 = (*m_next)[name]; |
173 | if (!v2.empty() && !v2.defaulted()) |
174 | return v2; |
175 | else return v; |
176 | } else { |
177 | return v; |
178 | } |
179 | } |
180 | |
181 | void |
182 | abstract_variables_map::next(abstract_variables_map* next) |
183 | { |
184 | m_next = next; |
185 | } |
186 | |
187 | variables_map::variables_map() |
188 | {} |
189 | |
190 | variables_map::variables_map(const abstract_variables_map* next) |
191 | : abstract_variables_map(next) |
192 | {} |
193 | |
194 | void variables_map::clear() |
195 | { |
196 | std::map<std::string, variable_value>::clear(); |
197 | m_final.clear(); |
198 | m_required.clear(); |
199 | } |
200 | |
201 | const variable_value& |
202 | variables_map::get(const std::string& name) const |
203 | { |
204 | static variable_value empty; |
205 | const_iterator i = this->find(name); |
206 | if (i == this->end()) |
207 | return empty; |
208 | else |
209 | return i->second; |
210 | } |
211 | |
212 | void |
213 | variables_map::notify() |
214 | { |
215 | // This checks if all required options occur |
216 | for (map<string, string>::const_iterator r = m_required.begin(); |
217 | r != m_required.end(); |
218 | ++r) |
219 | { |
220 | const string& opt = r->first; |
221 | const string& display_opt = r->second; |
222 | map<string, variable_value>::const_iterator iter = find(opt); |
223 | if (iter == end() || iter->second.empty()) |
224 | { |
225 | boost::throw_exception(required_option(display_opt)); |
226 | |
227 | } |
228 | } |
229 | |
230 | // Lastly, run notify actions. |
231 | for (map<string, variable_value>::iterator k = begin(); |
232 | k != end(); |
233 | ++k) |
234 | { |
235 | /* Users might wish to use variables_map to store their own values |
236 | that are not parsed, and therefore will not have value_semantics |
237 | defined. Do not crash on such values. In multi-module programs, |
238 | one module might add custom values, and the 'notify' function |
239 | will be called after that, so we check that value_sematics is |
240 | not NULL. See: |
241 | https://svn.boost.org/trac/boost/ticket/2782 |
242 | */ |
243 | if (k->second.m_value_semantic) |
244 | k->second.m_value_semantic->notify(k->second.value()); |
245 | } |
246 | } |
247 | |
248 | }} |
249 | |