mirror of
https://github.com/RPCS3/llvm-mirror.git
synced 2024-10-22 12:33:33 +02:00
fd9ead3b6f
string_ostream is a safe and efficient string builder that combines opaque stack storage with a built-in ostream interface. small_string_ostream<bytes> additionally permits an explicit stack storage size other than the default 128 bytes to be provided. Beyond that, storage is transferred to the heap. This convenient class can be used in most places an std::string+raw_string_ostream pair or SmallString<>+raw_svector_ostream pair would previously have been used, in order to guarantee consistent access without byte truncation. The patch also converts much of LLVM to use the new facility. These changes include several probable bug fixes for truncated output, a programming error that's no longer possible with the new interface. llvm-svn: 211749
121 lines
3.2 KiB
C++
121 lines
3.2 KiB
C++
//===--- Arg.cpp - Argument Implementations -------------------------------===//
|
|
//
|
|
// The LLVM Compiler Infrastructure
|
|
//
|
|
// This file is distributed under the University of Illinois Open Source
|
|
// License. See LICENSE.TXT for details.
|
|
//
|
|
//===----------------------------------------------------------------------===//
|
|
|
|
#include "llvm/Option/Arg.h"
|
|
#include "llvm/ADT/SmallString.h"
|
|
#include "llvm/ADT/Twine.h"
|
|
#include "llvm/Option/ArgList.h"
|
|
#include "llvm/Option/Option.h"
|
|
#include "llvm/Support/raw_ostream.h"
|
|
|
|
using namespace llvm;
|
|
using namespace llvm::opt;
|
|
|
|
Arg::Arg(const Option _Opt, StringRef S, unsigned _Index, const Arg *_BaseArg)
|
|
: Opt(_Opt), BaseArg(_BaseArg), Spelling(S), Index(_Index),
|
|
Claimed(false), OwnsValues(false) {
|
|
}
|
|
|
|
Arg::Arg(const Option _Opt, StringRef S, unsigned _Index,
|
|
const char *Value0, const Arg *_BaseArg)
|
|
: Opt(_Opt), BaseArg(_BaseArg), Spelling(S), Index(_Index),
|
|
Claimed(false), OwnsValues(false) {
|
|
Values.push_back(Value0);
|
|
}
|
|
|
|
Arg::Arg(const Option _Opt, StringRef S, unsigned _Index,
|
|
const char *Value0, const char *Value1, const Arg *_BaseArg)
|
|
: Opt(_Opt), BaseArg(_BaseArg), Spelling(S), Index(_Index),
|
|
Claimed(false), OwnsValues(false) {
|
|
Values.push_back(Value0);
|
|
Values.push_back(Value1);
|
|
}
|
|
|
|
Arg::~Arg() {
|
|
if (OwnsValues) {
|
|
for (unsigned i = 0, e = Values.size(); i != e; ++i)
|
|
delete[] Values[i];
|
|
}
|
|
}
|
|
|
|
void Arg::dump() const {
|
|
llvm::errs() << "<";
|
|
|
|
llvm::errs() << " Opt:";
|
|
Opt.dump();
|
|
|
|
llvm::errs() << " Index:" << Index;
|
|
|
|
llvm::errs() << " Values: [";
|
|
for (unsigned i = 0, e = Values.size(); i != e; ++i) {
|
|
if (i) llvm::errs() << ", ";
|
|
llvm::errs() << "'" << Values[i] << "'";
|
|
}
|
|
|
|
llvm::errs() << "]>\n";
|
|
}
|
|
|
|
std::string Arg::getAsString(const ArgList &Args) const {
|
|
small_string_ostream<256> OS;
|
|
|
|
ArgStringList ASL;
|
|
render(Args, ASL);
|
|
for (ArgStringList::iterator
|
|
it = ASL.begin(), ie = ASL.end(); it != ie; ++it) {
|
|
if (it != ASL.begin())
|
|
OS << ' ';
|
|
OS << *it;
|
|
}
|
|
|
|
return OS.str();
|
|
}
|
|
|
|
void Arg::renderAsInput(const ArgList &Args, ArgStringList &Output) const {
|
|
if (!getOption().hasNoOptAsInput()) {
|
|
render(Args, Output);
|
|
return;
|
|
}
|
|
|
|
for (unsigned i = 0, e = getNumValues(); i != e; ++i)
|
|
Output.push_back(getValue(i));
|
|
}
|
|
|
|
void Arg::render(const ArgList &Args, ArgStringList &Output) const {
|
|
switch (getOption().getRenderStyle()) {
|
|
case Option::RenderValuesStyle:
|
|
for (unsigned i = 0, e = getNumValues(); i != e; ++i)
|
|
Output.push_back(getValue(i));
|
|
break;
|
|
|
|
case Option::RenderCommaJoinedStyle: {
|
|
small_string_ostream<256> OS;
|
|
OS << getSpelling();
|
|
for (unsigned i = 0, e = getNumValues(); i != e; ++i) {
|
|
if (i) OS << ',';
|
|
OS << getValue(i);
|
|
}
|
|
Output.push_back(Args.MakeArgString(OS.str()));
|
|
break;
|
|
}
|
|
|
|
case Option::RenderJoinedStyle:
|
|
Output.push_back(Args.GetOrMakeJoinedArgString(
|
|
getIndex(), getSpelling(), getValue(0)));
|
|
for (unsigned i = 1, e = getNumValues(); i != e; ++i)
|
|
Output.push_back(getValue(i));
|
|
break;
|
|
|
|
case Option::RenderSeparateStyle:
|
|
Output.push_back(Args.MakeArgString(getSpelling()));
|
|
for (unsigned i = 0, e = getNumValues(); i != e; ++i)
|
|
Output.push_back(getValue(i));
|
|
break;
|
|
}
|
|
}
|