commit 83390ec00b6e244dec5ab055ffe910645156da8a
author Ted Kremenek <kremenek@apple.com> Fri Feb 22 20:00:31 2008 +0000
committer Ted Kremenek <kremenek@apple.com> Fri Feb 22 20:00:31 2008 +0000
tree d5dda966642e8eed29e428ee657d9e85973b2337
parent e060a3548b3b7030a622071edf58e55c1121019e

clang driver options --dump-cfg and --view-cfg now (optionally) use the
--analyze-function option to dump/view the CFGs of specific functions.


git-svn-id: https://llvm.org/svn/llvm-project/cfe/trunk@47497 91177308-0d34-0410-b5e6-96231b3b80d8

Driver/ASTConsumers.cpp

diff --git a/Driver/ASTConsumers.cpp b/Driver/ASTConsumers.cpp
index 2b205f0..316bcd4 100644
--- a/Driver/ASTConsumers.cpp
+++ b/Driver/ASTConsumers.cpp
@@ -453,7 +453,11 @@
 namespace {
 
 class CFGVisitor : public ASTConsumer {
+  std::string FName;
 public:
+  CFGVisitor(const std::string& fname) : FName(fname) {}
+  CFGVisitor() : FName("") {}
+  
   // CFG Visitor interface to be implemented by subclass.
   virtual void VisitCFG(CFG& C, FunctionDecl& FD) = 0;
   virtual bool printFuncDeclStart() { return true; }
@@ -465,8 +469,12 @@
 
 void CFGVisitor::HandleTopLevelDecl(Decl *D) {
   FunctionDecl *FD = dyn_cast<FunctionDecl>(D);
+
   if (!FD || !FD->getBody())
     return;
+  
+  if (FName.size() > 0 && FName != FD->getIdentifier()->getName())
+    return;
       
   if (printFuncDeclStart()) {
     DeclPrinter().PrintFunctionDeclStart(FD);
@@ -485,7 +493,8 @@
   class CFGDumper : public CFGVisitor {
     const bool UseGraphviz;
   public:
-    CFGDumper(bool use_graphviz) : UseGraphviz(use_graphviz) {}
+    CFGDumper(bool use_graphviz, const std::string& fname) 
+     : CFGVisitor(fname), UseGraphviz(use_graphviz) {}
     
     virtual void VisitCFG(CFG& C, FunctionDecl&) {
       if (UseGraphviz)
@@ -496,8 +505,8 @@
   }; 
 } // end anonymous namespace 
   
-ASTConsumer *clang::CreateCFGDumper(bool ViewGraphs) {
-  return new CFGDumper(ViewGraphs);
+ASTConsumer *clang::CreateCFGDumper(bool ViewGraphs, const std::string& FName) {
+  return new CFGDumper(ViewGraphs, FName);
 }
 
 //===----------------------------------------------------------------------===//
@@ -582,10 +591,10 @@
     Diagnostic &Diags;
     ASTContext* Ctx;
     bool Visualize;
-    std::string FName;
   public:
-    GRSimpleValsVisitor(Diagnostic &diags, const std::string& fname, bool visualize)
-      : Diags(diags), Visualize(visualize), FName(fname) {}
+    GRSimpleValsVisitor(Diagnostic &diags, const std::string& fname,
+                        bool visualize)
+      : CFGVisitor(fname), Diags(diags), Visualize(visualize) {}
     
     virtual void Initialize(ASTContext &Context) { Ctx = &Context; }    
     virtual void VisitCFG(CFG& C, FunctionDecl&);
@@ -601,8 +610,6 @@
 }
 
 void GRSimpleValsVisitor::VisitCFG(CFG& C, FunctionDecl& FD) {
-  if (FName.size() > 0 && FName != FD.getIdentifier()->getName())
-    return;
   
   SourceLocation Loc = FD.getLocation();