pythonのwaveモジュールでの書き込みについて

Question

前提・実現したいこと

読み込んだwavファイルに何らかの処理をした後別名のwavファイルとして保存したい

発生している問題・エラーメッセージ

Traceback (most recent call last):
  File "fir.py", line 40, in <module>
    save(y, fs, 1024, "fir_test.wav")
  File "fir.py", line 19, in save
    wf.setsampwidth(bit // 8)
  File "/home/hoge/.pyenv/versions/anaconda3-5.0.1/lib/python3.6/wave.py", line 343, in setsampwidth
    raise Error('bad sample width')
wave.Error: bad sample width
Exception ignored in: <bound method Wave_write.__del__ of <wave.Wave_write object at 0x7fa577d73e80>>
Traceback (most recent call last):
  File "/home/hoge/.pyenv/versions/anaconda3-5.0.1/lib/python3.6/wave.py", line 316, in __del__
    self.close()
  File "/home/hoge/.pyenv/versions/anaconda3-5.0.1/lib/python3.6/wave.py", line 434, in close
    self._ensure_header_written(0)
  File "/home/hoge/.pyenv/versions/anaconda3-5.0.1/lib/python3.6/wave.py", line 454, in _ensure_header_written
    raise Error('sample width not specified')
wave.Error: sample width not specified

該当のソースコード

#coding:utf-8
import wave
import struct
from pylab import *

def fir(x,b):
	y = [0.0] * len(x)
	N = len(b) - 1
	for n in range(len(x)):
		for i in range(N+1):
			if n - i >= 0:
				y[n] += b[i] * x[n - i]
	return y

def save(data, fs, bit, filename):
    """波形データをWAVEファイルへ出力"""
    wf = wave.open(filename, "w")
    wf.setnchannels(1)
    wf.setsampwidth(bit // 8)
    wf.setframerate(fs)
    wf.writeframes(data)
    wf.close()

if __name__ == '__main__':
    wf = wave.open("data/VOICEACTRESS100_092.wav", "r")
    fs = wf.getnframes()

    x = wf.readframes(wf.getnframes())
    x = frombuffer(x, dtype="int16") // 32768.0 #-1 ~ 1

    # FIRフィルタをかける
    b = [1, -0.97]
    y = fir(x, b)

    # 正規化前のバイナリデータに戻す
    y = [int(v * 32767.0) for v in y]
    y = struct.pack("h" * len(y), *y)

    # 音声を保存
    save(y, fs, 1024, "fir_test.wav")

試したこと

Wave_write.setsampwidth(n)がc言語のmallocのように出力先の容量を確保していると考え、元データの695484bytes（685.5kB）が入るよう1024を指定した。

補足情報（FW/ツールのバージョンなど）

こちらのサイトを参考に、サイトで使われていたwavファイルでは実行できたものの、手元のwavファイルでは実行できなかった。
http://aidiary.hatenablog.com/entry/20111023/1319334639

Answer 1

とりあえず、

Wave_write.setsampwidth(n)がc言語のmallocのように出力先の容量を確保していると考え、元データの695484bytes（685.5kB）が入るよう1024を指定した。

違います。

wf.setsampwidth() は１サンプルあたりのバイト数を入れます。
今回（というか一般的なWave）の場合、１サンプルあたりは 16bit (= 2Byte) ですので、seve() 関数の第3引数には 16 を設定するべきでしょう。

次に、seve()の第2引数の値は wf.setframerate() に入力されますので、サンプリング周波数を入力します。この値は多分元となるWavファイルと同じ値を入力するべきなのですが、現状は

fs = wf.getnframes()

が引き渡されております。（getnframes() は全体のフレーム数です）
ですので部分は

wf.getframerate()

の戻り値に修正する必要があるかとおもいます。

Comments

[vega]

動きました！
回答してくださりありがとうございました！