From 32fb674e836efd96770bd240ac8c465aadab0c52 Mon Sep 17 00:00:00 2001 From: Visa Tuominen Date: Mon, 8 Aug 2022 15:47:58 +0300 Subject: [PATCH] match loop index by value, add newlines to header --- src/wireviz/Harness.py | 14 +++++--------- 1 file changed, 5 insertions(+), 9 deletions(-) diff --git a/src/wireviz/Harness.py b/src/wireviz/Harness.py index af02399..4153193 100644 --- a/src/wireviz/Harness.py +++ b/src/wireviz/Harness.py @@ -93,8 +93,8 @@ class Harness: def create_graph(self) -> Graph: dot = Graph() - dot.body.append(f'// Graph generated by {APP_NAME} {__version__}') - dot.body.append(f'// {APP_URL}') + dot.body.append(f'// Graph generated by {APP_NAME} {__version__}\n') + dot.body.append(f'// {APP_URL}\n') dot.attr('graph', rankdir='LR', ranksep='2', bgcolor=wv_colors.translate_color(self.options.bgcolor, "HEX"), @@ -189,13 +189,9 @@ class Harness: else: raise Exception('No side for loops') for loop in connector.loops: - if type(loop[0]) == str: - i_loop_0 = connector.pins.index(loop[0]) - i_loop_1 = connector.pins.index(loop[1]) - else: - i_loop_0 = loop[0] - i_loop_1 = loop[1] - dot.edge(f'{connector.name}:p{i_loop_0}{loop_side}:{loop_dir}', + i_loop_0 = connector.pins.index(loop[0])+1 + i_loop_1 = connector.pins.index(loop[1])+1 + dot.edge(f'{connector.name}:p{i_loop_0}{loop_side}:{loop_dir}', f'{connector.name}:p{i_loop_1}{loop_side}:{loop_dir}')