dddlmZddlmZmZmZddlmZdd ZddZ d dZ d!dZ dddd"dZ dddd"dZ dddd"dZdddd"dZdS)#) annotations)CallableHashableSequence)is_none pattern_leninttext_len common_charstranspositionsreturnfloatcP|dz}d}|||z z }|||z z }|||z |z z }|dz S)Ng@)rr r r sims ^/home/feoh/.local/pipx/venvs/poetry/lib/python3.11/site-packages/rapidfuzz/distance/Jaro_py.py_jaro_calculate_similarityr sOqN C<+ %%C<( ""CL> )\ 99C 9 score_cutoffboolcZ|r|sdSt||t||d}||kS)zC filter matches below score_cutoff based on string lengths Fr)rmin)rr rrs r_jaro_length_filterrsG hu $Xs;991  C , rc:|sdSt|||d}||kS)zY filter matches below score_cutoff based on string lengths and common characters Fr)r)rr r rrs r_jaro_common_char_filterr#s/ u $[(L! L LC , rs1Sequence[Hashable]s22tuple[Sequence[Hashable], Sequence[Hashable], int]ct|}t|}d}||kr|dzdz }|||zkr |d||z}n|dzdz }|||zkr |d||z}|||fS)zB find bounds and skip out of bound parts of the sequences rrN)len)rr rr bounds r _jaro_boundsr&0s b''K2wwH E+A ! kE) ) ))kE))*Bq 1$ E) ) )&h&&'B r5=rN processorrr((Callable[..., Sequence[Hashable]] | None float | Nonec\t|st|rdS|||}||}|d}t|}t|}t|||sdS|dkr%|dkrt|d|dkSt ||\}}}dg|z}dg|z}d} t |D]i\} } t d| |z } t| |z|dz } t| | dzD]'}||s||| krdx|| <||<| dz } n(jt||| |sdSdx}}t |D]@\} }|r9t||D]}||r|dz}n|| ||kr|dz }At||| |S)a Calculates the jaro similarity Parameters ---------- s1 : Sequence[Hashable] First string to compare. s2 : Sequence[Hashable] Second string to compare. processor: callable, optional Optional callable that is used to preprocess the strings before comparing them. Default is None, which deactivates this behaviour. score_cutoff : float, optional Optional argument for a score threshold as a float between 0 and 1.0. For ratio < score_cutoff 0 is returned instead. Default is None, which deactivates this behaviour. Returns ------- similarity : float similarity between s1 and s2 as a float between 0 and 1.0 rNrr#FT) rr$rrr& enumeratemaxrrangerr)rr r(rrr r%s1_flagss2_flagsr is1_chlowhijk trans_counts1_fs r similarityr9Gs2:r{{gbkks Yr]] Yr]] b''K2wwH {Hl C CqaHMMRUbe^$$$ R((MBEw$Hw!HLbMM5!QY UHqL ) )sBF##  AA; 2a5E>>,00 hqk!  $K< V VqA X&&!!4  !1h''  A;AAE!u1~~q %k8\; W WWrc(t||||S)a Calculates the normalized jaro similarity Parameters ---------- s1 : Sequence[Hashable] First string to compare. s2 : Sequence[Hashable] Second string to compare. processor: callable, optional Optional callable that is used to preprocess the strings before comparing them. Default is None, which deactivates this behaviour. score_cutoff : float, optional Optional argument for a score threshold as a float between 0 and 1.0. For ratio < score_cutoff 0 is returned instead. Default is None, which deactivates this behaviour. Returns ------- normalized similarity : float normalized similarity between s1 and s2 as a float between 0 and 1.0 r')r9rr r(rs rnormalized_similarityr<s: b" M M MMrct|st|rdS|||}||}||dkrdnd|z }t|||}d|z }|||kr|ndS)a Calculates the jaro distance Parameters ---------- s1 : Sequence[Hashable] First string to compare. s2 : Sequence[Hashable] Second string to compare. processor: callable, optional Optional callable that is used to preprocess the strings before comparing them. Default is None, which deactivates this behaviour. score_cutoff : float, optional Optional argument for a score threshold as a float between 0 and 1.0. For ratio < score_cutoff 0 is returned instead. Default is None, which deactivates this behaviour. Returns ------- distance : float distance between s1 and s2 as a float between 1.0 and 0.0 g?N)r)rr9)rr r(rcutoff_distancerdists rdistancer@s:r{{gbkks Yr]] Yr]]%););# BT R/ : : :C 9D (DL,@,@44sJrc(t||||S)a Calculates the normalized jaro distance Parameters ---------- s1 : Sequence[Hashable] First string to compare. s2 : Sequence[Hashable] Second string to compare. processor: callable, optional Optional callable that is used to preprocess the strings before comparing them. Default is None, which deactivates this behaviour. score_cutoff : float, optional Optional argument for a score threshold as a float between 0 and 1.0. For ratio < score_cutoff 0 is returned instead. Default is None, which deactivates this behaviour. Returns ------- normalized distance : float normalized distance between s1 and s2 as a float between 1.0 and 0.0 r')r@r;s rnormalized_distancerBs: Bil K K KKr) rr r r r r r r r r)rr r r rrr r) rr r r r r rrr r)rrr rr r!) rrr rr(r)rr*r r) __future__rtypingrrrrapidfuzz._utilsrrrrr&r9r<r@rBrrrrFs^#"""""//////////$$$$$$        6;?!% RXRXRXRXRXRXr;?!% NNNNNNH;?!% )K)K)K)K)K)K`;?!% LLLLLLLLr