cuviddec.c File Reference

#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 "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 int	cuvid_decode_frame (AVCodecContext *avctx, void *data, int *got_frame, AVPacket *avpkt)
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 *	cuvid_hw_configs []

Macro Definition Documentation

#define cudaVideoSurfaceFormat_YUV444   2

Definition at line 41 of file cuviddec.c.

#define cudaVideoSurfaceFormat_YUV444_16Bit   3

Definition at line 42 of file cuviddec.c.

#define CHECK_CU

(

x

)

FF_CUDA_CHECK_DL(avctx, ctx->cudl, x)

Definition at line 104 of file cuviddec.c.

Referenced by cuvid_decode_init(), cuvid_decode_packet(), cuvid_handle_picture_decode(), cuvid_output_frame(), cuvid_test_capabilities(), vulkan_device_derive(), vulkan_free_internal(), vulkan_map_frame_to_mem(), and vulkan_transfer_data_to().

#define OFFSET

(

x

)

offsetof(CuvidContext, x)

Definition at line 1070 of file cuviddec.c.

#define VD   AV_OPT_FLAG_VIDEO_PARAM | AV_OPT_FLAG_DECODING_PARAM

Definition at line 1071 of file cuviddec.c.

Referenced by decode_flush().

#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, \
    }; \
    AVCodec ff_##x##_cuvid_decoder = { \
        .name           = #x "_cuvid", \
        .long_name      = NULL_IF_CONFIG_SMALL("Nvidia CUVID " #X " decoder"), \
        .type           = AVMEDIA_TYPE_VIDEO, \
        .id             = AV_CODEC_ID_##X, \
        .priv_data_size = sizeof(CuvidContext), \
        .priv_class     = &x##_cuvid_class, \
        .init           = cuvid_decode_init, \
        .close          = cuvid_decode_end, \
        .decode         = cuvid_decode_frame, \
        .receive_frame  = cuvid_output_frame, \
        .flush          = cuvid_flush, \
        .bsfs           = bsf_name, \
        .capabilities   = AV_CODEC_CAP_DELAY | AV_CODEC_CAP_AVOID_PROBING | AV_CODEC_CAP_HARDWARE, \
        .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, \
        .wrapper_name   = "cuvid", \
    };

Definition at line 1098 of file cuviddec.c.

Function Documentation

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

static

Definition at line 106 of file cuviddec.c.

Referenced by cuvid_decode_init().

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

static

Definition at line 328 of file cuviddec.c.

Referenced by cuvid_decode_init().

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

static

Definition at line 344 of file cuviddec.c.

Referenced by cuvid_decode_init().

static int cuvid_is_buffer_full ( AVCodecContext *  avctx )

static

Definition at line 370 of file cuviddec.c.

Referenced by cuvid_decode_packet(), and cuvid_output_frame().

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

static

Definition at line 381 of file cuviddec.c.

Referenced by cuvid_decode_frame(), and cuvid_output_frame().

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

static

Definition at line 446 of file cuviddec.c.

Referenced by cuvid_decode_frame().

static int cuvid_decode_frame ( AVCodecContext *  avctx, void *  data, int *  got_frame, AVPacket *  avpkt  )

static

Definition at line 638 of file cuviddec.c.

static av_cold int cuvid_decode_end ( AVCodecContext *  avctx )

static

Definition at line 669 of file cuviddec.c.

Referenced by cuvid_decode_init().

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

static

Definition at line 693 of file cuviddec.c.

Referenced by cuvid_decode_init().

static av_cold int cuvid_decode_init ( AVCodecContext *  avctx )

static

Definition at line 787 of file cuviddec.c.

static void cuvid_flush ( AVCodecContext *  avctx )

static

Definition at line 1014 of file cuviddec.c.

Variable Documentation

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 1072 of file cuviddec.c.

const AVCodecHWConfigInternal* 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 1085 of file cuviddec.c.