kadircet created this revision.
kadircet added a reviewer: ilya-biryukov.
Herald added subscribers: cfe-commits, arphaman, jkorous, MaskRay, ioeric.
Repository:
rCTE Clang Tools Extra
https://reviews.llvm.org/D50555
Files:
clangd/CodeComplete.cpp
clangd/Quality.cpp
clangd/Quality.h
unittests/clangd/CodeCompleteTests.cpp
Index: unittests/clangd/CodeCompleteTests.cpp
===================================================================
--- unittests/clangd/CodeCompleteTests.cpp
+++ unittests/clangd/CodeCompleteTests.cpp
@@ -1438,6 +1438,28 @@
}
}
+TEST(SignatureHelpTest, OverloadsOrdering) {
+ const auto Results = signatures(R"cpp(
+ void foo(int x);
+ void foo(int x, float y);
+ void foo(float x, int y);
+ void foo(float x, float y);
+ void foo(int x, int y = 0);
+ int main() { foo(^); }
+ )cpp");
+ EXPECT_THAT(
+ Results.signatures,
+ ElementsAre(
+ Sig("foo(int x) -> void", {"int x"}),
+ Sig("foo(int x, int y = 0) -> void", {"int x", "int y = 0"}),
+ Sig("foo(float x, int y) -> void", {"float x", "int y"}),
+ Sig("foo(int x, float y) -> void", {"int x", "float y"}),
+ Sig("foo(float x, float y) -> void", {"float x", "float y"})));
+ // We always prefer the first signature.
+ EXPECT_EQ(0, Results.activeSignature);
+ EXPECT_EQ(0, Results.activeParameter);
+}
+
} // namespace
} // namespace clangd
} // namespace clang
Index: clangd/Quality.h
===================================================================
--- clangd/Quality.h
+++ clangd/Quality.h
@@ -163,6 +163,17 @@
/// LSP. (The highest score compares smallest so it sorts at the top).
std::string sortText(float Score, llvm::StringRef Tiebreak = "");
+struct SignatureQualitySignals {
+ uint32_t NumberOfParameters = 0;
+ uint32_t NumberOfOptionalParameters = 0;
+ bool ContainsActiveParameter = false;
+ CodeCompleteConsumer::OverloadCandidate::CandidateKind Kind;
+
+ float evaluate() const;
+};
+llvm::raw_ostream &operator<<(llvm::raw_ostream &,
+ const SignatureQualitySignals &);
+
} // namespace clangd
} // namespace clang
Index: clangd/Quality.cpp
===================================================================
--- clangd/Quality.cpp
+++ clangd/Quality.cpp
@@ -401,5 +401,36 @@
return S;
}
+llvm::raw_ostream &operator<<(llvm::raw_ostream &OS,
+ const SignatureQualitySignals &S) {
+ OS << formatv("=== Signature Quality: {0}\n", S.evaluate());
+ OS << formatv("\tNumber of parameters: {0}\n", S.NumberOfParameters);
+ OS << formatv("\tNumber of optional parameters: {0}\n",
+ S.NumberOfOptionalParameters);
+ OS << formatv("\tContains active parameter: {0}\n",
+ S.ContainsActiveParameter);
+ OS << formatv("\tKind: {0}\n", S.Kind);
+ return OS;
+}
+
+float SignatureQualitySignals::evaluate() const {
+ using OC = CodeCompleteConsumer::OverloadCandidate;
+ float score = 1;
+ score /= NumberOfParameters + 1;
+ if (ContainsActiveParameter)
+ score *= 10;
+ switch (Kind) {
+ case OC::CK_Function:
+ score *= 10;
+ break;
+ case OC::CK_FunctionType:
+ score *= 5;
+ break;
+ case OC::CK_FunctionTemplate:
+ break;
+ }
+ return score;
+}
+
} // namespace clangd
} // namespace clang
Index: clangd/CodeComplete.cpp
===================================================================
--- clangd/CodeComplete.cpp
+++ clangd/CodeComplete.cpp
@@ -128,16 +128,16 @@
/// Get the optional chunk as a string. This function is possibly recursive.
///
/// The parameter info for each parameter is appended to the Parameters.
-std::string
-getOptionalParameters(const CodeCompletionString &CCS,
- std::vector<ParameterInformation> &Parameters) {
+std::string getOptionalParameters(const CodeCompletionString &CCS,
+ std::vector<ParameterInformation> &Parameters,
+ SignatureQualitySignals &Signal) {
std::string Result;
for (const auto &Chunk : CCS) {
switch (Chunk.Kind) {
case CodeCompletionString::CK_Optional:
assert(Chunk.Optional &&
"Expected the optional code completion string to be non-null.");
- Result += getOptionalParameters(*Chunk.Optional, Parameters);
+ Result += getOptionalParameters(*Chunk.Optional, Parameters, Signal);
break;
case CodeCompletionString::CK_VerticalSpace:
break;
@@ -153,6 +153,8 @@
ParameterInformation Info;
Info.label = Chunk.Text;
Parameters.push_back(std::move(Info));
+ Signal.ContainsActiveParameter = true;
+ Signal.NumberOfOptionalParameters++;
break;
}
default:
@@ -679,6 +681,24 @@
llvm::unique_function<void()> ResultsCallback;
};
+using SignatureInformationScore = float;
+using ScoredSignature =
+ std::pair<SignatureInformationScore, SignatureInformation>;
+struct ScoredSignatureGreater {
+ bool operator()(const ScoredSignature &L, const ScoredSignature &R) {
+ // Ordering follows:
+ // - High score is better.
+ // - Shorter signature is better.
+ // - Alphebatically smaller is better.
+ if (L.first == R.first) {
+ if (L.second.label.size() == R.second.label.size())
+ return L.second.label < R.second.label;
+ return L.second.label.size() < R.second.label.size();
+ }
+ return L.first > R.first;
+ }
+};
+
class SignatureHelpCollector final : public CodeCompleteConsumer {
public:
@@ -692,6 +712,8 @@
void ProcessOverloadCandidates(Sema &S, unsigned CurrentArg,
OverloadCandidate *Candidates,
unsigned NumCandidates) override {
+ TopN<ScoredSignature, ScoredSignatureGreater> Top(
+ std::numeric_limits<size_t>::max());
SigHelp.signatures.reserve(NumCandidates);
// FIXME(rwols): How can we determine the "active overload candidate"?
// Right now the overloaded candidates seem to be provided in a "best fit"
@@ -706,11 +728,14 @@
CurrentArg, S, *Allocator, CCTUInfo, true);
assert(CCS && "Expected the CodeCompletionString to be non-null");
// FIXME: for headers, we need to get a comment from the index.
- SigHelp.signatures.push_back(processOverloadCandidate(
+ Top.push(processOverloadCandidate(
Candidate, *CCS,
getParameterDocComment(S.getASTContext(), Candidate, CurrentArg,
/*CommentsFromHeaders=*/false)));
}
+ for (const auto SS : std::move(Top).items()) {
+ SigHelp.signatures.push_back(SS.second);
+ }
}
GlobalCodeCompletionAllocator &getAllocator() override { return *Allocator; }
@@ -720,14 +745,15 @@
private:
// FIXME(ioeric): consider moving CodeCompletionString logic here to
// CompletionString.h.
- SignatureInformation
- processOverloadCandidate(const OverloadCandidate &Candidate,
- const CodeCompletionString &CCS,
- llvm::StringRef DocComment) const {
+ ScoredSignature processOverloadCandidate(const OverloadCandidate &Candidate,
+ const CodeCompletionString &CCS,
+ llvm::StringRef DocComment) const {
SignatureInformation Result;
+ SignatureQualitySignals Signal;
const char *ReturnType = nullptr;
Result.documentation = formatDocumentation(CCS, DocComment);
+ Signal.Kind = Candidate.getKind();
for (const auto &Chunk : CCS) {
switch (Chunk.Kind) {
@@ -749,14 +775,16 @@
ParameterInformation Info;
Info.label = Chunk.Text;
Result.parameters.push_back(std::move(Info));
+ Signal.NumberOfParameters++;
+ Signal.ContainsActiveParameter = true;
break;
}
case CodeCompletionString::CK_Optional: {
// The rest of the parameters are defaulted/optional.
assert(Chunk.Optional &&
"Expected the optional code completion string to be non-null.");
Result.label +=
- getOptionalParameters(*Chunk.Optional, Result.parameters);
+ getOptionalParameters(*Chunk.Optional, Result.parameters, Signal);
break;
}
case CodeCompletionString::CK_VerticalSpace:
@@ -770,7 +798,8 @@
Result.label += " -> ";
Result.label += ReturnType;
}
- return Result;
+ dlog("Signal for {0}: {1}", Result, Signal);
+ return {Signal.evaluate(), Result};
}
SignatureHelp &SigHelp;
_______________________________________________
cfe-commits mailing list
cfe-commits@lists.llvm.org
http://lists.llvm.org/cgi-bin/mailman/listinfo/cfe-commits
