FFmpeg
nvenc.h
Go to the documentation of this file.
1 /*
2  * This file is part of FFmpeg.
3  *
4  * FFmpeg is free software; you can redistribute it and/or
5  * modify it under the terms of the GNU Lesser General Public
6  * License as published by the Free Software Foundation; either
7  * version 2.1 of the License, or (at your option) any later version.
8  *
9  * FFmpeg is distributed in the hope that it will be useful,
10  * but WITHOUT ANY WARRANTY; without even the implied warranty of
11  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
12  * Lesser General Public License for more details.
13  *
14  * You should have received a copy of the GNU Lesser General Public
15  * License along with FFmpeg; if not, write to the Free Software
16  * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
17  */
18 
19 #ifndef AVCODEC_NVENC_H
20 #define AVCODEC_NVENC_H
21 
22 #include "config.h"
23 
24 #if CONFIG_D3D11VA
25 #define COBJMACROS
27 #else
28 typedef void ID3D11Device;
29 #endif
30 
31 #include <ffnvcodec/nvEncodeAPI.h>
32 
34 #include "libavutil/fifo.h"
35 #include "libavutil/opt.h"
36 #include "hwconfig.h"
37 
38 #include "avcodec.h"
39 
40 #define MAX_REGISTERED_FRAMES 64
41 #define RC_MODE_DEPRECATED 0x800000
42 #define RCD(rc_mode) ((rc_mode) | RC_MODE_DEPRECATED)
43 
44 #define NVENCAPI_CHECK_VERSION(major, minor) \
45  ((major) < NVENCAPI_MAJOR_VERSION || ((major) == NVENCAPI_MAJOR_VERSION && (minor) <= NVENCAPI_MINOR_VERSION))
46 
47 // SDK 8.1 compile time feature checks
48 #if NVENCAPI_CHECK_VERSION(8, 1)
49 #define NVENC_HAVE_BFRAME_REF_MODE
50 #define NVENC_HAVE_QP_MAP_MODE
51 #endif
52 
53 // SDK 9.0 compile time feature checks
54 #if NVENCAPI_CHECK_VERSION(9, 0)
55 #define NVENC_HAVE_HEVC_BFRAME_REF_MODE
56 #endif
57 
58 // SDK 9.1 compile time feature checks
59 #if NVENCAPI_CHECK_VERSION(9, 1)
60 #define NVENC_HAVE_MULTIPLE_REF_FRAMES
61 #define NVENC_HAVE_CUSTREAM_PTR
62 #define NVENC_HAVE_GETLASTERRORSTRING
63 #endif
64 
65 // SDK 10.0 compile time feature checks
66 #if NVENCAPI_CHECK_VERSION(10, 0)
67 #define NVENC_HAVE_NEW_PRESETS
68 #define NVENC_HAVE_MULTIPASS
69 #define NVENC_HAVE_LDKFS
70 #define NVENC_HAVE_H264_LVL6
71 #endif
72 
73 typedef struct NvencSurface
74 {
75  NV_ENC_INPUT_PTR input_surface;
77  int reg_idx;
78  int width;
79  int height;
80  int pitch;
81 
82  NV_ENC_OUTPUT_PTR output_surface;
83  NV_ENC_BUFFER_FORMAT format;
84 } NvencSurface;
85 
86 typedef struct NvencDynLoadFunctions
87 {
88  CudaFunctions *cuda_dl;
89  NvencFunctions *nvenc_dl;
90 
91  NV_ENCODE_API_FUNCTION_LIST nvenc_funcs;
94 
95 enum {
106  PRESET_LOSSLESS_DEFAULT, // lossless presets must be the last ones
108 #ifdef NVENC_HAVE_NEW_PRESETS
109  PRESET_P1,
110  PRESET_P2,
111  PRESET_P3,
112  PRESET_P4,
113  PRESET_P5,
114  PRESET_P6,
115  PRESET_P7,
116 #endif
117 };
118 
119 enum {
124 };
125 
126 enum {
130 };
131 
132 enum {
137 
139 };
140 
141 enum {
144 };
145 
146 typedef struct NvencContext
147 {
149 
151 
152  NV_ENC_INITIALIZE_PARAMS init_encode_params;
153  NV_ENC_CONFIG encode_config;
154  CUcontext cu_context;
156  CUstream cu_stream;
158 
160 
163 
168 
169  struct {
170  void *ptr;
172  NV_ENC_REGISTERED_PTR regptr;
173  int mapped;
174  NV_ENC_MAP_INPUT_RESOURCE in_map;
175  } registered_frames[MAX_REGISTERED_FRAMES];
177 
178  /* the actual data pixel format, different from
179  * AVCodecContext.pix_fmt when using hwaccel frames on input */
180  enum AVPixelFormat data_pix_fmt;
181 
183 
184  void *nvencoder;
185 
186  int preset;
187  int profile;
188  int level;
189  int tier;
190  int rc;
191  int cbr;
192  int twopass;
193  int device;
194  int flags;
197  int aq;
200  int b_adapt;
203  int nonref_p;
206  float quality;
207  int aud;
212  int cqp;
214  int coder;
216  int a53_cc;
217  int s12m_tc;
218  int dpb_size;
221  int ldkfs;
222 } NvencContext;
223 
225 
227 
229 
231 
232 extern const enum AVPixelFormat ff_nvenc_pix_fmts[];
233 extern const AVCodecHWConfigInternal *const ff_nvenc_hw_configs[];
234 
235 #endif /* AVCODEC_NVENC_H */
int no_scenecut
Definition: nvenc.h:198
AVClass * avclass
Definition: nvenc.h:148
void * nvencoder
Definition: nvenc.h:184
int support_dyn_bitrate
Definition: nvenc.h:182
enum AVPixelFormat ff_nvenc_pix_fmts[]
Definition: nvenc.c:46
int twopass
Definition: nvenc.h:192
NV_ENC_BUFFER_FORMAT format
Definition: nvenc.h:83
int ff_nvenc_encode_init(AVCodecContext *avctx)
Definition: nvenc.c:1629
int height
Definition: nvenc.h:79
This structure describes decoded (raw) audio or video data.
Definition: frame.h:314
AVFifoBuffer * timestamp_list
Definition: nvenc.h:167
NvencFunctions * nvenc_dl
Definition: nvenc.h:89
int mapped
Definition: nvenc.h:173
AVFrame * in_ref
Definition: nvenc.h:76
An API-specific header for AV_HWDEVICE_TYPE_D3D11VA.
int forced_idr
Definition: nvenc.h:199
NV_ENCODE_API_FUNCTION_LIST nvenc_funcs
Definition: nvenc.h:91
NvencDynLoadFunctions nvenc_dload_funcs
Definition: nvenc.h:150
ID3D11Device * d3d11_device
Definition: nvenc.h:157
static AVPacket pkt
int init_qp_b
Definition: nvenc.h:210
int preset
Definition: nvenc.h:186
int pitch
Definition: nvenc.h:80
int nvenc_device_count
Definition: nvenc.h:92
NV_ENC_INPUT_PTR input_surface
Definition: nvenc.h:75
void ff_nvenc_encode_flush(AVCodecContext *avctx)
Definition: nvenc.c:2345
int aq
Definition: nvenc.h:197
int b_ref_mode
Definition: nvenc.h:215
CUcontext cu_context
Definition: nvenc.h:154
AVFifoBuffer * unused_surface_queue
Definition: nvenc.h:164
AVOptions.
int init_qp_p
Definition: nvenc.h:209
int ff_nvenc_encode_close(AVCodecContext *avctx)
Definition: nvenc.c:1552
int ff_nvenc_receive_packet(AVCodecContext *avctx, AVPacket *pkt)
Definition: nvenc.c:2294
float quality
Definition: nvenc.h:206
NV_ENC_INITIALIZE_PARAMS init_encode_params
Definition: nvenc.h:152
#define MAX_REGISTERED_FRAMES
Definition: nvenc.h:40
AVFifoBuffer * output_surface_ready_queue
Definition: nvenc.h:166
CUcontext cu_context_internal
Definition: nvenc.h:155
int ptr_index
Definition: nvenc.h:171
int async_depth
Definition: nvenc.h:195
void * ptr
Definition: nvenc.h:170
int coder
Definition: nvenc.h:214
int rc
Definition: nvenc.h:190
int nb_registered_frames
Definition: nvenc.h:176
int level
Definition: nvenc.h:188
void ID3D11Device
Definition: nvenc.h:28
int bluray_compat
Definition: nvenc.h:208
int aq_strength
Definition: nvenc.h:205
int flags
Definition: nvenc.h:194
NV_ENC_REGISTERED_PTR regptr
Definition: nvenc.h:172
const AVCodecHWConfigInternal *const ff_nvenc_hw_configs[]
Definition: nvenc.c:62
int dpb_size
Definition: nvenc.h:218
int profile
Definition: nvenc.h:187
AVFifoBuffer * output_surface_queue
Definition: nvenc.h:165
int tuning_info
Definition: nvenc.h:219
CudaFunctions * cuda_dl
Definition: nvenc.h:88
NV_ENC_CONFIG encode_config
Definition: nvenc.h:153
Libavcodec external API header.
int strict_gop
Definition: nvenc.h:204
int temporal_aq
Definition: nvenc.h:201
main external API structure.
Definition: avcodec.h:531
int init_qp_i
Definition: nvenc.h:211
a very simple circular buffer FIFO implementation
Describe the class of an AVClass context structure.
Definition: log.h:67
int width
Definition: nvenc.h:78
int multipass
Definition: nvenc.h:220
int nonref_p
Definition: nvenc.h:203
int cbr
Definition: nvenc.h:191
int s12m_tc
Definition: nvenc.h:217
int reg_idx
Definition: nvenc.h:77
int b_adapt
Definition: nvenc.h:200
int weighted_pred
Definition: nvenc.h:213
int rc_lookahead
Definition: nvenc.h:196
NvencSurface * surfaces
Definition: nvenc.h:162
NV_ENC_MAP_INPUT_RESOURCE in_map
Definition: nvenc.h:174
CUstream cu_stream
Definition: nvenc.h:156
int device
Definition: nvenc.h:193
int nb_surfaces
Definition: nvenc.h:161
int aud
Definition: nvenc.h:207
int cqp
Definition: nvenc.h:212
int tier
Definition: nvenc.h:189
int ldkfs
Definition: nvenc.h:221
int a53_cc
Definition: nvenc.h:216
AVFrame * frame
Definition: nvenc.h:159
AVPixelFormat
Pixel format.
Definition: pixfmt.h:64
This structure stores compressed data.
Definition: packet.h:340
NV_ENC_OUTPUT_PTR output_surface
Definition: nvenc.h:82
int zerolatency
Definition: nvenc.h:202