Soustava.py

#import Teleso
from Teleso import Teleso
import numpy as np
import matplotlib.pyplot as plt
from typing import List
class Soustava:
    """
        Trida zahrnujici nekolik teles
    """
    def __init__(self):
        """
            Konstruktor - jen vynuluje pocet teles
        """
        self._telesa: List[Teleso]=[] # prazdny list
        self._xCent=[]
        self._yCent=[]
        self._zCent=[]
        self._t=[]
    
    def __str__(self):
        text=f"Soustava ma {len(self._telesa)} teles: "
        for teleso in self._telesa:
            text=text+teleso._jmeno+","
        return(text)

    def pridej_teleso(self,teleso:'Teleso'):
        """
            Prida teleso do soustavy teles
        """
        if (isinstance(teleso,Teleso)):
            self._telesa.append(teleso)
        else:
            print("Nejde")

    def hmotny_stred(self):
        M=0.0
        r=np.array([0,0,0])
        for teleso in self._telesa:
            r=r+teleso.pozice()*teleso.hmotnost();
            M=M+teleso.hmotnost()
        return(r/M)  # hmotny stred


    def spocitej_sily_na_telesa(self):
        """
            Spocita sily pusobici na jednotliva telesa soustavy
        """
        for teleso in self._telesa:
            F=np.array([0,0,0])  # inicializuj silu
            for teleso2 in self._telesa:
                if (teleso is teleso2):
                    #print("teleso a teleso2 jsou stejne objekty, nemohou pusobit vzajemne")
                    continue
                F=F+teleso.sila(teleso2)
            teleso.setSila(F)

    def pohni_telesy(self,t,dt):
        """
            spocita sily pusobici na telesa a pohne telesy o dany casovy krok
        """
        self.spocitej_sily_na_telesa()
        for teleso in self._telesa:
            teleso.pohni(t,dt)
        rCent=self.hmotny_stred()
        self._xCent.append(rCent[0])
        self._yCent.append(rCent[1])
        self._zCent.append(rCent[2])
        self._t.append(t)

    def zobraz_soustavu(self):
        """
            Nakresli obrazek soustavy
        """
        plt.figure(1,figsize=[20,10])
        for teleso in self._telesa:
            teleso.zobraz(plt)
        plt.plot(self._xCent,self._yCent)
        plt.axis('equal')
        plt.show()

r=np.array([0,0,0])
v=np.array([0,0,0])
zeme=Teleso("Zeme",5.972e24,r,v)

r=np.array([384400e3,0,0])
T=27.321582*24*60*60  # obezna doba mesice
s=2*np.pi*r[0]
v_mesic=s/T
v=np.array([0.0,v_mesic,0])
mesic=Teleso("Mesic",7.347e22,r,v)
print(zeme)


soustava=Soustava()
soustava.pridej_teleso(4)
soustava.pridej_teleso(zeme)
print(soustava)
soustava.pridej_teleso(mesic)
print(soustava)

soustava.spocitej_sily_na_telesa()
dt=1*60*60  # 1 hodina
t=0
while t<=5.5*T:
    t=t+dt
    soustava.pohni_telesy(t,dt)
    
r=soustava.hmotny_stred()    
soustava.zobraz_soustavu()