# -*- coding: utf-8 -*-
"""
# @Time : 2022/12/29 17:27
# @Author : WangYK
# @Site : 
# @File : check_speed.py.py
# @Software: PyCharm
# @Desc:    本文件實(shí)現(xiàn)對(duì)比三種常見(jiàn)的都區(qū)語(yǔ)音方式的速度區(qū)別

"""
import librosa
import soundfile as sf
from scipy.io import wavfile
from time import process_time
from tqdm import tqdm


# 讀取方式1
def librosa_loa(filename):
    wav_data,fs=librosa.load(filename,sr=16000)     #float64類(lèi)型
    #wav_data:[-0.03305054 -0.03561401 -0.03814697]
    return wav_data,fs

#讀取方式2
def soundfile_load(filename):
    wav_data,fs=sf.read(filename)                   #float64類(lèi)型
    #wav_data:[-0.03305054 -0.03561401 -0.03814697]
    return wav_data

#讀取方式3
def wavfile_load(filename):
    fs,wav_data=wavfile.read(filename)              #int16類(lèi)型
    #轉(zhuǎn)為float64類(lèi)型
    wav_data=wav_data/(32768)
    #wav_data:[-0.03305054 -0.03561401 -0.038114697]
    return wav_data

if __name__ == '__main__':
    filename = ['01.wav']
    filenames = filename * 10000   #重復(fù)1萬(wàn)遍,用于對(duì)比各個(gè)方式耗時(shí)
    for filename in tqdm(filenames):
        wav_data=soundfile_load(filename)           # soundfile  讀取耗時(shí)   1.7031s
        wav_data=librosa_loa(filename)              # librosa    讀取耗時(shí)   416.23s
        wav_data=wavfile_load(filename)             # wavfile    讀取耗時(shí)   1.6875s

    print('當(dāng)前程序耗時(shí)：{:.9}s'.format(process_time()))

# 音頻的播放，本實(shí)驗(yàn)使用pyaudio（代碼相對(duì)matlab較麻煩，后期簡(jiǎn)化）
import pyaudio
import wave

chunk = 1024
wf = wave.open(r"C:\Users\zyf\Desktop\Jupyter\1.wav", 'rb')
p = pyaudio.PyAudio()

# 打開(kāi)聲音輸出流
stream = p.open(format = p.get_format_from_width(wf.getsampwidth()),
? ? ? ? ? ? ? ? channels = wf.getnchannels(),
? ? ? ? ? ? ? ? rate = wf.getframerate(),
? ? ? ? ? ? ? ? output = True)

# 寫(xiě)聲音輸出流到聲卡進(jìn)行播放
while True:
? ? data = wf.readframes(chunk)
? ? if data == "":
? ? ? ? break
? ? stream.write(data)

stream.stop_stream()
stream.close()
p.terminate() ? # 關(guān)閉PyAudio