FFmpeg
Data Structures | Macros | Functions | Variables
nvenc.c File Reference
#include "config.h"
#include "nvenc.h"
#include "hevc_sei.h"
#include "libavutil/hwcontext_cuda.h"
#include "libavutil/hwcontext.h"
#include "libavutil/cuda_check.h"
#include "libavutil/imgutils.h"
#include "libavutil/mem.h"
#include "libavutil/pixdesc.h"
#include "atsc_a53.h"
#include "encode.h"
#include "internal.h"
#include "packet_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 IS_GBRP(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 (AVCodecContext *avctx, NVENCSTATUS err, const char *error_string)
 
static void nvenc_map_preset (NvencContext *ctx)
 
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 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, int sei_count)
 
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 int prepare_sei_data_array (AVCodecContext *avctx, const AVFrame *frame)
 
static void reconfig_encoder (AVCodecContext *avctx, const AVFrame *frame)
 
static int nvenc_send_frame (AVCodecContext *avctx, const AVFrame *frame)
 
int ff_nvenc_receive_packet (AVCodecContext *avctx, AVPacket *pkt)
 
av_cold void ff_nvenc_encode_flush (AVCodecContext *avctx)
 

Variables

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

Macro Definition Documentation

◆ CHECK_CU

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

Definition at line 38 of file nvenc.c.

◆ NVENC_CAP

#define NVENC_CAP   0x30

Definition at line 40 of file nvenc.c.

◆ 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 41 of file nvenc.c.

◆ IS_10BIT

#define IS_10BIT (   pix_fmt)
Value:
pix_fmt == AV_PIX_FMT_P016 || \
pix_fmt == AV_PIX_FMT_YUV444P16 || \
pix_fmt == AV_PIX_FMT_GBRP16)

Definition at line 73 of file nvenc.c.

◆ IS_YUV444

#define IS_YUV444 (   pix_fmt)
Value:
pix_fmt == AV_PIX_FMT_YUV444P16 || \
pix_fmt == AV_PIX_FMT_GBRP || \
pix_fmt == AV_PIX_FMT_GBRP16)

Definition at line 78 of file nvenc.c.

◆ IS_GBRP

#define IS_GBRP (   pix_fmt)
Value:

Definition at line 83 of file nvenc.c.

◆ PRESET_ALIAS

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

Definition at line 159 of file nvenc.c.

◆ PRESET

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

Definition at line 162 of file nvenc.c.

Function Documentation

◆ nvenc_map_error()

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

Definition at line 119 of file nvenc.c.

Referenced by nvenc_print_error().

◆ nvenc_print_error()

static int nvenc_print_error ( AVCodecContext avctx,
NVENCSTATUS  err,
const char *  error_string 
)
static

◆ nvenc_map_preset()

static void nvenc_map_preset ( NvencContext ctx)
static

Definition at line 164 of file nvenc.c.

Referenced by nvenc_setup_device().

◆ nvenc_print_driver_requirement()

static void nvenc_print_driver_requirement ( AVCodecContext avctx,
int  level 
)
static

Definition at line 218 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 274 of file nvenc.c.

Referenced by ff_nvenc_encode_init().

◆ nvenc_push_context()

static int nvenc_push_context ( AVCodecContext avctx)
static

◆ nvenc_pop_context()

static int nvenc_pop_context ( AVCodecContext avctx)
static

◆ nvenc_open_session()

static av_cold int nvenc_open_session ( AVCodecContext avctx)
static

Definition at line 341 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 367 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 403 of file nvenc.c.

Referenced by nvenc_check_capabilities().

◆ nvenc_check_capabilities()

static int nvenc_check_capabilities ( AVCodecContext avctx)
static

Definition at line 420 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 569 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 644 of file nvenc.c.

Referenced by ff_nvenc_encode_init().

◆ set_constqp()

static av_cold void set_constqp ( AVCodecContext avctx)
static

Definition at line 755 of file nvenc.c.

Referenced by nvenc_override_rate_control().

◆ set_vbr()

static av_cold void set_vbr ( AVCodecContext avctx)
static

Definition at line 788 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 850 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 864 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 895 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 934 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 1087 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 1199 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 1305 of file nvenc.c.

Referenced by nvenc_setup_encoder().

◆ compute_dar()

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

Definition at line 1318 of file nvenc.c.

Referenced by nvenc_setup_encoder(), and reconfig_encoder().

◆ nvenc_setup_encoder()

static av_cold int nvenc_setup_encoder ( AVCodecContext avctx)
static

Definition at line 1332 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 1495 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 1520 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 1576 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 1617 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 1650 of file nvenc.c.

◆ ff_nvenc_encode_init()

av_cold int ff_nvenc_encode_init ( AVCodecContext avctx)

Definition at line 1729 of file nvenc.c.

◆ get_free_frame()

static NvencSurface* get_free_frame ( NvencContext ctx)
static

Definition at line 1776 of file nvenc.c.

Referenced by 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 1788 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 1818 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 1852 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 1906 of file nvenc.c.

Referenced by nvenc_send_frame().

◆ nvenc_codec_specific_pic_params()

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

Definition at line 1968 of file nvenc.c.

Referenced by nvenc_send_frame().

◆ timestamp_queue_enqueue()

static void timestamp_queue_enqueue ( AVFifoBuffer queue,
int64_t  timestamp 
)
inlinestatic

Definition at line 2001 of file nvenc.c.

Referenced by nvenc_send_frame().

◆ timestamp_queue_dequeue()

static int64_t timestamp_queue_dequeue ( AVFifoBuffer queue)
inlinestatic

Definition at line 2006 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 2015 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 2029 of file nvenc.c.

Referenced by ff_nvenc_receive_packet().

◆ output_ready()

static int output_ready ( AVCodecContext avctx,
int  flush 
)
static

Definition at line 2150 of file nvenc.c.

Referenced by ff_nvenc_receive_packet().

◆ prepare_sei_data_array()

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

Definition at line 2162 of file nvenc.c.

Referenced by nvenc_send_frame().

◆ reconfig_encoder()

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

Definition at line 2257 of file nvenc.c.

Referenced by nvenc_send_frame().

◆ nvenc_send_frame()

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

Definition at line 2349 of file nvenc.c.

Referenced by ff_nvenc_encode_flush(), and ff_nvenc_receive_packet().

◆ ff_nvenc_receive_packet()

int ff_nvenc_receive_packet ( AVCodecContext avctx,
AVPacket pkt 
)

Definition at line 2457 of file nvenc.c.

◆ ff_nvenc_encode_flush()

av_cold void ff_nvenc_encode_flush ( AVCodecContext avctx)

Definition at line 2508 of file nvenc.c.

Variable Documentation

◆ ff_nvenc_pix_fmts

enum AVPixelFormat ff_nvenc_pix_fmts[]

◆ ff_nvenc_hw_configs

const AVCodecHWConfigInternal* const ff_nvenc_hw_configs[]
Initial value:

Definition at line 63 of file nvenc.c.

◆ nverr

NVENCSTATUS nverr

Definition at line 87 of file nvenc.c.

Referenced by nvenc_map_error().

◆ averr

int averr

Definition at line 88 of file nvenc.c.

Referenced by print_tls_error().

◆ desc

const char* desc

Definition at line 89 of file nvenc.c.

Referenced by nvenc_map_error(), and nvenc_print_error().

◆ 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().

AV_PIX_FMT_CUDA
@ AV_PIX_FMT_CUDA
HW acceleration through CUDA.
Definition: pixfmt.h:225
AVERROR
Filter the word “frame” indicates either a video frame or a group of audio as stored in an AVFrame structure Format for each input and each output the list of supported formats For video that means pixel format For audio that means channel sample they are references to shared objects When the negotiation mechanism computes the intersection of the formats supported at each end of a all references to both lists are replaced with a reference to the intersection And when a single format is eventually chosen for a link amongst the remaining all references to the list are updated That means that if a filter requires that its input and output have the same format amongst a supported all it has to do is use a reference to the same list of formats query_formats can leave some formats unset and return AVERROR(EAGAIN) to cause the negotiation mechanism toagain later. That can be used by filters with complex requirements to use the format negotiated on one link to set the formats supported on another. Frame references ownership and permissions
NONE
@ NONE
Definition: af_afade.c:54
AVERROR_UNKNOWN
#define AVERROR_UNKNOWN
Unknown error, typically from an external library.
Definition: error.h:73
AVERROR_BUFFER_TOO_SMALL
#define AVERROR_BUFFER_TOO_SMALL
Buffer too small.
Definition: error.h:53
AV_PIX_FMT_YUV444P16
#define AV_PIX_FMT_YUV444P16
Definition: pixfmt.h:402
AV_PIX_FMT_0BGR32
#define AV_PIX_FMT_0BGR32
Definition: pixfmt.h:367
pix_fmt
static enum AVPixelFormat pix_fmt
Definition: demuxing_decoding.c:41
AV_PIX_FMT_YUV420P
@ AV_PIX_FMT_YUV420P
planar YUV 4:2:0, 12bpp, (1 Cr & Cb sample per 2x2 Y samples)
Definition: pixfmt.h:66
AV_PIX_FMT_GBRP16
#define AV_PIX_FMT_GBRP16
Definition: pixfmt.h:408
NULL
#define NULL
Definition: coverity.c:32
HW_CONFIG_ENCODER_DEVICE
#define HW_CONFIG_ENCODER_DEVICE(format, device_type_)
Definition: hwconfig.h:96
HW_CONFIG_ENCODER_FRAMES
#define HW_CONFIG_ENCODER_FRAMES(format, device_type_)
Definition: hwconfig.h:99
AV_PIX_FMT_P016
#define AV_PIX_FMT_P016
Definition: pixfmt.h:439
AV_PIX_FMT_NV12
@ AV_PIX_FMT_NV12
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
AV_PIX_FMT_0RGB32
#define AV_PIX_FMT_0RGB32
Definition: pixfmt.h:366
AV_PIX_FMT_NONE
@ AV_PIX_FMT_NONE
Definition: pixfmt.h:65
AV_PIX_FMT_YUV444P
@ AV_PIX_FMT_YUV444P
planar YUV 4:4:4, 24bpp, (1 Cr & Cb sample per 1x1 Y samples)
Definition: pixfmt.h:71
AV_PIX_FMT_P010
#define AV_PIX_FMT_P010
Definition: pixfmt.h:438
AV_PIX_FMT_GBRP
@ AV_PIX_FMT_GBRP
planar GBR 4:4:4 24bpp
Definition: pixfmt.h:158