Merge pull request #8335 from The-OpenROAD-Project-staging/odb-gen-includes

odb: generate some includes based on the schema

Author: maliberty

src/odb/src/codeGenerator/gen.py

@@ -268,6 +268,22 @@ def generate_relations(schema):
             make_child_next_field(child, relation)
 
 
+def add_include(klass, key, include, cpp=False, sys=False):
+    for field in klass["fields"]:
+        if key in field["type"]:
+            if sys:
+                klass["h_sys_includes"].insert(0, include)
+            else:
+                klass["h_includes"].insert(0, include)
+
+            if cpp:
+                if sys:
+                    klass["cpp_sys_includes"].insert(0, include)
+                else:
+                    klass["cpp_includes"].insert(0, include)
+            break
+
+
 def preprocess_klass(klass):
     klass["declared_classes"].insert(0, "dbIStream")
     klass["declared_classes"].insert(1, "dbOStream")
@@ -280,6 +296,11 @@ def preprocess_klass(klass):
     klass["cpp_includes"].extend(["dbTable.h", "dbTable.hpp", "odb/db.h", f"{name}.h"])
     if klass["hasBitFields"]:
         klass["cpp_sys_includes"].extend(["cstdint", "cstring"])
+    add_include(klass, "dbObject", "dbCore.h", cpp=True)
+    add_include(klass, "dbId<", "odb/dbId.h")
+    add_include(klass, "dbVector", "dbVector.h")
+    add_include(klass, "dbObjectType", "odb/dbObject.h", cpp=True)
+    add_include(klass, "tuple", "tuple", cpp=True, sys=True)
     for field in klass["fields"]:
         if field.get("table", False):
             page_size_part = f", {field['page_size']}" if "page_size" in field else ""

src/odb/src/db/dbAccessPoint.cpp

@@ -5,11 +5,14 @@
 #include "dbAccessPoint.h"
 
 #include <array>
+#include <tuple>
 
+#include "dbCore.h"
 #include "dbDatabase.h"
 #include "dbTable.h"
 #include "dbTable.hpp"
 #include "odb/db.h"
+#include "odb/dbObject.h"
 #include "odb/dbTypes.h"
 // User Code Begin Includes
 #include <algorithm>

src/odb/src/db/dbAccessPoint.h

@@ -5,9 +5,12 @@
 #pragma once
 
 #include <array>
+#include <tuple>
 
 #include "dbCore.h"
 #include "dbVector.h"
+#include "odb/dbId.h"
+#include "odb/dbObject.h"
 #include "odb/odb.h"
 // User Code Begin Includes
 #include <utility>

src/odb/src/db/dbBusPort.h

@@ -5,6 +5,7 @@
 #pragma once
 
 #include "dbCore.h"
+#include "odb/dbId.h"
 #include "odb/odb.h"
 // User Code Begin Includes
 #include "dbModuleBusPortModBTermItr.h"

src/odb/src/db/dbChip.h

@@ -8,6 +8,7 @@
 #include <unordered_map>
 
 #include "dbCore.h"
+#include "odb/dbId.h"
 #include "odb/geom.h"
 #include "odb/odb.h"
 

src/odb/src/db/dbChipInst.h

@@ -8,6 +8,7 @@
 #include <unordered_map>
 
 #include "dbCore.h"
+#include "odb/dbId.h"
 #include "odb/dbTypes.h"
 #include "odb/geom.h"
 #include "odb/odb.h"

src/odb/src/db/dbDatabase.h

@@ -6,6 +6,7 @@
 
 #include "dbCore.h"
 #include "dbHashTable.h"
+#include "odb/dbId.h"
 #include "odb/odb.h"
 // User Code Begin Includes
 #include <iostream>

src/odb/src/db/dbGCellGrid.h

@@ -7,7 +7,9 @@
 #include <map>
 
 #include "dbCore.h"
+#include "dbVector.h"
 #include "odb/db.h"
+#include "odb/dbId.h"
 #include "odb/odb.h"
 // User Code Begin Includes
 #include "dbVector.h"

src/odb/src/db/dbGDSARef.h

@@ -11,6 +11,7 @@
 
 #include "dbCore.h"
 #include "dbGDSStructure.h"
+#include "odb/dbId.h"
 #include "odb/odb.h"
 
 namespace odb {

src/odb/src/db/dbGDSSRef.h

@@ -11,6 +11,7 @@
 
 #include "dbCore.h"
 #include "dbGDSStructure.h"
+#include "odb/dbId.h"
 #include "odb/odb.h"
 
 namespace odb {