bitfield.hpp

Go to the documentation of this file.

#pragma once
 
#include <cstdint>
#include <ostream>
#include <vector>
 
#include "tfriend_fw.hpp"
 
namespace fur::download::bitfield {
class Bitfield {
 public:
  explicit Bitfield(int64_t len);
 
  Bitfield(std::vector<uint8_t> storage, int64_t len)
      : len{len}, storage{std::move(storage)} {}
 
  void set(int64_t index, bool value);
 
  void set(int64_t index) { set(index, true); }
  void unset(int64_t index) { set(index, false); }
 
  [[nodiscard]] bool get(int64_t index) const;
 
  [[nodiscard]] std::vector<uint8_t> get_bytes() const {
    // Basically just return a copy of `storage`.
    return storage;
  }
 
  friend std::ostream& operator<<(std::ostream& os, const Bitfield& bitfield);
 
  const int64_t len;
 
 private:
  std::vector<uint8_t> storage;
 
  // Befriend this class so the unit tests are able to access private members.
  friend TestingFriend;
};
}  // namespace fur::download::bitfield