forked from OSchip/llvm-project
144 lines
3.9 KiB
C++
144 lines
3.9 KiB
C++
//===--- ASTVisitor.h - Visitor for an ASTContext ---------------*- C++ -*-===//
|
|
//
|
|
// The LLVM Compiler Infrastructure
|
|
//
|
|
// This file is distributed under the University of Illinois Open Source
|
|
// License. See LICENSE.TXT for details.
|
|
//
|
|
//===----------------------------------------------------------------------===//
|
|
//
|
|
// This file defines the ASTVisitor interface.
|
|
//
|
|
//===----------------------------------------------------------------------===//
|
|
|
|
#ifndef LLVM_CLANG_INDEX_ASTVISITOR_H
|
|
#define LLVM_CLANG_INDEX_ASTVISITOR_H
|
|
|
|
#include "clang/AST/DeclVisitor.h"
|
|
#include "clang/AST/StmtVisitor.h"
|
|
#include "clang/AST/TypeLocVisitor.h"
|
|
|
|
namespace clang {
|
|
|
|
namespace idx {
|
|
|
|
/// \brief Traverses the full AST, both Decls and Stmts.
|
|
template<typename ImplClass>
|
|
class ASTVisitor : public DeclVisitor<ImplClass>,
|
|
public StmtVisitor<ImplClass>,
|
|
public TypeLocVisitor<ImplClass> {
|
|
public:
|
|
ASTVisitor() : CurrentDecl(0) { }
|
|
|
|
Decl *CurrentDecl;
|
|
|
|
typedef ASTVisitor<ImplClass> Base;
|
|
typedef DeclVisitor<ImplClass> BaseDeclVisitor;
|
|
typedef StmtVisitor<ImplClass> BaseStmtVisitor;
|
|
typedef TypeLocVisitor<ImplClass> BaseTypeLocVisitor;
|
|
|
|
using BaseStmtVisitor::Visit;
|
|
|
|
//===--------------------------------------------------------------------===//
|
|
// DeclVisitor
|
|
//===--------------------------------------------------------------------===//
|
|
|
|
void Visit(Decl *D) {
|
|
Decl *PrevDecl = CurrentDecl;
|
|
CurrentDecl = D;
|
|
BaseDeclVisitor::Visit(D);
|
|
CurrentDecl = PrevDecl;
|
|
}
|
|
|
|
void VisitDeclaratorDecl(DeclaratorDecl *D) {
|
|
BaseDeclVisitor::VisitDeclaratorDecl(D);
|
|
if (TypeSourceInfo *TInfo = D->getTypeSourceInfo())
|
|
Visit(TInfo->getTypeLoc());
|
|
}
|
|
|
|
void VisitFunctionDecl(FunctionDecl *D) {
|
|
BaseDeclVisitor::VisitFunctionDecl(D);
|
|
if (D->isThisDeclarationADefinition())
|
|
Visit(D->getBody());
|
|
}
|
|
|
|
void VisitObjCMethodDecl(ObjCMethodDecl *D) {
|
|
BaseDeclVisitor::VisitObjCMethodDecl(D);
|
|
if (D->getBody())
|
|
Visit(D->getBody());
|
|
}
|
|
|
|
void VisitBlockDecl(BlockDecl *D) {
|
|
BaseDeclVisitor::VisitBlockDecl(D);
|
|
Visit(D->getBody());
|
|
}
|
|
|
|
void VisitVarDecl(VarDecl *D) {
|
|
BaseDeclVisitor::VisitVarDecl(D);
|
|
if (Expr *Init = D->getInit())
|
|
Visit(Init);
|
|
}
|
|
|
|
void VisitDecl(Decl *D) {
|
|
if (isa<FunctionDecl>(D) || isa<ObjCMethodDecl>(D) || isa<BlockDecl>(D))
|
|
return;
|
|
|
|
if (DeclContext *DC = dyn_cast<DeclContext>(D))
|
|
static_cast<ImplClass*>(this)->VisitDeclContext(DC);
|
|
}
|
|
|
|
void VisitDeclContext(DeclContext *DC) {
|
|
for (DeclContext::decl_iterator
|
|
I = DC->decls_begin(), E = DC->decls_end(); I != E; ++I)
|
|
Visit(*I);
|
|
}
|
|
|
|
//===--------------------------------------------------------------------===//
|
|
// StmtVisitor
|
|
//===--------------------------------------------------------------------===//
|
|
|
|
void VisitDeclStmt(DeclStmt *Node) {
|
|
for (DeclStmt::decl_iterator
|
|
I = Node->decl_begin(), E = Node->decl_end(); I != E; ++I)
|
|
Visit(*I);
|
|
}
|
|
|
|
void VisitBlockExpr(BlockExpr *Node) {
|
|
// The BlockDecl is also visited by 'VisitDeclContext()'. No need to visit it twice.
|
|
}
|
|
|
|
void VisitStmt(Stmt *Node) {
|
|
for (Stmt::child_range I = Node->children(); I; ++I)
|
|
if (*I)
|
|
Visit(*I);
|
|
}
|
|
|
|
//===--------------------------------------------------------------------===//
|
|
// TypeLocVisitor
|
|
//===--------------------------------------------------------------------===//
|
|
|
|
void Visit(TypeLoc TL) {
|
|
for (; TL; TL = TL.getNextTypeLoc())
|
|
BaseTypeLocVisitor::Visit(TL);
|
|
}
|
|
|
|
void VisitArrayLoc(ArrayTypeLoc TL) {
|
|
BaseTypeLocVisitor::VisitArrayTypeLoc(TL);
|
|
if (TL.getSizeExpr())
|
|
Visit(TL.getSizeExpr());
|
|
}
|
|
|
|
void VisitFunctionTypeLoc(FunctionTypeLoc TL) {
|
|
BaseTypeLocVisitor::VisitFunctionTypeLoc(TL);
|
|
for (unsigned i = 0; i != TL.getNumArgs(); ++i)
|
|
Visit(TL.getArg(i));
|
|
}
|
|
|
|
};
|
|
|
|
} // namespace idx
|
|
|
|
} // namespace clang
|
|
|
|
#endif
|