o Pεi @sUdZddlZddlZddlZddlZddlZddlZddlZddl m Z m Z m Z m Z mZmZmZmZddlmZejeejefZdaejeed< ddejejejeeefdefdd ZeZd edejejefd d Z e!d krej"eej#ddddSdS)z5Functions for comparing phonemes by a distance metricN) _CONSONANTS _DATA_DIR_SCHWAS_VOWELS FEATURE_KEYS ConsonantSchwaVowel) to_vector_CLOSESTsymbolsreturncsddl}|s tttt}t|}i}|D]}|j|vrqt|||j<qt t| }t |j d}d|td<d|td<d|td<d |td <d|td <d|td <|jj|d d|dt|fddtD}|S)zfCreate mapping from each IPA symbol to a list of other IPA symbols reverse ordered by feature distancerNg? vowel_place vowel_heightg{Gz? vowel_roundedg333333?consonant_placeconsonant_voicedconsonant_sounds_like minkowski)metricpwcs.i|]\}fdd|DqS)cs g|] }|kr|qSr).0j) dist_symbolssrG/home/ubuntu/.local/lib/python3.10/site-packages/gruut_ipa/distances.py @s z-create_closest...)argsort)ridistr)rr ?sz"create_closest..)sklearn.metrics itertoolschainrrrlistipar npvstackvaluesonesshapermetricspairwise_distanceskeys enumerate)r sklearn symbol_listvectorssymbolmatrixrclosestrr#rcreate_closests.         r:r*c Cst+tdur&td}t|d }t|aWdn1s!wYWdn1s0wYtdus;Jt|S)zKGet a list of IPA symbols that are closest, ordered by increasing distance.Nzphoneme_distances.json.gzr) _CLOSEST_LOCKr rgzipopenjsonloadget)r* closest_path closest_filerrr get_closestJs   rD__main__F)indent ensure_ascii)N)$__doc__r=r'r?sys threadingtypingnumpyr+gruut_ipa.constantsrrrrrrrr gruut_ipa.featuresr MappingstrSequence _CLOSEST_TYPEr Optional__annotations__IterableUnionr:Lockr<rD__name__dumpstdoutrrrrs2(  +