[fix] fix load NvCodec API failed on Windows

This commit is contained in:
dijunkun
2024-08-13 16:10:00 +08:00
parent 9e30203e90
commit 4624d4f27f
6 changed files with 633 additions and 1672 deletions

View File

@@ -0,0 +1,313 @@
#include "nvcodec_api.h"
#include "log.h"
TcuInit cuInit_ld = NULL;
TcuDeviceGet cuDeviceGet_ld = NULL;
TcuDeviceGetCount cuDeviceGetCount_ld = NULL;
TcuCtxCreate cuCtxCreate_ld = NULL;
TcuGetErrorName cuGetErrorName_ld = NULL;
TcuCtxPushCurrent cuCtxPushCurrent_ld = NULL;
TcuCtxPopCurrent cuCtxPopCurrent_ld = NULL;
TcuMemAlloc cuMemAlloc_ld = NULL;
TcuMemAllocPitch cuMemAllocPitch_ld = NULL;
TcuMemFree cuMemFree_ld = NULL;
TcuMemcpy2DAsync cuMemcpy2DAsync_ld = NULL;
TcuStreamSynchronize cuStreamSynchronize_ld = NULL;
TcuMemcpy2D cuMemcpy2D_ld = NULL;
TcuMemcpy2DUnaligned cuMemcpy2DUnaligned_ld = NULL;
TcuvidCtxLockCreate cuvidCtxLockCreate_ld = NULL;
TcuvidGetDecoderCaps cuvidGetDecoderCaps_ld = NULL;
TcuvidCreateDecoder cuvidCreateDecoder_ld = NULL;
TcuvidDestroyDecoder cuvidDestroyDecoder_ld = NULL;
TcuvidDecodePicture cuvidDecodePicture_ld = NULL;
TcuvidGetDecodeStatus cuvidGetDecodeStatus_ld = NULL;
TcuvidReconfigureDecoder cuvidReconfigureDecoder_ld = NULL;
TcuvidMapVideoFrame64 cuvidMapVideoFrame64_ld = NULL;
TcuvidUnmapVideoFrame64 cuvidUnmapVideoFrame64_ld = NULL;
TcuvidCtxLockDestroy cuvidCtxLockDestroy_ld = NULL;
TcuvidCreateVideoParser cuvidCreateVideoParser_ld = NULL;
TcuvidParseVideoData cuvidParseVideoData_ld = NULL;
TcuvidDestroyVideoParser cuvidDestroyVideoParser_ld = NULL;
TNvEncodeAPICreateInstance NvEncodeAPICreateInstance_ld = NULL;
TNvEncodeAPIGetMaxSupportedVersion NvEncodeAPIGetMaxSupportedVersion_ld = NULL;
static HMODULE nvcuda_dll = NULL;
static HMODULE nvcuvid_dll = NULL;
static HMODULE nvEncodeAPI64_dll = NULL;
int LoadNvCodecDll() {
// Load library
nvcuda_dll = LoadLibrary(TEXT("nvcuda.dll"));
if (nvcuda_dll == NULL) {
LOG_ERROR("Unable to load nvcuda.dll");
return -1;
}
cuInit_ld = (TcuInit)GetProcAddress(nvcuda_dll, "cuInit");
if (cuInit_ld == NULL) {
LOG_ERROR("Unable to find function cuInit()");
FreeLibrary(nvcuda_dll);
return -1;
}
cuDeviceGet_ld = (TcuDeviceGet)GetProcAddress(nvcuda_dll, "cuDeviceGet");
if (cuDeviceGet_ld == NULL) {
LOG_ERROR("Unable to find function cuDeviceGet()");
FreeLibrary(nvcuda_dll);
return -1;
}
cuDeviceGetCount_ld =
(TcuDeviceGetCount)GetProcAddress(nvcuda_dll, "cuDeviceGetCount");
if (cuDeviceGetCount_ld == NULL) {
LOG_ERROR("Unable to find function cuDeviceGetCount()");
FreeLibrary(nvcuda_dll);
return -1;
}
cuCtxCreate_ld = (TcuCtxCreate)GetProcAddress(nvcuda_dll, "cuCtxCreate_v2");
if (cuCtxCreate_ld == NULL) {
LOG_ERROR("Unable to find function cuCtxCreate()");
FreeLibrary(nvcuda_dll);
return -1;
}
cuGetErrorName_ld =
(TcuGetErrorName)GetProcAddress(nvcuda_dll, "cuGetErrorName");
if (cuGetErrorName_ld == NULL) {
LOG_ERROR("Unable to find function cuGetErrorName()");
FreeLibrary(nvcuda_dll);
return -1;
}
cuCtxPushCurrent_ld =
(TcuCtxPushCurrent)GetProcAddress(nvcuda_dll, "cuCtxPushCurrent_v2");
if (cuCtxPushCurrent_ld == NULL) {
LOG_ERROR("Unable to find function cuCtxPushCurrent()");
FreeLibrary(nvcuda_dll);
return -1;
}
cuCtxPopCurrent_ld =
(TcuCtxPopCurrent)GetProcAddress(nvcuda_dll, "cuCtxPopCurrent_v2");
if (cuCtxPopCurrent_ld == NULL) {
LOG_ERROR("Unable to find function cuCtxPopCurrent()");
FreeLibrary(nvcuda_dll);
return -1;
}
cuMemAlloc_ld = (TcuMemAlloc)GetProcAddress(nvcuda_dll, "cuMemAlloc_v2");
if (cuMemAlloc_ld == NULL) {
LOG_ERROR("Unable to find function cuMemAlloc()");
FreeLibrary(nvcuda_dll);
return -1;
}
cuMemAllocPitch_ld =
(TcuMemAllocPitch)GetProcAddress(nvcuda_dll, "cuMemAllocPitch_v2");
if (cuMemAllocPitch_ld == NULL) {
LOG_ERROR("Unable to find function cuMemAllocPitch()");
FreeLibrary(nvcuda_dll);
return -1;
}
cuMemFree_ld = (TcuMemFree)GetProcAddress(nvcuda_dll, "cuMemFree_v2");
if (cuMemFree_ld == NULL) {
LOG_ERROR("Unable to find function cuMemFree()");
FreeLibrary(nvcuda_dll);
return -1;
}
cuMemcpy2DAsync_ld =
(TcuMemcpy2DAsync)GetProcAddress(nvcuda_dll, "cuMemcpy2DAsync_v2");
if (cuMemcpy2DAsync_ld == NULL) {
LOG_ERROR("Unable to find function cuMemcpy2DAsync()");
FreeLibrary(nvcuda_dll);
return -1;
}
cuStreamSynchronize_ld =
(TcuStreamSynchronize)GetProcAddress(nvcuda_dll, "cuStreamSynchronize");
if (cuStreamSynchronize_ld == NULL) {
LOG_ERROR("Unable to find function cuStreamSynchronize()");
FreeLibrary(nvcuda_dll);
return -1;
}
cuMemcpy2D_ld = (TcuMemcpy2D)GetProcAddress(nvcuda_dll, "cuMemcpy2D_v2");
if (cuMemcpy2D_ld == NULL) {
LOG_ERROR("Unable to find function cuMemcpy2D()");
FreeLibrary(nvcuda_dll);
return -1;
}
cuMemcpy2DUnaligned_ld = (TcuMemcpy2DUnaligned)GetProcAddress(
nvcuda_dll, "cuMemcpy2DUnaligned_v2");
if (cuMemcpy2DUnaligned_ld == NULL) {
LOG_ERROR("Unable to find function cuMemcpy2DUnaligned()");
FreeLibrary(nvcuda_dll);
return -1;
}
//
nvcuvid_dll = LoadLibrary(TEXT("nvcuvid.dll"));
if (nvcuvid_dll == NULL) {
LOG_ERROR("Unable to load nvcuvid.dll");
return -1;
}
cuvidCtxLockCreate_ld =
(TcuvidCtxLockCreate)GetProcAddress(nvcuvid_dll, "cuvidCtxLockCreate");
if (cuvidCtxLockCreate_ld == NULL) {
LOG_ERROR("Unable to find function cuvidCtxLockCreate()");
FreeLibrary(nvcuvid_dll);
return -1;
}
cuvidGetDecoderCaps_ld =
(TcuvidGetDecoderCaps)GetProcAddress(nvcuvid_dll, "cuvidGetDecoderCaps");
if (cuvidGetDecoderCaps_ld == NULL) {
LOG_ERROR("Unable to find function cuvidGetDecoderCaps()");
FreeLibrary(nvcuvid_dll);
return -1;
}
cuvidCreateDecoder_ld =
(TcuvidCreateDecoder)GetProcAddress(nvcuvid_dll, "cuvidCreateDecoder");
if (cuvidCreateDecoder_ld == NULL) {
LOG_ERROR("Unable to find function cuvidCreateDecoder()");
FreeLibrary(nvcuvid_dll);
return -1;
}
cuvidDestroyDecoder_ld =
(TcuvidDestroyDecoder)GetProcAddress(nvcuvid_dll, "cuvidDestroyDecoder");
if (cuvidDestroyDecoder_ld == NULL) {
LOG_ERROR("Unable to find function cuvidDestroyDecoder()");
FreeLibrary(nvcuvid_dll);
return -1;
}
cuvidDecodePicture_ld =
(TcuvidDecodePicture)GetProcAddress(nvcuvid_dll, "cuvidDecodePicture");
if (cuvidDecodePicture_ld == NULL) {
LOG_ERROR("Unable to find function cuvidDecodePicture()");
FreeLibrary(nvcuvid_dll);
return -1;
}
cuvidGetDecodeStatus_ld = (TcuvidGetDecodeStatus)GetProcAddress(
nvcuvid_dll, "cuvidGetDecodeStatus");
if (cuvidGetDecodeStatus_ld == NULL) {
LOG_ERROR("Unable to find function cuvidGetDecodeStatus()");
FreeLibrary(nvcuvid_dll);
return -1;
}
cuvidReconfigureDecoder_ld = (TcuvidReconfigureDecoder)GetProcAddress(
nvcuvid_dll, "cuvidReconfigureDecoder");
if (cuvidReconfigureDecoder_ld == NULL) {
LOG_ERROR("Unable to find function cuvidReconfigureDecoder()");
FreeLibrary(nvcuvid_dll);
return -1;
}
cuvidMapVideoFrame64_ld = (TcuvidMapVideoFrame64)GetProcAddress(
nvcuvid_dll, "cuvidMapVideoFrame64");
if (cuvidMapVideoFrame64_ld == NULL) {
LOG_ERROR("Unable to find function cuvidMapVideoFrame64()");
FreeLibrary(nvcuvid_dll);
return -1;
}
cuvidUnmapVideoFrame64_ld = (TcuvidUnmapVideoFrame64)GetProcAddress(
nvcuvid_dll, "cuvidUnmapVideoFrame64");
if (cuvidUnmapVideoFrame64_ld == NULL) {
LOG_ERROR("Unable to find function cuvidUnmapVideoFrame64()");
FreeLibrary(nvcuvid_dll);
return -1;
}
cuvidCtxLockDestroy_ld =
(TcuvidCtxLockDestroy)GetProcAddress(nvcuvid_dll, "cuvidCtxLockDestroy");
if (cuvidCtxLockDestroy_ld == NULL) {
LOG_ERROR("Unable to find function cuvidCtxLockDestroy()");
FreeLibrary(nvcuvid_dll);
return -1;
}
cuvidCreateVideoParser_ld = (TcuvidCreateVideoParser)GetProcAddress(
nvcuvid_dll, "cuvidCreateVideoParser");
if (cuvidCreateVideoParser_ld == NULL) {
LOG_ERROR("Unable to find function cuvidCreateVideoParser()");
FreeLibrary(nvcuvid_dll);
return -1;
}
cuvidParseVideoData_ld =
(TcuvidParseVideoData)GetProcAddress(nvcuvid_dll, "cuvidParseVideoData");
if (cuvidParseVideoData_ld == NULL) {
LOG_ERROR("Unable to find function cuvidParseVideoData()");
FreeLibrary(nvcuvid_dll);
return -1;
}
cuvidDestroyVideoParser_ld = (TcuvidDestroyVideoParser)GetProcAddress(
nvcuvid_dll, "cuvidDestroyVideoParser");
if (cuvidDestroyVideoParser_ld == NULL) {
LOG_ERROR("Unable to find function cuvidDestroyVideoParser()");
FreeLibrary(nvcuvid_dll);
return -1;
}
//
nvEncodeAPI64_dll = LoadLibrary(TEXT("nvEncodeAPI64.dll"));
if (nvEncodeAPI64_dll == NULL) {
LOG_ERROR("Unable to load nvEncodeAPI64.dll");
return -1;
}
NvEncodeAPICreateInstance_ld = (TNvEncodeAPICreateInstance)GetProcAddress(
nvEncodeAPI64_dll, "NvEncodeAPICreateInstance");
if (NvEncodeAPICreateInstance_ld == NULL) {
LOG_ERROR("Unable to find function NvEncodeAPICreateInstance()");
FreeLibrary(nvEncodeAPI64_dll);
return -1;
}
NvEncodeAPIGetMaxSupportedVersion_ld =
(TNvEncodeAPIGetMaxSupportedVersion)GetProcAddress(
nvEncodeAPI64_dll, "NvEncodeAPIGetMaxSupportedVersion");
if (NvEncodeAPIGetMaxSupportedVersion_ld == NULL) {
LOG_ERROR("Unable to find function NvEncodeAPIGetMaxSupportedVersion()");
FreeLibrary(nvEncodeAPI64_dll);
return -1;
}
LOG_INFO("Load NvCodec API success");
return 0;
}
int ReleaseNvCodecDll() {
if (nvcuda_dll != NULL) {
FreeLibrary(nvcuda_dll);
nvcuda_dll = NULL;
}
if (nvcuvid_dll != NULL) {
FreeLibrary(nvcuvid_dll);
nvcuvid_dll = NULL;
}
if (nvEncodeAPI64_dll != NULL) {
FreeLibrary(nvEncodeAPI64_dll);
nvEncodeAPI64_dll = NULL;
}
LOG_INFO("Release NvCodec API success");
return 0;
}

View File

@@ -0,0 +1,118 @@
/*
* @Author: DI JUNKUN
* @Date: 2024-08-12
* Copyright (c) 2024 by DI JUNKUN, All Rights Reserved.
*/
#ifndef _NVCODEC_API_H_
#define _NVCODEC_API_H_
#include <Windows.h>
#include <iostream>
#include "cuda.h"
#include "cuviddec.h"
#include "nvEncodeAPI.h"
#include "nvcuvid.h"
// nvcuda.dll
typedef CUresult (*TcuInit)(unsigned int Flags);
typedef CUresult (*TcuDeviceGet)(CUdevice *device, int ordinal);
typedef CUresult (*TcuDeviceGetCount)(int *count);
typedef CUresult (*TcuCtxCreate)(CUcontext *pctx, unsigned int flags,
CUdevice dev);
typedef CUresult (*TcuGetErrorName)(CUresult error, const char **pStr);
typedef CUresult (*TcuCtxPushCurrent)(CUcontext ctx);
typedef CUresult (*TcuCtxPopCurrent)(CUcontext *pctx);
typedef CUresult (*TcuMemAlloc)(CUdeviceptr *dptr, size_t bytesize);
typedef CUresult (*TcuMemAllocPitch)(CUdeviceptr *dptr, size_t *pPitch,
size_t WidthInBytes, size_t Height,
unsigned int ElementSizeBytes);
typedef CUresult (*TcuMemFree)(CUdeviceptr dptr);
typedef CUresult (*TcuMemcpy2DAsync)(const CUDA_MEMCPY2D *pCopy,
CUstream hStream);
typedef CUresult (*TcuStreamSynchronize)(CUstream hStream);
typedef CUresult (*TcuMemcpy2D)(const CUDA_MEMCPY2D *pCopy);
typedef CUresult (*TcuMemcpy2DUnaligned)(const CUDA_MEMCPY2D *pCopy);
extern TcuInit cuInit_ld;
extern TcuDeviceGet cuDeviceGet_ld;
extern TcuDeviceGetCount cuDeviceGetCount_ld;
extern TcuCtxCreate cuCtxCreate_ld;
extern TcuGetErrorName cuGetErrorName_ld;
extern TcuCtxPushCurrent cuCtxPushCurrent_ld;
extern TcuCtxPopCurrent cuCtxPopCurrent_ld;
extern TcuMemAlloc cuMemAlloc_ld;
extern TcuMemAllocPitch cuMemAllocPitch_ld;
extern TcuMemFree cuMemFree_ld;
extern TcuMemcpy2DAsync cuMemcpy2DAsync_ld;
extern TcuStreamSynchronize cuStreamSynchronize_ld;
extern TcuMemcpy2D cuMemcpy2D_ld;
extern TcuMemcpy2DUnaligned cuMemcpy2DUnaligned_ld;
// nvcuvid.dll
typedef CUresult (*TcuvidCtxLockCreate)(CUvideoctxlock *pLock, CUcontext ctx);
typedef CUresult (*TcuvidGetDecoderCaps)(CUVIDDECODECAPS *pdc);
typedef CUresult (*TcuvidCreateDecoder)(CUvideodecoder *phDecoder,
CUVIDDECODECREATEINFO *pdci);
typedef CUresult (*TcuvidDestroyDecoder)(CUvideodecoder hDecoder);
typedef CUresult (*TcuvidDecodePicture)(CUvideodecoder hDecoder,
CUVIDPICPARAMS *pPicParams);
typedef CUresult (*TcuvidGetDecodeStatus)(CUvideodecoder hDecoder, int nPicIdx,
CUVIDGETDECODESTATUS *pDecodeStatus);
typedef CUresult (*TcuvidReconfigureDecoder)(
CUvideodecoder hDecoder, CUVIDRECONFIGUREDECODERINFO *pDecReconfigParams);
typedef CUresult (*TcuvidMapVideoFrame64)(CUvideodecoder hDecoder, int nPicIdx,
unsigned long long *pDevPtr,
unsigned int *pPitch,
CUVIDPROCPARAMS *pVPP);
typedef CUresult (*TcuvidUnmapVideoFrame64)(CUvideodecoder hDecoder,
unsigned long long DevPtr);
typedef CUresult (*TcuvidCtxLockDestroy)(CUvideoctxlock lck);
typedef CUresult (*TcuvidCreateVideoParser)(CUvideoparser *pObj,
CUVIDPARSERPARAMS *pParams);
typedef CUresult (*TcuvidParseVideoData)(CUvideoparser obj,
CUVIDSOURCEDATAPACKET *pPacket);
typedef CUresult (*TcuvidDestroyVideoParser)(CUvideoparser obj);
extern TcuvidCtxLockCreate cuvidCtxLockCreate_ld;
extern TcuvidGetDecoderCaps cuvidGetDecoderCaps_ld;
extern TcuvidCreateDecoder cuvidCreateDecoder_ld;
extern TcuvidDestroyDecoder cuvidDestroyDecoder_ld;
extern TcuvidDecodePicture cuvidDecodePicture_ld;
extern TcuvidGetDecodeStatus cuvidGetDecodeStatus_ld;
extern TcuvidReconfigureDecoder cuvidReconfigureDecoder_ld;
extern TcuvidMapVideoFrame64 cuvidMapVideoFrame64_ld;
extern TcuvidUnmapVideoFrame64 cuvidUnmapVideoFrame64_ld;
extern TcuvidCtxLockDestroy cuvidCtxLockDestroy_ld;
extern TcuvidCreateVideoParser cuvidCreateVideoParser_ld;
extern TcuvidParseVideoData cuvidParseVideoData_ld;
extern TcuvidDestroyVideoParser cuvidDestroyVideoParser_ld;
// nvEncodeAPI64.dll
typedef NVENCSTATUS (*TNvEncodeAPICreateInstance)(
NV_ENCODE_API_FUNCTION_LIST *functionList);
typedef NVENCSTATUS (*TNvEncodeAPIGetMaxSupportedVersion)(uint32_t *version);
extern TNvEncodeAPICreateInstance NvEncodeAPICreateInstance_ld;
extern TNvEncodeAPIGetMaxSupportedVersion NvEncodeAPIGetMaxSupportedVersion_ld;
int LoadNvCodecDll();
int ReleaseNvCodecDll();
#endif

View File

@@ -228,7 +228,6 @@ int PeerConnection::CreateVideoCodec(bool hardware_acceleration) {
"MacOS not support hardware acceleration, use default software codec"); "MacOS not support hardware acceleration, use default software codec");
} }
#else #else
InitNvCodecApi();
#endif #endif
if (av1_encoding_) { if (av1_encoding_) {
@@ -237,7 +236,7 @@ int PeerConnection::CreateVideoCodec(bool hardware_acceleration) {
LOG_WARN("Only support software codec for AV1"); LOG_WARN("Only support software codec for AV1");
} else { } else {
if (hardware_acceleration_) { if (hardware_acceleration_) {
if (0 == InitNvCodecApi()) { if (0 == LoadNvCodecDll()) {
video_encoder_ = VideoEncoderFactory::CreateVideoEncoder(true, false); video_encoder_ = VideoEncoderFactory::CreateVideoEncoder(true, false);
video_decoder_ = VideoDecoderFactory::CreateVideoDecoder(true, false); video_decoder_ = VideoDecoderFactory::CreateVideoDecoder(true, false);
} else { } else {
@@ -614,6 +613,10 @@ int PeerConnection::Destroy() {
delete nv12_data_; delete nv12_data_;
nv12_data_ = nullptr; nv12_data_ = nullptr;
} }
if (hardware_acceleration_) {
ReleaseNvCodecDll();
}
return 0; return 0;
} }

File diff suppressed because it is too large Load Diff

View File

@@ -1,367 +0,0 @@
/*
* @Author: DI JUNKUN
* @Date: 2024-08-12
* Copyright (c) 2024 by DI JUNKUN, All Rights Reserved.
*/
#ifndef _NVCODEC_API_H_
#define _NVCODEC_API_H_
#include <Windows.h>
#include <iostream>
#include "cuda.h"
#include "cuviddec.h"
#include "nvEncodeAPI.h"
#include "nvcuvid.h"
typedef CUresult (*TcuInit)(unsigned int Flags);
typedef CUresult (*TcuDeviceGet)(CUdevice *device, int ordinal);
typedef CUresult (*TcuDeviceGetCount)(int *count);
typedef CUresult (*TcuCtxCreate)(CUcontext *pctx, unsigned int flags,
CUdevice dev);
typedef CUresult (*TcuGetErrorName)(CUresult error, const char **pStr);
typedef CUresult (*TcuCtxPushCurrent)(CUcontext ctx);
typedef CUresult (*TcuCtxPopCurrent)(CUcontext *pctx);
typedef CUresult (*TcuMemAlloc)(CUdeviceptr *dptr, size_t bytesize);
typedef CUresult (*TcuMemAllocPitch)(CUdeviceptr *dptr, size_t *pPitch,
size_t WidthInBytes, size_t Height,
unsigned int ElementSizeBytes);
typedef CUresult (*TcuMemFree)(CUdeviceptr dptr);
typedef CUresult (*TcuMemcpy2DAsync)(const CUDA_MEMCPY2D *pCopy,
CUstream hStream);
typedef CUresult (*TcuStreamSynchronize)(CUstream hStream);
typedef CUresult (*TcuMemcpy2D)(const CUDA_MEMCPY2D *pCopy);
typedef CUresult (*TcuMemcpy2DUnaligned)(const CUDA_MEMCPY2D *pCopy);
// API
static TcuInit cuInit_ld;
static TcuDeviceGet cuDeviceGet_ld;
static TcuDeviceGetCount cuDeviceGetCount_ld;
static TcuCtxCreate cuCtxCreate_ld;
static TcuGetErrorName cuGetErrorName_ld;
static TcuCtxPushCurrent cuCtxPushCurrent_ld;
static TcuCtxPopCurrent cuCtxPopCurrent_ld;
static TcuMemAlloc cuMemAlloc_ld;
static TcuMemAllocPitch cuMemAllocPitch_ld;
static TcuMemFree cuMemFree_ld;
static TcuMemcpy2DAsync cuMemcpy2DAsync_ld;
static TcuStreamSynchronize cuStreamSynchronize_ld;
static TcuMemcpy2D cuMemcpy2D_ld;
static TcuMemcpy2DUnaligned cuMemcpy2DUnaligned_ld;
//
typedef CUresult (*TcuvidCtxLockCreate)(CUvideoctxlock *pLock, CUcontext ctx);
typedef CUresult (*TcuvidGetDecoderCaps)(CUVIDDECODECAPS *pdc);
typedef CUresult (*TcuvidCreateDecoder)(CUvideodecoder *phDecoder,
CUVIDDECODECREATEINFO *pdci);
typedef CUresult (*TcuvidDestroyDecoder)(CUvideodecoder hDecoder);
typedef CUresult (*TcuvidDecodePicture)(CUvideodecoder hDecoder,
CUVIDPICPARAMS *pPicParams);
typedef CUresult (*TcuvidGetDecodeStatus)(CUvideodecoder hDecoder, int nPicIdx,
CUVIDGETDECODESTATUS *pDecodeStatus);
typedef CUresult (*TcuvidReconfigureDecoder)(
CUvideodecoder hDecoder, CUVIDRECONFIGUREDECODERINFO *pDecReconfigParams);
typedef CUresult (*TcuvidMapVideoFrame64)(CUvideodecoder hDecoder, int nPicIdx,
unsigned long long *pDevPtr,
unsigned int *pPitch,
CUVIDPROCPARAMS *pVPP);
typedef CUresult (*TcuvidUnmapVideoFrame64)(CUvideodecoder hDecoder,
unsigned long long DevPtr);
typedef CUresult (*TcuvidCtxLockDestroy)(CUvideoctxlock lck);
typedef CUresult (*TcuvidCreateVideoParser)(CUvideoparser *pObj,
CUVIDPARSERPARAMS *pParams);
typedef CUresult (*TcuvidParseVideoData)(CUvideoparser obj,
CUVIDSOURCEDATAPACKET *pPacket);
typedef CUresult (*TcuvidDestroyVideoParser)(CUvideoparser obj);
//
static TcuvidCtxLockCreate cuvidCtxLockCreate_ld;
static TcuvidGetDecoderCaps cuvidGetDecoderCaps_ld;
static TcuvidCreateDecoder cuvidCreateDecoder_ld;
static TcuvidDestroyDecoder cuvidDestroyDecoder_ld;
static TcuvidDecodePicture cuvidDecodePicture_ld;
static TcuvidGetDecodeStatus cuvidGetDecodeStatus_ld;
static TcuvidReconfigureDecoder cuvidReconfigureDecoder_ld;
static TcuvidMapVideoFrame64 cuvidMapVideoFrame64_ld;
static TcuvidUnmapVideoFrame64 cuvidUnmapVideoFrame64_ld;
static TcuvidCtxLockDestroy cuvidCtxLockDestroy_ld;
static TcuvidCreateVideoParser cuvidCreateVideoParser_ld;
static TcuvidParseVideoData cuvidParseVideoData_ld;
static TcuvidDestroyVideoParser cuvidDestroyVideoParser_ld;
//
typedef NVENCSTATUS (*TNvEncodeAPICreateInstance)(
NV_ENCODE_API_FUNCTION_LIST *functionList);
typedef NVENCSTATUS (*TNvEncodeAPIGetMaxSupportedVersion)(uint32_t *version);
//
static TNvEncodeAPICreateInstance NvEncodeAPICreateInstance_ld;
static TNvEncodeAPIGetMaxSupportedVersion NvEncodeAPIGetMaxSupportedVersion_ld;
static int InitNvCodecApi() {
// Load library
HMODULE nvcuda_dll = LoadLibrary(TEXT("nvcuda.dll"));
if (nvcuda_dll == NULL) {
std::cerr << "Unable to load nvcuda.dll!" << std::endl;
return -1;
}
cuInit_ld = (TcuInit)GetProcAddress(nvcuda_dll, "cuInit");
if (cuInit_ld == NULL) {
std::cerr << "Unable to find function!" << std::endl;
FreeLibrary(nvcuda_dll);
return -1;
}
cuDeviceGet_ld = (TcuDeviceGet)GetProcAddress(nvcuda_dll, "cuDeviceGet");
if (cuDeviceGet_ld == NULL) {
std::cerr << "Unable to find function!" << std::endl;
FreeLibrary(nvcuda_dll);
return -1;
}
cuDeviceGetCount_ld =
(TcuDeviceGetCount)GetProcAddress(nvcuda_dll, "cuDeviceGetCount");
if (cuDeviceGetCount_ld == NULL) {
std::cerr << "Unable to find function!" << std::endl;
FreeLibrary(nvcuda_dll);
return -1;
}
cuCtxCreate_ld = (TcuCtxCreate)GetProcAddress(nvcuda_dll, "cuCtxCreate");
if (cuCtxCreate_ld == NULL) {
std::cerr << "Unable to find function!" << std::endl;
FreeLibrary(nvcuda_dll);
return -1;
}
cuGetErrorName_ld =
(TcuGetErrorName)GetProcAddress(nvcuda_dll, "cuGetErrorName");
if (cuGetErrorName_ld == NULL) {
std::cerr << "Unable to find function!" << std::endl;
FreeLibrary(nvcuda_dll);
return -1;
}
cuCtxPushCurrent_ld =
(TcuCtxPushCurrent)GetProcAddress(nvcuda_dll, "cuCtxPushCurrent");
if (cuCtxPushCurrent_ld == NULL) {
std::cerr << "Unable to find function!" << std::endl;
FreeLibrary(nvcuda_dll);
return -1;
}
cuCtxPopCurrent_ld =
(TcuCtxPopCurrent)GetProcAddress(nvcuda_dll, "cuCtxPopCurrent");
if (cuCtxPopCurrent_ld == NULL) {
std::cerr << "Unable to find function!" << std::endl;
FreeLibrary(nvcuda_dll);
return -1;
}
cuMemAlloc_ld = (TcuMemAlloc)GetProcAddress(nvcuda_dll, "cuMemAlloc");
if (cuMemAlloc_ld == NULL) {
std::cerr << "Unable to find function!" << std::endl;
FreeLibrary(nvcuda_dll);
return -1;
}
cuMemAllocPitch_ld =
(TcuMemAllocPitch)GetProcAddress(nvcuda_dll, "cuMemAllocPitch");
if (cuMemAllocPitch_ld == NULL) {
std::cerr << "Unable to find function!" << std::endl;
FreeLibrary(nvcuda_dll);
return -1;
}
cuMemFree_ld = (TcuMemFree)GetProcAddress(nvcuda_dll, "cuMemFree");
if (cuMemFree_ld == NULL) {
std::cerr << "Unable to find function!" << std::endl;
FreeLibrary(nvcuda_dll);
return -1;
}
cuMemcpy2DAsync_ld =
(TcuMemcpy2DAsync)GetProcAddress(nvcuda_dll, "cuMemcpy2DAsync");
if (cuMemcpy2DAsync_ld == NULL) {
std::cerr << "Unable to find function!" << std::endl;
FreeLibrary(nvcuda_dll);
return -1;
}
cuStreamSynchronize_ld =
(TcuStreamSynchronize)GetProcAddress(nvcuda_dll, "cuStreamSynchronize");
if (cuStreamSynchronize_ld == NULL) {
std::cerr << "Unable to find function!" << std::endl;
FreeLibrary(nvcuda_dll);
return -1;
}
cuMemcpy2D_ld = (TcuMemcpy2D)GetProcAddress(nvcuda_dll, "cuMemcpy2D");
if (cuMemcpy2D_ld == NULL) {
std::cerr << "Unable to find function!" << std::endl;
FreeLibrary(nvcuda_dll);
return -1;
}
cuMemcpy2DUnaligned_ld =
(TcuMemcpy2DUnaligned)GetProcAddress(nvcuda_dll, "cuMemcpy2DUnaligned");
if (cuMemcpy2DUnaligned_ld == NULL) {
std::cerr << "Unable to find function!" << std::endl;
FreeLibrary(nvcuda_dll);
return -1;
}
//
HMODULE nvcuvid_dll = LoadLibrary(TEXT("nvcuvid.dll"));
if (nvcuvid_dll == NULL) {
std::cerr << "Unable to load nvcuvid.dll!" << std::endl;
return -1;
}
cuvidCtxLockCreate_ld =
(TcuvidCtxLockCreate)GetProcAddress(nvcuda_dll, "cuvidCtxLockCreate");
if (cuvidCtxLockCreate_ld == NULL) {
std::cerr << "Unable to find function!" << std::endl;
FreeLibrary(nvcuda_dll);
return -1;
}
cuvidGetDecoderCaps_ld =
(TcuvidGetDecoderCaps)GetProcAddress(nvcuda_dll, "cuvidGetDecoderCaps");
if (cuvidGetDecoderCaps_ld == NULL) {
std::cerr << "Unable to find function!" << std::endl;
FreeLibrary(nvcuda_dll);
return -1;
}
cuvidCreateDecoder_ld =
(TcuvidCreateDecoder)GetProcAddress(nvcuda_dll, "cuvidCreateDecoder");
if (cuvidCreateDecoder_ld == NULL) {
std::cerr << "Unable to find function!" << std::endl;
FreeLibrary(nvcuda_dll);
return -1;
}
cuvidDestroyDecoder_ld =
(TcuvidDestroyDecoder)GetProcAddress(nvcuda_dll, "cuvidDestroyDecoder");
if (cuvidDestroyDecoder_ld == NULL) {
std::cerr << "Unable to find function!" << std::endl;
FreeLibrary(nvcuda_dll);
return -1;
}
cuvidDecodePicture_ld =
(TcuvidDecodePicture)GetProcAddress(nvcuda_dll, "cuvidDecodePicture");
if (cuvidDecodePicture_ld == NULL) {
std::cerr << "Unable to find function!" << std::endl;
FreeLibrary(nvcuda_dll);
return -1;
}
cuvidGetDecodeStatus_ld =
(TcuvidGetDecodeStatus)GetProcAddress(nvcuda_dll, "cuvidGetDecodeStatus");
if (cuvidGetDecodeStatus_ld == NULL) {
std::cerr << "Unable to find function!" << std::endl;
FreeLibrary(nvcuda_dll);
return -1;
}
cuvidReconfigureDecoder_ld = (TcuvidReconfigureDecoder)GetProcAddress(
nvcuda_dll, "cuvidReconfigureDecoder");
if (cuvidReconfigureDecoder_ld == NULL) {
std::cerr << "Unable to find function!" << std::endl;
FreeLibrary(nvcuda_dll);
return -1;
}
cuvidMapVideoFrame64_ld =
(TcuvidMapVideoFrame64)GetProcAddress(nvcuda_dll, "cuvidMapVideoFrame64");
if (cuvidMapVideoFrame64_ld == NULL) {
std::cerr << "Unable to find function!" << std::endl;
FreeLibrary(nvcuda_dll);
return -1;
}
cuvidUnmapVideoFrame64_ld = (TcuvidUnmapVideoFrame64)GetProcAddress(
nvcuda_dll, "cuvidUnmapVideoFrame64");
if (cuvidUnmapVideoFrame64_ld == NULL) {
std::cerr << "Unable to find function!" << std::endl;
FreeLibrary(nvcuda_dll);
return -1;
}
cuvidCtxLockDestroy_ld =
(TcuvidCtxLockDestroy)GetProcAddress(nvcuda_dll, "cuvidCtxLockDestroy");
if (cuvidCtxLockDestroy_ld == NULL) {
std::cerr << "Unable to find function!" << std::endl;
FreeLibrary(nvcuda_dll);
return -1;
}
cuvidCreateVideoParser_ld = (TcuvidCreateVideoParser)GetProcAddress(
nvcuda_dll, "cuvidCreateVideoParser");
if (cuvidCreateVideoParser_ld == NULL) {
std::cerr << "Unable to find function!" << std::endl;
FreeLibrary(nvcuda_dll);
return -1;
}
cuvidParseVideoData_ld =
(TcuvidParseVideoData)GetProcAddress(nvcuda_dll, "cuvidParseVideoData");
if (cuvidParseVideoData_ld == NULL) {
std::cerr << "Unable to find function!" << std::endl;
FreeLibrary(nvcuda_dll);
return -1;
}
cuvidDestroyVideoParser_ld = (TcuvidDestroyVideoParser)GetProcAddress(
nvcuda_dll, "cuvidDestroyVideoParser");
if (cuvidDestroyVideoParser_ld == NULL) {
std::cerr << "Unable to find function!" << std::endl;
FreeLibrary(nvcuda_dll);
return -1;
}
//
HMODULE nvEncodeAPI64_dll = LoadLibrary(TEXT("nvEncodeAPI64.dll"));
if (nvEncodeAPI64_dll == NULL) {
std::cerr << "Unable to load nvEncodeAPI64.dll!" << std::endl;
return -1;
}
NvEncodeAPICreateInstance_ld = (TNvEncodeAPICreateInstance)GetProcAddress(
nvcuda_dll, "NvEncodeAPICreateInstance");
if (NvEncodeAPICreateInstance_ld == NULL) {
std::cerr << "Unable to find function!" << std::endl;
FreeLibrary(nvcuda_dll);
return -1;
}
NvEncodeAPIGetMaxSupportedVersion_ld =
(TNvEncodeAPIGetMaxSupportedVersion)GetProcAddress(
nvcuda_dll, "NvEncodeAPIGetMaxSupportedVersion");
if (NvEncodeAPIGetMaxSupportedVersion_ld == NULL) {
std::cerr << "Unable to find function!" << std::endl;
FreeLibrary(nvcuda_dll);
return -1;
}
return 0;
}
#endif

View File

@@ -21,9 +21,7 @@ add_packages("vcpkg::libnice")
includes("thirdparty") includes("thirdparty")
if is_os("windows") then if is_os("windows") then
add_packages("cuda")
add_defines("_WEBSOCKETPP_CPP11_INTERNAL_") add_defines("_WEBSOCKETPP_CPP11_INTERNAL_")
add_requires("cuda")
elseif is_os("linux") then elseif is_os("linux") then
add_requires("glib", {system = true}) add_requires("glib", {system = true})
add_packages("glib", "cuda") add_packages("glib", "cuda")
@@ -113,7 +111,8 @@ target("media")
"src/media/video/encode/openh264/*.cpp", "src/media/video/encode/openh264/*.cpp",
"src/media/video/decode/openh264/*.cpp", "src/media/video/decode/openh264/*.cpp",
"src/media/video/encode/aom/*.cpp", "src/media/video/encode/aom/*.cpp",
"src/media/video/decode/dav1d/*.cpp") "src/media/video/decode/dav1d/*.cpp",
"src/media/nvcodec/*.cpp")
add_includedirs("src/media/video/encode", add_includedirs("src/media/video/encode",
"src/media/video/decode", "src/media/video/decode",
"src/media/video/encode/nvcodec", "src/media/video/encode/nvcodec",
@@ -122,6 +121,7 @@ target("media")
"src/media/video/decode/openh264", "src/media/video/decode/openh264",
"src/media/video/encode/aom", "src/media/video/encode/aom",
"src/media/video/decode/dav1d", "src/media/video/decode/dav1d",
"src/media/nvcodec",
"thirdparty/nvcodec/Interface", "thirdparty/nvcodec/Interface",
"thirdparty/nvcodec/Samples", {public = true}) "thirdparty/nvcodec/Samples", {public = true})
elseif is_os(("linux")) then elseif is_os(("linux")) then
@@ -132,7 +132,8 @@ target("media")
"src/media/video/encode/openh264/*.cpp", "src/media/video/encode/openh264/*.cpp",
"src/media/video/decode/openh264/*.cpp", "src/media/video/decode/openh264/*.cpp",
"src/media/video/encode/aom/*.cpp", "src/media/video/encode/aom/*.cpp",
"src/media/video/decode/dav1d/*.cpp") "src/media/video/decode/dav1d/*.cpp",
"src/media/nvcodec/*.cpp")
add_includedirs("src/media/video/encode", add_includedirs("src/media/video/encode",
"src/media/video/decode", "src/media/video/decode",
"src/media/video/encode/nvcodec", "src/media/video/encode/nvcodec",
@@ -141,6 +142,7 @@ target("media")
"src/media/video/decode/openh264", "src/media/video/decode/openh264",
"src/media/video/encode/aom", "src/media/video/encode/aom",
"src/media/video/decode/dav1d", "src/media/video/decode/dav1d",
"src/media/nvcodec",
"thirdparty/nvcodec/Interface", "thirdparty/nvcodec/Interface",
"thirdparty/nvcodec/Samples", {public = true}) "thirdparty/nvcodec/Samples", {public = true})
elseif is_os("macosx") then elseif is_os("macosx") then