aligned.h

#pragma once
 
#include <pixie/storage.h>
#include <pixie/storage/read_only_view.h>
 
#include <algorithm>
#include <array>
#include <cstddef>
#include <cstdint>
#include <limits>
#include <span>
#include <stdexcept>
#include <vector>
 
namespace pixie {
 
inline constexpr std::size_t kAlignedStorageLineBytes = 64;
inline constexpr std::size_t kAlignedStorageLineBits =
    kAlignedStorageLineBytes * 8;
inline constexpr std::size_t kAlignedStorageLineWords64 =
    kAlignedStorageLineBytes / sizeof(std::uint64_t);
inline constexpr std::size_t kAlignedStorageLineWords16 =
    kAlignedStorageLineBytes / sizeof(std::uint16_t);

struct alignas(kAlignedStorageLineBytes) CacheLine {
  std::array<std::byte, kAlignedStorageLineBytes> data{};
};
 
static_assert(alignof(CacheLine) == kAlignedStorageLineBytes);
static_assert(sizeof(CacheLine) == kAlignedStorageLineBytes);

class AlignedStorage : public StorageBase<AlignedStorage> {
 public:
  AlignedStorage() = default;

  explicit AlignedStorage(std::size_t size_bits)
      : logical_size_bytes_(bytes_for_bits(size_bits)),
        data_(lines_for_bits(size_bits)) {}

  explicit AlignedStorage(std::span<const std::byte> bytes)
      : logical_size_bytes_(bytes.size()),
        data_(lines_for_bytes(bytes.size())) {
    std::ranges::copy(bytes, writable_bytes_impl().begin());
  }

  explicit AlignedStorage(std::span<const std::uint64_t> words)
      : AlignedStorage(std::as_bytes(words)) {}

  std::size_t size_bytes_impl() const { return logical_size_bytes_; }

  position_type begin_position_impl() const { return 0; }

  position_type end_position_impl() const { return logical_size_bytes_; }

  std::size_t logical_size_bytes() const { return logical_size_bytes_; }

  std::size_t padded_size_bytes() const {
    return data_.size() * kAlignedStorageLineBytes;
  }

  ReadOnlyStorageView padded_view() const {
    return ReadOnlyStorageView(
        std::as_bytes(std::span<const CacheLine>(data_)));
  }

  std::span<const std::byte> as_bytes_impl() const {
    return std::as_bytes(std::span<const CacheLine>(data_))
        .first(logical_size_bytes_);
  }

  SplitSpan<const std::byte> segments_impl(std::size_t offset_bytes,
                                           std::size_t count_bytes) const {
    return SplitSpan(as_bytes_impl().subspan(offset_bytes, count_bytes));
  }

  SplitSpan<std::byte> segments_impl(std::size_t offset_bytes,
                                     std::size_t count_bytes) {
    return SplitSpan(writable_bytes_impl().subspan(offset_bytes, count_bytes));
  }

  ReadOnlyStorageView view_impl(std::size_t offset_bytes,
                                std::size_t count_bytes) const {
    if (offset_bytes > size_bytes_impl() ||
        count_bytes > size_bytes_impl() - offset_bytes) {
      throw std::out_of_range("Storage view is outside the allocation");
    }
    return ReadOnlyStorageView(
        as_bytes_impl().subspan(offset_bytes, count_bytes));
  }

  void resize_impl(std::size_t size_bits) {
    data_.resize(lines_for_bits(size_bits));
    logical_size_bytes_ = bytes_for_bits(size_bits);
  }

  std::span<std::byte> writable_bytes_impl() {
    return std::as_writable_bytes(std::span<CacheLine>(data_))
        .first(logical_size_bytes_);
  }

  std::span<std::uint16_t> writable_words16_impl() {
    return {reinterpret_cast<std::uint16_t*>(data_.data()),
            logical_size_bytes_ / sizeof(std::uint16_t)};
  }

  std::span<std::uint64_t> writable_words64_impl() {
    return {reinterpret_cast<std::uint64_t*>(data_.data()),
            logical_size_bytes_ / sizeof(std::uint64_t)};
  }

  std::size_t allocated_bytes_impl() const {
    return data_.capacity() * kAlignedStorageLineBytes;
  }

  void shrink_to_fit_impl() { data_.shrink_to_fit(); }

  std::span<CacheLine> as_lines() { return data_; }

  std::span<const CacheLine> as_lines() const { return data_; }

  static AlignedStorage deserialize_impl(BinaryReader& reader) {
    const std::size_t size = reader.read_size();
    if (size > std::numeric_limits<std::size_t>::max() / 8) {
      throw std::length_error("Serialized aligned storage is too large");
    }
    AlignedStorage result(size * 8);
    const std::span<const std::byte> bytes = reader.read_bytes(size);
    std::ranges::copy(bytes, result.writable_bytes().begin());
    return result;
  }
 
 private:
  static constexpr std::size_t bytes_for_bits(std::size_t size_bits) {
    return size_bits / 8 + (size_bits % 8 != 0);
  }
 
  static constexpr std::size_t lines_for_bits(std::size_t size_bits) {
    return lines_for_bytes(bytes_for_bits(size_bits));
  }
 
  static constexpr std::size_t lines_for_bytes(std::size_t size_bytes) {
    return size_bytes / kAlignedStorageLineBytes +
           (size_bytes % kAlignedStorageLineBytes != 0);
  }
 
  std::size_t logical_size_bytes_ = 0;
  std::vector<CacheLine> data_;
};
 
}  // namespace pixie