src/soc/decoder/power_fields.py

   1 from collections import OrderedDict, namedtuple
   2 from soc.decoder.power_enums import download_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)[subscript]
  12         else:
  13             return 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="fields.txt"):
 112         self.bitkls = bitkls
 113         self.bitargs = bitargs
 114         self.fname = download_wiki_file(fname)
 115
 116     def create_specs(self):
 117         self.forms, self.instrs = self.decode_fields()
 118         self.form_names = forms = self.instrs.keys()
 119         #print ("specs", self.forms, forms)
 120         for form in forms:
 121             fields = self.instrs[form]
 122             fk = fields.keys()
 123             Fields = namedtuple("Fields", fk)
 124             instr = Fields(**fields)
 125             setattr(self, "Form%s" % form, instr)
 126         # now add in some commonly-used fields (should be done automatically)
 127         # note that these should only be ones which are the same on all Forms
 128         # note: these are from microwatt insn_helpers.vhdl
 129         self.common_fields = {
 130             "RS": self.FormX.RS,
 131             "RT": self.FormX.RT,
 132             "RA": self.FormX.RA,
 133             "RB": self.FormX.RB,
 134             "SI": self.FormD.SI,
 135             "UI": self.FormD.UI,
 136             "L": self.FormD.L,
 137             "SH32": self.FormM.SH,
 138             "sh": self.FormMD.sh,
 139             "MB32": self.FormM.MB,
 140             "ME32": self.FormM.ME,
 141             "LI": self.FormI.LI,
 142             "LK": self.FormI.LK,
 143             "AA": self.FormB.AA,
 144             "Rc": self.FormX.Rc,
 145             "OE": self.FormXO.Rc,
 146             "BD": self.FormB.BD,
 147             "BF": self.FormX.BF,
 148             "CR": self.FormXL.XO,
 149             "BB": self.FormXL.BB,
 150             "BA": self.FormXL.BA,
 151             "BT": self.FormXL.BT,
 152             "FXM": self.FormXFX.FXM,
 153             "BO": self.FormXL.BO,
 154             "BI": self.FormXL.BI,
 155             "BH": self.FormXL.BH,
 156             "D": self.FormD.D,
 157             "DS": self.FormDS.DS,
 158             "TO": self.FormX.TO,
 159             "BC": self.FormA.BC,
 160             "SH": self.FormX.SH,
 161             "ME": self.FormM.ME,
 162             "MB": self.FormM.MB,
 163             "SPR": self.FormXFX.SPR}
 164         for k, v in self.common_fields.items():
 165             setattr(self, k, v)
 166
 167     def decode_fields(self):
 168         with open(self.fname) as f:
 169             txt = f.readlines()
 170         #print ("decode", txt)
 171         forms = {}
 172         reading_data = False
 173         for l in txt:
 174             l = l.strip()
 175             if len(l) == 0:
 176                 continue
 177             if reading_data:
 178                 if l[0] == '#':
 179                     reading_data = False
 180                 else:
 181                     forms[heading].append(l)
 182             if not reading_data:
 183                 assert l[0] == '#'
 184                 heading = l[1:].strip()
 185                 # if heading.startswith('1.6.28'): # skip instr fields for now
 186                 # break
 187                 heading = heading.split(' ')[-1]
 188                 reading_data = True
 189                 forms[heading] = []
 190
 191         res = {}
 192         inst = {}
 193
 194         for hdr, form in forms.items():
 195             if heading == 'Fields':
 196                 i = decode_instructions(form)
 197                 for form, field in i.items():
 198                     inst[form] = self.decode_instruction_fields(field)
 199             # else:
 200             #    res[hdr] = decode_form(form)
 201         return res, inst
 202
 203     def decode_instruction_fields(self, fields):
 204         res = {}
 205         for field in fields:
 206             f, spec = field.strip().split(" ")
 207             d = self.bitkls(*self.bitargs)
 208             idx = 0
 209             for s in spec[1:-1].split(","):
 210                 s = s.split(':')
 211                 if len(s) == 1:
 212                     d[idx] = int(s[0])
 213                     idx += 1
 214                 else:
 215                     start = int(s[0])
 216                     end = int(s[1])
 217                     while start <= end:
 218                         d[idx] = start
 219                         idx += 1
 220                         start += 1
 221             f = f.replace(",", "_")
 222             unique = find_unique(res, f)
 223             res[unique] = d
 224
 225         return res
 226
 227
 228 if __name__ == '__main__':
 229     dec = DecodeFields()
 230     dec.create_specs()
 231     forms, instrs = dec.forms, dec.instrs