Add new method getVAlueAsDag

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

2 files changed, 25 insertions, 3 deletions

diff --git a/utils/TableGen/Record.cpp b/utils/TableGen/Record.cpp
index 4032f40..a76c55c 100644
--- a/utils/TableGen/Record.cpp
+++ b/utils/TableGen/Record.cpp
@@ -613,10 +613,26 @@ bool Record::getValueAsBit(const std::string &FieldName) const {
     throw "Record '" + getName() + "' does not have a field named '" +
       FieldName + "!\n";
 
-  if (BitInit *DI = dynamic_cast<BitInit*>(R->getValue()))
-    return DI->getValue();
+  if (BitInit *BI = dynamic_cast<BitInit*>(R->getValue()))
+    return BI->getValue();
   throw "Record '" + getName() + "', field '" + FieldName +
-        "' does not have a list initializer!";
+        "' does not have a bit initializer!";
+}
+
+/// getValueAsDag - This method looks up the specified field and returns its
+/// value as an Dag, throwing an exception if the field does not exist or if
+/// the value is not the right type.
+///
+DagInit *Record::getValueAsDag(const std::string &FieldName) const {
+  const RecordVal *R = getValue(FieldName);
+  if (R == 0 || R->getValue() == 0)
+    throw "Record '" + getName() + "' does not have a field named '" +
+      FieldName + "!\n";
+
+  if (DagInit *DI = dynamic_cast<DagInit*>(R->getValue()))
+    return DI;
+  throw "Record '" + getName() + "', field '" + FieldName +
+        "' does not have a dag initializer!";
 }
 
 
diff --git a/utils/TableGen/Record.h b/utils/TableGen/Record.h
index b26e13e..b621451 100644
--- a/utils/TableGen/Record.h
+++ b/utils/TableGen/Record.h
@@ -759,6 +759,12 @@ public:
   /// the value is not the right type.
   ///
   int getValueAsInt(const std::string &FieldName) const;
+
+  /// getValueAsDag - This method looks up the specified field and returns its
+  /// value as an Dag, throwing an exception if the field does not exist or if
+  /// the value is not the right type.
+  ///
+  DagInit *getValueAsDag(const std::string &FieldName) const;
 };
 
 std::ostream &operator<<(std::ostream &OS, const Record &R);