added basic intra cluster routing visualization

Author: SamuelHo10

vpr/src/draw/draw_basic.cpp

@@ -607,6 +607,7 @@ void draw_routed_net(ParentNetId net_id, ezgl::renderer* g) {
 //Draws the set of rr_nodes specified, using the colors set in draw_state
 void draw_partial_route(const std::vector<RRNodeId>& rr_nodes_to_draw, ezgl::renderer* g) {
     t_draw_state* draw_state = get_draw_state_vars();
+    t_draw_coords* draw_coords = get_draw_coords_vars();
     auto& device_ctx = g_vpr_ctx.device();
     const auto& rr_graph = device_ctx.rr_graph;
 
@@ -632,11 +633,6 @@ void draw_partial_route(const std::vector<RRNodeId>& rr_nodes_to_draw, ezgl::ren
             for (int j = 1; j < height - 1; j++)
                 chany_track[i][j] = (-1);
     }
-    // VTR_LOG("Start");
-    // for (RRNodeId inode : rr_nodes_to_draw) {
-    //     VTR_LOG( "%d\n", rr_graph.node_type(inode));
-    // }
-    // VTR_LOG("End");
 
     for (size_t i = 1; i < rr_nodes_to_draw.size(); ++i) {
         RRNodeId inode = rr_nodes_to_draw[i];
@@ -659,27 +655,21 @@ void draw_partial_route(const std::vector<RRNodeId>& rr_nodes_to_draw, ezgl::ren
 
         ezgl::color color = draw_state->draw_rr_node[inode].color;
 
-
-
-
+        // Skip drawing sources and sinks
         if (rr_graph.node_type(inode) == e_rr_type::SINK || rr_graph.node_type(inode) == e_rr_type::SOURCE || rr_graph.node_type(prev_node) == e_rr_type::SINK || rr_graph.node_type(prev_node) == e_rr_type::SOURCE) {
             continue;
         }
 
         if (!is_inode_inter_cluster && !is_prev_node_inter_cluster) {
 
+            auto blk_id_pin_id1 = get_rr_node_cluster_blk_id_pb_graph_pin(inode);
+            auto blk_id_pin_id2 = get_rr_node_cluster_blk_id_pb_graph_pin(prev_node);
 
+            ezgl::point2d p1 = draw_coords->get_absolute_pin_location(blk_id_pin_id1.first, blk_id_pin_id1.second);
+            ezgl::point2d p2 = draw_coords->get_absolute_pin_location(blk_id_pin_id2.first, blk_id_pin_id2.second);
 
-            // AtomPinId pin_id1 = get_rr_node_atom_pin_id(inode);
-            // AtomPinId pin_id2 = get_rr_node_atom_pin_id(prev_node);
-
-            // VTR_ASSERT(pin_id1 != AtomPinId::INVALID() && pin_id2 != AtomPinId::INVALID());
-
-            // ezgl::point2d p1 = atom_pin_draw_coord(pin_id1);
-            // ezgl::point2d p2 = atom_pin_draw_coord(pin_id2);
-
-            // g->set_color(color, edge_visibility.alpha);
-            // g->draw_line(p1, p2);
+            g->set_color(color, edge_visibility.alpha);
+            g->draw_line(p1, p2);
 
             continue;
         }

vpr/src/draw/draw_types.cpp

@@ -156,9 +156,15 @@ ezgl::point2d t_draw_coords::get_absolute_pin_location( const ClusterBlockId clb
     t_pb_graph_node* pb_gnode = pb_graph_pin->parent_node;
     ezgl::rectangle pb_bbox = this->get_absolute_pb_bbox(clb_index, pb_gnode);
     int num_pins = pb_gnode->num_pins();
-    // int pin_index =
+    int num_pin_in_port = pb_graph_pin->pin_number;
+    int num_pins_in_port = pb_graph_pin->port->num_pins;
+    int num_port = pb_graph_pin->port->index;
+    int num_pin = num_pins_in_port * num_port + num_pin_in_port;
 
-    
+
+    float interval = pb_bbox.width() / (num_pins + 1);
+
+    return ezgl::point2d(interval * (num_pin + 1), 0.01 *this->tile_width) + pb_bbox.top_left();
 }
 
 ezgl::rectangle t_draw_coords::get_absolute_clb_bbox(const ClusterBlockId clb_index, const t_logical_block_type_ptr block_type) {

vpr/src/util/vpr_utils.cpp

@@ -1755,7 +1755,7 @@ RRNodeId get_atom_pin_rr_node_id(AtomPinId atom_pin_id) {
     return rr_node_id;
 }
 
-AtomPinId get_rr_node_atom_pin_id(RRNodeId rr_node_id) {
+std::pair<ClusterBlockId, t_pb_graph_pin*> get_rr_node_cluster_blk_id_pb_graph_pin(RRNodeId rr_node_id) {
     auto& device_ctx = g_vpr_ctx.device();
     auto& place_ctx = g_vpr_ctx.placement();
 
@@ -1778,6 +1778,14 @@ AtomPinId get_rr_node_atom_pin_id(RRNodeId rr_node_id) {
 
     VTR_ASSERT(pb_graph_pin);
 
+    return {blk_id, pb_graph_pin};
+}
+
+AtomPinId get_rr_node_atom_pin_id(RRNodeId rr_node_id) {
+    ClusterBlockId blk_id;
+    t_pb_graph_pin* pb_graph_pin;
+    std::tie(blk_id, pb_graph_pin) = get_rr_node_cluster_blk_id_pb_graph_pin(rr_node_id);
+
     AtomPinId atom_pin_id = find_atom_pin(blk_id, pb_graph_pin);
 
     return atom_pin_id;

vpr/src/util/vpr_utils.h

@@ -257,11 +257,19 @@ RRNodeId get_pin_rr_node_id(const RRSpatialLookup& rr_spatial_lookup,
 RRNodeId get_atom_pin_rr_node_id(AtomPinId atom_pin_id);
 
 /**
- * @brief Returns the atom pin ID for the given RR node ID.
+ * @brief Returns the cluster block ID and pb_graph_pin for the given RR node ID.
  * **Warning**: This function should be called only if flat-router is enabled,
  * since, otherwise, the routing resources inside clusters are not added to the RR graph.
  * @param rr_node_id The RR node ID.
  */
+std::pair<ClusterBlockId, t_pb_graph_pin*> get_rr_node_cluster_blk_id_pb_graph_pin(RRNodeId rr_node_id);
+
+/**
+ * @brief Returns the atom pin ID for the given RR node ID.
+ * **Warning**: This function should be called only if flat-router is enabled,
+ * since, otherwise, the routing resources inside clusters are not added to the RR graph. Also, not all RRNodes have an AtomPinId associated with them.
+ * @param rr_node_id The RR node ID.
+ */
 AtomPinId get_rr_node_atom_pin_id(RRNodeId rr_node_id);
 
 RRNodeId get_class_rr_node_id(const RRSpatialLookup& rr_spatial_lookup,