[cfe-commits] r75599 - /cfe/trunk/tools/index-test/index-test.cpp

[Argiris Kirtzidis]

Author: akirtzidis
Date: Mon Jul 13 22:19:30 2009
New Revision: 75599

URL: http://llvm.org/viewvc/llvm-project?rev=75599&view=rev
Log:
For index-test, if the ASTLocation points at a CallExpr, get a Decl out of it.

Modified:
    cfe/trunk/tools/index-test/index-test.cpp

Modified: cfe/trunk/tools/index-test/index-test.cpp
URL: http://llvm.org/viewvc/llvm-project/cfe/trunk/tools/index-test/index-test.cpp?rev=75599&r1=75598&r2=75599&view=diff

==============================================================================
--- cfe/trunk/tools/index-test/index-test.cpp (original)
+++ cfe/trunk/tools/index-test/index-test.cpp Mon Jul 13 22:19:30 2009
@@ -150,18 +150,27 @@
   }
 }
 
+static Decl *getDeclFromExpr(Stmt *E) {
+  if (DeclRefExpr *RefExpr = dyn_cast<DeclRefExpr>(E))
+    return RefExpr->getDecl();
+  if (MemberExpr *ME = dyn_cast<MemberExpr>(E))
+    return ME->getMemberDecl();
+  if (CallExpr *CE = dyn_cast<CallExpr>(E))
+    return getDeclFromExpr(CE->getCallee());
+  if (CastExpr *CE = dyn_cast<CastExpr>(E))
+    return getDeclFromExpr(CE->getSubExpr());
+  
+  return 0;
+}
+
 static void ProcessASTLocation(ASTLocation ASTLoc, IndexProvider &IdxProvider) {
   assert(ASTLoc.isValid());
 
   Decl *D = 0;
-  if (ASTLoc.isStmt()) {
-    if (DeclRefExpr *RefExpr = dyn_cast<DeclRefExpr>(ASTLoc.getStmt()))
-      D = RefExpr->getDecl();
-    else if (MemberExpr *ME = dyn_cast<MemberExpr>(ASTLoc.getStmt()))
-      D = ME->getMemberDecl();
-  } else {
+  if (ASTLoc.isStmt())
+    D = getDeclFromExpr(ASTLoc.getStmt());
+  else
     D = ASTLoc.getDecl();
-  }
   
   if (D == 0) {
     llvm::errs() << "Error: Couldn't get a Decl out of the ASTLocation";