o Ò·h\ã@sŽddlmZddlmZmZmZddlmZddlm Z dd d „Z d d d œddd„Z d d d œddd„Z d d d œddd„Z d d d œddd„Zd S)é)Ú annotations)ÚCallableÚHashableÚSequence)Úconv_sequences)Úis_noneÚs1úSequence[Hashable]Ús2ÚreturnÚintcCs |st|ƒSdt|ƒ>d}d}d}d}t|ƒ}dt|ƒd>}i}|j} d} |D]} | | dƒ| B|| <| dK} q)|D]R} | | dƒ} || @d>|@}| |@||A| B|B}||B}|||BB}||@}|||@dk7}|||@dk8}|d>dB}|d>}|||BB}||@}| }q;|S)Nér)ÚlenÚget)rr ÚVPÚVNÚD0ÚPM_j_oldÚcurrDistÚmaskÚblockÚ block_getÚxÚch1Úch2ÚPM_jÚTRÚHPÚHN©rúL/var/www/html/venv/lib/python3.10/site-packages/rapidfuzz/distance/OSA_py.pyÚ_osa_distance_hyrroe2003 s:   r!N)Ú processorÚ score_cutoffr"ú(Callable[..., Sequence[Hashable]] | Noner#ú int | NonecCsL|dur ||ƒ}||ƒ}t||ƒ\}}t||ƒ}|dus ||kr"|S|dS)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 )rr!)rr r"r#Údistrrr Údistance7s ( r'cCsb|dur ||ƒ}||ƒ}t||ƒ\}}tt|ƒt|ƒƒ}t||ƒ}||}|dus-||kr/|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 Nr)rÚmaxrr')rr r"r#Úmaximumr&Úsimrrr Ú similarityhs  r+ú float | NoneÚfloatcCs~t|ƒst|ƒr dS|dur||ƒ}||ƒ}t||ƒ\}}tt|ƒt|ƒƒ}t||ƒ}|r1||nd}|dus;||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 çð?Nrr )rrr(rr')rr r"r#r)r&Ú norm_distrrr Únormalized_distance“s r0cCsdt|ƒst|ƒr dS|dur||ƒ}||ƒ}t||ƒ\}}t||ƒ}d|}|dus.||kr0|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 gNr.r)rrr0)rr r"r#r/Únorm_simrrr Únormalized_similarityÀs r2)rr r r r r ) rr r r r"r$r#r%r r ) rr r r r"r$r#r,r r-)Ú __future__rÚtypingrrrÚrapidfuzz._common_pyrÚrapidfuzz._utilsrr!r'r+r0r2rrrr Ús"    /û5û/û1û