【Hmetis图分割处理网表-具体划分 以gate为节点构建超图】

基本数据结构

std::unordered_map<std::string, int> _gate_name_indexs; // net:(name,index)
std::unordered_map<int, std::string> _gate_index_names; // net:(index,name)
std::unordered_map<std::string, int> _gate_P;          // name->part

核心函数

#include <ot/timer/timer.hpp>
namespace ot
{
   

  /*
    输出gate的绝对排序文件，为之后的GNN做铺垫
  */
  //================输出gate的绝对排序文件//================
  void Timer::dump_gate_sort(std::ostream &os) const
  {
   
    std::shared_lock lock(_mutex);
    _dump_gate_sort(os);
  }
  void Timer::_dump_gate_sort(std::ostream &os) const
  {
   
    for (const auto &[gate_name, gate] : _gates)
    {
   
      os << gate_name << std::endl;
    }
  }

 
  


  //================输出超图，以gate节点的超图文件//================
  void Timer::dump_hypergraph_gate(std::ostream &os)
  {
   
    std::shared_lock lock(_mutex);
    _dump_hypergraph_gate(os);
  }
  //输出超图文件，以gate作为节点，net作为超边
  // os: 为输出文件
  // gate_sort
  void Timer::_dump_hypergraph_gate(std::ostream &os)
  {
   
    std::string gatename;
    std::string buffer;
    int net_size = 0;
    std::unordered_set<std::string> uset; // set，记录所有筛掉后超边所连的gate
    //遍历net，将net超边所连gate输出
    for (const auto &[net_name, net] : _nets)
    {
   
      if (net.is_normal_net())
      {
   
        for (const auto pin : net._pins)
        {
   
          std::string gateName = pin->gate()->name();
          
          auto gate = _gate_name_indexs.find(gateName);
          assert(gate != _gate_name_indexs.end());
          buffer += to_string(gate->second) + " ";
          uset.emplace(gateName);
        }
        buffer += "\n";
        net_size++;
      }
    }
    assert(uset.size() <= _gate_name_indexs.size());
    //输出net_size ,gate_size
    os << net_size << " " << _gate_name_indexs.size() << "\n";
    //输出超边所连接的所有gate信息（index start from 0）
    os << buffer;
  }


  
  //================读取gate的绝对排序文件//================
  void Timer::read_gate_sort(std::filesystem::path path)
  {
   
    std::shared_lock lock(_mutex);
    std::ifstream ifs(path);
    assert(ifs.good());
    std::string gatename;
    int index = 1; // start from 1 gate index
    while (getline(ifs, gatename