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

Improve handling of single connections in Cable.connect()

Browse Source
This commit is contained in:
Daniel Rojas 2020-05-24 21:46:02 +02:00
parent 06a9188e83
commit a2971e7ee2
1 changed files with 12 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

12
src/wireviz.py
View File

@ -191,6 +191,8 @@ class Node:
self.loops = [] self.loops = []
if pinout is None: if pinout is None:
if num_pins is None:
num_pins = 1
self.pinout = ('',) * num_pins self.pinout = ('',) * num_pins
else: else:
if num_pins is None: if num_pins is None:
@ -252,6 +254,9 @@ class Cable:
via = tuple(x+1 for x in range(len(self.colors))) via = tuple(x+1 for x in range(len(self.colors)))
if to_pin == 'auto': if to_pin == 'auto':
to_pin = tuple(x+1 for x in range(len(self.colors))) to_pin = tuple(x+1 for x in range(len(self.colors)))
from_pin = int2tuple(from_pin)
via = int2tuple(via)
to_pin = int2tuple(to_pin)
if len(from_pin) != len(to_pin): if len(from_pin) != len(to_pin):
raise Exception('from_pin must have the same number of elements as to_pin') raise Exception('from_pin must have the same number of elements as to_pin')
for i, x in enumerate(from_pin): for i, x in enumerate(from_pin):
@ -275,6 +280,13 @@ def nested(input):
s = '|'.join(l) s = '|'.join(l)
return s return s
def int2tuple(input):
if isinstance(input, tuple):
output = input
else:
output = (input,)
return output
def translate_color(input, color_mode): def translate_color(input, color_mode):
if input == '': if input == '':
output = '' output = ''