diff options
author | Chandler Carruth <chandlerc@gmail.com> | 2014-04-28 11:10:23 +0000 |
---|---|---|
committer | Chandler Carruth <chandlerc@gmail.com> | 2014-04-28 11:10:23 +0000 |
commit | db0b52c8e09e7cbb3452d4560dfeb1c933034794 (patch) | |
tree | 0640796b4a83a3e0298e58b3a2a1cdc05d32bf03 /lib/Analysis/LazyCallGraph.cpp | |
parent | e52aad42029beb74d50bc493df34b4971ed74b31 (diff) | |
download | llvm-db0b52c8e09e7cbb3452d4560dfeb1c933034794.tar.gz llvm-db0b52c8e09e7cbb3452d4560dfeb1c933034794.tar.bz2 llvm-db0b52c8e09e7cbb3452d4560dfeb1c933034794.tar.xz |
[LCG] Add the most basic of edge insertion to the lazy call graph. This
just handles the pre-DFS case. Also add some test cases for this case to
make sure it works.
git-svn-id: https://llvm.org/svn/llvm-project/llvm/trunk@207411 91177308-0d34-0410-b5e6-96231b3b80d8
Diffstat (limited to 'lib/Analysis/LazyCallGraph.cpp')
-rw-r--r-- | lib/Analysis/LazyCallGraph.cpp | 15 |
1 files changed, 15 insertions, 0 deletions
diff --git a/lib/Analysis/LazyCallGraph.cpp b/lib/Analysis/LazyCallGraph.cpp index b593069e66..05757e1e2f 100644 --- a/lib/Analysis/LazyCallGraph.cpp +++ b/lib/Analysis/LazyCallGraph.cpp @@ -75,6 +75,14 @@ LazyCallGraph::Node::Node(LazyCallGraph &G, Function &F) findCallees(Worklist, Visited, Callees, CalleeIndexMap); } +void LazyCallGraph::Node::insertEdgeInternal(Function &Callee) { + CalleeIndexMap.insert(std::make_pair(&Callee, Callees.size())); + if (Node *N = G->lookup(Callee)) + Callees.push_back(N); + else + Callees.push_back(&Callee); +} + void LazyCallGraph::Node::removeEdgeInternal(Function &Callee) { auto IndexMapI = CalleeIndexMap.find(&Callee); assert(IndexMapI != CalleeIndexMap.end() && @@ -353,6 +361,13 @@ LazyCallGraph::SCC::removeIntraSCCEdge(Node &CallerN, return ResultSCCs; } +void LazyCallGraph::insertEdge(Node &CallerN, Function &Callee) { + assert(SCCMap.empty() && DFSStack.empty() && + "This method cannot be called after SCCs have been formed!"); + + return CallerN.insertEdgeInternal(Callee); +} + void LazyCallGraph::removeEdge(Node &CallerN, Function &Callee) { assert(SCCMap.empty() && DFSStack.empty() && "This method cannot be called after SCCs have been formed!"); |