blob: 6f935620888f3092622d48994de6ea387ae2d7e9 [file] [log] [blame]
Samuel Benzaquena76d8cd2013-05-15 19:49:05 +00001//===--- Registry.cpp - Matcher registry -------------------------===//
Manuel Klimek24db0f02013-05-14 09:13:00 +00002//
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//
Samuel Benzaquena76d8cd2013-05-15 19:49:05 +00008//===------------------------------------------------------------===//
Manuel Klimek24db0f02013-05-14 09:13:00 +00009///
10/// \file
11/// \brief Registry map populated at static initialization time.
12///
Samuel Benzaquena76d8cd2013-05-15 19:49:05 +000013//===------------------------------------------------------------===//
Manuel Klimek24db0f02013-05-14 09:13:00 +000014
15#include "clang/ASTMatchers/Dynamic/Registry.h"
Manuel Klimek24db0f02013-05-14 09:13:00 +000016#include "Marshallers.h"
17#include "clang/ASTMatchers/ASTMatchers.h"
Aaron Ballman7596bcf2015-08-12 20:05:18 +000018#include "llvm/ADT/STLExtras.h"
Manuel Klimek24db0f02013-05-14 09:13:00 +000019#include "llvm/ADT/StringMap.h"
20#include "llvm/ADT/StringRef.h"
21#include "llvm/Support/ManagedStatic.h"
Peter Collingbourned32e28c2014-01-23 22:48:38 +000022#include <set>
Chandler Carruth5553d0d2014-01-07 11:51:46 +000023#include <utility>
Manuel Klimek24db0f02013-05-14 09:13:00 +000024
Peter Collingbourned32e28c2014-01-23 22:48:38 +000025using namespace clang::ast_type_traits;
26
Manuel Klimek24db0f02013-05-14 09:13:00 +000027namespace clang {
28namespace ast_matchers {
29namespace dynamic {
30namespace {
31
Peter Collingbournef43e6942013-11-23 01:34:36 +000032using internal::MatcherDescriptor;
Manuel Klimek24db0f02013-05-14 09:13:00 +000033
Justin Lebar82380d82016-10-10 16:26:40 +000034typedef llvm::StringMap<std::unique_ptr<const MatcherDescriptor>> ConstructorMap;
Manuel Klimek24db0f02013-05-14 09:13:00 +000035class RegistryMaps {
36public:
37 RegistryMaps();
38 ~RegistryMaps();
39
40 const ConstructorMap &constructors() const { return Constructors; }
41
42private:
Justin Lebar82380d82016-10-10 16:26:40 +000043 void registerMatcher(StringRef MatcherName,
44 std::unique_ptr<MatcherDescriptor> Callback);
45
Manuel Klimek24db0f02013-05-14 09:13:00 +000046 ConstructorMap Constructors;
47};
48
Justin Lebar82380d82016-10-10 16:26:40 +000049void RegistryMaps::registerMatcher(
50 StringRef MatcherName, std::unique_ptr<MatcherDescriptor> Callback) {
Samuel Benzaquenc31b3522013-06-04 15:46:22 +000051 assert(Constructors.find(MatcherName) == Constructors.end());
Justin Lebar82380d82016-10-10 16:26:40 +000052 Constructors[MatcherName] = std::move(Callback);
Manuel Klimek24db0f02013-05-14 09:13:00 +000053}
54
55#define REGISTER_MATCHER(name) \
56 registerMatcher(#name, internal::makeMatcherAutoMarshall( \
57 ::clang::ast_matchers::name, #name));
58
Samuel Benzaquene0b2c8e2013-07-22 16:13:57 +000059#define SPECIFIC_MATCHER_OVERLOAD(name, Id) \
Justin Lebar82380d82016-10-10 16:26:40 +000060 static_cast<::clang::ast_matchers::name##_Type##Id>( \
Samuel Benzaquene0b2c8e2013-07-22 16:13:57 +000061 ::clang::ast_matchers::name)
62
63#define REGISTER_OVERLOADED_2(name) \
64 do { \
Justin Lebar82380d82016-10-10 16:26:40 +000065 std::unique_ptr<MatcherDescriptor> Callbacks[] = { \
66 internal::makeMatcherAutoMarshall(SPECIFIC_MATCHER_OVERLOAD(name, 0), \
67 #name), \
68 internal::makeMatcherAutoMarshall(SPECIFIC_MATCHER_OVERLOAD(name, 1), \
69 #name)}; \
70 registerMatcher( \
71 #name, \
72 llvm::make_unique<internal::OverloadedMatcherDescriptor>(Callbacks)); \
Samuel Benzaquen7f8a5b12013-07-24 14:48:01 +000073 } while (0)
74
Manuel Klimek24db0f02013-05-14 09:13:00 +000075/// \brief Generate a registry map with all the known matchers.
76RegistryMaps::RegistryMaps() {
Samuel Benzaquenc31b3522013-06-04 15:46:22 +000077 // TODO: Here is the list of the missing matchers, grouped by reason.
78 //
Samuel Benzaquenc31b3522013-06-04 15:46:22 +000079 // Need Variant/Parser fixes:
80 // ofKind
81 //
Samuel Benzaquenc31b3522013-06-04 15:46:22 +000082 // Polymorphic + argument overload:
Samuel Benzaquenc31b3522013-06-04 15:46:22 +000083 // findAll
84 //
Samuel Benzaquenc31b3522013-06-04 15:46:22 +000085 // Other:
Samuel Benzaquenc31b3522013-06-04 15:46:22 +000086 // equals
87 // equalsNode
Manuel Klimek24db0f02013-05-14 09:13:00 +000088
Samuel Benzaquene0b2c8e2013-07-22 16:13:57 +000089 REGISTER_OVERLOADED_2(callee);
90 REGISTER_OVERLOADED_2(hasPrefix);
91 REGISTER_OVERLOADED_2(hasType);
92 REGISTER_OVERLOADED_2(isDerivedFrom);
93 REGISTER_OVERLOADED_2(isSameOrDerivedFrom);
Samuel Benzaquena0839352014-03-10 15:40:23 +000094 REGISTER_OVERLOADED_2(loc);
Samuel Benzaquene0b2c8e2013-07-22 16:13:57 +000095 REGISTER_OVERLOADED_2(pointsTo);
96 REGISTER_OVERLOADED_2(references);
97 REGISTER_OVERLOADED_2(thisPointerType);
98
Samuel Benzaquenc31b3522013-06-04 15:46:22 +000099 REGISTER_MATCHER(accessSpecDecl);
Aaron Ballmana35b8fc2016-03-09 17:11:51 +0000100 REGISTER_MATCHER(addrLabelExpr);
Samuel Benzaquenc31b3522013-06-04 15:46:22 +0000101 REGISTER_MATCHER(alignOfExpr);
Samuel Benzaquen4adca622013-08-28 18:42:04 +0000102 REGISTER_MATCHER(allOf);
103 REGISTER_MATCHER(anyOf);
Samuel Benzaquenc6f2c9b2013-06-21 15:51:31 +0000104 REGISTER_MATCHER(anything);
105 REGISTER_MATCHER(argumentCountIs);
Samuel Benzaquenc31b3522013-06-04 15:46:22 +0000106 REGISTER_MATCHER(arraySubscriptExpr);
107 REGISTER_MATCHER(arrayType);
Samuel Benzaquenc31b3522013-06-04 15:46:22 +0000108 REGISTER_MATCHER(asmStmt);
Manuel Klimek7735e402014-10-09 13:06:22 +0000109 REGISTER_MATCHER(asString);
Aaron Ballmana35b8fc2016-03-09 17:11:51 +0000110 REGISTER_MATCHER(atomicExpr);
Samuel Benzaquenc31b3522013-06-04 15:46:22 +0000111 REGISTER_MATCHER(atomicType);
112 REGISTER_MATCHER(autoType);
Manuel Klimek24db0f02013-05-14 09:13:00 +0000113 REGISTER_MATCHER(binaryOperator);
Aaron Ballmana35b8fc2016-03-09 17:11:51 +0000114 REGISTER_MATCHER(binaryConditionalOperator);
Samuel Benzaquenc31b3522013-06-04 15:46:22 +0000115 REGISTER_MATCHER(blockPointerType);
Samuel Benzaquenbd3232a2015-12-22 20:06:40 +0000116 REGISTER_MATCHER(booleanType);
Samuel Benzaquenc31b3522013-06-04 15:46:22 +0000117 REGISTER_MATCHER(breakStmt);
118 REGISTER_MATCHER(builtinType);
Manuel Klimek24db0f02013-05-14 09:13:00 +0000119 REGISTER_MATCHER(callExpr);
Samuel Benzaquen0d455abe2013-11-25 15:03:44 +0000120 REGISTER_MATCHER(caseStmt);
Samuel Benzaquenc31b3522013-06-04 15:46:22 +0000121 REGISTER_MATCHER(castExpr);
Manuel Klimek24db0f02013-05-14 09:13:00 +0000122 REGISTER_MATCHER(characterLiteral);
Samuel Benzaquenc31b3522013-06-04 15:46:22 +0000123 REGISTER_MATCHER(classTemplateDecl);
124 REGISTER_MATCHER(classTemplateSpecializationDecl);
125 REGISTER_MATCHER(complexType);
126 REGISTER_MATCHER(compoundLiteralExpr);
Manuel Klimek24db0f02013-05-14 09:13:00 +0000127 REGISTER_MATCHER(compoundStmt);
128 REGISTER_MATCHER(conditionalOperator);
Samuel Benzaquenc31b3522013-06-04 15:46:22 +0000129 REGISTER_MATCHER(constantArrayType);
Samuel Benzaquenc31b3522013-06-04 15:46:22 +0000130 REGISTER_MATCHER(containsDeclaration);
131 REGISTER_MATCHER(continueStmt);
Manuel Klimek7735e402014-10-09 13:06:22 +0000132 REGISTER_MATCHER(cStyleCastExpr);
Aaron Ballman512fb642015-09-17 13:30:52 +0000133 REGISTER_MATCHER(cudaKernelCallExpr);
134 REGISTER_MATCHER(cxxBindTemporaryExpr);
135 REGISTER_MATCHER(cxxBoolLiteral);
136 REGISTER_MATCHER(cxxCatchStmt);
137 REGISTER_MATCHER(cxxConstCastExpr);
138 REGISTER_MATCHER(cxxConstructExpr);
139 REGISTER_MATCHER(cxxConstructorDecl);
140 REGISTER_MATCHER(cxxConversionDecl);
141 REGISTER_MATCHER(cxxCtorInitializer);
142 REGISTER_MATCHER(cxxDefaultArgExpr);
143 REGISTER_MATCHER(cxxDeleteExpr);
144 REGISTER_MATCHER(cxxDestructorDecl);
145 REGISTER_MATCHER(cxxDynamicCastExpr);
146 REGISTER_MATCHER(cxxForRangeStmt);
147 REGISTER_MATCHER(cxxFunctionalCastExpr);
148 REGISTER_MATCHER(cxxMemberCallExpr);
149 REGISTER_MATCHER(cxxMethodDecl);
150 REGISTER_MATCHER(cxxNewExpr);
151 REGISTER_MATCHER(cxxNullPtrLiteralExpr);
152 REGISTER_MATCHER(cxxOperatorCallExpr);
153 REGISTER_MATCHER(cxxRecordDecl);
154 REGISTER_MATCHER(cxxReinterpretCastExpr);
155 REGISTER_MATCHER(cxxStaticCastExpr);
156 REGISTER_MATCHER(cxxTemporaryObjectExpr);
157 REGISTER_MATCHER(cxxThisExpr);
158 REGISTER_MATCHER(cxxThrowExpr);
159 REGISTER_MATCHER(cxxTryStmt);
160 REGISTER_MATCHER(cxxUnresolvedConstructExpr);
Matthias Gehre2cf7e802015-10-12 21:46:07 +0000161 REGISTER_MATCHER(decayedType);
Samuel Benzaquenc31b3522013-06-04 15:46:22 +0000162 REGISTER_MATCHER(decl);
Manuel Klimek7735e402014-10-09 13:06:22 +0000163 REGISTER_MATCHER(declaratorDecl);
Samuel Benzaquenc31b3522013-06-04 15:46:22 +0000164 REGISTER_MATCHER(declCountIs);
Manuel Klimek24db0f02013-05-14 09:13:00 +0000165 REGISTER_MATCHER(declRefExpr);
166 REGISTER_MATCHER(declStmt);
Samuel Benzaquen0d455abe2013-11-25 15:03:44 +0000167 REGISTER_MATCHER(defaultStmt);
Samuel Benzaquenc31b3522013-06-04 15:46:22 +0000168 REGISTER_MATCHER(dependentSizedArrayType);
Aaron Ballmana35b8fc2016-03-09 17:11:51 +0000169 REGISTER_MATCHER(designatedInitExpr);
170 REGISTER_MATCHER(designatorCountIs);
Manuel Klimek24db0f02013-05-14 09:13:00 +0000171 REGISTER_MATCHER(doStmt);
Samuel Benzaquen4adca622013-08-28 18:42:04 +0000172 REGISTER_MATCHER(eachOf);
Samuel Benzaquenc31b3522013-06-04 15:46:22 +0000173 REGISTER_MATCHER(elaboratedType);
174 REGISTER_MATCHER(enumConstantDecl);
175 REGISTER_MATCHER(enumDecl);
Haojian Wue775de82016-06-30 07:50:01 +0000176 REGISTER_MATCHER(enumType);
Samuel Benzaquen0d455abe2013-11-25 15:03:44 +0000177 REGISTER_MATCHER(equalsBoundNode);
Manuel Klimek7735e402014-10-09 13:06:22 +0000178 REGISTER_MATCHER(equalsIntegralValue);
Manuel Klimek24db0f02013-05-14 09:13:00 +0000179 REGISTER_MATCHER(explicitCastExpr);
180 REGISTER_MATCHER(expr);
Samuel Benzaquenf1066292014-04-02 13:12:14 +0000181 REGISTER_MATCHER(exprWithCleanups);
Manuel Klimek24db0f02013-05-14 09:13:00 +0000182 REGISTER_MATCHER(fieldDecl);
Daniel Jasper91f1c8c2013-07-26 18:52:58 +0000183 REGISTER_MATCHER(floatLiteral);
Samuel Benzaquenbd7d8872013-08-16 16:19:42 +0000184 REGISTER_MATCHER(forEach);
Manuel Klimekce28f9e2016-01-18 11:20:09 +0000185 REGISTER_MATCHER(forEachArgumentWithParam);
Samuel Benzaquen0d455abe2013-11-25 15:03:44 +0000186 REGISTER_MATCHER(forEachConstructorInitializer);
Samuel Benzaquenbd7d8872013-08-16 16:19:42 +0000187 REGISTER_MATCHER(forEachDescendant);
Samuel Benzaquen0d455abe2013-11-25 15:03:44 +0000188 REGISTER_MATCHER(forEachSwitchCase);
Samuel Benzaquen06e056c2013-07-17 14:28:00 +0000189 REGISTER_MATCHER(forField);
Gabor Horvath1b3f8db2016-05-04 11:59:39 +0000190 REGISTER_MATCHER(forFunction);
Manuel Klimek24db0f02013-05-14 09:13:00 +0000191 REGISTER_MATCHER(forStmt);
Samuel Benzaquen0d455abe2013-11-25 15:03:44 +0000192 REGISTER_MATCHER(friendDecl);
Manuel Klimek24db0f02013-05-14 09:13:00 +0000193 REGISTER_MATCHER(functionDecl);
Aaron Ballman7e7b7b22016-02-01 14:11:47 +0000194 REGISTER_MATCHER(functionProtoType);
Samuel Benzaquenc31b3522013-06-04 15:46:22 +0000195 REGISTER_MATCHER(functionTemplateDecl);
196 REGISTER_MATCHER(functionType);
Aaron Ballman880a65b2016-02-09 14:04:49 +0000197 REGISTER_MATCHER(gnuNullExpr);
Samuel Benzaquenc31b3522013-06-04 15:46:22 +0000198 REGISTER_MATCHER(gotoStmt);
Samuel Benzaquenbd7d8872013-08-16 16:19:42 +0000199 REGISTER_MATCHER(has);
200 REGISTER_MATCHER(hasAncestor);
Samuel Benzaquenc6f2c9b2013-06-21 15:51:31 +0000201 REGISTER_MATCHER(hasAnyArgument);
Samuel Benzaquen06e056c2013-07-17 14:28:00 +0000202 REGISTER_MATCHER(hasAnyConstructorInitializer);
Martin Bohme8cef2c22016-08-09 15:07:52 +0000203 REGISTER_MATCHER(hasAnyDeclaration);
Samuel Benzaquen922bef42016-02-22 21:13:02 +0000204 REGISTER_MATCHER(hasAnyName);
Manuel Klimek24db0f02013-05-14 09:13:00 +0000205 REGISTER_MATCHER(hasAnyParameter);
206 REGISTER_MATCHER(hasAnySubstatement);
Samuel Benzaquen21b3da02013-07-17 15:11:30 +0000207 REGISTER_MATCHER(hasAnyTemplateArgument);
Samuel Benzaquenc31b3522013-06-04 15:46:22 +0000208 REGISTER_MATCHER(hasAnyUsingShadowDecl);
Samuel Benzaquenc6f2c9b2013-06-21 15:51:31 +0000209 REGISTER_MATCHER(hasArgument);
Samuel Benzaquenc31b3522013-06-04 15:46:22 +0000210 REGISTER_MATCHER(hasArgumentOfType);
Manuel Klimek3fe8a382014-08-25 11:23:50 +0000211 REGISTER_MATCHER(hasAttr);
Aaron Ballman8e7f00b2015-11-18 17:56:55 +0000212 REGISTER_MATCHER(hasAutomaticStorageDuration);
Samuel Benzaquenc31b3522013-06-04 15:46:22 +0000213 REGISTER_MATCHER(hasBase);
Aaron Ballman5c574342016-07-06 18:25:16 +0000214 REGISTER_MATCHER(hasBitWidth);
Samuel Benzaquenc6f2c9b2013-06-21 15:51:31 +0000215 REGISTER_MATCHER(hasBody);
Samuel Benzaquenc31b3522013-06-04 15:46:22 +0000216 REGISTER_MATCHER(hasCanonicalType);
Samuel Benzaquen0d455abe2013-11-25 15:03:44 +0000217 REGISTER_MATCHER(hasCaseConstant);
Etienne Bergeron75e52722016-05-13 19:36:55 +0000218 REGISTER_MATCHER(hasCastKind);
Samuel Benzaquenc6f2c9b2013-06-21 15:51:31 +0000219 REGISTER_MATCHER(hasCondition);
Manuel Klimek24db0f02013-05-14 09:13:00 +0000220 REGISTER_MATCHER(hasConditionVariableStatement);
Matthias Gehre2cf7e802015-10-12 21:46:07 +0000221 REGISTER_MATCHER(hasDecayedType);
Samuel Benzaquen464c1cb2013-11-18 14:53:42 +0000222 REGISTER_MATCHER(hasDeclaration);
Manuel Klimek7735e402014-10-09 13:06:22 +0000223 REGISTER_MATCHER(hasDeclContext);
Samuel Benzaquen79656e12013-07-15 19:25:06 +0000224 REGISTER_MATCHER(hasDeducedType);
Samuel Benzaquenbd7d8872013-08-16 16:19:42 +0000225 REGISTER_MATCHER(hasDescendant);
Manuel Klimek24db0f02013-05-14 09:13:00 +0000226 REGISTER_MATCHER(hasDestinationType);
Aaron Ballmanabdbbbc2016-05-16 16:49:01 +0000227 REGISTER_MATCHER(hasDynamicExceptionSpec);
Manuel Klimek24db0f02013-05-14 09:13:00 +0000228 REGISTER_MATCHER(hasEitherOperand);
Samuel Benzaquen79656e12013-07-15 19:25:06 +0000229 REGISTER_MATCHER(hasElementType);
Samuel Benzaquene1e74932014-08-15 19:13:27 +0000230 REGISTER_MATCHER(hasElse);
Aaron Ballmana086b9f2016-08-17 13:10:42 +0000231 REGISTER_MATCHER(hasExternalFormalLinkage);
Manuel Klimek24db0f02013-05-14 09:13:00 +0000232 REGISTER_MATCHER(hasFalseExpression);
Samuel Benzaquenf56a2992014-06-05 18:22:14 +0000233 REGISTER_MATCHER(hasGlobalStorage);
Manuel Klimek24db0f02013-05-14 09:13:00 +0000234 REGISTER_MATCHER(hasImplicitDestinationType);
Malcolm Parsons4ca3d182016-12-24 13:35:14 +0000235 REGISTER_MATCHER(hasInClassInitializer);
Samuel Benzaquenc31b3522013-06-04 15:46:22 +0000236 REGISTER_MATCHER(hasIncrement);
237 REGISTER_MATCHER(hasIndex);
Manuel Klimek24db0f02013-05-14 09:13:00 +0000238 REGISTER_MATCHER(hasInitializer);
Manuel Klimekbfa43572015-03-12 15:48:15 +0000239 REGISTER_MATCHER(hasKeywordSelector);
Manuel Klimek24db0f02013-05-14 09:13:00 +0000240 REGISTER_MATCHER(hasLHS);
Samuel Benzaquenc31b3522013-06-04 15:46:22 +0000241 REGISTER_MATCHER(hasLocalQualifiers);
Samuel Benzaquenf56a2992014-06-05 18:22:14 +0000242 REGISTER_MATCHER(hasLocalStorage);
Samuel Benzaquenc31b3522013-06-04 15:46:22 +0000243 REGISTER_MATCHER(hasLoopInit);
Samuel Benzaquene1e74932014-08-15 19:13:27 +0000244 REGISTER_MATCHER(hasLoopVariable);
Samuel Benzaquenc31b3522013-06-04 15:46:22 +0000245 REGISTER_MATCHER(hasMethod);
Manuel Klimek24db0f02013-05-14 09:13:00 +0000246 REGISTER_MATCHER(hasName);
Manuel Klimekbfa43572015-03-12 15:48:15 +0000247 REGISTER_MATCHER(hasNullSelector);
Manuel Klimek24db0f02013-05-14 09:13:00 +0000248 REGISTER_MATCHER(hasObjectExpression);
Samuel Benzaquenc6f2c9b2013-06-21 15:51:31 +0000249 REGISTER_MATCHER(hasOperatorName);
250 REGISTER_MATCHER(hasOverloadedOperatorName);
Samuel Benzaquenc31b3522013-06-04 15:46:22 +0000251 REGISTER_MATCHER(hasParameter);
Samuel Benzaquenbd7d8872013-08-16 16:19:42 +0000252 REGISTER_MATCHER(hasParent);
Samuel Benzaquenc31b3522013-06-04 15:46:22 +0000253 REGISTER_MATCHER(hasQualifier);
Samuel Benzaquene1e74932014-08-15 19:13:27 +0000254 REGISTER_MATCHER(hasRangeInit);
Manuel Klimekbfa43572015-03-12 15:48:15 +0000255 REGISTER_MATCHER(hasReceiverType);
Malcolm Parsons77f039b2016-12-08 11:46:22 +0000256 REGISTER_MATCHER(hasReplacementType);
Alexander Kornienko976921d2016-03-22 11:03:03 +0000257 REGISTER_MATCHER(hasReturnValue);
Manuel Klimek7735e402014-10-09 13:06:22 +0000258 REGISTER_MATCHER(hasRHS);
Manuel Klimekbfa43572015-03-12 15:48:15 +0000259 REGISTER_MATCHER(hasSelector);
Samuel Benzaquenc31b3522013-06-04 15:46:22 +0000260 REGISTER_MATCHER(hasSingleDecl);
261 REGISTER_MATCHER(hasSize);
262 REGISTER_MATCHER(hasSizeExpr);
Manuel Klimek24db0f02013-05-14 09:13:00 +0000263 REGISTER_MATCHER(hasSourceExpression);
Aaron Ballman8e7f00b2015-11-18 17:56:55 +0000264 REGISTER_MATCHER(hasStaticStorageDuration);
Aaron Ballmana35b8fc2016-03-09 17:11:51 +0000265 REGISTER_MATCHER(hasSyntacticForm);
Samuel Benzaquenc31b3522013-06-04 15:46:22 +0000266 REGISTER_MATCHER(hasTargetDecl);
Samuel Benzaquen21b3da02013-07-17 15:11:30 +0000267 REGISTER_MATCHER(hasTemplateArgument);
Samuel Benzaquene1e74932014-08-15 19:13:27 +0000268 REGISTER_MATCHER(hasThen);
Aaron Ballman8e7f00b2015-11-18 17:56:55 +0000269 REGISTER_MATCHER(hasThreadStorageDuration);
Manuel Klimek24db0f02013-05-14 09:13:00 +0000270 REGISTER_MATCHER(hasTrueExpression);
Samuel Benzaquen0d455abe2013-11-25 15:03:44 +0000271 REGISTER_MATCHER(hasTypeLoc);
Manuel Klimek24db0f02013-05-14 09:13:00 +0000272 REGISTER_MATCHER(hasUnaryOperand);
Manuel Klimekbfa43572015-03-12 15:48:15 +0000273 REGISTER_MATCHER(hasUnarySelector);
Martin Bohme8cef2c22016-08-09 15:07:52 +0000274 REGISTER_MATCHER(hasUnderlyingDecl);
Manuel Klimeka37e1102016-12-01 15:45:06 +0000275 REGISTER_MATCHER(hasUnqualifiedDesugaredType);
Samuel Benzaquen79656e12013-07-15 19:25:06 +0000276 REGISTER_MATCHER(hasValueType);
Manuel Klimek24db0f02013-05-14 09:13:00 +0000277 REGISTER_MATCHER(ifStmt);
Cong Liu8a02efb2016-06-24 09:38:03 +0000278 REGISTER_MATCHER(ignoringImplicit);
Samuel Benzaquenc31b3522013-06-04 15:46:22 +0000279 REGISTER_MATCHER(ignoringImpCasts);
280 REGISTER_MATCHER(ignoringParenCasts);
281 REGISTER_MATCHER(ignoringParenImpCasts);
Aaron Ballmanba8dbbe2016-06-06 18:52:17 +0000282 REGISTER_MATCHER(ignoringParens);
Manuel Klimek24db0f02013-05-14 09:13:00 +0000283 REGISTER_MATCHER(implicitCastExpr);
Aaron Ballmana35b8fc2016-03-09 17:11:51 +0000284 REGISTER_MATCHER(implicitValueInitExpr);
Samuel Benzaquenc31b3522013-06-04 15:46:22 +0000285 REGISTER_MATCHER(incompleteArrayType);
286 REGISTER_MATCHER(initListExpr);
Aaron Ballmanb85be662015-09-11 11:51:24 +0000287 REGISTER_MATCHER(injectedClassNameType);
Samuel Benzaquen79656e12013-07-15 19:25:06 +0000288 REGISTER_MATCHER(innerType);
Manuel Klimek24db0f02013-05-14 09:13:00 +0000289 REGISTER_MATCHER(integerLiteral);
Aaron Ballman11825f22015-08-18 19:55:20 +0000290 REGISTER_MATCHER(isAnonymous);
Felix Bergercc9df3b2016-02-15 04:00:39 +0000291 REGISTER_MATCHER(isAnyCharacter);
292 REGISTER_MATCHER(isAnyPointer);
Manuel Klimek24db0f02013-05-14 09:13:00 +0000293 REGISTER_MATCHER(isArrow);
Aaron Ballman1ca258e2015-08-05 12:11:30 +0000294 REGISTER_MATCHER(isBaseInitializer);
Aaron Ballman5c574342016-07-06 18:25:16 +0000295 REGISTER_MATCHER(isBitField);
Aaron Ballman9b869aa2015-07-02 12:53:22 +0000296 REGISTER_MATCHER(isCatchAll);
Aaron Ballman512fb642015-09-17 13:30:52 +0000297 REGISTER_MATCHER(isClass);
Samuel Benzaquen0d455abe2013-11-25 15:03:44 +0000298 REGISTER_MATCHER(isConst);
Alexander Shaposhnikovae33dfa2017-04-16 19:05:17 +0000299 REGISTER_MATCHER(isConstexpr);
Manuel Klimek24db0f02013-05-14 09:13:00 +0000300 REGISTER_MATCHER(isConstQualified);
Aaron Ballman433485f2016-01-20 20:47:58 +0000301 REGISTER_MATCHER(isCopyAssignmentOperator);
Aaron Ballman1ca258e2015-08-05 12:11:30 +0000302 REGISTER_MATCHER(isCopyConstructor);
303 REGISTER_MATCHER(isDefaultConstructor);
Aaron Ballmaneb85b042016-01-18 20:37:44 +0000304 REGISTER_MATCHER(isDefaulted);
Samuel Benzaquenc6f2c9b2013-06-21 15:51:31 +0000305 REGISTER_MATCHER(isDefinition);
Samuel Benzaquen8e7f9962014-08-15 14:20:59 +0000306 REGISTER_MATCHER(isDeleted);
Aaron Ballman39918462015-07-15 17:11:21 +0000307 REGISTER_MATCHER(isExceptionVariable);
Aaron Ballman6f6d0b62015-08-11 21:09:52 +0000308 REGISTER_MATCHER(isExplicit);
Samuel Benzaquenc6f2c9b2013-06-21 15:51:31 +0000309 REGISTER_MATCHER(isExplicitTemplateSpecialization);
Peter Collingbourne564597f2014-02-20 19:18:03 +0000310 REGISTER_MATCHER(isExpr);
Samuel Benzaquenc31b3522013-06-04 15:46:22 +0000311 REGISTER_MATCHER(isExternC);
Aaron Ballman41143bb2015-07-24 12:35:41 +0000312 REGISTER_MATCHER(isFinal);
Aaron Ballman11825f22015-08-18 19:55:20 +0000313 REGISTER_MATCHER(isInline);
Manuel Klimek24db0f02013-05-14 09:13:00 +0000314 REGISTER_MATCHER(isImplicit);
Manuel Klimekd3aa1f42014-11-25 17:01:06 +0000315 REGISTER_MATCHER(isExpansionInFileMatching);
316 REGISTER_MATCHER(isExpansionInMainFile);
Benjamin Kramer7ab84762014-09-03 12:08:14 +0000317 REGISTER_MATCHER(isInstantiated);
Manuel Klimekd3aa1f42014-11-25 17:01:06 +0000318 REGISTER_MATCHER(isExpansionInSystemHeader);
Samuel Benzaquenc31b3522013-06-04 15:46:22 +0000319 REGISTER_MATCHER(isInteger);
Manuel Klimek7735e402014-10-09 13:06:22 +0000320 REGISTER_MATCHER(isIntegral);
Benjamin Kramer7ab84762014-09-03 12:08:14 +0000321 REGISTER_MATCHER(isInTemplateInstantiation);
Samuel Benzaquen49385c72016-06-28 14:08:56 +0000322 REGISTER_MATCHER(isLambda);
Peter Collingbourne1fec3df2014-02-06 21:52:24 +0000323 REGISTER_MATCHER(isListInitialization);
Aaron Ballmaned455d42015-08-11 20:42:00 +0000324 REGISTER_MATCHER(isMemberInitializer);
Aaron Ballman31bde872016-01-22 22:37:09 +0000325 REGISTER_MATCHER(isMoveAssignmentOperator);
Aaron Ballman1ca258e2015-08-05 12:11:30 +0000326 REGISTER_MATCHER(isMoveConstructor);
Aaron Ballmana60bcda2015-12-02 15:23:59 +0000327 REGISTER_MATCHER(isNoThrow);
Samuel Benzaquenc31b3522013-06-04 15:46:22 +0000328 REGISTER_MATCHER(isOverride);
329 REGISTER_MATCHER(isPrivate);
330 REGISTER_MATCHER(isProtected);
331 REGISTER_MATCHER(isPublic);
Samuel Benzaquene1e74932014-08-15 19:13:27 +0000332 REGISTER_MATCHER(isPure);
Clement Courbet42517592016-07-12 06:36:00 +0000333 REGISTER_MATCHER(isSignedInteger);
Aaron Ballman565019f2017-04-15 12:53:20 +0000334 REGISTER_MATCHER(isStaticStorageClass);
Aaron Ballman512fb642015-09-17 13:30:52 +0000335 REGISTER_MATCHER(isStruct);
Samuel Benzaquenc6f2c9b2013-06-21 15:51:31 +0000336 REGISTER_MATCHER(isTemplateInstantiation);
Aaron Ballman512fb642015-09-17 13:30:52 +0000337 REGISTER_MATCHER(isUnion);
Clement Courbet42517592016-07-12 06:36:00 +0000338 REGISTER_MATCHER(isUnsignedInteger);
Aaron Ballman3fd6c112015-10-05 14:41:27 +0000339 REGISTER_MATCHER(isVariadic);
Samuel Benzaquenc31b3522013-06-04 15:46:22 +0000340 REGISTER_MATCHER(isVirtual);
Nico Webera415a1d2016-01-21 17:56:24 +0000341 REGISTER_MATCHER(isVirtualAsWritten);
Aaron Ballman6290fc92015-11-23 17:09:24 +0000342 REGISTER_MATCHER(isVolatileQualified);
Samuel Benzaquen06e056c2013-07-17 14:28:00 +0000343 REGISTER_MATCHER(isWritten);
Aaron Ballmana35b8fc2016-03-09 17:11:51 +0000344 REGISTER_MATCHER(labelDecl);
Samuel Benzaquenc31b3522013-06-04 15:46:22 +0000345 REGISTER_MATCHER(labelStmt);
346 REGISTER_MATCHER(lambdaExpr);
Manuel Klimek7735e402014-10-09 13:06:22 +0000347 REGISTER_MATCHER(lValueReferenceType);
Samuel Benzaquenc31b3522013-06-04 15:46:22 +0000348 REGISTER_MATCHER(matchesName);
Manuel Klimekbfa43572015-03-12 15:48:15 +0000349 REGISTER_MATCHER(matchesSelector);
Samuel Benzaquenc31b3522013-06-04 15:46:22 +0000350 REGISTER_MATCHER(materializeTemporaryExpr);
Manuel Klimek24db0f02013-05-14 09:13:00 +0000351 REGISTER_MATCHER(member);
352 REGISTER_MATCHER(memberExpr);
Samuel Benzaquenc31b3522013-06-04 15:46:22 +0000353 REGISTER_MATCHER(memberPointerType);
Manuel Klimek24db0f02013-05-14 09:13:00 +0000354 REGISTER_MATCHER(namedDecl);
Aaron Ballman6c79f352015-08-28 19:39:21 +0000355 REGISTER_MATCHER(namespaceAliasDecl);
Samuel Benzaquenc31b3522013-06-04 15:46:22 +0000356 REGISTER_MATCHER(namespaceDecl);
Manuel Klimek7735e402014-10-09 13:06:22 +0000357 REGISTER_MATCHER(namesType);
Samuel Benzaquenc31b3522013-06-04 15:46:22 +0000358 REGISTER_MATCHER(nestedNameSpecifier);
359 REGISTER_MATCHER(nestedNameSpecifierLoc);
Aaron Ballman232e63d2016-02-16 21:02:23 +0000360 REGISTER_MATCHER(nullPointerConstant);
Samuel Benzaquenc31b3522013-06-04 15:46:22 +0000361 REGISTER_MATCHER(nullStmt);
Manuel Klimekbfa43572015-03-12 15:48:15 +0000362 REGISTER_MATCHER(numSelectorArgs);
Manuel Klimek24db0f02013-05-14 09:13:00 +0000363 REGISTER_MATCHER(ofClass);
Aaron Ballman9fd6ee62017-03-15 20:14:25 +0000364 REGISTER_MATCHER(objcCategoryDecl);
Aaron Ballmanb85be662015-09-11 11:51:24 +0000365 REGISTER_MATCHER(objcInterfaceDecl);
Aaron Ballman9fd6ee62017-03-15 20:14:25 +0000366 REGISTER_MATCHER(objcIvarDecl);
Manuel Klimekbfa43572015-03-12 15:48:15 +0000367 REGISTER_MATCHER(objcMessageExpr);
Aaron Ballman9fd6ee62017-03-15 20:14:25 +0000368 REGISTER_MATCHER(objcMethodDecl);
Aaron Ballmanb85be662015-09-11 11:51:24 +0000369 REGISTER_MATCHER(objcObjectPointerType);
Aaron Ballman9fd6ee62017-03-15 20:14:25 +0000370 REGISTER_MATCHER(objcPropertyDecl);
371 REGISTER_MATCHER(objcProtocolDecl);
Manuel Klimek24db0f02013-05-14 09:13:00 +0000372 REGISTER_MATCHER(on);
373 REGISTER_MATCHER(onImplicitObjectArgument);
Aaron Ballmana35b8fc2016-03-09 17:11:51 +0000374 REGISTER_MATCHER(opaqueValueExpr);
Samuel Benzaquenc31b3522013-06-04 15:46:22 +0000375 REGISTER_MATCHER(parameterCountIs);
Aaron Ballmane8295d72016-01-20 16:17:39 +0000376 REGISTER_MATCHER(parenExpr);
Aaron Ballmana35b8fc2016-03-09 17:11:51 +0000377 REGISTER_MATCHER(parenListExpr);
Samuel Benzaquenc31b3522013-06-04 15:46:22 +0000378 REGISTER_MATCHER(parenType);
Samuel Benzaquen0d455abe2013-11-25 15:03:44 +0000379 REGISTER_MATCHER(parmVarDecl);
Samuel Benzaquen79656e12013-07-15 19:25:06 +0000380 REGISTER_MATCHER(pointee);
Samuel Benzaquenc31b3522013-06-04 15:46:22 +0000381 REGISTER_MATCHER(pointerType);
Aaron Ballmana35b8fc2016-03-09 17:11:51 +0000382 REGISTER_MATCHER(predefinedExpr);
Samuel Benzaquenc31b3522013-06-04 15:46:22 +0000383 REGISTER_MATCHER(qualType);
Aaron Ballmaneb7e5d92016-02-18 16:36:01 +0000384 REGISTER_MATCHER(realFloatingPointType);
Manuel Klimek24db0f02013-05-14 09:13:00 +0000385 REGISTER_MATCHER(recordDecl);
Samuel Benzaquenc31b3522013-06-04 15:46:22 +0000386 REGISTER_MATCHER(recordType);
387 REGISTER_MATCHER(referenceType);
388 REGISTER_MATCHER(refersToDeclaration);
Manuel Klimek7735e402014-10-09 13:06:22 +0000389 REGISTER_MATCHER(refersToIntegralType);
Samuel Benzaquenc31b3522013-06-04 15:46:22 +0000390 REGISTER_MATCHER(refersToType);
Aaron Ballmana35b8fc2016-03-09 17:11:51 +0000391 REGISTER_MATCHER(requiresZeroInitialization);
Samuel Benzaquenc31b3522013-06-04 15:46:22 +0000392 REGISTER_MATCHER(returns);
Manuel Klimek7735e402014-10-09 13:06:22 +0000393 REGISTER_MATCHER(returnStmt);
394 REGISTER_MATCHER(rValueReferenceType);
Samuel Benzaquenc31b3522013-06-04 15:46:22 +0000395 REGISTER_MATCHER(sizeOfExpr);
396 REGISTER_MATCHER(specifiesNamespace);
397 REGISTER_MATCHER(specifiesType);
398 REGISTER_MATCHER(specifiesTypeLoc);
399 REGISTER_MATCHER(statementCountIs);
David Majnemer8423df92015-06-05 22:40:53 +0000400 REGISTER_MATCHER(staticAssertDecl);
Manuel Klimek24db0f02013-05-14 09:13:00 +0000401 REGISTER_MATCHER(stmt);
Aaron Ballmana35b8fc2016-03-09 17:11:51 +0000402 REGISTER_MATCHER(stmtExpr);
Manuel Klimek24db0f02013-05-14 09:13:00 +0000403 REGISTER_MATCHER(stringLiteral);
Samuel Benzaquene1e74932014-08-15 19:13:27 +0000404 REGISTER_MATCHER(substNonTypeTemplateParmExpr);
Samuel Benzaquenf8ec4542015-08-26 16:15:59 +0000405 REGISTER_MATCHER(substTemplateTypeParmType);
Manuel Klimek24db0f02013-05-14 09:13:00 +0000406 REGISTER_MATCHER(switchCase);
Samuel Benzaquenc31b3522013-06-04 15:46:22 +0000407 REGISTER_MATCHER(switchStmt);
Manuel Klimek7735e402014-10-09 13:06:22 +0000408 REGISTER_MATCHER(templateArgument);
Haojian Wub33b02e2016-07-29 15:45:11 +0000409 REGISTER_MATCHER(templateName);
Manuel Klimek7735e402014-10-09 13:06:22 +0000410 REGISTER_MATCHER(templateArgumentCountIs);
Samuel Benzaquenc31b3522013-06-04 15:46:22 +0000411 REGISTER_MATCHER(templateSpecializationType);
Martin Bohme565bdd92016-08-12 13:51:00 +0000412 REGISTER_MATCHER(templateTypeParmDecl);
Aaron Ballmanb85be662015-09-11 11:51:24 +0000413 REGISTER_MATCHER(templateTypeParmType);
Samuel Benzaquenc31b3522013-06-04 15:46:22 +0000414 REGISTER_MATCHER(throughUsingDecl);
Manuel Klimek24db0f02013-05-14 09:13:00 +0000415 REGISTER_MATCHER(to);
Samuel Benzaquenef621f42015-02-10 14:46:45 +0000416 REGISTER_MATCHER(translationUnitDecl);
Samuel Benzaquenc31b3522013-06-04 15:46:22 +0000417 REGISTER_MATCHER(type);
Manuel Klimekd3aa1f42014-11-25 17:01:06 +0000418 REGISTER_MATCHER(typedefDecl);
Aaron Ballman66eb58a2016-04-14 16:05:45 +0000419 REGISTER_MATCHER(typedefNameDecl);
Samuel Benzaquenc31b3522013-06-04 15:46:22 +0000420 REGISTER_MATCHER(typedefType);
Aaron Ballman66eb58a2016-04-14 16:05:45 +0000421 REGISTER_MATCHER(typeAliasDecl);
Eric Liu285f8042017-03-28 12:56:47 +0000422 REGISTER_MATCHER(typeAliasTemplateDecl);
Manuel Klimek7735e402014-10-09 13:06:22 +0000423 REGISTER_MATCHER(typeLoc);
Samuel Benzaquenc31b3522013-06-04 15:46:22 +0000424 REGISTER_MATCHER(unaryExprOrTypeTraitExpr);
Manuel Klimek24db0f02013-05-14 09:13:00 +0000425 REGISTER_MATCHER(unaryOperator);
Samuel Benzaquen0d455abe2013-11-25 15:03:44 +0000426 REGISTER_MATCHER(unaryTransformType);
Samuel Benzaquen4d058742013-11-22 14:41:48 +0000427 REGISTER_MATCHER(unless);
Haojian Wu7751c922016-05-18 12:53:59 +0000428 REGISTER_MATCHER(unresolvedLookupExpr);
Aaron Ballmanb85be662015-09-11 11:51:24 +0000429 REGISTER_MATCHER(unresolvedUsingTypenameDecl);
Samuel Benzaquen0d455abe2013-11-25 15:03:44 +0000430 REGISTER_MATCHER(unresolvedUsingValueDecl);
Samuel Benzaquenc31b3522013-06-04 15:46:22 +0000431 REGISTER_MATCHER(userDefinedLiteral);
432 REGISTER_MATCHER(usingDecl);
Samuel Benzaquene1e74932014-08-15 19:13:27 +0000433 REGISTER_MATCHER(usingDirectiveDecl);
Samuel Benzaquenc640ef52014-10-28 13:33:58 +0000434 REGISTER_MATCHER(valueDecl);
Manuel Klimek24db0f02013-05-14 09:13:00 +0000435 REGISTER_MATCHER(varDecl);
Samuel Benzaquenc31b3522013-06-04 15:46:22 +0000436 REGISTER_MATCHER(variableArrayType);
Samuel Benzaquenb405c082014-12-15 15:09:22 +0000437 REGISTER_MATCHER(voidType);
Manuel Klimek24db0f02013-05-14 09:13:00 +0000438 REGISTER_MATCHER(whileStmt);
Samuel Benzaquen06e056c2013-07-17 14:28:00 +0000439 REGISTER_MATCHER(withInitializer);
Manuel Klimek24db0f02013-05-14 09:13:00 +0000440}
441
Justin Lebar82380d82016-10-10 16:26:40 +0000442RegistryMaps::~RegistryMaps() {}
Manuel Klimek24db0f02013-05-14 09:13:00 +0000443
444static llvm::ManagedStatic<RegistryMaps> RegistryData;
445
446} // anonymous namespace
447
448// static
Samuel Benzaquenf434c4f2014-04-14 13:51:21 +0000449llvm::Optional<MatcherCtor> Registry::lookupMatcherCtor(StringRef MatcherName) {
Justin Lebar82380d82016-10-10 16:26:40 +0000450 auto it = RegistryData->constructors().find(MatcherName);
Samuel Benzaquenf434c4f2014-04-14 13:51:21 +0000451 return it == RegistryData->constructors().end()
452 ? llvm::Optional<MatcherCtor>()
Justin Lebar82380d82016-10-10 16:26:40 +0000453 : it->second.get();
Manuel Klimek24db0f02013-05-14 09:13:00 +0000454}
455
Peter Collingbourned32e28c2014-01-23 22:48:38 +0000456namespace {
457
458llvm::raw_ostream &operator<<(llvm::raw_ostream &OS,
459 const std::set<ASTNodeKind> &KS) {
460 unsigned Count = 0;
461 for (std::set<ASTNodeKind>::const_iterator I = KS.begin(), E = KS.end();
462 I != E; ++I) {
463 if (I != KS.begin())
464 OS << "|";
465 if (Count++ == 3) {
466 OS << "...";
467 break;
468 }
469 OS << *I;
470 }
471 return OS;
472}
473
Samuel Benzaquen646f23b2014-08-12 21:11:37 +0000474} // namespace
Peter Collingbourned32e28c2014-01-23 22:48:38 +0000475
Samuel Benzaquen646f23b2014-08-12 21:11:37 +0000476std::vector<ArgKind> Registry::getAcceptedCompletionTypes(
477 ArrayRef<std::pair<MatcherCtor, unsigned>> Context) {
Peter Collingbourned32e28c2014-01-23 22:48:38 +0000478 ASTNodeKind InitialTypes[] = {
Samuel Benzaquen646f23b2014-08-12 21:11:37 +0000479 ASTNodeKind::getFromNodeKind<Decl>(),
480 ASTNodeKind::getFromNodeKind<QualType>(),
481 ASTNodeKind::getFromNodeKind<Type>(),
482 ASTNodeKind::getFromNodeKind<Stmt>(),
483 ASTNodeKind::getFromNodeKind<NestedNameSpecifier>(),
484 ASTNodeKind::getFromNodeKind<NestedNameSpecifierLoc>(),
485 ASTNodeKind::getFromNodeKind<TypeLoc>()};
Peter Collingbourned32e28c2014-01-23 22:48:38 +0000486
487 // Starting with the above seed of acceptable top-level matcher types, compute
488 // the acceptable type set for the argument indicated by each context element.
Samuel Benzaquen646f23b2014-08-12 21:11:37 +0000489 std::set<ArgKind> TypeSet(std::begin(InitialTypes), std::end(InitialTypes));
490 for (const auto &CtxEntry : Context) {
491 MatcherCtor Ctor = CtxEntry.first;
492 unsigned ArgNumber = CtxEntry.second;
493 std::vector<ArgKind> NextTypeSet;
494 for (const ArgKind &Kind : TypeSet) {
495 if (Kind.getArgKind() == Kind.AK_Matcher &&
496 Ctor->isConvertibleTo(Kind.getMatcherKind()) &&
497 (Ctor->isVariadic() || ArgNumber < Ctor->getNumArgs()))
498 Ctor->getArgKinds(Kind.getMatcherKind(), ArgNumber, NextTypeSet);
Peter Collingbourned32e28c2014-01-23 22:48:38 +0000499 }
500 TypeSet.clear();
Samuel Benzaquen646f23b2014-08-12 21:11:37 +0000501 TypeSet.insert(NextTypeSet.begin(), NextTypeSet.end());
Peter Collingbourned32e28c2014-01-23 22:48:38 +0000502 }
Samuel Benzaquen646f23b2014-08-12 21:11:37 +0000503 return std::vector<ArgKind>(TypeSet.begin(), TypeSet.end());
504}
Peter Collingbourned32e28c2014-01-23 22:48:38 +0000505
Samuel Benzaquen646f23b2014-08-12 21:11:37 +0000506std::vector<MatcherCompletion>
507Registry::getMatcherCompletions(ArrayRef<ArgKind> AcceptedTypes) {
508 std::vector<MatcherCompletion> Completions;
Peter Collingbourned32e28c2014-01-23 22:48:38 +0000509
Samuel Benzaquen646f23b2014-08-12 21:11:37 +0000510 // Search the registry for acceptable matchers.
Aaron Ballman001f1682015-08-12 19:00:39 +0000511 for (const auto &M : RegistryData->constructors()) {
Justin Lebar82380d82016-10-10 16:26:40 +0000512 const MatcherDescriptor& Matcher = *M.getValue();
Aaron Ballman001f1682015-08-12 19:00:39 +0000513 StringRef Name = M.getKey();
514
Peter Collingbourned32e28c2014-01-23 22:48:38 +0000515 std::set<ASTNodeKind> RetKinds;
Justin Lebar82380d82016-10-10 16:26:40 +0000516 unsigned NumArgs = Matcher.isVariadic() ? 1 : Matcher.getNumArgs();
517 bool IsPolymorphic = Matcher.isPolymorphic();
Samuel Benzaquen646f23b2014-08-12 21:11:37 +0000518 std::vector<std::vector<ArgKind>> ArgsKinds(NumArgs);
Peter Collingbourned32e28c2014-01-23 22:48:38 +0000519 unsigned MaxSpecificity = 0;
Samuel Benzaquen646f23b2014-08-12 21:11:37 +0000520 for (const ArgKind& Kind : AcceptedTypes) {
521 if (Kind.getArgKind() != Kind.AK_Matcher)
522 continue;
Peter Collingbourned32e28c2014-01-23 22:48:38 +0000523 unsigned Specificity;
524 ASTNodeKind LeastDerivedKind;
Justin Lebar82380d82016-10-10 16:26:40 +0000525 if (Matcher.isConvertibleTo(Kind.getMatcherKind(), &Specificity,
526 &LeastDerivedKind)) {
Peter Collingbourned32e28c2014-01-23 22:48:38 +0000527 if (MaxSpecificity < Specificity)
528 MaxSpecificity = Specificity;
529 RetKinds.insert(LeastDerivedKind);
530 for (unsigned Arg = 0; Arg != NumArgs; ++Arg)
Justin Lebar82380d82016-10-10 16:26:40 +0000531 Matcher.getArgKinds(Kind.getMatcherKind(), Arg, ArgsKinds[Arg]);
Peter Collingbourned32e28c2014-01-23 22:48:38 +0000532 if (IsPolymorphic)
533 break;
534 }
535 }
536
537 if (!RetKinds.empty() && MaxSpecificity > 0) {
538 std::string Decl;
539 llvm::raw_string_ostream OS(Decl);
540
541 if (IsPolymorphic) {
Aaron Ballman001f1682015-08-12 19:00:39 +0000542 OS << "Matcher<T> " << Name << "(Matcher<T>";
Peter Collingbourned32e28c2014-01-23 22:48:38 +0000543 } else {
Aaron Ballman001f1682015-08-12 19:00:39 +0000544 OS << "Matcher<" << RetKinds << "> " << Name << "(";
Samuel Benzaquen646f23b2014-08-12 21:11:37 +0000545 for (const std::vector<ArgKind> &Arg : ArgsKinds) {
546 if (&Arg != &ArgsKinds[0])
Peter Collingbourned32e28c2014-01-23 22:48:38 +0000547 OS << ", ";
Samuel Benzaquenb63c2512014-10-09 22:08:52 +0000548
549 bool FirstArgKind = true;
550 std::set<ASTNodeKind> MatcherKinds;
551 // Two steps. First all non-matchers, then matchers only.
552 for (const ArgKind &AK : Arg) {
553 if (AK.getArgKind() == ArgKind::AK_Matcher) {
554 MatcherKinds.insert(AK.getMatcherKind());
555 } else {
556 if (!FirstArgKind) OS << "|";
557 FirstArgKind = false;
558 OS << AK.asString();
559 }
560 }
561 if (!MatcherKinds.empty()) {
562 if (!FirstArgKind) OS << "|";
Peter Collingbourned32e28c2014-01-23 22:48:38 +0000563 OS << "Matcher<" << MatcherKinds << ">";
Peter Collingbourned32e28c2014-01-23 22:48:38 +0000564 }
565 }
566 }
Justin Lebar82380d82016-10-10 16:26:40 +0000567 if (Matcher.isVariadic())
Peter Collingbourned32e28c2014-01-23 22:48:38 +0000568 OS << "...";
569 OS << ")";
570
Aaron Ballman001f1682015-08-12 19:00:39 +0000571 std::string TypedText = Name;
Peter Collingbourned32e28c2014-01-23 22:48:38 +0000572 TypedText += "(";
573 if (ArgsKinds.empty())
574 TypedText += ")";
Samuel Benzaquen646f23b2014-08-12 21:11:37 +0000575 else if (ArgsKinds[0][0].getArgKind() == ArgKind::AK_String)
Peter Collingbourned32e28c2014-01-23 22:48:38 +0000576 TypedText += "\"";
577
Samuel Benzaquen646f23b2014-08-12 21:11:37 +0000578 Completions.emplace_back(TypedText, OS.str(), MaxSpecificity);
Peter Collingbourned32e28c2014-01-23 22:48:38 +0000579 }
580 }
581
Samuel Benzaquen646f23b2014-08-12 21:11:37 +0000582 return Completions;
Peter Collingbourned32e28c2014-01-23 22:48:38 +0000583}
584
Samuel Benzaquen31edb512013-06-03 19:31:08 +0000585// static
Peter Collingbourne00cba4f2013-11-23 01:13:16 +0000586VariantMatcher Registry::constructMatcher(MatcherCtor Ctor,
Craig Toppere335f252015-10-04 04:53:55 +0000587 SourceRange NameRange,
Peter Collingbourne00cba4f2013-11-23 01:13:16 +0000588 ArrayRef<ParserValue> Args,
589 Diagnostics *Error) {
Peter Collingbournef43e6942013-11-23 01:34:36 +0000590 return Ctor->create(NameRange, Args, Error);
Peter Collingbourne00cba4f2013-11-23 01:13:16 +0000591}
592
593// static
594VariantMatcher Registry::constructBoundMatcher(MatcherCtor Ctor,
Craig Toppere335f252015-10-04 04:53:55 +0000595 SourceRange NameRange,
Samuel Benzaquen0239b692013-08-13 14:54:51 +0000596 StringRef BindID,
597 ArrayRef<ParserValue> Args,
598 Diagnostics *Error) {
Peter Collingbourne00cba4f2013-11-23 01:13:16 +0000599 VariantMatcher Out = constructMatcher(Ctor, NameRange, Args, Error);
Samuel Benzaquen0239b692013-08-13 14:54:51 +0000600 if (Out.isNull()) return Out;
Samuel Benzaquenc6f2c9b2013-06-21 15:51:31 +0000601
Samuel Benzaquenf34ac3e2013-10-29 14:37:15 +0000602 llvm::Optional<DynTypedMatcher> Result = Out.getSingleMatcher();
603 if (Result.hasValue()) {
604 llvm::Optional<DynTypedMatcher> Bound = Result->tryBind(BindID);
605 if (Bound.hasValue()) {
Samuel Benzaquen0239b692013-08-13 14:54:51 +0000606 return VariantMatcher::SingleMatcher(*Bound);
Samuel Benzaquenc6f2c9b2013-06-21 15:51:31 +0000607 }
Samuel Benzaquen31edb512013-06-03 19:31:08 +0000608 }
Samuel Benzaquena37bb8c2013-07-18 19:47:59 +0000609 Error->addError(NameRange, Error->ET_RegistryNotBindable);
Samuel Benzaquen0239b692013-08-13 14:54:51 +0000610 return VariantMatcher();
Samuel Benzaquen31edb512013-06-03 19:31:08 +0000611}
612
Manuel Klimek24db0f02013-05-14 09:13:00 +0000613} // namespace dynamic
614} // namespace ast_matchers
615} // namespace clang