src/soc/decoder/pseudo/pywriter.py

   1 # python code-writer for OpenPOWER ISA pseudo-code parsing
   2
   3 import os
   4 import sys
   5 from soc.decoder.pseudo.pagereader import ISA
   6 from soc.decoder.power_pseudo import convert_to_python
   7 from soc.decoder.orderedset import OrderedSet
   8 from soc.decoder.isa.caller import create_args
   9
  10 def get_isasrc_dir():
  11     fdir = os.path.abspath(os.path.dirname(__file__))
  12     fdir = os.path.split(fdir)[0]
  13     return os.path.join(fdir, "isa")
  14
  15
  16 header = """\
  17 # auto-generated by pywriter.py, do not edit or commit
  18
  19 from soc.decoder.isa.caller import inject, instruction_info
  20 from soc.decoder.helpers import (EXTS, EXTS64, EXTZ64, ROTL64, ROTL32, MASK,)
  21 from soc.decoder.selectable_int import SelectableInt
  22 from soc.decoder.selectable_int import selectconcat as concat
  23 from soc.decoder.orderedset import OrderedSet
  24
  25 class %s:
  26
  27 """
  28
  29 iinfo_template = """instruction_info(func=%s,
  30                 read_regs=%s,
  31                 uninit_regs=%s, write_regs=%s,
  32                 special_regs=%s, op_fields=%s,
  33                 form='%s',
  34                 asmregs=%s)"""
  35
  36 class PyISAWriter(ISA):
  37     def __init__(self):
  38         ISA.__init__(self)
  39         self.pages_written = []
  40
  41     def write_pysource(self, pagename):
  42         self.pages_written.append(pagename)
  43         instrs = isa.page[pagename]
  44         isadir = get_isasrc_dir()
  45         fname = os.path.join(isadir, "%s.py" % pagename)
  46         with open(fname, "w") as f:
  47             iinf = ''
  48             f.write(header % pagename) # write out header
  49             # go through all instructions
  50             for page in instrs:
  51                 d = self.instr[page]
  52                 print (fname, d.opcode)
  53                 pcode = '\n'.join(d.pcode) + '\n'
  54                 print (pcode)
  55                 pycode, rused = convert_to_python(pcode, d.form)
  56                 # create list of arguments to call
  57                 regs = list(rused['read_regs']) + list(rused['uninit_regs'])
  58                 regs += list(rused['special_regs'])
  59                 args = ', '.join(create_args(regs, 'self'))
  60                 # create list of arguments to return
  61                 retargs = ', '.join(create_args(rused['write_regs']))
  62                 # write out function.  pre-pend "op_" because some instrs are
  63                 # also python keywords (cmp).  also replace "." with "_"
  64                 op_fname ="op_%s" % page.replace(".", "_")
  65                 f.write("    @inject()\n")
  66                 f.write("    def %s(%s):\n" % (op_fname, args))
  67                 if 'NIA' in pycode:  # HACK - TODO fix
  68                     f.write("        global NIA\n")
  69                 pycode = pycode.split("\n")
  70                 pycode = '\n'.join(map(lambda x: "        %s" % x, pycode))
  71                 pycode = pycode.rstrip()
  72                 f.write(pycode + '\n')
  73                 if retargs:
  74                     f.write("        return (%s,)\n\n" % retargs)
  75                 else:
  76                     f.write("\n")
  77                 # accumulate the instruction info
  78                 ops = repr(rused['op_fields'])
  79                 iinfo = iinfo_template % (op_fname, rused['read_regs'],
  80                                           rused['uninit_regs'],
  81                                           rused['write_regs'],
  82                                           rused['special_regs'],
  83                                           ops, d.form, d.regs)
  84                 iinf += "    %s_instrs['%s'] = %s\n" % (pagename, page, iinfo)
  85             # write out initialisation of info, for ISACaller to use
  86             f.write("    %s_instrs = {}\n" % pagename)
  87             f.write(iinf)
  88
  89     def write_isa_class(self):
  90         isadir = get_isasrc_dir()
  91         fname = os.path.join(isadir, "all.py")
  92
  93         with open(fname, "w") as f:
  94             f.write('from soc.decoder.isa.caller import ISACaller\n')
  95             for page in self.pages_written:
  96                 f.write('from soc.decoder.isa.%s import %s\n' % (page, page))
  97             f.write('\n')
  98
  99             classes = ', '.join(['ISACaller'] + self.pages_written)
 100             f.write('class ISA(%s):\n' % classes)
 101             f.write('    def __init__(self, dec, regs):\n')
 102             f.write('        super().__init__(dec, regs)\n')
 103             f.write('        self.instrs = {\n')
 104             for page in self.pages_written:
 105                 f.write('            **self.%s_instrs,\n' % page)
 106             f.write('        }\n')
 107
 108
 109 if __name__ == '__main__':
 110     isa = PyISAWriter()
 111     if len(sys.argv) == 1: # quick way to do it
 112         print (dir(isa))
 113         sources = isa.page.keys()
 114     else:
 115         sources = sys.argv[1:]
 116     for source in sources:
 117         isa.write_pysource(source)
 118     isa.write_isa_class()