sammccall created this revision.
sammccall added a reviewer: kadircet.
Herald added subscribers: usaxena95, arphaman.
sammccall requested review of this revision.
Herald added subscribers: cfe-commits, MaskRay, ilya-biryukov.
Herald added a project: clang.
This requires a second index query for refs to overrides, as the refs
call doesn't tell you which ref points at which symbol.
Repository:
rG LLVM Github Monorepo
https://reviews.llvm.org/D95451
Files:
clang-tools-extra/clangd/XRefs.cpp
clang-tools-extra/clangd/unittests/XRefsTests.cpp
Index: clang-tools-extra/clangd/unittests/XRefsTests.cpp
===================================================================
--- clang-tools-extra/clangd/unittests/XRefsTests.cpp
+++ clang-tools-extra/clangd/unittests/XRefsTests.cpp
@@ -1872,7 +1872,7 @@
};
class Derived : public Base {
public:
- void $decl[[func]]() override; // FIXME: ref, not decl
+ void [[func]]() override;
};
void test(Derived* D) {
D->[[func]]();
Index: clang-tools-extra/clangd/XRefs.cpp
===================================================================
--- clang-tools-extra/clangd/XRefs.cpp
+++ clang-tools-extra/clangd/XRefs.cpp
@@ -872,6 +872,7 @@
struct Reference {
syntax::Token SpelledTok;
index::SymbolRoleSet Role;
+ SymbolID Target;
Range range(const SourceManager &SM) const {
return halfOpenToRange(SM, SpelledTok.range(SM).toCharRange(SM));
@@ -906,13 +907,15 @@
SourceLocation Loc,
index::IndexDataConsumer::ASTNodeInfo ASTNode) override {
const SourceManager &SM = AST.getSourceManager();
- if (!isInsideMainFile(Loc, SM) ||
- TargetIDs.find(getSymbolID(D)) == TargetIDs.end())
+ if (!isInsideMainFile(Loc, SM))
+ return true;
+ SymbolID ID = getSymbolID(D);
+ if (!TargetIDs.contains(ID))
return true;
const auto &TB = AST.getTokens();
Loc = SM.getFileLoc(Loc);
if (const auto *Tok = TB.spelledTokenAt(Loc))
- References.push_back({*Tok, Roles});
+ References.push_back({*Tok, Roles, ID});
return true;
}
@@ -1298,6 +1301,7 @@
}
RefsRequest Req;
+ llvm::DenseSet<SymbolID> Overrides;
const auto *IdentifierAtCursor =
syntax::spelledIdentifierTouching(*CurLoc, AST.getTokens());
@@ -1336,7 +1340,6 @@
if (auto ID = getSymbolID(D))
Targets.insert(ID);
- llvm::DenseSet<SymbolID> Overrides;
if (Index) {
RelationsRequest FindOverrides;
FindOverrides.Predicate = RelationKind::OverriddenBy;
@@ -1374,16 +1377,18 @@
ReferencesResult::Reference Result;
Result.Loc.range = Ref.range(SM);
Result.Loc.uri = URIMainFile;
- if (Ref.Role & static_cast<unsigned>(index::SymbolRole::Declaration))
- Result.Attributes |= ReferencesResult::Declaration;
- // clang-index doesn't report definitions as declarations, but they are.
- if (Ref.Role & static_cast<unsigned>(index::SymbolRole::Definition))
- Result.Attributes |=
- ReferencesResult::Definition | ReferencesResult::Declaration;
+ // Overrides are always considered references, not defs/decls.
+ if (!Overrides.contains(Ref.Target)) {
+ if (Ref.Role & static_cast<unsigned>(index::SymbolRole::Declaration))
+ Result.Attributes |= ReferencesResult::Declaration;
+ // clang-index doesn't report definitions as declarations, but they are.
+ if (Ref.Role & static_cast<unsigned>(index::SymbolRole::Definition))
+ Result.Attributes |=
+ ReferencesResult::Definition | ReferencesResult::Declaration;
+ }
Results.References.push_back(std::move(Result));
}
if (Index && Results.References.size() <= Limit) {
- Req.IDs = std::move(Overrides);
for (const Decl *D : Decls) {
// Not all symbols can be referenced from outside (e.g.
// function-locals).
@@ -1397,8 +1402,10 @@
}
}
// Now query the index for references from other files.
- if (!Req.IDs.empty() && Index && Results.References.size() <= Limit) {
- Req.Limit = Limit;
+ Req.Limit = Limit;
+ auto QueryIndex = [&](bool AllowAttributes) {
+ if (Req.IDs.empty() || !Index || Results.References.size() > Limit)
+ return;
Results.HasMore |= Index->refs(Req, [&](const Ref &R) {
// No need to continue process if we reach the limit.
if (Results.References.size() > Limit)
@@ -1409,15 +1416,22 @@
return;
ReferencesResult::Reference Result;
Result.Loc = std::move(*LSPLoc);
- if ((R.Kind & RefKind::Declaration) == RefKind::Declaration)
- Result.Attributes |= ReferencesResult::Declaration;
- // FIXME: our index should definitely store def | decl separately!
- if ((R.Kind & RefKind::Definition) == RefKind::Definition)
- Result.Attributes |=
- ReferencesResult::Declaration | ReferencesResult::Definition;
+ if (AllowAttributes) {
+ if ((R.Kind & RefKind::Declaration) == RefKind::Declaration)
+ Result.Attributes |= ReferencesResult::Declaration;
+ // FIXME: our index should definitely store def | decl separately!
+ if ((R.Kind & RefKind::Definition) == RefKind::Definition)
+ Result.Attributes |=
+ ReferencesResult::Declaration | ReferencesResult::Definition;
+ }
Results.References.push_back(std::move(Result));
});
- }
+ };
+ QueryIndex(/*AllowAttributes=*/true);
+ // FIXME: Currently we surface decls/defs/refs of derived methods.
+ // Maybe we'd prefer decls/defs of derived methods, and refs of base methods?
+ Req.IDs = std::move(Overrides);
+ QueryIndex(/*AllowAttributes=*/false);
if (Results.References.size() > Limit) {
Results.HasMore = true;
Results.References.resize(Limit);
_______________________________________________
cfe-commits mailing list
cfe-commits@lists.llvm.org
https://lists.llvm.org/cgi-bin/mailman/listinfo/cfe-commits
