|
@@ -405,14 +405,18 @@ private:
|
|
|
bool TraverseFunctionHelper(FunctionDecl *D);
|
|
|
bool TraverseVarHelper(VarDecl *D);
|
|
|
|
|
|
+ bool Walk(Stmt *S);
|
|
|
+
|
|
|
struct EnqueueJob {
|
|
|
Stmt *S;
|
|
|
Stmt::child_iterator StmtIt;
|
|
|
|
|
|
- EnqueueJob(Stmt *S) : S(S), StmtIt() {}
|
|
|
+ EnqueueJob(Stmt *S) : S(S), StmtIt() {
|
|
|
+ if (Expr *E = dyn_cast_or_null<Expr>(S))
|
|
|
+ S = E->IgnoreParens();
|
|
|
+ }
|
|
|
};
|
|
|
bool dataTraverse(Stmt *S);
|
|
|
- bool dataTraverseNode(Stmt *S, bool &EnqueueChildren);
|
|
|
};
|
|
|
|
|
|
template<typename Derived>
|
|
@@ -431,12 +435,7 @@ bool RecursiveASTVisitor<Derived>::dataTraverse(Stmt *S) {
|
|
|
|
|
|
if (getDerived().shouldUseDataRecursionFor(CurrS)) {
|
|
|
if (job.StmtIt == Stmt::child_iterator()) {
|
|
|
- bool EnqueueChildren = true;
|
|
|
- if (!dataTraverseNode(CurrS, EnqueueChildren)) return false;
|
|
|
- if (!EnqueueChildren) {
|
|
|
- Queue.pop_back();
|
|
|
- continue;
|
|
|
- }
|
|
|
+ if (!Walk(CurrS)) return false;
|
|
|
job.StmtIt = CurrS->child_begin();
|
|
|
} else {
|
|
|
++job.StmtIt;
|
|
@@ -457,16 +456,10 @@ bool RecursiveASTVisitor<Derived>::dataTraverse(Stmt *S) {
|
|
|
}
|
|
|
|
|
|
template<typename Derived>
|
|
|
-bool RecursiveASTVisitor<Derived>::dataTraverseNode(Stmt *S,
|
|
|
- bool &EnqueueChildren) {
|
|
|
+bool RecursiveASTVisitor<Derived>::Walk(Stmt *S) {
|
|
|
|
|
|
- // Dispatch to the corresponding WalkUpFrom* function only if the derived
|
|
|
- // class didn't override Traverse* (and thus the traversal is trivial).
|
|
|
#define DISPATCH_WALK(NAME, CLASS, VAR) \
|
|
|
- if (&RecursiveASTVisitor::Traverse##NAME == &Derived::Traverse##NAME) \
|
|
|
- return getDerived().WalkUpFrom##NAME(static_cast<CLASS*>(VAR)); \
|
|
|
- EnqueueChildren = false; \
|
|
|
- return getDerived().Traverse##NAME(static_cast<CLASS*>(VAR));
|
|
|
+ return getDerived().WalkUpFrom##NAME(static_cast<CLASS*>(VAR));
|
|
|
|
|
|
if (BinaryOperator *BinOp = dyn_cast<BinaryOperator>(S)) {
|
|
|
switch (BinOp->getOpcode()) {
|