1 //===--- Tool.cpp - The LLVM Compiler Driver --------------------*- C++ -*-===//
2 //
3 // The LLVM Compiler Infrastructure
4 //
5 // This file is distributed under the University of Illinois Open
6 // Source License. See LICENSE.TXT for details.
7 //
8 //===----------------------------------------------------------------------===//
9 //
10 // Tool base class - implementation details.
11 //
12 //===----------------------------------------------------------------------===//
13
14 #include "llvm/CompilerDriver/BuiltinOptions.h"
15 #include "llvm/CompilerDriver/Tool.h"
16
17 #include "llvm/ADT/StringExtras.h"
18 #include "llvm/Support/Path.h"
19
20 #include <algorithm>
21
22 using namespace llvm;
23 using namespace llvmc;
24
25 namespace {
MakeTempFile(const sys::Path & TempDir,const std::string & BaseName,const std::string & Suffix)26 sys::Path MakeTempFile(const sys::Path& TempDir, const std::string& BaseName,
27 const std::string& Suffix) {
28 sys::Path Out;
29
30 // Make sure we don't end up with path names like '/file.o' if the
31 // TempDir is empty.
32 if (TempDir.empty()) {
33 Out.set(BaseName);
34 }
35 else {
36 Out = TempDir;
37 Out.appendComponent(BaseName);
38 }
39 Out.appendSuffix(Suffix);
40 // NOTE: makeUnique always *creates* a unique temporary file,
41 // which is good, since there will be no races. However, some
42 // tools do not like it when the output file already exists, so
43 // they need to be placated with -f or something like that.
44 Out.makeUnique(true, NULL);
45 return Out;
46 }
47 }
48
OutFilename(const sys::Path & In,const sys::Path & TempDir,bool StopCompilation,const char * OutputSuffix) const49 sys::Path Tool::OutFilename(const sys::Path& In,
50 const sys::Path& TempDir,
51 bool StopCompilation,
52 const char* OutputSuffix) const {
53 sys::Path Out;
54
55 if (StopCompilation) {
56 if (!OutputFilename.empty()) {
57 Out.set(OutputFilename);
58 }
59 else if (IsJoin()) {
60 Out.set("a");
61 Out.appendSuffix(OutputSuffix);
62 }
63 else {
64 Out.set(sys::path::stem(In.str()));
65 Out.appendSuffix(OutputSuffix);
66 }
67 }
68 else {
69 if (IsJoin())
70 Out = MakeTempFile(TempDir, "tmp", OutputSuffix);
71 else
72 Out = MakeTempFile(TempDir, sys::path::stem(In.str()), OutputSuffix);
73 }
74 return Out;
75 }
76
77 namespace {
78 template <class A, class B>
CompareFirst(std::pair<A,B> p1,std::pair<A,B> p2)79 bool CompareFirst (std::pair<A,B> p1, std::pair<A,B> p2) {
80 return std::less<A>()(p1.first, p2.first);
81 }
82 }
83
SortArgs(ArgsVector & Args) const84 StrVector Tool::SortArgs(ArgsVector& Args) const {
85 StrVector Out;
86
87 // HACK: this won't be needed when we'll migrate away from CommandLine.
88 std::stable_sort(Args.begin(), Args.end(),
89 &CompareFirst<unsigned, std::string>);
90 for (ArgsVector::iterator B = Args.begin(), E = Args.end(); B != E; ++B) {
91 Out.push_back(B->second);
92 }
93
94 return Out;
95 }
96