]> code.delx.au - pulseaudio/blob - src/pulsecore/resampler.c
delx.au / code / pulseaudio / blob
summary | shortlog | log | commit | commitdiff | tree
history | raw | HEAD
resampler: fix identity check
[pulseaudio] / src / pulsecore / resampler.c
1 /***
2 This file is part of PulseAudio.
3
4 Copyright 2004-2006 Lennart Poettering
5
6 PulseAudio is free software; you can redistribute it and/or modify
7 it under the terms of the GNU Lesser General Public License as published
8 by the Free Software Foundation; either version 2.1 of the License,
9 or (at your option) any later version.
10
11 PulseAudio is distributed in the hope that it will be useful, but
12 WITHOUT ANY WARRANTY; without even the implied warranty of
13 MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
14 General Public License for more details.
15
16 You should have received a copy of the GNU Lesser General Public License
17 along with PulseAudio; if not, write to the Free Software
18 Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307
19 USA.
20 ***/
21
22 #ifdef HAVE_CONFIG_H
23 #include <config.h>
24 #endif
25
26 #include <string.h>
27
28 #ifdef HAVE_LIBSAMPLERATE
29 #include <samplerate.h>
30 #endif
31
32 #include <speex/speex_resampler.h>
33
34 #include <pulse/xmalloc.h>
35 #include <pulsecore/sconv.h>
36 #include <pulsecore/log.h>
37 #include <pulsecore/macro.h>
38 #include <pulsecore/strbuf.h>
39
40 #include "ffmpeg/avcodec.h"
41
42 #include "resampler.h"
43
44 /* Number of samples of extra space we allow the resamplers to return */
45 #define EXTRA_FRAMES 128
46
47 typedef void (*pa_do_remap_func_t) (pa_resampler *r, void *d, const void *s, unsigned n);
48
49 static void remap_channels_matrix (pa_resampler *r, void *dst, const void *src, unsigned n);
50 static void remap_mono_to_stereo(pa_resampler *r, void *dst, const void *src, unsigned n);
51
52 struct pa_resampler {
53 pa_resample_method_t method;
54 pa_resample_flags_t flags;
55
56 pa_sample_spec i_ss, o_ss;
57 pa_channel_map i_cm, o_cm;
58 size_t i_fz, o_fz, w_sz;
59 pa_mempool *mempool;
60
61 pa_memchunk buf1, buf2, buf3, buf4;
62 unsigned buf1_samples, buf2_samples, buf3_samples, buf4_samples;
63
64 pa_sample_format_t work_format;
65
66 pa_convert_func_t to_work_format_func;
67 pa_convert_func_t from_work_format_func;
68
69 float map_table_f[PA_CHANNELS_MAX][PA_CHANNELS_MAX];
70 int32_t map_table_i[PA_CHANNELS_MAX][PA_CHANNELS_MAX];
71 pa_bool_t map_required;
72 pa_do_remap_func_t do_remap;
73
74 void (*impl_free)(pa_resampler *r);
75 void (*impl_update_rates)(pa_resampler *r);
76 void (*impl_resample)(pa_resampler *r, const pa_memchunk *in, unsigned in_samples, pa_memchunk *out, unsigned *out_samples);
77 void (*impl_reset)(pa_resampler *r);
78
79 struct { /* data specific to the trivial resampler */
80 unsigned o_counter;
81 unsigned i_counter;
82 } trivial;
83
84 struct { /* data specific to the peak finder pseudo resampler */
85 unsigned o_counter;
86 unsigned i_counter;
87
88 float max_f[PA_CHANNELS_MAX];
89 int16_t max_i[PA_CHANNELS_MAX];
90
91 } peaks;
92
93 #ifdef HAVE_LIBSAMPLERATE
94 struct { /* data specific to libsamplerate */
95 SRC_STATE *state;
96 } src;
97 #endif
98
99 struct { /* data specific to speex */
100 SpeexResamplerState* state;
101 } speex;
102
103 struct { /* data specific to ffmpeg */
104 struct AVResampleContext *state;
105 pa_memchunk buf[PA_CHANNELS_MAX];
106 } ffmpeg;
107 };
108
109 static int copy_init(pa_resampler *r);
110 static int trivial_init(pa_resampler*r);
111 static int speex_init(pa_resampler*r);
112 static int ffmpeg_init(pa_resampler*r);
113 static int peaks_init(pa_resampler*r);
114 #ifdef HAVE_LIBSAMPLERATE
115 static int libsamplerate_init(pa_resampler*r);
116 #endif
117
118 static void calc_map_table(pa_resampler *r);
119
120 static int (* const init_table[])(pa_resampler*r) = {
121 #ifdef HAVE_LIBSAMPLERATE
122 [PA_RESAMPLER_SRC_SINC_BEST_QUALITY] = libsamplerate_init,
123 [PA_RESAMPLER_SRC_SINC_MEDIUM_QUALITY] = libsamplerate_init,
124 [PA_RESAMPLER_SRC_SINC_FASTEST] = libsamplerate_init,
125 [PA_RESAMPLER_SRC_ZERO_ORDER_HOLD] = libsamplerate_init,
126 [PA_RESAMPLER_SRC_LINEAR] = libsamplerate_init,
127 #else
128 [PA_RESAMPLER_SRC_SINC_BEST_QUALITY] = NULL,
129 [PA_RESAMPLER_SRC_SINC_MEDIUM_QUALITY] = NULL,
130 [PA_RESAMPLER_SRC_SINC_FASTEST] = NULL,
131 [PA_RESAMPLER_SRC_ZERO_ORDER_HOLD] = NULL,
132 [PA_RESAMPLER_SRC_LINEAR] = NULL,
133 #endif
134 [PA_RESAMPLER_TRIVIAL] = trivial_init,
135 [PA_RESAMPLER_SPEEX_FLOAT_BASE+0] = speex_init,
136 [PA_RESAMPLER_SPEEX_FLOAT_BASE+1] = speex_init,
137 [PA_RESAMPLER_SPEEX_FLOAT_BASE+2] = speex_init,
138 [PA_RESAMPLER_SPEEX_FLOAT_BASE+3] = speex_init,
139 [PA_RESAMPLER_SPEEX_FLOAT_BASE+4] = speex_init,
140 [PA_RESAMPLER_SPEEX_FLOAT_BASE+5] = speex_init,
141 [PA_RESAMPLER_SPEEX_FLOAT_BASE+6] = speex_init,
142 [PA_RESAMPLER_SPEEX_FLOAT_BASE+7] = speex_init,
143 [PA_RESAMPLER_SPEEX_FLOAT_BASE+8] = speex_init,
144 [PA_RESAMPLER_SPEEX_FLOAT_BASE+9] = speex_init,
145 [PA_RESAMPLER_SPEEX_FLOAT_BASE+10] = speex_init,
146 [PA_RESAMPLER_SPEEX_FIXED_BASE+0] = speex_init,
147 [PA_RESAMPLER_SPEEX_FIXED_BASE+1] = speex_init,
148 [PA_RESAMPLER_SPEEX_FIXED_BASE+2] = speex_init,
149 [PA_RESAMPLER_SPEEX_FIXED_BASE+3] = speex_init,
150 [PA_RESAMPLER_SPEEX_FIXED_BASE+4] = speex_init,
151 [PA_RESAMPLER_SPEEX_FIXED_BASE+5] = speex_init,
152 [PA_RESAMPLER_SPEEX_FIXED_BASE+6] = speex_init,
153 [PA_RESAMPLER_SPEEX_FIXED_BASE+7] = speex_init,
154 [PA_RESAMPLER_SPEEX_FIXED_BASE+8] = speex_init,
155 [PA_RESAMPLER_SPEEX_FIXED_BASE+9] = speex_init,
156 [PA_RESAMPLER_SPEEX_FIXED_BASE+10] = speex_init,
157 [PA_RESAMPLER_FFMPEG] = ffmpeg_init,
158 [PA_RESAMPLER_AUTO] = NULL,
159 [PA_RESAMPLER_COPY] = copy_init,
160 [PA_RESAMPLER_PEAKS] = peaks_init,
161 };
162
163 pa_resampler* pa_resampler_new(
164 pa_mempool *pool,
165 const pa_sample_spec *a,
166 const pa_channel_map *am,
167 const pa_sample_spec *b,
168 const pa_channel_map *bm,
169 pa_resample_method_t method,
170 pa_resample_flags_t flags) {
171
172 pa_resampler *r = NULL;
173
174 pa_assert(pool);
175 pa_assert(a);
176 pa_assert(b);
177 pa_assert(pa_sample_spec_valid(a));
178 pa_assert(pa_sample_spec_valid(b));
179 pa_assert(method >= 0);
180 pa_assert(method < PA_RESAMPLER_MAX);
181
182 /* Fix method */
183
184 if (!(flags & PA_RESAMPLER_VARIABLE_RATE) && a->rate == b->rate) {
185 pa_log_info("Forcing resampler 'copy', because of fixed, identical sample rates.");
186 method = PA_RESAMPLER_COPY;
187 }
188
189 if (!pa_resample_method_supported(method)) {
190 pa_log_warn("Support for resampler '%s' not compiled in, reverting to 'auto'.", pa_resample_method_to_string(method));
191 method = PA_RESAMPLER_AUTO;
192 }
193
194 if (method == PA_RESAMPLER_FFMPEG && (flags & PA_RESAMPLER_VARIABLE_RATE)) {
195 pa_log_info("Resampler 'ffmpeg' cannot do variable rate, reverting to resampler 'auto'.");
196 method = PA_RESAMPLER_AUTO;
197 }
198
199 if (method == PA_RESAMPLER_COPY && ((flags & PA_RESAMPLER_VARIABLE_RATE) || a->rate != b->rate)) {
200 pa_log_info("Resampler 'copy' cannot change sampling rate, reverting to resampler 'auto'.");
201 method = PA_RESAMPLER_AUTO;
202 }
203
204 if (method == PA_RESAMPLER_AUTO)
205 method = PA_RESAMPLER_SPEEX_FLOAT_BASE + 3;
206
207 r = pa_xnew(pa_resampler, 1);
208 r->mempool = pool;
209 r->method = method;
210 r->flags = flags;
211
212 r->impl_free = NULL;
213 r->impl_update_rates = NULL;
214 r->impl_resample = NULL;
215 r->impl_reset = NULL;
216
217 /* Fill sample specs */
218 r->i_ss = *a;
219 r->o_ss = *b;
220
221 if (am)
222 r->i_cm = *am;
223 else if (!pa_channel_map_init_auto(&r->i_cm, r->i_ss.channels, PA_CHANNEL_MAP_DEFAULT))
224 goto fail;
225
226 if (bm)
227 r->o_cm = *bm;
228 else if (!pa_channel_map_init_auto(&r->o_cm, r->o_ss.channels, PA_CHANNEL_MAP_DEFAULT))
229 goto fail;
230
231 r->i_fz = pa_frame_size(a);
232 r->o_fz = pa_frame_size(b);
233
234 pa_memchunk_reset(&r->buf1);
235 pa_memchunk_reset(&r->buf2);
236 pa_memchunk_reset(&r->buf3);
237 pa_memchunk_reset(&r->buf4);
238
239 r->buf1_samples = r->buf2_samples = r->buf3_samples = r->buf4_samples = 0;
240
241 calc_map_table(r);
242
243 pa_log_info("Using resampler '%s'", pa_resample_method_to_string(method));
244
245 if ((method >= PA_RESAMPLER_SPEEX_FIXED_BASE && method <= PA_RESAMPLER_SPEEX_FIXED_MAX) ||
246 (method == PA_RESAMPLER_FFMPEG))
247 r->work_format = PA_SAMPLE_S16NE;
248 else if (method == PA_RESAMPLER_TRIVIAL || method == PA_RESAMPLER_COPY || method == PA_RESAMPLER_PEAKS) {
249
250 if (r->map_required || a->format != b->format || method == PA_RESAMPLER_PEAKS) {
251
252 if (a->format == PA_SAMPLE_S32NE || a->format == PA_SAMPLE_S32RE ||
253 a->format == PA_SAMPLE_FLOAT32NE || a->format == PA_SAMPLE_FLOAT32RE ||
254 a->format == PA_SAMPLE_S24NE || a->format == PA_SAMPLE_S24RE ||
255 a->format == PA_SAMPLE_S24_32NE || a->format == PA_SAMPLE_S24_32RE ||
256 b->format == PA_SAMPLE_S32NE || b->format == PA_SAMPLE_S32RE ||
257 b->format == PA_SAMPLE_FLOAT32NE || b->format == PA_SAMPLE_FLOAT32RE ||
258 b->format == PA_SAMPLE_S24NE || b->format == PA_SAMPLE_S24RE ||
259 b->format == PA_SAMPLE_S24_32NE || b->format == PA_SAMPLE_S24_32RE)
260 r->work_format = PA_SAMPLE_FLOAT32NE;
261 else
262 r->work_format = PA_SAMPLE_S16NE;
263
264 } else
265 r->work_format = a->format;
266
267 } else
268 r->work_format = PA_SAMPLE_FLOAT32NE;
269
270 pa_log_info("Using %s as working format.", pa_sample_format_to_string(r->work_format));
271
272 r->w_sz = pa_sample_size_of_format(r->work_format);
273
274 if (r->i_ss.format == r->work_format)
275 r->to_work_format_func = NULL;
276 else if (r->work_format == PA_SAMPLE_FLOAT32NE) {
277 if (!(r->to_work_format_func = pa_get_convert_to_float32ne_function(r->i_ss.format)))
278 goto fail;
279 } else {
280 pa_assert(r->work_format == PA_SAMPLE_S16NE);
281 if (!(r->to_work_format_func = pa_get_convert_to_s16ne_function(r->i_ss.format)))
282 goto fail;
283 }
284
285 if (r->o_ss.format == r->work_format)
286 r->from_work_format_func = NULL;
287 else if (r->work_format == PA_SAMPLE_FLOAT32NE) {
288 if (!(r->from_work_format_func = pa_get_convert_from_float32ne_function(r->o_ss.format)))
289 goto fail;
290 } else {
291 pa_assert(r->work_format == PA_SAMPLE_S16NE);
292 if (!(r->from_work_format_func = pa_get_convert_from_s16ne_function(r->o_ss.format)))
293 goto fail;
294 }
295
296 /* initialize implementation */
297 if (init_table[method](r) < 0)
298 goto fail;
299
300 return r;
301
302 fail:
303 if (r)
304 pa_xfree(r);
305
306 return NULL;
307 }
308
309 void pa_resampler_free(pa_resampler *r) {
310 pa_assert(r);
311
312 if (r->impl_free)
313 r->impl_free(r);
314
315 if (r->buf1.memblock)
316 pa_memblock_unref(r->buf1.memblock);
317 if (r->buf2.memblock)
318 pa_memblock_unref(r->buf2.memblock);
319 if (r->buf3.memblock)
320 pa_memblock_unref(r->buf3.memblock);
321 if (r->buf4.memblock)
322 pa_memblock_unref(r->buf4.memblock);
323
324 pa_xfree(r);
325 }
326
327 void pa_resampler_set_input_rate(pa_resampler *r, uint32_t rate) {
328 pa_assert(r);
329 pa_assert(rate > 0);
330
331 if (r->i_ss.rate == rate)
332 return;
333
334 r->i_ss.rate = rate;
335
336 r->impl_update_rates(r);
337 }
338
339 void pa_resampler_set_output_rate(pa_resampler *r, uint32_t rate) {
340 pa_assert(r);
341 pa_assert(rate > 0);
342
343 if (r->o_ss.rate == rate)
344 return;
345
346 r->o_ss.rate = rate;
347
348 r->impl_update_rates(r);
349 }
350
351 size_t pa_resampler_request(pa_resampler *r, size_t out_length) {
352 pa_assert(r);
353
354 /* Let's round up here */
355
356 return (((((out_length + r->o_fz-1) / r->o_fz) * r->i_ss.rate) + r->o_ss.rate-1) / r->o_ss.rate) * r->i_fz;
357 }
358
359 size_t pa_resampler_result(pa_resampler *r, size_t in_length) {
360 pa_assert(r);
361
362 /* Let's round up here */
363
364 return (((((in_length + r->i_fz-1) / r->i_fz) * r->o_ss.rate) + r->i_ss.rate-1) / r->i_ss.rate) * r->o_fz;
365 }
366
367 size_t pa_resampler_max_block_size(pa_resampler *r) {
368 size_t block_size_max;
369 pa_sample_spec ss;
370 size_t fs;
371
372 pa_assert(r);
373
374 block_size_max = pa_mempool_block_size_max(r->mempool);
375
376 /* We deduce the "largest" sample spec we're using during the
377 * conversion */
378 ss.channels = (uint8_t) (PA_MAX(r->i_ss.channels, r->o_ss.channels));
379
380 /* We silently assume that the format enum is ordered by size */
381 ss.format = PA_MAX(r->i_ss.format, r->o_ss.format);
382 ss.format = PA_MAX(ss.format, r->work_format);
383
384 ss.rate = PA_MAX(r->i_ss.rate, r->o_ss.rate);
385
386 fs = pa_frame_size(&ss);
387
388 return (((block_size_max/fs - EXTRA_FRAMES)*r->i_ss.rate)/ss.rate)*r->i_fz;
389 }
390
391 void pa_resampler_reset(pa_resampler *r) {
392 pa_assert(r);
393
394 if (r->impl_reset)
395 r->impl_reset(r);
396 }
397
398 pa_resample_method_t pa_resampler_get_method(pa_resampler *r) {
399 pa_assert(r);
400
401 return r->method;
402 }
403
404 const pa_channel_map* pa_resampler_input_channel_map(pa_resampler *r) {
405 pa_assert(r);
406
407 return &r->i_cm;
408 }
409
410 const pa_sample_spec* pa_resampler_input_sample_spec(pa_resampler *r) {
411 pa_assert(r);
412
413 return &r->i_ss;
414 }
415
416 const pa_channel_map* pa_resampler_output_channel_map(pa_resampler *r) {
417 pa_assert(r);
418
419 return &r->o_cm;
420 }
421
422 const pa_sample_spec* pa_resampler_output_sample_spec(pa_resampler *r) {
423 pa_assert(r);
424
425 return &r->o_ss;
426 }
427
428 static const char * const resample_methods[] = {
429 "src-sinc-best-quality",
430 "src-sinc-medium-quality",
431 "src-sinc-fastest",
432 "src-zero-order-hold",
433 "src-linear",
434 "trivial",
435 "speex-float-0",
436 "speex-float-1",
437 "speex-float-2",
438 "speex-float-3",
439 "speex-float-4",
440 "speex-float-5",
441 "speex-float-6",
442 "speex-float-7",
443 "speex-float-8",
444 "speex-float-9",
445 "speex-float-10",
446 "speex-fixed-0",
447 "speex-fixed-1",
448 "speex-fixed-2",
449 "speex-fixed-3",
450 "speex-fixed-4",
451 "speex-fixed-5",
452 "speex-fixed-6",
453 "speex-fixed-7",
454 "speex-fixed-8",
455 "speex-fixed-9",
456 "speex-fixed-10",
457 "ffmpeg",
458 "auto",
459 "copy",
460 "peaks"
461 };
462
463 const char *pa_resample_method_to_string(pa_resample_method_t m) {
464
465 if (m < 0 || m >= PA_RESAMPLER_MAX)
466 return NULL;
467
468 return resample_methods[m];
469 }
470
471 int pa_resample_method_supported(pa_resample_method_t m) {
472
473 if (m < 0 || m >= PA_RESAMPLER_MAX)
474 return 0;
475
476 #ifndef HAVE_LIBSAMPLERATE
477 if (m <= PA_RESAMPLER_SRC_LINEAR)
478 return 0;
479 #endif
480
481 return 1;
482 }
483
484 pa_resample_method_t pa_parse_resample_method(const char *string) {
485 pa_resample_method_t m;
486
487 pa_assert(string);
488
489 for (m = 0; m < PA_RESAMPLER_MAX; m++)
490 if (!strcmp(string, resample_methods[m]))
491 return m;
492
493 if (!strcmp(string, "speex-fixed"))
494 return PA_RESAMPLER_SPEEX_FIXED_BASE + 3;
495
496 if (!strcmp(string, "speex-float"))
497 return PA_RESAMPLER_SPEEX_FLOAT_BASE + 3;
498
499 return PA_RESAMPLER_INVALID;
500 }
501
502 static pa_bool_t on_left(pa_channel_position_t p) {
503
504 return
505 p == PA_CHANNEL_POSITION_FRONT_LEFT ||
506 p == PA_CHANNEL_POSITION_REAR_LEFT ||
507 p == PA_CHANNEL_POSITION_FRONT_LEFT_OF_CENTER ||
508 p == PA_CHANNEL_POSITION_SIDE_LEFT ||
509 p == PA_CHANNEL_POSITION_TOP_FRONT_LEFT ||
510 p == PA_CHANNEL_POSITION_TOP_REAR_LEFT;
511 }
512
513 static pa_bool_t on_right(pa_channel_position_t p) {
514
515 return
516 p == PA_CHANNEL_POSITION_FRONT_RIGHT ||
517 p == PA_CHANNEL_POSITION_REAR_RIGHT ||
518 p == PA_CHANNEL_POSITION_FRONT_RIGHT_OF_CENTER ||
519 p == PA_CHANNEL_POSITION_SIDE_RIGHT ||
520 p == PA_CHANNEL_POSITION_TOP_FRONT_RIGHT ||
521 p == PA_CHANNEL_POSITION_TOP_REAR_RIGHT;
522 }
523
524 static pa_bool_t on_center(pa_channel_position_t p) {
525
526 return
527 p == PA_CHANNEL_POSITION_FRONT_CENTER ||
528 p == PA_CHANNEL_POSITION_REAR_CENTER ||
529 p == PA_CHANNEL_POSITION_TOP_CENTER ||
530 p == PA_CHANNEL_POSITION_TOP_FRONT_CENTER ||
531 p == PA_CHANNEL_POSITION_TOP_REAR_CENTER;
532 }
533
534 static pa_bool_t on_lfe(pa_channel_position_t p) {
535 return
536 p == PA_CHANNEL_POSITION_LFE;
537 }
538
539 static pa_bool_t on_front(pa_channel_position_t p) {
540 return
541 p == PA_CHANNEL_POSITION_FRONT_LEFT ||
542 p == PA_CHANNEL_POSITION_FRONT_RIGHT ||
543 p == PA_CHANNEL_POSITION_FRONT_CENTER ||
544 p == PA_CHANNEL_POSITION_TOP_FRONT_LEFT ||
545 p == PA_CHANNEL_POSITION_TOP_FRONT_RIGHT ||
546 p == PA_CHANNEL_POSITION_TOP_FRONT_CENTER ||
547 p == PA_CHANNEL_POSITION_FRONT_LEFT_OF_CENTER ||
548 p == PA_CHANNEL_POSITION_FRONT_RIGHT_OF_CENTER;
549 }
550
551 static pa_bool_t on_rear(pa_channel_position_t p) {
552 return
553 p == PA_CHANNEL_POSITION_REAR_LEFT ||
554 p == PA_CHANNEL_POSITION_REAR_RIGHT ||
555 p == PA_CHANNEL_POSITION_REAR_CENTER ||
556 p == PA_CHANNEL_POSITION_TOP_REAR_LEFT ||
557 p == PA_CHANNEL_POSITION_TOP_REAR_RIGHT ||
558 p == PA_CHANNEL_POSITION_TOP_REAR_CENTER;
559 }
560
561 static pa_bool_t on_side(pa_channel_position_t p) {
562 return
563 p == PA_CHANNEL_POSITION_SIDE_LEFT ||
564 p == PA_CHANNEL_POSITION_SIDE_RIGHT ||
565 p == PA_CHANNEL_POSITION_TOP_CENTER;
566 }
567
568 enum {
569 ON_FRONT,
570 ON_REAR,
571 ON_SIDE,
572 ON_OTHER
573 };
574
575 static int front_rear_side(pa_channel_position_t p) {
576 if (on_front(p))
577 return ON_FRONT;
578 if (on_rear(p))
579 return ON_REAR;
580 if (on_side(p))
581 return ON_SIDE;
582 return ON_OTHER;
583 }
584
585 static void calc_map_table(pa_resampler *r) {
586 unsigned oc, ic;
587 pa_bool_t ic_connected[PA_CHANNELS_MAX];
588 pa_bool_t remix;
589 pa_strbuf *s;
590 char *t;
591
592 pa_assert(r);
593
594 if (!(r->map_required = (r->i_ss.channels != r->o_ss.channels || (!(r->flags & PA_RESAMPLER_NO_REMAP) && !pa_channel_map_equal(&r->i_cm, &r->o_cm)))))
595 return;
596
597 memset(r->map_table_f, 0, sizeof(r->map_table_f));
598 memset(r->map_table_i, 0, sizeof(r->map_table_i));
599 memset(ic_connected, 0, sizeof(ic_connected));
600 remix = (r->flags & (PA_RESAMPLER_NO_REMAP|PA_RESAMPLER_NO_REMIX)) == 0;
601
602 for (oc = 0; oc < r->o_ss.channels; oc++) {
603 pa_bool_t oc_connected = FALSE;
604 pa_channel_position_t b = r->o_cm.map[oc];
605
606 for (ic = 0; ic < r->i_ss.channels; ic++) {
607 pa_channel_position_t a = r->i_cm.map[ic];
608
609 if (r->flags & PA_RESAMPLER_NO_REMAP) {
610 /* We shall not do any remapping. Hence, just check by index */
611
612 if (ic == oc)
613 r->map_table_f[oc][ic] = 1.0;
614
615 continue;
616 }
617
618 if (r->flags & PA_RESAMPLER_NO_REMIX) {
619 /* We shall not do any remixing. Hence, just check by name */
620
621 if (a == b)
622 r->map_table_f[oc][ic] = 1.0;
623
624 continue;
625 }
626
627 pa_assert(remix);
628
629 /* OK, we shall do the full monty: upmixing and
630 * downmixing. Our algorithm is relatively simple, does
631 * not do spacialization, delay elements or apply lowpass
632 * filters for LFE. Patches are always welcome,
633 * though. Oh, and it doesn't do any matrix
634 * decoding. (Which probably wouldn't make any sense
635 * anyway.)
636 *
637 * This code is not idempotent: downmixing an upmixed
638 * stereo stream is not identical to the original. The
639 * volume will not match, and the two channels will be a
640 * linear combination of both.
641 *
642 * This is losely based on random suggestions found on the
643 * Internet, such as this:
644 * http://www.halfgaar.net/surround-sound-in-linux and the
645 * alsa upmix plugin.
646 *
647 * The algorithm works basically like this:
648 *
649 * 1) Connect all channels with matching names.
650 *
651 * 2) Mono Handling:
652 * S:Mono: Copy into all D:channels
653 * D:Mono: Copy in all S:channels
654 *
655 * 3) Mix D:Left, D:Right:
656 * D:Left: If not connected, avg all S:Left
657 * D:Right: If not connected, avg all S:Right
658 *
659 * 4) Mix D:Center
660 * If not connected, avg all S:Center
661 * If still not connected, avg all S:Left, S:Right
662 *
663 * 5) Mix D:LFE
664 * If not connected, avg all S:*
665 *
666 * 6) Make sure S:Left/S:Right is used: S:Left/S:Right: If
667 * not connected, mix into all D:left and all D:right
668 * channels. Gain is 0.1, the current left and right
669 * should be multiplied by 0.9.
670 *
671 * 7) Make sure S:Center, S:LFE is used:
672 *
673 * S:Center, S:LFE: If not connected, mix into all
674 * D:left, all D:right, all D:center channels, gain is
675 * 0.375. The current (as result of 1..6) factors
676 * should be multiplied by 0.75. (Alt. suggestion: 0.25
677 * vs. 0.5) If C-front is only mixed into
678 * L-front/R-front if available, otherwise into all L/R
679 * channels. Similarly for C-rear.
680 *
681 * S: and D: shall relate to the source resp. destination channels.
682 *
683 * Rationale: 1, 2 are probably obvious. For 3: this
684 * copies front to rear if needed. For 4: we try to find
685 * some suitable C source for C, if we don't find any, we
686 * avg L and R. For 5: LFE is mixed from all channels. For
687 * 6: the rear channels should not be dropped entirely,
688 * however have only minimal impact. For 7: movies usually
689 * encode speech on the center channel. Thus we have to
690 * make sure this channel is distributed to L and R if not
691 * available in the output. Also, LFE is used to achieve a
692 * greater dynamic range, and thus we should try to do our
693 * best to pass it to L+R.
694 */
695
696 if (a == b || a == PA_CHANNEL_POSITION_MONO || b == PA_CHANNEL_POSITION_MONO) {
697 r->map_table_f[oc][ic] = 1.0;
698
699 oc_connected = TRUE;
700 ic_connected[ic] = TRUE;
701 }
702 }
703
704 if (!oc_connected && remix) {
705 /* OK, we shall remix */
706
707 /* Try to find matching input ports for this output port */
708
709 if (on_left(b)) {
710 unsigned n = 0;
711
712 /* We are not connected and on the left side, let's
713 * average all left side input channels. */
714
715 for (ic = 0; ic < r->i_ss.channels; ic++)
716 if (on_left(r->i_cm.map[ic]))
717 n++;
718
719 if (n > 0)
720 for (ic = 0; ic < r->i_ss.channels; ic++)
721 if (on_left(r->i_cm.map[ic])) {
722 r->map_table_f[oc][ic] = 1.0f / (float) n;
723 ic_connected[ic] = TRUE;
724 }
725
726 /* We ignore the case where there is no left input
727 * channel. Something is really wrong in this case
728 * anyway. */
729
730 } else if (on_right(b)) {
731 unsigned n = 0;
732
733 /* We are not connected and on the right side, let's
734 * average all right side input channels. */
735
736 for (ic = 0; ic < r->i_ss.channels; ic++)
737 if (on_right(r->i_cm.map[ic]))
738 n++;
739
740 if (n > 0)
741 for (ic = 0; ic < r->i_ss.channels; ic++)
742 if (on_right(r->i_cm.map[ic])) {
743 r->map_table_f[oc][ic] = 1.0f / (float) n;
744 ic_connected[ic] = TRUE;
745 }
746
747 /* We ignore the case where there is no right input
748 * channel. Something is really wrong in this case
749 * anyway. */
750
751 } else if (on_center(b)) {
752 unsigned n = 0;
753
754 /* We are not connected and at the center. Let's
755 * average all center input channels. */
756
757 for (ic = 0; ic < r->i_ss.channels; ic++)
758 if (on_center(r->i_cm.map[ic]))
759 n++;
760
761 if (n > 0) {
762 for (ic = 0; ic < r->i_ss.channels; ic++)
763 if (on_center(r->i_cm.map[ic])) {
764 r->map_table_f[oc][ic] = 1.0f / (float) n;
765 ic_connected[ic] = TRUE;
766 }
767 } else {
768
769 /* Hmm, no center channel around, let's synthesize
770 * it by mixing L and R.*/
771
772 n = 0;
773
774 for (ic = 0; ic < r->i_ss.channels; ic++)
775 if (on_left(r->i_cm.map[ic]) || on_right(r->i_cm.map[ic]))
776 n++;
777
778 if (n > 0)
779 for (ic = 0; ic < r->i_ss.channels; ic++)
780 if (on_left(r->i_cm.map[ic]) || on_right(r->i_cm.map[ic])) {
781 r->map_table_f[oc][ic] = 1.0f / (float) n;
782 ic_connected[ic] = TRUE;
783 }
784
785 /* We ignore the case where there is not even a
786 * left or right input channel. Something is
787 * really wrong in this case anyway. */
788 }
789
790 } else if (on_lfe(b)) {
791
792 /* We are not connected and an LFE. Let's average all
793 * channels for LFE. */
794
795 for (ic = 0; ic < r->i_ss.channels; ic++) {
796
797 if (!(r->flags & PA_RESAMPLER_NO_LFE))
798 r->map_table_f[oc][ic] = 1.0f / (float) r->i_ss.channels;
799 else
800 r->map_table_f[oc][ic] = 0;
801
802 /* Please note that a channel connected to LFE
803 * doesn't really count as connected. */
804 }
805 }
806 }
807 }
808
809 if (remix) {
810 unsigned
811 ic_unconnected_left = 0,
812 ic_unconnected_right = 0,
813 ic_unconnected_center = 0,
814 ic_unconnected_lfe = 0;
815
816 for (ic = 0; ic < r->i_ss.channels; ic++) {
817 pa_channel_position_t a = r->i_cm.map[ic];
818
819 if (ic_connected[ic])
820 continue;
821
822 if (on_left(a))
823 ic_unconnected_left++;
824 else if (on_right(a))
825 ic_unconnected_right++;
826 else if (on_center(a))
827 ic_unconnected_center++;
828 else if (on_lfe(a))
829 ic_unconnected_lfe++;
830 }
831
832 if (ic_unconnected_left > 0) {
833
834 /* OK, so there are unconnected input channels on the
835 * left. Let's multiply all already connected channels on
836 * the left side by .9 and add in our averaged unconnected
837 * channels multplied by .1 */
838
839 for (oc = 0; oc < r->o_ss.channels; oc++) {
840
841 if (!on_left(r->o_cm.map[oc]))
842 continue;
843
844 for (ic = 0; ic < r->i_ss.channels; ic++) {
845
846 if (ic_connected[ic]) {
847 r->map_table_f[oc][ic] *= .9f;
848 continue;
849 }
850
851 if (on_left(r->i_cm.map[ic]))
852 r->map_table_f[oc][ic] = .1f / (float) ic_unconnected_left;
853 }
854 }
855 }
856
857 if (ic_unconnected_right > 0) {
858
859 /* OK, so there are unconnected input channels on the
860 * right. Let's multiply all already connected channels on
861 * the right side by .9 and add in our averaged unconnected
862 * channels multplied by .1 */
863
864 for (oc = 0; oc < r->o_ss.channels; oc++) {
865
866 if (!on_right(r->o_cm.map[oc]))
867 continue;
868
869 for (ic = 0; ic < r->i_ss.channels; ic++) {
870
871 if (ic_connected[ic]) {
872 r->map_table_f[oc][ic] *= .9f;
873 continue;
874 }
875
876 if (on_right(r->i_cm.map[ic]))
877 r->map_table_f[oc][ic] = .1f / (float) ic_unconnected_right;
878 }
879 }
880 }
881
882 if (ic_unconnected_center > 0) {
883 pa_bool_t mixed_in = FALSE;
884
885 /* OK, so there are unconnected input channels on the
886 * center. Let's multiply all already connected channels on
887 * the center side by .9 and add in our averaged unconnected
888 * channels multplied by .1 */
889
890 for (oc = 0; oc < r->o_ss.channels; oc++) {
891
892 if (!on_center(r->o_cm.map[oc]))
893 continue;
894
895 for (ic = 0; ic < r->i_ss.channels; ic++) {
896
897 if (ic_connected[ic]) {
898 r->map_table_f[oc][ic] *= .9f;
899 continue;
900 }
901
902 if (on_center(r->i_cm.map[ic])) {
903 r->map_table_f[oc][ic] = .1f / (float) ic_unconnected_center;
904 mixed_in = TRUE;
905 }
906 }
907 }
908
909 if (!mixed_in) {
910 unsigned ncenter[PA_CHANNELS_MAX];
911 pa_bool_t found_frs[PA_CHANNELS_MAX];
912
913 memset(ncenter, 0, sizeof(ncenter));
914 memset(found_frs, 0, sizeof(found_frs));
915
916 /* Hmm, as it appears there was no center channel we
917 could mix our center channel in. In this case, mix
918 it into left and right. Using .375 and 0.75 as
919 factors. */
920
921 for (ic = 0; ic < r->i_ss.channels; ic++) {
922
923 if (ic_connected[ic])
924 continue;
925
926 if (!on_center(r->i_cm.map[ic]))
927 continue;
928
929 for (oc = 0; oc < r->o_ss.channels; oc++) {
930
931 if (!on_left(r->o_cm.map[oc]) && !on_right(r->o_cm.map[oc]))
932 continue;
933
934 if (front_rear_side(r->i_cm.map[ic]) == front_rear_side(r->o_cm.map[oc])) {
935 found_frs[ic] = TRUE;
936 break;
937 }
938 }
939
940 for (oc = 0; oc < r->o_ss.channels; oc++) {
941
942 if (!on_left(r->o_cm.map[oc]) && !on_right(r->o_cm.map[oc]))
943 continue;
944
945 if (!found_frs[ic] || front_rear_side(r->i_cm.map[ic]) == front_rear_side(r->o_cm.map[oc]))
946 ncenter[oc]++;
947 }
948 }
949
950 for (oc = 0; oc < r->o_ss.channels; oc++) {
951
952 if (!on_left(r->o_cm.map[oc]) && !on_right(r->o_cm.map[oc]))
953 continue;
954
955 if (ncenter[oc] <= 0)
956 continue;
957
958 for (ic = 0; ic < r->i_ss.channels; ic++) {
959
960 if (ic_connected[ic]) {
961 r->map_table_f[oc][ic] *= .75f;
962 continue;
963 }
964
965 if (!on_center(r->i_cm.map[ic]))
966 continue;
967
968 if (!found_frs[ic] || front_rear_side(r->i_cm.map[ic]) == front_rear_side(r->o_cm.map[oc]))
969 r->map_table_f[oc][ic] = .375f / (float) ncenter[oc];
970 }
971 }
972 }
973 }
974
975 if (ic_unconnected_lfe > 0 && !(r->flags & PA_RESAMPLER_NO_LFE)) {
976
977 /* OK, so there is an unconnected LFE channel. Let's mix
978 * it into all channels, with factor 0.375 */
979
980 for (ic = 0; ic < r->i_ss.channels; ic++) {
981
982 if (!on_lfe(r->i_cm.map[ic]))
983 continue;
984
985 for (oc = 0; oc < r->o_ss.channels; oc++)
986 r->map_table_f[oc][ic] = 0.375f / (float) ic_unconnected_lfe;
987 }
988 }
989 }
990 /* make an 16:16 int version of the matrix */
991 for (oc = 0; oc < r->o_ss.channels; oc++)
992 for (ic = 0; ic < r->i_ss.channels; ic++)
993 r->map_table_i[oc][ic] = (int32_t) (r->map_table_f[oc][ic] * 0x10000);
994
995 s = pa_strbuf_new();
996
997 pa_strbuf_printf(s, " ");
998 for (ic = 0; ic < r->i_ss.channels; ic++)
999 pa_strbuf_printf(s, " I%02u ", ic);
1000 pa_strbuf_puts(s, "\n +");
1001
1002 for (ic = 0; ic < r->i_ss.channels; ic++)
1003 pa_strbuf_printf(s, "------");
1004 pa_strbuf_puts(s, "\n");
1005
1006 for (oc = 0; oc < r->o_ss.channels; oc++) {
1007 pa_strbuf_printf(s, "O%02u |", oc);
1008
1009 for (ic = 0; ic < r->i_ss.channels; ic++)
1010 pa_strbuf_printf(s, " %1.3f", r->map_table_f[oc][ic]);
1011
1012 pa_strbuf_puts(s, "\n");
1013 }
1014
1015 pa_log_debug("Channel matrix:\n%s", t = pa_strbuf_tostring_free(s));
1016 pa_xfree(t);
1017
1018 /* find some common channel remappings, fall back to full matrix operation. */
1019 if (r->i_ss.channels == 1 && r->o_ss.channels == 2 &&
1020 r->map_table_f[0][0] >= 1.0 && r->map_table_f[1][0] >= 1.0) {
1021 r->do_remap = (pa_do_remap_func_t) remap_mono_to_stereo;;
1022 pa_log_debug("Using mono to stereo remapping");
1023 } else {
1024 r->do_remap = (pa_do_remap_func_t) remap_channels_matrix;
1025 pa_log_debug("Using generic matrix remapping");
1026 }
1027
1028 }
1029
1030 static pa_memchunk* convert_to_work_format(pa_resampler *r, pa_memchunk *input) {
1031 unsigned n_samples;
1032 void *src, *dst;
1033
1034 pa_assert(r);
1035 pa_assert(input);
1036 pa_assert(input->memblock);
1037
1038 /* Convert the incoming sample into the work sample format and place them in buf1 */
1039
1040 if (!r->to_work_format_func || !input->length)
1041 return input;
1042
1043 n_samples = (unsigned) ((input->length / r->i_fz) * r->i_ss.channels);
1044
1045 r->buf1.index = 0;
1046 r->buf1.length = r->w_sz * n_samples;
1047
1048 if (!r->buf1.memblock || r->buf1_samples < n_samples) {
1049 if (r->buf1.memblock)
1050 pa_memblock_unref(r->buf1.memblock);
1051
1052 r->buf1_samples = n_samples;
1053 r->buf1.memblock = pa_memblock_new(r->mempool, r->buf1.length);
1054 }
1055
1056 src = (uint8_t*) pa_memblock_acquire(input->memblock) + input->index;
1057 dst = (uint8_t*) pa_memblock_acquire(r->buf1.memblock);
1058
1059 r->to_work_format_func(n_samples, src, dst);
1060
1061 pa_memblock_release(input->memblock);
1062 pa_memblock_release(r->buf1.memblock);
1063
1064 return &r->buf1;
1065 }
1066
1067 static void remap_mono_to_stereo(pa_resampler *r, void *dst, const void *src, unsigned n) {
1068
1069 switch (r->work_format) {
1070 case PA_SAMPLE_FLOAT32NE:
1071 {
1072 float *d, *s;
1073
1074 d = (float *) dst;
1075 s = (float *) src;
1076
1077 for (; n > 0; n--, s++, d += 2)
1078 d[0] = d[1] = *s;
1079 break;
1080 }
1081 case PA_SAMPLE_S16NE:
1082 {
1083 int16_t *d, *s;
1084
1085 d = (int16_t *) dst;
1086 s = (int16_t *) src;
1087
1088 for (; n > 0; n--, s++, d += 2)
1089 d[0] = d[1] = *s;
1090 break;
1091 }
1092 default:
1093 pa_assert_not_reached();
1094 }
1095 }
1096
1097 static void remap_channels_matrix (pa_resampler *r, void *dst, const void *src, unsigned n) {
1098 unsigned oc, i;
1099 unsigned n_ic, n_oc;
1100
1101 n_ic = r->i_ss.channels;
1102 n_oc = r->o_ss.channels;
1103
1104 memset(dst, 0, r->buf2.length);
1105
1106 switch (r->work_format) {
1107 case PA_SAMPLE_FLOAT32NE:
1108 {
1109 float *d, *s;
1110
1111 for (oc = 0; oc < n_oc; oc++) {
1112 unsigned ic;
1113
1114 for (ic = 0; ic < n_ic; ic++) {
1115 float vol;
1116
1117 vol = r->map_table_f[oc][ic];
1118
1119 if (vol <= 0.0)
1120 continue;
1121
1122 d = (float *)dst + oc;
1123 s = (float *)src + ic;
1124
1125 for (i = n; i > 0; i--, s += n_ic, d += n_oc)
1126 *d += *s * vol;
1127 }
1128 }
1129
1130 break;
1131 }
1132 case PA_SAMPLE_S16NE:
1133 {
1134 int16_t *d, *s;
1135
1136 for (oc = 0; oc < n_oc; oc++) {
1137 unsigned ic;
1138
1139 for (ic = 0; ic < n_ic; ic++) {
1140 int32_t vol;
1141
1142 vol = r->map_table_i[oc][ic];
1143
1144 if (vol <= 0)
1145 continue;
1146
1147 d = (int16_t *)dst + oc;
1148 s = (int16_t *)src + ic;
1149
1150 if (vol >= 0x10000) {
1151 for (i = n; i > 0; i--, s += n_ic, d += n_oc)
1152 *d += *s;
1153 } else {
1154 for (i = n; i > 0; i--, s += n_ic, d += n_oc)
1155 *d += (int16_t) (((int32_t)*s * vol) >> 16);
1156 }
1157 }
1158 }
1159 break;
1160 }
1161 default:
1162 pa_assert_not_reached();
1163 }
1164 }
1165
1166 static pa_memchunk *remap_channels(pa_resampler *r, pa_memchunk *input) {
1167 unsigned in_n_samples, out_n_samples, n_frames;
1168 void *src, *dst;
1169
1170 pa_assert(r);
1171 pa_assert(input);
1172 pa_assert(input->memblock);
1173
1174 /* Remap channels and place the result int buf2 */
1175
1176 if (!r->map_required || !input->length)
1177 return input;
1178
1179 in_n_samples = (unsigned) (input->length / r->w_sz);
1180 n_frames = in_n_samples / r->i_ss.channels;
1181 out_n_samples = n_frames * r->o_ss.channels;
1182
1183 r->buf2.index = 0;
1184 r->buf2.length = r->w_sz * out_n_samples;
1185
1186 if (!r->buf2.memblock || r->buf2_samples < out_n_samples) {
1187 if (r->buf2.memblock)
1188 pa_memblock_unref(r->buf2.memblock);
1189
1190 r->buf2_samples = out_n_samples;
1191 r->buf2.memblock = pa_memblock_new(r->mempool, r->buf2.length);
1192 }
1193
1194 src = ((uint8_t*) pa_memblock_acquire(input->memblock) + input->index);
1195 dst = pa_memblock_acquire(r->buf2.memblock);
1196
1197 pa_assert (r->do_remap);
1198 r->do_remap (r, dst, src, n_frames);
1199
1200 pa_memblock_release(input->memblock);
1201 pa_memblock_release(r->buf2.memblock);
1202
1203 return &r->buf2;
1204 }
1205
1206 static pa_memchunk *resample(pa_resampler *r, pa_memchunk *input) {
1207 unsigned in_n_frames, in_n_samples;
1208 unsigned out_n_frames, out_n_samples;
1209
1210 pa_assert(r);
1211 pa_assert(input);
1212
1213 /* Resample the data and place the result in buf3 */
1214
1215 if (!r->impl_resample || !input->length)
1216 return input;
1217
1218 in_n_samples = (unsigned) (input->length / r->w_sz);
1219 in_n_frames = (unsigned) (in_n_samples / r->o_ss.channels);
1220
1221 out_n_frames = ((in_n_frames*r->o_ss.rate)/r->i_ss.rate)+EXTRA_FRAMES;
1222 out_n_samples = out_n_frames * r->o_ss.channels;
1223
1224 r->buf3.index = 0;
1225 r->buf3.length = r->w_sz * out_n_samples;
1226
1227 if (!r->buf3.memblock || r->buf3_samples < out_n_samples) {
1228 if (r->buf3.memblock)
1229 pa_memblock_unref(r->buf3.memblock);
1230
1231 r->buf3_samples = out_n_samples;
1232 r->buf3.memblock = pa_memblock_new(r->mempool, r->buf3.length);
1233 }
1234
1235 r->impl_resample(r, input, in_n_frames, &r->buf3, &out_n_frames);
1236 r->buf3.length = out_n_frames * r->w_sz * r->o_ss.channels;
1237
1238 return &r->buf3;
1239 }
1240
1241 static pa_memchunk *convert_from_work_format(pa_resampler *r, pa_memchunk *input) {
1242 unsigned n_samples, n_frames;
1243 void *src, *dst;
1244
1245 pa_assert(r);
1246 pa_assert(input);
1247
1248 /* Convert the data into the correct sample type and place the result in buf4 */
1249
1250 if (!r->from_work_format_func || !input->length)
1251 return input;
1252
1253 n_samples = (unsigned) (input->length / r->w_sz);
1254 n_frames = n_samples / r->o_ss.channels;
1255
1256 r->buf4.index = 0;
1257 r->buf4.length = r->o_fz * n_frames;
1258
1259 if (!r->buf4.memblock || r->buf4_samples < n_samples) {
1260 if (r->buf4.memblock)
1261 pa_memblock_unref(r->buf4.memblock);
1262
1263 r->buf4_samples = n_samples;
1264 r->buf4.memblock = pa_memblock_new(r->mempool, r->buf4.length);
1265 }
1266
1267 src = (uint8_t*) pa_memblock_acquire(input->memblock) + input->index;
1268 dst = pa_memblock_acquire(r->buf4.memblock);
1269 r->from_work_format_func(n_samples, src, dst);
1270 pa_memblock_release(input->memblock);
1271 pa_memblock_release(r->buf4.memblock);
1272
1273 r->buf4.length = r->o_fz * n_frames;
1274
1275 return &r->buf4;
1276 }
1277
1278 void pa_resampler_run(pa_resampler *r, const pa_memchunk *in, pa_memchunk *out) {
1279 pa_memchunk *buf;
1280
1281 pa_assert(r);
1282 pa_assert(in);
1283 pa_assert(out);
1284 pa_assert(in->length);
1285 pa_assert(in->memblock);
1286 pa_assert(in->length % r->i_fz == 0);
1287
1288 buf = (pa_memchunk*) in;
1289 buf = convert_to_work_format(r, buf);
1290 buf = remap_channels(r, buf);
1291 buf = resample(r, buf);
1292
1293 if (buf->length) {
1294 buf = convert_from_work_format(r, buf);
1295 *out = *buf;
1296
1297 if (buf == in)
1298 pa_memblock_ref(buf->memblock);
1299 else
1300 pa_memchunk_reset(buf);
1301 } else
1302 pa_memchunk_reset(out);
1303 }
1304
1305 /*** libsamplerate based implementation ***/
1306
1307 #ifdef HAVE_LIBSAMPLERATE
1308 static void libsamplerate_resample(pa_resampler *r, const pa_memchunk *input, unsigned in_n_frames, pa_memchunk *output, unsigned *out_n_frames) {
1309 SRC_DATA data;
1310
1311 pa_assert(r);
1312 pa_assert(input);
1313 pa_assert(output);
1314 pa_assert(out_n_frames);
1315
1316 memset(&data, 0, sizeof(data));
1317
1318 data.data_in = (float*) ((uint8_t*) pa_memblock_acquire(input->memblock) + input->index);
1319 data.input_frames = (long int) in_n_frames;
1320
1321 data.data_out = (float*) ((uint8_t*) pa_memblock_acquire(output->memblock) + output->index);
1322 data.output_frames = (long int) *out_n_frames;
1323
1324 data.src_ratio = (double) r->o_ss.rate / r->i_ss.rate;
1325 data.end_of_input = 0;
1326
1327 pa_assert_se(src_process(r->src.state, &data) == 0);
1328 pa_assert((unsigned) data.input_frames_used == in_n_frames);
1329
1330 pa_memblock_release(input->memblock);
1331 pa_memblock_release(output->memblock);
1332
1333 *out_n_frames = (unsigned) data.output_frames_gen;
1334 }
1335
1336 static void libsamplerate_update_rates(pa_resampler *r) {
1337 pa_assert(r);
1338
1339 pa_assert_se(src_set_ratio(r->src.state, (double) r->o_ss.rate / r->i_ss.rate) == 0);
1340 }
1341
1342 static void libsamplerate_reset(pa_resampler *r) {
1343 pa_assert(r);
1344
1345 pa_assert_se(src_reset(r->src.state) == 0);
1346 }
1347
1348 static void libsamplerate_free(pa_resampler *r) {
1349 pa_assert(r);
1350
1351 if (r->src.state)
1352 src_delete(r->src.state);
1353 }
1354
1355 static int libsamplerate_init(pa_resampler *r) {
1356 int err;
1357
1358 pa_assert(r);
1359
1360 if (!(r->src.state = src_new(r->method, r->o_ss.channels, &err)))
1361 return -1;
1362
1363 r->impl_free = libsamplerate_free;
1364 r->impl_update_rates = libsamplerate_update_rates;
1365 r->impl_resample = libsamplerate_resample;
1366 r->impl_reset = libsamplerate_reset;
1367
1368 return 0;
1369 }
1370 #endif
1371
1372 /*** speex based implementation ***/
1373
1374 static void speex_resample_float(pa_resampler *r, const pa_memchunk *input, unsigned in_n_frames, pa_memchunk *output, unsigned *out_n_frames) {
1375 float *in, *out;
1376 uint32_t inf = in_n_frames, outf = *out_n_frames;
1377
1378 pa_assert(r);
1379 pa_assert(input);
1380 pa_assert(output);
1381 pa_assert(out_n_frames);
1382
1383 in = (float*) ((uint8_t*) pa_memblock_acquire(input->memblock) + input->index);
1384 out = (float*) ((uint8_t*) pa_memblock_acquire(output->memblock) + output->index);
1385
1386 pa_assert_se(speex_resampler_process_interleaved_float(r->speex.state, in, &inf, out, &outf) == 0);
1387
1388 pa_memblock_release(input->memblock);
1389 pa_memblock_release(output->memblock);
1390
1391 pa_assert(inf == in_n_frames);
1392 *out_n_frames = outf;
1393 }
1394
1395 static void speex_resample_int(pa_resampler *r, const pa_memchunk *input, unsigned in_n_frames, pa_memchunk *output, unsigned *out_n_frames) {
1396 int16_t *in, *out;
1397 uint32_t inf = in_n_frames, outf = *out_n_frames;
1398
1399 pa_assert(r);
1400 pa_assert(input);
1401 pa_assert(output);
1402 pa_assert(out_n_frames);
1403
1404 in = (int16_t*) ((uint8_t*) pa_memblock_acquire(input->memblock) + input->index);
1405 out = (int16_t*) ((uint8_t*) pa_memblock_acquire(output->memblock) + output->index);
1406
1407 pa_assert_se(speex_resampler_process_interleaved_int(r->speex.state, in, &inf, out, &outf) == 0);
1408
1409 pa_memblock_release(input->memblock);
1410 pa_memblock_release(output->memblock);
1411
1412 pa_assert(inf == in_n_frames);
1413 *out_n_frames = outf;
1414 }
1415
1416 static void speex_update_rates(pa_resampler *r) {
1417 pa_assert(r);
1418
1419 pa_assert_se(speex_resampler_set_rate(r->speex.state, r->i_ss.rate, r->o_ss.rate) == 0);
1420 }
1421
1422 static void speex_reset(pa_resampler *r) {
1423 pa_assert(r);
1424
1425 pa_assert_se(speex_resampler_reset_mem(r->speex.state) == 0);
1426 }
1427
1428 static void speex_free(pa_resampler *r) {
1429 pa_assert(r);
1430
1431 if (!r->speex.state)
1432 return;
1433
1434 speex_resampler_destroy(r->speex.state);
1435 }
1436
1437 static int speex_init(pa_resampler *r) {
1438 int q, err;
1439
1440 pa_assert(r);
1441
1442 r->impl_free = speex_free;
1443 r->impl_update_rates = speex_update_rates;
1444 r->impl_reset = speex_reset;
1445
1446 if (r->method >= PA_RESAMPLER_SPEEX_FIXED_BASE && r->method <= PA_RESAMPLER_SPEEX_FIXED_MAX) {
1447
1448 q = r->method - PA_RESAMPLER_SPEEX_FIXED_BASE;
1449 r->impl_resample = speex_resample_int;
1450
1451 } else {
1452 pa_assert(r->method >= PA_RESAMPLER_SPEEX_FLOAT_BASE && r->method <= PA_RESAMPLER_SPEEX_FLOAT_MAX);
1453
1454 q = r->method - PA_RESAMPLER_SPEEX_FLOAT_BASE;
1455 r->impl_resample = speex_resample_float;
1456 }
1457
1458 pa_log_info("Choosing speex quality setting %i.", q);
1459
1460 if (!(r->speex.state = speex_resampler_init(r->o_ss.channels, r->i_ss.rate, r->o_ss.rate, q, &err)))
1461 return -1;
1462
1463 return 0;
1464 }
1465
1466 /* Trivial implementation */
1467
1468 static void trivial_resample(pa_resampler *r, const pa_memchunk *input, unsigned in_n_frames, pa_memchunk *output, unsigned *out_n_frames) {
1469 size_t fz;
1470 unsigned o_index;
1471 void *src, *dst;
1472
1473 pa_assert(r);
1474 pa_assert(input);
1475 pa_assert(output);
1476 pa_assert(out_n_frames);
1477
1478 fz = r->w_sz * r->o_ss.channels;
1479
1480 src = (uint8_t*) pa_memblock_acquire(input->memblock) + input->index;
1481 dst = (uint8_t*) pa_memblock_acquire(output->memblock) + output->index;
1482
1483 for (o_index = 0;; o_index++, r->trivial.o_counter++) {
1484 unsigned j;
1485
1486 j = ((r->trivial.o_counter * r->i_ss.rate) / r->o_ss.rate);
1487 j = j > r->trivial.i_counter ? j - r->trivial.i_counter : 0;
1488
1489 if (j >= in_n_frames)
1490 break;
1491
1492 pa_assert(o_index * fz < pa_memblock_get_length(output->memblock));
1493
1494 memcpy((uint8_t*) dst + fz * o_index,
1495 (uint8_t*) src + fz * j, (int) fz);
1496 }
1497
1498 pa_memblock_release(input->memblock);
1499 pa_memblock_release(output->memblock);
1500
1501 *out_n_frames = o_index;
1502
1503 r->trivial.i_counter += in_n_frames;
1504
1505 /* Normalize counters */
1506 while (r->trivial.i_counter >= r->i_ss.rate) {
1507 pa_assert(r->trivial.o_counter >= r->o_ss.rate);
1508
1509 r->trivial.i_counter -= r->i_ss.rate;
1510 r->trivial.o_counter -= r->o_ss.rate;
1511 }
1512 }
1513
1514 static void trivial_update_rates_or_reset(pa_resampler *r) {
1515 pa_assert(r);
1516
1517 r->trivial.i_counter = 0;
1518 r->trivial.o_counter = 0;
1519 }
1520
1521 static int trivial_init(pa_resampler*r) {
1522 pa_assert(r);
1523
1524 r->trivial.o_counter = r->trivial.i_counter = 0;
1525
1526 r->impl_resample = trivial_resample;
1527 r->impl_update_rates = trivial_update_rates_or_reset;
1528 r->impl_reset = trivial_update_rates_or_reset;
1529
1530 return 0;
1531 }
1532
1533 /* Peak finder implementation */
1534
1535 static void peaks_resample(pa_resampler *r, const pa_memchunk *input, unsigned in_n_frames, pa_memchunk *output, unsigned *out_n_frames) {
1536 size_t fz;
1537 unsigned o_index;
1538 void *src, *dst;
1539 unsigned start = 0;
1540
1541 pa_assert(r);
1542 pa_assert(input);
1543 pa_assert(output);
1544 pa_assert(out_n_frames);
1545
1546 fz = r->w_sz * r->o_ss.channels;
1547
1548 src = (uint8_t*) pa_memblock_acquire(input->memblock) + input->index;
1549 dst = (uint8_t*) pa_memblock_acquire(output->memblock) + output->index;
1550
1551 for (o_index = 0;; o_index++, r->peaks.o_counter++) {
1552 unsigned j;
1553
1554 j = ((r->peaks.o_counter * r->i_ss.rate) / r->o_ss.rate);
1555
1556 if (j > r->peaks.i_counter)
1557 j -= r->peaks.i_counter;
1558 else
1559 j = 0;
1560
1561 pa_assert(o_index * fz < pa_memblock_get_length(output->memblock));
1562
1563 if (r->work_format == PA_SAMPLE_S16NE) {
1564 unsigned i, c;
1565 int16_t *s = (int16_t*) ((uint8_t*) src + fz * start);
1566 int16_t *d = (int16_t*) ((uint8_t*) dst + fz * o_index);
1567
1568 for (i = start; i <= j && i < in_n_frames; i++)
1569
1570 for (c = 0; c < r->o_ss.channels; c++, s++) {
1571 int16_t n;
1572
1573 n = (int16_t) (*s < 0 ? -*s : *s);
1574
1575 if (PA_UNLIKELY(n > r->peaks.max_i[c]))
1576 r->peaks.max_i[c] = n;
1577 }
1578
1579 if (i >= in_n_frames)
1580 break;
1581
1582 for (c = 0; c < r->o_ss.channels; c++, d++) {
1583 *d = r->peaks.max_i[c];
1584 r->peaks.max_i[c] = 0;
1585 }
1586
1587 } else {
1588 unsigned i, c;
1589 float *s = (float*) ((uint8_t*) src + fz * start);
1590 float *d = (float*) ((uint8_t*) dst + fz * o_index);
1591
1592 pa_assert(r->work_format == PA_SAMPLE_FLOAT32NE);
1593
1594 for (i = start; i <= j && i < in_n_frames; i++)
1595 for (c = 0; c < r->o_ss.channels; c++, s++) {
1596 float n = fabsf(*s);
1597
1598 if (n > r->peaks.max_f[c])
1599 r->peaks.max_f[c] = n;
1600 }
1601
1602 if (i >= in_n_frames)
1603 break;
1604
1605 for (c = 0; c < r->o_ss.channels; c++, d++) {
1606 *d = r->peaks.max_f[c];
1607 r->peaks.max_f[c] = 0;
1608 }
1609 }
1610
1611 start = j;
1612 }
1613
1614 pa_memblock_release(input->memblock);
1615 pa_memblock_release(output->memblock);
1616
1617 *out_n_frames = o_index;
1618
1619 r->peaks.i_counter += in_n_frames;
1620
1621 /* Normalize counters */
1622 while (r->peaks.i_counter >= r->i_ss.rate) {
1623 pa_assert(r->peaks.o_counter >= r->o_ss.rate);
1624
1625 r->peaks.i_counter -= r->i_ss.rate;
1626 r->peaks.o_counter -= r->o_ss.rate;
1627 }
1628 }
1629
1630 static void peaks_update_rates_or_reset(pa_resampler *r) {
1631 pa_assert(r);
1632
1633 r->peaks.i_counter = 0;
1634 r->peaks.o_counter = 0;
1635 }
1636
1637 static int peaks_init(pa_resampler*r) {
1638 pa_assert(r);
1639
1640 r->peaks.o_counter = r->peaks.i_counter = 0;
1641 memset(r->peaks.max_i, 0, sizeof(r->peaks.max_i));
1642 memset(r->peaks.max_f, 0, sizeof(r->peaks.max_f));
1643
1644 r->impl_resample = peaks_resample;
1645 r->impl_update_rates = peaks_update_rates_or_reset;
1646 r->impl_reset = peaks_update_rates_or_reset;
1647
1648 return 0;
1649 }
1650
1651 /*** ffmpeg based implementation ***/
1652
1653 static void ffmpeg_resample(pa_resampler *r, const pa_memchunk *input, unsigned in_n_frames, pa_memchunk *output, unsigned *out_n_frames) {
1654 unsigned used_frames = 0, c;
1655
1656 pa_assert(r);
1657 pa_assert(input);
1658 pa_assert(output);
1659 pa_assert(out_n_frames);
1660
1661 for (c = 0; c < r->o_ss.channels; c++) {
1662 unsigned u;
1663 pa_memblock *b, *w;
1664 int16_t *p, *t, *k, *q, *s;
1665 int consumed_frames;
1666 unsigned in, l;
1667
1668 /* Allocate a new block */
1669 b = pa_memblock_new(r->mempool, r->ffmpeg.buf[c].length + in_n_frames * sizeof(int16_t));
1670 p = pa_memblock_acquire(b);
1671
1672 /* Copy the remaining data into it */
1673 l = (unsigned) r->ffmpeg.buf[c].length;
1674 if (r->ffmpeg.buf[c].memblock) {
1675 t = (int16_t*) ((uint8_t*) pa_memblock_acquire(r->ffmpeg.buf[c].memblock) + r->ffmpeg.buf[c].index);
1676 memcpy(p, t, l);
1677 pa_memblock_release(r->ffmpeg.buf[c].memblock);
1678 pa_memblock_unref(r->ffmpeg.buf[c].memblock);
1679 pa_memchunk_reset(&r->ffmpeg.buf[c]);
1680 }
1681
1682 /* Now append the new data, splitting up channels */
1683 t = ((int16_t*) ((uint8_t*) pa_memblock_acquire(input->memblock) + input->index)) + c;
1684 k = (int16_t*) ((uint8_t*) p + l);
1685 for (u = 0; u < in_n_frames; u++) {
1686 *k = *t;
1687 t += r->o_ss.channels;
1688 k ++;
1689 }
1690 pa_memblock_release(input->memblock);
1691
1692 /* Calculate the resulting number of frames */
1693 in = (unsigned) in_n_frames + l / (unsigned) sizeof(int16_t);
1694
1695 /* Allocate buffer for the result */
1696 w = pa_memblock_new(r->mempool, *out_n_frames * sizeof(int16_t));
1697 q = pa_memblock_acquire(w);
1698
1699 /* Now, resample */
1700 used_frames = (unsigned) av_resample(r->ffmpeg.state,
1701 q, p,
1702 &consumed_frames,
1703 (int) in, (int) *out_n_frames,
1704 c >= (unsigned) (r->o_ss.channels-1));
1705
1706 pa_memblock_release(b);
1707
1708 /* Now store the remaining samples away */
1709 pa_assert(consumed_frames <= (int) in);
1710 if (consumed_frames < (int) in) {
1711 r->ffmpeg.buf[c].memblock = b;
1712 r->ffmpeg.buf[c].index = (size_t) consumed_frames * sizeof(int16_t);
1713 r->ffmpeg.buf[c].length = (size_t) (in - (unsigned) consumed_frames) * sizeof(int16_t);
1714 } else
1715 pa_memblock_unref(b);
1716
1717 /* And place the results in the output buffer */
1718 s = (short*) ((uint8_t*) pa_memblock_acquire(output->memblock) + output->index) + c;
1719 for (u = 0; u < used_frames; u++) {
1720 *s = *q;
1721 q++;
1722 s += r->o_ss.channels;
1723 }
1724 pa_memblock_release(output->memblock);
1725 pa_memblock_release(w);
1726 pa_memblock_unref(w);
1727 }
1728
1729 *out_n_frames = used_frames;
1730 }
1731
1732 static void ffmpeg_free(pa_resampler *r) {
1733 unsigned c;
1734
1735 pa_assert(r);
1736
1737 if (r->ffmpeg.state)
1738 av_resample_close(r->ffmpeg.state);
1739
1740 for (c = 0; c < PA_ELEMENTSOF(r->ffmpeg.buf); c++)
1741 if (r->ffmpeg.buf[c].memblock)
1742 pa_memblock_unref(r->ffmpeg.buf[c].memblock);
1743 }
1744
1745 static int ffmpeg_init(pa_resampler *r) {
1746 unsigned c;
1747
1748 pa_assert(r);
1749
1750 /* We could probably implement different quality levels by
1751 * adjusting the filter parameters here. However, ffmpeg
1752 * internally only uses these hardcoded values, so let's use them
1753 * here for now as well until ffmpeg makes this configurable. */
1754
1755 if (!(r->ffmpeg.state = av_resample_init((int) r->o_ss.rate, (int) r->i_ss.rate, 16, 10, 0, 0.8)))
1756 return -1;
1757
1758 r->impl_free = ffmpeg_free;
1759 r->impl_resample = ffmpeg_resample;
1760
1761 for (c = 0; c < PA_ELEMENTSOF(r->ffmpeg.buf); c++)
1762 pa_memchunk_reset(&r->ffmpeg.buf[c]);
1763
1764 return 0;
1765 }
1766
1767 /*** copy (noop) implementation ***/
1768
1769 static int copy_init(pa_resampler *r) {
1770 pa_assert(r);
1771
1772 pa_assert(r->o_ss.rate == r->i_ss.rate);
1773
1774 return 0;
1775 }