vf_fftdnoiz.c File Reference

#include <float.h>
#include "libavutil/common.h"
#include "libavutil/imgutils.h"
#include "libavutil/opt.h"
#include "libavutil/pixdesc.h"
#include "libavutil/tx.h"
#include "internal.h"
#include "window_func.h"

Go to the source code of this file.

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

Macros
#define	MAX_BLOCK   256
#define	MAX_THREADS   32
#define	OFFSET(x)   offsetof(FFTdnoizContext, x)
#define	FLAGS   AV_OPT_FLAG_FILTERING_PARAM|AV_OPT_FLAG_VIDEO_PARAM
#define	TFLAGS   AV_OPT_FLAG_FILTERING_PARAM|AV_OPT_FLAG_VIDEO_PARAM|AV_OPT_FLAG_RUNTIME_PARAM

Enumerations
enum	BufferTypes { CURRENT, PREV, NEXT, BSIZE }

Functions
	AVFILTER_DEFINE_CLASS (fftdnoiz)
static void	import_row8 (AVComplexFloat *dst, uint8_t *src, int rw, float scale, float *win, int off)
static void	export_row8 (AVComplexFloat *src, uint8_t *dst, int rw, int depth, float *win)
static void	import_row16 (AVComplexFloat *dst, uint8_t *srcp, int rw, float scale, float *win, int off)
static void	export_row16 (AVComplexFloat *src, uint8_t *dstp, int rw, int depth, float *win)
static int	config_input (AVFilterLink *inlink)
static void	import_block (FFTdnoizContext *s, uint8_t *srcp, int src_linesize, float *buffer, int buffer_linesize, int plane, int jobnr, int y, int x)
static void	export_block (FFTdnoizContext *s, uint8_t *dstp, int dst_linesize, float *buffer, int buffer_linesize, int plane, int jobnr, int y, int x)
static void	filter_block3d2 (FFTdnoizContext *s, int plane, float *pbuffer, float *nbuffer, int jobnr)
static void	filter_block3d1 (FFTdnoizContext *s, int plane, float *pbuffer, int jobnr)
static void	filter_block2d (FFTdnoizContext *s, int plane, int jobnr)
static int	denoise (AVFilterContext *ctx, void *arg, int jobnr, int nb_jobs)
static int	filter_frame (AVFilterLink *inlink, AVFrame *in)
static int	request_frame (AVFilterLink *outlink)
static av_cold void	uninit (AVFilterContext *ctx)

Variables
static const AVOption	fftdnoiz_options []
static enum AVPixelFormat	pix_fmts []
static const AVFilterPad	fftdnoiz_inputs []
static const AVFilterPad	fftdnoiz_outputs []
const AVFilter	ff_vf_fftdnoiz

Macro Definition Documentation

MAX_BLOCK

#define MAX_BLOCK   256

Definition at line 29 of file vf_fftdnoiz.c.

MAX_THREADS

#define MAX_THREADS   32

Definition at line 30 of file vf_fftdnoiz.c.

OFFSET

#define OFFSET

(

x

)

offsetof(FFTdnoizContext, x)

Definition at line 84 of file vf_fftdnoiz.c.

FLAGS

#define FLAGS   AV_OPT_FLAG_FILTERING_PARAM|AV_OPT_FLAG_VIDEO_PARAM

Definition at line 85 of file vf_fftdnoiz.c.

TFLAGS

#define TFLAGS   AV_OPT_FLAG_FILTERING_PARAM|AV_OPT_FLAG_VIDEO_PARAM|AV_OPT_FLAG_RUNTIME_PARAM

Definition at line 86 of file vf_fftdnoiz.c.

Enumeration Type Documentation

BufferTypes

enum BufferTypes

Enumerator
CURRENT
PREV
NEXT
BSIZE

Definition at line 32 of file vf_fftdnoiz.c.

Function Documentation

AVFILTER_DEFINE_CLASS()

AVFILTER_DEFINE_CLASS

(

fftdnoiz

)

import_row8()

static void import_row8 ( AVComplexFloat *  dst, uint8_t *  src, int  rw, float  scale, float *  win, int  off  )

static

Definition at line 145 of file vf_fftdnoiz.c.

Referenced by config_input().

export_row8()

static void export_row8 ( AVComplexFloat *  src, uint8_t *  dst, int  rw, int  depth, float *  win  )

static

Definition at line 155 of file vf_fftdnoiz.c.

Referenced by config_input().

import_row16()

static void import_row16 ( AVComplexFloat *  dst, uint8_t *  srcp, int  rw, float  scale, float *  win, int  off  )

static

Definition at line 161 of file vf_fftdnoiz.c.

Referenced by config_input().

export_row16()

static void export_row16 ( AVComplexFloat *  src, uint8_t *  dstp, int  rw, int  depth, float *  win  )

static

Definition at line 173 of file vf_fftdnoiz.c.

Referenced by config_input().

config_input()

static int config_input ( AVFilterLink *  inlink )

static

Definition at line 181 of file vf_fftdnoiz.c.

import_block()

static void import_block ( FFTdnoizContext *  s, uint8_t *  srcp, int  src_linesize, float *  buffer, int  buffer_linesize, int  plane, int  jobnr, int  y, int  x  )

static

Definition at line 269 of file vf_fftdnoiz.c.

Referenced by denoise().

export_block()

static void export_block ( FFTdnoizContext *  s, uint8_t *  dstp, int  dst_linesize, float *  buffer, int  buffer_linesize, int  plane, int  jobnr, int  y, int  x  )

static

Definition at line 332 of file vf_fftdnoiz.c.

Referenced by denoise().

filter_block3d2()

static void filter_block3d2 ( FFTdnoizContext *  s, int  plane, float *  pbuffer, float *  nbuffer, int  jobnr  )

static

Definition at line 379 of file vf_fftdnoiz.c.

Referenced by denoise().

filter_block3d1()

static void filter_block3d1 ( FFTdnoizContext *  s, int  plane, float *  pbuffer, int  jobnr  )

static

Definition at line 441 of file vf_fftdnoiz.c.

Referenced by denoise().

filter_block2d()

static void filter_block2d ( FFTdnoizContext *  s, int  plane, int  jobnr  )

static

Definition at line 498 of file vf_fftdnoiz.c.

Referenced by denoise().

denoise()

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

static

Definition at line 534 of file vf_fftdnoiz.c.

Referenced by filter_frame().

filter_frame()

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

static

Definition at line 588 of file vf_fftdnoiz.c.

Referenced by request_frame().

request_frame()

static int request_frame ( AVFilterLink *  outlink )

static

Definition at line 665 of file vf_fftdnoiz.c.

uninit()

static av_cold void uninit ( AVFilterContext *  ctx )

static

Definition at line 694 of file vf_fftdnoiz.c.

Variable Documentation

fftdnoiz_options

const AVOption fftdnoiz_options[]

static

Initial value:

= {
    { "sigma",   "set denoise strength",
        OFFSET(sigma),      AV_OPT_TYPE_FLOAT, {.dbl=1},        0, 100, .flags = TFLAGS },
    { "amount",  "set amount of denoising",
        OFFSET(amount),     AV_OPT_TYPE_FLOAT, {.dbl=1},     0.01,   1, .flags = TFLAGS },
    { "block",   "set block size",
        OFFSET(block_size), AV_OPT_TYPE_INT,   {.i64=32}, 8, MAX_BLOCK, .flags = FLAGS },
    { "overlap", "set block overlap",
        OFFSET(overlap),    AV_OPT_TYPE_FLOAT, {.dbl=0.5},    0.2, 0.8, .flags = FLAGS },
    { "method",  "set method of denoising",
        OFFSET(method),     AV_OPT_TYPE_INT,   {.i64=0},        0,   1, .flags = TFLAGS, "method" },
    { "wiener", "wiener method",
        0,                  AV_OPT_TYPE_CONST, {.i64=0},        0,   0, .flags = TFLAGS, "method" },
    { "hard",   "hard thresholding",
        0,                  AV_OPT_TYPE_CONST, {.i64=1},        0,   0, .flags = TFLAGS, "method" },
    { "prev",    "set number of previous frames for temporal denoising",
        OFFSET(nb_prev),    AV_OPT_TYPE_INT,   {.i64=0},        0,   1, .flags = FLAGS },
    { "next",    "set number of next frames for temporal denoising",
        OFFSET(nb_next),    AV_OPT_TYPE_INT,   {.i64=0},        0,   1, .flags = FLAGS },
    { "planes",  "set planes to filter",
        OFFSET(planesf),    AV_OPT_TYPE_INT,   {.i64=7},        0,  15, .flags = TFLAGS },
    WIN_FUNC_OPTION("window", OFFSET(window), FLAGS, WFUNC_HANNING),
    { NULL }
}

Definition at line 87 of file vf_fftdnoiz.c.

pix_fmts

enum AVPixelFormat pix_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 114 of file vf_fftdnoiz.c.

fftdnoiz_inputs

const AVFilterPad fftdnoiz_inputs[]

static

Initial value:

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

Definition at line 725 of file vf_fftdnoiz.c.

fftdnoiz_outputs

const AVFilterPad fftdnoiz_outputs[]

static

Initial value:

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

Definition at line 734 of file vf_fftdnoiz.c.

ff_vf_fftdnoiz

const AVFilter ff_vf_fftdnoiz

Initial value:

= {
    .name          = "fftdnoiz",
    .description   = NULL_IF_CONFIG_SMALL("Denoise frames using 3D FFT."),
    .priv_size     = sizeof(FFTdnoizContext),
    .uninit        = uninit,
    FILTER_INPUTS(fftdnoiz_inputs),
    FILTER_OUTPUTS(fftdnoiz_outputs),
    FILTER_PIXFMTS_ARRAY(pix_fmts),
    .priv_class    = &fftdnoiz_class,
    .flags         = AVFILTER_FLAG_SUPPORT_TIMELINE_INTERNAL |
                     AVFILTER_FLAG_SLICE_THREADS,
    .process_command = ff_filter_process_command,
}

Definition at line 742 of file vf_fftdnoiz.c.