o j@s:ddlZddlmZmZddlmZGdddejZdS)N)ListTuple) Punctuationc@seZdZdZedfddZddZeddZ e e j d d Z ee j d d Zee j d dZe e j ddZddZe j ddZdeeeeffddZdefddZd$dededefddZd%d!efd"d#ZdS)&BasePhonemizeravBase phonemizer class Phonemization follows the following steps: 1. Preprocessing: - remove empty lines - remove punctuation - keep track of punctuation marks 2. Phonemization: - convert text to phonemes 3. Postprocessing: - join phonemes - restore punctuation marks Args: language (str): Language used by the phonemizer. punctuations (List[str]): List of punctuation marks to be preserved. keep_puncs (bool): Whether to preserve punctuation marks or not. FcCs:|s td||||_||_t||_dS)Nz{} not installed on your system) is_available RuntimeErrorformatname_init_language _language _keep_puncsr _punctuator)selflanguage punctuations keep_puncsrV/home/kuhnn/.local/lib/python3.10/site-packages/TTS/tts/utils/text/phonemizers/base.py__init__"s  zBasePhonemizer.__init__cCs(||std|d|d|S)zpLanguage initialization This method may be overloaded in child classes (see Segments backend) z language "z" is not supported by the z backend)is_supported_languagerr rrrrrr .s zBasePhonemizer._init_languagecCs|jS)z9The language code configured to be used for phonemization)r )rrrrr8szBasePhonemizer.languagecCdS)zThe name of the backendNrrrrrr =zBasePhonemizer.namecCr)z9Returns True if the backend is installed, False otherwiseNrclsrrrrCrzBasePhonemizer.is_availablecCr)z;Return the backend version as a tuple (major, minor, patch)NrrrrrversionIrzBasePhonemizer.versioncCr)z@Return a dict of language codes -> name supported by the backendNrrrrrsupported_languagesOrz"BasePhonemizer.supported_languagescCs ||vS)z6Returns True if `language` is supported by the backend)rrrrrrUs z$BasePhonemizer.is_supported_languagecCr)zThe main phonemization methodNr)rtext separatorrrr _phonemizeYszBasePhonemizer._phonemizereturncCs,|}|jr |j|S|j|ggfS)zPreprocess the text before phonemization 1. remove spaces 2. remove punctuation Override this if you need a different behaviour )stripr r strip_to_restore)rrrrr_phonemize_preprocess]s z$BasePhonemizer._phonemize_preprocesscCs |jr |j||dS|dS)zgPostprocess the raw phonemized output Override this if you need a different behaviour r)r r restore)r phonemizedrrrr_phonemize_postprocessksz%BasePhonemizer._phonemize_postprocess|NrrcCsB||\}}g}|D] }|||}||q |||}|S)a&Returns the `text` phonemized for the given language Args: text (str): Text to be phonemized. separator (str): string separator used between phonemes. Default to '_'. Returns: (str): Phonemized text )r#rappendr&)rrrrrr%tprrr phonemizets    zBasePhonemizer.phonemizerlevelcCs6d|}t|d|jt|d|dS)N z| > phoneme language: z| > phoneme backend: )printrr )rr,indentrrr print_logsszBasePhonemizer.print_logs)r'N)r)__name__ __module__ __qualname____doc__r default_puncsrr propertyr staticmethodabcabstractmethodr classmethodrrrrrrrstrr#r&r+intr0rrrrrs2        r)r8typingrrTTS.tts.utils.text.punctuationrABCrrrrrs