From 928c5e8a493df6c4003f4717d5a67c2e99294075 Mon Sep 17 00:00:00 2001 From: Thibault Barnouin Date: Fri, 22 Oct 2021 16:29:44 +0200 Subject: [PATCH] debug get methods and initialisation in main --- lib/objects.py | 2 +- main.py | 11 ++++++++--- 2 files changed, 9 insertions(+), 4 deletions(-) diff --git a/lib/objects.py b/lib/objects.py index dae354f..66a2d99 100755 --- a/lib/objects.py +++ b/lib/objects.py @@ -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]) diff --git a/main.py b/main.py index 9b40db8..e5e2df3 100755 --- a/main.py +++ b/main.py @@ -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__':