blob: f6234d087bfb97b927d3b4bb547281a3f684dfee (
plain)
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
|
/* nvenc_common.c
*
* Copyright (c) 2003-2021 HandBrake Team
* This file is part of the HandBrake source code.
* Homepage: <http://handbrake.fr/>.
* It may be used under the terms of the GNU General Public License v2.
* For full terms see the file COPYING file or visit http://www.gnu.org/licenses/gpl-2.0.html
*/
#include "handbrake/hbffmpeg.h"
#include "handbrake/handbrake.h"
#if HB_PROJECT_FEATURE_NVENC
#include <ffnvcodec/nvEncodeAPI.h>
#include <ffnvcodec/dynlink_loader.h>
#endif
static int is_nvenc_available = -1;
int hb_nvenc_h264_available()
{
#if HB_PROJECT_FEATURE_NVENC
return hb_check_nvenc_available();
#else
return is_nvenc_available;
#endif
}
int hb_nvenc_h265_available()
{
#if HB_PROJECT_FEATURE_NVENC
return hb_check_nvenc_available();
#else
return is_nvenc_available;
#endif
}
int hb_check_nvenc_available()
{
if (is_hardware_disabled())
{
return 0;
}
if (is_nvenc_available != -1)
{
return is_nvenc_available;
}
#if HB_PROJECT_FEATURE_NVENC
uint32_t nvenc_ver;
void *context = NULL;
NvencFunctions *nvenc_dl = NULL;
int loadErr = nvenc_load_functions(&nvenc_dl, context);
if (loadErr < 0) {
return 0;
}
NVENCSTATUS apiErr = nvenc_dl->NvEncodeAPIGetMaxSupportedVersion(&nvenc_ver);
if (apiErr != NV_ENC_SUCCESS) {
is_nvenc_available = 0;
return 0;
} else {
hb_deep_log(1, "NVENC version %d.%d\n", nvenc_ver >> 4, nvenc_ver & 0xf);
is_nvenc_available = 1;
return 1;
}
return 1;
#else
return 0;
#endif
}
|