cutter/src/common/PythonManager.cpp

186 lines
4.8 KiB
C++
Raw Normal View History

#ifdef CUTTER_ENABLE_PYTHON
#include <cassert>
2018-06-23 16:59:23 +00:00
#include "PythonAPI.h"
#include "PythonManager.h"
#include "Cutter.h"
2018-06-23 16:59:23 +00:00
#include <marshal.h>
2018-06-24 09:51:16 +00:00
#include <QDebug>
2018-06-23 16:59:23 +00:00
#include <QFile>
#include <QDebug>
#include <QCoreApplication>
#include <QDir>
2018-06-23 16:59:23 +00:00
#ifdef CUTTER_ENABLE_PYTHON_BINDINGS
#include <shiboken.h>
#include <pyside.h>
#include <signalmanager.h>
#endif
2018-06-23 16:59:23 +00:00
#include "QtResImporter.h"
static PythonManager *uniqueInstance = nullptr;
2018-06-23 16:59:23 +00:00
PythonManager *PythonManager::getInstance()
{
if (!uniqueInstance) {
uniqueInstance = new PythonManager();
}
2018-06-23 16:59:23 +00:00
return uniqueInstance;
}
PythonManager::PythonManager()
{
}
PythonManager::~PythonManager()
{
}
void PythonManager::initPythonHome()
{
#if defined(APPIMAGE) || defined(MACOS_PYTHON_FRAMEWORK_BUNDLED)
if (customPythonHome.isNull()) {
auto pythonHomeDir = QDir(QCoreApplication::applicationDirPath());
# ifdef APPIMAGE
// Executable is in appdir/bin
pythonHomeDir.cdUp();
qInfo() << "Setting PYTHONHOME =" << pythonHomeDir.absolutePath() << " for AppImage.";
# else // MACOS_PYTHON_FRAMEWORK_BUNDLED
// @executable_path/../Frameworks/Python.framework/Versions/Current
pythonHomeDir.cd("../Frameworks/Python.framework/Versions/Current");
qInfo() << "Setting PYTHONHOME =" << pythonHomeDir.absolutePath() <<
" for macOS Application Bundle.";
# endif
customPythonHome = pythonHomeDir.absolutePath();
}
#endif
if (!customPythonHome.isNull()) {
qInfo() << "PYTHONHOME =" << customPythonHome;
pythonHome = Py_DecodeLocale(customPythonHome.toLocal8Bit().constData(), nullptr);
Py_SetPythonHome(pythonHome);
}
}
2019-02-13 22:24:22 +00:00
#ifdef CUTTER_ENABLE_PYTHON_BINDINGS
extern "C" PyObject *PyInit_CutterBindings();
2019-02-13 22:24:22 +00:00
#endif
2018-06-23 16:59:23 +00:00
void PythonManager::initialize()
{
initPythonHome();
PyImport_AppendInittab("_cutter", &PyInit_api);
PyImport_AppendInittab("_qtres", &PyInit_qtres);
2019-02-13 22:24:22 +00:00
#ifdef CUTTER_ENABLE_PYTHON_BINDINGS
PyImport_AppendInittab("CutterBindings", &PyInit_CutterBindings);
2019-02-13 22:24:22 +00:00
#endif
2018-06-23 16:59:23 +00:00
Py_Initialize();
PyEval_InitThreads();
2019-02-09 13:05:06 +00:00
pyThreadStateCounter = 1; // we have the thread now => 1
2018-06-23 16:59:23 +00:00
RegQtResImporter();
2018-06-24 09:51:16 +00:00
saveThread();
2018-06-23 16:59:23 +00:00
}
#ifdef CUTTER_ENABLE_PYTHON_BINDINGS
static void pySideDestructionVisitor(SbkObject* pyObj, void* data)
{
void **realData = reinterpret_cast<void**>(data);
auto pyQApp = reinterpret_cast<SbkObject*>(realData[0]);
auto pyQObjectType = reinterpret_cast<PyTypeObject*>(realData[1]);
if (pyObj == pyQApp || !PyObject_TypeCheck(pyObj, pyQObjectType)) {
return;
}
if (!Shiboken::Object::hasOwnership(pyObj) || !Shiboken::Object::isValid(pyObj, false)) {
return;
}
const char *reprStr = "";
PyObject *repr = PyObject_Repr(reinterpret_cast<PyObject *>(pyObj));
if (repr) {
reprStr = PyUnicode_AsUTF8(repr);
}
qWarning() << "Warning: QObject from Python remaining (leaked from plugin?):" << reprStr;
if (repr) {
Py_DecRef(repr);
}
Shiboken::Object::setValidCpp(pyObj, false);
Py_BEGIN_ALLOW_THREADS
Shiboken::callCppDestructor<QObject>(Shiboken::Object::cppPointer(pyObj, pyQObjectType));
Py_END_ALLOW_THREADS
};
#endif
2019-02-03 13:00:40 +00:00
void PythonManager::shutdown()
{
emit willShutDown();
restoreThread();
#ifdef CUTTER_ENABLE_PYTHON_BINDINGS
// This is necessary to prevent a segfault when the CutterCore instance is deleted after the Shiboken::BindingManager
Core()->setProperty("_PySideInvalidatePtr", QVariant());
// see PySide::destroyQCoreApplication()
PySide::SignalManager::instance().clear();
Shiboken::BindingManager& bm = Shiboken::BindingManager::instance();
SbkObject* pyQApp = bm.retrieveWrapper(QCoreApplication::instance());
PyTypeObject* pyQObjectType = Shiboken::Conversions::getPythonTypeObject("QObject*");
void* data[2] = {pyQApp, pyQObjectType};
bm.visitAllPyObjects(&pySideDestructionVisitor, &data);
PySide::runCleanupFunctions();
#endif
2019-02-03 13:00:40 +00:00
Py_Finalize();
if (pythonHome) {
PyMem_RawFree(pythonHome);
}
}
2018-06-23 16:59:23 +00:00
void PythonManager::addPythonPath(char *path) {
2018-06-24 09:51:16 +00:00
restoreThread();
2018-06-23 16:59:23 +00:00
PyObject *sysModule = PyImport_ImportModule("sys");
if (!sysModule) {
return;
}
PyObject *pythonPath = PyObject_GetAttrString(sysModule, "path");
if (!pythonPath) {
return;
}
PyObject *append = PyObject_GetAttrString(pythonPath, "append");
if (!append) {
return;
}
PyEval_CallFunction(append, "(s)", path);
2018-06-24 09:51:16 +00:00
saveThread();
2018-06-23 16:59:23 +00:00
}
2018-06-24 09:51:16 +00:00
void PythonManager::restoreThread()
2018-06-23 16:59:23 +00:00
{
2019-02-09 13:05:06 +00:00
pyThreadStateCounter++;
if (pyThreadStateCounter == 1 && pyThreadState) {
2018-06-23 16:59:23 +00:00
PyEval_RestoreThread(pyThreadState);
}
2018-06-24 09:51:16 +00:00
}
2018-06-23 16:59:23 +00:00
2018-06-24 09:51:16 +00:00
void PythonManager::saveThread()
{
2019-02-09 13:05:06 +00:00
pyThreadStateCounter--;
assert(pyThreadStateCounter >= 0);
if (pyThreadStateCounter == 0) {
pyThreadState = PyEval_SaveThread();
}
2018-06-23 16:59:23 +00:00
}
#endif