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
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
|
#include <windows.h>
#include <stdio.h>
// Function pointer typedefs for CUDA functions
typedef int (*InitCUDAFunc)();
typedef void (*CleanupCUDAFunc)();
typedef void (*UpdateVerticesFunc)(float* vertices, int numVertices, float time, float majorRadius, float minorRadius);
typedef void (*UpdateColorsFunc)(float* colors, int numVertices, float time);
// Global variables
static HMODULE cudaDLL = NULL;
static InitCUDAFunc initCUDAPtr = NULL;
static CleanupCUDAFunc cleanupCUDAPtr = NULL;
static UpdateVerticesFunc updateVerticesPtr = NULL;
static UpdateColorsFunc updateColorsPtr = NULL;
static int dllLoaded = 0;
// Load CUDA DLL and get function pointers
static int loadCUDADLL() {
if (dllLoaded) return 1;
printf("Loading CUDA DLL...\n");
cudaDLL = LoadLibrary("cuda_export.dll");
if (!cudaDLL) {
printf("Failed to load cuda_export.dll\n");
return 0;
}
initCUDAPtr = (InitCUDAFunc)GetProcAddress(cudaDLL, "initCUDA");
cleanupCUDAPtr = (CleanupCUDAFunc)GetProcAddress(cudaDLL, "cleanupCUDA");
updateVerticesPtr = (UpdateVerticesFunc)GetProcAddress(cudaDLL, "updateTorusVerticesCUDA");
updateColorsPtr = (UpdateColorsFunc)GetProcAddress(cudaDLL, "updateTorusColorsCUDA");
if (!initCUDAPtr || !cleanupCUDAPtr || !updateVerticesPtr || !updateColorsPtr) {
printf("Failed to get function pointers from CUDA DLL\n");
FreeLibrary(cudaDLL);
cudaDLL = NULL;
return 0;
}
printf("CUDA DLL loaded successfully\n");
dllLoaded = 1;
return 1;
}
// Wrapper functions
int initCUDA() {
printf("initCUDA wrapper called\n");
if (!loadCUDADLL()) {
printf("Failed to load CUDA DLL in initCUDA\n");
return 0;
}
printf("Calling CUDA initCUDA function\n");
int result = initCUDAPtr();
printf("CUDA initCUDA returned: %d\n", result);
return result;
}
void cleanupCUDA() {
printf("cleanupCUDA wrapper called\n");
if (dllLoaded && cleanupCUDAPtr) {
cleanupCUDAPtr();
}
if (cudaDLL) {
FreeLibrary(cudaDLL);
cudaDLL = NULL;
dllLoaded = 0;
}
printf("CUDA cleanup completed\n");
}
void updateTorusVerticesCUDA(float* vertices, int numVertices, float time, float majorRadius, float minorRadius) {
// printf("updateTorusVerticesCUDA wrapper called with %d vertices\n", numVertices);
if (dllLoaded && updateVerticesPtr) {
updateVerticesPtr(vertices, numVertices, time, majorRadius, minorRadius);
// printf("CUDA vertex update completed\n");
} else {
printf("CUDA DLL not loaded, skipping vertex update\n");
}
}
void updateTorusColorsCUDA(float* colors, int numVertices, float time) {
// printf("updateTorusColorsCUDA wrapper called with %d vertices\n", numVertices);
if (dllLoaded && updateColorsPtr) {
updateColorsPtr(colors, numVertices, time);
// printf("CUDA color update completed\n");
} else {
printf("CUDA DLL not loaded, skipping color update\n");
}
}
|
