NNPartMgr.hpp

Go to the documentation of this file.

 
#pragma once
 
// **Special code for two-pin nets**
 
#include <cstdint>  // for uint8_t
#include <span>     // for span
// #include <xnetwork/classes/graph.hpp>
 
// forward declare
// template <typename graph_t> struct Netlist;
// using SimpleNetlist = Netlist<xnetwork::SimpleGraph>;
 
enum class LegalCheck;
 
template <typename Gnl, typename GainMgr, typename ConstrMgr>  //
class NNPartMgr {
  public:
    using GainCalc_ = typename GainMgr::GainCalc_;
    using GainMgr_ = GainMgr;
    using ConstrMgr_ = ConstrMgr;
 
    // using Der = Derived<Gnl, GainMgr, ConstrMgr>;
 
  protected:
    const Gnl& hyprgraph;
    GainMgr& gain_mgr;
    ConstrMgr& validator;
    size_t num_parts;
    // std::vector<std::uint8_t> snapshot;
    // std::vector<std::uint8_t> part;
 
  public:
    int total_cost{};
 
    NNPartMgr(const Gnl& hyprgraph, GainMgr& gain_mgr, ConstrMgr& constr_mgr, size_t num_parts)
        : hyprgraph{hyprgraph}, gain_mgr{gain_mgr}, validator{constr_mgr}, num_parts{num_parts} {}
 
    void init(std::span<std::uint8_t> part);
 
    auto legalize(std::span<std::uint8_t> part) -> LegalCheck;
 
    void optimize(std::span<std::uint8_t> part);
 
  private:
    void _optimize_1pass(std::span<std::uint8_t> part);
 
    auto final_check(std::span<const std::uint8_t> part) -> bool {
        return this->validator.final_check(part);
    }
};