FFmpeg
Data Structures | Macros | Functions | Variables
vf_colorspace_cuda.c File Reference
#include <string.h>
#include "libavutil/common.h"
#include "libavutil/cuda_check.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 "internal.h"
#include "cuda/load_helper.h"

Go to the source code of this file.

Data Structures

struct  CUDAColorspaceContext
 

Macros

#define DIV_UP(a, b)   (((a) + (b)-1) / (b))
 
#define BLOCKX   32
 
#define BLOCKY   16
 
#define CHECK_CU(x)   FF_CUDA_CHECK_DL(ctx, s->hwctx->internal->cuda_dl, x)
 
#define OFFSET(x)   offsetof(CUDAColorspaceContext, x)
 
#define FLAGS   (AV_OPT_FLAG_FILTERING_PARAM | AV_OPT_FLAG_VIDEO_PARAM)
 

Functions

static av_cold int cudacolorspace_init (AVFilterContext *ctx)
 
static av_cold void cudacolorspace_uninit (AVFilterContext *ctx)
 
static av_cold int init_hwframe_ctx (CUDAColorspaceContext *s, AVBufferRef *device_ctx, int width, int height)
 
static int format_is_supported (enum AVPixelFormat fmt)
 
static av_cold int init_processing_chain (AVFilterContext *ctx, int width, int height)
 
static av_cold int cudacolorspace_load_functions (AVFilterContext *ctx)
 
static av_cold int cudacolorspace_config_props (AVFilterLink *outlink)
 
static int conv_cuda_convert (AVFilterContext *ctx, AVFrame *out, AVFrame *in)
 
static int cudacolorspace_conv (AVFilterContext *ctx, AVFrame *out, AVFrame *in)
 
static int cudacolorspace_filter_frame (AVFilterLink *link, AVFrame *in)
 

Variables

static enum AVPixelFormat supported_formats []
 
static const AVOption options []
 
static const AVClass cudacolorspace_class
 
static const AVFilterPad cudacolorspace_inputs []
 
static const AVFilterPad cudacolorspace_outputs []
 
const AVFilter ff_vf_colorspace_cuda
 

Macro Definition Documentation

◆ DIV_UP

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

Definition at line 44 of file vf_colorspace_cuda.c.

◆ BLOCKX

#define BLOCKX   32

Definition at line 45 of file vf_colorspace_cuda.c.

◆ BLOCKY

#define BLOCKY   16

Definition at line 46 of file vf_colorspace_cuda.c.

◆ CHECK_CU

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

Definition at line 48 of file vf_colorspace_cuda.c.

◆ OFFSET

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

Definition at line 382 of file vf_colorspace_cuda.c.

◆ FLAGS

#define FLAGS   (AV_OPT_FLAG_FILTERING_PARAM | AV_OPT_FLAG_VIDEO_PARAM)

Definition at line 383 of file vf_colorspace_cuda.c.

Function Documentation

◆ cudacolorspace_init()

static av_cold int cudacolorspace_init ( AVFilterContext *  ctx)
static

Definition at line 69 of file vf_colorspace_cuda.c.

◆ cudacolorspace_uninit()

static av_cold void cudacolorspace_uninit ( AVFilterContext *  ctx)
static

Definition at line 84 of file vf_colorspace_cuda.c.

◆ init_hwframe_ctx()

static av_cold int init_hwframe_ctx ( CUDAColorspaceContext *  s,
AVBufferRef *  device_ctx,
int  width,
int  height 
)
static

Definition at line 103 of file vf_colorspace_cuda.c.

Referenced by init_processing_chain().

◆ format_is_supported()

static int format_is_supported ( enum AVPixelFormat  fmt)
static

Definition at line 142 of file vf_colorspace_cuda.c.

Referenced by init_processing_chain().

◆ init_processing_chain()

static av_cold int init_processing_chain ( AVFilterContext *  ctx,
int  width,
int  height 
)
static

Definition at line 151 of file vf_colorspace_cuda.c.

Referenced by cudacolorspace_config_props().

◆ cudacolorspace_load_functions()

static av_cold int cudacolorspace_load_functions ( AVFilterContext *  ctx)
static

Definition at line 191 of file vf_colorspace_cuda.c.

Referenced by cudacolorspace_config_props().

◆ cudacolorspace_config_props()

static av_cold int cudacolorspace_config_props ( AVFilterLink *  outlink)
static

Definition at line 224 of file vf_colorspace_cuda.c.

◆ conv_cuda_convert()

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

Definition at line 259 of file vf_colorspace_cuda.c.

Referenced by cudacolorspace_conv().

◆ cudacolorspace_conv()

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

Definition at line 313 of file vf_colorspace_cuda.c.

Referenced by cudacolorspace_filter_frame().

◆ cudacolorspace_filter_frame()

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

Definition at line 342 of file vf_colorspace_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,
}

Definition at line 38 of file vf_colorspace_cuda.c.

Referenced by format_is_supported().

◆ options

const AVOption options[]
static
Initial value:
= {
{"range", "Output video range", OFFSET(range), AV_OPT_TYPE_INT, { .i64 = AVCOL_RANGE_UNSPECIFIED }, AVCOL_RANGE_UNSPECIFIED, AVCOL_RANGE_NB - 1, FLAGS, .unit = "range"},
{"tv", "Limited range", 0, AV_OPT_TYPE_CONST, { .i64 = AVCOL_RANGE_MPEG }, 0, 0, FLAGS, .unit = "range"},
{"mpeg", "Limited range", 0, AV_OPT_TYPE_CONST, { .i64 = AVCOL_RANGE_MPEG }, 0, 0, FLAGS, .unit = "range"},
{"pc", "Full range", 0, AV_OPT_TYPE_CONST, { .i64 = AVCOL_RANGE_JPEG }, 0, 0, FLAGS, .unit = "range"},
{"jpeg", "Full range", 0, AV_OPT_TYPE_CONST, { .i64 = AVCOL_RANGE_JPEG }, 0, 0, FLAGS, .unit = "range"},
{NULL},
}

Definition at line 384 of file vf_colorspace_cuda.c.

◆ cudacolorspace_class

const AVClass cudacolorspace_class
static
Initial value:
= {
.class_name = "colorspace_cuda",
.item_name = av_default_item_name,
.option = options,
.version = LIBAVUTIL_VERSION_INT,
}

Definition at line 393 of file vf_colorspace_cuda.c.

◆ cudacolorspace_inputs

const AVFilterPad cudacolorspace_inputs[]
static
Initial value:
= {
{
.name = "default",
.type = AVMEDIA_TYPE_VIDEO,
.filter_frame = cudacolorspace_filter_frame,
},
}

Definition at line 400 of file vf_colorspace_cuda.c.

◆ cudacolorspace_outputs

const AVFilterPad cudacolorspace_outputs[]
static
Initial value:
= {
{
.name = "default",
.type = AVMEDIA_TYPE_VIDEO,
.config_props = cudacolorspace_config_props,
},
}

Definition at line 408 of file vf_colorspace_cuda.c.

◆ ff_vf_colorspace_cuda

const AVFilter ff_vf_colorspace_cuda
Initial value:
= {
.name = "colorspace_cuda",
.description = NULL_IF_CONFIG_SMALL("CUDA accelerated video color converter"),
.init = cudacolorspace_init,
.uninit = cudacolorspace_uninit,
.priv_size = sizeof(CUDAColorspaceContext),
.priv_class = &cudacolorspace_class,
FILTER_INPUTS(cudacolorspace_inputs),
FILTER_OUTPUTS(cudacolorspace_outputs),
FILTER_SINGLE_PIXFMT(AV_PIX_FMT_CUDA),
.flags_internal = FF_FILTER_FLAG_HWFRAME_AWARE,
}

Definition at line 416 of file vf_colorspace_cuda.c.

AV_PIX_FMT_CUDA
@ AV_PIX_FMT_CUDA
HW acceleration through CUDA.
Definition: pixfmt.h:260
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:351
cudacolorspace_config_props
static av_cold int cudacolorspace_config_props(AVFilterLink *outlink)
Definition: vf_colorspace_cuda.c:224
cudacolorspace_uninit
static av_cold void cudacolorspace_uninit(AVFilterContext *ctx)
Definition: vf_colorspace_cuda.c:84
AVCOL_RANGE_JPEG
@ AVCOL_RANGE_JPEG
Full range content.
Definition: pixfmt.h:683
OFFSET
#define OFFSET(x)
Definition: vf_colorspace_cuda.c:382
FLAGS
#define FLAGS
Definition: vf_colorspace_cuda.c:383
AVCOL_RANGE_NB
@ AVCOL_RANGE_NB
Not part of ABI.
Definition: pixfmt.h:684
AV_PIX_FMT_YUV420P
@ AV_PIX_FMT_YUV420P
planar YUV 4:2:0, 12bpp, (1 Cr & Cb sample per 2x2 Y samples)
Definition: pixfmt.h:73
FILTER_INPUTS
#define FILTER_INPUTS(array)
Definition: internal.h:182
LIBAVUTIL_VERSION_INT
#define LIBAVUTIL_VERSION_INT
Definition: version.h:85
NULL
#define NULL
Definition: coverity.c:32
av_default_item_name
const char * av_default_item_name(void *ptr)
Return the context name.
Definition: log.c:237
CUDAColorspaceContext
Definition: vf_colorspace_cuda.c:50
AVCOL_RANGE_UNSPECIFIED
@ AVCOL_RANGE_UNSPECIFIED
Definition: pixfmt.h:649
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:106
range
enum AVColorRange range
Definition: mediacodec_wrapper.c:2557
FILTER_SINGLE_PIXFMT
#define FILTER_SINGLE_PIXFMT(pix_fmt_)
Definition: internal.h:172
cudacolorspace_inputs
static const AVFilterPad cudacolorspace_inputs[]
Definition: vf_colorspace_cuda.c:400
AVCOL_RANGE_MPEG
@ AVCOL_RANGE_MPEG
Narrow or limited range content.
Definition: pixfmt.h:666
AV_PIX_FMT_NV12
@ 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:96
cudacolorspace_init
static av_cold int cudacolorspace_init(AVFilterContext *ctx)
Definition: vf_colorspace_cuda.c:69
cudacolorspace_filter_frame
static int cudacolorspace_filter_frame(AVFilterLink *link, AVFrame *in)
Definition: vf_colorspace_cuda.c:342
AV_OPT_TYPE_INT
@ AV_OPT_TYPE_INT
Definition: opt.h:235
AV_PIX_FMT_YUV444P
@ AV_PIX_FMT_YUV444P
planar YUV 4:4:4, 24bpp, (1 Cr & Cb sample per 1x1 Y samples)
Definition: pixfmt.h:78
AVMEDIA_TYPE_VIDEO
@ AVMEDIA_TYPE_VIDEO
Definition: avutil.h:201
FILTER_OUTPUTS
#define FILTER_OUTPUTS(array)
Definition: internal.h:183
cudacolorspace_class
static const AVClass cudacolorspace_class
Definition: vf_colorspace_cuda.c:393
cudacolorspace_outputs
static const AVFilterPad cudacolorspace_outputs[]
Definition: vf_colorspace_cuda.c:408
AV_OPT_TYPE_CONST
@ AV_OPT_TYPE_CONST
Definition: opt.h:244
options
static const AVOption options[]
Definition: vf_colorspace_cuda.c:384
Generated on Thu Sep 26 2024 23:16:16 for FFmpeg by   doxygen 1.8.17