diff --git a/clang-tools-extra/clangd/XRefs.cpp b/clang-tools-extra/clangd/XRefs.cpp --- a/clang-tools-extra/clangd/XRefs.cpp +++ b/clang-tools-extra/clangd/XRefs.cpp @@ -144,6 +144,21 @@ return Result; } +std::vector +getExplicitDeclAtPosition(ParsedAST &AST, SourceLocation Pos, + DeclRelationSet Relations) { + FileID FID; + unsigned Offset; + std::tie(FID, Offset) = AST.getSourceManager().getDecomposedSpellingLoc(Pos); + SelectionTree Selection(AST.getASTContext(), AST.getTokens(), Offset); + std::vector Result; + if (const SelectionTree::Node *N = Selection.commonAncestor()) { + auto Decls = explicitReferenceTargets(N->ASTNode, Relations); + Result.assign(Decls.begin(), Decls.end()); + } + return Result; +} + llvm::Optional makeLocation(ASTContext &AST, SourceLocation TokLoc, llvm::StringRef TUPath) { const SourceManager &SourceMgr = AST.getSourceManager(); @@ -672,9 +687,12 @@ const SourceManager &SM = AST.getSourceManager(); SourceLocation SourceLocationBeg = SM.getMacroArgExpandedLocation( getBeginningOfIdentifier(Pos, SM, AST.getLangOpts())); - DeclRelationSet Relations = - DeclRelation::TemplatePattern | DeclRelation::Underlying; - auto Decls = getDeclAtPosition(AST, SourceLocationBeg, Relations); + // Note: getExplicitDeclAtPosition() will search for both template + // instantiations and template patterns, and prefer the former if available + // (generally, one will be available for non-dependent specializations of a + // class template). + DeclRelationSet Relations = DeclRelation::Underlying; + auto Decls = getExplicitDeclAtPosition(AST, SourceLocationBeg, Relations); if (Decls.empty()) return nullptr; @@ -700,6 +718,14 @@ std::vector typeParents(const CXXRecordDecl *CXXRD) { std::vector Result; + // If this is an invalid instantiation, instantiation of the bases + // may not have succeeded, so fall back to the template pattern. + if (auto *CTSD = dyn_cast(CXXRD)) { + if (CTSD->isInvalidDecl()) { + CXXRD = CTSD->getSpecializedTemplate()->getTemplatedDecl(); + } + } + for (auto Base : CXXRD->bases()) { const CXXRecordDecl *ParentDecl = nullptr; @@ -754,6 +780,11 @@ Result->children.emplace(); if (Index) { + // The index does not store relationships between implicit + // specializations, so if we have one, use the template pattern instead. + if (auto *CTSD = dyn_cast(CXXRD)) + CXXRD = CTSD->getTemplateInstantiationPattern(); + if (Optional ID = getSymbolID(CXXRD)) fillSubTypes(*ID, *Result->children, Index, ResolveLevels, TUPath); } diff --git a/clang-tools-extra/clangd/unittests/TypeHierarchyTests.cpp b/clang-tools-extra/clangd/unittests/TypeHierarchyTests.cpp --- a/clang-tools-extra/clangd/unittests/TypeHierarchyTests.cpp +++ b/clang-tools-extra/clangd/unittests/TypeHierarchyTests.cpp @@ -409,17 +409,21 @@ ASSERT_TRUE(!AST.getDiagnostics().empty()); // Make sure getTypeHierarchy() doesn't get into an infinite recursion. - // FIXME(nridge): It would be preferable if the type hierarchy gave us type - // names (e.g. "S<0>" for the child and "S<1>" for the parent) rather than - // template names (e.g. "S"). + // The parent is reported as "S" because "S<0>" is an invalid instantiation. + // We then iterate once more and find "S" again before detecting the + // recursion. llvm::Optional Result = getTypeHierarchy( AST, Source.points()[0], 0, TypeHierarchyDirection::Parents); ASSERT_TRUE(bool(Result)); EXPECT_THAT( *Result, - AllOf(WithName("S"), WithKind(SymbolKind::Struct), - Parents(AllOf(WithName("S"), WithKind(SymbolKind::Struct), - SelectionRangeIs(Source.range("SDef")), Parents())))); + AllOf(WithName("S<0>"), WithKind(SymbolKind::Struct), + Parents( + AllOf(WithName("S"), WithKind(SymbolKind::Struct), + SelectionRangeIs(Source.range("SDef")), + Parents(AllOf(WithName("S"), WithKind(SymbolKind::Struct), + SelectionRangeIs(Source.range("SDef")), + Parents())))))); } TEST(TypeHierarchy, RecursiveHierarchyBounded) { @@ -449,9 +453,12 @@ ASSERT_TRUE(bool(Result)); EXPECT_THAT( *Result, - AllOf(WithName("S"), WithKind(SymbolKind::Struct), - Parents(AllOf(WithName("S"), WithKind(SymbolKind::Struct), - SelectionRangeIs(Source.range("SDef")), Parents())))); + AllOf(WithName("S<2>"), WithKind(SymbolKind::Struct), + Parents(AllOf( + WithName("S<1>"), WithKind(SymbolKind::Struct), + SelectionRangeIs(Source.range("SDef")), + Parents(AllOf(WithName("S<0>"), WithKind(SymbolKind::Struct), + Parents())))))); Result = getTypeHierarchy(AST, Source.point("SRefDependent"), 0, TypeHierarchyDirection::Parents); ASSERT_TRUE(bool(Result)); @@ -462,6 +469,83 @@ SelectionRangeIs(Source.range("SDef")), Parents())))); } +TEST(TypeHierarchy, DeriveFromImplicitSpec) { + Annotations Source(R"cpp( + template + struct Parent {}; + + struct Child : Parent {}; + + Parent Fo^o; + )cpp"); + + TestTU TU = TestTU::withCode(Source.code()); + auto AST = TU.build(); + auto Index = TU.index(); + ASSERT_TRUE(AST.getDiagnostics().empty()); + + llvm::Optional Result = getTypeHierarchy( + AST, Source.points()[0], 2, TypeHierarchyDirection::Children, Index.get(), + testPath(TU.Filename)); + ASSERT_TRUE(bool(Result)); + EXPECT_THAT(*Result, + AllOf(WithName("Parent"), WithKind(SymbolKind::Struct), + Children(AllOf(WithName("Child"), + WithKind(SymbolKind::Struct), Children())))); +} + +TEST(TypeHierarchy, DeriveFromPartialSpec) { + Annotations Source(R"cpp( + template struct Parent {}; + template struct Parent {}; + + struct Child : Parent {}; + + Parent Fo^o; + )cpp"); + + TestTU TU = TestTU::withCode(Source.code()); + auto AST = TU.build(); + auto Index = TU.index(); + ASSERT_TRUE(AST.getDiagnostics().empty()); + + llvm::Optional Result = getTypeHierarchy( + AST, Source.points()[0], 2, TypeHierarchyDirection::Children, Index.get(), + testPath(TU.Filename)); + ASSERT_TRUE(bool(Result)); + EXPECT_THAT(*Result, AllOf(WithName("Parent"), + WithKind(SymbolKind::Struct), Children())); +} + +TEST(TypeHierarchy, DeriveFromTemplate) { + Annotations Source(R"cpp( + template + struct Parent {}; + + template + struct Child : Parent {}; + + Parent Fo^o; + )cpp"); + + TestTU TU = TestTU::withCode(Source.code()); + auto AST = TU.build(); + auto Index = TU.index(); + ASSERT_TRUE(AST.getDiagnostics().empty()); + + // FIXME: We'd like this to return the implicit specialization Child, + // but currently libIndex does not expose relationships between + // implicit specializations. + llvm::Optional Result = getTypeHierarchy( + AST, Source.points()[0], 2, TypeHierarchyDirection::Children, Index.get(), + testPath(TU.Filename)); + ASSERT_TRUE(bool(Result)); + EXPECT_THAT(*Result, + AllOf(WithName("Parent"), WithKind(SymbolKind::Struct), + Children(AllOf(WithName("Child"), + WithKind(SymbolKind::Struct), Children())))); +} + SymbolID findSymbolIDByName(SymbolIndex *Index, llvm::StringRef Name, llvm::StringRef TemplateArgs = "") { SymbolID Result;