FFmpeg
 All Data Structures Namespaces Files Functions Variables Typedefs Enumerations Enumerator Macros Groups Pages
Data Structures | Macros | Functions | Variables
nvenc.c File Reference
#include "config.h"
#include "nvenc.h"
#include "libavutil/hwcontext_cuda.h"
#include "libavutil/hwcontext.h"
#include "libavutil/cuda_check.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 CHECK_CU(x)   FF_CUDA_CHECK_DL(avctx, dl_fn->cuda_dl, x)
 
#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 void compute_dar (AVCodecContext *avctx, int *dw, int *dh)
 
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 NvencSurfaceget_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, NV_ENC_SEI_PAYLOAD *sei_data)
 
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)
 
static void reconfig_encoder (AVCodecContext *avctx, const AVFrame *frame)
 
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

#define CHECK_CU (   x)    FF_CUDA_CHECK_DL(avctx, dl_fn->cuda_dl, x)
#define NVENC_CAP   0x30

Definition at line 37 of file nvenc.c.

Referenced by nvenc_check_device().

#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 38 of file nvenc.c.

Referenced by nvenc_setup_h264_config(), and nvenc_setup_hevc_config().

#define IS_10BIT (   pix_fmt)
Value:
static enum AVPixelFormat pix_fmt
#define AV_PIX_FMT_P016
Definition: pixfmt.h:427
#define AV_PIX_FMT_P010
Definition: pixfmt.h:426
#define AV_PIX_FMT_YUV444P16
Definition: pixfmt.h:392

Definition at line 58 of file nvenc.c.

Referenced by nvenc_check_capabilities(), and nvenc_setup_hevc_config().

#define IS_YUV444 (   pix_fmt)
Value:
static enum AVPixelFormat pix_fmt
planar YUV 4:4:4, 24bpp, (1 Cr & Cb sample per 1x1 Y samples)
Definition: pixfmt.h:71
#define AV_PIX_FMT_YUV444P16
Definition: pixfmt.h:392

Definition at line 62 of file nvenc.c.

Referenced by nvenc_check_capabilities(), and nvenc_setup_hevc_config().

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

Definition at line 575 of file nvenc.c.

Referenced by nvenc_map_preset().

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

Definition at line 578 of file nvenc.c.

Referenced by nvenc_map_preset().

Function Documentation

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

Definition at line 98 of file nvenc.c.

Referenced by nvenc_print_error().

static int nvenc_print_error ( void log_ctx,
NVENCSTATUS  err,
const char *  error_string 
)
static
static void nvenc_print_driver_requirement ( AVCodecContext avctx,
int  level 
)
static

Definition at line 123 of file nvenc.c.

Referenced by nvenc_load_libraries().

static av_cold int nvenc_load_libraries ( AVCodecContext avctx)
static

Definition at line 141 of file nvenc.c.

Referenced by ff_nvenc_encode_init().

static int nvenc_push_context ( AVCodecContext avctx)
static
static int nvenc_pop_context ( AVCodecContext avctx)
static
static av_cold int nvenc_open_session ( AVCodecContext avctx)
static

Definition at line 208 of file nvenc.c.

Referenced by nvenc_check_device(), and nvenc_setup_device().

static int nvenc_check_codec_support ( AVCodecContext avctx)
static

Definition at line 234 of file nvenc.c.

Referenced by nvenc_check_capabilities().

static int nvenc_check_cap ( AVCodecContext avctx,
NV_ENC_CAPS  cap 
)
static

Definition at line 270 of file nvenc.c.

Referenced by nvenc_check_capabilities().

static int nvenc_check_capabilities ( AVCodecContext avctx)
static

Definition at line 287 of file nvenc.c.

Referenced by nvenc_check_device(), and nvenc_setup_device().

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

Definition at line 391 of file nvenc.c.

Referenced by nvenc_setup_device().

static av_cold int nvenc_setup_device ( AVCodecContext avctx)
static

Definition at line 465 of file nvenc.c.

Referenced by ff_nvenc_encode_init().

static void nvenc_map_preset ( NvencContext ctx)
static

Definition at line 580 of file nvenc.c.

Referenced by nvenc_setup_encoder().

static av_cold void set_constqp ( AVCodecContext avctx)
static

Definition at line 606 of file nvenc.c.

Referenced by nvenc_override_rate_control().

static av_cold void set_vbr ( AVCodecContext avctx)
static

Definition at line 639 of file nvenc.c.

Referenced by nvenc_override_rate_control(), and nvenc_setup_rate_control().

static av_cold void set_lossless ( AVCodecContext avctx)
static

Definition at line 701 of file nvenc.c.

Referenced by nvenc_setup_rate_control().

static void nvenc_override_rate_control ( AVCodecContext avctx)
static

Definition at line 715 of file nvenc.c.

Referenced by nvenc_setup_rate_control().

static av_cold int nvenc_recalc_surfaces ( AVCodecContext avctx)
static

Definition at line 746 of file nvenc.c.

Referenced by nvenc_setup_encoder().

static av_cold void nvenc_setup_rate_control ( AVCodecContext avctx)
static

Definition at line 785 of file nvenc.c.

Referenced by nvenc_setup_encoder().

static av_cold int nvenc_setup_h264_config ( AVCodecContext avctx)
static

Definition at line 904 of file nvenc.c.

Referenced by nvenc_setup_codec_config().

static av_cold int nvenc_setup_hevc_config ( AVCodecContext avctx)
static

Definition at line 996 of file nvenc.c.

Referenced by nvenc_setup_codec_config().

static av_cold int nvenc_setup_codec_config ( AVCodecContext avctx)
static

Definition at line 1076 of file nvenc.c.

Referenced by nvenc_setup_encoder().

static void compute_dar ( AVCodecContext avctx,
int dw,
int dh 
)
static

Definition at line 1089 of file nvenc.c.

Referenced by nvenc_setup_encoder(), and reconfig_encoder().

static av_cold int nvenc_setup_encoder ( AVCodecContext avctx)
static

Definition at line 1103 of file nvenc.c.

Referenced by ff_nvenc_encode_init().

static NV_ENC_BUFFER_FORMAT nvenc_map_buffer_format ( enum AVPixelFormat  pix_fmt)
static

Definition at line 1226 of file nvenc.c.

Referenced by nvenc_alloc_surface(), and nvenc_register_frame().

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

Definition at line 1249 of file nvenc.c.

Referenced by nvenc_setup_surfaces().

static av_cold int nvenc_setup_surfaces ( AVCodecContext avctx)
static

Definition at line 1306 of file nvenc.c.

Referenced by ff_nvenc_encode_init().

static av_cold int nvenc_setup_extradata ( AVCodecContext avctx)
static

Definition at line 1347 of file nvenc.c.

Referenced by ff_nvenc_encode_init().

av_cold int ff_nvenc_encode_close ( AVCodecContext avctx)

Definition at line 1380 of file nvenc.c.

av_cold int ff_nvenc_encode_init ( AVCodecContext avctx)

Definition at line 1455 of file nvenc.c.

Referenced by nvenc_old_init().

static NvencSurface* get_free_frame ( NvencContext ctx)
static

Definition at line 1498 of file nvenc.c.

Referenced by ff_nvenc_send_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 1510 of file nvenc.c.

Referenced by nvenc_upload_frame().

static int nvenc_find_free_reg_resource ( AVCodecContext avctx)
static

Definition at line 1540 of file nvenc.c.

Referenced by nvenc_register_frame().

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

Definition at line 1570 of file nvenc.c.

Referenced by nvenc_upload_frame().

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

Definition at line 1624 of file nvenc.c.

Referenced by ff_nvenc_send_frame().

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

Definition at line 1686 of file nvenc.c.

Referenced by ff_nvenc_send_frame().

static void timestamp_queue_enqueue ( AVFifoBuffer queue,
int64_t  timestamp 
)
inlinestatic

Definition at line 1718 of file nvenc.c.

Referenced by ff_nvenc_send_frame().

static int64_t timestamp_queue_dequeue ( AVFifoBuffer queue)
inlinestatic

Definition at line 1723 of file nvenc.c.

Referenced by nvenc_set_timestamp(), and process_output_surface().

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

Definition at line 1732 of file nvenc.c.

Referenced by process_output_surface().

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

Definition at line 1766 of file nvenc.c.

Referenced by ff_nvenc_receive_packet().

static int output_ready ( AVCodecContext avctx,
int  flush 
)
static

Definition at line 1898 of file nvenc.c.

Referenced by ff_nvenc_receive_packet().

static void reconfig_encoder ( AVCodecContext avctx,
const AVFrame frame 
)
static

Definition at line 1916 of file nvenc.c.

Referenced by ff_nvenc_send_frame().

int ff_nvenc_send_frame ( AVCodecContext avctx,
const AVFrame frame 
)

Definition at line 2008 of file nvenc.c.

Referenced by ff_nvenc_encode_frame().

int ff_nvenc_receive_packet ( AVCodecContext avctx,
AVPacket pkt 
)

Definition at line 2136 of file nvenc.c.

Referenced by ff_nvenc_encode_frame().

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

Definition at line 2172 of file nvenc.c.

Variable Documentation

enum AVPixelFormat ff_nvenc_pix_fmts[]
Initial value:
= {
}
planar YUV 4:4:4, 24bpp, (1 Cr & Cb sample per 1x1 Y samples)
Definition: pixfmt.h:71
#define AV_PIX_FMT_P016
Definition: pixfmt.h:427
#define AV_PIX_FMT_P010
Definition: pixfmt.h:426
#define AV_PIX_FMT_YUV444P16
Definition: pixfmt.h:392
planar YUV 4:2:0, 12bpp, 1 plane for Y and 1 plane for the UV components, which are interleaved (firs...
Definition: pixfmt.h:89
#define AV_PIX_FMT_0BGR32
Definition: pixfmt.h:357
HW acceleration through CUDA.
Definition: pixfmt.h:235
planar YUV 4:2:0, 12bpp, (1 Cr & Cb sample per 2x2 Y samples)
Definition: pixfmt.h:66
#define AV_PIX_FMT_0RGB32
Definition: pixfmt.h:356

Definition at line 42 of file nvenc.c.

NVENCSTATUS nverr

Definition at line 66 of file nvenc.c.

Referenced by nvenc_map_error().

int averr

Definition at line 67 of file nvenc.c.

const char* desc

Definition at line 68 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(), 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(), decode_frame(), descriptor_compare(), drm_map_frame(), encode_frame(), encode_init(), ff_all_formats(), ff_boxblur_eval_filter_params(), 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_video_buffer(), h264_metadata_update_sps(), 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(), is16BPS(), isALPHA(), isAnyRGB(), isBayer(), isBE(), isFloat(), isGray(), isNBPS(), isPacked(), isPackedRGB(), isPlanar(), isPlanarRGB(), isPlanarYUV(), isRGB(), isSemiPlanarYUV(), isYUV(), kmsgrab_free_desc(), kmsgrab_read_packet(), libopenjpeg_copy_to_packed16(), libopenjpeg_copyto16(), libopenjpeg_decode_frame(), libopenjpeg_ispacked(), libopenjpeg_matches_pix_fmt(), list_formats(), lut2_config_output(), main(), map_pixel_format(), mj2_create_image(), nvenc_print_error(), opencl_frames_uninit(), opencl_get_buffer(), opencl_get_plane_format(), 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(), read_packet_vs(), 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_encode_h265_init_sequence_params(), vaapi_encode_mjpeg_init_picture_params(), vaapi_encode_profile_entrypoint(), vaapi_frames_init(), vaapi_map_frame(), vaapi_pix_fmt_from_fourcc(), 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().

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" },
}
#define AVERROR(e)
Definition: error.h:43
#define AVERROR_BUFFER_TOO_SMALL
Buffer too small.
Definition: error.h:51
#define AVERROR_UNKNOWN
Unknown error, typically from an external library.
Definition: error.h:71

Referenced by nvenc_map_error().