#! /bin/python3
# A script that flies a plane
import time
import krpc

# The IP of the KRPC server (if not same computer)
remote = '192.168.1.10'
# Make connection
conn = krpc.connect(name='Plane autopilot', address=remote)

# Important objects
vessel = conn.space_center.active_vessel
body = vessel.orbit.body
r_frame = body.reference_frame
surf_vel = conn.get_call(getattr, vessel.flight(r_frame), 'horizontal_speed')
srf_altitude = conn.get_call(getattr, vessel.flight(), 'surface_altitude')
vert_vel = conn.get_call(getattr, vessel.flight(r_frame), 'vertical_speed')
print(vessel.name)

vessel.control.brakes = True

vessel.auto_pilot.target_pitch_and_heading(0, 90)
vessel.auto_pilot.engage()
#vessel.auto_pilot.sas = True
vessel.control.throttle = 1

print('Launching!')
vessel.control.activate_next_stage()
time.sleep(3)
vessel.control.brakes = False


expr = conn.krpc.Expression.greater_than(
    conn.krpc.Expression.call(surf_vel),
    conn.krpc.Expression.constant_double(60))
event = conn.krpc.add_event(expr)
with event.condition:
    event.wait()

print('Tilt up')
vessel.auto_pilot.target_pitch_and_heading(20, 90)


expr = conn.krpc.Expression.greater_than(
    conn.krpc.Expression.call(srf_altitude),
    conn.krpc.Expression.constant_double(40))
event = conn.krpc.add_event(expr)
with event.condition:
    event.wait()

print('Gear up')
vessel.control.gear = False

expr = conn.krpc.Expression.greater_than(
    conn.krpc.Expression.call(surf_vel),
    conn.krpc.Expression.constant_double(200))
event = conn.krpc.add_event(expr)
with event.condition:
    event.wait()

print('glide')
vessel.auto_pilot.target_pitch_and_heading(5, 90)
vessel.control.throttle = 0.20
time.sleep(35)

print('turn')
vessel.control.roll = 0.03
vessel.auto_pilot.target_pitch_and_heading(5, 120)
time.sleep(8)
vessel.auto_pilot.target_pitch_and_heading(5, 180)
time.sleep(8)
vessel.auto_pilot.target_pitch_and_heading(5, 210)
time.sleep(8)
vessel.auto_pilot.target_pitch_and_heading(2, 270)

vessel.control.roll = 0.00
vessel.control.throttle = 0.025

while vessel.flight(r_frame).horizontal_speed > 90:
    if vessel.flight(r_frame).surface_altitude > 100:
        print('Nose down')
        vessel.auto_pilot.target_pitch_and_heading(-5, 270)
    elif vessel.flight(r_frame).surface_altitude < 75:
        print('Nose up')
        vessel.auto_pilot.target_pitch_and_heading(7, 270)
    time.sleep(1)

vessel.control.gear = True
vessel.auto_pilot.target_pitch_and_heading(3, 270)

expr = conn.krpc.Expression.less_than(
    conn.krpc.Expression.call(srf_altitude),
    conn.krpc.Expression.constant_double(30))
event = conn.krpc.add_event(expr)
with event.condition:
    event.wait()

vessel.control.brakes = True

vessel.control.throttle = 0

conn.close()