Return null on failure, instead of aborting.
[oota-llvm.git] / lib / VMCore / PassManagerT.h
index 70f2cde66ea2707f394ed5438e01b9576ac48db6..83061deac16a2cc306b6153bfef3b06ea68760cd 100644 (file)
-//===- llvm/PassManager.h - Container for Passes -----------------*- C++ -*--=//
+//===- PassManagerT.h - Container for Passes ---------------------*- C++ -*--=//
 //
-// This file defines the PassManager class.  This class is used to hold,
+// This file defines the PassManagerT class.  This class is used to hold,
 // maintain, and optimize execution of Pass's.  The PassManager class ensures
 // that analysis results are available before a pass runs, and that Pass's are
 // destroyed when the PassManager is destroyed.
 //
-// The PassManagerT template is instantiated three times to do its job.
+// The PassManagerT template is instantiated three times to do its job.  The
+// public PassManager class is a Pimpl around the PassManagerT<Module> interface
+// to avoid having all of the PassManager clients being exposed to the
+// implementation details herein.
 //
 //===----------------------------------------------------------------------===//
 
-#ifndef LLVM_PASSMANAGER_H
-#define LLVM_PASSMANAGER_H
+#ifndef LLVM_PASSMANAGER_T_H
+#define LLVM_PASSMANAGER_T_H
 
 #include "llvm/Pass.h"
-#include <string>
+#include "Support/CommandLine.h"
+#include <algorithm>
+#include <iostream>
+class Annotable;
 
-// PassManager - Top level PassManagerT instantiation intended to be used.
-typedef PassManagerT<Module> PassManager;
+//===----------------------------------------------------------------------===//
+// Pass debugging information.  Often it is useful to find out what pass is
+// running when a crash occurs in a utility.  When this library is compiled with
+// debugging on, a command line option (--debug-pass) is enabled that causes the
+// pass name to be printed before it executes.
+//
+
+// Different debug levels that can be enabled...
+enum PassDebugLevel {
+  None, Arguments, Structure, Executions, Details
+};
 
+static cl::opt<enum PassDebugLevel>
+PassDebugging("debug-pass", cl::Hidden,
+              cl::desc("Print PassManager debugging information"),
+              cl::values(
+  clEnumVal(None      , "disable debug output"),
+  clEnumVal(Arguments , "print pass arguments to pass to 'opt'"),
+  clEnumVal(Structure , "print pass structure before run()"),
+  clEnumVal(Executions, "print pass name before it is executed"),
+  clEnumVal(Details   , "print pass details when it is executed"),
+                         0));
 
 //===----------------------------------------------------------------------===//
-// PMDebug class - a set of debugging functions that are enabled when compiling
-// with -g on.  If compiling at -O, all functions are inlined noops.
+// PMDebug class - a set of debugging functions, that are not to be
+// instantiated by the template.
 //
 struct PMDebug {
-#ifdef NDEBUG
-  inline static void PrintPassStructure(Pass *) {}
-  inline static void PrintPassInformation(unsigned,const char*,Pass*,Value*) {}
-  inline static void PrintAnalysisSetInfo(unsigned,const char*,
-                                          const Pass::AnalysisSet &) {}
-#else
-  // If compiled in debug mode, these functions can be enabled by setting
-  // -debug-pass on the command line of the tool being used.
+  static void PerformPassStartupStuff(Pass *P) {
+    // If debugging is enabled, print out argument information...
+    if (PassDebugging >= Arguments) {
+      std::cerr << "Pass Arguments: ";
+      PrintArgumentInformation(P);
+      std::cerr << "\n";
+
+      // Print the pass execution structure
+      if (PassDebugging >= Structure)
+        P->dumpPassStructure();
+    }
+  }
+
+  static void PrintArgumentInformation(const Pass *P);
+  static void PrintPassInformation(unsigned,const char*,Pass *, Annotable *);
+  static void PrintAnalysisSetInfo(unsigned,const char*,Pass *P,
+                                   const std::vector<AnalysisID> &);
+};
+
+
+//===----------------------------------------------------------------------===//
+// TimingInfo Class - This class is used to calculate information about the
+// amount of time each pass takes to execute.  This only happens when
+// -time-passes is enabled on the command line.
+//
+struct TimeRecord {      // TimeRecord - Data we collect and print for each pass
+  double Elapsed;        // Wall clock time elapsed in seconds
+  double UserTime;       // User time elapsed
+  double SystemTime;     // System time elapsed
+  unsigned long MaxRSS;  // Maximum resident set size (in bytes)
+  unsigned long RSSTemp; // Temp for calculating maxrss
+
+  TimeRecord() : Elapsed(0), UserTime(0), SystemTime(0), MaxRSS(0) {}
+  void passStart(const TimeRecord &T);
+  void passEnd(const TimeRecord &T);
+  void sum(const TimeRecord &TR);
+  bool operator<(const TimeRecord &TR) const {
+    return UserTime+SystemTime < TR.UserTime+TR.SystemTime;
+  }
+
+  void print(const char *PassName, const TimeRecord &TotalTime) const;
+};
+
+class TimingInfo {
+  std::map<Pass*, TimeRecord> TimingData;
+  TimingInfo() {}   // Private ctor, must use create member
+public:
+  // Create method.  If Timing is enabled, this creates and returns a new timing
+  // object, otherwise it returns null.
   //
-  static void PrintPassStructure(Pass *P);
-  static void PrintPassInformation(unsigned,const char*,Pass *, Value *);
-  static void PrintAnalysisSetInfo(unsigned,const char*,const Pass::AnalysisSet&);
-#endif
+  static TimingInfo *create();
+
+  // TimingDtor - Print out information about timing information
+  ~TimingInfo();
+
+  void passStarted(Pass *P);
+  void passEnded(Pass *P);
 };
 
 
@@ -54,20 +123,20 @@ template<class UnitType> class PassManagerTraits;   // Do not define.
 //
 template<typename UnitType>
 class PassManagerT : public PassManagerTraits<UnitType>,public AnalysisResolver{
-  typedef typename PassManagerTraits<UnitType>::PassClass       PassClass;
-  typedef typename PassManagerTraits<UnitType>::SubPassClass SubPassClass;
-  typedef typename PassManagerTraits<UnitType>::BatcherClass BatcherClass;
-  typedef typename PassManagerTraits<UnitType>::ParentClass   ParentClass;
-  typedef          PassManagerTraits<UnitType>                     Traits;
+  typedef PassManagerTraits<UnitType> Traits;
+  typedef typename Traits::PassClass       PassClass;
+  typedef typename Traits::SubPassClass SubPassClass;
+  typedef typename Traits::BatcherClass BatcherClass;
+  typedef typename Traits::ParentClass   ParentClass;
 
-  friend typename PassManagerTraits<UnitType>::PassClass;
-  friend typename PassManagerTraits<UnitType>::SubPassClass;  
-  friend class PassManagerTraits<UnitType>;
+  friend typename Traits::PassClass;
+  friend typename Traits::SubPassClass;  
+  friend class Traits;
 
-  std::vector<PassClass*> Passes;    // List of pass's to run
+  std::vector<PassClass*> Passes;    // List of passes to run
 
   // The parent of this pass manager...
-  const ParentClass *Parent;
+  ParentClass * const Parent;
 
   // The current batcher if one is in use, or null
   BatcherClass *Batcher;
@@ -78,12 +147,18 @@ class PassManagerT : public PassManagerTraits<UnitType>,public AnalysisResolver{
   //
   std::map<AnalysisID, Pass*> CurrentAnalyses;
 
+  // LastUseOf - This map keeps track of the last usage in our pipeline of a
+  // particular pass.  When executing passes, the memory for .first is free'd
+  // after .second is run.
+  //
+  std::map<Pass*, Pass*> LastUseOf;
+
 public:
   PassManagerT(ParentClass *Par = 0) : Parent(Par), Batcher(0) {}
   ~PassManagerT() {
     // Delete all of the contained passes...
-    for (std::vector<PassClass*>::iterator I = Passes.begin(), E = Passes.end();
-         I != E; ++I)
+    for (typename std::vector<PassClass*>::iterator
+           I = Passes.begin(), E = Passes.end(); I != E; ++I)
       delete *I;
   }
 
@@ -94,85 +169,110 @@ public:
     closeBatcher();
     CurrentAnalyses.clear();
 
+    // LastUserOf - This contains the inverted LastUseOfMap...
+    std::map<Pass *, std::vector<Pass*> > LastUserOf;
+    for (std::map<Pass*, Pass*>::iterator I = LastUseOf.begin(),
+                                          E = LastUseOf.end(); I != E; ++I)
+      LastUserOf[I->second].push_back(I->first);
+
+
     // Output debug information...
-    if (Parent == 0) PMDebug::PrintPassStructure(this);
+    if (Parent == 0) PMDebug::PerformPassStartupStuff(this);
 
     // Run all of the passes
     for (unsigned i = 0, e = Passes.size(); i < e; ++i) {
       PassClass *P = Passes[i];
       
-      PMDebug::PrintPassInformation(getDepth(), "Executing Pass", P, (Value*)M);
+      PMDebug::PrintPassInformation(getDepth(), "Executing Pass", P,
+                                    (Annotable*)M);
 
       // Get information about what analyses the pass uses...
-      std::vector<AnalysisID> Required, Destroyed, Provided;
-      P->getAnalysisUsageInfo(Required, Destroyed, Provided);
-      
-      PMDebug::PrintAnalysisSetInfo(getDepth(), "Required", Required);
+      AnalysisUsage AnUsage;
+      P->getAnalysisUsage(AnUsage);
+      PMDebug::PrintAnalysisSetInfo(getDepth(), "Required", P,
+                                    AnUsage.getRequiredSet());
 
 #ifndef NDEBUG
       // All Required analyses should be available to the pass as it runs!
-      for (Pass::AnalysisSet::iterator I = Required.begin(), 
-                                       E = Required.end(); I != E; ++I) {
+      for (std::vector<AnalysisID>::const_iterator
+             I = AnUsage.getRequiredSet().begin(), 
+             E = AnUsage.getRequiredSet().end(); I != E; ++I) {
         assert(getAnalysisOrNullUp(*I) && "Analysis used but not available!");
       }
 #endif
 
       // Run the sub pass!
-      MadeChanges |= Traits::runPass(P, M);
-
-      PMDebug::PrintAnalysisSetInfo(getDepth(), "Destroyed", Destroyed);
-      PMDebug::PrintAnalysisSetInfo(getDepth(), "Provided", Provided);
+      startPass(P);
+      bool Changed = runPass(P, M);
+      endPass(P);
+      MadeChanges |= Changed;
+
+      if (Changed)
+        PMDebug::PrintPassInformation(getDepth()+1, "Made Modification", P,
+                                      (Annotable*)M);
+      PMDebug::PrintAnalysisSetInfo(getDepth(), "Preserved", P,
+                                    AnUsage.getPreservedSet());
+
+
+      // Erase all analyses not in the preserved set...
+      if (!AnUsage.preservesAll()) {
+        const std::vector<AnalysisID> &PreservedSet = AnUsage.getPreservedSet();
+        for (std::map<AnalysisID, Pass*>::iterator I = CurrentAnalyses.begin(),
+               E = CurrentAnalyses.end(); I != E; )
+          if (std::find(PreservedSet.begin(), PreservedSet.end(), I->first) !=
+              PreservedSet.end())
+            ++I; // This analysis is preserved, leave it in the available set...
+          else {
+#if MAP_DOESNT_HAVE_BROKEN_ERASE_MEMBER
+            I = CurrentAnalyses.erase(I);   // Analysis not preserved!
+#else
+            // GCC 2.95.3 STL doesn't have correct erase member!
+            CurrentAnalyses.erase(I);
+            I = CurrentAnalyses.begin();
+#endif
+          }
+      }
 
-      // Erase all analyses in the destroyed set...
-      for (Pass::AnalysisSet::iterator I = Destroyed.begin(), 
-             E = Destroyed.end(); I != E; ++I)
-        CurrentAnalyses.erase(*I);
-      
-      // Add all analyses in the provided set...
-      for (Pass::AnalysisSet::iterator I = Provided.begin(),
-             E = Provided.end(); I != E; ++I)
-        CurrentAnalyses[*I] = P;
+      // Add the current pass to the set of passes that have been run, and are
+      // thus available to users.
+      //
+      if (const PassInfo *PI = P->getPassInfo())
+        CurrentAnalyses[PI] = P;
+
+      // Free memory for any passes that we are the last use of...
+      std::vector<Pass*> &DeadPass = LastUserOf[P];
+      for (std::vector<Pass*>::iterator I = DeadPass.begin(),E = DeadPass.end();
+           I != E; ++I) {
+        PMDebug::PrintPassInformation(getDepth()+1, "Freeing Pass", *I,
+                                      (Annotable*)M);
+        (*I)->releaseMemory();
+      }
     }
     return MadeChanges;
   }
 
-  // add - Add a pass to the queue of passes to run.  This passes ownership of
-  // the Pass to the PassManager.  When the PassManager is destroyed, the pass
-  // will be destroyed as well, so there is no need to delete the pass.  Also,
-  // all passes MUST be new'd.
-  //
-  void add(PassClass *P) {
-    // Get information about what analyses the pass uses...
-    std::vector<AnalysisID> Required, Destroyed, Provided;
-    P->getAnalysisUsageInfo(Required, Destroyed, Provided);
-
-    // Loop over all of the analyses used by this pass,
-    for (std::vector<AnalysisID>::iterator I = Required.begin(),
-                                           E = Required.end(); I != E; ++I) {
-      if (getAnalysisOrNullDown(*I) == 0)
-        add((PassClass*)I->createPass());
-    }
-
-    // Tell the pass to add itself to this PassManager... the way it does so
-    // depends on the class of the pass, and is critical to laying out passes in
-    // an optimal order..
-    //
-    P->addToPassManager(this, Destroyed, Provided);
-  }
-
-#ifndef NDEBUG
   // dumpPassStructure - Implement the -debug-passes=PassStructure option
   virtual void dumpPassStructure(unsigned Offset = 0) {
-    std::cerr << std::string(Offset*2, ' ') << "Pass Manager\n";
-    for (std::vector<PassClass*>::iterator I = Passes.begin(), E = Passes.end();
-         I != E; ++I)
-      (*I)->dumpPassStructure(Offset+1);
+    std::cerr << std::string(Offset*2, ' ') << Traits::getPMName()
+              << " Pass Manager\n";
+    for (typename std::vector<PassClass*>::iterator
+           I = Passes.begin(), E = Passes.end(); I != E; ++I) {
+      PassClass *P = *I;
+      P->dumpPassStructure(Offset+1);
+
+      // Loop through and see which classes are destroyed after this one...
+      for (std::map<Pass*, Pass*>::iterator I = LastUseOf.begin(),
+                                            E = LastUseOf.end(); I != E; ++I) {
+        if (P == I->second) {
+          std::cerr << "--" << std::string(Offset*2, ' ');
+          I->first->dumpPassStructure(0);
+        }
+      }
+    }
   }
-#endif
 
-public:
-  Pass *getAnalysisOrNullDown(AnalysisID ID) {
-    std::map<AnalysisID, Pass*>::iterator I = CurrentAnalyses.find(ID);
+  Pass *getAnalysisOrNullDown(AnalysisID ID) const {
+    std::map<AnalysisID, Pass*>::const_iterator I = CurrentAnalyses.find(ID);
     if (I == CurrentAnalyses.end()) {
       if (Batcher)
         return ((AnalysisResolver*)Batcher)->getAnalysisOrNullDown(ID);
@@ -181,19 +281,86 @@ public:
     return I->second;
   }
 
-  Pass *getAnalysisOrNullUp(AnalysisID ID) {
-    std::map<AnalysisID, Pass*>::iterator I = CurrentAnalyses.find(ID);
+  Pass *getAnalysisOrNullUp(AnalysisID ID) const {
+    std::map<AnalysisID, Pass*>::const_iterator I = CurrentAnalyses.find(ID);
     if (I == CurrentAnalyses.end()) {
       if (Parent)
-        return ((AnalysisResolver*)Parent)->getAnalysisOrNullUp(ID);
+        return Parent->getAnalysisOrNullUp(ID);
       return 0;
     }
     return I->second;
   }
 
+  // {start/end}Pass - Called when a pass is started, it just propogates
+  // information up to the top level PassManagerT object to tell it that a pass
+  // has started or ended.  This is used to gather timing information about
+  // passes.
+  //
+  void startPass(Pass *P) {
+    if (Parent) Parent->startPass(P);
+    else PassStarted(P);
+  }
+  void endPass(Pass *P) {
+    if (Parent) Parent->endPass(P);
+    else PassEnded(P);
+  }
+
+  // markPassUsed - Inform higher level pass managers (and ourselves)
+  // that these analyses are being used by this pass.  This is used to
+  // make sure that analyses are not free'd before we have to use
+  // them...
+  //
+  void markPassUsed(AnalysisID P, Pass *User) {
+    std::map<AnalysisID, Pass*>::iterator I = CurrentAnalyses.find(P);
+    if (I != CurrentAnalyses.end()) {
+      LastUseOf[I->second] = User;    // Local pass, extend the lifetime
+    } else {
+      // Pass not in current available set, must be a higher level pass
+      // available to us, propogate to parent pass manager...  We tell the
+      // parent that we (the passmanager) are using the analysis so that it
+      // frees the analysis AFTER this pass manager runs.
+      //
+      assert(Parent != 0 && "Pass available but not found! "
+             "Did your analysis pass 'Provide' itself?");
+      Parent->markPassUsed(P, this);
+    }
+  }
+
+  // Return the number of parent PassManagers that exist
   virtual unsigned getDepth() const {
     if (Parent == 0) return 0;
-    return 1 + ((AnalysisResolver*)Parent)->getDepth();
+    return 1 + Parent->getDepth();
+  }
+
+  virtual unsigned getNumContainedPasses() const { return Passes.size(); }
+  virtual const Pass *getContainedPass(unsigned N) const {
+    assert(N < Passes.size() && "Pass number out of range!");
+    return Passes[N];
+  }
+
+  // add - Add a pass to the queue of passes to run.  This passes ownership of
+  // the Pass to the PassManager.  When the PassManager is destroyed, the pass
+  // will be destroyed as well, so there is no need to delete the pass.  This
+  // implies that all passes MUST be new'd.
+  //
+  void add(PassClass *P) {
+    // Get information about what analyses the pass uses...
+    AnalysisUsage AnUsage;
+    P->getAnalysisUsage(AnUsage);
+    const std::vector<AnalysisID> &Required = AnUsage.getRequiredSet();
+
+    // Loop over all of the analyses used by this pass,
+    for (std::vector<AnalysisID>::const_iterator I = Required.begin(),
+           E = Required.end(); I != E; ++I) {
+      if (getAnalysisOrNullDown(*I) == 0)
+        add((PassClass*)(*I)->createPass());
+    }
+
+    // Tell the pass to add itself to this PassManager... the way it does so
+    // depends on the class of the pass, and is critical to laying out passes in
+    // an optimal order..
+    //
+    P->addToPassManager(this, AnUsage);
   }
 
 private:
@@ -206,15 +373,17 @@ private:
   //
   // For generic Pass subclasses (which are interprocedural passes), we simply
   // add the pass to the end of the pass list and terminate any accumulation of
-  // MethodPasses that are present.
+  // FunctionPass's that are present.
   //
-  void addPass(PassClass *P, Pass::AnalysisSet &Destroyed,
-               Pass::AnalysisSet &Provided) {
-    // Providers are analysis classes which are forbidden to modify the module
-    // they are operating on, so they are allowed to be reordered to before the
-    // batcher...
+  void addPass(PassClass *P, AnalysisUsage &AnUsage) {
+    const std::vector<AnalysisID> &RequiredSet = AnUsage.getRequiredSet();
+
+    // FIXME: If this pass being added isn't killed by any of the passes in the
+    // batcher class then we can reorder to pass to execute before the batcher
+    // does, which will potentially allow us to batch more passes!
     //
-    if (Batcher && Provided.empty())
+    //const std::vector<AnalysisID> &ProvidedSet = AnUsage.getProvidedSet();
+    if (Batcher /*&& ProvidedSet.empty()*/)
       closeBatcher();                     // This pass cannot be batched!
     
     // Set the Resolver instance variable in the Pass so that it knows where to 
@@ -223,27 +392,49 @@ private:
     setAnalysisResolver(P, this);
     Passes.push_back(P);
 
-    // Erase all analyses in the destroyed set...
-    for (std::vector<AnalysisID>::iterator I = Destroyed.begin(), 
-           E = Destroyed.end(); I != E; ++I)
-      CurrentAnalyses.erase(*I);
+    // Inform higher level pass managers (and ourselves) that these analyses are
+    // being used by this pass.  This is used to make sure that analyses are not
+    // free'd before we have to use them...
+    //
+    for (std::vector<AnalysisID>::const_iterator I = RequiredSet.begin(),
+           E = RequiredSet.end(); I != E; ++I)
+      markPassUsed(*I, P);     // Mark *I as used by P
+
+    // Erase all analyses not in the preserved set...
+    if (!AnUsage.preservesAll()) {
+      const std::vector<AnalysisID> &PreservedSet = AnUsage.getPreservedSet();
+      for (std::map<AnalysisID, Pass*>::iterator I = CurrentAnalyses.begin(),
+             E = CurrentAnalyses.end(); I != E; )
+        if (std::find(PreservedSet.begin(), PreservedSet.end(), I->first) !=
+            PreservedSet.end())
+          ++I;  // This analysis is preserved, leave it in the available set...
+        else {
+#if MAP_DOESNT_HAVE_BROKEN_ERASE_MEMBER
+          I = CurrentAnalyses.erase(I);   // Analysis not preserved!
+#else
+          CurrentAnalyses.erase(I);// GCC 2.95.3 STL doesn't have correct erase!
+          I = CurrentAnalyses.begin();
+#endif
+        }
+    }
+
+    // Add this pass to the currently available set...
+    if (const PassInfo *PI = P->getPassInfo())
+      CurrentAnalyses[PI] = P;
 
-    // Add all analyses in the provided set...
-    for (std::vector<AnalysisID>::iterator I = Provided.begin(),
-           E = Provided.end(); I != E; ++I)
-      CurrentAnalyses[*I] = P;
+    // For now assume that our results are never used...
+    LastUseOf[P] = P;
   }
   
-  // For MethodPass subclasses, we must be sure to batch the MethodPasses
-  // together in a MethodPassBatcher object so that all of the analyses are run
-  // together a method at a time.
+  // For FunctionPass subclasses, we must be sure to batch the FunctionPass's
+  // together in a BatcherClass object so that all of the analyses are run
+  // together a function at a time.
   //
-  void addPass(SubPassClass *MP, Pass::AnalysisSet &Destroyed,
-               Pass::AnalysisSet &Provided) {
+  void addPass(SubPassClass *MP, AnalysisUsage &AnUsage) {
     if (Batcher == 0) // If we don't have a batcher yet, make one now.
       Batcher = new BatcherClass(this);
-    // The Batcher will queue them passes up
-    MP->addToPassManager(Batcher, Destroyed, Provided);
+    // The Batcher will queue the passes up
+    MP->addToPassManager(Batcher, AnUsage);
   }
 
   // closeBatcher - Terminate the batcher that is being worked on.
@@ -280,7 +471,7 @@ template<> struct PassManagerTraits<BasicBlock> : public BasicBlockPass {
   typedef PassClass BatcherClass;
 
   // ParentClass - The type of the parent PassManager...
-  typedef PassManagerT<Method> ParentClass;
+  typedef PassManagerT<Function> ParentClass;
 
   // PMType - The type of the passmanager that subclasses this class
   typedef PassManagerT<BasicBlock> PMType;
@@ -288,26 +479,39 @@ template<> struct PassManagerTraits<BasicBlock> : public BasicBlockPass {
   // runPass - Specify how the pass should be run on the UnitType
   static bool runPass(PassClass *P, BasicBlock *M) {
     // todo, init and finalize
-    return P->runOnBasicBlock(M);
+    return P->runOnBasicBlock(*M);
   }
 
+  // Dummy implementation of PassStarted/PassEnded
+  static void PassStarted(Pass *P) {}
+  static void PassEnded(Pass *P) {}
+
+  // getPMName() - Return the name of the unit the PassManager operates on for
+  // debugging.
+  const char *getPMName() const { return "BasicBlock"; }
+  virtual const char *getPassName() const { return "BasicBlock Pass Manager"; }
+
   // Implement the BasicBlockPass interface...
-  virtual bool doInitialization(Module *M);
-  virtual bool runOnBasicBlock(BasicBlock *BB);
-  virtual bool doFinalization(Module *M);
+  virtual bool doInitialization(Module &M);
+  virtual bool runOnBasicBlock(BasicBlock &BB);
+  virtual bool doFinalization(Module &M);
+
+  virtual void getAnalysisUsage(AnalysisUsage &AU) const {
+    AU.setPreservesAll();
+  }
 };
 
 
 
 //===----------------------------------------------------------------------===//
-// PassManagerTraits<Method> Specialization
+// PassManagerTraits<Function> Specialization
 //
-// This pass manager is used to group together all of the MethodPass's
+// This pass manager is used to group together all of the FunctionPass's
 // into a single unit.
 //
-template<> struct PassManagerTraits<Method> : public MethodPass {
+template<> struct PassManagerTraits<Function> : public FunctionPass {
   // PassClass - The type of passes tracked by this PassManager
-  typedef MethodPass PassClass;
+  typedef FunctionPass PassClass;
 
   // SubPassClass - The types of classes that should be collated together
   typedef BasicBlockPass SubPassClass;
@@ -319,17 +523,30 @@ template<> struct PassManagerTraits<Method> : public MethodPass {
   typedef PassManagerT<Module> ParentClass;
 
   // PMType - The type of the passmanager that subclasses this class
-  typedef PassManagerT<Method> PMType;
+  typedef PassManagerT<Function> PMType;
 
   // runPass - Specify how the pass should be run on the UnitType
-  static bool runPass(PassClass *P, Method *M) {
-    return P->runOnMethod(M);
+  static bool runPass(PassClass *P, Function *F) {
+    return P->runOnFunction(*F);
   }
 
-  // Implement the MethodPass interface...
-  virtual bool doInitialization(Module *M);
-  virtual bool runOnMethod(Method *M);
-  virtual bool doFinalization(Module *M);
+  // Dummy implementation of PassStarted/PassEnded
+  static void PassStarted(Pass *P) {}
+  static void PassEnded(Pass *P) {}
+
+  // getPMName() - Return the name of the unit the PassManager operates on for
+  // debugging.
+  const char *getPMName() const { return "Function"; }
+  virtual const char *getPassName() const { return "Function Pass Manager"; }
+
+  // Implement the FunctionPass interface...
+  virtual bool doInitialization(Module &M);
+  virtual bool runOnFunction(Function &F);
+  virtual bool doFinalization(Module &M);
+
+  virtual void getAnalysisUsage(AnalysisUsage &AU) const {
+    AU.setPreservesAll();
+  }
 };
 
 
@@ -344,21 +561,53 @@ template<> struct PassManagerTraits<Module> : public Pass {
   typedef Pass PassClass;
 
   // SubPassClass - The types of classes that should be collated together
-  typedef MethodPass SubPassClass;
+  typedef FunctionPass SubPassClass;
 
   // BatcherClass - The type to use for collation of subtypes...
-  typedef PassManagerT<Method> BatcherClass;
+  typedef PassManagerT<Function> BatcherClass;
 
   // ParentClass - The type of the parent PassManager...
-  typedef void ParentClass;
+  typedef AnalysisResolver ParentClass;
 
   // runPass - Specify how the pass should be run on the UnitType
-  static bool runPass(PassClass *P, Module *M) { return P->run(M); }
+  static bool runPass(PassClass *P, Module *M) { return P->run(*M); }
+
+  // getPMName() - Return the name of the unit the PassManager operates on for
+  // debugging.
+  const char *getPMName() const { return "Module"; }
+  virtual const char *getPassName() const { return "Module Pass Manager"; }
+
+  // TimingInformation - This data member maintains timing information for each
+  // of the passes that is executed.
+  //
+  TimingInfo *TimeInfo;
+
+  // PassStarted/Ended - This callback is notified any time a pass is started
+  // or stops.  This is used to collect timing information about the different
+  // passes being executed.
+  //
+  void PassStarted(Pass *P) {
+    if (TimeInfo) TimeInfo->passStarted(P);
+  }
+  void PassEnded(Pass *P) {
+    if (TimeInfo) TimeInfo->passEnded(P);
+  }
 
-  // run - Implement the Pass interface...
-  virtual bool run(Module *M) {
-    return ((PassManagerT<Module>*)this)->runOnUnit(M);
+  // run - Implement the PassManager interface...
+  bool run(Module &M) {
+    TimeInfo = TimingInfo::create();
+    bool Result = ((PassManagerT<Module>*)this)->runOnUnit(&M);
+    if (TimeInfo) {
+      delete TimeInfo;
+      TimeInfo = 0;
+    }
+    return Result;
   }
+
+  // PassManagerTraits constructor - Create a timing info object if the user
+  // specified timing info should be collected on the command line.
+  //
+  PassManagerTraits() : TimeInfo(0) {}
 };
 
 
@@ -369,18 +618,18 @@ template<> struct PassManagerTraits<Module> : public Pass {
 
 // PassManagerTraits<BasicBlock> Implementations
 //
-inline bool PassManagerTraits<BasicBlock>::doInitialization(Module *M) {
+inline bool PassManagerTraits<BasicBlock>::doInitialization(Module &M) {
   bool Changed = false;
   for (unsigned i = 0, e = ((PMType*)this)->Passes.size(); i != e; ++i)
     ((PMType*)this)->Passes[i]->doInitialization(M);
   return Changed;
 }
 
-inline bool PassManagerTraits<BasicBlock>::runOnBasicBlock(BasicBlock *BB) {
-  return ((PMType*)this)->runOnUnit(BB);
+inline bool PassManagerTraits<BasicBlock>::runOnBasicBlock(BasicBlock &BB) {
+  return ((PMType*)this)->runOnUnit(&BB);
 }
 
-inline bool PassManagerTraits<BasicBlock>::doFinalization(Module *M) {
+inline bool PassManagerTraits<BasicBlock>::doFinalization(Module &M) {
   bool Changed = false;
   for (unsigned i = 0, e = ((PMType*)this)->Passes.size(); i != e; ++i)
     ((PMType*)this)->Passes[i]->doFinalization(M);
@@ -388,20 +637,20 @@ inline bool PassManagerTraits<BasicBlock>::doFinalization(Module *M) {
 }
 
 
-// PassManagerTraits<Method> Implementations
+// PassManagerTraits<Function> Implementations
 //
-inline bool PassManagerTraits<Method>::doInitialization(Module *M) {
+inline bool PassManagerTraits<Function>::doInitialization(Module &M) {
   bool Changed = false;
   for (unsigned i = 0, e = ((PMType*)this)->Passes.size(); i != e; ++i)
     ((PMType*)this)->Passes[i]->doInitialization(M);
   return Changed;
 }
 
-inline bool PassManagerTraits<Method>::runOnMethod(Method *M) {
-  return ((PMType*)this)->runOnUnit(M);
+inline bool PassManagerTraits<Function>::runOnFunction(Function &F) {
+  return ((PMType*)this)->runOnUnit(&F);
 }
 
-inline bool PassManagerTraits<Method>::doFinalization(Module *M) {
+inline bool PassManagerTraits<Function>::doFinalization(Module &M) {
   bool Changed = false;
   for (unsigned i = 0, e = ((PMType*)this)->Passes.size(); i != e; ++i)
     ((PMType*)this)->Passes[i]->doFinalization(M);