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
17using namespace llvm;
18using namespace llvm::opt;
19
20Arg::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
25Arg::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
32Arg::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
40Arg::~Arg() {
41  if (OwnsValues) {
42    for (unsigned i = 0, e = Values.size(); i != e; ++i)
43      delete[] Values[i];
44  }
45}
46
47void 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
64std::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
80void 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
90void 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}
123