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
|
/*
A* -------------------------------------------------------------------
B* This file contains source code for the PyMOL computer program
C* Copyright (c) Schrodinger, LLC.
D* -------------------------------------------------------------------
E* It is unlawful to modify or remove this copyright notice.
F* -------------------------------------------------------------------
G* Please see the accompanying LICENSE file for further information.
H* -------------------------------------------------------------------
I* Additional authors of this source file include:
-*
-*
-*
Z* -------------------------------------------------------------------
*/
#ifndef _H_os_python
#define _H_os_python
#include "os_predef.h"
#ifdef _PYMOL_NOPY
typedef int PyObject;
#undef _PYMOL_NUMPY
#else
// Python.h will redefine those, undef to avoid compiler warning
#undef _POSIX_C_SOURCE
#undef _XOPEN_SOURCE
#include"Python.h"
#include<pythread.h>
#include <string.h>
#if PY_MAJOR_VERSION >= 3
# define PyInt_Check PyLong_Check
# define PyInt_FromLong PyLong_FromLong
# define PyInt_AsLong PyLong_AsLong
# define PyInt_AS_LONG PyLong_AS_LONG
# define PyNumber_Int PyNumber_Long
# define PyString_Check PyUnicode_Check
# define PyString_Size PyUnicode_GetLength
# define PyString_GET_SIZE PyUnicode_GetLength
# define PyString_FromString PyUnicode_FromString
# define PyString_FromStringAndSize PyUnicode_FromStringAndSize
# define PyString_InternFromString PyUnicode_InternFromString
# define PyString_AsString PyUnicode_AsUTF8
# define PyString_AS_STRING PyUnicode_AsUTF8
# define PyCObject_AsVoidPtr(capsule) PyCapsule_GetPointer(capsule, "name")
// Warning: PyCObject_FromVoidPtr destructor takes `void* p` argument,
// but PyCapsule_New destructor takes `PyObject* capsule` argument!
// The "capsulethunk.h" documentation is wrong about using the same destructor.
# define PyCObject_FromVoidPtr(p, d) PyCapsule_New(p, "name", NULL)
# define PyCObject_Check PyCapsule_CheckExact
# define PyEval_EvalCode(o, ...) PyEval_EvalCode((PyObject*)o, __VA_ARGS__)
# define Py_TPFLAGS_HAVE_ITER 0
#endif
/*
* For compatibility with the pickletools, this type represents
* an optionally owned C string and has to be returned by value.
*/
class SomeString {
const char * m_str;
mutable int m_length;
public:
SomeString(const char * s, int len=-1) : m_str(s), m_length(len) {}
inline const char * data() const { return m_str; }
inline const char * c_str() const { return m_str; }
operator const char * () const { return m_str; } // allows assignment to std::string
inline size_t length() const {
if (m_length == -1) {
m_length = m_str ? strlen(m_str) : 0;
}
return m_length;
}
};
inline SomeString PyString_AsSomeString(PyObject * o) {
return PyString_AsString(o);
}
inline SomeString PyBytes_AsSomeString(PyObject * o) {
return SomeString(PyBytes_AsString(o), PyBytes_Size(o));
}
namespace pymol {
/**
* Destruction policy for unique_ptr<PyObject, pymol::pyobject_delete>
*/
struct pyobject_delete {
void operator()(PyObject* o) const { Py_XDECREF(o); }
};
} // namespace pymol
#define unique_PyObject_ptr std::unique_ptr<PyObject, pymol::pyobject_delete>
#endif
#define PYOBJECT_CALLMETHOD(o, m, ...) PyObject_CallMethod(o, (char*)m, (char*)__VA_ARGS__)
#define PYOBJECT_CALLFUNCTION(o, ...) PyObject_CallFunction(o, (char*)__VA_ARGS__)
#endif
|
