write mux width and read back in

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

diff --git a/src/spec/gen.py b/src/spec/gen.py
index 2c1335f6d0d89a11504794c774ff1562a5dec55d..e1dc561111b6ffa425aed312ebb30e5b2cc4877c 100644 (file)
--- a/src/spec/gen.py
+++ b/src/spec/gen.py
@@ -3,7 +3,8 @@ import os.path
 from spec.interfaces import Pinouts
 
 
-def specgen(of, pth, pinouts, bankspec, pinbanks, fixedpins):
+def specgen(of, pth, pinouts, bankspec, muxwidths, pinbanks, fixedpins,
+            fastbus):
     """ generates a specification of pinouts (tsv files)
         for reading in by pinmux.
 
@@ -28,10 +29,18 @@ def specgen(of, pth, pinouts, bankspec, pinbanks, fixedpins):
     #print pinouts.ganged.items()
     if not os.path.exists(pth):
         os.makedirs(pth)
+    with open(os.path.join(pth, 'muxwidths.txt'), 'w') as f:
+        for k, v in pinouts.muxwidths.items():
+            f.write("%s\t%d\n" % (k, v))
     with open(os.path.join(pth, 'interfaces.txt'), 'w') as f:
         for k in pinouts.fnspec.keys():
             s = pinouts.fnspec[k]
-            f.write("%s\t%d\n" % (k.lower(), len(s)))
+            line = [k.lower(), str(len(s))]
+            for b in fastbus:
+                if b.startswith(k.lower()):
+                    line.append(b)
+            line = '\t'.join(line)
+            f.write("%s\n" % line)
             s0 = s[list(s.keys())[0]]  # hack, take first
             gangedgroup = pinouts.ganged[k]
             with open(os.path.join(pth, '%s.txt' % k.lower()), 'w') as g:
@@ -85,5 +94,6 @@ def specgen(of, pth, pinouts, bankspec, pinbanks, fixedpins):
         keys = sorted(bankspec.keys())
         for bank in keys:
             pinstart = bankspec[bank]
-            of.write("* %s %d %d\n" % (bank, pinstart, pinbanks[bank]))
-            g.write("%s\t%d\t%d\n" % (bank, pinstart, pinbanks[bank]))
+            wid = muxwidths[bank]
+            of.write("* %s %d %d %d\n" % (bank, pinstart, pinbanks[bank], wid))
+            g.write("%s\t%d\t%d\t%d\n" % (bank, pinstart, pinbanks[bank], wid))