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
18using namespace llvm;
19using namespace llvm::opt;
20
21Arg::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
25Arg::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
32Arg::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
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::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)
65LLVM_DUMP_METHOD void Arg::dump() const { print(dbgs()); }
66#endif
67
68std::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
87void 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
96void 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