aboutsummaryrefslogtreecommitdiffstats
path: root/include/Support/CommandLine.h
blob: 84a3bc9b59df51f6fe43b50cf8a76f715186d5c9 (plain)
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
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
196
197
198
199
200
201
202
203
204
205
206
207
208
209
210
211
212
213
214
215
216
217
218
219
220
221
222
223
224
225
226
227
228
229
230
231
232
233
234
235
236
237
238
239
240
241
242
243
244
245
246
247
248
249
250
251
252
253
254
255
256
257
258
259
260
261
262
263
264
265
266
267
268
269
270
271
272
273
274
275
276
277
278
279
280
281
282
283
284
285
286
287
288
289
290
291
292
293
294
295
296
297
298
299
300
301
302
303
304
305
306
307
308
309
310
311
312
313
314
315
316
317
318
319
320
321
322
323
324
325
326
327
328
329
330
331
332
333
334
335
336
337
338
339
340
341
342
343
344
345
346
347
348
349
350
351
352
353
354
355
356
357
358
359
360
361
362
363
364
365
366
367
368
369
370
371
372
373
374
375
376
377
378
379
380
381
382
383
384
385
386
387
388
389
390
391
392
393
394
395
396
397
398
399
//===- Support/CommandLine.h - Flexible Command line parser ------*- C++ -*--=//
//
// This class implements a command line argument processor that is useful when
// creating a tool.  It provides a simple, minimalistic interface that is easily
// extensible and supports nonlocal (library) command line options.
//
// Note that rather than trying to figure out what this code does, you could try
// reading the library documentation located in docs/CommandLine.html
//
//===----------------------------------------------------------------------===//

#ifndef LLVM_SUPPORT_COMMANDLINE_H
#define LLVM_SUPPORT_COMMANDLINE_H

#include <string>
#include <vector>
#include <utility>
#include <stdarg.h>

namespace cl {   // Short namespace to make usage concise

//===----------------------------------------------------------------------===//
// ParseCommandLineOptions - Minimalistic command line option processing entry
//
void cl::ParseCommandLineOptions(int &argc, char **argv,
				 const char *Overview = 0,
				 int Flags = 0);

// ParserOptions - This set of option is use to control global behavior of the
// command line processor.
//
enum ParserOptions {
  // DisableSingleLetterArgGrouping - With this option enabled, multiple letter
  // options are allowed to bunch together with only a single hyphen for the
  // whole group.  This allows emulation of the behavior that ls uses for
  // example:  ls -la === ls -l -a    Providing this option, disables this.
  //
  DisableSingleLetterArgGrouping = 0x0001,

  // EnableSingleLetterArgValue - This option allows arguments that are
  // otherwise unrecognized to match single letter flags that take a value. 
  // This is useful for cases like a linker, where options are typically of the
  // form '-lfoo' or '-L../../include' where -l or -L are the actual flags.
  //
  EnableSingleLetterArgValue     = 0x0002,
};


//===----------------------------------------------------------------------===//
// Global flags permitted to be passed to command line arguments

enum FlagsOptions {
  NoFlags         = 0x00,      // Marker to make explicit that we have no flags
  Default         = 0x00,      // Equally, marker to use the default flags

  GlobalsMask     = 0x80, 
};

enum NumOccurances {           // Flags for the number of occurances allowed...
  Optional        = 0x01,      // Zero or One occurance
  ZeroOrMore      = 0x02,      // Zero or more occurances allowed
  Required        = 0x03,      // One occurance required
  OneOrMore       = 0x04,      // One or more occurances required

  // ConsumeAfter - Marker for a null ("") flag that can be used to indicate
  // that anything that matches the null marker starts a sequence of options
  // that all get sent to the null marker.  Thus, for example, all arguments
  // to LLI are processed until a filename is found.  Once a filename is found,
  // all of the succeeding arguments are passed, unprocessed, to the null flag.
  //
  ConsumeAfter    = 0x05,

  OccurancesMask  = 0x07,
};

enum ValueExpected {           // Is a value required for the option?
  ValueOptional   = 0x08,      // The value can oppear... or not
  ValueRequired   = 0x10,      // The value is required to appear!
  ValueDisallowed = 0x18,      // A value may not be specified (for flags)
  ValueMask       = 0x18,
};

enum OptionHidden {            // Control whether -help shows this option
  NotHidden       = 0x20,      // Option included in --help & --help-hidden
  Hidden          = 0x40,      // -help doesn't, but --help-hidden does
  ReallyHidden    = 0x60,      // Neither --help nor --help-hidden show this arg
  HiddenMask      = 0x60,
};


//===----------------------------------------------------------------------===//
// Option Base class
//
class Alias;
class Option {
  friend void cl::ParseCommandLineOptions(int &, char **, const char *, int);
  friend class Alias;

  // handleOccurances - Overriden by subclasses to handle the value passed into
  // an argument.  Should return true if there was an error processing the
  // argument and the program should exit.
  //
  virtual bool handleOccurance(const char *ArgName, const string &Arg) = 0;

  virtual enum NumOccurances getNumOccurancesFlagDefault() const { 
    return Optional;
  }
  virtual enum ValueExpected getValueExpectedFlagDefault() const {
    return ValueOptional; 
  }
  virtual enum OptionHidden getOptionHiddenFlagDefault() const {
    return NotHidden;
  }

  int NumOccurances;          // The number of times specified
  const int Flags;            // Flags for the argument
public:
  const char * const ArgStr;  // The argument string itself (ex: "help", "o")
  const char * const HelpStr; // The descriptive text message for --help

  inline enum NumOccurances getNumOccurancesFlag() const {
    int NO = Flags & OccurancesMask;
    return NO ? (enum NumOccurances)NO : getNumOccurancesFlagDefault();
  }
  inline enum ValueExpected getValueExpectedFlag() const {
    int VE = Flags & ValueMask;
    return VE ? (enum ValueExpected)VE : getValueExpectedFlagDefault();
  }
  inline enum OptionHidden getOptionHiddenFlag() const {
    int OH = Flags & HiddenMask;
    return OH ? (enum OptionHidden)OH : getOptionHiddenFlagDefault();
  }

protected:
  Option(const char *ArgStr, const char *Message, int Flags);
  Option(int flags) : NumOccurances(0), Flags(flags), ArgStr(""), HelpStr("") {}

public:
  // Return the width of the option tag for printing...
  virtual unsigned getOptionWidth() const;

  // printOptionInfo - Print out information about this option.  The 
  // to-be-maintained width is specified.
  //
  virtual void printOptionInfo(unsigned GlobalWidth) const;

  // addOccurance - Wrapper around handleOccurance that enforces Flags
  //
  bool addOccurance(const char *ArgName, const string &Value);

  // Prints option name followed by message.  Always returns true.
  bool error(string Message, const char *ArgName = 0);

public:
  inline int getNumOccurances() const { return NumOccurances; }
  virtual ~Option() {}
};


//===----------------------------------------------------------------------===//
// Aliased command line option (alias this name to a preexisting name)
//
class Alias : public Option {
  Option &AliasFor;
  virtual bool handleOccurance(const char *ArgName, const string &Arg) {
    return AliasFor.handleOccurance(AliasFor.ArgStr, Arg);
  }
  virtual enum OptionHidden getOptionHiddenFlagDefault() const {return Hidden;}
public:
  inline Alias(const char *ArgStr, const char *Message, int Flags,
	       Option &aliasFor) : Option(ArgStr, Message, Flags), 
				   AliasFor(aliasFor) {}
};

//===----------------------------------------------------------------------===//
// Boolean/flag command line option
//
class Flag : public Option {
  bool Value;
  virtual bool handleOccurance(const char *ArgName, const string &Arg);
public:
  inline Flag(const char *ArgStr, const char *Message, int Flags = 0, 
	      bool DefaultVal = 0) : Option(ArgStr, Message, Flags), 
				     Value(DefaultVal) {}
  operator const bool() const { return Value; }
  inline bool operator=(bool Val) { Value = Val; return Val; }
};



//===----------------------------------------------------------------------===//
// Integer valued command line option
//
class Int : public Option {
  int Value;
  virtual bool handleOccurance(const char *ArgName, const string &Arg);
  virtual enum ValueExpected getValueExpectedFlagDefault() const {
    return ValueRequired; 
  }
public:
  inline Int(const char *ArgStr, const char *Help, int Flags = 0,
	     int DefaultVal = 0) : Option(ArgStr, Help, Flags),
				   Value(DefaultVal) {}
  inline operator int() const { return Value; }
  inline int operator=(int Val) { Value = Val; return Val; }
};


//===----------------------------------------------------------------------===//
// String valued command line option
//
class String : public Option, public string {
  virtual bool handleOccurance(const char *ArgName, const string &Arg);
  virtual enum ValueExpected getValueExpectedFlagDefault() const {
    return ValueRequired; 
  }
public:
  inline String(const char *ArgStr, const char *Help, int Flags = 0, 
		const char *DefaultVal = "") 
    : Option(ArgStr, Help, Flags), string(DefaultVal) {}

  inline const string &operator=(const string &Val) { 
    return string::operator=(Val);
  }
};


//===----------------------------------------------------------------------===//
// String list command line option
//
class StringList : public Option, public vector<string> {

  virtual enum NumOccurances getNumOccurancesFlagDefault() const { 
    return ZeroOrMore;
  }
  virtual enum ValueExpected getValueExpectedFlagDefault() const {
    return ValueRequired;
  }
  virtual bool handleOccurance(const char *ArgName, const string &Arg);

public:
  inline StringList(const char *ArgStr, const char *Help, int Flags = 0)
    : Option(ArgStr, Help, Flags) {}
};


//===----------------------------------------------------------------------===//
// Enum valued command line option
//
#define clEnumVal(ENUMVAL, DESC) #ENUMVAL, ENUMVAL, DESC
#define clEnumValN(ENUMVAL, FLAGNAME, DESC) FLAGNAME, ENUMVAL, DESC

// EnumBase - Base class for all enum/varargs related argument types...
class EnumBase : public Option {
protected:
  // Use a vector instead of a map, because the lists should be short,
  // the overhead is less, and most importantly, it keeps them in the order
  // inserted so we can print our option out nicely.
  vector<pair<const char *, pair<int, const char *> > > ValueMap;

  inline EnumBase(const char *ArgStr, const char *Help, int Flags)
    : Option(ArgStr, Help, Flags) {}
  inline EnumBase(int Flags) : Option(Flags) {}

  // processValues - Incorporate the specifed varargs arglist into the 
  // ValueMap.
  //
  void processValues(va_list Vals);

  // registerArgs - notify the system about these new arguments
  void registerArgs();

public:
  // Turn an enum into the arg name that activates it
  const char *getArgName(int ID) const;
  const char *getArgDescription(int ID) const;
};

class EnumValueBase : public EnumBase {
protected:
  int Value;
  inline EnumValueBase(const char *ArgStr, const char *Help, int Flags)
    : EnumBase(ArgStr, Help, Flags) {}
  inline EnumValueBase(int Flags) : EnumBase(Flags) {}

  // handleOccurance - Set Value to the enum value specified by Arg
  virtual bool handleOccurance(const char *ArgName, const string &Arg);

  // Return the width of the option tag for printing...
  virtual unsigned getOptionWidth() const;

  // printOptionInfo - Print out information about this option.  The 
  // to-be-maintained width is specified.
  //
  virtual void printOptionInfo(unsigned GlobalWidth) const;
};

template <class E>  // The enum we are representing
class Enum : public EnumValueBase {
  virtual enum ValueExpected getValueExpectedFlagDefault() const {
    return ValueRequired;
  }
public:
  inline Enum(const char *ArgStr, int Flags, const char *Help, ...)
    : EnumValueBase(ArgStr, Help, Flags) {
    va_list Values;
    va_start(Values, Help);
    processValues(Values);
    va_end(Values);
    Value = ValueMap.front().second.first; // Grab default value
  }

  inline operator E() const { return (E)Value; }
  inline E operator=(E Val) { Value = Val; return Val; }
};


//===----------------------------------------------------------------------===//
// Enum flags command line option
//
class EnumFlagsBase : public EnumValueBase {
  virtual enum ValueExpected getValueExpectedFlagDefault() const {
    return ValueDisallowed;
  }
protected:
  virtual bool handleOccurance(const char *ArgName, const string &Arg);
  inline EnumFlagsBase(int Flags) : EnumValueBase(Flags) {}

  // Return the width of the option tag for printing...
  virtual unsigned getOptionWidth() const;

  // printOptionInfo - Print out information about this option.  The 
  // to-be-maintained width is specified.
  //
  virtual void printOptionInfo(unsigned GlobalWidth) const;
};

template <class E>  // The enum we are representing
class EnumFlags : public EnumFlagsBase {
public:
  inline EnumFlags(int Flags, ...) : EnumFlagsBase(Flags) {
    va_list Values;
    va_start(Values, Flags);
    processValues(Values);
    va_end(Values);
    registerArgs();
    Value = ValueMap.front().second.first; // Grab default value
  }

  inline operator E() const { return (E)Value; }
  inline E operator=(E Val) { Value = Val; return Val; }
};


//===----------------------------------------------------------------------===//
// Enum list command line option
//
class EnumListBase : public EnumBase {
  virtual enum NumOccurances getNumOccurancesFlagDefault() const { 
    return ZeroOrMore;
  }
  virtual enum ValueExpected getValueExpectedFlagDefault() const {
    return ValueDisallowed;
  }
protected:
  vector<int> Values;  // The options specified so far.

  inline EnumListBase(int Flags) 
    : EnumBase(Flags) {}
  virtual bool handleOccurance(const char *ArgName, const string &Arg);

  // Return the width of the option tag for printing...
  virtual unsigned getOptionWidth() const;

  // printOptionInfo - Print out information about this option.  The 
  // to-be-maintained width is specified.
  //
  virtual void printOptionInfo(unsigned GlobalWidth) const;
public:
  inline unsigned size() { return Values.size(); }
};

template <class E>  // The enum we are representing
class EnumList : public EnumListBase {
public:
  inline EnumList(int Flags, ...) : EnumListBase(Flags) {
    va_list Values;
    va_start(Values, Flags);
    processValues(Values);
    va_end(Values);
    registerArgs();
  }
  inline E  operator[](unsigned i) const { return (E)Values[i]; }
  inline E &operator[](unsigned i)       { return (E&)Values[i]; }
};

} // End namespace cl

#endif