Tibeuleu/KozaiLidov
1
0
Fork 0
Code Activity

modify initialisation parameters and add gif animation saving

Browse Source
This commit is contained in:
Thibault Barnouin
2021-10-30 16:42:12 +02:00
parent 9a16400a89
commit d538dfe7a4
5 changed files with 42 additions and 18 deletions
Download Patch File Download Diff File Expand all files Collapse all files

17
main.py
View File

@@ -8,34 +8,35 @@ from lib.objects import Body, System
def main():
#initialisation
m = np.array([1e5, 1e5, 0.1])
m = np.array([1, 1, 1e-5])
x1 = np.array([-1, 0, 0])
x2 = np.array([1, 0, 0])
x3 = np.array([100, 0, 0])
q = np.array([x1, x2, x3])
v1 = np.array([0, 0, 0])
v2 = np.array([0, 1, 0])
v1 = np.array([0, -0.35, 0])
v2 = np.array([0, 0.35, 0])
v3 = np.array([0, 0, 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()
E, L = frogleap(10, 0.01, dyn_syst, recover_param=True, display=True)
duration, step = 100, 0.01
E, L = frogleap(duration, step, dyn_syst, recover_param=True, display=True)
fig1 = plt.figure()
ax1 = fig1.add_subplot(111)
ax1.plot(np.arange(E.shape[0]), E, label=r"$E_m$")
ax1.plot(np.arange(E.shape[0])/duration, E, label=r"$E_m$")
ax1.legend()
fig2 = plt.figure()
ax2 = fig2.add_subplot(111)
ax2.plot(np.arange(L.shape[0]), np.sum(L**2,axis=1), label=r"$L^2$")
ax2.plot(np.arange(L.shape[0])/duration, np.sum(L**2,axis=1), label=r"$L^2$")
ax2.legend()
plt.show()
plt.show(block=True)
return 0
if __name__ == '__main__':