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() << ",\t// " << i+1 << "\n";
  38
  39   OS << "  };\n";
  40   if (!Namespace.empty())
  41     OS << "}\n";
  42 }
  43
  44 static Record *getRegisterInfo(RecordKeeper &RC) {
  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   return RegisterInfos[0];
  51 }
  52
  53 void RegisterInfoEmitter::runHeader(std::ostream &OS) {
  54   EmitSourceHeader("Register Information Header Fragment", OS);
  55
  56   std::string ClassName =
  57     getRegisterInfo(Records)->getValueAsString("ClassName");
  58
  59   OS << "#include \"llvm/Target/MRegisterInfo.h\"\n\n";
  60
  61   OS << "struct " << ClassName << " : public MRegisterInfo {\n"
  62      << "  " << ClassName
  63      << "(int CallFrameSetupOpcode = -1, int CallFrameDestroyOpcode = -1);\n"
  64      << "  const unsigned* getCalleeSaveRegs() const;\n"
  65      << "};\n\n";
  66 }
  67
  68 static std::string getQualifiedName(Record *R) {
  69   std::string Namespace = R->getValueAsString("Namespace");
  70   if (Namespace.empty()) return R->getName();
  71   return Namespace + "::" + R->getName();
  72 }
  73
  74 // RegisterInfoEmitter::run - Main register file description emitter.
  75 //
  76 void RegisterInfoEmitter::run(std::ostream &OS) {
  77   EmitSourceHeader("Register Information Source Fragment", OS);
  78
  79   // Start out by emitting each of the register classes... to do this, we build
  80   // a set of registers which belong to a register class, this is to ensure that
  81   // each register is only in a single register class.
  82   //
  83   std::vector<Record*> RegisterClasses =
  84     Records.getAllDerivedDefinitions("RegisterClass");
  85
  86   std::vector<Record*> Registers = Records.getAllDerivedDefinitions("Register");
  87   Record *RegisterClass = Records.getClass("Register");
  88
  89   std::set<Record*> RegistersFound;
  90   std::vector<std::string> RegClassNames;
  91
  92   // Loop over all of the register classes... emitting each one.
  93   OS << "namespace {     // Register classes...\n";
  94
  95   for (unsigned rc = 0, e = RegisterClasses.size(); rc != e; ++rc) {
  96     Record *RC = RegisterClasses[rc];
  97     std::string Name = RC->getName();
  98     if (Name[9] == '.') {
  99       static unsigned AnonCounter = 0;
 100       Name = "AnonRegClass_"+utostr(AnonCounter++);
 101     }
 102
 103     RegClassNames.push_back(Name);
 104
 105     // Emit the register list now...
 106     OS << "  // " << Name << " Register Class...\n  const unsigned " << Name
 107        << "[] = {\n    ";
 108     ListInit *RegList = RC->getValueAsListInit("MemberList");
 109     for (unsigned i = 0, e = RegList->getSize(); i != e; ++i) {
 110       Record *Reg = RegList->getElement(i);
 111       if (!Reg->isSubClassOf(RegisterClass))
 112         throw "Register Class member '" + Reg->getName() +
 113               " does not derive from the Register class!";
 114       if (RegistersFound.count(Reg))
 115         throw "Register '" + Reg->getName() +
 116               "' included in multiple register classes!";
 117       RegistersFound.insert(Reg);
 118       OS << getQualifiedName(Reg) << ", ";
 119     }
 120     OS << "\n  };\n\n";
 121
 122     OS << "  struct " << Name << "Class : public TargetRegisterClass {\n"
 123        << "    " << Name << "Class() : TargetRegisterClass("
 124        << RC->getValueAsInt("Size")/8 << ", " << RC->getValueAsInt("Alignment")
 125        << ", " << Name << ", " << Name << " + " << RegList->getSize()
 126        << ") {}\n";
 127
 128     if (CodeInit *CI = dynamic_cast<CodeInit*>(RC->getValueInit("Methods")))
 129       OS << CI->getValue();
 130     else
 131       throw "Expected 'code' fragment for 'Methods' value in register class '"+
 132             RC->getName() + "'!";
 133
 134     OS << "  } " << Name << "Instance;\n\n";
 135   }
 136
 137   OS << "  const TargetRegisterClass* const RegisterClasses[] = {\n";
 138   for (unsigned i = 0, e = RegClassNames.size(); i != e; ++i)
 139     OS << "    &" << RegClassNames[i] << "Instance,\n";
 140   OS << "  };\n";
 141
 142   // Emit register class aliases...
 143   std::vector<Record*> RegisterAliasesRecs =
 144     Records.getAllDerivedDefinitions("RegisterAliases");
 145   std::map<Record*, std::set<Record*> > RegisterAliases;
 146
 147   for (unsigned i = 0, e = RegisterAliasesRecs.size(); i != e; ++i) {
 148     Record *AS = RegisterAliasesRecs[i];
 149     Record *R = AS->getValueAsDef("Reg");
 150     ListInit *LI = AS->getValueAsListInit("Aliases");
 151
 152     // Add information that R aliases all of the elements in the list... and
 153     // that everything in the list aliases R.
 154     for (unsigned j = 0, e = LI->getSize(); j != e; ++j) {
 155       if (RegisterAliases[R].count(LI->getElement(j)))
 156         std::cerr << "Warning: register alias between " << getQualifiedName(R)
 157                   << " and " << getQualifiedName(LI->getElement(j))
 158                   << " specified multiple times!\n";
 159       RegisterAliases[R].insert(LI->getElement(j));
 160
 161       if (RegisterAliases[LI->getElement(j)].count(R))
 162         std::cerr << "Warning: register alias between " << getQualifiedName(R)
 163                   << " and " << getQualifiedName(LI->getElement(j))
 164                   << " specified multiple times!\n";
 165       RegisterAliases[LI->getElement(j)].insert(R);
 166     }
 167   }
 168
 169   if (!RegisterAliases.empty())
 170     OS << "\n\n  // Register Alias Sets...\n";
 171
 172   // Loop over all of the registers which have aliases, emitting the alias list
 173   // to memory.
 174   for (std::map<Record*, std::set<Record*> >::iterator
 175          I = RegisterAliases.begin(), E = RegisterAliases.end(); I != E; ++I) {
 176     OS << "  const unsigned " << I->first->getName() << "_AliasSet[] = { ";
 177     for (std::set<Record*>::iterator ASI = I->second.begin(),
 178            E = I->second.end(); ASI != E; ++ASI)
 179       OS << getQualifiedName(*ASI) << ", ";
 180     OS << "0 };\n";
 181   }
 182
 183   OS << "\n  const MRegisterDesc RegisterDescriptors[] = { // Descriptors\n";
 184   OS << "    { \"NOREG\",\t0,\t\t0,\t0 },\n";
 185   // Now that register alias sets have been emitted, emit the register
 186   // descriptors now.
 187   for (unsigned i = 0, e = Registers.size(); i != e; ++i) {
 188     Record *Reg = Registers[i];
 189     OS << "    { \"" << Reg->getName() << "\",\t";
 190     if (RegisterAliases.count(Reg))
 191       OS << Reg->getName() << "_AliasSet,\t";
 192     else
 193       OS << "0,\t\t";
 194     OS << "0, 0 },\n";
 195   }
 196   OS << "  };\n";      // End of register descriptors...
 197   OS << "}\n\n";       // End of anonymous namespace...
 198
 199   Record *RegisterInfo = getRegisterInfo(Records);
 200   std::string ClassName = RegisterInfo->getValueAsString("ClassName");
 201
 202   // Emit the constructor of the class...
 203   OS << ClassName << "::" << ClassName
 204      << "(int CallFrameSetupOpcode, int CallFrameDestroyOpcode)\n"
 205      << "  : MRegisterInfo(RegisterDescriptors, " << Registers.size()+1
 206      << ", RegisterClasses, RegisterClasses+" << RegClassNames.size() << ",\n "
 207      << "                 CallFrameSetupOpcode, CallFrameDestroyOpcode) {}\n\n";
 208
 209   // Emit the getCalleeSaveRegs method...
 210   OS << "const unsigned* " << ClassName << "::getCalleeSaveRegs() const {\n"
 211      << "  static const unsigned CalleeSaveRegs[] = {\n    ";
 212
 213   ListInit *LI = RegisterInfo->getValueAsListInit("CalleeSavedRegisters");
 214   for (unsigned i = 0, e = LI->getSize(); i != e; ++i)
 215     OS << getQualifiedName(LI->getElement(i)) << ", ";
 216   OS << " 0\n  };\n  return CalleeSaveRegs;\n}\n\n";
 217 }