YangWX/crossdesk
1
0
Fork 0
mirror of https://github.com/kunkundi/crossdesk.git synced 2025-10-27 12:45:35 +08:00
Code Issues Packages Projects Releases Wiki Activity

Add hardware accerleration valiation

Browse Source
This commit is contained in:
dijunkun
2023-09-21 11:26:08 +08:00
parent b4318cc8d6
commit 156172accb
7 changed files with 90 additions and 14 deletions
Download Patch File Download Diff File Expand all files Collapse all files

23
src/media/video/encode/video_encoder_factory.cpp
View File

@@ -10,8 +10,29 @@ VideoEncoderFactory::~VideoEncoderFactory() {}
std::unique_ptr<VideoEncoder> VideoEncoderFactory::CreateVideoEncoder(
bool hardware_acceleration) {
if (hardware_acceleration) {
return std::make_unique<NvidiaVideoEncoder>(NvidiaVideoEncoder());
if (CheckIsHardwareAccerlerationSupported()) {
return std::make_unique<NvidiaVideoEncoder>(NvidiaVideoEncoder());
} else {
return nullptr;
}
} else {
return std::make_unique<FFmpegVideoEncoder>(FFmpegVideoEncoder());
}
}
bool VideoEncoderFactory::CheckIsHardwareAccerlerationSupported() {
CUresult cuResult;
NV_ENCODE_API_FUNCTION_LIST functionList = {NV_ENCODE_API_FUNCTION_LIST_VER};
cuResult = cuInit(0);
if (cuResult != CUDA_SUCCESS) {
return false;
}
NVENCSTATUS nvEncStatus = NvEncodeAPICreateInstance(&functionList);
if (nvEncStatus != NV_ENC_SUCCESS) {
return false;
}
return true;
}

2
src/media/video/encode/video_encoder_factory.h
View File

@@ -9,6 +9,8 @@ class VideoEncoderFactory {
static std::unique_ptr<VideoEncoder> CreateVideoEncoder(
bool hardware_acceleration);
static bool CheckIsHardwareAccerlerationSupported();
};
#endif