mirror of
https://github.com/rizinorg/cutter.git
synced 2024-12-22 12:56:11 +00:00
ab27e09b91
Sometimes it is not necessary to include the whole Cutter.h file Hence, it's been splitted so you can include only what you require E.g. #include "core/CutterCommon.h" to have access to the common types
88 lines
2.3 KiB
C++
88 lines
2.3 KiB
C++
|
|
#ifdef CUTTER_ENABLE_JUPYTER
|
|
|
|
#include <Python.h>
|
|
#include <csignal>
|
|
|
|
#include "core/Cutter.h"
|
|
#include "NestedIPyKernel.h"
|
|
#include "QtResImporter.h"
|
|
|
|
NestedIPyKernel *NestedIPyKernel::start(const QStringList &argv)
|
|
{
|
|
PyThreadState *parentThreadState = PyThreadState_Get();
|
|
|
|
PyThreadState *threadState = Py_NewInterpreter();
|
|
if (!threadState) {
|
|
qWarning() << "Could not create subinterpreter.";
|
|
return nullptr;
|
|
}
|
|
|
|
RegQtResImporter();
|
|
auto cutterIPykernelModule = QtResImport("cutter_ipykernel");
|
|
if (!cutterIPykernelModule) {
|
|
qWarning() << "Could not import cutter_ipykernel.";
|
|
return nullptr;
|
|
}
|
|
|
|
auto kernel = new NestedIPyKernel(cutterIPykernelModule, argv);
|
|
|
|
PyThreadState_Swap(parentThreadState);
|
|
|
|
return kernel;
|
|
}
|
|
|
|
NestedIPyKernel::NestedIPyKernel(PyObject *cutterIPykernelModule, const QStringList &argv)
|
|
{
|
|
threadState = PyThreadState_Get();
|
|
|
|
auto launchFunc = PyObject_GetAttrString(cutterIPykernelModule, "launch_ipykernel");
|
|
|
|
PyObject *argvListObject = PyList_New(argv.size());
|
|
for (int i = 0; i < argv.size(); i++) {
|
|
QString s = argv[i];
|
|
PyList_SetItem(argvListObject, i, PyUnicode_DecodeUTF8(s.toUtf8().constData(), s.length(),
|
|
nullptr));
|
|
}
|
|
|
|
kernel = PyObject_CallFunction(launchFunc, "O", argvListObject);
|
|
}
|
|
|
|
NestedIPyKernel::~NestedIPyKernel()
|
|
{
|
|
auto parentThreadState = PyThreadState_Swap(threadState);
|
|
auto ret = PyObject_CallMethod(kernel, "cleanup", nullptr);
|
|
if (!ret) {
|
|
PyErr_Print();
|
|
}
|
|
PyThreadState_Swap(parentThreadState);
|
|
}
|
|
|
|
void NestedIPyKernel::sendSignal(long signum)
|
|
{
|
|
auto parentThreadState = PyThreadState_Swap(threadState);
|
|
auto ret = PyObject_CallMethod(kernel, "send_signal", "l", signum);
|
|
if (!ret) {
|
|
PyErr_Print();
|
|
}
|
|
PyThreadState_Swap(parentThreadState);
|
|
}
|
|
|
|
QVariant NestedIPyKernel::poll()
|
|
{
|
|
QVariant ret;
|
|
auto parentThreadState = PyThreadState_Swap(threadState);
|
|
PyObject *pyRet = PyObject_CallMethod(kernel, "poll", nullptr);
|
|
if (pyRet) {
|
|
if (PyLong_Check(pyRet)) {
|
|
ret = (qlonglong)PyLong_AsLong(pyRet);
|
|
}
|
|
} else {
|
|
PyErr_Print();
|
|
}
|
|
PyThreadState_Swap(parentThreadState);
|
|
return ret;
|
|
}
|
|
|
|
#endif
|