FFmpeg  4.2.2
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 
37 #include "avcodec.h"
38 
39 #define MAX_REGISTERED_FRAMES 64
40 #define RC_MODE_DEPRECATED 0x800000
41 #define RCD(rc_mode) ((rc_mode) | RC_MODE_DEPRECATED)
42 
43 #define NVENCAPI_CHECK_VERSION(major, minor) \
44  ((major) < NVENCAPI_MAJOR_VERSION || ((major) == NVENCAPI_MAJOR_VERSION && (minor) <= NVENCAPI_MINOR_VERSION))
45 
46 // SDK 8.1 compile time feature checks
47 #if NVENCAPI_CHECK_VERSION(8, 1)
48 #define NVENC_HAVE_BFRAME_REF_MODE
49 #define NVENC_HAVE_QP_MAP_MODE
50 #endif
51 
52 // SDK 9.0 compile time feature checks
53 #if NVENCAPI_CHECK_VERSION(9, 0)
54 #define NVENC_HAVE_HEVC_BFRAME_REF_MODE
55 #endif
56 
57 typedef struct NvencSurface
58 {
59  NV_ENC_INPUT_PTR input_surface;
61  int reg_idx;
62  int width;
63  int height;
64  int pitch;
65 
66  NV_ENC_OUTPUT_PTR output_surface;
67  NV_ENC_BUFFER_FORMAT format;
68  int size;
69 } NvencSurface;
70 
71 typedef struct NvencDynLoadFunctions
72 {
73  CudaFunctions *cuda_dl;
74  NvencFunctions *nvenc_dl;
75 
76  NV_ENCODE_API_FUNCTION_LIST nvenc_funcs;
79 
80 enum {
91  PRESET_LOSSLESS_DEFAULT, // lossless presets must be the last ones
93 };
94 
95 enum {
100 };
101 
102 enum {
106 };
107 
108 enum {
113 };
114 
115 enum {
118 };
119 
120 typedef struct NvencContext
121 {
123 
125 
126  NV_ENC_INITIALIZE_PARAMS init_encode_params;
127  NV_ENC_CONFIG encode_config;
128  CUcontext cu_context;
131 
134 
139 
141 
142  struct {
143  void *ptr;
145  NV_ENC_REGISTERED_PTR regptr;
146  int mapped;
147  NV_ENC_MAP_INPUT_RESOURCE in_map;
148  } registered_frames[MAX_REGISTERED_FRAMES];
150 
151  /* the actual data pixel format, different from
152  * AVCodecContext.pix_fmt when using hwaccel frames on input */
153  enum AVPixelFormat data_pix_fmt;
154 
155  /* timestamps of the first two frames, for computing the first dts
156  * when B-frames are present */
157  int64_t initial_pts[2];
159 
161 
162  void *nvencoder;
163 
164  int preset;
165  int profile;
166  int level;
167  int tier;
168  int rc;
169  int cbr;
170  int twopass;
171  int device;
172  int flags;
175  int aq;
178  int b_adapt;
181  int nonref_p;
184  float quality;
185  int aud;
190  int cqp;
192  int coder;
194  int a53_cc;
195 } NvencContext;
196 
198 
200 
201 int ff_nvenc_send_frame(AVCodecContext *avctx, const AVFrame *frame);
202 
204 
206  const AVFrame *frame, int *got_packet);
207 
208 extern const enum AVPixelFormat ff_nvenc_pix_fmts[];
209 
210 #endif /* AVCODEC_NVENC_H */
int no_scenecut
Definition: nvenc.h:176
AVClass * avclass
Definition: nvenc.h:122
void * nvencoder
Definition: nvenc.h:162
int support_dyn_bitrate
Definition: nvenc.h:160
enum AVPixelFormat ff_nvenc_pix_fmts[]
Definition: nvenc.c:42
int twopass
Definition: nvenc.h:170
NV_ENC_BUFFER_FORMAT format
Definition: nvenc.h:67
int ff_nvenc_encode_init(AVCodecContext *avctx)
Definition: nvenc.c:1479
int height
Definition: nvenc.h:63
This structure describes decoded (raw) audio or video data.
Definition: frame.h:295
AVFifoBuffer * timestamp_list
Definition: nvenc.h:138
NvencFunctions * nvenc_dl
Definition: nvenc.h:74
int mapped
Definition: nvenc.h:146
AVFrame * in_ref
Definition: nvenc.h:60
An API-specific header for AV_HWDEVICE_TYPE_D3D11VA.
int encoder_flushing
Definition: nvenc.h:140
int forced_idr
Definition: nvenc.h:177
NV_ENCODE_API_FUNCTION_LIST nvenc_funcs
Definition: nvenc.h:76
NvencDynLoadFunctions nvenc_dload_funcs
Definition: nvenc.h:124
ID3D11Device * d3d11_device
Definition: nvenc.h:130
int first_packet_output
Definition: nvenc.h:158
static AVPacket pkt
int init_qp_b
Definition: nvenc.h:188
int preset
Definition: nvenc.h:164
int pitch
Definition: nvenc.h:64
int nvenc_device_count
Definition: nvenc.h:77
NV_ENC_INPUT_PTR input_surface
Definition: nvenc.h:59
int aq
Definition: nvenc.h:175
int b_ref_mode
Definition: nvenc.h:193
CUcontext cu_context
Definition: nvenc.h:128
AVFifoBuffer * unused_surface_queue
Definition: nvenc.h:135
AVOptions.
int init_qp_p
Definition: nvenc.h:187
int ff_nvenc_encode_close(AVCodecContext *avctx)
Definition: nvenc.c:1404
int ff_nvenc_receive_packet(AVCodecContext *avctx, AVPacket *pkt)
Definition: nvenc.c:2161
float quality
Definition: nvenc.h:184
NV_ENC_INITIALIZE_PARAMS init_encode_params
Definition: nvenc.h:126
static AVFrame * frame
#define MAX_REGISTERED_FRAMES
Definition: nvenc.h:39
AVFifoBuffer * output_surface_ready_queue
Definition: nvenc.h:137
CUcontext cu_context_internal
Definition: nvenc.h:129
int ptr_index
Definition: nvenc.h:144
int async_depth
Definition: nvenc.h:173
void * ptr
Definition: nvenc.h:143
int coder
Definition: nvenc.h:192
int rc
Definition: nvenc.h:168
int nb_registered_frames
Definition: nvenc.h:149
int level
Definition: nvenc.h:166
void ID3D11Device
Definition: nvenc.h:28
int bluray_compat
Definition: nvenc.h:186
int aq_strength
Definition: nvenc.h:183
int ff_nvenc_encode_frame(AVCodecContext *avctx, AVPacket *pkt, const AVFrame *frame, int *got_packet)
Definition: nvenc.c:2197
int ff_nvenc_send_frame(AVCodecContext *avctx, const AVFrame *frame)
Definition: nvenc.c:2033
int flags
Definition: nvenc.h:172
NV_ENC_REGISTERED_PTR regptr
Definition: nvenc.h:145
int profile
Definition: nvenc.h:165
AVFifoBuffer * output_surface_queue
Definition: nvenc.h:136
CudaFunctions * cuda_dl
Definition: nvenc.h:73
NV_ENC_CONFIG encode_config
Definition: nvenc.h:127
Libavcodec external API header.
int strict_gop
Definition: nvenc.h:182
int temporal_aq
Definition: nvenc.h:179
main external API structure.
Definition: avcodec.h:1565
int init_qp_i
Definition: nvenc.h:189
a very simple circular buffer FIFO implementation
Describe the class of an AVClass context structure.
Definition: log.h:67
int width
Definition: nvenc.h:62
int nonref_p
Definition: nvenc.h:181
int cbr
Definition: nvenc.h:169
int reg_idx
Definition: nvenc.h:61
int b_adapt
Definition: nvenc.h:178
int weighted_pred
Definition: nvenc.h:191
int rc_lookahead
Definition: nvenc.h:174
int size
Definition: nvenc.h:68
NvencSurface * surfaces
Definition: nvenc.h:133
NV_ENC_MAP_INPUT_RESOURCE in_map
Definition: nvenc.h:147
int device
Definition: nvenc.h:171
int nb_surfaces
Definition: nvenc.h:132
int aud
Definition: nvenc.h:185
int cqp
Definition: nvenc.h:190
int tier
Definition: nvenc.h:167
int a53_cc
Definition: nvenc.h:194
AVPixelFormat
Pixel format.
Definition: pixfmt.h:64
This structure stores compressed data.
Definition: avcodec.h:1454
NV_ENC_OUTPUT_PTR output_surface
Definition: nvenc.h:66
int zerolatency
Definition: nvenc.h:180