mirrors/NewPipe
1
0
Fork 0
mirror of https://github.com/TeamNewPipe/NewPipe synced 2024-06-25 22:53:20 +00:00
Code Issues Releases Wiki Activity

Calculate search score using streams.

Browse Source 
Co-authored-by: Stypox <stypox@pm.me>
This commit is contained in:
Isira Seneviratne 2022-08-03 06:33:54 +05:30
parent b9b09d325a
commit 408a71cfdc
1 changed files with 10 additions and 25 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

35
app/src/main/java/org/schabi/newpipe/settings/preferencesearch/PreferenceFuzzySearchFunction.java
View File

@ -1,6 +1,7 @@
package org.schabi.newpipe.settings.preferencesearch; package org.schabi.newpipe.settings.preferencesearch;
import android.text.TextUtils; import android.text.TextUtils;
import android.util.Pair;
import org.apache.commons.text.similarity.FuzzyScore; import org.apache.commons.text.similarity.FuzzyScore;
@ -8,6 +9,7 @@ import java.util.Comparator;
import java.util.Locale; import java.util.Locale;
import java.util.Map; import java.util.Map;
import java.util.function.Function; import java.util.function.Function;
import java.util.stream.Collectors;
import java.util.stream.Stream; import java.util.stream.Stream;
public class PreferenceFuzzySearchFunction public class PreferenceFuzzySearchFunction
@ -72,39 +74,22 @@ public class PreferenceFuzzySearchFunction
); );
private final PreferenceSearchItem item; private final PreferenceSearchItem item;
private final float score; private final double score;
FuzzySearchSpecificDTO( FuzzySearchSpecificDTO(final PreferenceSearchItem item, final String keyword) {
final PreferenceSearchItem item,
final String keyword) {
this.item = item; this.item = item;
this.score = WEIGHT_MAP.entrySet().stream()
float attributeScoreSum = 0; .map(entry -> new Pair<>(entry.getKey().apply(item), entry.getValue()))
int countOfAttributesWithScore = 0; .filter(pair -> !pair.first.isEmpty())
for (final Map.Entry<Function<PreferenceSearchItem, String>, Float> we .collect(Collectors.averagingDouble(pair ->
: WEIGHT_MAP.entrySet()) { FUZZY_SCORE.fuzzyScore(pair.first, keyword) * pair.second));
final String valueToProcess = we.getKey().apply(item);
if (valueToProcess.isEmpty()) {
continue;
}
attributeScoreSum +=
FUZZY_SCORE.fuzzyScore(valueToProcess, keyword) * we.getValue();
countOfAttributesWithScore++;
}
if (countOfAttributesWithScore != 0) {
this.score = attributeScoreSum / countOfAttributesWithScore;
} else {
this.score = 0;
}
} }
public PreferenceSearchItem getItem() { public PreferenceSearchItem getItem() {
return item; return item;
} }
public float getScore() { public double getScore() {
return score; return score;
} }
} }