Tibeuleu/KozaiLidov
1
0
Fork 0
Code Activity

minor modification + system clas update comshift

Browse Source
This commit is contained in:
Alex_Hubert
2021-11-19 16:39:52 +01:00
parent ba7d999420
commit 6da8a27606
1 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
lib/objects.py
View File

@@ -93,14 +93,14 @@ class System(Body):
def COMV(self): #return center of mass velocity in cartesian np_array def COMV(self): #return center of mass velocity in cartesian np_array
coord = np.zeros(3) coord = np.zeros(3)
for body in self.bodylist: for body in self.bodylist:
coord = coord + body.p coord = coord + body.v
coord = coord/self.M coord = coord/self.M
return coord return coord
def COMShift(self): #Shift coordinates of bodies in system to COM frame and set COM at rest def COMShift(self): #Shift coordinates of bodies in system to COM frame and set COM at rest
for body in self.bodylist: for body in self.bodylist:
body.q = body.q - self.COM body.q = body.q - self.COM
body.p = body.p - self.COMV body.v = body.v - self.COMV
@property @property
def L(self): #return angular momentum of bodies in system def L(self): #return angular momentum of bodies in system