Details | Last modification | View Log | RSS feed
Rev | Author | Line No. | Line |
---|---|---|---|
14 | pmbaty | 1 | //===- llvm/Analysis/DominanceFrontier.h - Dominator Frontiers --*- C++ -*-===// |
2 | // |
||
3 | // Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions. |
||
4 | // See https://llvm.org/LICENSE.txt for license information. |
||
5 | // SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception |
||
6 | // |
||
7 | //===----------------------------------------------------------------------===// |
||
8 | // |
||
9 | // This is the generic implementation of the DominanceFrontier class, which |
||
10 | // calculate and holds the dominance frontier for a function for. |
||
11 | // |
||
12 | // This should be considered deprecated, don't add any more uses of this data |
||
13 | // structure. |
||
14 | // |
||
15 | //===----------------------------------------------------------------------===// |
||
16 | |||
17 | #ifndef LLVM_ANALYSIS_DOMINANCEFRONTIERIMPL_H |
||
18 | #define LLVM_ANALYSIS_DOMINANCEFRONTIERIMPL_H |
||
19 | |||
20 | #include "llvm/ADT/SmallPtrSet.h" |
||
21 | #include "llvm/Analysis/DominanceFrontier.h" |
||
22 | #include "llvm/Config/llvm-config.h" |
||
23 | #include "llvm/Support/Debug.h" |
||
24 | #include "llvm/Support/GenericDomTree.h" |
||
25 | #include "llvm/Support/raw_ostream.h" |
||
26 | #include <cassert> |
||
27 | #include <set> |
||
28 | #include <utility> |
||
29 | #include <vector> |
||
30 | |||
31 | namespace llvm { |
||
32 | |||
33 | template <class BlockT> |
||
34 | class DFCalculateWorkObject { |
||
35 | public: |
||
36 | using DomTreeNodeT = DomTreeNodeBase<BlockT>; |
||
37 | |||
38 | DFCalculateWorkObject(BlockT *B, BlockT *P, const DomTreeNodeT *N, |
||
39 | const DomTreeNodeT *PN) |
||
40 | : currentBB(B), parentBB(P), Node(N), parentNode(PN) {} |
||
41 | |||
42 | BlockT *currentBB; |
||
43 | BlockT *parentBB; |
||
44 | const DomTreeNodeT *Node; |
||
45 | const DomTreeNodeT *parentNode; |
||
46 | }; |
||
47 | |||
48 | template <class BlockT, bool IsPostDom> |
||
49 | void DominanceFrontierBase<BlockT, IsPostDom>::removeBlock(BlockT *BB) { |
||
50 | assert(find(BB) != end() && "Block is not in DominanceFrontier!"); |
||
51 | for (iterator I = begin(), E = end(); I != E; ++I) |
||
52 | I->second.erase(BB); |
||
53 | Frontiers.erase(BB); |
||
54 | } |
||
55 | |||
56 | template <class BlockT, bool IsPostDom> |
||
57 | void DominanceFrontierBase<BlockT, IsPostDom>::addToFrontier(iterator I, |
||
58 | BlockT *Node) { |
||
59 | assert(I != end() && "BB is not in DominanceFrontier!"); |
||
60 | assert(I->second.count(Node) && "Node is not in DominanceFrontier of BB"); |
||
61 | I->second.erase(Node); |
||
62 | } |
||
63 | |||
64 | template <class BlockT, bool IsPostDom> |
||
65 | void DominanceFrontierBase<BlockT, IsPostDom>::removeFromFrontier( |
||
66 | iterator I, BlockT *Node) { |
||
67 | assert(I != end() && "BB is not in DominanceFrontier!"); |
||
68 | assert(I->second.count(Node) && "Node is not in DominanceFrontier of BB"); |
||
69 | I->second.erase(Node); |
||
70 | } |
||
71 | |||
72 | template <class BlockT, bool IsPostDom> |
||
73 | bool DominanceFrontierBase<BlockT, IsPostDom>::compareDomSet( |
||
74 | DomSetType &DS1, const DomSetType &DS2) const { |
||
75 | std::set<BlockT *> tmpSet; |
||
76 | for (BlockT *BB : DS2) |
||
77 | tmpSet.insert(BB); |
||
78 | |||
79 | for (typename DomSetType::const_iterator I = DS1.begin(), E = DS1.end(); |
||
80 | I != E;) { |
||
81 | BlockT *Node = *I++; |
||
82 | |||
83 | if (tmpSet.erase(Node) == 0) |
||
84 | // Node is in DS1 but tnot in DS2. |
||
85 | return true; |
||
86 | } |
||
87 | |||
88 | if (!tmpSet.empty()) { |
||
89 | // There are nodes that are in DS2 but not in DS1. |
||
90 | return true; |
||
91 | } |
||
92 | |||
93 | // DS1 and DS2 matches. |
||
94 | return false; |
||
95 | } |
||
96 | |||
97 | template <class BlockT, bool IsPostDom> |
||
98 | bool DominanceFrontierBase<BlockT, IsPostDom>::compare( |
||
99 | DominanceFrontierBase<BlockT, IsPostDom> &Other) const { |
||
100 | DomSetMapType tmpFrontiers; |
||
101 | for (typename DomSetMapType::const_iterator I = Other.begin(), |
||
102 | E = Other.end(); |
||
103 | I != E; ++I) |
||
104 | tmpFrontiers.insert(std::make_pair(I->first, I->second)); |
||
105 | |||
106 | for (typename DomSetMapType::iterator I = tmpFrontiers.begin(), |
||
107 | E = tmpFrontiers.end(); |
||
108 | I != E;) { |
||
109 | BlockT *Node = I->first; |
||
110 | const_iterator DFI = find(Node); |
||
111 | if (DFI == end()) |
||
112 | return true; |
||
113 | |||
114 | if (compareDomSet(I->second, DFI->second)) |
||
115 | return true; |
||
116 | |||
117 | ++I; |
||
118 | tmpFrontiers.erase(Node); |
||
119 | } |
||
120 | |||
121 | if (!tmpFrontiers.empty()) |
||
122 | return true; |
||
123 | |||
124 | return false; |
||
125 | } |
||
126 | |||
127 | template <class BlockT, bool IsPostDom> |
||
128 | void DominanceFrontierBase<BlockT, IsPostDom>::print(raw_ostream &OS) const { |
||
129 | for (const_iterator I = begin(), E = end(); I != E; ++I) { |
||
130 | OS << " DomFrontier for BB "; |
||
131 | if (I->first) |
||
132 | I->first->printAsOperand(OS, false); |
||
133 | else |
||
134 | OS << " <<exit node>>"; |
||
135 | OS << " is:\t"; |
||
136 | |||
137 | const std::set<BlockT *> &BBs = I->second; |
||
138 | |||
139 | for (const BlockT *BB : BBs) { |
||
140 | OS << ' '; |
||
141 | if (BB) |
||
142 | BB->printAsOperand(OS, false); |
||
143 | else |
||
144 | OS << "<<exit node>>"; |
||
145 | } |
||
146 | OS << '\n'; |
||
147 | } |
||
148 | } |
||
149 | |||
150 | #if !defined(NDEBUG) || defined(LLVM_ENABLE_DUMP) |
||
151 | template <class BlockT, bool IsPostDom> |
||
152 | void DominanceFrontierBase<BlockT, IsPostDom>::dump() const { |
||
153 | print(dbgs()); |
||
154 | } |
||
155 | #endif |
||
156 | |||
157 | template <class BlockT> |
||
158 | const typename ForwardDominanceFrontierBase<BlockT>::DomSetType & |
||
159 | ForwardDominanceFrontierBase<BlockT>::calculate(const DomTreeT &DT, |
||
160 | const DomTreeNodeT *Node) { |
||
161 | BlockT *BB = Node->getBlock(); |
||
162 | DomSetType *Result = nullptr; |
||
163 | |||
164 | std::vector<DFCalculateWorkObject<BlockT>> workList; |
||
165 | SmallPtrSet<BlockT *, 32> visited; |
||
166 | |||
167 | workList.push_back(DFCalculateWorkObject<BlockT>(BB, nullptr, Node, nullptr)); |
||
168 | do { |
||
169 | DFCalculateWorkObject<BlockT> *currentW = &workList.back(); |
||
170 | assert(currentW && "Missing work object."); |
||
171 | |||
172 | BlockT *currentBB = currentW->currentBB; |
||
173 | BlockT *parentBB = currentW->parentBB; |
||
174 | const DomTreeNodeT *currentNode = currentW->Node; |
||
175 | const DomTreeNodeT *parentNode = currentW->parentNode; |
||
176 | assert(currentBB && "Invalid work object. Missing current Basic Block"); |
||
177 | assert(currentNode && "Invalid work object. Missing current Node"); |
||
178 | DomSetType &S = this->Frontiers[currentBB]; |
||
179 | |||
180 | // Visit each block only once. |
||
181 | if (visited.insert(currentBB).second) { |
||
182 | // Loop over CFG successors to calculate DFlocal[currentNode] |
||
183 | for (const auto Succ : children<BlockT *>(currentBB)) { |
||
184 | // Does Node immediately dominate this successor? |
||
185 | if (DT[Succ]->getIDom() != currentNode) |
||
186 | S.insert(Succ); |
||
187 | } |
||
188 | } |
||
189 | |||
190 | // At this point, S is DFlocal. Now we union in DFup's of our children... |
||
191 | // Loop through and visit the nodes that Node immediately dominates (Node's |
||
192 | // children in the IDomTree) |
||
193 | bool visitChild = false; |
||
194 | for (typename DomTreeNodeT::const_iterator NI = currentNode->begin(), |
||
195 | NE = currentNode->end(); |
||
196 | NI != NE; ++NI) { |
||
197 | DomTreeNodeT *IDominee = *NI; |
||
198 | BlockT *childBB = IDominee->getBlock(); |
||
199 | if (visited.count(childBB) == 0) { |
||
200 | workList.push_back(DFCalculateWorkObject<BlockT>( |
||
201 | childBB, currentBB, IDominee, currentNode)); |
||
202 | visitChild = true; |
||
203 | } |
||
204 | } |
||
205 | |||
206 | // If all children are visited or there is any child then pop this block |
||
207 | // from the workList. |
||
208 | if (!visitChild) { |
||
209 | if (!parentBB) { |
||
210 | Result = &S; |
||
211 | break; |
||
212 | } |
||
213 | |||
214 | typename DomSetType::const_iterator CDFI = S.begin(), CDFE = S.end(); |
||
215 | DomSetType &parentSet = this->Frontiers[parentBB]; |
||
216 | for (; CDFI != CDFE; ++CDFI) { |
||
217 | if (!DT.properlyDominates(parentNode, DT[*CDFI])) |
||
218 | parentSet.insert(*CDFI); |
||
219 | } |
||
220 | workList.pop_back(); |
||
221 | } |
||
222 | |||
223 | } while (!workList.empty()); |
||
224 | |||
225 | return *Result; |
||
226 | } |
||
227 | |||
228 | } // end namespace llvm |
||
229 | |||
230 | #endif // LLVM_ANALYSIS_DOMINANCEFRONTIERIMPL_H |