lib/Option/Arg.cpp

   1 //===- Arg.cpp - Argument Implementations ---------------------------------===//
   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 #include "llvm/ADT/SmallString.h"
  10 #include "llvm/Config/llvm-config.h"
  11 #include "llvm/Option/Arg.h"
  12 #include "llvm/Option/ArgList.h"
  13 #include "llvm/Option/Option.h"
  14 #include "llvm/Support/Compiler.h"
  15 #include "llvm/Support/Debug.h"
  16 #include "llvm/Support/raw_ostream.h"
  17
  18 using namespace llvm;
  19 using namespace llvm::opt;
  20
  21 Arg::Arg(const Option Opt, StringRef S, unsigned Index, const Arg *BaseArg)
  22     : Opt(Opt), BaseArg(BaseArg), Spelling(S), Index(Index), Claimed(false),
  23       OwnsValues(false) {}
  24
  25 Arg::Arg(const Option Opt, StringRef S, unsigned Index, const char *Value0,
  26          const Arg *BaseArg)
  27     : Opt(Opt), BaseArg(BaseArg), Spelling(S), Index(Index), Claimed(false),
  28       OwnsValues(false) {
  29   Values.push_back(Value0);
  30 }
  31
  32 Arg::Arg(const Option Opt, StringRef S, unsigned Index, const char *Value0,
  33          const char *Value1, const Arg *BaseArg)
  34     : Opt(Opt), BaseArg(BaseArg), Spelling(S), Index(Index), Claimed(false),
  35       OwnsValues(false) {
  36   Values.push_back(Value0);
  37   Values.push_back(Value1);
  38 }
  39
  40 Arg::~Arg() {
  41   if (OwnsValues) {
  42     for (unsigned i = 0, e = Values.size(); i != e; ++i)
  43       delete[] Values[i];
  44   }
  45 }
  46
  47 void Arg::print(raw_ostream& O) const {
  48   O << "<";
  49
  50   O << " Opt:";
  51   Opt.print(O);
  52
  53   O << " Index:" << Index;
  54
  55   O << " Values: [";
  56   for (unsigned i = 0, e = Values.size(); i != e; ++i) {
  57     if (i) O << ", ";
  58     O << "'" << Values[i] << "'";
  59   }
  60
  61   O << "]>\n";
  62 }
  63
  64 #if !defined(NDEBUG) || defined(LLVM_ENABLE_DUMP)
  65 LLVM_DUMP_METHOD void Arg::dump() const { print(dbgs()); }
  66 #endif
  67
  68 std::string Arg::getAsString(const ArgList &Args) const {
  69   if (Alias)
  70     return Alias->getAsString(Args);
  71
  72   SmallString<256> Res;
  73   raw_svector_ostream OS(Res);
  74
  75   ArgStringList ASL;
  76   render(Args, ASL);
  77   for (ArgStringList::iterator
  78          it = ASL.begin(), ie = ASL.end(); it != ie; ++it) {
  79     if (it != ASL.begin())
  80       OS << ' ';
  81     OS << *it;
  82   }
  83
  84   return OS.str();
  85 }
  86
  87 void Arg::renderAsInput(const ArgList &Args, ArgStringList &Output) const {
  88   if (!getOption().hasNoOptAsInput()) {
  89     render(Args, Output);
  90     return;
  91   }
  92
  93   Output.append(Values.begin(), Values.end());
  94 }
  95
  96 void Arg::render(const ArgList &Args, ArgStringList &Output) const {
  97   switch (getOption().getRenderStyle()) {
  98   case Option::RenderValuesStyle:
  99     Output.append(Values.begin(), Values.end());
 100     break;
 101
 102   case Option::RenderCommaJoinedStyle: {
 103     SmallString<256> Res;
 104     raw_svector_ostream OS(Res);
 105     OS << getSpelling();
 106     for (unsigned i = 0, e = getNumValues(); i != e; ++i) {
 107       if (i) OS << ',';
 108       OS << getValue(i);
 109     }
 110     Output.push_back(Args.MakeArgString(OS.str()));
 111     break;
 112   }
 113
 114  case Option::RenderJoinedStyle:
 115     Output.push_back(Args.GetOrMakeJoinedArgString(
 116                        getIndex(), getSpelling(), getValue(0)));
 117     Output.append(Values.begin() + 1, Values.end());
 118     break;
 119
 120   case Option::RenderSeparateStyle:
 121     Output.push_back(Args.MakeArgString(getSpelling()));
 122     Output.append(Values.begin(), Values.end());
 123     break;
 124   }
 125 }