Tibeuleu/KozaiLidov
1
0
Fork 0
Code Activity

debug get methods and initialisation in main

Browse Source
This commit is contained in:
Thibault Barnouin
2021-10-22 16:29:44 +02:00
parent 3e40f412e6
commit 928c5e8a49
2 changed files with 9 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
lib/objects.py
View File

@@ -26,7 +26,7 @@ class System:
self.bodylist = bodylist self.bodylist = bodylist
def get_masses(self): #return the masses of each object def get_masses(self): #return the masses of each object
return np.array([body.m for body in self.bodylit]) return np.array([body.m for body in self.bodylist])
def get_positions(self): #return the positions of the bodies def get_positions(self): #return the positions of the bodies
return np.array([body.q for body in self.bodylist]) return np.array([body.q for body in self.bodylist])

11
main.py
View File

@@ -3,7 +3,7 @@
from sys import exit as sysexit from sys import exit as sysexit
import numpy as np import numpy as np
from lib.integrator import frogleap from lib.integrator import frogleap
import lib.objects from lib.objects import Body, System
def main(): def main():
#initialisation #initialisation
@@ -17,9 +17,14 @@ def main():
v1 = np.array([0, 0, 0]) v1 = np.array([0, 0, 0])
v2 = np.array([0, 0, 0]) v2 = np.array([0, 0, 0])
v3 = np.array([0, 0, 0]) v3 = np.array([0, 0, 0])
p = m*np.array([v1, v2, v3]) v = np.array([v1, v2, v3])
q, p = frogleap(10, 0.01, m, q, p, display=True) bodylist = []
for i in range(3):
bodylist.append(Body(m[i], q[i], v[i]))
dyn_syst = System(bodylist)
new_dyn_syst = frogleap(10, 0.01, dyn_syst, display=True)
return 0 return 0
if __name__ == '__main__': if __name__ == '__main__':