from libc.stdint cimport uint64_t, uint32_t
from cymem.cymem cimport Pool

ctypedef uint64_t key_t

cdef struct BloomStruct:
    key_t* bitfield
    key_t hcount # hash count, number of hash functions
    key_t length
    uint32_t seed


cdef class BloomFilter:
    cdef Pool mem
    cdef BloomStruct* c_bloom
    # Thread-unsafe variant of __contains__
    cdef inline bint contains(self, key_t item) noexcept nogil

# Low-level thread-unsafe C API.
# If you use this API and expose it to Python, you must provide external
# synchronization (e.g. with a lock or critical section).

cdef void bloom_init(Pool mem, BloomStruct* bloom, key_t hcount, key_t length, uint32_t seed) except *

cdef bint bloom_contains(const BloomStruct* bloom, key_t item) noexcept nogil

cdef void bloom_add(BloomStruct* bloom, key_t item) noexcept nogil