sammccall updated this revision to Diff 217645.
sammccall added a comment.
remove printfs
Repository:
rG LLVM Github Monorepo
CHANGES SINCE LAST ACTION
https://reviews.llvm.org/D66872/new/
https://reviews.llvm.org/D66872
Files:
clang-tools-extra/clangd/Selection.cpp
clang-tools-extra/clangd/unittests/SelectionTests.cpp
clang-tools-extra/clangd/unittests/TweakTests.cpp
Index: clang-tools-extra/clangd/unittests/TweakTests.cpp
===================================================================
--- clang-tools-extra/clangd/unittests/TweakTests.cpp
+++ clang-tools-extra/clangd/unittests/TweakTests.cpp
@@ -562,6 +562,9 @@
struct Visible {};
}
}
+ template <typename T> class Allocator;
+ template <typename T, typename alloc = Allocator<T>> class Vec{};
+ using VecInt = Vec<int>;
)cpp";
EXPECT_AVAILABLE("^a^u^t^o^ i = 0;");
@@ -597,6 +600,11 @@
// replace array types
EXPECT_EQ(apply(R"cpp(au^to x = "test")cpp"),
R"cpp(const char * x = "test")cpp");
+ // Default template parameters aren't printed.
+ EXPECT_EQ(apply(R"cpp(au^to x = Vec<int>();)cpp"),
+ R"cpp(Vec<int> x = Vec<int>();)cpp");
+ EXPECT_EQ(apply(R"cpp(au^to x = VecInt();)cpp"),
+ R"cpp(VecInt x = VecInt();)cpp");
}
} // namespace
Index: clang-tools-extra/clangd/unittests/SelectionTests.cpp
===================================================================
--- clang-tools-extra/clangd/unittests/SelectionTests.cpp
+++ clang-tools-extra/clangd/unittests/SelectionTests.cpp
@@ -329,6 +329,8 @@
#define ECHO(X) X
ECHO(EC^HO([[$C[[int]]) EC^HO(a]]));
]])cpp",
+ R"cpp( $C[[^$C[[int]] a^]]; )cpp",
+ R"cpp( $C[[^$C[[int]] a = $C[[5]]^]]; )cpp",
};
for (const char *C : Cases) {
Annotations Test(C);
Index: clang-tools-extra/clangd/Selection.cpp
===================================================================
--- clang-tools-extra/clangd/Selection.cpp
+++ clang-tools-extra/clangd/Selection.cpp
@@ -60,13 +60,13 @@
// Associates any tokens overlapping [Begin, End) with an AST node.
// Tokens that were already claimed by another AST node are not claimed again.
- // Returns whether the node is selected in the sense of SelectionTree.
- SelectionTree::Selection claim(unsigned Begin, unsigned End) {
+ // Updates Result if the node is selected in the sense of SelectionTree.
+ void claim(unsigned Begin, unsigned End, SelectionTree::Selection &Result) {
assert(Begin <= End);
// Fast-path for missing the selection entirely.
if (Begin >= SelEnd || End <= SelBegin)
- return SelectionTree::Unselected;
+ return;
// We will consider the range (at least partially) selected if it hit any
// selected and previously unclaimed token.
@@ -97,9 +97,13 @@
}
}
- if (!ClaimedAnyToken)
- return SelectionTree::Unselected;
- return PartialSelection ? SelectionTree::Partial : SelectionTree::Complete;
+ // If some tokens were previously claimed (Result != Unselected), we may
+ // upgrade from Partial->Complete, even if no new tokens were claimed.
+ // Important for [[int a]].
+ if (ClaimedAnyToken || Result) {
+ Result = std::max(Result, PartialSelection ? SelectionTree::Partial
+ : SelectionTree::Complete);
+ }
}
private:
@@ -161,7 +165,9 @@
assert(V.Stack.size() == 1 && "Unpaired push/pop?");
assert(V.Stack.top() == &V.Nodes.front());
// We selected TUDecl if tokens were unclaimed (or the file is empty).
- if (V.Nodes.size() == 1 || V.Claimed.claim(Begin, End)) {
+ SelectionTree::Selection UnclaimedTokens = SelectionTree::Unselected;
+ V.Claimed.claim(Begin, End, UnclaimedTokens);
+ if (UnclaimedTokens || V.Nodes.size() == 1) {
StringRef FileContent = AST.getSourceManager().getBufferData(File);
// Don't require the trailing newlines to be selected.
bool SelectedAll = Begin == 0 && End >= FileContent.rtrim().size();
@@ -317,10 +323,11 @@
Nodes.emplace_back();
Nodes.back().ASTNode = std::move(Node);
Nodes.back().Parent = Stack.top();
- // Early hit detection never selects the whole node.
Stack.push(&Nodes.back());
- Nodes.back().Selected =
- claimRange(Early) ? SelectionTree::Partial : SelectionTree::Unselected;
+ claimRange(Early, Nodes.back().Selected);
+ // Early hit detection never selects the whole node.
+ if (Nodes.back().Selected)
+ Nodes.back().Selected = SelectionTree::Partial;
}
// Pops a node off the ancestor stack, and finalizes it. Pairs with push().
@@ -328,8 +335,7 @@
void pop() {
Node &N = *Stack.top();
dlog("{1}pop: {0}", printNodeToString(N.ASTNode, PrintPolicy), indent(-1));
- if (auto Sel = claimRange(N.ASTNode.getSourceRange()))
- N.Selected = Sel;
+ claimRange(N.ASTNode.getSourceRange(), N.Selected);
if (N.Selected || !N.Children.empty()) {
// Attach to the tree.
N.Parent->Children.push_back(&N);
@@ -359,9 +365,10 @@
// Perform hit-testing of a complete Node against the selection.
// This runs for every node in the AST, and must be fast in common cases.
// This is usually called from pop(), so we can take children into account.
- SelectionTree::Selection claimRange(SourceRange S) {
+ // The existing state of Result is relevant (early/late claims can interact).
+ void claimRange(SourceRange S, SelectionTree::Selection &Result) {
if (!S.isValid())
- return SelectionTree::Unselected;
+ return;
// toHalfOpenFileRange() allows selection of constructs in macro args. e.g:
// #define LOOP_FOREVER(Body) for(;;) { Body }
// void IncrementLots(int &x) {
@@ -375,17 +382,16 @@
auto E = SM.getDecomposedLoc(Range->getEnd());
// Otherwise, nodes in macro expansions can't be selected.
if (B.first != SelFile || E.first != SelFile)
- return SelectionTree::Unselected;
+ return;
// Attempt to claim the remaining range. If there's nothing to claim, only
// children were selected.
- SelectionTree::Selection Result = Claimed.claim(B.second, E.second);
+ Claimed.claim(B.second, E.second, Result);
if (Result)
dlog("{1}hit selection: {0}",
SourceRange(SM.getComposedLoc(B.first, B.second),
SM.getComposedLoc(E.first, E.second))
.printToString(SM),
indent());
- return Result;
}
std::string indent(int Offset = 0) {
_______________________________________________
cfe-commits mailing list
cfe-commits@lists.llvm.org
https://lists.llvm.org/cgi-bin/mailman/listinfo/cfe-commits
