From a5cc2e3b826528b4e4492f74703cb4a0582343de Mon Sep 17 00:00:00 2001 From: Lukas Winkler Date: Sun, 27 Dec 2020 13:40:46 +0100 Subject: [PATCH] improve graph output --- graph.py | 18 +++++++++++++++--- 1 file changed, 15 insertions(+), 3 deletions(-) diff --git a/graph.py b/graph.py index 9d9cc98..f2fdd5b 100644 --- a/graph.py +++ b/graph.py @@ -3,11 +3,23 @@ from graphviz import Digraph from extradata import ExtraData from utils import filename_from_argv -ed = ExtraData.load(filename_from_argv().with_suffix(".extra.json")) +fn = filename_from_argv() +ed = ExtraData.load(fn.with_suffix(".extra.json")) dot = Digraph(comment='Collisions') for merged, originals in ed.tree._tree.items(): + first_parent = True for parent in originals["parents"]: - dot.edge(str(parent), str(merged)) + meta = originals["meta"] + water_ret = meta["water_retention"] + mass_ret = meta["mass_retention"] + if first_parent: -dot.render('graph.gv', view=True) + label = f"{water_ret:.2f}/{mass_ret:.2f}" + first_parent = False + else: + label = None + dot.edge(str(parent), str(merged), xlabel=label) + +# dot.engine = 'neato' +dot.render(fn.with_suffix(".gv"), view=False, format="svg")