warehack.ing/WireViz
2
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Loop color codes

Browse Source
This commit is contained in:
Daniel Rojas 2020-05-22 20:02:34 +02:00
parent 5fc93a8bce
commit 81b2cb5423
1 changed files with 16 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

18
src/wireviz.py
View File

@ -171,7 +171,7 @@ class Cable:
if color_code is None and colors is None:
self.colors = ('',) * num_wires
else:
if colors is None:
if colors is None: # no custom color pallet was specified
if num_wires is None:
raise Exception('Unknown number of wires')
else:
@ -179,9 +179,21 @@ class Cable:
if color_code not in COLOR_CODES:
raise Exception('Unknown color code')
else:
if num_wires is None:
self.colors = colors
cc = COLOR_CODES[color_code]
cc = tuple(cc)
if num_wires <= len(cc):
self.colors = cc[:num_wires]
else:
n = num_wires / len(cc) + 1
cc = cc * int(n)
self.colors = cc[:num_wires]
else: # custom color pallet was specified
if num_wires is None: # assume number of wires = number of items in custom pallet
self.colors = colors
else: # number of wires was specified
# TODO: loop through colors if num_wires > len(colors)
self.colors = colors[:num_wires]
def connect(self, from_name, from_pin, via, to_name, to_pin):