120 lines
3.7 KiB
Python
120 lines
3.7 KiB
Python
from time import sleep
|
|
from enum import Enum
|
|
|
|
from maneuvers.grapple import GrappleManeuver
|
|
from maneuvers.rendezvous import RendezvousManeuver
|
|
from maneuvers.approach_rcs import ApproachRCSManeuver
|
|
from maneuvers.docking import DockingManeuver
|
|
from maneuvers.comsat import ComsatManeuver
|
|
|
|
from lib import get_contract, get_rescuee_vessel, get_body, get_connexion
|
|
|
|
|
|
class MissionType(Enum):
|
|
transport = 1
|
|
orbit_rescue = 2
|
|
comsat = 3
|
|
|
|
|
|
class MissionStatus(Enum):
|
|
Backlog = 1
|
|
Assigned = 2
|
|
InProgress = 3
|
|
Done = 4
|
|
|
|
|
|
class MissionReport:
|
|
mission_status = None
|
|
new_vessel_id = None
|
|
new_missions = []
|
|
|
|
def __int__(self):
|
|
pass
|
|
|
|
|
|
class Mission:
|
|
pass
|
|
|
|
|
|
class TransportMission(Mission):
|
|
type = MissionType.transport
|
|
|
|
def __init__(self, kerbal_name, destination):
|
|
self.kerbal_name = kerbal_name
|
|
self.destination = destination
|
|
|
|
|
|
class RescueMission(Mission):
|
|
type = MissionType.orbit_rescue
|
|
|
|
def __init__(self, contract_title):
|
|
self.contract_title = contract_title
|
|
self.contract = get_contract(contract_title)
|
|
rescuee_first_name = contract_title.split()[1]
|
|
self.rescuee_vessel = get_rescuee_vessel(rescuee_first_name)
|
|
self.rescuee_vessel_name = self.rescuee_vessel.name
|
|
self.rescuee_name = self.rescuee_vessel.crew[0].name
|
|
|
|
def execute(self, mission_control):
|
|
conn = get_connexion()
|
|
sc = conn.space_center
|
|
|
|
if sc.active_vessel.name != mission_control.vessel.name:
|
|
sc.active_vessel = mission_control.vessel
|
|
|
|
if sc.target_vessel is None or sc.target_vessel.name != self.rescuee_vessel.name:
|
|
sc.target_vessel = self.rescuee_vessel
|
|
|
|
rendezvous_done = RendezvousManeuver(conn, mission_control).execute()
|
|
if rendezvous_done:
|
|
sc.active_vessel = mission_control.deliver_probe(self.rescuee_vessel)
|
|
|
|
sc.target_vessel = self.rescuee_vessel
|
|
sc.active_vessel.parts.controlling = mission_control.get_grappling()
|
|
|
|
target_grapple_done = GrappleManeuver(conn, mission_control, sc.target_vessel).start()
|
|
if target_grapple_done:
|
|
mothership_docking_done = DockingManeuver(conn,
|
|
mission_control,
|
|
mission_control.get_probe_port(),
|
|
mission_control.get_bay_port()).start()
|
|
|
|
if mothership_docking_done:
|
|
mission_control.probe = None
|
|
kerbal = mission_control.get_kerbal(self.rescuee_name)
|
|
sc.transfer_crew(kerbal, mission_control.get_part_with_free_seat())
|
|
mission_control.toggle_grappling(False)
|
|
mission_control.recover_probe()
|
|
|
|
# Destroy Capsule
|
|
report = MissionReport()
|
|
report.mission_status = MissionStatus.Done
|
|
report.new_missions.append(
|
|
TransportMission(self.rescuee_name, 'kerbin_orbit')
|
|
)
|
|
|
|
return report
|
|
|
|
|
|
class ComSatNetworkMission:
|
|
type = MissionType.comsat
|
|
|
|
def __init__(self, body_name):
|
|
self.body = get_body(body_name)
|
|
|
|
def execute(self, mission_control):
|
|
conn = get_connexion()
|
|
sc = conn.space_center
|
|
|
|
if sc.active_vessel.name != mission_control.vessel.name:
|
|
sc.active_vessel = mission_control.vessel
|
|
|
|
mission_control.toggle_engines(True)
|
|
|
|
comsat_done = ComsatManeuver(conn, mission_control, self.body).execute()
|
|
if comsat_done:
|
|
report = MissionReport()
|
|
report.mission_status = MissionStatus.Done
|
|
|
|
return report
|