brick_pool.hpp

Go to the documentation of this file.

 
#pragma once
#include "brick.hpp"
#include "voxel_constants.hpp"
 
#include <atomic>
#include <bit>
#include <cstdint>
#include <memory>
#include <span>
#include <toast/export.hpp>
#include <vector>
 
namespace voxel {
 
inline constexpr size_t k_brick_material_bytes = static_cast<size_t>(k_brick_voxel_count);
 
inline constexpr uint32_t k_invalid_brick = k_brick_payload_mask;

class TOAST_API BrickPool {
public:
  explicit BrickPool(uint32_t capacity);

  [[nodiscard]]
  auto allocate() -> uint32_t;

  void free(uint32_t id);
 
  [[nodiscard]]
  auto material(uint32_t id) -> std::span<uint8_t, k_brick_material_bytes>;
 
  [[nodiscard]]
  auto material(uint32_t id) const -> std::span<const uint8_t, k_brick_material_bytes>;
 
  [[nodiscard]]
  auto occupancy(uint32_t id) -> BrickOccupancy&;
 
  [[nodiscard]]
  auto occupancy(uint32_t id) const -> const BrickOccupancy&;
 
  [[nodiscard]]
  auto capacity() const noexcept -> uint32_t {
    return m_capacity;
  }
 
  [[nodiscard]]
  auto allocatedCount() const noexcept -> uint32_t {
    return m_allocated.load(std::memory_order_relaxed);
  }
 
  [[nodiscard]]
  auto freeCount() const noexcept -> uint32_t {
    return m_capacity - allocatedCount();
  }
 
  [[nodiscard]]
  auto isValid(uint32_t id) const noexcept -> bool {
    return id < m_next_unused.load(std::memory_order_acquire);
  }
 
private:
  void clearBrick(uint32_t id);
 
  uint32_t m_capacity = 0;
 
  std::atomic<uint32_t> m_next_unused {0};
 
  std::atomic<uint32_t> m_allocated {0};
 
  std::atomic<uint64_t> m_free_head;
  std::unique_ptr<std::atomic<uint32_t>[]> m_next_free;    // NOLINT(modernize-avoid-c-arrays)
  std::unique_ptr<std::atomic<bool>[]> m_is_free;          // NOLINT(modernize-avoid-c-arrays)
  std::unique_ptr<uint8_t[]> m_material;                   // NOLINT(modernize-avoid-c-arrays)
  std::vector<BrickOccupancy> m_occupancy;
};
 
static_assert(std::endian::native == std::endian::little, "the brick material layer assumes a little-endian host");
 
}