Registry.cpp revision 321369 
1//===--- Registry.cpp - Matcher registry -------------------------===//
2//
3//                     The LLVM Compiler Infrastructure
4//
5// This file is distributed under the University of Illinois Open Source
6// License. See LICENSE.TXT for details.
7//
8//===------------------------------------------------------------===//
9///
10/// \file
11/// \brief Registry map populated at static initialization time.
12///
13//===------------------------------------------------------------===//
14
15#include "clang/ASTMatchers/Dynamic/Registry.h"
16#include "Marshallers.h"
17#include "clang/ASTMatchers/ASTMatchers.h"
18#include "llvm/ADT/STLExtras.h"
19#include "llvm/ADT/StringMap.h"
20#include "llvm/ADT/StringRef.h"
21#include "llvm/Support/ManagedStatic.h"
22#include <set>
23#include <utility>
24
25using namespace clang::ast_type_traits;
26
27namespace clang {
28namespace ast_matchers {
29namespace dynamic {
30namespace {
31
32using internal::MatcherDescriptor;
33
34typedef llvm::StringMap<std::unique_ptr<const MatcherDescriptor>> ConstructorMap;
35class RegistryMaps {
36public:
37  RegistryMaps();
38  ~RegistryMaps();
39
40  const ConstructorMap &constructors() const { return Constructors; }
41
42private:
43  void registerMatcher(StringRef MatcherName,
44                       std::unique_ptr<MatcherDescriptor> Callback);
45
46  ConstructorMap Constructors;
47};
48
49void RegistryMaps::registerMatcher(
50    StringRef MatcherName, std::unique_ptr<MatcherDescriptor> Callback) {
51  assert(Constructors.find(MatcherName) == Constructors.end());
52  Constructors[MatcherName] = std::move(Callback);
53}
54
55#define REGISTER_MATCHER(name)                                                 \
56  registerMatcher(#name, internal::makeMatcherAutoMarshall(                    \
57                             ::clang::ast_matchers::name, #name));
58
59#define REGISTER_MATCHER_OVERLOAD(name)                                        \
60  registerMatcher(#name,                                                       \
61      llvm::make_unique<internal::OverloadedMatcherDescriptor>(name##Callbacks))
62
63#define SPECIFIC_MATCHER_OVERLOAD(name, Id)                                    \
64  static_cast<::clang::ast_matchers::name##_Type##Id>(                         \
65      ::clang::ast_matchers::name)
66
67#define MATCHER_OVERLOAD_ENTRY(name, Id)                                       \
68        internal::makeMatcherAutoMarshall(SPECIFIC_MATCHER_OVERLOAD(name, Id), \
69                                          #name)
70
71#define REGISTER_OVERLOADED_2(name)                                            \
72  do {                                                                         \
73    std::unique_ptr<MatcherDescriptor> name##Callbacks[] = {                   \
74        MATCHER_OVERLOAD_ENTRY(name, 0),                                       \
75        MATCHER_OVERLOAD_ENTRY(name, 1)};                                      \
76    REGISTER_MATCHER_OVERLOAD(name);                                           \
77  } while (0)
78
79/// \brief Generate a registry map with all the known matchers.
80RegistryMaps::RegistryMaps() {
81  // TODO: Here is the list of the missing matchers, grouped by reason.
82  //
83  // Need Variant/Parser fixes:
84  // ofKind
85  //
86  // Polymorphic + argument overload:
87  // findAll
88  //
89  // Other:
90  // equalsNode
91
92  REGISTER_OVERLOADED_2(callee);
93  REGISTER_OVERLOADED_2(hasPrefix);
94  REGISTER_OVERLOADED_2(hasType);
95  REGISTER_OVERLOADED_2(isDerivedFrom);
96  REGISTER_OVERLOADED_2(isSameOrDerivedFrom);
97  REGISTER_OVERLOADED_2(loc);
98  REGISTER_OVERLOADED_2(pointsTo);
99  REGISTER_OVERLOADED_2(references);
100  REGISTER_OVERLOADED_2(thisPointerType);
101
102  std::unique_ptr<MatcherDescriptor> equalsCallbacks[] = {
103      MATCHER_OVERLOAD_ENTRY(equals, 0),
104      MATCHER_OVERLOAD_ENTRY(equals, 1),
105      MATCHER_OVERLOAD_ENTRY(equals, 2),
106  };
107  REGISTER_MATCHER_OVERLOAD(equals);
108
109  REGISTER_MATCHER(accessSpecDecl);
110  REGISTER_MATCHER(addrLabelExpr);
111  REGISTER_MATCHER(alignOfExpr);
112  REGISTER_MATCHER(allOf);
113  REGISTER_MATCHER(anyOf);
114  REGISTER_MATCHER(anything);
115  REGISTER_MATCHER(argumentCountIs);
116  REGISTER_MATCHER(arraySubscriptExpr);
117  REGISTER_MATCHER(arrayType);
118  REGISTER_MATCHER(asmStmt);
119  REGISTER_MATCHER(asString);
120  REGISTER_MATCHER(atomicExpr);
121  REGISTER_MATCHER(atomicType);
122  REGISTER_MATCHER(autoType);
123  REGISTER_MATCHER(binaryOperator);
124  REGISTER_MATCHER(binaryConditionalOperator);
125  REGISTER_MATCHER(blockPointerType);
126  REGISTER_MATCHER(booleanType);
127  REGISTER_MATCHER(breakStmt);
128  REGISTER_MATCHER(builtinType);
129  REGISTER_MATCHER(callExpr);
130  REGISTER_MATCHER(caseStmt);
131  REGISTER_MATCHER(castExpr);
132  REGISTER_MATCHER(characterLiteral);
133  REGISTER_MATCHER(classTemplateDecl);
134  REGISTER_MATCHER(classTemplateSpecializationDecl);
135  REGISTER_MATCHER(complexType);
136  REGISTER_MATCHER(compoundLiteralExpr);
137  REGISTER_MATCHER(compoundStmt);
138  REGISTER_MATCHER(conditionalOperator);
139  REGISTER_MATCHER(constantArrayType);
140  REGISTER_MATCHER(containsDeclaration);
141  REGISTER_MATCHER(continueStmt);
142  REGISTER_MATCHER(cStyleCastExpr);
143  REGISTER_MATCHER(cudaKernelCallExpr);
144  REGISTER_MATCHER(cxxBindTemporaryExpr);
145  REGISTER_MATCHER(cxxBoolLiteral);
146  REGISTER_MATCHER(cxxCatchStmt);
147  REGISTER_MATCHER(cxxConstCastExpr);
148  REGISTER_MATCHER(cxxConstructExpr);
149  REGISTER_MATCHER(cxxConstructorDecl);
150  REGISTER_MATCHER(cxxConversionDecl);
151  REGISTER_MATCHER(cxxCtorInitializer);
152  REGISTER_MATCHER(cxxDefaultArgExpr);
153  REGISTER_MATCHER(cxxDeleteExpr);
154  REGISTER_MATCHER(cxxDestructorDecl);
155  REGISTER_MATCHER(cxxDynamicCastExpr);
156  REGISTER_MATCHER(cxxForRangeStmt);
157  REGISTER_MATCHER(cxxFunctionalCastExpr);
158  REGISTER_MATCHER(cxxMemberCallExpr);
159  REGISTER_MATCHER(cxxMethodDecl);
160  REGISTER_MATCHER(cxxNewExpr);
161  REGISTER_MATCHER(cxxNullPtrLiteralExpr);
162  REGISTER_MATCHER(cxxOperatorCallExpr);
163  REGISTER_MATCHER(cxxRecordDecl);
164  REGISTER_MATCHER(cxxReinterpretCastExpr);
165  REGISTER_MATCHER(cxxStaticCastExpr);
166  REGISTER_MATCHER(cxxStdInitializerListExpr);
167  REGISTER_MATCHER(cxxTemporaryObjectExpr);
168  REGISTER_MATCHER(cxxThisExpr);
169  REGISTER_MATCHER(cxxThrowExpr);
170  REGISTER_MATCHER(cxxTryStmt);
171  REGISTER_MATCHER(cxxUnresolvedConstructExpr);
172  REGISTER_MATCHER(decayedType);
173  REGISTER_MATCHER(decl);
174  REGISTER_MATCHER(declaratorDecl);
175  REGISTER_MATCHER(declCountIs);
176  REGISTER_MATCHER(declRefExpr);
177  REGISTER_MATCHER(declStmt);
178  REGISTER_MATCHER(defaultStmt);
179  REGISTER_MATCHER(dependentSizedArrayType);
180  REGISTER_MATCHER(designatedInitExpr);
181  REGISTER_MATCHER(designatorCountIs);
182  REGISTER_MATCHER(doStmt);
183  REGISTER_MATCHER(eachOf);
184  REGISTER_MATCHER(elaboratedType);
185  REGISTER_MATCHER(enumConstantDecl);
186  REGISTER_MATCHER(enumDecl);
187  REGISTER_MATCHER(enumType);
188  REGISTER_MATCHER(equalsBoundNode);
189  REGISTER_MATCHER(equalsIntegralValue);
190  REGISTER_MATCHER(explicitCastExpr);
191  REGISTER_MATCHER(expr);
192  REGISTER_MATCHER(exprWithCleanups);
193  REGISTER_MATCHER(fieldDecl);
194  REGISTER_MATCHER(floatLiteral);
195  REGISTER_MATCHER(forEach);
196  REGISTER_MATCHER(forEachArgumentWithParam);
197  REGISTER_MATCHER(forEachConstructorInitializer);
198  REGISTER_MATCHER(forEachDescendant);
199  REGISTER_MATCHER(forEachSwitchCase);
200  REGISTER_MATCHER(forField);
201  REGISTER_MATCHER(forFunction);
202  REGISTER_MATCHER(forStmt);
203  REGISTER_MATCHER(friendDecl);
204  REGISTER_MATCHER(functionDecl);
205  REGISTER_MATCHER(functionProtoType);
206  REGISTER_MATCHER(functionTemplateDecl);
207  REGISTER_MATCHER(functionType);
208  REGISTER_MATCHER(gnuNullExpr);
209  REGISTER_MATCHER(gotoStmt);
210  REGISTER_MATCHER(has);
211  REGISTER_MATCHER(hasAncestor);
212  REGISTER_MATCHER(hasAnyArgument);
213  REGISTER_MATCHER(hasAnyConstructorInitializer);
214  REGISTER_MATCHER(hasAnyDeclaration);
215  REGISTER_MATCHER(hasAnyName);
216  REGISTER_MATCHER(hasAnyParameter);
217  REGISTER_MATCHER(hasAnySubstatement);
218  REGISTER_MATCHER(hasAnyTemplateArgument);
219  REGISTER_MATCHER(hasAnyUsingShadowDecl);
220  REGISTER_MATCHER(hasArgument);
221  REGISTER_MATCHER(hasArgumentOfType);
222  REGISTER_MATCHER(hasAttr);
223  REGISTER_MATCHER(hasAutomaticStorageDuration);
224  REGISTER_MATCHER(hasBase);
225  REGISTER_MATCHER(hasBitWidth);
226  REGISTER_MATCHER(hasBody);
227  REGISTER_MATCHER(hasCanonicalType);
228  REGISTER_MATCHER(hasCaseConstant);
229  REGISTER_MATCHER(hasCastKind);
230  REGISTER_MATCHER(hasCondition);
231  REGISTER_MATCHER(hasConditionVariableStatement);
232  REGISTER_MATCHER(hasDecayedType);
233  REGISTER_MATCHER(hasDeclaration);
234  REGISTER_MATCHER(hasDeclContext);
235  REGISTER_MATCHER(hasDeducedType);
236  REGISTER_MATCHER(hasDescendant);
237  REGISTER_MATCHER(hasDestinationType);
238  REGISTER_MATCHER(hasDynamicExceptionSpec);
239  REGISTER_MATCHER(hasEitherOperand);
240  REGISTER_MATCHER(hasElementType);
241  REGISTER_MATCHER(hasElse);
242  REGISTER_MATCHER(hasExternalFormalLinkage);
243  REGISTER_MATCHER(hasFalseExpression);
244  REGISTER_MATCHER(hasGlobalStorage);
245  REGISTER_MATCHER(hasImplicitDestinationType);
246  REGISTER_MATCHER(hasInClassInitializer);
247  REGISTER_MATCHER(hasIncrement);
248  REGISTER_MATCHER(hasIndex);
249  REGISTER_MATCHER(hasInitializer);
250  REGISTER_MATCHER(hasKeywordSelector);
251  REGISTER_MATCHER(hasLHS);
252  REGISTER_MATCHER(hasLocalQualifiers);
253  REGISTER_MATCHER(hasLocalStorage);
254  REGISTER_MATCHER(hasLoopInit);
255  REGISTER_MATCHER(hasLoopVariable);
256  REGISTER_MATCHER(hasMethod);
257  REGISTER_MATCHER(hasName);
258  REGISTER_MATCHER(hasNullSelector);
259  REGISTER_MATCHER(hasObjectExpression);
260  REGISTER_MATCHER(hasOperatorName);
261  REGISTER_MATCHER(hasOverloadedOperatorName);
262  REGISTER_MATCHER(hasParameter);
263  REGISTER_MATCHER(hasParent);
264  REGISTER_MATCHER(hasQualifier);
265  REGISTER_MATCHER(hasRangeInit);
266  REGISTER_MATCHER(hasReceiverType);
267  REGISTER_MATCHER(hasReplacementType);
268  REGISTER_MATCHER(hasReturnValue);
269  REGISTER_MATCHER(hasRHS);
270  REGISTER_MATCHER(hasSelector);
271  REGISTER_MATCHER(hasSingleDecl);
272  REGISTER_MATCHER(hasSize);
273  REGISTER_MATCHER(hasSizeExpr);
274  REGISTER_MATCHER(hasSourceExpression);
275  REGISTER_MATCHER(hasStaticStorageDuration);
276  REGISTER_MATCHER(hasSyntacticForm);
277  REGISTER_MATCHER(hasTargetDecl);
278  REGISTER_MATCHER(hasTemplateArgument);
279  REGISTER_MATCHER(hasThen);
280  REGISTER_MATCHER(hasThreadStorageDuration);
281  REGISTER_MATCHER(hasTrueExpression);
282  REGISTER_MATCHER(hasTypeLoc);
283  REGISTER_MATCHER(hasUnaryOperand);
284  REGISTER_MATCHER(hasUnarySelector);
285  REGISTER_MATCHER(hasUnderlyingDecl);
286  REGISTER_MATCHER(hasUnqualifiedDesugaredType);
287  REGISTER_MATCHER(hasValueType);
288  REGISTER_MATCHER(ifStmt);
289  REGISTER_MATCHER(ignoringImplicit);
290  REGISTER_MATCHER(ignoringImpCasts);
291  REGISTER_MATCHER(ignoringParenCasts);
292  REGISTER_MATCHER(ignoringParenImpCasts);
293  REGISTER_MATCHER(ignoringParens);
294  REGISTER_MATCHER(implicitCastExpr);
295  REGISTER_MATCHER(implicitValueInitExpr);
296  REGISTER_MATCHER(incompleteArrayType);
297  REGISTER_MATCHER(initListExpr);
298  REGISTER_MATCHER(injectedClassNameType);
299  REGISTER_MATCHER(innerType);
300  REGISTER_MATCHER(integerLiteral);
301  REGISTER_MATCHER(isAnonymous);
302  REGISTER_MATCHER(isAnyCharacter);
303  REGISTER_MATCHER(isAnyPointer);
304  REGISTER_MATCHER(isArrow);
305  REGISTER_MATCHER(isBaseInitializer);
306  REGISTER_MATCHER(isBitField);
307  REGISTER_MATCHER(isCatchAll);
308  REGISTER_MATCHER(isClass);
309  REGISTER_MATCHER(isConst);
310  REGISTER_MATCHER(isConstexpr);
311  REGISTER_MATCHER(isConstQualified);
312  REGISTER_MATCHER(isCopyAssignmentOperator);
313  REGISTER_MATCHER(isCopyConstructor);
314  REGISTER_MATCHER(isDefaultConstructor);
315  REGISTER_MATCHER(isDefaulted);
316  REGISTER_MATCHER(isDefinition);
317  REGISTER_MATCHER(isDeleted);
318  REGISTER_MATCHER(isExceptionVariable);
319  REGISTER_MATCHER(isExplicit);
320  REGISTER_MATCHER(isExplicitTemplateSpecialization);
321  REGISTER_MATCHER(isExpr);
322  REGISTER_MATCHER(isExternC);
323  REGISTER_MATCHER(isFinal);
324  REGISTER_MATCHER(isInline);
325  REGISTER_MATCHER(isImplicit);
326  REGISTER_MATCHER(isExpansionInFileMatching);
327  REGISTER_MATCHER(isExpansionInMainFile);
328  REGISTER_MATCHER(isInstantiated);
329  REGISTER_MATCHER(isExpansionInSystemHeader);
330  REGISTER_MATCHER(isInteger);
331  REGISTER_MATCHER(isIntegral);
332  REGISTER_MATCHER(isInTemplateInstantiation);
333  REGISTER_MATCHER(isLambda);
334  REGISTER_MATCHER(isListInitialization);
335  REGISTER_MATCHER(isMemberInitializer);
336  REGISTER_MATCHER(isMoveAssignmentOperator);
337  REGISTER_MATCHER(isMoveConstructor);
338  REGISTER_MATCHER(isNoThrow);
339  REGISTER_MATCHER(isOverride);
340  REGISTER_MATCHER(isPrivate);
341  REGISTER_MATCHER(isProtected);
342  REGISTER_MATCHER(isPublic);
343  REGISTER_MATCHER(isPure);
344  REGISTER_MATCHER(isSignedInteger);
345  REGISTER_MATCHER(isStaticStorageClass);
346  REGISTER_MATCHER(isStruct);
347  REGISTER_MATCHER(isTemplateInstantiation);
348  REGISTER_MATCHER(isUnion);
349  REGISTER_MATCHER(isUnsignedInteger);
350  REGISTER_MATCHER(isVariadic);
351  REGISTER_MATCHER(isVirtual);
352  REGISTER_MATCHER(isVirtualAsWritten);
353  REGISTER_MATCHER(isVolatileQualified);
354  REGISTER_MATCHER(isWritten);
355  REGISTER_MATCHER(labelDecl);
356  REGISTER_MATCHER(labelStmt);
357  REGISTER_MATCHER(lambdaExpr);
358  REGISTER_MATCHER(lValueReferenceType);
359  REGISTER_MATCHER(matchesName);
360  REGISTER_MATCHER(matchesSelector);
361  REGISTER_MATCHER(materializeTemporaryExpr);
362  REGISTER_MATCHER(member);
363  REGISTER_MATCHER(memberExpr);
364  REGISTER_MATCHER(memberPointerType);
365  REGISTER_MATCHER(namedDecl);
366  REGISTER_MATCHER(namespaceAliasDecl);
367  REGISTER_MATCHER(namespaceDecl);
368  REGISTER_MATCHER(namesType);
369  REGISTER_MATCHER(nestedNameSpecifier);
370  REGISTER_MATCHER(nestedNameSpecifierLoc);
371  REGISTER_MATCHER(nullPointerConstant);
372  REGISTER_MATCHER(nullStmt);
373  REGISTER_MATCHER(numSelectorArgs);
374  REGISTER_MATCHER(ofClass);
375  REGISTER_MATCHER(objcCategoryDecl);
376  REGISTER_MATCHER(objcInterfaceDecl);
377  REGISTER_MATCHER(objcIvarDecl);
378  REGISTER_MATCHER(objcMessageExpr);
379  REGISTER_MATCHER(objcMethodDecl);
380  REGISTER_MATCHER(objcObjectPointerType);
381  REGISTER_MATCHER(objcPropertyDecl);
382  REGISTER_MATCHER(objcProtocolDecl);
383  REGISTER_MATCHER(on);
384  REGISTER_MATCHER(onImplicitObjectArgument);
385  REGISTER_MATCHER(opaqueValueExpr);
386  REGISTER_MATCHER(parameterCountIs);
387  REGISTER_MATCHER(parenExpr);
388  REGISTER_MATCHER(parenListExpr);
389  REGISTER_MATCHER(parenType);
390  REGISTER_MATCHER(parmVarDecl);
391  REGISTER_MATCHER(pointee);
392  REGISTER_MATCHER(pointerType);
393  REGISTER_MATCHER(predefinedExpr);
394  REGISTER_MATCHER(qualType);
395  REGISTER_MATCHER(realFloatingPointType);
396  REGISTER_MATCHER(recordDecl);
397  REGISTER_MATCHER(recordType);
398  REGISTER_MATCHER(referenceType);
399  REGISTER_MATCHER(refersToDeclaration);
400  REGISTER_MATCHER(refersToIntegralType);
401  REGISTER_MATCHER(refersToType);
402  REGISTER_MATCHER(requiresZeroInitialization);
403  REGISTER_MATCHER(returns);
404  REGISTER_MATCHER(returnStmt);
405  REGISTER_MATCHER(rValueReferenceType);
406  REGISTER_MATCHER(sizeOfExpr);
407  REGISTER_MATCHER(specifiesNamespace);
408  REGISTER_MATCHER(specifiesType);
409  REGISTER_MATCHER(specifiesTypeLoc);
410  REGISTER_MATCHER(statementCountIs);
411  REGISTER_MATCHER(staticAssertDecl);
412  REGISTER_MATCHER(stmt);
413  REGISTER_MATCHER(stmtExpr);
414  REGISTER_MATCHER(stringLiteral);
415  REGISTER_MATCHER(substNonTypeTemplateParmExpr);
416  REGISTER_MATCHER(substTemplateTypeParmType);
417  REGISTER_MATCHER(switchCase);
418  REGISTER_MATCHER(switchStmt);
419  REGISTER_MATCHER(templateArgument);
420  REGISTER_MATCHER(templateName);
421  REGISTER_MATCHER(templateArgumentCountIs);
422  REGISTER_MATCHER(templateSpecializationType);
423  REGISTER_MATCHER(templateTypeParmDecl);
424  REGISTER_MATCHER(templateTypeParmType);
425  REGISTER_MATCHER(throughUsingDecl);
426  REGISTER_MATCHER(to);
427  REGISTER_MATCHER(translationUnitDecl);
428  REGISTER_MATCHER(type);
429  REGISTER_MATCHER(typedefDecl);
430  REGISTER_MATCHER(typedefNameDecl);
431  REGISTER_MATCHER(typedefType);
432  REGISTER_MATCHER(typeAliasDecl);
433  REGISTER_MATCHER(typeAliasTemplateDecl);
434  REGISTER_MATCHER(typeLoc);
435  REGISTER_MATCHER(unaryExprOrTypeTraitExpr);
436  REGISTER_MATCHER(unaryOperator);
437  REGISTER_MATCHER(unaryTransformType);
438  REGISTER_MATCHER(unless);
439  REGISTER_MATCHER(unresolvedLookupExpr);
440  REGISTER_MATCHER(unresolvedUsingTypenameDecl);
441  REGISTER_MATCHER(unresolvedUsingValueDecl);
442  REGISTER_MATCHER(userDefinedLiteral);
443  REGISTER_MATCHER(usingDecl);
444  REGISTER_MATCHER(usingDirectiveDecl);
445  REGISTER_MATCHER(valueDecl);
446  REGISTER_MATCHER(varDecl);
447  REGISTER_MATCHER(variableArrayType);
448  REGISTER_MATCHER(voidType);
449  REGISTER_MATCHER(whileStmt);
450  REGISTER_MATCHER(withInitializer);
451}
452
453RegistryMaps::~RegistryMaps() {}
454
455static llvm::ManagedStatic<RegistryMaps> RegistryData;
456
457} // anonymous namespace
458
459// static
460llvm::Optional<MatcherCtor> Registry::lookupMatcherCtor(StringRef MatcherName) {
461  auto it = RegistryData->constructors().find(MatcherName);
462  return it == RegistryData->constructors().end()
463             ? llvm::Optional<MatcherCtor>()
464             : it->second.get();
465}
466
467namespace {
468
469llvm::raw_ostream &operator<<(llvm::raw_ostream &OS,
470                              const std::set<ASTNodeKind> &KS) {
471  unsigned Count = 0;
472  for (std::set<ASTNodeKind>::const_iterator I = KS.begin(), E = KS.end();
473       I != E; ++I) {
474    if (I != KS.begin())
475      OS << "|";
476    if (Count++ == 3) {
477      OS << "...";
478      break;
479    }
480    OS << *I;
481  }
482  return OS;
483}
484
485}  // namespace
486
487std::vector<ArgKind> Registry::getAcceptedCompletionTypes(
488    ArrayRef<std::pair<MatcherCtor, unsigned>> Context) {
489  ASTNodeKind InitialTypes[] = {
490      ASTNodeKind::getFromNodeKind<Decl>(),
491      ASTNodeKind::getFromNodeKind<QualType>(),
492      ASTNodeKind::getFromNodeKind<Type>(),
493      ASTNodeKind::getFromNodeKind<Stmt>(),
494      ASTNodeKind::getFromNodeKind<NestedNameSpecifier>(),
495      ASTNodeKind::getFromNodeKind<NestedNameSpecifierLoc>(),
496      ASTNodeKind::getFromNodeKind<TypeLoc>()};
497
498  // Starting with the above seed of acceptable top-level matcher types, compute
499  // the acceptable type set for the argument indicated by each context element.
500  std::set<ArgKind> TypeSet(std::begin(InitialTypes), std::end(InitialTypes));
501  for (const auto &CtxEntry : Context) {
502    MatcherCtor Ctor = CtxEntry.first;
503    unsigned ArgNumber = CtxEntry.second;
504    std::vector<ArgKind> NextTypeSet;
505    for (const ArgKind &Kind : TypeSet) {
506      if (Kind.getArgKind() == Kind.AK_Matcher &&
507          Ctor->isConvertibleTo(Kind.getMatcherKind()) &&
508          (Ctor->isVariadic() || ArgNumber < Ctor->getNumArgs()))
509        Ctor->getArgKinds(Kind.getMatcherKind(), ArgNumber, NextTypeSet);
510    }
511    TypeSet.clear();
512    TypeSet.insert(NextTypeSet.begin(), NextTypeSet.end());
513  }
514  return std::vector<ArgKind>(TypeSet.begin(), TypeSet.end());
515}
516
517std::vector<MatcherCompletion>
518Registry::getMatcherCompletions(ArrayRef<ArgKind> AcceptedTypes) {
519  std::vector<MatcherCompletion> Completions;
520
521  // Search the registry for acceptable matchers.
522  for (const auto &M : RegistryData->constructors()) {
523    const MatcherDescriptor& Matcher = *M.getValue();
524    StringRef Name = M.getKey();
525
526    std::set<ASTNodeKind> RetKinds;
527    unsigned NumArgs = Matcher.isVariadic() ? 1 : Matcher.getNumArgs();
528    bool IsPolymorphic = Matcher.isPolymorphic();
529    std::vector<std::vector<ArgKind>> ArgsKinds(NumArgs);
530    unsigned MaxSpecificity = 0;
531    for (const ArgKind& Kind : AcceptedTypes) {
532      if (Kind.getArgKind() != Kind.AK_Matcher)
533        continue;
534      unsigned Specificity;
535      ASTNodeKind LeastDerivedKind;
536      if (Matcher.isConvertibleTo(Kind.getMatcherKind(), &Specificity,
537                                  &LeastDerivedKind)) {
538        if (MaxSpecificity < Specificity)
539          MaxSpecificity = Specificity;
540        RetKinds.insert(LeastDerivedKind);
541        for (unsigned Arg = 0; Arg != NumArgs; ++Arg)
542          Matcher.getArgKinds(Kind.getMatcherKind(), Arg, ArgsKinds[Arg]);
543        if (IsPolymorphic)
544          break;
545      }
546    }
547
548    if (!RetKinds.empty() && MaxSpecificity > 0) {
549      std::string Decl;
550      llvm::raw_string_ostream OS(Decl);
551
552      if (IsPolymorphic) {
553        OS << "Matcher<T> " << Name << "(Matcher<T>";
554      } else {
555        OS << "Matcher<" << RetKinds << "> " << Name << "(";
556        for (const std::vector<ArgKind> &Arg : ArgsKinds) {
557          if (&Arg != &ArgsKinds[0])
558            OS << ", ";
559
560          bool FirstArgKind = true;
561          std::set<ASTNodeKind> MatcherKinds;
562          // Two steps. First all non-matchers, then matchers only.
563          for (const ArgKind &AK : Arg) {
564            if (AK.getArgKind() == ArgKind::AK_Matcher) {
565              MatcherKinds.insert(AK.getMatcherKind());
566            } else {
567              if (!FirstArgKind) OS << "|";
568              FirstArgKind = false;
569              OS << AK.asString();
570            }
571          }
572          if (!MatcherKinds.empty()) {
573            if (!FirstArgKind) OS << "|";
574            OS << "Matcher<" << MatcherKinds << ">";
575          }
576        }
577      }
578      if (Matcher.isVariadic())
579        OS << "...";
580      OS << ")";
581
582      std::string TypedText = Name;
583      TypedText += "(";
584      if (ArgsKinds.empty())
585        TypedText += ")";
586      else if (ArgsKinds[0][0].getArgKind() == ArgKind::AK_String)
587        TypedText += "\"";
588
589      Completions.emplace_back(TypedText, OS.str(), MaxSpecificity);
590    }
591  }
592
593  return Completions;
594}
595
596// static
597VariantMatcher Registry::constructMatcher(MatcherCtor Ctor,
598                                          SourceRange NameRange,
599                                          ArrayRef<ParserValue> Args,
600                                          Diagnostics *Error) {
601  return Ctor->create(NameRange, Args, Error);
602}
603
604// static
605VariantMatcher Registry::constructBoundMatcher(MatcherCtor Ctor,
606                                               SourceRange NameRange,
607                                               StringRef BindID,
608                                               ArrayRef<ParserValue> Args,
609                                               Diagnostics *Error) {
610  VariantMatcher Out = constructMatcher(Ctor, NameRange, Args, Error);
611  if (Out.isNull()) return Out;
612
613  llvm::Optional<DynTypedMatcher> Result = Out.getSingleMatcher();
614  if (Result.hasValue()) {
615    llvm::Optional<DynTypedMatcher> Bound = Result->tryBind(BindID);
616    if (Bound.hasValue()) {
617      return VariantMatcher::SingleMatcher(*Bound);
618    }
619  }
620  Error->addError(NameRange, Error->ET_RegistryNotBindable);
621  return VariantMatcher();
622}
623
624}  // namespace dynamic
625}  // namespace ast_matchers
626}  // namespace clang
627