blob: 9e64de9c441f82e966200bedf616f399b90b717a [file] [log] [blame]
Guido van Rossum4acc25b2000-02-02 15:10:15 +00001"""Stuff to parse AIFF-C and AIFF files.
2
3Unless explicitly stated otherwise, the description below is true
4both for AIFF-C files and AIFF files.
5
6An AIFF-C file has the following structure.
7
8 +-----------------+
9 | FORM |
10 +-----------------+
11 | <size> |
12 +----+------------+
13 | | AIFC |
14 | +------------+
15 | | <chunks> |
16 | | . |
17 | | . |
18 | | . |
19 +----+------------+
20
21An AIFF file has the string "AIFF" instead of "AIFC".
22
23A chunk consists of an identifier (4 bytes) followed by a size (4 bytes,
24big endian order), followed by the data. The size field does not include
25the size of the 8 byte header.
26
27The following chunk types are recognized.
28
29 FVER
30 <version number of AIFF-C defining document> (AIFF-C only).
31 MARK
32 <# of markers> (2 bytes)
33 list of markers:
34 <marker ID> (2 bytes, must be > 0)
35 <position> (4 bytes)
36 <marker name> ("pstring")
37 COMM
38 <# of channels> (2 bytes)
39 <# of sound frames> (4 bytes)
40 <size of the samples> (2 bytes)
41 <sampling frequency> (10 bytes, IEEE 80-bit extended
42 floating point)
43 in AIFF-C files only:
44 <compression type> (4 bytes)
45 <human-readable version of compression type> ("pstring")
46 SSND
47 <offset> (4 bytes, not used by this program)
48 <blocksize> (4 bytes, not used by this program)
49 <sound data>
50
51A pstring consists of 1 byte length, a string of characters, and 0 or 1
52byte pad to make the total length even.
53
54Usage.
55
56Reading AIFF files:
57 f = aifc.open(file, 'r')
58where file is either the name of a file or an open file pointer.
59The open file pointer must have methods read(), seek(), and close().
60In some types of audio files, if the setpos() method is not used,
61the seek() method is not necessary.
62
63This returns an instance of a class with the following public methods:
64 getnchannels() -- returns number of audio channels (1 for
65 mono, 2 for stereo)
66 getsampwidth() -- returns sample width in bytes
67 getframerate() -- returns sampling frequency
68 getnframes() -- returns number of audio frames
69 getcomptype() -- returns compression type ('NONE' for AIFF files)
70 getcompname() -- returns human-readable version of
71 compression type ('not compressed' for AIFF files)
R David Murray4d35e752013-07-25 16:12:01 -040072 getparams() -- returns a namedtuple consisting of all of the
Guido van Rossum4acc25b2000-02-02 15:10:15 +000073 above in the above order
74 getmarkers() -- get the list of marks in the audio file or None
75 if there are no marks
76 getmark(id) -- get mark with the specified id (raises an error
77 if the mark does not exist)
78 readframes(n) -- returns at most n frames of audio
79 rewind() -- rewind to the beginning of the audio stream
80 setpos(pos) -- seek to the specified position
81 tell() -- return the current position
82 close() -- close the instance (make it unusable)
83The position returned by tell(), the position given to setpos() and
84the position of marks are all compatible and have nothing to do with
Thomas Wouters7e474022000-07-16 12:04:32 +000085the actual position in the file.
Guido van Rossum4acc25b2000-02-02 15:10:15 +000086The close() method is called automatically when the class instance
87is destroyed.
88
89Writing AIFF files:
90 f = aifc.open(file, 'w')
91where file is either the name of a file or an open file pointer.
92The open file pointer must have methods write(), tell(), seek(), and
93close().
94
95This returns an instance of a class with the following public methods:
96 aiff() -- create an AIFF file (AIFF-C default)
97 aifc() -- create an AIFF-C file
98 setnchannels(n) -- set the number of channels
99 setsampwidth(n) -- set the sample width
100 setframerate(n) -- set the frame rate
101 setnframes(n) -- set the number of frames
102 setcomptype(type, name)
103 -- set the compression type and the
104 human-readable compression type
105 setparams(tuple)
106 -- set all parameters at once
107 setmark(id, pos, name)
108 -- add specified mark to the list of marks
109 tell() -- return current position in output file (useful
110 in combination with setmark())
111 writeframesraw(data)
112 -- write audio frames without pathing up the
113 file header
114 writeframes(data)
115 -- write audio frames and patch up the file header
116 close() -- patch up the file header and close the
117 output file
118You should set the parameters before the first writeframesraw or
119writeframes. The total number of frames does not need to be set,
120but when it is set to the correct value, the header does not have to
121be patched up.
122It is best to first set all parameters, perhaps possibly the
123compression type, and then write audio frames using writeframesraw.
124When all frames have been written, either call writeframes('') or
125close() to patch up the sizes in the header.
Ezio Melotti30b9d5d2013-08-17 15:50:46 +0300126Marks can be added anytime. If there are any marks, you must call
Guido van Rossum4acc25b2000-02-02 15:10:15 +0000127close() after all frames have been written.
128The close() method is called automatically when the class instance
129is destroyed.
130
131When a file is opened with the extension '.aiff', an AIFF file is
132written, otherwise an AIFF-C file is written. This default can be
133changed by calling aiff() or aifc() before the first writeframes or
134writeframesraw.
135"""
Sjoerd Mullendereeabe7e1993-01-22 12:53:11 +0000136
Guido van Rossum36bb1811996-12-31 05:57:34 +0000137import struct
Georg Brandl1a3284e2007-12-02 09:40:06 +0000138import builtins
Ezio Melotti48d578c2012-03-12 23:57:18 +0200139import warnings
Sjoerd Mullendereeabe7e1993-01-22 12:53:11 +0000140
Georg Brandl2095cfe2008-06-07 19:01:03 +0000141__all__ = ["Error", "open", "openfp"]
Skip Montanaroe99d5ea2001-01-20 19:54:20 +0000142
Fred Drake227b1202000-08-17 05:06:49 +0000143class Error(Exception):
144 pass
Sjoerd Mullendereeabe7e1993-01-22 12:53:11 +0000145
Guido van Rossume2a383d2007-01-15 16:59:06 +0000146_AIFC_version = 0xA2805140 # Version 1 of AIFF-C
Sjoerd Mullendereeabe7e1993-01-22 12:53:11 +0000147
Sjoerd Mullendereeabe7e1993-01-22 12:53:11 +0000148def _read_long(file):
Guido van Rossum4acc25b2000-02-02 15:10:15 +0000149 try:
150 return struct.unpack('>l', file.read(4))[0]
151 except struct.error:
152 raise EOFError
Sjoerd Mullendereeabe7e1993-01-22 12:53:11 +0000153
154def _read_ulong(file):
Guido van Rossum4acc25b2000-02-02 15:10:15 +0000155 try:
156 return struct.unpack('>L', file.read(4))[0]
157 except struct.error:
158 raise EOFError
Sjoerd Mullendereeabe7e1993-01-22 12:53:11 +0000159
160def _read_short(file):
Guido van Rossum4acc25b2000-02-02 15:10:15 +0000161 try:
162 return struct.unpack('>h', file.read(2))[0]
163 except struct.error:
164 raise EOFError
Sjoerd Mullendereeabe7e1993-01-22 12:53:11 +0000165
Antoine Pitrou03757ec2012-01-17 17:13:04 +0100166def _read_ushort(file):
167 try:
168 return struct.unpack('>H', file.read(2))[0]
169 except struct.error:
170 raise EOFError
171
Sjoerd Mullendereeabe7e1993-01-22 12:53:11 +0000172def _read_string(file):
Guido van Rossum4acc25b2000-02-02 15:10:15 +0000173 length = ord(file.read(1))
174 if length == 0:
Georg Brandl2095cfe2008-06-07 19:01:03 +0000175 data = b''
Guido van Rossum4acc25b2000-02-02 15:10:15 +0000176 else:
177 data = file.read(length)
178 if length & 1 == 0:
179 dummy = file.read(1)
180 return data
Sjoerd Mullendereeabe7e1993-01-22 12:53:11 +0000181
182_HUGE_VAL = 1.79769313486231e+308 # See <limits.h>
183
184def _read_float(f): # 10 bytes
Guido van Rossum4acc25b2000-02-02 15:10:15 +0000185 expon = _read_short(f) # 2 bytes
186 sign = 1
187 if expon < 0:
188 sign = -1
189 expon = expon + 0x8000
190 himant = _read_ulong(f) # 4 bytes
191 lomant = _read_ulong(f) # 4 bytes
192 if expon == himant == lomant == 0:
193 f = 0.0
194 elif expon == 0x7FFF:
195 f = _HUGE_VAL
196 else:
197 expon = expon - 16383
Guido van Rossume2a383d2007-01-15 16:59:06 +0000198 f = (himant * 0x100000000 + lomant) * pow(2.0, expon - 63)
Guido van Rossum4acc25b2000-02-02 15:10:15 +0000199 return sign * f
Sjoerd Mullendereeabe7e1993-01-22 12:53:11 +0000200
201def _write_short(f, x):
Guido van Rossum4acc25b2000-02-02 15:10:15 +0000202 f.write(struct.pack('>h', x))
Sjoerd Mullendereeabe7e1993-01-22 12:53:11 +0000203
Antoine Pitrou03757ec2012-01-17 17:13:04 +0100204def _write_ushort(f, x):
205 f.write(struct.pack('>H', x))
206
Sjoerd Mullendereeabe7e1993-01-22 12:53:11 +0000207def _write_long(f, x):
Antoine Pitrou03757ec2012-01-17 17:13:04 +0100208 f.write(struct.pack('>l', x))
209
210def _write_ulong(f, x):
Guido van Rossum4acc25b2000-02-02 15:10:15 +0000211 f.write(struct.pack('>L', x))
Sjoerd Mullendereeabe7e1993-01-22 12:53:11 +0000212
213def _write_string(f, s):
Thomas Wouters4d70c3d2006-06-08 14:42:34 +0000214 if len(s) > 255:
215 raise ValueError("string exceeds maximum pstring length")
Antoine Pitrou03757ec2012-01-17 17:13:04 +0100216 f.write(struct.pack('B', len(s)))
Guido van Rossum4acc25b2000-02-02 15:10:15 +0000217 f.write(s)
218 if len(s) & 1 == 0:
Georg Brandl2095cfe2008-06-07 19:01:03 +0000219 f.write(b'\x00')
Sjoerd Mullendereeabe7e1993-01-22 12:53:11 +0000220
221def _write_float(f, x):
Guido van Rossum4acc25b2000-02-02 15:10:15 +0000222 import math
223 if x < 0:
224 sign = 0x8000
225 x = x * -1
226 else:
227 sign = 0
228 if x == 0:
229 expon = 0
230 himant = 0
231 lomant = 0
232 else:
233 fmant, expon = math.frexp(x)
Antoine Pitrou03757ec2012-01-17 17:13:04 +0100234 if expon > 16384 or fmant >= 1 or fmant != fmant: # Infinity or NaN
Guido van Rossum4acc25b2000-02-02 15:10:15 +0000235 expon = sign|0x7FFF
236 himant = 0
237 lomant = 0
238 else: # Finite
239 expon = expon + 16382
240 if expon < 0: # denormalized
241 fmant = math.ldexp(fmant, expon)
242 expon = 0
243 expon = expon | sign
244 fmant = math.ldexp(fmant, 32)
245 fsmant = math.floor(fmant)
Guido van Rossume2a383d2007-01-15 16:59:06 +0000246 himant = int(fsmant)
Guido van Rossum4acc25b2000-02-02 15:10:15 +0000247 fmant = math.ldexp(fmant - fsmant, 32)
248 fsmant = math.floor(fmant)
Guido van Rossume2a383d2007-01-15 16:59:06 +0000249 lomant = int(fsmant)
Antoine Pitrou03757ec2012-01-17 17:13:04 +0100250 _write_ushort(f, expon)
251 _write_ulong(f, himant)
252 _write_ulong(f, lomant)
Sjoerd Mullendereeabe7e1993-01-22 12:53:11 +0000253
Guido van Rossum8ea7bb81999-06-09 13:32:28 +0000254from chunk import Chunk
R David Murray4d35e752013-07-25 16:12:01 -0400255from collections import namedtuple
256
257_aifc_params = namedtuple('_aifc_params',
258 'nchannels sampwidth framerate nframes comptype compname')
259
Sjoerd Mullendereeabe7e1993-01-22 12:53:11 +0000260
Guido van Rossumd3166071993-05-24 14:16:22 +0000261class Aifc_read:
Guido van Rossum4acc25b2000-02-02 15:10:15 +0000262 # Variables used in this class:
263 #
264 # These variables are available to the user though appropriate
265 # methods of this class:
266 # _file -- the open file with methods read(), close(), and seek()
267 # set through the __init__() method
268 # _nchannels -- the number of audio channels
269 # available through the getnchannels() method
270 # _nframes -- the number of audio frames
271 # available through the getnframes() method
272 # _sampwidth -- the number of bytes per audio sample
273 # available through the getsampwidth() method
274 # _framerate -- the sampling frequency
275 # available through the getframerate() method
276 # _comptype -- the AIFF-C compression type ('NONE' if AIFF)
277 # available through the getcomptype() method
278 # _compname -- the human-readable AIFF-C compression type
279 # available through the getcomptype() method
280 # _markers -- the marks in the audio file
281 # available through the getmarkers() and getmark()
282 # methods
283 # _soundpos -- the position in the audio stream
284 # available through the tell() method, set through the
285 # setpos() method
286 #
287 # These variables are used internally only:
288 # _version -- the AIFF-C version number
289 # _decomp -- the decompressor from builtin module cl
290 # _comm_chunk_read -- 1 iff the COMM chunk has been read
291 # _aifc -- 1 iff reading an AIFF-C file
292 # _ssnd_seek_needed -- 1 iff positioned correctly in audio
293 # file for readframes()
294 # _ssnd_chunk -- instantiation of a chunk class for the SSND chunk
295 # _framesize -- size of one frame in the file
Sjoerd Mullender2a451411993-12-20 09:36:01 +0000296
Guido van Rossum4acc25b2000-02-02 15:10:15 +0000297 def initfp(self, file):
298 self._version = 0
Guido van Rossum4acc25b2000-02-02 15:10:15 +0000299 self._convert = None
300 self._markers = []
301 self._soundpos = 0
R. David Murray99352742009-05-07 18:24:38 +0000302 self._file = file
303 chunk = Chunk(file)
304 if chunk.getname() != b'FORM':
Collin Winterce36ad82007-08-30 01:19:48 +0000305 raise Error('file does not start with FORM id')
R. David Murray99352742009-05-07 18:24:38 +0000306 formdata = chunk.read(4)
Georg Brandl2095cfe2008-06-07 19:01:03 +0000307 if formdata == b'AIFF':
Guido van Rossum4acc25b2000-02-02 15:10:15 +0000308 self._aifc = 0
Georg Brandl2095cfe2008-06-07 19:01:03 +0000309 elif formdata == b'AIFC':
Guido van Rossum4acc25b2000-02-02 15:10:15 +0000310 self._aifc = 1
311 else:
Collin Winterce36ad82007-08-30 01:19:48 +0000312 raise Error('not an AIFF or AIFF-C file')
Guido van Rossum4acc25b2000-02-02 15:10:15 +0000313 self._comm_chunk_read = 0
314 while 1:
315 self._ssnd_seek_needed = 1
316 try:
317 chunk = Chunk(self._file)
318 except EOFError:
319 break
320 chunkname = chunk.getname()
Georg Brandl2095cfe2008-06-07 19:01:03 +0000321 if chunkname == b'COMM':
Guido van Rossum4acc25b2000-02-02 15:10:15 +0000322 self._read_comm_chunk(chunk)
323 self._comm_chunk_read = 1
Georg Brandl2095cfe2008-06-07 19:01:03 +0000324 elif chunkname == b'SSND':
Guido van Rossum4acc25b2000-02-02 15:10:15 +0000325 self._ssnd_chunk = chunk
326 dummy = chunk.read(8)
327 self._ssnd_seek_needed = 0
Georg Brandl2095cfe2008-06-07 19:01:03 +0000328 elif chunkname == b'FVER':
Guido van Rossum820819c2002-08-12 22:11:28 +0000329 self._version = _read_ulong(chunk)
Georg Brandl2095cfe2008-06-07 19:01:03 +0000330 elif chunkname == b'MARK':
Guido van Rossum4acc25b2000-02-02 15:10:15 +0000331 self._readmark(chunk)
Guido van Rossum4acc25b2000-02-02 15:10:15 +0000332 chunk.skip()
333 if not self._comm_chunk_read or not self._ssnd_chunk:
Collin Winterce36ad82007-08-30 01:19:48 +0000334 raise Error('COMM chunk and/or SSND chunk missing')
Sjoerd Mullendereeabe7e1993-01-22 12:53:11 +0000335
Guido van Rossum4acc25b2000-02-02 15:10:15 +0000336 def __init__(self, f):
Georg Brandl2095cfe2008-06-07 19:01:03 +0000337 if isinstance(f, str):
Georg Brandl1a3284e2007-12-02 09:40:06 +0000338 f = builtins.open(f, 'rb')
Guido van Rossum4acc25b2000-02-02 15:10:15 +0000339 # else, assume it is an open file object already
340 self.initfp(f)
Sjoerd Mullendereeabe7e1993-01-22 12:53:11 +0000341
Serhiy Storchaka44c66c72012-12-29 22:54:49 +0200342 def __enter__(self):
343 return self
344
345 def __exit__(self, *args):
346 self.close()
347
Guido van Rossum4acc25b2000-02-02 15:10:15 +0000348 #
349 # User visible methods.
350 #
351 def getfp(self):
352 return self._file
Sjoerd Mullendereeabe7e1993-01-22 12:53:11 +0000353
Guido van Rossum4acc25b2000-02-02 15:10:15 +0000354 def rewind(self):
355 self._ssnd_seek_needed = 1
356 self._soundpos = 0
Sjoerd Mullendereeabe7e1993-01-22 12:53:11 +0000357
Guido van Rossum4acc25b2000-02-02 15:10:15 +0000358 def close(self):
Benjamin Peterson1d1285d2009-05-07 11:53:38 +0000359 self._file.close()
Sjoerd Mullendereeabe7e1993-01-22 12:53:11 +0000360
Guido van Rossum4acc25b2000-02-02 15:10:15 +0000361 def tell(self):
362 return self._soundpos
Sjoerd Mullendereeabe7e1993-01-22 12:53:11 +0000363
Guido van Rossum4acc25b2000-02-02 15:10:15 +0000364 def getnchannels(self):
365 return self._nchannels
Sjoerd Mullendereeabe7e1993-01-22 12:53:11 +0000366
Guido van Rossum4acc25b2000-02-02 15:10:15 +0000367 def getnframes(self):
368 return self._nframes
Sjoerd Mullendereeabe7e1993-01-22 12:53:11 +0000369
Guido van Rossum4acc25b2000-02-02 15:10:15 +0000370 def getsampwidth(self):
371 return self._sampwidth
Sjoerd Mullendereeabe7e1993-01-22 12:53:11 +0000372
Guido van Rossum4acc25b2000-02-02 15:10:15 +0000373 def getframerate(self):
374 return self._framerate
Sjoerd Mullendereeabe7e1993-01-22 12:53:11 +0000375
Guido van Rossum4acc25b2000-02-02 15:10:15 +0000376 def getcomptype(self):
377 return self._comptype
Sjoerd Mullendereeabe7e1993-01-22 12:53:11 +0000378
Guido van Rossum4acc25b2000-02-02 15:10:15 +0000379 def getcompname(self):
380 return self._compname
Sjoerd Mullendereeabe7e1993-01-22 12:53:11 +0000381
Guido van Rossum4acc25b2000-02-02 15:10:15 +0000382## def getversion(self):
383## return self._version
Sjoerd Mullendereeabe7e1993-01-22 12:53:11 +0000384
Guido van Rossum4acc25b2000-02-02 15:10:15 +0000385 def getparams(self):
R David Murray4d35e752013-07-25 16:12:01 -0400386 return _aifc_params(self.getnchannels(), self.getsampwidth(),
387 self.getframerate(), self.getnframes(),
388 self.getcomptype(), self.getcompname())
Sjoerd Mullendereeabe7e1993-01-22 12:53:11 +0000389
Guido van Rossum4acc25b2000-02-02 15:10:15 +0000390 def getmarkers(self):
391 if len(self._markers) == 0:
392 return None
393 return self._markers
Sjoerd Mullendereeabe7e1993-01-22 12:53:11 +0000394
Guido van Rossum4acc25b2000-02-02 15:10:15 +0000395 def getmark(self, id):
396 for marker in self._markers:
397 if id == marker[0]:
398 return marker
Georg Brandl2095cfe2008-06-07 19:01:03 +0000399 raise Error('marker {0!r} does not exist'.format(id))
Sjoerd Mullendereeabe7e1993-01-22 12:53:11 +0000400
Guido van Rossum4acc25b2000-02-02 15:10:15 +0000401 def setpos(self, pos):
402 if pos < 0 or pos > self._nframes:
Collin Winterce36ad82007-08-30 01:19:48 +0000403 raise Error('position not in range')
Guido van Rossum4acc25b2000-02-02 15:10:15 +0000404 self._soundpos = pos
405 self._ssnd_seek_needed = 1
Sjoerd Mullendereeabe7e1993-01-22 12:53:11 +0000406
Guido van Rossum4acc25b2000-02-02 15:10:15 +0000407 def readframes(self, nframes):
408 if self._ssnd_seek_needed:
409 self._ssnd_chunk.seek(0)
410 dummy = self._ssnd_chunk.read(8)
411 pos = self._soundpos * self._framesize
412 if pos:
Guido van Rossum2663c132000-03-07 15:19:31 +0000413 self._ssnd_chunk.seek(pos + 8)
Guido van Rossum4acc25b2000-02-02 15:10:15 +0000414 self._ssnd_seek_needed = 0
415 if nframes == 0:
Georg Brandl2095cfe2008-06-07 19:01:03 +0000416 return b''
Guido van Rossum4acc25b2000-02-02 15:10:15 +0000417 data = self._ssnd_chunk.read(nframes * self._framesize)
418 if self._convert and data:
419 data = self._convert(data)
Georg Brandl2095cfe2008-06-07 19:01:03 +0000420 self._soundpos = self._soundpos + len(data) // (self._nchannels
421 * self._sampwidth)
Guido van Rossum4acc25b2000-02-02 15:10:15 +0000422 return data
Sjoerd Mullendereeabe7e1993-01-22 12:53:11 +0000423
Guido van Rossum4acc25b2000-02-02 15:10:15 +0000424 #
425 # Internal methods.
426 #
Sjoerd Mullender2a451411993-12-20 09:36:01 +0000427
Georg Brandl2095cfe2008-06-07 19:01:03 +0000428 def _alaw2lin(self, data):
429 import audioop
430 return audioop.alaw2lin(data, 2)
Sjoerd Mullender43bf0bc1993-12-13 11:42:39 +0000431
Guido van Rossum4acc25b2000-02-02 15:10:15 +0000432 def _ulaw2lin(self, data):
433 import audioop
434 return audioop.ulaw2lin(data, 2)
Sjoerd Mullender43bf0bc1993-12-13 11:42:39 +0000435
Guido van Rossum4acc25b2000-02-02 15:10:15 +0000436 def _adpcm2lin(self, data):
437 import audioop
438 if not hasattr(self, '_adpcmstate'):
439 # first time
440 self._adpcmstate = None
Georg Brandl2095cfe2008-06-07 19:01:03 +0000441 data, self._adpcmstate = audioop.adpcm2lin(data, 2, self._adpcmstate)
Guido van Rossum4acc25b2000-02-02 15:10:15 +0000442 return data
Sjoerd Mullender1f057541994-09-06 16:17:51 +0000443
Guido van Rossum4acc25b2000-02-02 15:10:15 +0000444 def _read_comm_chunk(self, chunk):
445 self._nchannels = _read_short(chunk)
446 self._nframes = _read_long(chunk)
Georg Brandl2095cfe2008-06-07 19:01:03 +0000447 self._sampwidth = (_read_short(chunk) + 7) // 8
Guido van Rossum4acc25b2000-02-02 15:10:15 +0000448 self._framerate = int(_read_float(chunk))
449 self._framesize = self._nchannels * self._sampwidth
450 if self._aifc:
451 #DEBUG: SGI's soundeditor produces a bad size :-(
452 kludge = 0
453 if chunk.chunksize == 18:
454 kludge = 1
Ezio Melotti48d578c2012-03-12 23:57:18 +0200455 warnings.warn('Warning: bad COMM chunk size')
Guido van Rossum4acc25b2000-02-02 15:10:15 +0000456 chunk.chunksize = 23
457 #DEBUG end
458 self._comptype = chunk.read(4)
459 #DEBUG start
460 if kludge:
461 length = ord(chunk.file.read(1))
462 if length & 1 == 0:
463 length = length + 1
464 chunk.chunksize = chunk.chunksize + length
465 chunk.file.seek(-1, 1)
466 #DEBUG end
467 self._compname = _read_string(chunk)
Georg Brandl2095cfe2008-06-07 19:01:03 +0000468 if self._comptype != b'NONE':
469 if self._comptype == b'G722':
470 self._convert = self._adpcm2lin
Georg Brandl2095cfe2008-06-07 19:01:03 +0000471 elif self._comptype in (b'ulaw', b'ULAW'):
472 self._convert = self._ulaw2lin
Georg Brandl2095cfe2008-06-07 19:01:03 +0000473 elif self._comptype in (b'alaw', b'ALAW'):
474 self._convert = self._alaw2lin
Guido van Rossum4acc25b2000-02-02 15:10:15 +0000475 else:
Collin Winterce36ad82007-08-30 01:19:48 +0000476 raise Error('unsupported compression type')
Serhiy Storchaka4b532592013-10-12 18:21:33 +0300477 self._sampwidth = 2
Guido van Rossum4acc25b2000-02-02 15:10:15 +0000478 else:
Georg Brandl2095cfe2008-06-07 19:01:03 +0000479 self._comptype = b'NONE'
480 self._compname = b'not compressed'
Sjoerd Mullendereeabe7e1993-01-22 12:53:11 +0000481
Guido van Rossum4acc25b2000-02-02 15:10:15 +0000482 def _readmark(self, chunk):
483 nmarkers = _read_short(chunk)
484 # Some files appear to contain invalid counts.
485 # Cope with this by testing for EOF.
486 try:
487 for i in range(nmarkers):
488 id = _read_short(chunk)
489 pos = _read_long(chunk)
490 name = _read_string(chunk)
491 if pos or name:
492 # some files appear to have
493 # dummy markers consisting of
494 # a position 0 and name ''
495 self._markers.append((id, pos, name))
496 except EOFError:
Ezio Melotti48d578c2012-03-12 23:57:18 +0200497 w = ('Warning: MARK chunk contains only %s marker%s instead of %s' %
498 (len(self._markers), '' if len(self._markers) == 1 else 's',
499 nmarkers))
500 warnings.warn(w)
Sjoerd Mullendereeabe7e1993-01-22 12:53:11 +0000501
Guido van Rossumd3166071993-05-24 14:16:22 +0000502class Aifc_write:
Guido van Rossum4acc25b2000-02-02 15:10:15 +0000503 # Variables used in this class:
504 #
505 # These variables are user settable through appropriate methods
506 # of this class:
507 # _file -- the open file with methods write(), close(), tell(), seek()
508 # set through the __init__() method
509 # _comptype -- the AIFF-C compression type ('NONE' in AIFF)
510 # set through the setcomptype() or setparams() method
511 # _compname -- the human-readable AIFF-C compression type
512 # set through the setcomptype() or setparams() method
513 # _nchannels -- the number of audio channels
514 # set through the setnchannels() or setparams() method
515 # _sampwidth -- the number of bytes per audio sample
516 # set through the setsampwidth() or setparams() method
517 # _framerate -- the sampling frequency
518 # set through the setframerate() or setparams() method
519 # _nframes -- the number of audio frames written to the header
520 # set through the setnframes() or setparams() method
521 # _aifc -- whether we're writing an AIFF-C file or an AIFF file
522 # set through the aifc() method, reset through the
523 # aiff() method
524 #
525 # These variables are used internally only:
526 # _version -- the AIFF-C version number
527 # _comp -- the compressor from builtin module cl
528 # _nframeswritten -- the number of audio frames actually written
529 # _datalength -- the size of the audio samples written to the header
530 # _datawritten -- the size of the audio samples actually written
Sjoerd Mullendereeabe7e1993-01-22 12:53:11 +0000531
Guido van Rossum4acc25b2000-02-02 15:10:15 +0000532 def __init__(self, f):
Georg Brandl2095cfe2008-06-07 19:01:03 +0000533 if isinstance(f, str):
Guido van Rossum4acc25b2000-02-02 15:10:15 +0000534 filename = f
Georg Brandl1a3284e2007-12-02 09:40:06 +0000535 f = builtins.open(f, 'wb')
Guido van Rossum4acc25b2000-02-02 15:10:15 +0000536 else:
537 # else, assume it is an open file object already
538 filename = '???'
539 self.initfp(f)
540 if filename[-5:] == '.aiff':
541 self._aifc = 0
542 else:
543 self._aifc = 1
Sjoerd Mullendereeabe7e1993-01-22 12:53:11 +0000544
Guido van Rossum4acc25b2000-02-02 15:10:15 +0000545 def initfp(self, file):
546 self._file = file
547 self._version = _AIFC_version
Georg Brandl2095cfe2008-06-07 19:01:03 +0000548 self._comptype = b'NONE'
549 self._compname = b'not compressed'
Guido van Rossum4acc25b2000-02-02 15:10:15 +0000550 self._convert = None
551 self._nchannels = 0
552 self._sampwidth = 0
553 self._framerate = 0
554 self._nframes = 0
555 self._nframeswritten = 0
556 self._datawritten = 0
557 self._datalength = 0
558 self._markers = []
559 self._marklength = 0
560 self._aifc = 1 # AIFF-C is default
Sjoerd Mullendereeabe7e1993-01-22 12:53:11 +0000561
Guido van Rossum4acc25b2000-02-02 15:10:15 +0000562 def __del__(self):
Sandro Tosi70efbef2012-01-01 22:53:08 +0100563 self.close()
Sjoerd Mullender43bf0bc1993-12-13 11:42:39 +0000564
Serhiy Storchaka44c66c72012-12-29 22:54:49 +0200565 def __enter__(self):
566 return self
567
568 def __exit__(self, *args):
569 self.close()
570
Guido van Rossum4acc25b2000-02-02 15:10:15 +0000571 #
572 # User visible methods.
573 #
574 def aiff(self):
575 if self._nframeswritten:
Collin Winterce36ad82007-08-30 01:19:48 +0000576 raise Error('cannot change parameters after starting to write')
Guido van Rossum4acc25b2000-02-02 15:10:15 +0000577 self._aifc = 0
Sjoerd Mullendereeabe7e1993-01-22 12:53:11 +0000578
Guido van Rossum4acc25b2000-02-02 15:10:15 +0000579 def aifc(self):
580 if self._nframeswritten:
Collin Winterce36ad82007-08-30 01:19:48 +0000581 raise Error('cannot change parameters after starting to write')
Guido van Rossum4acc25b2000-02-02 15:10:15 +0000582 self._aifc = 1
Sjoerd Mullendereeabe7e1993-01-22 12:53:11 +0000583
Guido van Rossum4acc25b2000-02-02 15:10:15 +0000584 def setnchannels(self, nchannels):
585 if self._nframeswritten:
Collin Winterce36ad82007-08-30 01:19:48 +0000586 raise Error('cannot change parameters after starting to write')
Guido van Rossum4acc25b2000-02-02 15:10:15 +0000587 if nchannels < 1:
Collin Winterce36ad82007-08-30 01:19:48 +0000588 raise Error('bad # of channels')
Guido van Rossum4acc25b2000-02-02 15:10:15 +0000589 self._nchannels = nchannels
Sjoerd Mullendereeabe7e1993-01-22 12:53:11 +0000590
Guido van Rossum4acc25b2000-02-02 15:10:15 +0000591 def getnchannels(self):
592 if not self._nchannels:
Collin Winterce36ad82007-08-30 01:19:48 +0000593 raise Error('number of channels not set')
Guido van Rossum4acc25b2000-02-02 15:10:15 +0000594 return self._nchannels
Sjoerd Mullendereeabe7e1993-01-22 12:53:11 +0000595
Guido van Rossum4acc25b2000-02-02 15:10:15 +0000596 def setsampwidth(self, sampwidth):
597 if self._nframeswritten:
Collin Winterce36ad82007-08-30 01:19:48 +0000598 raise Error('cannot change parameters after starting to write')
Guido van Rossum4acc25b2000-02-02 15:10:15 +0000599 if sampwidth < 1 or sampwidth > 4:
Collin Winterce36ad82007-08-30 01:19:48 +0000600 raise Error('bad sample width')
Guido van Rossum4acc25b2000-02-02 15:10:15 +0000601 self._sampwidth = sampwidth
Sjoerd Mullendereeabe7e1993-01-22 12:53:11 +0000602
Guido van Rossum4acc25b2000-02-02 15:10:15 +0000603 def getsampwidth(self):
604 if not self._sampwidth:
Collin Winterce36ad82007-08-30 01:19:48 +0000605 raise Error('sample width not set')
Guido van Rossum4acc25b2000-02-02 15:10:15 +0000606 return self._sampwidth
Sjoerd Mullendereeabe7e1993-01-22 12:53:11 +0000607
Guido van Rossum4acc25b2000-02-02 15:10:15 +0000608 def setframerate(self, framerate):
609 if self._nframeswritten:
Collin Winterce36ad82007-08-30 01:19:48 +0000610 raise Error('cannot change parameters after starting to write')
Guido van Rossum4acc25b2000-02-02 15:10:15 +0000611 if framerate <= 0:
Collin Winterce36ad82007-08-30 01:19:48 +0000612 raise Error('bad frame rate')
Guido van Rossum4acc25b2000-02-02 15:10:15 +0000613 self._framerate = framerate
Sjoerd Mullendereeabe7e1993-01-22 12:53:11 +0000614
Guido van Rossum4acc25b2000-02-02 15:10:15 +0000615 def getframerate(self):
616 if not self._framerate:
Collin Winterce36ad82007-08-30 01:19:48 +0000617 raise Error('frame rate not set')
Guido van Rossum4acc25b2000-02-02 15:10:15 +0000618 return self._framerate
Sjoerd Mullendereeabe7e1993-01-22 12:53:11 +0000619
Guido van Rossum4acc25b2000-02-02 15:10:15 +0000620 def setnframes(self, nframes):
621 if self._nframeswritten:
Collin Winterce36ad82007-08-30 01:19:48 +0000622 raise Error('cannot change parameters after starting to write')
Guido van Rossum4acc25b2000-02-02 15:10:15 +0000623 self._nframes = nframes
Sjoerd Mullendereeabe7e1993-01-22 12:53:11 +0000624
Guido van Rossum4acc25b2000-02-02 15:10:15 +0000625 def getnframes(self):
626 return self._nframeswritten
Sjoerd Mullendereeabe7e1993-01-22 12:53:11 +0000627
Guido van Rossum4acc25b2000-02-02 15:10:15 +0000628 def setcomptype(self, comptype, compname):
629 if self._nframeswritten:
Collin Winterce36ad82007-08-30 01:19:48 +0000630 raise Error('cannot change parameters after starting to write')
Georg Brandl2095cfe2008-06-07 19:01:03 +0000631 if comptype not in (b'NONE', b'ulaw', b'ULAW',
632 b'alaw', b'ALAW', b'G722'):
Collin Winterce36ad82007-08-30 01:19:48 +0000633 raise Error('unsupported compression type')
Guido van Rossum4acc25b2000-02-02 15:10:15 +0000634 self._comptype = comptype
635 self._compname = compname
Sjoerd Mullendereeabe7e1993-01-22 12:53:11 +0000636
Guido van Rossum4acc25b2000-02-02 15:10:15 +0000637 def getcomptype(self):
638 return self._comptype
Sjoerd Mullendereeabe7e1993-01-22 12:53:11 +0000639
Guido van Rossum4acc25b2000-02-02 15:10:15 +0000640 def getcompname(self):
641 return self._compname
Sjoerd Mullendereeabe7e1993-01-22 12:53:11 +0000642
Guido van Rossum4acc25b2000-02-02 15:10:15 +0000643## def setversion(self, version):
644## if self._nframeswritten:
645## raise Error, 'cannot change parameters after starting to write'
646## self._version = version
Sjoerd Mullendereeabe7e1993-01-22 12:53:11 +0000647
Guido van Rossum1bc535d2007-05-15 18:46:22 +0000648 def setparams(self, params):
649 nchannels, sampwidth, framerate, nframes, comptype, compname = params
Guido van Rossum4acc25b2000-02-02 15:10:15 +0000650 if self._nframeswritten:
Collin Winterce36ad82007-08-30 01:19:48 +0000651 raise Error('cannot change parameters after starting to write')
Georg Brandl2095cfe2008-06-07 19:01:03 +0000652 if comptype not in (b'NONE', b'ulaw', b'ULAW',
653 b'alaw', b'ALAW', b'G722'):
Collin Winterce36ad82007-08-30 01:19:48 +0000654 raise Error('unsupported compression type')
Guido van Rossum4acc25b2000-02-02 15:10:15 +0000655 self.setnchannels(nchannels)
656 self.setsampwidth(sampwidth)
657 self.setframerate(framerate)
658 self.setnframes(nframes)
659 self.setcomptype(comptype, compname)
Sjoerd Mullendereeabe7e1993-01-22 12:53:11 +0000660
Guido van Rossum4acc25b2000-02-02 15:10:15 +0000661 def getparams(self):
662 if not self._nchannels or not self._sampwidth or not self._framerate:
Collin Winterce36ad82007-08-30 01:19:48 +0000663 raise Error('not all parameters set')
R David Murray4d35e752013-07-25 16:12:01 -0400664 return _aifc_params(self._nchannels, self._sampwidth, self._framerate,
665 self._nframes, self._comptype, self._compname)
Sjoerd Mullendereeabe7e1993-01-22 12:53:11 +0000666
Guido van Rossum4acc25b2000-02-02 15:10:15 +0000667 def setmark(self, id, pos, name):
668 if id <= 0:
Collin Winterce36ad82007-08-30 01:19:48 +0000669 raise Error('marker ID must be > 0')
Guido van Rossum4acc25b2000-02-02 15:10:15 +0000670 if pos < 0:
Collin Winterce36ad82007-08-30 01:19:48 +0000671 raise Error('marker position must be >= 0')
Sandro Tosi70efbef2012-01-01 22:53:08 +0100672 if not isinstance(name, bytes):
673 raise Error('marker name must be bytes')
Guido van Rossum4acc25b2000-02-02 15:10:15 +0000674 for i in range(len(self._markers)):
675 if id == self._markers[i][0]:
676 self._markers[i] = id, pos, name
677 return
678 self._markers.append((id, pos, name))
Sjoerd Mullendereeabe7e1993-01-22 12:53:11 +0000679
Guido van Rossum4acc25b2000-02-02 15:10:15 +0000680 def getmark(self, id):
681 for marker in self._markers:
682 if id == marker[0]:
683 return marker
Georg Brandl2095cfe2008-06-07 19:01:03 +0000684 raise Error('marker {0!r} does not exist'.format(id))
Sjoerd Mullendereeabe7e1993-01-22 12:53:11 +0000685
Guido van Rossum4acc25b2000-02-02 15:10:15 +0000686 def getmarkers(self):
687 if len(self._markers) == 0:
688 return None
689 return self._markers
Tim Peters146965a2001-01-14 18:09:23 +0000690
Guido van Rossum4acc25b2000-02-02 15:10:15 +0000691 def tell(self):
692 return self._nframeswritten
Sjoerd Mullender43bf0bc1993-12-13 11:42:39 +0000693
Guido van Rossum4acc25b2000-02-02 15:10:15 +0000694 def writeframesraw(self, data):
Serhiy Storchaka452bab42013-11-16 14:01:31 +0200695 if not isinstance(data, (bytes, bytearray)):
696 data = memoryview(data).cast('B')
Guido van Rossum4acc25b2000-02-02 15:10:15 +0000697 self._ensure_header_written(len(data))
Georg Brandl2095cfe2008-06-07 19:01:03 +0000698 nframes = len(data) // (self._sampwidth * self._nchannels)
Guido van Rossum4acc25b2000-02-02 15:10:15 +0000699 if self._convert:
700 data = self._convert(data)
701 self._file.write(data)
702 self._nframeswritten = self._nframeswritten + nframes
703 self._datawritten = self._datawritten + len(data)
Sjoerd Mullendereeabe7e1993-01-22 12:53:11 +0000704
Guido van Rossum4acc25b2000-02-02 15:10:15 +0000705 def writeframes(self, data):
706 self.writeframesraw(data)
707 if self._nframeswritten != self._nframes or \
708 self._datalength != self._datawritten:
709 self._patchheader()
Sjoerd Mullendereeabe7e1993-01-22 12:53:11 +0000710
Guido van Rossum4acc25b2000-02-02 15:10:15 +0000711 def close(self):
Serhiy Storchaka051722d2012-12-29 22:30:56 +0200712 if self._file is None:
713 return
714 try:
Sandro Tosi70efbef2012-01-01 22:53:08 +0100715 self._ensure_header_written(0)
716 if self._datawritten & 1:
717 # quick pad to even size
718 self._file.write(b'\x00')
719 self._datawritten = self._datawritten + 1
720 self._writemarkers()
721 if self._nframeswritten != self._nframes or \
722 self._datalength != self._datawritten or \
723 self._marklength:
724 self._patchheader()
Serhiy Storchaka051722d2012-12-29 22:30:56 +0200725 finally:
Sandro Tosi70efbef2012-01-01 22:53:08 +0100726 # Prevent ref cycles
727 self._convert = None
Serhiy Storchaka051722d2012-12-29 22:30:56 +0200728 f = self._file
Sandro Tosi70efbef2012-01-01 22:53:08 +0100729 self._file = None
Serhiy Storchaka051722d2012-12-29 22:30:56 +0200730 f.close()
Sjoerd Mullendereeabe7e1993-01-22 12:53:11 +0000731
Guido van Rossum4acc25b2000-02-02 15:10:15 +0000732 #
733 # Internal methods.
734 #
Sjoerd Mullender2a451411993-12-20 09:36:01 +0000735
Georg Brandl2095cfe2008-06-07 19:01:03 +0000736 def _lin2alaw(self, data):
737 import audioop
738 return audioop.lin2alaw(data, 2)
Sjoerd Mullender43bf0bc1993-12-13 11:42:39 +0000739
Guido van Rossum4acc25b2000-02-02 15:10:15 +0000740 def _lin2ulaw(self, data):
741 import audioop
742 return audioop.lin2ulaw(data, 2)
Sjoerd Mullender43bf0bc1993-12-13 11:42:39 +0000743
Guido van Rossum4acc25b2000-02-02 15:10:15 +0000744 def _lin2adpcm(self, data):
745 import audioop
746 if not hasattr(self, '_adpcmstate'):
747 self._adpcmstate = None
Georg Brandl2095cfe2008-06-07 19:01:03 +0000748 data, self._adpcmstate = audioop.lin2adpcm(data, 2, self._adpcmstate)
Guido van Rossum4acc25b2000-02-02 15:10:15 +0000749 return data
Sjoerd Mullender1f057541994-09-06 16:17:51 +0000750
Guido van Rossum4acc25b2000-02-02 15:10:15 +0000751 def _ensure_header_written(self, datasize):
752 if not self._nframeswritten:
Sandro Tosibdd53542012-01-01 18:04:37 +0100753 if self._comptype in (b'ULAW', b'ulaw', b'ALAW', b'alaw', b'G722'):
Guido van Rossum4acc25b2000-02-02 15:10:15 +0000754 if not self._sampwidth:
755 self._sampwidth = 2
756 if self._sampwidth != 2:
Georg Brandl2095cfe2008-06-07 19:01:03 +0000757 raise Error('sample width must be 2 when compressing '
Sandro Tosibdd53542012-01-01 18:04:37 +0100758 'with ulaw/ULAW, alaw/ALAW or G7.22 (ADPCM)')
Guido van Rossum4acc25b2000-02-02 15:10:15 +0000759 if not self._nchannels:
Collin Winterce36ad82007-08-30 01:19:48 +0000760 raise Error('# channels not specified')
Guido van Rossum4acc25b2000-02-02 15:10:15 +0000761 if not self._sampwidth:
Collin Winterce36ad82007-08-30 01:19:48 +0000762 raise Error('sample width not specified')
Guido van Rossum4acc25b2000-02-02 15:10:15 +0000763 if not self._framerate:
Collin Winterce36ad82007-08-30 01:19:48 +0000764 raise Error('sampling rate not specified')
Guido van Rossum4acc25b2000-02-02 15:10:15 +0000765 self._write_header(datasize)
Guido van Rossum52fc1f61993-06-17 12:38:10 +0000766
Guido van Rossum4acc25b2000-02-02 15:10:15 +0000767 def _init_compression(self):
Georg Brandl2095cfe2008-06-07 19:01:03 +0000768 if self._comptype == b'G722':
Guido van Rossum4acc25b2000-02-02 15:10:15 +0000769 self._convert = self._lin2adpcm
Georg Brandl2095cfe2008-06-07 19:01:03 +0000770 elif self._comptype in (b'ulaw', b'ULAW'):
771 self._convert = self._lin2ulaw
772 elif self._comptype in (b'alaw', b'ALAW'):
773 self._convert = self._lin2alaw
Sjoerd Mullender43bf0bc1993-12-13 11:42:39 +0000774
Guido van Rossum4acc25b2000-02-02 15:10:15 +0000775 def _write_header(self, initlength):
Georg Brandl2095cfe2008-06-07 19:01:03 +0000776 if self._aifc and self._comptype != b'NONE':
Guido van Rossum4acc25b2000-02-02 15:10:15 +0000777 self._init_compression()
Georg Brandl2095cfe2008-06-07 19:01:03 +0000778 self._file.write(b'FORM')
Guido van Rossum4acc25b2000-02-02 15:10:15 +0000779 if not self._nframes:
Georg Brandl2095cfe2008-06-07 19:01:03 +0000780 self._nframes = initlength // (self._nchannels * self._sampwidth)
Guido van Rossum4acc25b2000-02-02 15:10:15 +0000781 self._datalength = self._nframes * self._nchannels * self._sampwidth
782 if self._datalength & 1:
783 self._datalength = self._datalength + 1
784 if self._aifc:
Georg Brandl2095cfe2008-06-07 19:01:03 +0000785 if self._comptype in (b'ulaw', b'ULAW', b'alaw', b'ALAW'):
786 self._datalength = self._datalength // 2
Guido van Rossum4acc25b2000-02-02 15:10:15 +0000787 if self._datalength & 1:
788 self._datalength = self._datalength + 1
Georg Brandl2095cfe2008-06-07 19:01:03 +0000789 elif self._comptype == b'G722':
790 self._datalength = (self._datalength + 3) // 4
Guido van Rossum4acc25b2000-02-02 15:10:15 +0000791 if self._datalength & 1:
792 self._datalength = self._datalength + 1
Serhiy Storchaka84d28b42013-12-14 20:35:04 +0200793 try:
794 self._form_length_pos = self._file.tell()
795 except (AttributeError, OSError):
796 self._form_length_pos = None
Guido van Rossum4acc25b2000-02-02 15:10:15 +0000797 commlength = self._write_form_length(self._datalength)
798 if self._aifc:
Georg Brandl2095cfe2008-06-07 19:01:03 +0000799 self._file.write(b'AIFC')
800 self._file.write(b'FVER')
Antoine Pitrou03757ec2012-01-17 17:13:04 +0100801 _write_ulong(self._file, 4)
802 _write_ulong(self._file, self._version)
Guido van Rossum4acc25b2000-02-02 15:10:15 +0000803 else:
Georg Brandl2095cfe2008-06-07 19:01:03 +0000804 self._file.write(b'AIFF')
805 self._file.write(b'COMM')
Antoine Pitrou03757ec2012-01-17 17:13:04 +0100806 _write_ulong(self._file, commlength)
Guido van Rossum4acc25b2000-02-02 15:10:15 +0000807 _write_short(self._file, self._nchannels)
Serhiy Storchaka84d28b42013-12-14 20:35:04 +0200808 if self._form_length_pos is not None:
809 self._nframes_pos = self._file.tell()
Antoine Pitrou03757ec2012-01-17 17:13:04 +0100810 _write_ulong(self._file, self._nframes)
Serhiy Storchaka4b532592013-10-12 18:21:33 +0300811 if self._comptype in (b'ULAW', b'ulaw', b'ALAW', b'alaw', b'G722'):
812 _write_short(self._file, 8)
813 else:
814 _write_short(self._file, self._sampwidth * 8)
Guido van Rossum4acc25b2000-02-02 15:10:15 +0000815 _write_float(self._file, self._framerate)
816 if self._aifc:
817 self._file.write(self._comptype)
818 _write_string(self._file, self._compname)
Georg Brandl2095cfe2008-06-07 19:01:03 +0000819 self._file.write(b'SSND')
Serhiy Storchaka84d28b42013-12-14 20:35:04 +0200820 if self._form_length_pos is not None:
821 self._ssnd_length_pos = self._file.tell()
Antoine Pitrou03757ec2012-01-17 17:13:04 +0100822 _write_ulong(self._file, self._datalength + 8)
823 _write_ulong(self._file, 0)
824 _write_ulong(self._file, 0)
Sjoerd Mullendereeabe7e1993-01-22 12:53:11 +0000825
Guido van Rossum4acc25b2000-02-02 15:10:15 +0000826 def _write_form_length(self, datalength):
827 if self._aifc:
828 commlength = 18 + 5 + len(self._compname)
829 if commlength & 1:
830 commlength = commlength + 1
831 verslength = 12
832 else:
833 commlength = 18
834 verslength = 0
Antoine Pitrou03757ec2012-01-17 17:13:04 +0100835 _write_ulong(self._file, 4 + verslength + self._marklength + \
836 8 + commlength + 16 + datalength)
Guido van Rossum4acc25b2000-02-02 15:10:15 +0000837 return commlength
Sjoerd Mullendereeabe7e1993-01-22 12:53:11 +0000838
Guido van Rossum4acc25b2000-02-02 15:10:15 +0000839 def _patchheader(self):
840 curpos = self._file.tell()
841 if self._datawritten & 1:
842 datalength = self._datawritten + 1
Georg Brandl2095cfe2008-06-07 19:01:03 +0000843 self._file.write(b'\x00')
Guido van Rossum4acc25b2000-02-02 15:10:15 +0000844 else:
845 datalength = self._datawritten
846 if datalength == self._datalength and \
847 self._nframes == self._nframeswritten and \
848 self._marklength == 0:
849 self._file.seek(curpos, 0)
850 return
851 self._file.seek(self._form_length_pos, 0)
852 dummy = self._write_form_length(datalength)
853 self._file.seek(self._nframes_pos, 0)
Antoine Pitrou03757ec2012-01-17 17:13:04 +0100854 _write_ulong(self._file, self._nframeswritten)
Guido van Rossum4acc25b2000-02-02 15:10:15 +0000855 self._file.seek(self._ssnd_length_pos, 0)
Antoine Pitrou03757ec2012-01-17 17:13:04 +0100856 _write_ulong(self._file, datalength + 8)
Guido van Rossum4acc25b2000-02-02 15:10:15 +0000857 self._file.seek(curpos, 0)
858 self._nframes = self._nframeswritten
859 self._datalength = datalength
Sjoerd Mullendereeabe7e1993-01-22 12:53:11 +0000860
Guido van Rossum4acc25b2000-02-02 15:10:15 +0000861 def _writemarkers(self):
862 if len(self._markers) == 0:
863 return
Georg Brandl2095cfe2008-06-07 19:01:03 +0000864 self._file.write(b'MARK')
Guido van Rossum4acc25b2000-02-02 15:10:15 +0000865 length = 2
866 for marker in self._markers:
867 id, pos, name = marker
868 length = length + len(name) + 1 + 6
869 if len(name) & 1 == 0:
870 length = length + 1
Antoine Pitrou03757ec2012-01-17 17:13:04 +0100871 _write_ulong(self._file, length)
Guido van Rossum4acc25b2000-02-02 15:10:15 +0000872 self._marklength = length + 8
873 _write_short(self._file, len(self._markers))
874 for marker in self._markers:
875 id, pos, name = marker
876 _write_short(self._file, id)
Antoine Pitrou03757ec2012-01-17 17:13:04 +0100877 _write_ulong(self._file, pos)
Guido van Rossum4acc25b2000-02-02 15:10:15 +0000878 _write_string(self._file, name)
Sjoerd Mullendereeabe7e1993-01-22 12:53:11 +0000879
Fred Drake43161351999-06-22 21:23:23 +0000880def open(f, mode=None):
Guido van Rossum4acc25b2000-02-02 15:10:15 +0000881 if mode is None:
882 if hasattr(f, 'mode'):
883 mode = f.mode
884 else:
885 mode = 'rb'
886 if mode in ('r', 'rb'):
887 return Aifc_read(f)
888 elif mode in ('w', 'wb'):
889 return Aifc_write(f)
890 else:
Collin Winterce36ad82007-08-30 01:19:48 +0000891 raise Error("mode must be 'r', 'rb', 'w', or 'wb'")
Sjoerd Mullendereeabe7e1993-01-22 12:53:11 +0000892
Guido van Rossum7bc817d1993-12-17 15:25:27 +0000893openfp = open # B/W compatibility
Guido van Rossum36bb1811996-12-31 05:57:34 +0000894
895if __name__ == '__main__':
Guido van Rossum4acc25b2000-02-02 15:10:15 +0000896 import sys
897 if not sys.argv[1:]:
898 sys.argv.append('/usr/demos/data/audio/bach.aiff')
899 fn = sys.argv[1]
Serhiy Storchaka58b3ebf2013-08-25 19:16:01 +0300900 with open(fn, 'r') as f:
Serhiy Storchakab33baf12013-08-25 19:12:56 +0300901 print("Reading", fn)
902 print("nchannels =", f.getnchannels())
903 print("nframes =", f.getnframes())
904 print("sampwidth =", f.getsampwidth())
905 print("framerate =", f.getframerate())
906 print("comptype =", f.getcomptype())
907 print("compname =", f.getcompname())
908 if sys.argv[2:]:
909 gn = sys.argv[2]
910 print("Writing", gn)
Serhiy Storchaka58b3ebf2013-08-25 19:16:01 +0300911 with open(gn, 'w') as g:
Serhiy Storchakab33baf12013-08-25 19:12:56 +0300912 g.setparams(f.getparams())
913 while 1:
914 data = f.readframes(1024)
915 if not data:
916 break
917 g.writeframes(data)
Serhiy Storchakab33baf12013-08-25 19:12:56 +0300918 print("Done.")