如何从wav文件中获取wav样本?

时间:2010-06-17 05:26:20

标签: python wav

我想知道如何从.wav文件中获取样本,以便执行两个.wav文件的窗口连接。

任何人都可以告诉我们如何做到这一点吗?

4 个答案:

答案 0 :(得分:13)

标准库的wave模块是关键:在代码顶部import wave之后,wave.open('the.wav', 'r')会返回一个“wave read”对象,您可以从中读取使用.readframes方法的帧,它返回一个字节字符串,这些字节是样本...以波形文件具有的任何格式(您可以确定与使用{{1}将帧分解为样本相关的两个参数通道数的方法,以及每个样本的字节数的.getnchannels

将字符串转换为数值序列的最佳方法是使用.getsampwidth模块,以及(分别)array'B',{{1每个样本1,2,4个字节(在32位构建的Python上;您可以使用数组对象的'H'值来仔细检查)。如果你有不同的'L'可以提供的样本宽度,你需要切片字节字符串(适当填充每个小片的值为0的字节)并使用struct模块代替(但那是更笨,更慢,所以如果可以,请使用itemsize

答案 1 :(得分:2)

您可以使用wave模块。首先,您应该阅读元数据,例如样本大小或通道数。使用readframes()方法,您可以读取样本,但只能作为字节字符串。根据样本格式,您必须使用struct.unpack()将其转换为样本。

或者,如果您希望将样本作为浮点数的数组,则可以使用SciPy的io.wavfile模块。

答案 2 :(得分:2)

这是一个从波形文件中读取样本的功能(使用单声道和立体声测试):

def read_samples(wave_file, nb_frames):
    frame_data = wave_file.readframes(nb_frames)
    if frame_data:
        sample_width = wave_file.getsampwidth()
        nb_samples = len(frame_data) // sample_width
        format = {1:"%db", 2:"<%dh", 4:"<%dl"}[sample_width] % nb_samples
        return struct.unpack(format, frame_data)
    else:
        return ()

这里是完整的脚本,可以窗口混合或连接多个.wav文件。所有输入文件都需要具有相同的参数(通道数和样本宽度)。

import argparse
import itertools
import struct
import sys
import wave

def _struct_format(sample_width, nb_samples):
    return {1:"%db", 2:"<%dh", 4:"<%dl"}[sample_width] % nb_samples

def _mix_samples(samples):
    return sum(samples)//len(samples)

def read_samples(wave_file, nb_frames):
    frame_data = wave_file.readframes(nb_frames)
    if frame_data:
        sample_width = wave_file.getsampwidth()
        nb_samples = len(frame_data) // sample_width
        format = _struct_format(sample_width, nb_samples)
        return struct.unpack(format, frame_data)
    else:
        return ()

def write_samples(wave_file, samples, sample_width):
    format = _struct_format(sample_width, len(samples))
    frame_data = struct.pack(format, *samples)
    wave_file.writeframes(frame_data)

def compatible_input_wave_files(input_wave_files):
    nchannels, sampwidth, framerate, nframes, comptype, compname = input_wave_files[0].getparams()
    for input_wave_file in input_wave_files[1:]:
        nc,sw,fr,nf,ct,cn = input_wave_file.getparams()
        if (nc,sw,fr,ct,cn) != (nchannels, sampwidth, framerate, comptype, compname):
            return False
    return True

def mix_wave_files(output_wave_file, input_wave_files, buffer_size):
    output_wave_file.setparams(input_wave_files[0].getparams())
    sampwidth = input_wave_files[0].getsampwidth()
    max_nb_frames = max([input_wave_file.getnframes() for input_wave_file in input_wave_files])
    for frame_window in xrange(max_nb_frames // buffer_size + 1):
        all_samples = [read_samples(wave_file, buffer_size) for wave_file in input_wave_files]
        mixed_samples = [_mix_samples(samples) for samples in itertools.izip_longest(*all_samples, fillvalue=0)]
        write_samples(output_wave_file, mixed_samples, sampwidth)

def concatenate_wave_files(output_wave_file, input_wave_files, buffer_size):
    output_wave_file.setparams(input_wave_files[0].getparams())
    sampwidth = input_wave_files[0].getsampwidth()
    for input_wave_file in input_wave_files:
        nb_frames = input_wave_file.getnframes()
        for frame_window in xrange(nb_frames // buffer_size + 1):
            samples = read_samples(input_wave_file, buffer_size)
            if samples:
                write_samples(output_wave_file, samples, sampwidth)

def argument_parser():
    parser = argparse.ArgumentParser(description='Mix or concatenate multiple .wav files')
    parser.add_argument('command', choices = ("mix", "concat"), help='command')
    parser.add_argument('output_file', help='ouput .wav file')
    parser.add_argument('input_files', metavar="input_file", help='input .wav files', nargs="+")
    parser.add_argument('--buffer_size', type=int, help='nb of frames to read per iteration', default=1000)
    return parser

if __name__ == '__main__':
    args = argument_parser().parse_args()

    input_wave_files = [wave.open(name,"rb") for name in args.input_files]
    if not compatible_input_wave_files(input_wave_files):
        print "ERROR: mixed wave files must have the same params."
        sys.exit(2)

    output_wave_file = wave.open(args.output_file, "wb")
    if args.command == "mix":
        mix_wave_files(output_wave_file, input_wave_files, args.buffer_size)
    elif args.command == "concat":
        concatenate_wave_files(output_wave_file, input_wave_files, args.buffer_size)

    output_wave_file.close()
    for input_wave_file in input_wave_files:
        input_wave_file.close()

答案 3 :(得分:0)

在读取样本后(例如使用波形模块,更多细节here),您可能希望将值缩放在-1和1之间(这是音频信号的惯例)。

在这种情况下,您可以添加:

# scale to -1.0 -- 1.0
max_nb_bit = float(2**(nb_bits-1))  
samples = signal_int / (max_nb_bit + 1.0) 

nb_bits位深度和signal_int整数值。