vf_mestimate.c File Reference

#include "motion_estimation.h"
#include "libavcodec/mathops.h"
#include "libavutil/common.h"
#include "libavutil/mem.h"
#include "libavutil/opt.h"
#include "libavutil/motion_vector.h"
#include "avfilter.h"
#include "filters.h"
#include "video.h"

Go to the source code of this file.

Data Structures
struct	MEContext
	Copyright (c) 2016 Davinder Singh (DSM_) <ds.mudhar<@gmail.com> More...

Macros
#define	OFFSET(x)   offsetof(MEContext, x)
#define	FLAGS   AV_OPT_FLAG_VIDEO_PARAM|AV_OPT_FLAG_FILTERING_PARAM
#define	CONST(name, help, val, u)   { name, help, 0, AV_OPT_TYPE_CONST, {.i64=val}, 0, 0, FLAGS, .unit = u }
#define	SEARCH_MV(method)
#define	ADD_PRED(preds, px, py)

Functions
	AVFILTER_DEFINE_CLASS (mestimate)
static int	config_input (AVFilterLink *inlink)
static void	add_mv_data (AVMotionVector *mv, int mb_size, int x, int y, int x_mv, int y_mv, int dir)
static int	filter_frame (AVFilterLink *inlink, AVFrame *frame)
static av_cold void	uninit (AVFilterContext *ctx)

Variables
static const AVOption	mestimate_options []
static enum AVPixelFormat	pix_fmts []
static const AVFilterPad	mestimate_inputs []
const AVFilter	ff_vf_mestimate

Macro Definition Documentation

OFFSET

#define OFFSET

(

x

)

offsetof(MEContext, x)

Definition at line 46 of file vf_mestimate.c.

FLAGS

#define FLAGS   AV_OPT_FLAG_VIDEO_PARAM|AV_OPT_FLAG_FILTERING_PARAM

Definition at line 47 of file vf_mestimate.c.

CONST

#define CONST	(		name,
			help,
			val,
			u
	)		{ name, help, 0, AV_OPT_TYPE_CONST, {.i64=val}, 0, 0, FLAGS, .unit = u }

Definition at line 48 of file vf_mestimate.c.

SEARCH_MV

#define SEARCH_MV

(

method

)

Value:

    do {\
        for (mb_y = 0; mb_y < s->b_height; mb_y++)\
            for (mb_x = 0; mb_x < s->b_width; mb_x++) {\
                const int x_mb = mb_x << s->log2_mb_size;\
                const int y_mb = mb_y << s->log2_mb_size;\
                int mv[2] = {x_mb, y_mb};\
                ff_me_search_##method(me_ctx, x_mb, y_mb, mv);\
                add_mv_data(((AVMotionVector *) sd->data) + mv_count++, me_ctx->mb_size, x_mb, y_mb, mv[0], mv[1], dir);\
            }\
    } while (0)

Definition at line 119 of file vf_mestimate.c.

ADD_PRED

#define ADD_PRED	(		preds,
			px,
			py
	)

Value:

    do {\
        preds.mvs[preds.nb][0] = px;\
        preds.mvs[preds.nb][1] = py;\
        preds.nb++;\
    } while(0)

Definition at line 131 of file vf_mestimate.c.

Function Documentation

AVFILTER_DEFINE_CLASS()

AVFILTER_DEFINE_CLASS

(

mestimate

)

config_input()

static int config_input ( AVFilterLink *  inlink )

static

Definition at line 80 of file vf_mestimate.c.

add_mv_data()

static void add_mv_data ( AVMotionVector *  mv, int  mb_size, int  x, int  y, int  x_mv, int  y_mv, int  dir  )

static

Definition at line 106 of file vf_mestimate.c.

Referenced by filter_frame().

filter_frame()

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

static

Definition at line 138 of file vf_mestimate.c.

uninit()

static av_cold void uninit ( AVFilterContext *  ctx )

static

Definition at line 329 of file vf_mestimate.c.

Variable Documentation

mestimate_options

const AVOption mestimate_options[]

static

Initial value:

= {
    { "method", "motion estimation method", OFFSET(method), AV_OPT_TYPE_INT, {.i64 = AV_ME_METHOD_ESA}, AV_ME_METHOD_ESA, AV_ME_METHOD_UMH, FLAGS, .unit = "method" },
        CONST("esa",   "exhaustive search",                  AV_ME_METHOD_ESA,      "method"),
        CONST("tss",   "three step search",                  AV_ME_METHOD_TSS,      "method"),
        CONST("tdls",  "two dimensional logarithmic search", AV_ME_METHOD_TDLS,     "method"),
        CONST("ntss",  "new three step search",              AV_ME_METHOD_NTSS,     "method"),
        CONST("fss",   "four step search",                   AV_ME_METHOD_FSS,      "method"),
        CONST("ds",    "diamond search",                     AV_ME_METHOD_DS,       "method"),
        CONST("hexbs", "hexagon-based search",               AV_ME_METHOD_HEXBS,    "method"),
        CONST("epzs",  "enhanced predictive zonal search",   AV_ME_METHOD_EPZS,     "method"),
        CONST("umh",   "uneven multi-hexagon search",        AV_ME_METHOD_UMH,      "method"),
    { "mb_size", "macroblock size", OFFSET(mb_size), AV_OPT_TYPE_INT, {.i64 = 16}, 8, INT_MAX, FLAGS },
    { "search_param", "search parameter", OFFSET(search_param), AV_OPT_TYPE_INT, {.i64 = 7}, 4, INT_MAX, FLAGS },
    { NULL }
}

Definition at line 50 of file vf_mestimate.c.

pix_fmts

enum AVPixelFormat pix_fmts[]

static

Initial value:

= {
    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_YUVJ444P, AV_PIX_FMT_YUVJ440P,
    AV_PIX_FMT_YUVJ422P, AV_PIX_FMT_YUVJ420P,
    AV_PIX_FMT_YUVJ411P,
    AV_PIX_FMT_YUVA420P, AV_PIX_FMT_YUVA422P, AV_PIX_FMT_YUVA444P,
    AV_PIX_FMT_GRAY8,
    AV_PIX_FMT_NONE
}

Definition at line 68 of file vf_mestimate.c.

mestimate_inputs

const AVFilterPad mestimate_inputs[]

static

Initial value:

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

Definition at line 342 of file vf_mestimate.c.

ff_vf_mestimate

const AVFilter ff_vf_mestimate

Initial value:

= {
    .name          = "mestimate",
    .description   = NULL_IF_CONFIG_SMALL("Generate motion vectors."),
    .priv_size     = sizeof(MEContext),
    .priv_class    = &mestimate_class,
    .uninit        = uninit,
    .flags         = AVFILTER_FLAG_METADATA_ONLY,
    FILTER_INPUTS(mestimate_inputs),
    FILTER_OUTPUTS(ff_video_default_filterpad),
    FILTER_PIXFMTS_ARRAY(pix_fmts),
}

Definition at line 351 of file vf_mestimate.c.