AtomicInt

CMakeLists.txt

@@ -1,5 +1,5 @@
 cmake_minimum_required(VERSION 3.18)
-project(cereggii-dev)
+project(cereggii)
 
 
 add_subdirectory(src)

pyproject.toml

@@ -103,6 +103,8 @@ ignore = [
   "T201",
   # Import block is un-sorted or un-formatted
   "I001",
+  # Unnecessary `list` / `tuple` / `dict` call (rewrite as a literal)
+  "C408",
 ]
 unfixable = [
   # Don't touch unused imports

src/CMakeLists.txt

@@ -20,8 +20,6 @@ find_package(Python3 REQUIRED COMPONENTS Development.Module)
 
 # Add the module to compile
 Python3_add_library(_cereggii MODULE
-        "cereggii/cereggii.c"
-        "cereggii/atomic_ref.c"
         "cereggii/atomic_dict/atomic_dict.c"
         "cereggii/atomic_dict/blocks.c"
         "cereggii/atomic_dict/delete.c"
@@ -32,6 +30,10 @@ Python3_add_library(_cereggii MODULE
         "cereggii/atomic_dict/node_ops.c"
         "cereggii/atomic_dict/node_sizes_table.c"
         "cereggii/atomic_dict/reservation_buffer.c"
+        "cereggii/atomic_int/atomic_int.c"
+        "cereggii/atomic_int/handle.c"
+        "cereggii/atomic_ref.c"
+        "cereggii/cereggii.c"
 )
 
 # Install the module

src/cereggii/__init__.py

@@ -13,8 +13,10 @@
 
 from .__about__ import __license__, __version__, __version_tuple__  # noqa: F401
 from .atomic_dict import AtomicDict  # noqa: F401
+from .atomic_int import AtomicInt, AtomicIntHandle  # noqa: F401
 from .atomic_ref import AtomicRef  # noqa: F401
 
+
 if not getattr(sys.flags, "nogil", False):
     warnings.warn(
         "this library is meant to be used with nogil python: https://github.com/colesbury/nogil", stacklevel=1

src/cereggii/_cereggii.pyi

@@ -1,9 +1,11 @@
-from typing import NewType
+from typing import Callable, NewType, SupportsComplex, SupportsFloat, SupportsInt
 
 Key = NewType("Key", object)
 Value = NewType("Value", object)
 Cancel = NewType("Cancel", object)
 
+Number = SupportsInt | SupportsFloat | SupportsComplex
+
 class AtomicDict:
     """A thread-safe dictionary (hashmap), that's almost-lock-free™."""
 
@@ -128,8 +130,52 @@ class AtomicDict:
 class AtomicRef:
     """An object reference that may be updated atomically."""
 
-    def __init__(self, reference: object): ...
-    def compare_and_set(self, expected: object, new: object) -> bool: ...
+    def __init__(self, reference: object | None = None): ...
+    def compare_and_set(self, expected: object, updated: object) -> bool: ...
     def get(self) -> object: ...
-    def get_and_set(self, new: object) -> object: ...
-    def set(self, new: object): ...  # noqa: A003
+    def get_and_set(self, updated: object) -> object: ...
+    def set(self, updated: object): ...  # noqa: A003
+
+class AtomicInt(int):
+    """An int that may be updated atomically."""
+
+    def __init__(self, initial_value: int | None = None): ...
+    def compare_and_set(self, expected: int, updated: int) -> bool: ...
+    def get(self) -> int: ...
+    def get_and_set(self, updated: int) -> int: ...
+    def set(self, updated: int) -> None: ...  # noqa: A003
+    def increment_and_get(self, amount: int | None = None) -> int: ...
+    def get_and_increment(self, amount: int | None = None) -> int: ...
+    def decrement_and_get(self, amount: int | None = None) -> int: ...
+    def get_and_decrement(self, amount: int | None = None) -> int: ...
+    def update_and_get(self, other: Callable[[int], int]) -> int: ...
+    def get_and_update(self, other: Callable[[int], int]) -> int: ...
+    def get_handle(self) -> AtomicIntHandle: ...
+    def __itruediv__(self, other):
+        raise NotImplementedError
+    def as_integer_ratio(self):
+        raise NotImplementedError
+    def bit_length(self):
+        raise NotImplementedError
+    def conjugate(self):
+        raise NotImplementedError
+    @classmethod
+    def from_bytes(cls, *args, **kwargs):
+        raise NotImplementedError
+    def to_bytes(self, *args, **kwargs):
+        raise NotImplementedError
+    @property
+    def denominator(self):
+        raise NotImplementedError
+    @property
+    def numerator(self):
+        raise NotImplementedError
+    @property
+    def imag(self):
+        raise NotImplementedError
+    @property
+    def real(self):
+        raise NotImplementedError
+
+class AtomicIntHandle(AtomicInt):
+    pass

src/cereggii/atomic_int/__init__.py

@@ -0,0 +1,24 @@
+# SPDX-FileCopyrightText: 2023-present dpdani <[email protected]>
+#
+# SPDX-License-Identifier: Apache-2.0
+
+import warnings
+
+
+try:
+    from cereggii import _cereggii
+except ImportError as exc:  # building sdist (without compiled modules)
+
+    class AtomicInt:
+        def __init__(self):
+            print("dummy")
+
+    class AtomicIntHandle:
+        def __init__(self):
+            print("dummy")
+
+    warnings.warn(str(exc), stacklevel=1)  # "UserWarning: No module named 'cereggii'" is expected during sdist build
+
+else:
+    AtomicInt = _cereggii.AtomicInt
+    AtomicIntHandle = _cereggii.AtomicIntHandle