wave.py revision ab1f4674ad14eb1430489a458c7654aa1ac9c51e
1e7b146fb3bdca62a0d5ecc06dbf3348e5a4fe757Guido van Rossum"""Stuff to parse WAVE files. 2e7b146fb3bdca62a0d5ecc06dbf3348e5a4fe757Guido van Rossum 3e7b146fb3bdca62a0d5ecc06dbf3348e5a4fe757Guido van RossumUsage. 4e7b146fb3bdca62a0d5ecc06dbf3348e5a4fe757Guido van Rossum 5e7b146fb3bdca62a0d5ecc06dbf3348e5a4fe757Guido van RossumReading WAVE files: 6e7b146fb3bdca62a0d5ecc06dbf3348e5a4fe757Guido van Rossum f = wave.open(file, 'r') 7e7b146fb3bdca62a0d5ecc06dbf3348e5a4fe757Guido van Rossumwhere file is either the name of a file or an open file pointer. 8e7b146fb3bdca62a0d5ecc06dbf3348e5a4fe757Guido van RossumThe open file pointer must have methods read(), seek(), and close(). 9e7b146fb3bdca62a0d5ecc06dbf3348e5a4fe757Guido van RossumWhen the setpos() and rewind() methods are not used, the seek() 10e7b146fb3bdca62a0d5ecc06dbf3348e5a4fe757Guido van Rossummethod is not necessary. 11e7b146fb3bdca62a0d5ecc06dbf3348e5a4fe757Guido van Rossum 12e7b146fb3bdca62a0d5ecc06dbf3348e5a4fe757Guido van RossumThis returns an instance of a class with the following public methods: 13e7b146fb3bdca62a0d5ecc06dbf3348e5a4fe757Guido van Rossum getnchannels() -- returns number of audio channels (1 for 14e7b146fb3bdca62a0d5ecc06dbf3348e5a4fe757Guido van Rossum mono, 2 for stereo) 15e7b146fb3bdca62a0d5ecc06dbf3348e5a4fe757Guido van Rossum getsampwidth() -- returns sample width in bytes 16e7b146fb3bdca62a0d5ecc06dbf3348e5a4fe757Guido van Rossum getframerate() -- returns sampling frequency 17e7b146fb3bdca62a0d5ecc06dbf3348e5a4fe757Guido van Rossum getnframes() -- returns number of audio frames 18e7b146fb3bdca62a0d5ecc06dbf3348e5a4fe757Guido van Rossum getcomptype() -- returns compression type ('NONE' for linear samples) 19e7b146fb3bdca62a0d5ecc06dbf3348e5a4fe757Guido van Rossum getcompname() -- returns human-readable version of 20e7b146fb3bdca62a0d5ecc06dbf3348e5a4fe757Guido van Rossum compression type ('not compressed' linear samples) 21e7b146fb3bdca62a0d5ecc06dbf3348e5a4fe757Guido van Rossum getparams() -- returns a tuple consisting of all of the 22e7b146fb3bdca62a0d5ecc06dbf3348e5a4fe757Guido van Rossum above in the above order 23e7b146fb3bdca62a0d5ecc06dbf3348e5a4fe757Guido van Rossum getmarkers() -- returns None (for compatibility with the 24e7b146fb3bdca62a0d5ecc06dbf3348e5a4fe757Guido van Rossum aifc module) 25e7b146fb3bdca62a0d5ecc06dbf3348e5a4fe757Guido van Rossum getmark(id) -- raises an error since the mark does not 26e7b146fb3bdca62a0d5ecc06dbf3348e5a4fe757Guido van Rossum exist (for compatibility with the aifc module) 27e7b146fb3bdca62a0d5ecc06dbf3348e5a4fe757Guido van Rossum readframes(n) -- returns at most n frames of audio 28e7b146fb3bdca62a0d5ecc06dbf3348e5a4fe757Guido van Rossum rewind() -- rewind to the beginning of the audio stream 29e7b146fb3bdca62a0d5ecc06dbf3348e5a4fe757Guido van Rossum setpos(pos) -- seek to the specified position 30e7b146fb3bdca62a0d5ecc06dbf3348e5a4fe757Guido van Rossum tell() -- return the current position 31e7b146fb3bdca62a0d5ecc06dbf3348e5a4fe757Guido van Rossum close() -- close the instance (make it unusable) 32e7b146fb3bdca62a0d5ecc06dbf3348e5a4fe757Guido van RossumThe position returned by tell() and the position given to setpos() 337e47402264cf87b9bbb61fc9ff610af08add7c7bThomas Woutersare compatible and have nothing to do with the actual position in the 34e7b146fb3bdca62a0d5ecc06dbf3348e5a4fe757Guido van Rossumfile. 35e7b146fb3bdca62a0d5ecc06dbf3348e5a4fe757Guido van RossumThe close() method is called automatically when the class instance 36e7b146fb3bdca62a0d5ecc06dbf3348e5a4fe757Guido van Rossumis destroyed. 37e7b146fb3bdca62a0d5ecc06dbf3348e5a4fe757Guido van Rossum 38e7b146fb3bdca62a0d5ecc06dbf3348e5a4fe757Guido van RossumWriting WAVE files: 39e7b146fb3bdca62a0d5ecc06dbf3348e5a4fe757Guido van Rossum f = wave.open(file, 'w') 40e7b146fb3bdca62a0d5ecc06dbf3348e5a4fe757Guido van Rossumwhere file is either the name of a file or an open file pointer. 41e7b146fb3bdca62a0d5ecc06dbf3348e5a4fe757Guido van RossumThe open file pointer must have methods write(), tell(), seek(), and 42e7b146fb3bdca62a0d5ecc06dbf3348e5a4fe757Guido van Rossumclose(). 43e7b146fb3bdca62a0d5ecc06dbf3348e5a4fe757Guido van Rossum 44e7b146fb3bdca62a0d5ecc06dbf3348e5a4fe757Guido van RossumThis returns an instance of a class with the following public methods: 45e7b146fb3bdca62a0d5ecc06dbf3348e5a4fe757Guido van Rossum setnchannels(n) -- set the number of channels 46e7b146fb3bdca62a0d5ecc06dbf3348e5a4fe757Guido van Rossum setsampwidth(n) -- set the sample width 47e7b146fb3bdca62a0d5ecc06dbf3348e5a4fe757Guido van Rossum setframerate(n) -- set the frame rate 48e7b146fb3bdca62a0d5ecc06dbf3348e5a4fe757Guido van Rossum setnframes(n) -- set the number of frames 49e7b146fb3bdca62a0d5ecc06dbf3348e5a4fe757Guido van Rossum setcomptype(type, name) 50e7b146fb3bdca62a0d5ecc06dbf3348e5a4fe757Guido van Rossum -- set the compression type and the 51e7b146fb3bdca62a0d5ecc06dbf3348e5a4fe757Guido van Rossum human-readable compression type 52e7b146fb3bdca62a0d5ecc06dbf3348e5a4fe757Guido van Rossum setparams(tuple) 53e7b146fb3bdca62a0d5ecc06dbf3348e5a4fe757Guido van Rossum -- set all parameters at once 54e7b146fb3bdca62a0d5ecc06dbf3348e5a4fe757Guido van Rossum tell() -- return current position in output file 55e7b146fb3bdca62a0d5ecc06dbf3348e5a4fe757Guido van Rossum writeframesraw(data) 56e7b146fb3bdca62a0d5ecc06dbf3348e5a4fe757Guido van Rossum -- write audio frames without pathing up the 57e7b146fb3bdca62a0d5ecc06dbf3348e5a4fe757Guido van Rossum file header 58e7b146fb3bdca62a0d5ecc06dbf3348e5a4fe757Guido van Rossum writeframes(data) 59e7b146fb3bdca62a0d5ecc06dbf3348e5a4fe757Guido van Rossum -- write audio frames and patch up the file header 60e7b146fb3bdca62a0d5ecc06dbf3348e5a4fe757Guido van Rossum close() -- patch up the file header and close the 61e7b146fb3bdca62a0d5ecc06dbf3348e5a4fe757Guido van Rossum output file 62e7b146fb3bdca62a0d5ecc06dbf3348e5a4fe757Guido van RossumYou should set the parameters before the first writeframesraw or 63e7b146fb3bdca62a0d5ecc06dbf3348e5a4fe757Guido van Rossumwriteframes. The total number of frames does not need to be set, 64e7b146fb3bdca62a0d5ecc06dbf3348e5a4fe757Guido van Rossumbut when it is set to the correct value, the header does not have to 65e7b146fb3bdca62a0d5ecc06dbf3348e5a4fe757Guido van Rossumbe patched up. 66e7b146fb3bdca62a0d5ecc06dbf3348e5a4fe757Guido van RossumIt is best to first set all parameters, perhaps possibly the 67e7b146fb3bdca62a0d5ecc06dbf3348e5a4fe757Guido van Rossumcompression type, and then write audio frames using writeframesraw. 68e7b146fb3bdca62a0d5ecc06dbf3348e5a4fe757Guido van RossumWhen all frames have been written, either call writeframes('') or 69e7b146fb3bdca62a0d5ecc06dbf3348e5a4fe757Guido van Rossumclose() to patch up the sizes in the header. 70e7b146fb3bdca62a0d5ecc06dbf3348e5a4fe757Guido van RossumThe close() method is called automatically when the class instance 71e7b146fb3bdca62a0d5ecc06dbf3348e5a4fe757Guido van Rossumis destroyed. 72e7b146fb3bdca62a0d5ecc06dbf3348e5a4fe757Guido van Rossum""" 733ed23cc158f6609742229a7d912ace5533595a86Guido van Rossum 743ed23cc158f6609742229a7d912ace5533595a86Guido van Rossumimport __builtin__ 753ed23cc158f6609742229a7d912ace5533595a86Guido van Rossum 7640fc16059f04ee8fda0b5956cc4883eb21ca8f8cSkip Montanaro__all__ = ["open", "openfp", "Error"] 7740fc16059f04ee8fda0b5956cc4883eb21ca8f8cSkip Montanaro 789b8d801c37fa29420848ebc1b50c601893b36287Fred Drakeclass Error(Exception): 799b8d801c37fa29420848ebc1b50c601893b36287Fred Drake pass 803ed23cc158f6609742229a7d912ace5533595a86Guido van Rossum 813ed23cc158f6609742229a7d912ace5533595a86Guido van RossumWAVE_FORMAT_PCM = 0x0001 823ed23cc158f6609742229a7d912ace5533595a86Guido van Rossum 833ed23cc158f6609742229a7d912ace5533595a86Guido van Rossum_array_fmts = None, 'b', 'h', None, 'l' 843ed23cc158f6609742229a7d912ace5533595a86Guido van Rossum 85ebb9c922cb1f1b72fe929cfba223e42df7492a1cGuido van Rossum# Determine endian-ness 86ebb9c922cb1f1b72fe929cfba223e42df7492a1cGuido van Rossumimport struct 87ebb9c922cb1f1b72fe929cfba223e42df7492a1cGuido van Rossumif struct.pack("h", 1) == "\000\001": 88e7b146fb3bdca62a0d5ecc06dbf3348e5a4fe757Guido van Rossum big_endian = 1 89ebb9c922cb1f1b72fe929cfba223e42df7492a1cGuido van Rossumelse: 90e7b146fb3bdca62a0d5ecc06dbf3348e5a4fe757Guido van Rossum big_endian = 0 91ebb9c922cb1f1b72fe929cfba223e42df7492a1cGuido van Rossum 923601e88cb3a41f805216e9b9feda591f678f4014Guido van Rossumfrom chunk import Chunk 933ed23cc158f6609742229a7d912ace5533595a86Guido van Rossum 943ed23cc158f6609742229a7d912ace5533595a86Guido van Rossumclass Wave_read: 95e7b146fb3bdca62a0d5ecc06dbf3348e5a4fe757Guido van Rossum """Variables used in this class: 96e7b146fb3bdca62a0d5ecc06dbf3348e5a4fe757Guido van Rossum 97e7b146fb3bdca62a0d5ecc06dbf3348e5a4fe757Guido van Rossum These variables are available to the user though appropriate 98e7b146fb3bdca62a0d5ecc06dbf3348e5a4fe757Guido van Rossum methods of this class: 99e7b146fb3bdca62a0d5ecc06dbf3348e5a4fe757Guido van Rossum _file -- the open file with methods read(), close(), and seek() 100e7b146fb3bdca62a0d5ecc06dbf3348e5a4fe757Guido van Rossum set through the __init__() method 101e7b146fb3bdca62a0d5ecc06dbf3348e5a4fe757Guido van Rossum _nchannels -- the number of audio channels 102e7b146fb3bdca62a0d5ecc06dbf3348e5a4fe757Guido van Rossum available through the getnchannels() method 103e7b146fb3bdca62a0d5ecc06dbf3348e5a4fe757Guido van Rossum _nframes -- the number of audio frames 104e7b146fb3bdca62a0d5ecc06dbf3348e5a4fe757Guido van Rossum available through the getnframes() method 105e7b146fb3bdca62a0d5ecc06dbf3348e5a4fe757Guido van Rossum _sampwidth -- the number of bytes per audio sample 106e7b146fb3bdca62a0d5ecc06dbf3348e5a4fe757Guido van Rossum available through the getsampwidth() method 107e7b146fb3bdca62a0d5ecc06dbf3348e5a4fe757Guido van Rossum _framerate -- the sampling frequency 108e7b146fb3bdca62a0d5ecc06dbf3348e5a4fe757Guido van Rossum available through the getframerate() method 109e7b146fb3bdca62a0d5ecc06dbf3348e5a4fe757Guido van Rossum _comptype -- the AIFF-C compression type ('NONE' if AIFF) 110e7b146fb3bdca62a0d5ecc06dbf3348e5a4fe757Guido van Rossum available through the getcomptype() method 111e7b146fb3bdca62a0d5ecc06dbf3348e5a4fe757Guido van Rossum _compname -- the human-readable AIFF-C compression type 112e7b146fb3bdca62a0d5ecc06dbf3348e5a4fe757Guido van Rossum available through the getcomptype() method 113e7b146fb3bdca62a0d5ecc06dbf3348e5a4fe757Guido van Rossum _soundpos -- the position in the audio stream 114e7b146fb3bdca62a0d5ecc06dbf3348e5a4fe757Guido van Rossum available through the tell() method, set through the 115e7b146fb3bdca62a0d5ecc06dbf3348e5a4fe757Guido van Rossum setpos() method 116e7b146fb3bdca62a0d5ecc06dbf3348e5a4fe757Guido van Rossum 117e7b146fb3bdca62a0d5ecc06dbf3348e5a4fe757Guido van Rossum These variables are used internally only: 118e7b146fb3bdca62a0d5ecc06dbf3348e5a4fe757Guido van Rossum _fmt_chunk_read -- 1 iff the FMT chunk has been read 119e7b146fb3bdca62a0d5ecc06dbf3348e5a4fe757Guido van Rossum _data_seek_needed -- 1 iff positioned correctly in audio 120e7b146fb3bdca62a0d5ecc06dbf3348e5a4fe757Guido van Rossum file for readframes() 121e7b146fb3bdca62a0d5ecc06dbf3348e5a4fe757Guido van Rossum _data_chunk -- instantiation of a chunk class for the DATA chunk 122e7b146fb3bdca62a0d5ecc06dbf3348e5a4fe757Guido van Rossum _framesize -- size of one frame in the file 123e7b146fb3bdca62a0d5ecc06dbf3348e5a4fe757Guido van Rossum """ 124e7b146fb3bdca62a0d5ecc06dbf3348e5a4fe757Guido van Rossum 125e7b146fb3bdca62a0d5ecc06dbf3348e5a4fe757Guido van Rossum def initfp(self, file): 126e7b146fb3bdca62a0d5ecc06dbf3348e5a4fe757Guido van Rossum self._convert = None 127e7b146fb3bdca62a0d5ecc06dbf3348e5a4fe757Guido van Rossum self._soundpos = 0 128e7b146fb3bdca62a0d5ecc06dbf3348e5a4fe757Guido van Rossum self._file = Chunk(file, bigendian = 0) 129e7b146fb3bdca62a0d5ecc06dbf3348e5a4fe757Guido van Rossum if self._file.getname() != 'RIFF': 130e7b146fb3bdca62a0d5ecc06dbf3348e5a4fe757Guido van Rossum raise Error, 'file does not start with RIFF id' 131e7b146fb3bdca62a0d5ecc06dbf3348e5a4fe757Guido van Rossum if self._file.read(4) != 'WAVE': 132e7b146fb3bdca62a0d5ecc06dbf3348e5a4fe757Guido van Rossum raise Error, 'not a WAVE file' 133e7b146fb3bdca62a0d5ecc06dbf3348e5a4fe757Guido van Rossum self._fmt_chunk_read = 0 134e7b146fb3bdca62a0d5ecc06dbf3348e5a4fe757Guido van Rossum self._data_chunk = None 135e7b146fb3bdca62a0d5ecc06dbf3348e5a4fe757Guido van Rossum while 1: 136e7b146fb3bdca62a0d5ecc06dbf3348e5a4fe757Guido van Rossum self._data_seek_needed = 1 137e7b146fb3bdca62a0d5ecc06dbf3348e5a4fe757Guido van Rossum try: 138e7b146fb3bdca62a0d5ecc06dbf3348e5a4fe757Guido van Rossum chunk = Chunk(self._file, bigendian = 0) 139e7b146fb3bdca62a0d5ecc06dbf3348e5a4fe757Guido van Rossum except EOFError: 140e7b146fb3bdca62a0d5ecc06dbf3348e5a4fe757Guido van Rossum break 141e7b146fb3bdca62a0d5ecc06dbf3348e5a4fe757Guido van Rossum chunkname = chunk.getname() 142e7b146fb3bdca62a0d5ecc06dbf3348e5a4fe757Guido van Rossum if chunkname == 'fmt ': 143e7b146fb3bdca62a0d5ecc06dbf3348e5a4fe757Guido van Rossum self._read_fmt_chunk(chunk) 144e7b146fb3bdca62a0d5ecc06dbf3348e5a4fe757Guido van Rossum self._fmt_chunk_read = 1 145e7b146fb3bdca62a0d5ecc06dbf3348e5a4fe757Guido van Rossum elif chunkname == 'data': 146e7b146fb3bdca62a0d5ecc06dbf3348e5a4fe757Guido van Rossum if not self._fmt_chunk_read: 147e7b146fb3bdca62a0d5ecc06dbf3348e5a4fe757Guido van Rossum raise Error, 'data chunk before fmt chunk' 148e7b146fb3bdca62a0d5ecc06dbf3348e5a4fe757Guido van Rossum self._data_chunk = chunk 14954e54c6877329e105406c48490f218faff59db39Guido van Rossum self._nframes = chunk.chunksize // self._framesize 150e7b146fb3bdca62a0d5ecc06dbf3348e5a4fe757Guido van Rossum self._data_seek_needed = 0 151e7b146fb3bdca62a0d5ecc06dbf3348e5a4fe757Guido van Rossum break 152e7b146fb3bdca62a0d5ecc06dbf3348e5a4fe757Guido van Rossum chunk.skip() 153e7b146fb3bdca62a0d5ecc06dbf3348e5a4fe757Guido van Rossum if not self._fmt_chunk_read or not self._data_chunk: 154e7b146fb3bdca62a0d5ecc06dbf3348e5a4fe757Guido van Rossum raise Error, 'fmt chunk and/or data chunk missing' 155e7b146fb3bdca62a0d5ecc06dbf3348e5a4fe757Guido van Rossum 156e7b146fb3bdca62a0d5ecc06dbf3348e5a4fe757Guido van Rossum def __init__(self, f): 157cfc4178e843f5f2b0f190bacd222c0f72ab91e53Tim Peters self._i_opened_the_file = None 1580e67fd478f04d8634ea1e196e6ade2c4394984eeNeal Norwitz if isinstance(f, basestring): 159e7b146fb3bdca62a0d5ecc06dbf3348e5a4fe757Guido van Rossum f = __builtin__.open(f, 'rb') 160cfc4178e843f5f2b0f190bacd222c0f72ab91e53Tim Peters self._i_opened_the_file = f 161e7b146fb3bdca62a0d5ecc06dbf3348e5a4fe757Guido van Rossum # else, assume it is an open file object already 162ab1f4674ad14eb1430489a458c7654aa1ac9c51eGeorg Brandl try: 163ab1f4674ad14eb1430489a458c7654aa1ac9c51eGeorg Brandl self.initfp(f) 164ab1f4674ad14eb1430489a458c7654aa1ac9c51eGeorg Brandl except: 165ab1f4674ad14eb1430489a458c7654aa1ac9c51eGeorg Brandl if self._i_opened_the_file: 166ab1f4674ad14eb1430489a458c7654aa1ac9c51eGeorg Brandl f.close() 167ab1f4674ad14eb1430489a458c7654aa1ac9c51eGeorg Brandl raise 168e7b146fb3bdca62a0d5ecc06dbf3348e5a4fe757Guido van Rossum 169cfc4178e843f5f2b0f190bacd222c0f72ab91e53Tim Peters def __del__(self): 170cfc4178e843f5f2b0f190bacd222c0f72ab91e53Tim Peters self.close() 171e7b146fb3bdca62a0d5ecc06dbf3348e5a4fe757Guido van Rossum # 172e7b146fb3bdca62a0d5ecc06dbf3348e5a4fe757Guido van Rossum # User visible methods. 173e7b146fb3bdca62a0d5ecc06dbf3348e5a4fe757Guido van Rossum # 174e7b146fb3bdca62a0d5ecc06dbf3348e5a4fe757Guido van Rossum def getfp(self): 175e7b146fb3bdca62a0d5ecc06dbf3348e5a4fe757Guido van Rossum return self._file 176e7b146fb3bdca62a0d5ecc06dbf3348e5a4fe757Guido van Rossum 177e7b146fb3bdca62a0d5ecc06dbf3348e5a4fe757Guido van Rossum def rewind(self): 178e7b146fb3bdca62a0d5ecc06dbf3348e5a4fe757Guido van Rossum self._data_seek_needed = 1 179e7b146fb3bdca62a0d5ecc06dbf3348e5a4fe757Guido van Rossum self._soundpos = 0 180e7b146fb3bdca62a0d5ecc06dbf3348e5a4fe757Guido van Rossum 181e7b146fb3bdca62a0d5ecc06dbf3348e5a4fe757Guido van Rossum def close(self): 182cfc4178e843f5f2b0f190bacd222c0f72ab91e53Tim Peters if self._i_opened_the_file: 183cfc4178e843f5f2b0f190bacd222c0f72ab91e53Tim Peters self._i_opened_the_file.close() 184cfc4178e843f5f2b0f190bacd222c0f72ab91e53Tim Peters self._i_opened_the_file = None 185e7b146fb3bdca62a0d5ecc06dbf3348e5a4fe757Guido van Rossum self._file = None 186e7b146fb3bdca62a0d5ecc06dbf3348e5a4fe757Guido van Rossum 187e7b146fb3bdca62a0d5ecc06dbf3348e5a4fe757Guido van Rossum def tell(self): 188e7b146fb3bdca62a0d5ecc06dbf3348e5a4fe757Guido van Rossum return self._soundpos 189e7b146fb3bdca62a0d5ecc06dbf3348e5a4fe757Guido van Rossum 190e7b146fb3bdca62a0d5ecc06dbf3348e5a4fe757Guido van Rossum def getnchannels(self): 191e7b146fb3bdca62a0d5ecc06dbf3348e5a4fe757Guido van Rossum return self._nchannels 192e7b146fb3bdca62a0d5ecc06dbf3348e5a4fe757Guido van Rossum 193e7b146fb3bdca62a0d5ecc06dbf3348e5a4fe757Guido van Rossum def getnframes(self): 194e7b146fb3bdca62a0d5ecc06dbf3348e5a4fe757Guido van Rossum return self._nframes 195e7b146fb3bdca62a0d5ecc06dbf3348e5a4fe757Guido van Rossum 196e7b146fb3bdca62a0d5ecc06dbf3348e5a4fe757Guido van Rossum def getsampwidth(self): 197e7b146fb3bdca62a0d5ecc06dbf3348e5a4fe757Guido van Rossum return self._sampwidth 198e7b146fb3bdca62a0d5ecc06dbf3348e5a4fe757Guido van Rossum 199e7b146fb3bdca62a0d5ecc06dbf3348e5a4fe757Guido van Rossum def getframerate(self): 200e7b146fb3bdca62a0d5ecc06dbf3348e5a4fe757Guido van Rossum return self._framerate 201e7b146fb3bdca62a0d5ecc06dbf3348e5a4fe757Guido van Rossum 202e7b146fb3bdca62a0d5ecc06dbf3348e5a4fe757Guido van Rossum def getcomptype(self): 203e7b146fb3bdca62a0d5ecc06dbf3348e5a4fe757Guido van Rossum return self._comptype 204e7b146fb3bdca62a0d5ecc06dbf3348e5a4fe757Guido van Rossum 205e7b146fb3bdca62a0d5ecc06dbf3348e5a4fe757Guido van Rossum def getcompname(self): 206e7b146fb3bdca62a0d5ecc06dbf3348e5a4fe757Guido van Rossum return self._compname 207e7b146fb3bdca62a0d5ecc06dbf3348e5a4fe757Guido van Rossum 208e7b146fb3bdca62a0d5ecc06dbf3348e5a4fe757Guido van Rossum def getparams(self): 209e7b146fb3bdca62a0d5ecc06dbf3348e5a4fe757Guido van Rossum return self.getnchannels(), self.getsampwidth(), \ 210e7b146fb3bdca62a0d5ecc06dbf3348e5a4fe757Guido van Rossum self.getframerate(), self.getnframes(), \ 211e7b146fb3bdca62a0d5ecc06dbf3348e5a4fe757Guido van Rossum self.getcomptype(), self.getcompname() 212e7b146fb3bdca62a0d5ecc06dbf3348e5a4fe757Guido van Rossum 213e7b146fb3bdca62a0d5ecc06dbf3348e5a4fe757Guido van Rossum def getmarkers(self): 214e7b146fb3bdca62a0d5ecc06dbf3348e5a4fe757Guido van Rossum return None 215e7b146fb3bdca62a0d5ecc06dbf3348e5a4fe757Guido van Rossum 216e7b146fb3bdca62a0d5ecc06dbf3348e5a4fe757Guido van Rossum def getmark(self, id): 217e7b146fb3bdca62a0d5ecc06dbf3348e5a4fe757Guido van Rossum raise Error, 'no marks' 218e7b146fb3bdca62a0d5ecc06dbf3348e5a4fe757Guido van Rossum 219e7b146fb3bdca62a0d5ecc06dbf3348e5a4fe757Guido van Rossum def setpos(self, pos): 220e7b146fb3bdca62a0d5ecc06dbf3348e5a4fe757Guido van Rossum if pos < 0 or pos > self._nframes: 221e7b146fb3bdca62a0d5ecc06dbf3348e5a4fe757Guido van Rossum raise Error, 'position not in range' 222e7b146fb3bdca62a0d5ecc06dbf3348e5a4fe757Guido van Rossum self._soundpos = pos 223e7b146fb3bdca62a0d5ecc06dbf3348e5a4fe757Guido van Rossum self._data_seek_needed = 1 224e7b146fb3bdca62a0d5ecc06dbf3348e5a4fe757Guido van Rossum 225e7b146fb3bdca62a0d5ecc06dbf3348e5a4fe757Guido van Rossum def readframes(self, nframes): 226e7b146fb3bdca62a0d5ecc06dbf3348e5a4fe757Guido van Rossum if self._data_seek_needed: 227e7b146fb3bdca62a0d5ecc06dbf3348e5a4fe757Guido van Rossum self._data_chunk.seek(0, 0) 228e7b146fb3bdca62a0d5ecc06dbf3348e5a4fe757Guido van Rossum pos = self._soundpos * self._framesize 229e7b146fb3bdca62a0d5ecc06dbf3348e5a4fe757Guido van Rossum if pos: 230e7b146fb3bdca62a0d5ecc06dbf3348e5a4fe757Guido van Rossum self._data_chunk.seek(pos, 0) 231e7b146fb3bdca62a0d5ecc06dbf3348e5a4fe757Guido van Rossum self._data_seek_needed = 0 232e7b146fb3bdca62a0d5ecc06dbf3348e5a4fe757Guido van Rossum if nframes == 0: 233e7b146fb3bdca62a0d5ecc06dbf3348e5a4fe757Guido van Rossum return '' 234e7b146fb3bdca62a0d5ecc06dbf3348e5a4fe757Guido van Rossum if self._sampwidth > 1 and big_endian: 235e7b146fb3bdca62a0d5ecc06dbf3348e5a4fe757Guido van Rossum # unfortunately the fromfile() method does not take 236e7b146fb3bdca62a0d5ecc06dbf3348e5a4fe757Guido van Rossum # something that only looks like a file object, so 237e7b146fb3bdca62a0d5ecc06dbf3348e5a4fe757Guido van Rossum # we have to reach into the innards of the chunk object 238e7b146fb3bdca62a0d5ecc06dbf3348e5a4fe757Guido van Rossum import array 239e7b146fb3bdca62a0d5ecc06dbf3348e5a4fe757Guido van Rossum chunk = self._data_chunk 240e7b146fb3bdca62a0d5ecc06dbf3348e5a4fe757Guido van Rossum data = array.array(_array_fmts[self._sampwidth]) 241e7b146fb3bdca62a0d5ecc06dbf3348e5a4fe757Guido van Rossum nitems = nframes * self._nchannels 242e7b146fb3bdca62a0d5ecc06dbf3348e5a4fe757Guido van Rossum if nitems * self._sampwidth > chunk.chunksize - chunk.size_read: 243e7b146fb3bdca62a0d5ecc06dbf3348e5a4fe757Guido van Rossum nitems = (chunk.chunksize - chunk.size_read) / self._sampwidth 244e7b146fb3bdca62a0d5ecc06dbf3348e5a4fe757Guido van Rossum data.fromfile(chunk.file.file, nitems) 245e7b146fb3bdca62a0d5ecc06dbf3348e5a4fe757Guido van Rossum # "tell" data chunk how much was read 246e7b146fb3bdca62a0d5ecc06dbf3348e5a4fe757Guido van Rossum chunk.size_read = chunk.size_read + nitems * self._sampwidth 247e7b146fb3bdca62a0d5ecc06dbf3348e5a4fe757Guido van Rossum # do the same for the outermost chunk 248e7b146fb3bdca62a0d5ecc06dbf3348e5a4fe757Guido van Rossum chunk = chunk.file 249e7b146fb3bdca62a0d5ecc06dbf3348e5a4fe757Guido van Rossum chunk.size_read = chunk.size_read + nitems * self._sampwidth 250e7b146fb3bdca62a0d5ecc06dbf3348e5a4fe757Guido van Rossum data.byteswap() 251e7b146fb3bdca62a0d5ecc06dbf3348e5a4fe757Guido van Rossum data = data.tostring() 252e7b146fb3bdca62a0d5ecc06dbf3348e5a4fe757Guido van Rossum else: 253e7b146fb3bdca62a0d5ecc06dbf3348e5a4fe757Guido van Rossum data = self._data_chunk.read(nframes * self._framesize) 254e7b146fb3bdca62a0d5ecc06dbf3348e5a4fe757Guido van Rossum if self._convert and data: 255e7b146fb3bdca62a0d5ecc06dbf3348e5a4fe757Guido van Rossum data = self._convert(data) 25654e54c6877329e105406c48490f218faff59db39Guido van Rossum self._soundpos = self._soundpos + len(data) // (self._nchannels * self._sampwidth) 257e7b146fb3bdca62a0d5ecc06dbf3348e5a4fe757Guido van Rossum return data 258e7b146fb3bdca62a0d5ecc06dbf3348e5a4fe757Guido van Rossum 259e7b146fb3bdca62a0d5ecc06dbf3348e5a4fe757Guido van Rossum # 260e7b146fb3bdca62a0d5ecc06dbf3348e5a4fe757Guido van Rossum # Internal methods. 261e7b146fb3bdca62a0d5ecc06dbf3348e5a4fe757Guido van Rossum # 262e7b146fb3bdca62a0d5ecc06dbf3348e5a4fe757Guido van Rossum 263e7b146fb3bdca62a0d5ecc06dbf3348e5a4fe757Guido van Rossum def _read_fmt_chunk(self, chunk): 264e7b146fb3bdca62a0d5ecc06dbf3348e5a4fe757Guido van Rossum wFormatTag, self._nchannels, self._framerate, dwAvgBytesPerSec, wBlockAlign = struct.unpack('<hhllh', chunk.read(14)) 265e7b146fb3bdca62a0d5ecc06dbf3348e5a4fe757Guido van Rossum if wFormatTag == WAVE_FORMAT_PCM: 266e7b146fb3bdca62a0d5ecc06dbf3348e5a4fe757Guido van Rossum sampwidth = struct.unpack('<h', chunk.read(2))[0] 26754e54c6877329e105406c48490f218faff59db39Guido van Rossum self._sampwidth = (sampwidth + 7) // 8 268e7b146fb3bdca62a0d5ecc06dbf3348e5a4fe757Guido van Rossum else: 26970a6b49821a3226f55e9716f32d802d06640cb89Walter Dörwald raise Error, 'unknown format: %r' % (wFormatTag,) 270e7b146fb3bdca62a0d5ecc06dbf3348e5a4fe757Guido van Rossum self._framesize = self._nchannels * self._sampwidth 271e7b146fb3bdca62a0d5ecc06dbf3348e5a4fe757Guido van Rossum self._comptype = 'NONE' 272e7b146fb3bdca62a0d5ecc06dbf3348e5a4fe757Guido van Rossum self._compname = 'not compressed' 2733ed23cc158f6609742229a7d912ace5533595a86Guido van Rossum 2743ed23cc158f6609742229a7d912ace5533595a86Guido van Rossumclass Wave_write: 275e7b146fb3bdca62a0d5ecc06dbf3348e5a4fe757Guido van Rossum """Variables used in this class: 276e7b146fb3bdca62a0d5ecc06dbf3348e5a4fe757Guido van Rossum 277e7b146fb3bdca62a0d5ecc06dbf3348e5a4fe757Guido van Rossum These variables are user settable through appropriate methods 278e7b146fb3bdca62a0d5ecc06dbf3348e5a4fe757Guido van Rossum of this class: 279e7b146fb3bdca62a0d5ecc06dbf3348e5a4fe757Guido van Rossum _file -- the open file with methods write(), close(), tell(), seek() 280e7b146fb3bdca62a0d5ecc06dbf3348e5a4fe757Guido van Rossum set through the __init__() method 281e7b146fb3bdca62a0d5ecc06dbf3348e5a4fe757Guido van Rossum _comptype -- the AIFF-C compression type ('NONE' in AIFF) 282e7b146fb3bdca62a0d5ecc06dbf3348e5a4fe757Guido van Rossum set through the setcomptype() or setparams() method 283e7b146fb3bdca62a0d5ecc06dbf3348e5a4fe757Guido van Rossum _compname -- the human-readable AIFF-C compression type 284e7b146fb3bdca62a0d5ecc06dbf3348e5a4fe757Guido van Rossum set through the setcomptype() or setparams() method 285e7b146fb3bdca62a0d5ecc06dbf3348e5a4fe757Guido van Rossum _nchannels -- the number of audio channels 286e7b146fb3bdca62a0d5ecc06dbf3348e5a4fe757Guido van Rossum set through the setnchannels() or setparams() method 287e7b146fb3bdca62a0d5ecc06dbf3348e5a4fe757Guido van Rossum _sampwidth -- the number of bytes per audio sample 288e7b146fb3bdca62a0d5ecc06dbf3348e5a4fe757Guido van Rossum set through the setsampwidth() or setparams() method 289e7b146fb3bdca62a0d5ecc06dbf3348e5a4fe757Guido van Rossum _framerate -- the sampling frequency 290e7b146fb3bdca62a0d5ecc06dbf3348e5a4fe757Guido van Rossum set through the setframerate() or setparams() method 291e7b146fb3bdca62a0d5ecc06dbf3348e5a4fe757Guido van Rossum _nframes -- the number of audio frames written to the header 292e7b146fb3bdca62a0d5ecc06dbf3348e5a4fe757Guido van Rossum set through the setnframes() or setparams() method 293e7b146fb3bdca62a0d5ecc06dbf3348e5a4fe757Guido van Rossum 294e7b146fb3bdca62a0d5ecc06dbf3348e5a4fe757Guido van Rossum These variables are used internally only: 295e7b146fb3bdca62a0d5ecc06dbf3348e5a4fe757Guido van Rossum _datalength -- the size of the audio samples written to the header 296e7b146fb3bdca62a0d5ecc06dbf3348e5a4fe757Guido van Rossum _nframeswritten -- the number of frames actually written 297e7b146fb3bdca62a0d5ecc06dbf3348e5a4fe757Guido van Rossum _datawritten -- the size of the audio samples actually written 298e7b146fb3bdca62a0d5ecc06dbf3348e5a4fe757Guido van Rossum """ 299e7b146fb3bdca62a0d5ecc06dbf3348e5a4fe757Guido van Rossum 300e7b146fb3bdca62a0d5ecc06dbf3348e5a4fe757Guido van Rossum def __init__(self, f): 301cfc4178e843f5f2b0f190bacd222c0f72ab91e53Tim Peters self._i_opened_the_file = None 3020e67fd478f04d8634ea1e196e6ade2c4394984eeNeal Norwitz if isinstance(f, basestring): 303e7b146fb3bdca62a0d5ecc06dbf3348e5a4fe757Guido van Rossum f = __builtin__.open(f, 'wb') 304cfc4178e843f5f2b0f190bacd222c0f72ab91e53Tim Peters self._i_opened_the_file = f 305ab1f4674ad14eb1430489a458c7654aa1ac9c51eGeorg Brandl try: 306ab1f4674ad14eb1430489a458c7654aa1ac9c51eGeorg Brandl self.initfp(f) 307ab1f4674ad14eb1430489a458c7654aa1ac9c51eGeorg Brandl except: 308ab1f4674ad14eb1430489a458c7654aa1ac9c51eGeorg Brandl if self._i_opened_the_file: 309ab1f4674ad14eb1430489a458c7654aa1ac9c51eGeorg Brandl f.close() 310ab1f4674ad14eb1430489a458c7654aa1ac9c51eGeorg Brandl raise 311e7b146fb3bdca62a0d5ecc06dbf3348e5a4fe757Guido van Rossum 312e7b146fb3bdca62a0d5ecc06dbf3348e5a4fe757Guido van Rossum def initfp(self, file): 313e7b146fb3bdca62a0d5ecc06dbf3348e5a4fe757Guido van Rossum self._file = file 314e7b146fb3bdca62a0d5ecc06dbf3348e5a4fe757Guido van Rossum self._convert = None 315e7b146fb3bdca62a0d5ecc06dbf3348e5a4fe757Guido van Rossum self._nchannels = 0 316e7b146fb3bdca62a0d5ecc06dbf3348e5a4fe757Guido van Rossum self._sampwidth = 0 317e7b146fb3bdca62a0d5ecc06dbf3348e5a4fe757Guido van Rossum self._framerate = 0 318e7b146fb3bdca62a0d5ecc06dbf3348e5a4fe757Guido van Rossum self._nframes = 0 319e7b146fb3bdca62a0d5ecc06dbf3348e5a4fe757Guido van Rossum self._nframeswritten = 0 320e7b146fb3bdca62a0d5ecc06dbf3348e5a4fe757Guido van Rossum self._datawritten = 0 321e7b146fb3bdca62a0d5ecc06dbf3348e5a4fe757Guido van Rossum self._datalength = 0 322e7b146fb3bdca62a0d5ecc06dbf3348e5a4fe757Guido van Rossum 323e7b146fb3bdca62a0d5ecc06dbf3348e5a4fe757Guido van Rossum def __del__(self): 324cfc4178e843f5f2b0f190bacd222c0f72ab91e53Tim Peters self.close() 325e7b146fb3bdca62a0d5ecc06dbf3348e5a4fe757Guido van Rossum 326e7b146fb3bdca62a0d5ecc06dbf3348e5a4fe757Guido van Rossum # 327e7b146fb3bdca62a0d5ecc06dbf3348e5a4fe757Guido van Rossum # User visible methods. 328e7b146fb3bdca62a0d5ecc06dbf3348e5a4fe757Guido van Rossum # 329e7b146fb3bdca62a0d5ecc06dbf3348e5a4fe757Guido van Rossum def setnchannels(self, nchannels): 330e7b146fb3bdca62a0d5ecc06dbf3348e5a4fe757Guido van Rossum if self._datawritten: 331e7b146fb3bdca62a0d5ecc06dbf3348e5a4fe757Guido van Rossum raise Error, 'cannot change parameters after starting to write' 332e7b146fb3bdca62a0d5ecc06dbf3348e5a4fe757Guido van Rossum if nchannels < 1: 333e7b146fb3bdca62a0d5ecc06dbf3348e5a4fe757Guido van Rossum raise Error, 'bad # of channels' 334e7b146fb3bdca62a0d5ecc06dbf3348e5a4fe757Guido van Rossum self._nchannels = nchannels 335e7b146fb3bdca62a0d5ecc06dbf3348e5a4fe757Guido van Rossum 336e7b146fb3bdca62a0d5ecc06dbf3348e5a4fe757Guido van Rossum def getnchannels(self): 337e7b146fb3bdca62a0d5ecc06dbf3348e5a4fe757Guido van Rossum if not self._nchannels: 338e7b146fb3bdca62a0d5ecc06dbf3348e5a4fe757Guido van Rossum raise Error, 'number of channels not set' 339e7b146fb3bdca62a0d5ecc06dbf3348e5a4fe757Guido van Rossum return self._nchannels 340e7b146fb3bdca62a0d5ecc06dbf3348e5a4fe757Guido van Rossum 341e7b146fb3bdca62a0d5ecc06dbf3348e5a4fe757Guido van Rossum def setsampwidth(self, sampwidth): 342e7b146fb3bdca62a0d5ecc06dbf3348e5a4fe757Guido van Rossum if self._datawritten: 343e7b146fb3bdca62a0d5ecc06dbf3348e5a4fe757Guido van Rossum raise Error, 'cannot change parameters after starting to write' 344e7b146fb3bdca62a0d5ecc06dbf3348e5a4fe757Guido van Rossum if sampwidth < 1 or sampwidth > 4: 345e7b146fb3bdca62a0d5ecc06dbf3348e5a4fe757Guido van Rossum raise Error, 'bad sample width' 346e7b146fb3bdca62a0d5ecc06dbf3348e5a4fe757Guido van Rossum self._sampwidth = sampwidth 347e7b146fb3bdca62a0d5ecc06dbf3348e5a4fe757Guido van Rossum 348e7b146fb3bdca62a0d5ecc06dbf3348e5a4fe757Guido van Rossum def getsampwidth(self): 349e7b146fb3bdca62a0d5ecc06dbf3348e5a4fe757Guido van Rossum if not self._sampwidth: 350e7b146fb3bdca62a0d5ecc06dbf3348e5a4fe757Guido van Rossum raise Error, 'sample width not set' 351e7b146fb3bdca62a0d5ecc06dbf3348e5a4fe757Guido van Rossum return self._sampwidth 352e7b146fb3bdca62a0d5ecc06dbf3348e5a4fe757Guido van Rossum 353e7b146fb3bdca62a0d5ecc06dbf3348e5a4fe757Guido van Rossum def setframerate(self, framerate): 354e7b146fb3bdca62a0d5ecc06dbf3348e5a4fe757Guido van Rossum if self._datawritten: 355e7b146fb3bdca62a0d5ecc06dbf3348e5a4fe757Guido van Rossum raise Error, 'cannot change parameters after starting to write' 356e7b146fb3bdca62a0d5ecc06dbf3348e5a4fe757Guido van Rossum if framerate <= 0: 357e7b146fb3bdca62a0d5ecc06dbf3348e5a4fe757Guido van Rossum raise Error, 'bad frame rate' 358e7b146fb3bdca62a0d5ecc06dbf3348e5a4fe757Guido van Rossum self._framerate = framerate 359e7b146fb3bdca62a0d5ecc06dbf3348e5a4fe757Guido van Rossum 360e7b146fb3bdca62a0d5ecc06dbf3348e5a4fe757Guido van Rossum def getframerate(self): 361e7b146fb3bdca62a0d5ecc06dbf3348e5a4fe757Guido van Rossum if not self._framerate: 362e7b146fb3bdca62a0d5ecc06dbf3348e5a4fe757Guido van Rossum raise Error, 'frame rate not set' 363e7b146fb3bdca62a0d5ecc06dbf3348e5a4fe757Guido van Rossum return self._framerate 364e7b146fb3bdca62a0d5ecc06dbf3348e5a4fe757Guido van Rossum 365e7b146fb3bdca62a0d5ecc06dbf3348e5a4fe757Guido van Rossum def setnframes(self, nframes): 366e7b146fb3bdca62a0d5ecc06dbf3348e5a4fe757Guido van Rossum if self._datawritten: 367e7b146fb3bdca62a0d5ecc06dbf3348e5a4fe757Guido van Rossum raise Error, 'cannot change parameters after starting to write' 368e7b146fb3bdca62a0d5ecc06dbf3348e5a4fe757Guido van Rossum self._nframes = nframes 369e7b146fb3bdca62a0d5ecc06dbf3348e5a4fe757Guido van Rossum 370e7b146fb3bdca62a0d5ecc06dbf3348e5a4fe757Guido van Rossum def getnframes(self): 371e7b146fb3bdca62a0d5ecc06dbf3348e5a4fe757Guido van Rossum return self._nframeswritten 372e7b146fb3bdca62a0d5ecc06dbf3348e5a4fe757Guido van Rossum 373e7b146fb3bdca62a0d5ecc06dbf3348e5a4fe757Guido van Rossum def setcomptype(self, comptype, compname): 374e7b146fb3bdca62a0d5ecc06dbf3348e5a4fe757Guido van Rossum if self._datawritten: 375e7b146fb3bdca62a0d5ecc06dbf3348e5a4fe757Guido van Rossum raise Error, 'cannot change parameters after starting to write' 376e7b146fb3bdca62a0d5ecc06dbf3348e5a4fe757Guido van Rossum if comptype not in ('NONE',): 377e7b146fb3bdca62a0d5ecc06dbf3348e5a4fe757Guido van Rossum raise Error, 'unsupported compression type' 378e7b146fb3bdca62a0d5ecc06dbf3348e5a4fe757Guido van Rossum self._comptype = comptype 379e7b146fb3bdca62a0d5ecc06dbf3348e5a4fe757Guido van Rossum self._compname = compname 380e7b146fb3bdca62a0d5ecc06dbf3348e5a4fe757Guido van Rossum 381e7b146fb3bdca62a0d5ecc06dbf3348e5a4fe757Guido van Rossum def getcomptype(self): 382e7b146fb3bdca62a0d5ecc06dbf3348e5a4fe757Guido van Rossum return self._comptype 383e7b146fb3bdca62a0d5ecc06dbf3348e5a4fe757Guido van Rossum 384e7b146fb3bdca62a0d5ecc06dbf3348e5a4fe757Guido van Rossum def getcompname(self): 385e7b146fb3bdca62a0d5ecc06dbf3348e5a4fe757Guido van Rossum return self._compname 386e7b146fb3bdca62a0d5ecc06dbf3348e5a4fe757Guido van Rossum 387e7b146fb3bdca62a0d5ecc06dbf3348e5a4fe757Guido van Rossum def setparams(self, (nchannels, sampwidth, framerate, nframes, comptype, compname)): 388e7b146fb3bdca62a0d5ecc06dbf3348e5a4fe757Guido van Rossum if self._datawritten: 389e7b146fb3bdca62a0d5ecc06dbf3348e5a4fe757Guido van Rossum raise Error, 'cannot change parameters after starting to write' 390e7b146fb3bdca62a0d5ecc06dbf3348e5a4fe757Guido van Rossum self.setnchannels(nchannels) 391e7b146fb3bdca62a0d5ecc06dbf3348e5a4fe757Guido van Rossum self.setsampwidth(sampwidth) 392e7b146fb3bdca62a0d5ecc06dbf3348e5a4fe757Guido van Rossum self.setframerate(framerate) 393e7b146fb3bdca62a0d5ecc06dbf3348e5a4fe757Guido van Rossum self.setnframes(nframes) 394e7b146fb3bdca62a0d5ecc06dbf3348e5a4fe757Guido van Rossum self.setcomptype(comptype, compname) 395e7b146fb3bdca62a0d5ecc06dbf3348e5a4fe757Guido van Rossum 396e7b146fb3bdca62a0d5ecc06dbf3348e5a4fe757Guido van Rossum def getparams(self): 397e7b146fb3bdca62a0d5ecc06dbf3348e5a4fe757Guido van Rossum if not self._nchannels or not self._sampwidth or not self._framerate: 398e7b146fb3bdca62a0d5ecc06dbf3348e5a4fe757Guido van Rossum raise Error, 'not all parameters set' 399e7b146fb3bdca62a0d5ecc06dbf3348e5a4fe757Guido van Rossum return self._nchannels, self._sampwidth, self._framerate, \ 400e7b146fb3bdca62a0d5ecc06dbf3348e5a4fe757Guido van Rossum self._nframes, self._comptype, self._compname 401e7b146fb3bdca62a0d5ecc06dbf3348e5a4fe757Guido van Rossum 402e7b146fb3bdca62a0d5ecc06dbf3348e5a4fe757Guido van Rossum def setmark(self, id, pos, name): 403e7b146fb3bdca62a0d5ecc06dbf3348e5a4fe757Guido van Rossum raise Error, 'setmark() not supported' 404e7b146fb3bdca62a0d5ecc06dbf3348e5a4fe757Guido van Rossum 405e7b146fb3bdca62a0d5ecc06dbf3348e5a4fe757Guido van Rossum def getmark(self, id): 406e7b146fb3bdca62a0d5ecc06dbf3348e5a4fe757Guido van Rossum raise Error, 'no marks' 407e7b146fb3bdca62a0d5ecc06dbf3348e5a4fe757Guido van Rossum 408e7b146fb3bdca62a0d5ecc06dbf3348e5a4fe757Guido van Rossum def getmarkers(self): 409e7b146fb3bdca62a0d5ecc06dbf3348e5a4fe757Guido van Rossum return None 410e119006e7dc0df0a5ff6b60764b2ce3cd9477688Tim Peters 411e7b146fb3bdca62a0d5ecc06dbf3348e5a4fe757Guido van Rossum def tell(self): 412e7b146fb3bdca62a0d5ecc06dbf3348e5a4fe757Guido van Rossum return self._nframeswritten 413e7b146fb3bdca62a0d5ecc06dbf3348e5a4fe757Guido van Rossum 414e7b146fb3bdca62a0d5ecc06dbf3348e5a4fe757Guido van Rossum def writeframesraw(self, data): 415e7b146fb3bdca62a0d5ecc06dbf3348e5a4fe757Guido van Rossum self._ensure_header_written(len(data)) 41654e54c6877329e105406c48490f218faff59db39Guido van Rossum nframes = len(data) // (self._sampwidth * self._nchannels) 417e7b146fb3bdca62a0d5ecc06dbf3348e5a4fe757Guido van Rossum if self._convert: 418e7b146fb3bdca62a0d5ecc06dbf3348e5a4fe757Guido van Rossum data = self._convert(data) 419e7b146fb3bdca62a0d5ecc06dbf3348e5a4fe757Guido van Rossum if self._sampwidth > 1 and big_endian: 420e7b146fb3bdca62a0d5ecc06dbf3348e5a4fe757Guido van Rossum import array 421e7b146fb3bdca62a0d5ecc06dbf3348e5a4fe757Guido van Rossum data = array.array(_array_fmts[self._sampwidth], data) 422e7b146fb3bdca62a0d5ecc06dbf3348e5a4fe757Guido van Rossum data.byteswap() 423e7b146fb3bdca62a0d5ecc06dbf3348e5a4fe757Guido van Rossum data.tofile(self._file) 424e7b146fb3bdca62a0d5ecc06dbf3348e5a4fe757Guido van Rossum self._datawritten = self._datawritten + len(data) * self._sampwidth 425e7b146fb3bdca62a0d5ecc06dbf3348e5a4fe757Guido van Rossum else: 426e7b146fb3bdca62a0d5ecc06dbf3348e5a4fe757Guido van Rossum self._file.write(data) 427e7b146fb3bdca62a0d5ecc06dbf3348e5a4fe757Guido van Rossum self._datawritten = self._datawritten + len(data) 428e7b146fb3bdca62a0d5ecc06dbf3348e5a4fe757Guido van Rossum self._nframeswritten = self._nframeswritten + nframes 429e7b146fb3bdca62a0d5ecc06dbf3348e5a4fe757Guido van Rossum 430e7b146fb3bdca62a0d5ecc06dbf3348e5a4fe757Guido van Rossum def writeframes(self, data): 431e7b146fb3bdca62a0d5ecc06dbf3348e5a4fe757Guido van Rossum self.writeframesraw(data) 432e7b146fb3bdca62a0d5ecc06dbf3348e5a4fe757Guido van Rossum if self._datalength != self._datawritten: 433e7b146fb3bdca62a0d5ecc06dbf3348e5a4fe757Guido van Rossum self._patchheader() 434e7b146fb3bdca62a0d5ecc06dbf3348e5a4fe757Guido van Rossum 435e7b146fb3bdca62a0d5ecc06dbf3348e5a4fe757Guido van Rossum def close(self): 436cfc4178e843f5f2b0f190bacd222c0f72ab91e53Tim Peters if self._file: 437cfc4178e843f5f2b0f190bacd222c0f72ab91e53Tim Peters self._ensure_header_written(0) 438cfc4178e843f5f2b0f190bacd222c0f72ab91e53Tim Peters if self._datalength != self._datawritten: 439cfc4178e843f5f2b0f190bacd222c0f72ab91e53Tim Peters self._patchheader() 440cfc4178e843f5f2b0f190bacd222c0f72ab91e53Tim Peters self._file.flush() 441cfc4178e843f5f2b0f190bacd222c0f72ab91e53Tim Peters self._file = None 442cfc4178e843f5f2b0f190bacd222c0f72ab91e53Tim Peters if self._i_opened_the_file: 443cfc4178e843f5f2b0f190bacd222c0f72ab91e53Tim Peters self._i_opened_the_file.close() 444cfc4178e843f5f2b0f190bacd222c0f72ab91e53Tim Peters self._i_opened_the_file = None 445e7b146fb3bdca62a0d5ecc06dbf3348e5a4fe757Guido van Rossum 446e7b146fb3bdca62a0d5ecc06dbf3348e5a4fe757Guido van Rossum # 447e7b146fb3bdca62a0d5ecc06dbf3348e5a4fe757Guido van Rossum # Internal methods. 448e7b146fb3bdca62a0d5ecc06dbf3348e5a4fe757Guido van Rossum # 449e7b146fb3bdca62a0d5ecc06dbf3348e5a4fe757Guido van Rossum 450e7b146fb3bdca62a0d5ecc06dbf3348e5a4fe757Guido van Rossum def _ensure_header_written(self, datasize): 451e7b146fb3bdca62a0d5ecc06dbf3348e5a4fe757Guido van Rossum if not self._datawritten: 452e7b146fb3bdca62a0d5ecc06dbf3348e5a4fe757Guido van Rossum if not self._nchannels: 453e7b146fb3bdca62a0d5ecc06dbf3348e5a4fe757Guido van Rossum raise Error, '# channels not specified' 454e7b146fb3bdca62a0d5ecc06dbf3348e5a4fe757Guido van Rossum if not self._sampwidth: 455e7b146fb3bdca62a0d5ecc06dbf3348e5a4fe757Guido van Rossum raise Error, 'sample width not specified' 456e7b146fb3bdca62a0d5ecc06dbf3348e5a4fe757Guido van Rossum if not self._framerate: 457e7b146fb3bdca62a0d5ecc06dbf3348e5a4fe757Guido van Rossum raise Error, 'sampling rate not specified' 458e7b146fb3bdca62a0d5ecc06dbf3348e5a4fe757Guido van Rossum self._write_header(datasize) 459e7b146fb3bdca62a0d5ecc06dbf3348e5a4fe757Guido van Rossum 460e7b146fb3bdca62a0d5ecc06dbf3348e5a4fe757Guido van Rossum def _write_header(self, initlength): 461e7b146fb3bdca62a0d5ecc06dbf3348e5a4fe757Guido van Rossum self._file.write('RIFF') 462e7b146fb3bdca62a0d5ecc06dbf3348e5a4fe757Guido van Rossum if not self._nframes: 463e7b146fb3bdca62a0d5ecc06dbf3348e5a4fe757Guido van Rossum self._nframes = initlength / (self._nchannels * self._sampwidth) 464e7b146fb3bdca62a0d5ecc06dbf3348e5a4fe757Guido van Rossum self._datalength = self._nframes * self._nchannels * self._sampwidth 465e7b146fb3bdca62a0d5ecc06dbf3348e5a4fe757Guido van Rossum self._form_length_pos = self._file.tell() 466eca576c68be8299b006251eeea0f23068520ed57Guido van Rossum self._file.write(struct.pack('<l4s4slhhllhh4s', 467e7b146fb3bdca62a0d5ecc06dbf3348e5a4fe757Guido van Rossum 36 + self._datalength, 'WAVE', 'fmt ', 16, 468e7b146fb3bdca62a0d5ecc06dbf3348e5a4fe757Guido van Rossum WAVE_FORMAT_PCM, self._nchannels, self._framerate, 469e7b146fb3bdca62a0d5ecc06dbf3348e5a4fe757Guido van Rossum self._nchannels * self._framerate * self._sampwidth, 470e7b146fb3bdca62a0d5ecc06dbf3348e5a4fe757Guido van Rossum self._nchannels * self._sampwidth, 471e7b146fb3bdca62a0d5ecc06dbf3348e5a4fe757Guido van Rossum self._sampwidth * 8, 'data')) 472e7b146fb3bdca62a0d5ecc06dbf3348e5a4fe757Guido van Rossum self._data_length_pos = self._file.tell() 473e7b146fb3bdca62a0d5ecc06dbf3348e5a4fe757Guido van Rossum self._file.write(struct.pack('<l', self._datalength)) 474e7b146fb3bdca62a0d5ecc06dbf3348e5a4fe757Guido van Rossum 475e7b146fb3bdca62a0d5ecc06dbf3348e5a4fe757Guido van Rossum def _patchheader(self): 476e7b146fb3bdca62a0d5ecc06dbf3348e5a4fe757Guido van Rossum if self._datawritten == self._datalength: 477e7b146fb3bdca62a0d5ecc06dbf3348e5a4fe757Guido van Rossum return 478e7b146fb3bdca62a0d5ecc06dbf3348e5a4fe757Guido van Rossum curpos = self._file.tell() 479e7b146fb3bdca62a0d5ecc06dbf3348e5a4fe757Guido van Rossum self._file.seek(self._form_length_pos, 0) 480e7b146fb3bdca62a0d5ecc06dbf3348e5a4fe757Guido van Rossum self._file.write(struct.pack('<l', 36 + self._datawritten)) 481e7b146fb3bdca62a0d5ecc06dbf3348e5a4fe757Guido van Rossum self._file.seek(self._data_length_pos, 0) 482e7b146fb3bdca62a0d5ecc06dbf3348e5a4fe757Guido van Rossum self._file.write(struct.pack('<l', self._datawritten)) 483e7b146fb3bdca62a0d5ecc06dbf3348e5a4fe757Guido van Rossum self._file.seek(curpos, 0) 484e7b146fb3bdca62a0d5ecc06dbf3348e5a4fe757Guido van Rossum self._datalength = self._datawritten 4853ed23cc158f6609742229a7d912ace5533595a86Guido van Rossum 486f9607821ade54ec75d8d91952b64ecf1bfde4923Fred Drakedef open(f, mode=None): 487e7b146fb3bdca62a0d5ecc06dbf3348e5a4fe757Guido van Rossum if mode is None: 488e7b146fb3bdca62a0d5ecc06dbf3348e5a4fe757Guido van Rossum if hasattr(f, 'mode'): 489e7b146fb3bdca62a0d5ecc06dbf3348e5a4fe757Guido van Rossum mode = f.mode 490e7b146fb3bdca62a0d5ecc06dbf3348e5a4fe757Guido van Rossum else: 491e7b146fb3bdca62a0d5ecc06dbf3348e5a4fe757Guido van Rossum mode = 'rb' 492e7b146fb3bdca62a0d5ecc06dbf3348e5a4fe757Guido van Rossum if mode in ('r', 'rb'): 493e7b146fb3bdca62a0d5ecc06dbf3348e5a4fe757Guido van Rossum return Wave_read(f) 494e7b146fb3bdca62a0d5ecc06dbf3348e5a4fe757Guido van Rossum elif mode in ('w', 'wb'): 495e7b146fb3bdca62a0d5ecc06dbf3348e5a4fe757Guido van Rossum return Wave_write(f) 496e7b146fb3bdca62a0d5ecc06dbf3348e5a4fe757Guido van Rossum else: 497e7b146fb3bdca62a0d5ecc06dbf3348e5a4fe757Guido van Rossum raise Error, "mode must be 'r', 'rb', 'w', or 'wb'" 4983ed23cc158f6609742229a7d912ace5533595a86Guido van Rossum 4993ed23cc158f6609742229a7d912ace5533595a86Guido van Rossumopenfp = open # B/W compatibility 500