From cc3b3492df85d6c5cface59923b60739400e83f8 Mon Sep 17 00:00:00 2001 From: tarepan Date: Mon, 18 Dec 2023 12:24:11 +0000 Subject: [PATCH] =?UTF-8?q?Refactor:=20=E7=B0=A1=E6=98=93docstring=20?= =?UTF-8?q?=E3=81=A8=E5=8D=98=E7=B4=94=E5=A4=89=E6=95=B0=E5=90=8D?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- voicevox_engine/tts_pipeline/tts_engine.py | 121 +++------------------ 1 file changed, 17 insertions(+), 104 deletions(-) diff --git a/voicevox_engine/tts_pipeline/tts_engine.py b/voicevox_engine/tts_pipeline/tts_engine.py index bbd1d537e..a65b688b3 100644 --- a/voicevox_engine/tts_pipeline/tts_engine.py +++ b/voicevox_engine/tts_pipeline/tts_engine.py @@ -123,19 +123,7 @@ def generate_silence_mora(length: float) -> Mora: def apply_prepost_silence(moras: list[Mora], query: AudioQuery) -> list[Mora]: - """ - 前後無音（`prePhonemeLength` & `postPhonemeLength`）の適用 - Parameters - ---------- - moras : List[Mora] - モーラ時系列 - query : AudioQuery - 音声合成用のクエリ - Returns - ------- - moras : List[Mora] - 前後無音が付加されたモーラ時系列 - """ + """モーラ系列へ音声合成用のクエリがもつ前後無音（`prePhonemeLength` & `postPhonemeLength`）を付加する""" pre_silence_moras = [generate_silence_mora(query.prePhonemeLength)] post_silence_moras = [generate_silence_mora(query.postPhonemeLength)] moras = pre_silence_moras + moras + post_silence_moras @@ -143,19 +131,7 @@ def apply_prepost_silence(moras: list[Mora], query: AudioQuery) -> list[Mora]: def apply_speed_scale(moras: list[Mora], query: AudioQuery) -> list[Mora]: - """ - 話速スケール（`speedScale`）の適用 - Parameters - ---------- - moras : list[Mora] - モーラ系列 - query : AudioQuery - 音声合成用のクエリ - Returns - ------- - moras : list[Mora] - 話速スケールが適用されたモーラ系列 - """ + """モーラ系列へ音声合成用のクエリがもつ話速スケール（`speedScale`）を適用する""" for mora in moras: mora.vowel_length /= query.speedScale if mora.consonant_length: @@ -209,38 +185,14 @@ def calc_frame_per_mora(mora: Mora) -> ndarray: def apply_pitch_scale(moras: list[Mora], query: AudioQuery) -> list[Mora]: - """ - 音高スケール（`pitchScale`）の適用 - Parameters - ---------- - moras : list[Mora] - モーラ系列 - query : AudioQuery - 音声合成用のクエリ - Returns - ------- - moras : list[Mora] - 音高スケールが適用されたモーラ系列 - """ + """モーラ系列へ音声合成用のクエリがもつ音高スケール（`pitchScale`）を適用する""" for mora in moras: mora.pitch *= 2**query.pitchScale return moras def apply_intonation_scale(moras: list[Mora], query: AudioQuery) -> list[Mora]: - """ - 抑揚スケール（`intonationScale`）の適用 - Parameters - ---------- - moras : list[Mora] - モーラ系列 - query : AudioQuery - 音声合成用のクエリ - Returns - ------- - moras : list[Mora] - 抑揚スケールが適用されたモーラ系列 - """ + """モーラ系列へ音声合成用のクエリがもつ抑揚スケール（`intonationScale`）を適用する""" # 有声音素 (f0>0) の平均値に対する乖離度をスケール voiced = list(filter(lambda mora: mora.pitch > 0, moras)) mean_f0 = numpy.mean(list(map(lambda mora: mora.pitch, voiced))).item() @@ -274,19 +226,7 @@ def calc_frame_pitch(moras: list[Mora]) -> ndarray: def apply_volume_scale(wave: numpy.ndarray, query: AudioQuery) -> numpy.ndarray: - """ - 音量スケール（`volumeScale`）の適用 - Parameters - ---------- - wave : numpy.ndarray - 音声波形 - query : AudioQuery - 音声合成用のクエリ - Returns - ------- - wave : numpy.ndarray - 音量スケールが適用された音声波形 - """ + """音声波形へ音声合成用のクエリがもつ音量スケール（`volumeScale`）を適用する""" wave *= query.volumeScale return wave @@ -317,43 +257,16 @@ def calc_frame_phoneme(phonemes: List[OjtPhoneme], frame_per_phoneme: numpy.ndar def apply_output_sampling_rate( wave: ndarray, sr_wave: int, query: AudioQuery ) -> ndarray: - """ - 出力サンプリングレート（`outputSamplingRate`）の適用 - Parameters - ---------- - wave : ndarray - 音声波形 - sr_wave : int - `wave`のサンプリングレート - query : AudioQuery - 音声合成用のクエリ - Returns - ------- - wave : ndarray - 出力サンプリングレートが適用された音声波形 - """ + """音声波形へ音声合成用のクエリがもつ出力サンプリングレート（`outputSamplingRate`）を適用する""" # サンプリングレート一致のときはスルー if sr_wave == query.outputSamplingRate: return wave - wave = resample(wave, sr_wave, query.outputSamplingRate) return wave def apply_output_stereo(wave: ndarray, query: AudioQuery) -> ndarray: - """ - ステレオ出力（`outputStereo`）の適用 - Parameters - ---------- - wave : ndarray - 音声波形 - query : AudioQuery - 音声合成用のクエリ - Returns - ------- - wave : ndarray - ステレオ出力設定が適用された音声波形 - """ + """音声波形へ音声合成用のクエリがもつステレオ出力設定（`outputStereo`）を適用する""" if query.outputStereo: wave = numpy.array([wave, wave]).T return wave @@ -373,20 +286,20 @@ def query_to_decoder_feature(query: AudioQuery) -> tuple[ndarray, ndarray]: f0 : ndarray フレームごとの基本周波数、shape=(Frame,) """ - flatten_moras = to_flatten_moras(query.accent_phrases) + moras = to_flatten_moras(query.accent_phrases) - flatten_moras = apply_prepost_silence(flatten_moras, query) - flatten_moras = apply_speed_scale(flatten_moras, query) - flatten_moras = apply_pitch_scale(flatten_moras, query) - flatten_moras = apply_intonation_scale(flatten_moras, query) + moras = apply_prepost_silence(moras, query) + moras = apply_speed_scale(moras, query) + moras = apply_pitch_scale(moras, query) + moras = apply_intonation_scale(moras, query) - phoneme_data_list = to_flatten_phonemes(flatten_moras) + phonemes = to_flatten_phonemes(moras) - frame_per_phoneme = calc_frame_per_phoneme(flatten_moras) - f0 = calc_frame_pitch(flatten_moras) - phoneme = calc_frame_phoneme(phoneme_data_list, frame_per_phoneme) + frame_per_phoneme = calc_frame_per_phoneme(moras) + f0 = calc_frame_pitch(moras) + phonemes = calc_frame_phoneme(phonemes, frame_per_phoneme) - return phoneme, f0 + return phonemes, f0 def raw_wave_to_output_wave(query: AudioQuery, wave: ndarray, sr_wave: int) -> ndarray: