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 <dlfcn.h>
#include "libavutil/hwcontext.h"
#include "libavutil/imgutils.h"
#include "libavutil/avassert.h"
#include "libavutil/mem.h"
#include "internal.h"
#include "nvenc.h"

Go to the source code of this file.

Data Structures

struct  GUIDTuple
 

Macros

#define CUDA_LIBNAME   "libcuda.so"
 
#define NVENC_LIBNAME   "libnvidia-encode.so"
 
#define NVENC_CAP   0x30
 
#define IS_CBR(rc)
 
#define LOAD_LIBRARY(l, path)
 
#define LOAD_SYMBOL(fun, lib, symbol)
 

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 av_cold int nvenc_load_libraries (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 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 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 *inSurf, NV_ENC_LOCK_INPUT_BUFFER *lockBufferParams, 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_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 CUDA_LIBNAME   "libcuda.so"

Definition at line 40 of file nvenc.c.

Referenced by nvenc_load_libraries().

#define NVENC_LIBNAME   "libnvidia-encode.so"

Definition at line 41 of file nvenc.c.

Referenced by nvenc_load_libraries().

#define NVENC_CAP   0x30

Definition at line 51 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_2_PASS_QUALITY || \
rc == NV_ENC_PARAMS_RC_2_PASS_FRAMESIZE_CAP)

Definition at line 52 of file nvenc.c.

Referenced by nvenc_setup_h264_config(), and nvenc_setup_hevc_config().

#define LOAD_LIBRARY (   l,
  path 
)
Value:
do { \
if (!((l) = dlopen(path, RTLD_LAZY))) { \
"Cannot load %s\n", \
path); \
} \
} while (0)
#define av_log(a,...)
#define AV_LOG_ERROR
Something went wrong and cannot losslessly be recovered.
Definition: log.h:176
return
if(ret< 0)
Definition: vf_mcdeint.c:282
#define AVERROR_UNKNOWN
Unknown error, typically from an external library.
Definition: error.h:71

Definition at line 56 of file nvenc.c.

Referenced by nvenc_load_libraries().

#define LOAD_SYMBOL (   fun,
  lib,
  symbol 
)
Value:
do { \
if (!((fun) = dlsym(lib, symbol))) { \
"Cannot load %s\n", \
symbol); \
} \
} while (0)
#define av_log(a,...)
#define AV_LOG_ERROR
Something went wrong and cannot losslessly be recovered.
Definition: log.h:176
return
if(ret< 0)
Definition: vf_mcdeint.c:282
#define AVERROR_UNKNOWN
Unknown error, typically from an external library.
Definition: error.h:71

Definition at line 66 of file nvenc.c.

Referenced by nvenc_load_libraries().

Function Documentation

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

Definition at line 119 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 av_cold int nvenc_load_libraries ( AVCodecContext avctx)
static

Definition at line 144 of file nvenc.c.

Referenced by ff_nvenc_encode_init().

static av_cold int nvenc_open_session ( AVCodecContext avctx)
static

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

Referenced by nvenc_check_capabilities().

static int nvenc_check_cap ( AVCodecContext avctx,
NV_ENC_CAPS  cap 
)
static

Definition at line 247 of file nvenc.c.

Referenced by nvenc_check_capabilities().

static int nvenc_check_capabilities ( AVCodecContext avctx)
static

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

Referenced by nvenc_setup_device().

static av_cold int nvenc_setup_device ( AVCodecContext avctx)
static

Definition at line 396 of file nvenc.c.

Referenced by ff_nvenc_encode_init().

static void nvenc_map_preset ( NvencContext ctx)
static

Definition at line 486 of file nvenc.c.

Referenced by nvenc_setup_encoder().

static av_cold void set_constqp ( AVCodecContext avctx)
static

Definition at line 509 of file nvenc.c.

Referenced by nvenc_override_rate_control().

static av_cold void set_vbr ( AVCodecContext avctx)
static

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

Referenced by nvenc_setup_rate_control().

static void nvenc_override_rate_control ( AVCodecContext avctx)
static

Definition at line 582 of file nvenc.c.

Referenced by nvenc_setup_rate_control().

static av_cold void nvenc_setup_rate_control ( AVCodecContext avctx)
static

Definition at line 625 of file nvenc.c.

Referenced by nvenc_setup_encoder().

static av_cold int nvenc_setup_h264_config ( AVCodecContext avctx)
static

Definition at line 678 of file nvenc.c.

Referenced by nvenc_setup_codec_config().

static av_cold int nvenc_setup_hevc_config ( AVCodecContext avctx)
static

Definition at line 762 of file nvenc.c.

Referenced by nvenc_setup_codec_config().

static av_cold int nvenc_setup_codec_config ( AVCodecContext avctx)
static

Definition at line 814 of file nvenc.c.

Referenced by nvenc_setup_encoder().

static av_cold int nvenc_setup_encoder ( AVCodecContext avctx)
static

Definition at line 827 of file nvenc.c.

Referenced by ff_nvenc_encode_init().

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

Definition at line 942 of file nvenc.c.

Referenced by nvenc_setup_surfaces().

static av_cold int nvenc_setup_surfaces ( AVCodecContext avctx)
static

Definition at line 1015 of file nvenc.c.

Referenced by ff_nvenc_encode_init().

static av_cold int nvenc_setup_extradata ( AVCodecContext avctx)
static

Definition at line 1047 of file nvenc.c.

Referenced by ff_nvenc_encode_init().

av_cold int ff_nvenc_encode_close ( AVCodecContext avctx)

Definition at line 1080 of file nvenc.c.

av_cold int ff_nvenc_encode_init ( AVCodecContext avctx)

Definition at line 1157 of file nvenc.c.

Referenced by nvenc_old_init().

static NvencSurface* get_free_frame ( NvencContext ctx)
static

Definition at line 1195 of file nvenc.c.

Referenced by ff_nvenc_encode_frame().

static int nvenc_copy_frame ( AVCodecContext avctx,
NvencSurface inSurf,
NV_ENC_LOCK_INPUT_BUFFER *  lockBufferParams,
const AVFrame frame 
)
static

Definition at line 1209 of file nvenc.c.

Referenced by nvenc_upload_frame().

static int nvenc_find_free_reg_resource ( AVCodecContext avctx)
static

Definition at line 1265 of file nvenc.c.

Referenced by nvenc_register_frame().

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

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

Referenced by ff_nvenc_encode_frame().

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

Definition at line 1385 of file nvenc.c.

Referenced by ff_nvenc_encode_frame().

static void timestamp_queue_enqueue ( AVFifoBuffer queue,
int64_t  timestamp 
)
inlinestatic

Definition at line 1406 of file nvenc.c.

Referenced by ff_nvenc_encode_frame().

static int64_t timestamp_queue_dequeue ( AVFifoBuffer queue)
inlinestatic

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

Referenced by process_output_surface().

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

Definition at line 1454 of file nvenc.c.

Referenced by ff_nvenc_encode_frame().

static int output_ready ( AVCodecContext avctx,
int  flush 
)
static

Definition at line 1565 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 1583 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:67
planar YUV 4:2:0, 12bpp, 1 plane for Y and 1 plane for the UV components, which are interleaved (firs...
Definition: pixfmt.h:90
planar YUV 4:2:0, 12bpp, (1 Cr & Cb sample per 2x2 Y samples)
Definition: pixfmt.h:62

Definition at line 76 of file nvenc.c.

NVENCSTATUS nverr

Definition at line 87 of file nvenc.c.

Referenced by nvenc_map_error().

int averr

Definition at line 88 of file nvenc.c.

const char* desc

Definition at line 89 of file nvenc.c.

Referenced by asf_read_picture(), asf_write_header1(), av_bsf_init(), av_image_alloc(), av_image_copy(), av_image_copy_to_buffer(), 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_descriptor_get_by_name(), avcodec_get_chroma_sub_sample(), avcodec_get_type(), avcodec_profile_name(), check_image_pointers(), choose_pixel_fmt(), config_input(), config_input_ref(), config_out_props(), config_output(), config_props(), copy_picture_field(), create_all_formats(), decode_frame(), dxva2_create_decoder(), 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_get_format(), ff_h264_draw_horiz_band(), ff_hevc_output_frame(), ff_id3v2_write_apic(), ff_mediacodec_dec_init(), ff_qsv_init_internal_session(), ff_sws_alphablendaway(), ff_sws_init_output_funcs(), ff_video_frame_pool_get(), ff_video_frame_pool_init(), filter_frame(), find_codec_or_die(), 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(), hScale16To15_c(), hScale16To19_c(), hwdownload_query_formats(), init(), init_image(), init_muxer(), is16BPS(), is9_OR_10BPS(), is_hwaccel_pix_fmt(), isALPHA(), isBE(), isPackedRGB(), isPlanar(), isPlanarRGB(), isPlanarYUV(), isRGB(), isYUV(), 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(), opengl_draw(), opengl_fill_color_map(), opengl_init_context(), output_frame(), pix_fmt_match(), print_codecs(), print_digraph(), query_formats(), raw_decode(), raw_encode_init(), raw_init_decoder(), read_gab2_sub(), rgb48Toxyz12(), rv10_write_header(), show_codecs(), show_help_codec(), show_help_muxer(), shuffleplanes_config_input(), smv_img_pnt(), smvjpeg_decode_frame(), swscale(), tiff_unpack_strip(), unsupported_codec(), usePal(), validate_avframe_allocation(), verify_md5(), video_frame_cksum(), video_get_buffer(), write_header(), write_packet(), xwd_encode_frame(), xyz12Torgb48(), 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(ENOBUFS), "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_UNKNOWN
Unknown error, typically from an external library.
Definition: error.h:71

Referenced by nvenc_map_error().