add pin-to-litex json map

[pinmux.git] / src / spec / ifaceprint.py

diff --git a/src/spec/ifaceprint.py b/src/spec/ifaceprint.py
index 3d3cfb4b27662b58e348dacc8202eb8e2184c0bc..808985ce5f4fa884939bbb217e73672508f385dd 100644 (file)
--- a/src/spec/ifaceprint.py
+++ b/src/spec/ifaceprint.py
@@ -1,7 +1,7 @@
 #!/usr/bin/env python
 
 from copy import deepcopy
-
+from collections import OrderedDict
 
 def display(of, pins, banksel=None, muxwidth=4):
     of.write("""\
@@ -89,6 +89,7 @@ def map_name(pinmap, fn, fblower, pin, rename):
 
 def python_pindict(of, pinmap, pins, function_names, dname, remap):
 
+    res = OrderedDict()
     of.write("\n%s = OrderedDict()\n" % dname)
 
     for k, pingroup in pins.byspec.items():
@@ -97,9 +98,12 @@ def python_pindict(of, pinmap, pins, function_names, dname, remap):
             a = "%s%s" % (a, n)
         fblower = a.lower()
         of.write("%s['%s'] = [ " % (dname, fblower))
+        res[fblower] = []
         count = 0
         for i, p in enumerate(pingroup):
-            of.write("'%s', " % map_name(pinmap, k[0], fblower, p, remap))
+            name = map_name(pinmap, k[0], fblower, p, remap)
+            res[fblower].append(name)
+            of.write("'%s', " % name)
             count += 1
             if count == 4 and i != len(pingroup)-1:
                 of.write("\n                ")
@@ -107,6 +111,7 @@ def python_pindict(of, pinmap, pins, function_names, dname, remap):
         of.write("]\n")
         print "    dict %s" % dname, a, n, pingroup
     of.write("\n\n")
+    return res
 
 def python_dict_fns(of, pinmap, pins, function_names):
     of.write("# auto-generated by Libre-SOC pinmux program: do not edit\n")
@@ -129,8 +134,28 @@ def python_dict_fns(of, pinmap, pins, function_names):
     print "by spec", pins.byspec
     print pinmap
 
-    python_pindict(of, {}, pins, function_names, 'pindict', False)
-    python_pindict(of, pinmap, pins, function_names, 'litexdict', True)
+    pd = python_pindict(of, {}, pins, function_names, 'pindict', False)
+    ld = python_pindict(of, pinmap, pins, function_names, 'litexdict', True)
+
+    print "pd", pd
+    print "ld", ld
+    # process results and create name map
+    litexmap = OrderedDict()
+    for k in pd.keys():
+        pl = pd[k]
+        ll = ld[k]
+        for pname, lname in zip(pl, ll):
+            pname = "%s_%s" % (k, pname[:-1]) # strip direction +/-/*
+            lname = lname[:-1] # strip direction +/-/*
+            if k in ['eint', 'pwm', 'gpio', 'vdd', 'vss']: # sigh
+                lname = "%s_%s" % (k, lname)
+            litexmap[pname] = lname
+    print "litexmap", litexmap
+    of.write("litexmap = {\n")
+    for k, v in litexmap.items():
+        of.write("\t'%s': '%s',\n" % (k, v))
+    of.write("}\n")
+    return litexmap
 
 
 def display_fns(of, bankspec, pins, function_names):