1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
|
//! a simple fuzzy pattern matcher for filename filtering / sorting.
//! It's not meant for file contents but for small strings (less than 1000 chars)
//! such as file names.
use crate::patterns::Match;
use std::fmt::{self, Write};
// weights used in match score computing
const BONUS_MATCH: i32 = 50_000;
const BONUS_EXACT: i32 = 1_000;
const BONUS_START: i32 = 10;
const BONUS_START_WORD: i32 = 5;
const BONUS_CANDIDATE_LENGTH: i32 = -1; // per char
const BONUS_LENGTH: i32 = -10; // per char of length of the match
const BONUS_NB_HOLES: i32 = -30; // there's also a max on that number
#[derive(Debug, Clone)]
pub struct FuzzyPattern {
lc_chars: Box<[char]>, // lowercase characters
max_nb_holes: usize,
}
impl fmt::Display for FuzzyPattern {
fn fmt(&self, f: &mut fmt::Formatter) -> fmt::Result {
for &c in self.lc_chars.iter() {
f.write_char(c)?
}
Ok(())
}
}
impl FuzzyPattern {
pub fn from(pat: &str) -> FuzzyPattern {
let lc_chars: Vec<char> = pat.chars().map(|c| c.to_ascii_lowercase()).collect();
let lc_chars = lc_chars.into_boxed_slice();
let max_nb_holes = match lc_chars.len() {
1 => 0,
2 => 1,
3 => 2,
4 => 3,
5 => 3,
6 => 4,
7 => 4,
8 => 4,
_ => lc_chars.len() * 4 / 7,
};
FuzzyPattern {
lc_chars,
max_nb_holes,
}
}
fn match_starting_at_index(
&self,
cand_chars: &[char],
start_idx: usize, // start index in candidate
) -> Option<Match> {
if cand_chars[start_idx] != self.lc_chars[0] {
return None;
}
let mut pos: Vec<usize> = vec![]; // positions of matching chars in candidate
pos.push(start_idx);
let mut d = 1;
let mut nb_holes = 0;
for pat_idx in 1..self.lc_chars.len() {
let hole_start = d;
loop {
let cand_idx = start_idx + d;
if cand_idx == cand_chars.len() {
return None;
}
d += 1;
if cand_chars[cand_idx] == self.lc_chars[pat_idx] {
pos.push(cand_idx);
break;
}
}
if hole_start + 1 != d {
// note that there's no absolute guarantee we found the minimal
// number of holes. The algorithm isn't perfect
if nb_holes >= self.max_nb_holes {
return None;
}
nb_holes += 1;
}
}
let mut score = BONUS_MATCH;
score += BONUS_CANDIDATE_LENGTH * (cand_chars.len() as i32);
score += BONUS_NB_HOLES * (nb_holes as i32);
let match_len = (d as i32) - 1;
score += match_len * BONUS_LENGTH;
if start_idx == 0 {
score += BONUS_START;
if cand_chars.len() == self.lc_chars.len() {
score += BONUS_EXACT;
}
} else {
let previous = cand_chars[start_idx - 1];
if previous == '_' || previous == ' ' || previous == '-' {
score += BONUS_START_WORD;
}
}
Some(Match { score, pos })
}
// return a match if the pattern can be found in the candidate string.
// The algorithm tries to return the best one. For example if you search
// "abc" in "ababaca-abc", the returned match would be at the end.
pub fn find(&self, candidate: &str) -> Option<Match> {
let cand_chars: Vec<char> = candidate.chars().map(|c| c.to_ascii_lowercase()).collect();
if cand_chars.len() < self.lc_chars.len() {
return None;
}
let mut best_score = 0;
let mut best_match: Option<Match> = None;
for start_idx in 0..=cand_chars.len() - self.lc_chars.len() {
let sm = self.match_starting_at_index(&cand_chars, start_idx);
if let Some(m) = sm {
if m.score > best_score {
best_score = m.score;
best_match = Some(m);
}
}
}
best_match
}
// return the number of results we should find before starting to
// sort them (unless time is runing out).
pub fn optimal_result_number(&self, targeted_size: usize) -> usize {
20 * targeted_size
}
}
|