mirror of
https://repo.dec05eba.com/gpu-screen-recorder
synced 2026-05-05 22:40:42 +09:00
Remove dependency on glew and glfw, move external files to external directory
This commit is contained in:
dec05eba
@@ -1,42 +1,29 @@
|
||||
#pragma once
|
||||
|
||||
#include "LibraryLoader.hpp"
|
||||
|
||||
#include <cuda.h>
|
||||
#include <cudaGL.h>
|
||||
#include <dlfcn.h>
|
||||
#include <stdio.h>
|
||||
|
||||
typedef CUresult CUDAAPI (*CUINIT)(unsigned int Flags);
|
||||
typedef CUresult CUDAAPI (*CUDEVICEGETCOUNT)(int *count);
|
||||
typedef CUresult CUDAAPI (*CUDEVICEGET)(CUdevice *device, int ordinal);
|
||||
typedef CUresult CUDAAPI (*CUCTXCREATE_V2)(CUcontext *pctx, unsigned int flags, CUdevice dev);
|
||||
typedef CUresult CUDAAPI (*CUCTXPUSHCURRENT_V2)(CUcontext ctx);
|
||||
typedef CUresult CUDAAPI (*CUCTXPOPCURRENT_V2)(CUcontext *pctx);
|
||||
typedef CUresult CUDAAPI (*CUGETERRORSTRING)(CUresult error, const char **pStr);
|
||||
typedef CUresult CUDAAPI (*CUMEMSETD8_V2)(CUdeviceptr dstDevice, unsigned char uc, size_t N);
|
||||
typedef CUresult CUDAAPI (*CUMEMCPY2D_V2)(const CUDA_MEMCPY2D *pCopy);
|
||||
|
||||
typedef CUresult CUDAAPI (*CUGRAPHICSGLREGISTERIMAGE)(CUgraphicsResource *pCudaResource, GLuint image, GLenum target, unsigned int Flags);
|
||||
typedef CUresult CUDAAPI (*CUGRAPHICSRESOURCESETMAPFLAGS)(CUgraphicsResource resource, unsigned int flags);
|
||||
typedef CUresult CUDAAPI (*CUGRAPHICSMAPRESOURCES)(unsigned int count, CUgraphicsResource *resources, CUstream hStream);
|
||||
typedef CUresult CUDAAPI (*CUGRAPHICSUNREGISTERRESOURCE)(CUgraphicsResource resource);
|
||||
typedef CUresult CUDAAPI (*CUGRAPHICSSUBRESOURCEGETMAPPEDARRAY)(CUarray *pArray, CUgraphicsResource resource, unsigned int arrayIndex, unsigned int mipLevel);
|
||||
typedef struct CUgraphicsResource_st *CUgraphicsResource;
|
||||
|
||||
struct Cuda {
|
||||
CUINIT cuInit;
|
||||
CUDEVICEGETCOUNT cuDeviceGetCount;
|
||||
CUDEVICEGET cuDeviceGet;
|
||||
CUCTXCREATE_V2 cuCtxCreate_v2;
|
||||
CUCTXPUSHCURRENT_V2 cuCtxPushCurrent_v2;
|
||||
CUCTXPOPCURRENT_V2 cuCtxPopCurrent_v2;
|
||||
CUGETERRORSTRING cuGetErrorString;
|
||||
CUMEMSETD8_V2 cuMemsetD8_v2;
|
||||
CUMEMCPY2D_V2 cuMemcpy2D_v2;
|
||||
CUresult CUDAAPI (*cuInit)(unsigned int Flags);
|
||||
CUresult CUDAAPI (*cuDeviceGetCount)(int *count);
|
||||
CUresult CUDAAPI (*cuDeviceGet)(CUdevice *device, int ordinal);
|
||||
CUresult CUDAAPI (*cuCtxCreate_v2)(CUcontext *pctx, unsigned int flags, CUdevice dev);
|
||||
CUresult CUDAAPI (*cuCtxPushCurrent_v2)(CUcontext ctx);
|
||||
CUresult CUDAAPI (*cuCtxPopCurrent_v2)(CUcontext *pctx);
|
||||
CUresult CUDAAPI (*cuGetErrorString)(CUresult error, const char **pStr);
|
||||
CUresult CUDAAPI (*cuMemsetD8_v2)(CUdeviceptr dstDevice, unsigned char uc, size_t N);
|
||||
CUresult CUDAAPI (*cuMemcpy2D_v2)(const CUDA_MEMCPY2D *pCopy);
|
||||
|
||||
CUGRAPHICSGLREGISTERIMAGE cuGraphicsGLRegisterImage;
|
||||
CUGRAPHICSRESOURCESETMAPFLAGS cuGraphicsResourceSetMapFlags;
|
||||
CUGRAPHICSMAPRESOURCES cuGraphicsMapResources;
|
||||
CUGRAPHICSUNREGISTERRESOURCE cuGraphicsUnregisterResource;
|
||||
CUGRAPHICSSUBRESOURCEGETMAPPEDARRAY cuGraphicsSubResourceGetMappedArray;
|
||||
CUresult CUDAAPI (*cuGraphicsGLRegisterImage)(CUgraphicsResource *pCudaResource, unsigned int image, unsigned int target, unsigned int Flags);
|
||||
CUresult CUDAAPI (*cuGraphicsResourceSetMapFlags)(CUgraphicsResource resource, unsigned int flags);
|
||||
CUresult CUDAAPI (*cuGraphicsMapResources)(unsigned int count, CUgraphicsResource *resources, CUstream hStream);
|
||||
CUresult CUDAAPI (*cuGraphicsUnregisterResource)(CUgraphicsResource resource);
|
||||
CUresult CUDAAPI (*cuGraphicsSubResourceGetMappedArray)(CUarray *pArray, CUgraphicsResource resource, unsigned int arrayIndex, unsigned int mipLevel);
|
||||
|
||||
~Cuda() {
|
||||
if(library)
|
||||
@@ -52,80 +39,39 @@ struct Cuda {
|
||||
if(!lib) {
|
||||
lib = dlopen("libcuda.so.1", RTLD_LAZY);
|
||||
if(!lib) {
|
||||
fprintf(stderr, "Error: failed to load libcuda.so and libcuda.so.1, error: %s\n", dlerror());
|
||||
fprintf(stderr, "Error: failed to load libcuda.so/libcuda.so.1, error: %s\n", dlerror());
|
||||
return false;
|
||||
}
|
||||
}
|
||||
|
||||
cuInit = (CUINIT)load_symbol(lib, "cuInit");
|
||||
if(!cuInit)
|
||||
goto fail;
|
||||
dlsym_assign required_dlsym[] = {
|
||||
{ (void**)&cuInit, "cuInit" },
|
||||
{ (void**)&cuDeviceGetCount, "cuDeviceGetCount" },
|
||||
{ (void**)&cuDeviceGet, "cuDeviceGet" },
|
||||
{ (void**)&cuCtxCreate_v2, "cuCtxCreate_v2" },
|
||||
{ (void**)&cuCtxPushCurrent_v2, "cuCtxPushCurrent_v2" },
|
||||
{ (void**)&cuCtxPopCurrent_v2, "cuCtxPopCurrent_v2" },
|
||||
{ (void**)&cuGetErrorString, "cuGetErrorString" },
|
||||
{ (void**)&cuMemsetD8_v2, "cuMemsetD8_v2" },
|
||||
{ (void**)&cuMemcpy2D_v2, "cuMemcpy2D_v2" },
|
||||
|
||||
cuDeviceGetCount = (CUDEVICEGETCOUNT)load_symbol(lib, "cuDeviceGetCount");
|
||||
if(!cuDeviceGetCount)
|
||||
goto fail;
|
||||
{ (void**)&cuGraphicsGLRegisterImage, "cuGraphicsGLRegisterImage" },
|
||||
{ (void**)&cuGraphicsResourceSetMapFlags, "cuGraphicsResourceSetMapFlags" },
|
||||
{ (void**)&cuGraphicsMapResources, "cuGraphicsMapResources" },
|
||||
{ (void**)&cuGraphicsUnregisterResource, "cuGraphicsUnregisterResource" },
|
||||
{ (void**)&cuGraphicsSubResourceGetMappedArray, "cuGraphicsSubResourceGetMappedArray" },
|
||||
|
||||
cuDeviceGet = (CUDEVICEGET)load_symbol(lib, "cuDeviceGet");
|
||||
if(!cuDeviceGet)
|
||||
goto fail;
|
||||
{ NULL, NULL }
|
||||
};
|
||||
|
||||
cuCtxCreate_v2 = (CUCTXCREATE_V2)load_symbol(lib, "cuCtxCreate_v2");
|
||||
if(!cuCtxCreate_v2)
|
||||
goto fail;
|
||||
|
||||
cuCtxPushCurrent_v2 = (CUCTXPUSHCURRENT_V2)load_symbol(lib, "cuCtxPushCurrent_v2");
|
||||
if(!cuCtxPushCurrent_v2)
|
||||
goto fail;
|
||||
|
||||
cuCtxPopCurrent_v2 = (CUCTXPOPCURRENT_V2)load_symbol(lib, "cuCtxPopCurrent_v2");
|
||||
if(!cuCtxPopCurrent_v2)
|
||||
goto fail;
|
||||
|
||||
cuGetErrorString = (CUGETERRORSTRING)load_symbol(lib, "cuGetErrorString");
|
||||
if(!cuGetErrorString)
|
||||
goto fail;
|
||||
|
||||
cuMemsetD8_v2 = (CUMEMSETD8_V2)load_symbol(lib, "cuMemsetD8_v2");
|
||||
if(!cuMemsetD8_v2)
|
||||
goto fail;
|
||||
|
||||
cuMemcpy2D_v2 = (CUMEMCPY2D_V2)load_symbol(lib, "cuMemcpy2D_v2");
|
||||
if(!cuMemcpy2D_v2)
|
||||
goto fail;
|
||||
|
||||
cuGraphicsGLRegisterImage = (CUGRAPHICSGLREGISTERIMAGE)load_symbol(lib, "cuGraphicsGLRegisterImage");
|
||||
if(!cuGraphicsGLRegisterImage)
|
||||
goto fail;
|
||||
|
||||
cuGraphicsResourceSetMapFlags = (CUGRAPHICSRESOURCESETMAPFLAGS)load_symbol(lib, "cuGraphicsResourceSetMapFlags");
|
||||
if(!cuGraphicsResourceSetMapFlags)
|
||||
goto fail;
|
||||
|
||||
cuGraphicsMapResources = (CUGRAPHICSMAPRESOURCES)load_symbol(lib, "cuGraphicsMapResources");
|
||||
if(!cuGraphicsMapResources)
|
||||
goto fail;
|
||||
|
||||
cuGraphicsUnregisterResource = (CUGRAPHICSUNREGISTERRESOURCE)load_symbol(lib, "cuGraphicsUnregisterResource");
|
||||
if(!cuGraphicsUnregisterResource)
|
||||
goto fail;
|
||||
|
||||
cuGraphicsSubResourceGetMappedArray = (CUGRAPHICSSUBRESOURCEGETMAPPEDARRAY)load_symbol(lib, "cuGraphicsSubResourceGetMappedArray");
|
||||
if(!cuGraphicsSubResourceGetMappedArray)
|
||||
goto fail;
|
||||
|
||||
library = lib;
|
||||
return true;
|
||||
|
||||
fail:
|
||||
dlclose(lib);
|
||||
return false;
|
||||
}
|
||||
private:
|
||||
void* load_symbol(void *library, const char *symbol) {
|
||||
void *sym = dlsym(library, symbol);
|
||||
if(!sym)
|
||||
fprintf(stderr, "Error: missing required symbol %s from libcuda.so\n", symbol);
|
||||
return sym;
|
||||
if(dlsym_load_list(lib, required_dlsym)) {
|
||||
library = lib;
|
||||
return true;
|
||||
} else {
|
||||
fprintf(stderr, "Error: missing required symbols in libcuda.so\n");
|
||||
dlclose(lib);
|
||||
return false;
|
||||
}
|
||||
}
|
||||
private:
|
||||
void *library = nullptr;
|
||||
|
||||
Reference in New Issue
Block a user