PythonSubInterpretersAndThreads/PythonInterpreter.cpp at main · neundorf/PythonSubInterpretersAndThreads · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
#include "PythonInterpreter.h"

#include <Python.h>
#include <fmt/color.h>
#include <fmt/printf.h>

RootPythonInterpreter* PythonInterpreter::s_rootInterpreter = nullptr;


RootPythonInterpreter::RootPythonInterpreter()
{
   Py_Initialize();
   m_mainThreadState = Py_NewInterpreter();
   if (m_mainThreadState == nullptr)
   {
      fmt::print(fmt::fg(fmt::color::red), "Failed to create the main interpreter\n");
   }
   fmt::print("main thread state: {0}\n", (void*)m_mainThreadState);

   PyRun_SimpleString("print('hello python')");
}


RootPythonInterpreter::~RootPythonInterpreter()
{
   Py_FinalizeEx();
}


PythonInterpreter::PythonInterpreter()
{
   if (s_rootInterpreter == nullptr)
   {
      fmt::print(fmt::fg(fmt::color::red), "Python not initialized !\n");
      return;
   }

   static PyInterpreterConfig py_config = {
        .use_main_obmalloc = 0,
        .allow_fork = 0,
        .allow_exec = 0,
        .allow_threads = 1,
        .allow_daemon_threads = 0,
        .check_multi_interp_extensions = 1,
        .gil = PyInterpreterConfig_OWN_GIL,
   };
   const PyThreadState* state1 = nullptr; // PyThreadState_Get();
   PyStatus status = Py_NewInterpreterFromConfig(&m_threadState, &py_config);
   const PyThreadState* state2 = PyThreadState_Get();
   fmt::print("new thread state: {0}\n", (void*)m_threadState);
   fmt::print("new thread state1: {0} state2: {1}\n", (void*)state1, (void*)state2);

}


PythonInterpreter::~PythonInterpreter()
{
}


void PythonInterpreter::initialize()
{
   if (s_rootInterpreter != nullptr)
   {
      fmt::print(fmt::fg(fmt::color::green), "Python already initialized.\n");
      return;
   }

   s_rootInterpreter = new RootPythonInterpreter();

}


int PythonInterpreter::runSimpleString(const char* _pythonCode)
{
   PyThreadState* prevState = PyThreadState_Get();
   const bool differentState = (prevState != m_threadState);

   if (differentState)
   {
      PyThreadState_Swap(m_threadState);
   }

   int result = PyRun_SimpleString(_pythonCode);

   if (differentState)
   {
      PyThreadState_Swap(prevState);
   }
   return result;
}