src/soc/decoder/pseudo/pywriter.py

   1 # python code-writer for OpenPOWER ISA pseudo-code parsing
   2
   3 import os
   4 from soc.decoder.pseudo.pagereader import ISA
   5 from soc.decoder.power_pseudo import convert_to_python
   6
   7 def get_isasrc_dir():
   8     fdir = os.path.abspath(os.path.dirname(__file__))
   9     fdir = os.path.split(fdir)[0]
  10     return os.path.join(fdir, "isa")
  11
  12 def create_args(reglist, extra=None):
  13     args = set()
  14     for reg in reglist:
  15         args.add(reg)
  16     args = list(args)
  17     if extra:
  18         args = [extra] + args
  19     return ', '.join(args)
  20
  21 header = """\
  22 # auto-generated by pywriter.py, do not edit or commit
  23
  24 from soc.decoder.isa import ISACaller
  25 from soc.decoder.helpers import (EXTS64, EXTZ64, ROTL64, ROTL32, MASK,)
  26
  27 class %s(ISACaller):
  28
  29 """
  30
  31 class PyISAWriter(ISA):
  32     def __init__(self):
  33         ISA.__init__(self)
  34
  35     def write_pysource(self, pagename):
  36         instrs = isa.page[pagename]
  37         isadir = get_isasrc_dir()
  38         fname = os.path.join(isadir, "%s.py" % pagename)
  39         with open(fname, "w") as f:
  40             iinf = ''
  41             f.write(header % pagename) # write out header
  42             # go through all instructions
  43             for page in instrs:
  44                 d = self.instr[page]
  45                 print (fname, d.opcode)
  46                 pcode = '\n'.join(d.pcode) + '\n'
  47                 print (pcode)
  48                 pycode, rused = convert_to_python(pcode)
  49                 # create list of arguments to call
  50                 regs = rused['read_regs'] + rused['uninit_regs']
  51                 args = create_args(regs, 'self')
  52                 # create list of arguments to return
  53                 retargs = create_args(rused['write_regs'])
  54                 f.write("    def %s(%s):\n" % (page.replace(".", "_"), args))
  55                 pycode = pycode.split("\n")
  56                 pycode = '\n'.join(map(lambda x: "        %s" % x, pycode))
  57                 pycode = pycode.rstrip()
  58                 f.write(pycode + '\n')
  59                 if retargs:
  60                     f.write("        return (%s,)\n\n" % retargs)
  61                 else:
  62                     f.write("\n")
  63                 # accumulate the instruction info
  64                 iinfo = "(%s, %s, %s, %s)" % \
  65                             (page, rused['read_regs'],
  66                             rused['uninit_regs'], rused['write_regs'])
  67                 iinf += "    instrs['%s'] = %s\n" % (pagename, iinfo)
  68             # write out initialisation of info, for ISACaller to use
  69             f.write("    instrs = {}\n")
  70             f.write(iinf)
  71
  72 if __name__ == '__main__':
  73     isa = PyISAWriter()
  74     isa.write_pysource('comparefixed')
  75     isa.write_pysource('fixedarith')