From 6af67dcc3519b699de8d4036bfe493aea692c190 Mon Sep 17 00:00:00 2001 From: Marton Balint Date: Sat, 2 Mar 2019 00:10:21 +0100 Subject: avfilter/af_astats: add support for optimized min/max/peak calculation Signed-off-by: Marton Balint --- libavfilter/af_astats.c | 26 +++++++++++++++++++++----- 1 file changed, 21 insertions(+), 5 deletions(-) diff --git a/libavfilter/af_astats.c b/libavfilter/af_astats.c index 9915a7965e..92368793c2 100644 --- a/libavfilter/af_astats.c +++ b/libavfilter/af_astats.c @@ -49,6 +49,8 @@ #define MEASURE_ZERO_CROSSINGS_RATE (1 << 17) #define MEASURE_NUMBER_OF_SAMPLES (1 << 18) +#define MEASURE_MINMAXPEAK (MEASURE_MIN_LEVEL | MEASURE_MAX_LEVEL | MEASURE_PEAK_LEVEL) + typedef struct ChannelStats { double last; double last_non_zero; @@ -216,6 +218,14 @@ static void bit_depth(AudioStatsContext *s, uint64_t mask, uint64_t imask, AVRat depth->num++; } +static inline void update_minmax(AudioStatsContext *s, ChannelStats *p, double d) +{ + if (d < p->min) + p->min = d; + if (d > p->max) + p->max = d; +} + static inline void update_stat(AudioStatsContext *s, ChannelStats *p, double d, double nd, int64_t i) { if (d < p->min) { @@ -406,26 +416,32 @@ static void set_metadata(AudioStatsContext *s, AVDictionary **metadata) set_meta(metadata, 0, "Overall.Number_of_samples", "%f", nb_samples / s->nb_channels); } -#define UPDATE_STATS_P(type, double_sample, normalized_sample, int_sample) \ +#define UPDATE_STATS_P(type, update_func, channel_func) \ for (int c = 0; c < channels; c++) { \ ChannelStats *p = &s->chstats[c]; \ const type *src = (const type *)data[c]; \ const type * const srcend = src + samples; \ for (; src < srcend; src++) \ - update_stat(s, p, double_sample, normalized_sample, int_sample); \ + update_func; \ + channel_func; \ } -#define UPDATE_STATS_I(type, double_sample, normalized_sample, int_sample) \ +#define UPDATE_STATS_I(type, update_func, channel_func) \ for (int c = 0; c < channels; c++) { \ ChannelStats *p = &s->chstats[c]; \ const type *src = (const type *)data[0]; \ const type * const srcend = src + samples * channels; \ for (src += c; src < srcend; src += channels) \ - update_stat(s, p, double_sample, normalized_sample, int_sample); \ + update_func; \ + channel_func; \ } #define UPDATE_STATS(planar, type, sample, normalizer_suffix, int_sample) \ - UPDATE_STATS_##planar(type, sample, sample normalizer_suffix, int_sample); + if ((s->measure_overall | s->measure_perchannel) & ~MEASURE_MINMAXPEAK) { \ + UPDATE_STATS_##planar(type, update_stat(s, p, sample, sample normalizer_suffix, int_sample), ); \ + } else { \ + UPDATE_STATS_##planar(type, update_minmax(s, p, sample), p->nmin = p->min normalizer_suffix; p->nmax = p->max normalizer_suffix;); \ + } static int filter_frame(AVFilterLink *inlink, AVFrame *buf) { -- cgit v1.2.3