PyUtils Namespace Reference

Functions
PyObject *	createNumpyArray (const std::vector< double > &data)
void	import_bornagain (const std::string &path="")
	Imports BornAgain from given location. If path is empty, tries to rely on PYTHONPATH. More...
std::string	pythonRuntimeInfo ()
	Returns multi-line string representing PATH, PYTHONPATH, sys.path and other info. More...
std::string	pythonStackTrace ()
	Returns string representing python stack trace. More...
std::string	toString (char *c)
	Converts char to string. In the case of nullptr will return an empty string. More...
std::string	toString (PyObject *obj)
	Converts PyObject into string, if possible, or throws exception. More...
std::string	toString (wchar_t *c)
std::vector< std::string >	toVectorString (PyObject *obj)
	Converts PyObject into vector of strings, if possible, or throws exception. More...

Function Documentation

createNumpyArray()

PyObject * PyUtils::createNumpyArray

(

const std::vector< double > &

data

)

Definition at line 172 of file PyUtils.cpp.

{
    const size_t ndim(1);
    npy_int ndim_numpy = ndim;
    npy_intp* ndimsizes_numpy = new npy_intp[ndim];
    ndimsizes_numpy[0] = data.size();
 
    // creating standalone numpy array
    PyObject* pyarray = PyArray_SimpleNew(ndim_numpy, ndimsizes_numpy, NPY_DOUBLE);
    delete[] ndimsizes_numpy;
    if (pyarray == nullptr)
        throw std::runtime_error("ExportOutputData() -> Panic in PyArray_SimpleNew");
 
    // getting pointer to data buffer of numpy array
    double* array_buffer = (double*)PyArray_DATA((PyArrayObject*)pyarray);
 
    for (size_t index = 0; index < data.size(); ++index)
        *array_buffer++ = data[index];
 
    return pyarray;
}

Referenced by Histogram1D::binCentersNumpy(), Histogram1D::binErrorsNumpy(), and Histogram1D::binValuesNumpy().

import_bornagain()

void PyUtils::import_bornagain

(

const std::string &

path = ""

)

Imports BornAgain from given location. If path is empty, tries to rely on PYTHONPATH.

Definition at line 71 of file PyUtils.cpp.

{
    if (!Py_IsInitialized()) {
        Py_InitializeEx(0);
 
        if (!path.empty()) {
            PyObject* sysPath = PySys_GetObject((char*)"path");
            PyList_Append(sysPath, PyString_FromString(path.c_str()));
        }
 
        // Stores signal handler before numpy's mess it up.
        // This is to make ctrl-c working from terminal.
#ifndef _WIN32
        PyOS_sighandler_t sighandler;
        sighandler = PyOS_getsig(SIGINT);
#endif
        PyObject* pmod = PyImport_ImportModule("bornagain");
        if (!pmod) {
            PyErr_Print();
            throw std::runtime_error("Can't load bornagain");
        }
 
        // restores single handler to make ctr-c alive.
#ifndef _WIN32
        PyOS_setsig(SIGINT, sighandler);
#endif
    }
}

Referenced by PyImport::createFromPython(), and PyImport::listOfFunctions().

pythonRuntimeInfo()

std::string PyUtils::pythonRuntimeInfo

(

)

Returns multi-line string representing PATH, PYTHONPATH, sys.path and other info.

Definition at line 100 of file PyUtils.cpp.

{
    Py_InitializeEx(0);
 
    std::stringstream result;
 
    // Runtime environment
    result << std::string(60, '=') << "\n";
    result << "PATH: " << SysUtils::getenv("PATH") << "\n";
    result << "PYTHONPATH: " << SysUtils::getenv("PYTHONPATH") << "\n";
    result << "PYTHONHOME: " << SysUtils::getenv("PYTHONHOME") << "\n";
 
    // Embedded Python details
    result << "Py_GetProgramName(): " << PyUtils::toString(Py_GetProgramName()) << "\n";
    result << "Py_GetProgramFullPath(): " << PyUtils::toString(Py_GetProgramFullPath()) << "\n";
    result << "Py_GetPath(): " << PyUtils::toString(Py_GetPath()) << "\n";
    result << "Py_GetPythonHome(): " << PyUtils::toString(Py_GetPythonHome()) << "\n";
 
    // Runtime Python's sys.path
    PyObject* sysPath = PySys_GetObject((char*)"path");
    auto content = PyUtils::toVectorString(sysPath);
    result << "sys.path: ";
    for (auto s : content)
        result << s << ",";
    result << "\n";
 
    return result.str();
}

References SysUtils::getenv(), toString(), and toVectorString().

Referenced by pythonStackTrace().

Here is the call graph for this function:

pythonStackTrace()

std::string PyUtils::pythonStackTrace

(

)

Returns string representing python stack trace.

Definition at line 132 of file PyUtils.cpp.

{
    std::stringstream result;
 
    if (PyErr_Occurred()) {
        PyObject *ptype, *pvalue, *ptraceback, *pystr;
 
        PyErr_Fetch(&ptype, &pvalue, &ptraceback);
        pystr = PyObject_Str(pvalue);
        if (char* str = PyString_AsString(pystr))
            result << std::string(str) << "\n";
 
        PyObject* module_name = PyString_FromString("traceback");
        PyObject* pyth_module = PyImport_Import(module_name);
        Py_DecRef(module_name);
 
        if (pyth_module) {
            result << "\n";
            PyObject* pyth_func = PyObject_GetAttrString(pyth_module, "format_exception");
            if (pyth_func && PyCallable_Check(pyth_func)) {
                PyObject* pyth_val =
                    PyObject_CallFunctionObjArgs(pyth_func, ptype, pvalue, ptraceback, NULL);
                if (pyth_val) {
                    pystr = PyObject_Str(pyth_val);
                    if (char* str = PyString_AsString(pystr))
                        result << std::string(str);
                    Py_DecRef(pyth_val);
                }
            }
            result << "\n";
        }
    }
 
    result << "\n";
    result << pythonRuntimeInfo();
    result << "\n";
 
    return result.str();
}

References pythonRuntimeInfo().

Here is the call graph for this function:

toString() [1/3]

std::string PyUtils::toString

(

char *

c

)

Converts char to string. In the case of nullptr will return an empty string.

Definition at line 55 of file PyUtils.cpp.

{
    if (c)
        return c;
    else
        return "";
}

toString() [2/3]

std::string PyUtils::toString

(

PyObject *

obj

)

Converts PyObject into string, if possible, or throws exception.

Definition at line 24 of file PyUtils.cpp.

{
    std::string result;
    PyObject* pyStr = PyUnicode_AsEncodedString(obj, "utf-8", "Error ~");
    result = std::string(PyBytes_AsString(pyStr));
    Py_DecRef(pyStr);
    return result;
}

Referenced by IntensityDataCanvas::applyPersistentSettings(), ModelView::JsonModelConverter::from_json(), PyImport::listOfFunctions(), pythonRuntimeInfo(), gui2::ParserPropertyWidget::readSettings(), ModelView::ComboProperty::stringOfSelections(), ModelView::ComboProperty::stringOfValues(), TEST_F(), and toVectorString().

toString() [3/3]

std::string PyUtils::toString

(

wchar_t *

c

)

Definition at line 63 of file PyUtils.cpp.

{
    if (!c)
        return "";
    std::wstring wstr(c);
    return std::string(wstr.begin(), wstr.end());
}

toVectorString()

std::vector< std::string > PyUtils::toVectorString

(

PyObject *

obj

)

Converts PyObject into vector of strings, if possible, or throws exception.

Definition at line 33 of file PyUtils.cpp.

{
    std::vector<std::string> result;
 
    if (PyTuple_Check(obj)) {
        for (Py_ssize_t i = 0; i < PyTuple_Size(obj); i++) {
            PyObject* value = PyTuple_GetItem(obj, i);
            result.push_back(toString(value));
        }
 
    } else if (PyList_Check(obj)) {
        for (Py_ssize_t i = 0; i < PyList_Size(obj); i++) {
            PyObject* value = PyList_GetItem(obj, i);
            result.push_back(toString(value));
        }
 
    } else
        throw std::runtime_error("PyUtils::toVectorString() -> Error. Unexpected object.");
 
    return result;
}

References toString().

Referenced by pythonRuntimeInfo().

Here is the call graph for this function: