HierNetlist.hpp

Go to the documentation of this file.

 
#pragma once
 
#include <ckpttn/array_like.hpp>  // for ShiftArray
#include <cstdint>                // for uint8_t
#include <netlistx/netlist.hpp>   // for Netlist, Netlist<>::nodeview_t
#include <py2cpp/dict.hpp>        // for dict
#include <py2cpp/set.hpp>         // for set
#include <span>                   // for span
// #include <type_traits>                 // for move
#include <vector>                      // for vector
#include <xnetwork/classes/graph.hpp>  // for SimpleGraph, Graph, Graph<>::n...
 
// using node_t = int;
 
// struct PartInfo
// {
//     std::vector<std::uint8_t> part;
//     py::set<node_t> extern_nets;
// };
 
template <typename graph_t> class HierNetlist : public Netlist<graph_t> {
  public:
    using nodeview_t = typename graph_t::nodeview_t;
    using node_t = typename graph_t::node_t;
    using index_t = typename nodeview_t::key_type;
 
    /* For multi-level algorithms */
    const Netlist<graph_t>* parent;
    std::vector<node_t> node_up_map;
    std::vector<node_t> node_down_map;
    py::dict<index_t, node_t> cluster_down_map;
    ShiftArray<std::vector<uint32_t>> net_weight;
 
    HierNetlist(graph_t gr, const nodeview_t& modules, const nodeview_t& nets);
 
    void projection_down(std::span<const std::uint8_t> part,
                         std::span<std::uint8_t> part_down) const;
 
    void projection_up(std::span<const std::uint8_t> part, std::span<std::uint8_t> part_up) const;
 
    auto get_net_weight(const node_t& net) const -> uint32_t {
        return this->net_weight.empty() ? 1U : this->net_weight[net];
    }
};
 
template <typename graph_t>
HierNetlist<graph_t>::HierNetlist(graph_t gr, const nodeview_t& modules, const nodeview_t& nets)
    : Netlist<graph_t>{std::move(gr), modules, nets} {}
 
// template <typename graph_t>
// HierNetlist<graph_t>::HierNetlist(graph_t gr, uint32_t numModules, uint32_t
// numNets)
//     : Netlist<graph_t> {std::move(gr), py::range<uint32_t>(numModules),
//           py::range<uint32_t>(numModules, numModules + numNets)}
// {
// }
 
using SimpleHierNetlist = HierNetlist<xnetwork::SimpleGraph>;