Compute total number of nodes and # call nodes

llvm-svn: 4592

1 files changed, 6 insertions, 0 deletions

diff --git a/llvm/lib/Analysis/DataStructure/Printer.cpp b/llvm/lib/Analysis/DataStructure/Printer.cpp
index f5c1ec49cf1..edee17212d4 100644
--- a/llvm/lib/Analysis/DataStructure/Printer.cpp
+++ b/llvm/lib/Analysis/DataStructure/Printer.cpp
@@ -167,9 +167,12 @@ static void printCollection(const Collection &C, std::ostream &O,
     return;
   }
 
+  unsigned TotalNumNodes = 0, TotalCallNodes = 0;
   for (Module::const_iterator I = M->begin(), E = M->end(); I != E; ++I)
     if (!I->isExternal()) {
       DSGraph &Gr = C.getDSGraph((Function&)*I);
+      TotalNumNodes += Gr.getGraphSize();
+      TotalCallNodes += Gr.getFunctionCalls().size();
       if (I->getName() == "main" || !OnlyPrintMain)
         Gr.writeGraphToFile(O, Prefix+I->getName());
       else {
@@ -177,6 +180,9 @@ static void printCollection(const Collection &C, std::ostream &O,
           << Gr.getGraphSize() << "+" << Gr.getFunctionCalls().size() << "]\n";
       }
     }
+
+  O << "\nGraphs contain [" << TotalNumNodes << "+" << TotalCallNodes 
+    << "] nodes total\n";
 }