-
Notifications
You must be signed in to change notification settings - Fork 2
/
orbit.py
86 lines (73 loc) · 2.79 KB
/
orbit.py
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
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
import time
from datetime import datetime, timedelta
import krpc
import helpers
"""
This script takes a two-stage rocket and launches it into orbit. The following
assumptions are made:
- The first stage uses solid fuel boosters
- The second stage has RCS available
- There's enough fuel available to get into orbit
- The stages are in the following order:
- Solid boosters
- Decouple & boost liquid fuel
"""
HEADING_NORTH = 0
HEADING_EAST = 90
HEADING_SOUTH = 180
HEADING_WEST = 270
def launch(connection, vessel, heading, target_altitude):
"""
Launch a given vessel into orbit at a given target altitude.
:params connection: A krpc connection
:params vessel: A vessel object
:params heading: The heading of the orbit
:params target_altitude: The target apoapsis and periapsis altitude in meters
"""
# Setup heading, control and throttle
start_time = datetime.now()
vessel.auto_pilot.engage()
vessel.auto_pilot.target_pitch_and_heading(90, heading)
vessel.control.throttle = 1
print(f"Delta-v: {helpers.get_estimated_delta_v(connection, vessel)}")
# Countdown...
print("3...")
time.sleep(1)
print("2...")
time.sleep(1)
print("1...")
time.sleep(1)
print("Launch!")
vessel.control.activate_next_stage()
# Reduce thrusters and set pitch for orbit
helpers.wait_for_altitude_more_than(connection, vessel, 3000)
vessel.control.throttle = 0.7
vessel.auto_pilot.target_pitch = 45
# Decouple external fuel tanks when empty
helpers.wait_for_fuel_less_than(connection, vessel, "SolidFuel", 0.1)
vessel.control.activate_next_stage()
# Keep boosting until we reach the target orbit altitude
helpers.wait_for_apoapsis_more_than(connection, vessel, target_altitude)
vessel.auto_pilot.target_pitch = 0
vessel.control.throttle = 0
vessel.control.rcs = True
vessel.auto_pilot.disengage()
time.sleep(1)
# There appears to be an issue when creating a multiple maneuver nodes and
# monitoring its time_to attributes. For now, we create an empty maneuver
# node and pass that into the helpers.even_orbit function.
kerbin_node = vessel.control.add_node(connection.space_center.ut)
helpers.even_orbit(connection, vessel, kerbin_node)
# In stable orbit
vessel.control.rcs = False
vessel.auto_pilot.disengage()
launch_duration = datetime.now() - start_time
print(f"Stable orbit achieved in {launch_duration}")
print(
f"Delta-v left: {helpers.get_estimated_delta_v(connection, vessel, sea_level_impulse=False)}"
)
if __name__ == "__main__":
server_ip_address = "replace_with_your_server_ip_address"
connection = krpc.connect(address=server_ip_address)
vessel = connection.space_center.active_vessel
launch(connection, vessel, HEADING_EAST, 80000)