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

10
lib/LeapFrog.py
View File

@@ -33,7 +33,7 @@ def LP(dyn_syst, dt):
Drift(dyn_syst, dt / 2)
dyn_syst.time = dyn_syst.time + dt
def leapfrog(dyn_syst, bin_syst, duration, dt, recover_param=False, display=False, savename=None):
def leapfrog(dyn_syst, bin_syst, duration, dt, recover_param=False, display=False, savename=None, gif=False):
if display:
try:
system("mkdir tmp")
@@ -56,11 +56,15 @@ def leapfrog(dyn_syst, bin_syst, duration, dt, recover_param=False, display=Fals
ecc[j] = bin_syst.ecc
if display and j % 10 == 0:
if gif:
step = j
else:
step = None
# display progression
if len(dyn_syst.bodylist) == 1:
d.on_running(dyn_syst, step=j, label="{0:.2f} years".format(j*dt/yr))
d.on_running(dyn_syst, step=step, label="{0:.2f} years".format(j*dt/yr))
else:
d.on_running(dyn_syst, step=j, label="{0:.2f} years".format(j*dt/yr))
d.on_running(dyn_syst, step=step, label="{0:.2f} years".format(j*dt/yr))
if display:
d.close()
if not savename is None:

10
lib/hermite.py
View File

@@ -79,7 +79,7 @@ def HPC(dyn_syst, dt): # update position and velocities of bodies in system wit
dyn_syst.time = dyn_syst.time + dt
def hermite(dyn_syst, bin_syst, duration, dt, recover_param=False, display=False, savename=None):
def hermite(dyn_syst, bin_syst, duration, dt, recover_param=False, display=False, savename=None, gif=False):
if display:
try:
system("mkdir tmp")
@@ -103,11 +103,15 @@ def hermite(dyn_syst, bin_syst, duration, dt, recover_param=False, display=False
ecc[j] = bin_syst.ecc
if display and j % 10 == 0:
if gif:
step = j
else:
step = None
# display progression
if len(dyn_syst.bodylist) == 1:
d.on_running(dyn_syst, step=j, label="{0:.2f} yearq".format(j*dt))
d.on_running(dyn_syst, step=step, label="{0:.2f} years".format(j*dt/yr))
else:
d.on_running(dyn_syst, step=j, label="{0:.2f} years".format(j*dt))
d.on_running(dyn_syst, step=step, label="{0:.2f} years".format(j*dt/yr))
if display:
d.close()
if not savename is None: