"""
python.exe -m pip install --upgrade pip
pip install pyaudio
pip install matplotlib
https://matplotlib.org/3.1.1/gallery/animation/simple_anim.html
https://matplotlib.org/stable/api/figure_api.html

"""

# encoding: utf-8
# 版权所有 2023 涂聚文有限公司
# 许可信息查看：
# 描述：
# Author    : geovindu,Geovin Du 涂聚文.
# IDE       : PyCharm 2023.1 python 311
# Datetime  : 2023/8/9 21:03
# User      : geovindu
# Product   : PyCharm
# Project   : pythonTkinterDemo
# File      : AudioHelper.py
# explain   : 学习

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


class audioHelper(object):
    """
     if __name__ == '__main__':
   # 录制5s的声音
   getAudio(5)
   # 播放声音
   play_audio('录音.wav')
   # 录制5s的音乐
   fig, line = plotInit()
   # 播放音乐
   play_audio('录音.wav')

      
    """



    def __init__(self):
        """

        """


    def getAudio(slef,sec):
        """
        录音
        :return:
        """
        # 创建对象
        p = pyaudio.PyAudio()
        # 创建流：采样位，声道数，采样频率，缓冲区大小，input=True
        stream = p.open(format=pyaudio.paInt16, channels=1, rate=16000, input=True, frames_per_buffer=1024)
        # 创建式打开音频文件
        wf = wave.open('录音.wav', 'wb')
        # 设置声道数
        wf.setnchannels(1)
        # 设置采样位数
        wf.setsampwidth(p.get_sample_size(pyaudio.paInt16))
        wf.setframerate(16000)
        print('开始录音')
        for w in range(int(16000 * sec / 1024)):
             data = stream.read(1024)
             wf.writeframes(data)
        print('录音结束')
        stream.stop_stream()
        stream.close()
        p.terminate()
        wf.close()
        # return 'test.wav'



    def play_audio(self,file):
        """
        播放音乐
        :return:
        """
        p = pyaudio.PyAudio()
        wf = wave.open(file, 'rb')
        stream = p.open(format=p.get_format_from_width(wf.getsampwidth()),
                       channels=wf.getnchannels(),
                       rate=wf.getframerate(),
                       output=True
                      )
        data = wf.readframes(1024)
        while len(data) > 0:
           stream.write(data)
           data = wf.readframes(1024)
        stream.stop_stream()
        stream.close()
        p.terminate()
        wf.close()

    def play_audio(sslf,file):
        """
        播放音乐
        :return:
        """
        fig, ax = plt.subplots()
        x = np.arange(0, 2 * np.pi, 0.01)
        line, = ax.plot(x, np.sin(x))
        p = pyaudio.PyAudio()
        wf = wave.open(file, 'rb')
        stream = p.open(format=p.get_format_from_width(wf.getsampwidth()),
                       channels=wf.getnchannels(),
                       rate=wf.getframerate(),
                       output=True
                      )
        data = wf.readframes(1024)
        while len(data) > 0:
           stream.write(data)
           data = wf.readframes(1024)
           data = np.frombuffer(data, dtype=np.int16)
           if len(data) <= 0:
               break
           print(data)
           line.set_ydata(data)
           fig.canvas.draw()
           fig.canvas.flush_events()
           plt.pause(0.01)

        stream.stop_stream()
        stream.close()
        p.terminate()
        wf.close()


    def plotInit(self):
        """
         录制5s的音乐
        :return:
        """
        CHUNK = 1024
        mpl.rcParams['toolbar'] = 'None'
        fig, ax = plt.subplots(figsize=(12, 3))
        ax.set_xlim(0, CHUNK - 1)
        ax.set_ylim(-2 ** 15, 2 ** 15)
        plt.subplots_adjust(left=0.001, top=0.999, right=0.999, bottom=0.001)
        plt.get_current_fig_manager().set_window_title('wave')
        x = np.arange(0, CHUNK)
        line, = ax.plot(x, np.random.rand(CHUNK), color='#C04851')
        plt.axis('off')
        plt.ion()
        plt.show()
        return fig, line