rr_graph_util: add util class

Signed-off-by: Alessandro Comodi <[email protected]>

Author: acomodi

vpr/src/route/rr_graph_util.cpp

@@ -2,14 +2,15 @@
 #include <random>
 #include <algorithm>
 
+#include "rr_graph_util.h"
+
 #include "vtr_memory.h"
 #include "vtr_time.h"
 
 #include "vpr_types.h"
 #include "vpr_error.h"
 
 #include "globals.h"
-#include "rr_graph_util.h"
 
 int seg_index_of_cblock(t_rr_type from_rr_type, int to_node) {
     /* Returns the segment number (distance along the channel) of the connection *
@@ -177,3 +178,20 @@ void reorder_rr_graph_nodes(const t_router_opts& router_opts) {
                                std::get<2>(edge));
     });
 }
+
+void t_rr_graph_util::init_fan_in_list() {
+    auto& rr_nodes = g_vpr_ctx.device().rr_nodes;
+
+    node_fan_in_list_.resize(rr_nodes.size(), std::vector<RREdgeId>(0));
+    node_fan_in_list_.shrink_to_fit();
+
+    //Walk the graph and increment fanin on all dwnstream nodes
+    rr_nodes.for_each_edge(
+        [&](RREdgeId edge, __attribute__((unused)) RRNodeId src, RRNodeId sink) {
+            node_fan_in_list_[sink].push_back(edge);
+        });
+}
+
+t_rr_graph_util::~t_rr_graph_util() {
+    node_fan_in_list_.clear();
+}

vpr/src/route/rr_graph_util.h

@@ -1,10 +1,31 @@
 #ifndef RR_GRAPH_UTIL_H
 #define RR_GRAPH_UTIL_H
 
+#include "vpr_types.h"
+
 int seg_index_of_cblock(t_rr_type from_rr_type, int to_node);
 
 int seg_index_of_sblock(int from_node, int to_node);
 
 void reorder_rr_graph_nodes(const t_router_opts& router_opts);
 
+// t_rr_graph_util is a utility class that collects utility
+// methods that do not strictly belong to the RR graph storage,
+// as they can use its public interface.
+class t_rr_graph_util {
+    public:
+        ~t_rr_graph_util();
+
+        void init_fan_in_list();
+
+        /* Returns a fan-in edge list corresponding to a specific node */
+        std::vector<RREdgeId> fan_in_list(RRNodeId id) const {
+            return node_fan_in_list_[id];
+        }
+
+    private:
+        // Fan in edges for each RR node.
+        vtr::vector<RRNodeId, std::vector<RREdgeId>> node_fan_in_list_;
+};
+
 #endif