82742294 |
#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
|
2eec9812 |
void RENAME(swri_noise_shaping)(SwrContext *s, AudioData *dsts, const AudioData *srcs, const AudioData *noises, int count){ |
cc4a4172 |
int pos = s->dither.ns_pos;
int i, j, ch; |
3ab19706 |
int taps = s->dither.ns_taps;
float S = s->dither.ns_scale;
float S_1 = s->dither.ns_scale_1; |
82742294 |
|
c8737d34 |
av_assert2((taps&3) != 2);
av_assert2((taps&3) != 3 || s->dither.ns_coeffs[taps] == 0);
|
82742294 |
for (ch=0; ch<srcs->ch_count; ch++) { |
fca51256 |
const float *noise = ((const float *)noises->ch[ch]) + s->dither.noise_pos; |
558aa6ca |
const DELEM *src = (const DELEM*)srcs->ch[ch];
DELEM *dst = (DELEM*)dsts->ch[ch]; |
68ff7d26 |
float *ns_errors = s->dither.ns_errors[ch]; |
ef7fdc8c |
const float *ns_coeffs = s->dither.ns_coeffs; |
3ab19706 |
pos = s->dither.ns_pos; |
82742294 |
for (i=0; i<count; i++) { |
558aa6ca |
double d1, d = src[i]*S_1; |
c8737d34 |
for(j=0; j<taps-2; j+=4) {
d -= ns_coeffs[j ] * ns_errors[pos + j ]
+ns_coeffs[j + 1] * ns_errors[pos + j + 1]
+ns_coeffs[j + 2] * ns_errors[pos + j + 2]
+ns_coeffs[j + 3] * ns_errors[pos + j + 3];
}
if(j < taps) |
ef7fdc8c |
d -= ns_coeffs[j] * ns_errors[pos + j]; |
b5d9e5d0 |
pos = pos ? pos - 1 : taps - 1; |
0b60fc94 |
d1 = rint(d + noise[i]); |
68ff7d26 |
ns_errors[pos + taps] = ns_errors[pos] = d1 - d; |
0b60fc94 |
d1 *= S; |
82742294 |
CLIP(d1); |
558aa6ca |
dst[i] = d1; |
82742294 |
}
}
|
3ab19706 |
s->dither.ns_pos = pos; |
82742294 |
}
#undef RENAME
#undef DELEM
#undef CLIP |