[llvm-commits] CVS: llvm/lib/Analysis/CFGPrinter.cpp

[Chris Lattner]

Changes in directory llvm/lib/Analysis:

CFGPrinter.cpp updated: 1.4 -> 1.5

---
Log message:

Finegrainify namespacification
Add new -print-cfg-only pass


---
Diffs of the changes:  (+23 -5)

Index: llvm/lib/Analysis/CFGPrinter.cpp
diff -u llvm/lib/Analysis/CFGPrinter.cpp:1.4 llvm/lib/Analysis/CFGPrinter.cpp:1.5
--- llvm/lib/Analysis/CFGPrinter.cpp:1.4	Tue Nov 11 16:41:31 2003
+++ llvm/lib/Analysis/CFGPrinter.cpp	Thu Dec 11 15:48:18 2003
@@ -25,8 +25,7 @@
 #include "llvm/Support/CFG.h"
 #include <sstream>
 #include <fstream>
-
-namespace llvm {
+using namespace llvm;
 
 /// CFGOnly flag - This is used to control whether or not the CFG graph printer
 /// prints out the contents of basic blocks or not.  This is acceptable because
@@ -34,6 +33,7 @@
 ///
 static bool CFGOnly = false;
 
+namespace llvm {
 template<>
 struct DOTGraphTraits<const Function*> : public DefaultDOTGraphTraits {
   static std::string getGraphName(const Function *F) {
@@ -87,6 +87,7 @@
     return "";
   }
 };
+}
 
 namespace {
   struct CFGPrinter : public FunctionPass {
@@ -112,7 +113,26 @@
 
   RegisterAnalysis<CFGPrinter> P1("print-cfg",
                                   "Print CFG of function to 'dot' file");
-};
+
+  struct CFGOnlyPrinter : public CFGPrinter {
+    virtual bool runOnFunction(Function &F) {
+      bool OldCFGOnly = CFGOnly;
+      CFGOnly = true;
+      CFGPrinter::runOnFunction(F);
+      CFGOnly = OldCFGOnly;
+      return false;
+    }
+    void print(std::ostream &OS) const {}
+    
+    virtual void getAnalysisUsage(AnalysisUsage &AU) const {
+      AU.setPreservesAll();
+    }
+  };
+
+  RegisterAnalysis<CFGOnlyPrinter>
+  P2("print-cfg-only",
+     "Print CFG of function to 'dot' file (with no function bodies)");
+}
 
 /// viewCFG - This function is meant for use from the debugger.  You can just
 /// say 'call F->viewCFG()' and a ghostview window should pop up from the
@@ -153,5 +173,3 @@
   viewCFG();
   CFGOnly = false;
 }
-
-} // End llvm namespace