Mikhail Glushenkov | f188178 | 2009-03-02 09:01:14 +0000 | [diff] [blame] | 1 | //===--- Main.cpp - The LLVM Compiler Driver -------------------*- C++ -*-===// |
Anton Korobeynikov | ac67b7e | 2008-03-23 08:57:20 +0000 | [diff] [blame] | 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 | // This tool provides a single point of access to the LLVM |
| 11 | // compilation tools. It has many options. To discover the options |
| 12 | // supported please refer to the tools' manual page or run the tool |
| 13 | // with the --help option. |
| 14 | // |
| 15 | //===----------------------------------------------------------------------===// |
| 16 | |
Mikhail Glushenkov | 4a1a77c | 2008-09-22 20:50:40 +0000 | [diff] [blame] | 17 | #include "llvm/CompilerDriver/CompilationGraph.h" |
Mikhail Glushenkov | f188178 | 2009-03-02 09:01:14 +0000 | [diff] [blame] | 18 | #include "llvm/CompilerDriver/Error.h" |
Mikhail Glushenkov | 4a1a77c | 2008-09-22 20:50:40 +0000 | [diff] [blame] | 19 | #include "llvm/CompilerDriver/Plugin.h" |
Anton Korobeynikov | ac67b7e | 2008-03-23 08:57:20 +0000 | [diff] [blame] | 20 | |
| 21 | #include "llvm/System/Path.h" |
| 22 | #include "llvm/Support/CommandLine.h" |
Mikhail Glushenkov | c82ce4a | 2008-09-22 20:49:34 +0000 | [diff] [blame] | 23 | #include "llvm/Support/PluginLoader.h" |
Anton Korobeynikov | ac67b7e | 2008-03-23 08:57:20 +0000 | [diff] [blame] | 24 | |
| 25 | #include <iostream> |
| 26 | #include <stdexcept> |
| 27 | #include <string> |
| 28 | |
Mikhail Glushenkov | b90cd83 | 2008-05-06 16:34:12 +0000 | [diff] [blame] | 29 | namespace cl = llvm::cl; |
| 30 | namespace sys = llvm::sys; |
Mikhail Glushenkov | be9d9a1 | 2008-05-06 18:08:59 +0000 | [diff] [blame] | 31 | using namespace llvmc; |
Anton Korobeynikov | ac67b7e | 2008-03-23 08:57:20 +0000 | [diff] [blame] | 32 | |
Mikhail Glushenkov | 0d08db0 | 2008-05-06 16:35:25 +0000 | [diff] [blame] | 33 | // Built-in command-line options. |
Mikhail Glushenkov | b90cd83 | 2008-05-06 16:34:12 +0000 | [diff] [blame] | 34 | // External linkage here is intentional. |
Mikhail Glushenkov | 0d08db0 | 2008-05-06 16:35:25 +0000 | [diff] [blame] | 35 | |
| 36 | cl::list<std::string> InputFilenames(cl::Positional, cl::desc("<input file>"), |
Mikhail Glushenkov | d7bb87a | 2008-05-06 17:44:16 +0000 | [diff] [blame] | 37 | cl::ZeroOrMore); |
Anton Korobeynikov | ac67b7e | 2008-03-23 08:57:20 +0000 | [diff] [blame] | 38 | cl::opt<std::string> OutputFilename("o", cl::desc("Output file name"), |
Mikhail Glushenkov | 15b064d | 2009-01-14 02:02:16 +0000 | [diff] [blame] | 39 | cl::value_desc("file"), cl::Prefix); |
Mikhail Glushenkov | 87416b4 | 2008-05-06 18:10:53 +0000 | [diff] [blame] | 40 | cl::list<std::string> Languages("x", |
| 41 | cl::desc("Specify the language of the following input files"), |
| 42 | cl::ZeroOrMore); |
Mikhail Glushenkov | 7ef3606 | 2008-05-30 18:48:52 +0000 | [diff] [blame] | 43 | cl::opt<bool> DryRun("dry-run", |
Mikhail Glushenkov | 74a0c28 | 2008-05-30 19:56:27 +0000 | [diff] [blame] | 44 | cl::desc("Only pretend to run commands")); |
Mikhail Glushenkov | 0d08db0 | 2008-05-06 16:35:25 +0000 | [diff] [blame] | 45 | cl::opt<bool> VerboseMode("v", |
| 46 | cl::desc("Enable verbose mode")); |
Mikhail Glushenkov | f8c430b | 2009-01-09 16:16:27 +0000 | [diff] [blame] | 47 | |
| 48 | cl::opt<bool> CheckGraph("check-graph", |
| 49 | cl::desc("Check the compilation graph for errors"), |
| 50 | cl::Hidden); |
Mikhail Glushenkov | 0d08db0 | 2008-05-06 16:35:25 +0000 | [diff] [blame] | 51 | cl::opt<bool> WriteGraph("write-graph", |
Mikhail Glushenkov | 3d68822 | 2008-05-06 18:07:48 +0000 | [diff] [blame] | 52 | cl::desc("Write compilation-graph.dot file"), |
Mikhail Glushenkov | 0d08db0 | 2008-05-06 16:35:25 +0000 | [diff] [blame] | 53 | cl::Hidden); |
| 54 | cl::opt<bool> ViewGraph("view-graph", |
| 55 | cl::desc("Show compilation graph in GhostView"), |
| 56 | cl::Hidden); |
Mikhail Glushenkov | 7329610 | 2008-05-30 06:29:17 +0000 | [diff] [blame] | 57 | cl::opt<bool> SaveTemps("save-temps", |
| 58 | cl::desc("Keep temporary files"), |
| 59 | cl::Hidden); |
Anton Korobeynikov | ac67b7e | 2008-03-23 08:57:20 +0000 | [diff] [blame] | 60 | |
| 61 | namespace { |
Mikhail Glushenkov | 4561ab5 | 2008-05-07 21:50:19 +0000 | [diff] [blame] | 62 | /// BuildTargets - A small wrapper for CompilationGraph::Build. |
Mikhail Glushenkov | 11a353a | 2008-09-22 20:47:46 +0000 | [diff] [blame] | 63 | int BuildTargets(CompilationGraph& graph, const LanguageMap& langMap) { |
Anton Korobeynikov | ac67b7e | 2008-03-23 08:57:20 +0000 | [diff] [blame] | 64 | int ret; |
Mikhail Glushenkov | 7329610 | 2008-05-30 06:29:17 +0000 | [diff] [blame] | 65 | const sys::Path& tempDir = SaveTemps |
| 66 | ? sys::Path("") |
| 67 | : sys::Path(sys::Path::GetTemporaryDirectory()); |
Anton Korobeynikov | ac67b7e | 2008-03-23 08:57:20 +0000 | [diff] [blame] | 68 | |
| 69 | try { |
Mikhail Glushenkov | 11a353a | 2008-09-22 20:47:46 +0000 | [diff] [blame] | 70 | ret = graph.Build(tempDir, langMap); |
Anton Korobeynikov | ac67b7e | 2008-03-23 08:57:20 +0000 | [diff] [blame] | 71 | } |
| 72 | catch(...) { |
| 73 | tempDir.eraseFromDisk(true); |
| 74 | throw; |
| 75 | } |
| 76 | |
Mikhail Glushenkov | 7329610 | 2008-05-30 06:29:17 +0000 | [diff] [blame] | 77 | if (!SaveTemps) |
| 78 | tempDir.eraseFromDisk(true); |
Anton Korobeynikov | ac67b7e | 2008-03-23 08:57:20 +0000 | [diff] [blame] | 79 | return ret; |
| 80 | } |
| 81 | } |
| 82 | |
| 83 | int main(int argc, char** argv) { |
| 84 | try { |
Mikhail Glushenkov | 11a353a | 2008-09-22 20:47:46 +0000 | [diff] [blame] | 85 | LanguageMap langMap; |
Anton Korobeynikov | ac67b7e | 2008-03-23 08:57:20 +0000 | [diff] [blame] | 86 | CompilationGraph graph; |
| 87 | |
Mikhail Glushenkov | 4eaa389 | 2008-05-30 06:11:45 +0000 | [diff] [blame] | 88 | cl::ParseCommandLineOptions |
| 89 | (argc, argv, "LLVM Compiler Driver (Work In Progress)", true); |
Mikhail Glushenkov | 11a353a | 2008-09-22 20:47:46 +0000 | [diff] [blame] | 90 | |
Mikhail Glushenkov | 14ef059 | 2008-09-22 20:51:19 +0000 | [diff] [blame] | 91 | PluginLoader Plugins; |
| 92 | Plugins.PopulateLanguageMap(langMap); |
| 93 | Plugins.PopulateCompilationGraph(graph); |
Mikhail Glushenkov | 0d08db0 | 2008-05-06 16:35:25 +0000 | [diff] [blame] | 94 | |
Mikhail Glushenkov | f8c430b | 2009-01-09 16:16:27 +0000 | [diff] [blame] | 95 | if (CheckGraph) { |
| 96 | return graph.Check(); |
Mikhail Glushenkov | d7bb87a | 2008-05-06 17:44:16 +0000 | [diff] [blame] | 97 | } |
| 98 | |
| 99 | if (ViewGraph) { |
Mikhail Glushenkov | 0d08db0 | 2008-05-06 16:35:25 +0000 | [diff] [blame] | 100 | graph.viewGraph(); |
Mikhail Glushenkov | f8c430b | 2009-01-09 16:16:27 +0000 | [diff] [blame] | 101 | if (!WriteGraph) |
| 102 | return 0; |
| 103 | } |
| 104 | |
| 105 | if (WriteGraph) { |
| 106 | graph.writeGraph(); |
Mikhail Glushenkov | d7bb87a | 2008-05-06 17:44:16 +0000 | [diff] [blame] | 107 | return 0; |
| 108 | } |
| 109 | |
| 110 | if (InputFilenames.empty()) { |
Mikhail Glushenkov | ed3ba40 | 2008-05-30 06:26:35 +0000 | [diff] [blame] | 111 | throw std::runtime_error("no input files"); |
Mikhail Glushenkov | d7bb87a | 2008-05-06 17:44:16 +0000 | [diff] [blame] | 112 | } |
Mikhail Glushenkov | 0d08db0 | 2008-05-06 16:35:25 +0000 | [diff] [blame] | 113 | |
Mikhail Glushenkov | 11a353a | 2008-09-22 20:47:46 +0000 | [diff] [blame] | 114 | return BuildTargets(graph, langMap); |
Anton Korobeynikov | ac67b7e | 2008-03-23 08:57:20 +0000 | [diff] [blame] | 115 | } |
Mikhail Glushenkov | a673037 | 2008-05-12 16:31:42 +0000 | [diff] [blame] | 116 | catch(llvmc::error_code& ec) { |
| 117 | return ec.code(); |
| 118 | } |
Anton Korobeynikov | ac67b7e | 2008-03-23 08:57:20 +0000 | [diff] [blame] | 119 | catch(const std::exception& ex) { |
Mikhail Glushenkov | ed3ba40 | 2008-05-30 06:26:35 +0000 | [diff] [blame] | 120 | std::cerr << argv[0] << ": " << ex.what() << '\n'; |
Anton Korobeynikov | ac67b7e | 2008-03-23 08:57:20 +0000 | [diff] [blame] | 121 | } |
| 122 | catch(...) { |
Mikhail Glushenkov | ed3ba40 | 2008-05-30 06:26:35 +0000 | [diff] [blame] | 123 | std::cerr << argv[0] << ": unknown error!\n"; |
Anton Korobeynikov | ac67b7e | 2008-03-23 08:57:20 +0000 | [diff] [blame] | 124 | } |
Mikhail Glushenkov | d818878 | 2008-05-06 17:25:23 +0000 | [diff] [blame] | 125 | return 1; |
Anton Korobeynikov | ac67b7e | 2008-03-23 08:57:20 +0000 | [diff] [blame] | 126 | } |