blob: ca6cfb3df82773f8a2222987bf715e2d272622c4 [file] [log] [blame]
Georg Brandl116aa622007-08-15 14:28:22 +00001:mod:`audioop` --- Manipulate raw audio data
2============================================
3
4.. module:: audioop
5 :synopsis: Manipulate raw audio data.
6
7
8The :mod:`audioop` module contains some useful operations on sound fragments.
Serhiy Storchakaeaea5e92013-10-19 21:10:46 +03009It operates on sound fragments consisting of signed integer samples 8, 16, 24
Serhiy Storchaka711e91b2013-11-10 21:44:36 +020010or 32 bits wide, stored in :term:`bytes-like object`\ s. All scalar items are
11integers, unless specified otherwise.
Serhiy Storchakaeaea5e92013-10-19 21:10:46 +030012
13.. versionchanged:: 3.4
14 Support for 24-bit samples was added.
Georg Brandl116aa622007-08-15 14:28:22 +000015
Serhiy Storchaka711e91b2013-11-10 21:44:36 +020016.. versionchanged:: 3.4
17 Any :term:`bytes-like object`\ s are now accepted by all functions in this
18 module. Strings no more supported.
19
Georg Brandl116aa622007-08-15 14:28:22 +000020.. index::
21 single: Intel/DVI ADPCM
22 single: ADPCM, Intel/DVI
23 single: a-LAW
24 single: u-LAW
25
26This module provides support for a-LAW, u-LAW and Intel/DVI ADPCM encodings.
27
Christian Heimes5b5e81c2007-12-31 16:14:33 +000028.. This para is mostly here to provide an excuse for the index entries...
Georg Brandl116aa622007-08-15 14:28:22 +000029
30A few of the more complicated operations only take 16-bit samples, otherwise the
31sample size (in bytes) is always a parameter of the operation.
32
33The module defines the following variables and functions:
34
35
36.. exception:: error
37
38 This exception is raised on all errors, such as unknown number of bytes per
39 sample, etc.
40
41
42.. function:: add(fragment1, fragment2, width)
43
44 Return a fragment which is the addition of the two samples passed as parameters.
Serhiy Storchakaeaea5e92013-10-19 21:10:46 +030045 *width* is the sample width in bytes, either ``1``, ``2``, ``3`` or ``4``. Both
Serhiy Storchaka01ad6222013-02-09 11:10:53 +020046 fragments should have the same length. Samples are truncated in case of overflow.
Georg Brandl116aa622007-08-15 14:28:22 +000047
48
49.. function:: adpcm2lin(adpcmfragment, width, state)
50
51 Decode an Intel/DVI ADPCM coded fragment to a linear fragment. See the
52 description of :func:`lin2adpcm` for details on ADPCM coding. Return a tuple
53 ``(sample, newstate)`` where the sample has the width specified in *width*.
54
55
56.. function:: alaw2lin(fragment, width)
57
58 Convert sound fragments in a-LAW encoding to linearly encoded sound fragments.
59 a-LAW encoding always uses 8 bits samples, so *width* refers only to the sample
60 width of the output fragment here.
61
Georg Brandl116aa622007-08-15 14:28:22 +000062
63.. function:: avg(fragment, width)
64
65 Return the average over all samples in the fragment.
66
67
68.. function:: avgpp(fragment, width)
69
70 Return the average peak-peak value over all samples in the fragment. No
71 filtering is done, so the usefulness of this routine is questionable.
72
73
74.. function:: bias(fragment, width, bias)
75
76 Return a fragment that is the original fragment with a bias added to each
Serhiy Storchaka01ad6222013-02-09 11:10:53 +020077 sample. Samples wrap around in case of overflow.
Georg Brandl116aa622007-08-15 14:28:22 +000078
79
80.. function:: cross(fragment, width)
81
82 Return the number of zero crossings in the fragment passed as an argument.
83
84
85.. function:: findfactor(fragment, reference)
86
87 Return a factor *F* such that ``rms(add(fragment, mul(reference, -F)))`` is
88 minimal, i.e., return the factor with which you should multiply *reference* to
89 make it match as well as possible to *fragment*. The fragments should both
90 contain 2-byte samples.
91
92 The time taken by this routine is proportional to ``len(fragment)``.
93
94
95.. function:: findfit(fragment, reference)
96
97 Try to match *reference* as well as possible to a portion of *fragment* (which
98 should be the longer fragment). This is (conceptually) done by taking slices
99 out of *fragment*, using :func:`findfactor` to compute the best match, and
100 minimizing the result. The fragments should both contain 2-byte samples.
101 Return a tuple ``(offset, factor)`` where *offset* is the (integer) offset into
102 *fragment* where the optimal match started and *factor* is the (floating-point)
103 factor as per :func:`findfactor`.
104
105
106.. function:: findmax(fragment, length)
107
108 Search *fragment* for a slice of length *length* samples (not bytes!) with
109 maximum energy, i.e., return *i* for which ``rms(fragment[i*2:(i+length)*2])``
110 is maximal. The fragments should both contain 2-byte samples.
111
112 The routine takes time proportional to ``len(fragment)``.
113
114
115.. function:: getsample(fragment, width, index)
116
117 Return the value of sample *index* from the fragment.
118
119
120.. function:: lin2adpcm(fragment, width, state)
121
122 Convert samples to 4 bit Intel/DVI ADPCM encoding. ADPCM coding is an adaptive
123 coding scheme, whereby each 4 bit number is the difference between one sample
124 and the next, divided by a (varying) step. The Intel/DVI ADPCM algorithm has
125 been selected for use by the IMA, so it may well become a standard.
126
127 *state* is a tuple containing the state of the coder. The coder returns a tuple
128 ``(adpcmfrag, newstate)``, and the *newstate* should be passed to the next call
129 of :func:`lin2adpcm`. In the initial call, ``None`` can be passed as the state.
130 *adpcmfrag* is the ADPCM coded fragment packed 2 4-bit values per byte.
131
132
133.. function:: lin2alaw(fragment, width)
134
135 Convert samples in the audio fragment to a-LAW encoding and return this as a
Serhiy Storchakac8bd74d2012-12-27 20:43:36 +0200136 bytes object. a-LAW is an audio encoding format whereby you get a dynamic
Georg Brandl116aa622007-08-15 14:28:22 +0000137 range of about 13 bits using only 8 bit samples. It is used by the Sun audio
138 hardware, among others.
139
Georg Brandl116aa622007-08-15 14:28:22 +0000140
141.. function:: lin2lin(fragment, width, newwidth)
142
Serhiy Storchakaeaea5e92013-10-19 21:10:46 +0300143 Convert samples between 1-, 2-, 3- and 4-byte formats.
Georg Brandl116aa622007-08-15 14:28:22 +0000144
Christian Heimescc47b052008-03-25 14:56:36 +0000145 .. note::
146
Serhiy Storchakaeaea5e92013-10-19 21:10:46 +0300147 In some audio formats, such as .WAV files, 16, 24 and 32 bit samples are
Christian Heimescc47b052008-03-25 14:56:36 +0000148 signed, but 8 bit samples are unsigned. So when converting to 8 bit wide
149 samples for these formats, you need to also add 128 to the result::
150
151 new_frames = audioop.lin2lin(frames, old_width, 1)
152 new_frames = audioop.bias(new_frames, 1, 128)
153
Serhiy Storchakaeaea5e92013-10-19 21:10:46 +0300154 The same, in reverse, has to be applied when converting from 8 to 16, 24
155 or 32 bit width samples.
Christian Heimescc47b052008-03-25 14:56:36 +0000156
Georg Brandl116aa622007-08-15 14:28:22 +0000157
158.. function:: lin2ulaw(fragment, width)
159
160 Convert samples in the audio fragment to u-LAW encoding and return this as a
Serhiy Storchakac8bd74d2012-12-27 20:43:36 +0200161 bytes object. u-LAW is an audio encoding format whereby you get a dynamic
Georg Brandl116aa622007-08-15 14:28:22 +0000162 range of about 14 bits using only 8 bit samples. It is used by the Sun audio
163 hardware, among others.
164
165
Georg Brandl116aa622007-08-15 14:28:22 +0000166.. function:: max(fragment, width)
167
168 Return the maximum of the *absolute value* of all samples in a fragment.
169
170
171.. function:: maxpp(fragment, width)
172
173 Return the maximum peak-peak value in the sound fragment.
174
175
Ezio Melottie0035a22012-12-14 20:18:46 +0200176.. function:: minmax(fragment, width)
177
178 Return a tuple consisting of the minimum and maximum values of all samples in
179 the sound fragment.
180
181
Georg Brandl116aa622007-08-15 14:28:22 +0000182.. function:: mul(fragment, width, factor)
183
184 Return a fragment that has all samples in the original fragment multiplied by
Serhiy Storchaka01ad6222013-02-09 11:10:53 +0200185 the floating-point value *factor*. Samples are truncated in case of overflow.
Georg Brandl116aa622007-08-15 14:28:22 +0000186
187
188.. function:: ratecv(fragment, width, nchannels, inrate, outrate, state[, weightA[, weightB]])
189
190 Convert the frame rate of the input fragment.
191
192 *state* is a tuple containing the state of the converter. The converter returns
193 a tuple ``(newfragment, newstate)``, and *newstate* should be passed to the next
194 call of :func:`ratecv`. The initial call should pass ``None`` as the state.
195
196 The *weightA* and *weightB* arguments are parameters for a simple digital filter
197 and default to ``1`` and ``0`` respectively.
198
199
200.. function:: reverse(fragment, width)
201
202 Reverse the samples in a fragment and returns the modified fragment.
203
204
205.. function:: rms(fragment, width)
206
207 Return the root-mean-square of the fragment, i.e. ``sqrt(sum(S_i^2)/n)``.
208
209 This is a measure of the power in an audio signal.
210
211
212.. function:: tomono(fragment, width, lfactor, rfactor)
213
214 Convert a stereo fragment to a mono fragment. The left channel is multiplied by
215 *lfactor* and the right channel by *rfactor* before adding the two channels to
216 give a mono signal.
217
218
219.. function:: tostereo(fragment, width, lfactor, rfactor)
220
221 Generate a stereo fragment from a mono fragment. Each pair of samples in the
222 stereo fragment are computed from the mono sample, whereby left channel samples
223 are multiplied by *lfactor* and right channel samples by *rfactor*.
224
225
226.. function:: ulaw2lin(fragment, width)
227
228 Convert sound fragments in u-LAW encoding to linearly encoded sound fragments.
229 u-LAW encoding always uses 8 bits samples, so *width* refers only to the sample
230 width of the output fragment here.
231
Georg Brandl502d9a52009-07-26 15:02:41 +0000232Note that operations such as :func:`.mul` or :func:`.max` make no distinction
Georg Brandl116aa622007-08-15 14:28:22 +0000233between mono and stereo fragments, i.e. all samples are treated equal. If this
234is a problem the stereo fragment should be split into two mono fragments first
235and recombined later. Here is an example of how to do that::
236
237 def mul_stereo(sample, width, lfactor, rfactor):
238 lsample = audioop.tomono(sample, width, 1, 0)
239 rsample = audioop.tomono(sample, width, 0, 1)
Georg Brandlf3d00872010-10-17 10:07:29 +0000240 lsample = audioop.mul(lsample, width, lfactor)
241 rsample = audioop.mul(rsample, width, rfactor)
Georg Brandl116aa622007-08-15 14:28:22 +0000242 lsample = audioop.tostereo(lsample, width, 1, 0)
243 rsample = audioop.tostereo(rsample, width, 0, 1)
244 return audioop.add(lsample, rsample, width)
245
246If you use the ADPCM coder to build network packets and you want your protocol
247to be stateless (i.e. to be able to tolerate packet loss) you should not only
248transmit the data but also the state. Note that you should send the *initial*
249state (the one you passed to :func:`lin2adpcm`) along to the decoder, not the
250final state (as returned by the coder). If you want to use
Serhiy Storchakabfdcd432013-10-13 23:09:14 +0300251:class:`struct.Struct` to store the state in binary you can code the first
Georg Brandl116aa622007-08-15 14:28:22 +0000252element (the predicted value) in 16 bits and the second (the delta index) in 8.
253
254The ADPCM coders have never been tried against other ADPCM coders, only against
255themselves. It could well be that I misinterpreted the standards in which case
256they will not be interoperable with the respective standards.
257
258The :func:`find\*` routines might look a bit funny at first sight. They are
259primarily meant to do echo cancellation. A reasonably fast way to do this is to
260pick the most energetic piece of the output sample, locate that in the input
261sample and subtract the whole output sample from the input sample::
262
263 def echocancel(outputdata, inputdata):
264 pos = audioop.findmax(outputdata, 800) # one tenth second
265 out_test = outputdata[pos*2:]
266 in_test = inputdata[pos*2:]
267 ipos, factor = audioop.findfit(in_test, out_test)
268 # Optional (for better cancellation):
Georg Brandl48310cd2009-01-03 21:18:54 +0000269 # factor = audioop.findfactor(in_test[ipos*2:ipos*2+len(out_test)],
Georg Brandl116aa622007-08-15 14:28:22 +0000270 # out_test)
271 prefill = '\0'*(pos+ipos)*2
272 postfill = '\0'*(len(inputdata)-len(prefill)-len(outputdata))
273 outputdata = prefill + audioop.mul(outputdata,2,-factor) + postfill
274 return audioop.add(inputdata, outputdata, 2)
275