class

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
+