PyCyBase 0.1.8.post4__cp313-cp313-win_amd64.whl

cbase/backports/pylong.h

@@ -0,0 +1,79 @@
+#ifndef PY_BACKPORTS_LONG_H
+#define PY_BACKPORTS_LONG_H
+
+#include <Python.h>
+#include <cbase/int128.h>
+
+static inline PyObject* PyLong_FromUInt128(uint128_t value) {
+#if PY_VERSION_HEX >= 0x030D0000
+    return PyLong_FromNativeBytes(&value, sizeof(value), Py_ASNATIVEBYTES_LITTLE_ENDIAN | Py_ASNATIVEBYTES_UNSIGNED_BUFFER);
+#else
+    return _PyLong_FromByteArray(
+        (const unsigned char*) &value,
+        sizeof(value),
+        1, /* little_endian */
+        0
+    ); /* unsigned */
+#endif
+}
+
+static inline PyObject* PyLong_FromInt128(int128_t value) {
+#if PY_VERSION_HEX >= 0x030D0000
+    return PyLong_FromNativeBytes(&value, sizeof(value), Py_ASNATIVEBYTES_LITTLE_ENDIAN);
+#else
+    return _PyLong_FromByteArray(
+        (const unsigned char*) &value,
+        sizeof(value),
+        1, /* little_endian */
+        1
+    ); /* signed */
+#endif
+}
+
+static inline Py_ssize_t PyLong_AsUInt128(PyObject* obj, uint128_t* value) {
+#if PY_VERSION_HEX >= 0x030D0000
+    return PyLong_AsNativeBytes(obj, value, sizeof(*value), Py_ASNATIVEBYTES_LITTLE_ENDIAN | Py_ASNATIVEBYTES_UNSIGNED_BUFFER);
+#else
+    if (!PyLong_Check(obj)) {
+        PyErr_Format(PyExc_TypeError, "expected int, got %.200s", Py_TYPE(obj)->tp_name);
+        return -1;
+    }
+
+    if (_PyLong_AsByteArray(
+            (PyLongObject*) obj,
+            (unsigned char*) value,
+            sizeof(*value),
+            1, /* little_endian */
+            0
+        ) < 0) {
+        return -1;
+    }
+
+    return sizeof(*value);
+#endif
+}
+
+static inline Py_ssize_t PyLong_AsInt128(PyObject* obj, int128_t* value) {
+#if PY_VERSION_HEX >= 0x030D0000
+    return PyLong_AsNativeBytes(obj, value, sizeof(*value), Py_ASNATIVEBYTES_LITTLE_ENDIAN);
+#else
+    if (!PyLong_Check(obj)) {
+        PyErr_Format(PyExc_TypeError, "expected int, got %.200s", Py_TYPE(obj)->tp_name);
+        return -1;
+    }
+
+    if (_PyLong_AsByteArray(
+            (PyLongObject*) obj,
+            (unsigned char*) value,
+            sizeof(*value),
+            1, /* little_endian */
+            1
+        ) < 0) {
+        return -1;
+    }
+
+    return sizeof(*value);
+#endif
+}
+
+#endif  // PY_BACKPORTS_LONG_H

cbase/backports/pylong.pxd

@@ -0,0 +1,23 @@
+from cpython.object cimport PyObject
+
+from cbase.int128 cimport uint128_t, int128_t
+
+
+cdef extern from "cbase/backports/pylong.h":
+    PyObject*  PyLong_FromUInt128(uint128_t value)
+    PyObject*  PyLong_FromInt128(int128_t value)
+    Py_ssize_t PyLong_AsUInt128(PyObject* obj, uint128_t* value)
+    Py_ssize_t PyLong_AsInt128(PyObject* obj, int128_t* value)
+
+
+cdef object _UINT128_MAX
+cdef object _INT128_MAX
+cdef object _INT128_MIN
+
+cpdef object pylong_from_uint128(bytes buf)
+
+cpdef object pylong_from_int128(bytes buf)
+
+cpdef bytes pylong_as_uint128(object val)
+
+cpdef bytes pylong_as_int128(object val)

cbase/backports/pylong.pyx

@@ -0,0 +1,61 @@
+from cpython.bytes cimport PyBytes_FromStringAndSize
+
+from cbase.int128 cimport c_read_uint128, c_write_uint128, c_read_int128, c_write_int128
+
+
+globals()['_UINT128_MAX'] = _UINT128_MAX = (1 << 128) - 1
+globals()['_INT128_MAX'] = _INT128_MAX = (1 << 127) - 1
+globals()['_INT128_MIN'] = _INT128_MIN = -(1 << 127)
+
+
+cpdef object pylong_from_uint128(bytes buf):
+    if len(buf) != 16:
+        raise ValueError(f"expected 16 bytes, got {len(buf)}")
+    cdef uint128_t val = c_read_uint128(<const char*> buf)
+    return <object> PyLong_FromUInt128(val)
+
+
+cpdef object pylong_from_int128(bytes buf):
+    if len(buf) != 16:
+        raise ValueError(f"expected 16 bytes, got {len(buf)}")
+    cdef int128_t val = c_read_int128(<const char*> buf)
+    return <object> PyLong_FromInt128(val)
+
+
+cpdef bytes pylong_as_uint128(object val):
+    if not isinstance(val, int):
+        raise TypeError(f"expected int, got {type(val).__name__}")
+
+    if val < 0:
+        raise OverflowError("can't convert negative int to uint128_t")
+    if val > _UINT128_MAX:
+        raise OverflowError("int too large for uint128_t")
+
+    cdef uint128_t v
+    cdef Py_ssize_t nbytes = PyLong_AsUInt128(<PyObject*> val, &v)
+
+    if nbytes < 0:
+        raise OverflowError("int out of range for uint128_t")
+    if nbytes > <Py_ssize_t> sizeof(uint128_t):
+        raise OverflowError("int too large for uint128_t")
+    cdef bytes out = PyBytes_FromStringAndSize(NULL, 16)
+    c_write_uint128(<char*> out, v)
+    return out
+
+
+cpdef bytes pylong_as_int128(object val):
+    if not isinstance(val, int):
+        raise TypeError(f"expected int, got {type(val).__name__}")
+
+    if val < _INT128_MIN or val > _INT128_MAX:
+        raise OverflowError("int out of range for int128_t")
+
+    cdef int128_t v
+    cdef Py_ssize_t nbytes = PyLong_AsInt128(<PyObject*> val, &v)
+    if nbytes < 0:
+        raise OverflowError("int out of range for int128_t")
+    if nbytes > <Py_ssize_t> sizeof(int128_t):
+        raise OverflowError("int too large for int128_t")
+    cdef bytes out = PyBytes_FromStringAndSize(NULL, 16)
+    c_write_int128(<char*> out, v)
+    return out

cbase/bytemap/__infra__.pxd

@@ -0,0 +1,42 @@
+from .c_bytemap cimport (
+    MIN_BYTEMAP_CAPACITY, DEFAULT_BYTEMAP_CAPACITY, BYTEMAP_GROWTH_FACTOR,
+    MAX_BYTEMAP_CAPACITY, BYTEMAP_SALT_MAGIC,
+
+    bytemap_entry, bytemap_entry_ex,
+    bytemap_callback_event, bytemap_callback_func, bytemap_ex_callback_func,
+    bytemap_callback_ctx, bytemap_ex_callback_ctx,
+    bytemap, bytemap_ex,
+    bytemap_ret_code,
+
+    c_bytemap_hash, c_bytemap_clone_key, c_bytemap_free_key,
+    c_bytemap_gen_seq_id,
+    c_bytemap_entry_at, c_bytemap_entry_next, c_bytemap_entry_first,
+    c_bytemap_invoke_callbacks,
+
+    c_bytemap_ex_init, c_bytemap_ex_dealloc,
+    c_bytemap_ex_new, c_bytemap_ex_clear, c_bytemap_ex_free,
+    c_bytemap_ex_register_callback, c_bytemap_ex_unregister_callback,
+    c_bytemap_ex_get, c_bytemap_ex_get_ptr, c_bytemap_ex_contains,
+    c_bytemap_ex_rehash, c_bytemap_ex_set, c_bytemap_ex_pop, c_bytemap_ex_pop_ptr,
+    c_bytemap_ex_len, c_bytemap_ex_clone,
+
+    c_bytemap_ex_set_double, c_bytemap_ex_get_double, c_bytemap_ex_pop_double,
+
+    c_bytemap_new, c_bytemap_clear, c_bytemap_free,
+    c_bytemap_register_callback, c_bytemap_unregister_callback,
+    c_bytemap_get, c_bytemap_contains, c_bytemap_rehash,
+    c_bytemap_set, c_bytemap_pop, c_bytemap_len, c_bytemap_clone,
+
+    c_bytemap_first, c_bytemap_last, c_bytemap_next, c_bytemap_prev,
+    c_bytemap_entry_value,
+    c_bytemap_entry_value_as_ptr, c_bytemap_entry_value_as_uintptr,
+    c_bytemap_entry_value_as_double,
+    c_bytemap_entry_value_from_ptr, c_bytemap_entry_value_from_uintptr,
+    c_bytemap_entry_value_from_double,
+
+    _ByteMapBase,
+    ByteMapEx, ByteMapExDouble, ByteMap,
+    _BoundByteMapBase,
+    BoundByteMapEx, BoundByteMapExDouble, BoundByteMap, BoundByteSet,
+    ByteMapPerformanceTestToolkit,
+)

cbase/bytemap/__init__.pxd

@@ -0,0 +1,42 @@
+from .c_bytemap cimport (
+    MIN_BYTEMAP_CAPACITY, DEFAULT_BYTEMAP_CAPACITY, BYTEMAP_GROWTH_FACTOR,
+    MAX_BYTEMAP_CAPACITY, BYTEMAP_SALT_MAGIC,
+
+    bytemap_entry, bytemap_entry_ex,
+    bytemap_callback_event, bytemap_callback_func, bytemap_ex_callback_func,
+    bytemap_callback_ctx, bytemap_ex_callback_ctx,
+    bytemap, bytemap_ex,
+    bytemap_ret_code,
+
+    c_bytemap_hash, c_bytemap_clone_key, c_bytemap_free_key,
+    c_bytemap_gen_seq_id,
+    c_bytemap_entry_at, c_bytemap_entry_next, c_bytemap_entry_first,
+    c_bytemap_invoke_callbacks,
+
+    c_bytemap_ex_init, c_bytemap_ex_dealloc,
+    c_bytemap_ex_new, c_bytemap_ex_clear, c_bytemap_ex_free,
+    c_bytemap_ex_register_callback, c_bytemap_ex_unregister_callback,
+    c_bytemap_ex_get, c_bytemap_ex_get_ptr, c_bytemap_ex_contains,
+    c_bytemap_ex_rehash, c_bytemap_ex_set, c_bytemap_ex_pop, c_bytemap_ex_pop_ptr,
+    c_bytemap_ex_len, c_bytemap_ex_clone,
+
+    c_bytemap_ex_set_double, c_bytemap_ex_get_double, c_bytemap_ex_pop_double,
+
+    c_bytemap_new, c_bytemap_clear, c_bytemap_free,
+    c_bytemap_register_callback, c_bytemap_unregister_callback,
+    c_bytemap_get, c_bytemap_contains, c_bytemap_rehash,
+    c_bytemap_set, c_bytemap_pop, c_bytemap_len, c_bytemap_clone,
+
+    c_bytemap_first, c_bytemap_last, c_bytemap_next, c_bytemap_prev,
+    c_bytemap_entry_value,
+    c_bytemap_entry_value_as_ptr, c_bytemap_entry_value_as_uintptr,
+    c_bytemap_entry_value_as_double,
+    c_bytemap_entry_value_from_ptr, c_bytemap_entry_value_from_uintptr,
+    c_bytemap_entry_value_from_double,
+
+    _ByteMapBase,
+    ByteMapEx, ByteMapExDouble, ByteMap,
+    _BoundByteMapBase,
+    BoundByteMapEx, BoundByteMapExDouble, BoundByteMap, BoundByteSet,
+    ByteMapPerformanceTestToolkit,
+)

cbase/bytemap/__init__.py

@@ -0,0 +1,21 @@
+from .c_bytemap import (
+    ByteMapEx,
+    ByteMapExDouble,
+    ByteMap,
+    BoundByteMapEx,
+    BoundByteMapExDouble,
+    BoundByteMap,
+    BoundByteSet,
+    ByteMapPerformanceTestToolkit,
+)
+
+__all__ = [
+    'ByteMapEx',
+    'ByteMapExDouble',
+    'ByteMap',
+    'BoundByteMapEx',
+    'BoundByteMapExDouble',
+    'BoundByteMap',
+    'BoundByteSet',
+    'ByteMapPerformanceTestToolkit',
+]