aboutsummaryrefslogtreecommitdiffstats
path: root/libswresample/dither_template.c
blob: 7e99c507004d83afc017845bc6082787cc59a07c (plain) (blame)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
#if defined(TEMPLATE_DITHER_DBL)
#    define RENAME(N) N ## _double
#    define DELEM  double
#    define CLIP(v)

#elif defined(TEMPLATE_DITHER_FLT)
#    define RENAME(N) N ## _float
#    define DELEM  float
#    define CLIP(v)

#elif defined(TEMPLATE_DITHER_S32)
#    define RENAME(N) N ## _int32
#    define DELEM  int32_t
#    define CLIP(v) v = FFMAX(FFMIN(v, INT32_MAX), INT32_MIN)

#elif defined(TEMPLATE_DITHER_S16)
#    define RENAME(N) N ## _int16
#    define DELEM  int16_t
#    define CLIP(v) v = FFMAX(FFMIN(v, INT16_MAX), INT16_MIN)

#else
ERROR
#endif

void RENAME(swri_noise_shaping)(SwrContext *s, AudioData *srcs, AudioData *noises, int count){
    int i, j, pos, ch;
    int taps  = s->ns_taps;
    float S   = s->ns_scale;
    float S_1 = s->ns_scale_1;

    for (ch=0; ch<srcs->ch_count; ch++) {
        const float *noise = ((const float *)noises->ch[ch]) + s->dither_pos;
        DELEM *data = (DELEM*)srcs->ch[ch];
        pos  = s->ns_pos;
        for (i=0; i<count; i++) {
            double d1, d = data[i];
            for(j=0; j<taps; j++)
                d -= s->ns_coeffs[j] * s->ns_errors[ch][pos + j];
            pos = pos ? pos - 1 : pos - 1 + taps;
            d1 = rint((d + noise[i]) * S_1)*S;
            s->ns_errors[ch][pos + taps] = s->ns_errors[ch][pos] = d1 - d;
            CLIP(d1);
            data[i] = d1;
        }
    }

    s->ns_pos = pos;
}

#undef RENAME
#undef DELEM
#undef CLIP