diff options
Diffstat (limited to 'lib/apps/fuzzy.vala')
-rw-r--r-- | lib/apps/fuzzy.vala | 73 |
1 files changed, 73 insertions, 0 deletions
diff --git a/lib/apps/fuzzy.vala b/lib/apps/fuzzy.vala new file mode 100644 index 0000000..7745320 --- /dev/null +++ b/lib/apps/fuzzy.vala @@ -0,0 +1,73 @@ +namespace AstalApps { +private int fuzzy_match_string(string pattern, string str) { + const int unmatched_letter_penalty = -1; + int score = 100; + + if (pattern.length == 0) return score; + if (str.length < pattern.length) return int.MIN; + + bool found = fuzzy_match_recurse(pattern, str, score, true, out score); + score += unmatched_letter_penalty * (str.length - pattern.length); + + if(!found) score = -10; + + return score; +} + +private bool fuzzy_match_recurse(string pattern, string str, int score, bool first_char, out int result) { + result = score; + if (pattern.length == 0) return true; + + int match_idx = 0; + int offset = 0; + unichar search = pattern.casefold().get_char(0); + int best_score = int.MIN; + + while ((match_idx = str.casefold().substring(offset).index_of_char(search)) >= 0) { + offset += match_idx; + int subscore; + bool found = fuzzy_match_recurse( + pattern.substring(1), + str.substring(offset + 1), + compute_score(offset, first_char, str, offset), false, out subscore); + if(!found) break; + best_score = int.max(best_score, subscore); + offset++; + } + + if (best_score == int.MIN) return false; + result += best_score; + return true; +} + +private int compute_score(int jump, bool first_char, string match, int idx) { + const int adjacency_bonus = 15; + const int separator_bonus = 30; + const int camel_bonus = 30; + const int first_letter_bonus = 15; + const int leading_letter_penalty = -5; + const int max_leading_letter_penalty = -15; + + int score = 0; + + if (!first_char && jump == 0) { + score += adjacency_bonus; + } + if (!first_char || jump > 0) { + if (match[idx].isupper() && match[idx-1].islower()) { + score += camel_bonus; + } + if (match[idx].isalnum() && !match[idx-1].isalnum()) { + score += separator_bonus; + } + } + if (first_char && jump == 0) { + score += first_letter_bonus; + } + if (first_char) { + score += int.max(leading_letter_penalty * jump, max_leading_letter_penalty); + } + + return score; +} +} |