Seperate analysis wrapper stuff to AnalysisWrappers.cpp


git-svn-id: https://llvm.org/svn/llvm-project/llvm/trunk@3127 91177308-0d34-0410-b5e6-96231b3b80d8
diff --git a/tools/opt/AnalysisWrappers.cpp b/tools/opt/AnalysisWrappers.cpp
new file mode 100644
index 0000000..db69517
--- /dev/null
+++ b/tools/opt/AnalysisWrappers.cpp
@@ -0,0 +1,101 @@
+//===- AnalysisWrappers.cpp - Wrappers around non-pass analyses -----------===//
+//
+// This file defines pass wrappers around LLVM analyses that don't make sense to
+// be passes.  It provides a nice standard pass interface to these classes so
+// that they can be printed out by analyze.
+//
+// These classes are seperated out of analyze.cpp so that it is more clear which
+// code is the integral part of the analyze tool, and which part of the code is
+// just making it so more passes are available.
+//
+//===----------------------------------------------------------------------===//
+
+#include "llvm/iPHINode.h"
+#include "llvm/Type.h"
+#include "llvm/Assembly/Writer.h"
+#include "llvm/Analysis/InstForest.h"
+#include "llvm/Analysis/Expressions.h"
+#include "llvm/Analysis/InductionVariable.h"
+#include "llvm/Analysis/LoopInfo.h"
+#include "llvm/Support/InstIterator.h"
+
+namespace {
+  struct InstForestHelper : public FunctionPass {
+    Function *F;
+    virtual bool runOnFunction(Function &Func) { F = &Func; return false; }
+
+    void print(std::ostream &OS) const {
+      std::cout << InstForest<char>(F);
+    }
+    
+    virtual void getAnalysisUsage(AnalysisUsage &AU) const {
+      AU.setPreservesAll();
+    }
+  };
+
+  RegisterAnalysis<InstForestHelper> P1("instforest", "InstForest Printer");
+
+  struct IndVars : public FunctionPass {
+    Function *F;
+    LoopInfo *LI;
+    virtual bool runOnFunction(Function &Func) {
+      F = &Func; LI = &getAnalysis<LoopInfo>();
+      return false;
+    }
+
+    void print(std::ostream &OS) const {
+      for (inst_iterator I = inst_begin(*F), E = inst_end(*F); I != E; ++I)
+        if (PHINode *PN = dyn_cast<PHINode>(*I)) {
+          InductionVariable IV(PN, LI);
+          if (IV.InductionType != InductionVariable::Unknown)
+            IV.print(OS);
+        }
+    }
+    
+    void getAnalysisUsage(AnalysisUsage &AU) const {
+      AU.addRequired(LoopInfo::ID);
+      AU.setPreservesAll();
+    }
+  };
+
+  RegisterAnalysis<IndVars> P6("indvars", "Induction Variable Analysis");
+
+
+  struct Exprs : public FunctionPass {
+    Function *F;
+    virtual bool runOnFunction(Function &Func) { F = &Func; return false; }
+
+    void print(std::ostream &OS) const {
+      OS << "Classified expressions for: " << F->getName() << "\n";
+      for (inst_iterator I = inst_begin(*F), E = inst_end(*F); I != E; ++I) {
+        OS << *I;
+      
+        if ((*I)->getType() == Type::VoidTy) continue;
+        analysis::ExprType R = analysis::ClassifyExpression(*I);
+        if (R.Var == *I) continue;  // Doesn't tell us anything
+      
+        OS << "\t\tExpr =";
+        switch (R.ExprTy) {
+        case analysis::ExprType::ScaledLinear:
+          WriteAsOperand(OS << "(", (Value*)R.Scale) << " ) *";
+          // fall through
+        case analysis::ExprType::Linear:
+          WriteAsOperand(OS << "(", R.Var) << " )";
+          if (R.Offset == 0) break;
+          else OS << " +";
+          // fall through
+        case analysis::ExprType::Constant:
+          if (R.Offset) WriteAsOperand(OS, (Value*)R.Offset);
+          else OS << " 0";
+          break;
+        }
+        OS << "\n\n";
+      }
+    }
+    virtual void getAnalysisUsage(AnalysisUsage &AU) const {
+      AU.setPreservesAll();
+    }
+  };
+
+  RegisterAnalysis<Exprs> P7("exprs", "Expression Printer");
+}