-
Notifications
You must be signed in to change notification settings - Fork 0
/
bitsetIterator.go
54 lines (45 loc) · 978 Bytes
/
bitsetIterator.go
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
package cof
import (
"github.com/RoaringBitmap/roaring"
)
type bitsetIterator struct {
size int
rows int
rbm *roaring.Bitmap
vector []ColumnValue
}
func NewBitsetIterator(vectorSize int, start, end int64, buf []byte) (Iterator, error) {
rbm := roaring.New()
err := rbm.UnmarshalBinary(buf[start:end])
if err != nil {
return nil, err
}
vals := make([]ColumnValue, vectorSize)
return &bitsetIterator{
rows: 0,
size: vectorSize,
vector: vals,
rbm: rbm,
}, nil
}
func (iter *bitsetIterator) Reset() {
for idx, cv := range iter.vector {
cv.Reset()
iter.vector[idx] = cv
}
}
func (iter *bitsetIterator) Next() (*[]ColumnValue, error) {
iter.Reset()
for i := 0; i < iter.size; i++ {
val := iter.rbm.ContainsInt(iter.rows)
iter.vector[i] = ColumnValue{Index: i, Value: val}
iter.rows++
}
return &iter.vector, nil
}
func (iter *bitsetIterator) Err() error {
return nil
}
func (iter *bitsetIterator) Pos() int64 {
return 0
}