diff --git a/llvm/include/llvm/Analysis/DebugInfo.h b/llvm/include/llvm/Analysis/DebugInfo.h index a9b21b9ed1bc..5c8b31ad5e2b 100644 --- a/llvm/include/llvm/Analysis/DebugInfo.h +++ b/llvm/include/llvm/Analysis/DebugInfo.h @@ -643,6 +643,9 @@ namespace llvm { // addSubprogram - Add subprgoram into SPs. bool addSubprogram(DISubprogram SP); + /// addType - Add type into Tys. + bool addType(DIType DT); + public: typedef SmallVector::iterator iterator; iterator compile_unit_begin() { return CUs.begin(); } @@ -651,15 +654,19 @@ namespace llvm { iterator subprogram_end() { return SPs.end(); } iterator global_variable_begin() { return GVs.begin(); } iterator global_variable_end() { return GVs.end(); } + iterator type_begin() { return TYs.begin(); } + iterator type_end() { return TYs.end(); } unsigned compile_unit_count() { return CUs.size(); } unsigned global_variable_count() { return GVs.size(); } unsigned subprogram_count() { return SPs.size(); } + unsigned type_count() { return TYs.size(); } private: SmallVector CUs; // Compile Units SmallVector SPs; // Subprograms - SmallVector GVs; // Global Variables; + SmallVector GVs; // Global Variables + SmallVector TYs; // Types SmallPtrSet NodesSeen; }; diff --git a/llvm/lib/Analysis/DebugInfo.cpp b/llvm/lib/Analysis/DebugInfo.cpp index b061827c32a6..3d9c93378fba 100644 --- a/llvm/lib/Analysis/DebugInfo.cpp +++ b/llvm/lib/Analysis/DebugInfo.cpp @@ -955,9 +955,7 @@ void DebugInfoFinder::processModule(Module &M) { /// processType - Process DIType. void DebugInfoFinder::processType(DIType DT) { - if (DT.isNull()) - return; - if (!NodesSeen.insert(DT.getGV())) + if (!addType(DT)) return; addCompileUnit(DT.getCompileUnit()); @@ -1028,6 +1026,18 @@ void DebugInfoFinder::processDeclare(DbgDeclareInst *DDI) { processType(DV.getType()); } +/// addType - Add type into Tys. +bool DebugInfoFinder::addType(DIType DT) { + if (DT.isNull()) + return false; + + if (!NodesSeen.insert(DT.getGV())) + return false; + + TYs.push_back(DT.getGV()); + return true; +} + /// addCompileUnit - Add compile unit into CUs. bool DebugInfoFinder::addCompileUnit(DICompileUnit CU) { if (CU.isNull())