matcomp-entrega/Corpo_Celeste.py at main · joao27lex/matcomp-entrega · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
class Corpo_Celeste:
    def __init__(
        self,
        massa,
        raio,
        color,
        name,
        pos_x,
        pos_y,
        vel_x=0.0,
        vel_y=0.0,
    ):
        self.massa = massa
        self.raio = raio
        self.color = color
        self.name = name
        self.pos_x = float(pos_x)
        self.pos_y = float(pos_y)
        self.vel_x = float(vel_x)
        self.vel_y = float(vel_y)
        self.trace = []

    # essa funcao é pro equacoes_movimento
    # x, y, vx, vy = estado
    def return_estado(self):
        info = [self.pos_x, self.pos_y, self.vel_x, self.vel_y]
        return info

    def return_pos(self):
        info = [self.pos_x, self.pos_y]
        return info

    def return_vel(self):
        info = [self.vel_x, self.vel_y]
        return info

    def return_name(self):
        name = self.name
        return name