scripts: edtlib: gen_defines: Add model name macros based on compat

Follow up to 5b5aa6e adding model name
and existence macros for all compatibles of a node that match an entry
in vendor prefixes.

Signed-off-by: Maureen Helm <maureen.helm@intel.com>

Author: MaureenHelm

doc/build/dts/macros.bnf

@@ -57,6 +57,8 @@ node-macro =/ %s"DT_N" path-id %s"_PARTITION_ID" DIGIT
 node-macro =/ %s"DT_N" path-id %s"_COMPAT_MATCHES_" dt-name
 node-macro =/ %s"DT_N" path-id %s"_COMPAT_VENDOR_IDX_" DIGIT "_EXISTS"
 node-macro =/ %s"DT_N" path-id %s"_COMPAT_VENDOR_IDX_" DIGIT
+node-macro =/ %s"DT_N" path-id %s"_COMPAT_MODEL_IDX_" DIGIT "_EXISTS"
+node-macro =/ %s"DT_N" path-id %s"_COMPAT_MODEL_IDX_" DIGIT
 ; Every non-root node gets one of these macros, which expands to the node
 ; identifier for that node's parent in the devicetree.
 node-macro =/ %s"DT_N" path-id %s"_PARENT"

scripts/dts/gen_defines.py

@@ -519,6 +519,11 @@ def write_compatibles(node):
             out_dt_define(f"{node.z_path_id}_COMPAT_VENDOR_IDX_{i}",
                           quote_str(node.edt.compat2vendor[compat]))
 
+        if node.edt.compat2model[compat]:
+            out_dt_define(f"{node.z_path_id}_COMPAT_MODEL_IDX_{i}_EXISTS", 1)
+            out_dt_define(f"{node.z_path_id}_COMPAT_MODEL_IDX_{i}",
+                          quote_str(node.edt.compat2model[compat]))
+
 def write_children(node):
     # Writes helper macros for dealing with node's children.
 

scripts/dts/python-devicetree/src/devicetree/edtlib.py

@@ -110,6 +110,10 @@ class EDT:
       A collections.defaultdict that maps each 'compatible' string that appears
       on some Node to a vendor name parsed from vendor_prefixes.
 
+    compat2model:
+      A collections.defaultdict that maps each 'compatible' string that appears
+      on some Node to a model name parsed from that compatible.
+
     label2node:
       A collections.OrderedDict that maps a node label to the node with
       that label.
@@ -455,6 +459,7 @@ def _init_luts(self):
         self.compat2nodes = defaultdict(list)
         self.compat2okay = defaultdict(list)
         self.compat2vendor = defaultdict(str)
+        self.compat2model = defaultdict(str)
 
         for node in self.nodes:
             for label in node.labels:
@@ -477,9 +482,10 @@ def _init_luts(self):
                          f"'{compat_re}'")
 
                 if ',' in compat and self._vendor_prefixes:
-                    vendor = compat.split(',', 1)[0]
+                    vendor, model = compat.split(',', 1)
                     if vendor in self._vendor_prefixes:
                         self.compat2vendor[compat] = self._vendor_prefixes[vendor]
+                        self.compat2model[compat] = model
 
                     # As an exception, the root node can have whatever
                     # compatibles it wants. Other nodes get checked.