124 lines
		
	
	
		
			3.2 KiB
		
	
	
	
		
			C++
		
	
	
	
			
		
		
	
	
			124 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/ADT/SmallString.h"
 | 
						|
#include "llvm/Config/llvm-config.h"
 | 
						|
#include "llvm/Option/Arg.h"
 | 
						|
#include "llvm/Option/ArgList.h"
 | 
						|
#include "llvm/Option/Option.h"
 | 
						|
#include "llvm/Support/Compiler.h"
 | 
						|
#include "llvm/Support/Debug.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::print(raw_ostream& O) const {
 | 
						|
  O << "<";
 | 
						|
 | 
						|
  O << " Opt:";
 | 
						|
  Opt.print(O);
 | 
						|
 | 
						|
  O << " Index:" << Index;
 | 
						|
 | 
						|
  O << " Values: [";
 | 
						|
  for (unsigned i = 0, e = Values.size(); i != e; ++i) {
 | 
						|
    if (i) O << ", ";
 | 
						|
    O << "'" << Values[i] << "'";
 | 
						|
  }
 | 
						|
 | 
						|
  O << "]>\n";
 | 
						|
}
 | 
						|
 | 
						|
#if !defined(NDEBUG) || defined(LLVM_ENABLE_DUMP)
 | 
						|
LLVM_DUMP_METHOD void Arg::dump() const { print(dbgs()); }
 | 
						|
#endif
 | 
						|
 | 
						|
std::string Arg::getAsString(const ArgList &Args) const {
 | 
						|
  SmallString<256> Res;
 | 
						|
  raw_svector_ostream OS(Res);
 | 
						|
 | 
						|
  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;
 | 
						|
  }
 | 
						|
 | 
						|
  Output.append(Values.begin(), Values.end());
 | 
						|
}
 | 
						|
 | 
						|
void Arg::render(const ArgList &Args, ArgStringList &Output) const {
 | 
						|
  switch (getOption().getRenderStyle()) {
 | 
						|
  case Option::RenderValuesStyle:
 | 
						|
    Output.append(Values.begin(), Values.end());
 | 
						|
    break;
 | 
						|
 | 
						|
  case Option::RenderCommaJoinedStyle: {
 | 
						|
    SmallString<256> Res;
 | 
						|
    raw_svector_ostream OS(Res);
 | 
						|
    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)));
 | 
						|
    Output.append(Values.begin() + 1, Values.end());
 | 
						|
    break;
 | 
						|
 | 
						|
  case Option::RenderSeparateStyle:
 | 
						|
    Output.push_back(Args.MakeArgString(getSpelling()));
 | 
						|
    Output.append(Values.begin(), Values.end());
 | 
						|
    break;
 | 
						|
  }
 | 
						|
}
 |