diff options
| -rw-r--r-- | include/clang/AST/ParentMap.h | 9 | ||||
| -rw-r--r-- | lib/AST/ParentMap.cpp | 5 |
2 files changed, 12 insertions, 2 deletions
diff --git a/include/clang/AST/ParentMap.h b/include/clang/AST/ParentMap.h index 199a360ca0..c669991ccc 100644 --- a/include/clang/AST/ParentMap.h +++ b/include/clang/AST/ParentMap.h @@ -24,11 +24,16 @@ public: ParentMap(Stmt* ASTRoot); ~ParentMap(); - Stmt* getParent(Stmt*) const; + Stmt *getParent(Stmt*) const; + Stmt *getParentIgnoreParens(Stmt *) const; - const Stmt* getParent(const Stmt* S) const { + const Stmt *getParent(const Stmt* S) const { return getParent(const_cast<Stmt*>(S)); } + + const Stmt *getParentIgnoreParens(const Stmt *S) const { + return getParentIgnoreParens(const_cast<Stmt*>(S)); + } bool hasParent(Stmt* S) const { return getParent(S) != 0; diff --git a/lib/AST/ParentMap.cpp b/lib/AST/ParentMap.cpp index 939e6f9be5..9d87daa0bf 100644 --- a/lib/AST/ParentMap.cpp +++ b/lib/AST/ParentMap.cpp @@ -46,6 +46,11 @@ Stmt* ParentMap::getParent(Stmt* S) const { return I == M->end() ? 0 : I->second; } +Stmt *ParentMap::getParentIgnoreParens(Stmt *S) const { + do { S = getParent(S); } while (S && isa<ParenExpr>(S)); + return S; +} + bool ParentMap::isConsumedExpr(Expr* E) const { Stmt *P = getParent(E); Stmt *DirectChild = E; |