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
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
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
import numpy as np
from lib.integrator import frogleap
import lib.objects
from lib.objects import Body, System
def main():
#initialisation
@@ -17,9 +17,14 @@ def main():
v1 = np.array([0, 0, 0])
v2 = 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
if __name__ == '__main__':