maintaining distinct arrays for muxed and dedicated cells

[pinmux.git] / src / pinmux_generator.py

diff --git a/src/pinmux_generator.py b/src/pinmux_generator.py
index dbea8d5609ebd274dc64e9c4a267a81b7713c5e4..0de361f180501b82218acb0ec2ea9f1050995c67 100644 (file)
--- a/src/pinmux_generator.py
+++ b/src/pinmux_generator.py
@@ -31,6 +31,8 @@ if not os.path.exists("bsv_src"):
     os.makedirs("bsv_src")
 
 bsv_file = open("./bsv_src/pinmux.bsv", "w")
+
+
 header = '''
 /*
    This BSV file has been generated by the PinMux tool available at:
@@ -72,8 +74,8 @@ bsv_file.write('''
       // of muxes happening per IO. For now we have a generalized width
       // where each IO will have the same number of muxes.''')
 
-for i in range(0, N_IO):
-    bsv_file.write(mux_interface.format(i))
+for cell in muxed_cells:
+    bsv_file.write(mux_interface.format(cell[0], len(cell) - 1))
 
 bsv_file.write('''
 
@@ -115,8 +117,8 @@ bsv_file.write('''
       // the followins wires capture the pin-mux selection
       // values for each mux assigned to a CELL
 ''')
-for i in range(0, N_IO):
-    bsv_file.write(muxwire.format(i))
+for cell in muxed_cells:
+    bsv_file.write(muxwire.format(cell[0], len(cell) - 1))
 
 
 bsv_file.write(
@@ -153,8 +155,8 @@ bsv_file.write('''
 ''')
 # ====================================================================
 # ================= interface definitions for each method =============#
-for i in range(0, N_IO):
-    bsv_file.write(mux_interface_def.format(i))
+for cell in muxed_cells:
+    bsv_file.write(mux_interface_def.format(cell[0], len(cell) - 1))
 for i in range(0, N_IO):
     bsv_file.write(io_interface_def.format(i))
 for i in range(0, N_UART):
@@ -166,3 +168,4 @@ for i in range(0, N_TWI):
 bsv_file.write(footer)
 print("BSV file successfully generated: bsv_src/pinmux.bsv")
 # ======================================================================
+bsv_file.close()