Pi"4dZddlZddlZddlZddlZddlZddlZddlZddlZddl m Z m Z ddl m Z ddlmZGddZdS)zWrapper on espeak-ng libraryN)TupleDict) EspeakAPI) EspeakVoicecfeZdZdZdZdZedZedZdZ de fdZ e d e fd Ze d Zd Zed eeeeffdZedZedZedZejdddZdZdZdde ded e fdZde fdZdS) 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 Ncd|_d|_d|_t||_d|_d|_dSN)_version _data_path_voicerlibrary_espeak_libc_ _tempfile_selfs u/home/jaya/work/projects/VOICE-AGENT/VIET/agent-env/lib/python3.11/site-packages/phonemizer/backend/espeak/wrapper.py__init__zEspeakWrapper.__init__4sF*.  !00  c|jbtjdkrtjjn;tjtj d|_|jS)Nwin32c) rsysplatformctypeswindllmsvcrtcdll LoadLibraryutil find_libraryrs r_libczEspeakWrapper._libcBsU ; (+ (?(? $$ '' (@(@(E(EFF K{rc|j ? "'9;;DO  T_do.C D D Drc,|j|j|jdS)z/For pickling, when phonemizing on multiple jobs)version data_pathvoice)r r r rs r __getstate__zEspeakWrapper.__getstate__Ss"}["" "rstatec:||d|_|d|_|d|_|jrXd|jjvr)||jjdddS||jjdSdS)z1For unpickling, when phonemizing on multiple jobsr,r-r.mbN)rr r r identifier set_voicelanguage)rr0s r __setstate__zEspeakWrapper.__setstate__Zs i(  ,Gn ; 5t{---t{5abb9:::::t{344444  5 5rrc||_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_LIBRARYclsrs r set_libraryzEspeakWrapper.set_libraryfs&rc|jr|jSdtjvr~tjtjd}|rtj|tjstd|d| Stj dptj d}|std|S)aReturns the espeak library used as backend The following precedence rule applies for library lookup: 1. As specified by BaseEspeakBackend.set_library() 2. Or as specified by the environment variable PHONEMIZER_ESPEAK_LIBRARY 3. Or the default espeak library found on the system Raises ------ RuntimeError if the espeak library cannot be found or if the environment variable PHONEMIZER_ESPEAK_LIBRARY is set to a non-readable file PHONEMIZER_ESPEAK_LIBRARYzPHONEMIZER_ESPEAK_LIBRARY=z is not a readable filez espeak-ngespeakzfailed to find espeak library) r9osenvironpathlibPathis_fileaccessR_OK RuntimeErrorresolverr!r"r:s rrzEspeakWrapper.libraryus$   '& & &"* 4 4l2:.I#JKKGOO%% /")GRW*E*E /"....///??$$ $ ((553 ((22  1/11 1rc|j\}}tj||_|jstd| dd dd}td| dD|_ dS) z9Initializes version and dapa path from the espeak libraryz(failed to retrieve espeak data directory rz-devc34K|]}t|VdSr )int).0vs r z8EspeakWrapper._fetch_version_and_path..s(AAc!ffAAAAAAr.N) rinforBrCdecoder is_dirrGstripsplitreplacetupler )rr,r-s r_fetch_version_and_pathz%EspeakWrapper._fetch_version_and_paths!\..00",y'7'7'9'9::%%'' KIJJ J..""((**0055a8@@LLAAgmmC.@.@AAAAA rreturncF|j||jS)z?The espeak version as a tuple of integers (major, minor, patch))r rYrs rr,zEspeakWrapper.versions& =  ( ( * * *}rc|jjS)z-The espeak library as a pathlib.Path instance)r library_pathrs rr]zEspeakWrapper.library_paths|((rcF|j||jS)z4The espeak data directory as a pathlib.Path instance)r rYrs rr-zEspeakWrapper.data_paths& ? "  ( ( * * *rc|jS)zsThe configured voice as an EspeakVoice instance If `set_voice` has not been called, returns None )r rs rr.zEspeakWrapper.voices {r)maxsizec |r"t|}|j|pd}d}g}||r||j}|tt j|j ddt j|j ddt j|j |dz }|||S)z>Voices available for phonemization, as a list of `EspeakVoice`)r6Nr_rJ)namer6r4) r to_ctypesr list_voicescontentsappendr@fsdecoderdrW languagesr4)rrdvoicesindexavailable_voicesr.s rrmzEspeakWrapper.available_voicess  :---7799D))$,$77Um 5M*E  # #K[,,44S#>>U_55abb9;u'788%:%:%: ; ; ; QJE Um  rcd|vr d|dD}n1i}|D]}|j|vr|j||j< ||}n"#t$rt d|ddwxYw|j|ddkrt d |d|}|st d |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 r2c8i|]}|jdd|jS)r3N)r4)rNr.s r z+EspeakWrapper.set_voice..s9>>> $e&6>>>rmbrolazinvalid voice code ""Nutf8rzfailed to load voice ") rmr6r4KeyErrorrGrset_voice_by_nameencode _get_voicer )r voice_code availabler. voice_names rr5zEspeakWrapper.set_voicesU :  >>!228<<>>>III..00 A A>22050@Ien- O":.JJ O O OCjCCCDD$ N O < ) )**;*;F*C*C D D I I666688 8!! GE EEEFF F s A A?cn|j}|jrtj|SdS)znReturns the current voice used for phonemization If no voice has been set up, returns None. N)rget_current_voicerdr from_ctypes)rr.s rrwzEspeakWrapper._get_voices7  ..00 : 2*511 1trFtexttiec:|jtd|r|jdkrtdtjtj|d}d}|jdkrd}n-|rdtd d zz}ntd d zd z}g}|jj Q|j |||}|r'| | |jj Qd |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)rc0r3z,tie option only compatible with espeak>=1.49rsrcu͡rbrJ)r.rGr,rpointerc_char_prvordrgvaluertext_to_phonemesrhrSjoin)rr~rtext_ptr text_mode phonemes_moderesultphonemess rrzEspeakWrapper.text_to_phonemes s3* : 344 4  @4<:-->@@ @>&/$++f2E2E"F"FGG <: % %,MM  1,s4yyA~=MMHHMD0M%1|44)]44H 1 hoo//000 %1 xxrc|jdkrtd|jtdtjtjg|jj_tj|jj_ tjg|jj _tj |jj _ |j d|j|j j|j j}|jdt'ddzz||jtj|d tjt-|d ztjd }|j ||dkrtd |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. )rc1z not compatible with espeak<=1.48Nrrrbrrsrczfailed to synthetize)r,rGr.rrr#fopenargtypesc_void_prestypefclosec_intr*truncaterdrvmoderset_phoneme_tracer synthetizec_size_tlenc_uintseekreadrSrU)rr~file_pstatus phonemizeds rrzEspeakWrapper.synthetize@s" <' ! !ABB B : 344 4&,_fo$F !#)?  &,o%6 "$*L ! """!! N  & & ( ( N  & & ( (** &&y3s88q='@&III(( ODKK// 0 0 OCIIM * * M$  !! &!!! Q;;566 6 A^((**113399;; rr )F)__name__ __module__ __qualname____doc__r9rpropertyr#r*r/rr7 classmethodstrr<rrYrrMr,r]r-r. functools lru_cachermr5rwboolrrrrrrs$O   XX""" 5$ 5 5 5 5 &# & & &[ &""["H B B BsC}-X ))X)X XY&&&   '& $(((T   2 2 S2 t2 2 2 2 2 h2s222222rr)rr ctypes.utilrr@rBrr%r'typingrrphonemizer.backend.espeak.apirphonemizer.backend.espeak.voicerrrrrrs#"   333333777777SSSSSSSSSSr