vf_nlmeans.c File Reference

#include "libavutil/avassert.h"
#include "libavutil/mem.h"
#include "libavutil/opt.h"
#include "libavutil/pixdesc.h"
#include "avfilter.h"
#include "filters.h"
#include "vf_nlmeans.h"
#include "vf_nlmeans_init.h"
#include "video.h"

Go to the source code of this file.

Data Structures
struct	NLMeansContext
struct	thread_data

Macros
#define	OFFSET(x)
#define	FLAGS   AV_OPT_FLAG_FILTERING_PARAM|AV_OPT_FLAG_VIDEO_PARAM
#define	CHECK_ODD_FIELD(field, name)

Functions
	AVFILTER_DEFINE_CLASS (nlmeans)
static void	compute_unsafe_ssd_integral_image (uint32_t *dst, ptrdiff_t dst_linesize_32, int startx, int starty, const uint8_t *src, ptrdiff_t linesize, int offx, int offy, int r, int sw, int sh, int w, int h)
	Compute squared difference of an unsafe area (the zone nor s1 nor s2 could be readable).
static void	compute_ssd_integral_image (const NLMeansDSPContext *dsp, uint32_t *ii, ptrdiff_t ii_linesize_32, const uint8_t *src, ptrdiff_t linesize, int offx, int offy, int e, int w, int h)
static int	config_input (AVFilterLink *inlink)
static int	nlmeans_slice (AVFilterContext *ctx, void *arg, int jobnr, int nb_jobs)
static void	weight_averages (uint8_t *dst, ptrdiff_t dst_linesize, const uint8_t *src, ptrdiff_t src_linesize, float *total_weight, float *sum, ptrdiff_t linesize, int w, int h)
static int	nlmeans_plane (AVFilterContext *ctx, int w, int h, int p, int r, uint8_t *dst, ptrdiff_t dst_linesize, const uint8_t *src, ptrdiff_t src_linesize)
static int	filter_frame (AVFilterLink *inlink, AVFrame *in)
static av_cold int	init (AVFilterContext *ctx)
static av_cold void	uninit (AVFilterContext *ctx)

Variables
static const AVOption	nlmeans_options []
static enum AVPixelFormat	pix_fmts []
static const AVFilterPad	nlmeans_inputs []
const FFFilter	ff_vf_nlmeans

Macro Definition Documentation

OFFSET

#define OFFSET

(

x

)

Value:

offsetof(NLMeansContext, x)

Definition at line 64 of file vf_nlmeans.c.

FLAGS

#define FLAGS   AV_OPT_FLAG_FILTERING_PARAM|AV_OPT_FLAG_VIDEO_PARAM

Definition at line 65 of file vf_nlmeans.c.

CHECK_ODD_FIELD

#define CHECK_ODD_FIELD	(		field,
			name )

Value:

    do {                       \
    if (!(s->field & 1)) {                                      \
        s->field |= 1;                                          \
        av_log(ctx, AV_LOG_WARNING, name " size must be odd, "  \
               "setting it to %d\n", s->field);                 \
    }                                                           \
} while (0)

Definition at line 413 of file vf_nlmeans.c.

Referenced by init().

Function Documentation

AVFILTER_DEFINE_CLASS()

AVFILTER_DEFINE_CLASS

(

nlmeans

)

compute_unsafe_ssd_integral_image()

static void compute_unsafe_ssd_integral_image ( uint32_t * dst, ptrdiff_t dst_linesize_32, int startx, int starty, const uint8_t * src, ptrdiff_t linesize, int offx, int offy, int r, int sw, int sh, int w, int h )

inlinestatic

Compute squared difference of an unsafe area (the zone nor s1 nor s2 could be readable).

On the other hand, the line above dst and the column to its left are always readable.

There is little point in having this function SIMDified as it is likely too complex and only handle small portions of the image.

Parameters

dst	integral image
dst_linesize_32	integral image linesize (in 32-bit integers unit)
startx	integral starting x position
starty	integral starting y position
src	source plane buffer
linesize	source plane linesize
offx	source offsetting in x
offy	source offsetting in y @paran r absolute maximum source offsetting
sw	source width
sh	source height
w	width to compute
h	height to compute

Definition at line 112 of file vf_nlmeans.c.

Referenced by compute_ssd_integral_image(), and main().

compute_ssd_integral_image()

static void compute_ssd_integral_image ( const NLMeansDSPContext * dsp, uint32_t * ii, ptrdiff_t ii_linesize_32, const uint8_t * src, ptrdiff_t linesize, int offx, int offy, int e, int w, int h )

static

Definition at line 152 of file vf_nlmeans.c.

Referenced by main(), and nlmeans_plane().

config_input()

static int config_input ( AVFilterLink * inlink )

static

Definition at line 224 of file vf_nlmeans.c.

nlmeans_slice()

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

static

Definition at line 288 of file vf_nlmeans.c.

Referenced by nlmeans_plane().

weight_averages()

static void weight_averages ( uint8_t * dst, ptrdiff_t dst_linesize, const uint8_t * src, ptrdiff_t src_linesize, float * total_weight, float * sum, ptrdiff_t linesize, int w, int h )

static

Definition at line 324 of file vf_nlmeans.c.

Referenced by nlmeans_plane().

nlmeans_plane()

static int nlmeans_plane ( AVFilterContext * ctx, int w, int h, int p, int r, uint8_t * dst, ptrdiff_t dst_linesize, const uint8_t * src, ptrdiff_t src_linesize )

static

Definition at line 343 of file vf_nlmeans.c.

Referenced by filter_frame().

filter_frame()

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

static

Definition at line 386 of file vf_nlmeans.c.

init()

static av_cold int init ( AVFilterContext * ctx )

static

Definition at line 421 of file vf_nlmeans.c.

uninit()

static av_cold void uninit ( AVFilterContext * ctx )

static

Definition at line 457 of file vf_nlmeans.c.

Variable Documentation

nlmeans_options

const AVOption nlmeans_options[]

static

Initial value:

= {
    { "s",  "denoising strength", OFFSET(sigma), AV_OPT_TYPE_DOUBLE, { .dbl = 1.0 }, 1.0, 30.0, FLAGS },
    { "p",  "patch size",                   OFFSET(patch_size),    AV_OPT_TYPE_INT, { .i64 = 3*2+1 }, 0, 99, FLAGS },
    { "pc", "patch size for chroma planes", OFFSET(patch_size_uv), AV_OPT_TYPE_INT, { .i64 = 0 },     0, 99, FLAGS },
    { "r",  "research window",                   OFFSET(research_size),    AV_OPT_TYPE_INT, { .i64 = 7*2+1 }, 0, 99, FLAGS },
    { "rc", "research window for chroma planes", OFFSET(research_size_uv), AV_OPT_TYPE_INT, { .i64 = 0 },     0, 99, FLAGS },
    { NULL }
}

Definition at line 66 of file vf_nlmeans.c.

pix_fmts

enum AVPixelFormat pix_fmts[]

static

Initial value:

= {
    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_YUVJ444P, AV_PIX_FMT_YUVJ440P,
    AV_PIX_FMT_YUVJ422P, AV_PIX_FMT_YUVJ420P,
    AV_PIX_FMT_YUVJ411P,
    AV_PIX_FMT_GRAY8, AV_PIX_FMT_GBRP,
    AV_PIX_FMT_NONE
}

Definition at line 77 of file vf_nlmeans.c.

nlmeans_inputs

const AVFilterPad nlmeans_inputs[]

static

Initial value:

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

Definition at line 466 of file vf_nlmeans.c.

ff_vf_nlmeans

const FFFilter ff_vf_nlmeans

Initial value:

= {
    .p.name        = "nlmeans",
    .p.description = NULL_IF_CONFIG_SMALL("Non-local means denoiser."),
    .p.priv_class  = &nlmeans_class,
    .p.flags       = AVFILTER_FLAG_SUPPORT_TIMELINE_GENERIC | AVFILTER_FLAG_SLICE_THREADS,
    .priv_size     = sizeof(NLMeansContext),
    .init          = init,
    .uninit        = uninit,
    FILTER_INPUTS(nlmeans_inputs),
    FILTER_OUTPUTS(ff_video_default_filterpad),
    FILTER_PIXFMTS_ARRAY(pix_fmts),
}

Definition at line 475 of file vf_nlmeans.c.