1
0

plot for multiple step values at time

This commit is contained in:
Thibault Barnouin
2021-11-14 00:06:22 +01:00
parent d5feffb0fe
commit 48b6027406
10 changed files with 21 additions and 11 deletions

14
main.py
View File

@@ -9,7 +9,7 @@ from lib.units import *
def main():
#initialisation
m = np.array([1., 1., 0.])*Ms # Masses in Solar mass
m = np.array([1., 1., 0])*Ms # Masses in Solar mass
a = np.array([1., 1., 5.])*au # Semi-major axis in astronomical units
e = np.array([0., 0., 1./4.]) # Eccentricity
psi = np.array([0., 0., 0.])*np.pi/180. # Inclination of the orbital plane in degrees
@@ -30,11 +30,13 @@ def main():
dyn_syst = System(bodylist)
dyn_syst.COMShift()
duration, step = 100*yr, 1e4
#E, L = dyn_syst.leapfrog(duration, step, recover_param=True, display=True)
E, L = dyn_syst.hermite(duration,step, recover_param=True, display=True)
parameters = [duration, step, dyn_syst, "hermite"]
display_parameters(E, L, parameters=parameters, savename="3bodies_hermite")
duration, step1, step2 = 100*yr, 1e4, 1e5
E1, L1 = dyn_syst.leapfrog(duration, step1, recover_param=True)#, display=True)
E2, L2 = dyn_syst.leapfrog(duration, step2, recover_param=True)#, display=True)
#E1, L1 = dyn_syst.hermite(duration, step1, recover_param=True)#, display=True)
#E2, L2 = dyn_syst.hermite(duration, step2, recover_param=True)#, display=True)
parameters = [duration, [step1, step2], dyn_syst, "leapfrog"]
display_parameters([E1, E2], [L1, L2], parameters=parameters, savename="3bodies_leapfrog")
return 0