drt: Add drt layer length utility

Add a DRT analogue of `grt::route_layer_lenghts` for RC correlation
experiments.

Signed-off-by: Martin Povišer <[email protected]>

Author: povik

src/drt/include/triton_route/TritonRoute.h

@@ -20,6 +20,7 @@ class dbDatabase;
 class dbInst;
 class dbBTerm;
 class dbNet;
+class dbWire;
 }  // namespace odb
 
 namespace utl {
@@ -163,6 +164,7 @@ class TritonRoute
                  const std::list<std::unique_ptr<frMarker>>& markers,
                  const std::string& marker_name,
                  odb::Rect drcBox = odb::Rect(0, 0, 0, 0)) const;
+  std::vector<int> routeLayerLengths(odb::dbWire* wire) const;
   void checkDRC(const char* filename,
                 int x1,
                 int y1,

src/drt/src/TritonRoute.cpp

@@ -1385,4 +1385,24 @@ void TritonRoute::reportDRC(const std::string& file_name,
   tool_category->writeTR(file_name);
 }
 
+std::vector<int> TritonRoute::routeLayerLengths(odb::dbWire* wire) const
+{
+  std::vector<int> lengths;
+  lengths.resize(db_->getTech()->getLayerCount());
+  odb::dbWireShapeItr shapes;
+  odb::dbShape s;
+
+  for (shapes.begin(wire); shapes.next(s);) {
+    if (!s.isVia()) {
+      lengths[s.getTechLayer()->getNumber()] += s.getLength();
+    } else {
+      if (s.getTechVia()) {
+        lengths[s.getTechVia()->getBottomLayer()->getNumber() + 1] += 1;
+      }
+    }
+  }
+
+  return lengths;
+}
+
 }  // namespace drt

src/drt/src/TritonRoute.i

@@ -13,6 +13,10 @@
 
 %include "../../Exception.i"
 
+%import <stl.i>
+%import <std_vector.i>
+%template(vector_int) std::vector<int>;
+
 %inline %{
 
 int detailed_route_num_drvs()
@@ -218,4 +222,12 @@ void check_drc_cmd(const char* drc_file, int x1, int y1, int x2, int y2, const c
   const int num_threads = ord::OpenRoad::openRoad()->getThreadCount();
   router->checkDRC(drc_file, x1, y1, x2, y2, marker_name, num_threads);
 }
+
+std::vector<int>
+route_layer_lengths(odb::dbWire* db_wire)
+{
+  auto* router = ord::OpenRoad::openRoad()->getTritonRoute();
+  return router->routeLayerLengths(db_wire);
+}
+
 %} // inline