pyo3_ffi/
setobject.rs

1use crate::object::*;
2#[cfg(not(any(Py_LIMITED_API, PyPy, GraalPy)))]
3use crate::pyport::Py_hash_t;
4use crate::pyport::Py_ssize_t;
5use std::ffi::c_int;
6
7pub const PySet_MINSIZE: usize = 8;
8
9#[cfg(not(any(Py_LIMITED_API, PyPy, GraalPy)))]
10#[repr(C)]
11#[derive(Debug)]
12pub struct setentry {
13    pub key: *mut PyObject,
14    pub hash: Py_hash_t,
15}
16
17#[cfg(not(any(Py_LIMITED_API, PyPy, GraalPy)))]
18#[repr(C)]
19#[derive(Debug)]
20pub struct PySetObject {
21    pub ob_base: PyObject,
22    pub fill: Py_ssize_t,
23    pub used: Py_ssize_t,
24    pub mask: Py_ssize_t,
25    pub table: *mut setentry,
26    pub hash: Py_hash_t,
27    pub finger: Py_ssize_t,
28    pub smalltable: [setentry; PySet_MINSIZE],
29    pub weakreflist: *mut PyObject,
30}
31
32// skipped
33#[inline]
34#[cfg(all(not(any(PyPy, GraalPy)), not(Py_LIMITED_API)))]
35pub unsafe fn PySet_GET_SIZE(so: *mut PyObject) -> Py_ssize_t {
36    debug_assert_eq!(PyAnySet_Check(so), 1);
37    let so = so.cast::<PySetObject>();
38    (*so).used
39}
40
41// skipped _PySet_Dummy
42
43extern_libpython! {
44    #[cfg(not(Py_LIMITED_API))]
45    #[cfg_attr(PyPy, link_name = "_PyPySet_NextEntry")]
46    pub fn _PySet_NextEntry(
47        set: *mut PyObject,
48        pos: *mut Py_ssize_t,
49        key: *mut *mut PyObject,
50        hash: *mut super::Py_hash_t,
51    ) -> c_int;
52
53    // skipped non-limited _PySet_Update
54}
55
56extern_libpython! {
57    #[cfg_attr(PyPy, link_name = "PyPySet_Type")]
58    pub static mut PySet_Type: PyTypeObject;
59    #[cfg_attr(PyPy, link_name = "PyPyFrozenSet_Type")]
60    pub static mut PyFrozenSet_Type: PyTypeObject;
61    pub static mut PySetIter_Type: PyTypeObject;
62}
63
64extern_libpython! {
65    #[cfg_attr(PyPy, link_name = "PyPySet_New")]
66    pub fn PySet_New(arg1: *mut PyObject) -> *mut PyObject;
67    #[cfg_attr(PyPy, link_name = "PyPyFrozenSet_New")]
68    pub fn PyFrozenSet_New(arg1: *mut PyObject) -> *mut PyObject;
69
70    #[cfg_attr(PyPy, link_name = "PyPySet_Add")]
71    pub fn PySet_Add(set: *mut PyObject, key: *mut PyObject) -> c_int;
72    #[cfg_attr(PyPy, link_name = "PyPySet_Clear")]
73    pub fn PySet_Clear(set: *mut PyObject) -> c_int;
74    #[cfg_attr(PyPy, link_name = "PyPySet_Contains")]
75    pub fn PySet_Contains(anyset: *mut PyObject, key: *mut PyObject) -> c_int;
76    #[cfg_attr(PyPy, link_name = "PyPySet_Discard")]
77    pub fn PySet_Discard(set: *mut PyObject, key: *mut PyObject) -> c_int;
78    #[cfg_attr(PyPy, link_name = "PyPySet_Pop")]
79    pub fn PySet_Pop(set: *mut PyObject) -> *mut PyObject;
80    #[cfg_attr(PyPy, link_name = "PyPySet_Size")]
81    pub fn PySet_Size(anyset: *mut PyObject) -> Py_ssize_t;
82
83    #[cfg(PyPy)]
84    #[link_name = "PyPyFrozenSet_CheckExact"]
85    pub fn PyFrozenSet_CheckExact(ob: *mut PyObject) -> c_int;
86}
87
88#[inline]
89#[cfg(not(any(PyPy, GraalPy)))]
90pub unsafe fn PyFrozenSet_CheckExact(ob: *mut PyObject) -> c_int {
91    (Py_TYPE(ob) == &raw mut PyFrozenSet_Type) as c_int
92}
93
94extern_libpython! {
95    #[cfg(PyPy)]
96    #[link_name = "PyPyFrozenSet_Check"]
97    pub fn PyFrozenSet_Check(ob: *mut PyObject) -> c_int;
98}
99
100#[inline]
101#[cfg(not(PyPy))]
102pub unsafe fn PyFrozenSet_Check(ob: *mut PyObject) -> c_int {
103    (Py_TYPE(ob) == &raw mut PyFrozenSet_Type
104        || PyType_IsSubtype(Py_TYPE(ob), &raw mut PyFrozenSet_Type) != 0) as c_int
105}
106
107extern_libpython! {
108    #[cfg(PyPy)]
109    #[link_name = "PyPyAnySet_CheckExact"]
110    pub fn PyAnySet_CheckExact(ob: *mut PyObject) -> c_int;
111}
112
113#[inline]
114#[cfg(not(PyPy))]
115pub unsafe fn PyAnySet_CheckExact(ob: *mut PyObject) -> c_int {
116    (Py_TYPE(ob) == &raw mut PySet_Type || Py_TYPE(ob) == &raw mut PyFrozenSet_Type) as c_int
117}
118
119#[inline]
120pub unsafe fn PyAnySet_Check(ob: *mut PyObject) -> c_int {
121    (PyAnySet_CheckExact(ob) != 0
122        || PyType_IsSubtype(Py_TYPE(ob), &raw mut PySet_Type) != 0
123        || PyType_IsSubtype(Py_TYPE(ob), &raw mut PyFrozenSet_Type) != 0) as c_int
124}
125
126#[inline]
127#[cfg(Py_3_10)]
128pub unsafe fn PySet_CheckExact(op: *mut PyObject) -> c_int {
129    crate::Py_IS_TYPE(op, &raw mut PySet_Type)
130}
131
132extern_libpython! {
133    #[cfg(PyPy)]
134    #[link_name = "PyPySet_Check"]
135    pub fn PySet_Check(ob: *mut PyObject) -> c_int;
136}
137
138#[inline]
139#[cfg(not(PyPy))]
140pub unsafe fn PySet_Check(ob: *mut PyObject) -> c_int {
141    (Py_TYPE(ob) == &raw mut PySet_Type || PyType_IsSubtype(Py_TYPE(ob), &raw mut PySet_Type) != 0)
142        as c_int
143}
pyo3_ffi/setobject.rs

pyo3_ffi/
setobject.rs
