BitIncrementIterator.h

 
#pragma once
 
#include <algorithm>
#include <cstddef>
#include <iosfwd>
#include <iterator>
#include <vector>
 
namespace dlinear {
 
class BitIncrementIterator {
 public:
  using iterator_category = std::input_iterator_tag;
  using value_type = std::vector<bool>;
  using reference = value_type const &;
  using pointer = value_type const *;
  using difference_type = ptrdiff_t;
 
  explicit BitIncrementIterator(std::size_t n)
      : vector_(n, false), fixed_(n, false), starting_vector_(n, false), ending_vector_(n, true) {}
  explicit BitIncrementIterator(std::vector<bool> starting_value);
 
  explicit operator bool() const { return !vector_.empty(); }
 
  pointer operator->() const { return &vector_; }
  reference operator*() const { return vector_; }
 
  bool operator==(const BitIncrementIterator &rhs) const { return vector_ == rhs.vector_; }
  bool operator!=(const BitIncrementIterator &rhs) const { return vector_ != rhs.vector_; }
 
  BitIncrementIterator &operator++();
  const BitIncrementIterator operator++(int);
 
  BitIncrementIterator &operator--();
  const BitIncrementIterator operator--(int);
 
  bool operator[](std::size_t i) const;
 
  [[nodiscard]] bool IsFixed(std::size_t i) const { return fixed_[i]; }
 
  void Set(std::size_t i, bool value, bool force = false) { vector_[i] = force || !fixed_[i] ? value : vector_[i]; }
 
  void SetFixed(std::size_t i, bool fixed) { fixed_[i] = fixed; }
 
  bool Learn(std::size_t i);
  bool Learn(std::size_t i, bool value);
 
 private:
  void ResetNonFixedRight(std::size_t start_pos = 0);
  void ResetNonFixedLeft(std::size_t start_pos);
  void ResetNonFixed();
  [[nodiscard]] bool IsDone() const;
 
  void UpdateVector(std::size_t i, bool value);
 
  std::vector<bool> vector_;           
  std::vector<bool> fixed_;            
  std::vector<bool> starting_vector_;  
  std::vector<bool> ending_vector_;    
};
 
std::vector<bool> &operator++(std::vector<bool> &vector);
std::vector<bool> &operator--(std::vector<bool> &vector);
std::ostream &operator<<(std::ostream &os, const BitIncrementIterator &it);
 
}  // namespace dlinear
 
#ifdef DLINEAR_INCLUDE_FMT
 
#include "dlinear/util/logging.h"
 
OSTREAM_FORMATTER(dlinear::BitIncrementIterator)
 
#endif