swr: add seperate in/out pointers to the noise shaping code
[ffmpeg.git] / libswresample / dither_template.c
1
2 #if defined(TEMPLATE_DITHER_DBL)
3 #    define RENAME(N) N ## _double
4 #    define DELEM  double
5 #    define CLIP(v)
6
7 #elif defined(TEMPLATE_DITHER_FLT)
8 #    define RENAME(N) N ## _float
9 #    define DELEM  float
10 #    define CLIP(v)
11
12 #elif defined(TEMPLATE_DITHER_S32)
13 #    define RENAME(N) N ## _int32
14 #    define DELEM  int32_t
15 #    define CLIP(v) v = FFMAX(FFMIN(v, INT32_MAX), INT32_MIN)
16
17 #elif defined(TEMPLATE_DITHER_S16)
18 #    define RENAME(N) N ## _int16
19 #    define DELEM  int16_t
20 #    define CLIP(v) v = FFMAX(FFMIN(v, INT16_MAX), INT16_MIN)
21
22 #else
23 ERROR
24 #endif
25
26 void RENAME(swri_noise_shaping)(SwrContext *s, AudioData *dsts, const AudioData *srcs, AudioData *noises, int count){
27     int i, j, pos, ch;
28     int taps  = s->dither.ns_taps;
29     float S   = s->dither.ns_scale;
30     float S_1 = s->dither.ns_scale_1;
31
32     for (ch=0; ch<srcs->ch_count; ch++) {
33         const float *noise = ((const float *)noises->ch[ch]) + s->dither.noise_pos;
34         const DELEM *src = (const DELEM*)srcs->ch[ch];
35         DELEM *dst = (DELEM*)dsts->ch[ch];
36         pos  = s->dither.ns_pos;
37         for (i=0; i<count; i++) {
38             double d1, d = src[i]*S_1;
39             for(j=0; j<taps; j++)
40                 d -= s->dither.ns_coeffs[j] * s->dither.ns_errors[ch][pos + j];
41             pos = pos ? pos - 1 : pos - 1 + taps;
42             d1 = rint(d + noise[i]);
43             s->dither.ns_errors[ch][pos + taps] = s->dither.ns_errors[ch][pos] = d1 - d;
44             d1 *= S;
45             CLIP(d1);
46             dst[i] = d1;
47         }
48     }
49
50     s->dither.ns_pos = pos;
51 }
52
53 #undef RENAME
54 #undef DELEM
55 #undef CLIP