nvenc.c File Reference

#include "config.h"
#include "nvenc.h"
#include "libavutil/hwcontext_cuda.h"
#include "libavutil/hwcontext.h"
#include "libavutil/imgutils.h"
#include "libavutil/avassert.h"
#include "libavutil/mem.h"
#include "libavutil/pixdesc.h"
#include "internal.h"

Go to the source code of this file.

Data Structures
struct	GUIDTuple

Macros
#define	NVENC_CAP   0x30
#define	IS_CBR(rc)
#define	IS_10BIT(pix_fmt)
#define	IS_YUV444(pix_fmt)
#define	PRESET_ALIAS(alias, name, ...)   [PRESET_ ## alias] = { NV_ENC_PRESET_ ## name ## _GUID, __VA_ARGS__ }
#define	PRESET(name, ...)   PRESET_ALIAS(name, name, __VA_ARGS__)

Functions
static int	nvenc_map_error (NVENCSTATUS err, const char **desc)
static int	nvenc_print_error (void *log_ctx, NVENCSTATUS err, const char *error_string)
static void	nvenc_print_driver_requirement (AVCodecContext *avctx, int level)
static av_cold int	nvenc_load_libraries (AVCodecContext *avctx)
static int	nvenc_push_context (AVCodecContext *avctx)
static int	nvenc_pop_context (AVCodecContext *avctx)
static av_cold int	nvenc_open_session (AVCodecContext *avctx)
static int	nvenc_check_codec_support (AVCodecContext *avctx)
static int	nvenc_check_cap (AVCodecContext *avctx, NV_ENC_CAPS cap)
static int	nvenc_check_capabilities (AVCodecContext *avctx)
static av_cold int	nvenc_check_device (AVCodecContext *avctx, int idx)
static av_cold int	nvenc_setup_device (AVCodecContext *avctx)
static void	nvenc_map_preset (NvencContext *ctx)
static av_cold void	set_constqp (AVCodecContext *avctx)
static av_cold void	set_vbr (AVCodecContext *avctx)
static av_cold void	set_lossless (AVCodecContext *avctx)
static void	nvenc_override_rate_control (AVCodecContext *avctx)
static av_cold int	nvenc_recalc_surfaces (AVCodecContext *avctx)
static av_cold void	nvenc_setup_rate_control (AVCodecContext *avctx)
static av_cold int	nvenc_setup_h264_config (AVCodecContext *avctx)
static av_cold int	nvenc_setup_hevc_config (AVCodecContext *avctx)
static av_cold int	nvenc_setup_codec_config (AVCodecContext *avctx)
static av_cold int	nvenc_setup_encoder (AVCodecContext *avctx)
static NV_ENC_BUFFER_FORMAT	nvenc_map_buffer_format (enum AVPixelFormat pix_fmt)
static av_cold int	nvenc_alloc_surface (AVCodecContext *avctx, int idx)
static av_cold int	nvenc_setup_surfaces (AVCodecContext *avctx)
static av_cold int	nvenc_setup_extradata (AVCodecContext *avctx)
av_cold int	ff_nvenc_encode_close (AVCodecContext *avctx)
av_cold int	ff_nvenc_encode_init (AVCodecContext *avctx)
static NvencSurface *	get_free_frame (NvencContext *ctx)
static int	nvenc_copy_frame (AVCodecContext *avctx, NvencSurface *nv_surface, NV_ENC_LOCK_INPUT_BUFFER *lock_buffer_params, const AVFrame *frame)
static int	nvenc_find_free_reg_resource (AVCodecContext *avctx)
static int	nvenc_register_frame (AVCodecContext *avctx, const AVFrame *frame)
static int	nvenc_upload_frame (AVCodecContext *avctx, const AVFrame *frame, NvencSurface *nvenc_frame)
static void	nvenc_codec_specific_pic_params (AVCodecContext *avctx, NV_ENC_PIC_PARAMS *params)
static void	timestamp_queue_enqueue (AVFifoBuffer *queue, int64_t timestamp)
static int64_t	timestamp_queue_dequeue (AVFifoBuffer *queue)
static int	nvenc_set_timestamp (AVCodecContext *avctx, NV_ENC_LOCK_BITSTREAM *params, AVPacket *pkt)
static int	process_output_surface (AVCodecContext *avctx, AVPacket *pkt, NvencSurface *tmpoutsurf)
static int	output_ready (AVCodecContext *avctx, int flush)
int	ff_nvenc_send_frame (AVCodecContext *avctx, const AVFrame *frame)
int	ff_nvenc_receive_packet (AVCodecContext *avctx, AVPacket *pkt)
int	ff_nvenc_encode_frame (AVCodecContext *avctx, AVPacket *pkt, const AVFrame *frame, int *got_packet)

Variables
enum AVPixelFormat	ff_nvenc_pix_fmts []
struct {
NVENCSTATUS   nverr
int   averr
const char *   desc
}	nvenc_errors []

Macro Definition Documentation

NVENC_CAP

#define NVENC_CAP   0x30

Definition at line 34 of file nvenc.c.

Referenced by nvenc_check_device().

IS_CBR

#define IS_CBR

(

rc

)

Value:

(rc == NV_ENC_PARAMS_RC_CBR ||             \
                    rc == NV_ENC_PARAMS_RC_CBR_LOWDELAY_HQ || \
                    rc == NV_ENC_PARAMS_RC_CBR_HQ)

Definition at line 35 of file nvenc.c.

Referenced by nvenc_setup_h264_config(), and nvenc_setup_hevc_config().

IS_10BIT

#define IS_10BIT

(

pix_fmt

)

Value:

(pix_fmt == AV_PIX_FMT_P010    || \
                            pix_fmt == AV_PIX_FMT_P016    || \
                            pix_fmt == AV_PIX_FMT_YUV444P16)

Definition at line 55 of file nvenc.c.

Referenced by nvenc_check_capabilities(), and nvenc_setup_hevc_config().

IS_YUV444

#define IS_YUV444

(

pix_fmt

)

Value:

(pix_fmt == AV_PIX_FMT_YUV444P || \
                            pix_fmt == AV_PIX_FMT_YUV444P16)

Definition at line 59 of file nvenc.c.

Referenced by nvenc_check_capabilities(), and nvenc_setup_hevc_config().

PRESET_ALIAS

#define PRESET_ALIAS	(		alias,
			name,
			...
	)		[PRESET_ ## alias] = { NV_ENC_PRESET_ ## name ## _GUID, __VA_ARGS__ }

Definition at line 601 of file nvenc.c.

Referenced by nvenc_map_preset().

PRESET

#define PRESET	(		name,
			...
	)		PRESET_ALIAS(name, name, __VA_ARGS__)

Definition at line 604 of file nvenc.c.

Referenced by nvenc_map_preset().

Function Documentation

nvenc_map_error()

static int nvenc_map_error ( NVENCSTATUS  err, const char **  desc  )

static

Definition at line 95 of file nvenc.c.

Referenced by nvenc_print_error().

nvenc_print_error()

static int nvenc_print_error ( void *  log_ctx, NVENCSTATUS  err, const char *  error_string  )

static

Definition at line 110 of file nvenc.c.

Referenced by ff_nvenc_send_frame(), nvenc_alloc_surface(), nvenc_find_free_reg_resource(), nvenc_load_libraries(), nvenc_open_session(), nvenc_register_frame(), nvenc_setup_encoder(), nvenc_setup_extradata(), nvenc_upload_frame(), and process_output_surface().

nvenc_print_driver_requirement()

static void nvenc_print_driver_requirement ( AVCodecContext *  avctx, int  level  )

static

Definition at line 120 of file nvenc.c.

Referenced by nvenc_load_libraries().

nvenc_load_libraries()

static av_cold int nvenc_load_libraries ( AVCodecContext *  avctx )

static

Definition at line 138 of file nvenc.c.

Referenced by ff_nvenc_encode_init().

nvenc_push_context()

static int nvenc_push_context ( AVCodecContext *  avctx )

static

Definition at line 182 of file nvenc.c.

Referenced by ff_nvenc_encode_close(), ff_nvenc_receive_packet(), ff_nvenc_send_frame(), nvenc_check_device(), nvenc_setup_encoder(), and nvenc_setup_surfaces().

nvenc_pop_context()

static int nvenc_pop_context ( AVCodecContext *  avctx )

static

Definition at line 200 of file nvenc.c.

Referenced by ff_nvenc_encode_close(), ff_nvenc_receive_packet(), ff_nvenc_send_frame(), nvenc_check_device(), nvenc_setup_encoder(), and nvenc_setup_surfaces().

nvenc_open_session()

static av_cold int nvenc_open_session ( AVCodecContext *  avctx )

static

Definition at line 219 of file nvenc.c.

Referenced by nvenc_check_device(), and nvenc_setup_device().

nvenc_check_codec_support()

static int nvenc_check_codec_support ( AVCodecContext *  avctx )

static

Definition at line 245 of file nvenc.c.

Referenced by nvenc_check_capabilities().

nvenc_check_cap()

static int nvenc_check_cap ( AVCodecContext *  avctx, NV_ENC_CAPS  cap  )

static

Definition at line 281 of file nvenc.c.

Referenced by nvenc_check_capabilities().

nvenc_check_capabilities()

static int nvenc_check_capabilities ( AVCodecContext *  avctx )

static

Definition at line 298 of file nvenc.c.

Referenced by nvenc_check_device(), and nvenc_setup_device().

nvenc_check_device()

static av_cold int nvenc_check_device ( AVCodecContext *  avctx, int  idx  )

static

Definition at line 400 of file nvenc.c.

Referenced by nvenc_setup_device().

nvenc_setup_device()

static av_cold int nvenc_setup_device ( AVCodecContext *  avctx )

static

Definition at line 485 of file nvenc.c.

Referenced by ff_nvenc_encode_init().

nvenc_map_preset()

static void nvenc_map_preset ( NvencContext *  ctx )

static

Definition at line 606 of file nvenc.c.

Referenced by nvenc_setup_encoder().

set_constqp()

static av_cold void set_constqp ( AVCodecContext *  avctx )

static

Definition at line 632 of file nvenc.c.

Referenced by nvenc_override_rate_control().

set_vbr()

static av_cold void set_vbr ( AVCodecContext *  avctx )

static

Definition at line 665 of file nvenc.c.

Referenced by nvenc_override_rate_control(), and nvenc_setup_rate_control().

set_lossless()

static av_cold void set_lossless ( AVCodecContext *  avctx )

static

Definition at line 727 of file nvenc.c.

Referenced by nvenc_setup_rate_control().

nvenc_override_rate_control()

static void nvenc_override_rate_control ( AVCodecContext *  avctx )

static

Definition at line 741 of file nvenc.c.

Referenced by nvenc_setup_rate_control().

nvenc_recalc_surfaces()

static av_cold int nvenc_recalc_surfaces ( AVCodecContext *  avctx )

static

Definition at line 772 of file nvenc.c.

Referenced by nvenc_setup_encoder().

nvenc_setup_rate_control()

static av_cold void nvenc_setup_rate_control ( AVCodecContext *  avctx )

static

Definition at line 811 of file nvenc.c.

Referenced by nvenc_setup_encoder().

nvenc_setup_h264_config()

static av_cold int nvenc_setup_h264_config ( AVCodecContext *  avctx )

static

Definition at line 930 of file nvenc.c.

Referenced by nvenc_setup_codec_config().

nvenc_setup_hevc_config()

static av_cold int nvenc_setup_hevc_config ( AVCodecContext *  avctx )

static

Definition at line 1022 of file nvenc.c.

Referenced by nvenc_setup_codec_config().

nvenc_setup_codec_config()

static av_cold int nvenc_setup_codec_config ( AVCodecContext *  avctx )

static

Definition at line 1102 of file nvenc.c.

Referenced by nvenc_setup_encoder().

nvenc_setup_encoder()

static av_cold int nvenc_setup_encoder ( AVCodecContext *  avctx )

static

Definition at line 1115 of file nvenc.c.

Referenced by ff_nvenc_encode_init().

nvenc_map_buffer_format()

static NV_ENC_BUFFER_FORMAT nvenc_map_buffer_format ( enum AVPixelFormat  pix_fmt )

static

Definition at line 1244 of file nvenc.c.

Referenced by nvenc_alloc_surface(), and nvenc_register_frame().

nvenc_alloc_surface()

static av_cold int nvenc_alloc_surface ( AVCodecContext *  avctx, int  idx  )

static

Definition at line 1267 of file nvenc.c.

Referenced by nvenc_setup_surfaces().

nvenc_setup_surfaces()

static av_cold int nvenc_setup_surfaces ( AVCodecContext *  avctx )

static

Definition at line 1324 of file nvenc.c.

Referenced by ff_nvenc_encode_init().

nvenc_setup_extradata()

static av_cold int nvenc_setup_extradata ( AVCodecContext *  avctx )

static

Definition at line 1365 of file nvenc.c.

Referenced by ff_nvenc_encode_init().

ff_nvenc_encode_close()

av_cold int ff_nvenc_encode_close

(

AVCodecContext *

avctx

)

Definition at line 1398 of file nvenc.c.

Referenced by nvenc_old_init().

ff_nvenc_encode_init()

av_cold int ff_nvenc_encode_init

(

AVCodecContext *

avctx

)

Definition at line 1473 of file nvenc.c.

Referenced by nvenc_old_init().

get_free_frame()

static NvencSurface* get_free_frame ( NvencContext *  ctx )

static

Definition at line 1516 of file nvenc.c.

Referenced by ff_nvenc_send_frame().

nvenc_copy_frame()

static int nvenc_copy_frame ( AVCodecContext *  avctx, NvencSurface *  nv_surface, NV_ENC_LOCK_INPUT_BUFFER *  lock_buffer_params, const AVFrame *  frame  )

static

Definition at line 1528 of file nvenc.c.

Referenced by nvenc_upload_frame().

nvenc_find_free_reg_resource()

static int nvenc_find_free_reg_resource ( AVCodecContext *  avctx )

static

Definition at line 1558 of file nvenc.c.

Referenced by nvenc_register_frame().

nvenc_register_frame()

static int nvenc_register_frame ( AVCodecContext *  avctx, const AVFrame *  frame  )

static

Definition at line 1588 of file nvenc.c.

Referenced by nvenc_upload_frame().

nvenc_upload_frame()

static int nvenc_upload_frame ( AVCodecContext *  avctx, const AVFrame *  frame, NvencSurface *  nvenc_frame  )

static

Definition at line 1642 of file nvenc.c.

Referenced by ff_nvenc_send_frame().

nvenc_codec_specific_pic_params()

static void nvenc_codec_specific_pic_params ( AVCodecContext *  avctx, NV_ENC_PIC_PARAMS *  params  )

static

Definition at line 1704 of file nvenc.c.

Referenced by ff_nvenc_send_frame().

timestamp_queue_enqueue()

static void timestamp_queue_enqueue ( AVFifoBuffer *  queue, int64_t  timestamp  )

inlinestatic

Definition at line 1725 of file nvenc.c.

Referenced by ff_nvenc_send_frame().

timestamp_queue_dequeue()

static int64_t timestamp_queue_dequeue ( AVFifoBuffer *  queue )

inlinestatic

Definition at line 1730 of file nvenc.c.

Referenced by nvenc_set_timestamp(), and process_output_surface().

nvenc_set_timestamp()

static int nvenc_set_timestamp ( AVCodecContext *  avctx, NV_ENC_LOCK_BITSTREAM *  params, AVPacket *  pkt  )

static

Definition at line 1739 of file nvenc.c.

Referenced by process_output_surface().

process_output_surface()

static int process_output_surface ( AVCodecContext *  avctx, AVPacket *  pkt, NvencSurface *  tmpoutsurf  )

static

Definition at line 1773 of file nvenc.c.

Referenced by ff_nvenc_receive_packet().

output_ready()

static int output_ready ( AVCodecContext *  avctx, int  flush  )

static

Definition at line 1905 of file nvenc.c.

Referenced by ff_nvenc_receive_packet().

ff_nvenc_send_frame()

int ff_nvenc_send_frame	(	AVCodecContext *	avctx,
		const AVFrame *	frame
	)

Definition at line 1923 of file nvenc.c.

Referenced by ff_nvenc_encode_frame(), and nvenc_old_init().

ff_nvenc_receive_packet()

int ff_nvenc_receive_packet	(	AVCodecContext *	avctx,
		AVPacket *	pkt
	)

Definition at line 2026 of file nvenc.c.

Referenced by ff_nvenc_encode_frame(), and nvenc_old_init().

ff_nvenc_encode_frame()

int ff_nvenc_encode_frame	(	AVCodecContext *	avctx,
		AVPacket *	pkt,
		const AVFrame *	frame,
		int *	got_packet
	)

Definition at line 2062 of file nvenc.c.

Referenced by nvenc_old_init().

Variable Documentation

ff_nvenc_pix_fmts

enum AVPixelFormat ff_nvenc_pix_fmts[]

Initial value:

= {
    AV_PIX_FMT_YUV420P,
    AV_PIX_FMT_NV12,
    AV_PIX_FMT_P010,
    AV_PIX_FMT_YUV444P,
    AV_PIX_FMT_P016,      
    AV_PIX_FMT_YUV444P16, 
    AV_PIX_FMT_0RGB32,
    AV_PIX_FMT_0BGR32,
    AV_PIX_FMT_CUDA,



    AV_PIX_FMT_NONE
}

Definition at line 39 of file nvenc.c.

Referenced by nvenc_old_init().

nverr

NVENCSTATUS nverr

Definition at line 63 of file nvenc.c.

Referenced by nvenc_map_error().

averr

int averr

Definition at line 64 of file nvenc.c.

desc

const char* desc

Definition at line 65 of file nvenc.c.

Referenced by asf_read_picture(), asf_write_header1(), av_bsf_init(), av_frame_apply_cropping(), av_image_alloc(), av_image_copy_to_buffer(), av_image_fill_black(), av_image_fill_linesizes(), av_image_fill_pointers(), av_image_get_buffer_size(), av_image_get_linesize(), av_picture_crop(), av_picture_pad(), av_pix_fmt_count_planes(), av_pix_fmt_get_chroma_sub_sample(), av_pix_fmt_swap_endianness(), avcodec_align_dimensions(), avcodec_align_dimensions2(), avcodec_default_get_format(), avcodec_descriptor_get_by_name(), avcodec_get_chroma_sub_sample(), avcodec_get_type(), avcodec_profile_name(), bufref_wrap_interface(), call_resize_kernel(), check_image_pointers(), choose_pixel_fmt(), config_input(), config_input0(), config_input1(), config_input_main(), config_input_ref(), config_inputx(), config_inputy(), config_out_props(), config_output(), config_props(), config_props_output(), copy_picture_field(), create_all_formats(), d3d11va_device_create(), d3d11va_get_buffer(), d3d11va_transfer_data(), decode_frame(), descriptor_compare(), drm_map_frame(), encode_frame(), encode_init(), ff_all_formats(), ff_color_frame(), ff_draw_color(), ff_draw_horiz_band(), ff_draw_init(), ff_flac_parse_picture(), ff_frame_pool_get(), ff_frame_pool_video_init(), ff_get_format(), ff_guess_coded_bitrate(), ff_h264_draw_horiz_band(), ff_id3v2_write_apic(), ff_mediacodec_dec_init(), ff_qsv_process_data(), ff_scale_eval_dimensions(), ff_sws_alphablendaway(), ff_sws_init_output_funcs(), ff_vmafmotion_init(), fill_frameinfo_by_link(), filter_frame(), find_codec_or_die(), flac_write_picture(), flush_encoders(), frame_end(), geq_config_props(), get_aspect_ratio(), get_bit_depth(), get_codecs_sorted(), get_deinterleaved_format(), get_format(), get_pix_fmt_depth(), get_sockaddr(), get_video_buffer(), have_alpha_planar(), hScale16To15_c(), hScale16To19_c(), hwdownload_query_formats(), hwmap_config_output(), image_copy(), image_copy_16_to_8(), init(), init_image(), init_muxer(), init_video_param(), init_video_param_jpeg(), kmsgrab_free_desc(), kmsgrab_read_packet(), libopenjpeg_copy_to_packed16(), libopenjpeg_copyto16(), libopenjpeg_decode_frame(), libopenjpeg_ispacked(), libopenjpeg_matches_pix_fmt(), list_formats(), main(), map_pixel_format(), mj2_create_image(), nvenc_print_error(), opencl_frames_uninit(), opencl_get_buffer(), opencl_get_plane_format(), opencl_map_frame(), opencl_mem_flags_for_mapping(), opencl_pool_alloc(), opencl_pool_free(), opengl_draw(), opengl_fill_color_map(), opengl_init_context(), pix_fmt_match(), print_codecs(), print_digraph(), qsv_decode_init(), qsv_init_surface(), qsv_map_from(), query_formats(), raw_decode(), raw_encode_init(), raw_init_decoder(), read_gab2_sub(), request_frame(), rgb48Toxyz12(), rkmpp_release_frame(), rkmpp_retrieve_frame(), rv10_write_header(), show_codecs(), show_help_codec(), show_help_muxer(), shuffleplanes_config_input(), smv_img_pnt(), smvjpeg_decode_frame(), swscale(), thumbnail_kernel(), tiff_unpack_strip(), transcode_init(), unsharp_opencl_make_filter_params(), unsupported_codec(), v4l2_read_header(), vaapi_map_to_memory(), validate_avframe_allocation(), verify_md5(), video_frame_cksum(), video_get_buffer(), wrap_texture_buf(), write_header(), write_packet(), xwd_encode_frame(), xyz12Torgb48(), yuv2gbrp16_full_X_c(), and yuv2gbrp_full_X_c().

nvenc_errors

const { ... } nvenc_errors[]

Initial value:

= {
    { NV_ENC_SUCCESS,                      0,                "success"                  },
    { NV_ENC_ERR_NO_ENCODE_DEVICE,         AVERROR(ENOENT),  "no encode device"         },
    { NV_ENC_ERR_UNSUPPORTED_DEVICE,       AVERROR(ENOSYS),  "unsupported device"       },
    { NV_ENC_ERR_INVALID_ENCODERDEVICE,    AVERROR(EINVAL),  "invalid encoder device"   },
    { NV_ENC_ERR_INVALID_DEVICE,           AVERROR(EINVAL),  "invalid device"           },
    { NV_ENC_ERR_DEVICE_NOT_EXIST,         AVERROR(EIO),     "device does not exist"    },
    { NV_ENC_ERR_INVALID_PTR,              AVERROR(EFAULT),  "invalid ptr"              },
    { NV_ENC_ERR_INVALID_EVENT,            AVERROR(EINVAL),  "invalid event"            },
    { NV_ENC_ERR_INVALID_PARAM,            AVERROR(EINVAL),  "invalid param"            },
    { NV_ENC_ERR_INVALID_CALL,             AVERROR(EINVAL),  "invalid call"             },
    { NV_ENC_ERR_OUT_OF_MEMORY,            AVERROR(ENOMEM),  "out of memory"            },
    { NV_ENC_ERR_ENCODER_NOT_INITIALIZED,  AVERROR(EINVAL),  "encoder not initialized"  },
    { NV_ENC_ERR_UNSUPPORTED_PARAM,        AVERROR(ENOSYS),  "unsupported param"        },
    { NV_ENC_ERR_LOCK_BUSY,                AVERROR(EAGAIN),  "lock busy"                },
    { NV_ENC_ERR_NOT_ENOUGH_BUFFER,        AVERROR_BUFFER_TOO_SMALL, "not enough buffer"},
    { NV_ENC_ERR_INVALID_VERSION,          AVERROR(EINVAL),  "invalid version"          },
    { NV_ENC_ERR_MAP_FAILED,               AVERROR(EIO),     "map failed"               },
    { NV_ENC_ERR_NEED_MORE_INPUT,          AVERROR(EAGAIN),  "need more input"          },
    { NV_ENC_ERR_ENCODER_BUSY,             AVERROR(EAGAIN),  "encoder busy"             },
    { NV_ENC_ERR_EVENT_NOT_REGISTERD,      AVERROR(EBADF),   "event not registered"     },
    { NV_ENC_ERR_GENERIC,                  AVERROR_UNKNOWN,  "generic error"            },
    { NV_ENC_ERR_INCOMPATIBLE_CLIENT_KEY,  AVERROR(EINVAL),  "incompatible client key"  },
    { NV_ENC_ERR_UNIMPLEMENTED,            AVERROR(ENOSYS),  "unimplemented"            },
    { NV_ENC_ERR_RESOURCE_REGISTER_FAILED, AVERROR(EIO),     "resource register failed" },
    { NV_ENC_ERR_RESOURCE_NOT_REGISTERED,  AVERROR(EBADF),   "resource not registered"  },
    { NV_ENC_ERR_RESOURCE_NOT_MAPPED,      AVERROR(EBADF),   "resource not mapped"      },
}

Referenced by nvenc_map_error().