utils/TableGen/RegisterInfoEmitter.cpp

   1 //===- RegisterInfoEmitter.cpp - Generate a Register File Desc. -*- C++ -*-===//
   2 //
   3 // This tablegen backend is responsible for emitting a description of a target
   4 // register file for a code generator.  It uses instances of the Register,
   5 // RegisterAliases, and RegisterClass classes to gather this information.
   6 //
   7 //===----------------------------------------------------------------------===//
   8
   9 #include "RegisterInfoEmitter.h"
  10 #include "Record.h"
  11 #include "Support/StringExtras.h"
  12 #include <set>
  13
  14 static void EmitSourceHeader(const std::string &Desc, std::ostream &o) {
  15   o << "//===- TableGen'erated file -------------------------------------*-"
  16        " C++ -*-===//\n//\n// " << Desc << "\n//\n// Automatically generate"
  17        "d file, do not edit!\n//\n//===------------------------------------"
  18        "----------------------------------===//\n\n";
  19 }
  20
  21 // runEnums - Print out enum values for all of the registers.
  22 void RegisterInfoEmitter::runEnums(std::ostream &OS) {
  23   std::vector<Record*> Registers = Records.getAllDerivedDefinitions("Register");
  24
  25   if (Registers.size() == 0)
  26     throw std::string("No 'Register' subclasses defined!");
  27
  28   std::string Namespace = Registers[0]->getValueAsString("Namespace");
  29
  30   EmitSourceHeader("Target Register Enum Values", OS);
  31
  32   if (!Namespace.empty())
  33     OS << "namespace " << Namespace << " {\n";
  34   OS << "  enum {\n    NoRegister,\n";
  35
  36   for (unsigned i = 0, e = Registers.size(); i != e; ++i)
  37     OS << "    " << Registers[i]->getName() << ",\n";
  38
  39   OS << "  };\n";
  40   if (!Namespace.empty())
  41     OS << "}\n";
  42 }
  43
  44 void RegisterInfoEmitter::runHeader(std::ostream &OS) {
  45   std::vector<Record*> RegisterInfos =
  46     Records.getAllDerivedDefinitions("RegisterInfo");
  47
  48   if (RegisterInfos.size() != 1)
  49     throw std::string("ERROR: Multiple subclasses of RegisterInfo defined!");
  50
  51   EmitSourceHeader("Register Information Header Fragment", OS);
  52
  53   std::string ClassName = RegisterInfos[0]->getValueAsString("ClassName");
  54
  55   OS << "#include \"llvm/Target/MRegisterInfo.h\"\n\n";
  56
  57   OS << "struct " << ClassName << " : public MRegisterInfo {\n"
  58      << "  " << ClassName << "();\n"
  59      << "  const unsigned* getCalleeSaveRegs() const;\n"
  60      << "};\n\n";
  61 }
  62
  63 static std::string getQualifiedRecordName(Record *R) {
  64   std::string Namespace = R->getValueAsString("Namespace");
  65   if (Namespace.empty()) return R->getName();
  66   return Namespace + "::" + R->getName();
  67 }
  68
  69 // RegisterInfoEmitter::run - Main register file description emitter.
  70 //
  71 void RegisterInfoEmitter::run(std::ostream &OS) {
  72   EmitSourceHeader("Register Information Source Fragment", OS);
  73
  74   // Start out by emitting each of the register classes... to do this, we build
  75   // a set of registers which belong to a register class, this is to ensure that
  76   // each register is only in a single register class.
  77   //
  78   std::vector<Record*> RegisterClasses =
  79     Records.getAllDerivedDefinitions("RegisterClass");
  80
  81   std::vector<Record*> Registers = Records.getAllDerivedDefinitions("Register");
  82   Record *RegisterClass = Records.getClass("Register");
  83
  84   std::set<Record*> RegistersFound;
  85
  86   // Loop over all of the register classes... emitting each one.
  87   OS << "namespace {     // Register classes...\n";
  88   std::vector<std::string> RegisterClassNames;
  89   for (unsigned rc = 0, e = RegisterClasses.size(); rc != e; ++rc) {
  90     Record *RC = RegisterClasses[rc];
  91     std::string Name = RC->getName();
  92     if (Name[9] == '.') {
  93       static unsigned AnonCounter = 0;
  94       Name = "AnonRegClass_"+utostr(AnonCounter++);
  95     }
  96
  97     // Emit the register list now...
  98     OS << "  // " << Name << " Register Class...\n  const unsigned " << Name
  99        << "[] = {\n    ";
 100     ListInit *RegList = RC->getValueAsListInit("MemberList");
 101     for (unsigned i = 0, e = RegList->getSize(); i != e; ++i) {
 102       Record *Reg = RegList->getElement(i);
 103       if (!Reg->isSubClassOf(RegisterClass))
 104         throw "Register Class member '" + Reg->getName() +
 105               " does not derive from the Register class!";
 106       if (RegistersFound.count(Reg))
 107         throw "Register '" + Reg->getName() +
 108               "' included in multiple register classes!";
 109       OS << getQualifiedRecordName(Reg) << ", ";
 110     }
 111     OS << "\n  };\n\n";
 112
 113     OS << "  struct " << Name << "Class : public TargetRegisterClass {\n"
 114        << "    " << Name << "Class() : TargetRegisterClass("
 115        << RC->getValueAsInt("Size")/8 << ", " << RC->getValueAsInt("Alignment")
 116        << ", " << Name << ", " << Name << " + " << RegList->getSize()
 117        << ") {}\n";
 118
 119     if (CodeInit *CI = dynamic_cast<CodeInit*>(RC->getValueInit("Methods")))
 120       OS << CI->getValue();
 121     else
 122       throw "Expected 'code' fragment for 'Methods' value in register class '"+
 123             RC->getName() + "'!";
 124
 125     OS << "  } " << Name << "Instance;\n\n";
 126
 127   }
 128
 129   OS << "}\n";         // End of anonymous namespace...
 130 }