initial commit

crypto/vm/cells/LevelMask.h

@@ -0,0 +1,84 @@
+/*
+    This file is part of TON Blockchain Library.
+
+    TON Blockchain Library is free software: you can redistribute it and/or modify
+    it under the terms of the GNU Lesser General Public License as published by
+    the Free Software Foundation, either version 2 of the License, or
+    (at your option) any later version.
+
+    TON Blockchain Library is distributed in the hope that it will be useful,
+    but WITHOUT ANY WARRANTY; without even the implied warranty of
+    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
+    GNU Lesser General Public License for more details.
+
+    You should have received a copy of the GNU Lesser General Public License
+    along with TON Blockchain Library.  If not, see <http://www.gnu.org/licenses/>.
+
+    Copyright 2017-2019 Telegram Systems LLP
+*/
+#pragma once
+
+#include "td/utils/bits.h"
+#include "td/utils/int_types.h"
+#include "td/utils/StringBuilder.h"
+
+namespace td {
+class StringBuilder;
+}
+
+namespace vm {
+namespace detail {
+class LevelMask {
+ public:
+  explicit LevelMask(td::uint32 new_mask = 0) : mask_(new_mask) {
+  }
+  td::uint32 get_mask() const {
+    return mask_;
+  }
+  td::uint32 get_level() const {
+    return 32 - td::count_leading_zeroes32(mask_);
+  }
+  td::uint32 get_hash_i() const {
+    return td::count_bits32(mask_);
+  }
+  td::uint32 get_hashes_count() const {
+    return get_hash_i() + 1;
+  }
+  LevelMask apply(td::uint32 level) const {
+    DCHECK(level < 32);
+    return LevelMask{mask_ & ((1u << level) - 1)};
+  }
+  LevelMask apply_or(LevelMask other) const {
+    return LevelMask{mask_ | other.mask_};
+  }
+  LevelMask shift_right() const {
+    return LevelMask{mask_ >> 1};
+  }
+  bool is_significant(td::uint32 level) const {
+    DCHECK(level < 32);
+    bool res = level == 0 || ((mask_ >> (level - 1)) % 2 != 0);
+    CHECK(res == (apply(level).get_level() == level));
+    return res;
+  }
+
+  bool operator==(const LevelMask& other) const {
+    return mask_ == other.mask_;
+  }
+  bool operator!=(const LevelMask& other) const {
+    return !(*this == other);
+  }
+
+  static LevelMask one_level(td::uint32 level) {
+    DCHECK(level < 32);
+    if (level == 0) {
+      return LevelMask(0);
+    }
+    return LevelMask(1 << (level - 1));
+  }
+
+ private:
+  td::uint32 mask_;
+};
+td::StringBuilder& operator<<(td::StringBuilder& sb, LevelMask level_mask);
+}  // namespace detail
+}  // namespace vm