vf_hqdn3d.c File Reference

#include <float.h>
#include "config.h"
#include "libavutil/attributes.h"
#include "libavutil/common.h"
#include "libavutil/pixdesc.h"
#include "libavutil/intreadwrite.h"
#include "libavutil/opt.h"
#include "avfilter.h"
#include "formats.h"
#include "internal.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)   offsetof(HQDN3DContext, 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 []
static const AVFilterPad	avfilter_vf_hqdn3d_outputs []
const AVFilter	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.

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.

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) {                                                        \
            av_frame_free(&out);                                              \
            if (!direct)                                                      \
                av_frame_free(&in);                                           \
            return ret;                                                       \
        }                                                                     \
    } while (0)

Definition at line 156 of file vf_hqdn3d.c.

PARAM1_DEFAULT

#define PARAM1_DEFAULT   4.0

Definition at line 192 of file vf_hqdn3d.c.

PARAM2_DEFAULT

#define PARAM2_DEFAULT   3.0

Definition at line 193 of file vf_hqdn3d.c.

PARAM3_DEFAULT

#define PARAM3_DEFAULT   6.0

Definition at line 194 of file vf_hqdn3d.c.

OFFSET

#define OFFSET

(

x

)

offsetof(HQDN3DContext, x)

Definition at line 364 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 365 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.

Referenced by denoise_spatial(), and denoise_temporal().

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 175 of file vf_hqdn3d.c.

Referenced by calc_coefs().

init()

static av_cold int init ( AVFilterContext *  ctx )

static

Definition at line 196 of file vf_hqdn3d.c.

uninit()

static av_cold void uninit ( AVFilterContext *  ctx )

static

Definition at line 216 of file vf_hqdn3d.c.

Referenced by config_input().

calc_coefs()

static void calc_coefs ( AVFilterContext *  ctx )

static

Definition at line 246 of file vf_hqdn3d.c.

Referenced by config_input(), and process_command().

config_input()

static int config_input ( AVFilterLink *  inlink )

static

Definition at line 254 of file vf_hqdn3d.c.

do_denoise()

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

static

Definition at line 293 of file vf_hqdn3d.c.

Referenced by filter_frame().

filter_frame()

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

static

Definition at line 312 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 350 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 232 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 366 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 376 of file vf_hqdn3d.c.

avfilter_vf_hqdn3d_outputs

const AVFilterPad avfilter_vf_hqdn3d_outputs[]

static

Initial value:

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

Definition at line 386 of file vf_hqdn3d.c.

ff_vf_hqdn3d

const AVFilter ff_vf_hqdn3d

Initial value:

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

Definition at line 393 of file vf_hqdn3d.c.