diff options
author | Chris Lattner <sabre@nondot.org> | 2002-09-12 17:06:40 +0000 |
---|---|---|
committer | Chris Lattner <sabre@nondot.org> | 2002-09-12 17:06:40 +0000 |
commit | e0a54f88a0d8b8d68b2b9d44b0eb5e234e0cac74 (patch) | |
tree | 4fab155837105e01a98f786eae942c46b7ad074d /lib | |
parent | 287d4436c5afff553019f8971f382e36e02bc0af (diff) | |
download | llvm-e0a54f88a0d8b8d68b2b9d44b0eb5e234e0cac74.tar.gz llvm-e0a54f88a0d8b8d68b2b9d44b0eb5e234e0cac74.tar.bz2 llvm-e0a54f88a0d8b8d68b2b9d44b0eb5e234e0cac74.tar.xz |
Add a new BasicBlockPass::doInitialization/Finalization(Function &) pair of
methods that may be useful for BasicBlockPasses.
git-svn-id: https://llvm.org/svn/llvm-project/llvm/trunk@3688 91177308-0d34-0410-b5e6-96231b3b80d8
Diffstat (limited to 'lib')
-rw-r--r-- | lib/VMCore/Pass.cpp | 10 | ||||
-rw-r--r-- | lib/VMCore/PassManagerT.h | 16 |
2 files changed, 22 insertions, 4 deletions
diff --git a/lib/VMCore/Pass.cpp b/lib/VMCore/Pass.cpp index b673484f21..19b564e61f 100644 --- a/lib/VMCore/Pass.cpp +++ b/lib/VMCore/Pass.cpp @@ -341,18 +341,20 @@ void FunctionPass::addToPassManager(PassManagerT<Function> *PM, // function. // bool BasicBlockPass::runOnFunction(Function &F) { - bool Changed = false; + bool Changed = doInitialization(F); for (Function::iterator I = F.begin(), E = F.end(); I != E; ++I) Changed |= runOnBasicBlock(*I); - return Changed; + return Changed | doFinalization(F); } // To run directly on the basic block, we initialize, runOnBasicBlock, then // finalize. // bool BasicBlockPass::run(BasicBlock &BB) { - Module &M = *BB.getParent()->getParent(); - return doInitialization(M) | runOnBasicBlock(BB) | doFinalization(M); + Function &F = *BB.getParent(); + Module &M = *F.getParent(); + return doInitialization(M) | doInitialization(F) | runOnBasicBlock(BB) | + doFinalization(F) | doFinalization(M); } void BasicBlockPass::addToPassManager(PassManagerT<Function> *PM, diff --git a/lib/VMCore/PassManagerT.h b/lib/VMCore/PassManagerT.h index 4a2ad88671..afff535391 100644 --- a/lib/VMCore/PassManagerT.h +++ b/lib/VMCore/PassManagerT.h @@ -525,7 +525,9 @@ template<> struct PassManagerTraits<BasicBlock> : public BasicBlockPass { // Implement the BasicBlockPass interface... virtual bool doInitialization(Module &M); + virtual bool doInitialization(Function &F); virtual bool runOnBasicBlock(BasicBlock &BB); + virtual bool doFinalization(Function &F); virtual bool doFinalization(Module &M); virtual void getAnalysisUsage(AnalysisUsage &AU) const { @@ -657,10 +659,24 @@ inline bool PassManagerTraits<BasicBlock>::doInitialization(Module &M) { return Changed; } +inline bool PassManagerTraits<BasicBlock>::doInitialization(Function &F) { + bool Changed = false; + for (unsigned i = 0, e = ((PMType*)this)->Passes.size(); i != e; ++i) + ((PMType*)this)->Passes[i]->doInitialization(F); + return Changed; +} + inline bool PassManagerTraits<BasicBlock>::runOnBasicBlock(BasicBlock &BB) { return ((PMType*)this)->runOnUnit(&BB); } +inline bool PassManagerTraits<BasicBlock>::doFinalization(Function &F) { + bool Changed = false; + for (unsigned i = 0, e = ((PMType*)this)->Passes.size(); i != e; ++i) + ((PMType*)this)->Passes[i]->doFinalization(F); + return Changed; +} + inline bool PassManagerTraits<BasicBlock>::doFinalization(Module &M) { bool Changed = false; for (unsigned i = 0, e = ((PMType*)this)->Passes.size(); i != e; ++i) |