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

Merge branch 'dev'

Browse Source
This commit is contained in:
Daniel Rojas 2020-05-22 17:29:14 +02:00
commit f11dce97d2
5 changed files with 81 additions and 76 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

14
src/example1.py Normal file
View File

@ -0,0 +1,14 @@
import wireviz
X1 = wireviz.Node('X1', type='D-Sub', gender='female', pinout=('DCD','RX','TX','DTR','GND','DSR','RTS','CTS','RI'), ports_right=True)
X2 = wireviz.Node('X2', type='Molex KK 254', gender='female', pinout=('GND','RX','TX','NC','OUT','IN'), ports_left=True)
W1 = wireviz.Cable('W1', mm2=0.25, length=0.2, show_name=True, show_pinout=True, num_wires=3, color_code='DIN', shield=True)
# Option 1: define wires and shield in one line
# W1.connect(X1,(5,2,3,5),(1,2,3,'s'),X2,(1,3,2,None))
# Option 2: define wires and shield separately
W1.connect(X1,(5,2,3),'auto',X2,(1,3,2)) # wires
W1.connect(X1,(5,),('s',),X2,(None,)) # shield
X2.loop(5,6)
objects = [X1, X2, W1]
wireviz.output(objects)

32
src/example2.py Normal file
View File

@ -0,0 +1,32 @@
import wireviz
PINOUT_I2C = ('GND','+5V','SCL','SDA')
COLORS_I2C = ('BK', 'RD', 'YE', 'GN')
PINOUT_SPI_DATAONLY = ('MISO','MOSI','SCK')
X1 = wireviz.Node('X1',type='Molex KK 254', gender='female', pinout=(
'GND',
'+5V',
'SCL',
'SDA',
'MISO',
'MOSI',
'SCK',
'N/C'
), ports_right=True)
X2 = wireviz.Node('X2', type='Molex KK 254', gender='female', pinout=PINOUT_I2C, ports_left=True)
X3 = wireviz.Node('X3', type='Molex KK 254', gender='female', pinout=PINOUT_I2C, ports_left=True)
X4 = wireviz.Node('X4', type='Molex KK 254', gender='female', pinout=('GND','+12V')+PINOUT_SPI_DATAONLY, ports_left=True)
X5 = wireviz.Node('X5', type='Molex Micro-Fit', gender='male', pinout=('GND','+12V'), ports_right=True)
W1 = wireviz.Cable('W1', mm2=0.14, length=0.2, colors=COLORS_I2C)
W2 = wireviz.Cable('W2', mm2=0.14, length=0.2, colors=COLORS_I2C)
W3 = wireviz.Cable('W3', mm2=0.14, length=0.2, colors=('BK','BU','OG','VT'))
W4 = wireviz.Cable('W4', mm2=0.5, length=0.35, colors=('BK','RD'))
W1.connect(X1,(1,2,3,4),'auto',X2,'auto')
W2.connect(X1,(1,2,3,4),'auto',X3,'auto')
W3.connect(X1,(1,5,6,7),'auto',X4,(1,3,4,5))
W4.connect(X5,'auto','auto',X4,'auto')
objects = [X1, X2, X3, X4, X5, W1, W2, W3, W4]
wireviz.output(objects)

15
src/example3.py Normal file
View File

@ -0,0 +1,15 @@
import wireviz
wireviz.color_mode = 'full' # short/SHORT/full/FULL/hex/HEX
X1 = wireviz.Node('X1', num_pins=10, ports_right=True)
X2 = wireviz.Node('X2', num_pins=10, ports_left=True)
W1 = wireviz.Cable('W1', num_wires=10, color_code='IEC')
W1.connect_all_straight(X1,X2)
X3 = wireviz.Node('X3', num_pins=10, ports_right=True)
X4 = wireviz.Node('X4', num_pins=10, ports_left=True)
W2 = wireviz.Cable('W2', num_wires=10, color_code='DIN')
W2.connect_all_straight(X3,X4)
objects = [X1, X2, W1, X3, X4, W2]
wireviz.output(objects)

76
src/test.py
View File

@ -1,76 +0,0 @@
import wireviz
PINOUT_I2C = ('GND','+5V','SCL','SDA')
COLORS_I2C = ('BK', 'RD', 'YE', 'GN')
PINOUT_SPI_DATAONLY = ('MISO','MOSI','SCK')
wireviz.color_mode = 'SHORT' # short/SHORT/full/FULL/hex/HEX
# example 1
X1 = wireviz.Node('X1', type='D-Sub', gender='female', pinout=('DCD','RX','TX','DTR','GND','DSR','RTS','CTS','RI'), ports_right=True)
X2 = wireviz.Node('X2', type='Molex KK 254', gender='female', pinout=('GND','RX','TX','NC','OUT','IN'), ports_left=True)
W1 = wireviz.Cable('W1', mm2=0.25, length=0.2, show_name=True, show_pinout=True, num_wires=3, color_code='DIN', shield=True)
# Option 1: define wires and shield in one line
# W1.connect(X1,(5,2,3,5),(1,2,3,'s'),X2,(1,3,2,None))
# Option 2: define wires and shield separately
W1.connect(X1,(5,2,3),'auto',X2,(1,3,2)) # wires
W1.connect(X1,(5,),('s',),X2,(None,)) # shield
X2.loop(5,6)
objects = [X1, X2, W1]
# example 2
# X1 = wireviz.Node('X1',type='Molex KK 254', gender='female', pinout=(
# 'GND',
# '+5V',
# 'SCL',
# 'SDA',
# 'MISO',
# 'MOSI',
# 'SCK',
# 'N/C'
# ), ports_right=True)
# X2 = wireviz.Node('X2', type='Molex KK 254', gender='female', pinout=PINOUT_I2C, ports_left=True)
# X3 = wireviz.Node('X3', type='Molex KK 254', gender='female', pinout=PINOUT_I2C, ports_left=True)
# X4 = wireviz.Node('X4', type='Molex KK 254', gender='female', pinout=('GND','+12V')+PINOUT_SPI_DATAONLY, ports_left=True)
# X5 = wireviz.Node('X5', type='Molex Micro-Fit', gender='male', pinout=('GND','+12V'), ports_right=True)
# W1 = wireviz.Cable('W1', mm2=0.14, length=0.2, colors=COLORS_I2C)
# W2 = wireviz.Cable('W2', mm2=0.14, length=0.2, colors=COLORS_I2C)
# W3 = wireviz.Cable('W3', mm2=0.14, length=0.2, colors=('BK','BU','OG','VT'))
# W4 = wireviz.Cable('W4', mm2=0.5, length=0.35, colors=('BK','RD'))
# W1.connect(X1,(1,2,3,4),'auto',X2,'auto')
# W2.connect(X1,(1,2,3,4),'auto',X3,'auto')
# W3.connect(X1,(1,5,6,7),'auto',X4,(1,3,4,5))
# W4.connect(X5,'auto','auto',X4,'auto')
# objects = [X1, X2, X3, X4, X5, W1, W2, W3, W4]
# example 3
# X1 = wireviz.Node('X1', num_pins=10, ports_right=True)
# X2 = wireviz.Node('X2', num_pins=10, ports_left=True)
# W1 = wireviz.Cable('W1', num_wires=10, color_code='IEC')
# W1.connect_all_straight(X1,X2)
# X3 = wireviz.Node('X3', num_pins=10, ports_right=True)
# X4 = wireviz.Node('X4', num_pins=10, ports_left=True)
# W2 = wireviz.Cable('W2', num_wires=10, color_code='DIN')
# W2.connect_all_straight(X3,X4)
# objects = [X1, X2, W1, X3, X4, W2]
with open('output/output.dot','w') as f:
with open('input/header.dot','r') as infile:
for line in infile:
f.write(line)
f.write('\n\n')
for o in objects:
f.write(o.graphviz() + '\n')
f.write('\n\n')
with open('input/footer.dot','r') as infile:
for line in infile:
f.write(line)
# print output file
# with open('output/output.dot','r') as f:
# for line in f:
# print(line)

20
src/wireviz.py
View File

@ -281,3 +281,23 @@ class Cable:
s = s + '}'
return s
def output(objects, print_to_screen=False):
with open('output/output.dot','w') as f:
with open('input/header.dot','r') as infile:
for line in infile:
f.write(line)
f.write('\n\n')
for o in objects:
f.write(o.graphviz() + '\n')
f.write('\n\n')
with open('input/footer.dot','r') as infile:
for line in infile:
f.write(line)
if print_to_screen == True:
with open('output/output.dot','r') as f:
for line in f:
print(line)