nvenc 0.1.0

Wrapper for the nvidia NVENC API
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
196
197
198
199
200
201
202
203
204
205
206
207
208
209
210
211
212
213
214

use std::{ffi::c_char, mem::MaybeUninit, os::raw::c_void};

use crate::{
    stdcall,
    sys::{
        enums::{NVencBufferFormat, NVencTuningInfo},
        result::NVencResult,
        structs::{
            Guid, NVencCapsParam, NVencCreateBitstreamBuffer, NVencCreateInputBuffer,
            NVencInitializeParams, NVencLockBitStream, NVencLockInputBuffer, NVencMapInputResource,
            NVencOpenEncodeSessionExParams, NVencPicParams, NVencPresetConfig,
            NVencRegisterResource,
        },
    },
};

#[repr(C)]
#[derive(Clone)]
pub struct NVencFunctionList {
    pub version: u32,
    rsvd: u32,
    #[deprecated]
    nvenc_open_encode_session: NvencOpenEncodeSession,
    pub nvenc_get_encoder_guid_count: NvencGetEncodeGuidCount,
    nvenc_get_encoder_profile_guid_count: NvencGetEncodeProfileGuidCount,
    nvenc_get_encoder_profile_guids: NvencGetEncodeProfileGuids,
    pub nvenc_get_encoder_guids: NvencGetEncodeGuids,
    nvenc_get_input_format_count: NvencGetInputFormatCount,
    nvenc_get_input_formats: NvencGetInputFormats,
    nvenc_get_encode_caps: NvencGetEncodeCaps,
    pub nvenc_get_encode_preset_count: NvencGetEncodePresetCount,
    pub nvenc_get_encode_preset_guids: NvencGetEncodePresetGuids,
    nvenc_get_encode_preset_config: NvencGetEncodePresetConfig,
    pub nvenc_initialize_encoder: NvencInitializeEncoder,
    pub nvenc_create_input_buffer: NvencCreateInputBuffer,
    pub nvenc_destroy_input_buffer: NvencDestroyInputBuffer,
    pub nvenc_create_bit_stream_buffer: NvencCreateBitStreamBuffer,
    pub nvenc_destory_bit_stream_buffer: NvencDestroyBitStreamBuffer,
    pub nvenc_encode_picture: NvencEncodePicture,
    pub nvenc_lock_bit_stream: NvencLockBitStream,
    pub nvenc_unlock_bit_stream: NvencUnlockBitStream,
    pub nvenc_lock_input_buffer: NvencLockInputBuffer,
    pub nvenc_unlock_input_buffer: NvencUnlockInputBuffer,
    nvenc_get_encode_stats: NvencGetEncodeStats,
    nvenc_get_sequence_params: NvencGetSequenceParams,
    nvenc_register_async_event: NvencRegisterAsyncEvent,
    nvenc_unregister_async_event: NvencUnregisterAsyncEvent,
    pub nvenc_map_input_resource: NvencMapInputResource,
    pub nvenc_unmap_input_resource: NvencUnmapInputResource,
    pub nvenc_destroy_encoder: NvencDestroyEncoder,
    nvenc_invalidate_ref_frames: NvencInvalidateRefFrames,
    pub nvenc_open_encode_session_ex: NvencOpenEncodeSessionEx,
    pub nvenc_register_resource: NvencRegisterResource,
    pub nvenc_unregister_resource: NvencUnRegisterResource,
    nvenc_reconfigure_encoder: NvencReconfigureEncoder,
    rsvd1: *mut c_void,
    nvenc_create_mv_buffer: NvencCreateMVBuffer,
    nvenc_destory_mv_buffer: NvencDestoryMVBuffer,
    nvenc_run_motion_estimation_only: NvencRunMotionEstimationOnly,
    pub nvenc_get_last_error: NvencGetLastError,
    nvenc_set_io_cuda_streams: NvencSetIOCudaStreams,
    pub nvenc_get_encode_preset_config_ex: NvencGetEncodePresetConfigEx,
    nvenc_get_sequence_params_ex: NvencGetSequenceParamsEx,
    nvenc_store_encoder_state: NvencStoreEncoderState,
    nvenc_look_ahead_picture: NvencLookAheadPicture,
    rsvd2: [*mut c_void; 275],
}

type NvencOpenEncodeSession =
    stdcall!(fn(device: *mut c_void, device_type: i32, encoder: *mut *mut c_void) -> NVencResult);
type NvencGetEncodeGuidCount =
    stdcall!(fn(encoder: *mut c_void, encoder_guid_count: *mut u32) -> NVencResult);
type NvencGetEncodeGuids = stdcall!(
    fn(
        encoder: *mut c_void,
        guids: *mut Guid,
        guid_array_size: u32,
        guid_count: *mut u32,
    ) -> NVencResult
);
type NvencGetEncodeProfileGuidCount = stdcall!(
    fn(
        encoder: *mut c_void,
        encode_guid: Guid,
        encoder_profile_guid_count: *mut u32,
    ) -> NVencResult
);
type NvencGetEncodeProfileGuids = stdcall!(
    fn(
        encoder: *mut c_void,
        profile_guids: *mut Guid,
        guid_array_size: u32,
        guid_count: *mut u32,
    ) -> NVencResult
);
type NvencGetInputFormatCount = stdcall!(
    fn(encoder: *mut c_void, encoder_guid: Guid, input_fmt_count: *mut u32) -> NVencResult
);
type NvencGetInputFormats = stdcall!(
    fn(
        encoder: *mut c_void,
        input_formats: *mut NVencBufferFormat,
        input_format_array_size: u32,
        input_format_count: *mut u32,
    ) -> NVencResult
);
type NvencGetEncodeCaps = stdcall!(
    fn(
        encoder: *mut c_void,
        encoder_guid: Guid,
        caps_param: *mut NVencCapsParam,
        caps_val: *mut i32,
    ) -> NVencResult
);
type NvencGetEncodePresetCount = stdcall!(
    fn(
        encoder: *mut c_void,
        encoder_guid: Guid,
        encoder_preset_guid_count: *mut u32,
    ) -> NVencResult
);
type NvencGetEncodePresetGuids = stdcall!(
    fn(
        encoder: *mut c_void,
        encode_guid: Guid,
        preset_guids: *mut Guid,
        guid_array_size: u32,
        encoder_preset_guid_count: *mut u32,
    ) -> NVencResult
);
type NvencGetEncodePresetConfig = stdcall!(
    fn(
        encoder: *mut c_void,
        encode_guid: Guid,
        preset_guid: Guid,
        preset_config: *mut NVencPresetConfig,
    ) -> NVencResult
);
type NvencGetEncodePresetConfigEx = stdcall!(
    fn(
        encoder: *mut c_void,
        encode_guid: Guid,
        preset_guid: Guid,
        tuning_info: NVencTuningInfo,
        preset_config: *mut MaybeUninit<NVencPresetConfig>,
    ) -> NVencResult
);
type NvencInitializeEncoder = stdcall!(
    fn(encoder: *mut c_void, create_encode_params: *mut NVencInitializeParams) -> NVencResult
);
type NvencCreateInputBuffer = stdcall!(
    fn(
        encoder: *mut c_void,
        create_input_buffer_params: *mut NVencCreateInputBuffer,
    ) -> NVencResult
);
type NvencDestroyInputBuffer =
    stdcall!(fn(encoder: *mut c_void, input_buffer: *mut c_void) -> NVencResult);
type NvencCreateBitStreamBuffer = stdcall!(
    fn(
        encoder: *mut c_void,
        create_bit_stream_buffer_params: *mut NVencCreateBitstreamBuffer,
    ) -> NVencResult
);
type NvencDestroyBitStreamBuffer =
    stdcall!(fn(encoder: *mut c_void, bitstream_buffer: *mut c_void) -> NVencResult);
type NvencEncodePicture = stdcall!(
    fn(encoder: *mut c_void, encode_pic_params: *mut MaybeUninit<NVencPicParams>) -> NVencResult
);
type NvencLockBitStream = stdcall!(
    fn(
        encoder: *mut c_void,
        lock_bitstream_buffer_params: *mut MaybeUninit<NVencLockBitStream>,
    ) -> NVencResult
);
type NvencUnlockBitStream =
    stdcall!(fn(encoder: *mut c_void, bitstream_buffer: *mut c_void) -> NVencResult);
type NvencLockInputBuffer = stdcall!(
    fn(encoder: *mut c_void, lock_input_buffer_params: *mut NVencLockInputBuffer) -> NVencResult
);
type NvencUnlockInputBuffer =
    stdcall!(fn(encoder: *mut c_void, input_buffer: *mut c_void) -> NVencResult);
type NvencGetEncodeStats = stdcall!(fn(encoder: *mut c_void) -> NVencResult);
type NvencGetSequenceParams = stdcall!(fn(encoder: *mut c_void) -> NVencResult);
type NvencRegisterAsyncEvent = stdcall!(fn(encoder: *mut c_void) -> NVencResult);
type NvencUnregisterAsyncEvent = stdcall!(fn(encoder: *mut c_void) -> NVencResult);
type NvencMapInputResource = stdcall!(
    fn(encoder: *mut c_void, map_input_resource_params: *mut NVencMapInputResource) -> NVencResult
);
type NvencUnmapInputResource =
    stdcall!(fn(encoder: *mut c_void, mapped_input_buffer: *mut c_void) -> NVencResult);
type NvencDestroyEncoder = stdcall!(fn(encoder: *mut c_void) -> NVencResult);
type NvencInvalidateRefFrames = stdcall!(fn(encoder: *mut c_void) -> NVencResult);
type NvencOpenEncodeSessionEx = stdcall!(
    fn(
        open_session_ex_params: *mut NVencOpenEncodeSessionExParams,
        encoder: *mut *mut c_void,
    ) -> NVencResult
);
type NvencRegisterResource = stdcall!(
    fn(encoder: *mut c_void, register_res_params: *mut NVencRegisterResource) -> NVencResult
);
type NvencUnRegisterResource =
    stdcall!(fn(encoder: *mut c_void, registered_res: *mut c_void) -> NVencResult);
type NvencReconfigureEncoder = stdcall!(fn(encoder: *mut c_void) -> NVencResult);

type NvencCreateMVBuffer = stdcall!(fn(encoder: *mut c_void) -> NVencResult);
type NvencDestoryMVBuffer = stdcall!(fn(encoder: *mut c_void) -> NVencResult);
type NvencRunMotionEstimationOnly = stdcall!(fn(encoder: *mut c_void) -> NVencResult);
type NvencGetLastError = stdcall!(fn(encoder: *mut c_void) -> *const c_char);
type NvencSetIOCudaStreams = stdcall!(fn(encoder: *mut c_void) -> NVencResult);
type NvencGetSequenceParamsEx = stdcall!(fn(encoder: *mut c_void) -> NVencResult);
type NvencStoreEncoderState = stdcall!(fn(encoder: *mut c_void) -> NVencResult);
type NvencLookAheadPicture = stdcall!(fn(encoder: *mut c_void) -> NVencResult);