cuviddec.c File Reference

#include "config_components.h"
#include "compat/cuda/dynlink_loader.h"
#include "libavutil/buffer.h"
#include "libavutil/mathematics.h"
#include "libavutil/hwcontext.h"
#include "libavutil/hwcontext_cuda_internal.h"
#include "libavutil/cuda_check.h"
#include "libavutil/fifo.h"
#include "libavutil/log.h"
#include "libavutil/opt.h"
#include "libavutil/pixdesc.h"
#include "avcodec.h"
#include "bsf.h"
#include "codec_internal.h"
#include "decode.h"
#include "hwconfig.h"
#include "nvdec.h"
#include "internal.h"

Go to the source code of this file.

Data Structures
struct	CuvidContext
struct	CuvidParsedFrame

Macros
#define	cudaVideoSurfaceFormat_YUV444   2
#define	cudaVideoSurfaceFormat_YUV444_16Bit   3
#define	CHECK_CU(x)   FF_CUDA_CHECK_DL(avctx, ctx->cudl, x)
#define	OFFSET(x)   offsetof(CuvidContext, x)
#define	VD   AV_OPT_FLAG_VIDEO_PARAM | AV_OPT_FLAG_DECODING_PARAM
#define	DEFINE_CUVID_CODEC(x, X, bsf_name)

Functions
static int CUDAAPI	cuvid_handle_video_sequence (void *opaque, CUVIDEOFORMAT *format)
static int CUDAAPI	cuvid_handle_picture_decode (void *opaque, CUVIDPICPARAMS *picparams)
static int CUDAAPI	cuvid_handle_picture_display (void *opaque, CUVIDPARSERDISPINFO *dispinfo)
static int	cuvid_is_buffer_full (AVCodecContext *avctx)
static int	cuvid_decode_packet (AVCodecContext *avctx, const AVPacket *avpkt)
static int	cuvid_output_frame (AVCodecContext *avctx, AVFrame *frame)
static av_cold int	cuvid_decode_end (AVCodecContext *avctx)
static int	cuvid_test_capabilities (AVCodecContext *avctx, const CUVIDPARSERPARAMS *cuparseinfo, int probed_width, int probed_height, int bit_depth)
static av_cold int	cuvid_decode_init (AVCodecContext *avctx)
static void	cuvid_flush (AVCodecContext *avctx)

Variables
static const AVOption	options []
static const AVCodecHWConfigInternal *const	cuvid_hw_configs []

Macro Definition Documentation

cudaVideoSurfaceFormat_YUV444

#define cudaVideoSurfaceFormat_YUV444   2

Definition at line 45 of file cuviddec.c.

cudaVideoSurfaceFormat_YUV444_16Bit

#define cudaVideoSurfaceFormat_YUV444_16Bit   3

Definition at line 46 of file cuviddec.c.

CHECK_CU

#define CHECK_CU

(

x

)

FF_CUDA_CHECK_DL(avctx, ctx->cudl, x)

Definition at line 116 of file cuviddec.c.

OFFSET

#define OFFSET

(

x

)

offsetof(CuvidContext, x)

Definition at line 1084 of file cuviddec.c.

VD

#define VD   AV_OPT_FLAG_VIDEO_PARAM | AV_OPT_FLAG_DECODING_PARAM

Definition at line 1085 of file cuviddec.c.

DEFINE_CUVID_CODEC

#define DEFINE_CUVID_CODEC	(		x,
			X,
			bsf_name
	)

Value:

    static const AVClass x##_cuvid_class = { \
        .class_name = #x "_cuvid", \
        .item_name = av_default_item_name, \
        .option = options, \
        .version = LIBAVUTIL_VERSION_INT, \
    }; \
    const FFCodec ff_##x##_cuvid_decoder = { \
        .p.name         = #x "_cuvid", \
        .p.long_name    = NULL_IF_CONFIG_SMALL("Nvidia CUVID " #X " decoder"), \
        .p.type         = AVMEDIA_TYPE_VIDEO, \
        .p.id           = AV_CODEC_ID_##X, \
        .priv_data_size = sizeof(CuvidContext), \
        .p.priv_class   = &x##_cuvid_class, \
        .init           = cuvid_decode_init, \
        .close          = cuvid_decode_end, \
        FF_CODEC_RECEIVE_FRAME_CB(cuvid_output_frame), \
        .flush          = cuvid_flush, \
        .bsfs           = bsf_name, \
        .p.capabilities = AV_CODEC_CAP_DELAY | AV_CODEC_CAP_AVOID_PROBING | AV_CODEC_CAP_HARDWARE, \
        .caps_internal  = FF_CODEC_CAP_SETS_FRAME_PROPS, \
        .p.pix_fmts     = (const enum AVPixelFormat[]){ AV_PIX_FMT_CUDA, \
                                                        AV_PIX_FMT_NV12, \
                                                        AV_PIX_FMT_P010, \
                                                        AV_PIX_FMT_P016, \
                                                        AV_PIX_FMT_NONE }, \
        .hw_configs     = cuvid_hw_configs, \
        .p.wrapper_name = "cuvid", \
    };

Definition at line 1112 of file cuviddec.c.

Function Documentation

cuvid_handle_video_sequence()

static int CUDAAPI cuvid_handle_video_sequence ( void *  opaque, CUVIDEOFORMAT *  format  )

static

Definition at line 118 of file cuviddec.c.

Referenced by cuvid_decode_init().

cuvid_handle_picture_decode()

static int CUDAAPI cuvid_handle_picture_decode ( void *  opaque, CUVIDPICPARAMS *  picparams  )

static

Definition at line 340 of file cuviddec.c.

Referenced by cuvid_decode_init().

cuvid_handle_picture_display()

static int CUDAAPI cuvid_handle_picture_display ( void *  opaque, CUVIDPARSERDISPINFO *  dispinfo  )

static

Definition at line 357 of file cuviddec.c.

Referenced by cuvid_decode_init().

cuvid_is_buffer_full()

static int cuvid_is_buffer_full ( AVCodecContext *  avctx )

static

Definition at line 383 of file cuviddec.c.

Referenced by cuvid_decode_packet(), and cuvid_output_frame().

cuvid_decode_packet()

static int cuvid_decode_packet ( AVCodecContext *  avctx, const AVPacket *  avpkt  )

static

Definition at line 394 of file cuviddec.c.

Referenced by cuvid_output_frame().

cuvid_output_frame()

static int cuvid_output_frame ( AVCodecContext *  avctx, AVFrame *  frame  )

static

Definition at line 459 of file cuviddec.c.

cuvid_decode_end()

static av_cold int cuvid_decode_end ( AVCodecContext *  avctx )

static

Definition at line 655 of file cuviddec.c.

Referenced by cuvid_decode_init().

cuvid_test_capabilities()

static int cuvid_test_capabilities ( AVCodecContext *  avctx, const CUVIDPARSERPARAMS *  cuparseinfo, int  probed_width, int  probed_height, int  bit_depth  )

static

Definition at line 689 of file cuviddec.c.

Referenced by cuvid_decode_init().

cuvid_decode_init()

static av_cold int cuvid_decode_init ( AVCodecContext *  avctx )

static

Definition at line 783 of file cuviddec.c.

cuvid_flush()

static void cuvid_flush ( AVCodecContext *  avctx )

static

Definition at line 1034 of file cuviddec.c.

Variable Documentation

options

const AVOption options[]

static

Initial value:

= {
    { "deint",    "Set deinterlacing mode", OFFSET(deint_mode), AV_OPT_TYPE_INT,   { .i64 = cudaVideoDeinterlaceMode_Weave    }, cudaVideoDeinterlaceMode_Weave, cudaVideoDeinterlaceMode_Adaptive, VD, "deint" },
    { "weave",    "Weave deinterlacing (do nothing)",        0, AV_OPT_TYPE_CONST, { .i64 = cudaVideoDeinterlaceMode_Weave    }, 0, 0, VD, "deint" },
    { "bob",      "Bob deinterlacing",                       0, AV_OPT_TYPE_CONST, { .i64 = cudaVideoDeinterlaceMode_Bob      }, 0, 0, VD, "deint" },
    { "adaptive", "Adaptive deinterlacing",                  0, AV_OPT_TYPE_CONST, { .i64 = cudaVideoDeinterlaceMode_Adaptive }, 0, 0, VD, "deint" },
    { "gpu",      "GPU to be used for decoding", OFFSET(cu_gpu), AV_OPT_TYPE_STRING, { .str = NULL }, 0, 0, VD },
    { "surfaces", "Maximum surfaces to be used for decoding", OFFSET(nb_surfaces), AV_OPT_TYPE_INT, { .i64 = 25 }, 0, INT_MAX, VD },
    { "drop_second_field", "Drop second field when deinterlacing", OFFSET(drop_second_field), AV_OPT_TYPE_BOOL, { .i64 = 0 }, 0, 1, VD },
    { "crop",     "Crop (top)x(bottom)x(left)x(right)", OFFSET(crop_expr), AV_OPT_TYPE_STRING, { .str = NULL }, 0, 0, VD },
    { "resize",   "Resize (width)x(height)", OFFSET(resize_expr), AV_OPT_TYPE_STRING, { .str = NULL }, 0, 0, VD },
    { NULL }
}

Definition at line 1086 of file cuviddec.c.

cuvid_hw_configs

const AVCodecHWConfigInternal* const cuvid_hw_configs[]

static

Initial value:

= {
    &(const AVCodecHWConfigInternal) {
        .public = {
            .pix_fmt     = AV_PIX_FMT_CUDA,
            .methods     = AV_CODEC_HW_CONFIG_METHOD_HW_DEVICE_CTX |
                           AV_CODEC_HW_CONFIG_METHOD_INTERNAL,
            .device_type = AV_HWDEVICE_TYPE_CUDA
        },
        .hwaccel = NULL,
    },
    NULL
}

Definition at line 1099 of file cuviddec.c.