src/soc/decoder/power_fields.py

   1 from collections import OrderedDict, namedtuple
   2 from soc.decoder.power_enums import find_wiki_file
   3
   4
   5 class BitRange(OrderedDict):
   6     """BitRange: remaps from straight indices (0,1,2..) to bit numbers
   7     """
   8
   9     def __getitem__(self, subscript):
  10         if isinstance(subscript, slice):
  11             return list(self.values())[subscript]
  12         else:
  13             return OrderedDict.__getitem__(self, subscript)
  14
  15
  16 def decode_instructions(form):
  17     res = {}
  18     accum = []
  19     for l in form:
  20         if l.strip().startswith("Formats"):
  21             l = l.strip().split(":")[-1]
  22             l = l.replace(" ", "")
  23             l = l.split(",")
  24             for fmt in l:
  25                 if fmt not in res:
  26                     res[fmt] = [accum[0]]
  27                 else:
  28                     res[fmt].append(accum[0])
  29             accum = []
  30         else:
  31             accum.append(l.strip())
  32     return res
  33
  34
  35 def decode_form_header(hdr):
  36     res = {}
  37     count = 0
  38     hdr = hdr.strip()
  39     for f in hdr.split("|"):
  40         if not f:
  41             continue
  42         if f[0].isdigit():
  43             idx = int(f.strip().split(' ')[0])
  44             res[count] = idx
  45         count += len(f) + 1
  46     return res
  47
  48
  49 def find_unique(d, key):
  50     if key not in d:
  51         return key
  52     idx = 1
  53     while "%s_%d" % (key, idx) in d:
  54         idx += 1
  55     return "%s_%d" % (key, idx)
  56
  57
  58 def decode_line(header, line):
  59     line = line.strip()
  60     res = {}
  61     count = 0
  62     prev_fieldname = None
  63     for f in line.split("|"):
  64         if not f:
  65             continue
  66         end = count + len(f) + 1
  67         fieldname = f.strip()
  68         if not fieldname or fieldname.startswith('/'):
  69             if prev_fieldname is not None:
  70                 res[prev_fieldname] = (res[prev_fieldname], header[count])
  71                 prev_fieldname = None
  72             count = end
  73             continue
  74         bitstart = header[count]
  75         if prev_fieldname is not None:
  76             res[prev_fieldname] = (res[prev_fieldname], bitstart)
  77         res[fieldname] = bitstart
  78         count = end
  79         prev_fieldname = fieldname
  80     res[prev_fieldname] = (bitstart, 32)
  81     return res
  82
  83
  84 def decode_form(form):
  85     header = decode_form_header(form[0])
  86     res = []
  87     for line in form[1:]:
  88         dec = decode_line(header, line)
  89         if dec:
  90             res.append(dec)
  91     fields = {}
  92     falternate = {}
  93     for l in res:
  94         for k, (start, end) in l.items():
  95             if k in fields:
  96                 if (start, end) == fields[k]:
  97                     continue  # already in and matching for this Form
  98                 if k in falternate:
  99                     alternate = "%s_%d" % (k, falternate[k])
 100                     if (start, end) == fields[alternate]:
 101                         continue
 102                 falternate[k] = fidx = falternate.get(k, 0) + 1
 103                 fields["%s_%d" % (k, fidx)] = (start, end)
 104             else:
 105                 fields[k] = (start, end)
 106     return fields
 107
 108
 109 class DecodeFields:
 110
 111     def __init__(self, bitkls=BitRange, bitargs=(), fname=None,
 112                  name_on_wiki=None):
 113         self.bitkls = bitkls
 114         self.bitargs = bitargs
 115         if fname is None:
 116             assert name_on_wiki is None
 117             fname = "fields.txt"
 118             name_on_wiki = "fields.text"
 119         self.fname = find_wiki_file(name_on_wiki)
 120
 121     @property
 122     def form_names(self):
 123         return self.instrs.keys()
 124
 125     def create_specs(self):
 126         self.forms, self.instrs = self.decode_fields()
 127         forms = self.form_names
 128         #print ("specs", self.forms, forms)
 129         for form in forms:
 130             fields = self.instrs[form]
 131             fk = fields.keys()
 132             Fields = namedtuple("Fields", fk)
 133             instr = Fields(**fields)
 134             setattr(self, "Form%s" % form, instr)
 135         # now add in some commonly-used fields (should be done automatically)
 136         # note that these should only be ones which are the same on all Forms
 137         # note: these are from microwatt insn_helpers.vhdl
 138         self.common_fields = {
 139             "RS": self.FormX.RS,
 140             "RT": self.FormX.RT,
 141             "RA": self.FormX.RA,
 142             "RB": self.FormX.RB,
 143             "SI": self.FormD.SI,
 144             "UI": self.FormD.UI,
 145             "L": self.FormD.L,
 146             "SH32": self.FormM.SH,
 147             "sh": self.FormMD.sh,
 148             "MB32": self.FormM.MB,
 149             "ME32": self.FormM.ME,
 150             "LI": self.FormI.LI,
 151             "LK": self.FormI.LK,
 152             "AA": self.FormB.AA,
 153             "Rc": self.FormX.Rc,
 154             "OE": self.FormXO.OE,
 155             "BD": self.FormB.BD,
 156             "BF": self.FormX.BF,
 157             "CR": self.FormXL.XO,
 158             "BB": self.FormXL.BB,
 159             "BA": self.FormXL.BA,
 160             "BT": self.FormXL.BT,
 161             "FXM": self.FormXFX.FXM,
 162             "BO": self.FormXL.BO,
 163             "BI": self.FormXL.BI,
 164             "BH": self.FormXL.BH,
 165             "D": self.FormD.D,
 166             "DS": self.FormDS.DS,
 167             "TO": self.FormX.TO,
 168             "BC": self.FormA.BC,
 169             "SH": self.FormX.SH,
 170             "ME": self.FormM.ME,
 171             "MB": self.FormM.MB,
 172             "SPR": self.FormXFX.SPR}
 173         for k, v in self.common_fields.items():
 174             setattr(self, k, v)
 175
 176     def decode_fields(self):
 177         with open(self.fname) as f:
 178             txt = f.readlines()
 179         #print ("decode", txt)
 180         forms = {}
 181         reading_data = False
 182         for l in txt:
 183             l = l.strip()
 184             if len(l) == 0:
 185                 continue
 186             if reading_data:
 187                 if l[0] == '#':
 188                     reading_data = False
 189                 else:
 190                     forms[heading].append(l)
 191             if not reading_data:
 192                 assert l[0] == '#'
 193                 heading = l[1:].strip()
 194                 # if heading.startswith('1.6.28'): # skip instr fields for now
 195                 #     break
 196                 heading = heading.split(' ')[-1]
 197                 reading_data = True
 198                 forms[heading] = []
 199
 200         res = {}
 201         inst = {}
 202
 203         for hdr, form in forms.items():
 204             if heading == 'Fields':
 205                 i = decode_instructions(form)
 206                 for form, field in i.items():
 207                     inst[form] = self.decode_instruction_fields(field)
 208             # else:
 209             #     res[hdr] = decode_form(form)
 210         return res, inst
 211
 212     def decode_instruction_fields(self, fields):
 213         res = {}
 214         for field in fields:
 215             f, spec = field.strip().split(" ")
 216             ss = spec[1:-1].split(",")
 217             fs = f.split(",")
 218             if len(fs) > 1:
 219                 individualfields = []
 220                 for f0, s0 in zip(fs, ss):
 221                     txt = "%s (%s)" % (f0, s0)
 222                     individualfields.append(txt)
 223                 if len(fs) > 1:
 224                     res.update(self.decode_instruction_fields(
 225                         individualfields))
 226             d = self.bitkls(*self.bitargs)
 227             idx = 0
 228             for s in ss:
 229                 s = s.split(':')
 230                 if len(s) == 1:
 231                     d[idx] = int(s[0])
 232                     idx += 1
 233                 else:
 234                     start = int(s[0])
 235                     end = int(s[1])
 236                     while start <= end:
 237                         d[idx] = start
 238                         idx += 1
 239                         start += 1
 240             f = f.replace(",", "_")
 241             unique = find_unique(res, f)
 242             res[unique] = d
 243
 244         return res
 245
 246
 247 if __name__ == '__main__':
 248     dec = DecodeFields()
 249     dec.create_specs()
 250     forms, instrs = dec.forms, dec.instrs
 251     for form, fields in instrs.items():
 252         print("Form", form)
 253         for field, bits in fields.items():
 254             print("\tfield", field, bits)