libmotiva 0.1.1

Sanctioned entities matching utilities
Documentation 
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

use bumpalo::{
  Bump,
  collections::{CollectIn, Vec},
};
use itertools::Itertools;
use libmotiva_macros::scoring_feature;
use rphonetic::Metaphone;

use crate::{
  matching::{Feature, comparers::compare_name_phonetic_tuples, extractors},
  model::{Entity, HasProperties, SearchEntity},
};

#[scoring_feature(PersonNamePhoneticMatch, name = "person_name_phonetic_match")]
fn score_feature(&self, bump: &Bump, lhs: &SearchEntity, rhs: &Entity) -> f64 {
  if !lhs.schema.is_a("Person") && !rhs.schema.is_a("Person") {
    return 0.0;
  }

  let lhs_names = extractors::clean_names(lhs.names_and_aliases().iter()).collect_in::<Vec<_>>(bump);
  let rhs_names = extractors::clean_names(rhs.names_and_aliases().iter()).collect_in::<Vec<_>>(bump);

  let metaphone = Metaphone::new(None);
  let lhs_phone = extractors::phonetic_names_tuples(&metaphone, lhs_names.iter());
  let rhs_phone = extractors::phonetic_names_tuples(&metaphone, rhs_names.iter());

  let mut score = 0.0f64;

  lhs_phone.iter().cartesian_product(rhs_phone.iter()).for_each(|(ls, rs)| {
    let mut matched = 0;
    let mut comp = rs.clone();

    for (l_name, l_phone) in ls {
      for (r_name, r_phone) in comp.clone().iter() {
        if compare_name_phonetic_tuples((l_name, l_phone.as_deref()), (r_name, r_phone.as_deref())) {
          matched += 1;

          if let Some(index) = comp.iter().position(|x| *x == (r_name, r_phone.clone())) {
            comp.remove(index);
          }

          break;
        }
      }
    }

    score = score.max(matched as f64 / ls.len() as f64);
  });

  score
}