wrapper_generator_module.cpp File Reference

#include "openFileStream.h"
#include "header_generator.h"
#include "wrapper_generator_module.h"

Include dependency graph for wrapper_generator_module.cpp:

Go to the source code of this file.

Functions
void	project_wrapper_moduleAddObject (std::ofstream &fs, const PVecClassConfig &vecClassConfig)
	Create the Py_INCREF for all classes of the module.
bool	project_wrapper_moduleGeneratorMain (const PPath &fileName, const ProjectParam &projectParam)
	Create the main wrapper module.
void	project_wrapper_modulePyIncref (std::ofstream &fs, const PVecClassConfig &vecClassConfig)
	Create the Py_INCREF for all classes of the module.
void	project_wrapper_modulePyTypeReady (std::ofstream &fs, const PVecClassConfig &vecClassConfig)
	Create the PyType_Ready for all classes of the module.

Function Documentation

project_wrapper_moduleAddObject()

void project_wrapper_moduleAddObject	(	std::ofstream &	fs,
		const PVecClassConfig &	vecClassConfig )

Create the Py_INCREF for all classes of the module.

Parameters

[out]	fs	: file to be modified
	vecClassConfig	: vector of class configuration

Definition at line 52 of file wrapper_generator_module.cpp.

                                                                                              {
        for(PVecClassConfig::const_iterator it(vecClassConfig.begin()); it != vecClassConfig.end(); ++it){
                if(it->getIsEnum()){
                        fs << "\tPyModule_AddObject(m, \""+it->getName()+"\", "+it->getName()+"EnumType);\n";
                }else{
                        fs << "\tPyModule_AddObject(m, \""+it->getName()+"\", (PyObject *)&WP"+it->getName()+"Type);\n";
                }
        }
}

Referenced by project_wrapper_moduleGeneratorMain().

Here is the caller graph for this function:

project_wrapper_moduleGeneratorMain()

bool project_wrapper_moduleGeneratorMain	(	const PPath &	fileName,
		const ProjectParam &	projectParam )

Create the main wrapper module.

Parameters

fileName	: path of the module
projectParam	: config of the project

Returns

true on success, false otherwise

Definition at line 67 of file wrapper_generator_module.cpp.

                                                                                                   {
        PString moduleName(projectParam.mode.moduleName);
        std::ofstream fs;
        if(!openFileStream(fs, fileName)){return false;}
        licenceSave(fs);
        fs << "\n";
//      fs << "#define NPY_NO_DEPRECATED_API NPY_1_7_API_VERSION\n";
//      fs << "#ifndef DISABLE_COOL_ARRAY\n";
//      fs << "#\tdefine PY_ARRAY_UNIQUE_SYMBOL "+moduleName+"_ARRAY_API\n";
//      fs << "#endif\n";
        fs << "\n";
        fs << "#include <Python.h>\n";
        fs << "#include \"structmember.h\"\n";
//      fs << "#include <numpy/arrayobject.h>\n";
        fs << "\n";
        fs << "#include <string>\n";
        fs << "#include <iostream>\n";
        //TODO : put the includes of all class definitions
        for(const PDataConfig & config : projectParam.vecDataConfig){
                fs << "#include \"" << config.getFileName().getFileName().eraseExtension() <<"_wrapper.h\"\n";
        }
        fs << "\n";
//      fs << "#include \""+ baseImplInclude + "_wrapper.h\"\n\n";
        
        fs << "static PyMethodDef _"+moduleName+"_methods[] = {\n";
        fs << "\t{NULL, NULL}\n";
        fs << "};\n\n";
 
        fs << "static PyModuleDef _"+moduleName+"_module = {\n";
        fs << "\tPyModuleDef_HEAD_INIT,\n";
        fs << "\t\""+moduleName+"\",\n";
        fs << "\t\"\",\n";
        fs << "\t-1,\n";
        fs << "\t_"+moduleName+"_methods,\n";
        fs << "\tNULL,\n";
        fs << "\tNULL,\n";
        fs << "\tNULL,\n";
        fs << "\tNULL\n";
        fs << "};\n\n";
        
        fs << "///Create the python module "+moduleName+"\n";
        fs << "/**\t@return python module "+moduleName+"\n";
        fs << "*/\n";
        fs << "PyMODINIT_FUNC PyInit_"+moduleName+"(void){\n";
//      fs << "\timport_array();\n";    //For numpy stuff
        fs << "\n";
        for(const PDataConfig & config : projectParam.vecDataConfig){
                project_wrapper_modulePyTypeReady(fs, config.getVecClassConfig());
        }
        fs << "\t\n";
        fs << "\tPyObject * m = PyModule_Create(&_"+moduleName+"_module);\n";
        fs << "\tif(m == NULL){\n";
        fs << "\t\treturn NULL;\n";
        fs << "\t}\n";
        fs << "\t\n";
        for(const PDataConfig & config : projectParam.vecDataConfig){
                project_wrapper_modulePyIncref(fs, config.getVecClassConfig());
        }
        fs << "\t\n";
        for(const PDataConfig & config : projectParam.vecDataConfig){
                project_wrapper_moduleAddObject(fs, config.getVecClassConfig());
        }
        fs << "\t\n";
        fs << "\treturn m;\n";
        fs << "}\n\n";
        fs.close();
        return true;
}

References PDataConfig::getFileName(), PDataConfig::getVecClassConfig(), licenceSave(), ProjectParam::mode, GeneratorMode::moduleName, project_wrapper_moduleAddObject(), project_wrapper_modulePyIncref(), project_wrapper_modulePyTypeReady(), and ProjectParam::vecDataConfig.

Referenced by wrapper_generator_class_full().

Here is the call graph for this function:

Here is the caller graph for this function:

project_wrapper_modulePyIncref()

void project_wrapper_modulePyIncref	(	std::ofstream &	fs,
		const PVecClassConfig &	vecClassConfig )

Create the Py_INCREF for all classes of the module.

Parameters

[out]	fs	: file to be modified
	vecClassConfig	: vector of class configuration

Definition at line 38 of file wrapper_generator_module.cpp.

                                                                                             {
        for(PVecClassConfig::const_iterator it(vecClassConfig.begin()); it != vecClassConfig.end(); ++it){
                if(it->getIsEnum()){
                        fs << "\t//Py_INCREF("+it->getName()+"EnumType);\t//2: An other segmentation fault of python, when pytest is collecting tests\n";
                }else{
                        fs << "\tPy_INCREF(&WP"+it->getName()+"Type);\n";
                }
        }
}

Referenced by project_wrapper_moduleGeneratorMain().

Here is the caller graph for this function:

project_wrapper_modulePyTypeReady()

void project_wrapper_modulePyTypeReady	(	std::ofstream &	fs,
		const PVecClassConfig &	vecClassConfig )

Create the PyType_Ready for all classes of the module.

Parameters

[out]	fs	: file to be modified
	vecClassConfig	: vector of class configuration

Definition at line 17 of file wrapper_generator_module.cpp.

                                                                                                {
        for(PVecClassConfig::const_iterator it(vecClassConfig.begin()); it != vecClassConfig.end(); ++it){
                if(it->getIsEnum()){
                        std::string varName(it->getName()+"EnumType");
                        fs << "\t//PyObject* "+varName+" = (PyObject*)Py_TYPE(WP"+it->getName()+"_newC(NULL, NULL, NULL));\t//1: A nice segmentation fault of python\n";
                        fs << "\tPyObject* "+varName+" = WP"+it->getName()+"_newC(NULL, NULL, NULL);\t//2: alone => SystemError: initialization of pystereodata raised unreported exception, while importing generated module\n";
                        fs << "\tif("+varName+" == NULL){" << std::endl;
                        fs << "\t\tstd::cerr << \"Cannot create enum '"<<it->getName()<<"'\" << std::endl;" << std::endl;
                        fs << "\t\treturn NULL;" << std::endl;
                        fs << "\t}" << std::endl;
//                      fs << "\tif(PyType_Ready("+varName+") < 0){std::cerr << \"Type for enum '"<<it->getName()<<"' not ready\" << std::endl;return NULL;}\n";
                }else{
                        fs << "\tif(PyType_Ready(&WP"+it->getName()+"Type) < 0){return NULL;}\n";
                }
        }
}

Referenced by project_wrapper_moduleGeneratorMain().

Here is the caller graph for this function: