X-Git-Url: https://git.libre-soc.org/?a=blobdiff_plain;f=src%2Fparse.py;h=8d67b01e0b2e2809b39120a4804c9646133a9074;hb=24c4414dee7455efbdbafb555517020fa405a4f4;hp=a6875f36cacb60ce6ea0e49ca219d60e8e8c1bb9;hpb=a940d019c8053f170f30ca096ac972fbf642faa5;p=pinmux.git

diff --git a/src/parse.py b/src/parse.py
index a6875f3..8d67b01 100644
--- a/src/parse.py
+++ b/src/parse.py
@@ -26,7 +26,7 @@ class Parse(object):
         self.muxed_cells_bank = []
         self.dedicated_cells = []
         self.pinnumbers = []
-        self.bankwidths = {} 
+        self.bankwidths = {}
         self.banksize = {}
         self.bankstart = {}
 
@@ -39,7 +39,7 @@ class Parse(object):
                 self.bankwidths[line1[0]] = int(line1[3])
                 self.banksize[line1[0]] = int(line1[2])
                 self.bankstart[line1[0]] = int(line1[1])
-            
+
         # == capture the number of IO cells required == #
         fname = 'pinmap.txt'
         if pth:
@@ -53,9 +53,9 @@ class Parse(object):
                 self.muxed_cells_bank.append(line1[1])
                 self.muxed_cells_width.append(int(line1[2]))
                 # XXX TODO: dedicated pins in separate file
-                #if len(line1) == 2:  # dedicated
+                # if len(line1) == 2:  # dedicated
                 #    self.dedicated_cells.append(line1)
-                #else:
+                # else:
                 for i in range(3, len(line1)):
                     # XXX HORRIBLE HACK!!
                     if line1[i].startswith('pwm'):
@@ -77,7 +77,7 @@ class Parse(object):
         #print("Muxer bit width: " + str(self.cell_bitwidth))
         print("Muxed IOs: " + str(len(self.muxed_cells)))
         print("Dedicated IOs: " + str(len(self.dedicated_cells)))
-        #sys.exit(0)
+        # sys.exit(0)
 
     def do_checks(self):
         """ Multiple checks to see if the user has not screwed up
@@ -120,7 +120,13 @@ class Parse(object):
         return int(math.log(max_num_cells + 1, 2))
 
     def get_muxwidth(self, cellnum):
-        return self.muxed_cells_width[cellnum]
+        return self.muxed_cells_width[int(cellnum)]
+
+    def get_muxbitwidth(self, cellnum):
+        wid = self.get_muxwidth(cellnum)
+        if wid == 1:
+            return 0
+        return int(math.log(wid + 1, 2))
 
 
 if __name__ == '__main__':