Mirror/MarginaliaSearch
1
0
Fork 0
mirror of https://github.com/MarginaliaSearch/MarginaliaSearch.git synced 2025-02-23 21:18:58 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

(qs, WIP) Tidy it up a bit

Browse Source
This commit is contained in:
Viktor Lofgren 2024-03-28 14:18:26 +01:00
parent 0bd1e15cce
commit 98a1adbf81
2 changed files with 20 additions and 8 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
code/functions/search-query/java/nu/marginalia/functions/searchquery/query_parser/model/QWordGraph.java
View File

@ -225,7 +225,8 @@ public class QWordGraph implements Iterable<QWord> {
}
public Comparator<QWord> topologicalComparator() {
return Comparator.comparing(sortOrder::get);
Comparator<QWord> comp = Comparator.comparing(sortOrder::get);
return comp.thenComparing(QWord::ord);
}
}

25
code/functions/search-query/java/nu/marginalia/functions/searchquery/query_parser/model/QWordPathsRenderer.java
View File

@ -21,6 +21,13 @@ class QWordPathsRenderer {
return new QWordPathsRenderer(graph).render(graph.reachability());
}
private static String render(Collection<QWordPath> paths,
QWordGraph.ReachabilityData reachability)
{
return new QWordPathsRenderer(paths).render(reachability);
}
/** Render the paths into a human-readable infix-style expression.
* <p></p>
* This method is recursive, but the recursion depth is limited by the
@ -34,8 +41,7 @@ class QWordPathsRenderer {
// Find the commonality of words in the paths
Map<QWord, Integer> commonality = paths.stream().flatMap(QWordPath::stream)
.collect(Collectors.groupingBy(w -> w, Collectors.summingInt(w -> 1)));
Map<QWord, Integer> commonality = nodeCommonality();
// Break the words into two categories: those that are common to all paths, and those that are not
@ -72,10 +78,7 @@ class QWordPathsRenderer {
}
// Recurse into the non-overlapping portions
if (!nonOverlappingPortions.isEmpty()) {
var wp = new QWordPathsRenderer(nonOverlappingPortions);
resultJoiner.add(wp.render(reachability));
}
resultJoiner.add(render(nonOverlappingPortions, reachability));
}
} else if (commonality.size() > 1) { // The case where no words are common to all paths
@ -117,8 +120,10 @@ class QWordPathsRenderer {
.sorted(Map.Entry.comparingByKey(reachability.topologicalComparator())) // Sort by topological order to ensure consistent output
.map(e -> {
String commonWord = e.getKey().word();
// Recurse into the branches:
String branchPart = new QWordPathsRenderer(e.getValue()).render(reachability);
String branchPart = render(e.getValue(), reachability);
return STR."\{commonWord} \{branchPart}";
})
.collect(Collectors.joining(" | ", " ( ", " ) "));
@ -130,4 +135,10 @@ class QWordPathsRenderer {
return resultJoiner.toString().replaceAll("\\s+", " ").trim();
}
/** Compute how many paths each word is part of */
private Map<QWord, Integer> nodeCommonality() {
return paths.stream().flatMap(QWordPath::stream)
.collect(Collectors.groupingBy(w -> w, Collectors.summingInt(w -> 1)));
}
}