Tibeuleu/KozaiLidov
1
0
Fork 0
Code Activity

Update display with readable data

Browse Source
This commit is contained in:
Thibault Barnouin
2021-11-12 17:08:07 +01:00
parent 966b6493ac
commit 27844f5cb8
21 changed files with 91 additions and 39 deletions
Download Patch File Download Diff File Expand all files Collapse all files

38
main.py
View File

@@ -4,50 +4,38 @@ from sys import exit as sysexit
import numpy as np
import matplotlib.pyplot as plt
from lib.objects import Body, System
globals()['G'] = 6.67e-11 #Gravitational constant in SI units
globals()['Ms'] = 2e30 #Solar mass in kg
globals()['au'] = 1.5e11 #Astronomical unit in m
from lib.plots import display_parameters
from lib.units import *
def main():
#initialisation
m = np.array([1, 1, 0.1])*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., 80.])*np.pi/180. # Inclination of the orbital plane in degrees
psi = np.array([0., 0., 0.])*np.pi/180. # Inclination of the orbital plane in degrees
x1 = np.array([0., -1., 0.])*a[0]
x2 = np.array([0., 1., 0.])*a[1]
x3 = np.array([np.cos(psi[2]), 0., np.sin(psi[2])])*a[2]
q = np.array([x1, x2, x3])
v1 = np.array([-np.sqrt(G*m[1]**2/((m[0]+m[1])*np.sqrt(np.sum((q[0]-q[1])**2)))), 0., 0.])
v2 = np.array([np.sqrt(G*m[0]**2/((m[0]+m[1])*np.sqrt(np.sum((q[0]-q[1])**2)))), 0., 0.])
v1 = np.array([np.sqrt(G*m[1]**2/((m[0]+m[1])*np.sqrt(np.sum((q[0]-q[1])**2)))), 0., 0.])
v2 = np.array([-np.sqrt(G*m[0]**2/((m[0]+m[1])*np.sqrt(np.sum((q[0]-q[1])**2)))), 0., 0.])
v3 = np.array([0., np.sqrt(G*(m[0]+m[1])*(2./np.sqrt(np.sum(q[2]**2))-1./a[2])), 0.])
v = np.array([v1, v2, v3])
bodylist = []
for i in range(3):
for i in range(2):
bodylist.append(Body(m[i], q[i], v[i]))
dyn_syst = System(bodylist)
dyn_syst.COMShift()
duration, step = 100*3e7, 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)
plt.close()
fig1 = plt.figure(figsize=(30,15))
ax1 = fig1.add_subplot(111)
ax1.plot(np.arange(E.shape[0])/duration, E, label=r"$E_m$")
ax1.legend()
fig1.savefig("plots/Em.png",bbox_inches="tight")
fig2 = plt.figure(figsize=(30,15))
ax2 = fig2.add_subplot(111)
ax2.plot(np.arange(L.shape[0])/duration, np.sum(L**2,axis=1), label=r"$L^2$")
ax2.legend()
fig2.savefig("plots/L2.png",bbox_inches="tight")
plt.show(block=True)
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="2bodies_hermite")
return 0
if __name__ == '__main__':