waveio.py

#!/usr/bin/python
# -*- coding:utf-8 -*-
'''
Created on 2015/04/26

@author: drumichiro
'''

import wave
import struct
# TODO Implement WaveIn


class WaveOut:
    def __init__(self, name, samplingRate):
        # Setup parameters
        self.name = name
        self.samplingRate = samplingRate
        self.bits = 16
        self.channels = 1
        self.dones = 0
        self.SHORT_MAX = int(2**15)

        # Initialize by preset parameters
        self.wavo = wave.open(name, "w")
        self.wavo.setnchannels(self.channels)
        self.wavo.setsampwidth(self.bits/8)
        self.wavo.setframerate(self.samplingRate)
        self.wavo.setnframes(self.samplingRate*self.channels)

    def __del__(self):
        self.wavo.close()

    def write(self, signal):
        if not self.isOpened():
            # Full. The handle has been already closed.
            return False

        # Written length of data.
        shortSignal = signal*self.SHORT_MAX
        data = struct.pack("h"*len(shortSignal), *shortSignal)
        self.wavo.writeframes(data)
        return True