?  search:
summary | log | graphiclog1 | graphiclog2 | commit | commitdiff | tree | refs | edit | fork
blame | history | raw | HEAD
Saved and restored logging._handlerList at the same time as saving/restoring logging...
[python.git] / Modules / audioop.c
blob8d5a3055250cdaabfb374cc1a0551918910a1dba
1
2 /* audioopmodule - Module to detect peak values in arrays */
3
4 #include "Python.h"
5
6 #if SIZEOF_INT == 4
7 typedef int Py_Int32;
8 typedef unsigned int Py_UInt32;
9 #else
10 #if SIZEOF_LONG == 4
11 typedef long Py_Int32;
12 typedef unsigned long Py_UInt32;
13 #else
14 #error "No 4-byte integral type"
15 #endif
16 #endif
17
18 #if defined(__CHAR_UNSIGNED__)
19 #if defined(signed)
20 /* This module currently does not work on systems where only unsigned
21 characters are available. Take it out of Setup. Sorry. */
22 #endif
23 #endif
24
25 /* Code shamelessly stolen from sox,
26 ** (c) Craig Reese, Joe Campbell and Jeff Poskanzer 1989 */
27
28 #define MINLIN -32768
29 #define MAXLIN 32767
30 #define LINCLIP(x) do { if ( x < MINLIN ) x = MINLIN ; \
31 else if ( x > MAXLIN ) x = MAXLIN; \
32 } while ( 0 )
33
34 static unsigned char st_linear_to_ulaw(int sample);
35
36 /*
37 ** This macro converts from ulaw to 16 bit linear, faster.
38 **
39 ** Jef Poskanzer
40 ** 23 October 1989
41 **
42 ** Input: 8 bit ulaw sample
43 ** Output: signed 16 bit linear sample
44 */
45 #define st_ulaw_to_linear(ulawbyte) ulaw_table[ulawbyte]
46
47 static int ulaw_table[256] = {
48 -32124, -31100, -30076, -29052, -28028, -27004, -25980, -24956,
49 -23932, -22908, -21884, -20860, -19836, -18812, -17788, -16764,
50 -15996, -15484, -14972, -14460, -13948, -13436, -12924, -12412,
51 -11900, -11388, -10876, -10364, -9852, -9340, -8828, -8316,
52 -7932, -7676, -7420, -7164, -6908, -6652, -6396, -6140,
53 -5884, -5628, -5372, -5116, -4860, -4604, -4348, -4092,
54 -3900, -3772, -3644, -3516, -3388, -3260, -3132, -3004,
55 -2876, -2748, -2620, -2492, -2364, -2236, -2108, -1980,
56 -1884, -1820, -1756, -1692, -1628, -1564, -1500, -1436,
57 -1372, -1308, -1244, -1180, -1116, -1052, -988, -924,
58 -876, -844, -812, -780, -748, -716, -684, -652,
59 -620, -588, -556, -524, -492, -460, -428, -396,
60 -372, -356, -340, -324, -308, -292, -276, -260,
61 -244, -228, -212, -196, -180, -164, -148, -132,
62 -120, -112, -104, -96, -88, -80, -72, -64,
63 -56, -48, -40, -32, -24, -16, -8, 0,
64 32124, 31100, 30076, 29052, 28028, 27004, 25980, 24956,
65 23932, 22908, 21884, 20860, 19836, 18812, 17788, 16764,
66 15996, 15484, 14972, 14460, 13948, 13436, 12924, 12412,
67 11900, 11388, 10876, 10364, 9852, 9340, 8828, 8316,
68 7932, 7676, 7420, 7164, 6908, 6652, 6396, 6140,
69 5884, 5628, 5372, 5116, 4860, 4604, 4348, 4092,
70 3900, 3772, 3644, 3516, 3388, 3260, 3132, 3004,
71 2876, 2748, 2620, 2492, 2364, 2236, 2108, 1980,
72 1884, 1820, 1756, 1692, 1628, 1564, 1500, 1436,
73 1372, 1308, 1244, 1180, 1116, 1052, 988, 924,
74 876, 844, 812, 780, 748, 716, 684, 652,
75 620, 588, 556, 524, 492, 460, 428, 396,
76 372, 356, 340, 324, 308, 292, 276, 260,
77 244, 228, 212, 196, 180, 164, 148, 132,
78 120, 112, 104, 96, 88, 80, 72, 64,
79 56, 48, 40, 32, 24, 16, 8, 0 };
80
81 /* #define ZEROTRAP */ /* turn on the trap as per the MIL-STD */
82 #define BIAS 0x84 /* define the add-in bias for 16 bit samples */
83 #define CLIP 32635
84
85 static unsigned char
86 st_linear_to_ulaw(int sample)
87 {
88 static int exp_lut[256] = {0,0,1,1,2,2,2,2,3,3,3,3,3,3,3,3,
89 4,4,4,4,4,4,4,4,4,4,4,4,4,4,4,4,
90 5,5,5,5,5,5,5,5,5,5,5,5,5,5,5,5,
91 5,5,5,5,5,5,5,5,5,5,5,5,5,5,5,5,
92 6,6,6,6,6,6,6,6,6,6,6,6,6,6,6,6,
93 6,6,6,6,6,6,6,6,6,6,6,6,6,6,6,6,
94 6,6,6,6,6,6,6,6,6,6,6,6,6,6,6,6,
95 6,6,6,6,6,6,6,6,6,6,6,6,6,6,6,6,
96 7,7,7,7,7,7,7,7,7,7,7,7,7,7,7,7,
97 7,7,7,7,7,7,7,7,7,7,7,7,7,7,7,7,
98 7,7,7,7,7,7,7,7,7,7,7,7,7,7,7,7,
99 7,7,7,7,7,7,7,7,7,7,7,7,7,7,7,7,
100 7,7,7,7,7,7,7,7,7,7,7,7,7,7,7,7,
101 7,7,7,7,7,7,7,7,7,7,7,7,7,7,7,7,
102 7,7,7,7,7,7,7,7,7,7,7,7,7,7,7,7,
103 7,7,7,7,7,7,7,7,7,7,7,7,7,7,7,7};
104 int sign, exponent, mantissa;
105 unsigned char ulawbyte;
106
107 /* Get the sample into sign-magnitude. */
108 sign = (sample >> 8) & 0x80; /* set aside the sign */
109 if ( sign != 0 ) sample = -sample; /* get magnitude */
110 if ( sample > CLIP ) sample = CLIP; /* clip the magnitude */
111
112 /* Convert from 16 bit linear to ulaw. */
113 sample = sample + BIAS;
114 exponent = exp_lut[( sample >> 7 ) & 0xFF];
115 mantissa = ( sample >> ( exponent + 3 ) ) & 0x0F;
116 ulawbyte = ~ ( sign | ( exponent << 4 ) | mantissa );
117 #ifdef ZEROTRAP
118 if ( ulawbyte == 0 ) ulawbyte = 0x02; /* optional CCITT trap */
119 #endif
120
121 return ulawbyte;
122 }
123 /* End of code taken from sox */
124
125 /* Intel ADPCM step variation table */
126 static int indexTable[16] = {
127 -1, -1, -1, -1, 2, 4, 6, 8,
128 -1, -1, -1, -1, 2, 4, 6, 8,
129 };
130
131 static int stepsizeTable[89] = {
132 7, 8, 9, 10, 11, 12, 13, 14, 16, 17,
133 19, 21, 23, 25, 28, 31, 34, 37, 41, 45,
134 50, 55, 60, 66, 73, 80, 88, 97, 107, 118,
135 130, 143, 157, 173, 190, 209, 230, 253, 279, 307,
136 337, 371, 408, 449, 494, 544, 598, 658, 724, 796,
137 876, 963, 1060, 1166, 1282, 1411, 1552, 1707, 1878, 2066,
138 2272, 2499, 2749, 3024, 3327, 3660, 4026, 4428, 4871, 5358,
139 5894, 6484, 7132, 7845, 8630, 9493, 10442, 11487, 12635, 13899,
140 15289, 16818, 18500, 20350, 22385, 24623, 27086, 29794, 32767
141 };
142
143 #define CHARP(cp, i) ((signed char *)(cp+i))
144 #define SHORTP(cp, i) ((short *)(cp+i))
145 #define LONGP(cp, i) ((Py_Int32 *)(cp+i))
146
147
148
149 static PyObject *AudioopError;
150
151 static PyObject *
152 audioop_getsample(PyObject *self, PyObject *args)
153 {
154 signed char *cp;
155 int len, size, val = 0;
156 int i;
157
158 if ( !PyArg_Parse(args, "(s#ii)", &cp, &len, &size, &i) )
159 return 0;
160 if ( size != 1 && size != 2 && size != 4 ) {
161 PyErr_SetString(AudioopError, "Size should be 1, 2 or 4");
162 return 0;
163 }
164 if ( i < 0 || i >= len/size ) {
165 PyErr_SetString(AudioopError, "Index out of range");
166 return 0;
167 }
168 if ( size == 1 ) val = (int)*CHARP(cp, i);
169 else if ( size == 2 ) val = (int)*SHORTP(cp, i*2);
170 else if ( size == 4 ) val = (int)*LONGP(cp, i*4);
171 return PyInt_FromLong(val);
172 }
173
174 static PyObject *
175 audioop_max(PyObject *self, PyObject *args)
176 {
177 signed char *cp;
178 int len, size, val = 0;
179 int i;
180 int max = 0;
181
182 if ( !PyArg_Parse(args, "(s#i)", &cp, &len, &size) )
183 return 0;
184 if ( size != 1 && size != 2 && size != 4 ) {
185 PyErr_SetString(AudioopError, "Size should be 1, 2 or 4");
186 return 0;
187 }
188 for ( i=0; i<len; i+= size) {
189 if ( size == 1 ) val = (int)*CHARP(cp, i);
190 else if ( size == 2 ) val = (int)*SHORTP(cp, i);
191 else if ( size == 4 ) val = (int)*LONGP(cp, i);
192 if ( val < 0 ) val = (-val);
193 if ( val > max ) max = val;
194 }
195 return PyInt_FromLong(max);
196 }
197
198 static PyObject *
199 audioop_minmax(PyObject *self, PyObject *args)
200 {
201 signed char *cp;
202 int len, size, val = 0;
203 int i;
204 int min = 0x7fffffff, max = -0x7fffffff;
205
206 if (!PyArg_Parse(args, "(s#i)", &cp, &len, &size))
207 return NULL;
208 if (size != 1 && size != 2 && size != 4) {
209 PyErr_SetString(AudioopError, "Size should be 1, 2 or 4");
210 return NULL;
211 }
212 for (i = 0; i < len; i += size) {
213 if (size == 1) val = (int) *CHARP(cp, i);
214 else if (size == 2) val = (int) *SHORTP(cp, i);
215 else if (size == 4) val = (int) *LONGP(cp, i);
216 if (val > max) max = val;
217 if (val < min) min = val;
218 }
219 return Py_BuildValue("(ii)", min, max);
220 }
221
222 static PyObject *
223 audioop_avg(PyObject *self, PyObject *args)
224 {
225 signed char *cp;
226 int len, size, val = 0;
227 int i;
228 double avg = 0.0;
229
230 if ( !PyArg_Parse(args, "(s#i)", &cp, &len, &size) )
231 return 0;
232 if ( size != 1 && size != 2 && size != 4 ) {
233 PyErr_SetString(AudioopError, "Size should be 1, 2 or 4");
234 return 0;
235 }
236 for ( i=0; i<len; i+= size) {
237 if ( size == 1 ) val = (int)*CHARP(cp, i);
238 else if ( size == 2 ) val = (int)*SHORTP(cp, i);
239 else if ( size == 4 ) val = (int)*LONGP(cp, i);
240 avg += val;
241 }
242 if ( len == 0 )
243 val = 0;
244 else
245 val = (int)(avg / (double)(len/size));
246 return PyInt_FromLong(val);
247 }
248
249 static PyObject *
250 audioop_rms(PyObject *self, PyObject *args)
251 {
252 signed char *cp;
253 int len, size, val = 0;
254 int i;
255 double sum_squares = 0.0;
256
257 if ( !PyArg_Parse(args, "(s#i)", &cp, &len, &size) )
258 return 0;
259 if ( size != 1 && size != 2 && size != 4 ) {
260 PyErr_SetString(AudioopError, "Size should be 1, 2 or 4");
261 return 0;
262 }
263 for ( i=0; i<len; i+= size) {
264 if ( size == 1 ) val = (int)*CHARP(cp, i);
265 else if ( size == 2 ) val = (int)*SHORTP(cp, i);
266 else if ( size == 4 ) val = (int)*LONGP(cp, i);
267 sum_squares += (double)val*(double)val;
268 }
269 if ( len == 0 )
270 val = 0;
271 else
272 val = (int)sqrt(sum_squares / (double)(len/size));
273 return PyInt_FromLong(val);
274 }
275
276 static double _sum2(short *a, short *b, int len)
277 {
278 int i;
279 double sum = 0.0;
280
281 for( i=0; i<len; i++) {
282 sum = sum + (double)a[i]*(double)b[i];
283 }
284 return sum;
285 }
286
287 /*
288 ** Findfit tries to locate a sample within another sample. Its main use
289 ** is in echo-cancellation (to find the feedback of the output signal in
290 ** the input signal).
291 ** The method used is as follows:
292 **
293 ** let R be the reference signal (length n) and A the input signal (length N)
294 ** with N > n, and let all sums be over i from 0 to n-1.
295 **
296 ** Now, for each j in {0..N-n} we compute a factor fj so that -fj*R matches A
297 ** as good as possible, i.e. sum( (A[j+i]+fj*R[i])^2 ) is minimal. This
298 ** equation gives fj = sum( A[j+i]R[i] ) / sum(R[i]^2).
299 **
300 ** Next, we compute the relative distance between the original signal and
301 ** the modified signal and minimize that over j:
302 ** vj = sum( (A[j+i]-fj*R[i])^2 ) / sum( A[j+i]^2 ) =>
303 ** vj = ( sum(A[j+i]^2)*sum(R[i]^2) - sum(A[j+i]R[i])^2 ) / sum( A[j+i]^2 )
304 **
305 ** In the code variables correspond as follows:
306 ** cp1 A
307 ** cp2 R
308 ** len1 N
309 ** len2 n
310 ** aj_m1 A[j-1]
311 ** aj_lm1 A[j+n-1]
312 ** sum_ri_2 sum(R[i]^2)
313 ** sum_aij_2 sum(A[i+j]^2)
314 ** sum_aij_ri sum(A[i+j]R[i])
315 **
316 ** sum_ri is calculated once, sum_aij_2 is updated each step and sum_aij_ri
317 ** is completely recalculated each step.
318 */
319 static PyObject *
320 audioop_findfit(PyObject *self, PyObject *args)
321 {
322 short *cp1, *cp2;
323 int len1, len2;
324 int j, best_j;
325 double aj_m1, aj_lm1;
326 double sum_ri_2, sum_aij_2, sum_aij_ri, result, best_result, factor;
327
328 if ( !PyArg_Parse(args, "(s#s#)", &cp1, &len1, &cp2, &len2) )
329 return 0;
330 if ( len1 & 1 || len2 & 1 ) {
331 PyErr_SetString(AudioopError, "Strings should be even-sized");
332 return 0;
333 }
334 len1 >>= 1;
335 len2 >>= 1;
336
337 if ( len1 < len2 ) {
338 PyErr_SetString(AudioopError, "First sample should be longer");
339 return 0;
340 }
341 sum_ri_2 = _sum2(cp2, cp2, len2);
342 sum_aij_2 = _sum2(cp1, cp1, len2);
343 sum_aij_ri = _sum2(cp1, cp2, len2);
344
345 result = (sum_ri_2*sum_aij_2 - sum_aij_ri*sum_aij_ri) / sum_aij_2;
346
347 best_result = result;
348 best_j = 0;
349 j = 0;
350
351 for ( j=1; j<=len1-len2; j++) {
352 aj_m1 = (double)cp1[j-1];
353 aj_lm1 = (double)cp1[j+len2-1];
354
355 sum_aij_2 = sum_aij_2 + aj_lm1*aj_lm1 - aj_m1*aj_m1;
356 sum_aij_ri = _sum2(cp1+j, cp2, len2);
357
358 result = (sum_ri_2*sum_aij_2 - sum_aij_ri*sum_aij_ri)
359 / sum_aij_2;
360
361 if ( result < best_result ) {
362 best_result = result;
363 best_j = j;
364 }
365
366 }
367
368 factor = _sum2(cp1+best_j, cp2, len2) / sum_ri_2;
369
370 return Py_BuildValue("(if)", best_j, factor);
371 }
372
373 /*
374 ** findfactor finds a factor f so that the energy in A-fB is minimal.
375 ** See the comment for findfit for details.
376 */
377 static PyObject *
378 audioop_findfactor(PyObject *self, PyObject *args)
379 {
380 short *cp1, *cp2;
381 int len1, len2;
382 double sum_ri_2, sum_aij_ri, result;
383
384 if ( !PyArg_Parse(args, "(s#s#)", &cp1, &len1, &cp2, &len2) )
385 return 0;
386 if ( len1 & 1 || len2 & 1 ) {
387 PyErr_SetString(AudioopError, "Strings should be even-sized");
388 return 0;
389 }
390 if ( len1 != len2 ) {
391 PyErr_SetString(AudioopError, "Samples should be same size");
392 return 0;
393 }
394 len2 >>= 1;
395 sum_ri_2 = _sum2(cp2, cp2, len2);
396 sum_aij_ri = _sum2(cp1, cp2, len2);
397
398 result = sum_aij_ri / sum_ri_2;
399
400 return PyFloat_FromDouble(result);
401 }
402
403 /*
404 ** findmax returns the index of the n-sized segment of the input sample
405 ** that contains the most energy.
406 */
407 static PyObject *
408 audioop_findmax(PyObject *self, PyObject *args)
409 {
410 short *cp1;
411 int len1, len2;
412 int j, best_j;
413 double aj_m1, aj_lm1;
414 double result, best_result;
415
416 if ( !PyArg_Parse(args, "(s#i)", &cp1, &len1, &len2) )
417 return 0;
418 if ( len1 & 1 ) {
419 PyErr_SetString(AudioopError, "Strings should be even-sized");
420 return 0;
421 }
422 len1 >>= 1;
423
424 if ( len1 < len2 ) {
425 PyErr_SetString(AudioopError, "Input sample should be longer");
426 return 0;
427 }
428
429 result = _sum2(cp1, cp1, len2);
430
431 best_result = result;
432 best_j = 0;
433 j = 0;
434
435 for ( j=1; j<=len1-len2; j++) {
436 aj_m1 = (double)cp1[j-1];
437 aj_lm1 = (double)cp1[j+len2-1];
438
439 result = result + aj_lm1*aj_lm1 - aj_m1*aj_m1;
440
441 if ( result > best_result ) {
442 best_result = result;
443 best_j = j;
444 }
445
446 }
447
448 return PyInt_FromLong(best_j);
449 }
450
451 static PyObject *
452 audioop_avgpp(PyObject *self, PyObject *args)
453 {
454 signed char *cp;
455 int len, size, val = 0, prevval = 0, prevextremevalid = 0,
456 prevextreme = 0;
457 int i;
458 double avg = 0.0;
459 int diff, prevdiff, extremediff, nextreme = 0;
460
461 if ( !PyArg_Parse(args, "(s#i)", &cp, &len, &size) )
462 return 0;
463 if ( size != 1 && size != 2 && size != 4 ) {
464 PyErr_SetString(AudioopError, "Size should be 1, 2 or 4");
465 return 0;
466 }
467 /* Compute first delta value ahead. Also automatically makes us
468 ** skip the first extreme value
469 */
470 if ( size == 1 ) prevval = (int)*CHARP(cp, 0);
471 else if ( size == 2 ) prevval = (int)*SHORTP(cp, 0);
472 else if ( size == 4 ) prevval = (int)*LONGP(cp, 0);
473 if ( size == 1 ) val = (int)*CHARP(cp, size);
474 else if ( size == 2 ) val = (int)*SHORTP(cp, size);
475 else if ( size == 4 ) val = (int)*LONGP(cp, size);
476 prevdiff = val - prevval;
477
478 for ( i=size; i<len; i+= size) {
479 if ( size == 1 ) val = (int)*CHARP(cp, i);
480 else if ( size == 2 ) val = (int)*SHORTP(cp, i);
481 else if ( size == 4 ) val = (int)*LONGP(cp, i);
482 diff = val - prevval;
483 if ( diff*prevdiff < 0 ) {
484 /* Derivative changed sign. Compute difference to last
485 ** extreme value and remember.
486 */
487 if ( prevextremevalid ) {
488 extremediff = prevval - prevextreme;
489 if ( extremediff < 0 )
490 extremediff = -extremediff;
491 avg += extremediff;
492 nextreme++;
493 }
494 prevextremevalid = 1;
495 prevextreme = prevval;
496 }
497 prevval = val;
498 if ( diff != 0 )
499 prevdiff = diff;
500 }
501 if ( nextreme == 0 )
502 val = 0;
503 else
504 val = (int)(avg / (double)nextreme);
505 return PyInt_FromLong(val);
506 }
507
508 static PyObject *
509 audioop_maxpp(PyObject *self, PyObject *args)
510 {
511 signed char *cp;
512 int len, size, val = 0, prevval = 0, prevextremevalid = 0,
513 prevextreme = 0;
514 int i;
515 int max = 0;
516 int diff, prevdiff, extremediff;
517
518 if ( !PyArg_Parse(args, "(s#i)", &cp, &len, &size) )
519 return 0;
520 if ( size != 1 && size != 2 && size != 4 ) {
521 PyErr_SetString(AudioopError, "Size should be 1, 2 or 4");
522 return 0;
523 }
524 /* Compute first delta value ahead. Also automatically makes us
525 ** skip the first extreme value
526 */
527 if ( size == 1 ) prevval = (int)*CHARP(cp, 0);
528 else if ( size == 2 ) prevval = (int)*SHORTP(cp, 0);
529 else if ( size == 4 ) prevval = (int)*LONGP(cp, 0);
530 if ( size == 1 ) val = (int)*CHARP(cp, size);
531 else if ( size == 2 ) val = (int)*SHORTP(cp, size);
532 else if ( size == 4 ) val = (int)*LONGP(cp, size);
533 prevdiff = val - prevval;
534
535 for ( i=size; i<len; i+= size) {
536 if ( size == 1 ) val = (int)*CHARP(cp, i);
537 else if ( size == 2 ) val = (int)*SHORTP(cp, i);
538 else if ( size == 4 ) val = (int)*LONGP(cp, i);
539 diff = val - prevval;
540 if ( diff*prevdiff < 0 ) {
541 /* Derivative changed sign. Compute difference to
542 ** last extreme value and remember.
543 */
544 if ( prevextremevalid ) {
545 extremediff = prevval - prevextreme;
546 if ( extremediff < 0 )
547 extremediff = -extremediff;
548 if ( extremediff > max )
549 max = extremediff;
550 }
551 prevextremevalid = 1;
552 prevextreme = prevval;
553 }
554 prevval = val;
555 if ( diff != 0 )
556 prevdiff = diff;
557 }
558 return PyInt_FromLong(max);
559 }
560
561 static PyObject *
562 audioop_cross(PyObject *self, PyObject *args)
563 {
564 signed char *cp;
565 int len, size, val = 0;
566 int i;
567 int prevval, ncross;
568
569 if ( !PyArg_Parse(args, "(s#i)", &cp, &len, &size) )
570 return 0;
571 if ( size != 1 && size != 2 && size != 4 ) {
572 PyErr_SetString(AudioopError, "Size should be 1, 2 or 4");
573 return 0;
574 }
575 ncross = -1;
576 prevval = 17; /* Anything <> 0,1 */
577 for ( i=0; i<len; i+= size) {
578 if ( size == 1 ) val = ((int)*CHARP(cp, i)) >> 7;
579 else if ( size == 2 ) val = ((int)*SHORTP(cp, i)) >> 15;
580 else if ( size == 4 ) val = ((int)*LONGP(cp, i)) >> 31;
581 val = val & 1;
582 if ( val != prevval ) ncross++;
583 prevval = val;
584 }
585 return PyInt_FromLong(ncross);
586 }
587
588 static PyObject *
589 audioop_mul(PyObject *self, PyObject *args)
590 {
591 signed char *cp, *ncp;
592 int len, size, val = 0;
593 double factor, fval, maxval;
594 PyObject *rv;
595 int i;
596
597 if ( !PyArg_Parse(args, "(s#id)", &cp, &len, &size, &factor ) )
598 return 0;
599
600 if ( size == 1 ) maxval = (double) 0x7f;
601 else if ( size == 2 ) maxval = (double) 0x7fff;
602 else if ( size == 4 ) maxval = (double) 0x7fffffff;
603 else {
604 PyErr_SetString(AudioopError, "Size should be 1, 2 or 4");
605 return 0;
606 }
607
608 rv = PyString_FromStringAndSize(NULL, len);
609 if ( rv == 0 )
610 return 0;
611 ncp = (signed char *)PyString_AsString(rv);
612
613
614 for ( i=0; i < len; i += size ) {
615 if ( size == 1 ) val = (int)*CHARP(cp, i);
616 else if ( size == 2 ) val = (int)*SHORTP(cp, i);
617 else if ( size == 4 ) val = (int)*LONGP(cp, i);
618 fval = (double)val*factor;
619 if ( fval > maxval ) fval = maxval;
620 else if ( fval < -maxval ) fval = -maxval;
621 val = (int)fval;
622 if ( size == 1 ) *CHARP(ncp, i) = (signed char)val;
623 else if ( size == 2 ) *SHORTP(ncp, i) = (short)val;
624 else if ( size == 4 ) *LONGP(ncp, i) = (Py_Int32)val;
625 }
626 return rv;
627 }
628
629 static PyObject *
630 audioop_tomono(PyObject *self, PyObject *args)
631 {
632 signed char *cp, *ncp;
633 int len, size, val1 = 0, val2 = 0;
634 double fac1, fac2, fval, maxval;
635 PyObject *rv;
636 int i;
637
638 if ( !PyArg_Parse(args, "(s#idd)", &cp, &len, &size, &fac1, &fac2 ) )
639 return 0;
640
641 if ( size == 1 ) maxval = (double) 0x7f;
642 else if ( size == 2 ) maxval = (double) 0x7fff;
643 else if ( size == 4 ) maxval = (double) 0x7fffffff;
644 else {
645 PyErr_SetString(AudioopError, "Size should be 1, 2 or 4");
646 return 0;
647 }
648
649 rv = PyString_FromStringAndSize(NULL, len/2);
650 if ( rv == 0 )
651 return 0;
652 ncp = (signed char *)PyString_AsString(rv);
653
654
655 for ( i=0; i < len; i += size*2 ) {
656 if ( size == 1 ) val1 = (int)*CHARP(cp, i);
657 else if ( size == 2 ) val1 = (int)*SHORTP(cp, i);
658 else if ( size == 4 ) val1 = (int)*LONGP(cp, i);
659 if ( size == 1 ) val2 = (int)*CHARP(cp, i+1);
660 else if ( size == 2 ) val2 = (int)*SHORTP(cp, i+2);
661 else if ( size == 4 ) val2 = (int)*LONGP(cp, i+4);
662 fval = (double)val1*fac1 + (double)val2*fac2;
663 if ( fval > maxval ) fval = maxval;
664 else if ( fval < -maxval ) fval = -maxval;
665 val1 = (int)fval;
666 if ( size == 1 ) *CHARP(ncp, i/2) = (signed char)val1;
667 else if ( size == 2 ) *SHORTP(ncp, i/2) = (short)val1;
668 else if ( size == 4 ) *LONGP(ncp, i/2)= (Py_Int32)val1;
669 }
670 return rv;
671 }
672
673 static PyObject *
674 audioop_tostereo(PyObject *self, PyObject *args)
675 {
676 signed char *cp, *ncp;
677 int len, size, val1, val2, val = 0;
678 double fac1, fac2, fval, maxval;
679 PyObject *rv;
680 int i;
681
682 if ( !PyArg_Parse(args, "(s#idd)", &cp, &len, &size, &fac1, &fac2 ) )
683 return 0;
684
685 if ( size == 1 ) maxval = (double) 0x7f;
686 else if ( size == 2 ) maxval = (double) 0x7fff;
687 else if ( size == 4 ) maxval = (double) 0x7fffffff;
688 else {
689 PyErr_SetString(AudioopError, "Size should be 1, 2 or 4");
690 return 0;
691 }
692
693 rv = PyString_FromStringAndSize(NULL, len*2);
694 if ( rv == 0 )
695 return 0;
696 ncp = (signed char *)PyString_AsString(rv);
697
698
699 for ( i=0; i < len; i += size ) {
700 if ( size == 1 ) val = (int)*CHARP(cp, i);
701 else if ( size == 2 ) val = (int)*SHORTP(cp, i);
702 else if ( size == 4 ) val = (int)*LONGP(cp, i);
703
704 fval = (double)val*fac1;
705 if ( fval > maxval ) fval = maxval;
706 else if ( fval < -maxval ) fval = -maxval;
707 val1 = (int)fval;
708
709 fval = (double)val*fac2;
710 if ( fval > maxval ) fval = maxval;
711 else if ( fval < -maxval ) fval = -maxval;
712 val2 = (int)fval;
713
714 if ( size == 1 ) *CHARP(ncp, i*2) = (signed char)val1;
715 else if ( size == 2 ) *SHORTP(ncp, i*2) = (short)val1;
716 else if ( size == 4 ) *LONGP(ncp, i*2) = (Py_Int32)val1;
717
718 if ( size == 1 ) *CHARP(ncp, i*2+1) = (signed char)val2;
719 else if ( size == 2 ) *SHORTP(ncp, i*2+2) = (short)val2;
720 else if ( size == 4 ) *LONGP(ncp, i*2+4) = (Py_Int32)val2;
721 }
722 return rv;
723 }
724
725 static PyObject *
726 audioop_add(PyObject *self, PyObject *args)
727 {
728 signed char *cp1, *cp2, *ncp;
729 int len1, len2, size, val1 = 0, val2 = 0, maxval, newval;
730 PyObject *rv;
731 int i;
732
733 if ( !PyArg_Parse(args, "(s#s#i)",
734 &cp1, &len1, &cp2, &len2, &size ) )
735 return 0;
736
737 if ( len1 != len2 ) {
738 PyErr_SetString(AudioopError, "Lengths should be the same");
739 return 0;
740 }
741
742 if ( size == 1 ) maxval = 0x7f;
743 else if ( size == 2 ) maxval = 0x7fff;
744 else if ( size == 4 ) maxval = 0x7fffffff;
745 else {
746 PyErr_SetString(AudioopError, "Size should be 1, 2 or 4");
747 return 0;
748 }
749
750 rv = PyString_FromStringAndSize(NULL, len1);
751 if ( rv == 0 )
752 return 0;
753 ncp = (signed char *)PyString_AsString(rv);
754
755 for ( i=0; i < len1; i += size ) {
756 if ( size == 1 ) val1 = (int)*CHARP(cp1, i);
757 else if ( size == 2 ) val1 = (int)*SHORTP(cp1, i);
758 else if ( size == 4 ) val1 = (int)*LONGP(cp1, i);
759
760 if ( size == 1 ) val2 = (int)*CHARP(cp2, i);
761 else if ( size == 2 ) val2 = (int)*SHORTP(cp2, i);
762 else if ( size == 4 ) val2 = (int)*LONGP(cp2, i);
763
764 newval = val1 + val2;
765 /* truncate in case of overflow */
766 if (newval > maxval) newval = maxval;
767 else if (newval < -maxval) newval = -maxval;
768 else if (size == 4 && (newval^val1) < 0 && (newval^val2) < 0)
769 newval = val1 > 0 ? maxval : - maxval;
770
771 if ( size == 1 ) *CHARP(ncp, i) = (signed char)newval;
772 else if ( size == 2 ) *SHORTP(ncp, i) = (short)newval;
773 else if ( size == 4 ) *LONGP(ncp, i) = (Py_Int32)newval;
774 }
775 return rv;
776 }
777
778 static PyObject *
779 audioop_bias(PyObject *self, PyObject *args)
780 {
781 signed char *cp, *ncp;
782 int len, size, val = 0;
783 PyObject *rv;
784 int i;
785 int bias;
786
787 if ( !PyArg_Parse(args, "(s#ii)",
788 &cp, &len, &size , &bias) )
789 return 0;
790
791 if ( size != 1 && size != 2 && size != 4) {
792 PyErr_SetString(AudioopError, "Size should be 1, 2 or 4");
793 return 0;
794 }
795
796 rv = PyString_FromStringAndSize(NULL, len);
797 if ( rv == 0 )
798 return 0;
799 ncp = (signed char *)PyString_AsString(rv);
800
801
802 for ( i=0; i < len; i += size ) {
803 if ( size == 1 ) val = (int)*CHARP(cp, i);
804 else if ( size == 2 ) val = (int)*SHORTP(cp, i);
805 else if ( size == 4 ) val = (int)*LONGP(cp, i);
806
807 if ( size == 1 ) *CHARP(ncp, i) = (signed char)(val+bias);
808 else if ( size == 2 ) *SHORTP(ncp, i) = (short)(val+bias);
809 else if ( size == 4 ) *LONGP(ncp, i) = (Py_Int32)(val+bias);
810 }
811 return rv;
812 }
813
814 static PyObject *
815 audioop_reverse(PyObject *self, PyObject *args)
816 {
817 signed char *cp;
818 unsigned char *ncp;
819 int len, size, val = 0;
820 PyObject *rv;
821 int i, j;
822
823 if ( !PyArg_Parse(args, "(s#i)",
824 &cp, &len, &size) )
825 return 0;
826
827 if ( size != 1 && size != 2 && size != 4 ) {
828 PyErr_SetString(AudioopError, "Size should be 1, 2 or 4");
829 return 0;
830 }
831
832 rv = PyString_FromStringAndSize(NULL, len);
833 if ( rv == 0 )
834 return 0;
835 ncp = (unsigned char *)PyString_AsString(rv);
836
837 for ( i=0; i < len; i += size ) {
838 if ( size == 1 ) val = ((int)*CHARP(cp, i)) << 8;
839 else if ( size == 2 ) val = (int)*SHORTP(cp, i);
840 else if ( size == 4 ) val = ((int)*LONGP(cp, i)) >> 16;
841
842 j = len - i - size;
843
844 if ( size == 1 ) *CHARP(ncp, j) = (signed char)(val >> 8);
845 else if ( size == 2 ) *SHORTP(ncp, j) = (short)(val);
846 else if ( size == 4 ) *LONGP(ncp, j) = (Py_Int32)(val<<16);
847 }
848 return rv;
849 }
850
851 static PyObject *
852 audioop_lin2lin(PyObject *self, PyObject *args)
853 {
854 signed char *cp;
855 unsigned char *ncp;
856 int len, size, size2, val = 0;
857 PyObject *rv;
858 int i, j;
859
860 if ( !PyArg_Parse(args, "(s#ii)",
861 &cp, &len, &size, &size2) )
862 return 0;
863
864 if ( (size != 1 && size != 2 && size != 4) ||
865 (size2 != 1 && size2 != 2 && size2 != 4)) {
866 PyErr_SetString(AudioopError, "Size should be 1, 2 or 4");
867 return 0;
868 }
869
870 rv = PyString_FromStringAndSize(NULL, (len/size)*size2);
871 if ( rv == 0 )
872 return 0;
873 ncp = (unsigned char *)PyString_AsString(rv);
874
875 for ( i=0, j=0; i < len; i += size, j += size2 ) {
876 if ( size == 1 ) val = ((int)*CHARP(cp, i)) << 8;
877 else if ( size == 2 ) val = (int)*SHORTP(cp, i);
878 else if ( size == 4 ) val = ((int)*LONGP(cp, i)) >> 16;
879
880 if ( size2 == 1 ) *CHARP(ncp, j) = (signed char)(val >> 8);
881 else if ( size2 == 2 ) *SHORTP(ncp, j) = (short)(val);
882 else if ( size2 == 4 ) *LONGP(ncp, j) = (Py_Int32)(val<<16);
883 }
884 return rv;
885 }
886
887 static int
888 gcd(int a, int b)
889 {
890 while (b > 0) {
891 int tmp = a % b;
892 a = b;
893 b = tmp;
894 }
895 return a;
896 }
897
898 static PyObject *
899 audioop_ratecv(PyObject *self, PyObject *args)
900 {
901 char *cp, *ncp;
902 int len, size, nchannels, inrate, outrate, weightA, weightB;
903 int chan, d, *prev_i, *cur_i, cur_o;
904 PyObject *state, *samps, *str, *rv = NULL;
905 int bytes_per_frame;
906
907 weightA = 1;
908 weightB = 0;
909 if (!PyArg_ParseTuple(args, "s#iiiiO|ii:ratecv", &cp, &len, &size, &nchannels,
910 &inrate, &outrate, &state, &weightA, &weightB))
911 return NULL;
912 if (size != 1 && size != 2 && size != 4) {
913 PyErr_SetString(AudioopError, "Size should be 1, 2 or 4");
914 return NULL;
915 }
916 if (nchannels < 1) {
917 PyErr_SetString(AudioopError, "# of channels should be >= 1");
918 return NULL;
919 }
920 bytes_per_frame = size * nchannels;
921 if (bytes_per_frame / nchannels != size) {
922 /* This overflow test is rigorously correct because
923 both multiplicands are >= 1. Use the argument names
924 from the docs for the error msg. */
925 PyErr_SetString(PyExc_OverflowError,
926 "width * nchannels too big for a C int");
927 return NULL;
928 }
929 if (weightA < 1 || weightB < 0) {
930 PyErr_SetString(AudioopError,
931 "weightA should be >= 1, weightB should be >= 0");
932 return NULL;
933 }
934 if (len % bytes_per_frame != 0) {
935 PyErr_SetString(AudioopError, "not a whole number of frames");
936 return NULL;
937 }
938 if (inrate <= 0 || outrate <= 0) {
939 PyErr_SetString(AudioopError, "sampling rate not > 0");
940 return NULL;
941 }
942 /* divide inrate and outrate by their greatest common divisor */
943 d = gcd(inrate, outrate);
944 inrate /= d;
945 outrate /= d;
946
947 prev_i = (int *) malloc(nchannels * sizeof(int));
948 cur_i = (int *) malloc(nchannels * sizeof(int));
949 if (prev_i == NULL || cur_i == NULL) {
950 (void) PyErr_NoMemory();
951 goto exit;
952 }
953
954 len /= bytes_per_frame; /* # of frames */
955
956 if (state == Py_None) {
957 d = -outrate;
958 for (chan = 0; chan < nchannels; chan++)
959 prev_i[chan] = cur_i[chan] = 0;
960 }
961 else {
962 if (!PyArg_ParseTuple(state,
963 "iO!;audioop.ratecv: illegal state argument",
964 &d, &PyTuple_Type, &samps))
965 goto exit;
966 if (PyTuple_Size(samps) != nchannels) {
967 PyErr_SetString(AudioopError,
968 "illegal state argument");
969 goto exit;
970 }
971 for (chan = 0; chan < nchannels; chan++) {
972 if (!PyArg_ParseTuple(PyTuple_GetItem(samps, chan),
973 "ii:ratecv",&prev_i[chan],&cur_i[chan]))
974 goto exit;
975 }
976 }
977
978 /* str <- Space for the output buffer. */
979 {
980 /* There are len input frames, so we need (mathematically)
981 ceiling(len*outrate/inrate) output frames, and each frame
982 requires bytes_per_frame bytes. Computing this
983 without spurious overflow is the challenge; we can
984 settle for a reasonable upper bound, though. */
985 int ceiling; /* the number of output frames */
986 int nbytes; /* the number of output bytes needed */
987 int q = len / inrate;
988 /* Now len = q * inrate + r exactly (with r = len % inrate),
989 and this is less than q * inrate + inrate = (q+1)*inrate.
990 So a reasonable upper bound on len*outrate/inrate is
991 ((q+1)*inrate)*outrate/inrate =
992 (q+1)*outrate.
993 */
994 ceiling = (q+1) * outrate;
995 nbytes = ceiling * bytes_per_frame;
996 /* See whether anything overflowed; if not, get the space. */
997 if (q+1 < 0 ||
998 ceiling / outrate != q+1 ||
999 nbytes / bytes_per_frame != ceiling)
1000 str = NULL;
1001 else
1002 str = PyString_FromStringAndSize(NULL, nbytes);
1003
1004 if (str == NULL) {
1005 PyErr_SetString(PyExc_MemoryError,
1006 "not enough memory for output buffer");
1007 goto exit;
1008 }
1009 }
1010 ncp = PyString_AsString(str);
1011
1012 for (;;) {
1013 while (d < 0) {
1014 if (len == 0) {
1015 samps = PyTuple_New(nchannels);
1016 for (chan = 0; chan < nchannels; chan++)
1017 PyTuple_SetItem(samps, chan,
1018 Py_BuildValue("(ii)",
1019 prev_i[chan],
1020 cur_i[chan]));
1021 if (PyErr_Occurred())
1022 goto exit;
1023 len = ncp - PyString_AsString(str);
1024 if (len == 0) {
1025 /*don't want to resize to zero length*/
1026 rv = PyString_FromStringAndSize("", 0);
1027 Py_DECREF(str);
1028 str = rv;
1029 } else if (_PyString_Resize(&str, len) < 0)
1030 goto exit;
1031 rv = Py_BuildValue("(O(iO))", str, d, samps);
1032 Py_DECREF(samps);
1033 Py_DECREF(str);
1034 goto exit; /* return rv */
1035 }
1036 for (chan = 0; chan < nchannels; chan++) {
1037 prev_i[chan] = cur_i[chan];
1038 if (size == 1)
1039 cur_i[chan] = ((int)*CHARP(cp, 0)) << 8;
1040 else if (size == 2)
1041 cur_i[chan] = (int)*SHORTP(cp, 0);
1042 else if (size == 4)
1043 cur_i[chan] = ((int)*LONGP(cp, 0)) >> 16;
1044 cp += size;
1045 /* implements a simple digital filter */
1046 cur_i[chan] =
1047 (weightA * cur_i[chan] +
1048 weightB * prev_i[chan]) /
1049 (weightA + weightB);
1050 }
1051 len--;
1052 d += outrate;
1053 }
1054 while (d >= 0) {
1055 for (chan = 0; chan < nchannels; chan++) {
1056 cur_o = (prev_i[chan] * d +
1057 cur_i[chan] * (outrate - d)) /
1058 outrate;
1059 if (size == 1)
1060 *CHARP(ncp, 0) = (signed char)(cur_o >> 8);
1061 else if (size == 2)
1062 *SHORTP(ncp, 0) = (short)(cur_o);
1063 else if (size == 4)
1064 *LONGP(ncp, 0) = (Py_Int32)(cur_o<<16);
1065 ncp += size;
1066 }
1067 d -= inrate;
1068 }
1069 }
1070 exit:
1071 if (prev_i != NULL)
1072 free(prev_i);
1073 if (cur_i != NULL)
1074 free(cur_i);
1075 return rv;
1076 }
1077
1078 static PyObject *
1079 audioop_lin2ulaw(PyObject *self, PyObject *args)
1080 {
1081 signed char *cp;
1082 unsigned char *ncp;
1083 int len, size, val = 0;
1084 PyObject *rv;
1085 int i;
1086
1087 if ( !PyArg_Parse(args, "(s#i)",
1088 &cp, &len, &size) )
1089 return 0;
1090
1091 if ( size != 1 && size != 2 && size != 4) {
1092 PyErr_SetString(AudioopError, "Size should be 1, 2 or 4");
1093 return 0;
1094 }
1095
1096 rv = PyString_FromStringAndSize(NULL, len/size);
1097 if ( rv == 0 )
1098 return 0;
1099 ncp = (unsigned char *)PyString_AsString(rv);
1100
1101 for ( i=0; i < len; i += size ) {
1102 if ( size == 1 ) val = ((int)*CHARP(cp, i)) << 8;
1103 else if ( size == 2 ) val = (int)*SHORTP(cp, i);
1104 else if ( size == 4 ) val = ((int)*LONGP(cp, i)) >> 16;
1105
1106 *ncp++ = st_linear_to_ulaw(val);
1107 }
1108 return rv;
1109 }
1110
1111 static PyObject *
1112 audioop_ulaw2lin(PyObject *self, PyObject *args)
1113 {
1114 unsigned char *cp;
1115 unsigned char cval;
1116 signed char *ncp;
1117 int len, size, val;
1118 PyObject *rv;
1119 int i;
1120
1121 if ( !PyArg_Parse(args, "(s#i)",
1122 &cp, &len, &size) )
1123 return 0;
1124
1125 if ( size != 1 && size != 2 && size != 4) {
1126 PyErr_SetString(AudioopError, "Size should be 1, 2 or 4");
1127 return 0;
1128 }
1129
1130 rv = PyString_FromStringAndSize(NULL, len*size);
1131 if ( rv == 0 )
1132 return 0;
1133 ncp = (signed char *)PyString_AsString(rv);
1134
1135 for ( i=0; i < len*size; i += size ) {
1136 cval = *cp++;
1137 val = st_ulaw_to_linear(cval);
1138
1139 if ( size == 1 ) *CHARP(ncp, i) = (signed char)(val >> 8);
1140 else if ( size == 2 ) *SHORTP(ncp, i) = (short)(val);
1141 else if ( size == 4 ) *LONGP(ncp, i) = (Py_Int32)(val<<16);
1142 }
1143 return rv;
1144 }
1145
1146 static PyObject *
1147 audioop_lin2adpcm(PyObject *self, PyObject *args)
1148 {
1149 signed char *cp;
1150 signed char *ncp;
1151 int len, size, val = 0, step, valpred, delta,
1152 index, sign, vpdiff, diff;
1153 PyObject *rv, *state, *str;
1154 int i, outputbuffer = 0, bufferstep;
1155
1156 if ( !PyArg_Parse(args, "(s#iO)",
1157 &cp, &len, &size, &state) )
1158 return 0;
1159
1160
1161 if ( size != 1 && size != 2 && size != 4) {
1162 PyErr_SetString(AudioopError, "Size should be 1, 2 or 4");
1163 return 0;
1164 }
1165
1166 str = PyString_FromStringAndSize(NULL, len/(size*2));
1167 if ( str == 0 )
1168 return 0;
1169 ncp = (signed char *)PyString_AsString(str);
1170
1171 /* Decode state, should have (value, step) */
1172 if ( state == Py_None ) {
1173 /* First time, it seems. Set defaults */
1174 valpred = 0;
1175 step = 7;
1176 index = 0;
1177 } else if ( !PyArg_Parse(state, "(ii)", &valpred, &index) )
1178 return 0;
1179
1180 step = stepsizeTable[index];
1181 bufferstep = 1;
1182
1183 for ( i=0; i < len; i += size ) {
1184 if ( size == 1 ) val = ((int)*CHARP(cp, i)) << 8;
1185 else if ( size == 2 ) val = (int)*SHORTP(cp, i);
1186 else if ( size == 4 ) val = ((int)*LONGP(cp, i)) >> 16;
1187
1188 /* Step 1 - compute difference with previous value */
1189 diff = val - valpred;
1190 sign = (diff < 0) ? 8 : 0;
1191 if ( sign ) diff = (-diff);
1192
1193 /* Step 2 - Divide and clamp */
1194 /* Note:
1195 ** This code *approximately* computes:
1196 ** delta = diff*4/step;
1197 ** vpdiff = (delta+0.5)*step/4;
1198 ** but in shift step bits are dropped. The net result of this
1199 ** is that even if you have fast mul/div hardware you cannot
1200 ** put it to good use since the fixup would be too expensive.
1201 */
1202 delta = 0;
1203 vpdiff = (step >> 3);
1204
1205 if ( diff >= step ) {
1206 delta = 4;
1207 diff -= step;
1208 vpdiff += step;
1209 }
1210 step >>= 1;
1211 if ( diff >= step ) {
1212 delta |= 2;
1213 diff -= step;
1214 vpdiff += step;
1215 }
1216 step >>= 1;
1217 if ( diff >= step ) {
1218 delta |= 1;
1219 vpdiff += step;
1220 }
1221
1222 /* Step 3 - Update previous value */
1223 if ( sign )
1224 valpred -= vpdiff;
1225 else
1226 valpred += vpdiff;
1227
1228 /* Step 4 - Clamp previous value to 16 bits */
1229 if ( valpred > 32767 )
1230 valpred = 32767;
1231 else if ( valpred < -32768 )
1232 valpred = -32768;
1233
1234 /* Step 5 - Assemble value, update index and step values */
1235 delta |= sign;
1236
1237 index += indexTable[delta];
1238 if ( index < 0 ) index = 0;
1239 if ( index > 88 ) index = 88;
1240 step = stepsizeTable[index];
1241
1242 /* Step 6 - Output value */
1243 if ( bufferstep ) {
1244 outputbuffer = (delta << 4) & 0xf0;
1245 } else {
1246 *ncp++ = (delta & 0x0f) | outputbuffer;
1247 }
1248 bufferstep = !bufferstep;
1249 }
1250 rv = Py_BuildValue("(O(ii))", str, valpred, index);
1251 Py_DECREF(str);
1252 return rv;
1253 }
1254
1255 static PyObject *
1256 audioop_adpcm2lin(PyObject *self, PyObject *args)
1257 {
1258 signed char *cp;
1259 signed char *ncp;
1260 int len, size, valpred, step, delta, index, sign, vpdiff;
1261 PyObject *rv, *str, *state;
1262 int i, inputbuffer = 0, bufferstep;
1263
1264 if ( !PyArg_Parse(args, "(s#iO)",
1265 &cp, &len, &size, &state) )
1266 return 0;
1267
1268 if ( size != 1 && size != 2 && size != 4) {
1269 PyErr_SetString(AudioopError, "Size should be 1, 2 or 4");
1270 return 0;
1271 }
1272
1273 /* Decode state, should have (value, step) */
1274 if ( state == Py_None ) {
1275 /* First time, it seems. Set defaults */
1276 valpred = 0;
1277 step = 7;
1278 index = 0;
1279 } else if ( !PyArg_Parse(state, "(ii)", &valpred, &index) )
1280 return 0;
1281
1282 str = PyString_FromStringAndSize(NULL, len*size*2);
1283 if ( str == 0 )
1284 return 0;
1285 ncp = (signed char *)PyString_AsString(str);
1286
1287 step = stepsizeTable[index];
1288 bufferstep = 0;
1289
1290 for ( i=0; i < len*size*2; i += size ) {
1291 /* Step 1 - get the delta value and compute next index */
1292 if ( bufferstep ) {
1293 delta = inputbuffer & 0xf;
1294 } else {
1295 inputbuffer = *cp++;
1296 delta = (inputbuffer >> 4) & 0xf;
1297 }
1298
1299 bufferstep = !bufferstep;
1300
1301 /* Step 2 - Find new index value (for later) */
1302 index += indexTable[delta];
1303 if ( index < 0 ) index = 0;
1304 if ( index > 88 ) index = 88;
1305
1306 /* Step 3 - Separate sign and magnitude */
1307 sign = delta & 8;
1308 delta = delta & 7;
1309
1310 /* Step 4 - Compute difference and new predicted value */
1311 /*
1312 ** Computes 'vpdiff = (delta+0.5)*step/4', but see comment
1313 ** in adpcm_coder.
1314 */
1315 vpdiff = step >> 3;
1316 if ( delta & 4 ) vpdiff += step;
1317 if ( delta & 2 ) vpdiff += step>>1;
1318 if ( delta & 1 ) vpdiff += step>>2;
1319
1320 if ( sign )
1321 valpred -= vpdiff;
1322 else
1323 valpred += vpdiff;
1324
1325 /* Step 5 - clamp output value */
1326 if ( valpred > 32767 )
1327 valpred = 32767;
1328 else if ( valpred < -32768 )
1329 valpred = -32768;
1330
1331 /* Step 6 - Update step value */
1332 step = stepsizeTable[index];
1333
1334 /* Step 6 - Output value */
1335 if ( size == 1 ) *CHARP(ncp, i) = (signed char)(valpred >> 8);
1336 else if ( size == 2 ) *SHORTP(ncp, i) = (short)(valpred);
1337 else if ( size == 4 ) *LONGP(ncp, i) = (Py_Int32)(valpred<<16);
1338 }
1339
1340 rv = Py_BuildValue("(O(ii))", str, valpred, index);
1341 Py_DECREF(str);
1342 return rv;
1343 }
1344
1345 static PyMethodDef audioop_methods[] = {
1346 { "max", audioop_max, METH_OLDARGS },
1347 { "minmax", audioop_minmax, METH_OLDARGS },
1348 { "avg", audioop_avg, METH_OLDARGS },
1349 { "maxpp", audioop_maxpp, METH_OLDARGS },
1350 { "avgpp", audioop_avgpp, METH_OLDARGS },
1351 { "rms", audioop_rms, METH_OLDARGS },
1352 { "findfit", audioop_findfit, METH_OLDARGS },
1353 { "findmax", audioop_findmax, METH_OLDARGS },
1354 { "findfactor", audioop_findfactor, METH_OLDARGS },
1355 { "cross", audioop_cross, METH_OLDARGS },
1356 { "mul", audioop_mul, METH_OLDARGS },
1357 { "add", audioop_add, METH_OLDARGS },
1358 { "bias", audioop_bias, METH_OLDARGS },
1359 { "ulaw2lin", audioop_ulaw2lin, METH_OLDARGS },
1360 { "lin2ulaw", audioop_lin2ulaw, METH_OLDARGS },
1361 { "lin2lin", audioop_lin2lin, METH_OLDARGS },
1362 { "adpcm2lin", audioop_adpcm2lin, METH_OLDARGS },
1363 { "lin2adpcm", audioop_lin2adpcm, METH_OLDARGS },
1364 { "tomono", audioop_tomono, METH_OLDARGS },
1365 { "tostereo", audioop_tostereo, METH_OLDARGS },
1366 { "getsample", audioop_getsample, METH_OLDARGS },
1367 { "reverse", audioop_reverse, METH_OLDARGS },
1368 { "ratecv", audioop_ratecv, METH_VARARGS },
1369 { 0, 0 }
1370 };
1371
1372 PyMODINIT_FUNC
1373 initaudioop(void)
1374 {
1375 PyObject *m, *d;
1376 m = Py_InitModule("audioop", audioop_methods);
1377 if (m == NULL)
1378 return;
1379 d = PyModule_GetDict(m);
1380 AudioopError = PyErr_NewException("audioop.error", NULL, NULL);
1381 if (AudioopError != NULL)
1382 PyDict_SetItemString(d,"error",AudioopError);
1383 }
python git mirror