Add a new Record::getValueAsCode method to mirror the other getValueAs*
methods. Use it to simplify some code.
git-svn-id: https://llvm.org/svn/llvm-project/llvm/trunk@23336 91177308-0d34-0410-b5e6-96231b3b80d8
diff --git a/utils/TableGen/CodeGenTarget.cpp b/utils/TableGen/CodeGenTarget.cpp
index 61e237f..7c153ec 100644
--- a/utils/TableGen/CodeGenTarget.cpp
+++ b/utils/TableGen/CodeGenTarget.cpp
@@ -154,17 +154,8 @@
SpillAlignment = R->getValueAsInt("Alignment");
VT = getValueType(R->getValueAsDef("RegType"));
- if (CodeInit *CI = dynamic_cast<CodeInit*>(R->getValueInit("MethodBodies")))
- MethodBodies = CI->getValue();
- else
- throw "Expected 'code' fragment for 'MethodBodies' value in register "
- "class '" + getName() + "'!";
-
- if (CodeInit *CI = dynamic_cast<CodeInit*>(R->getValueInit("MethodProtos")))
- MethodProtos = CI->getValue();
- else
- throw "Expected 'code' fragment for 'MethodProtos' value in register "
- "class '" + getName() + "'!";
+ MethodBodies = R->getValueAsCode("MethodBodies");
+ MethodProtos = R->getValueAsCode("MethodProtos");
ListInit *RegList = R->getValueAsListInit("MemberList");
for (unsigned i = 0, e = RegList->getSize(); i != e; ++i) {
diff --git a/utils/TableGen/Record.cpp b/utils/TableGen/Record.cpp
index 077476d..c5f0565 100644
--- a/utils/TableGen/Record.cpp
+++ b/utils/TableGen/Record.cpp
@@ -773,6 +773,18 @@
"' does not have a dag initializer!";
}
+std::string Record::getValueAsCode(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 (const CodeInit *CI = dynamic_cast<const CodeInit*>(R->getValue()))
+ return CI->getValue();
+ throw "Record `" + getName() + "', field `" + FieldName +
+ "' does not have a code initializer!";
+}
+
void RecordKeeper::dump() const { std::cerr << *this; }
diff --git a/utils/TableGen/Record.h b/utils/TableGen/Record.h
index 4ca0890..edd875a 100644
--- a/utils/TableGen/Record.h
+++ b/utils/TableGen/Record.h
@@ -1023,6 +1023,12 @@
/// the value is not the right type.
///
DagInit *getValueAsDag(const std::string &FieldName) const;
+
+ /// getValueAsCode - This method looks up the specified field and returns
+ /// its value as the string data in a CodeInit, throwing an exception if the
+ /// field does not exist or if the value is not a code object.
+ ///
+ std::string getValueAsCode(const std::string &FieldName) const;
};
std::ostream &operator<<(std::ostream &OS, const Record &R);