/* Wrap one class */
int vtkWrapPython_WrapOneClass(
FILE *fp, const char *module, const char *classname,
ClassInfo *data, FileInfo *finfo, HierarchyInfo *hinfo,
int is_vtkobject)
{
int class_has_new = 0;
int i;
/* recursive handling of templated classes */
if (data->Template)
{
return vtkWrapPython_WrapTemplatedClass(fp, data, finfo, hinfo);
}
/* verify wrappability */
if (!is_vtkobject && !vtkWrapPython_IsSpecialTypeWrappable(data))
{
return 0;
}
/* declare items to be exported or imported */
if (is_vtkobject)
{
vtkWrapPython_ExportVTKClass(fp, data, hinfo);
}
/* prototype for the docstring function */
fprintf(fp,
"\n"
"static const char **Py%s_Doc();\n"
"\n",
classname);
/* check for New() function */
for (i = 0; i < data->NumberOfFunctions; i++)
{
if (data->Functions[i]->Name &&
data->Functions[i]->Access == VTK_ACCESS_PUBLIC &&
strcmp("New",data->Functions[i]->Name) == 0 &&
data->Functions[i]->NumberOfParameters == 0)
{
class_has_new = 1;
}
}
/* create any enum types defined in the class */
for (i = 0; i < data->NumberOfEnums; i++)
{
if (data->Enums[i]->Access == VTK_ACCESS_PUBLIC)
{
vtkWrapPython_GenerateEnumType(
fp, module, classname, data->Enums[i]);
}
}
/* now output all the methods are wrappable */
vtkWrapPython_GenerateMethods(
fp, classname, data, finfo, hinfo, is_vtkobject, 0);
/* output the class initilization function for VTK objects */
if (is_vtkobject)
{
vtkWrapPython_GenerateObjectType(
fp, module, classname);
vtkWrapPython_GenerateObjectNew(
fp, classname, data, hinfo, class_has_new);
}
/* output the class initilization function for special objects */
else
{
vtkWrapPython_GenerateSpecialType(
fp, module, classname, data, finfo, hinfo);
}
/* the docstring for the class, as a static var ending in "Doc" */
fprintf(fp,
"const char **Py%s_Doc()\n"
"{\n"
" static const char *docstring[] = {\n",
classname);
vtkWrapPython_ClassDoc(fp, finfo, data, hinfo, is_vtkobject);
fprintf(fp,
" NULL\n"
" };\n"
"\n"
" return docstring;\n"
"}\n"
"\n");
return 1;
}
int main(int argc, char *argv[])
{
ClassInfo *wrappedClasses[MAX_WRAPPED_CLASSES];
unsigned char wrapAsVTKObject[MAX_WRAPPED_CLASSES];
ClassInfo *data = NULL;
NamespaceInfo *contents;
OptionInfo *options;
HierarchyInfo *hinfo = NULL;
FileInfo *file_info;
FILE *fp;
const char *module = "vtkCommonCore";
const char *name;
char *name_from_file = NULL;
int numberOfWrappedClasses = 0;
int numberOfWrappedNamespaces = 0;
int wrapped_anything = 0;
int i, j;
size_t k, m;
int is_vtkobject;
/* pre-define a macro to identify the language */
vtkParse_DefineMacro("__VTK_WRAP_PYTHON__", 0);
/* get command-line args and parse the header file */
file_info = vtkParse_Main(argc, argv);
/* get the command-line options */
options = vtkParse_GetCommandLineOptions();
/* get the output file */
fp = fopen(options->OutputFileName, "w");
if (!fp)
{
fprintf(stderr, "Error opening output file %s\n", options->OutputFileName);
exit(1);
}
/* get the hierarchy info for accurate typing */
if (options->HierarchyFileNames)
{
hinfo = vtkParseHierarchy_ReadFiles(
options->NumberOfHierarchyFileNames, options->HierarchyFileNames);
}
/* get the filename without the extension */
name = file_info->FileName;
m = strlen(name);
for (k = m; k > 0; k--)
{
if (name[k] == '.') { break; }
}
if (k > 0) { m = k; }
for (k = m; k > 0; k--)
{
if (!((name[k-1] >= 'a' && name[k-1] <= 'z') ||
(name[k-1] >= 'A' && name[k-1] <= 'Z') ||
(name[k-1] >= '0' && name[k-1] <= '9') ||
name[k-1] == '_')) { break; }
}
name_from_file = (char *)malloc(m - k + 1);
strncpy(name_from_file, &name[k], m - k);
name_from_file[m-k] = '\0';
name = name_from_file;
/* get the global namespace */
contents = file_info->Contents;
/* use the hierarchy file to expand typedefs */
if (hinfo)
{
for (i = 0; i < contents->NumberOfClasses; i++)
{
vtkWrap_ApplyUsingDeclarations(contents->Classes[i], file_info, hinfo);
}
for (i = 0; i < contents->NumberOfClasses; i++)
{
vtkWrap_ExpandTypedefs(contents->Classes[i], file_info, hinfo);
}
}
/* the VTK_WRAPPING_CXX tells header files where they're included from */
fprintf(fp,
"// python wrapper for %s\n//\n"
"#define VTK_WRAPPING_CXX\n",
name);
/* unless this is vtkObjectBase.h, define VTK_STREAMS_FWD_ONLY */
if (strcmp("vtkObjectBase", name) != 0)
{
/* Block inclusion of full streams. */
fprintf(fp,
"#define VTK_STREAMS_FWD_ONLY\n");
}
/* lots of important utility functions are defined in vtkPythonArgs.h */
fprintf(fp,
"#include \"vtkPythonArgs.h\"\n"
"#include \"vtkPythonOverload.h\"\n"
"#include \"vtkConfigure.h\"\n"
"#include <cstddef>\n"
"#include <sstream>\n");
/* vtkPythonCommand is needed to wrap vtkObject.h */
if (strcmp("vtkObject", name) == 0)
{
fprintf(fp,
"#include \"vtkPythonCommand.h\"\n");
}
/* generate includes for any special types that are used */
vtkWrapPython_GenerateSpecialHeaders(fp, file_info, hinfo);
/* the header file for the wrapped class */
fprintf(fp,
"#include \"%s.h\"\n\n",
name);
/* do the export of the main entry point */
fprintf(fp,
"extern \"C\" { %s void PyVTKAddFile_%s(PyObject *); }\n",
"VTK_ABI_EXPORT", name);
/* get the module that is being wrapped */
data = file_info->MainClass;
if (!data && file_info->Contents->NumberOfClasses > 0)
{
data = file_info->Contents->Classes[0];
}
if (data && hinfo)
{
module = vtkWrapPython_ClassModule(hinfo, data->Name);
}
/* Identify all enum types that are used by methods */
vtkWrapPython_MarkAllEnums(file_info->Contents, hinfo);
/* Wrap any enum types defined in the global namespace */
for (i = 0; i < contents->NumberOfEnums; i++)
{
vtkWrapPython_GenerateEnumType(fp, module, NULL, contents->Enums[i]);
}
/* Wrap any namespaces */
for (i = 0; i < contents->NumberOfNamespaces; i++)
{
if (contents->Namespaces[i]->NumberOfConstants > 0)
{
vtkWrapPython_WrapNamespace(fp, module, contents->Namespaces[i]);
numberOfWrappedNamespaces++;
}
}
/* Check for all special classes before any classes are wrapped */
for (i = 0; i < contents->NumberOfClasses; i++)
{
data = contents->Classes[i];
/* guess whether type is a vtkobject */
is_vtkobject = (data == file_info->MainClass ? 1 : 0);
if (hinfo)
{
is_vtkobject = vtkWrap_IsTypeOf(hinfo, data->Name, "vtkObjectBase");
}
if (!is_vtkobject)
{
/* mark class as abstract only if it has pure virtual methods */
/* (does not check for inherited pure virtual methods) */
data->IsAbstract = 0;
for (j = 0; j < data->NumberOfFunctions; j++)
{
FunctionInfo *func = data->Functions[j];
if (func && func->IsPureVirtual)
{
data->IsAbstract = 1;
break;
}
}
}
wrapAsVTKObject[i] = (is_vtkobject ? 1 : 0);
}
/* Wrap all of the classes in the file */
for (i = 0; i < contents->NumberOfClasses; i++)
{
data = contents->Classes[i];
is_vtkobject = wrapAsVTKObject[i];
/* if "hinfo" is present, wrap everything, else just the main class */
if (hinfo || data == file_info->MainClass)
{
if (vtkWrapPython_WrapOneClass(
fp, module, data->Name, data, file_info, hinfo, is_vtkobject))
{
/* re-index wrapAsVTKObject for wrapped classes */
wrapAsVTKObject[numberOfWrappedClasses] = (is_vtkobject ? 1 : 0);
wrappedClasses[numberOfWrappedClasses++] = data;
}
}
}
/* The function for adding everything to the module dict */
wrapped_anything = (numberOfWrappedClasses ||
numberOfWrappedNamespaces ||
contents->NumberOfConstants);
fprintf(fp,
"void PyVTKAddFile_%s(\n"
" PyObject *%s)\n"
"{\n"
"%s",
name,
(wrapped_anything ? "dict" : ""),
(wrapped_anything ? " PyObject *o;\n" : ""));
/* Add all of the namespaces */
for (j = 0; j < contents->NumberOfNamespaces; j++)
{
if (contents->Namespaces[j]->NumberOfConstants > 0)
{
fprintf(fp,
" o = PyVTKNamespace_%s();\n"
" if (o && PyDict_SetItemString(dict, \"%s\", o) != 0)\n"
" {\n"
" Py_DECREF(o);\n"
" }\n"
"\n",
contents->Namespaces[j]->Name,
contents->Namespaces[j]->Name);
}
}
/* Add all of the classes that have been wrapped */
for (i = 0; i < numberOfWrappedClasses; i++)
{
data = wrappedClasses[i];
is_vtkobject = wrapAsVTKObject[i];
if (data->Template)
{
/* Template generator */
fprintf(fp,
" o = Py%s_TemplateNew();\n"
"\n",
data->Name);
/* Add template specializations to dict */
fprintf(fp,
" if (o)\n"
" {\n"
" PyObject *l = PyObject_CallMethod(o, (char *)\"values\", 0);\n"
" Py_ssize_t n = PyList_GET_SIZE(l);\n"
" for (Py_ssize_t i = 0; i < n; i++)\n"
" {\n"
" PyObject *ot = PyList_GET_ITEM(l, i);\n"
" const char *nt = NULL;\n"
" if (PyType_Check(ot))\n"
" {\n"
" nt = ((PyTypeObject *)ot)->tp_name;\n"
" }\n"
" else if (PyCFunction_Check(ot))\n"
" {\n"
" nt = ((PyCFunctionObject *)ot)->m_ml->ml_name;\n"
" }\n"
" if (nt)\n"
" {\n"
" PyDict_SetItemString(dict, nt, ot);\n"
" }\n"
" }\n"
" Py_DECREF(l);\n"
" }\n"
"\n");
}
else if (is_vtkobject)
{
/* Class is derived from vtkObjectBase */
fprintf(fp,
" o = Py%s_ClassNew();\n"
"\n",
data->Name);
}
else
{
/* Classes that are not derived from vtkObjectBase */
fprintf(fp,
" o = Py%s_TypeNew();\n"
"\n",
data->Name);
}
fprintf(fp,
" if (o && PyDict_SetItemString(dict, \"%s\", o) != 0)\n"
" {\n"
" Py_DECREF(o);\n"
" }\n"
"\n",
data->Name);
}
/* add any enum types defined in the file */
vtkWrapPython_AddPublicEnumTypes(fp, " ", "dict", "o", contents);
/* add any constants defined in the file */
vtkWrapPython_AddPublicConstants(fp, " ", "dict", "o", contents);
/* close the AddFile function */
fprintf(fp,
"}\n\n");
free(name_from_file);
vtkParse_Free(file_info);
return 0;
}
