Add the ability to emit register file enums
[oota-llvm.git] / 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 <set>
12
13 static void EmitSourceHeader(const std::string &Desc, std::ostream &o) {
14   o << "//===- TableGen'erated file -------------------------------------*-"
15        " C++ -*-===//\n//\n// " << Desc << "\n//\n// Automatically generate"
16        "d file, do not edit!\n//\n//===------------------------------------"
17        "----------------------------------===//\n\n";
18 }
19
20 // runEnums - Print out enum values for all of the registers.
21 void RegisterInfoEmitter::runEnums(std::ostream &OS) {
22   std::vector<Record*> Registers = Records.getAllDerivedDefinitions("Register");
23
24   if (Registers.size() == 0)
25     throw std::string("No 'Register' subclasses defined!");
26
27   std::string Namespace = Registers[0]->getValueAsString("Namespace");
28
29   EmitSourceHeader("Target Register Enum Values", OS);
30
31   if (!Namespace.empty())
32     OS << "namespace " << Namespace << " {\n";
33   OS << "  enum {\n    NoRegister,\n";
34
35   for (unsigned i = 0, e = Registers.size(); i != e; ++i)
36     OS << "    " << Registers[i]->getName() << ",\n";
37   
38   OS << "  };\n";
39   if (!Namespace.empty())
40     OS << "}\n";
41 }
42
43 void RegisterInfoEmitter::runHeader(std::ostream &OS) {
44   std::vector<Record*> RegisterInfos =
45     Records.getAllDerivedDefinitions("RegisterInfo");
46
47   if (RegisterInfos.size() != 1)
48     throw std::string("ERROR: Multiple subclasses of RegisterInfo defined!");
49
50   EmitSourceHeader("Register Information Header Fragment", OS);
51
52   std::string ClassName = RegisterInfos[0]->getValueAsString("ClassName");
53
54   OS << "#include \"llvm/Target/MRegisterInfo.h\"\n\n";
55
56   OS << "struct " << ClassName << " : public MRegisterInfo {\n"
57      << "  " << ClassName << "();\n"
58      << "  const unsigned* getCalleeSaveRegs() const;\n"
59      << "};\n\n";
60 }
61
62 // RegisterInfoEmitter::run - Main register file description emitter.
63 //
64 void RegisterInfoEmitter::run(std::ostream &OS) {
65   EmitSourceHeader("Register Information Source Fragment", OS);
66
67   // Start out by emitting each of the register classes... to do this, we build
68   // a set of registers which belong to a register class, this is to ensure that
69   // each register is only in a single register class.
70   //
71   std::vector<Record*> RegisterClasses =
72     Records.getAllDerivedDefinitions("RegisterClass");
73
74   std::set<Record*> RegistersFound;
75
76   // Loop over all of the register classes... emitting each one.
77   OS << "namespace {     // Register classes...\n";
78   std::vector<std::string> RegisterClassNames;
79   for (unsigned rc = 0, e = RegisterClasses.size(); rc != e; ++rc) {
80     Record *RC = RegisterClasses[rc];
81     std::string Name = RC->getName();
82     //if (Name[
83
84   }
85
86   OS << "}\n";         // End of anonymous namespace...
87 }