Refactor code into a new CallExpr::getDirectCallee() method. Simplify some
code with the new method.
git-svn-id: https://llvm.org/svn/llvm-project/cfe/trunk@76164 91177308-0d34-0410-b5e6-96231b3b80d8
diff --git a/lib/Analysis/CallGraph.cpp b/lib/Analysis/CallGraph.cpp
index 16f14d7..d49e8ec 100644
--- a/lib/Analysis/CallGraph.cpp
+++ b/lib/Analysis/CallGraph.cpp
@@ -66,21 +66,10 @@
}
void CGBuilder::VisitCallExpr(CallExpr *CE) {
- Expr *Callee = CE->getCallee();
-
- if (CastExpr *CE = dyn_cast<CastExpr>(Callee))
- Callee = CE->getSubExpr();
-
- if (DeclRefExpr *DRE = dyn_cast<DeclRefExpr>(Callee)) {
- Decl *D = DRE->getDecl();
- if (FunctionDecl *CalleeDecl = dyn_cast<FunctionDecl>(D)) {
-
- Entity *Ent = Entity::get(CalleeDecl, G.getProgram());
-
- CallGraphNode *CalleeNode = G.getOrInsertFunction(Ent);
-
- CallerNode->addCallee(ASTLocation(FD, CE), CalleeNode);
- }
+ if (FunctionDecl *CalleeDecl = CE->getDirectCallee()) {
+ Entity *Ent = Entity::get(CalleeDecl, G.getProgram());
+ CallGraphNode *CalleeNode = G.getOrInsertFunction(Ent);
+ CallerNode->addCallee(ASTLocation(FD, CE), CalleeNode);
}
}