avfilter/vf_xmedian: remove limitation of only odd number of inputs

This commit is contained in:
Paul B Mahol 2019-06-02 11:03:08 +02:00
parent 415886588f
commit cbaa60329a
2 changed files with 13 additions and 10 deletions

View File

@ -18465,9 +18465,10 @@ Pick median pixels from several input videos.
The filter accept the following options:
@table @option
@item nb_inputs
Set number of inputs. This must be odd number.
@item inputs
Set number of inputs.
Default is 3. Allowed range is from 3 to 255.
If number of inputs is even number, than result will be mean value between two median values.
@item planes
Set which planes to filter. Default value is @code{15}, by which all planes are processed.

View File

@ -88,10 +88,6 @@ static av_cold int init(AVFilterContext *ctx)
XMedianContext *s = ctx->priv;
int ret;
if (!(s->nb_inputs & 1))
av_log(s, AV_LOG_WARNING, "nb_intputs: %d is not odd number.\n", s->nb_inputs);
s->nb_inputs = s->nb_inputs | 1;
s->radius = s->nb_inputs / 2;
s->frames = av_calloc(s->nb_inputs, sizeof(*s->frames));
if (!s->frames)
@ -156,7 +152,10 @@ static int median_frames16(AVFilterContext *ctx, void *arg, int jobnr, int nb_jo
}
AV_QSORT(values, nb_inputs, int, comparei);
dst[x] = values[radius];
if (radius & 1)
dst[x] = values[radius];
else
dst[x] = (values[radius] + values[radius - 1]) >> 1;
}
dst += out->linesize[p] / 2;
@ -195,7 +194,10 @@ static int median_frames8(AVFilterContext *ctx, void *arg, int jobnr, int nb_job
values[i] = in[i]->data[p][y * in[i]->linesize[p] + x];
AV_QSORT(values, nb_inputs, int, comparei);
dst[x] = values[radius];
if (radius & 1)
dst[x] = values[radius];
else
dst[x] = (values[radius] + values[radius - 1]) >> 1;
}
dst += out->linesize[p];
@ -319,8 +321,8 @@ static int activate(AVFilterContext *ctx)
#define FLAGS AV_OPT_FLAG_VIDEO_PARAM | AV_OPT_FLAG_FILTERING_PARAM
static const AVOption xmedian_options[] = {
{ "nb_inputs", "set number of inputs", OFFSET(nb_inputs), AV_OPT_TYPE_INT, {.i64=3}, 3, 255, .flags = FLAGS },
{ "planes", "set planes to filter", OFFSET(planes), AV_OPT_TYPE_INT, {.i64=15}, 0, 15, .flags = FLAGS },
{ "inputs", "set number of inputs", OFFSET(nb_inputs), AV_OPT_TYPE_INT, {.i64=3}, 3, 255, .flags = FLAGS },
{ "planes", "set planes to filter", OFFSET(planes), AV_OPT_TYPE_INT, {.i64=15}, 0, 15, .flags = FLAGS },
{ NULL },
};