1 //===- Main.cpp - Top-Level TableGen implementation -----------------------===// 2 // 3 // The LLVM Compiler Infrastructure 4 // 5 // This file is distributed under the University of Illinois Open Source 6 // License. See LICENSE.TXT for details. 7 // 8 //===----------------------------------------------------------------------===// 9 // 10 // TableGen is a tool which can be used to build up a description of something, 11 // then invoke one or more "tablegen backends" to emit information about the 12 // description in some predefined format. In practice, this is used by the LLVM 13 // code generators to automate generation of a code generator through a 14 // high-level description of the target. 15 // 16 //===----------------------------------------------------------------------===// 17 18 #include "TGParser.h" 19 #include "llvm/ADT/OwningPtr.h" 20 #include "llvm/Support/CommandLine.h" 21 #include "llvm/Support/MemoryBuffer.h" 22 #include "llvm/Support/ToolOutputFile.h" 23 #include "llvm/Support/system_error.h" 24 #include "llvm/TableGen/Error.h" 25 #include "llvm/TableGen/Main.h" 26 #include "llvm/TableGen/Record.h" 27 #include <algorithm> 28 #include <cstdio> 29 using namespace llvm; 30 31 namespace { 32 cl::opt<std::string> 33 OutputFilename("o", cl::desc("Output filename"), cl::value_desc("filename"), 34 cl::init("-")); 35 36 cl::opt<std::string> 37 DependFilename("d", 38 cl::desc("Dependency filename"), 39 cl::value_desc("filename"), 40 cl::init("")); 41 42 cl::opt<std::string> 43 InputFilename(cl::Positional, cl::desc("<input file>"), cl::init("-")); 44 45 cl::list<std::string> 46 IncludeDirs("I", cl::desc("Directory of include files"), 47 cl::value_desc("directory"), cl::Prefix); 48 } 49 50 /// \brief Create a dependency file for `-d` option. 51 /// 52 /// This functionality is really only for the benefit of the build system. 53 /// It is similar to GCC's `-M*` family of options. 54 static int createDependencyFile(const TGParser &Parser, const char *argv0) { 55 if (OutputFilename == "-") { 56 errs() << argv0 << ": the option -d must be used together with -o\n"; 57 return 1; 58 } 59 std::string Error; 60 tool_output_file DepOut(DependFilename.c_str(), Error); 61 if (!Error.empty()) { 62 errs() << argv0 << ": error opening " << DependFilename 63 << ":" << Error << "\n"; 64 return 1; 65 } 66 DepOut.os() << OutputFilename << ":"; 67 const TGLexer::DependenciesMapTy &Dependencies = Parser.getDependencies(); 68 for (TGLexer::DependenciesMapTy::const_iterator I = Dependencies.begin(), 69 E = Dependencies.end(); 70 I != E; ++I) { 71 DepOut.os() << " " << I->first; 72 } 73 DepOut.os() << "\n"; 74 DepOut.keep(); 75 return 0; 76 } 77 78 namespace llvm { 79 80 int TableGenMain(char *argv0, TableGenMainFn *MainFn) { 81 RecordKeeper Records; 82 83 // Parse the input file. 84 OwningPtr<MemoryBuffer> File; 85 if (error_code ec = 86 MemoryBuffer::getFileOrSTDIN(InputFilename, File)) { 87 errs() << "Could not open input file '" << InputFilename << "': " 88 << ec.message() <<"\n"; 89 return 1; 90 } 91 MemoryBuffer *F = File.take(); 92 93 // Tell SrcMgr about this buffer, which is what TGParser will pick up. 94 SrcMgr.AddNewSourceBuffer(F, SMLoc()); 95 96 // Record the location of the include directory so that the lexer can find 97 // it later. 98 SrcMgr.setIncludeDirs(IncludeDirs); 99 100 TGParser Parser(SrcMgr, Records); 101 102 if (Parser.ParseFile()) 103 return 1; 104 105 std::string Error; 106 tool_output_file Out(OutputFilename.c_str(), Error); 107 if (!Error.empty()) { 108 errs() << argv0 << ": error opening " << OutputFilename 109 << ":" << Error << "\n"; 110 return 1; 111 } 112 if (!DependFilename.empty()) { 113 if (int Ret = createDependencyFile(Parser, argv0)) 114 return Ret; 115 } 116 117 if (MainFn(Out.os(), Records)) 118 return 1; 119 120 if (ErrorsPrinted > 0) { 121 errs() << argv0 << ": " << ErrorsPrinted << " errors.\n"; 122 return 1; 123 } 124 125 // Declare success. 126 Out.keep(); 127 return 0; 128 } 129 130 } 131
