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
|
#ifndef _REF_UTILS_HPP_
#define _REF_UTILS_HPP_
PyObject* GetPyObjectPointerNoDebugInfo(bool isDebug, PyObject* object) {
if (object != nullptr && isDebug) {
// debug builds have 2 extra pointers at the front that we don't care about
return (PyObject*)((size_t*)object + 2);
}
return object;
}
void DecRef(PyObject* object, bool isDebug) {
auto noDebug = GetPyObjectPointerNoDebugInfo(isDebug, object);
if (noDebug != nullptr && --noDebug->ob_refcnt == 0) {
((PyTypeObject*)GetPyObjectPointerNoDebugInfo(isDebug, noDebug->ob_type))->tp_dealloc(object);
}
}
void IncRef(PyObject* object) {
object->ob_refcnt++;
}
class PyObjectHolder {
private:
PyObject* _object;
public:
bool _isDebug;
PyObjectHolder(bool isDebug) {
_object = nullptr;
_isDebug = isDebug;
}
PyObjectHolder(bool isDebug, PyObject *object) {
_object = object;
_isDebug = isDebug;
};
PyObjectHolder(bool isDebug, PyObject *object, bool addRef) {
_object = object;
_isDebug = isDebug;
if (_object != nullptr && addRef) {
GetPyObjectPointerNoDebugInfo(_isDebug, _object)->ob_refcnt++;
}
};
PyObject* ToPython() {
return _object;
}
~PyObjectHolder() {
DecRef(_object, _isDebug);
}
PyObject* operator* () {
return GetPyObjectPointerNoDebugInfo(_isDebug, _object);
}
};
#endif //_REF_UTILS_HPP_
|
