Add get masses/positions/momenta methods and modify frogleap input/output

2021-10-22 16:23:08 +02:00
parent e1431ce1b4
commit 3e40f412e6
3 changed files with 26 additions and 8 deletions
--- a/main.py
+++ b/main.py
@@ -3,10 +3,11 @@
 from sys import exit as sysexit
 import numpy as np
 from lib.integrator import frogleap
+import lib.objects

 def main():
    #initialisation
-    m = np.array([1e10, 1, 1])
+    m = np.array([1e10, 1, 0])

    x1 = np.array([0, 0, 0])
    x2 = np.array([1, 0, 0])