blob: 3b64ac611e815b2aa6cd6fe8e4f56b55c81e5fc0 (
plain)
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
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
|
/*
* Copyright (C) 2005-2018 Team Kodi
* This file is part of Kodi - https://kodi.tv
*
* SPDX-License-Identifier: GPL-2.0-or-later
* See LICENSES/README.md for more information.
*/
#include "PyContext.h"
#include "utils/log.h"
#include <Python.h>
namespace XBMCAddon
{
namespace Python
{
struct PyContextState
{
inline explicit PyContextState(bool pcreatedByGilRelease = false) :
state(NULL), createdByGilRelease(pcreatedByGilRelease) {}
int value = 0;
PyThreadState* state;
int gilReleasedDepth = 0;
bool createdByGilRelease;
};
static thread_local PyContextState* tlsPyContextState;
void* PyContext::enterContext()
{
PyContextState* cur = tlsPyContextState;
if (cur == NULL)
{
cur = new PyContextState();
tlsPyContextState = cur;
}
// increment the count
cur->value++;
return cur;
}
void PyContext::leaveContext()
{
// here we ASSUME that the constructor was called.
PyContextState* cur = tlsPyContextState;
cur->value--;
int curlevel = cur->value;
// this is a hack but ...
if (curlevel < 0)
{
CLog::Log(LOGERROR, "FATAL: PyContext closed more than opened");
curlevel = cur->value = 0;
}
if (curlevel == 0)
{
// clear the tlsPyContextState
tlsPyContextState = NULL;
delete cur;
}
}
void PyGILLock::releaseGil()
{
PyContextState* cur = tlsPyContextState;
// This means we're not within the python context, but
// because we may be in a thread spawned by python itself,
// we need to handle this.
if (!cur)
{
cur = static_cast<PyContextState*>(PyContext::enterContext());
cur->createdByGilRelease = true;
}
if (cur->gilReleasedDepth == 0) // true if we are at the outermost
{
PyThreadState* _save;
// this macro sets _save
{
Py_UNBLOCK_THREADS
}
cur->state = _save;
}
cur->gilReleasedDepth++; // the first time this goes to 1
}
void PyGILLock::acquireGil()
{
PyContextState* cur = tlsPyContextState;
// it's not possible for cur to be NULL (and if it is, we want to fail anyway).
// decrement the depth and make sure we're in the right place.
cur->gilReleasedDepth--;
if (cur->gilReleasedDepth == 0) // are we back to zero?
{
PyThreadState* _save = cur->state;
// This macros uses _save
{
Py_BLOCK_THREADS
}
cur->state = NULL; // clear the state to indicate we've reacquired the gil
// we clear it only if we created it on this level.
if (cur->createdByGilRelease)
PyContext::leaveContext();
}
}
}
}
|
