clang 19.0.0git
RecursiveSymbolVisitor.h
Go to the documentation of this file.
1//===--- RecursiveSymbolVisitor.h - Clang refactoring library -------------===//
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/// \file
10/// A wrapper class around \c RecursiveASTVisitor that visits each
11/// occurrences of a named symbol.
12///
13//===----------------------------------------------------------------------===//
14
15#ifndef LLVM_CLANG_TOOLING_REFACTORING_RECURSIVESYMBOLVISITOR_H
16#define LLVM_CLANG_TOOLING_REFACTORING_RECURSIVESYMBOLVISITOR_H
17
18#include "clang/AST/AST.h"
20#include "clang/Lex/Lexer.h"
21
22namespace clang {
23namespace tooling {
24
25/// Traverses the AST and visits the occurrence of each named symbol in the
26/// given nodes.
27template <typename T>
29 : public RecursiveASTVisitor<RecursiveSymbolVisitor<T>> {
31
32public:
34 : SM(SM), LangOpts(LangOpts) {}
35
37 ArrayRef<SourceRange> NameRanges) {
38 return true;
39 }
40
41 // Declaration visitors:
42
43 bool VisitNamedDecl(const NamedDecl *D) {
44 return isa<CXXConversionDecl>(D) ? true : visit(D, D->getLocation());
45 }
46
48 for (const auto *Initializer : CD->inits()) {
49 // Ignore implicit initializers.
50 if (!Initializer->isWritten())
51 continue;
52 if (const FieldDecl *FD = Initializer->getMember()) {
53 if (!visit(FD, Initializer->getSourceLocation(),
54 Lexer::getLocForEndOfToken(Initializer->getSourceLocation(),
55 0, SM, LangOpts)))
56 return false;
57 }
58 }
59 return true;
60 }
61
62 // Expression visitors:
63
65 return visit(Expr->getFoundDecl(), Expr->getLocation());
66 }
67
69 return visit(Expr->getFoundDecl().getDecl(), Expr->getMemberLoc());
70 }
71
73 for (unsigned I = 0, E = S->getNumComponents(); I != E; ++I) {
74 const OffsetOfNode &Component = S->getComponent(I);
75 if (Component.getKind() == OffsetOfNode::Field) {
76 if (!visit(Component.getField(), Component.getEndLoc()))
77 return false;
78 }
79 // FIXME: Try to resolve dependent field references.
80 }
81 return true;
82 }
83
84 // Other visitors:
85
86 bool VisitTypeLoc(const TypeLoc Loc) {
87 const SourceLocation TypeBeginLoc = Loc.getBeginLoc();
88 const SourceLocation TypeEndLoc =
89 Lexer::getLocForEndOfToken(TypeBeginLoc, 0, SM, LangOpts);
90 if (const auto *TemplateTypeParm =
91 dyn_cast<TemplateTypeParmType>(Loc.getType())) {
92 if (!visit(TemplateTypeParm->getDecl(), TypeBeginLoc, TypeEndLoc))
93 return false;
94 }
95 if (const auto *TemplateSpecType =
96 dyn_cast<TemplateSpecializationType>(Loc.getType())) {
97 if (!visit(TemplateSpecType->getTemplateName().getAsTemplateDecl(),
98 TypeBeginLoc, TypeEndLoc))
99 return false;
100 }
101 if (const Type *TP = Loc.getTypePtr()) {
102 if (TP->getTypeClass() == clang::Type::Record)
103 return visit(TP->getAsCXXRecordDecl(), TypeBeginLoc, TypeEndLoc);
104 }
105 return true;
106 }
107
109 const SourceLocation TypeEndLoc =
110 Lexer::getLocForEndOfToken(TL.getBeginLoc(), 0, SM, LangOpts);
111 return visit(TL.getTypedefNameDecl(), TL.getBeginLoc(), TypeEndLoc);
112 }
113
115 // The base visitor will visit NNSL prefixes, so we should only look at
116 // the current NNS.
117 if (NNS) {
119 if (!visit(ND, NNS.getLocalBeginLoc(), NNS.getLocalEndLoc()))
120 return false;
121 }
123 }
124
126 for (const DesignatedInitExpr::Designator &D : E->designators()) {
127 if (D.isFieldDesignator()) {
128 if (const FieldDecl *Decl = D.getFieldDecl()) {
129 if (!visit(Decl, D.getFieldLoc(), D.getFieldLoc()))
130 return false;
131 }
132 }
133 }
134 return true;
135 }
136
137private:
138 const SourceManager &SM;
139 const LangOptions &LangOpts;
140
141 bool visit(const NamedDecl *ND, SourceLocation BeginLoc,
142 SourceLocation EndLoc) {
143 return static_cast<T *>(this)->visitSymbolOccurrence(
144 ND, SourceRange(BeginLoc, EndLoc));
145 }
146 bool visit(const NamedDecl *ND, SourceLocation Loc) {
147 return visit(ND, Loc, Lexer::getLocForEndOfToken(Loc, 0, SM, LangOpts));
148 }
149};
150
151} // end namespace tooling
152} // end namespace clang
153
154#endif // LLVM_CLANG_TOOLING_REFACTORING_RECURSIVESYMBOLVISITOR_H
#define SM(sm)
Definition: Cuda.cpp:82
Represents a C++ constructor within a class.
Definition: DeclCXX.h:2528
A reference to a declared variable, function, enum, etc.
Definition: Expr.h:1260
Decl - This represents one declaration (or definition), e.g.
Definition: DeclBase.h:85
SourceLocation getLocation() const
Definition: DeclBase.h:444
Represents a single C99 designator.
Definition: Expr.h:5129
Represents a C99 designated initializer expression.
Definition: Expr.h:5086
llvm::MutableArrayRef< Designator > designators()
Definition: Expr.h:5319
This represents one expression.
Definition: Expr.h:110
Represents a member of a struct/union/class.
Definition: Decl.h:3025
Keeps track of the various options that can be enabled, which controls the dialect of C or C++ that i...
Definition: LangOptions.h:418
static SourceLocation getLocForEndOfToken(SourceLocation Loc, unsigned Offset, const SourceManager &SM, const LangOptions &LangOpts)
Computes the source location just past the end of the token at this source location.
Definition: Lexer.cpp:850
MemberExpr - [C99 6.5.2.3] Structure and Union Members.
Definition: Expr.h:3182
This represents a decl that may have a name.
Definition: Decl.h:249
Represent a C++ namespace.
Definition: Decl.h:547
A C++ nested-name-specifier augmented with source location information.
SourceLocation getLocalEndLoc() const
Retrieve the location of the end of this component of the nested-name-specifier.
NestedNameSpecifier * getNestedNameSpecifier() const
Retrieve the nested-name-specifier to which this instance refers.
SourceLocation getLocalBeginLoc() const
Retrieve the location of the beginning of this component of the nested-name-specifier.
NamespaceDecl * getAsNamespace() const
Retrieve the namespace stored in this nested name specifier.
OffsetOfExpr - [C99 7.17] - This represents an expression of the form offsetof(record-type,...
Definition: Expr.h:2464
Helper class for OffsetOfExpr.
Definition: Expr.h:2358
FieldDecl * getField() const
For a field offsetof node, returns the field.
Definition: Expr.h:2422
@ Field
A field.
Definition: Expr.h:2365
Kind getKind() const
Determine what kind of offsetof node this is.
Definition: Expr.h:2412
SourceLocation getEndLoc() const LLVM_READONLY
Definition: Expr.h:2445
A class that does preorder or postorder depth-first traversal on the entire Clang AST and visits each...
bool TraverseNestedNameSpecifierLoc(NestedNameSpecifierLoc NNS)
Recursively visit a C++ nested-name-specifier with location information.
Encodes a location in the source.
This class handles loading and caching of source files into memory.
A trivial tuple used to represent a source range.
Base wrapper for a particular "section" of type source info.
Definition: TypeLoc.h:59
QualType getType() const
Get the type for which this source info wrapper provides information.
Definition: TypeLoc.h:133
SourceLocation getBeginLoc() const
Get the begin source location.
Definition: TypeLoc.cpp:192
const Type * getTypePtr() const
Definition: TypeLoc.h:137
The base class of the type hierarchy.
Definition: Type.h:1606
Wrapper for source info for typedefs.
Definition: TypeLoc.h:693
TypedefNameDecl * getTypedefNameDecl() const
Definition: TypeLoc.h:695
Traverses the AST and visits the occurrence of each named symbol in the given nodes.
bool VisitOffsetOfExpr(const OffsetOfExpr *S)
bool VisitMemberExpr(const MemberExpr *Expr)
bool VisitDesignatedInitExpr(const DesignatedInitExpr *E)
bool VisitDeclRefExpr(const DeclRefExpr *Expr)
RecursiveSymbolVisitor(const SourceManager &SM, const LangOptions &LangOpts)
bool TraverseNestedNameSpecifierLoc(NestedNameSpecifierLoc NNS)
bool VisitCXXConstructorDecl(const CXXConstructorDecl *CD)
bool visitSymbolOccurrence(const NamedDecl *ND, ArrayRef< SourceRange > NameRanges)
The JSON file list parser is used to communicate input to InstallAPI.
#define true
Definition: stdbool.h:21