import wave #調(diào)用wave模塊
import matplotlib.pyplot as plt #調(diào)用matplotlib.pyplot模塊作為Plt
import numpy as np  #調(diào)用numpy模塊記作np
import scipy.signal as signal
import pyaudio

%matplotlib inline
plt.rcParams['font.sans-serif'] = ['SimHei']  # 用來(lái)正常顯示中文標(biāo)簽
plt.rcParams['axes.unicode_minus'] = False  # 用來(lái)正常顯示符號(hào)

f = wave.open(r"C:\Users\zyf\Desktop\Jupyter\1.wav", "rb")#讀取語(yǔ)音文件
params = f.getparams() #返回音頻參數(shù)
nchannels, sampwidth, framerate, nframes = params[:4] #賦值聲道數(shù)，量化位數(shù)，采樣頻率，采樣點(diǎn)數(shù)
print(nchannels,sampwidth,framerate,nframes)# 輸出聲道數(shù)，量化位數(shù)，采樣頻率，采樣點(diǎn)數(shù)

str_data = f.readframes(nframes) # 讀取nframes個(gè)數(shù)據(jù)，返回字符串格式
f.close()
wave_data = np.frombuffer(str_data, dtype=np.short) # 將字符串轉(zhuǎn)換為數(shù)組，得到一維的short類(lèi)型的數(shù)組
wave_data = wave_data * 1.0 / (max(abs(wave_data))) # 賦值的歸一化
time = np.arange(0, nframes) * (1.0 / framerate) # 最后通過(guò)采樣點(diǎn)數(shù)和取樣頻率計(jì)算出每個(gè)取樣的時(shí)間
# 整合左聲道和右聲道的數(shù)據(jù),如果語(yǔ)音為雙通道語(yǔ)音，具體代碼需做調(diào)整
#wave_data = np.reshape(wave_data, [nframes, nchannels])
# wave_data.shape = (-1, 2)   # -1的意思就是沒(méi)有指定,根據(jù)另一個(gè)維度的數(shù)量進(jìn)行分割
plt.figure() # 單通道語(yǔ)音波形圖
plt.plot(time, wave_data[:])
plt.xlabel("時(shí)間/s",fontsize=14)
plt.ylabel("幅度",fontsize=14)
plt.title("波形圖",fontsize=14)
plt.grid()  # 標(biāo)尺

plt.tight_layout()  # 緊密布局
plt.show()

# 音頻的播放，本實(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

# 音頻文件的寫(xiě)入、存儲(chǔ)
# 使用wave庫(kù)，相關(guān)的庫(kù)還有l(wèi)ibrosa、scipy等，讀寫(xiě)操作上的差異參閱博客： https://blog.csdn.net/weixin_38346042/article/details/119906391
import wave
import numpy as np
import scipy.signal as signal

framerate = 44100   # 采樣頻率
time = 10           # 持續(xù)時(shí)間

t = np.arange(0, time, 1.0/framerate)

# 調(diào)用scipy.signal庫(kù)中的chrip函數(shù)，
# 產(chǎn)生長(zhǎng)度為10秒、取樣頻率為44.1kHz、100Hz到1kHz的頻率掃描波
wave_data = signal.chirp(t, 100, time, 1000, method='linear') * 10000

# 由于chrip函數(shù)返回的數(shù)組為float64型，
# 需要調(diào)用數(shù)組的astype方法將其轉(zhuǎn)換為short型。
wave_data = wave_data.astype(np.short)

# 打開(kāi)WAV音頻用來(lái)寫(xiě)操作
f = wave.open(r"sweep.wav", "wb")

f.setnchannels(1)           # 配置聲道數(shù)
f.setsampwidth(2)           # 配置量化位數(shù)
f.setframerate(framerate)   # 配置取樣頻率
comptype = "NONE"
compname = "not compressed"

# 也可以用setparams一次性配置所有參數(shù)
# outwave.setparams((1, 2, framerate, nframes,comptype, compname))

# 將wav_data轉(zhuǎn)換為二進(jìn)制數(shù)據(jù)寫(xiě)入文件
f.writeframes(wave_data.tobytes())
f.close()

# 加窗分幀（接上）
# 語(yǔ)音分幀、加窗
wlen=512 # 每幀信號(hào)長(zhǎng)度
inc=128  # 幀移
signal_length=len(wave_data) #信號(hào)總長(zhǎng)度
print(signal_length)
if signal_length<=wlen: #若信號(hào)長(zhǎng)度小于一個(gè)幀的長(zhǎng)度，則幀數(shù)定義為1
        nf=1
else:                 #否則，計(jì)算幀的總長(zhǎng)度
        nf=int(np.ceil((1.0*signal_length-wlen+inc)/inc))   # nf 為幀數(shù)
# np.ceil向上取整，所以會(huì)導(dǎo)致實(shí)際分幀后的長(zhǎng)度大于信號(hào)本身的長(zhǎng)度，所以要對(duì)原來(lái)的信號(hào)進(jìn)行補(bǔ)零
pad_length=int((nf-1)*inc+wlen) #所有幀加起來(lái)總的鋪平后的長(zhǎng)度
zeros=np.zeros((pad_length-signal_length,)) #不夠的長(zhǎng)度使用0填補(bǔ)，類(lèi)似于FFT中的擴(kuò)充數(shù)組操作
pad_signal=np.concatenate((wave_data,zeros)) #填補(bǔ)后的信號(hào)記為pad_signal
indices=np.tile(np.arange(0,wlen),(nf,1))+np.tile(np.arange(0,nf*inc,inc),(wlen,1)).T  #相當(dāng)于對(duì)所有幀的時(shí)間點(diǎn)進(jìn)行抽取，得到nf*wlen長(zhǎng)度的矩陣
indices=np.array(indices,dtype=np.int32) #將indices轉(zhuǎn)化為矩陣
frames=pad_signal[indices] #得到幀信號(hào),587*512的矩陣信號(hào)
#a=frames[30:31]
#print(frames.shape)
winfunc = signal.hamming(wlen) # 調(diào)用窗函數(shù)，本初以漢明窗為例
#print(winfunc.shape) 
win=np.tile(winfunc,(nf,1)) #窗函數(shù)為一維數(shù)組（512，），因此需要按照信號(hào)幀數(shù)進(jìn)行變換得到（587*512）矩陣信號(hào)
#print(win.shape)
my = frames*win   # 這里的*指的是計(jì)算矩陣的數(shù)量積（即對(duì)位相乘）。
# python中矩陣運(yùn)算分為兩種形式，一是np.array，而是np.matrix
# ----------------------------------
#  繪制分幀加窗后的圖像（接上）
# 因?yàn)榉謳哟昂蟮男盘?hào)為587*512的矩陣信號(hào)，為了繪圖，將其轉(zhuǎn)換為一維信號(hào)
t=my.flatten()  
t=t.T
print(t.shape)
time = np.arange(0, len(t)) * (1.0 / framerate)   # 調(diào)整時(shí)間軸
plt.figure()
plt.plot(time,t,c="g")
plt.grid()
plt.show()

import numpy as np
from scipy.io import wavfile
import matplotlib.pyplot as plt

%matplotlib inline
sampling_freq, audio = wavfile.read(r"C:\Users\zyf\Desktop\Jupyter\1.wav")   # 讀取文件

audio = audio / np.max(audio)   # 歸一化，標(biāo)準(zhǔn)化

# 應(yīng)用傅里葉變換
fft_signal = np.fft.fft(audio)
print(fft_signal)

fft_signal = abs(fft_signal)
print(fft_signal)

# 建立時(shí)間軸
Freq = np.arange(0, len(fft_signal))

# 繪制語(yǔ)音信號(hào)的
plt.figure()
plt.plot(Freq, fft_signal, color='blue')
plt.xlabel('Freq (in kHz)')
plt.ylabel('Amplitude')
plt.show()

import wave
import matplotlib.pyplot as plt
import numpy as np

f = wave.open(r"C:\Users\zyf\Desktop\Jupyter\1.wav", "rb")
params = f.getparams()
nchannels, sampwidth, framerate, nframes = params[:4]
strData = f.readframes(nframes)#讀取音頻，字符串格式
waveData = np.fromstring(strData,dtype=np.int16)#將字符串轉(zhuǎn)化為int
waveData = waveData*1.0/(max(abs(waveData)))#wave幅值歸一化
waveData = np.reshape(waveData,[nframes,nchannels]).T
f.close()

plt.specgram(waveData[0],Fs = framerate, scale_by_freq = True, sides = 'default')
plt.ylabel('Frequency(Hz)')
plt.xlabel('Time(s)')
plt.colorbar()
plt.show()