PyImport Namespace Reference

Functions
std::unique_ptr< MultiLayer >	createFromPython (const std::string &script, const std::string &functionName, const std::string &path="")
	Creates a multi layer by running python code in embedded interpreter. More...
std::vector< std::string >	listOfFunctions (const std::string &script, const std::string &path="")
	Returns list of functions defined in the script. More...

Function Documentation

createFromPython()

std::unique_ptr< MultiLayer > PyImport::createFromPython	(	const std::string &	script,
		const std::string &	functionName,
		const std::string &	path = ""
	)

Creates a multi layer by running python code in embedded interpreter.

Parameters

script	Python script
functionName	A function name in this script which produces a MultiLayer
path	A path to import BornAgain library. If empty, relies on PYTHONPATH

Definition at line 34 of file PyImport.cpp.

{
    PyUtils::import_bornagain(path);
 
    PyObject* pCompiledFn = Py_CompileString(script.c_str(), "", Py_file_input);
    if (!pCompiledFn)
        throw std::runtime_error(error_description("Can't compile snippet"));
 
    // create a module
    PyObject* pModule = PyImport_ExecCodeModule((char*)"test", pCompiledFn);
    if (!pModule) {
        Py_DecRef(pCompiledFn);
        throw std::runtime_error(error_description("Can't exec module"));
    }
 
    // locate the "get_simulation" function (it's an attribute of the module)
    PyObject* pAddFn = PyObject_GetAttrString(pModule, functionName.c_str());
    if (!pAddFn)
        throw std::runtime_error("Can't locate compiled function");
 
    PyObject* instance = PyObject_CallFunctionObjArgs(pAddFn, NULL);
    if (!instance) {
        Py_DecRef(pAddFn);
        Py_DecRef(pModule);
        Py_DecRef(pCompiledFn);
        throw std::runtime_error(error_description("Can't call function"));
    }
 
    // clean up
    Py_DecRef(pAddFn);
    Py_DecRef(pModule);
    Py_DecRef(pCompiledFn);
 
    void* argp1 = 0;
    swig_type_info* pTypeInfo = SWIG_TypeQuery("MultiLayer *");
 
    const int res = SWIG_ConvertPtr(instance, &argp1, pTypeInfo, 0);
    if (!SWIG_IsOK(res)) {
        Py_DecRef(instance);
        throw std::runtime_error("SWIG failed to extract a MultiLayer.");
    }
 
    MultiLayer* multilayer = reinterpret_cast<MultiLayer*>(argp1);
    std::unique_ptr<MultiLayer> result(multilayer->clone());
 
    Py_DecRef(instance);
 
    return result;
}

References MultiLayer::clone(), and PyUtils::import_bornagain().

Referenced by PyImportAssistant::createMultiLayer().

Here is the call graph for this function:

listOfFunctions()

std::vector< std::string > PyImport::listOfFunctions	(	const std::string &	script,
		const std::string &	path = ""
	)

Returns list of functions defined in the script.

Parameters

script	Python script
path	A path to import BornAgain library. If empty, relies on PYTHONPATH

Definition at line 86 of file PyImport.cpp.

{
    PyUtils::import_bornagain(path);
 
    PyObject* pCompiledFn = Py_CompileString(script.c_str(), "", Py_file_input);
    if (!pCompiledFn)
        throw std::runtime_error(error_description("Can't compile snippet"));
 
    // create a module
    PyObject* pModule = PyImport_ExecCodeModule((char*)"test", pCompiledFn);
    if (!pModule) {
        Py_DecRef(pCompiledFn);
        throw std::runtime_error(error_description("Can't exec module"));
    }
 
    PyObject* dict = PyModule_GetDict(pModule);
    if (!dict)
        throw std::runtime_error("Can't get dictionary from module");
 
    std::vector<std::string> result;
    PyObject *key, *value;
    Py_ssize_t pos = 0;
    while (PyDict_Next(dict, &pos, &key, &value)) {
        if (PyCallable_Check(value)) {
            std::string func_name = PyUtils::toString(key);
            if (func_name.find("__") == std::string::npos)
                result.push_back(func_name);
        }
    }
 
    Py_DecRef(dict);
    Py_DecRef(pModule);
    Py_DecRef(pCompiledFn);
 
    return result;
}

References PyUtils::import_bornagain(), and PyUtils::toString().

Referenced by PyImportAssistant::getPySampleFunctionName().

Here is the call graph for this function: