lib/Option/Arg.cpp

   1 //===--- Arg.cpp - Argument Implementations -------------------------------===//
   2 //
   3 //                     The LLVM Compiler Infrastructure
   4 //
   5 // This file is distributed under the University of Illinois Open Source
   6 // License. See LICENSE.TXT for details.
   7 //
   8 //===----------------------------------------------------------------------===//
   9
  10 #include "llvm/Option/Arg.h"
  11 #include "llvm/ADT/SmallString.h"
  12 #include "llvm/ADT/Twine.h"
  13 #include "llvm/Option/ArgList.h"
  14 #include "llvm/Option/Option.h"
  15 #include "llvm/Support/raw_ostream.h"
  16
  17 using namespace llvm;
  18 using namespace llvm::opt;
  19
  20 Arg::Arg(const Option _Opt, StringRef S, unsigned _Index, const Arg *_BaseArg)
  21   : Opt(_Opt), BaseArg(_BaseArg), Spelling(S), Index(_Index),
  22     Claimed(false), OwnsValues(false) {
  23 }
  24
  25 Arg::Arg(const Option _Opt, StringRef S, unsigned _Index,
  26          const char *Value0, const Arg *_BaseArg)
  27   : Opt(_Opt), BaseArg(_BaseArg), Spelling(S), Index(_Index),
  28     Claimed(false), OwnsValues(false) {
  29   Values.push_back(Value0);
  30 }
  31
  32 Arg::Arg(const Option _Opt, StringRef S, unsigned _Index,
  33          const char *Value0, const char *Value1, const Arg *_BaseArg)
  34   : Opt(_Opt), BaseArg(_BaseArg), Spelling(S), Index(_Index),
  35     Claimed(false), 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::dump() const {
  48   llvm::errs() << "<";
  49
  50   llvm::errs() << " Opt:";
  51   Opt.dump();
  52
  53   llvm::errs() << " Index:" << Index;
  54
  55   llvm::errs() << " Values: [";
  56   for (unsigned i = 0, e = Values.size(); i != e; ++i) {
  57     if (i) llvm::errs() << ", ";
  58     llvm::errs() << "'" << Values[i] << "'";
  59   }
  60
  61   llvm::errs() << "]>\n";
  62 }
  63
  64 std::string Arg::getAsString(const ArgList &Args) const {
  65   SmallString<256> Res;
  66   llvm::raw_svector_ostream OS(Res);
  67
  68   ArgStringList ASL;
  69   render(Args, ASL);
  70   for (ArgStringList::iterator
  71          it = ASL.begin(), ie = ASL.end(); it != ie; ++it) {
  72     if (it != ASL.begin())
  73       OS << ' ';
  74     OS << *it;
  75   }
  76
  77   return OS.str();
  78 }
  79
  80 void Arg::renderAsInput(const ArgList &Args, ArgStringList &Output) const {
  81   if (!getOption().hasNoOptAsInput()) {
  82     render(Args, Output);
  83     return;
  84   }
  85
  86   for (unsigned i = 0, e = getNumValues(); i != e; ++i)
  87     Output.push_back(getValue(i));
  88 }
  89
  90 void Arg::render(const ArgList &Args, ArgStringList &Output) const {
  91   switch (getOption().getRenderStyle()) {
  92   case Option::RenderValuesStyle:
  93     for (unsigned i = 0, e = getNumValues(); i != e; ++i)
  94       Output.push_back(getValue(i));
  95     break;
  96
  97   case Option::RenderCommaJoinedStyle: {
  98     SmallString<256> Res;
  99     llvm::raw_svector_ostream OS(Res);
 100     OS << getSpelling();
 101     for (unsigned i = 0, e = getNumValues(); i != e; ++i) {
 102       if (i) OS << ',';
 103       OS << getValue(i);
 104     }
 105     Output.push_back(Args.MakeArgString(OS.str()));
 106     break;
 107   }
 108
 109  case Option::RenderJoinedStyle:
 110     Output.push_back(Args.GetOrMakeJoinedArgString(
 111                        getIndex(), getSpelling(), getValue(0)));
 112     for (unsigned i = 1, e = getNumValues(); i != e; ++i)
 113       Output.push_back(getValue(i));
 114     break;
 115
 116   case Option::RenderSeparateStyle:
 117     Output.push_back(Args.MakeArgString(getSpelling()));
 118     for (unsigned i = 0, e = getNumValues(); i != e; ++i)
 119       Output.push_back(getValue(i));
 120     break;
 121   }
 122 }