averello/swift-mirror
1
0
Fork 0
mirror of https://github.com/apple/swift.git synced 2025-12-21 12:14:44 +01:00
Code Issues Packages Projects Releases Wiki Activity

SILModule: Use llvm::MapVector to store Functions.

Browse Source 
(instead of slumming with a DenseMap/vector pair)
Re: r3322

Swift SVN r3331
This commit is contained in:
Joe Groff
2012-12-03 17:16:34 +00:00
parent fdb8982f35
commit ec1e1aedc4
3 changed files with 19 additions and 12 deletions
Download Patch File Download Diff File Expand all files Collapse all files

24
include/swift/SIL/SILModule.h
View File

@@ -19,7 +19,7 @@
#include "swift/SIL/SILBase.h"
#include "llvm/ADT/ArrayRef.h"
#include "llvm/ADT/DenseMap.h"
#include "llvm/ADT/MapVector.h"
namespace swift {
class TranslationUnit;
@@ -44,9 +44,7 @@ private:
ASTContext &Context;
/// The collection of all Functions in the module.
llvm::DenseMap<ValueDecl*, Function*> functions;
/// FIXME a hack so that SILModule::print dumps functions in source order
std::vector<ValueDecl*> functionDecls;
llvm::MapVector<ValueDecl*, Function*> functions;
// Intentionally marked private so that we need to use 'constructSIL()'
// to construct a SILModule.
@@ -60,13 +58,23 @@ public:
ASTContext &getContext() const { return Context; }
llvm::ArrayRef<ValueDecl*> getFunctionDecls() const {
return functionDecls;
/// Returns true if a Function was generated from the given declaration.
bool hasFunction(ValueDecl *decl) const {
return functions.find(decl) != functions.end();
}
llvm::DenseMap<ValueDecl*, Function*> const &getFunctions() const {
return functions;
/// Returns a pointer to the Function generated from the given declaration.
Function *getFunction(ValueDecl *decl) const {
auto found = functions.find(decl);
assert(found != functions.end() && "no Function generated for Decl");
return found->second;
}
typedef llvm::MapVector<ValueDecl*, Function*>::const_iterator iterator;
typedef iterator const_iterator;
iterator begin() const { return functions.begin(); }
iterator end() const { return functions.end(); }
/// verify - Run the SIL verifier to make sure that all Functions follow
/// invariants.