用struct重建我的wave文件

时间:2014-12-11 14:03:28

标签: python audio struct

我的目标是读取波形文件并通过在-1到1范围内的每个数据位上添加一个随机数来编辑它的数据,希望产生一些失真,然后将其保存为编辑波文件。我像这样读取和编辑波形文件:

riffTag = fileIn.read(4)
if riffTag != 'RIFF':
    print 'not a valid RIFF file'
    exit(1)

riffLength = struct.unpack('<L', fileIn.read(4))[0]
riffType = fileIn.read(4)
if riffType != 'WAVE':
    print 'not a WAV file'
    exit(1)

# now read children
while fileIn.tell() < 8 + riffLength:
    tag = fileIn.read(4)
    length = struct.unpack('<L', fileIn.read(4))[0]

    if tag == 'fmt ':  # format element
        fmtData = fileIn.read(length)
        fmt, numChannels, sampleRate, byteRate, blockAlign, bitsPerSample = struct.unpack('<HHLLHH', fmtData)
        stHeaderFields['AudioFormat'] = fmt
        stHeaderFields['NumChannels'] = numChannels
        stHeaderFields['SampleRate'] = sampleRate
        stHeaderFields['ByteRate'] = byteRate
        stHeaderFields['BlockAlign'] = blockAlign
        stHeaderFields['BitsPerSample'] = bitsPerSample

    elif tag == 'data': # data element
        rawData = fileIn.read(length)

    else: # some other element, just skip it
        fileIn.seek(length, 1)

numChannels = stHeaderFields['NumChannels']

# some sanity checks
assert(stHeaderFields['BitsPerSample'] == 16)
assert(numChannels * stHeaderFields['BitsPerSample'] == blockAlign * 8)

samples = []
edited_samples = []

for offset in range(0, len(rawData), blockAlign):
    samples.append(struct.unpack('<h', rawData[offset:offset+blockAlign]))

for sample in samples:
    edited_samples.append(sample[0] + random.randint(-1, 1))

完成此操作后,我尝试通过执行以下操作将数据保存为新编辑的wave文件:

foo = []
for sample in edited_samples:
    foo.append(struct.pack('<h', int(sample)))

with open(fileIn.name + ' edited.wav', 'w') as file_out:
    file_out.write('RIFF')
    file_out.write(struct.pack('<L', riffLength))
    file_out.write('WAVE')
    file_out.write(ur'fmt\u0020')
    file_out.write(struct.pack('<H', fmt))
    file_out.write(struct.pack('<H', numChannels))
    file_out.write(struct.pack('<L', sampleRate))
    file_out.write(struct.pack('<L', byteRate))
    file_out.write(struct.pack('<H', blockAlign))
    file_out.write(struct.pack('<H', bitsPerSample))
    file_out.write('data')
    for item in foo:
        file_out.write(item)

虽然它没有给我任何错误,但我无法在媒体播放器中播放新的wave文件。当我尝试打开我的新wave文件时,我在fmt, numChannels, sampleRate, byteRate, blockAlign, bitsPerSample = struct.unpack('<HHLLHH', fmtData)行遇到了错误error: unpack requires a string argument of length 16。我想我正在构建错误的wave文件。我该如何正确构建它?

1 个答案:

答案 0 :(得分:0)

除非您出于某些其他原因(获得处理二进制文件的经验等)而打算自己编写.wav文件的支持,否则不要这样做。 Python附带了wave模块,可以处理所有文件格式问题并让您只使用数据。