Registry.cpp revision 314564
1259701Sdim//===--- Registry.cpp - Matcher registry -------------------------===//
2259701Sdim//
3259701Sdim//                     The LLVM Compiler Infrastructure
4259701Sdim//
5259701Sdim// This file is distributed under the University of Illinois Open Source
6259701Sdim// License. See LICENSE.TXT for details.
7259701Sdim//
8259701Sdim//===------------------------------------------------------------===//
9259701Sdim///
10259701Sdim/// \file
11259701Sdim/// \brief Registry map populated at static initialization time.
12259701Sdim///
13259701Sdim//===------------------------------------------------------------===//
14259701Sdim
15259701Sdim#include "clang/ASTMatchers/Dynamic/Registry.h"
16259701Sdim#include "Marshallers.h"
17259701Sdim#include "clang/ASTMatchers/ASTMatchers.h"
18296417Sdim#include "llvm/ADT/STLExtras.h"
19259701Sdim#include "llvm/ADT/StringMap.h"
20259701Sdim#include "llvm/ADT/StringRef.h"
21259701Sdim#include "llvm/Support/ManagedStatic.h"
22276479Sdim#include <set>
23276479Sdim#include <utility>
24259701Sdim
25276479Sdimusing namespace clang::ast_type_traits;
26276479Sdim
27259701Sdimnamespace clang {
28259701Sdimnamespace ast_matchers {
29259701Sdimnamespace dynamic {
30259701Sdimnamespace {
31259701Sdim
32276479Sdimusing internal::MatcherDescriptor;
33259701Sdim
34314564Sdimtypedef llvm::StringMap<std::unique_ptr<const MatcherDescriptor>> ConstructorMap;
35259701Sdimclass RegistryMaps {
36259701Sdimpublic:
37259701Sdim  RegistryMaps();
38259701Sdim  ~RegistryMaps();
39259701Sdim
40259701Sdim  const ConstructorMap &constructors() const { return Constructors; }
41259701Sdim
42259701Sdimprivate:
43314564Sdim  void registerMatcher(StringRef MatcherName,
44314564Sdim                       std::unique_ptr<MatcherDescriptor> Callback);
45314564Sdim
46259701Sdim  ConstructorMap Constructors;
47259701Sdim};
48259701Sdim
49314564Sdimvoid RegistryMaps::registerMatcher(
50314564Sdim    StringRef MatcherName, std::unique_ptr<MatcherDescriptor> Callback) {
51259701Sdim  assert(Constructors.find(MatcherName) == Constructors.end());
52314564Sdim  Constructors[MatcherName] = std::move(Callback);
53259701Sdim}
54259701Sdim
55259701Sdim#define REGISTER_MATCHER(name)                                                 \
56259701Sdim  registerMatcher(#name, internal::makeMatcherAutoMarshall(                    \
57259701Sdim                             ::clang::ast_matchers::name, #name));
58259701Sdim
59259701Sdim#define SPECIFIC_MATCHER_OVERLOAD(name, Id)                                    \
60314564Sdim  static_cast<::clang::ast_matchers::name##_Type##Id>(                         \
61259701Sdim      ::clang::ast_matchers::name)
62259701Sdim
63259701Sdim#define REGISTER_OVERLOADED_2(name)                                            \
64259701Sdim  do {                                                                         \
65314564Sdim    std::unique_ptr<MatcherDescriptor> Callbacks[] = {                         \
66314564Sdim        internal::makeMatcherAutoMarshall(SPECIFIC_MATCHER_OVERLOAD(name, 0),  \
67314564Sdim                                          #name),                              \
68314564Sdim        internal::makeMatcherAutoMarshall(SPECIFIC_MATCHER_OVERLOAD(name, 1),  \
69314564Sdim                                          #name)};                             \
70314564Sdim    registerMatcher(                                                           \
71314564Sdim        #name,                                                                 \
72314564Sdim        llvm::make_unique<internal::OverloadedMatcherDescriptor>(Callbacks));  \
73259701Sdim  } while (0)
74259701Sdim
75259701Sdim/// \brief Generate a registry map with all the known matchers.
76259701SdimRegistryMaps::RegistryMaps() {
77259701Sdim  // TODO: Here is the list of the missing matchers, grouped by reason.
78259701Sdim  //
79259701Sdim  // Need Variant/Parser fixes:
80259701Sdim  // ofKind
81259701Sdim  //
82259701Sdim  // Polymorphic + argument overload:
83259701Sdim  // findAll
84259701Sdim  //
85259701Sdim  // Other:
86259701Sdim  // equals
87259701Sdim  // equalsNode
88259701Sdim
89259701Sdim  REGISTER_OVERLOADED_2(callee);
90259701Sdim  REGISTER_OVERLOADED_2(hasPrefix);
91259701Sdim  REGISTER_OVERLOADED_2(hasType);
92259701Sdim  REGISTER_OVERLOADED_2(isDerivedFrom);
93259701Sdim  REGISTER_OVERLOADED_2(isSameOrDerivedFrom);
94276479Sdim  REGISTER_OVERLOADED_2(loc);
95259701Sdim  REGISTER_OVERLOADED_2(pointsTo);
96259701Sdim  REGISTER_OVERLOADED_2(references);
97259701Sdim  REGISTER_OVERLOADED_2(thisPointerType);
98259701Sdim
99259701Sdim  REGISTER_MATCHER(accessSpecDecl);
100309124Sdim  REGISTER_MATCHER(addrLabelExpr);
101259701Sdim  REGISTER_MATCHER(alignOfExpr);
102259701Sdim  REGISTER_MATCHER(allOf);
103259701Sdim  REGISTER_MATCHER(anyOf);
104259701Sdim  REGISTER_MATCHER(anything);
105259701Sdim  REGISTER_MATCHER(argumentCountIs);
106259701Sdim  REGISTER_MATCHER(arraySubscriptExpr);
107259701Sdim  REGISTER_MATCHER(arrayType);
108280031Sdim  REGISTER_MATCHER(asmStmt);
109259701Sdim  REGISTER_MATCHER(asString);
110309124Sdim  REGISTER_MATCHER(atomicExpr);
111259701Sdim  REGISTER_MATCHER(atomicType);
112259701Sdim  REGISTER_MATCHER(autoType);
113259701Sdim  REGISTER_MATCHER(binaryOperator);
114309124Sdim  REGISTER_MATCHER(binaryConditionalOperator);
115259701Sdim  REGISTER_MATCHER(blockPointerType);
116296417Sdim  REGISTER_MATCHER(booleanType);
117259701Sdim  REGISTER_MATCHER(breakStmt);
118259701Sdim  REGISTER_MATCHER(builtinType);
119259701Sdim  REGISTER_MATCHER(callExpr);
120276479Sdim  REGISTER_MATCHER(caseStmt);
121259701Sdim  REGISTER_MATCHER(castExpr);
122259701Sdim  REGISTER_MATCHER(characterLiteral);
123259701Sdim  REGISTER_MATCHER(classTemplateDecl);
124259701Sdim  REGISTER_MATCHER(classTemplateSpecializationDecl);
125259701Sdim  REGISTER_MATCHER(complexType);
126259701Sdim  REGISTER_MATCHER(compoundLiteralExpr);
127259701Sdim  REGISTER_MATCHER(compoundStmt);
128259701Sdim  REGISTER_MATCHER(conditionalOperator);
129280031Sdim  REGISTER_MATCHER(constantArrayType);
130259701Sdim  REGISTER_MATCHER(containsDeclaration);
131259701Sdim  REGISTER_MATCHER(continueStmt);
132280031Sdim  REGISTER_MATCHER(cStyleCastExpr);
133296417Sdim  REGISTER_MATCHER(cudaKernelCallExpr);
134296417Sdim  REGISTER_MATCHER(cxxBindTemporaryExpr);
135296417Sdim  REGISTER_MATCHER(cxxBoolLiteral);
136296417Sdim  REGISTER_MATCHER(cxxCatchStmt);
137296417Sdim  REGISTER_MATCHER(cxxConstCastExpr);
138296417Sdim  REGISTER_MATCHER(cxxConstructExpr);
139296417Sdim  REGISTER_MATCHER(cxxConstructorDecl);
140296417Sdim  REGISTER_MATCHER(cxxConversionDecl);
141296417Sdim  REGISTER_MATCHER(cxxCtorInitializer);
142296417Sdim  REGISTER_MATCHER(cxxDefaultArgExpr);
143296417Sdim  REGISTER_MATCHER(cxxDeleteExpr);
144296417Sdim  REGISTER_MATCHER(cxxDestructorDecl);
145296417Sdim  REGISTER_MATCHER(cxxDynamicCastExpr);
146296417Sdim  REGISTER_MATCHER(cxxForRangeStmt);
147296417Sdim  REGISTER_MATCHER(cxxFunctionalCastExpr);
148296417Sdim  REGISTER_MATCHER(cxxMemberCallExpr);
149296417Sdim  REGISTER_MATCHER(cxxMethodDecl);
150296417Sdim  REGISTER_MATCHER(cxxNewExpr);
151296417Sdim  REGISTER_MATCHER(cxxNullPtrLiteralExpr);
152296417Sdim  REGISTER_MATCHER(cxxOperatorCallExpr);
153296417Sdim  REGISTER_MATCHER(cxxRecordDecl);
154296417Sdim  REGISTER_MATCHER(cxxReinterpretCastExpr);
155296417Sdim  REGISTER_MATCHER(cxxStaticCastExpr);
156296417Sdim  REGISTER_MATCHER(cxxTemporaryObjectExpr);
157296417Sdim  REGISTER_MATCHER(cxxThisExpr);
158296417Sdim  REGISTER_MATCHER(cxxThrowExpr);
159296417Sdim  REGISTER_MATCHER(cxxTryStmt);
160296417Sdim  REGISTER_MATCHER(cxxUnresolvedConstructExpr);
161296417Sdim  REGISTER_MATCHER(decayedType);
162259701Sdim  REGISTER_MATCHER(decl);
163280031Sdim  REGISTER_MATCHER(declaratorDecl);
164259701Sdim  REGISTER_MATCHER(declCountIs);
165259701Sdim  REGISTER_MATCHER(declRefExpr);
166259701Sdim  REGISTER_MATCHER(declStmt);
167276479Sdim  REGISTER_MATCHER(defaultStmt);
168259701Sdim  REGISTER_MATCHER(dependentSizedArrayType);
169309124Sdim  REGISTER_MATCHER(designatedInitExpr);
170309124Sdim  REGISTER_MATCHER(designatorCountIs);
171259701Sdim  REGISTER_MATCHER(doStmt);
172259701Sdim  REGISTER_MATCHER(eachOf);
173259701Sdim  REGISTER_MATCHER(elaboratedType);
174259701Sdim  REGISTER_MATCHER(enumConstantDecl);
175259701Sdim  REGISTER_MATCHER(enumDecl);
176309124Sdim  REGISTER_MATCHER(enumType);
177276479Sdim  REGISTER_MATCHER(equalsBoundNode);
178280031Sdim  REGISTER_MATCHER(equalsIntegralValue);
179259701Sdim  REGISTER_MATCHER(explicitCastExpr);
180259701Sdim  REGISTER_MATCHER(expr);
181276479Sdim  REGISTER_MATCHER(exprWithCleanups);
182259701Sdim  REGISTER_MATCHER(fieldDecl);
183259701Sdim  REGISTER_MATCHER(floatLiteral);
184259701Sdim  REGISTER_MATCHER(forEach);
185309124Sdim  REGISTER_MATCHER(forEachArgumentWithParam);
186276479Sdim  REGISTER_MATCHER(forEachConstructorInitializer);
187259701Sdim  REGISTER_MATCHER(forEachDescendant);
188276479Sdim  REGISTER_MATCHER(forEachSwitchCase);
189259701Sdim  REGISTER_MATCHER(forField);
190309124Sdim  REGISTER_MATCHER(forFunction);
191259701Sdim  REGISTER_MATCHER(forStmt);
192276479Sdim  REGISTER_MATCHER(friendDecl);
193259701Sdim  REGISTER_MATCHER(functionDecl);
194309124Sdim  REGISTER_MATCHER(functionProtoType);
195259701Sdim  REGISTER_MATCHER(functionTemplateDecl);
196259701Sdim  REGISTER_MATCHER(functionType);
197309124Sdim  REGISTER_MATCHER(gnuNullExpr);
198259701Sdim  REGISTER_MATCHER(gotoStmt);
199259701Sdim  REGISTER_MATCHER(has);
200259701Sdim  REGISTER_MATCHER(hasAncestor);
201259701Sdim  REGISTER_MATCHER(hasAnyArgument);
202259701Sdim  REGISTER_MATCHER(hasAnyConstructorInitializer);
203314564Sdim  REGISTER_MATCHER(hasAnyDeclaration);
204309124Sdim  REGISTER_MATCHER(hasAnyName);
205259701Sdim  REGISTER_MATCHER(hasAnyParameter);
206259701Sdim  REGISTER_MATCHER(hasAnySubstatement);
207259701Sdim  REGISTER_MATCHER(hasAnyTemplateArgument);
208259701Sdim  REGISTER_MATCHER(hasAnyUsingShadowDecl);
209259701Sdim  REGISTER_MATCHER(hasArgument);
210259701Sdim  REGISTER_MATCHER(hasArgumentOfType);
211280031Sdim  REGISTER_MATCHER(hasAttr);
212296417Sdim  REGISTER_MATCHER(hasAutomaticStorageDuration);
213259701Sdim  REGISTER_MATCHER(hasBase);
214309124Sdim  REGISTER_MATCHER(hasBitWidth);
215259701Sdim  REGISTER_MATCHER(hasBody);
216259701Sdim  REGISTER_MATCHER(hasCanonicalType);
217276479Sdim  REGISTER_MATCHER(hasCaseConstant);
218309124Sdim  REGISTER_MATCHER(hasCastKind);
219259701Sdim  REGISTER_MATCHER(hasCondition);
220259701Sdim  REGISTER_MATCHER(hasConditionVariableStatement);
221296417Sdim  REGISTER_MATCHER(hasDecayedType);
222280031Sdim  REGISTER_MATCHER(hasDeclaration);
223259701Sdim  REGISTER_MATCHER(hasDeclContext);
224259701Sdim  REGISTER_MATCHER(hasDeducedType);
225259701Sdim  REGISTER_MATCHER(hasDescendant);
226259701Sdim  REGISTER_MATCHER(hasDestinationType);
227309124Sdim  REGISTER_MATCHER(hasDynamicExceptionSpec);
228259701Sdim  REGISTER_MATCHER(hasEitherOperand);
229259701Sdim  REGISTER_MATCHER(hasElementType);
230280031Sdim  REGISTER_MATCHER(hasElse);
231314564Sdim  REGISTER_MATCHER(hasExternalFormalLinkage);
232259701Sdim  REGISTER_MATCHER(hasFalseExpression);
233276479Sdim  REGISTER_MATCHER(hasGlobalStorage);
234259701Sdim  REGISTER_MATCHER(hasImplicitDestinationType);
235314564Sdim  REGISTER_MATCHER(hasInClassInitializer);
236259701Sdim  REGISTER_MATCHER(hasIncrement);
237259701Sdim  REGISTER_MATCHER(hasIndex);
238259701Sdim  REGISTER_MATCHER(hasInitializer);
239288943Sdim  REGISTER_MATCHER(hasKeywordSelector);
240259701Sdim  REGISTER_MATCHER(hasLHS);
241259701Sdim  REGISTER_MATCHER(hasLocalQualifiers);
242276479Sdim  REGISTER_MATCHER(hasLocalStorage);
243259701Sdim  REGISTER_MATCHER(hasLoopInit);
244280031Sdim  REGISTER_MATCHER(hasLoopVariable);
245259701Sdim  REGISTER_MATCHER(hasMethod);
246259701Sdim  REGISTER_MATCHER(hasName);
247288943Sdim  REGISTER_MATCHER(hasNullSelector);
248259701Sdim  REGISTER_MATCHER(hasObjectExpression);
249259701Sdim  REGISTER_MATCHER(hasOperatorName);
250259701Sdim  REGISTER_MATCHER(hasOverloadedOperatorName);
251259701Sdim  REGISTER_MATCHER(hasParameter);
252259701Sdim  REGISTER_MATCHER(hasParent);
253259701Sdim  REGISTER_MATCHER(hasQualifier);
254280031Sdim  REGISTER_MATCHER(hasRangeInit);
255288943Sdim  REGISTER_MATCHER(hasReceiverType);
256314564Sdim  REGISTER_MATCHER(hasReplacementType);
257309124Sdim  REGISTER_MATCHER(hasReturnValue);
258259701Sdim  REGISTER_MATCHER(hasRHS);
259288943Sdim  REGISTER_MATCHER(hasSelector);
260259701Sdim  REGISTER_MATCHER(hasSingleDecl);
261259701Sdim  REGISTER_MATCHER(hasSize);
262259701Sdim  REGISTER_MATCHER(hasSizeExpr);
263259701Sdim  REGISTER_MATCHER(hasSourceExpression);
264296417Sdim  REGISTER_MATCHER(hasStaticStorageDuration);
265309124Sdim  REGISTER_MATCHER(hasSyntacticForm);
266259701Sdim  REGISTER_MATCHER(hasTargetDecl);
267259701Sdim  REGISTER_MATCHER(hasTemplateArgument);
268280031Sdim  REGISTER_MATCHER(hasThen);
269296417Sdim  REGISTER_MATCHER(hasThreadStorageDuration);
270259701Sdim  REGISTER_MATCHER(hasTrueExpression);
271276479Sdim  REGISTER_MATCHER(hasTypeLoc);
272259701Sdim  REGISTER_MATCHER(hasUnaryOperand);
273288943Sdim  REGISTER_MATCHER(hasUnarySelector);
274314564Sdim  REGISTER_MATCHER(hasUnderlyingDecl);
275314564Sdim  REGISTER_MATCHER(hasUnqualifiedDesugaredType);
276259701Sdim  REGISTER_MATCHER(hasValueType);
277259701Sdim  REGISTER_MATCHER(ifStmt);
278309124Sdim  REGISTER_MATCHER(ignoringImplicit);
279259701Sdim  REGISTER_MATCHER(ignoringImpCasts);
280259701Sdim  REGISTER_MATCHER(ignoringParenCasts);
281259701Sdim  REGISTER_MATCHER(ignoringParenImpCasts);
282309124Sdim  REGISTER_MATCHER(ignoringParens);
283259701Sdim  REGISTER_MATCHER(implicitCastExpr);
284309124Sdim  REGISTER_MATCHER(implicitValueInitExpr);
285259701Sdim  REGISTER_MATCHER(incompleteArrayType);
286259701Sdim  REGISTER_MATCHER(initListExpr);
287296417Sdim  REGISTER_MATCHER(injectedClassNameType);
288259701Sdim  REGISTER_MATCHER(innerType);
289259701Sdim  REGISTER_MATCHER(integerLiteral);
290296417Sdim  REGISTER_MATCHER(isAnonymous);
291309124Sdim  REGISTER_MATCHER(isAnyCharacter);
292309124Sdim  REGISTER_MATCHER(isAnyPointer);
293259701Sdim  REGISTER_MATCHER(isArrow);
294296417Sdim  REGISTER_MATCHER(isBaseInitializer);
295309124Sdim  REGISTER_MATCHER(isBitField);
296288943Sdim  REGISTER_MATCHER(isCatchAll);
297296417Sdim  REGISTER_MATCHER(isClass);
298276479Sdim  REGISTER_MATCHER(isConst);
299259701Sdim  REGISTER_MATCHER(isConstQualified);
300309124Sdim  REGISTER_MATCHER(isCopyAssignmentOperator);
301296417Sdim  REGISTER_MATCHER(isCopyConstructor);
302296417Sdim  REGISTER_MATCHER(isDefaultConstructor);
303309124Sdim  REGISTER_MATCHER(isDefaulted);
304259701Sdim  REGISTER_MATCHER(isDefinition);
305280031Sdim  REGISTER_MATCHER(isDeleted);
306296417Sdim  REGISTER_MATCHER(isExceptionVariable);
307296417Sdim  REGISTER_MATCHER(isExplicit);
308259701Sdim  REGISTER_MATCHER(isExplicitTemplateSpecialization);
309276479Sdim  REGISTER_MATCHER(isExpr);
310259701Sdim  REGISTER_MATCHER(isExternC);
311296417Sdim  REGISTER_MATCHER(isFinal);
312296417Sdim  REGISTER_MATCHER(isInline);
313259701Sdim  REGISTER_MATCHER(isImplicit);
314280031Sdim  REGISTER_MATCHER(isExpansionInFileMatching);
315280031Sdim  REGISTER_MATCHER(isExpansionInMainFile);
316280031Sdim  REGISTER_MATCHER(isInstantiated);
317280031Sdim  REGISTER_MATCHER(isExpansionInSystemHeader);
318259701Sdim  REGISTER_MATCHER(isInteger);
319280031Sdim  REGISTER_MATCHER(isIntegral);
320280031Sdim  REGISTER_MATCHER(isInTemplateInstantiation);
321309124Sdim  REGISTER_MATCHER(isLambda);
322276479Sdim  REGISTER_MATCHER(isListInitialization);
323296417Sdim  REGISTER_MATCHER(isMemberInitializer);
324309124Sdim  REGISTER_MATCHER(isMoveAssignmentOperator);
325296417Sdim  REGISTER_MATCHER(isMoveConstructor);
326296417Sdim  REGISTER_MATCHER(isNoThrow);
327259701Sdim  REGISTER_MATCHER(isOverride);
328259701Sdim  REGISTER_MATCHER(isPrivate);
329259701Sdim  REGISTER_MATCHER(isProtected);
330259701Sdim  REGISTER_MATCHER(isPublic);
331280031Sdim  REGISTER_MATCHER(isPure);
332309124Sdim  REGISTER_MATCHER(isSignedInteger);
333296417Sdim  REGISTER_MATCHER(isStruct);
334259701Sdim  REGISTER_MATCHER(isTemplateInstantiation);
335296417Sdim  REGISTER_MATCHER(isUnion);
336309124Sdim  REGISTER_MATCHER(isUnsignedInteger);
337296417Sdim  REGISTER_MATCHER(isVariadic);
338259701Sdim  REGISTER_MATCHER(isVirtual);
339309124Sdim  REGISTER_MATCHER(isVirtualAsWritten);
340296417Sdim  REGISTER_MATCHER(isVolatileQualified);
341259701Sdim  REGISTER_MATCHER(isWritten);
342309124Sdim  REGISTER_MATCHER(labelDecl);
343259701Sdim  REGISTER_MATCHER(labelStmt);
344259701Sdim  REGISTER_MATCHER(lambdaExpr);
345280031Sdim  REGISTER_MATCHER(lValueReferenceType);
346259701Sdim  REGISTER_MATCHER(matchesName);
347288943Sdim  REGISTER_MATCHER(matchesSelector);
348259701Sdim  REGISTER_MATCHER(materializeTemporaryExpr);
349259701Sdim  REGISTER_MATCHER(member);
350259701Sdim  REGISTER_MATCHER(memberExpr);
351259701Sdim  REGISTER_MATCHER(memberPointerType);
352259701Sdim  REGISTER_MATCHER(namedDecl);
353296417Sdim  REGISTER_MATCHER(namespaceAliasDecl);
354280031Sdim  REGISTER_MATCHER(namespaceDecl);
355259701Sdim  REGISTER_MATCHER(namesType);
356259701Sdim  REGISTER_MATCHER(nestedNameSpecifier);
357259701Sdim  REGISTER_MATCHER(nestedNameSpecifierLoc);
358309124Sdim  REGISTER_MATCHER(nullPointerConstant);
359259701Sdim  REGISTER_MATCHER(nullStmt);
360288943Sdim  REGISTER_MATCHER(numSelectorArgs);
361259701Sdim  REGISTER_MATCHER(ofClass);
362296417Sdim  REGISTER_MATCHER(objcInterfaceDecl);
363288943Sdim  REGISTER_MATCHER(objcMessageExpr);
364296417Sdim  REGISTER_MATCHER(objcObjectPointerType);
365259701Sdim  REGISTER_MATCHER(on);
366259701Sdim  REGISTER_MATCHER(onImplicitObjectArgument);
367309124Sdim  REGISTER_MATCHER(opaqueValueExpr);
368259701Sdim  REGISTER_MATCHER(parameterCountIs);
369309124Sdim  REGISTER_MATCHER(parenExpr);
370309124Sdim  REGISTER_MATCHER(parenListExpr);
371259701Sdim  REGISTER_MATCHER(parenType);
372276479Sdim  REGISTER_MATCHER(parmVarDecl);
373259701Sdim  REGISTER_MATCHER(pointee);
374259701Sdim  REGISTER_MATCHER(pointerType);
375309124Sdim  REGISTER_MATCHER(predefinedExpr);
376259701Sdim  REGISTER_MATCHER(qualType);
377309124Sdim  REGISTER_MATCHER(realFloatingPointType);
378259701Sdim  REGISTER_MATCHER(recordDecl);
379259701Sdim  REGISTER_MATCHER(recordType);
380259701Sdim  REGISTER_MATCHER(referenceType);
381259701Sdim  REGISTER_MATCHER(refersToDeclaration);
382280031Sdim  REGISTER_MATCHER(refersToIntegralType);
383259701Sdim  REGISTER_MATCHER(refersToType);
384309124Sdim  REGISTER_MATCHER(requiresZeroInitialization);
385280031Sdim  REGISTER_MATCHER(returns);
386259701Sdim  REGISTER_MATCHER(returnStmt);
387280031Sdim  REGISTER_MATCHER(rValueReferenceType);
388259701Sdim  REGISTER_MATCHER(sizeOfExpr);
389259701Sdim  REGISTER_MATCHER(specifiesNamespace);
390259701Sdim  REGISTER_MATCHER(specifiesType);
391259701Sdim  REGISTER_MATCHER(specifiesTypeLoc);
392259701Sdim  REGISTER_MATCHER(statementCountIs);
393288943Sdim  REGISTER_MATCHER(staticAssertDecl);
394259701Sdim  REGISTER_MATCHER(stmt);
395309124Sdim  REGISTER_MATCHER(stmtExpr);
396259701Sdim  REGISTER_MATCHER(stringLiteral);
397280031Sdim  REGISTER_MATCHER(substNonTypeTemplateParmExpr);
398296417Sdim  REGISTER_MATCHER(substTemplateTypeParmType);
399259701Sdim  REGISTER_MATCHER(switchCase);
400259701Sdim  REGISTER_MATCHER(switchStmt);
401280031Sdim  REGISTER_MATCHER(templateArgument);
402314564Sdim  REGISTER_MATCHER(templateName);
403280031Sdim  REGISTER_MATCHER(templateArgumentCountIs);
404259701Sdim  REGISTER_MATCHER(templateSpecializationType);
405314564Sdim  REGISTER_MATCHER(templateTypeParmDecl);
406296417Sdim  REGISTER_MATCHER(templateTypeParmType);
407259701Sdim  REGISTER_MATCHER(throughUsingDecl);
408259701Sdim  REGISTER_MATCHER(to);
409288943Sdim  REGISTER_MATCHER(translationUnitDecl);
410259701Sdim  REGISTER_MATCHER(type);
411280031Sdim  REGISTER_MATCHER(typedefDecl);
412309124Sdim  REGISTER_MATCHER(typedefNameDecl);
413280031Sdim  REGISTER_MATCHER(typedefType);
414309124Sdim  REGISTER_MATCHER(typeAliasDecl);
415259701Sdim  REGISTER_MATCHER(typeLoc);
416259701Sdim  REGISTER_MATCHER(unaryExprOrTypeTraitExpr);
417259701Sdim  REGISTER_MATCHER(unaryOperator);
418276479Sdim  REGISTER_MATCHER(unaryTransformType);
419276479Sdim  REGISTER_MATCHER(unless);
420309124Sdim  REGISTER_MATCHER(unresolvedLookupExpr);
421296417Sdim  REGISTER_MATCHER(unresolvedUsingTypenameDecl);
422276479Sdim  REGISTER_MATCHER(unresolvedUsingValueDecl);
423259701Sdim  REGISTER_MATCHER(userDefinedLiteral);
424259701Sdim  REGISTER_MATCHER(usingDecl);
425280031Sdim  REGISTER_MATCHER(usingDirectiveDecl);
426280031Sdim  REGISTER_MATCHER(valueDecl);
427259701Sdim  REGISTER_MATCHER(varDecl);
428259701Sdim  REGISTER_MATCHER(variableArrayType);
429280031Sdim  REGISTER_MATCHER(voidType);
430259701Sdim  REGISTER_MATCHER(whileStmt);
431259701Sdim  REGISTER_MATCHER(withInitializer);
432259701Sdim}
433259701Sdim
434314564SdimRegistryMaps::~RegistryMaps() {}
435259701Sdim
436259701Sdimstatic llvm::ManagedStatic<RegistryMaps> RegistryData;
437259701Sdim
438259701Sdim} // anonymous namespace
439259701Sdim
440259701Sdim// static
441276479Sdimllvm::Optional<MatcherCtor> Registry::lookupMatcherCtor(StringRef MatcherName) {
442314564Sdim  auto it = RegistryData->constructors().find(MatcherName);
443276479Sdim  return it == RegistryData->constructors().end()
444276479Sdim             ? llvm::Optional<MatcherCtor>()
445314564Sdim             : it->second.get();
446276479Sdim}
447276479Sdim
448276479Sdimnamespace {
449276479Sdim
450276479Sdimllvm::raw_ostream &operator<<(llvm::raw_ostream &OS,
451276479Sdim                              const std::set<ASTNodeKind> &KS) {
452276479Sdim  unsigned Count = 0;
453276479Sdim  for (std::set<ASTNodeKind>::const_iterator I = KS.begin(), E = KS.end();
454276479Sdim       I != E; ++I) {
455276479Sdim    if (I != KS.begin())
456276479Sdim      OS << "|";
457276479Sdim    if (Count++ == 3) {
458276479Sdim      OS << "...";
459276479Sdim      break;
460276479Sdim    }
461276479Sdim    OS << *I;
462259701Sdim  }
463276479Sdim  return OS;
464276479Sdim}
465259701Sdim
466280031Sdim}  // namespace
467276479Sdim
468280031Sdimstd::vector<ArgKind> Registry::getAcceptedCompletionTypes(
469280031Sdim    ArrayRef<std::pair<MatcherCtor, unsigned>> Context) {
470276479Sdim  ASTNodeKind InitialTypes[] = {
471280031Sdim      ASTNodeKind::getFromNodeKind<Decl>(),
472280031Sdim      ASTNodeKind::getFromNodeKind<QualType>(),
473280031Sdim      ASTNodeKind::getFromNodeKind<Type>(),
474280031Sdim      ASTNodeKind::getFromNodeKind<Stmt>(),
475280031Sdim      ASTNodeKind::getFromNodeKind<NestedNameSpecifier>(),
476280031Sdim      ASTNodeKind::getFromNodeKind<NestedNameSpecifierLoc>(),
477280031Sdim      ASTNodeKind::getFromNodeKind<TypeLoc>()};
478276479Sdim
479276479Sdim  // Starting with the above seed of acceptable top-level matcher types, compute
480276479Sdim  // the acceptable type set for the argument indicated by each context element.
481280031Sdim  std::set<ArgKind> TypeSet(std::begin(InitialTypes), std::end(InitialTypes));
482280031Sdim  for (const auto &CtxEntry : Context) {
483280031Sdim    MatcherCtor Ctor = CtxEntry.first;
484280031Sdim    unsigned ArgNumber = CtxEntry.second;
485280031Sdim    std::vector<ArgKind> NextTypeSet;
486280031Sdim    for (const ArgKind &Kind : TypeSet) {
487280031Sdim      if (Kind.getArgKind() == Kind.AK_Matcher &&
488280031Sdim          Ctor->isConvertibleTo(Kind.getMatcherKind()) &&
489280031Sdim          (Ctor->isVariadic() || ArgNumber < Ctor->getNumArgs()))
490280031Sdim        Ctor->getArgKinds(Kind.getMatcherKind(), ArgNumber, NextTypeSet);
491276479Sdim    }
492276479Sdim    TypeSet.clear();
493280031Sdim    TypeSet.insert(NextTypeSet.begin(), NextTypeSet.end());
494276479Sdim  }
495280031Sdim  return std::vector<ArgKind>(TypeSet.begin(), TypeSet.end());
496280031Sdim}
497276479Sdim
498280031Sdimstd::vector<MatcherCompletion>
499280031SdimRegistry::getMatcherCompletions(ArrayRef<ArgKind> AcceptedTypes) {
500280031Sdim  std::vector<MatcherCompletion> Completions;
501276479Sdim
502280031Sdim  // Search the registry for acceptable matchers.
503296417Sdim  for (const auto &M : RegistryData->constructors()) {
504314564Sdim    const MatcherDescriptor& Matcher = *M.getValue();
505296417Sdim    StringRef Name = M.getKey();
506296417Sdim
507276479Sdim    std::set<ASTNodeKind> RetKinds;
508314564Sdim    unsigned NumArgs = Matcher.isVariadic() ? 1 : Matcher.getNumArgs();
509314564Sdim    bool IsPolymorphic = Matcher.isPolymorphic();
510280031Sdim    std::vector<std::vector<ArgKind>> ArgsKinds(NumArgs);
511276479Sdim    unsigned MaxSpecificity = 0;
512280031Sdim    for (const ArgKind& Kind : AcceptedTypes) {
513280031Sdim      if (Kind.getArgKind() != Kind.AK_Matcher)
514280031Sdim        continue;
515276479Sdim      unsigned Specificity;
516276479Sdim      ASTNodeKind LeastDerivedKind;
517314564Sdim      if (Matcher.isConvertibleTo(Kind.getMatcherKind(), &Specificity,
518314564Sdim                                  &LeastDerivedKind)) {
519276479Sdim        if (MaxSpecificity < Specificity)
520276479Sdim          MaxSpecificity = Specificity;
521276479Sdim        RetKinds.insert(LeastDerivedKind);
522276479Sdim        for (unsigned Arg = 0; Arg != NumArgs; ++Arg)
523314564Sdim          Matcher.getArgKinds(Kind.getMatcherKind(), Arg, ArgsKinds[Arg]);
524276479Sdim        if (IsPolymorphic)
525276479Sdim          break;
526276479Sdim      }
527276479Sdim    }
528276479Sdim
529276479Sdim    if (!RetKinds.empty() && MaxSpecificity > 0) {
530276479Sdim      std::string Decl;
531276479Sdim      llvm::raw_string_ostream OS(Decl);
532276479Sdim
533276479Sdim      if (IsPolymorphic) {
534296417Sdim        OS << "Matcher<T> " << Name << "(Matcher<T>";
535276479Sdim      } else {
536296417Sdim        OS << "Matcher<" << RetKinds << "> " << Name << "(";
537280031Sdim        for (const std::vector<ArgKind> &Arg : ArgsKinds) {
538280031Sdim          if (&Arg != &ArgsKinds[0])
539276479Sdim            OS << ", ";
540280031Sdim
541280031Sdim          bool FirstArgKind = true;
542280031Sdim          std::set<ASTNodeKind> MatcherKinds;
543280031Sdim          // Two steps. First all non-matchers, then matchers only.
544280031Sdim          for (const ArgKind &AK : Arg) {
545280031Sdim            if (AK.getArgKind() == ArgKind::AK_Matcher) {
546280031Sdim              MatcherKinds.insert(AK.getMatcherKind());
547280031Sdim            } else {
548280031Sdim              if (!FirstArgKind) OS << "|";
549280031Sdim              FirstArgKind = false;
550280031Sdim              OS << AK.asString();
551280031Sdim            }
552280031Sdim          }
553280031Sdim          if (!MatcherKinds.empty()) {
554280031Sdim            if (!FirstArgKind) OS << "|";
555276479Sdim            OS << "Matcher<" << MatcherKinds << ">";
556276479Sdim          }
557276479Sdim        }
558276479Sdim      }
559314564Sdim      if (Matcher.isVariadic())
560276479Sdim        OS << "...";
561276479Sdim      OS << ")";
562276479Sdim
563296417Sdim      std::string TypedText = Name;
564276479Sdim      TypedText += "(";
565276479Sdim      if (ArgsKinds.empty())
566276479Sdim        TypedText += ")";
567280031Sdim      else if (ArgsKinds[0][0].getArgKind() == ArgKind::AK_String)
568276479Sdim        TypedText += "\"";
569276479Sdim
570280031Sdim      Completions.emplace_back(TypedText, OS.str(), MaxSpecificity);
571276479Sdim    }
572276479Sdim  }
573276479Sdim
574280031Sdim  return Completions;
575276479Sdim}
576276479Sdim
577259701Sdim// static
578276479SdimVariantMatcher Registry::constructMatcher(MatcherCtor Ctor,
579296417Sdim                                          SourceRange NameRange,
580276479Sdim                                          ArrayRef<ParserValue> Args,
581276479Sdim                                          Diagnostics *Error) {
582276479Sdim  return Ctor->create(NameRange, Args, Error);
583276479Sdim}
584276479Sdim
585276479Sdim// static
586276479SdimVariantMatcher Registry::constructBoundMatcher(MatcherCtor Ctor,
587296417Sdim                                               SourceRange NameRange,
588259701Sdim                                               StringRef BindID,
589259701Sdim                                               ArrayRef<ParserValue> Args,
590259701Sdim                                               Diagnostics *Error) {
591276479Sdim  VariantMatcher Out = constructMatcher(Ctor, NameRange, Args, Error);
592259701Sdim  if (Out.isNull()) return Out;
593259701Sdim
594259701Sdim  llvm::Optional<DynTypedMatcher> Result = Out.getSingleMatcher();
595259701Sdim  if (Result.hasValue()) {
596259701Sdim    llvm::Optional<DynTypedMatcher> Bound = Result->tryBind(BindID);
597259701Sdim    if (Bound.hasValue()) {
598259701Sdim      return VariantMatcher::SingleMatcher(*Bound);
599259701Sdim    }
600259701Sdim  }
601259701Sdim  Error->addError(NameRange, Error->ET_RegistryNotBindable);
602259701Sdim  return VariantMatcher();
603259701Sdim}
604259701Sdim
605259701Sdim}  // namespace dynamic
606259701Sdim}  // namespace ast_matchers
607259701Sdim}  // namespace clang
608