Add more helper methods

git-svn-id: https://llvm.org/svn/llvm-project/llvm/trunk@7626 91177308-0d34-0410-b5e6-96231b3b80d8

2 files changed, 30 insertions, 1 deletions

diff --git a/support/tools/TableGen/TableGenBackend.cpp b/support/tools/TableGen/TableGenBackend.cpp
index f00c6ee..c0a67e3 100644
--- a/support/tools/TableGen/TableGenBackend.cpp
+++ b/support/tools/TableGen/TableGenBackend.cpp
@@ -5,13 +5,32 @@
 //===----------------------------------------------------------------------===//
 
 #include "TableGenBackend.h"
+#include "Record.h"
 #include <iostream>
 
 void TableGenBackend::EmitSourceFileHeader(const std::string &Desc,
-                                           std::ostream &OS) {
+                                           std::ostream &OS) const {
   OS << "//===- TableGen'erated file -------------------------------------*-"
        " C++ -*-===//\n//\n// " << Desc << "\n//\n// Automatically generate"
        "d file, do not edit!\n//\n//===------------------------------------"
        "----------------------------------===//\n\n";
 }
 
+/// getQualifiedName - Return the name of the specified record, with a
+/// namespace qualifier if the record contains one.
+///
+std::string TableGenBackend::getQualifiedName(Record *R) const {
+  std::string Namespace = R->getValueAsString("Namespace");
+  if (Namespace.empty()) return R->getName();
+  return Namespace + "::" + R->getName();
+}
+
+/// getTarget - Return the current instance of the Target class.
+///
+Record *TableGenBackend::getTarget(RecordKeeper &RC) const {
+  std::vector<Record*> Targets = RC.getAllDerivedDefinitions("Target");
+
+  if (Targets.size() != 1)
+    throw std::string("ERROR: Multiple subclasses of Target defined!");
+  return Targets[0];
+}
diff --git a/support/tools/TableGen/TableGenBackend.h b/support/tools/TableGen/TableGenBackend.h
index b9e8c49..ae71d2c 100644
--- a/support/tools/TableGen/TableGenBackend.h
+++ b/support/tools/TableGen/TableGenBackend.h
@@ -10,6 +10,8 @@
 
 #include <string>
 #include <iosfwd>
+class Record;
+class RecordKeeper;
 
 struct TableGenBackend {
 
@@ -19,8 +21,16 @@ struct TableGenBackend {
 
 
 public:   // Useful helper routines...
+  /// EmitSourceFileHeader - Output a LLVM style file header to the specified
+  /// ostream.
   void EmitSourceFileHeader(const std::string &Desc, std::ostream &OS) const;
 
+  /// getQualifiedName - Return the name of the specified record, with a
+  /// namespace qualifier if the record contains one.
+  std::string getQualifiedName(Record *R) const;
+
+  /// getTarget - Return the current instance of the Target class.
+  Record *getTarget(RecordKeeper &RC) const;
 };
 
 #endif