Tibeuleu/KozaiLidov
1
0
Fork 0
Code Activity

general debug and display angular momentum and mechanical energy curves

Browse Source
This commit is contained in:
Thibault Barnouin
2021-10-30 13:24:02 +02:00
parent a82d316cc9
commit 9a16400a89
3 changed files with 20 additions and 10 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
lib/integrator.py
View File

@@ -34,7 +34,7 @@ def frogleap(duration, step, dyn_syst, recover_param=False, display=False):
q_array = dyn_syst.get_positions()
p_array = dyn_syst.get_momenta()
E = np.zeros(N)
L = np.zeros(N)
L = np.zeros((N,3))
if display:
d = DynamicUpdate()
@@ -53,13 +53,14 @@ def frogleap(duration, step, dyn_syst, recover_param=False, display=False):
body.p = p_array[i]
if body.m != 0.:
body.v = body.p/body.m
dyn_syst.COMShift()
E[j] = dyn_syst.Eval()
L[j] = dyn_syst.Lval()
if display:
# In center of mass frame
q_cm = np.array([0.,0.])#np.sum(m_array.reshape((q_array.shape[0],1))*q_array, axis=0)/m_array.sum()
q_cm = np.sum(m_array.reshape((q_array.shape[0],1))*q_array, axis=0)/m_array.sum()
# display progression
d.on_running(q_array[:,0]-q_cm[0], q_array[:,1]-q_cm[1])
time.sleep(0.01)