o mia;@s~dZddlZddlZddlZddlZddlZddlZddlZddlZddl m Z m Z ddl m Z ddlmZGdddZdS)zWrapper on espeak-ng libraryN)TupleDict) EspeakAPI) EspeakVoicec@seZdZdZdZdZddZeddZeddZ d d Z d e fd d Z e defddZe defddZe ddZddZedeeeeffddZeddZeddZedd Zejdd!d/d"d#Zd$d%Zd&d'Zd0d)ed*edefd+d,Zd)efd-d.Z dS)1 EspeakWrapperaWrapper on espeak shared library The aim of this wrapper is not to be exhaustive but to encapsulate the espeak functions required for phonemization. It relies on a espeak shared library (*.so on Linux, *.dylib on Mac and *.dll on Windows) that must be installed on the system. Use the function `EspeakWrapper.set_library()` before instanciation to customize the library to use. Raises ------ RuntimeError if the espeak shared library cannot be loaded NcCs4d|_d|_d|_t||j|_d|_d|_dSN) _version _data_path_voicerlibrary data_path_espeak_libc_ _tempfile_selfr[/home/kim/smarthome/.venv/lib/python3.10/site-packages/phonemizer/backend/espeak/wrapper.py__init__5s  zEspeakWrapper.__init__cCs8|jdurtjdkrtjjn tjtj d|_|jS)Nwin32c) rsysplatformctypesZwindllmsvcrtZcdllZ LoadLibraryutil find_libraryrrrr_libcCs zEspeakWrapper._libccCs,|jdurt|_t|j|jj|jSr)rtempfileNamedTemporaryFileweakreffinalizecloserrrr _tempfileKs  zEspeakWrapper._tempfilecCs|j|j|jdS)z/For pickling, when phonemizing on multiple jobs)versionr voice)rr r rrrr __getstate__TszEspeakWrapper.__getstate__statecCsh||d|_|d|_|d|_|jr2d|jjvr)||jjdddS||jjdSdS)z1For unpickling, when phonemizing on multiple jobsr$r r%mbN)rrr r identifier set_voicelanguage)rr'rrr __setstate__[s    zEspeakWrapper.__setstate__r cC ||_dS)a`Sets the espeak backend to use `library` If this is not set, the backend uses the default espeak shared library from the system installation. Parameters ---------- library (str or None) : the path to the espeak shared library to use as backend. Set `library` to None to restore the default. N)_ESPEAK_LIBRARYclsr rrr set_libraryg zEspeakWrapper.set_libraryr cCr.)afSets the path for the data to be used by the espeak backend. If this is not set, the backend uses the default data path from the system installation. Parameters ---------- data_path : str The path to the data to be used by the espeak backend. Set `data_path` to None to restore the default. N)_ESPEAK_DATA_PATH)r1r rrr set_data_pathvr3zEspeakWrapper.set_data_pathcCs||jr|jSdtjvr*ttjd}|rt|tjs&td|d| St j dp5t j d}|ssz8EspeakWrapper._fetch_version_and_path...N) r infor9r:decoder is_dirr>stripsplitreplacetupler)rr$r rrr_fetch_version_and_paths  z%EspeakWrapper._fetch_version_and_pathreturncCs|jdur ||jS)z?The espeak version as a tuple of integers (major, minor, patch)N)rrNrrrrr$s zEspeakWrapper.versioncCs|jjS)z-The espeak library as a pathlib.Path instance)r library_pathrrrrrPszEspeakWrapper.library_pathcCs|jr#t|j}|rt|jtjst|jd||_ n%dtj vrHttj d}|r;t|tjsCtd|d||_ |j durVt |drV| |j S)r6z is not a readable directoryZPHONEMIZER_ESPEAK_DATA_PATHzPHONEMIZER_ESPEAK_DATA_PATH=Nr ) r4r9r:rIr7r<r=r>r?r r8hasattrrN)rr rrrr s     zEspeakWrapper.data_pathcCs|jS)zsThe configured voice as an EspeakVoice instance If `set_voice` has not been called, returns None )r rrrrr%szEspeakWrapper.voice)maxsizecCs|r t|d}|j|pd}d}g}||rD||j}|tt|j ddt|j ddt|j d|d7}||s|S)z>Voices available for phonemization, as a list of `EspeakVoice`)r,Nr_r@)namer,r*) rZ to_ctypesr Z list_voicescontentsappendr7fsdecoderUrL languagesr*)rrUZvoicesindexavailable_voicesr%rrrr[s  zEspeakWrapper.available_voicescCsd|vrdd|dD}ni}|D] }|j|vr"|j||j<qz||}Wnty9td|ddw|j|dd krMtd |d|}|s[td |d||_ dS) a!Setup the voice to use for phonemization Parameters ---------- voice_code (str) : Must be a valid language code that is actually supported by espeak Raises ------ RuntimeError if the required voice cannot be initialized r(cSsi|] }|jdd|jqS)r)N)r*)rCr%rrr sz+EspeakWrapper.set_voice..Zmbrolazinvalid voice code ""Nutf8rzfailed to load voice ") r[r,r*KeyErrorr>r Zset_voice_by_nameencode _get_voicer )rZ voice_code availabler%Z voice_namerrrr+s,        zEspeakWrapper.set_voicecCs|j}|jr t|SdS)znReturns the current voice used for phonemization If no voice has been set up, returns None. N)r Zget_current_voicerUrZ from_ctypes)rr%rrrra.s  zEspeakWrapper._get_voiceFtexttiecCs|jdur td|r|jdkrtdtt|d}d}|jdkr)d}n|r4dtd d >B}ntd d >d B}g}|jj dur[|j |||}|rU| | |jj dusDd |S)uTranslates a text into phonemes, must call set_voice() first. This method is used by the Espeak backend. Wrapper on the espeak_TextToPhonemes function. Parameters ---------- text (str) : the text to phonemize tie (bool, optional) : When True use a '͡' character between consecutive characters of a single phoneme. Else separate phoneme with '_'. This option requires espeak>=1.49. Default to False. Returns ------- phonemes (str) : the phonemes for the text encoded in IPA, with '_' as phonemes separator (excepted if ``tie`` is True) and ' ' as word separator. Nno voice specified)rT0r)z,tie option only compatible with espeak>=1.49r^rTu͡rSr@)r%r>r$rZpointerc_char_pr`ordrVvaluer text_to_phonemesrWrHjoin)rrcrdZtext_ptr text_modeZ phonemes_moderesultZphonemesrrrrn9s,    zEspeakWrapper.text_to_phonemescCs|jdkr td|jdurtdtjtjg|jj_tj|jj_ tjg|jj _tj |jj _ |j d|j|j j|j j}|jdtdd>B||jt|d tt|d td }|j ||dkrvtd |j d|j }|S) aTranslates a text into phonemes, must call set_voice() first. Only compatible with espeak>=1.49. This method is used by the EspeakMbrola backend. Wrapper on the espeak_Synthesize function. Parameters ---------- text (str) : the text to phonemize Returns ------- phonemes (str) : the phonemes for the text encoded in SAMPA, with '_' as phonemes separator and no word separation. )rT1z not compatible with espeak<=1.48NrerrSrir^rTzfailed to synthetize)r$r>r%rrkrZfopenargtypesZc_void_prestypeZfcloseZc_intr#truncaterUr`moder Zset_phoneme_tracerl synthetizeZc_size_tlenZc_uintseekreadrHrJ)rrcZfile_pstatusZ phonemizedrrrrxms2         zEspeakWrapper.synthetizer)F)!__name__ __module__ __qualname____doc__r/r4rpropertyrr#r&rr- classmethodstrr2r5r rNrrBr$rPr r% functools lru_cacher[r+raboolrnrxrrrrrs@    $  #   * 4r)rrZ ctypes.utilrr7r9rrr typingrrZphonemizer.backend.espeak.apirZphonemizer.backend.espeak.voicerrrrrrs