vf_transpose_cuda.c File Reference

#include "libavutil/avassert.h"
#include "libavutil/common.h"
#include "libavutil/hwcontext.h"
#include "libavutil/hwcontext_cuda_internal.h"
#include "libavutil/cuda_check.h"
#include "libavutil/internal.h"
#include "libavutil/opt.h"
#include "libavutil/pixdesc.h"
#include "avfilter.h"
#include "filters.h"
#include "transpose.h"
#include "video.h"
#include "cuda/load_helper.h"

Go to the source code of this file.

Data Structures
struct	TransposeCUDAContext

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

Functions
static av_cold int	cudatranspose_init (AVFilterContext *ctx)
static av_cold void	cudatranspose_uninit (AVFilterContext *ctx)
static av_cold int	init_hwframe_ctx (TransposeCUDAContext *s, AVBufferRef *device_ctx, int width, int height, enum AVPixelFormat sw_format)
static int	format_is_supported (enum AVPixelFormat fmt)
static int	init_processing_chain (AVFilterContext *ctx, int out_width, int out_height)
static int	cudatranspose_config_props (AVFilterLink *outlink)
static CUresult	call_kernel (AVFilterContext *ctx, CUfunction cu_func, CUarray_format cu_format, int channels, int is_422_uv, CUdeviceptr dst0, CUdeviceptr dst1, int dst_width, int dst_height, int dst_pitch, CUdeviceptr src0, CUdeviceptr src1, int src_width, int src_height, int src_pitch)
static int	cudatranspose_rotate (AVFilterContext *ctx, AVFrame *out, AVFrame *in)
static int	cudatranspose_transpose (AVFilterContext *ctx, AVFrame *out, AVFrame *in)
static int	cudatranspose_filter_frame (AVFilterLink *link, AVFrame *in)
static AVFrame *	cudatranspose_get_video_buffer (AVFilterLink *inlink, int w, int h)
	AVFILTER_DEFINE_CLASS (cudatranspose)

Variables
static enum AVPixelFormat	supported_formats []
static const AVOption	cudatranspose_options []
static const AVFilterPad	cudatranspose_inputs []
static const AVFilterPad	cudatranspose_outputs []
const FFFilter	ff_vf_transpose_cuda

Macro Definition Documentation

DIV_UP

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

Definition at line 37 of file vf_transpose_cuda.c.

BLOCK_X

#define BLOCK_X   32

Definition at line 38 of file vf_transpose_cuda.c.

BLOCK_Y

#define BLOCK_Y   16

Definition at line 39 of file vf_transpose_cuda.c.

CHECK_CU

#define CHECK_CU

(

x

)

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

Definition at line 41 of file vf_transpose_cuda.c.

OFFSET

#define OFFSET

(

x

)

offsetof(TransposeCUDAContext, x)

Definition at line 513 of file vf_transpose_cuda.c.

FLAGS

#define FLAGS   (AV_OPT_FLAG_FILTERING_PARAM | AV_OPT_FLAG_VIDEO_PARAM)

Definition at line 514 of file vf_transpose_cuda.c.

Function Documentation

cudatranspose_init()

static av_cold int cudatranspose_init ( AVFilterContext *  ctx )

static

Definition at line 87 of file vf_transpose_cuda.c.

cudatranspose_uninit()

static av_cold void cudatranspose_uninit ( AVFilterContext *  ctx )

static

Definition at line 102 of file vf_transpose_cuda.c.

init_hwframe_ctx()

static av_cold int init_hwframe_ctx ( TransposeCUDAContext *  s, AVBufferRef *  device_ctx, int  width, int  height, enum AVPixelFormat  sw_format  )

static

Definition at line 119 of file vf_transpose_cuda.c.

Referenced by init_processing_chain().

format_is_supported()

static int format_is_supported ( enum AVPixelFormat  fmt )

static

Definition at line 159 of file vf_transpose_cuda.c.

Referenced by init_processing_chain().

init_processing_chain()

static int init_processing_chain ( AVFilterContext *  ctx, int  out_width, int  out_height  )

static

Definition at line 167 of file vf_transpose_cuda.c.

Referenced by cudatranspose_config_props().

cudatranspose_config_props()

static int cudatranspose_config_props ( AVFilterLink *  outlink )

static

Definition at line 208 of file vf_transpose_cuda.c.

call_kernel()

static CUresult call_kernel ( AVFilterContext *  ctx, CUfunction  cu_func, CUarray_format  cu_format, int  channels, int  is_422_uv, CUdeviceptr  dst0, CUdeviceptr  dst1, int  dst_width, int  dst_height, int  dst_pitch, CUdeviceptr  src0, CUdeviceptr  src1, int  src_width, int  src_height, int  src_pitch  )

static

Definition at line 303 of file vf_transpose_cuda.c.

Referenced by cudatranspose_rotate().

cudatranspose_rotate()

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

static

Definition at line 369 of file vf_transpose_cuda.c.

Referenced by cudatranspose_transpose().

cudatranspose_transpose()

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

static

Definition at line 430 of file vf_transpose_cuda.c.

Referenced by cudatranspose_filter_frame().

cudatranspose_filter_frame()

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

static

Definition at line 463 of file vf_transpose_cuda.c.

cudatranspose_get_video_buffer()

static AVFrame* cudatranspose_get_video_buffer ( AVFilterLink *  inlink, int  w, int  h  )

static

Definition at line 504 of file vf_transpose_cuda.c.

AVFILTER_DEFINE_CLASS()

AVFILTER_DEFINE_CLASS

(

cudatranspose

)

Variable Documentation

supported_formats

enum AVPixelFormat supported_formats[]

static

Initial value:

= {
    AV_PIX_FMT_YUV420P,
    AV_PIX_FMT_YUV422P,
    AV_PIX_FMT_YUV444P,
    AV_PIX_FMT_YUV420P10,
    AV_PIX_FMT_YUV422P10,
    AV_PIX_FMT_YUV444P10,
    AV_PIX_FMT_YUV444P16,
    AV_PIX_FMT_NV12,
    AV_PIX_FMT_NV16,
    AV_PIX_FMT_P010,
    AV_PIX_FMT_P210,
    AV_PIX_FMT_P016,
    AV_PIX_FMT_P216,
    AV_PIX_FMT_0RGB32,
    AV_PIX_FMT_0BGR32,
    AV_PIX_FMT_RGB32,
    AV_PIX_FMT_BGR32,
}

Definition at line 43 of file vf_transpose_cuda.c.

Referenced by format_is_supported().

cudatranspose_options

const AVOption cudatranspose_options[]

static

Initial value:

= {
    { "dir", "set transpose direction", OFFSET(dir), AV_OPT_TYPE_INT, { .i64 = TRANSPOSE_CCLOCK_FLIP }, 0, 6, FLAGS, .unit = "dir" },
        { "cclock_flip", "rotate counter-clockwise with vertical flip", 0, AV_OPT_TYPE_CONST, { .i64 = TRANSPOSE_CCLOCK_FLIP }, 0, 0, FLAGS, .unit = "dir" },
        { "clock",       "rotate clockwise",                            0, AV_OPT_TYPE_CONST, { .i64 = TRANSPOSE_CLOCK       }, 0, 0, FLAGS, .unit = "dir" },
        { "cclock",      "rotate counter-clockwise",                    0, AV_OPT_TYPE_CONST, { .i64 = TRANSPOSE_CCLOCK      }, 0, 0, FLAGS, .unit = "dir" },
        { "clock_flip",  "rotate clockwise with vertical flip",         0, AV_OPT_TYPE_CONST, { .i64 = TRANSPOSE_CLOCK_FLIP  }, 0, 0, FLAGS, .unit = "dir" },
        { "reversal",    "rotate by half-turn",                         0, AV_OPT_TYPE_CONST, { .i64 = TRANSPOSE_REVERSAL    }, 0, 0, FLAGS, .unit = "dir" },
        { "hflip",       "flip horizontally",                           0, AV_OPT_TYPE_CONST, { .i64 = TRANSPOSE_HFLIP       }, 0, 0, FLAGS, .unit = "dir" },
        { "vflip",       "flip vertically",                             0, AV_OPT_TYPE_CONST, { .i64 = TRANSPOSE_VFLIP       }, 0, 0, FLAGS, .unit = "dir" },
 
    { "passthrough", "do not apply transposition if the input matches the specified geometry", OFFSET(passthrough), AV_OPT_TYPE_INT, { .i64 = TRANSPOSE_PT_TYPE_NONE },  0, 2, FLAGS, .unit = "passthrough" },
        { "none",      "always apply transposition",  0, AV_OPT_TYPE_CONST, { .i64 = TRANSPOSE_PT_TYPE_NONE },      0, 0, FLAGS, .unit = "passthrough" },
        { "landscape", "preserve landscape geometry", 0, AV_OPT_TYPE_CONST, { .i64 = TRANSPOSE_PT_TYPE_LANDSCAPE }, 0, 0, FLAGS, .unit = "passthrough" },
        { "portrait",  "preserve portrait geometry",  0, AV_OPT_TYPE_CONST, { .i64 = TRANSPOSE_PT_TYPE_PORTRAIT },  0, 0, FLAGS, .unit = "passthrough" },
 
    { NULL },
}

Definition at line 516 of file vf_transpose_cuda.c.

cudatranspose_inputs

const AVFilterPad cudatranspose_inputs[]

static

Initial value:

= {
    {
        .name             = "default",
        .type             = AVMEDIA_TYPE_VIDEO,
        .filter_frame     = cudatranspose_filter_frame,
        .get_buffer.video = cudatranspose_get_video_buffer,
    },
}

Definition at line 536 of file vf_transpose_cuda.c.

cudatranspose_outputs

const AVFilterPad cudatranspose_outputs[]

static

Initial value:

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

Definition at line 545 of file vf_transpose_cuda.c.

ff_vf_transpose_cuda

const FFFilter ff_vf_transpose_cuda

Initial value:

= {
    .p.name         = "transpose_cuda",
    .p.description  = NULL_IF_CONFIG_SMALL("Transpose input video using CUDA"),
    .p.priv_class   = &cudatranspose_class,
    .init           = cudatranspose_init,
    .uninit         = cudatranspose_uninit,
    .priv_size      = sizeof(TransposeCUDAContext),
    FILTER_INPUTS(cudatranspose_inputs),
    FILTER_OUTPUTS(cudatranspose_outputs),
    FILTER_SINGLE_PIXFMT(AV_PIX_FMT_CUDA),
    .flags_internal = FF_FILTER_FLAG_HWFRAME_AWARE,
}

Definition at line 553 of file vf_transpose_cuda.c.