forked from Kitware/VTK
/
vtkWrapPythonInit.c
135 lines (113 loc) · 3.35 KB
/
vtkWrapPythonInit.c
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
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
#include <stdio.h>
#include <stdlib.h>
#include <string.h>
/* warning this code is also in getclasses.cxx under pcmaker */
/* this roputine creates the init file */
static void CreateInitFile(const char *libName,
int numFiles, char **files,
FILE *fout)
{
int i;
const char *prefix = "";
const char* dllexp = "VTK_ABI_EXPORT ";
fprintf(fout,"// Generated by vtkWrapPythonInit in VTK/Wrapping\n");
fprintf(fout,"#include \"vtkPython.h\"\n\n");
fprintf(fout,"#include \"vtkSystemIncludes.h\"\n");
fprintf(fout,"#include <string.h>\n");
fprintf(fout,"// Handle compiler warning messages, etc.\n"
"#if defined( _MSC_VER ) && !defined(VTK_DISPLAY_WIN32_WARNINGS)\n"
"#pragma warning ( disable : 4706 )\n"
"#endif // Windows Warnings \n\n");
for (i = 0; i < numFiles; i++)
{
fprintf(fout,"extern \"C\" {%sPyObject *PyVTKAddFile_%s(PyObject *, const char *); }\n", dllexp, files[i]);
}
fprintf(fout,"\nstatic PyMethodDef Py%s_ClassMethods[] = {\n", libName);
fprintf(fout,"{NULL, NULL, 0, NULL}};\n\n");
fprintf(fout,"static void real_init%s(const char *modulename);\n\n", libName);
for (;;)
{
fprintf(fout,"extern \"C\" { %svoid init%s%s(); }\n\n", dllexp, prefix, libName);
fprintf(fout,"void init%s%s()\n{\n", prefix, libName);
fprintf(fout," static const char modulename[] = \"%s%s\";\n", prefix, libName);
fprintf(fout," real_init%s(modulename);\n}\n\n", libName);
#if defined(__CYGWIN__) || !defined(_WIN32)
/* add a "lib" prefix for compatibility with old python scripts */
if (strcmp(prefix, "lib") != 0)
{
prefix = "lib";
continue;
}
#endif
break;
}
fprintf(fout,"void real_init%s(const char *modulename)\n{\n", libName);
/* module init function */
fprintf(fout," PyObject *m, *d;\n\n");
fprintf(fout," m = Py_InitModule((char*)modulename, Py%s_ClassMethods);\n",
libName);
fprintf(fout," d = PyModule_GetDict(m);\n");
fprintf(fout," if (!d)\n");
fprintf(fout," {\n");
fprintf(fout," Py_FatalError((char*)\"can't get dictionary for module %s\");\n",
libName);
fprintf(fout," }\n");
for (i = 0; i < numFiles; i++)
{
fprintf(fout," PyVTKAddFile_%s(d, modulename);\n",
files[i]);
}
fprintf(fout,"}\n\n");
}
int main(int argc,char *argv[])
{
FILE *file;
FILE *fout;
int numFiles = 0;
char libName[250];
char tmpVal[250];
char *files[4000];
if (argc < 3)
{
fprintf(stderr,"Usage: %s input_file output_file\n",argv[0]);
return 1;
}
file = fopen(argv[1],"r");
if (!file)
{
fprintf(stderr,"Input file %s could not be opened\n",argv[1]);
return 1;
}
fout = fopen(argv[2],"w");
if (!fout)
{
fclose(file);
return 1;
}
/* read the info from the file */
if (fscanf(file,"%s",libName) != 1)
{
fprintf(stderr,"Error getting libName\n");
fclose(file);
fclose(fout);
return 1;
}
/* read in the classes */
while (fscanf(file,"%s",tmpVal) != EOF)
{
files[numFiles] = strdup(tmpVal);
numFiles++;
}
/* close the file */
fclose(file);
/* extra object for the Common module */
if (strcmp(libName, "vtkCommonCorePython") == 0)
{
/* the PyVTKMutableObject */
files[numFiles] = strdup("mutable");
numFiles++;
}
CreateInitFile(libName, numFiles, files, fout);
fclose(fout);
return 0;
}