radv: trim buffer load result (fixes dota2)
[mesa.git] / src / amd / common / sid_tables.py
index 0a2b7ef1fe4cc3c15acdd69c3d8f15293d73af7f..4e53acefa44c560a75a6bf08f5aa13d545d28495 100644 (file)
@@ -25,8 +25,12 @@ CopyRight = '''
  */
 '''
 
-import sys
+import collections
+import functools
+import itertools
+import os.path
 import re
+import sys
 
 
 class StringTable:
@@ -131,91 +135,203 @@ class Field:
         self.s_name = s_name
         self.name = strip_prefix(s_name)
         self.values = []
-        self.varname_values = '%s__%s__values' % (reg.r_name.lower(), self.name.lower())
+
+    def format(self, string_table, idx_table):
+        if len(self.values):
+            values_offsets = []
+            for value in self.values:
+                while value[1] >= len(values_offsets):
+                    values_offsets.append(-1)
+                values_offsets[value[1]] = string_table.add(strip_prefix(value[0]))
+            return '{%s, %s(~0u), %s, %s}' % (
+                string_table.add(self.name), self.s_name,
+                len(values_offsets), idx_table.add(values_offsets))
+        else:
+            return '{%s, %s(~0u)}' % (string_table.add(self.name), self.s_name)
+
+    def __eq__(self, other):
+        return (self.s_name == other.s_name and
+                self.name == other.name and
+                len(self.values) == len(other.values) and
+                all(a[0] == b[0] and a[1] == b[1] for a, b, in zip(self.values, other.values)))
+
+    def __ne__(self, other):
+        return not (self == other)
+
+
+class FieldTable:
+    """
+    A class for collecting multiple arrays of register fields in a single big
+    array that is used by indexing (to avoid relocations in the resulting binary)
+    """
+    def __init__(self):
+        self.table = []
+        self.idxs = set()
+        self.name_to_idx = collections.defaultdict(lambda: [])
+
+    def add(self, array):
+        """
+        Add an array of Field objects, and return the index of where to find
+        the array in the table.
+        """
+        # Check if we can find the array in the table already
+        for base_idx in self.name_to_idx.get(array[0].name, []):
+            if base_idx + len(array) > len(self.table):
+                continue
+
+            for i, a in enumerate(array):
+                b = self.table[base_idx + i]
+                if a != b:
+                    break
+            else:
+                return base_idx
+
+        base_idx = len(self.table)
+        self.idxs.add(base_idx)
+
+        for field in array:
+            self.name_to_idx[field.name].append(len(self.table))
+            self.table.append(field)
+
+        return base_idx
+
+    def emit(self, filp, string_table, idx_table):
+        """
+        Write
+        static const struct si_field sid_fields_table[] = { ... };
+        to filp.
+        """
+        idxs = sorted(self.idxs) + [len(self.table)]
+
+        filp.write('static const struct si_field sid_fields_table[] = {\n')
+
+        for start, end in zip(idxs, idxs[1:]):
+            filp.write('\t/* %s */\n' % (start))
+            for field in self.table[start:end]:
+                filp.write('\t%s,\n' % (field.format(string_table, idx_table)))
+
+        filp.write('};\n')
+
 
 class Reg:
     def __init__(self, r_name):
         self.r_name = r_name
         self.name = strip_prefix(r_name)
         self.fields = []
-        self.own_fields = True
+
+    def __eq__(self, other):
+        if not isinstance(other, Reg):
+            return False
+        return (self.r_name == other.r_name and
+                self.name == other.name and
+                len(self.fields) == len(other.fields) and
+                all(a == b for a, b in zip(self.fields, other.fields)))
+
+    def __ne__(self, other):
+        return not (self == other)
 
 
 def strip_prefix(s):
     '''Strip prefix in the form ._.*_, e.g. R_001234_'''
     return s[s[2:].find('_')+3:]
 
-def parse(filename, regs, packets):
-    stream = open(filename)
-
-    for line in stream:
-        if not line.startswith('#define '):
-            continue
-
-        line = line[8:].strip()
-
-        if line.startswith('R_'):
-            name = line.split()[0]
 
-            for it in regs:
-                if it.r_name == name:
-                    reg = it
-                    break
-            else:
-                reg = Reg(name)
-                regs.append(reg)
-
-        elif line.startswith('S_'):
-            name = line[:line.find('(')]
+class Asic:
+    """
+    Store the registers of one ASIC class / group of classes.
+    """
+    def __init__(self, name):
+        self.name = name
+        self.registers = []
 
-            for it in reg.fields:
-                if it.s_name == name:
-                    field = it
-                    break
-            else:
-                field = Field(reg, name)
-                reg.fields.append(field)
+    def parse(self, filp, packets, older_asics):
+        """
+        Parse registers from the given header file. Packets are separately
+        stored in the packets array.
+        """
+        for line in filp:
+            if not line.startswith('#define '):
+                continue
 
-        elif line.startswith('V_'):
-            split = line.split()
-            name = split[0]
-            value = int(split[1], 0)
+            line = line[8:].strip()
 
-            for (n,v) in field.values:
-                if n == name:
-                    if v != value:
-                        sys.exit('Value mismatch: name = ' + name)
+            if line.startswith('R_'):
+                name = line.split()[0]
 
-            field.values.append((name, value))
+                for it in self.registers:
+                    if it.r_name == name:
+                        sys.exit('Duplicate register define: %s' % (name))
+                else:
+                    reg = Reg(name)
+                    self.registers.append(reg)
 
-        elif line.startswith('PKT3_') and line.find('0x') != -1 and line.find('(') == -1:
-            packets.append(line.split()[0])
+            elif line.startswith('S_'):
+                name = line[:line.find('(')]
 
-    # Copy fields to indexed registers which have their fields only defined
-    # at register index 0.
-    # For example, copy fields from CB_COLOR0_INFO to CB_COLORn_INFO, n > 0.
-    match_number = re.compile('[0-9]+')
-    reg_dict = dict()
+                for it in reg.fields:
+                    if it.s_name == name:
+                        sys.exit('Duplicate field define: %s' % (name))
+                else:
+                    field = Field(reg, name)
+                    reg.fields.append(field)
 
-    # Create a dict of registers with fields and '0' in their name
-    for reg in regs:
-        if len(reg.fields) and reg.name.find('0') != -1:
-            reg_dict[reg.name] = reg
+            elif line.startswith('V_'):
+                split = line.split()
+                name = split[0]
+                value = int(split[1], 0)
 
-    # Assign fields
-    for reg in regs:
-        if not len(reg.fields):
-            reg0 = reg_dict.get(match_number.sub('0', reg.name))
-            if reg0 != None:
-                reg.fields = reg0.fields
-                reg.fields_owner = reg0
-                reg.own_fields = False
+                for (n,v) in field.values:
+                    if n == name:
+                        sys.exit('Duplicate value define: name = ' + name)
 
+                field.values.append((name, value))
 
-def write_tables(regs, packets):
+            elif line.startswith('PKT3_') and line.find('0x') != -1 and line.find('(') == -1:
+                packets.append(line.split()[0])
 
+        # Copy values for corresponding fields from older ASICs if they were
+        # not redefined
+        for reg in self.registers:
+            old_reg = False
+            for field in reg.fields:
+                if len(field.values) > 0:
+                    continue
+                if old_reg is False:
+                    for old_reg in itertools.chain(
+                            *(asic.registers for asic in reversed(older_asics))):
+                        if old_reg.name == reg.name:
+                            break
+                    else:
+                        old_reg = None
+                if old_reg is not None:
+                    for old_field in old_reg.fields:
+                        if old_field.name == field.name:
+                            field.values = old_field.values
+                            break
+
+        # Copy fields to indexed registers which have their fields only defined
+        # at register index 0.
+        # For example, copy fields from CB_COLOR0_INFO to CB_COLORn_INFO, n > 0.
+        match_number = re.compile('[0-9]+')
+        reg_dict = dict()
+
+        # Create a dict of registers with fields and '0' in their name
+        for reg in self.registers:
+            if len(reg.fields) and reg.name.find('0') != -1:
+                reg_dict[reg.name] = reg
+
+        # Assign fields
+        for reg in self.registers:
+            if not len(reg.fields):
+                reg0 = reg_dict.get(match_number.sub('0', reg.name))
+                if reg0 != None:
+                    reg.fields = reg0.fields
+
+
+def write_tables(asics, packets):
     strings = StringTable()
     strings_offsets = IntTable("int")
+    fields = FieldTable()
 
     print '/* This file is autogenerated by sid_tables.py from sid.h. Do not edit directly. */'
     print
@@ -250,40 +366,28 @@ struct si_packet3 {
     print '};'
     print
 
-    print 'static const struct si_field sid_fields_table[] = {'
-
-    fields_idx = 0
-    for reg in regs:
-        if len(reg.fields) and reg.own_fields:
-            print '\t/* %s */' % (fields_idx)
-
-            reg.fields_idx = fields_idx
+    regs = {}
+    for asic in asics:
+        print 'static const struct si_reg %s_reg_table[] = {' % (asic.name)
+        for reg in asic.registers:
+            # Only output a register that was changed or added relative to
+            # the previous generation
+            previous = regs.get(reg.r_name, None)
+            if previous == reg:
+                continue
+
+            if len(reg.fields):
+                print '\t{%s, %s, %s, %s},' % (strings.add(reg.name), reg.r_name,
+                    len(reg.fields), fields.add(reg.fields))
+            else:
+                print '\t{%s, %s},' % (strings.add(reg.name), reg.r_name)
 
-            for field in reg.fields:
-                if len(field.values):
-                    values_offsets = []
-                    for value in field.values:
-                        while value[1] >= len(values_offsets):
-                            values_offsets.append(-1)
-                        values_offsets[value[1]] = strings.add(strip_prefix(value[0]))
-                    print '\t{%s, %s(~0u), %s, %s},' % (
-                        strings.add(field.name), field.s_name,
-                        len(values_offsets), strings_offsets.add(values_offsets))
-                else:
-                    print '\t{%s, %s(~0u)},' % (strings.add(field.name), field.s_name)
-                fields_idx += 1
+            regs[reg.r_name] = reg
+        print '};'
+        print
 
-    print '};'
-    print
+    fields.emit(sys.stdout, strings, strings_offsets)
 
-    print 'static const struct si_reg sid_reg_table[] = {'
-    for reg in regs:
-        if len(reg.fields):
-            print '\t{%s, %s, %s, %s},' % (strings.add(reg.name), reg.r_name,
-                len(reg.fields), reg.fields_idx if reg.own_fields else reg.fields_owner.fields_idx)
-        else:
-            print '\t{%s, %s},' % (strings.add(reg.name), reg.r_name)
-    print '};'
     print
 
     strings.emit(sys.stdout, "sid_strings")
@@ -297,11 +401,16 @@ struct si_packet3 {
 
 
 def main():
-    regs = []
+    asics = []
     packets = []
     for arg in sys.argv[1:]:
-        parse(arg, regs, packets)
-    write_tables(regs, packets)
+        basename = os.path.basename(arg)
+        m = re.match(r'(.*)\.h', basename)
+        asic = Asic(m.group(1))
+        with open(arg) as filp:
+            asic.parse(filp, packets, asics)
+        asics.append(asic)
+    write_tables(asics, packets)
 
 
 if __name__ == '__main__':