96 lines
3.3 KiB
Python
96 lines
3.3 KiB
Python
import math
|
|
|
|
from maneuver_scheduler import ManeuverScheduler
|
|
|
|
from . import MechJebManeuver
|
|
|
|
|
|
class ComsatManeuver(MechJebManeuver):
|
|
def __init__(self, conn, mission_control, target_body):
|
|
super().__init__(conn, mission_control)
|
|
self.target_body = target_body
|
|
|
|
body = self.target_body
|
|
if body.satellites:
|
|
lowest_sat = min(body.satellites, key=lambda sat: sat.orbit.periapsis)
|
|
max_orbit = lowest_sat.orbit.periapsis_altitude - lowest_sat.sphere_of_influence
|
|
else:
|
|
max_orbit = body.sphere_of_influence
|
|
|
|
self.target_altitude = max_orbit - ((5 / 100) * max_orbit)
|
|
|
|
def start(self):
|
|
self.plan_next_maneuver()
|
|
|
|
def plan_next_maneuver(self):
|
|
sc = self.conn.space_center
|
|
vessel = sc.active_vessel
|
|
|
|
if vessel.orbit.body.name != self.target_body.name:
|
|
raise NotImplementedError
|
|
|
|
if not math.isclose(vessel.orbit.apoapsis_altitude, self.target_altitude, rel_tol=.01):
|
|
SetOrbitApoapsis(self.conn, self.mission_control, self.target_body).prepare_maneuver()
|
|
elif not math.isclose(vessel.orbit.eccentricity, 0, abs_tol=.001) or self.vessel.control.current_stage > 1:
|
|
CircularizeOrbitAndDeliver(self.conn, self.mission_control, self.target_body).prepare_maneuver()
|
|
elif self.vessel.control.current_stage <= 1:
|
|
return True
|
|
|
|
return False
|
|
|
|
|
|
class SetOrbitApoapsis(ComsatManeuver):
|
|
name = "Set target orbit's apoapsis"
|
|
|
|
def prepare_maneuver(self):
|
|
oa = self.maneuver_planner.operation_apoapsis
|
|
oa.new_apoapsis = self.target_altitude
|
|
oa.time_selector.time_reference = self.mech_jeb.TimeReference.periapsis
|
|
nodes = oa.make_nodes()
|
|
|
|
node = nodes[0]
|
|
|
|
ManeuverScheduler.book_timeslot_for_node(self.vessel, node, self)
|
|
|
|
|
|
class CircularizeOrbitAndDeliver(ComsatManeuver):
|
|
name = "Circularize orbit and deliver comsat"
|
|
|
|
def prepare_maneuver(self):
|
|
oc = self.maneuver_planner.operation_circularize
|
|
oc.time_selector.time_reference = self.mech_jeb.TimeReference.apoapsis
|
|
nodes = oc.make_nodes()
|
|
|
|
node = nodes[0]
|
|
|
|
ManeuverScheduler.book_timeslot_for_node(self.vessel, node, self)
|
|
|
|
def execute(self) -> bool:
|
|
sc = self.conn.space_center
|
|
if sc.active_vessel.name != self.vessel.name:
|
|
sc.active_vessel = self.vessel
|
|
|
|
while self.vessel.control.nodes:
|
|
self._execute_node()
|
|
|
|
current_stage = self.vessel.control.current_stage
|
|
if current_stage > 1:
|
|
relay = self.vessel.control.activate_next_stage()
|
|
sc.active_vessel = relay[0]
|
|
sc.active_vessel.name = self.vessel.name + " " + current_stage
|
|
sc.active_vessel.control.solar_panels = True
|
|
sc.active_vessel = self.vessel
|
|
|
|
oro = self.maneuver_planner.operation_resonant_orbit
|
|
oro.resonance_numerator = 2
|
|
oro.resonance_denominator = 3
|
|
oro.time_selector.lead_time = 10
|
|
oro.time_selector.time_reference = self.mech_jeb.TimeReference.x_from_now
|
|
oro.make_nodes()
|
|
|
|
while self.vessel.control.nodes:
|
|
self._execute_node()
|
|
return self.plan_next_maneuver()
|
|
else:
|
|
return True
|