diff --git a/include/llvm/Module.h b/include/llvm/Module.h index b9c98814f15..8c50098dced 100644 --- a/include/llvm/Module.h +++ b/include/llvm/Module.h @@ -309,6 +309,9 @@ public: /// is none by that name. StructType *getTypeByName(StringRef Name) const; + /// getNumNamedStructTypes - Return the number of named struct types. + unsigned getNumNamedStructTypes() const; + /// @} /// @name Function Accessors /// @{ diff --git a/lib/VMCore/Module.cpp b/lib/VMCore/Module.cpp index 3c67191e09c..3b0926404be 100644 --- a/lib/VMCore/Module.cpp +++ b/lib/VMCore/Module.cpp @@ -481,76 +481,116 @@ namespace { // objects, we keep several helper maps. DenseSet VisitedConstants; DenseSet VisitedTypes; - std::vector &StructTypes; + + const Module *M; + unsigned NumNamedStructTypes; + unsigned StructTypesSize; public: TypeFinder(std::vector &structTypes) - : StructTypes(structTypes) {} + : StructTypes(structTypes), M(0), NumNamedStructTypes(0), + StructTypesSize(0) {} - void run(const Module &M) { + void run(const Module &Mod) { + M = &Mod; + NumNamedStructTypes = M->getNumNamedStructTypes(); + + if (NumNamedStructTypes == 0) + return; + // Get types from global variables. - for (Module::const_global_iterator I = M.global_begin(), - E = M.global_end(); I != E; ++I) { + for (Module::const_global_iterator I = M->global_begin(), + E = M->global_end(); I != E; ++I) { incorporateType(I->getType()); if (I->hasInitializer()) incorporateValue(I->getInitializer()); } // Get types from aliases. - for (Module::const_alias_iterator I = M.alias_begin(), - E = M.alias_end(); I != E; ++I) { + for (Module::const_alias_iterator I = M->alias_begin(), + E = M->alias_end(); I != E; ++I) { incorporateType(I->getType()); if (const Value *Aliasee = I->getAliasee()) incorporateValue(Aliasee); } + + if (foundAllStructTypes()) return; SmallVector, 4> MDForInst; // Get types from functions. - for (Module::const_iterator FI = M.begin(), E = M.end(); FI != E; ++FI) { + for (Module::const_iterator FI = M->begin(), + E = M->end(); FI != E; ++FI) { incorporateType(FI->getType()); - + if (foundAllStructTypes()) return; + for (Function::const_iterator BB = FI->begin(), E = FI->end(); BB != E;++BB) for (BasicBlock::const_iterator II = BB->begin(), E = BB->end(); II != E; ++II) { const Instruction &I = *II; + + // Incorporate the type of the instruction and all its operands. incorporateType(I.getType()); for (User::const_op_iterator OI = I.op_begin(), OE = I.op_end(); - OI != OE; ++OI) + OI != OE; ++OI) { incorporateValue(*OI); - + if (foundAllStructTypes()) return; + } + // Incorporate types hiding in metadata. I.getAllMetadataOtherThanDebugLoc(MDForInst); - for (unsigned i = 0, e = MDForInst.size(); i != e; ++i) + for (unsigned i = 0, e = MDForInst.size(); i != e; ++i) { incorporateMDNode(MDForInst[i].second); + if (foundAllStructTypes()) return; + } + MDForInst.clear(); } } - for (Module::const_named_metadata_iterator I = M.named_metadata_begin(), - E = M.named_metadata_end(); I != E; ++I) { + for (Module::const_named_metadata_iterator I = M->named_metadata_begin(), + E = M->named_metadata_end(); I != E; ++I) { const NamedMDNode *NMD = I; - for (unsigned i = 0, e = NMD->getNumOperands(); i != e; ++i) + for (unsigned i = 0, e = NMD->getNumOperands(); i != e; ++i) { incorporateMDNode(NMD->getOperand(i)); + if (foundAllStructTypes()) return; + } } } private: + bool foundAllStructTypes() { + return StructTypesSize == NumNamedStructTypes; + } + + bool addStructType(StructType *ST) { + if (foundAllStructTypes()) return false; + StructTypes.push_back(ST); + ++StructTypesSize; + return true; + } + void incorporateType(Type *Ty) { + if (StructTypesSize == NumNamedStructTypes) + return; + // Check to see if we're already visited this type. if (!VisitedTypes.insert(Ty).second) return; // If this is a structure or opaque type, add a name for the type. if (StructType *STy = dyn_cast(Ty)) - StructTypes.push_back(STy); + if (!addStructType(STy)) + return; // Recursively walk all contained types. for (Type::subtype_iterator I = Ty->subtype_begin(), - E = Ty->subtype_end(); I != E; ++I) + E = Ty->subtype_end(); I != E; ++I) { incorporateType(*I); + if (foundAllStructTypes()) return; + } } /// incorporateValue - This method is used to walk operand lists finding @@ -560,7 +600,9 @@ namespace { void incorporateValue(const Value *V) { if (const MDNode *M = dyn_cast(V)) return incorporateMDNode(M); - if (!isa(V) || isa(V)) return; + + if (!isa(V) || isa(V)) + return; // Already visited? if (!VisitedConstants.insert(V).second) @@ -577,15 +619,18 @@ namespace { } void incorporateMDNode(const MDNode *V) { - + if (foundAllStructTypes()) return; + // Already visited? if (!VisitedConstants.insert(V).second) return; // Look in operands for types. for (unsigned i = 0, e = V->getNumOperands(); i != e; ++i) - if (Value *Op = V->getOperand(i)) + if (Value *Op = V->getOperand(i)) { incorporateValue(Op); + if (foundAllStructTypes()) return; + } } }; } // end anonymous namespace diff --git a/lib/VMCore/Type.cpp b/lib/VMCore/Type.cpp index c6f35580e15..3f333aaf581 100644 --- a/lib/VMCore/Type.cpp +++ b/lib/VMCore/Type.cpp @@ -638,6 +638,10 @@ StructType *Module::getTypeByName(StringRef Name) const { return 0; } +/// getNumNamedStructTypes - Return the number of named struct types. +unsigned Module::getNumNamedStructTypes() const { + return getContext().pImpl->NamedStructTypes.size(); +} //===----------------------------------------------------------------------===// // CompositeType Implementation