diff --git a/clang/include/clang/Tooling/NodeIntrospection.h b/clang/include/clang/Tooling/NodeIntrospection.h --- a/clang/include/clang/Tooling/NodeIntrospection.h +++ b/clang/include/clang/Tooling/NodeIntrospection.h @@ -22,6 +22,7 @@ namespace clang { class Stmt; +class Decl; namespace tooling { @@ -78,6 +79,7 @@ namespace NodeIntrospection { NodeLocationAccessors GetLocations(clang::Stmt const *Object); +NodeLocationAccessors GetLocations(clang::Decl const *Object); NodeLocationAccessors GetLocations(clang::DynTypedNode const &Node); } // namespace NodeIntrospection } // namespace tooling diff --git a/clang/lib/Tooling/CMakeLists.txt b/clang/lib/Tooling/CMakeLists.txt --- a/clang/lib/Tooling/CMakeLists.txt +++ b/clang/lib/Tooling/CMakeLists.txt @@ -40,6 +40,9 @@ NodeLocationAccessors NodeIntrospection::GetLocations(clang::Stmt const *) { return {}; } +NodeLocationAccessors NodeIntrospection::GetLocations(clang::Decl const *) { + return {}; +} NodeLocationAccessors NodeIntrospection::GetLocations(clang::DynTypedNode const &) { return {}; diff --git a/clang/lib/Tooling/DumpTool/ASTSrcLocProcessor.cpp b/clang/lib/Tooling/DumpTool/ASTSrcLocProcessor.cpp --- a/clang/lib/Tooling/DumpTool/ASTSrcLocProcessor.cpp +++ b/clang/lib/Tooling/DumpTool/ASTSrcLocProcessor.cpp @@ -26,7 +26,8 @@ isDefinition(), isSameOrDerivedFrom( // TODO: Extend this with other clades - namedDecl(hasName("clang::Stmt")).bind("nodeClade")), + namedDecl(hasAnyName("clang::Stmt", "clang::Decl")) + .bind("nodeClade")), optionally(isDerivedFrom(cxxRecordDecl().bind("derivedFrom")))) .bind("className"), this); diff --git a/clang/lib/Tooling/DumpTool/generate_cxx_src_locs.py b/clang/lib/Tooling/DumpTool/generate_cxx_src_locs.py --- a/clang/lib/Tooling/DumpTool/generate_cxx_src_locs.py +++ b/clang/lib/Tooling/DumpTool/generate_cxx_src_locs.py @@ -177,6 +177,9 @@ NodeLocationAccessors NodeIntrospection::GetLocations(clang::Stmt const *) { return {}; } +NodeLocationAccessors NodeIntrospection::GetLocations(clang::Decl const *) { + return {}; +} NodeLocationAccessors NodeIntrospection::GetLocations(clang::DynTypedNode const &) { return {}; diff --git a/clang/unittests/Introspection/IntrospectionTest.cpp b/clang/unittests/Introspection/IntrospectionTest.cpp --- a/clang/unittests/Introspection/IntrospectionTest.cpp +++ b/clang/unittests/Introspection/IntrospectionTest.cpp @@ -42,7 +42,7 @@ #define STRING_LOCATION_PAIR(INSTANCE, LOC) Pair(#LOC, INSTANCE->LOC) -TEST(Introspection, SourceLocations) { +TEST(Introspection, SourceLocations_Stmt) { auto AST = buildASTFromCode("void foo() {} void bar() { foo(); }", "foo.cpp", std::make_shared()); auto &Ctx = AST->getASTContext(); @@ -79,3 +79,69 @@ EXPECT_THAT(ExpectedRanges, UnorderedElementsAre(STRING_LOCATION_PAIR( FooCall, getSourceRange()))); } + +TEST(Introspection, SourceLocations_Decl) { + auto AST = + buildASTFromCode(R"cpp( +namespace ns1 { +namespace ns2 { +template struct Foo {}; +template struct Bar { + struct Nested { + template + Foo method(int i, bool b) const noexcept(true); + }; +}; +} // namespace ns2 +} // namespace ns1 + +template +template +ns1::ns2::Foo ns1::ns2::Bar::Nested::method(int i, bool b) const + noexcept(true) {} +)cpp", + "foo.cpp", std::make_shared()); + auto &Ctx = AST->getASTContext(); + auto &TU = *Ctx.getTranslationUnitDecl(); + + auto BoundNodes = ast_matchers::match( + decl(hasDescendant( + cxxMethodDecl(hasName("method")).bind("method"))), + TU, Ctx); + + EXPECT_EQ(BoundNodes.size(), 1u); + + const auto *MethodDecl = BoundNodes[0].getNodeAs("method"); + + auto Result = NodeIntrospection::GetLocations(MethodDecl); + + if (Result.LocationAccessors.empty() && Result.RangeAccessors.empty()) { + return; + } + + auto ExpectedLocations = + FormatExpected(Result.LocationAccessors); + + EXPECT_THAT(ExpectedLocations, + UnorderedElementsAre( + STRING_LOCATION_PAIR(MethodDecl, getBeginLoc()), + STRING_LOCATION_PAIR(MethodDecl, getBodyRBrace()), + STRING_LOCATION_PAIR(MethodDecl, getEllipsisLoc()), + STRING_LOCATION_PAIR(MethodDecl, getInnerLocStart()), + STRING_LOCATION_PAIR(MethodDecl, getLocation()), + STRING_LOCATION_PAIR(MethodDecl, getOuterLocStart()), + STRING_LOCATION_PAIR(MethodDecl, getPointOfInstantiation()), + STRING_LOCATION_PAIR(MethodDecl, getTypeSpecEndLoc()), + STRING_LOCATION_PAIR(MethodDecl, getTypeSpecStartLoc()), + STRING_LOCATION_PAIR(MethodDecl, getEndLoc()))); + + auto ExpectedRanges = FormatExpected(Result.RangeAccessors); + + EXPECT_THAT( + ExpectedRanges, + UnorderedElementsAre( + STRING_LOCATION_PAIR(MethodDecl, getExceptionSpecSourceRange()), + STRING_LOCATION_PAIR(MethodDecl, getParametersSourceRange()), + STRING_LOCATION_PAIR(MethodDecl, getReturnTypeSourceRange()), + STRING_LOCATION_PAIR(MethodDecl, getSourceRange()))); +}