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