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
|
//
// Zinnia: Online hand recognition system with machine learning
//
// $Id: param.h 17 2009-04-05 11:40:32Z taku-ku $;
//
// Copyright(C) 2008 Taku Kudo <taku@chasen.org>
//
#ifndef zinnia_PARAM_H_
#define zinnia_PARAM_H_
#include <map>
#include <string>
#include <vector>
#include <sstream>
#include "scoped_ptr.h"
#include "common.h"
namespace zinnia {
namespace {
template <class Target, class Source>
Target lexical_cast(Source arg) {
std::stringstream interpreter;
Target result;
if (!(interpreter << arg) || !(interpreter >> result) ||
!(interpreter >> std::ws).eof()) {
scoped_ptr<Target> r(new Target()); // return default value
return *r;
}
return result;
}
template <>
std::string lexical_cast<std::string, std::string>(std::string arg) {
return arg;
}
}
struct Option {
const char *name;
char short_name;
const char *default_value;
const char *arg_description;
const char *description;
};
class Param {
private:
std::map<std::string, std::string> conf_;
std::vector<std::string> rest_;
std::string system_name_;
std::string help_;
std::string version_;
whatlog what_;
public:
bool open(int argc, char **argv, const Option *opt);
bool open(const char *arg, const Option *opt);
bool load(const char *filename);
void clear();
const std::vector<std::string>& rest_args() const { return rest_; }
const char* program_name() const { return system_name_.c_str(); }
const char *what() { return what_.str(); }
const char* help() const { return help_.c_str(); }
const char* version() const { return version_.c_str(); }
int help_version() const;
template <class T>
T get(const char *key) const {
std::map<std::string, std::string>::const_iterator it = conf_.find(key);
if (it == conf_.end()) {
scoped_ptr<T> r(new T());
return *r;
}
return lexical_cast<T, std::string>(it->second);
}
template <class T>
void set(const char* key, const T &value,
bool rewrite = true) {
std::string key2 = std::string(key);
if (rewrite || (!rewrite && conf_.find(key2) == conf_.end()))
conf_[key2] = lexical_cast<std::string, T>(value);
}
void dump_config(std::ostream *os) const;
explicit Param() {}
virtual ~Param() {}
};
}
#endif
|
