clang  10.0.0svn
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_REFACTOR_RECURSIVE_SYMBOL_VISITOR_H
16 #define LLVM_CLANG_TOOLING_REFACTOR_RECURSIVE_SYMBOL_VISITOR_H
17 
18 #include "clang/AST/AST.h"
20 #include "clang/Lex/Lexer.h"
21 
22 namespace clang {
23 namespace tooling {
24 
25 /// Traverses the AST and visits the occurrence of each named symbol in the
26 /// given nodes.
27 template <typename T>
29  : public RecursiveASTVisitor<RecursiveSymbolVisitor<T>> {
31 
32 public:
33  RecursiveSymbolVisitor(const SourceManager &SM, const LangOptions &LangOpts)
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 
125 private:
126  const SourceManager &SM;
127  const LangOptions &LangOpts;
128 
129  bool visit(const NamedDecl *ND, SourceLocation BeginLoc,
130  SourceLocation EndLoc) {
131  return static_cast<T *>(this)->visitSymbolOccurrence(
132  ND, SourceRange(BeginLoc, EndLoc));
133  }
134  bool visit(const NamedDecl *ND, SourceLocation Loc) {
135  return visit(ND, Loc, Lexer::getLocForEndOfToken(Loc, 0, SM, LangOpts));
136  }
137 };
138 
139 } // end namespace tooling
140 } // end namespace clang
141 
142 #endif // LLVM_CLANG_TOOLING_REFACTOR_RECURSIVE_SYMBOL_VISITOR_H
NamedDecl * getFoundDecl()
Get the NamedDecl through which this reference occurred.
Definition: Expr.h:1259
bool VisitCXXConstructorDecl(const CXXConstructorDecl *CD)
Traverses the AST and visits the occurrence of each named symbol in the given nodes.
NamedDecl * getDecl() const
The base class of the type hierarchy.
Definition: Type.h:1436
Represent a C++ namespace.
Definition: Decl.h:522
Wrapper for source info for typedefs.
Definition: TypeLoc.h:666
Represents a C++ constructor within a class.
Definition: DeclCXX.h:2365
SourceLocation getLocalBeginLoc() const
Retrieve the location of the beginning of this component of the nested-name-specifier.
bool VisitDeclRefExpr(const DeclRefExpr *Expr)
const Type * getTypePtr() const
Definition: TypeLoc.h:134
Base wrapper for a particular "section" of type source info.
Definition: TypeLoc.h:56
FieldDecl * getField() const
For a field offsetof node, returns the field.
Definition: Expr.h:2202
A C++ nested-name-specifier augmented with source location information.
Represents a member of a struct/union/class.
Definition: Decl.h:2643
RecursiveSymbolVisitor(const SourceManager &SM, const LangOptions &LangOpts)
bool TraverseNestedNameSpecifierLoc(NestedNameSpecifierLoc NNS)
Keeps track of the various options that can be enabled, which controls the dialect of C or C++ that i...
Definition: LangOptions.h:49
SourceLocation getLocalEndLoc() const
Retrieve the location of the end of this component of the nested-name-specifier.
Helper class for OffsetOfExpr.
Definition: Expr.h:2138
A class that does preorder or postorder depth-first traversal on the entire Clang AST and visits each...
SourceLocation getLocation() const
Definition: Expr.h:1230
SourceLocation getBeginLoc() const
Get the begin source location.
Definition: TypeLoc.cpp:189
NamespaceDecl * getAsNamespace() const
Retrieve the namespace stored in this nested name specifier.
DeclAccessPair getFoundDecl() const
Retrieves the declaration found by lookup.
Definition: Expr.h:2898
This represents one expression.
Definition: Expr.h:108
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:778
Kind getKind() const
Determine what kind of offsetof node this is.
Definition: Expr.h:2192
SourceLocation getMemberLoc() const
getMemberLoc - Return the location of the "member", in X->F, it is the location of &#39;F&#39;...
Definition: Expr.h:3000
const OffsetOfNode & getComponent(unsigned Idx) const
Definition: Expr.h:2291
SourceLocation getEndLoc() const LLVM_READONLY
Definition: Expr.h:2225
Encodes a location in the source.
bool VisitOffsetOfExpr(const OffsetOfExpr *S)
bool TraverseNestedNameSpecifierLoc(NestedNameSpecifierLoc NNS)
Recursively visit a C++ nested-name-specifier with location information.
Dataflow Directional Tag Classes.
NestedNameSpecifier * getNestedNameSpecifier() const
Retrieve the nested-name-specifier to which this instance refers.
bool VisitMemberExpr(const MemberExpr *Expr)
QualType getType() const
Get the type for which this source info wrapper provides information.
Definition: TypeLoc.h:130
TypedefNameDecl * getTypedefNameDecl() const
Definition: TypeLoc.h:670
MemberExpr - [C99 6.5.2.3] Structure and Union Members.
Definition: Expr.h:2811
unsigned getNumComponents() const
Definition: Expr.h:2301
A reference to a declared variable, function, enum, etc.
Definition: Expr.h:1146
A trivial tuple used to represent a source range.
This represents a decl that may have a name.
Definition: Decl.h:248
OffsetOfExpr - [C99 7.17] - This represents an expression of the form offsetof(record-type, member-designator).
Definition: Expr.h:2242
bool visitSymbolOccurrence(const NamedDecl *ND, ArrayRef< SourceRange > NameRanges)
This class handles loading and caching of source files into memory.
SourceLocation getLocation() const
Definition: DeclBase.h:429