vf_siti.c File Reference

#include <math.h>
#include "libavutil/imgutils.h"
#include "libavutil/internal.h"
#include "libavutil/opt.h"
#include "avfilter.h"
#include "internal.h"
#include "video.h"

Go to the source code of this file.

Data Structures
struct	SiTiContext

Macros
#define	CONVOLVE(bps)
#define	CALCULATE(bps)
#define	OFFSET(x)   offsetof(SiTiContext, x)
#define	FLAGS   AV_OPT_FLAG_VIDEO_PARAM|AV_OPT_FLAG_FILTERING_PARAM

Functions
static av_cold int	init (AVFilterContext *ctx)
static av_cold void	uninit (AVFilterContext *ctx)
static int	config_input (AVFilterLink *inlink)
static int	is_full_range (AVFrame *frame)
static uint16_t	convert_full_range (int factor, uint16_t y)
static void	convolve_sobel (SiTiContext *s, const uint8_t *src, float *dst, int linesize)
static void	calculate_motion (SiTiContext *s, const uint8_t *curr, float *motion_matrix, int linesize)
static float	std_deviation (float *img_metrics, int width, int height)
static void	set_meta (AVDictionary **metadata, const char *key, float d)
static int	filter_frame (AVFilterLink *inlink, AVFrame *frame)
	AVFILTER_DEFINE_CLASS (siti)

Variables
static const int	X_FILTER [9]
static const int	Y_FILTER [9]
static enum AVPixelFormat	pix_fmts []
static const AVOption	siti_options []
static const AVFilterPad	avfilter_vf_siti_inputs []
const AVFilter	ff_vf_siti

Detailed Description

Calculate Spatial Info (SI) and Temporal Info (TI) scores

Definition in file vf_siti.c.

Macro Definition Documentation

CONVOLVE

#define CONVOLVE

(

bps

)

Value:

    {                                                               \
        uint##bps##_t *vsrc = (uint##bps##_t*)src;                  \
        for (int j = 1; j < s->height - 1; j++) {                   \
            for (int i = 1; i < s->width - 1; i++) {                \
                x_conv_sum = 0.0;                                   \
                y_conv_sum = 0.0;                                   \
                for (int k = 0; k < filter_size; k++) {             \
                    ki = k % filter_width - 1;                      \
                    kj = floor(k / filter_width) - 1;               \
                    index = (j + kj) * stride + (i + ki);           \
                    data = s->full_range ? vsrc[index] : convert_full_range(factor, vsrc[index]); \
                    x_conv_sum += data * X_FILTER[k];               \
                    y_conv_sum += data * Y_FILTER[k];               \
                }                                                   \
                gradient = sqrt(x_conv_sum * x_conv_sum + y_conv_sum * y_conv_sum); \
                dst[(j - 1) * (s->width - 2) + (i - 1)] = gradient; \
            }                                                       \
        }                                                           \
    }

CALCULATE

#define CALCULATE

(

bps

)

Value:

    {                                                                \
        uint##bps##_t *vsrc = (uint##bps##_t*)curr;                  \
        uint##bps##_t *vdst = (uint##bps##_t*)s->prev_frame;         \
        for (int j = 0; j < s->height; j++) {                        \
            for (int i = 0; i < s->width; i++) {                     \
                motion = 0;                                          \
                curr_index = j * stride + i;                         \
                prev_index = j * s->width + i;                       \
                curr_data = s->full_range ? vsrc[curr_index] : convert_full_range(factor, vsrc[curr_index]); \
                if (s->nb_frames > 1)                                \
                    motion = curr_data - vdst[prev_index];           \
                vdst[prev_index] = curr_data;                        \
                motion_matrix[j * s->width + i] = motion;            \
            }                                                        \
        }                                                            \
    }

OFFSET

#define OFFSET

(

x

)

offsetof(SiTiContext, x)

Definition at line 311 of file vf_siti.c.

FLAGS

#define FLAGS   AV_OPT_FLAG_VIDEO_PARAM|AV_OPT_FLAG_FILTERING_PARAM

Definition at line 312 of file vf_siti.c.

Function Documentation

init()

static av_cold int init ( AVFilterContext *  ctx )

static

Definition at line 74 of file vf_siti.c.

uninit()

static av_cold void uninit ( AVFilterContext *  ctx )

static

Definition at line 83 of file vf_siti.c.

config_input()

static int config_input ( AVFilterLink *  inlink )

static

Definition at line 103 of file vf_siti.c.

is_full_range()

static int is_full_range ( AVFrame *  frame )

static

Definition at line 144 of file vf_siti.c.

Referenced by filter_frame().

convert_full_range()

static uint16_t convert_full_range ( int  factor, uint16_t  y  )

static

Definition at line 153 of file vf_siti.c.

convolve_sobel()

static void convolve_sobel ( SiTiContext *  s, const uint8_t *  src, float *  dst, int  linesize  )

static

Definition at line 169 of file vf_siti.c.

Referenced by filter_frame().

calculate_motion()

static void calculate_motion ( SiTiContext *  s, const uint8_t *  curr, float *  motion_matrix, int  linesize  )

static

Definition at line 214 of file vf_siti.c.

Referenced by filter_frame().

std_deviation()

static float std_deviation ( float *  img_metrics, int  width, int  height  )

static

Definition at line 251 of file vf_siti.c.

Referenced by filter_frame().

set_meta()

static void set_meta ( AVDictionary **  metadata, const char *  key, float  d  )

static

Definition at line 273 of file vf_siti.c.

Referenced by filter_frame().

filter_frame()

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

static

Definition at line 280 of file vf_siti.c.

AVFILTER_DEFINE_CLASS()

AVFILTER_DEFINE_CLASS

(

siti

)

Variable Documentation

X_FILTER

const int X_FILTER[9]

static

Initial value:

= {
    1, 0, -1,
    2, 0, -2,
    1, 0, -1
}

Definition at line 37 of file vf_siti.c.

Y_FILTER

const int Y_FILTER[9]

static

Initial value:

= {
    1, 2, 1,
    0, 0, 0,
    -1, -2, -1
}

Definition at line 43 of file vf_siti.c.

pix_fmts

enum AVPixelFormat pix_fmts[]

static

Initial value:

= {
    AV_PIX_FMT_YUV420P, AV_PIX_FMT_YUV422P,
    AV_PIX_FMT_YUVJ420P, AV_PIX_FMT_YUVJ422P,
    AV_PIX_FMT_YUV420P10, AV_PIX_FMT_YUV422P10,
    AV_PIX_FMT_NONE
}

Definition at line 67 of file vf_siti.c.

siti_options

const AVOption siti_options[]

static

Initial value:

= {
    { "print_summary", "Print summary showing average values", OFFSET(print_summary), AV_OPT_TYPE_BOOL, { .i64=0 }, 0, 1, FLAGS },
    { NULL }
}

Definition at line 314 of file vf_siti.c.

avfilter_vf_siti_inputs

const AVFilterPad avfilter_vf_siti_inputs[]

static

Initial value:

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

Definition at line 321 of file vf_siti.c.

ff_vf_siti

const AVFilter ff_vf_siti

Initial value:

= {
    .name          = "siti",
    .description   = NULL_IF_CONFIG_SMALL("Calculate spatial information (SI) and temporal information (TI)."),
    .priv_size     = sizeof(SiTiContext),
    .priv_class    = &siti_class,
    .init          = init,
    .uninit        = uninit,
    .flags         = AVFILTER_FLAG_METADATA_ONLY,
    FILTER_PIXFMTS_ARRAY(pix_fmts),
    FILTER_INPUTS(avfilter_vf_siti_inputs),
    FILTER_OUTPUTS(ff_video_default_filterpad),
}

Definition at line 330 of file vf_siti.c.