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
|
#pragma once
#include <util/system/defaults.h>
#include <util/generic/string.h>
#include <util/generic/vector.h>
// simplified options parser
// No 'optional argument' (e.g. "a::" in spec.) support;
// Supports '+' switch (see opt.h), does not support '-';
/** Typical use
Opt2 opt(argc, argv, "A:b:c", 3); <- 3 more arguments expected, opt.Pos[0], etc.
** Usage description for options is provided through functions that query values **
const char *a = opt.Arg('A', "<var_name> - usage of -A"); <- This option is required
int b = opt.Int('b', "<var_name> - usage of -b", 2); <- This option has default value, not required
bool c = opt.Has('c', "- usage of -c"); <- switches are always optional
** Additional argument names are provided in AutoUsage call **
** AutoUsage generages 'USAGE' text automatically **
if (opt.AutoUsage("<L> <M>")) <- Returns 1 if there was any error in getopt
return 1;
OR: opt.AutoUsageErr("<L> <M>"); <- Will terminate program for you :)
*/
// Note: struct Opt2Param can be moved to cpp-file
struct Opt2Param {
char opt;
bool HasArg;
bool IsFound;
bool IsNumeric;
bool IsRequired;
bool MultipleUse;
const char* DefValue;
TString DefValueStr;
TString HelpUsage;
TVector<const char*> ActualValue;
const char* LongOptName;
Opt2Param()
: HasArg(false)
, IsFound(0)
, IsNumeric(0)
, IsRequired(0)
, MultipleUse(0)
, DefValue(nullptr)
, LongOptName(nullptr)
{
}
};
struct IntRange {
int Left, Right;
IntRange() = delete;
IntRange(int both)
: Left(both)
, Right(both)
{
}
IntRange(int left, int right)
: Left(left)
, Right(right)
{
}
};
class Opt2 {
public:
Opt2() = default;
Opt2(int argc, char* const* argv, const char* optspec, IntRange free_args_num = -1, const char* long_alias = nullptr) {
Init(argc, argv, optspec, free_args_num, long_alias);
}
// Init throws exception only in case of incorrect optspec.
// In other cases, consult HasErrors or call AutoUsage()
void Init(int argc, char* const* argv, const char* optspec, IntRange free_args_num = -1, const char* long_alias = nullptr);
// In case of incorrect options, constructs and prints Usage text,
// usually to stderr (however, to stdout if '-?' switch was used), and returns 1.
int AutoUsage(const char* free_arg_names = "");
// same as AutoUsage but calls exit(1) instead of error code
void AutoUsageErr(const char* free_arg_names = "");
// For options with parameters
const char* Arg(char opt, const char* helpUsage, const char* defValue, bool required = false);
const char* Arg(char opt, const char* helpUsage) {
return Arg(opt, helpUsage, nullptr, true);
}
const char* Arg(char opt, const char* helpUsage, TString defValue, bool required = false);
// Options with parameters that can be specified several times
const TVector<const char*>& MArg(char opt, const char* helpUsage);
// Get() + strtol, may set up HasErrors
long Int(char opt, const char* helpUsage, long defValue, bool required = false);
long Int(char opt, const char* helpUsage) {
return Int(opt, helpUsage, 0, true);
}
// Get() + strtoul, may set up HasErrors
unsigned long UInt(char opt, const char* helpUsage, unsigned long defValue, bool required = false);
unsigned long UInt(char opt, const char* helpUsage) {
return UInt(opt, helpUsage, 0, true);
}
// For options w/o parameters
bool Has(char opt, const char* helpUsage);
// Add user defined error message and set error flag
void AddError(const char* message = nullptr);
public:
// non-option args
TVector<char*> Pos;
bool HasErrors;
private:
bool BadPosCount;
char UnknownOption;
char* UnknownLongOption;
char OptionMissingArg;
char OptionWrongArg;
char RequiredOptionMissing;
TVector<TString> UserErrorMessages;
protected:
int Argc;
char* const* Argv;
int MinArgs, MaxArgs;
ui8 SpecsMap[256];
TVector<Opt2Param> Specs;
TString alias_copy;
void EatArgv(const char* optspec, const char* long_alias);
void Clear();
Opt2Param& GetInternal(char opt, const char* defValue, const char* helpUsage, bool required);
};
|
