utils/TableGen/TableGen.cpp

   1 //===- TableGen.cpp - Top-Level TableGen implementation -------------------===//
   2 //
   3 //                     The LLVM Compiler Infrastructure
   4 //
   5 // This file was developed by the LLVM research group and is distributed under
   6 // the University of Illinois Open Source 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 "Record.h"
  19 #include "llvm/Support/CommandLine.h"
  20 #include "llvm/System/Signals.h"
  21 #include "llvm/Support/FileUtilities.h"
  22 #include "CodeEmitterGen.h"
  23 #include "RegisterInfoEmitter.h"
  24 #include "InstrInfoEmitter.h"
  25 #include "AsmWriterEmitter.h"
  26 #include "DAGISelEmitter.h"
  27 #include "SubtargetEmitter.h"
  28 #include "IntrinsicEmitter.h"
  29 #include <algorithm>
  30 #include <cstdio>
  31 #include <fstream>
  32 #include <ios>
  33 using namespace llvm;
  34
  35 enum ActionType {
  36   PrintRecords,
  37   GenEmitter,
  38   GenRegisterEnums, GenRegister, GenRegisterHeader,
  39   GenInstrEnums, GenInstrs, GenAsmWriter,
  40   GenDAGISel,
  41   GenSubtarget,
  42   GenIntrinsic,
  43   PrintEnums
  44 };
  45
  46 namespace {
  47   cl::opt<ActionType>
  48   Action(cl::desc("Action to perform:"),
  49          cl::values(clEnumValN(PrintRecords, "print-records",
  50                                "Print all records to stdout (default)"),
  51                     clEnumValN(GenEmitter, "gen-emitter",
  52                                "Generate machine code emitter"),
  53                     clEnumValN(GenRegisterEnums, "gen-register-enums",
  54                                "Generate enum values for registers"),
  55                     clEnumValN(GenRegister, "gen-register-desc",
  56                                "Generate a register info description"),
  57                     clEnumValN(GenRegisterHeader, "gen-register-desc-header",
  58                                "Generate a register info description header"),
  59                     clEnumValN(GenInstrEnums, "gen-instr-enums",
  60                                "Generate enum values for instructions"),
  61                     clEnumValN(GenInstrs, "gen-instr-desc",
  62                                "Generate instruction descriptions"),
  63                     clEnumValN(GenAsmWriter, "gen-asm-writer",
  64                                "Generate assembly writer"),
  65                     clEnumValN(GenDAGISel, "gen-dag-isel",
  66                                "Generate a DAG instruction selector"),
  67                     clEnumValN(GenSubtarget, "gen-subtarget",
  68                                "Generate subtarget enumerations"),
  69                     clEnumValN(GenIntrinsic, "gen-intrinsic",
  70                                "Generate intrinsic information"),
  71                     clEnumValN(PrintEnums, "print-enums",
  72                                "Print enum values for a class"),
  73                     clEnumValEnd));
  74
  75   cl::opt<std::string>
  76   Class("class", cl::desc("Print Enum list for this class"),
  77         cl::value_desc("class name"));
  78
  79   cl::opt<std::string>
  80   OutputFilename("o", cl::desc("Output filename"), cl::value_desc("filename"),
  81                  cl::init("-"));
  82
  83   cl::opt<std::string>
  84   InputFilename(cl::Positional, cl::desc("<input file>"), cl::init("-"));
  85
  86   cl::list<std::string>
  87   IncludeDirs("I", cl::desc("Directory of include files"),
  88               cl::value_desc("directory"), cl::Prefix);
  89 }
  90
  91 namespace llvm {
  92   void ParseFile(const std::string &Filename,
  93                  const std::vector<std::string> &IncludeDirs);
  94 }
  95
  96 RecordKeeper llvm::Records;
  97
  98 int main(int argc, char **argv) {
  99   cl::ParseCommandLineOptions(argc, argv);
 100   ParseFile(InputFilename, IncludeDirs);
 101
 102   std::ostream *Out = &std::cout;
 103   if (OutputFilename != "-") {
 104     Out = new std::ofstream(OutputFilename.c_str());
 105
 106     if (!Out->good()) {
 107       std::cerr << argv[0] << ": error opening " << OutputFilename << "!\n";
 108       return 1;
 109     }
 110
 111     // Make sure the file gets removed if *gasp* tablegen crashes...
 112     sys::RemoveFileOnSignal(sys::Path(OutputFilename));
 113   }
 114
 115   try {
 116     switch (Action) {
 117     case PrintRecords:
 118       *Out << Records;           // No argument, dump all contents
 119       break;
 120     case GenEmitter:
 121       CodeEmitterGen(Records).run(*Out);
 122       break;
 123
 124     case GenRegisterEnums:
 125       RegisterInfoEmitter(Records).runEnums(*Out);
 126       break;
 127     case GenRegister:
 128       RegisterInfoEmitter(Records).run(*Out);
 129       break;
 130     case GenRegisterHeader:
 131       RegisterInfoEmitter(Records).runHeader(*Out);
 132       break;
 133
 134     case GenInstrEnums:
 135       InstrInfoEmitter(Records).runEnums(*Out);
 136       break;
 137     case GenInstrs:
 138       InstrInfoEmitter(Records).run(*Out);
 139       break;
 140
 141     case GenAsmWriter:
 142       AsmWriterEmitter(Records).run(*Out);
 143       break;
 144
 145     case GenDAGISel:
 146       DAGISelEmitter(Records).run(*Out);
 147       break;
 148     case GenSubtarget:
 149       SubtargetEmitter(Records).run(*Out);
 150       break;
 151     case GenIntrinsic:
 152       IntrinsicEmitter(Records).run(*Out);
 153       break;
 154     case PrintEnums:
 155     {
 156       std::vector<Record*> Recs = Records.getAllDerivedDefinitions(Class);
 157       for (unsigned i = 0, e = Recs.size(); i != e; ++i)
 158         *Out << Recs[i]->getName() << ", ";
 159       *Out << "\n";
 160       break;
 161     }
 162     default:
 163       assert(1 && "Invalid Action");
 164       return 1;
 165     }
 166   } catch (const std::string &Error) {
 167     std::cerr << argv[0] << ": " << Error << "\n";
 168     if (Out != &std::cout) {
 169       delete Out;                             // Close the file
 170       std::remove(OutputFilename.c_str());    // Remove the file, it's broken
 171     }
 172     return 1;
 173   } catch (...) {
 174     std::cerr << argv[0] << ": Unknown unexpected exception occurred.\n";
 175     if (Out != &std::cout) {
 176       delete Out;                             // Close the file
 177       std::remove(OutputFilename.c_str());    // Remove the file, it's broken
 178     }
 179     return 2;
 180   }
 181
 182   if (Out != &std::cout) {
 183     delete Out;                               // Close the file
 184   }
 185   return 0;
 186 }