mirror of
https://github.com/Findus23/rebound-collisions.git
synced 2024-09-19 15:53:48 +02:00
improve graph output
This commit is contained in:
parent
1156546300
commit
a5cc2e3b82
1 changed files with 15 additions and 3 deletions
18
graph.py
18
graph.py
|
@ -3,11 +3,23 @@ from graphviz import Digraph
|
||||||
from extradata import ExtraData
|
from extradata import ExtraData
|
||||||
from utils import filename_from_argv
|
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')
|
dot = Digraph(comment='Collisions')
|
||||||
for merged, originals in ed.tree._tree.items():
|
for merged, originals in ed.tree._tree.items():
|
||||||
|
first_parent = True
|
||||||
for parent in originals["parents"]:
|
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")
|
||||||
|
|
Loading…
Reference in a new issue