9i\ddlmZddlmZmZmZddlmZddlm Z d dZ ddd d dZ ddd d d Z ddd dd Z ddd dd Zy)) annotations)CallableHashableSequence)conv_sequences)is_nonec|s t|Sdt|zdz }d}d}d}t|}dt|dz z}i}|j} d} |D]} | | d| z|| <| dz} |D]n} | | d} || zdz|z}| |z|z|z | z|z}||z}|||zz}||z}|||zdk7z }|||zdk7z}|dzdz}|dz}|||zz}||z}| }p|S)Nr)lenget)s1s2VPVND0PM_j_oldcurrDistmaskblock block_getxch1ch2PM_jTRHPHNs c/var/www/html/backtest/airagagent/rag_env/lib/python3.12/site-packages/rapidfuzz/distance/OSA_py.py_osa_distance_hyrroe2003r s` 2w s2w,! B B BH2wH R1 D!#E I AsA&*c  aa t|!X -r R2% - 2 "WBG*_ "W R$Y1$$R$Y1$$Ag] 1W BG*_ "W), ON) processor score_cutoffcx|||}||}t||\}}t||}|||kr|S|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 r )rr)r rr!r"dists rdistancer%7sVP r] r] B #FB #B +D (DL,@4W|VWGWWr c|||}||}t||\}}tt|t|}t||}||z }|||k\r|SdS)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 r)rmaxr r%)r rr!r"maximumr$sims r similarityr*hsm@ r] r] B #FB#b'3r7#G B D D.C'3,+>3FQFr ct|s t|ry|||}||}t||\}}tt|t|}t ||}|r||z nd}|||kr|SdS)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 ?rr )rrr'r r%)r rr!r"r(r$ norm_dists rnormalized_distancer.s>r{gbk r] r] B #FB#b'3r7#G B D")wqI%-l1J9RQRRr ct|s t|ry|||}||}t||\}}t||}d|z }|||k\r|SdS)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 gr,r)rrr.)r rr!r"r-norm_sims rnormalized_similarityr1sh>r{gbk r] r] B #FB#B+IYH$,L0H8PqPr )r Sequence[Hashable]rr2returnint) r r2rr2r!(Callable[..., Sequence[Hashable]] | Noner"z int | Noner3r4) r r2rr2r!r5r"z float | Noner3float) __future__rtypingrrrrapidfuzz._common_pyrrapidfuzz._utilsrrr%r*r.r1r rr<s##///$(^;?# .X.X.X8 .X  .X  .Xj;?# (G(G(G8 (G  (G  (G^;?!% *S*S*S8 *S  *S  *Sb;?!% )Q)Q)Q8 )Q  )Q  )Qr