Fix thinko.
[oota-llvm.git] / lib / VMCore / PassManager.cpp
1 //===- PassManager.cpp - LLVM Pass Infrastructure Implementation ----------===//
2 //
3 //                     The LLVM Compiler Infrastructure
4 //
5 // This file was developed by Devang Patel and is distributed under
6 // the University of Illinois Open Source License. See LICENSE.TXT for details.
7 //
8 //===----------------------------------------------------------------------===//
9 //
10 // This file implements the LLVM Pass Manager infrastructure. 
11 //
12 //===----------------------------------------------------------------------===//
13
14
15 #include "llvm/PassManager.h"
16 #include "llvm/Module.h"
17 #include "llvm/ModuleProvider.h"
18 #include "llvm/Support/Streams.h"
19 #include <vector>
20 #include <map>
21 using namespace llvm;
22
23 //===----------------------------------------------------------------------===//
24 // Overview:
25 // The Pass Manager Infrastructure manages passes. It's responsibilities are:
26 // 
27 //   o Manage optimization pass execution order
28 //   o Make required Analysis information available before pass P is run
29 //   o Release memory occupied by dead passes
30 //   o If Analysis information is dirtied by a pass then regenerate Analysis 
31 //     information before it is consumed by another pass.
32 //
33 // Pass Manager Infrastructure uses multipe pass managers. They are PassManager,
34 // FunctionPassManager, ModulePassManager, BasicBlockPassManager. This class 
35 // hierarcy uses multiple inheritance but pass managers do not derive from
36 // another pass manager.
37 //
38 // PassManager and FunctionPassManager are two top level pass manager that
39 // represents the external interface of this entire pass manager infrastucture.
40 //
41 // Important classes :
42 //
43 // [o] class PMTopLevelManager;
44 //
45 // Two top level managers, PassManager and FunctionPassManager, derive from 
46 // PMTopLevelManager. PMTopLevelManager manages information used by top level 
47 // managers such as last user info.
48 //
49 // [o] class PMDataManager;
50 //
51 // PMDataManager manages information, e.g. list of available analysis info, 
52 // used by a pass manager to manage execution order of passes. It also provides
53 // a place to implement common pass manager APIs. All pass managers derive from
54 // PMDataManager.
55 //
56 // [o] class BasicBlockPassManager : public FunctionPass, public PMDataManager;
57 //
58 // BasicBlockPassManager manages BasicBlockPasses.
59 //
60 // [o] class FunctionPassManager;
61 //
62 // This is a external interface used by JIT to manage FunctionPasses. This
63 // interface relies on FunctionPassManagerImpl to do all the tasks.
64 //
65 // [o] class FunctionPassManagerImpl : public ModulePass, PMDataManager,
66 //                                     public PMTopLevelManager;
67 //
68 // FunctionPassManagerImpl is a top level manager. It manages FunctionPasses
69 // and BasicBlockPassManagers.
70 //
71 // [o] class ModulePassManager : public Pass, public PMDataManager;
72 //
73 // ModulePassManager manages ModulePasses and FunctionPassManagerImpls.
74 //
75 // [o] class PassManager;
76 //
77 // This is a external interface used by various tools to manages passes. It
78 // relies on PassManagerImpl to do all the tasks.
79 //
80 // [o] class PassManagerImpl : public Pass, public PMDataManager,
81 //                             public PMDTopLevelManager
82 //
83 // PassManagerImpl is a top level pass manager responsible for managing
84 // ModulePassManagers.
85 //===----------------------------------------------------------------------===//
86
87 namespace llvm {
88
89 //===----------------------------------------------------------------------===//
90 // PMTopLevelManager
91 //
92 /// PMTopLevelManager manages LastUser info and collects common APIs used by
93 /// top level pass managers.
94 class PMTopLevelManager {
95
96 public:
97
98   inline std::vector<Pass *>::iterator passManagersBegin() { 
99     return PassManagers.begin(); 
100   }
101
102   inline std::vector<Pass *>::iterator passManagersEnd() { 
103     return PassManagers.end();
104   }
105
106   /// Schedule pass P for execution. Make sure that passes required by
107   /// P are run before P is run. Update analysis info maintained by
108   /// the manager. Remove dead passes. This is a recursive function.
109   void schedulePass(Pass *P, Pass *PM);
110
111   /// This is implemented by top level pass manager and used by 
112   /// schedulePass() to add analysis info passes that are not available.
113   virtual void addTopLevelPass(Pass  *P) = 0;
114
115   /// Set pass P as the last user of the given analysis passes.
116   void setLastUser(std::vector<Pass *> &AnalysisPasses, Pass *P);
117
118   /// Collect passes whose last user is P
119   void collectLastUses(std::vector<Pass *> &LastUses, Pass *P);
120
121   virtual ~PMTopLevelManager() {
122     PassManagers.clear();
123   }
124
125   /// Add immutable pass and initialize it.
126   inline void addImmutablePass(ImmutablePass *P) {
127     P->initializePass();
128     ImmutablePasses.push_back(P);
129   }
130
131   inline std::vector<ImmutablePass *>& getImmutablePasses() {
132     return ImmutablePasses;
133   }
134
135 private:
136   
137   /// Collection of pass managers
138   std::vector<Pass *> PassManagers;
139
140   // Map to keep track of last user of the analysis pass.
141   // LastUser->second is the last user of Lastuser->first.
142   std::map<Pass *, Pass *> LastUser;
143
144   /// Immutable passes are managed by top level manager.
145   std::vector<ImmutablePass *> ImmutablePasses;
146 };
147   
148 /// Set pass P as the last user of the given analysis passes.
149 void PMTopLevelManager::setLastUser(std::vector<Pass *> &AnalysisPasses, 
150                                     Pass *P) {
151
152   for (std::vector<Pass *>::iterator I = AnalysisPasses.begin(),
153          E = AnalysisPasses.end(); I != E; ++I) {
154     Pass *AP = *I;
155     LastUser[AP] = P;
156     // If AP is the last user of other passes then make P last user of
157     // such passes.
158     for (std::map<Pass *, Pass *>::iterator LUI = LastUser.begin(),
159            LUE = LastUser.end(); LUI != LUE; ++LUI) {
160       if (LUI->second == AP)
161         LastUser[LUI->first] = P;
162     }
163   }
164
165 }
166
167 /// Collect passes whose last user is P
168 void PMTopLevelManager::collectLastUses(std::vector<Pass *> &LastUses,
169                                             Pass *P) {
170    for (std::map<Pass *, Pass *>::iterator LUI = LastUser.begin(),
171           LUE = LastUser.end(); LUI != LUE; ++LUI)
172       if (LUI->second == P)
173         LastUses.push_back(LUI->first);
174 }
175
176 /// Schedule pass P for execution. Make sure that passes required by
177 /// P are run before P is run. Update analysis info maintained by
178 /// the manager. Remove dead passes. This is a recursive function.
179 void PMTopLevelManager::schedulePass(Pass *P, Pass *PM) {
180
181   // TODO : Allocate function manager for this pass, other wise required set
182   // may be inserted into previous function manager
183
184   AnalysisUsage AnUsage;
185   P->getAnalysisUsage(AnUsage);
186   const std::vector<AnalysisID> &RequiredSet = AnUsage.getRequiredSet();
187   for (std::vector<AnalysisID>::const_iterator I = RequiredSet.begin(),
188          E = RequiredSet.end(); I != E; ++I) {
189
190     Pass *AnalysisPass = NULL; // FIXME PM->getAnalysisPass(*I, true);
191     if (!AnalysisPass) {
192       // Schedule this analysis run first.
193       AnalysisPass = (*I)->createPass();
194       schedulePass(AnalysisPass, PM);
195     }
196   }
197
198   // Now all required passes are available.
199   addTopLevelPass(P);
200 }
201
202
203 //===----------------------------------------------------------------------===//
204 // PMDataManager
205
206 /// PMDataManager provides the common place to manage the analysis data
207 /// used by pass managers.
208 class PMDataManager {
209
210 public:
211
212   PMDataManager() : TPM(NULL) {
213     initializeAnalysisInfo();
214   }
215
216   /// Return true IFF pass P's required analysis set does not required new
217   /// manager.
218   bool manageablePass(Pass *P);
219
220   Pass *getAnalysisPass(AnalysisID AID) const {
221
222     std::map<AnalysisID, Pass*>::const_iterator I = 
223       AvailableAnalysis.find(AID);
224
225     if (I != AvailableAnalysis.end())
226       return NULL;
227     else
228       return I->second;
229   }
230
231   /// Augment AvailableAnalysis by adding analysis made available by pass P.
232   void recordAvailableAnalysis(Pass *P);
233
234   /// Remove Analysis that is not preserved by the pass
235   void removeNotPreservedAnalysis(Pass *P);
236   
237   /// Remove dead passes
238   void removeDeadPasses(Pass *P);
239
240   /// Add pass P into the PassVector. Update 
241   /// AvailableAnalysis appropriately if ProcessAnalysis is true.
242   void addPassToManager (Pass *P, bool ProcessAnalysis = true);
243
244   // Initialize available analysis information.
245   void initializeAnalysisInfo() { 
246     AvailableAnalysis.clear();
247     LastUser.clear();
248
249     // Include immutable passes into AvailableAnalysis vector.
250     std::vector<ImmutablePass *> &ImmutablePasses =  TPM->getImmutablePasses();
251     for (std::vector<ImmutablePass *>::iterator I = ImmutablePasses.begin(),
252            E = ImmutablePasses.end(); I != E; ++I) 
253       recordAvailableAnalysis(*I);
254   }
255
256   // All Required analyses should be available to the pass as it runs!  Here
257   // we fill in the AnalysisImpls member of the pass so that it can
258   // successfully use the getAnalysis() method to retrieve the
259   // implementations it needs.
260   //
261  void initializeAnalysisImpl(Pass *P);
262
263   inline std::vector<Pass *>::iterator passVectorBegin() { 
264     return PassVector.begin(); 
265   }
266
267   inline std::vector<Pass *>::iterator passVectorEnd() { 
268     return PassVector.end();
269   }
270
271   inline void setLastUser(Pass *P, Pass *LU) {
272     LastUser[P] = LU; 
273     // TODO : Check if pass P is available.
274   }
275
276   // Access toplevel manager
277   PMTopLevelManager *getTopLevelManager() { return TPM; }
278   void setTopLevelManager(PMTopLevelManager *T) { TPM = T; }
279
280 private:
281   // Set of available Analysis. This information is used while scheduling 
282   // pass. If a pass requires an analysis which is not not available then 
283   // equired analysis pass is scheduled to run before the pass itself is 
284   // scheduled to run.
285   std::map<AnalysisID, Pass*> AvailableAnalysis;
286
287   // Map to keep track of last user of the analysis pass.
288   // LastUser->second is the last user of Lastuser->first.
289   std::map<Pass *, Pass *> LastUser;
290
291   // Collection of pass that are managed by this manager
292   std::vector<Pass *> PassVector;
293
294   // Top level manager.
295   // TODO : Make it a reference.
296   PMTopLevelManager *TPM;
297 };
298
299 /// BasicBlockPassManager_New manages BasicBlockPass. It batches all the
300 /// pass together and sequence them to process one basic block before
301 /// processing next basic block.
302 class BasicBlockPassManager_New : public PMDataManager, 
303                                   public FunctionPass {
304
305 public:
306   BasicBlockPassManager_New() { }
307
308   /// Add a pass into a passmanager queue. 
309   bool addPass(Pass *p);
310   
311   /// Execute all of the passes scheduled for execution.  Keep track of
312   /// whether any of the passes modifies the function, and if so, return true.
313   bool runOnFunction(Function &F);
314
315   /// Return true IFF AnalysisID AID is currently available.
316   Pass *getAnalysisPassFromManager(AnalysisID AID);
317
318   /// Pass Manager itself does not invalidate any analysis info.
319   void getAnalysisUsage(AnalysisUsage &Info) const {
320     Info.setPreservesAll();
321   }
322
323 private:
324 };
325
326 /// FunctionPassManagerImpl_New manages FunctionPasses and BasicBlockPassManagers.
327 /// It batches all function passes and basic block pass managers together and
328 /// sequence them to process one function at a time before processing next
329 /// function.
330 class FunctionPassManagerImpl_New : public ModulePass, 
331                                     public PMDataManager,
332                                     public PMTopLevelManager {
333 public:
334   FunctionPassManagerImpl_New(ModuleProvider *P) { /* TODO */ }
335   FunctionPassManagerImpl_New() { 
336     activeBBPassManager = NULL;
337   }
338   ~FunctionPassManagerImpl_New() { /* TODO */ };
339  
340   inline void addTopLevelPass(Pass *P) { 
341     addPass(P);
342   }
343
344   /// add - Add a pass to the queue of passes to run.  This passes
345   /// ownership of the Pass to the PassManager.  When the
346   /// PassManager_X is destroyed, the pass will be destroyed as well, so
347   /// there is no need to delete the pass. (TODO delete passes.)
348   /// This implies that all passes MUST be allocated with 'new'.
349   void add(Pass *P) { 
350     schedulePass(P, this);
351   }
352
353   /// Add pass into the pass manager queue.
354   bool addPass(Pass *P);
355
356   /// Execute all of the passes scheduled for execution.  Keep
357   /// track of whether any of the passes modifies the function, and if
358   /// so, return true.
359   bool runOnModule(Module &M);
360   bool runOnFunction(Function &F);
361
362   /// Return true IFF AnalysisID AID is currently available.
363   Pass *getAnalysisPassFromManager(AnalysisID AID);
364
365   /// doInitialization - Run all of the initializers for the function passes.
366   ///
367   bool doInitialization(Module &M);
368   
369   /// doFinalization - Run all of the initializers for the function passes.
370   ///
371   bool doFinalization(Module &M);
372
373   /// Pass Manager itself does not invalidate any analysis info.
374   void getAnalysisUsage(AnalysisUsage &Info) const {
375     Info.setPreservesAll();
376   }
377
378 private:
379   // Active Pass Managers
380   BasicBlockPassManager_New *activeBBPassManager;
381 };
382
383 /// ModulePassManager_New manages ModulePasses and function pass managers.
384 /// It batches all Module passes  passes and function pass managers together and
385 /// sequence them to process one module.
386 class ModulePassManager_New : public PMDataManager {
387  
388 public:
389   ModulePassManager_New() { activeFunctionPassManager = NULL; }
390   
391   /// Add a pass into a passmanager queue. 
392   bool addPass(Pass *p);
393   
394   /// run - Execute all of the passes scheduled for execution.  Keep track of
395   /// whether any of the passes modifies the module, and if so, return true.
396   bool runOnModule(Module &M);
397
398   /// Return true IFF AnalysisID AID is currently available.
399   Pass *getAnalysisPassFromManager(AnalysisID AID);
400
401   /// Pass Manager itself does not invalidate any analysis info.
402   void getAnalysisUsage(AnalysisUsage &Info) const {
403     Info.setPreservesAll();
404   }
405
406 private:
407   // Active Pass Manager
408   FunctionPassManagerImpl_New *activeFunctionPassManager;
409 };
410
411 /// PassManager_New manages ModulePassManagers
412 class PassManagerImpl_New : public Pass,
413                             public PMDataManager,
414                             public PMTopLevelManager {
415
416 public:
417
418   /// add - Add a pass to the queue of passes to run.  This passes ownership of
419   /// the Pass to the PassManager.  When the PassManager is destroyed, the pass
420   /// will be destroyed as well, so there is no need to delete the pass.  This
421   /// implies that all passes MUST be allocated with 'new'.
422   void add(Pass *P) {
423     schedulePass(P, this);
424   }
425  
426   /// run - Execute all of the passes scheduled for execution.  Keep track of
427   /// whether any of the passes modifies the module, and if so, return true.
428   bool run(Module &M);
429
430   /// Return true IFF AnalysisID AID is currently available.
431   Pass *getAnalysisPassFromManager(AnalysisID AID);
432
433   /// Pass Manager itself does not invalidate any analysis info.
434   void getAnalysisUsage(AnalysisUsage &Info) const {
435     Info.setPreservesAll();
436   }
437
438   inline void addTopLevelPass(Pass *P) {
439     addPass(P);
440   }
441
442 private:
443
444   /// Add a pass into a passmanager queue.
445   bool addPass(Pass *p);
446
447   // Collection of pass managers
448   std::vector<ModulePassManager_New *> PassManagers;
449
450   // Active Pass Manager
451   ModulePassManager_New *activeManager;
452 };
453
454 } // End of llvm namespace
455
456 //===----------------------------------------------------------------------===//
457 // PMDataManager implementation
458
459 /// Return true IFF pass P's required analysis set does not required new
460 /// manager.
461 bool PMDataManager::manageablePass(Pass *P) {
462
463   // TODO 
464   // If this pass is not preserving information that is required by a
465   // pass maintained by higher level pass manager then do not insert
466   // this pass into current manager. Use new manager. For example,
467   // For example, If FunctionPass F is not preserving ModulePass Info M1
468   // that is used by another ModulePass M2 then do not insert F in
469   // current function pass manager.
470   return true;
471 }
472
473 /// Augement AvailableAnalysis by adding analysis made available by pass P.
474 void PMDataManager::recordAvailableAnalysis(Pass *P) {
475                                                 
476   if (const PassInfo *PI = P->getPassInfo()) {
477     AvailableAnalysis[PI] = P;
478
479     //This pass is the current implementation of all of the interfaces it
480     //implements as well.
481     const std::vector<const PassInfo*> &II = PI->getInterfacesImplemented();
482     for (unsigned i = 0, e = II.size(); i != e; ++i)
483       AvailableAnalysis[II[i]] = P;
484   }
485 }
486
487 /// Remove Analyss not preserved by Pass P
488 void PMDataManager::removeNotPreservedAnalysis(Pass *P) {
489   AnalysisUsage AnUsage;
490   P->getAnalysisUsage(AnUsage);
491
492   if (AnUsage.getPreservesAll())
493     return;
494
495   const std::vector<AnalysisID> &PreservedSet = AnUsage.getPreservedSet();
496   for (std::map<AnalysisID, Pass*>::iterator I = AvailableAnalysis.begin(),
497          E = AvailableAnalysis.end(); I != E; ++I ) {
498     if (std::find(PreservedSet.begin(), PreservedSet.end(), I->first) == 
499         PreservedSet.end()) {
500       // Remove this analysis
501       std::map<AnalysisID, Pass*>::iterator J = I++;
502       AvailableAnalysis.erase(J);
503     }
504   }
505 }
506
507 /// Remove analysis passes that are not used any longer
508 void PMDataManager::removeDeadPasses(Pass *P) {
509
510   for (std::map<Pass *, Pass *>::iterator I = LastUser.begin(),
511          E = LastUser.end(); I !=E; ++I) {
512     if (I->second == P) {
513       Pass *deadPass = I->first;
514       deadPass->releaseMemory();
515
516       std::map<AnalysisID, Pass*>::iterator Pos = 
517         AvailableAnalysis.find(deadPass->getPassInfo());
518       
519       assert (Pos != AvailableAnalysis.end() &&
520               "Pass is not available");
521       AvailableAnalysis.erase(Pos);
522     }
523   }
524 }
525
526 /// Add pass P into the PassVector. Update 
527 /// AvailableAnalysis appropriately if ProcessAnalysis is true.
528 void PMDataManager::addPassToManager(Pass *P, 
529                                      bool ProcessAnalysis) {
530
531   if (ProcessAnalysis) {
532     // Take a note of analysis required and made available by this pass
533     initializeAnalysisImpl(P);
534     recordAvailableAnalysis(P);
535
536     // Remove the analysis not preserved by this pass
537     removeNotPreservedAnalysis(P);
538   }
539
540   // Add pass
541   PassVector.push_back(P);
542 }
543
544 // All Required analyses should be available to the pass as it runs!  Here
545 // we fill in the AnalysisImpls member of the pass so that it can
546 // successfully use the getAnalysis() method to retrieve the
547 // implementations it needs.
548 //
549 void PMDataManager::initializeAnalysisImpl(Pass *P) {
550   AnalysisUsage AnUsage;
551   P->getAnalysisUsage(AnUsage);
552  
553   for (std::vector<const PassInfo *>::const_iterator
554          I = AnUsage.getRequiredSet().begin(),
555          E = AnUsage.getRequiredSet().end(); I != E; ++I) {
556     Pass *Impl = getAnalysisPass(*I);
557     if (Impl == 0)
558       assert(0 && "Analysis used but not available!");
559     // TODO:  P->AnalysisImpls.push_back(std::make_pair(*I, Impl));
560   }
561 }
562
563 //===----------------------------------------------------------------------===//
564 // BasicBlockPassManager_New implementation
565
566 /// Add pass P into PassVector and return true. If this pass is not
567 /// manageable by this manager then return false.
568 bool
569 BasicBlockPassManager_New::addPass(Pass *P) {
570
571   BasicBlockPass *BP = dynamic_cast<BasicBlockPass*>(P);
572   if (!BP)
573     return false;
574
575   // If this pass does not preserve anlysis that is used by other passes
576   // managed by this manager than it is not a suiable pass for this manager.
577   if (!manageablePass(P))
578     return false;
579
580   addPassToManager (BP);
581
582   return true;
583 }
584
585 /// Execute all of the passes scheduled for execution by invoking 
586 /// runOnBasicBlock method.  Keep track of whether any of the passes modifies 
587 /// the function, and if so, return true.
588 bool
589 BasicBlockPassManager_New::runOnFunction(Function &F) {
590
591   bool Changed = false;
592   initializeAnalysisInfo();
593
594   for (Function::iterator I = F.begin(), E = F.end(); I != E; ++I)
595     for (std::vector<Pass *>::iterator itr = passVectorBegin(),
596            e = passVectorEnd(); itr != e; ++itr) {
597       Pass *P = *itr;
598       
599       recordAvailableAnalysis(P);
600       BasicBlockPass *BP = dynamic_cast<BasicBlockPass*>(P);
601       Changed |= BP->runOnBasicBlock(*I);
602       removeNotPreservedAnalysis(P);
603       removeDeadPasses(P);
604     }
605   return Changed;
606 }
607
608 /// Return true IFF AnalysisID AID is currently available.
609 Pass * BasicBlockPassManager_New::getAnalysisPassFromManager(AnalysisID AID) {
610   return getAnalysisPass(AID);
611 }
612
613 //===----------------------------------------------------------------------===//
614 // FunctionPassManager_New implementation
615
616 /// Create new Function pass manager
617 FunctionPassManager_New::FunctionPassManager_New() {
618   FPM = new FunctionPassManagerImpl_New();
619 }
620
621 /// add - Add a pass to the queue of passes to run.  This passes
622 /// ownership of the Pass to the PassManager.  When the
623 /// PassManager_X is destroyed, the pass will be destroyed as well, so
624 /// there is no need to delete the pass. (TODO delete passes.)
625 /// This implies that all passes MUST be allocated with 'new'.
626 void FunctionPassManager_New::add(Pass *P) { 
627   FPM->add(P);
628 }
629
630 /// Execute all of the passes scheduled for execution.  Keep
631 /// track of whether any of the passes modifies the function, and if
632 /// so, return true.
633 bool FunctionPassManager_New::runOnModule(Module &M) {
634   return FPM->runOnModule(M);
635 }
636
637 /// run - Execute all of the passes scheduled for execution.  Keep
638 /// track of whether any of the passes modifies the function, and if
639 /// so, return true.
640 ///
641 bool FunctionPassManager_New::run(Function &F) {
642   std::string errstr;
643   if (MP->materializeFunction(&F, &errstr)) {
644     cerr << "Error reading bytecode file: " << errstr << "\n";
645     abort();
646   }
647   return FPM->runOnFunction(F);
648 }
649
650
651 /// doInitialization - Run all of the initializers for the function passes.
652 ///
653 bool FunctionPassManager_New::doInitialization() {
654   return FPM->doInitialization(*MP->getModule());
655 }
656
657 /// doFinalization - Run all of the initializers for the function passes.
658 ///
659 bool FunctionPassManager_New::doFinalization() {
660   return FPM->doFinalization(*MP->getModule());
661 }
662
663 //===----------------------------------------------------------------------===//
664 // FunctionPassManagerImpl_New implementation
665
666 /// Add pass P into the pass manager queue. If P is a BasicBlockPass then
667 /// either use it into active basic block pass manager or create new basic
668 /// block pass manager to handle pass P.
669 bool
670 FunctionPassManagerImpl_New::addPass(Pass *P) {
671
672   // If P is a BasicBlockPass then use BasicBlockPassManager_New.
673   if (BasicBlockPass *BP = dynamic_cast<BasicBlockPass*>(P)) {
674
675     if (!activeBBPassManager
676         || !activeBBPassManager->addPass(BP)) {
677
678       activeBBPassManager = new BasicBlockPassManager_New();
679       addPassToManager(activeBBPassManager, false);
680       if (!activeBBPassManager->addPass(BP))
681         assert(0 && "Unable to add Pass");
682     }
683     return true;
684   }
685
686   FunctionPass *FP = dynamic_cast<FunctionPass *>(P);
687   if (!FP)
688     return false;
689
690   // If this pass does not preserve anlysis that is used by other passes
691   // managed by this manager than it is not a suiable pass for this manager.
692   if (!manageablePass(P))
693     return false;
694
695   addPassToManager (FP);
696   activeBBPassManager = NULL;
697   return true;
698 }
699
700 /// Execute all of the passes scheduled for execution by invoking 
701 /// runOnFunction method.  Keep track of whether any of the passes modifies 
702 /// the function, and if so, return true.
703 bool FunctionPassManagerImpl_New::runOnModule(Module &M) {
704
705   bool Changed = false;
706   initializeAnalysisInfo();
707
708   for (Module::iterator I = M.begin(), E = M.end(); I != E; ++I)
709     for (std::vector<Pass *>::iterator itr = passVectorBegin(),
710            e = passVectorEnd(); itr != e; ++itr) {
711       Pass *P = *itr;
712       
713       recordAvailableAnalysis(P);
714       FunctionPass *FP = dynamic_cast<FunctionPass*>(P);
715       Changed |= FP->runOnFunction(*I);
716       removeNotPreservedAnalysis(P);
717       removeDeadPasses(P);
718     }
719   return Changed;
720 }
721
722 /// Execute all of the passes scheduled for execution by invoking 
723 /// runOnFunction method.  Keep track of whether any of the passes modifies 
724 /// the function, and if so, return true.
725 bool FunctionPassManagerImpl_New::runOnFunction(Function &F) {
726
727   bool Changed = false;
728   initializeAnalysisInfo();
729
730   for (std::vector<Pass *>::iterator itr = passVectorBegin(),
731          e = passVectorEnd(); itr != e; ++itr) {
732     Pass *P = *itr;
733     
734     recordAvailableAnalysis(P);
735     FunctionPass *FP = dynamic_cast<FunctionPass*>(P);
736     Changed |= FP->runOnFunction(F);
737     removeNotPreservedAnalysis(P);
738     removeDeadPasses(P);
739   }
740   return Changed;
741 }
742
743
744 /// Return true IFF AnalysisID AID is currently available.
745 Pass *FunctionPassManagerImpl_New::getAnalysisPassFromManager(AnalysisID AID) {
746
747   Pass *P = getAnalysisPass(AID);
748   if (P)
749     return P;
750
751   if (activeBBPassManager && 
752       activeBBPassManager->getAnalysisPass(AID) != 0)
753     return activeBBPassManager->getAnalysisPass(AID);
754
755   // TODO : Check inactive managers
756   return NULL;
757 }
758
759 inline bool FunctionPassManagerImpl_New::doInitialization(Module &M) {
760   bool Changed = false;
761
762   for (std::vector<Pass *>::iterator itr = passVectorBegin(),
763          e = passVectorEnd(); itr != e; ++itr) {
764     Pass *P = *itr;
765     
766     FunctionPass *FP = dynamic_cast<FunctionPass*>(P);
767     Changed |= FP->doInitialization(M);
768   }
769
770   return Changed;
771 }
772
773 inline bool FunctionPassManagerImpl_New::doFinalization(Module &M) {
774   bool Changed = false;
775
776   for (std::vector<Pass *>::iterator itr = passVectorBegin(),
777          e = passVectorEnd(); itr != e; ++itr) {
778     Pass *P = *itr;
779     
780     FunctionPass *FP = dynamic_cast<FunctionPass*>(P);
781     Changed |= FP->doFinalization(M);
782   }
783
784
785   return Changed;
786 }
787
788 //===----------------------------------------------------------------------===//
789 // ModulePassManager implementation
790
791 /// Add P into pass vector if it is manageble. If P is a FunctionPass
792 /// then use FunctionPassManagerImpl_New to manage it. Return false if P
793 /// is not manageable by this manager.
794 bool
795 ModulePassManager_New::addPass(Pass *P) {
796
797   // If P is FunctionPass then use function pass maanager.
798   if (FunctionPass *FP = dynamic_cast<FunctionPass*>(P)) {
799
800     activeFunctionPassManager = NULL;
801
802     if (!activeFunctionPassManager
803         || !activeFunctionPassManager->addPass(P)) {
804
805       activeFunctionPassManager = new FunctionPassManagerImpl_New();
806       addPassToManager(activeFunctionPassManager, false);
807       if (!activeFunctionPassManager->addPass(FP))
808         assert(0 && "Unable to add pass");
809     }
810     return true;
811   }
812
813   ModulePass *MP = dynamic_cast<ModulePass *>(P);
814   if (!MP)
815     return false;
816
817   // If this pass does not preserve anlysis that is used by other passes
818   // managed by this manager than it is not a suiable pass for this manager.
819   if (!manageablePass(P))
820     return false;
821
822   addPassToManager(MP);
823   activeFunctionPassManager = NULL;
824   return true;
825 }
826
827
828 /// Execute all of the passes scheduled for execution by invoking 
829 /// runOnModule method.  Keep track of whether any of the passes modifies 
830 /// the module, and if so, return true.
831 bool
832 ModulePassManager_New::runOnModule(Module &M) {
833   bool Changed = false;
834   initializeAnalysisInfo();
835
836   for (std::vector<Pass *>::iterator itr = passVectorBegin(),
837          e = passVectorEnd(); itr != e; ++itr) {
838     Pass *P = *itr;
839
840     recordAvailableAnalysis(P);
841     ModulePass *MP = dynamic_cast<ModulePass*>(P);
842     Changed |= MP->runOnModule(M);
843     removeNotPreservedAnalysis(P);
844     removeDeadPasses(P);
845   }
846   return Changed;
847 }
848
849 /// Return true IFF AnalysisID AID is currently available.
850 Pass *ModulePassManager_New::getAnalysisPassFromManager(AnalysisID AID) {
851
852   
853   Pass *P = getAnalysisPass(AID);
854   if (P)
855     return P;
856
857   if (activeFunctionPassManager && 
858       activeFunctionPassManager->getAnalysisPass(AID) != 0)
859     return activeFunctionPassManager->getAnalysisPass(AID);
860
861   // TODO : Check inactive managers
862   return NULL;
863 }
864
865 //===----------------------------------------------------------------------===//
866 // PassManagerImpl implementation
867
868 /// Return true IFF AnalysisID AID is currently available.
869 Pass *PassManagerImpl_New::getAnalysisPassFromManager(AnalysisID AID) {
870
871   Pass *P = NULL;
872   for (std::vector<ModulePassManager_New *>::iterator itr = PassManagers.begin(),
873          e = PassManagers.end(); !P && itr != e; ++itr)
874     P  = (*itr)->getAnalysisPassFromManager(AID);
875   return P;
876 }
877
878 // PassManager_New implementation
879 /// Add P into active pass manager or use new module pass manager to
880 /// manage it.
881 bool PassManagerImpl_New::addPass(Pass *P) {
882
883   if (!activeManager || !activeManager->addPass(P)) {
884     activeManager = new ModulePassManager_New();
885     PassManagers.push_back(activeManager);
886     return activeManager->addPass(P);
887   }
888   return true;
889 }
890
891 /// run - Execute all of the passes scheduled for execution.  Keep track of
892 /// whether any of the passes modifies the module, and if so, return true.
893 bool PassManagerImpl_New::run(Module &M) {
894
895   bool Changed = false;
896   for (std::vector<ModulePassManager_New *>::iterator itr = PassManagers.begin(),
897          e = PassManagers.end(); itr != e; ++itr) {
898     ModulePassManager_New *pm = *itr;
899     Changed |= pm->runOnModule(M);
900   }
901   return Changed;
902 }
903
904 //===----------------------------------------------------------------------===//
905 // PassManager implementation
906
907 /// Create new pass manager
908 PassManager_New::PassManager_New() {
909   PM = new PassManagerImpl_New();
910 }
911
912 /// add - Add a pass to the queue of passes to run.  This passes ownership of
913 /// the Pass to the PassManager.  When the PassManager is destroyed, the pass
914 /// will be destroyed as well, so there is no need to delete the pass.  This
915 /// implies that all passes MUST be allocated with 'new'.
916 void 
917 PassManager_New::add(Pass *P) {
918   PM->add(P);
919 }
920
921 /// run - Execute all of the passes scheduled for execution.  Keep track of
922 /// whether any of the passes modifies the module, and if so, return true.
923 bool
924 PassManager_New::run(Module &M) {
925   return PM->run(M);
926 }
927