| |
| |
| |
| |
|
|
| """ This code is modified from https://github.com/keithito/tacotron """ |
|
|
| import re |
|
|
|
|
| valid_symbols = [ |
| "AA", |
| "AA0", |
| "AA1", |
| "AA2", |
| "AE", |
| "AE0", |
| "AE1", |
| "AE2", |
| "AH", |
| "AH0", |
| "AH1", |
| "AH2", |
| "AO", |
| "AO0", |
| "AO1", |
| "AO2", |
| "AW", |
| "AW0", |
| "AW1", |
| "AW2", |
| "AY", |
| "AY0", |
| "AY1", |
| "AY2", |
| "B", |
| "CH", |
| "D", |
| "DH", |
| "EH", |
| "EH0", |
| "EH1", |
| "EH2", |
| "ER", |
| "ER0", |
| "ER1", |
| "ER2", |
| "EY", |
| "EY0", |
| "EY1", |
| "EY2", |
| "F", |
| "G", |
| "HH", |
| "IH", |
| "IH0", |
| "IH1", |
| "IH2", |
| "IY", |
| "IY0", |
| "IY1", |
| "IY2", |
| "JH", |
| "K", |
| "L", |
| "M", |
| "N", |
| "NG", |
| "OW", |
| "OW0", |
| "OW1", |
| "OW2", |
| "OY", |
| "OY0", |
| "OY1", |
| "OY2", |
| "P", |
| "R", |
| "S", |
| "SH", |
| "T", |
| "TH", |
| "UH", |
| "UH0", |
| "UH1", |
| "UH2", |
| "UW", |
| "UW0", |
| "UW1", |
| "UW2", |
| "V", |
| "W", |
| "Y", |
| "Z", |
| "ZH", |
| ] |
|
|
| _valid_symbol_set = set(valid_symbols) |
|
|
|
|
| class CMUDict: |
| """Thin wrapper around CMUDict data. http://www.speech.cs.cmu.edu/cgi-bin/cmudict""" |
|
|
| def __init__(self, file_or_path, keep_ambiguous=True): |
| if isinstance(file_or_path, str): |
| with open(file_or_path, encoding="latin-1") as f: |
| entries = _parse_cmudict(f) |
| else: |
| entries = _parse_cmudict(file_or_path) |
| if not keep_ambiguous: |
| entries = {word: pron for word, pron in entries.items() if len(pron) == 1} |
| self._entries = entries |
|
|
| def __len__(self): |
| return len(self._entries) |
|
|
| def lookup(self, word): |
| """Returns list of ARPAbet pronunciations of the given word.""" |
| return self._entries.get(word.upper()) |
|
|
|
|
| _alt_re = re.compile(r"\([0-9]+\)") |
|
|
|
|
| def _parse_cmudict(file): |
| cmudict = {} |
| for line in file: |
| if len(line) and (line[0] >= "A" and line[0] <= "Z" or line[0] == "'"): |
| parts = line.split(" ") |
| word = re.sub(_alt_re, "", parts[0]) |
| pronunciation = _get_pronunciation(parts[1]) |
| if pronunciation: |
| if word in cmudict: |
| cmudict[word].append(pronunciation) |
| else: |
| cmudict[word] = [pronunciation] |
| return cmudict |
|
|
|
|
| def _get_pronunciation(s): |
| parts = s.strip().split(" ") |
| for part in parts: |
| if part not in _valid_symbol_set: |
| return None |
| return " ".join(parts) |
|
|
