FMConstrMgr.hpp

Go to the documentation of this file.

 
#pragma once
 
#include <cstdint>  // for uint8_t
#include <span>     // for span
#include <vector>   // for vector
 
// #include "moveinfo.hpp"  // for MoveInfo
 
// forward declare
template <typename Node> struct MoveInfo;
template <typename Node> struct MoveInfoV;
 
enum class LegalCheck { NotSatisfied, GetBetter, AllSatisfied };
 
template <typename Gnl> class FMConstrMgr {
  private:
    const Gnl& hyprgraph;
    double bal_tol;
    unsigned int total_weight{0};
    unsigned int weight{};
 
  protected:
    std::vector<unsigned int> diff;
    unsigned int lowerbound{};
    std::uint8_t num_parts;
 
    using node_t = typename Gnl::node_t;
 
    FMConstrMgr(const Gnl& hyprgraph, double bal_tol) : FMConstrMgr(hyprgraph, bal_tol, 2) {}
 
    FMConstrMgr(const Gnl& hyprgraph, double bal_tol, std::uint8_t num_parts);
 
  public:
    auto init(std::span<const std::uint8_t> part) -> void;
 
    auto check_legal(const MoveInfoV<node_t>& move_info_v) -> LegalCheck;
 
    auto check_constraints(const MoveInfoV<node_t>& move_info_v) -> bool;
 
    auto update_move(const MoveInfoV<node_t>& move_info_v) -> void;
 
    auto final_check(std::span<const std::uint8_t> part) -> bool;
};