Tibeuleu/KozaiLidov
1
0
Fork 0
Code Activity

modify leapfrog to take velocities instead of momenta, add new initial conditions

Browse Source
This commit is contained in:
Thibault Barnouin
2021-11-05 17:12:16 +01:00
parent d5f948f4a1
commit fff1832e61
5 changed files with 51 additions and 41 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
lib/objects.py
View File

@@ -34,6 +34,9 @@ class System:
def get_positions(self): #return the positions of the bodies
return np.array([body.q for body in self.bodylist])
def get_velocities(self): #return the positions of the bodies
return np.array([body.v for body in self.bodylist])
def get_momenta(self): #return the momenta of the bodies
return np.array([body.p for body in self.bodylist])