From 1e35a3a43ca8ce9e0702496560b4f95d1644ec40 Mon Sep 17 00:00:00 2001
From: Alex_Hubert <alex.hubert@etu.unistra.fr>
Date: Fri, 15 Oct 2021 23:16:27 +0200
Subject: [PATCH] class

---
 LeapFrog_1.0.py |  2 ++
 lib/class.py    | 64 +++++++++++++++++++++++++++++++++++++++++++++++++
 2 files changed, 66 insertions(+)
 create mode 100644 lib/class.py

diff --git a/LeapFrog_1.0.py b/LeapFrog_1.0.py
index 941b888..93665b6 100644
--- a/LeapFrog_1.0.py
+++ b/LeapFrog_1.0.py
@@ -12,6 +12,8 @@ def main():
     v2 = np.array([1, 0, 0])
     v3 = np.array([1, 0, 0])
 
+    t = 0
+    dt = 0.1
 
 
 if __name__ == '__main__':
diff --git a/lib/class.py b/lib/class.py
new file mode 100644
index 0000000..6d723ff
--- /dev/null
+++ b/lib/class.py
@@ -0,0 +1,64 @@
+import numpy as np
+
+"""
+Class definition for physical atribute
+"""
+
+
+class Body:
+
+
+    
+    def __init__(self, mass, position, velocity):
+        self.m = mass
+        self.q = position
+        self.v = velocity
+        self.p = velocity*mass
+
+
+class System:
+
+    def __init__(self, bodylist):
+        self.bodylist = bodylist
+
+    def Mass(self): #return total system mass
+        mass = 0
+        for body in self.bodylist:
+            mass = mass + body.m
+        return mass
+
+    def COM(self): #return center of mass in cartesian np_array
+        coord = np.zeros(3)
+        for body in self.bodylist:
+            coord = coord + body.m*body.q
+        coord = coord/self.Mass
+        return coord
+
+
+    #def initialize(self):
+
+# initialisation mass
+m1 = 10
+m2 = 1
+m3 = 1
+    
+# initialisation position
+q1 = np.array([0, 0, 0])
+q2 = np.array([1, 0, 0])
+q3 = np.array([2, 0, 0])
+
+# initialisation velocity
+v1 = np.array([0, 0, 0])
+v2 = np.array([1, 0, 0])
+v3 = np.array([2, 0, 0])
+
+
+star1 = Body(m1,q1,v1)
+print('test')
+star2 = Body(m2,q2,v2)
+star3 = Body(m3,q3,v3)
+
+
+
+star2 = Body
+