1//===- llvm/Support/CommandLine.h - Command line handler --------*- C++ -*-===//
2//
3// Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions.
4// See https://llvm.org/LICENSE.txt for license information.
5// SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
6//
7//===----------------------------------------------------------------------===//
8//
9// This class implements a command line argument processor that is useful when
10// creating a tool. It provides a simple, minimalistic interface that is easily
11// extensible and supports nonlocal (library) command line options.
12//
13// Note that rather than trying to figure out what this code does, you should
14// read the library documentation located in docs/CommandLine.html or looks at
15// the many example usages in tools/*/*.cpp
16//
17//===----------------------------------------------------------------------===//
18
19#ifndef LLVM_SUPPORT_COMMANDLINE_H
20#define LLVM_SUPPORT_COMMANDLINE_H
21
22#include "llvm/ADT/ArrayRef.h"
23#include "llvm/ADT/STLExtras.h"
24#include "llvm/ADT/SmallPtrSet.h"
25#include "llvm/ADT/SmallVector.h"
26#include "llvm/ADT/StringMap.h"
27#include "llvm/ADT/StringRef.h"
28#include "llvm/ADT/Twine.h"
29#include "llvm/ADT/iterator_range.h"
30#include "llvm/Support/Compiler.h"
31#include "llvm/Support/ErrorHandling.h"
32#include "llvm/Support/StringSaver.h"
33#include "llvm/Support/raw_ostream.h"
34#include <cassert>
35#include <climits>
36#include <cstddef>
37#include <functional>
38#include <initializer_list>
39#include <string>
40#include <type_traits>
41#include <vector>
42
43namespace llvm {
44
45namespace vfs {
46class FileSystem;
47}
48
49class StringSaver;
50
51/// This namespace contains all of the command line option processing machinery.
52/// It is intentionally a short name to make qualified usage concise.
53namespace cl {
54
55//===----------------------------------------------------------------------===//
56// Command line option processing entry point.
57//
58// Returns true on success. Otherwise, this will print the error message to
59// stderr and exit if \p Errs is not set (nullptr by default), or print the
60// error message to \p Errs and return false if \p Errs is provided.
61//
62// If EnvVar is not nullptr, command-line options are also parsed from the
63// environment variable named by EnvVar. Precedence is given to occurrences
64// from argv. This precedence is currently implemented by parsing argv after
65// the environment variable, so it is only implemented correctly for options
66// that give precedence to later occurrences. If your program supports options
67// that give precedence to earlier occurrences, you will need to extend this
68// function to support it correctly.
69LLVM_ABI bool ParseCommandLineOptions(int argc, const char *const *argv,
70 StringRef Overview = "",
71 raw_ostream *Errs = nullptr,
72 const char *EnvVar = nullptr,
73 bool LongOptionsUseDoubleDash = false);
74
75// Function pointer type for printing version information.
76using VersionPrinterTy = std::function<void(raw_ostream &)>;
77
78///===---------------------------------------------------------------------===//
79/// Override the default (LLVM specific) version printer used to print out the
80/// version when --version is given on the command line. This allows other
81/// systems using the CommandLine utilities to print their own version string.
82LLVM_ABI void SetVersionPrinter(VersionPrinterTy func);
83
84///===---------------------------------------------------------------------===//
85/// Add an extra printer to use in addition to the default one. This can be
86/// called multiple times, and each time it adds a new function to the list
87/// which will be called after the basic LLVM version printing is complete.
88/// Each can then add additional information specific to the tool.
89LLVM_ABI void AddExtraVersionPrinter(VersionPrinterTy func);
90
91// Print option values.
92// With -print-options print the difference between option values and defaults.
93// With -print-all-options print all option values.
94// (Currently not perfect, but best-effort.)
95LLVM_ABI void PrintOptionValues();
96
97// Forward declaration - AddLiteralOption needs to be up here to make gcc happy.
98class Option;
99
100/// Adds a new option for parsing and provides the option it refers to.
101///
102/// \param O pointer to the option
103/// \param Name the string name for the option to handle during parsing
104///
105/// Literal options are used by some parsers to register special option values.
106/// This is how the PassNameParser registers pass names for opt.
107LLVM_ABI void AddLiteralOption(Option &O, StringRef Name);
108
109//===----------------------------------------------------------------------===//
110// Flags permitted to be passed to command line arguments
111//
112
113enum NumOccurrencesFlag { // Flags for the number of occurrences allowed
114 Optional = 0x00, // Zero or One occurrence
115 ZeroOrMore = 0x01, // Zero or more occurrences allowed
116 Required = 0x02, // One occurrence required
117 OneOrMore = 0x03, // One or more occurrences required
118
119 // Indicates that this option is fed anything that follows the last positional
120 // argument required by the application (it is an error if there are zero
121 // positional arguments, and a ConsumeAfter option is used).
122 // Thus, for example, all arguments to LLI are processed until a filename is
123 // found. Once a filename is found, all of the succeeding arguments are
124 // passed, unprocessed, to the ConsumeAfter option.
125 //
126 ConsumeAfter = 0x04
127};
128
129enum ValueExpected { // Is a value required for the option?
130 // zero reserved for the unspecified value
131 ValueOptional = 0x01, // The value can appear... or not
132 ValueRequired = 0x02, // The value is required to appear!
133 ValueDisallowed = 0x03 // A value may not be specified (for flags)
134};
135
136enum OptionHidden { // Control whether -help shows this option
137 NotHidden = 0x00, // Option included in -help & -help-hidden
138 Hidden = 0x01, // -help doesn't, but -help-hidden does
139 ReallyHidden = 0x02 // Neither -help nor -help-hidden show this arg
140};
141
142// This controls special features that the option might have that cause it to be
143// parsed differently...
144//
145// Prefix - This option allows arguments that are otherwise unrecognized to be
146// matched by options that are a prefix of the actual value. This is useful for
147// cases like a linker, where options are typically of the form '-lfoo' or
148// '-L../../include' where -l or -L are the actual flags. When prefix is
149// enabled, and used, the value for the flag comes from the suffix of the
150// argument.
151//
152// AlwaysPrefix - Only allow the behavior enabled by the Prefix flag and reject
153// the Option=Value form.
154//
155
156enum FormattingFlags {
157 NormalFormatting = 0x00, // Nothing special
158 Positional = 0x01, // Is a positional argument, no '-' required
159 Prefix = 0x02, // Can this option directly prefix its value?
160 AlwaysPrefix = 0x03 // Can this option only directly prefix its value?
161};
162
163enum MiscFlags { // Miscellaneous flags to adjust argument
164 CommaSeparated = 0x01, // Should this cl::list split between commas?
165 PositionalEatsArgs = 0x02, // Should this positional cl::list eat -args?
166 Sink = 0x04, // Should this cl::list eat all unknown options?
167
168 // Can this option group with other options?
169 // If this is enabled, multiple letter options are allowed to bunch together
170 // with only a single hyphen for the whole group. This allows emulation
171 // of the behavior that ls uses for example: ls -la === ls -l -a
172 Grouping = 0x08,
173
174 // Default option
175 DefaultOption = 0x10
176};
177
178//===----------------------------------------------------------------------===//
179//
180class OptionCategory {
181private:
182 StringRef const Name;
183 StringRef const Description;
184
185 LLVM_ABI void registerCategory();
186
187public:
188 OptionCategory(StringRef const Name,
189 StringRef const Description = "")
190 : Name(Name), Description(Description) {
191 registerCategory();
192 }
193
194 StringRef getName() const { return Name; }
195 StringRef getDescription() const { return Description; }
196};
197
198// The general Option Category (used as default category).
199LLVM_ABI OptionCategory &getGeneralCategory();
200
201//===----------------------------------------------------------------------===//
202//
203class SubCommand {
204private:
205 StringRef Name;
206 StringRef Description;
207
208protected:
209 LLVM_ABI void registerSubCommand();
210 LLVM_ABI void unregisterSubCommand();
211
212public:
213 SubCommand(StringRef Name, StringRef Description = "")
214 : Name(Name), Description(Description) {
215 registerSubCommand();
216 }
217 SubCommand() = default;
218
219 // Get the special subcommand representing no subcommand.
220 LLVM_ABI static SubCommand &getTopLevel();
221
222 // Get the special subcommand that can be used to put an option into all
223 // subcommands.
224 LLVM_ABI static SubCommand &getAll();
225
226 LLVM_ABI void reset();
227
228 LLVM_ABI explicit operator bool() const;
229
230 StringRef getName() const { return Name; }
231 StringRef getDescription() const { return Description; }
232
233 SmallVector<Option *, 4> PositionalOpts;
234 SmallVector<Option *, 4> SinkOpts;
235 StringMap<Option *> OptionsMap;
236
237 Option *ConsumeAfterOpt = nullptr; // The ConsumeAfter option if it exists.
238};
239
240class SubCommandGroup {
241 SmallVector<SubCommand *, 4> Subs;
242
243public:
244 SubCommandGroup(std::initializer_list<SubCommand *> IL) : Subs(IL) {}
245
246 ArrayRef<SubCommand *> getSubCommands() const { return Subs; }
247};
248
249//===----------------------------------------------------------------------===//
250//
251class LLVM_ABI Option {
252 friend class alias;
253
254 // Overriden by subclasses to handle the value passed into an argument. Should
255 // return true if there was an error processing the argument and the program
256 // should exit.
257 //
258 virtual bool handleOccurrence(unsigned pos, StringRef ArgName,
259 StringRef Arg) = 0;
260
261 virtual enum ValueExpected getValueExpectedFlagDefault() const {
262 return ValueOptional;
263 }
264
265 // Out of line virtual function to provide home for the class.
266 virtual void anchor();
267
268 uint16_t NumOccurrences; // The number of times specified
269 // Occurrences, HiddenFlag, and Formatting are all enum types but to avoid
270 // problems with signed enums in bitfields.
271 uint16_t Occurrences : 3; // enum NumOccurrencesFlag
272 // not using the enum type for 'Value' because zero is an implementation
273 // detail representing the non-value
274 uint16_t Value : 2;
275 uint16_t HiddenFlag : 2; // enum OptionHidden
276 uint16_t Formatting : 2; // enum FormattingFlags
277 uint16_t Misc : 5;
278 uint16_t FullyInitialized : 1; // Has addArgument been called?
279 uint16_t Position; // Position of last occurrence of the option
280 uint16_t AdditionalVals; // Greater than 0 for multi-valued option.
281
282public:
283 StringRef ArgStr; // The argument string itself (ex: "help", "o")
284 StringRef HelpStr; // The descriptive text message for -help
285 StringRef ValueStr; // String describing what the value of this option is
286 SmallVector<OptionCategory *, 1>
287 Categories; // The Categories this option belongs to
288 SmallPtrSet<SubCommand *, 1> Subs; // The subcommands this option belongs to.
289
290 inline enum NumOccurrencesFlag getNumOccurrencesFlag() const {
291 return (enum NumOccurrencesFlag)Occurrences;
292 }
293
294 inline enum ValueExpected getValueExpectedFlag() const {
295 return Value ? ((enum ValueExpected)Value) : getValueExpectedFlagDefault();
296 }
297
298 inline enum OptionHidden getOptionHiddenFlag() const {
299 return (enum OptionHidden)HiddenFlag;
300 }
301
302 inline enum FormattingFlags getFormattingFlag() const {
303 return (enum FormattingFlags)Formatting;
304 }
305
306 inline unsigned getMiscFlags() const { return Misc; }
307 inline unsigned getPosition() const { return Position; }
308 inline unsigned getNumAdditionalVals() const { return AdditionalVals; }
309
310 // Return true if the argstr != ""
311 bool hasArgStr() const { return !ArgStr.empty(); }
312 bool isPositional() const { return getFormattingFlag() == cl::Positional; }
313 bool isSink() const { return getMiscFlags() & cl::Sink; }
314 bool isDefaultOption() const { return getMiscFlags() & cl::DefaultOption; }
315
316 bool isConsumeAfter() const {
317 return getNumOccurrencesFlag() == cl::ConsumeAfter;
318 }
319
320 //-------------------------------------------------------------------------===
321 // Accessor functions set by OptionModifiers
322 //
323 void setArgStr(StringRef S);
324 void setDescription(StringRef S) { HelpStr = S; }
325 void setValueStr(StringRef S) { ValueStr = S; }
326 void setNumOccurrencesFlag(enum NumOccurrencesFlag Val) { Occurrences = Val; }
327 void setValueExpectedFlag(enum ValueExpected Val) { Value = Val; }
328 void setHiddenFlag(enum OptionHidden Val) { HiddenFlag = Val; }
329 void setFormattingFlag(enum FormattingFlags V) { Formatting = V; }
330 void setMiscFlag(enum MiscFlags M) { Misc |= M; }
331 void setPosition(unsigned pos) { Position = pos; }
332 void addCategory(OptionCategory &C);
333 void addSubCommand(SubCommand &S) { Subs.insert(Ptr: &S); }
334
335protected:
336 explicit Option(enum NumOccurrencesFlag OccurrencesFlag,
337 enum OptionHidden Hidden)
338 : NumOccurrences(0), Occurrences(OccurrencesFlag), Value(0),
339 HiddenFlag(Hidden), Formatting(NormalFormatting), Misc(0),
340 FullyInitialized(false), Position(0), AdditionalVals(0) {
341 Categories.push_back(Elt: &getGeneralCategory());
342 }
343
344 inline void setNumAdditionalVals(unsigned n) { AdditionalVals = n; }
345
346public:
347 virtual ~Option() = default;
348
349 // Register this argument with the commandline system.
350 //
351 void addArgument();
352
353 /// Unregisters this option from the CommandLine system.
354 ///
355 /// This option must have been the last option registered.
356 /// For testing purposes only.
357 void removeArgument();
358
359 // Return the width of the option tag for printing...
360 virtual size_t getOptionWidth() const = 0;
361
362 // Print out information about this option. The to-be-maintained width is
363 // specified.
364 //
365 virtual void printOptionInfo(size_t GlobalWidth) const = 0;
366
367 virtual void printOptionValue(size_t GlobalWidth, bool Force) const = 0;
368
369 virtual void setDefault() = 0;
370
371 // Prints the help string for an option.
372 //
373 // This maintains the Indent for multi-line descriptions.
374 // FirstLineIndentedBy is the count of chars of the first line
375 // i.e. the one containing the --<option name>.
376 static void printHelpStr(StringRef HelpStr, size_t Indent,
377 size_t FirstLineIndentedBy);
378
379 // Prints the help string for an enum value.
380 //
381 // This maintains the Indent for multi-line descriptions.
382 // FirstLineIndentedBy is the count of chars of the first line
383 // i.e. the one containing the =<value>.
384 static void printEnumValHelpStr(StringRef HelpStr, size_t Indent,
385 size_t FirstLineIndentedBy);
386
387 virtual void getExtraOptionNames(SmallVectorImpl<StringRef> &) {}
388
389 // Wrapper around handleOccurrence that enforces Flags.
390 //
391 virtual bool addOccurrence(unsigned pos, StringRef ArgName, StringRef Value,
392 bool MultiArg = false);
393
394 // Prints option name followed by message. Always returns true.
395 bool error(const Twine &Message, StringRef ArgName = StringRef(), raw_ostream &Errs = llvm::errs());
396 bool error(const Twine &Message, raw_ostream &Errs) {
397 return error(Message, ArgName: StringRef(), Errs);
398 }
399
400 inline int getNumOccurrences() const { return NumOccurrences; }
401 void reset();
402};
403
404//===----------------------------------------------------------------------===//
405// Command line option modifiers that can be used to modify the behavior of
406// command line option parsers...
407//
408
409// Modifier to set the description shown in the -help output...
410struct desc {
411 StringRef Desc;
412
413 desc(StringRef Str) : Desc(Str) {}
414
415 void apply(Option &O) const { O.setDescription(Desc); }
416};
417
418// Modifier to set the value description shown in the -help output...
419struct value_desc {
420 StringRef Desc;
421
422 value_desc(StringRef Str) : Desc(Str) {}
423
424 void apply(Option &O) const { O.setValueStr(Desc); }
425};
426
427// Specify a default (initial) value for the command line argument, if the
428// default constructor for the argument type does not give you what you want.
429// This is only valid on "opt" arguments, not on "list" arguments.
430template <class Ty> struct initializer {
431 const Ty &Init;
432 initializer(const Ty &Val) : Init(Val) {}
433
434 template <class Opt> void apply(Opt &O) const { O.setInitialValue(Init); }
435};
436
437template <class Ty> struct list_initializer {
438 ArrayRef<Ty> Inits;
439 list_initializer(ArrayRef<Ty> Vals) : Inits(Vals) {}
440
441 template <class Opt> void apply(Opt &O) const { O.setInitialValues(Inits); }
442};
443
444template <class Ty> initializer<Ty> init(const Ty &Val) {
445 return initializer<Ty>(Val);
446}
447
448template <class Ty>
449list_initializer<Ty> list_init(ArrayRef<Ty> Vals) {
450 return list_initializer<Ty>(Vals);
451}
452
453// Allow the user to specify which external variable they want to store the
454// results of the command line argument processing into, if they don't want to
455// store it in the option itself.
456template <class Ty> struct LocationClass {
457 Ty &Loc;
458
459 LocationClass(Ty &L) : Loc(L) {}
460
461 template <class Opt> void apply(Opt &O) const { O.setLocation(O, Loc); }
462};
463
464template <class Ty> LocationClass<Ty> location(Ty &L) {
465 return LocationClass<Ty>(L);
466}
467
468// Specify the Option category for the command line argument to belong to.
469struct cat {
470 OptionCategory &Category;
471
472 cat(OptionCategory &c) : Category(c) {}
473
474 template <class Opt> void apply(Opt &O) const { O.addCategory(Category); }
475};
476
477// Specify the subcommand that this option belongs to.
478struct sub {
479 SubCommand *Sub = nullptr;
480 SubCommandGroup *Group = nullptr;
481
482 sub(SubCommand &S) : Sub(&S) {}
483 sub(SubCommandGroup &G) : Group(&G) {}
484
485 template <class Opt> void apply(Opt &O) const {
486 if (Sub)
487 O.addSubCommand(*Sub);
488 else if (Group)
489 for (SubCommand *SC : Group->getSubCommands())
490 O.addSubCommand(*SC);
491 }
492};
493
494// Specify a callback function to be called when an option is seen.
495// Can be used to set other options automatically.
496template <typename R, typename Ty> struct cb {
497 std::function<R(Ty)> CB;
498
499 cb(std::function<R(Ty)> CB) : CB(CB) {}
500
501 template <typename Opt> void apply(Opt &O) const { O.setCallback(CB); }
502};
503
504namespace detail {
505template <typename F>
506struct callback_traits : public callback_traits<decltype(&F::operator())> {};
507
508template <typename R, typename C, typename... Args>
509struct callback_traits<R (C::*)(Args...) const> {
510 using result_type = R;
511 using arg_type = std::tuple_element_t<0, std::tuple<Args...>>;
512 static_assert(sizeof...(Args) == 1, "callback function must have one and only one parameter");
513 static_assert(std::is_same_v<result_type, void>,
514 "callback return type must be void");
515 static_assert(std::is_lvalue_reference_v<arg_type> &&
516 std::is_const_v<std::remove_reference_t<arg_type>>,
517 "callback arg_type must be a const lvalue reference");
518};
519} // namespace detail
520
521template <typename F>
522cb<typename detail::callback_traits<F>::result_type,
523 typename detail::callback_traits<F>::arg_type>
524callback(F CB) {
525 using result_type = typename detail::callback_traits<F>::result_type;
526 using arg_type = typename detail::callback_traits<F>::arg_type;
527 return cb<result_type, arg_type>(CB);
528}
529
530//===----------------------------------------------------------------------===//
531
532// Support value comparison outside the template.
533struct LLVM_ABI GenericOptionValue {
534 virtual bool compare(const GenericOptionValue &V) const = 0;
535
536protected:
537 GenericOptionValue() = default;
538 GenericOptionValue(const GenericOptionValue&) = default;
539 GenericOptionValue &operator=(const GenericOptionValue &) = default;
540 ~GenericOptionValue() = default;
541
542private:
543 virtual void anchor();
544};
545
546template <class DataType> struct OptionValue;
547
548// The default value safely does nothing. Option value printing is only
549// best-effort.
550template <class DataType, bool isClass>
551struct OptionValueBase : public GenericOptionValue {
552 // Temporary storage for argument passing.
553 using WrapperType = OptionValue<DataType>;
554
555 bool hasValue() const { return false; }
556
557 const DataType &getValue() const { llvm_unreachable("no default value"); }
558
559 // Some options may take their value from a different data type.
560 template <class DT> void setValue(const DT & /*V*/) {}
561
562 // Returns whether this instance matches the argument.
563 bool compare(const DataType & /*V*/) const { return false; }
564
565 bool compare(const GenericOptionValue & /*V*/) const override {
566 return false;
567 }
568
569protected:
570 ~OptionValueBase() = default;
571};
572
573// Simple copy of the option value.
574template <class DataType> class OptionValueCopy : public GenericOptionValue {
575 DataType Value;
576 bool Valid = false;
577
578protected:
579 OptionValueCopy(const OptionValueCopy&) = default;
580 OptionValueCopy &operator=(const OptionValueCopy &) = default;
581 ~OptionValueCopy() = default;
582
583public:
584 OptionValueCopy() = default;
585
586 bool hasValue() const { return Valid; }
587
588 const DataType &getValue() const {
589 assert(Valid && "invalid option value");
590 return Value;
591 }
592
593 void setValue(const DataType &V) {
594 Valid = true;
595 Value = V;
596 }
597
598 // Returns whether this instance matches V.
599 bool compare(const DataType &V) const { return Valid && (Value == V); }
600
601 bool compare(const GenericOptionValue &V) const override {
602 const OptionValueCopy<DataType> &VC =
603 static_cast<const OptionValueCopy<DataType> &>(V);
604 if (!VC.hasValue())
605 return false;
606 return compare(VC.getValue());
607 }
608};
609
610// Non-class option values.
611template <class DataType>
612struct OptionValueBase<DataType, false> : OptionValueCopy<DataType> {
613 using WrapperType = DataType;
614
615protected:
616 OptionValueBase() = default;
617 OptionValueBase(const OptionValueBase&) = default;
618 OptionValueBase &operator=(const OptionValueBase &) = default;
619 ~OptionValueBase() = default;
620};
621
622// Top-level option class.
623template <class DataType>
624struct OptionValue final
625 : OptionValueBase<DataType, std::is_class_v<DataType>> {
626 OptionValue() = default;
627
628 OptionValue(const DataType &V) { this->setValue(V); }
629
630 // Some options may take their value from a different data type.
631 template <class DT> OptionValue<DataType> &operator=(const DT &V) {
632 this->setValue(V);
633 return *this;
634 }
635};
636
637// Other safe-to-copy-by-value common option types.
638enum boolOrDefault { BOU_UNSET, BOU_TRUE, BOU_FALSE };
639template <>
640struct LLVM_ABI OptionValue<cl::boolOrDefault> final
641 : OptionValueCopy<cl::boolOrDefault> {
642 using WrapperType = cl::boolOrDefault;
643
644 OptionValue() = default;
645
646 OptionValue(const cl::boolOrDefault &V) { this->setValue(V); }
647
648 OptionValue<cl::boolOrDefault> &operator=(const cl::boolOrDefault &V) {
649 setValue(V);
650 return *this;
651 }
652
653private:
654 void anchor() override;
655};
656
657template <>
658struct LLVM_ABI OptionValue<std::string> final : OptionValueCopy<std::string> {
659 using WrapperType = StringRef;
660
661 OptionValue() = default;
662
663 OptionValue(const std::string &V) { this->setValue(V); }
664
665 OptionValue<std::string> &operator=(const std::string &V) {
666 setValue(V);
667 return *this;
668 }
669
670private:
671 void anchor() override;
672};
673
674//===----------------------------------------------------------------------===//
675// Enum valued command line option
676//
677
678// This represents a single enum value, using "int" as the underlying type.
679struct OptionEnumValue {
680 StringRef Name;
681 int Value;
682 StringRef Description;
683};
684
685#define clEnumVal(ENUMVAL, DESC) \
686 llvm::cl::OptionEnumValue { #ENUMVAL, int(ENUMVAL), DESC }
687#define clEnumValN(ENUMVAL, FLAGNAME, DESC) \
688 llvm::cl::OptionEnumValue { FLAGNAME, int(ENUMVAL), DESC }
689
690// For custom data types, allow specifying a group of values together as the
691// values that go into the mapping that the option handler uses.
692//
693class ValuesClass {
694 // Use a vector instead of a map, because the lists should be short,
695 // the overhead is less, and most importantly, it keeps them in the order
696 // inserted so we can print our option out nicely.
697 SmallVector<OptionEnumValue, 4> Values;
698
699public:
700 ValuesClass(std::initializer_list<OptionEnumValue> Options)
701 : Values(Options) {}
702
703 template <class Opt> void apply(Opt &O) const {
704 for (const auto &Value : Values)
705 O.getParser().addLiteralOption(Value.Name, Value.Value,
706 Value.Description);
707 }
708};
709
710/// Helper to build a ValuesClass by forwarding a variable number of arguments
711/// as an initializer list to the ValuesClass constructor.
712template <typename... OptsTy> ValuesClass values(OptsTy... Options) {
713 return ValuesClass({Options...});
714}
715
716//===----------------------------------------------------------------------===//
717// Parameterizable parser for different data types. By default, known data types
718// (string, int, bool) have specialized parsers, that do what you would expect.
719// The default parser, used for data types that are not built-in, uses a mapping
720// table to map specific options to values, which is used, among other things,
721// to handle enum types.
722
723//--------------------------------------------------
724// This class holds all the non-generic code that we do not need replicated for
725// every instance of the generic parser. This also allows us to put stuff into
726// CommandLine.cpp
727//
728class LLVM_ABI generic_parser_base {
729protected:
730 class GenericOptionInfo {
731 public:
732 GenericOptionInfo(StringRef name, StringRef helpStr)
733 : Name(name), HelpStr(helpStr) {}
734 StringRef Name;
735 StringRef HelpStr;
736 };
737
738public:
739 generic_parser_base(Option &O) : Owner(O) {}
740
741 virtual ~generic_parser_base() = default;
742 // Base class should have virtual-destructor
743
744 // Virtual function implemented by generic subclass to indicate how many
745 // entries are in Values.
746 //
747 virtual unsigned getNumOptions() const = 0;
748
749 // Return option name N.
750 virtual StringRef getOption(unsigned N) const = 0;
751
752 // Return description N
753 virtual StringRef getDescription(unsigned N) const = 0;
754
755 // Return the width of the option tag for printing...
756 virtual size_t getOptionWidth(const Option &O) const;
757
758 virtual const GenericOptionValue &getOptionValue(unsigned N) const = 0;
759
760 // Print out information about this option. The to-be-maintained width is
761 // specified.
762 //
763 virtual void printOptionInfo(const Option &O, size_t GlobalWidth) const;
764
765 void printGenericOptionDiff(const Option &O, const GenericOptionValue &V,
766 const GenericOptionValue &Default,
767 size_t GlobalWidth) const;
768
769 // Print the value of an option and it's default.
770 //
771 // Template definition ensures that the option and default have the same
772 // DataType (via the same AnyOptionValue).
773 template <class AnyOptionValue>
774 void printOptionDiff(const Option &O, const AnyOptionValue &V,
775 const AnyOptionValue &Default,
776 size_t GlobalWidth) const {
777 printGenericOptionDiff(O, V, Default, GlobalWidth);
778 }
779
780 void initialize() {}
781
782 void getExtraOptionNames(SmallVectorImpl<StringRef> &OptionNames) {
783 // If there has been no argstr specified, that means that we need to add an
784 // argument for every possible option. This ensures that our options are
785 // vectored to us.
786 if (!Owner.hasArgStr())
787 for (unsigned i = 0, e = getNumOptions(); i != e; ++i)
788 OptionNames.push_back(Elt: getOption(N: i));
789 }
790
791 enum ValueExpected getValueExpectedFlagDefault() const {
792 // If there is an ArgStr specified, then we are of the form:
793 //
794 // -opt=O2 or -opt O2 or -optO2
795 //
796 // In which case, the value is required. Otherwise if an arg str has not
797 // been specified, we are of the form:
798 //
799 // -O2 or O2 or -la (where -l and -a are separate options)
800 //
801 // If this is the case, we cannot allow a value.
802 //
803 if (Owner.hasArgStr())
804 return ValueRequired;
805 else
806 return ValueDisallowed;
807 }
808
809 // Return the option number corresponding to the specified
810 // argument string. If the option is not found, getNumOptions() is returned.
811 //
812 unsigned findOption(StringRef Name);
813
814protected:
815 Option &Owner;
816};
817
818// Default parser implementation - This implementation depends on having a
819// mapping of recognized options to values of some sort. In addition to this,
820// each entry in the mapping also tracks a help message that is printed with the
821// command line option for -help. Because this is a simple mapping parser, the
822// data type can be any unsupported type.
823//
824template <class DataType> class parser : public generic_parser_base {
825protected:
826 class OptionInfo : public GenericOptionInfo {
827 public:
828 OptionInfo(StringRef name, DataType v, StringRef helpStr)
829 : GenericOptionInfo(name, helpStr), V(v) {}
830
831 OptionValue<DataType> V;
832 };
833 SmallVector<OptionInfo, 8> Values;
834
835public:
836 parser(Option &O) : generic_parser_base(O) {}
837
838 using parser_data_type = DataType;
839
840 // Implement virtual functions needed by generic_parser_base
841 unsigned getNumOptions() const override { return unsigned(Values.size()); }
842 StringRef getOption(unsigned N) const override { return Values[N].Name; }
843 StringRef getDescription(unsigned N) const override {
844 return Values[N].HelpStr;
845 }
846
847 // Return the value of option name N.
848 const GenericOptionValue &getOptionValue(unsigned N) const override {
849 return Values[N].V;
850 }
851
852 // Return true on error.
853 bool parse(Option &O, StringRef ArgName, StringRef Arg, DataType &V) {
854 StringRef ArgVal;
855 if (Owner.hasArgStr())
856 ArgVal = Arg;
857 else
858 ArgVal = ArgName;
859
860 for (size_t i = 0, e = Values.size(); i != e; ++i)
861 if (Values[i].Name == ArgVal) {
862 V = Values[i].V.getValue();
863 return false;
864 }
865
866 return O.error(Message: "Cannot find option named '" + ArgVal + "'!");
867 }
868
869 /// Add an entry to the mapping table.
870 ///
871 template <class DT>
872 void addLiteralOption(StringRef Name, const DT &V, StringRef HelpStr) {
873#ifndef NDEBUG
874 if (findOption(Name) != Values.size())
875 report_fatal_error(reason: "Option '" + Name + "' already exists!");
876#endif
877 OptionInfo X(Name, static_cast<DataType>(V), HelpStr);
878 Values.push_back(X);
879 AddLiteralOption(Owner, Name);
880 }
881
882 /// Remove the specified option.
883 ///
884 void removeLiteralOption(StringRef Name) {
885 unsigned N = findOption(Name);
886 assert(N != Values.size() && "Option not found!");
887 Values.erase(Values.begin() + N);
888 }
889};
890
891//--------------------------------------------------
892// Super class of parsers to provide boilerplate code
893//
894class LLVM_ABI
895 basic_parser_impl { // non-template implementation of basic_parser<t>
896public:
897 basic_parser_impl(Option &) {}
898
899 virtual ~basic_parser_impl() = default;
900
901 enum ValueExpected getValueExpectedFlagDefault() const {
902 return ValueRequired;
903 }
904
905 void getExtraOptionNames(SmallVectorImpl<StringRef> &) {}
906
907 void initialize() {}
908
909 // Return the width of the option tag for printing...
910 size_t getOptionWidth(const Option &O) const;
911
912 // Print out information about this option. The to-be-maintained width is
913 // specified.
914 //
915 void printOptionInfo(const Option &O, size_t GlobalWidth) const;
916
917 // Print a placeholder for options that don't yet support printOptionDiff().
918 void printOptionNoValue(const Option &O, size_t GlobalWidth) const;
919
920 // Overload in subclass to provide a better default value.
921 virtual StringRef getValueName() const { return "value"; }
922
923 // An out-of-line virtual method to provide a 'home' for this class.
924 virtual void anchor();
925
926protected:
927 // A helper for basic_parser::printOptionDiff.
928 void printOptionName(const Option &O, size_t GlobalWidth) const;
929};
930
931// The real basic parser is just a template wrapper that provides a typedef for
932// the provided data type.
933//
934template <class DataType> class basic_parser : public basic_parser_impl {
935public:
936 using parser_data_type = DataType;
937 using OptVal = OptionValue<DataType>;
938
939 basic_parser(Option &O) : basic_parser_impl(O) {}
940};
941
942//--------------------------------------------------
943
944extern template class LLVM_TEMPLATE_ABI basic_parser<bool>;
945
946template <> class LLVM_ABI parser<bool> : public basic_parser<bool> {
947public:
948 parser(Option &O) : basic_parser(O) {}
949
950 // Return true on error.
951 bool parse(Option &O, StringRef ArgName, StringRef Arg, bool &Val);
952
953 void initialize() {}
954
955 enum ValueExpected getValueExpectedFlagDefault() const {
956 return ValueOptional;
957 }
958
959 // Do not print =<value> at all.
960 StringRef getValueName() const override { return StringRef(); }
961
962 void printOptionDiff(const Option &O, bool V, OptVal Default,
963 size_t GlobalWidth) const;
964
965 // An out-of-line virtual method to provide a 'home' for this class.
966 void anchor() override;
967};
968
969//--------------------------------------------------
970
971extern template class LLVM_TEMPLATE_ABI basic_parser<boolOrDefault>;
972
973template <>
974class LLVM_ABI parser<boolOrDefault> : public basic_parser<boolOrDefault> {
975public:
976 parser(Option &O) : basic_parser(O) {}
977
978 // Return true on error.
979 bool parse(Option &O, StringRef ArgName, StringRef Arg, boolOrDefault &Val);
980
981 enum ValueExpected getValueExpectedFlagDefault() const {
982 return ValueOptional;
983 }
984
985 // Do not print =<value> at all.
986 StringRef getValueName() const override { return StringRef(); }
987
988 void printOptionDiff(const Option &O, boolOrDefault V, OptVal Default,
989 size_t GlobalWidth) const;
990
991 // An out-of-line virtual method to provide a 'home' for this class.
992 void anchor() override;
993};
994
995//--------------------------------------------------
996
997extern template class LLVM_TEMPLATE_ABI basic_parser<int>;
998
999template <> class LLVM_ABI parser<int> : public basic_parser<int> {
1000public:
1001 parser(Option &O) : basic_parser(O) {}
1002
1003 // Return true on error.
1004 bool parse(Option &O, StringRef ArgName, StringRef Arg, int &Val);
1005
1006 // Overload in subclass to provide a better default value.
1007 StringRef getValueName() const override { return "int"; }
1008
1009 void printOptionDiff(const Option &O, int V, OptVal Default,
1010 size_t GlobalWidth) const;
1011
1012 // An out-of-line virtual method to provide a 'home' for this class.
1013 void anchor() override;
1014};
1015
1016//--------------------------------------------------
1017
1018extern template class LLVM_TEMPLATE_ABI basic_parser<long>;
1019
1020template <> class LLVM_ABI parser<long> final : public basic_parser<long> {
1021public:
1022 parser(Option &O) : basic_parser(O) {}
1023
1024 // Return true on error.
1025 bool parse(Option &O, StringRef ArgName, StringRef Arg, long &Val);
1026
1027 // Overload in subclass to provide a better default value.
1028 StringRef getValueName() const override { return "long"; }
1029
1030 void printOptionDiff(const Option &O, long V, OptVal Default,
1031 size_t GlobalWidth) const;
1032
1033 // An out-of-line virtual method to provide a 'home' for this class.
1034 void anchor() override;
1035};
1036
1037//--------------------------------------------------
1038
1039extern template class LLVM_TEMPLATE_ABI basic_parser<long long>;
1040
1041template <> class LLVM_ABI parser<long long> : public basic_parser<long long> {
1042public:
1043 parser(Option &O) : basic_parser(O) {}
1044
1045 // Return true on error.
1046 bool parse(Option &O, StringRef ArgName, StringRef Arg, long long &Val);
1047
1048 // Overload in subclass to provide a better default value.
1049 StringRef getValueName() const override { return "long"; }
1050
1051 void printOptionDiff(const Option &O, long long V, OptVal Default,
1052 size_t GlobalWidth) const;
1053
1054 // An out-of-line virtual method to provide a 'home' for this class.
1055 void anchor() override;
1056};
1057
1058//--------------------------------------------------
1059
1060extern template class LLVM_TEMPLATE_ABI basic_parser<unsigned>;
1061
1062template <> class LLVM_ABI parser<unsigned> : public basic_parser<unsigned> {
1063public:
1064 parser(Option &O) : basic_parser(O) {}
1065
1066 // Return true on error.
1067 bool parse(Option &O, StringRef ArgName, StringRef Arg, unsigned &Val);
1068
1069 // Overload in subclass to provide a better default value.
1070 StringRef getValueName() const override { return "uint"; }
1071
1072 void printOptionDiff(const Option &O, unsigned V, OptVal Default,
1073 size_t GlobalWidth) const;
1074
1075 // An out-of-line virtual method to provide a 'home' for this class.
1076 void anchor() override;
1077};
1078
1079//--------------------------------------------------
1080
1081extern template class LLVM_TEMPLATE_ABI basic_parser<unsigned long>;
1082
1083template <>
1084class LLVM_ABI parser<unsigned long> final
1085 : public basic_parser<unsigned long> {
1086public:
1087 parser(Option &O) : basic_parser(O) {}
1088
1089 // Return true on error.
1090 bool parse(Option &O, StringRef ArgName, StringRef Arg, unsigned long &Val);
1091
1092 // Overload in subclass to provide a better default value.
1093 StringRef getValueName() const override { return "ulong"; }
1094
1095 void printOptionDiff(const Option &O, unsigned long V, OptVal Default,
1096 size_t GlobalWidth) const;
1097
1098 // An out-of-line virtual method to provide a 'home' for this class.
1099 void anchor() override;
1100};
1101
1102//--------------------------------------------------
1103
1104extern template class LLVM_TEMPLATE_ABI basic_parser<unsigned long long>;
1105
1106template <>
1107class LLVM_ABI parser<unsigned long long>
1108 : public basic_parser<unsigned long long> {
1109public:
1110 parser(Option &O) : basic_parser(O) {}
1111
1112 // Return true on error.
1113 bool parse(Option &O, StringRef ArgName, StringRef Arg,
1114 unsigned long long &Val);
1115
1116 // Overload in subclass to provide a better default value.
1117 StringRef getValueName() const override { return "ulong"; }
1118
1119 void printOptionDiff(const Option &O, unsigned long long V, OptVal Default,
1120 size_t GlobalWidth) const;
1121
1122 // An out-of-line virtual method to provide a 'home' for this class.
1123 void anchor() override;
1124};
1125
1126//--------------------------------------------------
1127
1128extern template class LLVM_TEMPLATE_ABI basic_parser<double>;
1129
1130template <> class LLVM_ABI parser<double> : public basic_parser<double> {
1131public:
1132 parser(Option &O) : basic_parser(O) {}
1133
1134 // Return true on error.
1135 bool parse(Option &O, StringRef ArgName, StringRef Arg, double &Val);
1136
1137 // Overload in subclass to provide a better default value.
1138 StringRef getValueName() const override { return "number"; }
1139
1140 void printOptionDiff(const Option &O, double V, OptVal Default,
1141 size_t GlobalWidth) const;
1142
1143 // An out-of-line virtual method to provide a 'home' for this class.
1144 void anchor() override;
1145};
1146
1147//--------------------------------------------------
1148
1149extern template class LLVM_TEMPLATE_ABI basic_parser<float>;
1150
1151template <> class LLVM_ABI parser<float> : public basic_parser<float> {
1152public:
1153 parser(Option &O) : basic_parser(O) {}
1154
1155 // Return true on error.
1156 bool parse(Option &O, StringRef ArgName, StringRef Arg, float &Val);
1157
1158 // Overload in subclass to provide a better default value.
1159 StringRef getValueName() const override { return "number"; }
1160
1161 void printOptionDiff(const Option &O, float V, OptVal Default,
1162 size_t GlobalWidth) const;
1163
1164 // An out-of-line virtual method to provide a 'home' for this class.
1165 void anchor() override;
1166};
1167
1168//--------------------------------------------------
1169
1170extern template class LLVM_TEMPLATE_ABI basic_parser<std::string>;
1171
1172template <>
1173class LLVM_ABI parser<std::string> : public basic_parser<std::string> {
1174public:
1175 parser(Option &O) : basic_parser(O) {}
1176
1177 // Return true on error.
1178 bool parse(Option &, StringRef, StringRef Arg, std::string &Value) {
1179 Value = Arg.str();
1180 return false;
1181 }
1182
1183 // Overload in subclass to provide a better default value.
1184 StringRef getValueName() const override { return "string"; }
1185
1186 void printOptionDiff(const Option &O, StringRef V, const OptVal &Default,
1187 size_t GlobalWidth) const;
1188
1189 // An out-of-line virtual method to provide a 'home' for this class.
1190 void anchor() override;
1191};
1192
1193//--------------------------------------------------
1194
1195extern template class LLVM_TEMPLATE_ABI basic_parser<char>;
1196
1197template <> class LLVM_ABI parser<char> : public basic_parser<char> {
1198public:
1199 parser(Option &O) : basic_parser(O) {}
1200
1201 // Return true on error.
1202 bool parse(Option &, StringRef, StringRef Arg, char &Value) {
1203 Value = Arg[0];
1204 return false;
1205 }
1206
1207 // Overload in subclass to provide a better default value.
1208 StringRef getValueName() const override { return "char"; }
1209
1210 void printOptionDiff(const Option &O, char V, OptVal Default,
1211 size_t GlobalWidth) const;
1212
1213 // An out-of-line virtual method to provide a 'home' for this class.
1214 void anchor() override;
1215};
1216
1217//--------------------------------------------------
1218// This collection of wrappers is the intermediary between class opt and class
1219// parser to handle all the template nastiness.
1220
1221// This overloaded function is selected by the generic parser.
1222template <class ParserClass, class DT>
1223void printOptionDiff(const Option &O, const generic_parser_base &P, const DT &V,
1224 const OptionValue<DT> &Default, size_t GlobalWidth) {
1225 OptionValue<DT> OV = V;
1226 P.printOptionDiff(O, OV, Default, GlobalWidth);
1227}
1228
1229// This is instantiated for basic parsers when the parsed value has a different
1230// type than the option value. e.g. HelpPrinter.
1231template <class ParserDT, class ValDT> struct OptionDiffPrinter {
1232 void print(const Option &O, const parser<ParserDT> &P, const ValDT & /*V*/,
1233 const OptionValue<ValDT> & /*Default*/, size_t GlobalWidth) {
1234 P.printOptionNoValue(O, GlobalWidth);
1235 }
1236};
1237
1238// This is instantiated for basic parsers when the parsed value has the same
1239// type as the option value.
1240template <class DT> struct OptionDiffPrinter<DT, DT> {
1241 void print(const Option &O, const parser<DT> &P, const DT &V,
1242 const OptionValue<DT> &Default, size_t GlobalWidth) {
1243 P.printOptionDiff(O, V, Default, GlobalWidth);
1244 }
1245};
1246
1247// This overloaded function is selected by the basic parser, which may parse a
1248// different type than the option type.
1249template <class ParserClass, class ValDT>
1250void printOptionDiff(
1251 const Option &O,
1252 const basic_parser<typename ParserClass::parser_data_type> &P,
1253 const ValDT &V, const OptionValue<ValDT> &Default, size_t GlobalWidth) {
1254
1255 OptionDiffPrinter<typename ParserClass::parser_data_type, ValDT> printer;
1256 printer.print(O, static_cast<const ParserClass &>(P), V, Default,
1257 GlobalWidth);
1258}
1259
1260//===----------------------------------------------------------------------===//
1261// This class is used because we must use partial specialization to handle
1262// literal string arguments specially (const char* does not correctly respond to
1263// the apply method). Because the syntax to use this is a pain, we have the
1264// 'apply' method below to handle the nastiness...
1265//
1266template <class Mod> struct applicator {
1267 template <class Opt> static void opt(const Mod &M, Opt &O) { M.apply(O); }
1268};
1269
1270// Handle const char* as a special case...
1271template <unsigned n> struct applicator<char[n]> {
1272 template <class Opt> static void opt(StringRef Str, Opt &O) {
1273 O.setArgStr(Str);
1274 }
1275};
1276template <unsigned n> struct applicator<const char[n]> {
1277 template <class Opt> static void opt(StringRef Str, Opt &O) {
1278 O.setArgStr(Str);
1279 }
1280};
1281template <> struct applicator<StringRef > {
1282 template <class Opt> static void opt(StringRef Str, Opt &O) {
1283 O.setArgStr(Str);
1284 }
1285};
1286
1287template <> struct applicator<NumOccurrencesFlag> {
1288 static void opt(NumOccurrencesFlag N, Option &O) {
1289 O.setNumOccurrencesFlag(N);
1290 }
1291};
1292
1293template <> struct applicator<ValueExpected> {
1294 static void opt(ValueExpected VE, Option &O) { O.setValueExpectedFlag(VE); }
1295};
1296
1297template <> struct applicator<OptionHidden> {
1298 static void opt(OptionHidden OH, Option &O) { O.setHiddenFlag(OH); }
1299};
1300
1301template <> struct applicator<FormattingFlags> {
1302 static void opt(FormattingFlags FF, Option &O) { O.setFormattingFlag(FF); }
1303};
1304
1305template <> struct applicator<MiscFlags> {
1306 static void opt(MiscFlags MF, Option &O) {
1307 assert((MF != Grouping || O.ArgStr.size() == 1) &&
1308 "cl::Grouping can only apply to single character Options.");
1309 O.setMiscFlag(MF);
1310 }
1311};
1312
1313// Apply modifiers to an option in a type safe way.
1314template <class Opt, class Mod, class... Mods>
1315void apply(Opt *O, const Mod &M, const Mods &... Ms) {
1316 applicator<Mod>::opt(M, *O);
1317 apply(O, Ms...);
1318}
1319
1320template <class Opt, class Mod> void apply(Opt *O, const Mod &M) {
1321 applicator<Mod>::opt(M, *O);
1322}
1323
1324//===----------------------------------------------------------------------===//
1325// Default storage class definition: external storage. This implementation
1326// assumes the user will specify a variable to store the data into with the
1327// cl::location(x) modifier.
1328//
1329template <class DataType, bool ExternalStorage, bool isClass>
1330class opt_storage {
1331 DataType *Location = nullptr; // Where to store the object...
1332 OptionValue<DataType> Default;
1333
1334 void check_location() const {
1335 assert(Location && "cl::location(...) not specified for a command "
1336 "line option with external storage, "
1337 "or cl::init specified before cl::location()!!");
1338 }
1339
1340public:
1341 opt_storage() = default;
1342
1343 bool setLocation(Option &O, DataType &L) {
1344 if (Location)
1345 return O.error(Message: "cl::location(x) specified more than once!");
1346 Location = &L;
1347 Default = L;
1348 return false;
1349 }
1350
1351 template <class T> void setValue(const T &V, bool initial = false) {
1352 check_location();
1353 *Location = V;
1354 if (initial)
1355 Default = V;
1356 }
1357
1358 DataType &getValue() {
1359 check_location();
1360 return *Location;
1361 }
1362 const DataType &getValue() const {
1363 check_location();
1364 return *Location;
1365 }
1366
1367 operator DataType() const { return this->getValue(); }
1368
1369 const OptionValue<DataType> &getDefault() const { return Default; }
1370};
1371
1372// Define how to hold a class type object, such as a string. Since we can
1373// inherit from a class, we do so. This makes us exactly compatible with the
1374// object in all cases that it is used.
1375//
1376template <class DataType>
1377class opt_storage<DataType, false, true> : public DataType {
1378public:
1379 OptionValue<DataType> Default;
1380
1381 template <class T> void setValue(const T &V, bool initial = false) {
1382 DataType::operator=(V);
1383 if (initial)
1384 Default = V;
1385 }
1386
1387 DataType &getValue() { return *this; }
1388 const DataType &getValue() const { return *this; }
1389
1390 const OptionValue<DataType> &getDefault() const { return Default; }
1391};
1392
1393// Define a partial specialization to handle things we cannot inherit from. In
1394// this case, we store an instance through containment, and overload operators
1395// to get at the value.
1396//
1397template <class DataType> class opt_storage<DataType, false, false> {
1398public:
1399 DataType Value;
1400 OptionValue<DataType> Default;
1401
1402 // Make sure we initialize the value with the default constructor for the
1403 // type.
1404 opt_storage() : Value(DataType()), Default() {}
1405
1406 template <class T> void setValue(const T &V, bool initial = false) {
1407 Value = V;
1408 if (initial)
1409 Default = V;
1410 }
1411 DataType &getValue() { return Value; }
1412 DataType getValue() const { return Value; }
1413
1414 const OptionValue<DataType> &getDefault() const { return Default; }
1415
1416 operator DataType() const { return getValue(); }
1417
1418 // If the datatype is a pointer, support -> on it.
1419 DataType operator->() const { return Value; }
1420};
1421
1422//===----------------------------------------------------------------------===//
1423// A scalar command line option.
1424//
1425template <class DataType, bool ExternalStorage = false,
1426 class ParserClass = parser<DataType>>
1427class opt
1428 : public Option,
1429 public opt_storage<DataType, ExternalStorage, std::is_class_v<DataType>> {
1430 ParserClass Parser;
1431
1432 bool handleOccurrence(unsigned pos, StringRef ArgName,
1433 StringRef Arg) override {
1434 typename ParserClass::parser_data_type Val =
1435 typename ParserClass::parser_data_type();
1436 if (Parser.parse(*this, ArgName, Arg, Val))
1437 return true; // Parse error!
1438 this->setValue(Val);
1439 this->setPosition(pos);
1440 Callback(Val);
1441 return false;
1442 }
1443
1444 enum ValueExpected getValueExpectedFlagDefault() const override {
1445 return Parser.getValueExpectedFlagDefault();
1446 }
1447
1448 void getExtraOptionNames(SmallVectorImpl<StringRef> &OptionNames) override {
1449 return Parser.getExtraOptionNames(OptionNames);
1450 }
1451
1452 // Forward printing stuff to the parser...
1453 size_t getOptionWidth() const override {
1454 return Parser.getOptionWidth(*this);
1455 }
1456
1457 void printOptionInfo(size_t GlobalWidth) const override {
1458 Parser.printOptionInfo(*this, GlobalWidth);
1459 }
1460
1461 void printOptionValue(size_t GlobalWidth, bool Force) const override {
1462 if (Force || !this->getDefault().compare(this->getValue())) {
1463 cl::printOptionDiff<ParserClass>(*this, Parser, this->getValue(),
1464 this->getDefault(), GlobalWidth);
1465 }
1466 }
1467
1468 void setDefault() override {
1469 if constexpr (std::is_assignable_v<DataType &, DataType>) {
1470 const OptionValue<DataType> &V = this->getDefault();
1471 if (V.hasValue())
1472 this->setValue(V.getValue());
1473 else
1474 this->setValue(DataType());
1475 }
1476 }
1477
1478 void done() {
1479 addArgument();
1480 Parser.initialize();
1481 }
1482
1483public:
1484 // Command line options should not be copyable
1485 opt(const opt &) = delete;
1486 opt &operator=(const opt &) = delete;
1487
1488 // setInitialValue - Used by the cl::init modifier...
1489 void setInitialValue(const DataType &V) { this->setValue(V, true); }
1490
1491 ParserClass &getParser() { return Parser; }
1492
1493 template <class T> DataType &operator=(const T &Val) {
1494 this->setValue(Val);
1495 Callback(Val);
1496 return this->getValue();
1497 }
1498
1499 template <class... Mods>
1500 explicit opt(const Mods &... Ms)
1501 : Option(llvm::cl::Optional, NotHidden), Parser(*this) {
1502 apply(this, Ms...);
1503 done();
1504 }
1505
1506 void setCallback(
1507 std::function<void(const typename ParserClass::parser_data_type &)> CB) {
1508 Callback = CB;
1509 }
1510
1511 std::function<void(const typename ParserClass::parser_data_type &)> Callback =
1512 [](const typename ParserClass::parser_data_type &) {};
1513};
1514
1515extern template class opt<unsigned>;
1516extern template class opt<int>;
1517extern template class opt<std::string>;
1518extern template class opt<char>;
1519extern template class opt<bool>;
1520
1521//===----------------------------------------------------------------------===//
1522// Default storage class definition: external storage. This implementation
1523// assumes the user will specify a variable to store the data into with the
1524// cl::location(x) modifier.
1525//
1526template <class DataType, class StorageClass> class list_storage {
1527 StorageClass *Location = nullptr; // Where to store the object...
1528 std::vector<OptionValue<DataType>> Default =
1529 std::vector<OptionValue<DataType>>();
1530 bool DefaultAssigned = false;
1531
1532public:
1533 list_storage() = default;
1534
1535 void clear() {}
1536
1537 bool setLocation(Option &O, StorageClass &L) {
1538 if (Location)
1539 return O.error(Message: "cl::location(x) specified more than once!");
1540 Location = &L;
1541 return false;
1542 }
1543
1544 template <class T> void addValue(const T &V, bool initial = false) {
1545 assert(Location != nullptr &&
1546 "cl::location(...) not specified for a command "
1547 "line option with external storage!");
1548 Location->push_back(V);
1549 if (initial)
1550 Default.push_back(V);
1551 }
1552
1553 const std::vector<OptionValue<DataType>> &getDefault() const {
1554 return Default;
1555 }
1556
1557 void assignDefault() { DefaultAssigned = true; }
1558 void overwriteDefault() { DefaultAssigned = false; }
1559 bool isDefaultAssigned() { return DefaultAssigned; }
1560};
1561
1562// Define how to hold a class type object, such as a string.
1563// Originally this code inherited from std::vector. In transitioning to a new
1564// API for command line options we should change this. The new implementation
1565// of this list_storage specialization implements the minimum subset of the
1566// std::vector API required for all the current clients.
1567//
1568// FIXME: Reduce this API to a more narrow subset of std::vector
1569//
1570template <class DataType> class list_storage<DataType, bool> {
1571 std::vector<DataType> Storage;
1572 std::vector<OptionValue<DataType>> Default;
1573 bool DefaultAssigned = false;
1574
1575public:
1576 using iterator = typename std::vector<DataType>::iterator;
1577
1578 iterator begin() { return Storage.begin(); }
1579 iterator end() { return Storage.end(); }
1580
1581 using const_iterator = typename std::vector<DataType>::const_iterator;
1582
1583 const_iterator begin() const { return Storage.begin(); }
1584 const_iterator end() const { return Storage.end(); }
1585
1586 using size_type = typename std::vector<DataType>::size_type;
1587
1588 size_type size() const { return Storage.size(); }
1589
1590 bool empty() const { return Storage.empty(); }
1591
1592 void push_back(const DataType &value) { Storage.push_back(value); }
1593 void push_back(DataType &&value) { Storage.push_back(value); }
1594
1595 using reference = typename std::vector<DataType>::reference;
1596 using const_reference = typename std::vector<DataType>::const_reference;
1597
1598 reference operator[](size_type pos) { return Storage[pos]; }
1599 const_reference operator[](size_type pos) const { return Storage[pos]; }
1600
1601 void clear() {
1602 Storage.clear();
1603 }
1604
1605 iterator erase(const_iterator pos) { return Storage.erase(pos); }
1606 iterator erase(const_iterator first, const_iterator last) {
1607 return Storage.erase(first, last);
1608 }
1609
1610 iterator erase(iterator pos) { return Storage.erase(pos); }
1611 iterator erase(iterator first, iterator last) {
1612 return Storage.erase(first, last);
1613 }
1614
1615 iterator insert(const_iterator pos, const DataType &value) {
1616 return Storage.insert(pos, value);
1617 }
1618 iterator insert(const_iterator pos, DataType &&value) {
1619 return Storage.insert(pos, value);
1620 }
1621
1622 iterator insert(iterator pos, const DataType &value) {
1623 return Storage.insert(pos, value);
1624 }
1625 iterator insert(iterator pos, DataType &&value) {
1626 return Storage.insert(pos, value);
1627 }
1628
1629 reference front() { return Storage.front(); }
1630 const_reference front() const { return Storage.front(); }
1631
1632 operator std::vector<DataType> &() { return Storage; }
1633 operator ArrayRef<DataType>() const { return Storage; }
1634 std::vector<DataType> *operator&() { return &Storage; }
1635 const std::vector<DataType> *operator&() const { return &Storage; }
1636
1637 template <class T> void addValue(const T &V, bool initial = false) {
1638 Storage.push_back(V);
1639 if (initial)
1640 Default.push_back(OptionValue<DataType>(V));
1641 }
1642
1643 const std::vector<OptionValue<DataType>> &getDefault() const {
1644 return Default;
1645 }
1646
1647 void assignDefault() { DefaultAssigned = true; }
1648 void overwriteDefault() { DefaultAssigned = false; }
1649 bool isDefaultAssigned() { return DefaultAssigned; }
1650};
1651
1652//===----------------------------------------------------------------------===//
1653// A list of command line options.
1654//
1655template <class DataType, class StorageClass = bool,
1656 class ParserClass = parser<DataType>>
1657class list : public Option, public list_storage<DataType, StorageClass> {
1658 std::vector<unsigned> Positions;
1659 ParserClass Parser;
1660
1661 enum ValueExpected getValueExpectedFlagDefault() const override {
1662 return Parser.getValueExpectedFlagDefault();
1663 }
1664
1665 void getExtraOptionNames(SmallVectorImpl<StringRef> &OptionNames) override {
1666 return Parser.getExtraOptionNames(OptionNames);
1667 }
1668
1669 bool handleOccurrence(unsigned pos, StringRef ArgName,
1670 StringRef Arg) override {
1671 typename ParserClass::parser_data_type Val =
1672 typename ParserClass::parser_data_type();
1673 if (list_storage<DataType, StorageClass>::isDefaultAssigned()) {
1674 clear();
1675 list_storage<DataType, StorageClass>::overwriteDefault();
1676 }
1677 if (Parser.parse(*this, ArgName, Arg, Val))
1678 return true; // Parse Error!
1679 list_storage<DataType, StorageClass>::addValue(Val);
1680 setPosition(pos);
1681 Positions.push_back(x: pos);
1682 Callback(Val);
1683 return false;
1684 }
1685
1686 // Forward printing stuff to the parser...
1687 size_t getOptionWidth() const override {
1688 return Parser.getOptionWidth(*this);
1689 }
1690
1691 void printOptionInfo(size_t GlobalWidth) const override {
1692 Parser.printOptionInfo(*this, GlobalWidth);
1693 }
1694
1695 // Unimplemented: list options don't currently store their default value.
1696 void printOptionValue(size_t /*GlobalWidth*/, bool /*Force*/) const override {
1697 }
1698
1699 void setDefault() override {
1700 Positions.clear();
1701 list_storage<DataType, StorageClass>::clear();
1702 for (auto &Val : list_storage<DataType, StorageClass>::getDefault())
1703 list_storage<DataType, StorageClass>::addValue(Val.getValue());
1704 }
1705
1706 void done() {
1707 addArgument();
1708 Parser.initialize();
1709 }
1710
1711public:
1712 // Command line options should not be copyable
1713 list(const list &) = delete;
1714 list &operator=(const list &) = delete;
1715
1716 ParserClass &getParser() { return Parser; }
1717
1718 unsigned getPosition(unsigned optnum) const {
1719 assert(optnum < this->size() && "Invalid option index");
1720 return Positions[optnum];
1721 }
1722
1723 void clear() {
1724 Positions.clear();
1725 list_storage<DataType, StorageClass>::clear();
1726 }
1727
1728 // setInitialValues - Used by the cl::list_init modifier...
1729 void setInitialValues(ArrayRef<DataType> Vs) {
1730 assert(!(list_storage<DataType, StorageClass>::isDefaultAssigned()) &&
1731 "Cannot have two default values");
1732 list_storage<DataType, StorageClass>::assignDefault();
1733 for (auto &Val : Vs)
1734 list_storage<DataType, StorageClass>::addValue(Val, true);
1735 }
1736
1737 void setNumAdditionalVals(unsigned n) { Option::setNumAdditionalVals(n); }
1738
1739 template <class... Mods>
1740 explicit list(const Mods &... Ms)
1741 : Option(ZeroOrMore, NotHidden), Parser(*this) {
1742 apply(this, Ms...);
1743 done();
1744 }
1745
1746 void setCallback(
1747 std::function<void(const typename ParserClass::parser_data_type &)> CB) {
1748 Callback = CB;
1749 }
1750
1751 std::function<void(const typename ParserClass::parser_data_type &)> Callback =
1752 [](const typename ParserClass::parser_data_type &) {};
1753};
1754
1755// Modifier to set the number of additional values.
1756struct multi_val {
1757 unsigned AdditionalVals;
1758 explicit multi_val(unsigned N) : AdditionalVals(N) {}
1759
1760 template <typename D, typename S, typename P>
1761 void apply(list<D, S, P> &L) const {
1762 L.setNumAdditionalVals(AdditionalVals);
1763 }
1764};
1765
1766//===----------------------------------------------------------------------===//
1767// Default storage class definition: external storage. This implementation
1768// assumes the user will specify a variable to store the data into with the
1769// cl::location(x) modifier.
1770//
1771template <class DataType, class StorageClass> class bits_storage {
1772 unsigned *Location = nullptr; // Where to store the bits...
1773
1774 template <class T> static unsigned Bit(const T &V) {
1775 unsigned BitPos = static_cast<unsigned>(V);
1776 assert(BitPos < sizeof(unsigned) * CHAR_BIT &&
1777 "enum exceeds width of bit vector!");
1778 return 1 << BitPos;
1779 }
1780
1781public:
1782 bits_storage() = default;
1783
1784 bool setLocation(Option &O, unsigned &L) {
1785 if (Location)
1786 return O.error(Message: "cl::location(x) specified more than once!");
1787 Location = &L;
1788 return false;
1789 }
1790
1791 template <class T> void addValue(const T &V) {
1792 assert(Location != nullptr &&
1793 "cl::location(...) not specified for a command "
1794 "line option with external storage!");
1795 *Location |= Bit(V);
1796 }
1797
1798 unsigned getBits() { return *Location; }
1799
1800 void clear() {
1801 if (Location)
1802 *Location = 0;
1803 }
1804
1805 template <class T> bool isSet(const T &V) {
1806 return (*Location & Bit(V)) != 0;
1807 }
1808};
1809
1810// Define how to hold bits. Since we can inherit from a class, we do so.
1811// This makes us exactly compatible with the bits in all cases that it is used.
1812//
1813template <class DataType> class bits_storage<DataType, bool> {
1814 unsigned Bits{0}; // Where to store the bits...
1815
1816 template <class T> static unsigned Bit(const T &V) {
1817 unsigned BitPos = static_cast<unsigned>(V);
1818 assert(BitPos < sizeof(unsigned) * CHAR_BIT &&
1819 "enum exceeds width of bit vector!");
1820 return 1 << BitPos;
1821 }
1822
1823public:
1824 template <class T> void addValue(const T &V) { Bits |= Bit(V); }
1825
1826 unsigned getBits() { return Bits; }
1827
1828 void clear() { Bits = 0; }
1829
1830 template <class T> bool isSet(const T &V) { return (Bits & Bit(V)) != 0; }
1831};
1832
1833//===----------------------------------------------------------------------===//
1834// A bit vector of command options.
1835//
1836template <class DataType, class Storage = bool,
1837 class ParserClass = parser<DataType>>
1838class bits : public Option, public bits_storage<DataType, Storage> {
1839 std::vector<unsigned> Positions;
1840 ParserClass Parser;
1841
1842 enum ValueExpected getValueExpectedFlagDefault() const override {
1843 return Parser.getValueExpectedFlagDefault();
1844 }
1845
1846 void getExtraOptionNames(SmallVectorImpl<StringRef> &OptionNames) override {
1847 return Parser.getExtraOptionNames(OptionNames);
1848 }
1849
1850 bool handleOccurrence(unsigned pos, StringRef ArgName,
1851 StringRef Arg) override {
1852 typename ParserClass::parser_data_type Val =
1853 typename ParserClass::parser_data_type();
1854 if (Parser.parse(*this, ArgName, Arg, Val))
1855 return true; // Parse Error!
1856 this->addValue(Val);
1857 setPosition(pos);
1858 Positions.push_back(x: pos);
1859 Callback(Val);
1860 return false;
1861 }
1862
1863 // Forward printing stuff to the parser...
1864 size_t getOptionWidth() const override {
1865 return Parser.getOptionWidth(*this);
1866 }
1867
1868 void printOptionInfo(size_t GlobalWidth) const override {
1869 Parser.printOptionInfo(*this, GlobalWidth);
1870 }
1871
1872 // Unimplemented: bits options don't currently store their default values.
1873 void printOptionValue(size_t /*GlobalWidth*/, bool /*Force*/) const override {
1874 }
1875
1876 void setDefault() override { bits_storage<DataType, Storage>::clear(); }
1877
1878 void done() {
1879 addArgument();
1880 Parser.initialize();
1881 }
1882
1883public:
1884 // Command line options should not be copyable
1885 bits(const bits &) = delete;
1886 bits &operator=(const bits &) = delete;
1887
1888 ParserClass &getParser() { return Parser; }
1889
1890 unsigned getPosition(unsigned optnum) const {
1891 assert(optnum < this->size() && "Invalid option index");
1892 return Positions[optnum];
1893 }
1894
1895 template <class... Mods>
1896 explicit bits(const Mods &... Ms)
1897 : Option(ZeroOrMore, NotHidden), Parser(*this) {
1898 apply(this, Ms...);
1899 done();
1900 }
1901
1902 void setCallback(
1903 std::function<void(const typename ParserClass::parser_data_type &)> CB) {
1904 Callback = CB;
1905 }
1906
1907 std::function<void(const typename ParserClass::parser_data_type &)> Callback =
1908 [](const typename ParserClass::parser_data_type &) {};
1909};
1910
1911//===----------------------------------------------------------------------===//
1912// Aliased command line option (alias this name to a preexisting name)
1913//
1914
1915class LLVM_ABI alias : public Option {
1916 Option *AliasFor;
1917
1918 bool handleOccurrence(unsigned pos, StringRef /*ArgName*/,
1919 StringRef Arg) override {
1920 return AliasFor->handleOccurrence(pos, ArgName: AliasFor->ArgStr, Arg);
1921 }
1922
1923 bool addOccurrence(unsigned pos, StringRef /*ArgName*/, StringRef Value,
1924 bool MultiArg = false) override {
1925 return AliasFor->addOccurrence(pos, ArgName: AliasFor->ArgStr, Value, MultiArg);
1926 }
1927
1928 // Handle printing stuff...
1929 size_t getOptionWidth() const override;
1930 void printOptionInfo(size_t GlobalWidth) const override;
1931
1932 // Aliases do not need to print their values.
1933 void printOptionValue(size_t /*GlobalWidth*/, bool /*Force*/) const override {
1934 }
1935
1936 void setDefault() override { AliasFor->setDefault(); }
1937
1938 ValueExpected getValueExpectedFlagDefault() const override {
1939 return AliasFor->getValueExpectedFlag();
1940 }
1941
1942 void done() {
1943 if (!hasArgStr())
1944 error(Message: "cl::alias must have argument name specified!");
1945 if (!AliasFor)
1946 error(Message: "cl::alias must have an cl::aliasopt(option) specified!");
1947 if (!Subs.empty())
1948 error(Message: "cl::alias must not have cl::sub(), aliased option's cl::sub() will be used!");
1949 Subs = AliasFor->Subs;
1950 Categories = AliasFor->Categories;
1951 addArgument();
1952 }
1953
1954public:
1955 // Command line options should not be copyable
1956 alias(const alias &) = delete;
1957 alias &operator=(const alias &) = delete;
1958
1959 void setAliasFor(Option &O) {
1960 if (AliasFor)
1961 error(Message: "cl::alias must only have one cl::aliasopt(...) specified!");
1962 AliasFor = &O;
1963 }
1964
1965 template <class... Mods>
1966 explicit alias(const Mods &... Ms)
1967 : Option(Optional, Hidden), AliasFor(nullptr) {
1968 apply(this, Ms...);
1969 done();
1970 }
1971};
1972
1973// Modifier to set the option an alias aliases.
1974struct aliasopt {
1975 Option &Opt;
1976
1977 explicit aliasopt(Option &O) : Opt(O) {}
1978
1979 void apply(alias &A) const { A.setAliasFor(Opt); }
1980};
1981
1982// Provide additional help at the end of the normal help output. All occurrences
1983// of cl::extrahelp will be accumulated and printed to stderr at the end of the
1984// regular help, just before exit is called.
1985struct extrahelp {
1986 StringRef morehelp;
1987
1988 LLVM_ABI explicit extrahelp(StringRef help);
1989};
1990
1991LLVM_ABI void PrintVersionMessage();
1992
1993/// This function just prints the help message, exactly the same way as if the
1994/// -help or -help-hidden option had been given on the command line.
1995///
1996/// \param Hidden if true will print hidden options
1997/// \param Categorized if true print options in categories
1998LLVM_ABI void PrintHelpMessage(bool Hidden = false, bool Categorized = false);
1999
2000/// An array of optional enabled settings in the LLVM build configuration,
2001/// which may be of interest to compiler developers. For example, includes
2002/// "+assertions" if assertions are enabled. Used by printBuildConfig.
2003LLVM_ABI ArrayRef<StringRef> getCompilerBuildConfig();
2004
2005/// Prints the compiler build configuration.
2006/// Designed for compiler developers, not compiler end-users.
2007/// Intended to be used in --version output when enabled.
2008LLVM_ABI void printBuildConfig(raw_ostream &OS);
2009
2010//===----------------------------------------------------------------------===//
2011// Public interface for accessing registered options.
2012//
2013
2014/// Use this to get a StringMap to all registered named options
2015/// (e.g. -help).
2016///
2017/// \return A reference to the StringMap used by the cl APIs to parse options.
2018///
2019/// Access to unnamed arguments (i.e. positional) are not provided because
2020/// it is expected that the client already has access to these.
2021///
2022/// Typical usage:
2023/// \code
2024/// main(int argc,char* argv[]) {
2025/// StringMap<llvm::cl::Option*> &opts = llvm::cl::getRegisteredOptions();
2026/// assert(opts.count("help") == 1)
2027/// opts["help"]->setDescription("Show alphabetical help information")
2028/// // More code
2029/// llvm::cl::ParseCommandLineOptions(argc,argv);
2030/// //More code
2031/// }
2032/// \endcode
2033///
2034/// This interface is useful for modifying options in libraries that are out of
2035/// the control of the client. The options should be modified before calling
2036/// llvm::cl::ParseCommandLineOptions().
2037///
2038/// Hopefully this API can be deprecated soon. Any situation where options need
2039/// to be modified by tools or libraries should be handled by sane APIs rather
2040/// than just handing around a global list.
2041LLVM_ABI StringMap<Option *> &
2042getRegisteredOptions(SubCommand &Sub = SubCommand::getTopLevel());
2043
2044/// Use this to get all registered SubCommands from the provided parser.
2045///
2046/// \return A range of all SubCommand pointers registered with the parser.
2047///
2048/// Typical usage:
2049/// \code
2050/// main(int argc, char* argv[]) {
2051/// llvm::cl::ParseCommandLineOptions(argc, argv);
2052/// for (auto* S : llvm::cl::getRegisteredSubcommands()) {
2053/// if (*S) {
2054/// std::cout << "Executing subcommand: " << S->getName() << std::endl;
2055/// // Execute some function based on the name...
2056/// }
2057/// }
2058/// }
2059/// \endcode
2060///
2061/// This interface is useful for defining subcommands in libraries and
2062/// the dispatch from a single point (like in the main function).
2063LLVM_ABI iterator_range<typename SmallPtrSet<SubCommand *, 4>::iterator>
2064getRegisteredSubcommands();
2065
2066//===----------------------------------------------------------------------===//
2067// Standalone command line processing utilities.
2068//
2069
2070/// Tokenizes a command line that can contain escapes and quotes.
2071//
2072/// The quoting rules match those used by GCC and other tools that use
2073/// libiberty's buildargv() or expandargv() utilities, and do not match bash.
2074/// They differ from buildargv() on treatment of backslashes that do not escape
2075/// a special character to make it possible to accept most Windows file paths.
2076///
2077/// \param [in] Source The string to be split on whitespace with quotes.
2078/// \param [in] Saver Delegates back to the caller for saving parsed strings.
2079/// \param [in] MarkEOLs true if tokenizing a response file and you want end of
2080/// lines and end of the response file to be marked with a nullptr string.
2081/// \param [out] NewArgv All parsed strings are appended to NewArgv.
2082LLVM_ABI void TokenizeGNUCommandLine(StringRef Source, StringSaver &Saver,
2083 SmallVectorImpl<const char *> &NewArgv,
2084 bool MarkEOLs = false);
2085
2086/// Tokenizes a string of Windows command line arguments, which may contain
2087/// quotes and escaped quotes.
2088///
2089/// See MSDN docs for CommandLineToArgvW for information on the quoting rules.
2090/// http://msdn.microsoft.com/en-us/library/windows/desktop/17w5ykft(v=vs.85).aspx
2091///
2092/// For handling a full Windows command line including the executable name at
2093/// the start, see TokenizeWindowsCommandLineFull below.
2094///
2095/// \param [in] Source The string to be split on whitespace with quotes.
2096/// \param [in] Saver Delegates back to the caller for saving parsed strings.
2097/// \param [in] MarkEOLs true if tokenizing a response file and you want end of
2098/// lines and end of the response file to be marked with a nullptr string.
2099/// \param [out] NewArgv All parsed strings are appended to NewArgv.
2100LLVM_ABI void TokenizeWindowsCommandLine(StringRef Source, StringSaver &Saver,
2101 SmallVectorImpl<const char *> &NewArgv,
2102 bool MarkEOLs = false);
2103
2104/// Tokenizes a Windows command line while attempting to avoid copies. If no
2105/// quoting or escaping was used, this produces substrings of the original
2106/// string. If a token requires unquoting, it will be allocated with the
2107/// StringSaver.
2108LLVM_ABI void
2109TokenizeWindowsCommandLineNoCopy(StringRef Source, StringSaver &Saver,
2110 SmallVectorImpl<StringRef> &NewArgv);
2111
2112/// Tokenizes a Windows full command line, including command name at the start.
2113///
2114/// This uses the same syntax rules as TokenizeWindowsCommandLine for all but
2115/// the first token. But the first token is expected to be parsed as the
2116/// executable file name in the way CreateProcess would do it, rather than the
2117/// way the C library startup code would do it: CreateProcess does not consider
2118/// that \ is ever an escape character (because " is not a valid filename char,
2119/// hence there's never a need to escape it to be used literally).
2120///
2121/// Parameters are the same as for TokenizeWindowsCommandLine. In particular,
2122/// if you set MarkEOLs = true, then the first word of every line will be
2123/// parsed using the special rules for command names, making this function
2124/// suitable for parsing a file full of commands to execute.
2125LLVM_ABI void
2126TokenizeWindowsCommandLineFull(StringRef Source, StringSaver &Saver,
2127 SmallVectorImpl<const char *> &NewArgv,
2128 bool MarkEOLs = false);
2129
2130/// String tokenization function type. Should be compatible with either
2131/// Windows or Unix command line tokenizers.
2132using TokenizerCallback = void (*)(StringRef Source, StringSaver &Saver,
2133 SmallVectorImpl<const char *> &NewArgv,
2134 bool MarkEOLs);
2135
2136/// Tokenizes content of configuration file.
2137///
2138/// \param [in] Source The string representing content of config file.
2139/// \param [in] Saver Delegates back to the caller for saving parsed strings.
2140/// \param [out] NewArgv All parsed strings are appended to NewArgv.
2141/// \param [in] MarkEOLs Added for compatibility with TokenizerCallback.
2142///
2143/// It works like TokenizeGNUCommandLine with ability to skip comment lines.
2144///
2145LLVM_ABI void tokenizeConfigFile(StringRef Source, StringSaver &Saver,
2146 SmallVectorImpl<const char *> &NewArgv,
2147 bool MarkEOLs = false);
2148
2149/// Contains options that control response file expansion.
2150class ExpansionContext {
2151 /// Provides persistent storage for parsed strings.
2152 StringSaver Saver;
2153
2154 /// Tokenization strategy. Typically Unix or Windows.
2155 TokenizerCallback Tokenizer;
2156
2157 /// File system used for all file access when running the expansion.
2158 vfs::FileSystem *FS;
2159
2160 /// Path used to resolve relative rsp files. If empty, the file system
2161 /// current directory is used instead.
2162 StringRef CurrentDir;
2163
2164 /// Directories used for search of config files.
2165 ArrayRef<StringRef> SearchDirs;
2166
2167 /// True if names of nested response files must be resolved relative to
2168 /// including file.
2169 bool RelativeNames = false;
2170
2171 /// If true, mark end of lines and the end of the response file with nullptrs
2172 /// in the Argv vector.
2173 bool MarkEOLs = false;
2174
2175 /// If true, body of config file is expanded.
2176 bool InConfigFile = false;
2177
2178 llvm::Error expandResponseFile(StringRef FName,
2179 SmallVectorImpl<const char *> &NewArgv);
2180
2181public:
2182 LLVM_ABI ExpansionContext(BumpPtrAllocator &A, TokenizerCallback T);
2183
2184 ExpansionContext &setMarkEOLs(bool X) {
2185 MarkEOLs = X;
2186 return *this;
2187 }
2188
2189 ExpansionContext &setRelativeNames(bool X) {
2190 RelativeNames = X;
2191 return *this;
2192 }
2193
2194 ExpansionContext &setCurrentDir(StringRef X) {
2195 CurrentDir = X;
2196 return *this;
2197 }
2198
2199 ExpansionContext &setSearchDirs(ArrayRef<StringRef> X) {
2200 SearchDirs = X;
2201 return *this;
2202 }
2203
2204 ExpansionContext &setVFS(vfs::FileSystem *X) {
2205 FS = X;
2206 return *this;
2207 }
2208
2209 /// Looks for the specified configuration file.
2210 ///
2211 /// \param[in] FileName Name of the file to search for.
2212 /// \param[out] FilePath File absolute path, if it was found.
2213 /// \return True if file was found.
2214 ///
2215 /// If the specified file name contains a directory separator, it is searched
2216 /// for by its absolute path. Otherwise looks for file sequentially in
2217 /// directories specified by SearchDirs field.
2218 LLVM_ABI bool findConfigFile(StringRef FileName,
2219 SmallVectorImpl<char> &FilePath);
2220
2221 /// Reads command line options from the given configuration file.
2222 ///
2223 /// \param [in] CfgFile Path to configuration file.
2224 /// \param [out] Argv Array to which the read options are added.
2225 /// \return true if the file was successfully read.
2226 ///
2227 /// It reads content of the specified file, tokenizes it and expands "@file"
2228 /// commands resolving file names in them relative to the directory where
2229 /// CfgFilename resides. It also expands "<CFGDIR>" to the base path of the
2230 /// current config file.
2231 LLVM_ABI Error readConfigFile(StringRef CfgFile,
2232 SmallVectorImpl<const char *> &Argv);
2233
2234 /// Expands constructs "@file" in the provided array of arguments recursively.
2235 LLVM_ABI Error expandResponseFiles(SmallVectorImpl<const char *> &Argv);
2236};
2237
2238/// A convenience helper which concatenates the options specified by the
2239/// environment variable EnvVar and command line options, then expands
2240/// response files recursively.
2241/// \return true if all @files were expanded successfully or there were none.
2242LLVM_ABI bool expandResponseFiles(int Argc, const char *const *Argv,
2243 const char *EnvVar,
2244 SmallVectorImpl<const char *> &NewArgv);
2245
2246/// A convenience helper which supports the typical use case of expansion
2247/// function call.
2248LLVM_ABI bool ExpandResponseFiles(StringSaver &Saver,
2249 TokenizerCallback Tokenizer,
2250 SmallVectorImpl<const char *> &Argv);
2251
2252/// A convenience helper which concatenates the options specified by the
2253/// environment variable EnvVar and command line options, then expands response
2254/// files recursively. The tokenizer is a predefined GNU or Windows one.
2255/// \return true if all @files were expanded successfully or there were none.
2256LLVM_ABI bool expandResponseFiles(int Argc, const char *const *Argv,
2257 const char *EnvVar, StringSaver &Saver,
2258 SmallVectorImpl<const char *> &NewArgv);
2259
2260/// Mark all options not part of this category as cl::ReallyHidden.
2261///
2262/// \param Category the category of options to keep displaying
2263///
2264/// Some tools (like clang-format) like to be able to hide all options that are
2265/// not specific to the tool. This function allows a tool to specify a single
2266/// option category to display in the -help output.
2267LLVM_ABI void HideUnrelatedOptions(cl::OptionCategory &Category,
2268 SubCommand &Sub = SubCommand::getTopLevel());
2269
2270/// Mark all options not part of the categories as cl::ReallyHidden.
2271///
2272/// \param Categories the categories of options to keep displaying.
2273///
2274/// Some tools (like clang-format) like to be able to hide all options that are
2275/// not specific to the tool. This function allows a tool to specify a single
2276/// option category to display in the -help output.
2277LLVM_ABI void
2278HideUnrelatedOptions(ArrayRef<const cl::OptionCategory *> Categories,
2279 SubCommand &Sub = SubCommand::getTopLevel());
2280
2281/// Reset all command line options to a state that looks as if they have
2282/// never appeared on the command line. This is useful for being able to parse
2283/// a command line multiple times (especially useful for writing tests).
2284LLVM_ABI void ResetAllOptionOccurrences();
2285
2286/// Reset the command line parser back to its initial state. This
2287/// removes
2288/// all options, categories, and subcommands and returns the parser to a state
2289/// where no options are supported.
2290LLVM_ABI void ResetCommandLineParser();
2291
2292/// Parses `Arg` into the option handler `Handler`.
2293LLVM_ABI bool ProvidePositionalOption(Option *Handler, StringRef Arg, int i);
2294
2295} // end namespace cl
2296
2297} // end namespace llvm
2298
2299#endif // LLVM_SUPPORT_COMMANDLINE_H
2300

source code of llvm/include/llvm/Support/CommandLine.h