o ia;@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_selfrU/home/ubuntu/.local/lib/python3.10/site-packages/phonemizer/backend/espeak/wrapper.py__init__5s  zEspeakWrapper.__init__cCs8|jdurtjdkrtjjn tjtj d|_|jS)Nwin32c) rsysplatformctypeswindllmsvcrtcdll 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 cCr1)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)r4r rrr set_data_pathvr6zEspeakWrapper.set_data_pathcCs||jr|jSdtjvr*ttjd}|rt|tjs&td|d| St j dp5t j d}|ssz8EspeakWrapper._fetch_version_and_path...N) r infor>r?decoder is_dirrCstripsplitreplacetupler)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)rrSrrrrr's zEspeakWrapper.versioncCs|jjS)z-The espeak library as a pathlib.Path instance)r library_pathrrrrrUszEspeakWrapper.library_pathcCs|jr#t|j}|rt|jtjst|jd||_ n%dtj vrHttj d}|r;t|tjsCtd|d||_ |j durVt |drV| |j S)r9z is not a readable directoryPHONEMIZER_ESPEAK_DATA_PATHzPHONEMIZER_ESPEAK_DATA_PATH=Nr ) r7r>r?rNr<rArBrCrDr r=hasattrrS)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_rE)namer/r-) r to_ctypesr list_voicescontentsappendr<fsdecoder[rQ languagesr-)rr[voicesindexavailable_voicesr(rrrrds  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-)rHr(rrr sz+EspeakWrapper.set_voice..mbrolazinvalid voice code ""Nutf8rzfailed to load voice ") rdr/r-KeyErrorrCr set_voice_by_nameencode _get_voicer )r voice_code availabler( 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 get_current_voicer[r from_ctypes)rr(rrrrl.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)rZ0r,z,tie option only compatible with espeak>=1.49rhrZu͡rYrE)r(rCr'rpointerc_char_prkordr^valuer text_to_phonemesr_rMjoin)rrrrstext_ptr text_mode phonemes_moderesultphonemesrrrr~9s,    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. )rZ1z not compatible with espeak<=1.48NrtrrYrxrhrZzfailed to synthetize)r'rCr(rr{r fopenargtypesc_void_prestypefclosec_intr&truncater[rkmoder set_phoneme_tracer| synthetizec_size_tlenc_uintseekreadrMrO)rrrfile_pstatus phonemizedrrrrms2         zEspeakWrapper.synthetizer)F)!__name__ __module__ __qualname____doc__r2r7rpropertyr r&r)rr0 classmethodstrr5r8r rSrrGr'rUr r( functools lru_cacherdr.rlboolr~rrrrrrs@    $  #   * 4r)rr ctypes.utilrr<r>rr!r#typingrrphonemizer.backend.espeak.apirphonemizer.backend.espeak.voicerrrrrrs