mirror/rebound-collisions
1
0
Fork 0
mirror of https://github.com/Findus23/rebound-collisions.git synced 2024-09-19 15:53:48 +02:00
Code Releases Activity

fix graph

Browse source
This commit is contained in:
Lukas Winkler 2021-01-04 12:25:13 +01:00
parent 5ae33a1fdb
commit 160eb3504d
Signed by: lukas
GPG key ID: 54DE4D798D244853
1 changed files with 5 additions and 5 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

10
graph.py
View file

@ -1,6 +1,6 @@
from graphviz import Digraph
from extradata import ExtraData
from extradata import ExtraData, CollisionMeta
from utils import filename_from_argv
fn = filename_from_argv()
@ -10,9 +10,9 @@ dot = Digraph(comment='Collisions')
for merged, originals in ed.tree._tree.items():
first_parent = True
for parent in originals["parents"]:
meta = originals["meta"]
water_ret = meta["water_retention"]
mass_ret = meta["mass_retention"]
meta: CollisionMeta = originals["meta"]
water_ret = meta.water_retention
mass_ret = meta.mass_retention
if first_parent:
label = f"{water_ret:.2f}/{mass_ret:.2f}"
@ -22,4 +22,4 @@ for merged, originals in ed.tree._tree.items():
dot.edge(str(parent), str(merged), xlabel=label)
# dot.engine = 'neato'
dot.render(fn.with_suffix(".gv"), view=False, format="svg")
dot.render(fn.with_suffix(".gv"), view=True, format="svg")