wrapper_generator_module.cpp

Go to the documentation of this file.

/***************************************
        Auteur : Pierre Aubert
        Mail : pierre.aubert@lapp.in2p3.fr
        Licence : CeCILL-C
****************************************/
 
#include "openFileStream.h"
#include "header_generator.h"
 
#include "wrapper_generator_module.h"
 


void project_wrapper_modulePyTypeReady(std::ofstream & fs, const PVecClassConfig & vecClassConfig){
        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";
                }
        }
}


void project_wrapper_modulePyIncref(std::ofstream & fs, const PVecClassConfig & vecClassConfig){
        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";
                }
        }
}


void project_wrapper_moduleAddObject(std::ofstream & fs, const PVecClassConfig & vecClassConfig){
        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";
                }
        }
}


bool project_wrapper_moduleGeneratorMain(const PPath & fileName, const ProjectParam & projectParam){
        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;
}