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
|
#include "opt.h"
#include <util/system/progname.h>
#include <ctype.h>
using namespace NLastGetopt;
namespace {
struct TOptsNoDefault: public TOpts {
TOptsNoDefault(const TStringBuf& optstring = TStringBuf())
: TOpts(optstring)
{
}
};
}
void Opt::Init(int argc, char* argv[], const char* optString, const Ion* longOptions, bool longOnly, bool isOpen) {
Ions_ = longOptions;
Err = true;
GotError_ = false;
Ind = argc;
Opts_.Reset(new TOptsNoDefault(optString));
for (const Ion* o = longOptions; o != nullptr && o->name != nullptr; ++o) {
TOpt* opt;
if ((unsigned)o->val < 0x80 && isalnum(o->val)) {
opt = &Opts_->CharOption(char(o->val));
opt->AddLongName(o->name);
} else {
Opts_->AddLongOption(o->name);
opt = const_cast<TOpt*>(&Opts_->GetLongOption(o->name));
}
opt->HasArg_ = EHasArg(o->has_arg);
opt->UserValue(o);
}
Opts_->AllowSingleDashForLong_ = longOnly;
Opts_->AllowPlusForLong_ = true;
Opts_->AllowUnknownCharOptions_ = isOpen;
Opts_->AllowUnknownLongOptions_ = false;
OptsParser_.Reset(new TOptsParser(Opts_.Get(), argc, argv));
}
Opt::Opt(int argc, char* argv[], const char* optString, const Ion* longOptions, bool longOnly, bool isOpen) {
Init(argc, argv, optString, longOptions, longOnly, isOpen);
}
Opt::Opt(int argc, const char* argv[], const char* optString, const Ion* longOptions, bool longOnly, bool isOpen) {
Init(argc, (char**)argv, optString, longOptions, longOnly, isOpen);
}
int Opt::Get() {
return Get(nullptr);
}
int Opt::Get(int* longOptionIndex) {
if (GotError_)
return EOF;
Arg = nullptr;
try {
bool r = OptsParser_->Next();
Ind = (int)OptsParser_->Pos_;
if (!r) {
return EOF;
} else {
Arg = (char*)OptsParser_->CurVal();
if (!OptsParser_->CurOpt()) {
// possible if RETURN_IN_ORDER
return 1;
} else {
const Ion* ion = (const Ion*)OptsParser_->CurOpt()->UserValue();
if (longOptionIndex) {
*longOptionIndex = int(ion - Ions_);
}
char c = OptsParser_->CurOpt()->GetCharOr0();
return c != 0 ? c : ion->val;
}
}
} catch (const NLastGetopt::TException&) {
GotError_ = true;
if (Err)
Cerr << CurrentExceptionMessage() << Endl;
return '?';
}
}
void Opt::DummyHelp(IOutputStream& os) {
Opts_->PrintUsage(GetProgramName(), os);
}
int Opt::GetArgC() const {
return (int)OptsParser_->Argc_;
}
const char** Opt::GetArgV() const {
return OptsParser_->Argv_;
}
int opt_get_number(int& argc, char* argv[]) {
int num = -1;
for (int a = 1; a < argc; a++) {
if (*argv[a] == '-' && isdigit((ui8)argv[a][1])) {
char* ne;
num = strtol(argv[a] + 1, &ne, 10);
if (*ne) {
memmove(argv[a] + 1, ne, strlen(ne) + 1);
} else {
for (argc--; a < argc; a++)
argv[a] = argv[a + 1];
}
break;
}
}
return num;
}
|
