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       IgnoredTargetSpecific(false), 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       IgnoredTargetSpecific(false), 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       IgnoredTargetSpecific(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::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 std::string(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 }
126