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