add a DIVS function as separate and discrete from floor_div
[soc.git] / src / soc / decoder / pseudo / pywriter.py
index dade1fceff2b47031b9adc3c63404e40cdd6d245..f06d996f388399bb8cd078107bc967f394788bbb 100644 (file)
@@ -1,8 +1,13 @@
 # python code-writer for OpenPOWER ISA pseudo-code parsing
 
 import os
+import sys
+import shutil
+import subprocess
 from soc.decoder.pseudo.pagereader import ISA
 from soc.decoder.power_pseudo import convert_to_python
+from soc.decoder.orderedset import OrderedSet
+from soc.decoder.isa.caller import create_args
 
 def get_isasrc_dir():
     fdir = os.path.abspath(os.path.dirname(__file__))
@@ -10,30 +15,126 @@ def get_isasrc_dir():
     return os.path.join(fdir, "isa")
 
 
+header = """\
+# auto-generated by pywriter.py, do not edit or commit
+
+from soc.decoder.isa.caller import inject, instruction_info
+from soc.decoder.helpers import (EXTS, EXTS64, EXTZ64, ROTL64, ROTL32, MASK,
+                                 ne, eq, gt, ge, lt, le, ltu, gtu, length,
+                                 trunc_divs, trunc_rems, MULS, DIVS, MODS)
+from soc.decoder.selectable_int import SelectableInt
+from soc.decoder.selectable_int import selectconcat as concat
+from soc.decoder.orderedset import OrderedSet
+
+class %s:
+
+"""
+
+iinfo_template = """instruction_info(func=%s,
+                read_regs=%s,
+                uninit_regs=%s, write_regs=%s,
+                special_regs=%s, op_fields=%s,
+                form='%s',
+                asmregs=%s)"""
+
 class PyISAWriter(ISA):
     def __init__(self):
         ISA.__init__(self)
+        self.pages_written = []
 
     def write_pysource(self, pagename):
+        self.pages_written.append(pagename)
         instrs = isa.page[pagename]
         isadir = get_isasrc_dir()
         fname = os.path.join(isadir, "%s.py" % pagename)
         with open(fname, "w") as f:
-            f.write("class %s:\n" % pagename)
+            iinf = ''
+            f.write(header % pagename) # write out header
+            # go through all instructions
             for page in instrs:
                 d = self.instr[page]
-                print (fname, d.opcode)
+                print ("page", pagename, page, fname, d.opcode)
                 pcode = '\n'.join(d.pcode) + '\n'
                 print (pcode)
-                pycode, regsused = convert_to_python(pcode)
-                f.write("    #%s\n" % repr(regsused))
-                f.write("    def %s(self):\n" % page.replace(".", "_"))
+                incl_carry = pagename == 'fixedshift'
+                pycode, rused = convert_to_python(pcode, d.form, incl_carry)
+                # create list of arguments to call
+                regs = list(rused['read_regs']) + list(rused['uninit_regs'])
+                regs += list(rused['special_regs'])
+                args = ', '.join(create_args(regs, 'self'))
+                # create list of arguments to return
+                retargs = ', '.join(create_args(rused['write_regs']))
+                # write out function.  pre-pend "op_" because some instrs are
+                # also python keywords (cmp).  also replace "." with "_"
+                op_fname ="op_%s" % page.replace(".", "_")
+                f.write("    @inject()\n")
+                f.write("    def %s(%s):\n" % (op_fname, args))
+                if 'NIA' in pycode:  # HACK - TODO fix
+                    f.write("        global NIA\n")
                 pycode = pycode.split("\n")
                 pycode = '\n'.join(map(lambda x: "        %s" % x, pycode))
-                f.write("%s\n\n" % pycode)
+                pycode = pycode.rstrip()
+                f.write(pycode + '\n')
+                if retargs:
+                    f.write("        return (%s,)\n\n" % retargs)
+                else:
+                    f.write("\n")
+                # accumulate the instruction info
+                ops = repr(rused['op_fields'])
+                iinfo = iinfo_template % (op_fname, rused['read_regs'],
+                                          rused['uninit_regs'],
+                                          rused['write_regs'],
+                                          rused['special_regs'],
+                                          ops, d.form, d.regs)
+                iinf += "    %s_instrs['%s'] = %s\n" % (pagename, page, iinfo)
+            # write out initialisation of info, for ISACaller to use
+            f.write("    %s_instrs = {}\n" % pagename)
+            f.write(iinf)
+
+    def patch_if_needed(self, source):
+        isadir = get_isasrc_dir()
+        fname = os.path.join(isadir, "%s.py" % source)
+        patchname = os.path.join(isadir, "%s.patch" % source)
+
+        try:
+            with open(patchname, 'r') as patch:
+                newfname = fname + '.orig'
+                shutil.copyfile(fname, newfname)
+                subprocess.check_call(['patch', fname],
+                                      stdin=patch)
+        except:
+            pass
+
+
+    def write_isa_class(self):
+        isadir = get_isasrc_dir()
+        fname = os.path.join(isadir, "all.py")
+
+        with open(fname, "w") as f:
+            f.write('# auto-generated by pywriter.py: do not edit or commit\n')
+            f.write('from soc.decoder.isa.caller import ISACaller\n')
+            for page in self.pages_written:
+                f.write('from soc.decoder.isa.%s import %s\n' % (page, page))
+            f.write('\n')
+
+            classes = ', '.join(['ISACaller'] + self.pages_written)
+            f.write('class ISA(%s):\n' % classes)
+            f.write('    def __init__(self, *args, **kwargs):\n')
+            f.write('        super().__init__(*args, **kwargs)\n')
+            f.write('        self.instrs = {\n')
+            for page in self.pages_written:
+                f.write('            **self.%s_instrs,\n' % page)
+            f.write('        }\n')
 
 
 if __name__ == '__main__':
     isa = PyISAWriter()
-    isa.write_pysource('comparefixed')
-    isa.write_pysource('fixedarith')
+    if len(sys.argv) == 1: # quick way to do it
+        print (dir(isa))
+        sources = isa.page.keys()
+    else:
+        sources = sys.argv[1:]
+    for source in sources:
+        isa.write_pysource(source)
+        isa.patch_if_needed(source)
+    isa.write_isa_class()