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/Debug.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), Claimed(false),
  22       OwnsValues(false) {}
  23
  24 Arg::Arg(const Option Opt, StringRef S, unsigned Index, const char *Value0,
  25          const Arg *BaseArg)
  26     : Opt(Opt), BaseArg(BaseArg), Spelling(S), Index(Index), Claimed(false),
  27       OwnsValues(false) {
  28   Values.push_back(Value0);
  29 }
  30
  31 Arg::Arg(const Option Opt, StringRef S, unsigned Index, const char *Value0,
  32          const char *Value1, const Arg *BaseArg)
  33     : Opt(Opt), BaseArg(BaseArg), Spelling(S), Index(Index), Claimed(false),
  34       OwnsValues(false) {
  35   Values.push_back(Value0);
  36   Values.push_back(Value1);
  37 }
  38
  39 Arg::~Arg() {
  40   if (OwnsValues) {
  41     for (unsigned i = 0, e = Values.size(); i != e; ++i)
  42       delete[] Values[i];
  43   }
  44 }
  45
  46 void Arg::print(raw_ostream &OS) const {
  47   OS << "<";
  48
  49   OS << " Opt:";
  50   Opt.dump();
  51
  52   OS << " Index:" << Index;
  53
  54   OS << " Values: [";
  55   for (unsigned i = 0, e = Values.size(); i != e; ++i) {
  56     OS << "'" << Values[i] << "'";
  57     if (i != e - 1) llvm::errs() << ", ";
  58   }
  59   OS << "]>\n";
  60 }
  61
  62 void Arg::dump() const {
  63   print(llvm::dbgs());
  64 }
  65
  66 std::string Arg::getAsString(const ArgList &Args) const {
  67   SmallString<256> Res;
  68   llvm::raw_svector_ostream OS(Res);
  69
  70   ArgStringList ASL;
  71   render(Args, ASL);
  72   for (ArgStringList::iterator
  73          it = ASL.begin(), ie = ASL.end(); it != ie; ++it) {
  74     if (it != ASL.begin())
  75       OS << ' ';
  76     OS << *it;
  77   }
  78
  79   return OS.str();
  80 }
  81
  82 void Arg::renderAsInput(const ArgList &Args, ArgStringList &Output) const {
  83   if (!getOption().hasNoOptAsInput()) {
  84     render(Args, Output);
  85     return;
  86   }
  87
  88   Output.append(Values.begin(), Values.end());
  89 }
  90
  91 void Arg::render(const ArgList &Args, ArgStringList &Output) const {
  92   switch (getOption().getRenderStyle()) {
  93   case Option::RenderValuesStyle:
  94     Output.append(Values.begin(), Values.end());
  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     Output.append(Values.begin() + 1, Values.end());
 113     break;
 114
 115   case Option::RenderSeparateStyle:
 116     Output.push_back(Args.MakeArgString(getSpelling()));
 117     Output.append(Values.begin(), Values.end());
 118     break;
 119   }
 120 }