| from __future__ import print_function |
| |
| CopyRight = ''' |
| /* |
| * Copyright 2015 Advanced Micro Devices, Inc. |
| * |
| * Permission is hereby granted, free of charge, to any person obtaining a |
| * copy of this software and associated documentation files (the "Software"), |
| * to deal in the Software without restriction, including without limitation |
| * on the rights to use, copy, modify, merge, publish, distribute, sub |
| * license, and/or sell copies of the Software, and to permit persons to whom |
| * the Software is furnished to do so, subject to the following conditions: |
| * |
| * The above copyright notice and this permission notice (including the next |
| * paragraph) shall be included in all copies or substantial portions of the |
| * Software. |
| * |
| * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR |
| * IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY, |
| * FITNESS FOR A PARTICULAR PURPOSE AND NON-INFRINGEMENT. IN NO EVENT SHALL |
| * THE AUTHOR(S) AND/OR THEIR SUPPLIERS BE LIABLE FOR ANY CLAIM, |
| * DAMAGES OR OTHER LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR |
| * OTHERWISE, ARISING FROM, OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE |
| * USE OR OTHER DEALINGS IN THE SOFTWARE. |
| * |
| */ |
| ''' |
| |
| import sys |
| import re |
| |
| |
| class StringTable: |
| """ |
| A class for collecting multiple strings in a single larger string that is |
| used by indexing (to avoid relocations in the resulting binary) |
| """ |
| def __init__(self): |
| self.table = [] |
| self.length = 0 |
| |
| def add(self, string): |
| # We might get lucky with string being a suffix of a previously added string |
| for te in self.table: |
| if te[0].endswith(string): |
| idx = te[1] + len(te[0]) - len(string) |
| te[2].add(idx) |
| return idx |
| |
| idx = self.length |
| self.table.append((string, idx, set((idx,)))) |
| self.length += len(string) + 1 |
| |
| return idx |
| |
| def emit(self, filp, name, static=True): |
| """ |
| Write |
| [static] const char name[] = "..."; |
| to filp. |
| """ |
| fragments = [ |
| '"%s\\0" /* %s */' % ( |
| te[0].encode('unicode_escape').decode(), |
| ', '.join(str(idx) for idx in te[2]) |
| ) |
| for te in self.table |
| ] |
| filp.write('%sconst char %s[] =\n%s;\n' % ( |
| 'static ' if static else '', |
| name, |
| '\n'.join('\t' + fragment for fragment in fragments) |
| )) |
| |
| class IntTable: |
| """ |
| A class for collecting multiple arrays of integers in a single big array |
| that is used by indexing (to avoid relocations in the resulting binary) |
| """ |
| def __init__(self, typename): |
| self.typename = typename |
| self.table = [] |
| self.idxs = set() |
| |
| def add(self, array): |
| # We might get lucky and find the array somewhere in the existing data |
| try: |
| idx = 0 |
| while True: |
| idx = self.table.index(array[0], idx, len(self.table) - len(array) + 1) |
| |
| for i in range(1, len(array)): |
| if array[i] != self.table[idx + i]: |
| break |
| else: |
| self.idxs.add(idx) |
| return idx |
| |
| idx += 1 |
| except ValueError: |
| pass |
| |
| idx = len(self.table) |
| self.table += array |
| self.idxs.add(idx) |
| return idx |
| |
| def emit(self, filp, name, static=True): |
| """ |
| Write |
| [static] const typename name[] = { ... }; |
| to filp. |
| """ |
| idxs = sorted(self.idxs) + [len(self.table)] |
| |
| fragments = [ |
| ('\t/* %s */ %s' % ( |
| idxs[i], |
| ' '.join((str(elt) + ',') for elt in self.table[idxs[i]:idxs[i+1]]) |
| )) |
| for i in range(len(idxs) - 1) |
| ] |
| |
| filp.write('%sconst %s %s[] = {\n%s\n};\n' % ( |
| 'static ' if static else '', |
| self.typename, name, |
| '\n'.join(fragments) |
| )) |
| |
| class Field: |
| def __init__(self, reg, s_name): |
| 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()) |
| |
| 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 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('(')] |
| |
| for it in reg.fields: |
| if it.s_name == name: |
| field = it |
| break |
| else: |
| field = Field(reg, name) |
| reg.fields.append(field) |
| |
| elif line.startswith('V_'): |
| split = line.split() |
| name = split[0] |
| value = int(split[1], 0) |
| |
| for (n,v) in field.values: |
| if n == name: |
| if v != value: |
| sys.exit('Value mismatch: name = ' + name) |
| |
| field.values.append((name, value)) |
| |
| elif line.startswith('PKT3_') and line.find('0x') != -1 and line.find('(') == -1: |
| packets.append(line.split()[0]) |
| |
| # 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 regs: |
| if len(reg.fields) and reg.name.find('0') != -1: |
| reg_dict[reg.name] = reg |
| |
| # 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 |
| |
| |
| def write_tables(regs, packets): |
| |
| strings = StringTable() |
| strings_offsets = IntTable("int") |
| |
| print('/* This file is autogenerated by egd_tables.py from evergreend.h. Do not edit directly. */') |
| print() |
| print(CopyRight.strip()) |
| print(''' |
| #ifndef EG_TABLES_H |
| #define EG_TABLES_H |
| |
| struct eg_field { |
| unsigned name_offset; |
| unsigned mask; |
| unsigned num_values; |
| unsigned values_offset; /* offset into eg_strings_offsets */ |
| }; |
| |
| struct eg_reg { |
| unsigned name_offset; |
| unsigned offset; |
| unsigned num_fields; |
| unsigned fields_offset; |
| }; |
| |
| struct eg_packet3 { |
| unsigned name_offset; |
| unsigned op; |
| }; |
| ''') |
| |
| print('static const struct eg_packet3 packet3_table[] = {') |
| for pkt in packets: |
| print('\t{%s, %s},' % (strings.add(pkt[5:]), pkt)) |
| print('};') |
| print() |
| |
| print('static const struct eg_field egd_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 |
| |
| 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 |
| |
| print('};') |
| print() |
| |
| print('static const struct eg_reg egd_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, "egd_strings") |
| |
| print() |
| |
| strings_offsets.emit(sys.stdout, "egd_strings_offsets") |
| |
| print() |
| print('#endif') |
| |
| |
| def main(): |
| regs = [] |
| packets = [] |
| for arg in sys.argv[1:]: |
| parse(arg, regs, packets) |
| write_tables(regs, packets) |
| |
| |
| if __name__ == '__main__': |
| main() |
| |
| # kate: space-indent on; indent-width 4; replace-tabs on; |