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:36:28 +01:00
parent 1c20570200
commit ba7d999420
3 changed files with 4 additions and 7 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
lib/LeapFrog.py
View File

@@ -32,9 +32,6 @@ def LP(dyn_syst, dt):
Kick(dyn_syst, dt) Kick(dyn_syst, dt)
Drift(dyn_syst, dt / 2) Drift(dyn_syst, dt / 2)
dyn_syst.time = dyn_syst.time + dt dyn_syst.time = dyn_syst.time + dt
for body in dyn_syst.bodylist:
body.p = body.v * body.m
def leapfrog(dyn_syst, bin_syst, duration, dt, recover_param=False, display=False, savename=None): def leapfrog(dyn_syst, bin_syst, duration, dt, recover_param=False, display=False, savename=None):
if display: if display:

2
lib/hermite.py
View File

@@ -77,8 +77,6 @@ def HPC(dyn_syst, dt): # update position and velocities of bodies in system wit
Update_jp(dyn_syst) Update_jp(dyn_syst)
Correct(dyn_syst, dt) Correct(dyn_syst, dt)
dyn_syst.time = dyn_syst.time + dt dyn_syst.time = dyn_syst.time + dt
for body in dyn_syst.bodylist:
body.p = body.v * body.m
def hermite(dyn_syst, duration, dt, recover_param=False, display=False, savename=None): def hermite(dyn_syst, duration, dt, recover_param=False, display=False, savename=None):

6
lib/objects.py
View File

@@ -14,7 +14,6 @@ class Body:
self.m = mass self.m = mass
self.q = position self.q = position
self.v = velocity self.v = velocity
self.p = velocity * mass
self.a = np.zeros(3) self.a = np.zeros(3)
self.ap = np.zeros(3) self.ap = np.zeros(3)
self.j = np.zeros(3) self.j = np.zeros(3)
@@ -28,6 +27,10 @@ class Body:
def __str__(self): # Called upon "str(body)" def __str__(self): # Called upon "str(body)"
return r"Body of mass: {0:.1e} $M_\odot$".format(self.m/Ms) return r"Body of mass: {0:.1e} $M_\odot$".format(self.m/Ms)
@property
def p(self):
return self.v*self.m
class System(Body): class System(Body):
def __init__(self, bodylist, main = False, blackstyle=True): def __init__(self, bodylist, main = False, blackstyle=True):
@@ -121,7 +124,6 @@ class System(Body):
@property @property
def mu(self): def mu(self):
sum = 0
prod = 1 prod = 1
for body in self.bodylist: for body in self.bodylist:
prod = prod * body.m prod = prod * body.m