Documentation
¶
Index ¶
- type Accumulator
- type Bitmap
- func And(a, b *Bitmap) *Bitmap
- func AndNot(a, b *Bitmap) *Bitmap
- func AndOld(a, b *Bitmap) *Bitmap
- func CopresenceByMask(bms []*Bitmap, mask uint64) *Bitmap
- func CopresenceByMaskToBuf(bms []*Bitmap, mask uint64, buf []byte) *Bitmap
- func FastAnd(bitmaps ...*Bitmap) *Bitmap
- func FastOr(bitmaps ...*Bitmap) *Bitmap
- func FastParOr(numGo int, bitmaps ...*Bitmap) *Bitmap
- func FromBuffer(data []byte) *Bitmap
- func FromBufferUnlimited(buf []byte) *Bitmap
- func FromBufferWithCopy(src []byte) *Bitmap
- func FromSortedList(vals []uint64) *Bitmap
- func InitFromBufferUnlimited(dst *Bitmap, buf []byte) *Bitmap
- func MaskedAnd(a, b *Bitmap, mask uint64) *Bitmap
- func MaskedAndToBuf(a, b *Bitmap, mask uint64, buf []byte) *Bitmap
- func NewBitmap() *Bitmap
- func NewBitmapToBuf(buf []byte) *Bitmap
- func NewBitmapWith(numKeys int) *Bitmap
- func Or(a, b *Bitmap) *Bitmap
- func OrOld(a, b *Bitmap) *Bitmap
- func Prefill(maxX uint64) *Bitmap
- func (ra *Bitmap) And(bm *Bitmap) *Bitmap
- func (ra *Bitmap) AndConc(bm *Bitmap, maxConcurrency int) *Bitmap
- func (ra *Bitmap) AndMasked(b *Bitmap, mask uint64) *Bitmap
- func (ra *Bitmap) AndMaskedConc(b *Bitmap, mask uint64, maxConcurrency int) *Bitmap
- func (ra *Bitmap) AndNot(bm *Bitmap) *Bitmap
- func (ra *Bitmap) AndNotConc(bm *Bitmap, maxConcurrency int) *Bitmap
- func (ra *Bitmap) AndNotOld(bm *Bitmap)
- func (ra *Bitmap) AndOld(bm *Bitmap)
- func (ra *Bitmap) Cleanup()
- func (ra *Bitmap) Clone() *Bitmap
- func (ra *Bitmap) CloneToBuf(buf []byte) *Bitmap
- func (ra *Bitmap) Contains(x uint64) bool
- func (ra *Bitmap) ConvertToBitmapContainers()
- func (ra *Bitmap) Debug(x uint64) string
- func (ra *Bitmap) FillUp(maxX uint64)
- func (ra *Bitmap) GetCardinality() int
- func (src *Bitmap) InitCloneToBuf(dst *Bitmap, buf []byte) *Bitmap
- func (ra *Bitmap) Intersects(bm *Bitmap) bool
- func (ra *Bitmap) IntersectsMasked(bm *Bitmap, mask uint64) bool
- func (ra *Bitmap) IsEmpty() bool
- func (ra *Bitmap) LenInBytes() int
- func (r *Bitmap) ManyIterator() *ManyItr
- func (ra *Bitmap) Masked(mask uint64) *Bitmap
- func (ra *Bitmap) MaskedToBuf(mask uint64, buf []byte) *Bitmap
- func (ra *Bitmap) Maximum() uint64
- func (ra *Bitmap) Minimum() uint64
- func (ra *Bitmap) NewContainsCursor() *ContainsCursor
- func (bm *Bitmap) NewIterator() *Iterator
- func (bm *Bitmap) NewRangeIterators(numRanges int) []*Iterator
- func (ra *Bitmap) NumContainers() int
- func (ra *Bitmap) Or(bm *Bitmap) *Bitmap
- func (ra *Bitmap) OrConc(bm *Bitmap, maxConcurrency int) *Bitmap
- func (dst *Bitmap) OrOld(src *Bitmap)
- func (ra *Bitmap) Rank(x uint64) int
- func (ra *Bitmap) Remove(x uint64) bool
- func (ra *Bitmap) RemoveRange(lo, hi uint64)
- func (ra *Bitmap) Reset()
- func (ra *Bitmap) Select(x uint64) (uint64, error)
- func (ra *Bitmap) Set(x uint64) bool
- func (ra *Bitmap) SetMany(vals []uint64)
- func (bm *Bitmap) Split(externalSize func(start, end uint64) uint64, maxSz uint64) []*Bitmap
- func (ra *Bitmap) String() string
- func (ra *Bitmap) ToArray() []uint64
- func (ra *Bitmap) ToBuffer() []byte
- func (ra *Bitmap) ToBufferWithCopy() []byte
- func (ra *Bitmap) ZeroOut()
- type ContainsCursor
- type Iterator
- type ManyItr
Constants ¶
This section is empty.
Variables ¶
This section is empty.
Functions ¶
This section is empty.
Types ¶
type Accumulator ¶ added in v0.0.16
type Accumulator struct {
// contains filtered or unexported fields
}
Accumulator combines many bitmaps into one by depositing their elements into dense per-key staging containers instead of merging roaring structures pairwise. Merging N tiny bitmaps via Or/FastOr costs a structural container merge per source — for array containers that is an O(container size) memmove to insert even a single element, which makes N-way unions of near-singleton bitmaps quadratic in the source count. The accumulator makes every deposit O(1) by staging each touched 64K key range in one fixed-size dense block, then assembling the exact-size result once (Bitmap, BitmapToBuf or InitBitmapToBuf).
Total cost is O(total input elements + touched key ranges); staging memory is one 8KB block per touched key range (proportional to the spread of the result, not to the number of sources). In the sparse extreme — every element in its own range — that degrades to ~8KB per element, staging orders of magnitude more memory than the result it builds; FastOr or sort+FromSortedList are the better tools there. Reset caps only what is retained between unions, never the peak during one. Within one source, keys ascend, so its new ranges append cheaply; across sources, each new range that arrives out of order pays an O(existing ranges) insertion shift, so unions spanning many ranges fed from unsorted sources approach O(ranges²) in that term.
Usage:
acc := NewAccumulator()
for _, bm := range sources {
acc.Or(bm)
}
result := acc.Bitmap()
Or never retains its argument, so callers may release or reuse a source's memory as soon as Or returns. The zero Accumulator is ready to use; NewAccumulator is equivalent. An Accumulator is not safe for concurrent use.
func NewAccumulator ¶ added in v0.0.16
func NewAccumulator() *Accumulator
func (*Accumulator) Bitmap ¶ added in v0.0.16
func (acc *Accumulator) Bitmap() *Bitmap
Bitmap assembles and returns the union. The accumulator keeps its staging state afterwards; call Reset to reuse it for a new union.
func (*Accumulator) BitmapToBuf ¶ added in v0.0.16
func (acc *Accumulator) BitmapToBuf(get func(sizeBytes int) []byte) *Bitmap
BitmapToBuf assembles the union into a buffer obtained from get — for callers that pool result memory — and returns a bitmap backed by it. get is called exactly once, with the exact required size in bytes, and must not itself use the accumulator (doing so panics). Returning a buffer smaller than the requested size panics — the caller was told exactly how much is needed. The buffer is adopted to its full capacity (rounded down to even), not just its length — a length-limited view of a larger backing array hands the whole backing array to the result. The returned bitmap holds a reference to the buffer, so the buffer must not be reused until the bitmap is released. Mutating the result stays within the buffer's capacity until it needs to grow, at which point it migrates to the heap.
func (*Accumulator) InitBitmapToBuf ¶ added in v0.0.16
func (acc *Accumulator) InitBitmapToBuf(get func(sizeBytes int) (*Bitmap, []byte)) *Bitmap
InitBitmapToBuf is BitmapToBuf for callers that pool the result Bitmap struct together with its buffer: get returns both — typically from one pool entry — and the union is built into them, so a warm accumulator building into pooled memory allocates nothing. The struct must be non-nil; its previous fields are overwritten, never freed — it must not own anything the caller still needs. A buffer smaller than the requested size panics, as in BitmapToBuf.
func (*Accumulator) Or ¶ added in v0.0.16
func (acc *Accumulator) Or(bm *Bitmap)
Or deposits all elements of bm into the accumulator. bm is only read, never retained: it may be released or reused as soon as Or returns.
func (*Accumulator) Reset ¶ added in v0.0.16
func (acc *Accumulator) Reset()
Reset clears the accumulator for reuse. Up to maxRetainedBlocks staging blocks are cleared and kept as key-independent spares for the next union's ranges, whatever they are; everything else is released, so a pooled accumulator's memory and per-union cost stay bounded by the current union, not by every range it has ever served. Reset bounds retained memory only — the peak during a union is whatever its spread demands.
type Bitmap ¶
type Bitmap struct {
// contains filtered or unexported fields
}
func CopresenceByMask ¶ added in v0.0.14
CopresenceByMask returns a new bitmap containing every value v in the union of bms whose masked value (v & mask) is present in bms[i].Masked(mask) for every i. Original (unmasked) values are preserved. Equivalent in result to the (much more expensive) fold:
out := bms[0].Masked(mask)
for i := 1; i < len(bms); i++ {
out = out.And(bms[i].Masked(mask))
}
// then map each masked value back to every original v in any bms[i]
// whose v & mask matches.
but processes all inputs in one pass — avoiding intermediate bitmap allocations and walking each input only once. None of the bms are modified.
Mask shape requirement: mask & 0xFFFF == 0xFFFF. Other mask shapes are not supported and will yield incorrect results.
Edge cases:
- len(bms) == 0: empty result.
- len(bms) == 1: result is a clone of bms[0] (single-input co-presence is trivially the input).
- any input empty: empty result.
Algorithm: with mask_high = mask & 0xFFFFFFFFFFFF0000, group each input's containers by K & mask_high. Walk the masked-key entry slices of all N inputs in lockstep (multi-pointer max-key advance) to find groups present on every side. For each such group:
- compute pos_i = OR of positions across input i's containers in the group
- common_pos = AND_i (pos_i) — positions co-present in every side
- for each unique original key K appearing in any input's group, emit OR_i(bms[i][K]) AND common_pos at K (omitting empty results)
Step 2 short-circuits as soon as common_pos becomes empty.
func CopresenceByMaskToBuf ¶ added in v0.0.14
CopresenceByMaskToBuf is like CopresenceByMask but uses the provided byte slice as the underlying buffer for the result bitmap, avoiding the result's heap allocation when the buffer is large enough. If the buffer is too small the bitmap grows internally (and allocates), so correctness is unaffected — only the allocation-elision is.
Sizing guidance: a safe upper bound is the sum of input byte sizes (sum_i bms[i].LenInBytes()). The result's keys area is bounded by the sum of input keys areas, and each result container is bounded by the sum of its contributing inputs' containers (the per-key OR can't exceed it; the subsequent AND with commonPos can only shrink). In practice results are often much smaller — callers that know their workload's typical density can pass a smaller buffer and accept occasional growth.
Same mask shape requirement as CopresenceByMask: mask & 0xFFFF == 0xFFFF.
func FastOr ¶
FastOr would merge given Bitmaps into one Bitmap. This is faster than doing an OR over the bitmaps iteratively.
func FastParOr ¶
FastParOr would group up bitmaps and call FastOr on them concurrently. It would then merge the groups into final Bitmap. This approach is simpler and faster than operating at a container level, because we can't operate on array containers belonging to the same Bitmap concurrently because array containers can expand, leaving no clear boundaries.
If FastParOr is called with numGo=1, it just calls FastOr.
Experiments with numGo=4 shows that FastParOr would be 2x the speed of FastOr, but 4x the memory usage, even under 50% CPU usage. So, use wisely.
func FromBuffer ¶
FromBuffer returns a pointer to bitmap corresponding to the given buffer. This bitmap shouldn't be modified because it might corrupt the given buffer.
func FromBufferUnlimited ¶ added in v0.0.10
FromBufferUnlimited returns a pointer to bitmap corresponding to the given buffer. Entire buffer capacity is utlized for future bitmap modifications and expansions.
func FromBufferWithCopy ¶
FromBufferWithCopy creates a copy of the given buffer and returns a bitmap based on the copied buffer. This bitmap is safe for both read and write operations.
func FromSortedList ¶
func InitFromBufferUnlimited ¶ added in v0.0.16
InitFromBufferUnlimited re-points dst at buf — the in-place equivalent of FromBufferUnlimited. dst's previous content is discarded, never freed. A buf too small for even the empty bitmap (under 8 bytes) falls back to a heap allocation. Returns dst.
func MaskedAnd ¶ added in v0.0.14
MaskedAnd returns a new bitmap containing the AND of a and b, with mask applied to all keys. Keys that become equal after masking have their containers merged via OR. This is equivalent to Masked(And(a, b), mask) but allocates only one bitmap instead of two. Neither a nor b is modified. The lowest 16 bits of keys are always zeroed.
func MaskedAndToBuf ¶ added in v0.0.14
MaskedAndToBuf is like MaskedAnd but uses the provided byte slice as the underlying buffer for the result bitmap, avoiding heap allocation when the buffer is large enough. The lowest 16 bits of keys are always zeroed.
func NewBitmapToBuf ¶ added in v0.0.14
NewBitmapToBuf creates a new bitmap using the provided byte slice as the underlying data buffer. If the buffer is too small to hold the initial bitmap structure, a heap-allocated bitmap is returned instead. The _ptr field is set to keep a GC reference to buf, since the bitmap operates on an unsafe []uint16 view of the same memory.
func NewBitmapWith ¶
func (*Bitmap) And ¶
And performs in-place AND of ra and bm (ra &= bm). ra drives the two-pointer walk: ra containers with no matching bm key must be zeroed out, so every ra container must be visited.
func (*Bitmap) AndConc ¶ added in v0.0.9
AndConc performs And merge concurrently (ra &= bm). ra drives the walk: every ra container must be visited to zero out those with no matching bm key. Concurrency is calculated based on number of internal containers in ra, so that each goroutine handles at least [minContainersPerRoutine] containers. maxConcurrency limits concurrency calculated internally. If maxConcurrency <= 0, then calculated concurrency is not limited.
E.g.: dst bitmap has 100 containers. Internal concurrency = 100/24 = 4. For: - maxConcurrency = 2, there will be 2 goroutines executed - maxConcurrency = 6, there will be 4 goroutines executed
func (*Bitmap) AndMasked ¶ added in v0.0.14
AndMasked is equivalent to ra.And(b.Masked(mask)) but avoids allocating an intermediate masked bitmap. b is not modified.
For each key k in ra, AndMasked finds all keys k' in b where k'&mask == k, ORs their containers together, then ANDs the result into ra's container at k. If no key in b maps to k under the mask, ra's container at k is zeroed out. The lowest 16 bits of the mask are always ignored.
func (*Bitmap) AndMaskedConc ¶ added in v0.0.14
AndMaskedConc is like AndMasked but processes ra's containers concurrently. Concurrency is calculated based on number of internal containers in ra, so that each goroutine handles at least [minContainersPerRoutine] containers. maxConcurrency limits concurrency calculated internally. If maxConcurrency <= 0, then calculated concurrency is not limited.
func (*Bitmap) AndNot ¶
AndNot performs in-place AND-NOT of ra and bm (ra &^= bm). Neither bitmap has unmatched containers requiring mandatory action: unmatched ra containers are kept, unmatched bm containers are irrelevant. The smaller bitmap drives for a minor sequential optimisation.
func (*Bitmap) AndNotConc ¶ added in v0.0.9
AndNotConc performs AndNot merge concurrently (ra &^= bm). Neither bitmap has unmatched containers requiring mandatory action, so the smaller bitmap drives to minimise goroutine count. Concurrency is calculated based on number of internal containers in the smaller bitmap, so that each goroutine handles at least [minContainersPerRoutine] containers. maxConcurrency limits concurrency calculated internally. If maxConcurrency <= 0, then calculated concurrency is not limited.
E.g.: smaller bitmap has 100 containers. Internal concurrency = 100/24 = 4. For: - maxConcurrency = 2, there will be 2 goroutines executed - maxConcurrency = 6, there will be 4 goroutines executed
func (*Bitmap) CloneToBuf ¶ added in v0.0.9
func (*Bitmap) ConvertToBitmapContainers ¶ added in v0.0.3
func (ra *Bitmap) ConvertToBitmapContainers()
func (*Bitmap) FillUp ¶ added in v0.0.9
FillUp fill bitmap with elements (maximum-maxX], where maximum means last element. If bitmap is empty then [0-maxX] elements are added (reusing underlying data slice if big enough to fit all elements). If last element is >= than given maxX nothing is done.
func (*Bitmap) GetCardinality ¶
func (*Bitmap) InitCloneToBuf ¶ added in v0.0.16
InitCloneToBuf re-initializes dst over buf with a clone of src — the in-place equivalent of CloneToBuf. dst's previous content is discarded, never freed. A nil or zero-value src yields an empty bitmap over buf; a buf too small for the clone panics, too small for even the empty bitmap falls back to a heap allocation. Returns dst.
func (*Bitmap) Intersects ¶ added in v0.0.14
Intersects reports whether ra and bm share at least one element. Equivalent to !ra.Clone().And(bm).IsEmpty() but short-circuits at the first common element without allocating a result bitmap.
func (*Bitmap) IntersectsMasked ¶ added in v0.0.14
IntersectsMasked reports whether ra and bm share at least one element after applying mask to bm's keys. Equivalent to !ra.Clone().And(bm.Masked(mask)).IsEmpty() but short-circuits at the first common element without allocating. Containers in ra whose key has a non-zero dimension (low 16 bits != 0) are not checked — same behaviour as AndMasked.
func (*Bitmap) LenInBytes ¶ added in v0.0.9
func (*Bitmap) ManyIterator ¶
TODO: See if this is needed, we should remove this
func (*Bitmap) Masked ¶ added in v0.0.14
Masked applies the given mask to every key and returns a new bitmap. Keys that collapse to the same masked value have their containers merged via container-level OR operations.
func (*Bitmap) MaskedToBuf ¶ added in v0.0.14
MaskedToBuf is like Masked but uses the provided byte slice as the underlying buffer for the result bitmap, avoiding heap allocation when the buffer is large enough.
func (*Bitmap) NewContainsCursor ¶ added in v0.0.15
func (ra *Bitmap) NewContainsCursor() *ContainsCursor
NewContainsCursor returns a cursor bound to ra. A nil ra (or nil cursor) reports false, matching Bitmap.Contains.
func (*Bitmap) NewIterator ¶
func (*Bitmap) NewRangeIterators ¶
func (*Bitmap) NumContainers ¶ added in v0.0.14
func (*Bitmap) Or ¶
Or performs in-place OR of ra and bm (ra |= bm). bm drives the two-pointer walk: bm containers with no matching ra key must be added to ra, so every bm container must be visited. Ra containers with no matching bm key are left unchanged — no visit required.
func (*Bitmap) OrConc ¶ added in v0.0.9
OrConc performs Or merge concurrently (ra |= bm). bm drives the walk: every bm container must be visited to add those with no matching ra key. Concurrency is calculated based on number of internal containers in bm, so that each goroutine handles at least [minContainersPerRoutine] containers. maxConcurrency limits concurrency calculated internally. If maxConcurrency <= 0, then calculated concurrency is not limited.
E.g.: src bitmap has 100 containers. Internal concurrency = 100/24 = 4. For: - maxConcurrency = 2, there will be 2 goroutines executed - maxConcurrency = 6, there will be 4 goroutines executed
func (*Bitmap) RemoveRange ¶
Remove range removes [lo, hi) from the bitmap.
func (*Bitmap) Split ¶
Split splits the bitmap based on maxSz and the externalSize function. It splits the bitmap such that size of each split bitmap + external size corresponding to its elements approximately equal to maxSz (it can be greater than maxSz sometimes). The splits are returned in sorted order. externalSize is a function that should return the external size corresponding to elements in range [start, end]. External size is used to calculate the split boundaries.
func (*Bitmap) ToBufferWithCopy ¶
type ContainsCursor ¶ added in v0.0.15
type ContainsCursor struct {
// contains filtered or unexported fields
}
ContainsCursor accelerates repeated Contains on a fixed Bitmap when successive probes have locality: it returns the identical result to Bitmap.Contains for any x in any order, and is cheapest when probes repeat the previous container or advance past it (same-container repeats skip the key search; forward moves gallop from the cached position at both the container and the array-value level; backward moves fall back to the plain search).
Not safe for concurrent use; the bound Bitmap must not be mutated while the cursor is live.
func (*ContainsCursor) Contains ¶ added in v0.0.15
func (cur *ContainsCursor) Contains(x uint64) bool
Contains reports whether x is set, for any x in any order (cheapest when x repeats the previous container or advances past it).
func (*ContainsCursor) NextGeq ¶ added in v0.0.15
func (cur *ContainsCursor) NextGeq(x uint64) (uint64, bool)
NextGeq returns the smallest set value >= x and whether one exists. It shares the cursor's cache and leaves the cursor positioned on the returned value, so interleaving Contains and NextGeq with non-decreasing arguments keeps both on their fast paths. This is the successor primitive for leapfrog intersection: instead of probing candidates one by one, a caller can jump directly to the bitmap's next admissible value.
func (*ContainsCursor) Reset ¶ added in v0.0.15
func (cur *ContainsCursor) Reset(ra *Bitmap)
Reset re-binds the cursor to ra and clears its cache, letting callers embed a ContainsCursor by value and prime it without a heap allocation.