File: | build/source/clang-tools-extra/clangd/refactor/tweaks/ObjCMemberwiseInitializer.cpp |
Warning: | line 110, column 27 Called C++ object pointer is null |
Press '?' to see keyboard shortcuts
Keyboard shortcuts:
1 | //===--- ObjCMemberwiseInitializer.cpp ---------------------------*- C++-*-===// | ||||
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 | #include "ParsedAST.h" | ||||
10 | #include "SourceCode.h" | ||||
11 | #include "refactor/InsertionPoint.h" | ||||
12 | #include "refactor/Tweak.h" | ||||
13 | #include "support/Logger.h" | ||||
14 | #include "clang/AST/DeclObjC.h" | ||||
15 | #include "clang/AST/PrettyPrinter.h" | ||||
16 | #include "clang/Basic/LLVM.h" | ||||
17 | #include "clang/Basic/LangOptions.h" | ||||
18 | #include "clang/Basic/SourceLocation.h" | ||||
19 | #include "clang/Basic/SourceManager.h" | ||||
20 | #include "clang/Tooling/Core/Replacement.h" | ||||
21 | #include "llvm/ADT/StringRef.h" | ||||
22 | #include "llvm/ADT/iterator_range.h" | ||||
23 | #include "llvm/Support/Casting.h" | ||||
24 | #include "llvm/Support/Error.h" | ||||
25 | #include <optional> | ||||
26 | |||||
27 | namespace clang { | ||||
28 | namespace clangd { | ||||
29 | namespace { | ||||
30 | |||||
31 | static std::string capitalize(std::string Message) { | ||||
32 | if (!Message.empty()) | ||||
33 | Message[0] = llvm::toUpper(Message[0]); | ||||
34 | return Message; | ||||
35 | } | ||||
36 | |||||
37 | static std::string getTypeStr(const QualType &OrigT, const Decl &D, | ||||
38 | unsigned PropertyAttributes) { | ||||
39 | QualType T = OrigT; | ||||
40 | PrintingPolicy Policy(D.getASTContext().getLangOpts()); | ||||
41 | Policy.SuppressStrongLifetime = true; | ||||
42 | std::string Prefix; | ||||
43 | // If the nullability is specified via a property attribute, use the shorter | ||||
44 | // `nullable` form for the method parameter. | ||||
45 | if (PropertyAttributes & ObjCPropertyAttribute::kind_nullability) { | ||||
46 | if (auto Kind = AttributedType::stripOuterNullability(T)) { | ||||
47 | switch (*Kind) { | ||||
48 | case NullabilityKind::Nullable: | ||||
49 | Prefix = "nullable "; | ||||
50 | break; | ||||
51 | case NullabilityKind::NonNull: | ||||
52 | Prefix = "nonnull "; | ||||
53 | break; | ||||
54 | case NullabilityKind::Unspecified: | ||||
55 | Prefix = "null_unspecified "; | ||||
56 | break; | ||||
57 | case NullabilityKind::NullableResult: | ||||
58 | T = OrigT; | ||||
59 | break; | ||||
60 | } | ||||
61 | } | ||||
62 | } | ||||
63 | return Prefix + T.getAsString(Policy); | ||||
64 | } | ||||
65 | |||||
66 | struct MethodParameter { | ||||
67 | // Parameter name. | ||||
68 | llvm::StringRef Name; | ||||
69 | |||||
70 | // Type of the parameter. | ||||
71 | std::string Type; | ||||
72 | |||||
73 | // Assignment target (LHS). | ||||
74 | std::string Assignee; | ||||
75 | |||||
76 | MethodParameter(const ObjCIvarDecl &ID) { | ||||
77 | // Convention maps `@property int foo` to ivar `int _foo`, so drop the | ||||
78 | // leading `_` if there is one. | ||||
79 | Name = ID.getName(); | ||||
80 | Name.consume_front("_"); | ||||
81 | Type = getTypeStr(ID.getType(), ID, ObjCPropertyAttribute::kind_noattr); | ||||
82 | Assignee = ID.getName().str(); | ||||
83 | } | ||||
84 | MethodParameter(const ObjCPropertyDecl &PD) { | ||||
85 | Name = PD.getName(); | ||||
86 | Type = getTypeStr(PD.getType(), PD, PD.getPropertyAttributes()); | ||||
87 | if (const auto *ID = PD.getPropertyIvarDecl()) | ||||
88 | Assignee = ID->getName().str(); | ||||
89 | else // Could be a dynamic property or a property in a header. | ||||
90 | Assignee = ("self." + Name).str(); | ||||
91 | } | ||||
92 | static std::optional<MethodParameter> parameterFor(const Decl &D) { | ||||
93 | if (const auto *ID = dyn_cast<ObjCIvarDecl>(&D)) | ||||
94 | return MethodParameter(*ID); | ||||
95 | if (const auto *PD = dyn_cast<ObjCPropertyDecl>(&D)) | ||||
96 | if (PD->isInstanceProperty()) | ||||
97 | return MethodParameter(*PD); | ||||
98 | return std::nullopt; | ||||
99 | } | ||||
100 | }; | ||||
101 | |||||
102 | static SmallVector<MethodParameter, 8> | ||||
103 | getAllParams(const ObjCInterfaceDecl *ID) { | ||||
104 | SmallVector<MethodParameter, 8> Params; | ||||
105 | // Currently we only generate based on the ivars and properties declared | ||||
106 | // in the interface. We could consider expanding this to include visible | ||||
107 | // categories + class extensions in the future (see | ||||
108 | // all_declared_ivar_begin). | ||||
109 | llvm::DenseSet<llvm::StringRef> Names; | ||||
110 | for (const auto *Ivar : ID->ivars()) { | ||||
| |||||
111 | MethodParameter P(*Ivar); | ||||
112 | if (Names.insert(P.Name).second) | ||||
113 | Params.push_back(P); | ||||
114 | } | ||||
115 | for (const auto *Prop : ID->properties()) { | ||||
116 | MethodParameter P(*Prop); | ||||
117 | if (Names.insert(P.Name).second) | ||||
118 | Params.push_back(P); | ||||
119 | } | ||||
120 | return Params; | ||||
121 | } | ||||
122 | |||||
123 | static std::string | ||||
124 | initializerForParams(const SmallVector<MethodParameter, 8> &Params, | ||||
125 | bool GenerateImpl) { | ||||
126 | std::string Code; | ||||
127 | llvm::raw_string_ostream Stream(Code); | ||||
128 | |||||
129 | if (Params.empty()) { | ||||
130 | if (GenerateImpl) { | ||||
131 | Stream << | ||||
132 | R"cpp(- (instancetype)init { | ||||
133 | self = [super init]; | ||||
134 | if (self) { | ||||
135 | |||||
136 | } | ||||
137 | return self; | ||||
138 | })cpp"; | ||||
139 | } else { | ||||
140 | Stream << "- (instancetype)init;"; | ||||
141 | } | ||||
142 | } else { | ||||
143 | const auto &First = Params.front(); | ||||
144 | Stream << llvm::formatv("- (instancetype)initWith{0}:({1}){2}", | ||||
145 | capitalize(First.Name.trim().str()), First.Type, | ||||
146 | First.Name); | ||||
147 | for (const auto &It : llvm::drop_begin(Params)) | ||||
148 | Stream << llvm::formatv(" {0}:({1}){0}", It.Name, It.Type); | ||||
149 | |||||
150 | if (GenerateImpl) { | ||||
151 | Stream << | ||||
152 | R"cpp( { | ||||
153 | self = [super init]; | ||||
154 | if (self) {)cpp"; | ||||
155 | for (const auto &Param : Params) | ||||
156 | Stream << llvm::formatv("\n {0} = {1};", Param.Assignee, Param.Name); | ||||
157 | Stream << | ||||
158 | R"cpp( | ||||
159 | } | ||||
160 | return self; | ||||
161 | })cpp"; | ||||
162 | } else { | ||||
163 | Stream << ";"; | ||||
164 | } | ||||
165 | } | ||||
166 | Stream << "\n\n"; | ||||
167 | return Code; | ||||
168 | } | ||||
169 | |||||
170 | /// Generate an initializer for an Objective-C class based on selected | ||||
171 | /// properties and instance variables. | ||||
172 | class ObjCMemberwiseInitializer : public Tweak { | ||||
173 | public: | ||||
174 | const char *id() const final; | ||||
175 | llvm::StringLiteral kind() const override { | ||||
176 | return CodeAction::REFACTOR_KIND; | ||||
177 | } | ||||
178 | |||||
179 | bool prepare(const Selection &Inputs) override; | ||||
180 | Expected<Tweak::Effect> apply(const Selection &Inputs) override; | ||||
181 | std::string title() const override; | ||||
182 | |||||
183 | private: | ||||
184 | SmallVector<MethodParameter, 8> | ||||
185 | paramsForSelection(const SelectionTree::Node *N); | ||||
186 | |||||
187 | const ObjCInterfaceDecl *Interface = nullptr; | ||||
188 | |||||
189 | // Will be nullptr if running on an interface. | ||||
190 | const ObjCImplementationDecl *Impl = nullptr; | ||||
191 | }; | ||||
192 | |||||
193 | REGISTER_TWEAK(ObjCMemberwiseInitializer)::llvm::Registry<::clang::clangd::Tweak>::Add<ObjCMemberwiseInitializer > TweakRegistrationForObjCMemberwiseInitializer("ObjCMemberwiseInitializer" , ""); const char *ObjCMemberwiseInitializer::id() const { return "ObjCMemberwiseInitializer"; } | ||||
194 | |||||
195 | bool ObjCMemberwiseInitializer::prepare(const Selection &Inputs) { | ||||
196 | const SelectionTree::Node *N = Inputs.ASTSelection.commonAncestor(); | ||||
197 | if (!N) | ||||
198 | return false; | ||||
199 | const Decl *D = N->ASTNode.get<Decl>(); | ||||
200 | if (!D) | ||||
201 | return false; | ||||
202 | const auto &LangOpts = Inputs.AST->getLangOpts(); | ||||
203 | // Require ObjC w/ arc enabled since we don't emit retains. | ||||
204 | if (!LangOpts.ObjC || !LangOpts.ObjCAutoRefCount) | ||||
205 | return false; | ||||
206 | |||||
207 | // We support the following selected decls: | ||||
208 | // - ObjCInterfaceDecl/ObjCImplementationDecl only - generate for all | ||||
209 | // properties and ivars | ||||
210 | // | ||||
211 | // - Specific ObjCPropertyDecl(s)/ObjCIvarDecl(s) - generate only for those | ||||
212 | // selected. Note that if only one is selected, the common ancestor will be | ||||
213 | // the ObjCPropertyDecl/ObjCIvarDecl itself instead of the container. | ||||
214 | if (const auto *ID = dyn_cast<ObjCInterfaceDecl>(D)) { | ||||
215 | // Ignore forward declarations (@class Name;). | ||||
216 | if (!ID->isThisDeclarationADefinition()) | ||||
217 | return false; | ||||
218 | Interface = ID; | ||||
219 | } else if (const auto *ID = dyn_cast<ObjCImplementationDecl>(D)) { | ||||
220 | Interface = ID->getClassInterface(); | ||||
221 | Impl = ID; | ||||
222 | } else if (isa<ObjCPropertyDecl, ObjCIvarDecl>(D)) { | ||||
223 | const auto *DC = D->getDeclContext(); | ||||
224 | if (const auto *ID = dyn_cast<ObjCInterfaceDecl>(DC)) { | ||||
225 | Interface = ID; | ||||
226 | } else if (const auto *ID = dyn_cast<ObjCImplementationDecl>(DC)) { | ||||
227 | Interface = ID->getClassInterface(); | ||||
228 | Impl = ID; | ||||
229 | } | ||||
230 | } | ||||
231 | return Interface != nullptr; | ||||
232 | } | ||||
233 | |||||
234 | SmallVector<MethodParameter, 8> | ||||
235 | ObjCMemberwiseInitializer::paramsForSelection(const SelectionTree::Node *N) { | ||||
236 | SmallVector<MethodParameter, 8> Params; | ||||
237 | // Base case: selected a single ivar or property. | ||||
238 | if (const auto *D
| ||||
239 | if (auto Param = MethodParameter::parameterFor(*D)) { | ||||
240 | Params.push_back(*Param); | ||||
241 | return Params; | ||||
242 | } | ||||
243 | } | ||||
244 | const ObjCContainerDecl *Container = | ||||
245 | Impl ? static_cast<const ObjCContainerDecl *>(Impl) | ||||
246 | : static_cast<const ObjCContainerDecl *>(Interface); | ||||
247 | if (Container == N->ASTNode.get<ObjCContainerDecl>() && N->Children.empty()) | ||||
248 | return getAllParams(Interface); | ||||
249 | |||||
250 | llvm::DenseSet<llvm::StringRef> Names; | ||||
251 | // Check for selecting multiple ivars/properties. | ||||
252 | for (const auto *CNode : N->Children) { | ||||
253 | const Decl *D = CNode->ASTNode.get<Decl>(); | ||||
254 | if (!D) | ||||
255 | continue; | ||||
256 | if (auto P = MethodParameter::parameterFor(*D)) | ||||
257 | if (Names.insert(P->Name).second) | ||||
258 | Params.push_back(*P); | ||||
259 | } | ||||
260 | return Params; | ||||
261 | } | ||||
262 | |||||
263 | Expected<Tweak::Effect> | ||||
264 | ObjCMemberwiseInitializer::apply(const Selection &Inputs) { | ||||
265 | const auto &SM = Inputs.AST->getASTContext().getSourceManager(); | ||||
266 | const SelectionTree::Node *N = Inputs.ASTSelection.commonAncestor(); | ||||
267 | if (!N) | ||||
| |||||
268 | return error("Invalid selection"); | ||||
269 | |||||
270 | SmallVector<MethodParameter, 8> Params = paramsForSelection(N); | ||||
271 | |||||
272 | // Insert before the first non-init instance method. | ||||
273 | std::vector<Anchor> Anchors = { | ||||
274 | {[](const Decl *D) { | ||||
275 | if (const auto *MD = llvm::dyn_cast<ObjCMethodDecl>(D)) { | ||||
276 | return MD->getMethodFamily() != OMF_init && MD->isInstanceMethod(); | ||||
277 | } | ||||
278 | return false; | ||||
279 | }, | ||||
280 | Anchor::Above}}; | ||||
281 | Effect E; | ||||
282 | |||||
283 | auto InterfaceReplacement = | ||||
284 | insertDecl(initializerForParams(Params, /*GenerateImpl=*/false), | ||||
285 | *Interface, Anchors); | ||||
286 | if (!InterfaceReplacement) | ||||
287 | return InterfaceReplacement.takeError(); | ||||
288 | auto FE = Effect::fileEdit(SM, SM.getFileID(Interface->getLocation()), | ||||
289 | tooling::Replacements(*InterfaceReplacement)); | ||||
290 | if (!FE) | ||||
291 | return FE.takeError(); | ||||
292 | E.ApplyEdits.insert(std::move(*FE)); | ||||
293 | |||||
294 | if (Impl) { | ||||
295 | // If we see the class implementation, add the initializer there too. | ||||
296 | // FIXME: merging the edits is awkward, do this elsewhere. | ||||
297 | auto ImplReplacement = insertDecl( | ||||
298 | initializerForParams(Params, /*GenerateImpl=*/true), *Impl, Anchors); | ||||
299 | if (!ImplReplacement) | ||||
300 | return ImplReplacement.takeError(); | ||||
301 | |||||
302 | if (SM.isWrittenInSameFile(Interface->getLocation(), Impl->getLocation())) { | ||||
303 | // Merge with previous edit if they are in the same file. | ||||
304 | if (auto Err = | ||||
305 | E.ApplyEdits.begin()->second.Replacements.add(*ImplReplacement)) | ||||
306 | return std::move(Err); | ||||
307 | } else { | ||||
308 | // Generate a new edit if the interface and implementation are in | ||||
309 | // different files. | ||||
310 | auto FE = Effect::fileEdit(SM, SM.getFileID(Impl->getLocation()), | ||||
311 | tooling::Replacements(*ImplReplacement)); | ||||
312 | if (!FE) | ||||
313 | return FE.takeError(); | ||||
314 | E.ApplyEdits.insert(std::move(*FE)); | ||||
315 | } | ||||
316 | } | ||||
317 | return E; | ||||
318 | } | ||||
319 | |||||
320 | std::string ObjCMemberwiseInitializer::title() const { | ||||
321 | if (Impl) | ||||
322 | return "Generate memberwise initializer"; | ||||
323 | return "Declare memberwise initializer"; | ||||
324 | } | ||||
325 | |||||
326 | } // namespace | ||||
327 | } // namespace clangd | ||||
328 | } // namespace clang |
1 | //===--- ASTTypeTraits.h ----------------------------------------*- C++ -*-===// |
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 | // Provides a dynamic type identifier and a dynamically typed node container |
10 | // that can be used to store an AST base node at runtime in the same storage in |
11 | // a type safe way. |
12 | // |
13 | //===----------------------------------------------------------------------===// |
14 | |
15 | #ifndef LLVM_CLANG_AST_ASTTYPETRAITS_H |
16 | #define LLVM_CLANG_AST_ASTTYPETRAITS_H |
17 | |
18 | #include "clang/AST/ASTFwd.h" |
19 | #include "clang/AST/DeclCXX.h" |
20 | #include "clang/AST/LambdaCapture.h" |
21 | #include "clang/AST/NestedNameSpecifier.h" |
22 | #include "clang/AST/TemplateBase.h" |
23 | #include "clang/AST/TypeLoc.h" |
24 | #include "clang/Basic/LLVM.h" |
25 | #include "llvm/ADT/DenseMapInfo.h" |
26 | #include "llvm/Support/AlignOf.h" |
27 | |
28 | namespace llvm { |
29 | class raw_ostream; |
30 | } // namespace llvm |
31 | |
32 | namespace clang { |
33 | |
34 | struct PrintingPolicy; |
35 | |
36 | /// Defines how we descend a level in the AST when we pass |
37 | /// through expressions. |
38 | enum TraversalKind { |
39 | /// Will traverse all child nodes. |
40 | TK_AsIs, |
41 | |
42 | /// Ignore AST nodes not written in the source |
43 | TK_IgnoreUnlessSpelledInSource |
44 | }; |
45 | |
46 | /// Kind identifier. |
47 | /// |
48 | /// It can be constructed from any node kind and allows for runtime type |
49 | /// hierarchy checks. |
50 | /// Use getFromNodeKind<T>() to construct them. |
51 | class ASTNodeKind { |
52 | public: |
53 | /// Empty identifier. It matches nothing. |
54 | constexpr ASTNodeKind() : KindId(NKI_None) {} |
55 | |
56 | /// Construct an identifier for T. |
57 | template <class T> static constexpr ASTNodeKind getFromNodeKind() { |
58 | return ASTNodeKind(KindToKindId<T>::Id); |
59 | } |
60 | |
61 | /// \{ |
62 | /// Construct an identifier for the dynamic type of the node |
63 | static ASTNodeKind getFromNode(const Decl &D); |
64 | static ASTNodeKind getFromNode(const Stmt &S); |
65 | static ASTNodeKind getFromNode(const Type &T); |
66 | static ASTNodeKind getFromNode(const TypeLoc &T); |
67 | static ASTNodeKind getFromNode(const LambdaCapture &L); |
68 | static ASTNodeKind getFromNode(const OMPClause &C); |
69 | static ASTNodeKind getFromNode(const Attr &A); |
70 | /// \} |
71 | |
72 | /// Returns \c true if \c this and \c Other represent the same kind. |
73 | constexpr bool isSame(ASTNodeKind Other) const { |
74 | return KindId != NKI_None && KindId == Other.KindId; |
75 | } |
76 | |
77 | /// Returns \c true only for the default \c ASTNodeKind() |
78 | constexpr bool isNone() const { return KindId == NKI_None; } |
79 | |
80 | /// Returns \c true if \c this is a base kind of (or same as) \c Other. |
81 | /// \param Distance If non-null, used to return the distance between \c this |
82 | /// and \c Other in the class hierarchy. |
83 | bool isBaseOf(ASTNodeKind Other, unsigned *Distance = nullptr) const; |
84 | |
85 | /// String representation of the kind. |
86 | StringRef asStringRef() const; |
87 | |
88 | /// Strict weak ordering for ASTNodeKind. |
89 | constexpr bool operator<(const ASTNodeKind &Other) const { |
90 | return KindId < Other.KindId; |
91 | } |
92 | |
93 | /// Return the most derived type between \p Kind1 and \p Kind2. |
94 | /// |
95 | /// Return ASTNodeKind() if they are not related. |
96 | static ASTNodeKind getMostDerivedType(ASTNodeKind Kind1, ASTNodeKind Kind2); |
97 | |
98 | /// Return the most derived common ancestor between Kind1 and Kind2. |
99 | /// |
100 | /// Return ASTNodeKind() if they are not related. |
101 | static ASTNodeKind getMostDerivedCommonAncestor(ASTNodeKind Kind1, |
102 | ASTNodeKind Kind2); |
103 | |
104 | ASTNodeKind getCladeKind() const; |
105 | |
106 | /// Hooks for using ASTNodeKind as a key in a DenseMap. |
107 | struct DenseMapInfo { |
108 | // ASTNodeKind() is a good empty key because it is represented as a 0. |
109 | static inline ASTNodeKind getEmptyKey() { return ASTNodeKind(); } |
110 | // NKI_NumberOfKinds is not a valid value, so it is good for a |
111 | // tombstone key. |
112 | static inline ASTNodeKind getTombstoneKey() { |
113 | return ASTNodeKind(NKI_NumberOfKinds); |
114 | } |
115 | static unsigned getHashValue(const ASTNodeKind &Val) { return Val.KindId; } |
116 | static bool isEqual(const ASTNodeKind &LHS, const ASTNodeKind &RHS) { |
117 | return LHS.KindId == RHS.KindId; |
118 | } |
119 | }; |
120 | |
121 | /// Check if the given ASTNodeKind identifies a type that offers pointer |
122 | /// identity. This is useful for the fast path in DynTypedNode. |
123 | constexpr bool hasPointerIdentity() const { |
124 | return KindId > NKI_LastKindWithoutPointerIdentity; |
125 | } |
126 | |
127 | private: |
128 | /// Kind ids. |
129 | /// |
130 | /// Includes all possible base and derived kinds. |
131 | enum NodeKindId { |
132 | NKI_None, |
133 | NKI_TemplateArgument, |
134 | NKI_TemplateArgumentLoc, |
135 | NKI_LambdaCapture, |
136 | NKI_TemplateName, |
137 | NKI_NestedNameSpecifierLoc, |
138 | NKI_QualType, |
139 | #define TYPELOC(CLASS, PARENT) NKI_##CLASS##TypeLoc, |
140 | #include "clang/AST/TypeLocNodes.def" |
141 | NKI_TypeLoc, |
142 | NKI_LastKindWithoutPointerIdentity = NKI_TypeLoc, |
143 | NKI_CXXBaseSpecifier, |
144 | NKI_CXXCtorInitializer, |
145 | NKI_NestedNameSpecifier, |
146 | NKI_Decl, |
147 | #define DECL(DERIVED, BASE) NKI_##DERIVED##Decl, |
148 | #include "clang/AST/DeclNodes.inc" |
149 | NKI_Stmt, |
150 | #define STMT(DERIVED, BASE) NKI_##DERIVED, |
151 | #include "clang/AST/StmtNodes.inc" |
152 | NKI_Type, |
153 | #define TYPE(DERIVED, BASE) NKI_##DERIVED##Type, |
154 | #include "clang/AST/TypeNodes.inc" |
155 | NKI_OMPClause, |
156 | #define GEN_CLANG_CLAUSE_CLASS |
157 | #define CLAUSE_CLASS(Enum, Str, Class) NKI_##Class, |
158 | #include "llvm/Frontend/OpenMP/OMP.inc" |
159 | NKI_Attr, |
160 | #define ATTR(A) NKI_##A##Attr, |
161 | #include "clang/Basic/AttrList.inc" |
162 | NKI_ObjCProtocolLoc, |
163 | NKI_NumberOfKinds |
164 | }; |
165 | |
166 | /// Use getFromNodeKind<T>() to construct the kind. |
167 | constexpr ASTNodeKind(NodeKindId KindId) : KindId(KindId) {} |
168 | |
169 | /// Returns \c true if \c Base is a base kind of (or same as) \c |
170 | /// Derived. |
171 | /// \param Distance If non-null, used to return the distance between \c Base |
172 | /// and \c Derived in the class hierarchy. |
173 | static bool isBaseOf(NodeKindId Base, NodeKindId Derived, unsigned *Distance); |
174 | |
175 | /// Helper meta-function to convert a kind T to its enum value. |
176 | /// |
177 | /// This struct is specialized below for all known kinds. |
178 | template <class T> struct KindToKindId { |
179 | static const NodeKindId Id = NKI_None; |
180 | }; |
181 | template <class T> |
182 | struct KindToKindId<const T> : KindToKindId<T> {}; |
183 | |
184 | /// Per kind info. |
185 | struct KindInfo { |
186 | /// The id of the parent kind, or None if it has no parent. |
187 | NodeKindId ParentId; |
188 | /// Name of the kind. |
189 | const char *Name; |
190 | }; |
191 | static const KindInfo AllKindInfo[NKI_NumberOfKinds]; |
192 | |
193 | NodeKindId KindId; |
194 | }; |
195 | |
196 | #define KIND_TO_KIND_ID(Class) \ |
197 | template <> struct ASTNodeKind::KindToKindId<Class> { \ |
198 | static const NodeKindId Id = NKI_##Class; \ |
199 | }; |
200 | KIND_TO_KIND_ID(CXXCtorInitializer) |
201 | KIND_TO_KIND_ID(TemplateArgument) |
202 | KIND_TO_KIND_ID(TemplateArgumentLoc) |
203 | KIND_TO_KIND_ID(LambdaCapture) |
204 | KIND_TO_KIND_ID(TemplateName) |
205 | KIND_TO_KIND_ID(NestedNameSpecifier) |
206 | KIND_TO_KIND_ID(NestedNameSpecifierLoc) |
207 | KIND_TO_KIND_ID(QualType) |
208 | #define TYPELOC(CLASS, PARENT) KIND_TO_KIND_ID(CLASS##TypeLoc) |
209 | #include "clang/AST/TypeLocNodes.def" |
210 | KIND_TO_KIND_ID(TypeLoc) |
211 | KIND_TO_KIND_ID(Decl) |
212 | KIND_TO_KIND_ID(Stmt) |
213 | KIND_TO_KIND_ID(Type) |
214 | KIND_TO_KIND_ID(OMPClause) |
215 | KIND_TO_KIND_ID(Attr) |
216 | KIND_TO_KIND_ID(ObjCProtocolLoc) |
217 | KIND_TO_KIND_ID(CXXBaseSpecifier) |
218 | #define DECL(DERIVED, BASE) KIND_TO_KIND_ID(DERIVED##Decl) |
219 | #include "clang/AST/DeclNodes.inc" |
220 | #define STMT(DERIVED, BASE) KIND_TO_KIND_ID(DERIVED) |
221 | #include "clang/AST/StmtNodes.inc" |
222 | #define TYPE(DERIVED, BASE) KIND_TO_KIND_ID(DERIVED##Type) |
223 | #include "clang/AST/TypeNodes.inc" |
224 | #define GEN_CLANG_CLAUSE_CLASS |
225 | #define CLAUSE_CLASS(Enum, Str, Class) KIND_TO_KIND_ID(Class) |
226 | #include "llvm/Frontend/OpenMP/OMP.inc" |
227 | #define ATTR(A) KIND_TO_KIND_ID(A##Attr) |
228 | #include "clang/Basic/AttrList.inc" |
229 | #undef KIND_TO_KIND_ID |
230 | |
231 | inline raw_ostream &operator<<(raw_ostream &OS, ASTNodeKind K) { |
232 | OS << K.asStringRef(); |
233 | return OS; |
234 | } |
235 | |
236 | /// A dynamically typed AST node container. |
237 | /// |
238 | /// Stores an AST node in a type safe way. This allows writing code that |
239 | /// works with different kinds of AST nodes, despite the fact that they don't |
240 | /// have a common base class. |
241 | /// |
242 | /// Use \c create(Node) to create a \c DynTypedNode from an AST node, |
243 | /// and \c get<T>() to retrieve the node as type T if the types match. |
244 | /// |
245 | /// See \c ASTNodeKind for which node base types are currently supported; |
246 | /// You can create DynTypedNodes for all nodes in the inheritance hierarchy of |
247 | /// the supported base types. |
248 | class DynTypedNode { |
249 | public: |
250 | /// Creates a \c DynTypedNode from \c Node. |
251 | template <typename T> |
252 | static DynTypedNode create(const T &Node) { |
253 | return BaseConverter<T>::create(Node); |
254 | } |
255 | |
256 | /// Retrieve the stored node as type \c T. |
257 | /// |
258 | /// Returns NULL if the stored node does not have a type that is |
259 | /// convertible to \c T. |
260 | /// |
261 | /// For types that have identity via their pointer in the AST |
262 | /// (like \c Stmt, \c Decl, \c Type and \c NestedNameSpecifier) the returned |
263 | /// pointer points to the referenced AST node. |
264 | /// For other types (like \c QualType) the value is stored directly |
265 | /// in the \c DynTypedNode, and the returned pointer points at |
266 | /// the storage inside DynTypedNode. For those nodes, do not |
267 | /// use the pointer outside the scope of the DynTypedNode. |
268 | template <typename T> const T *get() const { |
269 | return BaseConverter<T>::get(NodeKind, &Storage); |
270 | } |
271 | |
272 | /// Retrieve the stored node as type \c T. |
273 | /// |
274 | /// Similar to \c get(), but asserts that the type is what we are expecting. |
275 | template <typename T> |
276 | const T &getUnchecked() const { |
277 | return BaseConverter<T>::getUnchecked(NodeKind, &Storage); |
278 | } |
279 | |
280 | ASTNodeKind getNodeKind() const { return NodeKind; } |
281 | |
282 | /// Returns a pointer that identifies the stored AST node. |
283 | /// |
284 | /// Note that this is not supported by all AST nodes. For AST nodes |
285 | /// that don't have a pointer-defined identity inside the AST, this |
286 | /// method returns NULL. |
287 | const void *getMemoizationData() const { |
288 | return NodeKind.hasPointerIdentity() |
289 | ? *reinterpret_cast<void *const *>(&Storage) |
290 | : nullptr; |
291 | } |
292 | |
293 | /// Prints the node to the given output stream. |
294 | void print(llvm::raw_ostream &OS, const PrintingPolicy &PP) const; |
295 | |
296 | /// Dumps the node to the given output stream. |
297 | void dump(llvm::raw_ostream &OS, const ASTContext &Context) const; |
298 | |
299 | /// For nodes which represent textual entities in the source code, |
300 | /// return their SourceRange. For all other nodes, return SourceRange(). |
301 | SourceRange getSourceRange() const; |
302 | |
303 | /// @{ |
304 | /// Imposes an order on \c DynTypedNode. |
305 | /// |
306 | /// Supports comparison of nodes that support memoization. |
307 | /// FIXME: Implement comparison for other node types (currently |
308 | /// only Stmt, Decl, Type and NestedNameSpecifier return memoization data). |
309 | bool operator<(const DynTypedNode &Other) const { |
310 | if (!NodeKind.isSame(Other.NodeKind)) |
311 | return NodeKind < Other.NodeKind; |
312 | |
313 | if (ASTNodeKind::getFromNodeKind<QualType>().isSame(NodeKind)) |
314 | return getUnchecked<QualType>().getAsOpaquePtr() < |
315 | Other.getUnchecked<QualType>().getAsOpaquePtr(); |
316 | |
317 | if (ASTNodeKind::getFromNodeKind<TypeLoc>().isBaseOf(NodeKind)) { |
318 | auto TLA = getUnchecked<TypeLoc>(); |
319 | auto TLB = Other.getUnchecked<TypeLoc>(); |
320 | return std::make_pair(TLA.getType().getAsOpaquePtr(), |
321 | TLA.getOpaqueData()) < |
322 | std::make_pair(TLB.getType().getAsOpaquePtr(), |
323 | TLB.getOpaqueData()); |
324 | } |
325 | |
326 | if (ASTNodeKind::getFromNodeKind<NestedNameSpecifierLoc>().isSame( |
327 | NodeKind)) { |
328 | auto NNSLA = getUnchecked<NestedNameSpecifierLoc>(); |
329 | auto NNSLB = Other.getUnchecked<NestedNameSpecifierLoc>(); |
330 | return std::make_pair(NNSLA.getNestedNameSpecifier(), |
331 | NNSLA.getOpaqueData()) < |
332 | std::make_pair(NNSLB.getNestedNameSpecifier(), |
333 | NNSLB.getOpaqueData()); |
334 | } |
335 | |
336 | assert(getMemoizationData() && Other.getMemoizationData())(static_cast <bool> (getMemoizationData() && Other .getMemoizationData()) ? void (0) : __assert_fail ("getMemoizationData() && Other.getMemoizationData()" , "clang/include/clang/AST/ASTTypeTraits.h", 336, __extension__ __PRETTY_FUNCTION__)); |
337 | return getMemoizationData() < Other.getMemoizationData(); |
338 | } |
339 | bool operator==(const DynTypedNode &Other) const { |
340 | // DynTypedNode::create() stores the exact kind of the node in NodeKind. |
341 | // If they contain the same node, their NodeKind must be the same. |
342 | if (!NodeKind.isSame(Other.NodeKind)) |
343 | return false; |
344 | |
345 | // FIXME: Implement for other types. |
346 | if (ASTNodeKind::getFromNodeKind<QualType>().isSame(NodeKind)) |
347 | return getUnchecked<QualType>() == Other.getUnchecked<QualType>(); |
348 | |
349 | if (ASTNodeKind::getFromNodeKind<TypeLoc>().isBaseOf(NodeKind)) |
350 | return getUnchecked<TypeLoc>() == Other.getUnchecked<TypeLoc>(); |
351 | |
352 | if (ASTNodeKind::getFromNodeKind<NestedNameSpecifierLoc>().isSame(NodeKind)) |
353 | return getUnchecked<NestedNameSpecifierLoc>() == |
354 | Other.getUnchecked<NestedNameSpecifierLoc>(); |
355 | |
356 | assert(getMemoizationData() && Other.getMemoizationData())(static_cast <bool> (getMemoizationData() && Other .getMemoizationData()) ? void (0) : __assert_fail ("getMemoizationData() && Other.getMemoizationData()" , "clang/include/clang/AST/ASTTypeTraits.h", 356, __extension__ __PRETTY_FUNCTION__)); |
357 | return getMemoizationData() == Other.getMemoizationData(); |
358 | } |
359 | bool operator!=(const DynTypedNode &Other) const { |
360 | return !operator==(Other); |
361 | } |
362 | /// @} |
363 | |
364 | /// Hooks for using DynTypedNode as a key in a DenseMap. |
365 | struct DenseMapInfo { |
366 | static inline DynTypedNode getEmptyKey() { |
367 | DynTypedNode Node; |
368 | Node.NodeKind = ASTNodeKind::DenseMapInfo::getEmptyKey(); |
369 | return Node; |
370 | } |
371 | static inline DynTypedNode getTombstoneKey() { |
372 | DynTypedNode Node; |
373 | Node.NodeKind = ASTNodeKind::DenseMapInfo::getTombstoneKey(); |
374 | return Node; |
375 | } |
376 | static unsigned getHashValue(const DynTypedNode &Val) { |
377 | // FIXME: Add hashing support for the remaining types. |
378 | if (ASTNodeKind::getFromNodeKind<TypeLoc>().isBaseOf(Val.NodeKind)) { |
379 | auto TL = Val.getUnchecked<TypeLoc>(); |
380 | return llvm::hash_combine(TL.getType().getAsOpaquePtr(), |
381 | TL.getOpaqueData()); |
382 | } |
383 | |
384 | if (ASTNodeKind::getFromNodeKind<NestedNameSpecifierLoc>().isSame( |
385 | Val.NodeKind)) { |
386 | auto NNSL = Val.getUnchecked<NestedNameSpecifierLoc>(); |
387 | return llvm::hash_combine(NNSL.getNestedNameSpecifier(), |
388 | NNSL.getOpaqueData()); |
389 | } |
390 | |
391 | assert(Val.getMemoizationData())(static_cast <bool> (Val.getMemoizationData()) ? void ( 0) : __assert_fail ("Val.getMemoizationData()", "clang/include/clang/AST/ASTTypeTraits.h" , 391, __extension__ __PRETTY_FUNCTION__)); |
392 | return llvm::hash_value(Val.getMemoizationData()); |
393 | } |
394 | static bool isEqual(const DynTypedNode &LHS, const DynTypedNode &RHS) { |
395 | auto Empty = ASTNodeKind::DenseMapInfo::getEmptyKey(); |
396 | auto TombStone = ASTNodeKind::DenseMapInfo::getTombstoneKey(); |
397 | return (ASTNodeKind::DenseMapInfo::isEqual(LHS.NodeKind, Empty) && |
398 | ASTNodeKind::DenseMapInfo::isEqual(RHS.NodeKind, Empty)) || |
399 | (ASTNodeKind::DenseMapInfo::isEqual(LHS.NodeKind, TombStone) && |
400 | ASTNodeKind::DenseMapInfo::isEqual(RHS.NodeKind, TombStone)) || |
401 | LHS == RHS; |
402 | } |
403 | }; |
404 | |
405 | private: |
406 | /// Takes care of converting from and to \c T. |
407 | template <typename T, typename EnablerT = void> struct BaseConverter; |
408 | |
409 | /// Converter that uses dyn_cast<T> from a stored BaseT*. |
410 | template <typename T, typename BaseT> struct DynCastPtrConverter { |
411 | static const T *get(ASTNodeKind NodeKind, const void *Storage) { |
412 | if (ASTNodeKind::getFromNodeKind<T>().isBaseOf(NodeKind)) |
413 | return &getUnchecked(NodeKind, Storage); |
414 | return nullptr; |
415 | } |
416 | static const T &getUnchecked(ASTNodeKind NodeKind, const void *Storage) { |
417 | assert(ASTNodeKind::getFromNodeKind<T>().isBaseOf(NodeKind))(static_cast <bool> (ASTNodeKind::getFromNodeKind<T> ().isBaseOf(NodeKind)) ? void (0) : __assert_fail ("ASTNodeKind::getFromNodeKind<T>().isBaseOf(NodeKind)" , "clang/include/clang/AST/ASTTypeTraits.h", 417, __extension__ __PRETTY_FUNCTION__)); |
418 | return *cast<T>(static_cast<const BaseT *>( |
419 | *reinterpret_cast<const void *const *>(Storage))); |
420 | } |
421 | static DynTypedNode create(const BaseT &Node) { |
422 | DynTypedNode Result; |
423 | Result.NodeKind = ASTNodeKind::getFromNode(Node); |
424 | new (&Result.Storage) const void *(&Node); |
425 | return Result; |
426 | } |
427 | }; |
428 | |
429 | /// Converter that stores T* (by pointer). |
430 | template <typename T> struct PtrConverter { |
431 | static const T *get(ASTNodeKind NodeKind, const void *Storage) { |
432 | if (ASTNodeKind::getFromNodeKind<T>().isSame(NodeKind)) |
433 | return &getUnchecked(NodeKind, Storage); |
434 | return nullptr; |
435 | } |
436 | static const T &getUnchecked(ASTNodeKind NodeKind, const void *Storage) { |
437 | assert(ASTNodeKind::getFromNodeKind<T>().isSame(NodeKind))(static_cast <bool> (ASTNodeKind::getFromNodeKind<T> ().isSame(NodeKind)) ? void (0) : __assert_fail ("ASTNodeKind::getFromNodeKind<T>().isSame(NodeKind)" , "clang/include/clang/AST/ASTTypeTraits.h", 437, __extension__ __PRETTY_FUNCTION__)); |
438 | return *static_cast<const T *>( |
439 | *reinterpret_cast<const void *const *>(Storage)); |
440 | } |
441 | static DynTypedNode create(const T &Node) { |
442 | DynTypedNode Result; |
443 | Result.NodeKind = ASTNodeKind::getFromNodeKind<T>(); |
444 | new (&Result.Storage) const void *(&Node); |
445 | return Result; |
446 | } |
447 | }; |
448 | |
449 | /// Converter that stores T (by value). |
450 | template <typename T> struct ValueConverter { |
451 | static const T *get(ASTNodeKind NodeKind, const void *Storage) { |
452 | if (ASTNodeKind::getFromNodeKind<T>().isSame(NodeKind)) |
453 | return reinterpret_cast<const T *>(Storage); |
454 | return nullptr; |
455 | } |
456 | static const T &getUnchecked(ASTNodeKind NodeKind, const void *Storage) { |
457 | assert(ASTNodeKind::getFromNodeKind<T>().isSame(NodeKind))(static_cast <bool> (ASTNodeKind::getFromNodeKind<T> ().isSame(NodeKind)) ? void (0) : __assert_fail ("ASTNodeKind::getFromNodeKind<T>().isSame(NodeKind)" , "clang/include/clang/AST/ASTTypeTraits.h", 457, __extension__ __PRETTY_FUNCTION__)); |
458 | return *reinterpret_cast<const T *>(Storage); |
459 | } |
460 | static DynTypedNode create(const T &Node) { |
461 | DynTypedNode Result; |
462 | Result.NodeKind = ASTNodeKind::getFromNodeKind<T>(); |
463 | new (&Result.Storage) T(Node); |
464 | return Result; |
465 | } |
466 | }; |
467 | |
468 | /// Converter that stores nodes by value. It must be possible to dynamically |
469 | /// cast the stored node within a type hierarchy without breaking (especially |
470 | /// through slicing). |
471 | template <typename T, typename BaseT, |
472 | typename = std::enable_if_t<(sizeof(T) == sizeof(BaseT))>> |
473 | struct DynCastValueConverter { |
474 | static const T *get(ASTNodeKind NodeKind, const void *Storage) { |
475 | if (ASTNodeKind::getFromNodeKind<T>().isBaseOf(NodeKind)) |
476 | return &getUnchecked(NodeKind, Storage); |
477 | return nullptr; |
478 | } |
479 | static const T &getUnchecked(ASTNodeKind NodeKind, const void *Storage) { |
480 | assert(ASTNodeKind::getFromNodeKind<T>().isBaseOf(NodeKind))(static_cast <bool> (ASTNodeKind::getFromNodeKind<T> ().isBaseOf(NodeKind)) ? void (0) : __assert_fail ("ASTNodeKind::getFromNodeKind<T>().isBaseOf(NodeKind)" , "clang/include/clang/AST/ASTTypeTraits.h", 480, __extension__ __PRETTY_FUNCTION__)); |
481 | return *static_cast<const T *>(reinterpret_cast<const BaseT *>(Storage)); |
482 | } |
483 | static DynTypedNode create(const T &Node) { |
484 | DynTypedNode Result; |
485 | Result.NodeKind = ASTNodeKind::getFromNode(Node); |
486 | new (&Result.Storage) T(Node); |
487 | return Result; |
488 | } |
489 | }; |
490 | |
491 | ASTNodeKind NodeKind; |
492 | |
493 | /// Stores the data of the node. |
494 | /// |
495 | /// Note that we can store \c Decls, \c Stmts, \c Types, |
496 | /// \c NestedNameSpecifiers and \c CXXCtorInitializer by pointer as they are |
497 | /// guaranteed to be unique pointers pointing to dedicated storage in the AST. |
498 | /// \c QualTypes, \c NestedNameSpecifierLocs, \c TypeLocs, |
499 | /// \c TemplateArguments and \c TemplateArgumentLocs on the other hand do not |
500 | /// have storage or unique pointers and thus need to be stored by value. |
501 | llvm::AlignedCharArrayUnion<const void *, TemplateArgument, |
502 | TemplateArgumentLoc, NestedNameSpecifierLoc, |
503 | QualType, TypeLoc, ObjCProtocolLoc> |
504 | Storage; |
505 | }; |
506 | |
507 | template <typename T> |
508 | struct DynTypedNode::BaseConverter< |
509 | T, std::enable_if_t<std::is_base_of<Decl, T>::value>> |
510 | : public DynCastPtrConverter<T, Decl> {}; |
511 | |
512 | template <typename T> |
513 | struct DynTypedNode::BaseConverter< |
514 | T, std::enable_if_t<std::is_base_of<Stmt, T>::value>> |
515 | : public DynCastPtrConverter<T, Stmt> {}; |
516 | |
517 | template <typename T> |
518 | struct DynTypedNode::BaseConverter< |
519 | T, std::enable_if_t<std::is_base_of<Type, T>::value>> |
520 | : public DynCastPtrConverter<T, Type> {}; |
521 | |
522 | template <typename T> |
523 | struct DynTypedNode::BaseConverter< |
524 | T, std::enable_if_t<std::is_base_of<OMPClause, T>::value>> |
525 | : public DynCastPtrConverter<T, OMPClause> {}; |
526 | |
527 | template <typename T> |
528 | struct DynTypedNode::BaseConverter< |
529 | T, std::enable_if_t<std::is_base_of<Attr, T>::value>> |
530 | : public DynCastPtrConverter<T, Attr> {}; |
531 | |
532 | template <> |
533 | struct DynTypedNode::BaseConverter< |
534 | NestedNameSpecifier, void> : public PtrConverter<NestedNameSpecifier> {}; |
535 | |
536 | template <> |
537 | struct DynTypedNode::BaseConverter< |
538 | CXXCtorInitializer, void> : public PtrConverter<CXXCtorInitializer> {}; |
539 | |
540 | template <> |
541 | struct DynTypedNode::BaseConverter< |
542 | TemplateArgument, void> : public ValueConverter<TemplateArgument> {}; |
543 | |
544 | template <> |
545 | struct DynTypedNode::BaseConverter<TemplateArgumentLoc, void> |
546 | : public ValueConverter<TemplateArgumentLoc> {}; |
547 | |
548 | template <> |
549 | struct DynTypedNode::BaseConverter<LambdaCapture, void> |
550 | : public ValueConverter<LambdaCapture> {}; |
551 | |
552 | template <> |
553 | struct DynTypedNode::BaseConverter< |
554 | TemplateName, void> : public ValueConverter<TemplateName> {}; |
555 | |
556 | template <> |
557 | struct DynTypedNode::BaseConverter< |
558 | NestedNameSpecifierLoc, |
559 | void> : public ValueConverter<NestedNameSpecifierLoc> {}; |
560 | |
561 | template <> |
562 | struct DynTypedNode::BaseConverter<QualType, |
563 | void> : public ValueConverter<QualType> {}; |
564 | |
565 | template <typename T> |
566 | struct DynTypedNode::BaseConverter< |
567 | T, std::enable_if_t<std::is_base_of<TypeLoc, T>::value>> |
568 | : public DynCastValueConverter<T, TypeLoc> {}; |
569 | |
570 | template <> |
571 | struct DynTypedNode::BaseConverter<CXXBaseSpecifier, void> |
572 | : public PtrConverter<CXXBaseSpecifier> {}; |
573 | |
574 | template <> |
575 | struct DynTypedNode::BaseConverter<ObjCProtocolLoc, void> |
576 | : public ValueConverter<ObjCProtocolLoc> {}; |
577 | |
578 | // The only operation we allow on unsupported types is \c get. |
579 | // This allows to conveniently use \c DynTypedNode when having an arbitrary |
580 | // AST node that is not supported, but prevents misuse - a user cannot create |
581 | // a DynTypedNode from arbitrary types. |
582 | template <typename T, typename EnablerT> struct DynTypedNode::BaseConverter { |
583 | static const T *get(ASTNodeKind NodeKind, const char Storage[]) { |
584 | return NULL__null; |
585 | } |
586 | }; |
587 | |
588 | } // end namespace clang |
589 | |
590 | namespace llvm { |
591 | |
592 | template <> |
593 | struct DenseMapInfo<clang::ASTNodeKind> : clang::ASTNodeKind::DenseMapInfo {}; |
594 | |
595 | template <> |
596 | struct DenseMapInfo<clang::DynTypedNode> : clang::DynTypedNode::DenseMapInfo {}; |
597 | |
598 | } // end namespace llvm |
599 | |
600 | #endif |