forked from Source-Python-Dev-Team/Source.Python
-
Notifications
You must be signed in to change notification settings - Fork 0
Expand file tree
/
Copy paththreads.cpp
More file actions
127 lines (109 loc) · 3.85 KB
/
Copy paththreads.cpp
File metadata and controls
127 lines (109 loc) · 3.85 KB
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
118
119
120
121
122
123
124
125
126
127
/**
* =============================================================================
* Source Python
* Copyright (C) 2012-2023 Source Python Development Team. All rights reserved.
* =============================================================================
*
* This program is free software; you can redistribute it and/or modify it under
* the terms of the GNU General Public License, version 3.0, as published by the
* Free Software Foundation.
*
* This program is distributed in the hope that it will be useful, but WITHOUT
* ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or FITNESS
* FOR A PARTICULAR PURPOSE. See the GNU General Public License for more
* details.
*
* You should have received a copy of the GNU General Public License along with
* this program. If not, see <http://www.gnu.org/licenses/>.
*
* As a special exception, the Source Python Team gives you permission
* to link the code of this program (as well as its derivative works) to
* "Half-Life 2," the "Source Engine," and any Game MODs that run on software
* by the Valve Corporation. You must obey the GNU General Public License in
* all respects for all other code used. Additionally, the Source.Python
* Development Team grants this exception to all derivative works.
*/
//-----------------------------------------------------------------------------
// Includes.
//-----------------------------------------------------------------------------
#include "sp_main.h"
#include "threads.h"
#include "utilities.h"
#include "utilities/wrap_macros.h"
#include "dbg.h"
#include "eiface.h"
#include "convar.h"
#include "tier0/threadtools.h"
#include ENGINE_INCLUDE_PATH(threads.h)
//-----------------------------------------------------------------------------
// Externals.
//-----------------------------------------------------------------------------
extern CGlobalVars *gpGlobals;
//-----------------------------------------------------------------------------
// ConVar "sp_thread_yielding" registration.
//-----------------------------------------------------------------------------
ConVar sp_thread_yielding(
"sp_thread_yielding", "0", FCVAR_NONE,
#ifndef TY_NotImplemented
"If enabled, yields remaining cycles to Python threads every frame."
#else
"Thread yielding is not implemented on '" XSTRINGIFY(SOURCE_ENGINE) "' at this time."
#endif
);
//-----------------------------------------------------------------------------
// CThreadYielder initialization.
//-----------------------------------------------------------------------------
unsigned long CThreadYielder::s_nRefCount = 0;
//-----------------------------------------------------------------------------
// CThreadYielder class.
//-----------------------------------------------------------------------------
PyObject *CThreadYielder::__enter__(PyObject *pSelf)
{
if (!sp_thread_yielding.GetBool()) {
return incref(pSelf);
}
#ifndef TY_NotImplemented
++s_nRefCount;
static bool s_bHooked = false;
if (!s_bHooked) {
WriteJMP((unsigned char *)::TY_Hook, (void *)CThreadYielder::ThreadSleep);
s_bHooked = true;
}
#else
static object warn = import("warnings").attr("warn");
warn("Thread yielding is not implemented on '" XSTRINGIFY(SOURCE_ENGINE) "' at this time.");
#endif
return incref(pSelf);
}
void CThreadYielder::__exit__(PyObject *, PyObject *, PyObject *, PyObject *)
{
#ifndef TY_NotImplemented
if (!s_nRefCount) {
return;
}
--s_nRefCount;
#endif
}
void CThreadYielder::ThreadSleep(unsigned nTime)
{
#ifndef TY_NotImplemented
if (!s_nRefCount || !ThreadInMainThread() || !PyGILState_Check()) {
TY_Sleep(nTime);
}
else {
Py_BEGIN_ALLOW_THREADS;
TY_Yield();
TY_Sleep(nTime);
Py_END_ALLOW_THREADS;
DevMsg(2, MSG_PREFIX "Yielded %d%s on tick %d!\n", nTime, XSTRINGIFY(TY_Unit), gpGlobals->framecount);
}
#endif
}
bool CThreadYielder::IsImplemented()
{
#ifndef TY_NotImplemented
return true;
#else
return false;
#endif
}