vf_atadenoise.c File Reference

Adaptive Temporal Averaging Denoiser, based on paper "Video Denoising Based on Adaptive Temporal Averaging" by David Bartovčak and Miroslav Vrankić More...

#include "libavutil/imgutils.h"
#include "libavutil/opt.h"
#include "libavutil/pixdesc.h"
#include "avfilter.h"
#include "bufferqueue.h"
#include "atadenoise.h"
#include "filters.h"
#include "video.h"

Go to the source code of this file.

Data Structures
struct	ATADenoiseContext
struct	ThreadData
	Used for passing data between threads. More...

Macros
#define	FF_BUFQUEUE_SIZE   129
#define	SIZE   FF_BUFQUEUE_SIZE
#define	OFFSET(x)
#define	FLAGS   AV_OPT_FLAG_VIDEO_PARAM|AV_OPT_FLAG_FILTERING_PARAM|AV_OPT_FLAG_RUNTIME_PARAM
#define	VF   AV_OPT_FLAG_VIDEO_PARAM|AV_OPT_FLAG_FILTERING_PARAM
#define	WFILTER_ROW(type, name)
#define	WFILTER_ROW_SERIAL(type, name)
#define	FILTER_ROW(type, name)
#define	FILTER_ROW_SERIAL(type, name)

Functions
	AVFILTER_DEFINE_CLASS (atadenoise)
static av_cold int	init (AVFilterContext *ctx)
static int	filter_slice (AVFilterContext *ctx, void *arg, int jobnr, int nb_jobs)
static int	config_input (AVFilterLink *inlink)
static int	filter_frame (AVFilterLink *inlink, AVFrame *buf)
static int	request_frame (AVFilterLink *outlink)
static av_cold void	uninit (AVFilterContext *ctx)
static int	process_command (AVFilterContext *ctx, const char *cmd, const char *arg, char *res, int res_len, int flags)

Variables
static const AVOption	atadenoise_options []
static enum AVPixelFormat	pixel_fmts []
static const AVFilterPad	inputs []
static const AVFilterPad	outputs []
const FFFilter	ff_vf_atadenoise

Detailed Description

Adaptive Temporal Averaging Denoiser, based on paper "Video Denoising Based on Adaptive Temporal Averaging" by David Bartovčak and Miroslav Vrankić

Definition in file vf_atadenoise.c.

Macro Definition Documentation

FF_BUFQUEUE_SIZE

#define FF_BUFQUEUE_SIZE   129

Definition at line 33 of file vf_atadenoise.c.

SIZE

#define SIZE   FF_BUFQUEUE_SIZE

Definition at line 40 of file vf_atadenoise.c.

OFFSET

#define OFFSET

(

x

)

Value:

offsetof(ATADenoiseContext, x)

Definition at line 68 of file vf_atadenoise.c.

FLAGS

#define FLAGS   AV_OPT_FLAG_VIDEO_PARAM|AV_OPT_FLAG_FILTERING_PARAM|AV_OPT_FLAG_RUNTIME_PARAM

Definition at line 69 of file vf_atadenoise.c.

VF

#define VF   AV_OPT_FLAG_VIDEO_PARAM|AV_OPT_FLAG_FILTERING_PARAM

Definition at line 70 of file vf_atadenoise.c.

WFILTER_ROW

#define WFILTER_ROW	(		type,
			name )

Definition at line 140 of file vf_atadenoise.c.

WFILTER_ROW_SERIAL

#define WFILTER_ROW_SERIAL	(		type,
			name )

Definition at line 187 of file vf_atadenoise.c.

FILTER_ROW

#define FILTER_ROW	(		type,
			name )

Definition at line 237 of file vf_atadenoise.c.

FILTER_ROW_SERIAL

#define FILTER_ROW_SERIAL	(		type,
			name )

Definition at line 284 of file vf_atadenoise.c.

Function Documentation

AVFILTER_DEFINE_CLASS()

AVFILTER_DEFINE_CLASS

(

atadenoise

)

init()

static av_cold int init ( AVFilterContext * ctx )

static

Definition at line 122 of file vf_atadenoise.c.

filter_slice()

static int filter_slice ( AVFilterContext * ctx, void * arg, int jobnr, int nb_jobs )

static

Definition at line 334 of file vf_atadenoise.c.

Referenced by blend_frame(), blend_frames(), config_input(), filter(), filter(), filter_frame(), filter_frame(), filter_frame(), filter_frame(), filter_frame(), filter_frame(), filter_frame(), filter_frame(), filter_frame(), filter_frame(), filter_frame(), filter_frame(), filter_frame(), filter_frame(), filter_frame(), filter_frame(), filter_slice(), filter_slice(), get_frame(), process_frame(), and process_frame().

config_input()

static int config_input ( AVFilterLink * inlink )

static

Definition at line 381 of file vf_atadenoise.c.

Referenced by process_command().

filter_frame()

static int filter_frame ( AVFilterLink * inlink, AVFrame * buf )

static

Definition at line 436 of file vf_atadenoise.c.

Referenced by request_frame().

request_frame()

static int request_frame ( AVFilterLink * outlink )

static

Definition at line 505 of file vf_atadenoise.c.

uninit()

static av_cold void uninit ( AVFilterContext * ctx )

static

Definition at line 525 of file vf_atadenoise.c.

process_command()

static int process_command ( AVFilterContext * ctx, const char * cmd, const char * arg, char * res, int res_len, int flags )

static

Definition at line 532 of file vf_atadenoise.c.

Variable Documentation

atadenoise_options

const AVOption atadenoise_options[]

static

Initial value:

= {
    { "0a", "set threshold A for 1st plane", OFFSET(fthra[0]), AV_OPT_TYPE_FLOAT, {.dbl=0.02}, 0, 0.3, FLAGS },
    { "0b", "set threshold B for 1st plane", OFFSET(fthrb[0]), AV_OPT_TYPE_FLOAT, {.dbl=0.04}, 0, 5.0, FLAGS },
    { "1a", "set threshold A for 2nd plane", OFFSET(fthra[1]), AV_OPT_TYPE_FLOAT, {.dbl=0.02}, 0, 0.3, FLAGS },
    { "1b", "set threshold B for 2nd plane", OFFSET(fthrb[1]), AV_OPT_TYPE_FLOAT, {.dbl=0.04}, 0, 5.0, FLAGS },
    { "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, VF    },
    { "p",  "set what planes to filter",     OFFSET(planes),   AV_OPT_TYPE_FLAGS, {.i64=7},    0, 15,  FLAGS },
    { "a",  "set variant of algorithm",      OFFSET(algorithm),AV_OPT_TYPE_INT,   {.i64=PARALLEL},  0, NB_ATAA-1, FLAGS, .unit = "a" },
    { "p",  "parallel",                      0,                AV_OPT_TYPE_CONST, {.i64=PARALLEL},  0, 0,         FLAGS, .unit = "a" },
    { "s",  "serial",                        0,                AV_OPT_TYPE_CONST, {.i64=SERIAL},    0, 0,         FLAGS, .unit = "a" },
    { "0s", "set sigma for 1st plane",       OFFSET(sigma[0]), AV_OPT_TYPE_FLOAT, {.dbl=INT16_MAX}, 0, INT16_MAX, FLAGS },
    { "1s", "set sigma for 2nd plane",       OFFSET(sigma[1]), AV_OPT_TYPE_FLOAT, {.dbl=INT16_MAX}, 0, INT16_MAX, FLAGS },
    { "2s", "set sigma for 3rd plane",       OFFSET(sigma[2]), AV_OPT_TYPE_FLOAT, {.dbl=INT16_MAX}, 0, INT16_MAX, FLAGS },
    { NULL }
}

Definition at line 72 of file vf_atadenoise.c.

pixel_fmts

enum AVPixelFormat pixel_fmts[]

static

Initial value:

= {
    AV_PIX_FMT_GRAY8,
    AV_PIX_FMT_GRAY9,
    AV_PIX_FMT_GRAY10,
    AV_PIX_FMT_GRAY12,
    AV_PIX_FMT_GRAY14,
    AV_PIX_FMT_GRAY16,
    AV_PIX_FMT_YUV410P, AV_PIX_FMT_YUV411P,
    AV_PIX_FMT_YUV420P, AV_PIX_FMT_YUV422P,
    AV_PIX_FMT_YUV440P, AV_PIX_FMT_YUV444P,
    AV_PIX_FMT_YUVJ420P, AV_PIX_FMT_YUVJ422P,
    AV_PIX_FMT_YUVJ440P, AV_PIX_FMT_YUVJ444P,
    AV_PIX_FMT_YUVJ411P,
    AV_PIX_FMT_YUV420P9, AV_PIX_FMT_YUV422P9, AV_PIX_FMT_YUV444P9,
    AV_PIX_FMT_YUV420P10, AV_PIX_FMT_YUV422P10, AV_PIX_FMT_YUV444P10,
    AV_PIX_FMT_YUV440P10,
    AV_PIX_FMT_YUV444P12, AV_PIX_FMT_YUV422P12, AV_PIX_FMT_YUV420P12,
    AV_PIX_FMT_YUV440P12,
    AV_PIX_FMT_YUV444P14, AV_PIX_FMT_YUV422P14, AV_PIX_FMT_YUV420P14,
    AV_PIX_FMT_YUV420P16, AV_PIX_FMT_YUV422P16, AV_PIX_FMT_YUV444P16,
    AV_PIX_FMT_GBRP, AV_PIX_FMT_GBRP9, AV_PIX_FMT_GBRP10,
    AV_PIX_FMT_GBRP12, AV_PIX_FMT_GBRP14, AV_PIX_FMT_GBRP16,
    AV_PIX_FMT_YUVA420P,  AV_PIX_FMT_YUVA422P,   AV_PIX_FMT_YUVA444P,
    AV_PIX_FMT_YUVA444P9, AV_PIX_FMT_YUVA444P10, AV_PIX_FMT_YUVA444P12, AV_PIX_FMT_YUVA444P16,
    AV_PIX_FMT_YUVA422P9, AV_PIX_FMT_YUVA422P10, AV_PIX_FMT_YUVA422P12, AV_PIX_FMT_YUVA422P16,
    AV_PIX_FMT_YUVA420P9, AV_PIX_FMT_YUVA420P10, AV_PIX_FMT_YUVA420P16,
    AV_PIX_FMT_GBRAP,     AV_PIX_FMT_GBRAP10,    AV_PIX_FMT_GBRAP12,    AV_PIX_FMT_GBRAP16,
    AV_PIX_FMT_NONE
}

Definition at line 92 of file vf_atadenoise.c.

inputs

const AVFilterPad inputs[]

static

Initial value:

= {
    {
        .name         = "default",
        .type         = AVMEDIA_TYPE_VIDEO,
        .filter_frame = filter_frame,
        .config_props = config_input,
    },
}

Definition at line 547 of file vf_atadenoise.c.

outputs

const AVFilterPad outputs[]

static

Initial value:

= {
    {
        .name          = "default",
        .type          = AVMEDIA_TYPE_VIDEO,
        .request_frame = request_frame,
    },
}

Definition at line 556 of file vf_atadenoise.c.

ff_vf_atadenoise

const FFFilter ff_vf_atadenoise

Initial value:

= {
    .p.name        = "atadenoise",
    .p.description = NULL_IF_CONFIG_SMALL("Apply an Adaptive Temporal Averaging Denoiser."),
    .p.priv_class  = &atadenoise_class,
    .p.flags       = AVFILTER_FLAG_SUPPORT_TIMELINE_INTERNAL | AVFILTER_FLAG_SLICE_THREADS,
    .priv_size     = sizeof(ATADenoiseContext),
    .init          = init,
    .uninit        = uninit,
    FILTER_INPUTS(inputs),
    FILTER_OUTPUTS(outputs),
    FILTER_PIXFMTS_ARRAY(pixel_fmts),
    .process_command = process_command,
}

Definition at line 564 of file vf_atadenoise.c.