Program Listing for File QuorumMessage.hpp

↰ Return to documentation for file (include/dish2/quorum/QuorumMessage.hpp)

#pragma once
#ifndef DISH2_QUORUM_QUORUMMESSAGE_HPP_INCLUDE
#define DISH2_QUORUM_QUORUMMESSAGE_HPP_INCLUDE

#include <algorithm>
#include <iostream>

#include "../../../third-party/conduit/include/uitsl/algorithm/unset_mask.hpp"
#include "../../../third-party/Empirical/include/emp/bits/BitSet.hpp"
#include "../../../third-party/signalgp-lite/include/sgpl/utility/ThreadLocalRandom.hpp"

#include "../config/cfg.hpp"

namespace dish2 {

template< typename Spec >
class QuorumMessage {

public:

  static constexpr size_t bitset_width = 64;

private:

  // TODO swtich back to emp::array after https://github.com/devosoft/Empirical/issues/408 is resolved
  std::array< emp::BitSet<bitset_width>, Spec::NLEV > data{};

public:

  QuorumMessage() = default;

  QuorumMessage(std::in_place_t) {
    auto& rand = sgpl::tlrand.Get();
    for ( size_t lev{}; lev < Spec::NLEV; ++lev ) {
      if ( rand.P( dish2::cfg.P_SET_QUORUM_BIT()[lev] ) ) {
        data[lev].Set( rand.GetUInt( data[lev].GetSize() ) );
      }
    }
  }

  bool operator==(const QuorumMessage& other) const {
    return data == other.data;
  }

  void UnsetMask( const QuorumMessage& mask ) {
    std::transform(
      std::begin( data ),
      std::end( data ),
      std::begin( mask.data ),
      std::begin( data ),
      [](const auto data_bits, const auto mask_bits){
        return uitsl::unset_mask( data_bits, mask_bits );
      }
    );
  }

  QuorumMessage operator~() const {
    QuorumMessage res;
    std::transform(
      std::begin( data ),
      std::end( data ),
      std::begin( res.data ),
      []( const auto& this_bits ){ return ~this_bits; }
    );
    return res;
  }

  QuorumMessage operator|(const QuorumMessage& other) const {
    QuorumMessage res;
    std::transform(
      std::begin( data ),
      std::end( data ),
      std::begin( other.data ),
      std::begin( res.data ),
      []( const auto& this_bits, const auto& other_bits ){
        return this_bits | other_bits;
      }
    );
    return res;
  }

  QuorumMessage operator&(const QuorumMessage& other) const {
    QuorumMessage res;
    std::transform(
      std::begin( data ),
      std::end( data ),
      std::begin( other.data ),
      std::begin( res.data ),
      []( const auto& this_bits, const auto& other_bits ){
        return this_bits & other_bits;
      }
    );
    return res;
  }

  const QuorumMessage& operator|=(const QuorumMessage& other) {
    std::transform(
      std::begin( data ),
      std::end( data ),
      std::begin( other.data ),
      std::begin( data ),
      []( const auto& this_bits, const auto& other_bits ){
        return this_bits | other_bits;
      }
    );
    return *this;
  }

  void Clear() { for (auto& bs : data) bs.Clear(); }

  void ClearLev( const size_t lev ) { data[ lev ].Clear(); }

  void SetBit( const size_t lev, const size_t which_bit ) {
    data[ lev ][ which_bit ] = 1;
  }

  size_t GetNumBits(const size_t lev) const { return data[ lev ].CountOnes(); }

  bool GetBit(const size_t lev, const size_t bit) const {
    return data[ lev ][ bit ];
  }

  size_t GetNumBits() const {
    return std::accumulate(
      std::begin( data ), std::end( data ),
      size_t{},
      [](const auto& accumulator, const auto& bitset){
        return accumulator + bitset.CountOnes();
      }
    );
  }

  friend std::ostream& operator<<(std::ostream &out, const QuorumMessage& qm){
    for ( const auto& bs : qm.data ) out << bs;
    return out;
  }


};

} // namespace dish2

#endif // #ifndef DISH2_QUORUM_QUORUMMESSAGE_HPP_INCLUDE