Add version of getMethodSummary() that can be used to query the summary for the

method currently being analyzed.


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

1 files changed, 14 insertions, 0 deletions

diff --git a/lib/Analysis/CFRefCount.cpp b/lib/Analysis/CFRefCount.cpp
index 721d80095a..149fe23465 100644
--- a/lib/Analysis/CFRefCount.cpp
+++ b/lib/Analysis/CFRefCount.cpp
@@ -748,6 +748,20 @@ public:
                                  ME->getClassInfo().first,
                                  ME->getMethodDecl(), ME->getType());
   }
+
+  /// getMethodSummary - This version of getMethodSummary is used to query
+  ///  the summary for the current method being analyzed.
+  RetainSummary *getMethodSummary(ObjCMethodDecl *MD) {
+    Selector S = MD->getSelector();
+    ObjCInterfaceDecl *ID = MD->getClassInterface();
+    IdentifierInfo *ClsName = ID->getIdentifier();
+    QualType ResultTy = MD->getResultType();
+    
+    if (MD->isInstanceMethod())
+      return getInstanceMethodSummary(S, ClsName, ID, MD, ResultTy);
+    else
+      return getClassMethodSummary(S, ClsName, ID, MD, ResultTy);
+  }
   
   RetainSummary* getCommonMethodSummary(ObjCMethodDecl* MD, Selector S,
                                         QualType RetTy);