专栏首页python3python(实操4):录音文件的读取、

python(实操4):录音文件的读取、

#!/usr/bin/python
# encoding:utf-8
# Filename : processAudio.py
# author by :morespeech
# python2.7
# platform:visual studio code, windows
# topic: practice every day
# detial: process audio

# !/usr/bin/env python
# -*- coding:utf-8 -*-

import os
import wave
import pyaudio
import pylab as pl
import numpy as np


class cProcessAudio:
    def __init__(self):
        pass

# public function
# read file to buffer
    def readWav(self, filename, mode):
        if not os.path.exists(filename):
            return
        else:
            fileHandle = wave.open(filename, mode)
            params = fileHandle.getparams()
            nchannels, sampwidth, samplerate, nsamples = params[:4]
            # read the data
            str_data = fileHandle.readframes(nsamples)
            fileHandle.close()

            wave_data = np.fromstring(str_data, dtype=np.short)
            if params[0] == 2:
                wave_data.shape = -1, 2
                wave_data = wave_data.T
            return wave_data

# write buffer to file, only mono
    def writeWav(self, outfilename, writemode, data, fs, nchannel):
        fileHandle = wave.open(outfilename, writemode)
        fileHandle.setnchannels(nchannel)
        fileHandle.setsampwidth(2)
        fileHandle.setframerate(fs)
        fileHandle.writeframes(data.tostring())
        fileHandle.close()

# record wav
    def recordWav(self, filename, time, fs, nchannel):
        pa = pyaudio.PyAudio()
        save_buffer = ''
        buffer_size = 1000
        stream = pa.open(
            format=pyaudio.paInt16,
            channels=1,
            rate=fs,
            input=True,
            frames_per_buffer=buffer_size)

        read_time_per_second = fs / buffer_size
        cnt = 0
        while cnt < time * read_time_per_second:
            str_data = stream.read(buffer_size)
            save_buffer += str_data
            cnt += 1

        wave_data = np.fromstring(save_buffer, dtype=np.short)
        self.writeWav(filename, "wb", wave_data, fs, nchannel)
        save_buffer = ''

# play wav
    def playWav(self, filename):
        fileHandle = wave.open(filename, "rb")
        p = pyaudio.PyAudio()
        stream = p.open(
            format=p.get_format_from_width(fileHandle.getsampwidth()),
            channels=fileHandle.getnchannels(),
            rate=fileHandle.getframerate(),
            output=True)
        data = fileHandle.readframes(1024)
        while data != '':
            stream.write(data)
            data = fileHandle.readframes(1024)
        stream.close()
        p.terminate()

# plot wav
    def plotWav(self, data, fs, nchannel):
        if 2 == nchannel:
            length = len(data[0])
            time = np.arange(0, length) * (1.0 / fs)
            pl.subplot(211)
            pl.plot(time, data[0])
            pl.subplot(212)
            pl.plot(time, data[1], c="g")
            pl.xlabel("time (seconds)")
            pl.show()
        else:
            length = len(data)
            time = np.arange(0, length) * (1.0 / fs)
            pl.plot(time, data)
            pl.show()


# demo
if __name__ == "__main__":
    data = ''
    data = cProcessAudio().readWav("1.wav", "rb")
    # cProcessAudio().writeWav("new_1.wav", "wb", data, 8000, 1)
    # cProcessAudio().plotWav(data, 8000, 2)
    # cProcessAudio().playWav("1.wav")
    cProcessAudio().recordWav("new_2.wav", 5, 8000, 1)

本文参与腾讯云自媒体分享计划,欢迎正在阅读的你也加入,一起分享。

我来说两句

0 条评论
登录 后参与评论

相关文章

  • Python_冒泡排序

    从小到大的排序:(最前面的数和一步步和后面的数比较,如果大于则交换,如果不大于则继续循环)

    py3study
  • Python算法题----列表去重

    有这样一个列表[1, 1, 1, 2, 3, 3, 2, 4, 3, 4, 5, 5, 5, 6, 7, 8, 9, 9, 9, 9, 10, 10], 写一个...

    py3study
  • 用Python可视化股票指标

    但是一个量化交易可以通过回测系统建立信心然后让其一如既往的运行,以达到让钱生钱的目的,并且是自动的。

    py3study
  • 讲讲切比雪夫定理

    前面讲了大数定理,讲了中心极限定理,有读者留言让讲讲切比雪夫定理,安排。这一篇就来讲讲切比雪夫定理。

    张俊红
  • 使用 Python 实现几种常见的排序算法

    冒泡排序是最为基础的排序算法,其核心思想就是相邻元素两两比较,把较大的元素放到后面,在一轮比较完成之后,最大的元素就位于最后一个位置了,就好像是气泡,慢慢的浮出...

    周萝卜
  • pandas的一些小知识

    生信编程日常
  • R语言绘制三线表

    在进行文章的撰写时,总是少不了一些表格的绘制,当然最标准的表格就是三线表了。我一般用word绘制三线表,但是R语言也可以且比较省时间。

    用户1359560
  • Python_冒泡排序

    从小到大的排序:(最前面的数和一步步和后面的数比较,如果大于则交换,如果不大于则继续循环)

    py3study
  • 稀疏矩阵及其实现

    李志伟
  • 利用Phaser开发微信小游戏(排行榜小结)

    版权声明:本文为博主原创文章,遵循 CC 4.0 BY-SA 版权协议,转载请附上原文出处链接和本声明...

    bering

扫码关注云+社区

领取腾讯云代金券