2 files changed, 8 insertions, 0 deletions
diff --git a/include/llvm/Analysis/DataStructure.h b/include/llvm/Analysis/DataStructure.h
index 3dce043..335477f 100644
--- a/include/llvm/Analysis/DataStructure.h
+++ b/include/llvm/Analysis/DataStructure.h
@@ -432,6 +432,10 @@ public:
 
   const PointerValSet &getRetNodes() const { return RetNode; }
 
+  unsigned getGraphSize() const {
+    return ArgNodes.size() + AllocNodes.size() + ShadowNodes.size() +
+      GlobalNodes.size() + CallNodes.size();
+  }
 
   void printFunction(std::ostream &O, const char *Label) const;
 };
diff --git a/include/llvm/Analysis/DataStructure/DataStructure.h b/include/llvm/Analysis/DataStructure/DataStructure.h
index 3dce043..335477f 100644
--- a/include/llvm/Analysis/DataStructure/DataStructure.h
+++ b/include/llvm/Analysis/DataStructure/DataStructure.h
@@ -432,6 +432,10 @@ public:
 
   const PointerValSet &getRetNodes() const { return RetNode; }
 
+  unsigned getGraphSize() const {
+    return ArgNodes.size() + AllocNodes.size() + ShadowNodes.size() +
+      GlobalNodes.size() + CallNodes.size();
+  }
 
   void printFunction(std::ostream &O, const char *Label) const;
 };