FFmpeg  4.0
Data Structures | Macros | Functions | Variables
vf_scale_cuda.c File Reference
#include <cuda.h>
#include <stdio.h>
#include <string.h>
#include "libavutil/avstring.h"
#include "libavutil/common.h"
#include "libavutil/hwcontext.h"
#include "libavutil/hwcontext_cuda_internal.h"
#include "libavutil/internal.h"
#include "libavutil/opt.h"
#include "libavutil/pixdesc.h"
#include "avfilter.h"
#include "formats.h"
#include "internal.h"
#include "scale.h"
#include "video.h"

Go to the source code of this file.

Data Structures

struct  CUDAScaleContext
 

Macros

#define DIV_UP(a, b)   ( ((a) + (b) - 1) / (b) )
 
#define ALIGN_UP(a, b)   (((a) + (b) - 1) & ~((b) - 1))
 
#define NUM_BUFFERS   2
 
#define BLOCKX   32
 
#define BLOCKY   16
 
#define OFFSET(x)   offsetof(CUDAScaleContext, x)
 
#define FLAGS   (AV_OPT_FLAG_FILTERING_PARAM|AV_OPT_FLAG_VIDEO_PARAM)
 

Functions

static av_cold int cudascale_init (AVFilterContext *ctx)
 
static av_cold void cudascale_uninit (AVFilterContext *ctx)
 
static int cudascale_query_formats (AVFilterContext *ctx)
 
static av_cold int init_stage (CUDAScaleContext *s, AVBufferRef *device_ctx)
 
static int format_is_supported (enum AVPixelFormat fmt)
 
static av_cold int init_processing_chain (AVFilterContext *ctx, int in_width, int in_height, int out_width, int out_height)
 
static av_cold int cudascale_config_props (AVFilterLink *outlink)
 
static int call_resize_kernel (CUDAScaleContext *s, CUfunction func, CUtexref tex, int channels, uint8_t *src_dptr, int src_width, int src_height, int src_pitch, uint8_t *dst_dptr, int dst_width, int dst_height, int dst_pitch, int pixel_size)
 
static int scalecuda_resize (AVFilterContext *ctx, AVFrame *out, AVFrame *in)
 
static int cudascale_scale (AVFilterContext *ctx, AVFrame *out, AVFrame *in)
 
static int cudascale_filter_frame (AVFilterLink *link, AVFrame *in)
 

Variables

static enum AVPixelFormat supported_formats []
 
static const AVOption options []
 
static const AVClass cudascale_class
 
static const AVFilterPad cudascale_inputs []
 
static const AVFilterPad cudascale_outputs []
 
AVFilter ff_vf_scale_cuda
 

Macro Definition Documentation

◆ DIV_UP

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

Definition at line 49 of file vf_scale_cuda.c.

Referenced by call_resize_kernel().

◆ ALIGN_UP

#define ALIGN_UP (   a,
  b 
)    (((a) + (b) - 1) & ~((b) - 1))

Definition at line 50 of file vf_scale_cuda.c.

Referenced by scalecuda_resize().

◆ NUM_BUFFERS

#define NUM_BUFFERS   2

Definition at line 51 of file vf_scale_cuda.c.

◆ BLOCKX

#define BLOCKX   32

Definition at line 52 of file vf_scale_cuda.c.

Referenced by call_resize_kernel().

◆ BLOCKY

#define BLOCKY   16

Definition at line 53 of file vf_scale_cuda.c.

Referenced by call_resize_kernel().

◆ OFFSET

#define OFFSET (   x)    offsetof(CUDAScaleContext, x)

Definition at line 508 of file vf_scale_cuda.c.

◆ FLAGS

#define FLAGS   (AV_OPT_FLAG_FILTERING_PARAM|AV_OPT_FLAG_VIDEO_PARAM)

Definition at line 509 of file vf_scale_cuda.c.

Function Documentation

◆ cudascale_init()

static av_cold int cudascale_init ( AVFilterContext *  ctx)
static

Definition at line 100 of file vf_scale_cuda.c.

◆ cudascale_uninit()

static av_cold void cudascale_uninit ( AVFilterContext *  ctx)
static

Definition at line 116 of file vf_scale_cuda.c.

◆ cudascale_query_formats()

static int cudascale_query_formats ( AVFilterContext *  ctx)
static

Definition at line 125 of file vf_scale_cuda.c.

◆ init_stage()

static av_cold int init_stage ( CUDAScaleContext *  s,
AVBufferRef *  device_ctx 
)
static

Definition at line 135 of file vf_scale_cuda.c.

Referenced by init_processing_chain().

◆ format_is_supported()

static int format_is_supported ( enum AVPixelFormat  fmt)
static

Definition at line 187 of file vf_scale_cuda.c.

Referenced by init_processing_chain().

◆ init_processing_chain()

static av_cold int init_processing_chain ( AVFilterContext *  ctx,
int  in_width,
int  in_height,
int  out_width,
int  out_height 
)
static

Definition at line 197 of file vf_scale_cuda.c.

Referenced by cudascale_config_props().

◆ cudascale_config_props()

static av_cold int cudascale_config_props ( AVFilterLink *  outlink)
static

Definition at line 250 of file vf_scale_cuda.c.

◆ call_resize_kernel()

static int call_resize_kernel ( CUDAScaleContext *  s,
CUfunction  func,
CUtexref  tex,
int  channels,
uint8_t *  src_dptr,
int  src_width,
int  src_height,
int  src_pitch,
uint8_t *  dst_dptr,
int  dst_width,
int  dst_height,
int  dst_pitch,
int  pixel_size 
)
static

Definition at line 342 of file vf_scale_cuda.c.

Referenced by scalecuda_resize().

◆ scalecuda_resize()

static int scalecuda_resize ( AVFilterContext *  ctx,
AVFrame *  out,
AVFrame *  in 
)
static

Definition at line 367 of file vf_scale_cuda.c.

Referenced by cudascale_scale().

◆ cudascale_scale()

static int cudascale_scale ( AVFilterContext *  ctx,
AVFrame *  out,
AVFrame *  in 
)
static

Definition at line 439 of file vf_scale_cuda.c.

Referenced by cudascale_filter_frame().

◆ cudascale_filter_frame()

static int cudascale_filter_frame ( AVFilterLink *  link,
AVFrame *  in 
)
static

Definition at line 464 of file vf_scale_cuda.c.

Variable Documentation

◆ supported_formats

enum AVPixelFormat supported_formats[]
static
Initial value:
= {
AV_PIX_FMT_YUV420P,
AV_PIX_FMT_NV12,
AV_PIX_FMT_YUV444P,
AV_PIX_FMT_P010,
AV_PIX_FMT_P016
}
AV_PIX_FMT_YUV444P
planar YUV 4:4:4, 24bpp, (1 Cr & Cb sample per 1x1 Y samples)
Definition: pixfmt.h:67
AV_PIX_FMT_P016
#define AV_PIX_FMT_P016
Definition: pixfmt.h:414
AV_PIX_FMT_P010
#define AV_PIX_FMT_P010
Definition: pixfmt.h:413
AV_PIX_FMT_NV12
planar YUV 4:2:0, 12bpp, 1 plane for Y and 1 plane for the UV components, which are interleaved (firs...
Definition: pixfmt.h:85
AV_PIX_FMT_YUV420P
planar YUV 4:2:0, 12bpp, (1 Cr & Cb sample per 2x2 Y samples)
Definition: pixfmt.h:62

Definition at line 41 of file vf_scale_cuda.c.

Referenced by format_is_supported().

◆ options

const AVOption options[]
static
Initial value:
= {
{ "w", "Output video width", OFFSET(w_expr), AV_OPT_TYPE_STRING, { .str = "iw" }, .flags = FLAGS },
{ "h", "Output video height", OFFSET(h_expr), AV_OPT_TYPE_STRING, { .str = "ih" }, .flags = FLAGS },
{ NULL },
}
NULL
#define NULL
Definition: coverity.c:32
FLAGS
#define FLAGS
Definition: vf_scale_cuda.c:509
AV_OPT_TYPE_STRING
Definition: opt.h:227
OFFSET
#define OFFSET(x)
Definition: vf_scale_cuda.c:508

Definition at line 510 of file vf_scale_cuda.c.

◆ cudascale_class

const AVClass cudascale_class
static
Initial value:
= {
.class_name = "cudascale",
.item_name = av_default_item_name,
.option = options,
.version = LIBAVUTIL_VERSION_INT,
}
LIBAVUTIL_VERSION_INT
#define LIBAVUTIL_VERSION_INT
Definition: version.h:85
av_default_item_name
const char * av_default_item_name(void *ptr)
Return the context name.
Definition: log.c:191
options
static const AVOption options[]
Definition: vf_scale_cuda.c:510

Definition at line 516 of file vf_scale_cuda.c.

◆ cudascale_inputs

const AVFilterPad cudascale_inputs[]
static
Initial value:
= {
{
.name = "default",
.type = AVMEDIA_TYPE_VIDEO,
.filter_frame = cudascale_filter_frame,
},
{ NULL }
}
NULL
#define NULL
Definition: coverity.c:32
cudascale_filter_frame
static int cudascale_filter_frame(AVFilterLink *link, AVFrame *in)
Definition: vf_scale_cuda.c:464
AVMEDIA_TYPE_VIDEO
Definition: avutil.h:201

Definition at line 523 of file vf_scale_cuda.c.

◆ cudascale_outputs

const AVFilterPad cudascale_outputs[]
static
Initial value:
= {
{
.name = "default",
.type = AVMEDIA_TYPE_VIDEO,
.config_props = cudascale_config_props,
},
{ NULL }
}
NULL
#define NULL
Definition: coverity.c:32
cudascale_config_props
static av_cold int cudascale_config_props(AVFilterLink *outlink)
Definition: vf_scale_cuda.c:250
AVMEDIA_TYPE_VIDEO
Definition: avutil.h:201

Definition at line 532 of file vf_scale_cuda.c.

◆ ff_vf_scale_cuda

AVFilter ff_vf_scale_cuda
Initial value:
= {
.name = "scale_cuda",
.description = NULL_IF_CONFIG_SMALL("GPU accelerated video resizer"),
.init = cudascale_init,
.uninit = cudascale_uninit,
.query_formats = cudascale_query_formats,
.priv_size = sizeof(CUDAScaleContext),
.priv_class = &cudascale_class,
.inputs = cudascale_inputs,
.outputs = cudascale_outputs,
.flags_internal = FF_FILTER_FLAG_HWFRAME_AWARE,
}
FF_FILTER_FLAG_HWFRAME_AWARE
#define FF_FILTER_FLAG_HWFRAME_AWARE
The filter is aware of hardware frames, and any hardware frame context should not be automatically pr...
Definition: internal.h:385
cudascale_outputs
static const AVFilterPad cudascale_outputs[]
Definition: vf_scale_cuda.c:532
NULL_IF_CONFIG_SMALL
#define NULL_IF_CONFIG_SMALL(x)
Return NULL if CONFIG_SMALL is true, otherwise the argument without modification. ...
Definition: internal.h:186
cudascale_class
static const AVClass cudascale_class
Definition: vf_scale_cuda.c:516
inputs
static const AVFilterPad inputs[]
Definition: af_acontrast.c:193
outputs
static const AVFilterPad outputs[]
Definition: af_acontrast.c:203
cudascale_uninit
static av_cold void cudascale_uninit(AVFilterContext *ctx)
Definition: vf_scale_cuda.c:116
cudascale_inputs
static const AVFilterPad cudascale_inputs[]
Definition: vf_scale_cuda.c:523
cudascale_init
static av_cold int cudascale_init(AVFilterContext *ctx)
Definition: vf_scale_cuda.c:100
cudascale_query_formats
static int cudascale_query_formats(AVFilterContext *ctx)
Definition: vf_scale_cuda.c:125

Definition at line 541 of file vf_scale_cuda.c.

Generated on Tue May 22 2018 07:07:14 for FFmpeg by   doxygen 1.8.13