fix naming
[pinmux.git] / src / bsv / peripheral_gen.py
index de01427d8bad4e09724b604265e6e744af092aeb..bd479eaf4a12b5e2883ae7e1fc3c70f4c9889261 100644 (file)
@@ -1,5 +1,10 @@
+import types
+from copy import deepcopy
+
+
 class PBase(object):
-    pass
+    def __init__(self, name):
+        self.name = name
 
     def axibase(self, name, ifacenum):
         name = name.upper()
@@ -16,8 +21,8 @@ class PBase(object):
         bname = self.axibase(name, ifacenum)
         bend = self.axiend(name, ifacenum)
         comment = "%d 32-bit regs" % self.num_axi_regs32()
-        return ("    `define%(bname)s 'h%(start)08X\n"
-                "    `define%(bend)s  'h%(end)08X // %(comment)s" % locals(),
+        return ("    `define %(bname)s 'h%(start)08X\n"
+                "    `define %(bend)s  'h%(end)08X // %(comment)s" % locals(),
                 offs)
 
     def axi_slave_name(self, name, ifacenum):
@@ -37,73 +42,281 @@ class PBase(object):
             return tuple2(True,fromInteger(valueOf({2})));
         else""".format(bname, bend, name)
 
+    def mk_pincon(self, name, count):
+        # TODO: really should be using bsv.interface_decl.Interfaces
+        # pin-naming rules.... logic here is hard-coded to duplicate
+        # it (see Interface.__init__ outen)
+        ret = []
+        for p in self.peripheral.pinspecs:
+            typ = p['type']
+            pname = p['name']
+            #n = "{0}{1}".format(self.name, self.mksuffix(name, count))
+            n = name  # "{0}{1}".format(self.name, self.mksuffix(name, count))
+            ret.append("    //%s %s" % (n, str(p)))
+            sname = self.peripheral.pname(pname).format(count)
+            ps = "pinmux.peripheral_side.%s" % sname
+            if typ == 'out' or typ == 'inout':
+                ret.append("    rule con_%s%d_%s_out;" % (name, count, pname))
+                fname = self.pinname_out(pname)
+                if fname:
+                    if p.get('outen'):
+                        ps_ = ps + '_out'
+                    else:
+                        ps_ = ps
+                    n_ = "{0}{1}".format(n, count)
+                    ret.append("      {0}({1}.{2});".format(ps_, n_, fname))
+                fname = None
+                if p.get('outen'):
+                    fname = self.pinname_outen(pname)
+                if fname:
+                    if isinstance(fname, str):
+                        fname = "{0}{1}.{2}".format(n, count, fname)
+                    fname = self.pinname_tweak(pname, 'outen', fname)
+                    ret.append("      {0}_outen({1});".format(ps, fname))
+                ret.append("    endrule")
+            if typ == 'in' or typ == 'inout':
+                fname = self.pinname_in(pname)
+                if fname:
+                    if p.get('outen'):
+                        ps_ = ps + '_in'
+                    else:
+                        ps_ = ps
+                    ret.append(
+                        "    rule con_%s%d_%s_in;" %
+                        (name, count, pname))
+                    n_ = "{0}{1}".format(n, count)
+                    ret.append("      {1}.{2}({0});".format(ps_, n_, fname))
+                    ret.append("    endrule")
+        return '\n'.join(ret)
+
+    def mk_cellconn(self, *args):
+        return ''
+
+    def mkslow_peripheral(self, size=0):
+        return ''
+
+    def mksuffix(self, name, i):
+        return i
+
+    def __mk_connection(self, con, aname):
+        txt = "        mkConnection (slow_fabric.v_to_slaves\n" + \
+            "                    [fromInteger(valueOf({1}))],\n" + \
+            "                    {0});"
+
+        print "PBase __mk_connection", self.name, aname
+        if not con:
+            return ''
+        return txt.format(con, aname)
+
+    def mk_connection(self, count, name=None):
+        if name is None:
+            name = self.name
+        print "PBase mk_conn", self.name, count
+        aname = self.axi_slave_name(name, count)
+        #dname = self.mksuffix(name, count)
+        #dname = "{0}{1}".format(name, dname)
+        con = self._mk_connection(name, count).format(count, aname)
+        return self.__mk_connection(con, aname)
+
+    def _mk_connection(self, name=None, count=0):
+        return ''
+
+    def pinname_out(self, pname):
+        return ''
+
+    def pinname_in(self, pname):
+        return ''
+
+    def pinname_outen(self, pname):
+        return ''
+
+    def pinname_tweak(self, pname, typ, txt):
+        return txt
+
 
 class uart(PBase):
-    def importfn(self):
+
+    def slowimport(self):
         return "          import Uart16550         :: *;"
 
-    def ifacedecl(self):
+    def slowifdecl(self):
         return "            interface RS232_PHY_Ifc uart{0}_coe;\n" + \
                "            method Bit#(1) uart{0}_intr;"
 
     def num_axi_regs32(self):
         return 8
 
+    def mkslow_peripheral(self, size=0):
+        return "        Uart16550_AXI4_Lite_Ifc uart{0} <- \n" + \
+               "                mkUart16550(clocked_by uart_clock,\n" + \
+               "                    reset_by uart_reset, sp_clock, sp_reset);"
+
+    def _mk_connection(self, name=None, count=0):
+        return "uart{0}.slave_axi_uart"
+
+    def pinname_out(self, pname):
+        return {'tx': 'coe_rs232.sout'}.get(pname, '')
+
+    def pinname_in(self, pname):
+        return {'rx': 'coe_rs232.sin'}.get(pname, '')
+
 
 class rs232(PBase):
-    def importfn(self):
+
+    def slowimport(self):
         return "        import Uart_bs::*;\n" + \
                "        import RS232_modified::*;"
 
-    def ifacedecl(self):
+    def slowifdecl(self):
         return "            interface RS232 uart{0}_coe;"
 
     def num_axi_regs32(self):
         return 2
 
+    def mkslow_peripheral(self, size=0):
+        return "        //Ifc_Uart_bs uart{0} <-" + \
+               "        //       mkUart_bs(clocked_by uart_clock,\n" + \
+               "        //          reset_by uart_reset,sp_clock, sp_reset);" +\
+               "        Ifc_Uart_bs uart{0} <-" + \
+               "                mkUart_bs(clocked_by sp_clock,\n" + \
+               "                    reset_by sp_reset, sp_clock, sp_reset);"
+
+    def _mk_connection(self, name=None, count=0):
+        return "uart{0}.slave_axi_uart"
+
+    def pinname_out(self, pname):
+        return {'tx': 'coe_rs232.sout'}.get(pname, '')
+
+    def pinname_in(self, pname):
+        return {'rx': 'coe_rs232.sin'}.get(pname, '')
+
 
 class twi(PBase):
-    def importfn(self):
+
+    def slowimport(self):
         return "        import I2C_top           :: *;"
 
-    def ifacedecl(self):
-        return "            interface I2C_out i2c{0}_out;\n" + \
-               "            method Bit#(1) i2c{0}_isint;"
+    def slowifdecl(self):
+        return "            interface I2C_out twi{0}_out;\n" + \
+               "            method Bit#(1) twi{0}_isint;"
 
     def num_axi_regs32(self):
         return 8
 
+    def mkslow_peripheral(self, size=0):
+        return "        I2C_IFC twi{0} <- mkI2CController();"
+
+    def _mk_connection(self, name=None, count=0):
+        return "twi{0}.slave_i2c_axi"
+
+    def pinname_out(self, pname):
+        return {'sda': 'out.sda_out',
+                'scl': 'out.scl_out'}.get(pname, '')
+
+    def pinname_in(self, pname):
+        return {'sda': 'out.sda_in',
+                'scl': 'out.scl_in'}.get(pname, '')
+
+    def pinname_outen(self, pname):
+        return {'sda': 'out.sda_outen',
+                'scl': 'out.scl_outen'}.get(pname, '')
+
+    def pinname_tweak(self, pname, typ, txt):
+        if typ == 'outen':
+            return "pack({0})".format(txt)
+        return txt
+
 
 class qspi(PBase):
-    def importfn(self):
+
+    def slowimport(self):
         return "        import qspi              :: *;"
 
-    def ifacedecl(self):
+    def slowifdecl(self):
         return "            interface QSPI_out qspi{0}_out;\n" + \
                "            method Bit#(1) qspi{0}_isint;"
 
     def num_axi_regs32(self):
         return 13
 
+    def mkslow_peripheral(self, size=0):
+        return "        Ifc_qspi qspi{0} <-  mkqspi();"
+
+    def _mk_connection(self, name=None, count=0):
+        return "qspi{0}.slave"
+
+    def pinname_out(self, pname):
+        return {'ck': 'out.clk_o',
+                'nss': 'out.ncs_o',
+                'io0': 'out.io_o[0]',
+                'io1': 'out.io_o[1]',
+                'io2': 'out.io_o[2]',
+                'io3': 'out.io_o[3]',
+                }.get(pname, '')
+
+    def pinname_outen(self, pname):
+        return {'ck': 1,
+                'nss': 1,
+                'io0': 'out.io_enable[0]',
+                'io1': 'out.io_enable[1]',
+                'io2': 'out.io_enable[2]',
+                'io3': 'out.io_enable[3]',
+                }.get(pname, '')
+
+    def mk_pincon(self, name, count):
+        ret = [PBase.mk_pincon(self, name, count)]
+        # special-case for gpio in, store in a temporary vector
+        plen = len(self.peripheral.pinspecs)
+        ret.append("    // XXX NSS and CLK are hard-coded master")
+        ret.append("    // TODO: must add qspi slave-mode")
+        ret.append("    // all ins done in one rule from 4-bitfield")
+        ret.append("    rule con_%s%d_io_in;" % (name, count))
+        ret.append("       {0}{1}.out.io_i({{".format(name, count))
+        for i, p in enumerate(self.peripheral.pinspecs):
+            typ = p['type']
+            pname = p['name']
+            if not pname.startswith('io'):
+                continue
+            idx = pname[1:]
+            n = name
+            sname = self.peripheral.pname(pname).format(count)
+            ps = "pinmux.peripheral_side.%s_in" % sname
+            comma = '' if i == 5 else ','
+            ret.append("            {0}{1}".format(ps, comma))
+        ret.append("        });")
+        ret.append("    endrule")
+        return '\n'.join(ret)
+
 
 class pwm(PBase):
-    def importfn(self):
+
+    def slowimport(self):
         return "        import pwm::*;"
 
-    def ifacedecl(self):
-        return "        interface PWMIO pwm_o;"
+    def slowifdecl(self):
+        return "        interface PWMIO pwm{0}_o;"
 
     def num_axi_regs32(self):
         return 4
 
+    def mkslow_peripheral(self, size=0):
+        return "        Ifc_PWM_bus pwm{0}_bus <- mkPWM_bus(sp_clock);"
+
+    def _mk_connection(self, name=None, count=0):
+        return "pwm{0}_bus.axi4_slave"
+
+    def pinname_out(self, pname):
+        return {'out': 'pwm_io.pwm_o'}.get(pname, '')
+
 
 class gpio(PBase):
-    def importfn(self):
+
+    def slowimport(self):
         return "     import pinmux::*;\n" + \
                "     import mux::*;\n" + \
                "     import gpio::*;\n"
 
-    def ifacedecl(self):
+    def slowifdecl(self):
         return "        interface GPIO_config#({1}) pad_config{0};"
 
     def num_axi_regs32(self):
@@ -114,10 +327,73 @@ class gpio(PBase):
             GPIO also has a muxer per bank
         """
         name = name.upper()
+        mname = 'mux' + name[4:]
+        mname = mname.upper()
+        print "AXIslavenum", name,  mname
         (ret, x) = PBase.axi_slave_idx(self, idx, name, ifacenum)
-        (ret2, x) = PBase.axi_slave_idx(self, idx, "mux", ifacenum)
+        (ret2, x) = PBase.axi_slave_idx(self, idx, mname, ifacenum)
         return ("%s\n%s" % (ret, ret2), 2)
 
+    def mkslow_peripheral(self, size=0):
+        print "gpioslow", self.peripheral,  dir(self.peripheral)
+        size = len(self.peripheral.pinspecs)
+        return "        MUX#(%d) mux{0} <- mkmux();\n" % size + \
+               "        GPIO#(%d) gpio{0} <- mkgpio();" % size
+
+    def mk_connection(self, count):
+        print "GPIO mk_conn", self.name, count
+        res = []
+        dname = self.mksuffix(self.name, count)
+        for i, n in enumerate(['gpio' + dname, 'mux' + dname]):
+            res.append(PBase.mk_connection(self, count, n))
+        return '\n'.join(res)
+
+    def _mk_connection(self, name=None, count=0):
+        n = self.mksuffix(name, count)
+        if name.startswith('gpio'):
+            return "gpio{0}.axi_slave".format(n)
+        if name.startswith('mux'):
+            return "mux{0}.axi_slave".format(n)
+
+    def mksuffix(self, name, i):
+        if name.startswith('mux'):
+            return name[3:]
+        return name[4:]
+
+    def mk_cellconn(self, cellnum, name, count):
+        ret = []
+        bank = self.mksuffix(name, count)
+        txt = "       pinmux.mux_lines.cell{0}_mux(mux{1}.mux_config.mux[{2}]);"
+        for p in self.peripheral.pinspecs:
+            ret.append(txt.format(cellnum, bank, p['name'][1:]))
+            cellnum += 1
+        return ("\n".join(ret), cellnum)
+
+    def pinname_out(self, pname):
+        return "func.gpio_out[{0}]".format(pname[1:])
+
+    def pinname_outen(self, pname):
+        return {'sda': 'out.sda_outen',
+                'scl': 'out.scl_outen'}.get(pname, '')
+
+    def mk_pincon(self, name, count):
+        ret = [PBase.mk_pincon(self, name, count)]
+        # special-case for gpio in, store in a temporary vector
+        plen = len(self.peripheral.pinspecs)
+        ret.append("    rule con_%s%d_in;" % (name, count))
+        ret.append("       Vector#({0},Bit#(1)) temp;".format(plen))
+        for p in self.peripheral.pinspecs:
+            typ = p['type']
+            pname = p['name']
+            idx = pname[1:]
+            n = name
+            sname = self.peripheral.pname(pname).format(count)
+            ps = "pinmux.peripheral_side.%s_in" % sname
+            ret.append("        temp[{0}]={1};".format(idx, ps))
+        ret.append("        {0}.func.gpio_in(temp);".format(name))
+        ret.append("    endrule")
+        return '\n'.join(ret)
+
 
 axi_slave_declarations = """\
 typedef  0  SlowMaster;
@@ -131,23 +407,45 @@ typedef  TAdd#(Plic_slave_num   ,`ifdef AXIEXP      1 `else 0 `endif )
 typedef TAdd#(AxiExp1_slave_num,1) Num_Slow_Slaves;
 """
 
+pinmux_cellrule = """\
+    rule connect_select_lines_pinmux;
+{0}
+    endrule
+"""
+
+
+class CallFn(object):
+    def __init__(self, peripheral, name):
+        self.peripheral = peripheral
+        self.name = name
+
+    def __call__(self, *args):
+        #print "__call__", self.name, self.peripheral.slow, args
+        if not self.peripheral.slow:
+            return ''
+        return getattr(self.peripheral.slow, self.name)(*args[1:])
+
 
 class PeripheralIface(object):
     def __init__(self, ifacename):
         self.slow = None
         slow = slowfactory.getcls(ifacename)
+        print "Iface", ifacename, slow
         if slow:
-            self.slow = slow()
+            self.slow = slow(ifacename)
+            self.slow.peripheral = self
+        for fname in ['slowimport', 'slowifdecl', 'mkslow_peripheral',
+                      'mk_connection', 'mk_cellconn', 'mk_pincon']:
+            fn = CallFn(self, fname)
+            setattr(self, fname, types.MethodType(fn, self))
 
-    def slowimport(self):
-        if not self.slow:
-            return ''
-        return self.slow.importfn().format()
+        #print "PeripheralIface"
+        #print dir(self)
 
-    def slowifdecl(self, count):
-        if not self.slow:
-            return ''
-        return self.slow.ifacedecl().format(count, self.ifacename)
+    def mksuffix(self, name, i):
+        if self.slow is None:
+            return i
+        return self.slow.mksuffix(name, i)
 
     def axi_reg_def(self, start, count):
         if not self.slow:
@@ -172,6 +470,7 @@ class PeripheralInterfaces(object):
     def slowimport(self, *args):
         ret = []
         for (name, count) in self.ifacecount:
+            #print "slowimport", name, self.data[name].slowimport
             ret.append(self.data[name].slowimport())
         return '\n'.join(list(filter(None, ret)))
 
@@ -179,7 +478,7 @@ class PeripheralInterfaces(object):
         ret = []
         for (name, count) in self.ifacecount:
             for i in range(count):
-                ret.append(self.data[name].slowifdecl(i))
+                ret.append(self.data[name].slowifdecl().format(i, name))
         return '\n'.join(list(filter(None, ret)))
 
     def axi_reg_def(self, *args):
@@ -188,7 +487,7 @@ class PeripheralInterfaces(object):
         for (name, count) in self.ifacecount:
             for i in range(count):
                 x = self.data[name].axi_reg_def(start, i)
-                print ("ifc", name, x)
+                #print ("ifc", name, x)
                 (rdef, offs) = x
                 ret.append(rdef)
                 start += offs
@@ -200,10 +499,10 @@ class PeripheralInterfaces(object):
         for (name, count) in self.ifacecount:
             for i in range(count):
                 (rdef, offs) = self.data[name].axi_slave_idx(start, i)
-                print ("ifc", name, rdef, offs)
+                #print ("ifc", name, rdef, offs)
                 ret.append(rdef)
                 start += offs
-        ret.append("typedef %d LastGen_slave_num" % (start - 1))
+        ret.append("typedef %d LastGen_slave_num;" % (start - 1))
         decls = '\n'.join(list(filter(None, ret)))
         return axi_slave_declarations.format(decls)
 
@@ -214,16 +513,72 @@ class PeripheralInterfaces(object):
                 ret.append(self.data[name].axi_addr_map(i))
         return '\n'.join(list(filter(None, ret)))
 
+    def mkslow_peripheral(self, *args):
+        ret = []
+        for (name, count) in self.ifacecount:
+            for i in range(count):
+                print "mkslow", name, count
+                x = self.data[name].mkslow_peripheral()
+                print name, count, x
+                suffix = self.data[name].mksuffix(name, i)
+                ret.append(x.format(suffix))
+        return '\n'.join(list(filter(None, ret)))
+
+    def mk_connection(self, *args):
+        ret = []
+        for (name, count) in self.ifacecount:
+            for i in range(count):
+                print "mk_conn", name, i
+                txt = self.data[name].mk_connection(i)
+                if name == 'gpioa':
+                    print "txt", txt
+                    print self.data[name].mk_connection
+                ret.append(txt)
+        return '\n'.join(list(filter(None, ret)))
+
+    def mk_cellconn(self):
+        ret = []
+        cellcount = 0
+        for (name, count) in self.ifacecount:
+            for i in range(count):
+                res = self.data[name].mk_cellconn(cellcount, name, i)
+                if not res:
+                    continue
+                (txt, cellcount) = res
+                ret.append(txt)
+        ret = '\n'.join(list(filter(None, ret)))
+        return pinmux_cellrule.format(ret)
+
+    def mk_pincon(self):
+        ret = []
+        for (name, count) in self.ifacecount:
+            for i in range(count):
+                txt = self.data[name].mk_pincon(name, i)
+                ret.append(txt)
+        return '\n'.join(list(filter(None, ret)))
+
 
 class PFactory(object):
     def getcls(self, name):
-        return {'uart': uart,
-                'rs232': rs232,
-                'twi': twi,
-                'qspi': qspi,
-                'pwm': pwm,
-                'gpio': gpio
-                }.get(name, None)
+        for k, v in {'uart': uart,
+                     'rs232': rs232,
+                     'twi': twi,
+                     'qspi': qspi,
+                     'pwm': pwm,
+                     'gpio': gpio
+                     }.items():
+            if name.startswith(k):
+                return v
+        return None
 
 
 slowfactory = PFactory()
+
+if __name__ == '__main__':
+    p = uart('uart')
+    print p.slowimport()
+    print p.slowifdecl()
+    i = PeripheralIface('uart')
+    print i, i.slow
+    i = PeripheralIface('gpioa')
+    print i, i.slow