vf_vif.c File Reference

Calculate VIF between two input videos. More...

#include <float.h>
#include "libavutil/internal.h"
#include "libavutil/mem.h"
#include "libavutil/opt.h"
#include "libavutil/pixdesc.h"
#include "avfilter.h"
#include "filters.h"
#include "framesync.h"

Go to the source code of this file.

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

Macros
#define	NUM_DATA_BUFS   13
#define	OFFSET(x)
#define	offset_fn(type, bits)
#define	PF(suf)

Functions
	FRAMESYNC_DEFINE_CLASS (vif, VIFContext, fs)
static void	vif_dec2 (const float *src, float *dst, int w, int h, int src_stride, int dst_stride)
static void	vif_statistic (const float *mu1_sq, const float *mu2_sq, const float *mu1_mu2, const float *xx_filt, const float *yy_filt, const float *xy_filt, float *num, float *den, int w, int h)
static void	vif_xx_yy_xy (const float *x, const float *y, float *xx, float *yy, float *xy, int w, int h)
static int	vif_filter1d (AVFilterContext *ctx, void *arg, int jobnr, int nb_jobs)
static int	compute_vif2 (AVFilterContext *ctx, const float *ref, const float *main, int w, int h, int ref_stride, int main_stride, float *score, float *const data_buf[NUM_DATA_BUFS], float **temp, int gnb_threads)
	offset_fn (uint8_t, 8)
static AVFrame *	do_vif (AVFilterContext *ctx, AVFrame *main, const AVFrame *ref)
static int	config_input_ref (AVFilterLink *inlink)
static int	process_frame (FFFrameSync *fs)
static int	config_output (AVFilterLink *outlink)
static int	activate (AVFilterContext *ctx)
static av_cold void	uninit (AVFilterContext *ctx)

Variables
static const AVOption	vif_options []
static const uint8_t	vif_filter1d_width1 [4] = { 17, 9, 5, 3 }
static const float	vif_filter1d_table [4][17]
static enum AVPixelFormat	pix_fmts []
static const AVFilterPad	vif_inputs []
static const AVFilterPad	vif_outputs []
const FFFilter	ff_vf_vif

Detailed Description

Calculate VIF between two input videos.

Definition in file vf_vif.c.

Macro Definition Documentation

NUM_DATA_BUFS

#define NUM_DATA_BUFS   13

Definition at line 38 of file vf_vif.c.

Referenced by compute_vif2(), config_input_ref(), and uninit().

OFFSET

#define OFFSET

(

x

)

Value:

offsetof(VIFContext, x)

Definition at line 58 of file vf_vif.c.

offset_fn

#define offset_fn	(		type,
			bits )

Value:

static void offset_##bits##bit(VIFContext *s,            \
                               const AVFrame *ref,       \
                               AVFrame *main, int stride)\
{                                                        \
    int w = s->width;                                    \
    int h = s->height;                                   \
                                                         \
    int ref_stride = ref->linesize[0];                   \
    int main_stride = main->linesize[0];                 \
                                                         \
    const type *ref_ptr = (const type *) ref->data[0];   \
    const type *main_ptr = (const type *) main->data[0]; \
                                            \
    const float factor = s->factor;         \
                                            \
    float *ref_ptr_data = s->ref_data;      \
    float *main_ptr_data = s->main_data;    \
                                            \
    for (int i = 0; i < h; i++) {           \
        for (int j = 0; j < w; j++) {       \
            ref_ptr_data[j] = ref_ptr[j] * factor - 128.f;   \
            main_ptr_data[j] = main_ptr[j] * factor - 128.f; \
        }                                   \
        ref_ptr += ref_stride / sizeof(type);   \
        ref_ptr_data += w;                      \
        main_ptr += main_stride / sizeof(type); \
        main_ptr_data += w;                     \
    } \
}

Definition at line 396 of file vf_vif.c.

Referenced by offset_fn().

PF

#define PF

(

suf

)

Value:

AV_PIX_FMT_YUV420##suf,  AV_PIX_FMT_YUV422##suf,  AV_PIX_FMT_YUV444##suf

Function Documentation

FRAMESYNC_DEFINE_CLASS()

FRAMESYNC_DEFINE_CLASS	(	vif	,
		VIFContext	,
		fs	)

vif_dec2()

static void vif_dec2 ( const float * src, float * dst, int w, int h, int src_stride, int dst_stride )

static

Definition at line 99 of file vf_vif.c.

Referenced by compute_vif2().

vif_statistic()

static void vif_statistic ( const float * mu1_sq, const float * mu2_sq, const float * mu1_mu2, const float * xx_filt, const float * yy_filt, const float * xy_filt, float * num, float * den, int w, int h )

static

Definition at line 110 of file vf_vif.c.

Referenced by compute_vif2().

vif_xx_yy_xy()

static void vif_xx_yy_xy ( const float * x, const float * y, float * xx, float * yy, float * xy, int w, int h )

static

Definition at line 183 of file vf_vif.c.

Referenced by compute_vif2().

vif_filter1d()

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

static

Vertical pass.

Horizontal pass.

Definition at line 207 of file vf_vif.c.

Referenced by compute_vif2().

compute_vif2()

static int compute_vif2 ( AVFilterContext * ctx, const float * ref, const float * main, int w, int h, int ref_stride, int main_stride, float * score, float *const data_buf[NUM_DATA_BUFS], float ** temp, int gnb_threads )

static

Definition at line 285 of file vf_vif.c.

Referenced by do_vif().

offset_fn()

offset_fn	(	uint8_t	,
		8	)

Definition at line 427 of file vf_vif.c.

do_vif()

static AVFrame * do_vif ( AVFilterContext * ctx, AVFrame * main, const AVFrame * ref )

static

Definition at line 437 of file vf_vif.c.

Referenced by process_frame().

config_input_ref()

static int config_input_ref ( AVFilterLink * inlink )

static

Definition at line 482 of file vf_vif.c.

process_frame()

static int process_frame ( FFFrameSync * fs )

static

Definition at line 525 of file vf_vif.c.

config_output()

static int config_output ( AVFilterLink * outlink )

static

Definition at line 549 of file vf_vif.c.

activate()

static int activate ( AVFilterContext * ctx )

static

Definition at line 582 of file vf_vif.c.

uninit()

static av_cold void uninit ( AVFilterContext * ctx )

static

Definition at line 588 of file vf_vif.c.

Variable Documentation

vif_options

const AVOption vif_options[]

static

Initial value:

= {
    { NULL }
}

Definition at line 60 of file vf_vif.c.

vif_filter1d_width1

const uint8_t vif_filter1d_width1[4] = { 17, 9, 5, 3 }

static

Definition at line 66 of file vf_vif.c.

Referenced by compute_vif2().

vif_filter1d_table

const float vif_filter1d_table[4][17]

static

Initial value:

=
{
    {
        0.00745626912, 0.0142655009, 0.0250313189, 0.0402820669, 0.0594526194,
        0.0804751068, 0.0999041125, 0.113746084, 0.118773937, 0.113746084,
        0.0999041125, 0.0804751068, 0.0594526194, 0.0402820669, 0.0250313189,
        0.0142655009, 0.00745626912
    },
    {
        0.0189780835, 0.0558981746, 0.120920904, 0.192116052, 0.224173605,
        0.192116052, 0.120920904, 0.0558981746, 0.0189780835
    },
    {
        0.054488685, 0.244201347, 0.402619958, 0.244201347, 0.054488685
    },
    {
        0.166378498, 0.667243004, 0.166378498
    }
}

Definition at line 68 of file vf_vif.c.

Referenced by compute_vif2().

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_YUV420P, AV_PIX_FMT_YUV422P, AV_PIX_FMT_YUV444P,
    AV_PIX_FMT_YUV440P, AV_PIX_FMT_YUV411P, AV_PIX_FMT_YUV410P,
    AV_PIX_FMT_YUVJ411P, AV_PIX_FMT_YUVJ420P, AV_PIX_FMT_YUVJ422P,
    AV_PIX_FMT_YUVJ440P, AV_PIX_FMT_YUVJ444P,
#define PF(suf)                                                                         
    PF(P9), PF(P10), PF(P12), PF(P14), PF(P16),
    AV_PIX_FMT_NONE
}

Definition at line 470 of file vf_vif.c.

vif_inputs

const AVFilterPad vif_inputs[]

static

Initial value:

= {
    {
        .name         = "main",
        .type         = AVMEDIA_TYPE_VIDEO,
    },{
        .name         = "reference",
        .type         = AVMEDIA_TYPE_VIDEO,
        .config_props = config_input_ref,
    },
}

Definition at line 612 of file vf_vif.c.

vif_outputs

const AVFilterPad vif_outputs[]

static

Initial value:

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

Definition at line 623 of file vf_vif.c.

ff_vf_vif

const FFFilter ff_vf_vif

Initial value:

= {
    .p.name        = "vif",
    .p.description = NULL_IF_CONFIG_SMALL("Calculate the VIF between two video streams."),
    .p.priv_class  = &vif_class,
    .p.flags       = AVFILTER_FLAG_SUPPORT_TIMELINE_INTERNAL |
                     AVFILTER_FLAG_SLICE_THREADS             |
                     AVFILTER_FLAG_METADATA_ONLY,
    .preinit       = vif_framesync_preinit,
    .uninit        = uninit,
    .priv_size     = sizeof(VIFContext),
    .activate      = activate,
    FILTER_INPUTS(vif_inputs),
    FILTER_OUTPUTS(vif_outputs),
    FILTER_PIXFMTS_ARRAY(pix_fmts),
}

Definition at line 631 of file vf_vif.c.