include/llvm/Pass.h

   1 //===- llvm/Pass.h - Base class for XForm Passes -----------------*- C++ -*--=//
   2 //
   3 // This file defines a base class that indicates that a specified class is a
   4 // transformation pass implementation.
   5 //
   6 // Pass's are designed this way so that it is possible to run passes in a cache
   7 // and organizationally optimal order without having to specify it at the front
   8 // end.  This allows arbitrary passes to be strung together and have them
   9 // executed as effeciently as possible.
  10 //
  11 // Passes should extend one of the classes below, depending on the guarantees
  12 // that it can make about what will be modified as it is run.  For example, most
  13 // global optimizations should derive from FunctionPass, because they do not add
  14 // or delete functions, they operate on the internals of the function.
  15 //
  16 // Note that this file #includes PassSupport.h and PassAnalysisSupport.h (at the
  17 // bottom), so the APIs exposed by these files are also automatically available
  18 // to all users of this file.
  19 //
  20 //===----------------------------------------------------------------------===//
  21
  22 #ifndef LLVM_PASS_H
  23 #define LLVM_PASS_H
  24
  25 #include <vector>
  26 #include <map>
  27 #include <iosfwd>
  28 #include <typeinfo>
  29 class Value;
  30 class BasicBlock;
  31 class Function;
  32 class Module;
  33 class AnalysisUsage;
  34 class PassInfo;
  35 class ImmutablePass;
  36 template<class UnitType> class PassManagerT;
  37 struct AnalysisResolver;
  38
  39 // AnalysisID - Use the PassInfo to identify a pass...
  40 typedef const PassInfo* AnalysisID;
  41
  42 //===----------------------------------------------------------------------===//
  43 /// Pass interface - Implemented by all 'passes'.  Subclass this if you are an
  44 /// interprocedural optimization or you do not fit into any of the more
  45 /// constrained passes described below.
  46 ///
  47 class Pass {
  48   friend class AnalysisResolver;
  49   AnalysisResolver *Resolver;  // AnalysisResolver this pass is owned by...
  50   const PassInfo *PassInfoCache;
  51
  52   // AnalysisImpls - This keeps track of which passes implement the interfaces
  53   // that are required by the current pass (to implement getAnalysis()).
  54   //
  55   std::vector<std::pair<const PassInfo*, Pass*> > AnalysisImpls;
  56
  57   void operator=(const Pass&);  // DO NOT IMPLEMENT
  58   Pass(const Pass &);           // DO NOT IMPLEMENT
  59 public:
  60   Pass() : Resolver(0), PassInfoCache(0) {}
  61   virtual ~Pass() {} // Destructor is virtual so we can be subclassed
  62
  63   /// getPassName - Return a nice clean name for a pass.  This usually
  64   /// implemented in terms of the name that is registered by one of the
  65   /// Registration templates, but can be overloaded directly, and if nothing
  66   /// else is available, C++ RTTI will be consulted to get a SOMEWHAT
  67   /// intelligable name for the pass.
  68   ///
  69   virtual const char *getPassName() const;
  70
  71   /// getPassInfo - Return the PassInfo data structure that corresponds to this
  72   /// pass...  If the pass has not been registered, this will return null.
  73   ///
  74   const PassInfo *getPassInfo() const;
  75
  76   /// run - Run this pass, returning true if a modification was made to the
  77   /// module argument.  This should be implemented by all concrete subclasses.
  78   ///
  79   virtual bool run(Module &M) = 0;
  80
  81   /// print - Print out the internal state of the pass.  This is called by
  82   /// Analyze to print out the contents of an analysis.  Otherwise it is not
  83   /// neccesary to implement this method.  Beware that the module pointer MAY be
  84   /// null.  This automatically forwards to a virtual function that does not
  85   /// provide the Module* in case the analysis doesn't need it it can just be
  86   /// ignored.
  87   ///
  88   virtual void print(std::ostream &O, const Module *M) const { print(O); }
  89   virtual void print(std::ostream &O) const;
  90   void dump() const; // dump - call print(std::cerr, 0);
  91
  92
  93   /// getAnalysisUsage - This function should be overriden by passes that need
  94   /// analysis information to do their job.  If a pass specifies that it uses a
  95   /// particular analysis result to this function, it can then use the
  96   /// getAnalysis<AnalysisType>() function, below.
  97   ///
  98   virtual void getAnalysisUsage(AnalysisUsage &Info) const {
  99     // By default, no analysis results are used, all are invalidated.
 100   }
 101
 102   /// releaseMemory() - This member can be implemented by a pass if it wants to
 103   /// be able to release its memory when it is no longer needed.  The default
 104   /// behavior of passes is to hold onto memory for the entire duration of their
 105   /// lifetime (which is the entire compile time).  For pipelined passes, this
 106   /// is not a big deal because that memory gets recycled every time the pass is
 107   /// invoked on another program unit.  For IP passes, it is more important to
 108   /// free memory when it is unused.
 109   ///
 110   /// Optionally implement this function to release pass memory when it is no
 111   /// longer used.
 112   ///
 113   virtual void releaseMemory() {}
 114
 115   // dumpPassStructure - Implement the -debug-passes=PassStructure option
 116   virtual void dumpPassStructure(unsigned Offset = 0);
 117
 118
 119   // getPassInfo - Static method to get the pass information from a class name.
 120   template<typename AnalysisClass>
 121   static const PassInfo *getClassPassInfo() {
 122     return lookupPassInfo(typeid(AnalysisClass));
 123   }
 124
 125   // lookupPassInfo - Return the pass info object for the specified pass class,
 126   // or null if it is not known.
 127   static const PassInfo *lookupPassInfo(const std::type_info &TI);
 128
 129   /// getAnalysisToUpdate<AnalysisType>() - This function is used by subclasses
 130   /// to get to the analysis information that might be around that needs to be
 131   /// updated.  This is different than getAnalysis in that it can fail (ie the
 132   /// analysis results haven't been computed), so should only be used if you
 133   /// provide the capability to update an analysis that exists.  This method is
 134   /// often used by transformation APIs to update analysis results for a pass
 135   /// automatically as the transform is performed.
 136   ///
 137   template<typename AnalysisType>
 138   AnalysisType *getAnalysisToUpdate() const {
 139     assert(Resolver && "Pass not resident in a PassManager object!");
 140     const PassInfo *PI = getClassPassInfo<AnalysisType>();
 141     if (PI == 0) return 0;
 142     return dynamic_cast<AnalysisType*>(Resolver->getAnalysisToUpdate(PI));
 143   }
 144
 145 protected:
 146
 147   /// getAnalysis<AnalysisType>() - This function is used by subclasses to get
 148   /// to the analysis information that they claim to use by overriding the
 149   /// getAnalysisUsage function.
 150   ///
 151   template<typename AnalysisType>
 152   AnalysisType &getAnalysis() const {
 153     assert(Resolver && "Pass has not been inserted into a PassManager object!");
 154     const PassInfo *PI = getClassPassInfo<AnalysisType>();
 155     return getAnalysisID<AnalysisType>(PI);
 156   }
 157
 158   template<typename AnalysisType>
 159   AnalysisType &getAnalysisID(const PassInfo *PI) const {
 160     assert(Resolver && "Pass has not been inserted into a PassManager object!");
 161     assert(PI && "getAnalysis for unregistered pass!");
 162
 163     // PI *must* appear in AnalysisImpls.  Because the number of passes used
 164     // should be a small number, we just do a linear search over a (dense)
 165     // vector.
 166     Pass *ResultPass = 0;
 167     for (unsigned i = 0; ; ++i) {
 168       assert(i != AnalysisImpls.size() &&
 169              "getAnalysis*() called on an analysis that we not "
 170              "'required' by pass!");
 171       if (AnalysisImpls[i].first == PI) {
 172         ResultPass = AnalysisImpls[i].second;
 173         break;
 174       }
 175     }
 176
 177     // Because the AnalysisType may not be a subclass of pass (for
 178     // AnalysisGroups), we must use dynamic_cast here to potentially adjust the
 179     // return pointer (because the class may multiply inherit, once from pass,
 180     // once from AnalysisType).
 181     //
 182     AnalysisType *Result = dynamic_cast<AnalysisType*>(ResultPass);
 183     assert(Result && "Pass does not implement interface required!");
 184     return *Result;
 185   }
 186
 187 private:
 188   friend class PassManagerT<Module>;
 189   friend class PassManagerT<Function>;
 190   friend class PassManagerT<BasicBlock>;
 191   virtual void addToPassManager(PassManagerT<Module> *PM, AnalysisUsage &AU);
 192 };
 193
 194 inline std::ostream &operator<<(std::ostream &OS, const Pass &P) {
 195   P.print(OS, 0); return OS;
 196 }
 197
 198
 199
 200 //===----------------------------------------------------------------------===//
 201 /// ImmutablePass class - This class is used to provide information that does
 202 /// not need to be run.  This is useful for things like target information and
 203 /// "basic" versions of AnalysisGroups.
 204 ///
 205 struct ImmutablePass : public Pass {
 206
 207   // ImmutablePasses are never run.
 208   virtual bool run(Module &M) { return false; }
 209
 210 private:
 211   friend class PassManagerT<Module>;
 212   virtual void addToPassManager(PassManagerT<Module> *PM, AnalysisUsage &AU);
 213 };
 214
 215
 216 //===----------------------------------------------------------------------===//
 217 /// FunctionPass class - This class is used to implement most global
 218 /// optimizations.  Optimizations should subclass this class if they meet the
 219 /// following constraints:
 220 ///
 221 ///  1. Optimizations are organized globally, ie a function at a time
 222 ///  2. Optimizing a function does not cause the addition or removal of any
 223 ///     functions in the module
 224 ///
 225 struct FunctionPass : public Pass {
 226   /// doInitialization - Virtual method overridden by subclasses to do
 227   /// any neccesary per-module initialization.
 228   ///
 229   virtual bool doInitialization(Module &M) { return false; }
 230
 231   /// runOnFunction - Virtual method overriden by subclasses to do the
 232   /// per-function processing of the pass.
 233   ///
 234   virtual bool runOnFunction(Function &F) = 0;
 235
 236   /// doFinalization - Virtual method overriden by subclasses to do any post
 237   /// processing needed after all passes have run.
 238   ///
 239   virtual bool doFinalization(Module &M) { return false; }
 240
 241   /// run - On a module, we run this pass by initializing, ronOnFunction'ing
 242   /// once for every function in the module, then by finalizing.
 243   ///
 244   virtual bool run(Module &M);
 245
 246   /// run - On a function, we simply initialize, run the function, then
 247   /// finalize.
 248   ///
 249   bool run(Function &F);
 250
 251 private:
 252   friend class PassManagerT<Module>;
 253   friend class PassManagerT<Function>;
 254   friend class PassManagerT<BasicBlock>;
 255   virtual void addToPassManager(PassManagerT<Module> *PM, AnalysisUsage &AU);
 256   virtual void addToPassManager(PassManagerT<Function> *PM, AnalysisUsage &AU);
 257 };
 258
 259
 260
 261 //===----------------------------------------------------------------------===//
 262 /// BasicBlockPass class - This class is used to implement most local
 263 /// optimizations.  Optimizations should subclass this class if they
 264 /// meet the following constraints:
 265 ///   1. Optimizations are local, operating on either a basic block or
 266 ///      instruction at a time.
 267 ///   2. Optimizations do not modify the CFG of the contained function, or any
 268 ///      other basic block in the function.
 269 ///   3. Optimizations conform to all of the constraints of FunctionPass's.
 270 ///
 271 struct BasicBlockPass : public FunctionPass {
 272   /// doInitialization - Virtual method overridden by subclasses to do
 273   /// any neccesary per-module initialization.
 274   ///
 275   virtual bool doInitialization(Module &M) { return false; }
 276
 277   /// doInitialization - Virtual method overridden by BasicBlockPass subclasses
 278   /// to do any neccesary per-function initialization.
 279   ///
 280   virtual bool doInitialization(Function &F) { return false; }
 281
 282   /// runOnBasicBlock - Virtual method overriden by subclasses to do the
 283   /// per-basicblock processing of the pass.
 284   ///
 285   virtual bool runOnBasicBlock(BasicBlock &BB) = 0;
 286
 287   /// doFinalization - Virtual method overriden by BasicBlockPass subclasses to
 288   /// do any post processing needed after all passes have run.
 289   ///
 290   virtual bool doFinalization(Function &F) { return false; }
 291
 292   /// doFinalization - Virtual method overriden by subclasses to do any post
 293   /// processing needed after all passes have run.
 294   ///
 295   virtual bool doFinalization(Module &M) { return false; }
 296
 297
 298   // To run this pass on a function, we simply call runOnBasicBlock once for
 299   // each function.
 300   //
 301   bool runOnFunction(Function &F);
 302
 303   /// To run directly on the basic block, we initialize, runOnBasicBlock, then
 304   /// finalize.
 305   ///
 306   bool run(BasicBlock &BB);
 307
 308 private:
 309   friend class PassManagerT<Function>;
 310   friend class PassManagerT<BasicBlock>;
 311   virtual void addToPassManager(PassManagerT<Function> *PM, AnalysisUsage &AU);
 312   virtual void addToPassManager(PassManagerT<BasicBlock> *PM,AnalysisUsage &AU);
 313 };
 314
 315 // Include support files that contain important APIs commonly used by Passes,
 316 // but that we want to seperate out to make it easier to read the header files.
 317 //
 318 #include "llvm/PassSupport.h"
 319 #include "llvm/PassAnalysisSupport.h"
 320
 321 #endif