chore(dict): Automate more cleanup

This commit is contained in:
Ed Page 2023-06-08 08:54:36 -05:00
parent 6cc3d16d3f
commit 09f096a096
3 changed files with 2245 additions and 2248 deletions

File diff suppressed because it is too large Load diff

File diff suppressed because it is too large Load diff

View file

@ -1,9 +1,10 @@
use std::collections::BTreeMap; use std::collections::BTreeMap;
use std::collections::BTreeSet;
use std::collections::HashMap; use std::collections::HashMap;
use std::collections::HashSet; use std::collections::HashSet;
use unicase::UniCase; use unicase::UniCase;
type Dict = BTreeMap<UniCase<String>, Vec<String>>; type Dict = BTreeMap<UniCase<String>, BTreeSet<String>>;
#[test] #[test]
fn verify() { fn verify() {
@ -30,7 +31,9 @@ fn generate<W: std::io::Write>(file: &mut W, dict: &[u8]) {
let mut typo = i.next().expect("typo").to_owned(); let mut typo = i.next().expect("typo").to_owned();
typo.make_ascii_lowercase(); typo.make_ascii_lowercase();
let typo = UniCase::new(typo); let typo = UniCase::new(typo);
rows.entry(typo).or_insert_with(Vec::new).extend(i.map(|c| { rows.entry(typo)
.or_insert_with(BTreeSet::new)
.extend(i.map(|c| {
let mut c = c.to_owned(); let mut c = c.to_owned();
c.make_ascii_lowercase(); c.make_ascii_lowercase();
c c
@ -41,7 +44,7 @@ fn generate<W: std::io::Write>(file: &mut W, dict: &[u8]) {
.into_iter() .into_iter()
.filter(|(t, _)| is_word(t)) .filter(|(t, _)| is_word(t))
.filter_map(|(t, c)| { .filter_map(|(t, c)| {
let new_c: Vec<_> = c.into_iter().filter(|c| is_word(c)).collect(); let new_c: BTreeSet<_> = c.into_iter().filter(|c| is_word(c)).collect();
if new_c.is_empty() { if new_c.is_empty() {
None None
} else { } else {
@ -53,7 +56,7 @@ fn generate<W: std::io::Write>(file: &mut W, dict: &[u8]) {
let varcon_words = varcon_words(); let varcon_words = varcon_words();
let allowed_words = allowed_words(); let allowed_words = allowed_words();
let word_variants = proper_word_variants(); let word_variants = proper_word_variants();
let rows: Dict = rows let rows: Vec<_> = rows
.into_iter() .into_iter()
.filter(|(typo, _)| { .filter(|(typo, _)| {
let is_disallowed = varcon_words.contains(&unicase::UniCase::new(typo)); let is_disallowed = varcon_words.contains(&unicase::UniCase::new(typo));
@ -71,29 +74,34 @@ fn generate<W: std::io::Write>(file: &mut W, dict: &[u8]) {
} }
}) })
.map(|(typo, corrections)| { .map(|(typo, corrections)| {
let mut new_corrections = vec![]; let mut new_corrections = BTreeSet::new();
for correction in corrections { for correction in corrections {
let correction = word_variants let correction = word_variants
.get(correction.as_str()) .get(correction.as_str())
.and_then(|words| find_best_match(&typo, correction.as_str(), words)) .and_then(|words| find_best_match(&typo, correction.as_str(), words))
.unwrap_or(&correction); .unwrap_or(&correction);
new_corrections.push(correction.to_owned()); new_corrections.insert(correction.to_owned());
} }
(typo, new_corrections) (typo, new_corrections)
}) })
.collect(); .collect();
let mut dict = Dict::new();
for (bad, good) in rows {
let current = dict.entry(bad).or_default();
current.extend(good);
}
let corrections: std::collections::HashSet<_> = let corrections: std::collections::HashSet<_> =
rows.values().flatten().map(ToOwned::to_owned).collect(); dict.values().flatten().map(ToOwned::to_owned).collect();
let rows: Vec<_> = rows let rows: Vec<_> = dict
.into_iter() .into_iter()
.filter(|(typo, _)| !corrections.contains(typo.as_str())) .filter(|(typo, _)| !corrections.contains(typo.as_str()))
.collect(); .collect();
let mut wtr = csv::WriterBuilder::new().flexible(true).from_writer(file); let mut wtr = csv::WriterBuilder::new().flexible(true).from_writer(file);
for (typo, corrections) in rows { for (typo, corrections) in rows {
let mut row = corrections; let mut row = vec![typo.as_str().to_owned()];
row.insert(0, typo.as_str().to_owned()); row.extend(corrections);
wtr.write_record(&row).unwrap(); wtr.write_record(&row).unwrap();
} }
wtr.flush().unwrap(); wtr.flush().unwrap();