Load cuda at runtime. Include cuda header files inside the project

1 files changed, 129 insertions, 0 deletions

diff --git a/include/CudaLibrary.hpp b/include/CudaLibrary.hpp
new file mode 100644
index 0000000..ac35113
--- /dev/null
+++ b/include/CudaLibrary.hpp
@@ -0,0 +1,129 @@
+#pragma once
+
+#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);
+
+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;
+
+    CUGRAPHICSGLREGISTERIMAGE cuGraphicsGLRegisterImage;
+    CUGRAPHICSRESOURCESETMAPFLAGS cuGraphicsResourceSetMapFlags;
+    CUGRAPHICSMAPRESOURCES cuGraphicsMapResources;
+    CUGRAPHICSUNREGISTERRESOURCE cuGraphicsUnregisterResource;
+    CUGRAPHICSSUBRESOURCEGETMAPPEDARRAY cuGraphicsSubResourceGetMappedArray;
+
+    ~Cuda() {
+        if(library)
+            dlclose(library);
+    }
+
+    bool load() {
+        if(library)
+            return true;
+
+        dlerror(); // clear
+        void *lib = dlopen("libcuda.so", RTLD_LAZY);
+        if(!lib) {
+            fprintf(stderr, "Error: failed to load libcuda.so, error: %s\n", dlerror());
+            return false;
+        }
+
+        cuInit = (CUINIT)load_symbol(lib, "cuInit");
+        if(!cuInit)
+            goto fail;
+
+        cuDeviceGetCount = (CUDEVICEGETCOUNT)load_symbol(lib, "cuDeviceGetCount");
+        if(!cuDeviceGetCount)
+            goto fail;
+
+        cuDeviceGet = (CUDEVICEGET)load_symbol(lib, "cuDeviceGet");
+        if(!cuDeviceGet)
+            goto fail;
+
+        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;
+    }
+private:
+    void *library = nullptr;
+};
\ No newline at end of file