diff --git a/src/wireviz/Harness.py b/src/wireviz/Harness.py index 4928653a..4c2bf756 100644 --- a/src/wireviz/Harness.py +++ b/src/wireviz/Harness.py @@ -202,11 +202,18 @@ def create_graph(self) -> Graph: html = f'{html}{bla}' html = f'{html}' - bgcolors = ['#000000'] + get_color_hex(connection_color, pad=pad) + ['#000000'] - html = f'{html}' - for j, bgcolor in enumerate(bgcolors[::-1]): # Reverse to match the curved wires when more than 2 colors - html = f'{html}' - html = html + '
' + colors = get_color_hex(connection_color) + html = f'{html}' + html = f'{html}' + numColors = len(colors) + for i in range(12) if numColors > 1 else range(1): #12 works well up to 4 colors + html = f'{html}' + + html = html + '
' + + #for j, bgcolor in enumerate(bgcolors[::-1]): # Reverse to match the curved wires when more than 2 colors + # html = f'{html}' + #html = html + '' if(cable.category == 'bundle'): # for bundles individual wires can have part information # create a list of wire parameters wireidentification = [] @@ -290,7 +297,7 @@ def create_graph(self) -> Graph: dot.edge (cable.name, connection_color.to_name) #cable color - dot.attr('edge', style="solid", penwidth="2.0", color=colors[0]) + dot.attr('edge', style="solid", penwidth="4.0" if isShield else "2.0", color=colors[0]) if leftConn: dot.edge(code_left_1, code_left_2) if rightConn: