vf_hqdn3d.c File Reference

high quality 3d video denoiser, ported from MPlayer libmpcodecs/vf_hqdn3d.c. More...

#include <float.h>
#include "config.h"
#include "libavutil/attributes.h"
#include "libavutil/common.h"
#include "libavutil/mem.h"
#include "libavutil/pixdesc.h"
#include "libavutil/intreadwrite.h"
#include "libavutil/opt.h"
#include "avfilter.h"
#include "filters.h"
#include "video.h"
#include "vf_hqdn3d.h"

Go to the source code of this file.

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

Macros
#define	LUT_BITS   (depth==16 ? 8 : 4)
#define	LOAD(x)
#define	STORE(x, val)
#define	denoise(...)
#define	PARAM1_DEFAULT   4.0
#define	PARAM2_DEFAULT   3.0
#define	PARAM3_DEFAULT   6.0
#define	OFFSET(x)
#define	FLAGS   AV_OPT_FLAG_VIDEO_PARAM | AV_OPT_FLAG_FILTERING_PARAM | AV_OPT_FLAG_RUNTIME_PARAM

Functions
static av_always_inline uint32_t	lowpass (int prev, int cur, int16_t *coef, int depth)
static av_always_inline void	denoise_temporal (uint8_t *src, uint8_t *dst, uint16_t *frame_ant, int w, int h, int sstride, int dstride, int16_t *temporal, int depth)
static av_always_inline void	denoise_spatial (HQDN3DContext *s, uint8_t *src, uint8_t *dst, uint16_t *line_ant, uint16_t *frame_ant, int w, int h, int sstride, int dstride, int16_t *spatial, int16_t *temporal, int depth)
static av_always_inline int	denoise_depth (HQDN3DContext *s, uint8_t *src, uint8_t *dst, uint16_t *line_ant, uint16_t **frame_ant_ptr, int w, int h, int sstride, int dstride, int16_t *spatial, int16_t *temporal, int depth)
static void	precalc_coefs (double dist25, int depth, int16_t *ct)
static av_cold int	init (AVFilterContext *ctx)
static av_cold void	uninit (AVFilterContext *ctx)
static void	calc_coefs (AVFilterContext *ctx)
static int	config_input (AVFilterLink *inlink)
static int	do_denoise (AVFilterContext *ctx, void *data, int job_nr, int n_jobs)
static int	filter_frame (AVFilterLink *inlink, AVFrame *in)
static int	process_command (AVFilterContext *ctx, const char *cmd, const char *args, char *res, int res_len, int flags)
	AVFILTER_DEFINE_CLASS (hqdn3d)

Variables
static enum AVPixelFormat	pix_fmts []
static const AVOption	hqdn3d_options []
static const AVFilterPad	avfilter_vf_hqdn3d_inputs []
const FFFilter	ff_vf_hqdn3d

Detailed Description

high quality 3d video denoiser, ported from MPlayer libmpcodecs/vf_hqdn3d.c.

Definition in file vf_hqdn3d.c.

Macro Definition Documentation

LUT_BITS

#define LUT_BITS   (depth==16 ? 8 : 4)

Definition at line 44 of file vf_hqdn3d.c.

LOAD

#define LOAD

(

x

)

Value:

                 (((depth == 8 ? src[x] : AV_RN16A(src + (x) * 2)) << (16 - depth))\
                 + (((1 << (16 - depth)) - 1) >> 1))

Definition at line 45 of file vf_hqdn3d.c.

Referenced by denoise_depth(), denoise_spatial(), and denoise_temporal().

STORE

#define STORE	(		x,
			val )

Value:

                                   (depth == 8 ? dst[x] = (val) >> (16 - depth) : \
                                   AV_WN16A(dst + (x) * 2, (val) >> (16 - depth)))

Definition at line 47 of file vf_hqdn3d.c.

Referenced by denoise_spatial(), and denoise_temporal().

denoise

#define denoise

(

...

)

Value:

    do {                                                                      \
        int ret = AVERROR_BUG;                                                \
        switch (s->depth) {                                                   \
            case  8: ret = denoise_depth(__VA_ARGS__,  8); break;             \
            case  9: ret = denoise_depth(__VA_ARGS__,  9); break;             \
            case 10: ret = denoise_depth(__VA_ARGS__, 10); break;             \
            case 12: ret = denoise_depth(__VA_ARGS__, 12); break;             \
            case 14: ret = denoise_depth(__VA_ARGS__, 14); break;             \
            case 16: ret = denoise_depth(__VA_ARGS__, 16); break;             \
        }                                                                     \
        if (ret < 0)                                                          \
            return ret;                                                       \
    } while (0)

Definition at line 155 of file vf_hqdn3d.c.

Referenced by denoise_vaapi_build_filter_params(), do_denoise(), and filter_frame().

PARAM1_DEFAULT

#define PARAM1_DEFAULT   4.0

Definition at line 187 of file vf_hqdn3d.c.

Referenced by init().

PARAM2_DEFAULT

#define PARAM2_DEFAULT   3.0

Definition at line 188 of file vf_hqdn3d.c.

Referenced by init().

PARAM3_DEFAULT

#define PARAM3_DEFAULT   6.0

Definition at line 189 of file vf_hqdn3d.c.

Referenced by init().

OFFSET

#define OFFSET

(

x

)

Value:

offsetof(HQDN3DContext, x)

Definition at line 386 of file vf_hqdn3d.c.

FLAGS

#define FLAGS   AV_OPT_FLAG_VIDEO_PARAM | AV_OPT_FLAG_FILTERING_PARAM | AV_OPT_FLAG_RUNTIME_PARAM

Definition at line 387 of file vf_hqdn3d.c.

Function Documentation

lowpass()

static av_always_inline uint32_t lowpass ( int prev, int cur, int16_t * coef, int depth )

static

Definition at line 51 of file vf_hqdn3d.c.

denoise_temporal()

static av_always_inline void denoise_temporal ( uint8_t * src, uint8_t * dst, uint16_t * frame_ant, int w, int h, int sstride, int dstride, int16_t * temporal, int depth )

static

Definition at line 58 of file vf_hqdn3d.c.

Referenced by denoise_depth().

denoise_spatial()

static av_always_inline void denoise_spatial ( HQDN3DContext * s, uint8_t * src, uint8_t * dst, uint16_t * line_ant, uint16_t * frame_ant, int w, int h, int sstride, int dstride, int16_t * spatial, int16_t * temporal, int depth )

static

Definition at line 80 of file vf_hqdn3d.c.

Referenced by denoise_depth().

denoise_depth()

static av_always_inline int denoise_depth ( HQDN3DContext * s, uint8_t * src, uint8_t * dst, uint16_t * line_ant, uint16_t ** frame_ant_ptr, int w, int h, int sstride, int dstride, int16_t * spatial, int16_t * temporal, int depth )

static

Definition at line 124 of file vf_hqdn3d.c.

precalc_coefs()

static void precalc_coefs ( double dist25, int depth, int16_t * ct )

static

Definition at line 170 of file vf_hqdn3d.c.

Referenced by calc_coefs().

init()

static av_cold int init ( AVFilterContext * ctx )

static

Definition at line 191 of file vf_hqdn3d.c.

uninit()

static av_cold void uninit ( AVFilterContext * ctx )

static

Definition at line 211 of file vf_hqdn3d.c.

Referenced by config_input().

calc_coefs()

static void calc_coefs ( AVFilterContext * ctx )

static

Definition at line 241 of file vf_hqdn3d.c.

Referenced by config_input(), and process_command().

config_input()

static int config_input ( AVFilterLink * inlink )

static

Definition at line 249 of file vf_hqdn3d.c.

Referenced by filter_frame().

do_denoise()

static int do_denoise ( AVFilterContext * ctx, void * data, int job_nr, int n_jobs )

static

Definition at line 291 of file vf_hqdn3d.c.

Referenced by filter_frame().

filter_frame()

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

static

Definition at line 309 of file vf_hqdn3d.c.

process_command()

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

static

Definition at line 372 of file vf_hqdn3d.c.

AVFILTER_DEFINE_CLASS()

AVFILTER_DEFINE_CLASS

(

hqdn3d

)

Variable Documentation

pix_fmts

enum AVPixelFormat pix_fmts[]

static

Initial value:

= {
    AV_PIX_FMT_YUV420P, AV_PIX_FMT_YUV422P, AV_PIX_FMT_YUV444P,
    AV_PIX_FMT_YUV410P, AV_PIX_FMT_YUV411P, AV_PIX_FMT_YUV440P,
    AV_PIX_FMT_YUVJ420P, AV_PIX_FMT_YUVJ422P, AV_PIX_FMT_YUVJ444P, AV_PIX_FMT_YUVJ440P,
    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_NONE
}

Definition at line 227 of file vf_hqdn3d.c.

hqdn3d_options

const AVOption hqdn3d_options[]

static

Initial value:

= {
    { "luma_spatial",   "spatial luma strength",    OFFSET(strength[LUMA_SPATIAL]),   AV_OPT_TYPE_DOUBLE, { .dbl = 0.0 }, 0, DBL_MAX, FLAGS },
    { "chroma_spatial", "spatial chroma strength",  OFFSET(strength[CHROMA_SPATIAL]), AV_OPT_TYPE_DOUBLE, { .dbl = 0.0 }, 0, DBL_MAX, FLAGS },
    { "luma_tmp",       "temporal luma strength",   OFFSET(strength[LUMA_TMP]),       AV_OPT_TYPE_DOUBLE, { .dbl = 0.0 }, 0, DBL_MAX, FLAGS },
    { "chroma_tmp",     "temporal chroma strength", OFFSET(strength[CHROMA_TMP]),     AV_OPT_TYPE_DOUBLE, { .dbl = 0.0 }, 0, DBL_MAX, FLAGS },
    { NULL }
}

Definition at line 388 of file vf_hqdn3d.c.

avfilter_vf_hqdn3d_inputs

const AVFilterPad avfilter_vf_hqdn3d_inputs[]

static

Initial value:

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

Definition at line 398 of file vf_hqdn3d.c.

ff_vf_hqdn3d

const FFFilter ff_vf_hqdn3d

Initial value:

= {
    .p.name        = "hqdn3d",
    .p.description = NULL_IF_CONFIG_SMALL("Apply a High Quality 3D Denoiser."),
    .p.priv_class  = &hqdn3d_class,
    .p.flags       = AVFILTER_FLAG_SUPPORT_TIMELINE_INTERNAL | AVFILTER_FLAG_SLICE_THREADS,
    .priv_size     = sizeof(HQDN3DContext),
    .init          = init,
    .uninit        = uninit,
    FILTER_INPUTS(avfilter_vf_hqdn3d_inputs),
    FILTER_OUTPUTS(ff_video_default_filterpad),
    FILTER_PIXFMTS_ARRAY(pix_fmts),
    .process_command = process_command,
}

Definition at line 408 of file vf_hqdn3d.c.