o S6iS@sdZddlZddlmZmZdedeeeeffddZd edee fd d Z dd ee de deefddZ dedefddZ dS)z!Text to viseme conversion module.N)KOREAN_PHONEME_MAP VISEME_SETcharreturncCsn|rd|kr dksdSdSt|d}|d}|dd}|d}d}d}d }||||||fS) zu Decompose a Korean character into its components. Returns: (initial consonant, vowel, final consonant) 가힣)r r iiLu9ㄱㄲㄴㄷㄸㄹㅁㅂㅃㅅㅆㅇㅈㅉㅊㅋㅌㅍㅎu?ㅏㅐㅑㅒㅓㅔㅕㅖㅗㅘㅙㅚㅛㅜㅝㅞㅟㅠㅡㅢㅣuR ㄱㄲㄳㄴㄵㄶㄷㄹㄺㄻㄼㄽㄾㄿㅀㅁㅂㅄㅅㅆㅇㅈㅊㅋㅌㅍㅎ)ordstrip)rcodeZ initial_idxZ vowel_idxZ final_idxinitialsZvowelsZfinalsrQ/var/www/tkim.planitai.co.jp/blog/20251208-make-apng-tool/./src/text_to_viseme.pydecompose_korean_chars   rtextc Csg}|D]}d|krdkrQnn?t|\}}}|r.|dkr.t|d}||d|d|r?t|d}||d|d|rPt|d}||d|dq|r|}id dd d d d dddddddddddddddddddddddddd d!d d d d"d#}||d}||d$|dq|d%vr|dd$d&dq|d'vr|dd(|dq|S))z Convert Korean text to a sequence of visemes with timing hints. Returns: List of dicts with 'viseme', 'duration', 'char' keys rruㅇrestg?)visemedurationrAg333333?aeEiIoOuUbMmpfFvlLrdtnsCzW)cjwg?z  z.,!?g333333?)rrgetappendisalphalower) rresultrinitialZvowelfinalrZ char_lowerZ viseme_maprrrkorean_text_to_visemes!s       r<viseme_sequencefpscCs<g}|D]}tdt|d|}||dg|q|S)z Convert viseme sequence with durations to frame-by-frame viseme list. Args: viseme_sequence: Output from korean_text_to_visemes fps: Frames per second Returns: List of viseme codes, one per frame rrr)maxintextend)r>r?framesitemZ num_framesrrrviseme_sequence_to_framesjs rErcCs t|dS)z&Get the description for a viseme code.zUnknown viseme)rr5)rrrrget_viseme_infos rF)r=)__doc__reconfigrrstrtuplerlistdictr<rArErFrrrrsK