o
    2wi                     @   sf   d Z ddlZddlmZmZmZmZmZmZ dd Z	e
dkr1ddlmZ e Zee	de dS dS )	z<
Convert English to Hangul
https://github.com/kyubyong/g2pK
    N)adjustcomposeto_choseongto_jungseongto_jongseongreconstructc                 C   s  t td| }|D ]}| }||vrq
|| d }t|}d}tt|D ]}|| }	|dkr9||d  nd}
|t|d k rI||d  nd}|t|d k rY||d  nd}d}d	}d
}d}|	dv r|
dd |v rz|dkrz|t|	7 }q(|
dd |v r|d dvr|t|	7 }q(|d dv r|t|	7 }|d7 }q(|t|	7 }q(|	dv r|t|	7 }|d |v r|d7 }q(|	dv r|t|	7 }|	dv r|d |v r|d7 }q(|	dkr|d dv r|d7 }q(|d |v r|d7 }q(|d7 }q(|	dkr|d |v r|d7 }q(|	dv r'|t|	7 }|d |v r&|	dv r"|d7 }q(|d7 }q(|	dv rF|	dv r?|d |v r?|t|	7 }q(|t|	7 }q(|	dkr|
dkrW|t|	7 }q(|d dv re|t|	7 }q(|
dv rq|t|	7 }q(|d |v r}|d7 }q(|dv r|d |vr|d 7 }q(|	d!kr|
d |v r|d"7 }|t	|	7 }|d |v r|d#7 }q(|	d$kr|d |v r|t|	7 }q(|	d d%v r|t	|	7 }q(|t|	7 }q(t
|}t|}td&d|}| ||} q
| S )'u   Convert a string such that English words inside are turned into Hangul.
    string: input string.
    cmu: cmu dict object.

    >>> convert_eng("그 사람 좀 old school이야", cmu)
    그 사람 좀 올드 스쿨이야
    z
[A-Za-z']+r       ^$   )AEAHAXEHIHIXUHAEIOUYBCDFGHJKLMNPQRSTVWXZz$BCDFGHJKLMNPQRSTVWXZPTKN	AEIOULRMNz$BCDFGHJKLMNPQRSTVWXYZu   ᅳBDG)SZFVTHDHSHZH)r   r   r   r   r   r   r   u   ᅵu   ᅲYr    )TSDZCHJH)r"   r#   )MNNGMNLz$BCDFGHJKLPQRSTVWXZu   ᆯᄅu	   ᆯ르ERu   ᄋu   ᄅRAEIOUu	   [ᄀ-ᇿ])setrefindalllowerr   rangelenr   r   r   r   r   subreplace)stringcmu	eng_wordseng_wordwordarpabetsphonemesretipp_prevp_nextp_next2short_vowelsvowels
consonantssyllable_final_or_consonants rG   I/home/ubuntu/sommelier/.venv/lib/python3.10/site-packages/g2pk/english.pyconvert_eng   s   
  















rI   __main__)cmudictu>   오늘 학교에서 밥을 먹고 집에 와서 game을 했다)__doc__r/   
g2pk.utilsr   r   r   r   r   r   rI   __name__nltk.corpusrK   dictr7   printrG   rG   rG   rH   <module>   s     	