From c60f0517fb03aa09f72a5885477569cf2d813d65 Mon Sep 17 00:00:00 2001 From: lanvent Date: Thu, 20 Apr 2023 23:22:08 +0800 Subject: [PATCH] refactor(audio_convert.py): remove redundant functions --- voice/audio_convert.py | 42 ++---------------------------------------- 1 file changed, 2 insertions(+), 40 deletions(-) diff --git a/voice/audio_convert.py b/voice/audio_convert.py index 241a3a6..ce0601d 100644 --- a/voice/audio_convert.py +++ b/voice/audio_convert.py @@ -62,52 +62,14 @@ def any_to_sil(any_path, sil_path): ): shutil.copy2(any_path, sil_path) return 10000 - if any_path.endswith(".wav"): - return pcm_to_sil(any_path, sil_path) - if any_path.endswith(".mp3"): - return mp3_to_sil(any_path, sil_path) - raise NotImplementedError("Not support file type: {}".format(any_path)) - - -def mp3_to_wav(mp3_path, wav_path): - """ - 把mp3格式转成pcm文件 - """ - audio = AudioSegment.from_mp3(mp3_path) - audio.export(wav_path, format="wav") - - -def pcm_to_sil(pcm_path, silk_path): - """ - wav 文件转成 silk - return 声音长度,毫秒 - """ - audio = AudioSegment.from_wav(pcm_path) - rate = find_closest_sil_supports(audio.frame_rate) - # Convert to PCM_s16 - pcm_s16 = audio.set_sample_width(2) - pcm_s16 = pcm_s16.set_frame_rate(rate) - wav_data = pcm_s16.raw_data - silk_data = pysilk.encode(wav_data, data_rate=rate, sample_rate=rate) - with open(silk_path, "wb") as f: - f.write(silk_data) - return audio.duration_seconds * 1000 - - -def mp3_to_sil(mp3_path, silk_path): - """ - mp3 文件转成 silk - return 声音长度,毫秒 - """ - audio = AudioSegment.from_mp3(mp3_path) + audio = AudioSegment.from_file(any_path) rate = find_closest_sil_supports(audio.frame_rate) # Convert to PCM_s16 pcm_s16 = audio.set_sample_width(2) pcm_s16 = pcm_s16.set_frame_rate(rate) wav_data = pcm_s16.raw_data silk_data = pysilk.encode(wav_data, data_rate=rate, sample_rate=rate) - # Save the silk file - with open(silk_path, "wb") as f: + with open(sil_path, "wb") as f: f.write(silk_data) return audio.duration_seconds * 1000