nvdec.c File Reference

#include "config.h"
#include "config_components.h"
#include <stdatomic.h>
#include "libavutil/avassert.h"
#include "libavutil/common.h"
#include "libavutil/error.h"
#include "libavutil/hwcontext.h"
#include "libavutil/hwcontext_cuda_internal.h"
#include "libavutil/cuda_check.h"
#include "libavutil/mem.h"
#include "libavutil/pixdesc.h"
#include "libavutil/pixfmt.h"
#include "avcodec.h"
#include "decode.h"
#include "nvdec.h"
#include "internal.h"
#include "libavutil/refstruct.h"

Go to the source code of this file.

Data Structures
struct	NVDECDecoder
struct	NVDECFramePool

Macros
#define	cudaVideoSurfaceFormat_YUV444   2
#define	cudaVideoSurfaceFormat_YUV444_16Bit   3
#define	CHECK_CU(x)   FF_CUDA_CHECK_DL(logctx, decoder->cudl, x)

Functions
static int	map_avcodec_id (enum AVCodecID id)
static int	map_chroma_format (enum AVPixelFormat pix_fmt)
static int	nvdec_test_capabilities (NVDECDecoder *decoder, CUVIDDECODECREATEINFO *params, void *logctx)
static void	nvdec_decoder_free (AVRefStructOpaque unused, void *obj)
static int	nvdec_decoder_create (NVDECDecoder **out, AVBufferRef *hw_device_ref, CUVIDDECODECREATEINFO *params, void *logctx)
static int	nvdec_decoder_frame_init (AVRefStructOpaque opaque, void *obj)
static void	nvdec_decoder_frame_pool_free (AVRefStructOpaque opaque)
int	ff_nvdec_decode_uninit (AVCodecContext *avctx)
static void	nvdec_free_dummy (struct AVHWFramesContext *ctx)
static AVBufferRef *	nvdec_alloc_dummy (size_t size)
static int	nvdec_init_hwframes (AVCodecContext *avctx, AVBufferRef **out_frames_ref, int dummy)
int	ff_nvdec_decode_init (AVCodecContext *avctx)
static void	nvdec_fdd_priv_free (void *priv)
static void	nvdec_unmap_mapped_frame (void *opaque, uint8_t *data)
static int	nvdec_retrieve_data (void *logctx, AVFrame *frame)
int	ff_nvdec_start_frame (AVCodecContext *avctx, AVFrame *frame)
int	ff_nvdec_start_frame_sep_ref (AVCodecContext *avctx, AVFrame *frame, int has_sep_ref)
int	ff_nvdec_end_frame (AVCodecContext *avctx)
int	ff_nvdec_simple_end_frame (AVCodecContext *avctx)
int	ff_nvdec_simple_decode_slice (AVCodecContext *avctx, const uint8_t *buffer, uint32_t size)
int	ff_nvdec_frame_params (AVCodecContext *avctx, AVBufferRef *hw_frames_ctx, enum AVPixelFormat hw_format, int dpb_size, int supports_444)
int	ff_nvdec_get_ref_idx (AVFrame *frame)

Macro Definition Documentation

cudaVideoSurfaceFormat_YUV444

#define cudaVideoSurfaceFormat_YUV444   2

Definition at line 45 of file nvdec.c.

cudaVideoSurfaceFormat_YUV444_16Bit

#define cudaVideoSurfaceFormat_YUV444_16Bit   3

Definition at line 46 of file nvdec.c.

CHECK_CU

#define CHECK_CU

(

x

)

FF_CUDA_CHECK_DL(logctx, decoder->cudl, x)

Definition at line 74 of file nvdec.c.

Function Documentation

map_avcodec_id()

static int map_avcodec_id ( enum AVCodecID  id )

static

Definition at line 76 of file nvdec.c.

Referenced by ff_nvdec_decode_init(), and ff_nvdec_frame_params().

map_chroma_format()

static int map_chroma_format ( enum AVPixelFormat  pix_fmt )

static

Definition at line 96 of file nvdec.c.

Referenced by ff_nvdec_decode_init(), and ff_nvdec_frame_params().

nvdec_test_capabilities()

static int nvdec_test_capabilities ( NVDECDecoder *  decoder, CUVIDDECODECREATEINFO *  params, void *  logctx  )

static

Definition at line 115 of file nvdec.c.

Referenced by nvdec_decoder_create().

nvdec_decoder_free()

static void nvdec_decoder_free ( AVRefStructOpaque  unused, void *  obj  )

static

Definition at line 175 of file nvdec.c.

Referenced by nvdec_decoder_create().

nvdec_decoder_create()

static int nvdec_decoder_create ( NVDECDecoder **  out, AVBufferRef *  hw_device_ref, CUVIDDECODECREATEINFO *  params, void *  logctx  )

static

Definition at line 222 of file nvdec.c.

Referenced by ff_nvdec_decode_init().

nvdec_decoder_frame_init()

static int nvdec_decoder_frame_init ( AVRefStructOpaque  opaque, void *  obj  )

static

Definition at line 279 of file nvdec.c.

nvdec_decoder_frame_pool_free()

static void nvdec_decoder_frame_pool_free ( AVRefStructOpaque  opaque )

static

Definition at line 292 of file nvdec.c.

ff_nvdec_decode_uninit()

int ff_nvdec_decode_uninit

(

AVCodecContext *

avctx

)

Definition at line 297 of file nvdec.c.

nvdec_free_dummy()

static void nvdec_free_dummy ( struct AVHWFramesContext *  ctx )

static

Definition at line 316 of file nvdec.c.

Referenced by nvdec_init_hwframes().

nvdec_alloc_dummy()

static AVBufferRef* nvdec_alloc_dummy ( size_t  size )

static

Definition at line 321 of file nvdec.c.

Referenced by nvdec_init_hwframes().

nvdec_init_hwframes()

static int nvdec_init_hwframes ( AVCodecContext *  avctx, AVBufferRef **  out_frames_ref, int  dummy  )

static

Definition at line 326 of file nvdec.c.

Referenced by ff_nvdec_decode_init().

ff_nvdec_decode_init()

int ff_nvdec_decode_init

(

AVCodecContext *

avctx

)

Definition at line 404 of file nvdec.c.

Referenced by nvdec_hevc_decode_init().

nvdec_fdd_priv_free()

static void nvdec_fdd_priv_free ( void *  priv )

static

Definition at line 714 of file nvdec.c.

Referenced by ff_nvdec_start_frame().

nvdec_unmap_mapped_frame()

static void nvdec_unmap_mapped_frame ( void *  opaque, uint8_t *  data  )

static

Definition at line 728 of file nvdec.c.

Referenced by nvdec_retrieve_data().

nvdec_retrieve_data()

static int nvdec_retrieve_data ( void *  logctx, AVFrame *  frame  )

static

Definition at line 771 of file nvdec.c.

Referenced by ff_nvdec_start_frame().

ff_nvdec_start_frame()

int ff_nvdec_start_frame	(	AVCodecContext *	avctx,
		AVFrame *	frame
	)

Definition at line 946 of file nvdec.c.

Referenced by ff_nvdec_start_frame_sep_ref(), nvdec_h264_start_frame(), nvdec_hevc_start_frame(), nvdec_mjpeg_start_frame(), nvdec_mpeg12_start_frame(), nvdec_mpeg4_start_frame(), nvdec_vc1_start_frame(), nvdec_vp8_start_frame(), and nvdec_vp9_start_frame().

ff_nvdec_start_frame_sep_ref()

int ff_nvdec_start_frame_sep_ref	(	AVCodecContext *	avctx,
		AVFrame *	frame,
		int	has_sep_ref
	)

Definition at line 984 of file nvdec.c.

Referenced by nvdec_av1_start_frame().

ff_nvdec_end_frame()

int ff_nvdec_end_frame

(

AVCodecContext *

avctx

)

Definition at line 1014 of file nvdec.c.

Referenced by ff_nvdec_simple_end_frame().

ff_nvdec_simple_end_frame()

int ff_nvdec_simple_end_frame

(

AVCodecContext *

avctx

)

Definition at line 1051 of file nvdec.c.

ff_nvdec_simple_decode_slice()

int ff_nvdec_simple_decode_slice	(	AVCodecContext *	avctx,
		const uint8_t *	buffer,
		uint32_t	size
	)

Definition at line 1061 of file nvdec.c.

Referenced by nvdec_mjpeg_start_frame(), and nvdec_mpeg4_start_frame().

ff_nvdec_frame_params()

int ff_nvdec_frame_params	(	AVCodecContext *	avctx,
		AVBufferRef *	hw_frames_ctx,
		enum AVPixelFormat	hw_format,
		int	dpb_size,
		int	supports_444
	)

Definition at line 1083 of file nvdec.c.

Referenced by nvdec_av1_frame_params(), nvdec_h264_frame_params(), nvdec_hevc_frame_params(), nvdec_mjpeg_frame_params(), nvdec_mpeg12_frame_params(), nvdec_mpeg4_frame_params(), nvdec_vc1_frame_params(), nvdec_vp8_frame_params(), and nvdec_vp9_frame_params().

ff_nvdec_get_ref_idx()

int ff_nvdec_get_ref_idx

(

AVFrame *

frame

)

Definition at line 1180 of file nvdec.c.

Referenced by nvdec_av1_start_frame(), nvdec_mpeg12_start_frame(), nvdec_mpeg4_start_frame(), nvdec_vc1_start_frame(), nvdec_vp9_start_frame(), and safe_get_ref_idx().