Keep track of all modules crated using a name to module map.

Add private member function getMoudle().

llvm-svn: 30130
This commit is contained in:
Devang Patel 2006-09-06 18:50:26 +00:00
parent d495bd075a
commit ee9f617cff
2 changed files with 24 additions and 1 deletions

View File

@ -83,6 +83,8 @@ namespace llvm {
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,
@ -91,9 +93,13 @@ namespace llvm {
std::vector<const char*> &exportList,
std::string &targetTriple);
private:
Module *getModule (const std::string &InputFilename);
private:
std::vector<Module *> modules;
NameToSymbolMap allSymbols;
NameToModuleMap allModules;
};
} // End llvm namespace

View File

@ -99,6 +99,23 @@ findExternalRefs(Value *value, std::set<std::string> &references,
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 @@ LinkTimeOptimizer::readLLVMObjectFile(const std::string &InputFilename,
NameToSymbolMap &symbols,
std::set<std::string> &references)
{
Module *m = ParseBytecodeFile(InputFilename);
Module *m = getModule(InputFilename);
if (!m)
return LTO_READ_FAILURE;