Keep track of all modules crated using a name to module map.
Add private member function getMoudle().
git-svn-id: https://llvm.org/svn/llvm-project/llvm/trunk@30130 91177308-0d34-0410-b5e6-96231b3b80d8
diff --git a/include/llvm/LinkTimeOptimizer.h b/include/llvm/LinkTimeOptimizer.h
index d20d21e..76a7af6 100644
--- a/include/llvm/LinkTimeOptimizer.h
+++ b/include/llvm/LinkTimeOptimizer.h
@@ -83,6 +83,8 @@
public:
typedef hash_map<const char*, LLVMSymbol*, hash<const char*>,
string_compare> NameToSymbolMap;
+ typedef hash_map<const char*, Module*, hash<const char*>,
+ string_compare> NameToModuleMap;
enum LTOStatus readLLVMObjectFile(const std::string &InputFilename,
NameToSymbolMap &symbols,
@@ -92,8 +94,12 @@
std::string &targetTriple);
private:
+ Module *getModule (const std::string &InputFilename);
+
+ private:
std::vector<Module *> modules;
NameToSymbolMap allSymbols;
+ NameToModuleMap allModules;
};
} // End llvm namespace
diff --git a/tools/lto/lto.cpp b/tools/lto/lto.cpp
index e49364e..96b449e 100644
--- a/tools/lto/lto.cpp
+++ b/tools/lto/lto.cpp
@@ -99,6 +99,23 @@
findExternalRefs(c->getOperand(i), references, mangler);
}
+/// InputFilename is a LLVM bytecode file. If Module with InputFilename is
+/// available then return it. Otherwise parseInputFilename.
+Module *
+LinkTimeOptimizer::getModule(const std::string &InputFilename)
+{
+ Module *m = NULL;
+
+ NameToModuleMap::iterator pos = allModules.find(InputFilename.c_str());
+ if (pos != allModules.end())
+ m = allModules[InputFilename.c_str()];
+ else {
+ m = ParseBytecodeFile(InputFilename);
+ allModules[InputFilename.c_str()] = m;
+ }
+ return m;
+}
+
/// InputFilename is a LLVM bytecode file. Read it using bytecode reader.
/// Collect global functions and symbol names in symbols vector.
/// Collect external references in references vector.
@@ -108,7 +125,7 @@
NameToSymbolMap &symbols,
std::set<std::string> &references)
{
- Module *m = ParseBytecodeFile(InputFilename);
+ Module *m = getModule(InputFilename);
if (!m)
return LTO_READ_FAILURE;