Tibeuleu/KozaiLidov
1
0
Fork 0
Code Activity

add parameter to save to gif or not

Browse Source
This commit is contained in:
Thibault Barnouin
2022-01-07 11:28:22 +01:00
parent deaf53f100
commit 209cb0aabb
4 changed files with 21 additions and 11 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
main.py
View File

@@ -32,6 +32,7 @@ def main():
integrator = "leapfrog"
n_bodies = 3
display = False
gif = False
savename = "{0:d}bodies_{1:s}".format(n_bodies, integrator)
#simulation start
@@ -46,9 +47,9 @@ def main():
if i != 0:
display = False
if integrator.lower() in ['leapfrog', 'frogleap', 'frog']:
E0, L0, sma, ecc = leapfrog(dyn_syst, bin_syst, duration, step0, recover_param=True, display=display, savename=savename)
E0, L0, sma, ecc = leapfrog(dyn_syst, bin_syst, duration, step0, recover_param=True, display=display, savename=savename, gif=gif)
elif integrator.lower() in ['hermite','herm']:
E0, L0, sma, ecc = hermite(dyn_syst, bin_syst, duration, step0, recover_param=True, display=display, savename=savename)
E0, L0, sma, ecc = hermite(dyn_syst, bin_syst, duration, step0, recover_param=True, display=display, savename=savename, gif=gif)
E.append(E0)
L.append(L0)