aboutsummaryrefslogtreecommitdiffstats
path: root/library/cpp/containers/bitseq/bititerator.h
diff options
context:
space:
mode:
authorpkalinnikov <pkalinnikov@yandex-team.ru>2022-02-10 16:50:15 +0300
committerDaniil Cherednik <dcherednik@yandex-team.ru>2022-02-10 16:50:15 +0300
commitd507a9366b2ab84411afe63fea9fba5498891e1b (patch)
tree5d5cb817648f650d76cf1076100726fd9b8448e8 /library/cpp/containers/bitseq/bititerator.h
parent9e33e026829d561d6fd46d72b88c367952e08075 (diff)
downloadydb-d507a9366b2ab84411afe63fea9fba5498891e1b.tar.gz
Restoring authorship annotation for <pkalinnikov@yandex-team.ru>. Commit 2 of 2.
Diffstat (limited to 'library/cpp/containers/bitseq/bititerator.h')
-rw-r--r--library/cpp/containers/bitseq/bititerator.h258
1 files changed, 129 insertions, 129 deletions
diff --git a/library/cpp/containers/bitseq/bititerator.h b/library/cpp/containers/bitseq/bititerator.h
index 616c9bbac4..52dadd3798 100644
--- a/library/cpp/containers/bitseq/bititerator.h
+++ b/library/cpp/containers/bitseq/bititerator.h
@@ -1,138 +1,138 @@
-#pragma once
-
-#include "traits.h"
-
+#pragma once
+
+#include "traits.h"
+
#include <library/cpp/pop_count/popcount.h>
-template <typename T>
-class TBitIterator {
-public:
- using TWord = T;
- using TTraits = TBitSeqTraits<TWord>;
-
-public:
- TBitIterator(const T* data = nullptr)
- : Current(0)
- , Mask(0)
- , Data(data)
+template <typename T>
+class TBitIterator {
+public:
+ using TWord = T;
+ using TTraits = TBitSeqTraits<TWord>;
+
+public:
+ TBitIterator(const T* data = nullptr)
+ : Current(0)
+ , Mask(0)
+ , Data(data)
{
}
-
- /// Get the word next to the one we are currenlty iterating over.
- const TWord* NextWord() const {
- return Data;
- }
-
- /// Get the next bit without moving the iterator.
- bool Peek() const {
- return Mask ? (Current & Mask) : (*Data & 1);
- }
-
- /// Get the next bit and move forward.
- /// TODO: Implement inversed iteration as well.
- bool Next() {
- if (!Mask) {
- Current = *Data++;
- Mask = 1;
- }
- const bool bit = Current & Mask;
- Mask <<= 1;
- return bit;
- }
-
- /// Get the next count bits without moving the iterator.
- TWord Peek(ui8 count) const {
- if (!count)
- return 0;
+
+ /// Get the word next to the one we are currenlty iterating over.
+ const TWord* NextWord() const {
+ return Data;
+ }
+
+ /// Get the next bit without moving the iterator.
+ bool Peek() const {
+ return Mask ? (Current & Mask) : (*Data & 1);
+ }
+
+ /// Get the next bit and move forward.
+ /// TODO: Implement inversed iteration as well.
+ bool Next() {
+ if (!Mask) {
+ Current = *Data++;
+ Mask = 1;
+ }
+ const bool bit = Current & Mask;
+ Mask <<= 1;
+ return bit;
+ }
+
+ /// Get the next count bits without moving the iterator.
+ TWord Peek(ui8 count) const {
+ if (!count)
+ return 0;
Y_VERIFY_DEBUG(count <= TTraits::NumBits);
-
- if (!Mask)
- return *Data & TTraits::ElemMask(count);
-
+
+ if (!Mask)
+ return *Data & TTraits::ElemMask(count);
+
auto usedBits = (size_t)PopCount(Mask - 1);
- TWord result = Current >> usedBits;
- auto leftInCurrent = TTraits::NumBits - usedBits;
- if (count <= leftInCurrent)
- return result & TTraits::ElemMask(count);
-
- count -= leftInCurrent;
- result |= (*Data & TTraits::ElemMask(count)) << leftInCurrent;
- return result;
- }
-
- /// Get the next count bits and move forward by count bits.
- TWord Read(ui8 count) {
- if (!count)
- return 0;
+ TWord result = Current >> usedBits;
+ auto leftInCurrent = TTraits::NumBits - usedBits;
+ if (count <= leftInCurrent)
+ return result & TTraits::ElemMask(count);
+
+ count -= leftInCurrent;
+ result |= (*Data & TTraits::ElemMask(count)) << leftInCurrent;
+ return result;
+ }
+
+ /// Get the next count bits and move forward by count bits.
+ TWord Read(ui8 count) {
+ if (!count)
+ return 0;
Y_VERIFY_DEBUG(count <= TTraits::NumBits);
-
- if (!Mask) {
- Current = *Data++;
- Mask = 1 << count;
- return Current & TTraits::ElemMask(count);
- }
-
+
+ if (!Mask) {
+ Current = *Data++;
+ Mask = 1 << count;
+ return Current & TTraits::ElemMask(count);
+ }
+
auto usedBits = (size_t)PopCount(Mask - 1);
- TWord result = Current >> usedBits;
- auto leftInCurrent = TTraits::NumBits - usedBits;
- if (count < leftInCurrent) {
- Mask <<= count;
- return result & TTraits::ElemMask(count);
- }
-
- count -= leftInCurrent;
- if (count) {
- Current = *Data++;
- Mask = 1 << count;
- result |= (Current & TTraits::ElemMask(count)) << leftInCurrent;
- } else {
- Mask = 0;
- }
-
- return result;
- }
-
- /// Move the iterator forward by count bits.
- void Forward(int count) {
- if (!count)
- return;
-
+ TWord result = Current >> usedBits;
+ auto leftInCurrent = TTraits::NumBits - usedBits;
+ if (count < leftInCurrent) {
+ Mask <<= count;
+ return result & TTraits::ElemMask(count);
+ }
+
+ count -= leftInCurrent;
+ if (count) {
+ Current = *Data++;
+ Mask = 1 << count;
+ result |= (Current & TTraits::ElemMask(count)) << leftInCurrent;
+ } else {
+ Mask = 0;
+ }
+
+ return result;
+ }
+
+ /// Move the iterator forward by count bits.
+ void Forward(int count) {
+ if (!count)
+ return;
+
int leftInCurrent = (size_t)PopCount(~(Mask - 1));
- if (count < leftInCurrent) {
- Mask <<= count;
- return;
- }
-
- count -= leftInCurrent;
- Data += count >> TTraits::DivShift;
- auto remainder = count & TTraits::ModMask;
-
- if (remainder) {
- Current = *Data++;
- Mask = 1 << remainder;
- } else {
- Current = 0;
- Mask = 0;
- }
- }
-
- /// Skip trailing bits of the current word and move by count words forward.
- void Align(int count = 0) {
- Current = 0;
- if (Mask)
- Mask = 0;
- Data += count;
- }
-
- /// Initialize the iterator.
- void Reset(const TWord* data) {
- Current = 0;
- Mask = 0;
- Data = data;
- }
-
-private:
- TWord Current;
- TWord Mask;
- const TWord* Data;
-};
+ if (count < leftInCurrent) {
+ Mask <<= count;
+ return;
+ }
+
+ count -= leftInCurrent;
+ Data += count >> TTraits::DivShift;
+ auto remainder = count & TTraits::ModMask;
+
+ if (remainder) {
+ Current = *Data++;
+ Mask = 1 << remainder;
+ } else {
+ Current = 0;
+ Mask = 0;
+ }
+ }
+
+ /// Skip trailing bits of the current word and move by count words forward.
+ void Align(int count = 0) {
+ Current = 0;
+ if (Mask)
+ Mask = 0;
+ Data += count;
+ }
+
+ /// Initialize the iterator.
+ void Reset(const TWord* data) {
+ Current = 0;
+ Mask = 0;
+ Data = data;
+ }
+
+private:
+ TWord Current;
+ TWord Mask;
+ const TWord* Data;
+};