dzddlmZddlmZmZmZddlmZdd Zd d d ddZ d d d ddZ d d d ddZ d d d ddZ d S)) annotations)CallableHashableSequence)is_nones1Sequence[Hashable]s2returnintc|st|Sdt|zdz }d}d}d}t|}dt|dz z}i}|j} d} |D]} | | d| z|| <| dz} |D]s} | | d} || zdz|z}| |z|z|z | z|z}||z}|||zz}||z}|||zdkz }|||zdkz}|dzdz}|dz}|||zz}||z}| }t|S)Nr)lenget)rr VPVND0PM_j_oldcurrDistmaskblock block_getxch1ch2PM_jTRHPHNs ]/home/feoh/.local/pipx/venvs/poetry/lib/python3.11/site-packages/rapidfuzz/distance/OSA_py.py_osa_distance_hyrroe2003r! su 2ww s2ww,! B B BH2wwH R1 D!#E I AYsA&&*c  aya  t|!X -r R2% - 2 "WBG*_ "W R$Y1$$R$Y1$$Ag] 1W BG*_ "W ON) processor score_cutoffr#(Callable[..., Sequence[Hashable]] | Noner$ int | Nonecp|||}||}t||}|||kr|n|dzS)a Calculates the optimal string alignment (OSA) 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 : int, optional Maximum distance between s1 and s2, that is considered as a result. If the distance is bigger than score_cutoff, score_cutoff + 1 is returned instead. Default is None, which deactivates this behaviour. Returns ------- distance : int distance between s1 and s2 Examples -------- Find the OSA distance between two strings: >>> from rapidfuzz.distance import OSA >>> OSA.distance("CA", "AC") 2 >>> OSA.distance("CA", "ABC") 3 Nr)r!)rr r#r$dists r distancer)6sSP Yr]] Yr]] #B + +D (DL,@,@44|VWGWWr"c|||}||}tt|t|}t||}||z }|||kr|ndS)a5 Calculates the optimal string alignment (OSA) similarity in the range [max, 0]. This is calculated as ``max(len1, len2) - 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 : int, optional Maximum distance between s1 and s2, that is considered as a result. If the similarity is smaller than score_cutoff, 0 is returned instead. Default is None, which deactivates this behaviour. Returns ------- similarity : int similarity between s1 and s2 Nr)maxrr))rr r#r$maximumr(sims r similarityr.fsq@ Yr]] Yr]]#b''3r77##G B  D D.C'3,+>+>33QFr" float | Nonefloatct|st|rdS|||}||}tt|t|}t||}|r||z nd}|||kr|ndS)aM Calculates a normalized optimal string alignment (OSA) similarity in the range [1, 0]. This is calculated as ``distance / max(len1, len2)``. 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 norm_dist > score_cutoff 1.0 is returned instead. Default is 1.0, which deactivates this behaviour. Returns ------- norm_dist : float normalized distance between s1 and s2 as a float between 0 and 1.0 ?Nrr)rr+rr))rr r#r$r,r( norm_dists r normalized_distancer4s>r{{gbkks Yr]] Yr]]#b''3r77##G B  D")0wqI%-l1J1J99QRRr"ct|st|rdS|||}||}t||}d|z }|||kr|ndS)aE Calculates a normalized optimal string alignment (OSA) similarity in the range [0, 1]. This is calculated as ``1 - normalized_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 norm_sim < score_cutoff 0 is returned instead. Default is 0, which deactivates this behaviour. Returns ------- norm_sim : float normalized similarity between s1 and s2 as a float between 0 and 1.0 gNr2r)rr4)rr r#r$r3norm_sims r normalized_similarityr7su>r{{gbkks Yr]] Yr]]#B++IYH$,L0H0H88qPr")rr r r r r ) rr r r r#r%r$r&r r ) rr r r r#r%r$r/r r0) __future__rtypingrrrrapidfuzz._utilsrr!r)r.r4r7r"r r<s#"""""//////////$$$$$$((((^;?# -X-X-X-X-X-Xh;?# 'G'G'G'G'G'G\;?!% )S)S)S)S)S)S`;?!% (Q(Q(Q(Q(Q(Q(Q(Qr"