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