o ҷh@sddlmZddlmZmZmZddlmZddlm Z dd d Z d d d dddZ d d d dddZ d d d dddZ d d d dddZd S)) annotations)CallableHashableSequence)conv_sequences)is_nones1Sequence[Hashable]s2returnintcCstt|t|d}i}|j}t|d}|g|}|g|}tt|}||d<tdt|dD]} ||}}d} |d} | |d<|} tdt|dD]y} || d|| d|| dk}|| dd}|| d}t|||}|| d|| dkr| } || d|| <| } n0||| dd}| }| |dkr|| | |}t||}n| |dkr| | |}t||}|| } ||| <qO| ||| d<q3|t|S)Nr)maxlengetlistrangemin)rr maxVal last_row_idlast_row_id_getsizeFRR1Ri last_col_id last_i2l1Tjdiagleftuptempkl transposer)[/var/www/html/venv/lib/python3.10/site-packages/rapidfuzz/distance/DamerauLevenshtein_py.py"_damerau_levenshtein_distance_zhao sF     $         r+N) processor score_cutoffr,(Callable[..., Sequence[Hashable]] | Noner- int | NonecCsL|dur ||}||}t||\}}t||}|dus ||kr"|S|dS)a Calculates the Damerau-Levenshtein 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 Damerau-Levenshtein distance between two strings: >>> from rapidfuzz.distance import DamerauLevenshtein >>> DamerauLevenshtein.distance("CA", "ABC") 2 Nr )rr+)rr r,r-distr)r)r*distance:s & r1cCsb|dur ||}||}t||\}}tt|t|}t||}||}|dus-||kr/|SdS)a* Calculates the Damerau-Levenshtein 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)rrrr1)rr r,r-maximumr0simr)r)r* similarityis  r4 float | NonefloatcCs~t|st|r dS|dur||}||}t||\}}tt|t|}t||}|r1||nd}|dus;||kr=|SdS)aB Calculates a normalized Damerau-Levenshtein 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 )rrrrr1)rr r,r-r2r0 norm_distr)r)r*normalized_distances r9cCsdt|st|r dS|dur||}||}t||\}}t||}d|}|dus.||kr0|SdS)a: Calculates a normalized Damerau-Levenshtein 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 gNr7r)rrr9)rr r,r-r8norm_simr)r)r*normalized_similaritys r;)rr r r r r ) rr r r r,r.r-r/r r ) rr r r r,r.r-r5r r6) __future__rtypingrrrrapidfuzz._common_pyrrapidfuzz._utilsrr+r1r4r9r;r)r)r)r*s"    23/1