68 lines
1.5 KiB
C++
68 lines
1.5 KiB
C++
|
#include "FitVector.hpp"
|
||
|
|
|
||
|
|
namespace xcdat {
|
||
|
|
|
||
|
|
FitVector::FitVector(const std::vector<id_type>& integers) {
|
||
|
|
if (integers.empty()) {
|
||
|
|
return;
|
||
|
|
}
|
||
|
|
|
||
|
|
width_ = 0;
|
||
|
|
auto max_value = *std::max_element(std::begin(integers), std::end(integers));
|
||
|
|
do {
|
||
|
|
++width_;
|
||
|
|
max_value >>= 1;
|
||
|
|
} while (max_value);
|
||
|
|
|
||
|
|
size_ = integers.size();
|
||
|
|
mask_ = (1U << width_) - 1;
|
||
|
|
std::vector<id_type> chunks(size_ * width_ / kChunkWidth + 1, 0);
|
||
|
|
|
||
|
|
for (id_type i = 0; i < size_; ++i) {
|
||
|
|
const auto chunk_pos = i * width_ / kChunkWidth;
|
||
|
|
const auto offset = i * width_ % kChunkWidth;
|
||
|
|
|
||
|
|
chunks[chunk_pos] &= ~(mask_ << offset);
|
||
|
|
chunks[chunk_pos] |= (integers[i] & mask_) << offset;
|
||
|
|
|
||
|
|
if (kChunkWidth < offset + width_) {
|
||
|
|
chunks[chunk_pos + 1] &= ~(mask_ >> (kChunkWidth - offset));
|
||
|
|
chunks[chunk_pos + 1] |= (integers[i] & mask_) >> (kChunkWidth - offset);
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
chunks_.steal(chunks);
|
||
|
|
}
|
||
|
|
|
||
|
|
size_t FitVector::size_in_bytes() const {
|
||
|
|
size_t ret = 0;
|
||
|
|
ret += chunks_.size_in_bytes();
|
||
|
|
ret += sizeof(size_);
|
||
|
|
ret += sizeof(width_);
|
||
|
|
ret += sizeof(mask_);
|
||
|
|
return ret;
|
||
|
|
}
|
||
|
|
|
||
|
|
void FitVector::write(std::ostream& os) const {
|
||
|
|
chunks_.write(os);
|
||
|
|
write_value(size_, os);
|
||
|
|
write_value(width_, os);
|
||
|
|
write_value(mask_, os);
|
||
|
|
}
|
||
|
|
|
||
|
|
void FitVector::read(std::istream& is) {
|
||
|
|
chunks_.read(is);
|
||
|
|
read_value(size_, is);
|
||
|
|
read_value(width_, is);
|
||
|
|
read_value(mask_, is);
|
||
|
|
}
|
||
|
|
|
||
|
|
void FitVector::swap(FitVector& rhs) {
|
||
|
|
chunks_.swap(rhs.chunks_);
|
||
|
|
std::swap(size_, rhs.size_);
|
||
|
|
std::swap(width_, rhs.width_);
|
||
|
|
std::swap(mask_, rhs.mask_);
|
||
|
|
}
|
||
|
|
|
||
|
|
} //namespace - xcdat
|