diff options
-rw-r--r-- | utils/TableGen/Record.cpp | 19 | ||||
-rw-r--r-- | utils/TableGen/Record.h | 6 |
2 files changed, 25 insertions, 0 deletions
diff --git a/utils/TableGen/Record.cpp b/utils/TableGen/Record.cpp index c5f0565749..109ed39251 100644 --- a/utils/TableGen/Record.cpp +++ b/utils/TableGen/Record.cpp @@ -709,6 +709,25 @@ ListInit *Record::getValueAsListInit(const std::string &FieldName) const { "' does not have a list initializer!"; } +/// getValueAsListDef - This method looks up the specified field and returns +/// its value as a vector of records, throwing an exception if the field does +/// not exist or if the value is not the right type. +/// +std::vector<Record*> Record::getValueAsListDef(const std::string &FieldName) + const { + ListInit *List = getValueAsListInit(FieldName); + std::vector<Record*> Defs; + for (unsigned i = 0; i < List->getSize(); i++) { + if (DefInit *DI = dynamic_cast<DefInit*>(List->getElement(i))) { + Defs.push_back(DI->getDef()); + } else { + throw "Record `" + getName() + "', field `" + FieldName + + "' list is not entirely DefInit!"; + } + } + return Defs; +} + /// getValueAsInt - This method looks up the specified field and returns its /// value as an int, throwing an exception if the field does not exist or if /// the value is not the right type. diff --git a/utils/TableGen/Record.h b/utils/TableGen/Record.h index edd875afce..ecd6d7020a 100644 --- a/utils/TableGen/Record.h +++ b/utils/TableGen/Record.h @@ -1000,6 +1000,12 @@ public: /// ListInit *getValueAsListInit(const std::string &FieldName) const; + /// getValueAsListDef - This method looks up the specified field and returns + /// its value as a vector of records, throwing an exception if the field does + /// not exist or if the value is not the right type. + /// + std::vector<Record*> getValueAsListDef(const std::string &FieldName) const; + /// getValueAsDef - This method looks up the specified field and returns its /// value as a Record, throwing an exception if the field does not exist or if /// the value is not the right type. |