diff options
| -rw-r--r-- | include/llvm/Module.h | 5 | ||||
| -rw-r--r-- | lib/VMCore/Module.cpp | 20 |
2 files changed, 25 insertions, 0 deletions
diff --git a/include/llvm/Module.h b/include/llvm/Module.h index d94500a..cd15178 100644 --- a/include/llvm/Module.h +++ b/include/llvm/Module.h @@ -69,6 +69,11 @@ public: // bool addTypeName(const std::string &Name, const Type *Ty); + // getTypeName - If there is at least one entry in the symbol table for the + // specified type, return it. + // + std::string getTypeName(const Type *Ty); + // Get the underlying elements of the Module... inline const GlobalListType &getGlobalList() const { return GlobalList; } inline GlobalListType &getGlobalList() { return GlobalList; } diff --git a/lib/VMCore/Module.cpp b/lib/VMCore/Module.cpp index 74b5f4b..f3d7cd9 100644 --- a/lib/VMCore/Module.cpp +++ b/lib/VMCore/Module.cpp @@ -86,6 +86,26 @@ bool Module::addTypeName(const std::string &Name, const Type *Ty) { return false; } +// getTypeName - If there is at least one entry in the symbol table for the +// specified type, return it. +// +std::string Module::getTypeName(const Type *Ty) { + const SymbolTable *ST = getSymbolTable(); + if (ST == 0) return ""; // No symbol table, must not have an entry... + if (ST->find(Type::TypeTy) == ST->end()) + return ""; // No names for types... + + SymbolTable::type_const_iterator TI = ST->type_begin(Type::TypeTy); + SymbolTable::type_const_iterator TE = ST->type_end(Type::TypeTy); + + while (TI != TE && TI->second != (const Value*)Ty) + ++TI; + + if (TI != TE) // Must have found an entry! + return TI->first; + return ""; // Must not have found anything... +} + // dropAllReferences() - This function causes all the subinstructions to "let // go" of all references that they are maintaining. This allows one to |