Home | History | Annotate | Download | only in llvm
      1 //===- llvm/PassAnalysisSupport.h - Analysis Pass Support code --*- C++ -*-===//
      2 //
      3 //                     The LLVM Compiler Infrastructure
      4 //
      5 // This file is distributed under the University of Illinois Open Source
      6 // License. See LICENSE.TXT for details.
      7 //
      8 //===----------------------------------------------------------------------===//
      9 //
     10 // This file defines stuff that is used to define and "use" Analysis Passes.
     11 // This file is automatically #included by Pass.h, so:
     12 //
     13 //           NO .CPP FILES SHOULD INCLUDE THIS FILE DIRECTLY
     14 //
     15 // Instead, #include Pass.h
     16 //
     17 //===----------------------------------------------------------------------===//
     18 
     19 #ifndef LLVM_PASSANALYSISSUPPORT_H
     20 #define LLVM_PASSANALYSISSUPPORT_H
     21 
     22 #include "llvm/ADT/SmallVector.h"
     23 #include "llvm/ADT/StringRef.h"
     24 #include "llvm/Pass.h"
     25 #include <vector>
     26 
     27 namespace llvm {
     28 
     29 //===----------------------------------------------------------------------===//
     30 /// Represent the analysis usage information of a pass.  This tracks analyses
     31 /// that the pass REQUIRES (must be available when the pass runs), REQUIRES
     32 /// TRANSITIVE (must be available throughout the lifetime of the pass), and
     33 /// analyses that the pass PRESERVES (the pass does not invalidate the results
     34 /// of these analyses).  This information is provided by a pass to the Pass
     35 /// infrastructure through the getAnalysisUsage virtual function.
     36 ///
     37 class AnalysisUsage {
     38 public:
     39   typedef SmallVectorImpl<AnalysisID> VectorType;
     40 
     41 private:
     42   /// Sets of analyses required and preserved by a pass
     43   // TODO: It's not clear that SmallVector is an appropriate data structure for
     44   // this usecase.  The sizes were picked to minimize wasted space, but are
     45   // otherwise fairly meaningless.
     46   SmallVector<AnalysisID, 8> Required;
     47   SmallVector<AnalysisID, 2> RequiredTransitive;
     48   SmallVector<AnalysisID, 2> Preserved;
     49   SmallVector<AnalysisID, 0> Used;
     50   bool PreservesAll;
     51 
     52 public:
     53   AnalysisUsage() : PreservesAll(false) {}
     54 
     55   ///@{
     56   /// Add the specified ID to the required set of the usage info for a pass.
     57   AnalysisUsage &addRequiredID(const void *ID);
     58   AnalysisUsage &addRequiredID(char &ID);
     59   template<class PassClass>
     60   AnalysisUsage &addRequired() {
     61     return addRequiredID(PassClass::ID);
     62   }
     63 
     64   AnalysisUsage &addRequiredTransitiveID(char &ID);
     65   template<class PassClass>
     66   AnalysisUsage &addRequiredTransitive() {
     67     return addRequiredTransitiveID(PassClass::ID);
     68   }
     69   ///@}
     70 
     71   ///@{
     72   /// Add the specified ID to the set of analyses preserved by this pass.
     73   AnalysisUsage &addPreservedID(const void *ID) {
     74     Preserved.push_back(ID);
     75     return *this;
     76   }
     77   AnalysisUsage &addPreservedID(char &ID) {
     78     Preserved.push_back(&ID);
     79     return *this;
     80   }
     81   /// Add the specified Pass class to the set of analyses preserved by this pass.
     82   template<class PassClass>
     83   AnalysisUsage &addPreserved() {
     84     Preserved.push_back(&PassClass::ID);
     85     return *this;
     86   }
     87   ///@}
     88 
     89   ///@{
     90   /// Add the specified ID to the set of analyses used by this pass if they are
     91   /// available..
     92   AnalysisUsage &addUsedIfAvailableID(const void *ID) {
     93     Used.push_back(ID);
     94     return *this;
     95   }
     96   AnalysisUsage &addUsedIfAvailableID(char &ID) {
     97     Used.push_back(&ID);
     98     return *this;
     99   }
    100   /// Add the specified Pass class to the set of analyses used by this pass.
    101   template<class PassClass>
    102   AnalysisUsage &addUsedIfAvailable() {
    103     Used.push_back(&PassClass::ID);
    104     return *this;
    105   }
    106   ///@}
    107 
    108   /// Add the Pass with the specified argument string to the set of analyses
    109   /// preserved by this pass. If no such Pass exists, do nothing. This can be
    110   /// useful when a pass is trivially preserved, but may not be linked in. Be
    111   /// careful about spelling!
    112   AnalysisUsage &addPreserved(StringRef Arg);
    113 
    114   /// Set by analyses that do not transform their input at all
    115   void setPreservesAll() { PreservesAll = true; }
    116 
    117   /// Determine whether a pass said it does not transform its input at all
    118   bool getPreservesAll() const { return PreservesAll; }
    119 
    120   /// This function should be called by the pass, iff they do not:
    121   ///
    122   ///  1. Add or remove basic blocks from the function
    123   ///  2. Modify terminator instructions in any way.
    124   ///
    125   /// This function annotates the AnalysisUsage info object to say that analyses
    126   /// that only depend on the CFG are preserved by this pass.
    127   ///
    128   void setPreservesCFG();
    129 
    130   const VectorType &getRequiredSet() const { return Required; }
    131   const VectorType &getRequiredTransitiveSet() const {
    132     return RequiredTransitive;
    133   }
    134   const VectorType &getPreservedSet() const { return Preserved; }
    135   const VectorType &getUsedSet() const { return Used; }
    136 };
    137 
    138 //===----------------------------------------------------------------------===//
    139 /// AnalysisResolver - Simple interface used by Pass objects to pull all
    140 /// analysis information out of pass manager that is responsible to manage
    141 /// the pass.
    142 ///
    143 class PMDataManager;
    144 class AnalysisResolver {
    145 private:
    146   AnalysisResolver() = delete;
    147 
    148 public:
    149   explicit AnalysisResolver(PMDataManager &P) : PM(P) { }
    150 
    151   inline PMDataManager &getPMDataManager() { return PM; }
    152 
    153   /// Find pass that is implementing PI.
    154   Pass *findImplPass(AnalysisID PI) {
    155     Pass *ResultPass = nullptr;
    156     for (unsigned i = 0; i < AnalysisImpls.size() ; ++i) {
    157       if (AnalysisImpls[i].first == PI) {
    158         ResultPass = AnalysisImpls[i].second;
    159         break;
    160       }
    161     }
    162     return ResultPass;
    163   }
    164 
    165   /// Find pass that is implementing PI. Initialize pass for Function F.
    166   Pass *findImplPass(Pass *P, AnalysisID PI, Function &F);
    167 
    168   void addAnalysisImplsPair(AnalysisID PI, Pass *P) {
    169     if (findImplPass(PI) == P)
    170       return;
    171     std::pair<AnalysisID, Pass*> pir = std::make_pair(PI,P);
    172     AnalysisImpls.push_back(pir);
    173   }
    174 
    175   /// Clear cache that is used to connect a pass to the the analysis (PassInfo).
    176   void clearAnalysisImpls() {
    177     AnalysisImpls.clear();
    178   }
    179 
    180   /// Return analysis result or null if it doesn't exist.
    181   Pass *getAnalysisIfAvailable(AnalysisID ID, bool Direction) const;
    182 
    183 private:
    184   /// This keeps track of which passes implements the interfaces that are
    185   /// required by the current pass (to implement getAnalysis()).
    186   std::vector<std::pair<AnalysisID, Pass*> > AnalysisImpls;
    187 
    188   /// PassManager that is used to resolve analysis info
    189   PMDataManager &PM;
    190 };
    191 
    192 /// getAnalysisIfAvailable<AnalysisType>() - Subclasses use this function to
    193 /// get analysis information that might be around, for example to update it.
    194 /// This is different than getAnalysis in that it can fail (if the analysis
    195 /// results haven't been computed), so should only be used if you can handle
    196 /// the case when the analysis is not available.  This method is often used by
    197 /// transformation APIs to update analysis results for a pass automatically as
    198 /// the transform is performed.
    199 ///
    200 template<typename AnalysisType>
    201 AnalysisType *Pass::getAnalysisIfAvailable() const {
    202   assert(Resolver && "Pass not resident in a PassManager object!");
    203 
    204   const void *PI = &AnalysisType::ID;
    205 
    206   Pass *ResultPass = Resolver->getAnalysisIfAvailable(PI, true);
    207   if (!ResultPass) return nullptr;
    208 
    209   // Because the AnalysisType may not be a subclass of pass (for
    210   // AnalysisGroups), we use getAdjustedAnalysisPointer here to potentially
    211   // adjust the return pointer (because the class may multiply inherit, once
    212   // from pass, once from AnalysisType).
    213   return (AnalysisType*)ResultPass->getAdjustedAnalysisPointer(PI);
    214 }
    215 
    216 /// getAnalysis<AnalysisType>() - This function is used by subclasses to get
    217 /// to the analysis information that they claim to use by overriding the
    218 /// getAnalysisUsage function.
    219 ///
    220 template<typename AnalysisType>
    221 AnalysisType &Pass::getAnalysis() const {
    222   assert(Resolver && "Pass has not been inserted into a PassManager object!");
    223   return getAnalysisID<AnalysisType>(&AnalysisType::ID);
    224 }
    225 
    226 template<typename AnalysisType>
    227 AnalysisType &Pass::getAnalysisID(AnalysisID PI) const {
    228   assert(PI && "getAnalysis for unregistered pass!");
    229   assert(Resolver&&"Pass has not been inserted into a PassManager object!");
    230   // PI *must* appear in AnalysisImpls.  Because the number of passes used
    231   // should be a small number, we just do a linear search over a (dense)
    232   // vector.
    233   Pass *ResultPass = Resolver->findImplPass(PI);
    234   assert (ResultPass &&
    235           "getAnalysis*() called on an analysis that was not "
    236           "'required' by pass!");
    237 
    238   // Because the AnalysisType may not be a subclass of pass (for
    239   // AnalysisGroups), we use getAdjustedAnalysisPointer here to potentially
    240   // adjust the return pointer (because the class may multiply inherit, once
    241   // from pass, once from AnalysisType).
    242   return *(AnalysisType*)ResultPass->getAdjustedAnalysisPointer(PI);
    243 }
    244 
    245 /// getAnalysis<AnalysisType>() - This function is used by subclasses to get
    246 /// to the analysis information that they claim to use by overriding the
    247 /// getAnalysisUsage function.
    248 ///
    249 template<typename AnalysisType>
    250 AnalysisType &Pass::getAnalysis(Function &F) {
    251   assert(Resolver &&"Pass has not been inserted into a PassManager object!");
    252 
    253   return getAnalysisID<AnalysisType>(&AnalysisType::ID, F);
    254 }
    255 
    256 template<typename AnalysisType>
    257 AnalysisType &Pass::getAnalysisID(AnalysisID PI, Function &F) {
    258   assert(PI && "getAnalysis for unregistered pass!");
    259   assert(Resolver && "Pass has not been inserted into a PassManager object!");
    260   // PI *must* appear in AnalysisImpls.  Because the number of passes used
    261   // should be a small number, we just do a linear search over a (dense)
    262   // vector.
    263   Pass *ResultPass = Resolver->findImplPass(this, PI, F);
    264   assert(ResultPass && "Unable to find requested analysis info");
    265 
    266   // Because the AnalysisType may not be a subclass of pass (for
    267   // AnalysisGroups), we use getAdjustedAnalysisPointer here to potentially
    268   // adjust the return pointer (because the class may multiply inherit, once
    269   // from pass, once from AnalysisType).
    270   return *(AnalysisType*)ResultPass->getAdjustedAnalysisPointer(PI);
    271 }
    272 
    273 } // End llvm namespace
    274 
    275 #endif
    276