94 lines
1.9 KiB
C
94 lines
1.9 KiB
C
|
//
|
||
|
|
// Copyright Aliaksei Levin (levlam@telegram.org), Arseny Smirnov (arseny30@gmail.com) 2014-2018
|
||
|
|
//
|
||
|
|
// Distributed under the Boost Software License, Version 1.0. (See accompanying
|
||
|
|
// file LICENSE_1_0.txt or copy at http://www.boost.org/LICENSE_1_0.txt)
|
||
|
|
//
|
||
|
|
#pragma once
|
||
|
|
|
||
|
|
#include "td/utils/common.h"
|
||
|
|
#include "td/utils/Slice.h"
|
||
|
|
|
||
|
|
#include <cstring>
|
||
|
|
|
||
|
|
namespace td {
|
||
|
|
|
||
|
|
template <size_t size>
|
||
|
|
struct UInt {
|
||
|
|
static_assert(size % 8 == 0, "size should be divisible by 8");
|
||
|
|
uint8 raw[size / 8];
|
||
|
|
|
||
|
|
Slice as_slice() const {
|
||
|
|
return Slice(raw, size / 8);
|
||
|
|
}
|
||
|
|
|
||
|
|
MutableSlice as_slice() {
|
||
|
|
return MutableSlice(raw, size / 8);
|
||
|
|
}
|
||
|
|
|
||
|
|
bool is_zero() const {
|
||
|
|
for (size_t i = 0; i < size / 8; i++) {
|
||
|
|
if (raw[i] != 0) {
|
||
|
|
return false;
|
||
|
|
}
|
||
|
|
}
|
||
|
|
return true;
|
||
|
|
}
|
||
|
|
void set_zero() {
|
||
|
|
for (size_t i = 0; i < size / 8; i++) {
|
||
|
|
raw[i] = 0;
|
||
|
|
}
|
||
|
|
}
|
||
|
|
static UInt zero() {
|
||
|
|
UInt v;
|
||
|
|
v.set_zero();
|
||
|
|
return v;
|
||
|
|
}
|
||
|
|
};
|
||
|
|
|
||
|
|
template <size_t size>
|
||
|
|
bool operator==(const UInt<size> &a, const UInt<size> &b) {
|
||
|
|
return std::memcmp(a.raw, b.raw, sizeof(a.raw)) == 0;
|
||
|
|
}
|
||
|
|
|
||
|
|
template <size_t size>
|
||
|
|
bool operator!=(const UInt<size> &a, const UInt<size> &b) {
|
||
|
|
return !(a == b);
|
||
|
|
}
|
||
|
|
|
||
|
|
template <size_t size>
|
||
|
|
td::UInt<size> operator^(const UInt<size> &a, const UInt<size> &b) {
|
||
|
|
td::UInt<size> res;
|
||
|
|
for (size_t i = 0; i < size / 8; i++) {
|
||
|
|
res.raw[i] = static_cast<uint8>(a.raw[i] ^ b.raw[i]);
|
||
|
|
}
|
||
|
|
return res;
|
||
|
|
}
|
||
|
|
|
||
|
|
template <size_t size>
|
||
|
|
int get_kth_bit(const UInt<size> &a, uint32 bit) {
|
||
|
|
uint8 b = a.raw[bit / 8];
|
||
|
|
bit &= 7;
|
||
|
|
return (b >> (7 - bit)) & 1;
|
||
|
|
}
|
||
|
|
|
||
|
|
template <size_t size>
|
||
|
|
Slice as_slice(const UInt<size> &value) {
|
||
|
|
return value.as_slice();
|
||
|
|
}
|
||
|
|
|
||
|
|
template <size_t size>
|
||
|
|
MutableSlice as_slice(UInt<size> &value) {
|
||
|
|
return value.as_slice();
|
||
|
|
}
|
||
|
|
|
||
|
|
template <size_t size>
|
||
|
|
bool operator<(const UInt<size> &a, const UInt<size> &b) {
|
||
|
|
return std::memcmp(a.raw, b.raw, sizeof(a.raw)) < 0;
|
||
|
|
}
|
||
|
|
|
||
|
|
using UInt128 = UInt<128>;
|
||
|
|
using UInt256 = UInt<256>;
|
||
|
|
|
||
|
|
} // namespace td
|