vf_thumbnail_cuda.c File Reference

#include "libavutil/hwcontext.h"
#include "libavutil/hwcontext_cuda_internal.h"
#include "libavutil/cuda_check.h"
#include "libavutil/mem.h"
#include "libavutil/opt.h"
#include "libavutil/pixdesc.h"
#include "avfilter.h"
#include "filters.h"
#include "cuda/load_helper.h"

Go to the source code of this file.

Data Structures
struct	thumb_frame
struct	ThumbnailCudaContext

Macros
#define	CHECK_CU(x)
#define	HIST_SIZE   (3*256)
#define	DIV_UP(a, b)
#define	BLOCKX   32
#define	BLOCKY   16
#define	OFFSET(x)
#define	FLAGS   AV_OPT_FLAG_VIDEO_PARAM|AV_OPT_FLAG_FILTERING_PARAM

Functions
	AVFILTER_DEFINE_CLASS (thumbnail_cuda)
static av_cold int	init (AVFilterContext *ctx)
static double	frame_sum_square_err (const int *hist, const double *median)
	Compute Sum-square deviation to estimate "closeness".
static AVFrame *	get_best_frame (AVFilterContext *ctx)
static int	thumbnail_kernel (AVFilterContext *ctx, CUfunction func, int channels, int *histogram, uint8_t *src_dptr, int src_width, int src_height, int src_pitch, int pixel_size)
static int	thumbnail (AVFilterContext *ctx, int *histogram, AVFrame *in)
static int	filter_frame (AVFilterLink *inlink, AVFrame *frame)
static av_cold void	uninit (AVFilterContext *ctx)
static int	request_frame (AVFilterLink *link)
static int	format_is_supported (enum AVPixelFormat fmt)
static int	config_props (AVFilterLink *inlink)

Variables
static enum AVPixelFormat	supported_formats []
static const AVOption	thumbnail_cuda_options []
static const AVFilterPad	thumbnail_cuda_inputs []
static const AVFilterPad	thumbnail_cuda_outputs []
const FFFilter	ff_vf_thumbnail_cuda

Macro Definition Documentation

CHECK_CU

#define CHECK_CU

(

x

)

Value:

FF_CUDA_CHECK_DL(ctx, s->hwctx->internal->cuda_dl, x)

Definition at line 35 of file vf_thumbnail_cuda.c.

Referenced by config_props(), filter_frame(), thumbnail_kernel(), and uninit().

HIST_SIZE

#define HIST_SIZE   (3*256)

Definition at line 37 of file vf_thumbnail_cuda.c.

DIV_UP

#define DIV_UP	(		a,
			b )

Value:

( ((a) + (b) - 1) / (b) )

Definition at line 38 of file vf_thumbnail_cuda.c.

Referenced by thumbnail_kernel().

BLOCKX

#define BLOCKX   32

Definition at line 39 of file vf_thumbnail_cuda.c.

BLOCKY

#define BLOCKY   16

Definition at line 40 of file vf_thumbnail_cuda.c.

OFFSET

#define OFFSET

(

x

)

Value:

offsetof(ThumbnailCudaContext, x)

Definition at line 81 of file vf_thumbnail_cuda.c.

FLAGS

#define FLAGS   AV_OPT_FLAG_VIDEO_PARAM|AV_OPT_FLAG_FILTERING_PARAM

Definition at line 82 of file vf_thumbnail_cuda.c.

Function Documentation

AVFILTER_DEFINE_CLASS()

AVFILTER_DEFINE_CLASS

(

thumbnail_cuda

)

init()

static av_cold int init ( AVFilterContext * ctx )

static

Definition at line 91 of file vf_thumbnail_cuda.c.

frame_sum_square_err()

static double frame_sum_square_err ( const int * hist, const double * median )

static

Compute Sum-square deviation to estimate "closeness".

Parameters

hist	color distribution histogram
median	average color distribution histogram

Returns

sum of squared errors

Definition at line 111 of file vf_thumbnail_cuda.c.

Referenced by get_best_frame().

get_best_frame()

static AVFrame * get_best_frame ( AVFilterContext * ctx )

static

Definition at line 123 of file vf_thumbnail_cuda.c.

Referenced by filter_frame(), and request_frame().

thumbnail_kernel()

static int thumbnail_kernel ( AVFilterContext * ctx, CUfunction func, int channels, int * histogram, uint8_t * src_dptr, int src_width, int src_height, int src_pitch, int pixel_size )

static

Definition at line 163 of file vf_thumbnail_cuda.c.

Referenced by thumbnail().

thumbnail()

static int thumbnail ( AVFilterContext * ctx, int * histogram, AVFrame * in )

static

Definition at line 203 of file vf_thumbnail_cuda.c.

Referenced by AVFILTER_DEFINE_CLASS(), and filter_frame().

filter_frame()

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

static

Definition at line 256 of file vf_thumbnail_cuda.c.

uninit()

static av_cold void uninit ( AVFilterContext * ctx )

static

Definition at line 313 of file vf_thumbnail_cuda.c.

request_frame()

static int request_frame ( AVFilterLink * link )

static

Definition at line 338 of file vf_thumbnail_cuda.c.

format_is_supported()

static int format_is_supported ( enum AVPixelFormat fmt )

static

Definition at line 355 of file vf_thumbnail_cuda.c.

Referenced by config_props().

config_props()

static int config_props ( AVFilterLink * inlink )

static

Definition at line 365 of file vf_thumbnail_cuda.c.

Variable Documentation

supported_formats

enum AVPixelFormat supported_formats[]

static

Initial value:

= {
    AV_PIX_FMT_NV12,
    AV_PIX_FMT_YUV420P,
    AV_PIX_FMT_YUV444P,
    AV_PIX_FMT_P010,
    AV_PIX_FMT_P012,
    AV_PIX_FMT_P016,
    AV_PIX_FMT_YUV444P10MSB,
    AV_PIX_FMT_YUV444P12MSB,
    AV_PIX_FMT_YUV444P16,
}

Definition at line 42 of file vf_thumbnail_cuda.c.

thumbnail_cuda_options

const AVOption thumbnail_cuda_options[]

static

Initial value:

= {
    { "n", "set the frames batch size", OFFSET(n_frames), AV_OPT_TYPE_INT, {.i64=100}, 2, INT_MAX, FLAGS },
    { NULL }
}

Definition at line 84 of file vf_thumbnail_cuda.c.

thumbnail_cuda_inputs

const AVFilterPad thumbnail_cuda_inputs[]

static

Initial value:

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

Definition at line 429 of file vf_thumbnail_cuda.c.

thumbnail_cuda_outputs

const AVFilterPad thumbnail_cuda_outputs[]

static

Initial value:

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

Definition at line 438 of file vf_thumbnail_cuda.c.

ff_vf_thumbnail_cuda

const FFFilter ff_vf_thumbnail_cuda

Initial value:

= {
    .p.name        = "thumbnail_cuda",
    .p.description = NULL_IF_CONFIG_SMALL("Select the most representative frame in a given sequence of consecutive frames using CUDA."),
    .p.priv_class  = &thumbnail_cuda_class,
    .priv_size     = sizeof(ThumbnailCudaContext),
    .init          = init,
    .uninit        = uninit,
    FILTER_INPUTS(thumbnail_cuda_inputs),
    FILTER_OUTPUTS(thumbnail_cuda_outputs),
    FILTER_SINGLE_PIXFMT(AV_PIX_FMT_CUDA),
    .flags_internal = FF_FILTER_FLAG_HWFRAME_AWARE,
}

Definition at line 446 of file vf_thumbnail_cuda.c.