xref: /openbsd-src/gnu/llvm/clang/lib/ASTMatchers/Dynamic/Registry.cpp (revision 12c855180aad702bbcca06e0398d774beeafb155)
1e5dd7070Spatrick //===- Registry.cpp - Matcher registry ------------------------------------===//
2e5dd7070Spatrick //
3e5dd7070Spatrick // Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions.
4e5dd7070Spatrick // See https://llvm.org/LICENSE.txt for license information.
5e5dd7070Spatrick // SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
6e5dd7070Spatrick //
7e5dd7070Spatrick //===----------------------------------------------------------------------===//
8e5dd7070Spatrick //
9e5dd7070Spatrick /// \file
10e5dd7070Spatrick /// Registry map populated at static initialization time.
11e5dd7070Spatrick //
12e5dd7070Spatrick //===----------------------------------------------------------------------===//
13e5dd7070Spatrick 
14e5dd7070Spatrick #include "clang/ASTMatchers/Dynamic/Registry.h"
15e5dd7070Spatrick #include "Marshallers.h"
16e5dd7070Spatrick #include "clang/AST/ASTTypeTraits.h"
17e5dd7070Spatrick #include "clang/ASTMatchers/ASTMatchers.h"
18e5dd7070Spatrick #include "clang/ASTMatchers/Dynamic/Diagnostics.h"
19e5dd7070Spatrick #include "clang/ASTMatchers/Dynamic/VariantValue.h"
20e5dd7070Spatrick #include "llvm/ADT/STLExtras.h"
21e5dd7070Spatrick #include "llvm/ADT/StringMap.h"
22e5dd7070Spatrick #include "llvm/ADT/StringRef.h"
23e5dd7070Spatrick #include "llvm/Support/ManagedStatic.h"
24e5dd7070Spatrick #include "llvm/Support/raw_ostream.h"
25e5dd7070Spatrick #include <cassert>
26e5dd7070Spatrick #include <iterator>
27e5dd7070Spatrick #include <memory>
28*12c85518Srobert #include <optional>
29e5dd7070Spatrick #include <set>
30e5dd7070Spatrick #include <string>
31e5dd7070Spatrick #include <utility>
32e5dd7070Spatrick #include <vector>
33e5dd7070Spatrick 
34e5dd7070Spatrick namespace clang {
35e5dd7070Spatrick namespace ast_matchers {
36e5dd7070Spatrick namespace dynamic {
37e5dd7070Spatrick 
38e5dd7070Spatrick namespace {
39e5dd7070Spatrick 
40e5dd7070Spatrick using internal::MatcherDescriptor;
41e5dd7070Spatrick 
42e5dd7070Spatrick using ConstructorMap =
43e5dd7070Spatrick     llvm::StringMap<std::unique_ptr<const MatcherDescriptor>>;
44e5dd7070Spatrick 
45e5dd7070Spatrick class RegistryMaps {
46e5dd7070Spatrick public:
47e5dd7070Spatrick   RegistryMaps();
48e5dd7070Spatrick   ~RegistryMaps();
49e5dd7070Spatrick 
constructors() const50e5dd7070Spatrick   const ConstructorMap &constructors() const { return Constructors; }
51e5dd7070Spatrick 
52e5dd7070Spatrick private:
53e5dd7070Spatrick   void registerMatcher(StringRef MatcherName,
54e5dd7070Spatrick                        std::unique_ptr<MatcherDescriptor> Callback);
55e5dd7070Spatrick 
56e5dd7070Spatrick   ConstructorMap Constructors;
57e5dd7070Spatrick };
58e5dd7070Spatrick 
59e5dd7070Spatrick } // namespace
60e5dd7070Spatrick 
registerMatcher(StringRef MatcherName,std::unique_ptr<MatcherDescriptor> Callback)61e5dd7070Spatrick void RegistryMaps::registerMatcher(
62e5dd7070Spatrick     StringRef MatcherName, std::unique_ptr<MatcherDescriptor> Callback) {
63e5dd7070Spatrick   assert(Constructors.find(MatcherName) == Constructors.end());
64e5dd7070Spatrick   Constructors[MatcherName] = std::move(Callback);
65e5dd7070Spatrick }
66e5dd7070Spatrick 
67e5dd7070Spatrick #define REGISTER_MATCHER(name)                                                 \
68e5dd7070Spatrick   registerMatcher(#name, internal::makeMatcherAutoMarshall(                    \
69e5dd7070Spatrick                              ::clang::ast_matchers::name, #name));
70e5dd7070Spatrick 
71e5dd7070Spatrick #define REGISTER_MATCHER_OVERLOAD(name)                                        \
72e5dd7070Spatrick   registerMatcher(#name,                                                       \
73e5dd7070Spatrick       std::make_unique<internal::OverloadedMatcherDescriptor>(name##Callbacks))
74e5dd7070Spatrick 
75e5dd7070Spatrick #define SPECIFIC_MATCHER_OVERLOAD(name, Id)                                    \
76e5dd7070Spatrick   static_cast<::clang::ast_matchers::name##_Type##Id>(                         \
77e5dd7070Spatrick       ::clang::ast_matchers::name)
78e5dd7070Spatrick 
79e5dd7070Spatrick #define MATCHER_OVERLOAD_ENTRY(name, Id)                                       \
80e5dd7070Spatrick         internal::makeMatcherAutoMarshall(SPECIFIC_MATCHER_OVERLOAD(name, Id), \
81e5dd7070Spatrick                                           #name)
82e5dd7070Spatrick 
83e5dd7070Spatrick #define REGISTER_OVERLOADED_2(name)                                            \
84e5dd7070Spatrick   do {                                                                         \
85e5dd7070Spatrick     std::unique_ptr<MatcherDescriptor> name##Callbacks[] = {                   \
86e5dd7070Spatrick         MATCHER_OVERLOAD_ENTRY(name, 0),                                       \
87e5dd7070Spatrick         MATCHER_OVERLOAD_ENTRY(name, 1)};                                      \
88e5dd7070Spatrick     REGISTER_MATCHER_OVERLOAD(name);                                           \
89e5dd7070Spatrick   } while (false)
90e5dd7070Spatrick 
91ec727ea7Spatrick #define REGISTER_REGEX_MATCHER(name)                                           \
92ec727ea7Spatrick   registerMatcher(#name, internal::makeMatcherRegexMarshall(name, name))
93ec727ea7Spatrick 
94e5dd7070Spatrick /// Generate a registry map with all the known matchers.
95e5dd7070Spatrick /// Please keep sorted alphabetically!
RegistryMaps()96e5dd7070Spatrick RegistryMaps::RegistryMaps() {
97e5dd7070Spatrick   // TODO: Here is the list of the missing matchers, grouped by reason.
98e5dd7070Spatrick   //
99e5dd7070Spatrick   // Polymorphic + argument overload:
100e5dd7070Spatrick   // findAll
101e5dd7070Spatrick   //
102e5dd7070Spatrick   // Other:
103e5dd7070Spatrick   // equalsNode
104e5dd7070Spatrick 
105a9ac8606Spatrick   registerMatcher("mapAnyOf",
106a9ac8606Spatrick                   std::make_unique<internal::MapAnyOfBuilderDescriptor>());
107a9ac8606Spatrick 
108e5dd7070Spatrick   REGISTER_OVERLOADED_2(callee);
109e5dd7070Spatrick   REGISTER_OVERLOADED_2(hasPrefix);
110e5dd7070Spatrick   REGISTER_OVERLOADED_2(hasType);
111e5dd7070Spatrick   REGISTER_OVERLOADED_2(ignoringParens);
112e5dd7070Spatrick   REGISTER_OVERLOADED_2(isDerivedFrom);
113e5dd7070Spatrick   REGISTER_OVERLOADED_2(isDirectlyDerivedFrom);
114e5dd7070Spatrick   REGISTER_OVERLOADED_2(isSameOrDerivedFrom);
115e5dd7070Spatrick   REGISTER_OVERLOADED_2(loc);
116e5dd7070Spatrick   REGISTER_OVERLOADED_2(pointsTo);
117e5dd7070Spatrick   REGISTER_OVERLOADED_2(references);
118e5dd7070Spatrick   REGISTER_OVERLOADED_2(thisPointerType);
119e5dd7070Spatrick 
120e5dd7070Spatrick   std::unique_ptr<MatcherDescriptor> equalsCallbacks[] = {
121e5dd7070Spatrick       MATCHER_OVERLOAD_ENTRY(equals, 0),
122e5dd7070Spatrick       MATCHER_OVERLOAD_ENTRY(equals, 1),
123e5dd7070Spatrick       MATCHER_OVERLOAD_ENTRY(equals, 2),
124e5dd7070Spatrick   };
125e5dd7070Spatrick   REGISTER_MATCHER_OVERLOAD(equals);
126e5dd7070Spatrick 
127ec727ea7Spatrick   REGISTER_REGEX_MATCHER(isExpansionInFileMatching);
128ec727ea7Spatrick   REGISTER_REGEX_MATCHER(matchesName);
129ec727ea7Spatrick   REGISTER_REGEX_MATCHER(matchesSelector);
130ec727ea7Spatrick 
131e5dd7070Spatrick   REGISTER_MATCHER(accessSpecDecl);
132e5dd7070Spatrick   REGISTER_MATCHER(addrLabelExpr);
133e5dd7070Spatrick   REGISTER_MATCHER(alignOfExpr);
134e5dd7070Spatrick   REGISTER_MATCHER(allOf);
135e5dd7070Spatrick   REGISTER_MATCHER(anyOf);
136e5dd7070Spatrick   REGISTER_MATCHER(anything);
137e5dd7070Spatrick   REGISTER_MATCHER(argumentCountIs);
138e5dd7070Spatrick   REGISTER_MATCHER(arraySubscriptExpr);
139e5dd7070Spatrick   REGISTER_MATCHER(arrayType);
140e5dd7070Spatrick   REGISTER_MATCHER(asString);
141e5dd7070Spatrick   REGISTER_MATCHER(asmStmt);
142e5dd7070Spatrick   REGISTER_MATCHER(atomicExpr);
143e5dd7070Spatrick   REGISTER_MATCHER(atomicType);
144*12c85518Srobert   REGISTER_MATCHER(attr);
145e5dd7070Spatrick   REGISTER_MATCHER(autoType);
146e5dd7070Spatrick   REGISTER_MATCHER(autoreleasePoolStmt)
147e5dd7070Spatrick   REGISTER_MATCHER(binaryConditionalOperator);
148e5dd7070Spatrick   REGISTER_MATCHER(binaryOperator);
149a9ac8606Spatrick   REGISTER_MATCHER(binaryOperation);
150a9ac8606Spatrick   REGISTER_MATCHER(bindingDecl);
151e5dd7070Spatrick   REGISTER_MATCHER(blockDecl);
152e5dd7070Spatrick   REGISTER_MATCHER(blockExpr);
153e5dd7070Spatrick   REGISTER_MATCHER(blockPointerType);
154e5dd7070Spatrick   REGISTER_MATCHER(booleanType);
155e5dd7070Spatrick   REGISTER_MATCHER(breakStmt);
156e5dd7070Spatrick   REGISTER_MATCHER(builtinType);
157e5dd7070Spatrick   REGISTER_MATCHER(cStyleCastExpr);
158e5dd7070Spatrick   REGISTER_MATCHER(callExpr);
159*12c85518Srobert   REGISTER_MATCHER(capturesThis);
160*12c85518Srobert   REGISTER_MATCHER(capturesVar);
161e5dd7070Spatrick   REGISTER_MATCHER(caseStmt);
162e5dd7070Spatrick   REGISTER_MATCHER(castExpr);
163e5dd7070Spatrick   REGISTER_MATCHER(characterLiteral);
164e5dd7070Spatrick   REGISTER_MATCHER(chooseExpr);
165e5dd7070Spatrick   REGISTER_MATCHER(classTemplateDecl);
166ec727ea7Spatrick   REGISTER_MATCHER(classTemplatePartialSpecializationDecl);
167e5dd7070Spatrick   REGISTER_MATCHER(classTemplateSpecializationDecl);
168e5dd7070Spatrick   REGISTER_MATCHER(complexType);
169e5dd7070Spatrick   REGISTER_MATCHER(compoundLiteralExpr);
170e5dd7070Spatrick   REGISTER_MATCHER(compoundStmt);
171a9ac8606Spatrick   REGISTER_MATCHER(coawaitExpr);
172e5dd7070Spatrick   REGISTER_MATCHER(conditionalOperator);
173e5dd7070Spatrick   REGISTER_MATCHER(constantArrayType);
174e5dd7070Spatrick   REGISTER_MATCHER(constantExpr);
175e5dd7070Spatrick   REGISTER_MATCHER(containsDeclaration);
176e5dd7070Spatrick   REGISTER_MATCHER(continueStmt);
177a9ac8606Spatrick   REGISTER_MATCHER(coreturnStmt);
178a9ac8606Spatrick   REGISTER_MATCHER(coyieldExpr);
179e5dd7070Spatrick   REGISTER_MATCHER(cudaKernelCallExpr);
180a9ac8606Spatrick   REGISTER_MATCHER(cxxBaseSpecifier);
181e5dd7070Spatrick   REGISTER_MATCHER(cxxBindTemporaryExpr);
182e5dd7070Spatrick   REGISTER_MATCHER(cxxBoolLiteral);
183e5dd7070Spatrick   REGISTER_MATCHER(cxxCatchStmt);
184e5dd7070Spatrick   REGISTER_MATCHER(cxxConstCastExpr);
185e5dd7070Spatrick   REGISTER_MATCHER(cxxConstructExpr);
186e5dd7070Spatrick   REGISTER_MATCHER(cxxConstructorDecl);
187e5dd7070Spatrick   REGISTER_MATCHER(cxxConversionDecl);
188e5dd7070Spatrick   REGISTER_MATCHER(cxxCtorInitializer);
189e5dd7070Spatrick   REGISTER_MATCHER(cxxDeductionGuideDecl);
190e5dd7070Spatrick   REGISTER_MATCHER(cxxDefaultArgExpr);
191e5dd7070Spatrick   REGISTER_MATCHER(cxxDeleteExpr);
192e5dd7070Spatrick   REGISTER_MATCHER(cxxDependentScopeMemberExpr);
193e5dd7070Spatrick   REGISTER_MATCHER(cxxDestructorDecl);
194e5dd7070Spatrick   REGISTER_MATCHER(cxxDynamicCastExpr);
195e5dd7070Spatrick   REGISTER_MATCHER(cxxForRangeStmt);
196e5dd7070Spatrick   REGISTER_MATCHER(cxxFunctionalCastExpr);
197e5dd7070Spatrick   REGISTER_MATCHER(cxxMemberCallExpr);
198e5dd7070Spatrick   REGISTER_MATCHER(cxxMethodDecl);
199e5dd7070Spatrick   REGISTER_MATCHER(cxxNewExpr);
200ec727ea7Spatrick   REGISTER_MATCHER(cxxNoexceptExpr);
201e5dd7070Spatrick   REGISTER_MATCHER(cxxNullPtrLiteralExpr);
202e5dd7070Spatrick   REGISTER_MATCHER(cxxOperatorCallExpr);
203e5dd7070Spatrick   REGISTER_MATCHER(cxxRecordDecl);
204e5dd7070Spatrick   REGISTER_MATCHER(cxxReinterpretCastExpr);
205a9ac8606Spatrick   REGISTER_MATCHER(cxxRewrittenBinaryOperator);
206e5dd7070Spatrick   REGISTER_MATCHER(cxxStaticCastExpr);
207e5dd7070Spatrick   REGISTER_MATCHER(cxxStdInitializerListExpr);
208e5dd7070Spatrick   REGISTER_MATCHER(cxxTemporaryObjectExpr);
209e5dd7070Spatrick   REGISTER_MATCHER(cxxThisExpr);
210e5dd7070Spatrick   REGISTER_MATCHER(cxxThrowExpr);
211e5dd7070Spatrick   REGISTER_MATCHER(cxxTryStmt);
212e5dd7070Spatrick   REGISTER_MATCHER(cxxUnresolvedConstructExpr);
213e5dd7070Spatrick   REGISTER_MATCHER(decayedType);
214e5dd7070Spatrick   REGISTER_MATCHER(decl);
215a9ac8606Spatrick   REGISTER_MATCHER(decompositionDecl);
216e5dd7070Spatrick   REGISTER_MATCHER(declCountIs);
217e5dd7070Spatrick   REGISTER_MATCHER(declRefExpr);
218e5dd7070Spatrick   REGISTER_MATCHER(declStmt);
219e5dd7070Spatrick   REGISTER_MATCHER(declaratorDecl);
220e5dd7070Spatrick   REGISTER_MATCHER(decltypeType);
221ec727ea7Spatrick   REGISTER_MATCHER(deducedTemplateSpecializationType);
222e5dd7070Spatrick   REGISTER_MATCHER(defaultStmt);
223a9ac8606Spatrick   REGISTER_MATCHER(dependentCoawaitExpr);
224e5dd7070Spatrick   REGISTER_MATCHER(dependentSizedArrayType);
225e5dd7070Spatrick   REGISTER_MATCHER(designatedInitExpr);
226e5dd7070Spatrick   REGISTER_MATCHER(designatorCountIs);
227e5dd7070Spatrick   REGISTER_MATCHER(doStmt);
228e5dd7070Spatrick   REGISTER_MATCHER(eachOf);
229e5dd7070Spatrick   REGISTER_MATCHER(elaboratedType);
230*12c85518Srobert   REGISTER_MATCHER(elaboratedTypeLoc);
231*12c85518Srobert   REGISTER_MATCHER(usingType);
232e5dd7070Spatrick   REGISTER_MATCHER(enumConstantDecl);
233e5dd7070Spatrick   REGISTER_MATCHER(enumDecl);
234e5dd7070Spatrick   REGISTER_MATCHER(enumType);
235e5dd7070Spatrick   REGISTER_MATCHER(equalsBoundNode);
236e5dd7070Spatrick   REGISTER_MATCHER(equalsIntegralValue);
237e5dd7070Spatrick   REGISTER_MATCHER(explicitCastExpr);
238e5dd7070Spatrick   REGISTER_MATCHER(expr);
239e5dd7070Spatrick   REGISTER_MATCHER(exprWithCleanups);
240e5dd7070Spatrick   REGISTER_MATCHER(fieldDecl);
241a9ac8606Spatrick   REGISTER_MATCHER(fixedPointLiteral);
242e5dd7070Spatrick   REGISTER_MATCHER(floatLiteral);
243a9ac8606Spatrick   REGISTER_MATCHER(forCallable);
244a9ac8606Spatrick   REGISTER_MATCHER(forDecomposition);
245e5dd7070Spatrick   REGISTER_MATCHER(forEach);
246e5dd7070Spatrick   REGISTER_MATCHER(forEachArgumentWithParam);
247a9ac8606Spatrick   REGISTER_MATCHER(forEachArgumentWithParamType);
248e5dd7070Spatrick   REGISTER_MATCHER(forEachConstructorInitializer);
249e5dd7070Spatrick   REGISTER_MATCHER(forEachDescendant);
250*12c85518Srobert   REGISTER_MATCHER(forEachLambdaCapture);
251e5dd7070Spatrick   REGISTER_MATCHER(forEachOverridden);
252e5dd7070Spatrick   REGISTER_MATCHER(forEachSwitchCase);
253*12c85518Srobert   REGISTER_MATCHER(forEachTemplateArgument);
254e5dd7070Spatrick   REGISTER_MATCHER(forField);
255e5dd7070Spatrick   REGISTER_MATCHER(forFunction);
256e5dd7070Spatrick   REGISTER_MATCHER(forStmt);
257e5dd7070Spatrick   REGISTER_MATCHER(friendDecl);
258e5dd7070Spatrick   REGISTER_MATCHER(functionDecl);
259e5dd7070Spatrick   REGISTER_MATCHER(functionProtoType);
260e5dd7070Spatrick   REGISTER_MATCHER(functionTemplateDecl);
261e5dd7070Spatrick   REGISTER_MATCHER(functionType);
262a9ac8606Spatrick   REGISTER_MATCHER(genericSelectionExpr);
263e5dd7070Spatrick   REGISTER_MATCHER(gnuNullExpr);
264e5dd7070Spatrick   REGISTER_MATCHER(gotoStmt);
265e5dd7070Spatrick   REGISTER_MATCHER(has);
266e5dd7070Spatrick   REGISTER_MATCHER(hasAncestor);
267e5dd7070Spatrick   REGISTER_MATCHER(hasAnyArgument);
268ec727ea7Spatrick   REGISTER_MATCHER(hasAnyBase);
269a9ac8606Spatrick   REGISTER_MATCHER(hasAnyBinding);
270a9ac8606Spatrick   REGISTER_MATCHER(hasAnyBody);
271*12c85518Srobert   REGISTER_MATCHER(hasAnyCapture);
272e5dd7070Spatrick   REGISTER_MATCHER(hasAnyClause);
273e5dd7070Spatrick   REGISTER_MATCHER(hasAnyConstructorInitializer);
274e5dd7070Spatrick   REGISTER_MATCHER(hasAnyDeclaration);
275e5dd7070Spatrick   REGISTER_MATCHER(hasAnyName);
276ec727ea7Spatrick   REGISTER_MATCHER(hasAnyOperatorName);
277ec727ea7Spatrick   REGISTER_MATCHER(hasAnyOverloadedOperatorName);
278e5dd7070Spatrick   REGISTER_MATCHER(hasAnyParameter);
279ec727ea7Spatrick   REGISTER_MATCHER(hasAnyPlacementArg);
280e5dd7070Spatrick   REGISTER_MATCHER(hasAnySelector);
281e5dd7070Spatrick   REGISTER_MATCHER(hasAnySubstatement);
282e5dd7070Spatrick   REGISTER_MATCHER(hasAnyTemplateArgument);
283*12c85518Srobert   REGISTER_MATCHER(hasAnyTemplateArgumentLoc);
284e5dd7070Spatrick   REGISTER_MATCHER(hasAnyUsingShadowDecl);
285e5dd7070Spatrick   REGISTER_MATCHER(hasArgument);
286e5dd7070Spatrick   REGISTER_MATCHER(hasArgumentOfType);
287e5dd7070Spatrick   REGISTER_MATCHER(hasArraySize);
288e5dd7070Spatrick   REGISTER_MATCHER(hasAttr);
289e5dd7070Spatrick   REGISTER_MATCHER(hasAutomaticStorageDuration);
290e5dd7070Spatrick   REGISTER_MATCHER(hasBase);
291a9ac8606Spatrick   REGISTER_MATCHER(hasBinding);
292e5dd7070Spatrick   REGISTER_MATCHER(hasBitWidth);
293e5dd7070Spatrick   REGISTER_MATCHER(hasBody);
294e5dd7070Spatrick   REGISTER_MATCHER(hasCanonicalType);
295e5dd7070Spatrick   REGISTER_MATCHER(hasCaseConstant);
296e5dd7070Spatrick   REGISTER_MATCHER(hasCastKind);
297e5dd7070Spatrick   REGISTER_MATCHER(hasCondition);
298e5dd7070Spatrick   REGISTER_MATCHER(hasConditionVariableStatement);
299e5dd7070Spatrick   REGISTER_MATCHER(hasDecayedType);
300e5dd7070Spatrick   REGISTER_MATCHER(hasDeclContext);
301e5dd7070Spatrick   REGISTER_MATCHER(hasDeclaration);
302e5dd7070Spatrick   REGISTER_MATCHER(hasDeducedType);
303e5dd7070Spatrick   REGISTER_MATCHER(hasDefaultArgument);
304e5dd7070Spatrick   REGISTER_MATCHER(hasDefinition);
305e5dd7070Spatrick   REGISTER_MATCHER(hasDescendant);
306e5dd7070Spatrick   REGISTER_MATCHER(hasDestinationType);
307ec727ea7Spatrick   REGISTER_MATCHER(hasDirectBase);
308e5dd7070Spatrick   REGISTER_MATCHER(hasDynamicExceptionSpec);
309e5dd7070Spatrick   REGISTER_MATCHER(hasEitherOperand);
310e5dd7070Spatrick   REGISTER_MATCHER(hasElementType);
311e5dd7070Spatrick   REGISTER_MATCHER(hasElse);
312e5dd7070Spatrick   REGISTER_MATCHER(hasExplicitSpecifier);
313e5dd7070Spatrick   REGISTER_MATCHER(hasExternalFormalLinkage);
314e5dd7070Spatrick   REGISTER_MATCHER(hasFalseExpression);
315e5dd7070Spatrick   REGISTER_MATCHER(hasGlobalStorage);
316e5dd7070Spatrick   REGISTER_MATCHER(hasImplicitDestinationType);
317e5dd7070Spatrick   REGISTER_MATCHER(hasInClassInitializer);
318e5dd7070Spatrick   REGISTER_MATCHER(hasIncrement);
319e5dd7070Spatrick   REGISTER_MATCHER(hasIndex);
320e5dd7070Spatrick   REGISTER_MATCHER(hasInit);
321e5dd7070Spatrick   REGISTER_MATCHER(hasInitializer);
322e5dd7070Spatrick   REGISTER_MATCHER(hasInitStatement);
323e5dd7070Spatrick   REGISTER_MATCHER(hasKeywordSelector);
324e5dd7070Spatrick   REGISTER_MATCHER(hasLHS);
325e5dd7070Spatrick   REGISTER_MATCHER(hasLocalQualifiers);
326e5dd7070Spatrick   REGISTER_MATCHER(hasLocalStorage);
327e5dd7070Spatrick   REGISTER_MATCHER(hasLoopInit);
328e5dd7070Spatrick   REGISTER_MATCHER(hasLoopVariable);
329a9ac8606Spatrick   REGISTER_MATCHER(hasMemberName);
330e5dd7070Spatrick   REGISTER_MATCHER(hasMethod);
331e5dd7070Spatrick   REGISTER_MATCHER(hasName);
332*12c85518Srobert   REGISTER_MATCHER(hasNamedTypeLoc);
333e5dd7070Spatrick   REGISTER_MATCHER(hasNullSelector);
334e5dd7070Spatrick   REGISTER_MATCHER(hasObjectExpression);
335ec727ea7Spatrick   REGISTER_MATCHER(hasOperands);
336e5dd7070Spatrick   REGISTER_MATCHER(hasOperatorName);
337e5dd7070Spatrick   REGISTER_MATCHER(hasOverloadedOperatorName);
338e5dd7070Spatrick   REGISTER_MATCHER(hasParameter);
339e5dd7070Spatrick   REGISTER_MATCHER(hasParent);
340*12c85518Srobert   REGISTER_MATCHER(hasPointeeLoc);
341e5dd7070Spatrick   REGISTER_MATCHER(hasQualifier);
342e5dd7070Spatrick   REGISTER_MATCHER(hasRHS);
343e5dd7070Spatrick   REGISTER_MATCHER(hasRangeInit);
344e5dd7070Spatrick   REGISTER_MATCHER(hasReceiver);
345e5dd7070Spatrick   REGISTER_MATCHER(hasReceiverType);
346*12c85518Srobert   REGISTER_MATCHER(hasReferentLoc);
347e5dd7070Spatrick   REGISTER_MATCHER(hasReplacementType);
348*12c85518Srobert   REGISTER_MATCHER(hasReturnTypeLoc);
349e5dd7070Spatrick   REGISTER_MATCHER(hasReturnValue);
350ec727ea7Spatrick   REGISTER_MATCHER(hasPlacementArg);
351e5dd7070Spatrick   REGISTER_MATCHER(hasSelector);
352e5dd7070Spatrick   REGISTER_MATCHER(hasSingleDecl);
353e5dd7070Spatrick   REGISTER_MATCHER(hasSize);
354e5dd7070Spatrick   REGISTER_MATCHER(hasSizeExpr);
355e5dd7070Spatrick   REGISTER_MATCHER(hasSourceExpression);
356e5dd7070Spatrick   REGISTER_MATCHER(hasSpecializedTemplate);
357e5dd7070Spatrick   REGISTER_MATCHER(hasStaticStorageDuration);
358e5dd7070Spatrick   REGISTER_MATCHER(hasStructuredBlock);
359e5dd7070Spatrick   REGISTER_MATCHER(hasSyntacticForm);
360e5dd7070Spatrick   REGISTER_MATCHER(hasTargetDecl);
361e5dd7070Spatrick   REGISTER_MATCHER(hasTemplateArgument);
362*12c85518Srobert   REGISTER_MATCHER(hasTemplateArgumentLoc);
363e5dd7070Spatrick   REGISTER_MATCHER(hasThen);
364e5dd7070Spatrick   REGISTER_MATCHER(hasThreadStorageDuration);
365e5dd7070Spatrick   REGISTER_MATCHER(hasTrailingReturn);
366e5dd7070Spatrick   REGISTER_MATCHER(hasTrueExpression);
367e5dd7070Spatrick   REGISTER_MATCHER(hasTypeLoc);
368e5dd7070Spatrick   REGISTER_MATCHER(hasUnaryOperand);
369e5dd7070Spatrick   REGISTER_MATCHER(hasUnarySelector);
370e5dd7070Spatrick   REGISTER_MATCHER(hasUnderlyingDecl);
371e5dd7070Spatrick   REGISTER_MATCHER(hasUnderlyingType);
372e5dd7070Spatrick   REGISTER_MATCHER(hasUnqualifiedDesugaredType);
373*12c85518Srobert   REGISTER_MATCHER(hasUnqualifiedLoc);
374e5dd7070Spatrick   REGISTER_MATCHER(hasValueType);
375e5dd7070Spatrick   REGISTER_MATCHER(ifStmt);
376e5dd7070Spatrick   REGISTER_MATCHER(ignoringElidableConstructorCall);
377e5dd7070Spatrick   REGISTER_MATCHER(ignoringImpCasts);
378e5dd7070Spatrick   REGISTER_MATCHER(ignoringImplicit);
379e5dd7070Spatrick   REGISTER_MATCHER(ignoringParenCasts);
380e5dd7070Spatrick   REGISTER_MATCHER(ignoringParenImpCasts);
381e5dd7070Spatrick   REGISTER_MATCHER(imaginaryLiteral);
382e5dd7070Spatrick   REGISTER_MATCHER(implicitCastExpr);
383e5dd7070Spatrick   REGISTER_MATCHER(implicitValueInitExpr);
384e5dd7070Spatrick   REGISTER_MATCHER(incompleteArrayType);
385e5dd7070Spatrick   REGISTER_MATCHER(indirectFieldDecl);
386e5dd7070Spatrick   REGISTER_MATCHER(initListExpr);
387e5dd7070Spatrick   REGISTER_MATCHER(injectedClassNameType);
388e5dd7070Spatrick   REGISTER_MATCHER(innerType);
389e5dd7070Spatrick   REGISTER_MATCHER(integerLiteral);
390a9ac8606Spatrick   REGISTER_MATCHER(invocation);
391e5dd7070Spatrick   REGISTER_MATCHER(isAllowedToContainClauseKind);
392e5dd7070Spatrick   REGISTER_MATCHER(isAnonymous);
393e5dd7070Spatrick   REGISTER_MATCHER(isAnyCharacter);
394e5dd7070Spatrick   REGISTER_MATCHER(isAnyPointer);
395e5dd7070Spatrick   REGISTER_MATCHER(isArray);
396e5dd7070Spatrick   REGISTER_MATCHER(isArrow);
397e5dd7070Spatrick   REGISTER_MATCHER(isAssignmentOperator);
398ec727ea7Spatrick   REGISTER_MATCHER(isAtPosition);
399e5dd7070Spatrick   REGISTER_MATCHER(isBaseInitializer);
400e5dd7070Spatrick   REGISTER_MATCHER(isBitField);
401e5dd7070Spatrick   REGISTER_MATCHER(isCatchAll);
402e5dd7070Spatrick   REGISTER_MATCHER(isClass);
403e5dd7070Spatrick   REGISTER_MATCHER(isClassMessage);
404e5dd7070Spatrick   REGISTER_MATCHER(isClassMethod);
405ec727ea7Spatrick   REGISTER_MATCHER(isComparisonOperator);
406e5dd7070Spatrick   REGISTER_MATCHER(isConst);
407e5dd7070Spatrick   REGISTER_MATCHER(isConstQualified);
408*12c85518Srobert   REGISTER_MATCHER(isConsteval);
409e5dd7070Spatrick   REGISTER_MATCHER(isConstexpr);
410*12c85518Srobert   REGISTER_MATCHER(isConstinit);
411e5dd7070Spatrick   REGISTER_MATCHER(isCopyAssignmentOperator);
412e5dd7070Spatrick   REGISTER_MATCHER(isCopyConstructor);
413e5dd7070Spatrick   REGISTER_MATCHER(isDefaultConstructor);
414e5dd7070Spatrick   REGISTER_MATCHER(isDefaulted);
415e5dd7070Spatrick   REGISTER_MATCHER(isDefinition);
416e5dd7070Spatrick   REGISTER_MATCHER(isDelegatingConstructor);
417e5dd7070Spatrick   REGISTER_MATCHER(isDeleted);
418ec727ea7Spatrick   REGISTER_MATCHER(isEnum);
419e5dd7070Spatrick   REGISTER_MATCHER(isExceptionVariable);
420ec727ea7Spatrick   REGISTER_MATCHER(isExpandedFromMacro);
421e5dd7070Spatrick   REGISTER_MATCHER(isExpansionInMainFile);
422e5dd7070Spatrick   REGISTER_MATCHER(isExpansionInSystemHeader);
423e5dd7070Spatrick   REGISTER_MATCHER(isExplicit);
424e5dd7070Spatrick   REGISTER_MATCHER(isExplicitTemplateSpecialization);
425e5dd7070Spatrick   REGISTER_MATCHER(isExpr);
426e5dd7070Spatrick   REGISTER_MATCHER(isExternC);
427e5dd7070Spatrick   REGISTER_MATCHER(isFinal);
428*12c85518Srobert   REGISTER_MATCHER(isPrivateKind);
429ec727ea7Spatrick   REGISTER_MATCHER(isFirstPrivateKind);
430e5dd7070Spatrick   REGISTER_MATCHER(isImplicit);
431*12c85518Srobert   REGISTER_MATCHER(isInAnonymousNamespace);
432e5dd7070Spatrick   REGISTER_MATCHER(isInStdNamespace);
433e5dd7070Spatrick   REGISTER_MATCHER(isInTemplateInstantiation);
434*12c85518Srobert   REGISTER_MATCHER(isInitCapture);
435e5dd7070Spatrick   REGISTER_MATCHER(isInline);
436e5dd7070Spatrick   REGISTER_MATCHER(isInstanceMessage);
437e5dd7070Spatrick   REGISTER_MATCHER(isInstanceMethod);
438e5dd7070Spatrick   REGISTER_MATCHER(isInstantiated);
439e5dd7070Spatrick   REGISTER_MATCHER(isInstantiationDependent);
440e5dd7070Spatrick   REGISTER_MATCHER(isInteger);
441e5dd7070Spatrick   REGISTER_MATCHER(isIntegral);
442e5dd7070Spatrick   REGISTER_MATCHER(isLambda);
443e5dd7070Spatrick   REGISTER_MATCHER(isListInitialization);
444e5dd7070Spatrick   REGISTER_MATCHER(isMain);
445e5dd7070Spatrick   REGISTER_MATCHER(isMemberInitializer);
446e5dd7070Spatrick   REGISTER_MATCHER(isMoveAssignmentOperator);
447e5dd7070Spatrick   REGISTER_MATCHER(isMoveConstructor);
448e5dd7070Spatrick   REGISTER_MATCHER(isNoReturn);
449e5dd7070Spatrick   REGISTER_MATCHER(isNoThrow);
450e5dd7070Spatrick   REGISTER_MATCHER(isNoneKind);
451e5dd7070Spatrick   REGISTER_MATCHER(isOverride);
452e5dd7070Spatrick   REGISTER_MATCHER(isPrivate);
453e5dd7070Spatrick   REGISTER_MATCHER(isProtected);
454e5dd7070Spatrick   REGISTER_MATCHER(isPublic);
455e5dd7070Spatrick   REGISTER_MATCHER(isPure);
456e5dd7070Spatrick   REGISTER_MATCHER(isScoped);
457e5dd7070Spatrick   REGISTER_MATCHER(isSharedKind);
458e5dd7070Spatrick   REGISTER_MATCHER(isSignedInteger);
459e5dd7070Spatrick   REGISTER_MATCHER(isStandaloneDirective);
460e5dd7070Spatrick   REGISTER_MATCHER(isStaticLocal);
461e5dd7070Spatrick   REGISTER_MATCHER(isStaticStorageClass);
462e5dd7070Spatrick   REGISTER_MATCHER(isStruct);
463e5dd7070Spatrick   REGISTER_MATCHER(isTemplateInstantiation);
464e5dd7070Spatrick   REGISTER_MATCHER(isTypeDependent);
465e5dd7070Spatrick   REGISTER_MATCHER(isUnion);
466e5dd7070Spatrick   REGISTER_MATCHER(isUnsignedInteger);
467e5dd7070Spatrick   REGISTER_MATCHER(isUserProvided);
468e5dd7070Spatrick   REGISTER_MATCHER(isValueDependent);
469e5dd7070Spatrick   REGISTER_MATCHER(isVariadic);
470e5dd7070Spatrick   REGISTER_MATCHER(isVirtual);
471e5dd7070Spatrick   REGISTER_MATCHER(isVirtualAsWritten);
472e5dd7070Spatrick   REGISTER_MATCHER(isVolatileQualified);
473a9ac8606Spatrick   REGISTER_MATCHER(isWeak);
474e5dd7070Spatrick   REGISTER_MATCHER(isWritten);
475e5dd7070Spatrick   REGISTER_MATCHER(lValueReferenceType);
476e5dd7070Spatrick   REGISTER_MATCHER(labelDecl);
477e5dd7070Spatrick   REGISTER_MATCHER(labelStmt);
478*12c85518Srobert   REGISTER_MATCHER(lambdaCapture);
479e5dd7070Spatrick   REGISTER_MATCHER(lambdaExpr);
480e5dd7070Spatrick   REGISTER_MATCHER(linkageSpecDecl);
481e5dd7070Spatrick   REGISTER_MATCHER(materializeTemporaryExpr);
482e5dd7070Spatrick   REGISTER_MATCHER(member);
483e5dd7070Spatrick   REGISTER_MATCHER(memberExpr);
484a9ac8606Spatrick   REGISTER_MATCHER(memberHasSameNameAsBoundNode);
485e5dd7070Spatrick   REGISTER_MATCHER(memberPointerType);
486e5dd7070Spatrick   REGISTER_MATCHER(namedDecl);
487e5dd7070Spatrick   REGISTER_MATCHER(namesType);
488e5dd7070Spatrick   REGISTER_MATCHER(namespaceAliasDecl);
489e5dd7070Spatrick   REGISTER_MATCHER(namespaceDecl);
490e5dd7070Spatrick   REGISTER_MATCHER(nestedNameSpecifier);
491e5dd7070Spatrick   REGISTER_MATCHER(nestedNameSpecifierLoc);
492e5dd7070Spatrick   REGISTER_MATCHER(nonTypeTemplateParmDecl);
493e5dd7070Spatrick   REGISTER_MATCHER(nullPointerConstant);
494e5dd7070Spatrick   REGISTER_MATCHER(nullStmt);
495e5dd7070Spatrick   REGISTER_MATCHER(numSelectorArgs);
496e5dd7070Spatrick   REGISTER_MATCHER(objcCatchStmt);
497e5dd7070Spatrick   REGISTER_MATCHER(objcCategoryDecl);
498e5dd7070Spatrick   REGISTER_MATCHER(objcCategoryImplDecl);
499e5dd7070Spatrick   REGISTER_MATCHER(objcFinallyStmt);
500e5dd7070Spatrick   REGISTER_MATCHER(objcImplementationDecl);
501e5dd7070Spatrick   REGISTER_MATCHER(objcInterfaceDecl);
502e5dd7070Spatrick   REGISTER_MATCHER(objcIvarDecl);
503e5dd7070Spatrick   REGISTER_MATCHER(objcIvarRefExpr);
504e5dd7070Spatrick   REGISTER_MATCHER(objcMessageExpr);
505e5dd7070Spatrick   REGISTER_MATCHER(objcMethodDecl);
506e5dd7070Spatrick   REGISTER_MATCHER(objcObjectPointerType);
507e5dd7070Spatrick   REGISTER_MATCHER(objcPropertyDecl);
508e5dd7070Spatrick   REGISTER_MATCHER(objcProtocolDecl);
509*12c85518Srobert   REGISTER_MATCHER(objcStringLiteral);
510e5dd7070Spatrick   REGISTER_MATCHER(objcThrowStmt);
511e5dd7070Spatrick   REGISTER_MATCHER(objcTryStmt);
512e5dd7070Spatrick   REGISTER_MATCHER(ofClass);
513ec727ea7Spatrick   REGISTER_MATCHER(ofKind);
514e5dd7070Spatrick   REGISTER_MATCHER(ompDefaultClause);
515e5dd7070Spatrick   REGISTER_MATCHER(ompExecutableDirective);
516e5dd7070Spatrick   REGISTER_MATCHER(on);
517e5dd7070Spatrick   REGISTER_MATCHER(onImplicitObjectArgument);
518e5dd7070Spatrick   REGISTER_MATCHER(opaqueValueExpr);
519e5dd7070Spatrick   REGISTER_MATCHER(optionally);
520e5dd7070Spatrick   REGISTER_MATCHER(parameterCountIs);
521e5dd7070Spatrick   REGISTER_MATCHER(parenExpr);
522e5dd7070Spatrick   REGISTER_MATCHER(parenListExpr);
523e5dd7070Spatrick   REGISTER_MATCHER(parenType);
524e5dd7070Spatrick   REGISTER_MATCHER(parmVarDecl);
525e5dd7070Spatrick   REGISTER_MATCHER(pointee);
526e5dd7070Spatrick   REGISTER_MATCHER(pointerType);
527*12c85518Srobert   REGISTER_MATCHER(pointerTypeLoc);
528e5dd7070Spatrick   REGISTER_MATCHER(predefinedExpr);
529e5dd7070Spatrick   REGISTER_MATCHER(qualType);
530*12c85518Srobert   REGISTER_MATCHER(qualifiedTypeLoc);
531e5dd7070Spatrick   REGISTER_MATCHER(rValueReferenceType);
532e5dd7070Spatrick   REGISTER_MATCHER(realFloatingPointType);
533e5dd7070Spatrick   REGISTER_MATCHER(recordDecl);
534e5dd7070Spatrick   REGISTER_MATCHER(recordType);
535e5dd7070Spatrick   REGISTER_MATCHER(referenceType);
536*12c85518Srobert   REGISTER_MATCHER(referenceTypeLoc);
537e5dd7070Spatrick   REGISTER_MATCHER(refersToDeclaration);
538e5dd7070Spatrick   REGISTER_MATCHER(refersToIntegralType);
539e5dd7070Spatrick   REGISTER_MATCHER(refersToTemplate);
540e5dd7070Spatrick   REGISTER_MATCHER(refersToType);
541e5dd7070Spatrick   REGISTER_MATCHER(requiresZeroInitialization);
542e5dd7070Spatrick   REGISTER_MATCHER(returnStmt);
543e5dd7070Spatrick   REGISTER_MATCHER(returns);
544e5dd7070Spatrick   REGISTER_MATCHER(sizeOfExpr);
545e5dd7070Spatrick   REGISTER_MATCHER(specifiesNamespace);
546e5dd7070Spatrick   REGISTER_MATCHER(specifiesType);
547e5dd7070Spatrick   REGISTER_MATCHER(specifiesTypeLoc);
548e5dd7070Spatrick   REGISTER_MATCHER(statementCountIs);
549e5dd7070Spatrick   REGISTER_MATCHER(staticAssertDecl);
550e5dd7070Spatrick   REGISTER_MATCHER(stmt);
551e5dd7070Spatrick   REGISTER_MATCHER(stmtExpr);
552e5dd7070Spatrick   REGISTER_MATCHER(stringLiteral);
553e5dd7070Spatrick   REGISTER_MATCHER(substNonTypeTemplateParmExpr);
554e5dd7070Spatrick   REGISTER_MATCHER(substTemplateTypeParmType);
555e5dd7070Spatrick   REGISTER_MATCHER(switchCase);
556e5dd7070Spatrick   REGISTER_MATCHER(switchStmt);
557ec727ea7Spatrick   REGISTER_MATCHER(tagDecl);
558e5dd7070Spatrick   REGISTER_MATCHER(tagType);
559e5dd7070Spatrick   REGISTER_MATCHER(templateArgument);
560e5dd7070Spatrick   REGISTER_MATCHER(templateArgumentCountIs);
561a9ac8606Spatrick   REGISTER_MATCHER(templateArgumentLoc);
562e5dd7070Spatrick   REGISTER_MATCHER(templateName);
563e5dd7070Spatrick   REGISTER_MATCHER(templateSpecializationType);
564*12c85518Srobert   REGISTER_MATCHER(templateSpecializationTypeLoc);
565a9ac8606Spatrick   REGISTER_MATCHER(templateTemplateParmDecl);
566e5dd7070Spatrick   REGISTER_MATCHER(templateTypeParmDecl);
567e5dd7070Spatrick   REGISTER_MATCHER(templateTypeParmType);
568e5dd7070Spatrick   REGISTER_MATCHER(throughUsingDecl);
569e5dd7070Spatrick   REGISTER_MATCHER(to);
570e5dd7070Spatrick   REGISTER_MATCHER(translationUnitDecl);
571e5dd7070Spatrick   REGISTER_MATCHER(type);
572e5dd7070Spatrick   REGISTER_MATCHER(typeAliasDecl);
573e5dd7070Spatrick   REGISTER_MATCHER(typeAliasTemplateDecl);
574e5dd7070Spatrick   REGISTER_MATCHER(typeLoc);
575e5dd7070Spatrick   REGISTER_MATCHER(typedefDecl);
576e5dd7070Spatrick   REGISTER_MATCHER(typedefNameDecl);
577e5dd7070Spatrick   REGISTER_MATCHER(typedefType);
578e5dd7070Spatrick   REGISTER_MATCHER(unaryExprOrTypeTraitExpr);
579e5dd7070Spatrick   REGISTER_MATCHER(unaryOperator);
580e5dd7070Spatrick   REGISTER_MATCHER(unaryTransformType);
581e5dd7070Spatrick   REGISTER_MATCHER(unless);
582e5dd7070Spatrick   REGISTER_MATCHER(unresolvedLookupExpr);
583e5dd7070Spatrick   REGISTER_MATCHER(unresolvedMemberExpr);
584e5dd7070Spatrick   REGISTER_MATCHER(unresolvedUsingTypenameDecl);
585e5dd7070Spatrick   REGISTER_MATCHER(unresolvedUsingValueDecl);
586e5dd7070Spatrick   REGISTER_MATCHER(userDefinedLiteral);
587e5dd7070Spatrick   REGISTER_MATCHER(usesADL);
588e5dd7070Spatrick   REGISTER_MATCHER(usingDecl);
589a9ac8606Spatrick   REGISTER_MATCHER(usingEnumDecl);
590e5dd7070Spatrick   REGISTER_MATCHER(usingDirectiveDecl);
591e5dd7070Spatrick   REGISTER_MATCHER(valueDecl);
592e5dd7070Spatrick   REGISTER_MATCHER(varDecl);
593e5dd7070Spatrick   REGISTER_MATCHER(variableArrayType);
594e5dd7070Spatrick   REGISTER_MATCHER(voidType);
595e5dd7070Spatrick   REGISTER_MATCHER(whileStmt);
596e5dd7070Spatrick   REGISTER_MATCHER(withInitializer);
597e5dd7070Spatrick }
598e5dd7070Spatrick 
599e5dd7070Spatrick RegistryMaps::~RegistryMaps() = default;
600e5dd7070Spatrick 
601e5dd7070Spatrick static llvm::ManagedStatic<RegistryMaps> RegistryData;
602e5dd7070Spatrick 
nodeMatcherType(MatcherCtor Ctor)603a9ac8606Spatrick ASTNodeKind Registry::nodeMatcherType(MatcherCtor Ctor) {
604a9ac8606Spatrick   return Ctor->nodeMatcherType();
605a9ac8606Spatrick }
606a9ac8606Spatrick 
MatcherDescriptorPtr(MatcherDescriptor * Ptr)607a9ac8606Spatrick internal::MatcherDescriptorPtr::MatcherDescriptorPtr(MatcherDescriptor *Ptr)
608a9ac8606Spatrick     : Ptr(Ptr) {}
609a9ac8606Spatrick 
~MatcherDescriptorPtr()610a9ac8606Spatrick internal::MatcherDescriptorPtr::~MatcherDescriptorPtr() { delete Ptr; }
611a9ac8606Spatrick 
isBuilderMatcher(MatcherCtor Ctor)612a9ac8606Spatrick bool Registry::isBuilderMatcher(MatcherCtor Ctor) {
613a9ac8606Spatrick   return Ctor->isBuilderMatcher();
614a9ac8606Spatrick }
615a9ac8606Spatrick 
616a9ac8606Spatrick internal::MatcherDescriptorPtr
buildMatcherCtor(MatcherCtor Ctor,SourceRange NameRange,ArrayRef<ParserValue> Args,Diagnostics * Error)617a9ac8606Spatrick Registry::buildMatcherCtor(MatcherCtor Ctor, SourceRange NameRange,
618a9ac8606Spatrick                            ArrayRef<ParserValue> Args, Diagnostics *Error) {
619a9ac8606Spatrick   return internal::MatcherDescriptorPtr(
620a9ac8606Spatrick       Ctor->buildMatcherCtor(NameRange, Args, Error).release());
621a9ac8606Spatrick }
622a9ac8606Spatrick 
623e5dd7070Spatrick // static
lookupMatcherCtor(StringRef MatcherName)624*12c85518Srobert std::optional<MatcherCtor> Registry::lookupMatcherCtor(StringRef MatcherName) {
625e5dd7070Spatrick   auto it = RegistryData->constructors().find(MatcherName);
626*12c85518Srobert   return it == RegistryData->constructors().end() ? std::optional<MatcherCtor>()
627e5dd7070Spatrick                                                   : it->second.get();
628e5dd7070Spatrick }
629e5dd7070Spatrick 
operator <<(llvm::raw_ostream & OS,const std::set<ASTNodeKind> & KS)630e5dd7070Spatrick static llvm::raw_ostream &operator<<(llvm::raw_ostream &OS,
631e5dd7070Spatrick                                      const std::set<ASTNodeKind> &KS) {
632e5dd7070Spatrick   unsigned Count = 0;
633e5dd7070Spatrick   for (std::set<ASTNodeKind>::const_iterator I = KS.begin(), E = KS.end();
634e5dd7070Spatrick        I != E; ++I) {
635e5dd7070Spatrick     if (I != KS.begin())
636e5dd7070Spatrick       OS << "|";
637e5dd7070Spatrick     if (Count++ == 3) {
638e5dd7070Spatrick       OS << "...";
639e5dd7070Spatrick       break;
640e5dd7070Spatrick     }
641e5dd7070Spatrick     OS << *I;
642e5dd7070Spatrick   }
643e5dd7070Spatrick   return OS;
644e5dd7070Spatrick }
645e5dd7070Spatrick 
getAcceptedCompletionTypes(ArrayRef<std::pair<MatcherCtor,unsigned>> Context)646e5dd7070Spatrick std::vector<ArgKind> Registry::getAcceptedCompletionTypes(
647e5dd7070Spatrick     ArrayRef<std::pair<MatcherCtor, unsigned>> Context) {
648e5dd7070Spatrick   ASTNodeKind InitialTypes[] = {
649e5dd7070Spatrick       ASTNodeKind::getFromNodeKind<Decl>(),
650e5dd7070Spatrick       ASTNodeKind::getFromNodeKind<QualType>(),
651e5dd7070Spatrick       ASTNodeKind::getFromNodeKind<Type>(),
652e5dd7070Spatrick       ASTNodeKind::getFromNodeKind<Stmt>(),
653e5dd7070Spatrick       ASTNodeKind::getFromNodeKind<NestedNameSpecifier>(),
654e5dd7070Spatrick       ASTNodeKind::getFromNodeKind<NestedNameSpecifierLoc>(),
655e5dd7070Spatrick       ASTNodeKind::getFromNodeKind<TypeLoc>()};
656e5dd7070Spatrick 
657e5dd7070Spatrick   // Starting with the above seed of acceptable top-level matcher types, compute
658e5dd7070Spatrick   // the acceptable type set for the argument indicated by each context element.
659a9ac8606Spatrick   std::set<ArgKind> TypeSet;
660a9ac8606Spatrick   for (auto IT : InitialTypes) {
661a9ac8606Spatrick     TypeSet.insert(ArgKind::MakeMatcherArg(IT));
662a9ac8606Spatrick   }
663e5dd7070Spatrick   for (const auto &CtxEntry : Context) {
664e5dd7070Spatrick     MatcherCtor Ctor = CtxEntry.first;
665e5dd7070Spatrick     unsigned ArgNumber = CtxEntry.second;
666e5dd7070Spatrick     std::vector<ArgKind> NextTypeSet;
667e5dd7070Spatrick     for (const ArgKind &Kind : TypeSet) {
668e5dd7070Spatrick       if (Kind.getArgKind() == Kind.AK_Matcher &&
669e5dd7070Spatrick           Ctor->isConvertibleTo(Kind.getMatcherKind()) &&
670e5dd7070Spatrick           (Ctor->isVariadic() || ArgNumber < Ctor->getNumArgs()))
671e5dd7070Spatrick         Ctor->getArgKinds(Kind.getMatcherKind(), ArgNumber, NextTypeSet);
672e5dd7070Spatrick     }
673e5dd7070Spatrick     TypeSet.clear();
674e5dd7070Spatrick     TypeSet.insert(NextTypeSet.begin(), NextTypeSet.end());
675e5dd7070Spatrick   }
676e5dd7070Spatrick   return std::vector<ArgKind>(TypeSet.begin(), TypeSet.end());
677e5dd7070Spatrick }
678e5dd7070Spatrick 
679e5dd7070Spatrick std::vector<MatcherCompletion>
getMatcherCompletions(ArrayRef<ArgKind> AcceptedTypes)680e5dd7070Spatrick Registry::getMatcherCompletions(ArrayRef<ArgKind> AcceptedTypes) {
681e5dd7070Spatrick   std::vector<MatcherCompletion> Completions;
682e5dd7070Spatrick 
683e5dd7070Spatrick   // Search the registry for acceptable matchers.
684e5dd7070Spatrick   for (const auto &M : RegistryData->constructors()) {
685e5dd7070Spatrick     const MatcherDescriptor& Matcher = *M.getValue();
686e5dd7070Spatrick     StringRef Name = M.getKey();
687e5dd7070Spatrick 
688e5dd7070Spatrick     std::set<ASTNodeKind> RetKinds;
689e5dd7070Spatrick     unsigned NumArgs = Matcher.isVariadic() ? 1 : Matcher.getNumArgs();
690e5dd7070Spatrick     bool IsPolymorphic = Matcher.isPolymorphic();
691e5dd7070Spatrick     std::vector<std::vector<ArgKind>> ArgsKinds(NumArgs);
692e5dd7070Spatrick     unsigned MaxSpecificity = 0;
693a9ac8606Spatrick     bool NodeArgs = false;
694e5dd7070Spatrick     for (const ArgKind& Kind : AcceptedTypes) {
695a9ac8606Spatrick       if (Kind.getArgKind() != Kind.AK_Matcher &&
696a9ac8606Spatrick           Kind.getArgKind() != Kind.AK_Node) {
697e5dd7070Spatrick         continue;
698a9ac8606Spatrick       }
699a9ac8606Spatrick 
700a9ac8606Spatrick       if (Kind.getArgKind() == Kind.AK_Node) {
701a9ac8606Spatrick         NodeArgs = true;
702a9ac8606Spatrick         unsigned Specificity;
703a9ac8606Spatrick         ASTNodeKind LeastDerivedKind;
704a9ac8606Spatrick         if (Matcher.isConvertibleTo(Kind.getNodeKind(), &Specificity,
705a9ac8606Spatrick                                     &LeastDerivedKind)) {
706a9ac8606Spatrick           if (MaxSpecificity < Specificity)
707a9ac8606Spatrick             MaxSpecificity = Specificity;
708a9ac8606Spatrick           RetKinds.insert(LeastDerivedKind);
709a9ac8606Spatrick           for (unsigned Arg = 0; Arg != NumArgs; ++Arg)
710a9ac8606Spatrick             Matcher.getArgKinds(Kind.getNodeKind(), Arg, ArgsKinds[Arg]);
711a9ac8606Spatrick           if (IsPolymorphic)
712a9ac8606Spatrick             break;
713a9ac8606Spatrick         }
714a9ac8606Spatrick       } else {
715e5dd7070Spatrick         unsigned Specificity;
716e5dd7070Spatrick         ASTNodeKind LeastDerivedKind;
717e5dd7070Spatrick         if (Matcher.isConvertibleTo(Kind.getMatcherKind(), &Specificity,
718e5dd7070Spatrick                                     &LeastDerivedKind)) {
719e5dd7070Spatrick           if (MaxSpecificity < Specificity)
720e5dd7070Spatrick             MaxSpecificity = Specificity;
721e5dd7070Spatrick           RetKinds.insert(LeastDerivedKind);
722e5dd7070Spatrick           for (unsigned Arg = 0; Arg != NumArgs; ++Arg)
723e5dd7070Spatrick             Matcher.getArgKinds(Kind.getMatcherKind(), Arg, ArgsKinds[Arg]);
724e5dd7070Spatrick           if (IsPolymorphic)
725e5dd7070Spatrick             break;
726e5dd7070Spatrick         }
727e5dd7070Spatrick       }
728a9ac8606Spatrick     }
729e5dd7070Spatrick 
730e5dd7070Spatrick     if (!RetKinds.empty() && MaxSpecificity > 0) {
731e5dd7070Spatrick       std::string Decl;
732e5dd7070Spatrick       llvm::raw_string_ostream OS(Decl);
733e5dd7070Spatrick 
734a9ac8606Spatrick       std::string TypedText = std::string(Name);
735a9ac8606Spatrick 
736a9ac8606Spatrick       if (NodeArgs) {
737a9ac8606Spatrick         OS << Name;
738a9ac8606Spatrick       } else {
739a9ac8606Spatrick 
740e5dd7070Spatrick         if (IsPolymorphic) {
741e5dd7070Spatrick           OS << "Matcher<T> " << Name << "(Matcher<T>";
742e5dd7070Spatrick         } else {
743e5dd7070Spatrick           OS << "Matcher<" << RetKinds << "> " << Name << "(";
744e5dd7070Spatrick           for (const std::vector<ArgKind> &Arg : ArgsKinds) {
745e5dd7070Spatrick             if (&Arg != &ArgsKinds[0])
746e5dd7070Spatrick               OS << ", ";
747e5dd7070Spatrick 
748e5dd7070Spatrick             bool FirstArgKind = true;
749e5dd7070Spatrick             std::set<ASTNodeKind> MatcherKinds;
750e5dd7070Spatrick             // Two steps. First all non-matchers, then matchers only.
751e5dd7070Spatrick             for (const ArgKind &AK : Arg) {
752e5dd7070Spatrick               if (AK.getArgKind() == ArgKind::AK_Matcher) {
753e5dd7070Spatrick                 MatcherKinds.insert(AK.getMatcherKind());
754e5dd7070Spatrick               } else {
755a9ac8606Spatrick                 if (!FirstArgKind)
756a9ac8606Spatrick                   OS << "|";
757e5dd7070Spatrick                 FirstArgKind = false;
758e5dd7070Spatrick                 OS << AK.asString();
759e5dd7070Spatrick               }
760e5dd7070Spatrick             }
761e5dd7070Spatrick             if (!MatcherKinds.empty()) {
762e5dd7070Spatrick               if (!FirstArgKind) OS << "|";
763e5dd7070Spatrick               OS << "Matcher<" << MatcherKinds << ">";
764e5dd7070Spatrick             }
765e5dd7070Spatrick           }
766e5dd7070Spatrick         }
767e5dd7070Spatrick         if (Matcher.isVariadic())
768e5dd7070Spatrick           OS << "...";
769e5dd7070Spatrick         OS << ")";
770e5dd7070Spatrick 
771e5dd7070Spatrick         TypedText += "(";
772e5dd7070Spatrick         if (ArgsKinds.empty())
773e5dd7070Spatrick           TypedText += ")";
774e5dd7070Spatrick         else if (ArgsKinds[0][0].getArgKind() == ArgKind::AK_String)
775e5dd7070Spatrick           TypedText += "\"";
776a9ac8606Spatrick       }
777e5dd7070Spatrick 
778e5dd7070Spatrick       Completions.emplace_back(TypedText, OS.str(), MaxSpecificity);
779e5dd7070Spatrick     }
780e5dd7070Spatrick   }
781e5dd7070Spatrick 
782e5dd7070Spatrick   return Completions;
783e5dd7070Spatrick }
784e5dd7070Spatrick 
constructMatcher(MatcherCtor Ctor,SourceRange NameRange,ArrayRef<ParserValue> Args,Diagnostics * Error)785e5dd7070Spatrick VariantMatcher Registry::constructMatcher(MatcherCtor Ctor,
786e5dd7070Spatrick                                           SourceRange NameRange,
787e5dd7070Spatrick                                           ArrayRef<ParserValue> Args,
788e5dd7070Spatrick                                           Diagnostics *Error) {
789e5dd7070Spatrick   return Ctor->create(NameRange, Args, Error);
790e5dd7070Spatrick }
791e5dd7070Spatrick 
constructBoundMatcher(MatcherCtor Ctor,SourceRange NameRange,StringRef BindID,ArrayRef<ParserValue> Args,Diagnostics * Error)792e5dd7070Spatrick VariantMatcher Registry::constructBoundMatcher(MatcherCtor Ctor,
793e5dd7070Spatrick                                                SourceRange NameRange,
794e5dd7070Spatrick                                                StringRef BindID,
795e5dd7070Spatrick                                                ArrayRef<ParserValue> Args,
796e5dd7070Spatrick                                                Diagnostics *Error) {
797e5dd7070Spatrick   VariantMatcher Out = constructMatcher(Ctor, NameRange, Args, Error);
798e5dd7070Spatrick   if (Out.isNull()) return Out;
799e5dd7070Spatrick 
800*12c85518Srobert   std::optional<DynTypedMatcher> Result = Out.getSingleMatcher();
801*12c85518Srobert   if (Result) {
802*12c85518Srobert     std::optional<DynTypedMatcher> Bound = Result->tryBind(BindID);
803*12c85518Srobert     if (Bound) {
804e5dd7070Spatrick       return VariantMatcher::SingleMatcher(*Bound);
805e5dd7070Spatrick     }
806e5dd7070Spatrick   }
807e5dd7070Spatrick   Error->addError(NameRange, Error->ET_RegistryNotBindable);
808e5dd7070Spatrick   return VariantMatcher();
809e5dd7070Spatrick }
810e5dd7070Spatrick 
811e5dd7070Spatrick } // namespace dynamic
812e5dd7070Spatrick } // namespace ast_matchers
813e5dd7070Spatrick } // namespace clang
814