Skip to main content

pyo3_ffi/
objimpl.rs

1use libc::size_t;
2use std::ffi::{c_int, c_void};
3
4use crate::object::*;
5use crate::pyport::Py_ssize_t;
6
7extern_libpython! {
8    #[cfg_attr(PyPy, link_name = "PyPyObject_Malloc")]
9    pub fn PyObject_Malloc(size: size_t) -> *mut c_void;
10    #[cfg_attr(PyPy, link_name = "PyPyObject_Calloc")]
11    pub fn PyObject_Calloc(nelem: size_t, elsize: size_t) -> *mut c_void;
12    #[cfg_attr(PyPy, link_name = "PyPyObject_Realloc")]
13    pub fn PyObject_Realloc(ptr: *mut c_void, new_size: size_t) -> *mut c_void;
14    #[cfg_attr(PyPy, link_name = "PyPyObject_Free")]
15    pub fn PyObject_Free(ptr: *mut c_void);
16
17    // skipped PyObject_MALLOC
18    // skipped PyObject_REALLOC
19    // skipped PyObject_FREE
20    // skipped PyObject_Del
21    // skipped PyObject_DEL
22
23    #[cfg_attr(PyPy, link_name = "PyPyObject_Init")]
24    pub fn PyObject_Init(arg1: *mut PyObject, arg2: *mut PyTypeObject) -> *mut PyObject;
25    #[cfg_attr(PyPy, link_name = "PyPyObject_InitVar")]
26    pub fn PyObject_InitVar(
27        arg1: *mut PyVarObject,
28        arg2: *mut PyTypeObject,
29        arg3: Py_ssize_t,
30    ) -> *mut PyVarObject;
31
32    // skipped PyObject_INIT
33    // skipped PyObject_INIT_VAR
34
35    #[cfg_attr(PyPy, link_name = "_PyPyObject_New")]
36    fn _PyObject_New(typeobj: *mut PyTypeObject) -> *mut PyObject;
37    #[cfg_attr(PyPy, link_name = "_PyPyObject_NewVar")]
38    fn _PyObject_NewVar(typeobj: *mut PyTypeObject, n: Py_ssize_t) -> *mut PyVarObject;
39}
40
41#[inline]
42pub unsafe fn PyObject_New<T>(typeobj: *mut PyTypeObject) -> *mut T {
43    _PyObject_New(typeobj).cast()
44}
45
46// skipped PyObject_NEW
47
48#[inline]
49pub unsafe fn PyObject_NewVar<T>(typeobj: *mut PyTypeObject, n: Py_ssize_t) -> *mut T {
50    _PyObject_NewVar(typeobj, n).cast()
51}
52
53// skipped PyObject_NEW_VAR
54
55extern_libpython! {
56    #[cfg_attr(PyPy, link_name = "PyPyGC_Collect")]
57    pub fn PyGC_Collect() -> Py_ssize_t;
58
59    #[cfg(Py_3_10)]
60    #[cfg_attr(PyPy, link_name = "PyPyGC_Enable")]
61    pub fn PyGC_Enable() -> c_int;
62
63    #[cfg(Py_3_10)]
64    #[cfg_attr(PyPy, link_name = "PyPyGC_Disable")]
65    pub fn PyGC_Disable() -> c_int;
66
67    #[cfg(Py_3_10)]
68    #[cfg_attr(PyPy, link_name = "PyPyGC_IsEnabled")]
69    pub fn PyGC_IsEnabled() -> c_int;
70}
71
72#[inline]
73pub unsafe fn PyType_IS_GC(t: *mut PyTypeObject) -> c_int {
74    PyType_HasFeature(t, Py_TPFLAGS_HAVE_GC)
75}
76
77extern_libpython! {
78    fn _PyObject_GC_Resize(op: *mut PyVarObject, n: Py_ssize_t) -> *mut PyVarObject;
79}
80
81#[inline]
82pub unsafe fn PyObject_GC_Resize<T>(op: *mut PyObject, n: Py_ssize_t) -> *mut T {
83    _PyObject_GC_Resize(op.cast(), n).cast()
84}
85
86extern_libpython! {
87    #[cfg_attr(PyPy, link_name = "_PyPyObject_GC_New")]
88    fn _PyObject_GC_New(typeobj: *mut PyTypeObject) -> *mut PyObject;
89    #[cfg_attr(PyPy, link_name = "_PyPyObject_GC_NewVar")]
90    fn _PyObject_GC_NewVar(typeobj: *mut PyTypeObject, n: Py_ssize_t) -> *mut PyVarObject;
91
92    #[cfg(not(PyPy))]
93    pub fn PyObject_GC_Track(arg1: *mut c_void);
94
95    #[cfg(not(PyPy))]
96    pub fn PyObject_GC_UnTrack(arg1: *mut c_void);
97
98    #[cfg_attr(PyPy, link_name = "PyPyObject_GC_Del")]
99    pub fn PyObject_GC_Del(arg1: *mut c_void);
100}
101
102#[inline]
103pub unsafe fn PyObject_GC_New<T>(typeobj: *mut PyTypeObject) -> *mut T {
104    _PyObject_GC_New(typeobj).cast()
105}
106
107#[inline]
108pub unsafe fn PyObject_GC_NewVar<T>(typeobj: *mut PyTypeObject, n: Py_ssize_t) -> *mut T {
109    _PyObject_GC_NewVar(typeobj, n).cast()
110}
111
112extern_libpython! {
113    #[cfg(any(all(Py_3_9, not(PyPy)), Py_3_10))] // added in 3.9, or 3.10 on PyPy
114    #[cfg_attr(PyPy, link_name = "PyPyObject_GC_IsTracked")]
115    pub fn PyObject_GC_IsTracked(arg1: *mut PyObject) -> c_int;
116    #[cfg(any(all(Py_3_9, not(PyPy)), Py_3_10))] // added in 3.9, or 3.10 on PyPy
117    #[cfg_attr(PyPy, link_name = "PyPyObject_GC_IsFinalized")]
118    pub fn PyObject_GC_IsFinalized(arg1: *mut PyObject) -> c_int;
119}
120
121// skipped Py_VISIT