/* * This file is part of FFmpeg. * * FFmpeg is free software; you can redistribute it and/or * modify it under the terms of the GNU Lesser General Public * License as published by the Free Software Foundation; either * version 2.1 of the License, or (at your option) any later version. * * FFmpeg is distributed in the hope that it will be useful, * but WITHOUT ANY WARRANTY; without even the implied warranty of * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU * Lesser General Public License for more details. * * You should have received a copy of the GNU Lesser General Public * License along with FFmpeg; if not, write to the Free Software * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA */ #ifndef AVCODEC_NVENC_H #define AVCODEC_NVENC_H #include "config.h" #if CONFIG_D3D11VA #define COBJMACROS #include "libavutil/hwcontext_d3d11va.h" #else typedef void ID3D11Device; #endif #include <ffnvcodec/nvEncodeAPI.h> #include "compat/cuda/dynlink_loader.h" #include "libavutil/fifo.h" #include "libavutil/opt.h" #include "hwconfig.h" #include "avcodec.h" #define MAX_REGISTERED_FRAMES 64 #define RC_MODE_DEPRECATED 0x800000 #define RCD(rc_mode) ((rc_mode) | RC_MODE_DEPRECATED) #define NVENCAPI_CHECK_VERSION(major, minor) \ ((major) < NVENCAPI_MAJOR_VERSION || ((major) == NVENCAPI_MAJOR_VERSION && (minor) <= NVENCAPI_MINOR_VERSION)) // SDK 8.1 compile time feature checks #if NVENCAPI_CHECK_VERSION(8, 1) #define NVENC_HAVE_BFRAME_REF_MODE #define NVENC_HAVE_QP_MAP_MODE #endif // SDK 9.0 compile time feature checks #if NVENCAPI_CHECK_VERSION(9, 0) #define NVENC_HAVE_HEVC_BFRAME_REF_MODE #endif // SDK 9.1 compile time feature checks #if NVENCAPI_CHECK_VERSION(9, 1) #define NVENC_HAVE_MULTIPLE_REF_FRAMES #define NVENC_HAVE_CUSTREAM_PTR #define NVENC_HAVE_GETLASTERRORSTRING #endif typedef struct NvencSurface { NV_ENC_INPUT_PTR input_surface; AVFrame *in_ref; int reg_idx; int width; int height; int pitch; NV_ENC_OUTPUT_PTR output_surface; NV_ENC_BUFFER_FORMAT format; int size; } NvencSurface; typedef struct NvencDynLoadFunctions { CudaFunctions *cuda_dl; NvencFunctions *nvenc_dl; NV_ENCODE_API_FUNCTION_LIST nvenc_funcs; int nvenc_device_count; } NvencDynLoadFunctions; enum { PRESET_DEFAULT = 0, PRESET_SLOW, PRESET_MEDIUM, PRESET_FAST, PRESET_HP, PRESET_HQ, PRESET_BD , PRESET_LOW_LATENCY_DEFAULT , PRESET_LOW_LATENCY_HQ , PRESET_LOW_LATENCY_HP, PRESET_LOSSLESS_DEFAULT, // lossless presets must be the last ones PRESET_LOSSLESS_HP, }; enum { NV_ENC_H264_PROFILE_BASELINE, NV_ENC_H264_PROFILE_MAIN, NV_ENC_H264_PROFILE_HIGH, NV_ENC_H264_PROFILE_HIGH_444P, }; enum { NV_ENC_HEVC_PROFILE_MAIN, NV_ENC_HEVC_PROFILE_MAIN_10, NV_ENC_HEVC_PROFILE_REXT, }; enum { NVENC_LOWLATENCY = 1, NVENC_LOSSLESS = 2, NVENC_ONE_PASS = 4, NVENC_TWO_PASSES = 8, }; enum { LIST_DEVICES = -2, ANY_DEVICE, }; typedef struct NvencContext { AVClass *avclass; NvencDynLoadFunctions nvenc_dload_funcs; NV_ENC_INITIALIZE_PARAMS init_encode_params; NV_ENC_CONFIG encode_config; CUcontext cu_context; CUcontext cu_context_internal; CUstream cu_stream; ID3D11Device *d3d11_device; int nb_surfaces; NvencSurface *surfaces; AVFifoBuffer *unused_surface_queue; AVFifoBuffer *output_surface_queue; AVFifoBuffer *output_surface_ready_queue; AVFifoBuffer *timestamp_list; int encoder_flushing; struct { void *ptr; int ptr_index; NV_ENC_REGISTERED_PTR regptr; int mapped; NV_ENC_MAP_INPUT_RESOURCE in_map; } registered_frames[MAX_REGISTERED_FRAMES]; int nb_registered_frames; /* the actual data pixel format, different from * AVCodecContext.pix_fmt when using hwaccel frames on input */ enum AVPixelFormat data_pix_fmt; int support_dyn_bitrate; void *nvencoder; int preset; int profile; int level; int tier; int rc; int cbr; int twopass; int device; int flags; int async_depth; int rc_lookahead; int aq; int no_scenecut; int forced_idr; int b_adapt; int temporal_aq; int zerolatency; int nonref_p; int strict_gop; int aq_strength; float quality; int aud; int bluray_compat; int init_qp_p; int init_qp_b; int init_qp_i; int cqp; int weighted_pred; int coder; int b_ref_mode; int a53_cc; int dpb_size; } NvencContext; int ff_nvenc_encode_init(AVCodecContext *avctx); int ff_nvenc_encode_close(AVCodecContext *avctx); 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); void ff_nvenc_encode_flush(AVCodecContext *avctx); extern const enum AVPixelFormat ff_nvenc_pix_fmts[]; extern const AVCodecHWConfigInternal *ff_nvenc_hw_configs[]; #endif /* AVCODEC_NVENC_H */