diff options
author | Paul B Mahol <onemda@gmail.com> | 2016-08-27 13:44:21 +0200 |
---|---|---|
committer | Paul B Mahol <onemda@gmail.com> | 2016-08-27 13:48:14 +0200 |
commit | b2c6a11fb60429c9698b9d1e8538d32d741e68c5 (patch) | |
tree | a47d5b529472c9f81af54e2ee281ebe16ab18560 /libavfilter/vf_atadenoise.c | |
parent | 69abf4f93cb67cc52ff55f318ae09f261e7ad27e (diff) | |
download | ffmpeg-b2c6a11fb60429c9698b9d1e8538d32d741e68c5.tar.gz |
avfilter/vf_atadenoise: add planes option
Make possible filtering only some planes.
Diffstat (limited to 'libavfilter/vf_atadenoise.c')
-rw-r--r-- | libavfilter/vf_atadenoise.c | 15 |
1 files changed, 15 insertions, 0 deletions
diff --git a/libavfilter/vf_atadenoise.c b/libavfilter/vf_atadenoise.c index cdea298a82..07da792a84 100644 --- a/libavfilter/vf_atadenoise.c +++ b/libavfilter/vf_atadenoise.c @@ -25,6 +25,7 @@ * David Bartovčak and Miroslav Vrankić */ +#include "libavutil/imgutils.h" #include "libavutil/opt.h" #include "libavutil/pixdesc.h" #include "avfilter.h" @@ -44,6 +45,7 @@ typedef struct ATADenoiseContext { float fthra[4], fthrb[4]; int thra[4], thrb[4]; + int planes; int nb_planes; int planewidth[4]; int planeheight[4]; @@ -68,6 +70,7 @@ static const AVOption atadenoise_options[] = { { "2a", "set threshold A for 3rd plane", OFFSET(fthra[2]), AV_OPT_TYPE_FLOAT, {.dbl=0.02}, 0, 0.3, FLAGS }, { "2b", "set threshold B for 3rd plane", OFFSET(fthrb[2]), AV_OPT_TYPE_FLOAT, {.dbl=0.04}, 0, 5.0, FLAGS }, { "s", "set how many frames to use", OFFSET(size), AV_OPT_TYPE_INT, {.i64=9}, 5, SIZE, FLAGS }, + { "p", "set what planes to filter", OFFSET(planes), AV_OPT_TYPE_FLAGS, {.i64=7}, 0, 15, FLAGS }, { NULL } }; @@ -141,6 +144,12 @@ static int filter_slice8(AVFilterContext *ctx, void *arg, int jobnr, int nb_jobs const int *linesize = (const int *)s->linesize[p]; const uint8_t *srcf[SIZE]; + if (!((1 << p) & s->planes)) { + av_image_copy_plane(dst, out->linesize[p], src, in->linesize[p], + w, slice_end - slice_start); + continue; + } + for (i = 0; i < size; i++) srcf[i] = data[i] + slice_start * linesize[i]; @@ -212,6 +221,12 @@ static int filter_slice16(AVFilterContext *ctx, void *arg, int jobnr, int nb_job const int *linesize = (const int *)s->linesize[p]; const uint16_t *srcf[SIZE]; + if (!((1 << p) & s->planes)) { + av_image_copy_plane((uint8_t *)dst, out->linesize[p], (uint8_t *)src, in->linesize[p], + w * 2, slice_end - slice_start); + continue; + } + for (i = 0; i < s->size; i++) srcf[i] = (const uint16_t *)(data[i] + slice_start * linesize[i]); |