vf_thumbnail_cuda.c File Reference

#include <cuda.h>
#include "libavutil/hwcontext.h"
#include "libavutil/hwcontext_cuda.h"
#include "libavutil/opt.h"
#include "libavutil/pixdesc.h"
#include "avfilter.h"
#include "internal.h"

Go to the source code of this file.

Data Structures
struct	thumb_frame
struct	ThumbnailCudaContext

Macros
#define	HIST_SIZE   (3*256)
#define	DIV_UP(a, b)   ( ((a) + (b) - 1) / (b) )
#define	BLOCKX   32
#define	BLOCKY   16
#define	OFFSET(x)   offsetof(ThumbnailCudaContext, 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". More...
static AVFrame *	get_best_frame (AVFilterContext *ctx)
static int	thumbnail_kernel (ThumbnailCudaContext *s, CUfunction func, CUtexref tex, 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)
static int	query_formats (AVFilterContext *ctx)

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

Macro Definition Documentation

#define HIST_SIZE   (3*256)

Definition at line 33 of file vf_thumbnail_cuda.c.

Referenced by config_props(), frame_sum_square_err(), and get_best_frame().

#define DIV_UP	(		a,
			b
	)		( ((a) + (b) - 1) / (b) )

Definition at line 34 of file vf_thumbnail_cuda.c.

Referenced by thumbnail_kernel().

#define BLOCKX   32

Definition at line 35 of file vf_thumbnail_cuda.c.

Referenced by thumbnail_kernel().

#define BLOCKY   16

Definition at line 36 of file vf_thumbnail_cuda.c.

Referenced by thumbnail_kernel().

#define OFFSET

(

x

)

offsetof(ThumbnailCudaContext, x)

Definition at line 75 of file vf_thumbnail_cuda.c.

#define FLAGS   AV_OPT_FLAG_VIDEO_PARAM|AV_OPT_FLAG_FILTERING_PARAM

Definition at line 76 of file vf_thumbnail_cuda.c.

Function Documentation

AVFILTER_DEFINE_CLASS

(

thumbnail_cuda

)

static av_cold int init ( AVFilterContext *  ctx )

static

Definition at line 85 of file vf_thumbnail_cuda.c.

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 105 of file vf_thumbnail_cuda.c.

Referenced by get_best_frame().

static AVFrame* get_best_frame ( AVFilterContext *  ctx )

static

Definition at line 117 of file vf_thumbnail_cuda.c.

Referenced by request_frame().

static int thumbnail_kernel ( ThumbnailCudaContext *  s, CUfunction  func, CUtexref  tex, int  channels, int *  histogram, uint8_t *  src_dptr, int  src_width, int  src_height, int  src_pitch, int  pixel_size  )

static

Definition at line 157 of file vf_thumbnail_cuda.c.

Referenced by thumbnail().

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

static

Definition at line 180 of file vf_thumbnail_cuda.c.

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

static

Definition at line 230 of file vf_thumbnail_cuda.c.

static av_cold void uninit ( AVFilterContext *  ctx )

static

Definition at line 289 of file vf_thumbnail_cuda.c.

static int request_frame ( AVFilterLink *  link )

static

Definition at line 309 of file vf_thumbnail_cuda.c.

static int format_is_supported ( enum AVPixelFormat  fmt )

static

Definition at line 326 of file vf_thumbnail_cuda.c.

Referenced by config_props().

static int config_props ( AVFilterLink *  inlink )

static

Definition at line 336 of file vf_thumbnail_cuda.c.

static int query_formats ( AVFilterContext *  ctx )

static

Definition at line 403 of file vf_thumbnail_cuda.c.

Variable Documentation

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_P016,
    AV_PIX_FMT_YUV444P16,
}

Definition at line 38 of file vf_thumbnail_cuda.c.

Referenced by format_is_supported().

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 78 of file vf_thumbnail_cuda.c.

const AVFilterPad thumbnail_cuda_inputs[]

static

Initial value:

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

Definition at line 415 of file vf_thumbnail_cuda.c.

const AVFilterPad thumbnail_cuda_outputs[]

static

Initial value:

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

Definition at line 425 of file vf_thumbnail_cuda.c.

AVFilter ff_vf_thumbnail_cuda

Initial value:

= {
    .name          = "thumbnail_cuda",
    .description   = NULL_IF_CONFIG_SMALL("Select the most representative frame in a given sequence of consecutive frames."),
    .priv_size     = sizeof(ThumbnailCudaContext),
    .init          = init,
    .uninit        = uninit,
    .query_formats = query_formats,
    .inputs        = thumbnail_cuda_inputs,
    .outputs       = thumbnail_cuda_outputs,
    .priv_class    = &thumbnail_cuda_class,
    .flags_internal = FF_FILTER_FLAG_HWFRAME_AWARE,
}

Definition at line 434 of file vf_thumbnail_cuda.c.