blob: 372e42164fc48f2b691c7e5b62cc8b64ee277610 (
plain) (
blame)
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
40
41
42
43
44
45
46
47
48
49
50
51
|
from stage import Stage
class Rocket():
def __init__(self, name: str, stages: [type[Stage]], payload_mass: int):
self.name = name
self.stages = stages
self.stages_spent = 0
self.payload_mass = payload_mass
def current_stage(self) -> type[Stage]:
return self.stages[0]
def top_stage(self) -> type[Stage]:
return self.stages[len(self.stages) - 1] #TODO: drag coef and cross sectional area of top stage
def perform_stage_separation(self, engines_on: bool):
if len(self.stages) > 1:
self.stages.pop(0)
self.stages_spent += 1
self.current_stage().engines_on = engines_on
def total_mass(self):
total_mass = self.payload_mass
for stage in self.stages:
total_mass += stage.total_mass()
return total_mass
def total_fuel(self):
fuel_mass = 0
for stage in self.stages:
fuel_mass += stage.fuel_mass
return fuel_mass
#maybe we should average this with the rest of the stages?
def rocket_x_drag_coefficient(self):
return self.top_stage().x_drag_coefficient
def rocket_x_cross_sectional_area(self):
total_cross_sectional_area = 0
for stage in self.stages:
total_cross_sectional_area += stage.x_cross_sectional_area
return total_cross_sectional_area
def rocket_y_drag_coefficient(self):
return self.top_stage().y_drag_coefficient
def rocket_y_cross_sectional_area(self):
return self.top_stage().y_cross_sectional_area
def __str__(self):
return "eue"
|
