From be125444cf48d0fbc80c2f438a6842ae74424f16 Mon Sep 17 00:00:00 2001 From: Argyrios Kyrtzidis Date: Tue, 14 Jul 2009 03:19:30 +0000 Subject: For index-test, if the ASTLocation points at a CallExpr, get a Decl out of it. git-svn-id: https://llvm.org/svn/llvm-project/cfe/trunk@75599 91177308-0d34-0410-b5e6-96231b3b80d8 --- tools/index-test/index-test.cpp | 23 ++++++++++++++++------- 1 file changed, 16 insertions(+), 7 deletions(-) (limited to 'tools/index-test/index-test.cpp') diff --git a/tools/index-test/index-test.cpp b/tools/index-test/index-test.cpp index b7471a1c9d..63c86aac46 100644 --- a/tools/index-test/index-test.cpp +++ b/tools/index-test/index-test.cpp @@ -150,18 +150,27 @@ static void ProcessDecl(Decl *D) { } } +static Decl *getDeclFromExpr(Stmt *E) { + if (DeclRefExpr *RefExpr = dyn_cast(E)) + return RefExpr->getDecl(); + if (MemberExpr *ME = dyn_cast(E)) + return ME->getMemberDecl(); + if (CallExpr *CE = dyn_cast(E)) + return getDeclFromExpr(CE->getCallee()); + if (CastExpr *CE = dyn_cast(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(ASTLoc.getStmt())) - D = RefExpr->getDecl(); - else if (MemberExpr *ME = dyn_cast(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"; -- cgit v1.2.3-18-g5258