'''
Fly ArduPlane QuadPlane in SITL
AP_FLAKE8_CLEAN
'''
import os
import numpy
import math
import copy
from pymavlink import mavutil
from pymavlink.rotmat import Vector3
import vehicle_test_suite
from vehicle_test_suite import Test
from vehicle_test_suite import AutoTestTimeoutException, NotAchievedException, PreconditionFailedException
import operator
testdir = os.path.dirname(os.path.realpath(__file__))
WIND = "0,180,0.2"
SITL_START_LOCATION = mavutil.location(-27.274439, 151.290064, 343, 8.7)
class AutoTestQuadPlane(vehicle_test_suite.TestSuite):
@staticmethod
def get_not_armable_mode_list():
return []
@staticmethod
def get_not_disarmed_settable_modes_list():
return []
@staticmethod
def get_no_position_not_settable_modes_list():
return []
@staticmethod
def get_position_armable_modes_list():
return []
@staticmethod
def get_normal_armable_modes_list():
return []
def vehicleinfo_key(self):
return 'ArduPlane'
def default_frame(self):
return "quadplane"
def test_filepath(self):
return os.path.realpath(__file__)
def sitl_start_location(self):
return SITL_START_LOCATION
def default_speedup(self):
'''QuadPlane seems to be race-free'''
return 100
def log_name(self):
return "QuadPlane"
def set_current_test_name(self, name):
self.current_test_name_directory = "ArduPlane_Tests/" + name + "/"
def apply_defaultfile_parameters(self):
pass
def defaults_filepath(self):
return self.model_defaults_filepath(self.frame)
def is_plane(self):
return True
def get_stick_arming_channel(self):
return int(self.get_parameter("RCMAP_YAW"))
def get_disarm_delay(self):
return int(self.get_parameter("LAND_DISARMDELAY"))
def set_autodisarm_delay(self, delay):
self.set_parameter("LAND_DISARMDELAY", delay)
def AirMode(self):
"""Check that plane.air_mode turns on and off as required"""
self.progress("########## Testing AirMode operation")
self.set_parameter("AHRS_EKF_TYPE", 10)
self.change_mode('QSTABILIZE')
self.wait_ready_to_arm()
"""
SPIN_ARM and SPIN_MIN default to 0.10 and 0.15
when armed with zero throttle in AirMode, motor PWM should be at SPIN_MIN
If AirMode is off, motor PWM will drop to SPIN_ARM
"""
self.progress("Verify that SERVO5 is Motor1 (default)")
motor1_servo_function_lp = 33
if (self.get_parameter('SERVO5_FUNCTION') != motor1_servo_function_lp):
raise PreconditionFailedException("SERVO5_FUNCTION not %d" % motor1_servo_function_lp)
self.progress("Verify that flightmode channel is 5 (default)")
default_fltmode_ch = 5
if (self.get_parameter("FLTMODE_CH") != default_fltmode_ch):
raise PreconditionFailedException("FLTMODE_CH not %d" % default_fltmode_ch)
"""When disarmed, motor PWM will drop to min_pwm"""
min_pwm = self.get_parameter("Q_M_PWM_MIN")
self.progress("Verify Motor1 is at min_pwm when disarmed")
self.wait_servo_channel_value(5, min_pwm, comparator=operator.eq)
armdisarm_option = 154
arm_ch = 8
self.set_parameter("RC%d_OPTION" % arm_ch, armdisarm_option)
self.progress("Configured RC%d as ARMDISARM switch" % arm_ch)
"""arm with GCS, record Motor1 SPIN_ARM PWM output and disarm"""
spool_delay = self.get_parameter("Q_M_SPOOL_TIME") + 0.25
self.zero_throttle()
self.arm_vehicle()
self.progress("Waiting for Motor1 to spool up to SPIN_ARM")
self.delay_sim_time(spool_delay)
spin_arm_pwm = self.wait_servo_channel_value(5, min_pwm, comparator=operator.gt)
self.progress("spin_arm_pwm: %d" % spin_arm_pwm)
self.disarm_vehicle()
"""arm with switch, record Motor1 SPIN_MIN PWM output and disarm"""
self.set_rc(8, 2000)
self.delay_sim_time(spool_delay)
self.progress("Waiting for Motor1 to spool up to SPIN_MIN")
spin_min_pwm = self.wait_servo_channel_value(5, spin_arm_pwm, comparator=operator.gt)
self.progress("spin_min_pwm: %d" % spin_min_pwm)
self.set_rc(8, 1000)
if (spin_arm_pwm >= spin_min_pwm):
raise PreconditionFailedException("SPIN_MIN pwm not greater than SPIN_ARM pwm")
self.start_subtest("Test auxswitch arming with AirMode Switch")
for mode in ('QSTABILIZE', 'QACRO'):
"""verify that arming with switch results in higher PWM output"""
self.progress("Testing %s mode" % mode)
self.change_mode(mode)
self.zero_throttle()
self.progress("Arming with switch at zero throttle")
self.arm_motors_with_switch(arm_ch)
self.progress("Waiting for Motor1 to speed up")
self.wait_servo_channel_value(5, spin_min_pwm, comparator=operator.ge)
self.progress("Verify that rudder disarm is disabled")
try:
self.disarm_motors_with_rc_input()
except NotAchievedException:
pass
if not self.armed():
raise NotAchievedException("Rudder disarm not disabled")
self.progress("Disarming with switch")
self.disarm_motors_with_switch(arm_ch)
self.progress("Waiting for Motor1 to stop")
self.wait_servo_channel_value(5, min_pwm, comparator=operator.le)
self.wait_ready_to_arm()
self.start_subtest("Verify that arming with switch does not spin motors in other modes")
arming_magthresh = self.get_parameter("ARMING_MAGTHRESH")
self.set_parameter("ARMING_MAGTHRESH", 0)
ahrs_trim_x = self.get_parameter("AHRS_TRIM_X")
self.set_parameter("AHRS_TRIM_X", math.radians(-60))
self.wait_roll(60, 1)
for mode in (
'ACRO',
'AUTOTUNE',
'AVOID_ADSB',
'CIRCLE',
'CRUISE',
'FBWA',
'FBWB',
'GUIDED',
'LOITER',
'QHOVER',
'QLOITER',
'STABILIZE',
'TRAINING',
):
self.progress("Testing %s mode" % mode)
self.change_mode(mode)
self.zero_throttle()
self.progress("Arming with switch at zero throttle")
self.arm_motors_with_switch(arm_ch)
self.progress("Waiting for Motor1 to (not) speed up")
self.delay_sim_time(spool_delay)
self.wait_servo_channel_value(5, spin_arm_pwm, comparator=operator.le)
self.wait_servo_channel_value(6, spin_arm_pwm, comparator=operator.le)
self.wait_servo_channel_value(7, spin_arm_pwm, comparator=operator.le)
self.wait_servo_channel_value(8, spin_arm_pwm, comparator=operator.le)
self.progress("Disarming with switch")
self.disarm_motors_with_switch(arm_ch)
self.progress("Waiting for Motor1 to stop")
self.wait_servo_channel_value(5, min_pwm, comparator=operator.le)
self.wait_ready_to_arm()
self.set_parameter("AHRS_TRIM_X", ahrs_trim_x)
self.set_parameter("ARMING_MAGTHRESH", arming_magthresh)
self.start_subtest("verify that AIRMODE auxswitch turns airmode on/off while armed")
"""set RC7_OPTION to AIRMODE"""
option_airmode = 84
self.set_parameter("RC7_OPTION", option_airmode)
for mode in ('QSTABILIZE', 'QACRO'):
self.progress("Testing %s mode" % mode)
self.change_mode(mode)
self.zero_throttle()
self.progress("Arming with GCS at zero throttle")
self.arm_vehicle()
self.progress("Turn airmode on with auxswitch")
self.set_rc(7, 2000)
self.progress("Waiting for Motor1 to speed up")
self.wait_servo_channel_value(5, spin_min_pwm, comparator=operator.ge)
self.progress("Turn airmode off with auxswitch")
self.set_rc(7, 1000)
self.progress("Waiting for Motor1 to slow down")
self.wait_servo_channel_value(5, spin_arm_pwm, comparator=operator.le)
self.disarm_vehicle()
self.wait_ready_to_arm()
self.start_subtest("Test GCS arming")
for mode in ('QSTABILIZE', 'QACRO'):
self.progress("Testing %s mode" % mode)
self.change_mode(mode)
self.zero_throttle()
self.progress("Arming with GCS at zero throttle")
self.arm_vehicle()
self.progress("Turn airmode on with auxswitch")
self.set_rc(7, 2000)
self.progress("Waiting for Motor1 to speed up")
self.wait_servo_channel_value(5, spin_min_pwm, comparator=operator.ge)
self.disarm_vehicle_expect_fail()
self.arm_vehicle()
self.progress("Verify that airmode is still on")
self.wait_servo_channel_value(5, spin_min_pwm, comparator=operator.ge)
self.disarm_vehicle(force=True)
self.wait_ready_to_arm()
def TestMotorMask(self):
"""Check operation of output_motor_mask"""
"""copter tailsitters will add condition: or (int(self.get_parameter('Q_TAILSIT_MOTMX')) & 1)"""
if not (int(self.get_parameter('Q_TILT_MASK')) & 1):
self.progress("output_motor_mask not in use")
return
self.progress("Testing output_motor_mask")
self.wait_ready_to_arm()
"""Default channel for Motor1 is 5"""
self.progress('Assert that SERVO5 is Motor1')
assert 33 == self.get_parameter('SERVO5_FUNCTION')
modes = ('MANUAL', 'FBWA', 'QHOVER')
for mode in modes:
self.progress("Testing %s mode" % mode)
self.change_mode(mode)
self.arm_vehicle()
self.progress("Raising throttle")
self.set_rc(3, 1800)
self.progress("Waiting for Motor1 to start")
self.wait_servo_channel_value(5, 1100, comparator=operator.gt)
self.set_rc(3, 1000)
self.disarm_vehicle()
self.wait_ready_to_arm()
def fly_mission(self, filename, fence=None, height_accuracy=-1):
"""Fly a mission from a file."""
self.progress("Flying mission %s" % filename)
num_wp = self.load_mission(filename)
if self.mavproxy is not None:
self.mavproxy.send('wp list\n')
if fence is not None:
self.load_fence(fence)
if self.mavproxy is not None:
self.mavproxy.send('fence list\n')
self.change_mode('AUTO')
self.wait_ready_to_arm()
self.arm_vehicle()
self.wait_waypoint(1, num_wp-1)
self.wait_disarmed(timeout=120)
def EXTENDED_SYS_STATE_SLT(self):
self.set_message_rate_hz(mavutil.mavlink.MAVLINK_MSG_ID_EXTENDED_SYS_STATE, 10)
self.change_mode("QHOVER")
self.assert_extended_sys_state(mavutil.mavlink.MAV_VTOL_STATE_MC,
mavutil.mavlink.MAV_LANDED_STATE_ON_GROUND)
self.change_mode("FBWA")
self.assert_extended_sys_state(mavutil.mavlink.MAV_VTOL_STATE_FW,
mavutil.mavlink.MAV_LANDED_STATE_ON_GROUND)
self.change_mode("QHOVER")
self.wait_ready_to_arm()
self.arm_vehicle()
self.assert_extended_sys_state(mavutil.mavlink.MAV_VTOL_STATE_MC,
mavutil.mavlink.MAV_LANDED_STATE_ON_GROUND)
self.change_mode("MANUAL")
self.assert_extended_sys_state(mavutil.mavlink.MAV_VTOL_STATE_FW,
mavutil.mavlink.MAV_LANDED_STATE_ON_GROUND)
self.change_mode("QHOVER")
self.progress("Taking off")
self.set_rc(3, 1750)
self.wait_altitude(1, 5, relative=True)
self.assert_extended_sys_state(mavutil.mavlink.MAV_VTOL_STATE_MC,
mavutil.mavlink.MAV_LANDED_STATE_IN_AIR)
self.wait_altitude(10, 15, relative=True)
self.progress("Transitioning to fixed wing")
self.change_mode("FBWA")
self.set_rc(3, 1900)
self.wait_extended_sys_state(mavutil.mavlink.MAV_VTOL_STATE_TRANSITION_TO_FW,
mavutil.mavlink.MAV_LANDED_STATE_IN_AIR)
self.wait_extended_sys_state(mavutil.mavlink.MAV_VTOL_STATE_FW,
mavutil.mavlink.MAV_LANDED_STATE_IN_AIR)
self.progress("Transitioning to multicopter")
self.set_rc(3, 1500)
self.change_mode("QHOVER")
self.wait_extended_sys_state(mavutil.mavlink.MAV_VTOL_STATE_MC,
mavutil.mavlink.MAV_LANDED_STATE_IN_AIR)
self.change_mode("QLAND")
self.wait_altitude(0, 2, relative=True, timeout=60)
self.wait_extended_sys_state(mavutil.mavlink.MAV_VTOL_STATE_MC,
mavutil.mavlink.MAV_LANDED_STATE_ON_GROUND,
timeout=30)
self.mav.motors_disarmed_wait()
def EXTENDED_SYS_STATE(self):
'''Check extended sys state works'''
self.EXTENDED_SYS_STATE_SLT()
def QAUTOTUNE(self):
'''test Plane QAutoTune mode'''
self.set_parameters({
"Q_AUTOTUNE_AGGR": 0.1,
"Q_AUTOTUNE_MIN_D": 0.0004,
"Q_A_RAT_RLL_P" : 0.15,
"Q_A_RAT_RLL_I" : 0.25,
"Q_A_RAT_RLL_D" : 0.002,
"Q_A_RAT_PIT_P" : 0.15,
"Q_A_RAT_PIT_I" : 0.25,
"Q_A_RAT_PIT_D" : 0.002,
"Q_A_RAT_YAW_P" : 0.18,
"Q_A_RAT_YAW_I" : 0.018,
"Q_A_ANG_RLL_P" : 4.5,
"Q_A_ANG_PIT_P" : 4.5,
})
parameter_values = self.get_parameters([
"Q_A_RAT_RLL_P",
"Q_A_RAT_RLL_I",
"Q_A_RAT_RLL_D",
"Q_A_ANG_RLL_P",
"Q_A_ACCEL_R_MAX",
"Q_A_RAT_PIT_P",
"Q_A_RAT_PIT_I",
"Q_A_RAT_PIT_D",
"Q_A_ANG_PIT_P",
"Q_A_ACCEL_P_MAX",
"Q_A_RAT_YAW_P",
"Q_A_RAT_YAW_I",
"Q_A_RAT_YAW_FLTE",
"Q_A_ANG_YAW_P",
"Q_A_ACCEL_Y_MAX",
])
self.set_parameters(parameter_values)
self.takeoff(15, mode='GUIDED')
self.set_rc(3, 1500)
self.change_mode("QLOITER")
tstart = self.get_sim_time()
self.context_collect('STATUSTEXT')
self.change_mode("QAUTOTUNE")
self.wait_text(
"AutoTune: (Success|Failed to level).*",
timeout=5000,
check_context=True,
regex=True,
)
if self.re_match.group(1) != "Success":
raise NotAchievedException("autotune did not succeed")
now = self.get_sim_time()
self.progress("AUTOTUNE OK (%u seconds)" % (now - tstart))
self.context_clear_collection('STATUSTEXT')
self.progress("Landing to save gains")
self.set_rc(3, 1200)
self.wait_speed_vector(
Vector3(float('nan'), float('nan'), 1.4),
timeout=5,
)
self.wait_speed_vector(
Vector3(0.0, 0.0, 0.0),
timeout=20,
)
distance = self.distance_to_home()
if distance > 20:
raise NotAchievedException("wandered from home (distance=%f)" %
(distance,))
self.set_rc(3, 1000)
tstart = self.get_sim_time()
while True:
now = self.get_sim_time_cached()
if now - tstart > 500:
raise NotAchievedException("Did not get success message")
self.send_mavlink_disarm_command()
try:
self.wait_text(
"AutoTune: Saved gains for Roll Pitch Yaw.*",
timeout=0.5,
check_context=True,
regex=True,
)
except AutoTestTimeoutException:
continue
break
self.wait_disarmed()
self.reboot_sitl()
def takeoff(self, height, mode, timeout=30):
"""climb to specified height and set throttle to 1500"""
self.set_current_waypoint(0, check_afterwards=False)
self.change_mode(mode)
self.wait_ready_to_arm()
self.arm_vehicle()
if mode == 'GUIDED':
self.user_takeoff(alt_min=height, timeout=timeout)
return
self.set_rc(3, 1800)
self.wait_altitude(height,
height+5,
relative=True,
timeout=timeout)
self.set_rc(3, 1500)
def do_RTL(self):
self.change_mode("QRTL")
self.wait_altitude(-5, 1, relative=True, timeout=60)
self.wait_disarmed()
self.zero_throttle()
def fly_home_land_and_disarm(self, timeout=30):
self.context_push()
self.change_mode('LOITER')
self.set_parameter('RTL_AUTOLAND', 2)
filename = "QuadPlaneDalbyRTL.txt"
self.progress("Using %s to fly home" % filename)
self.load_generic_mission(filename)
self.send_cmd_do_set_mode("RTL")
self.wait_mode('AUTO')
self.wait_current_waypoint(4)
self.wait_statustext('Land descend started')
self.wait_statustext('Land final started', timeout=60)
self.wait_disarmed(timeout=timeout)
self.clear_mission(mavutil.mavlink.MAV_MISSION_TYPE_MISSION)
self.set_current_waypoint(0, check_afterwards=False)
self.change_mode('MANUAL')
self.context_pop()
def wait_level_flight(self, accuracy=5, timeout=30):
"""Wait for level flight."""
tstart = self.get_sim_time()
self.progress("Waiting for level flight")
self.set_rc(1, 1500)
self.set_rc(2, 1500)
self.set_rc(4, 1500)
while self.get_sim_time_cached() < tstart + timeout:
m = self.assert_receive_message('ATTITUDE')
roll = math.degrees(m.roll)
pitch = math.degrees(m.pitch)
self.progress("Roll=%.1f Pitch=%.1f" % (roll, pitch))
if math.fabs(roll) <= accuracy and math.fabs(pitch) <= accuracy:
self.progress("Attained level flight")
return
raise NotAchievedException("Failed to attain level flight")
def fly_left_circuit(self):
"""Fly a left circuit, 200m on a side."""
self.mavproxy.send('switch 4\n')
self.change_mode('FBWA')
self.set_rc(3, 1700)
self.wait_level_flight()
self.progress("Flying left circuit")
for i in range(0, 4):
self.progress("Starting turn %u" % i)
self.set_rc(1, 1000)
self.wait_heading(270 - (90*i), accuracy=10)
self.set_rc(1, 1500)
self.progress("Starting leg %u" % i)
self.wait_distance(100, accuracy=20)
self.progress("Circuit complete")
self.change_mode('QHOVER')
self.set_rc(3, 1100)
self.wait_altitude(10, 15,
relative=True,
timeout=60)
self.set_rc(3, 1500)
def hover_and_check_matched_frequency(self, dblevel=-15, minhz=200, maxhz=300, fftLength=32, peakhz=None):
self.takeoff(10, mode="QHOVER")
hover_time = 15
tstart = self.get_sim_time()
self.progress("Hovering for %u seconds" % hover_time)
while self.get_sim_time_cached() < tstart + hover_time:
self.assert_receive_message('ATTITUDE')
vfr_hud = self.assert_receive_message('VFR_HUD')
tend = self.get_sim_time()
self.do_RTL()
psd = self.mavfft_fttd(1, 0, tstart * 1.0e6, tend * 1.0e6)
scale = 1000. / 1024.
sminhz = int(minhz * scale)
smaxhz = int(maxhz * scale)
freq = psd["F"][numpy.argmax(psd["X"][sminhz:smaxhz]) + sminhz]
peakdb = numpy.amax(psd["X"][sminhz:smaxhz])
if peakdb < dblevel or (peakhz is not None and abs(freq - peakhz) / peakhz > 0.05):
raise NotAchievedException("No motor peak, found %fHz at %fdB" % (freq, peakdb))
else:
self.progress("motor peak %fHz, thr %f%%, %fdB" % (freq, vfr_hud.throttle, peakdb))
mlog = self.dfreader_for_current_onboard_log()
freqDelta = 1000. / fftLength
pkAvg = freq
freqs = []
while True:
m = mlog.recv_match(
type='FTN1',
blocking=True,
condition="FTN1.TimeUS>%u and FTN1.TimeUS<%u" % (tstart * 1.0e6, tend * 1.0e6))
if m is None:
break
freqs.append(m.PkAvg)
pkAvg = numpy.median(numpy.asarray(freqs))
if abs(pkAvg - freq) > freqDelta:
raise NotAchievedException("FFT did not detect a motor peak at %f, found %f, wanted %f" % (dblevel, pkAvg, freq))
return freq
def GyroFFT(self):
"""Use dynamic harmonic notch to control motor noise."""
self.progress("Flying with gyro FFT - Gyro sample rate")
self.set_parameters({
"AHRS_EKF_TYPE": 10,
"INS_LOG_BAT_MASK": 3,
"INS_LOG_BAT_OPT": 0,
"INS_GYRO_FILTER": 100,
"LOG_BITMASK": 45054,
"LOG_DISARMED": 0,
"SIM_DRIFT_SPEED": 0,
"SIM_DRIFT_TIME": 0,
"SIM_GYR1_RND": 20,
"FFT_ENABLE": 1,
"FFT_MINHZ": 80,
"FFT_MAXHZ": 350,
"FFT_SNR_REF": 10,
"FFT_WINDOW_SIZE": 128,
"FFT_WINDOW_OLAP": 0.75,
})
self.set_parameters({
"SIM_VIB_FREQ_X": 250,
"SIM_VIB_FREQ_Y": 250,
"SIM_VIB_FREQ_Z": 250,
})
self.reboot_sitl()
self.hover_and_check_matched_frequency(-15, 100, 350, 128, 250)
self.set_parameters({
"SIM_VIB_MOT_MAX": 350,
"FFT_WINDOW_SIZE": 32,
"FFT_WINDOW_OLAP": 0.5,
})
self.reboot_sitl()
freq = self.hover_and_check_matched_frequency(-15, 200, 300, 32)
self.set_parameters({
"INS_LOG_BAT_OPT": 2,
"INS_HNTCH_ENABLE": 1,
"INS_HNTCH_FREQ": freq,
"INS_HNTCH_REF": 1.0,
"INS_HNTCH_ATT": 50,
"INS_HNTCH_BW": freq/2,
"INS_HNTCH_MODE": 4,
})
self.reboot_sitl()
self.takeoff(10, mode="QHOVER")
hover_time = 15
ignore_bins = 20
self.progress("Hovering for %u seconds" % hover_time)
tstart = self.get_sim_time()
while self.get_sim_time_cached() < tstart + hover_time:
self.assert_receive_message('ATTITUDE')
tend = self.get_sim_time()
self.do_RTL()
psd = self.mavfft_fttd(1, 0, tstart * 1.0e6, tend * 1.0e6)
freq = psd["F"][numpy.argmax(psd["X"][ignore_bins:]) + ignore_bins]
peakdB = numpy.amax(psd["X"][ignore_bins:])
if peakdB < -10:
self.progress("No motor peak, %f at %f dB" % (freq, peakdB))
else:
raise NotAchievedException("Detected peak at %f Hz of %.2f dB" % (freq, peakdB))
self.progress("Flying with gyro FFT - vtol to plane")
self.load_mission("quadplane-gyro-mission.txt")
if self.mavproxy is not None:
self.mavproxy.send('wp list\n')
self.change_mode('AUTO')
self.wait_ready_to_arm()
self.arm_vehicle()
self.wait_waypoint(1, 7, max_dist=60, timeout=1200)
self.wait_disarmed(timeout=120)
self.set_parameter("FFT_ENABLE", 0)
self.reboot_sitl()
def PIDTuning(self):
'''Test PID Tuning'''
self.change_mode("FBWA")
super(AutoTestQuadPlane, self).PIDTuning()
def ParameterChecks(self):
'''basic parameter checks'''
self.test_parameter_checks_poscontrol("Q_P")
self.context_push()
self.set_parameters({
"Q_RTL_MODE": 1,
"RTL_AUTOLAND": 2,
})
self.assert_prearm_failure("unset one of RTL_AUTOLAND or Q_RTL_MODE")
self.context_pop()
self.wait_ready_to_arm()
def rc_defaults(self):
ret = super(AutoTestQuadPlane, self).rc_defaults()
ret[3] = 1000
return ret
def default_mode(self):
return "MANUAL"
def disabled_tests(self):
return {
"FRSkyPassThrough": "Currently failing",
"CPUFailsafe": "servo channel values not scaled like ArduPlane",
"GyroFFT": "flapping test",
"ConfigErrorLoop": "failing because RC values not settable",
}
def BootInAUTO(self):
'''Test behaviour when booting in auto'''
self.load_mission("mission.txt")
self.set_parameters({
})
self.set_rc(5, 1000)
self.wait_mode('AUTO')
self.reboot_sitl()
self.wait_ready_to_arm()
self.delay_sim_time(20)
self.assert_current_waypoint(1)
self.arm_vehicle()
self.wait_altitude(9, 11, relative=True)
distance = self.distance_to_home()
if distance > 2:
raise NotAchievedException("wandered from home (distance=%f)" %
(distance,))
self.change_mode('QLAND')
self.wait_disarmed(timeout=60)
def PilotYaw(self):
'''Test pilot yaw in various modes'''
self.takeoff(10, mode="QLOITER")
self.set_parameter("STICK_MIXING", 0)
self.set_rc(4, 1700)
for mode in "QLOITER", "QHOVER":
self.wait_heading(45)
self.wait_heading(90)
self.wait_heading(180)
self.wait_heading(275)
self.set_rc(4, 1500)
self.do_RTL()
def FwdThrInVTOL(self):
'''test use of fwd motor throttle into wind'''
self.set_parameters({"SIM_WIND_SPD": 25,
"SIM_WIND_DIR": 360,
"Q_WVANE_ENABLE": 1,
"Q_WVANE_GAIN": 1,
"STICK_MIXING": 0,
"Q_FWD_THR_USE": 2})
self.takeoff(10, mode="QLOITER")
self.set_rc(2, 1000)
self.delay_sim_time(10)
fwd_thr_pwm = self.get_servo_channel_value(3)
if fwd_thr_pwm < 1150 :
raise NotAchievedException("fwd motor pwm command low, want >= 1150 got %f" % (fwd_thr_pwm))
m = self.assert_receive_message('ATTITUDE')
pitch = math.degrees(m.pitch)
if abs(pitch + 3.0) > 0.5 :
raise NotAchievedException("pitch should be -3.0 +- 0.5 deg, got %f" % (pitch))
self.set_rc(2, 1500)
self.delay_sim_time(5)
loc1 = self.mav.location()
self.set_parameter("SIM_ENGINE_FAIL", 1 << 2)
self.delay_sim_time(20)
self.change_mode('QLAND')
self.wait_disarmed(timeout=60)
loc2 = self.mav.location()
position_drift = self.get_distance(loc1, loc2)
if position_drift > 5.0 :
raise NotAchievedException("position drift high, want < 5.0 m got %f m" % (position_drift))
def Weathervane(self):
'''test nose-into-wind functionality'''
self.set_parameters({"SIM_WIND_SPD": 10,
"SIM_WIND_DIR": 240,
"Q_WVANE_ENABLE": 3,
"Q_WVANE_GAIN": 3,
"STICK_MIXING": 0})
self.takeoff(10, mode="QLOITER")
self.set_rc(4, 1700)
self.wait_heading(90)
self.set_rc(4, 1500)
self.wait_heading(150, accuracy=5, timeout=180)
self.do_RTL()
def CPUFailsafe(self):
'''In lockup Plane should copy RC inputs to RC outputs'''
self.plane_CPUFailsafe()
def QAssist(self):
'''QuadPlane Assist tests'''
self.takeoff(50, mode="QHOVER", timeout=120)
self.set_rc(3, 1800)
self.change_mode("FBWA")
self.set_parameter("STALL_PREVENTION", 0)
thr_min_pwm = self.get_parameter("Q_M_PWM_MIN")
lim_roll_deg = self.get_parameter("ROLL_LIMIT_DEG")
lim_pitch_down_deg = self.get_parameter("PTCH_LIM_MIN_DEG")
lim_pitch_up_deg = self.get_parameter("PTCH_LIM_MAX_DEG")
self.progress("Waiting for motors to stop (transition completion)")
self.wait_servo_channel_value(5,
thr_min_pwm,
timeout=30,
comparator=operator.eq)
self.delay_sim_time(5)
self.wait_servo_channel_value(5,
thr_min_pwm,
timeout=30,
comparator=operator.eq)
self.progress("Stopping forward motor to kill airspeed below limit")
self.set_rc(3, 1000)
self.progress("Waiting for qassist to kick in")
self.wait_servo_channel_value(5, 1400, timeout=30, comparator=operator.gt)
self.progress("Move forward again, check qassist stops")
self.set_rc(3, 1800)
self.progress("Checking qassist stops")
self.wait_servo_channel_value(5,
thr_min_pwm,
timeout=30,
comparator=operator.eq)
self.set_rc(3, 1300)
self.start_subtest("Test angle assist (roll)")
self.context_push()
self.context_collect('STATUSTEXT')
self.progress("Rolling over to %.0f degrees" % -lim_roll_deg)
self.set_rc(1, 1000)
self.wait_roll(-lim_roll_deg, 5)
self.progress("Killing aileron servo output to force qassist to help")
self.set_parameter("SERVO1_MIN", 1480)
self.set_parameter("SERVO1_MAX", 1480)
self.set_parameter("SERVO1_TRIM", 1480)
self.progress("Trying to roll over hard the other way")
self.set_rc(1, 2000)
self.progress("Waiting for qassist (angle) to kick in")
self.wait_servo_channel_value(5, 1100, timeout=30, comparator=operator.gt)
self.wait_statustext('Angle assist', check_context=True)
self.wait_roll(lim_roll_deg, 5)
self.context_pop()
self.set_rc(1, 1500)
self.progress("Checking qassist stops")
self.set_rc(3, 1800)
self.wait_servo_channel_value(
5,
thr_min_pwm,
timeout=60,
comparator=operator.eq,
)
self.set_rc(3, 1300)
self.start_subtest("Test angle assist (pitch-down)")
self.context_push()
self.context_collect('STATUSTEXT')
self.progress("Pitching down to %.0f degrees" % lim_pitch_down_deg)
self.set_rc(2, 1000)
self.wait_pitch(lim_pitch_down_deg, accuracy=5)
self.progress("Killing elevator servo output to force qassist to help")
self.set_parameters({
"SERVO2_MIN": 1480,
"SERVO2_MAX": 1480,
"SERVO2_TRIM": 1480,
})
self.progress("Trying to pitch up hard")
self.set_rc(2, 2000)
self.progress("Waiting for qassist (angle) to kick in")
self.wait_servo_channel_value(5, 1100, timeout=30, comparator=operator.gt)
self.wait_statustext('Angle assist', check_context=True)
self.set_rc(2, 1500)
self.wait_pitch(0, accuracy=5)
self.context_pop()
self.progress("Checking qassist stops")
self.set_rc(3, 1800)
self.wait_servo_channel_value(
5,
thr_min_pwm,
timeout=30,
comparator=operator.eq,
)
self.set_rc(3, 1300)
self.start_subtest("Test angle assist (pitch-up)")
self.context_push()
self.context_collect('STATUSTEXT')
self.progress("Pitching up to %.0f degrees" % lim_pitch_up_deg)
self.set_rc(3, 2000)
self.delay_sim_time(5)
self.change_mode('MANUAL')
self.context_push()
self.set_parameter("SIM_SPEEDUP", 1)
self.set_rc(2, 1550)
self.wait_pitch(lim_pitch_up_deg+5, accuracy=5)
self.context_pop()
self.progress("Killing elevator servo output to force qassist to help")
servo2_out = self.get_servo_channel_value(2)
self.set_parameters({
"SERVO2_MIN": servo2_out,
"SERVO2_MAX": servo2_out,
"SERVO2_TRIM": servo2_out,
})
self.change_mode('FBWA')
self.progress("Trying to pitch down hard")
self.set_rc(2, 1000)
self.progress("Waiting for qassist (angle) to kick in")
self.wait_servo_channel_value(5, 1100, timeout=30, comparator=operator.gt)
self.wait_statustext('Angle assist', check_context=True)
self.set_rc(2, 1500)
self.wait_pitch(0, accuracy=5)
self.context_pop()
self.progress("Checking qassist stops")
self.set_rc(3, 1800)
self.wait_servo_channel_value(
5,
thr_min_pwm,
timeout=30,
comparator=operator.eq,
)
self.set_rc(3, 1300)
self.context_push()
guided_loc = self.home_relative_loc_ne(0, 0)
guided_loc.alt = 60
self.change_mode("GUIDED")
self.send_do_reposition(guided_loc)
self.wait_altitude(58, 62, relative=True, timeout=120)
self.set_parameter("Q_ASSIST_ALT", 50)
guided_loc.alt = 40
self.send_do_reposition(guided_loc)
self.wait_statustext(r"Alt assist \d*.\d*m", regex=True, timeout=100)
self.set_parameter("Q_TRANS_FAIL_ACT", 1)
self.set_parameter("Q_TRANS_FAIL", 10)
self.wait_mode("QRTL")
self.context_pop()
self.wait_disarmed(timeout=200)
def LoiterAltQLand(self):
'''test loitering and qland with terrain involved'''
self.LoiterAltQLand_Terrain(
home="LakeGeorgeLookout",
ofs_n=0,
ofs_e=300,
)
self.LoiterAltQLand_Relative()
def LoiterAltQLand_Relative(self):
'''test failsafe where vehicle loiters in fixed-wing mode to a
specific altitude then changes mode to QLAND'''
self.set_parameters({
'BATT_MONITOR': 4,
'BATT_FS_LOW_ACT': 6,
})
self.reboot_sitl()
takeoff_alt = 5
self.takeoff(takeoff_alt, mode='QLOITER')
loc = self.mav.location()
self.location_offset_ne(loc, 500, 500)
new_alt = 100
initial_altitude = self.get_altitude(relative=False, timeout=2)
self.run_cmd_int(
mavutil.mavlink.MAV_CMD_DO_REPOSITION,
0,
1,
0,
0,
int(loc.lat * 1e7),
int(loc.lng * 1e7),
new_alt,
frame=mavutil.mavlink.MAV_FRAME_GLOBAL_RELATIVE_ALT_INT,
)
self.wait_altitude(
new_alt-1,
new_alt+1,
timeout=60,
relative=True,
minimum_duration=10)
self.wait_location(loc, timeout=120, accuracy=100)
self.progress("Triggering failsafe")
self.set_parameter('BATT_LOW_VOLT', 50)
self.wait_mode(25)
self.drain_mav()
m = self.assert_receive_message('POSITION_TARGET_GLOBAL_INT', very_verbose=True)
q_rtl_alt = self.get_parameter('Q_RTL_ALT')
expected_alt = initial_altitude - takeoff_alt + q_rtl_alt
if abs(m.alt - expected_alt) > 20:
raise NotAchievedException("Unexpected altitude; expected=%f got=%f" %
(expected_alt, m.alt))
self.assert_mode('LOITERALTQLAND')
self.wait_mode('QLAND')
alt = self.get_altitude(relative=True)
if abs(alt - q_rtl_alt) > 2:
raise NotAchievedException("qland too late; want=%f got=%f" %
(alt, q_rtl_alt))
self.wait_disarmed(timeout=300)
def LoiterAltQLand_Terrain(self,
home=None,
ofs_n=None,
ofs_e=None,
reposition_alt=100):
'''test failsafe where vehicle loiters in fixed-wing mode to a
specific altitude then changes mode to QLAND'''
self.context_push()
self.install_terrain_handlers_context()
self.set_parameters({
'BATT_MONITOR': 4,
'BATT_FS_LOW_ACT': 6,
'TERRAIN_FOLLOW': 1,
})
self.customise_SITL_commandline(
["--home", home]
)
takeoff_alt = 5
self.takeoff(takeoff_alt, mode='QLOITER')
loc = self.mav.location()
self.location_offset_ne(loc, ofs_n, ofs_e)
initial_altitude = self.get_altitude(relative=False, timeout=2)
self.run_cmd_int(
mavutil.mavlink.MAV_CMD_DO_REPOSITION,
0,
1,
0,
0,
int(loc.lat * 1e7),
int(loc.lng * 1e7),
reposition_alt,
frame=mavutil.mavlink.MAV_FRAME_GLOBAL_RELATIVE_ALT_INT,
)
self.wait_altitude(
reposition_alt-1,
reposition_alt+1,
timeout=60,
relative=True,
minimum_duration=10)
self.wait_location(loc, timeout=500, accuracy=100)
self.progress("Triggering failsafe")
self.set_parameter('BATT_LOW_VOLT', 50)
self.wait_mode(25)
terrain_alt = self.get_terrain_height(verbose=True)
self.drain_mav()
m = self.assert_receive_message('POSITION_TARGET_GLOBAL_INT', very_verbose=True)
q_rtl_alt = self.get_parameter('Q_RTL_ALT')
expected_alt = terrain_alt + q_rtl_alt
if abs(m.alt - expected_alt) > 20:
raise NotAchievedException("Unexpected altitude; expected=%f got=%f" %
(expected_alt, m.alt))
self.assert_mode('LOITERALTQLAND')
self.wait_mode('QLAND')
alt = initial_altitude + self.get_altitude(relative=True)
if abs(alt - expected_alt) > 10:
raise NotAchievedException("qland too late; want=%f got=%f" %
(expected_alt, alt))
self.wait_disarmed(timeout=300)
self.zero_throttle()
self.reset_SITL_commandline()
self.context_pop()
def GUIDEDToAUTO(self):
'''Test using GUIDED mode for takeoff before shifting to auto'''
self.load_mission("mission.txt")
self.takeoff(30, mode='GUIDED')
self.assert_not_receiving_message('CAMERA_FEEDBACK')
self.change_mode('AUTO')
self.wait_current_waypoint(3)
self.change_mode('QRTL')
self.wait_disarmed(timeout=240)
def Tailsitter(self):
'''tailsitter test'''
self.set_parameter('Q_FRAME_CLASS', 10)
self.set_parameter('Q_ENABLE', 1)
self.set_parameter('Q_TAILSIT_ENABLE', 1)
self.reboot_sitl()
self.wait_ready_to_arm()
value_before = self.get_servo_channel_value(3)
self.progress("Before: %u" % value_before)
self.change_mode('QHOVER')
tstart = self.get_sim_time()
while True:
now = self.get_sim_time_cached()
if now - tstart > 60:
break
value_after = self.get_servo_channel_value(3)
self.progress("After: t=%f output=%u" % ((now - tstart), value_after))
if value_before != value_after:
raise NotAchievedException("Changed throttle output on mode change to QHOVER")
self.disarm_vehicle()
def CopterTailsitter(self):
'''copter tailsitter test'''
self.customise_SITL_commandline(
[],
defaults_filepath=self.model_defaults_filepath('quadplane-copter_tailsitter'),
model="quadplane-copter_tailsitter",
wipe=True,
)
self.reboot_sitl()
self.wait_ready_to_arm()
self.takeoff(60, mode='GUIDED')
self.context_collect("STATUSTEXT")
self.progress("Starting QLAND")
self.change_mode("QLAND")
self.wait_statustext("Rangefinder engaged", check_context=True)
self.wait_disarmed(timeout=100)
def setup_ICEngine_vehicle(self):
'''restarts SITL with an IC Engine setup'''
model = "quadplane-ice"
self.customise_SITL_commandline(
[],
model=model,
defaults_filepath=self.model_defaults_filepath(model),
wipe=False,
)
def ICEngine(self):
'''Test ICE Engine support'''
rc_engine_start_chan = 11
self.setup_ICEngine_vehicle()
self.wait_ready_to_arm()
self.wait_rpm(1, 0, 0, minimum_duration=1)
self.arm_vehicle()
self.wait_rpm(1, 0, 0, minimum_duration=1)
self.context_collect("STATUSTEXT")
self.progress("Setting engine-start RC switch to HIGH")
self.set_rc(rc_engine_start_chan, 2000)
self.wait_statustext("Starting engine", check_context=True)
self.wait_rpm(1, 300, 400, minimum_duration=1)
self.progress("Setting engine-start RC switch to MID")
self.set_rc(rc_engine_start_chan, 1500)
self.progress("Setting full throttle")
self.set_rc(3, 2000)
self.wait_rpm(1, 6500, 7500, minimum_duration=30, timeout=40)
self.progress("Setting min-throttle")
self.set_rc(3, 1000)
self.wait_rpm(1, 65, 75, minimum_duration=1)
self.progress("Setting engine-start RC switch to LOW")
self.set_rc(rc_engine_start_chan, 1000)
self.wait_rpm(1, 0, 0, minimum_duration=1)
self.disarm_vehicle(force=True)
self.reboot_sitl()
self.start_subtest("Testing throttle out in manual mode")
self.change_mode('MANUAL')
self.set_rc(3, 1700)
self.wait_servo_channel_value(3, 2000)
self.set_parameter("ICE_OPTIONS", 4)
self.wait_servo_channel_value(3, 1300)
self.change_mode('FBWA')
self.wait_servo_channel_value(3, 2000)
self.start_subtest("Testing automatic restart")
max_tries = 4
self.set_parameter("ICE_STRT_MX_RTRY", max_tries)
rpm_chan = self.get_parameter("ICE_RPM_CHAN")
self.set_parameter("ICE_RPM_CHAN", 120)
self.set_rc(rc_engine_start_chan, 2000)
self.wait_statustext("Uncommanded engine stop")
self.wait_statustext("Starting engine")
self.set_parameter("ICE_RPM_CHAN", rpm_chan)
try:
self.wait_statustext("Uncommanded engine stop", timeout=30)
raise NotAchievedException("Engine stopped unexpectedly")
except AutoTestTimeoutException:
pass
self.context_stop_collecting("STATUSTEXT")
self.start_subtest("Testing automatic starter attempt limit")
for i in range(2):
self.context_collect("STATUSTEXT")
self.set_parameter("ICE_RPM_CHAN", 120)
self.set_rc(rc_engine_start_chan, 2000)
self.wait_statustext("Engine max crank attempts reached", check_context=True, timeout=30)
self.delay_sim_time(30)
messages = self.context_get().collections["STATUSTEXT"]
self.context_stop_collecting("STATUSTEXT")
attempts = 0
for m in messages:
if "Starting engine" == m.text:
attempts += 1
if attempts != max_tries:
raise NotAchievedException(f"Run {i+1}: Expected {max_tries} attempts, got {attempts}")
self.context_collect("STATUSTEXT")
self.set_rc(rc_engine_start_chan, 1000)
self.wait_statustext("ignition:0", check_context=True)
self.context_stop_collecting("STATUSTEXT")
def ICEngineMission(self):
'''Test ICE Engine Mission support'''
rc_engine_start_chan = 11
self.setup_ICEngine_vehicle()
self.load_mission("mission.txt")
self.wait_ready_to_arm()
self.set_rc(rc_engine_start_chan, 2000)
self.arm_vehicle()
self.change_mode('AUTO')
self.wait_disarmed(timeout=300)
def ICEngineRPMGovernor(self):
'''Test ICE idle and redline governor'''
self.setup_ICEngine_vehicle()
options = int(self.get_parameter("ICE_OPTIONS"))
options |= 1 << 2
self.set_parameter("ICE_OPTIONS", options)
self.start_subtest("ICEngine idle governor")
self.change_mode('QHOVER')
self.run_cmd(mavutil.mavlink.MAV_CMD_DO_ENGINE_CONTROL, p1=1)
deadband = 50
self.set_parameter("ICE_IDLE_DB", deadband)
for idle_rpm in (1000, 1500):
self.set_parameter("ICE_IDLE_RPM", idle_rpm)
self.wait_rpm(
1,
idle_rpm - 1.1 * deadband,
idle_rpm + 1.1 * deadband,
timeout=60,
minimum_duration=15,
)
self.start_subtest("ICEngine redline governor")
self.change_mode('MANUAL')
self.wait_ready_to_arm()
self.arm_vehicle()
self.set_rc(3, 2000)
for redline_rpm in (6500, 6000):
self.set_parameter("ICE_REDLINE_RPM", redline_rpm)
self.wait_rpm(
1,
redline_rpm - 2 * deadband,
redline_rpm,
timeout=60,
minimum_duration=15,
)
self.set_rc(3, 1000)
self.disarm_vehicle()
def MAV_CMD_DO_ENGINE_CONTROL(self):
'''test MAV_CMD_DO_ENGINE_CONTROL mavlink command'''
expected_idle_rpm_min = 65
expected_idle_rpm_max = 75
expected_starter_rpm_min = 345
expected_starter_rpm_max = 355
rc_engine_start_chan = 11
self.setup_ICEngine_vehicle()
self.wait_ready_to_arm()
for method in self.run_cmd, self.run_cmd_int:
self.change_mode('MANUAL')
self.set_rc(rc_engine_start_chan, 1500)
self.wait_rpm(1, 0, 0, minimum_duration=1)
self.arm_vehicle()
self.wait_rpm(1, 0, 0, minimum_duration=1)
self.start_subtest("Start motor")
method(mavutil.mavlink.MAV_CMD_DO_ENGINE_CONTROL, p1=1)
self.wait_rpm(1, expected_starter_rpm_min, expected_starter_rpm_max)
self.wait_rpm(1, expected_idle_rpm_min, expected_idle_rpm_max, minimum_duration=10)
self.start_subtest("try start motor again")
self.context_collect('STATUSTEXT')
method(mavutil.mavlink.MAV_CMD_DO_ENGINE_CONTROL, p1=1, want_result=mavutil.mavlink.MAV_RESULT_FAILED)
self.wait_statustext("already running", check_context=True)
self.context_stop_collecting('STATUSTEXT')
self.wait_rpm(1, expected_idle_rpm_min, expected_idle_rpm_max, minimum_duration=1)
self.start_subtest("Stop motor")
method(mavutil.mavlink.MAV_CMD_DO_ENGINE_CONTROL, p1=0)
self.wait_rpm(1, 0, 0, minimum_duration=1)
self.start_subtest("Stop motor (again)")
method(mavutil.mavlink.MAV_CMD_DO_ENGINE_CONTROL, p1=0)
self.wait_rpm(1, 0, 0, minimum_duration=1)
self.start_subtest("Check start chan control disable")
old_start_channel_value = self.get_rc_channel_value(rc_engine_start_chan)
self.set_rc(rc_engine_start_chan, 1000)
self.delay_sim_time(1)
self.context_collect('STATUSTEXT')
method(mavutil.mavlink.MAV_CMD_DO_ENGINE_CONTROL, p1=1, want_result=mavutil.mavlink.MAV_RESULT_FAILED)
self.wait_statustext("start control disabled", check_context=True)
self.context_stop_collecting('STATUSTEXT')
self.set_rc(rc_engine_start_chan, old_start_channel_value)
self.wait_rpm(1, 0, 0, minimum_duration=1)
self.start_subtest("test start-at-height")
self.wait_rpm(1, 0, 0, minimum_duration=1)
self.context_collect('STATUSTEXT')
method(
mavutil.mavlink.MAV_CMD_DO_ENGINE_CONTROL,
p1=1,
p3=15.5,
)
self.wait_statustext("height set to 15.5m", check_context=True)
self.wait_rpm(1, 0, 0, minimum_duration=2)
self.takeoff(20, mode='GUIDED')
self.wait_rpm(1, expected_starter_rpm_min, expected_starter_rpm_max, minimum_duration=1)
self.wait_statustext("Engine running", check_context=True)
self.context_stop_collecting('STATUSTEXT')
method(mavutil.mavlink.MAV_CMD_DO_ENGINE_CONTROL, p1=0)
self.wait_rpm(1, 0, 0, minimum_duration=1)
self.change_mode('QLAND')
self.wait_disarmed()
def Ship(self):
'''Ensure we can take off from simulated ship'''
self.context_push()
self.set_parameters({
'SIM_SHIP_ENABLE': 1,
'SIM_SHIP_SPEED': 1,
})
self.change_mode('QLOITER')
self.wait_ready_to_arm()
self.arm_vehicle()
self.set_rc(3, 1700)
self.wait_altitude(20, 30, relative=True)
self.disarm_vehicle(force=True)
self.context_pop()
self.reboot_sitl()
def MidAirDisarmDisallowed(self):
'''Check disarm behaviour in Q-mode'''
self.start_subtest("Basic arm in qloiter")
self.set_parameter("FLIGHT_OPTIONS", 0)
self.change_mode('QLOITER')
self.wait_ready_to_arm()
self.arm_vehicle()
self.disarm_vehicle()
self.context_push()
self.start_subtest("Ensure disarming in q-modes on ground works")
self.set_parameter("FLIGHT_OPTIONS", 1 << 11)
self.arm_vehicle()
self.disarm_vehicle()
self.context_pop()
self.start_subtest("Ensure no disarming mid-air")
self.arm_vehicle()
self.set_rc(3, 2000)
self.wait_altitude(5, 50, relative=True)
self.set_rc(3, 1000)
disarmed = False
try:
self.disarm_vehicle()
disarmed = True
except ValueError as e:
self.progress("Got %s" % repr(e))
if "Expected MAV_RESULT_ACCEPTED got MAV_RESULT_FAILED" not in str(e):
raise e
if disarmed:
raise NotAchievedException("Disarmed when we shouldn't have")
self.change_mode('QLAND')
self.wait_disarmed()
self.start_subtest("Check we can disarm after a short period on the ground")
self.takeoff(5, 'QHOVER')
self.change_mode('QLAND')
try:
self.set_message_rate_hz(mavutil.mavlink.MAVLINK_MSG_ID_EXTENDED_SYS_STATE, 10)
self.wait_extended_sys_state(
landed_state=mavutil.mavlink.MAV_LANDED_STATE_ON_GROUND,
vtol_state=mavutil.mavlink.MAV_VTOL_STATE_MC,
timeout=60
)
except Exception:
self.set_message_rate_hz(mavutil.mavlink.MAVLINK_MSG_ID_EXTENDED_SYS_STATE, 0)
raise
self.set_message_rate_hz(mavutil.mavlink.MAVLINK_MSG_ID_EXTENDED_SYS_STATE, -1)
self.disarm_vehicle()
def MAV_CMD_NAV_LOITER_TO_ALT(self, target_system=1, target_component=1):
'''ensure consecutive loiter to alts work'''
self.load_mission('mission.txt')
self.change_mode('AUTO')
self.wait_ready_to_arm()
self.arm_vehicle()
self.wait_current_waypoint(4, timeout=240)
self.assert_altitude(120, accuracy=5, relative=True)
self.delay_sim_time(30)
self.assert_altitude(120, accuracy=5, relative=True)
self.set_current_waypoint(5)
self.wait_altitude(altitude_min=65, altitude_max=75, relative=True)
if self.current_waypoint() != 5:
raise NotAchievedException("Should pass 90m before passing waypoint 5")
self.wait_disarmed(timeout=300)
def Mission(self):
'''fly the OBC 2016 mission in Dalby'''
self.load_mission("Dalby-OBC2016.txt")
self.load_fence("Dalby-OBC2016-fence.txt")
if self.mavproxy is not None:
self.mavproxy.send('wp list\n')
self.install_terrain_handlers_context()
self.wait_ready_to_arm()
self.arm_vehicle()
self.change_mode('AUTO')
self.wait_waypoint(1, 19, max_dist=60, timeout=1200)
self.wait_disarmed(timeout=120)
self.set_current_waypoint(20)
self.wait_ready_to_arm()
self.arm_vehicle()
self.wait_waypoint(20, 34, max_dist=60, timeout=1200)
self.wait_disarmed(timeout=120)
self.progress("Mission OK")
def VTOLLandSpiral(self):
'''check spiral-to-alt option for landing'''
self.fly_mission('mission.txt')
self.set_parameter('WP_LOITER_RAD', -self.get_parameter('WP_LOITER_RAD'))
self.set_current_waypoint(0, check_afterwards=False)
self.fly_mission('mission.txt')
def VTOLQuicktune(self):
'''VTOL Quicktune'''
self.install_applet_script_context("VTOL-quicktune.lua")
self.set_parameters({
"SCR_ENABLE": 1,
"SIM_SPEEDUP": 20,
"RC7_OPTION": 300,
})
self.reboot_sitl()
self.context_collect('STATUSTEXT')
self.set_parameters({
"QUIK_ENABLE" : 1,
"QUIK_DOUBLE_TIME" : 5,
})
self.scripting_restart()
self.wait_text("Quicktune for quadplane loaded", check_context=True)
self.wait_ready_to_arm()
self.change_mode("QLOITER")
self.arm_vehicle()
self.takeoff(20, 'QLOITER')
self.set_rc(7, 1500)
self.wait_text("Tuning: starting tune", check_context=True)
for axis in ['RLL', 'PIT', 'YAW']:
self.wait_text("Starting %s tune" % axis, check_context=True)
self.wait_text("Tuning: %s_D done" % axis, check_context=True, timeout=120)
self.wait_text("Tuning: %s_P done" % axis, check_context=True, timeout=120)
self.wait_text("Tuning: %s done" % axis, check_context=True, timeout=120)
self.wait_text("Tuning: YAW done", check_context=True, timeout=120)
self.run_auxfunc(300, 2)
self.wait_text("Tuning: saved", check_context=True)
self.change_mode("QLAND")
self.wait_disarmed(timeout=120)
def VTOLQuicktune_CPP(self):
'''VTOL Quicktune in C++'''
self.set_parameters({
"RC7_OPTION": 181,
"QWIK_ENABLE" : 1,
"QWIK_DOUBLE_TIME" : 5,
})
self.context_push()
self.context_collect('STATUSTEXT')
gain_mul = 0.5
soften_params = ['Q_A_RAT_RLL_P', 'Q_A_RAT_RLL_I', 'Q_A_RAT_RLL_D',
'Q_A_RAT_PIT_P', 'Q_A_RAT_PIT_I', 'Q_A_RAT_PIT_D',
'Q_A_RAT_YAW_P', 'Q_A_RAT_YAW_I']
original_values = self.get_parameters(soften_params)
softened_values = {}
for p in original_values.keys():
softened_values[p] = original_values[p] * gain_mul
self.set_parameters(softened_values)
self.wait_ready_to_arm()
self.change_mode("QLOITER")
self.set_rc(7, 1000)
self.arm_vehicle()
self.takeoff(20, 'QLOITER')
self.set_rc(7, 1500)
self.wait_text("Quicktune: starting tune", check_context=True)
for axis in ['Roll', 'Pitch', 'Yaw']:
self.wait_text("Starting %s tune" % axis, check_context=True)
self.wait_text("Quicktune: %s D done" % axis, check_context=True, timeout=120)
self.wait_text("Quicktune: %s P done" % axis, check_context=True, timeout=120)
self.wait_text("Quicktune: %s done" % axis, check_context=True, timeout=120)
new_values = self.get_parameters(soften_params)
for p in original_values.keys():
threshold = 0.8 * original_values[p]
self.progress("tuned param %s %.4f need %.4f" % (p, new_values[p], threshold))
if new_values[p] < threshold:
raise NotAchievedException(
"parameter %s %.4f not increased over %.4f" %
(p, new_values[p], threshold))
self.progress("ensure we are not overtuned")
self.set_parameters({
'SIM_ENGINE_MUL': 0.9,
'SIM_ENGINE_FAIL': 1 << 0,
})
self.delay_sim_time(5)
self.set_parameter('SIM_ENGINE_MUL', 1)
for i in range(5):
self.wait_heartbeat()
if self.statustext_in_collections("ABORTING"):
raise NotAchievedException("tune has aborted, overtuned")
self.progress("using aux fn for save tune")
self.run_auxfunc(181, 2)
self.wait_text("Quicktune: saved", check_context=True)
self.change_mode("QLAND")
self.wait_disarmed(timeout=120)
self.set_parameter("QWIK_ENABLE", 0)
self.context_pop()
self.reboot_sitl()
def PrecisionLanding(self):
'''VTOL precision landing'''
self.install_applet_script_context("plane_precland.lua")
here = self.mav.location()
target = self.offset_location_ne(here, 20, 0)
self.set_parameters({
"SCR_ENABLE": 1,
"PLND_ENABLED": 1,
"PLND_TYPE": 4,
"SIM_PLD_ENABLE": 1,
"SIM_PLD_LAT" : target.lat,
"SIM_PLD_LON" : target.lng,
"SIM_PLD_HEIGHT" : 0,
"SIM_PLD_ALT_LMT" : 50,
"SIM_PLD_DIST_LMT" : 30,
"RNGFND1_TYPE": 100,
"RNGFND1_PIN" : 0,
"RNGFND1_SCALING" : 12.2,
"RNGFND1_MAX" : 50.00,
"RNGFND_LANDING" : 1,
})
self.reboot_sitl()
self.set_parameters({
"PLND_ALT_CUTOFF" : 5,
"SIM_SPEEDUP" : 10,
})
self.context_collect('STATUSTEXT')
self.scripting_restart()
self.wait_text("PLND: Loaded", check_context=True)
self.wait_ready_to_arm()
self.change_mode("GUIDED")
self.arm_vehicle()
self.takeoff(60, 'GUIDED')
self.wait_altitude(58, 62, relative=True)
self.drain_mav()
self.change_mode("QRTL")
self.wait_text("PLND: Target Acquired", check_context=True, timeout=60)
self.wait_disarmed(timeout=180)
loc2 = self.mav.location()
error = self.get_distance(target, loc2)
self.progress("Target error %.1fm" % error)
if error > 2:
raise NotAchievedException("too far from target %.1fm" % error)
def ShipLanding(self):
'''ship landing test'''
self.install_applet_script_context("plane_ship_landing.lua")
self.set_parameters({
"SCR_ENABLE": 1,
"SIM_SHIP_ENABLE": 1,
"SIM_SHIP_SPEED": 5,
"Q_WP_SPEED": 700,
"Q_P_NE_POS_P": 0.25,
"Q_P_NE_VEL_D": 0.25,
"Q_P_NE_VEL_I": 0.25,
"Q_P_NE_VEL_P": 1.0,
"SIM_SHIP_DSIZE": 10,
"FOLL_ENABLE": 1,
"FOLL_SYSID": 17,
"FOLL_OFS_TYPE": 1,
"SIM_TERRAIN" : 0,
"TERRAIN_ENABLE" : 0,
})
self.load_mission("takeoff100.txt")
self.reboot_sitl(check_position=False)
self.context_collect('STATUSTEXT')
self.set_parameters({
"SHIP_ENABLE" : 1,
"SIM_SPEEDUP" : 10,
})
self.scripting_restart()
self.wait_text("ShipLanding: loaded", check_context=True)
self.wait_ready_to_arm()
self.change_mode("AUTO")
self.arm_vehicle()
self.wait_altitude(95, 105, relative=True, timeout=90)
self.drain_mav()
self.wait_text("Mission complete, changing mode to RTL", check_context=True, timeout=60)
self.wait_text("Descending for approach", check_context=True, timeout=60)
self.wait_text("Reached target altitude", check_context=True, timeout=120)
self.wait_text("Starting approach", check_context=True, timeout=120)
self.wait_text("Land complete", check_context=True, timeout=120)
self.wait_disarmed(timeout=180)
self.wait_groundspeed(4.8, 5.2)
tstart = self.get_sim_time_cached()
ship_gpi = None
vehicle_gpi = None
while ship_gpi is None or vehicle_gpi is None:
if self.get_sim_time_cached() - tstart > 5:
raise NotAchievedException("Did not get GPI for ship")
gpi = self.assert_receive_message('GLOBAL_POSITION_INT')
if gpi.get_srcSystem() == 17:
ship_gpi = gpi
elif gpi.get_srcSystem() == 1:
vehicle_gpi = gpi
distance = self.get_distance_int(vehicle_gpi, ship_gpi)
self.progress(f"{distance=}")
max_distance = 1.2
if distance > max_distance:
raise NotAchievedException(f"Did not land within {max_distance}m of ship {distance=}")
def RCDisableAirspeedUse(self):
'''check disabling airspeed using RC switch'''
self.set_parameter("RC9_OPTION", 106)
self.delay_sim_time(5)
self.set_rc(9, 1000)
self.wait_sensor_state(
mavutil.mavlink.MAV_SYS_STATUS_SENSOR_DIFFERENTIAL_PRESSURE,
True,
True,
True)
self.set_rc(9, 2000)
self.wait_sensor_state(
mavutil.mavlink.MAV_SYS_STATUS_SENSOR_DIFFERENTIAL_PRESSURE,
True,
False,
True)
self.set_rc(9, 1000)
self.wait_sensor_state(
mavutil.mavlink.MAV_SYS_STATUS_SENSOR_DIFFERENTIAL_PRESSURE,
True,
True,
True)
self.progress("Disabling airspeed sensor")
self.context_push()
self.set_rc(9, 2000)
self.set_parameters({
"COMPASS_ENABLE": 0,
"EK2_ENABLE": 0,
"AHRS_EKF_TYPE": 3,
"COMPASS_USE": 0,
"COMPASS_USE2": 0,
"COMPASS_USE3": 0,
"ARMING_SKIPCHK": 1 << 2,
})
self.reboot_sitl()
self.context_collect('STATUSTEXT')
self.wait_prearm_sys_status_healthy(timeout=120)
self.change_mode('QLOITER')
self.arm_vehicle()
self.set_rc(3, 2000)
self.wait_altitude(10, 30, relative=True)
self.change_mode('FBWA')
self.wait_statustext('Transition done')
self.wait_statustext('EKF3 active', check_context=True)
self.disarm_vehicle(force=True)
self.context_pop()
self.reboot_sitl()
def mission_MAV_CMD_DO_VTOL_TRANSITION(self):
'''mission item forces transition'''
wps = self.create_simple_relhome_mission([
(mavutil.mavlink.MAV_CMD_NAV_TAKEOFF, 0, 0, 30),
(mavutil.mavlink.MAV_CMD_NAV_WAYPOINT, 200, 0, 30),
self.create_MISSION_ITEM_INT(
mavutil.mavlink.MAV_CMD_DO_VTOL_TRANSITION,
p1=mavutil.mavlink.MAV_VTOL_STATE_MC
),
(mavutil.mavlink.MAV_CMD_NAV_WAYPOINT, 300, 200, 30),
self.create_MISSION_ITEM_INT(
mavutil.mavlink.MAV_CMD_DO_VTOL_TRANSITION,
p1=mavutil.mavlink.MAV_VTOL_STATE_FW
),
(mavutil.mavlink.MAV_CMD_NAV_WAYPOINT, 100, 200, 30),
(mavutil.mavlink.MAV_CMD_NAV_RETURN_TO_LAUNCH, 0, 0, 0),
])
self.check_mission_upload_download(wps)
self.change_mode('AUTO')
self.wait_ready_to_arm()
self.arm_vehicle()
self.wait_current_waypoint(4)
self.wait_servo_channel_value(5, 1200, comparator=operator.gt)
self.wait_current_waypoint(6)
self.wait_servo_channel_value(5, 1000, comparator=operator.eq, timeout=90)
self.fly_home_land_and_disarm()
def mavlink_MAV_CMD_DO_VTOL_TRANSITION(self):
'''mavlink command forces transition during mission'''
wps = self.create_simple_relhome_mission([
(mavutil.mavlink.MAV_CMD_NAV_TAKEOFF, 0, 0, 30),
(mavutil.mavlink.MAV_CMD_NAV_WAYPOINT, 2000, 0, 30),
(mavutil.mavlink.MAV_CMD_NAV_RETURN_TO_LAUNCH, 0, 0, 0),
])
self.check_mission_upload_download(wps)
self.change_mode('AUTO')
self.wait_ready_to_arm()
self.arm_vehicle()
self.wait_current_waypoint(2)
self.wait_servo_channel_value(5, 1000, comparator=operator.eq, timeout=90)
for command in self.run_cmd, self.run_cmd_int:
command(mavutil.mavlink.MAV_CMD_DO_VTOL_TRANSITION, p1=mavutil.mavlink.MAV_VTOL_STATE_MC)
self.wait_servo_channel_value(5, 1200, comparator=operator.gt, timeout=300)
command(mavutil.mavlink.MAV_CMD_DO_VTOL_TRANSITION, p1=mavutil.mavlink.MAV_VTOL_STATE_FW)
self.wait_servo_channel_value(5, 1000, comparator=operator.eq, timeout=90)
self.fly_home_land_and_disarm()
def TransitionMinThrottle(self):
'''Ensure that TKOFF_THR_MIN is applied during the forward transition'''
wps = self.create_simple_relhome_mission([
(mavutil.mavlink.MAV_CMD_NAV_TAKEOFF, 0, 0, 30),
(mavutil.mavlink.MAV_CMD_NAV_WAYPOINT, 2000, 0, 30),
(mavutil.mavlink.MAV_CMD_NAV_RETURN_TO_LAUNCH, 0, 0, 0),
])
self.check_mission_upload_download(wps)
self.set_parameter('TKOFF_THR_MIN', 80)
self.change_mode('AUTO')
self.wait_ready_to_arm()
self.arm_vehicle()
self.wait_current_waypoint(2)
self.delay_sim_time(5)
thr_min = self.get_parameter('TKOFF_THR_MIN')
self.wait_servo_channel_value(3, 1000+thr_min*10, comparator=operator.eq)
self.fly_home_land_and_disarm()
def BackTransitionMinThrottle(self):
'''Ensure min throttle is applied during back transition.'''
wps = self.create_simple_relhome_mission([
(mavutil.mavlink.MAV_CMD_NAV_TAKEOFF, 0, 0, 30),
(mavutil.mavlink.MAV_CMD_NAV_WAYPOINT, 2000, 0, 30),
(mavutil.mavlink.MAV_CMD_NAV_RETURN_TO_LAUNCH, 0, 0, 0),
])
self.check_mission_upload_download(wps)
self.set_parameter('Q_RTL_MODE', 1)
trim_pwm = 1000 + 10*self.get_parameter("TRIM_THROTTLE")
min_pwm = 1000 + 10*self.get_parameter("THR_MIN")
self.change_mode('AUTO')
self.wait_ready_to_arm()
self.arm_vehicle()
self.context_collect('STATUSTEXT')
self.wait_statustext("VTOL airbrake", check_context=True, timeout=300)
self.wait_servo_channel_value(3, trim_pwm, comparator=operator.le, timeout=1)
self.wait_statustext("VTOL position1", check_context=True, timeout=10)
self.wait_servo_channel_value(3, min_pwm+10, comparator=operator.le, timeout=1)
self.wait_disarmed(timeout=60)
def MAV_CMD_NAV_TAKEOFF(self):
'''test issuing takeoff command via mavlink'''
self.change_mode('GUIDED')
self.wait_ready_to_arm()
self.arm_vehicle()
self.run_cmd(mavutil.mavlink.MAV_CMD_NAV_TAKEOFF, p7=5)
self.wait_altitude(4.5, 5.5, minimum_duration=5, relative=True)
self.change_mode('QLAND')
self.wait_disarmed()
self.start_subtest("Check NAV_TAKEOFF is above current location, not home location")
self.change_mode('GUIDED')
self.wait_ready_to_arm()
current_alt_abs = self.get_altitude(relative=False)
loc = self.mav.location()
home_z_ofs = 20
self.run_cmd(
mavutil.mavlink.MAV_CMD_DO_SET_HOME,
p5=loc.lat,
p6=loc.lng,
p7=current_alt_abs + home_z_ofs,
)
self.arm_vehicle()
takeoff_alt = 5
self.run_cmd(mavutil.mavlink.MAV_CMD_NAV_TAKEOFF, p7=takeoff_alt)
self.wait_altitude(
current_alt_abs + takeoff_alt - 0.5,
current_alt_abs + takeoff_alt + 0.5,
minimum_duration=5,
relative=False,
)
self.change_mode('QLAND')
self.wait_disarmed()
self.reboot_sitl()
def Q_GUIDED_MODE(self):
'''test moving in VTOL mode with SET_POSITION_TARGET_GLOBAL_INT'''
self.set_parameter('Q_GUIDED_MODE', 1)
self.change_mode('GUIDED')
self.wait_ready_to_arm()
self.arm_vehicle()
self.run_cmd(mavutil.mavlink.MAV_CMD_NAV_TAKEOFF, p7=15)
self.wait_altitude(14, 16, relative=True)
loc = self.mav.location()
self.location_offset_ne(loc, 50, 50)
self.run_cmd_int(
mavutil.mavlink.MAV_CMD_DO_REPOSITION,
0,
1,
0,
0,
int(loc.lat * 1e7),
int(loc.lng * 1e7),
30,
frame=mavutil.mavlink.MAV_FRAME_GLOBAL_RELATIVE_ALT_INT,
)
self.wait_location(loc, timeout=120)
self.fly_home_land_and_disarm()
def DCMClimbRate(self):
'''Test the climb rate measurement in DCM with and without GPS'''
self.wait_ready_to_arm()
self.change_mode('QHOVER')
self.arm_vehicle()
self.set_rc(3, 2000)
self.wait_altitude(30, 50, relative=True)
self.set_rc(3, 1000)
self.wait_climbrate(-5, -0.5, timeout=10)
self.set_parameter('AHRS_EKF_TYPE', 0)
self.delay_sim_time(5)
self.set_rc(3, 2000)
self.wait_climbrate(0.5, 5, timeout=10)
self.set_parameters({
'SIM_GPS1_ENABLE': 0,
'SIM_GPS2_ENABLE': 0,
})
self.delay_sim_time(5)
self.set_rc(3, 1000)
self.wait_climbrate(-5, -0.5, timeout=10)
self.reboot_sitl(force=True)
def RTL_AUTOLAND_1(self):
'''test behaviour when RTL_AUTOLAND==1'''
self.set_parameters({
"RTL_AUTOLAND": 1,
})
wps = self.create_simple_relhome_mission([
(mavutil.mavlink.MAV_CMD_NAV_TAKEOFF, 0, 0, 30),
(mavutil.mavlink.MAV_CMD_NAV_WAYPOINT, 500, 0, 30),
(mavutil.mavlink.MAV_CMD_NAV_WAYPOINT, 1000, 0, 30),
self.create_MISSION_ITEM_INT(
mavutil.mavlink.MAV_CMD_DO_LAND_START,
),
(mavutil.mavlink.MAV_CMD_NAV_WAYPOINT, 0, -1000, 30),
(mavutil.mavlink.MAV_CMD_NAV_WAYPOINT, 0, -1300, 15),
(mavutil.mavlink.MAV_CMD_NAV_WAYPOINT, 0, -1600, 5),
(mavutil.mavlink.MAV_CMD_NAV_VTOL_LAND, 0, -1750, 0),
])
self.check_mission_upload_download(wps)
self.change_mode('AUTO')
self.wait_ready_to_arm()
self.arm_vehicle()
self.wait_current_waypoint(3)
self.change_mode('RTL')
self.wait_distance_to_home(100, 200, timeout=120)
self.wait_current_waypoint(7)
self.fly_home_land_and_disarm()
def send_reposition_to_loc(self, loc):
self.run_cmd_int(
mavutil.mavlink.MAV_CMD_DO_REPOSITION,
0,
1,
0,
0,
int(loc.lat * 1e7),
int(loc.lng * 1e7),
20,
frame=mavutil.mavlink.MAV_FRAME_GLOBAL_RELATIVE_ALT,
)
def reposition_to_loc(self, loc, accuracy=100):
self.send_reposition_to_loc(loc)
self.wait_location(
loc,
accuracy=accuracy,
minimum_duration=20,
timeout=120,
)
def AHRSFlyForwardFlag(self):
'''ensure FlyForward flag is set appropriately'''
self.set_parameters({
"LOG_DISARMED": 1,
"LOG_REPLAY": 1,
})
self.reboot_sitl()
self.assert_mode_is('FBWA')
self.delay_sim_time(10)
self.change_mode('QHOVER')
self.delay_sim_time(10)
self.wait_ready_to_arm()
self.arm_vehicle()
self.set_rc(3, 2000)
self.wait_altitude(20, 50, relative=True)
self.context_collect('STATUSTEXT')
self.change_mode('CRUISE')
self.set_rc(3, 1500)
self.wait_statustext('Transition started airspeed', check_context=True)
self.wait_statustext('Transition airspeed reached', check_context=True)
self.wait_statustext('Transition done', check_context=True)
self.delay_sim_time(5)
self.change_mode('QHOVER')
self.wait_airspeed(0, 5)
self.delay_sim_time(5)
mlog_path = self.current_onboard_log_filepath()
self.fly_home_land_and_disarm(timeout=600)
mlog = self.dfreader_for_path(mlog_path)
stage_require_fbwa = "require_fbwa"
stage_wait_qhover = "wait_qhover"
stage_verify_qhover_ff = "verify_qhover_ff"
stage_wait_cruise = "wait_cruise"
stage_cruise_wait_ff = "cruise_wait_ff"
stage_qhover2 = "qhover2"
stage_done = "done"
stage = stage_require_fbwa
msgs = {}
seen_flag_set_in_cruise = False
FF_BIT_MASK = (1 << 2)
while stage != stage_done:
m = mlog.recv_match()
if m is None:
raise NotAchievedException(f"Stuck in stage {stage}")
m_type = m.get_type()
msgs[m_type] = m
if stage == stage_require_fbwa:
if m_type == 'MODE':
if m.ModeNum == self.get_mode_from_mode_mapping('MANUAL'):
continue
fbwa_num = self.get_mode_from_mode_mapping('FBWA')
print(f"{m.ModeNum=} {fbwa_num=}")
if m.ModeNum != fbwa_num:
raise ValueError(f"wanted mode={fbwa_num} got={m.ModeNum}")
continue
if m_type == 'RFRN':
if not m.Flags & FF_BIT_MASK:
raise ValueError("Expected FF to be set in FBWA")
stage = stage_wait_qhover
continue
continue
if stage == stage_wait_qhover:
if m_type == 'MODE':
qhover_num = self.get_mode_from_mode_mapping('QHOVER')
print(f"want={qhover_num} got={m.ModeNum}")
if m.ModeNum == qhover_num:
stage = stage_verify_qhover_ff
continue
continue
continue
if stage == stage_verify_qhover_ff:
if m_type == 'RFRN':
if m.Flags & FF_BIT_MASK:
raise ValueError("Expected FF to be unset in QHOVER")
stage = stage_wait_cruise
continue
continue
if stage == stage_wait_cruise:
if m_type == 'MODE':
want_num = self.get_mode_from_mode_mapping('CRUISE')
if m.ModeNum == want_num:
stage = stage_cruise_wait_ff
cruise_wait_ff_start = msgs['ATT'].TimeUS*1e-6
continue
continue
continue
if stage == stage_cruise_wait_ff:
if m_type == 'MODE':
want_num = self.get_mode_from_mode_mapping('CRUISE')
if want_num != m.ModeNum:
if not seen_flag_set_in_cruise:
raise ValueError("Never saw FF get set")
if m.ModeNum == self.get_mode_from_mode_mapping('QHOVER'):
stage = stage_qhover2
continue
continue
if m_type == 'RFRN':
flag_set = m.Flags & FF_BIT_MASK
now = msgs['ATT'].TimeUS*1e-6
delta_t = now - cruise_wait_ff_start
if delta_t < 8:
if flag_set:
raise ValueError("Should not see bit set")
if delta_t > 10:
if not flag_set and not seen_flag_set_in_cruise:
raise ValueError("Should see bit set")
seen_flag_set_in_cruise = True
continue
continue
if stage == stage_qhover2:
'''bit should stay low for qhover 2'''
if m_type == 'RFRN':
flag_set = m.Flags & FF_BIT_MASK
if flag_set:
raise ValueError("ff should be low in qhover")
continue
if m_type == 'MODE':
if m.ModeNum != self.get_mode_from_mode_mapping('QHOVER'):
stage = stage_done
continue
continue
continue
raise NotAchievedException("Bad stage")
def RTL_AUTOLAND_1_FROM_GUIDED(self):
'''test behaviour when RTL_AUTOLAND==1 and entering from guided'''
self.set_parameters({
"RTL_AUTOLAND": 1,
})
wps = self.create_simple_relhome_mission([
(mavutil.mavlink.MAV_CMD_NAV_TAKEOFF, 0, 0, 30),
(mavutil.mavlink.MAV_CMD_NAV_WAYPOINT, 500, 0, 30),
(mavutil.mavlink.MAV_CMD_NAV_WAYPOINT, 1000, 0, 30),
self.create_MISSION_ITEM_INT(
mavutil.mavlink.MAV_CMD_DO_LAND_START,
),
(mavutil.mavlink.MAV_CMD_NAV_WAYPOINT, 0, -1000, 30),
(mavutil.mavlink.MAV_CMD_NAV_WAYPOINT, 0, -1300, 15),
(mavutil.mavlink.MAV_CMD_NAV_WAYPOINT, 0, -1600, 5),
(mavutil.mavlink.MAV_CMD_NAV_VTOL_LAND, 0, -1750, 0),
])
self.check_mission_upload_download(wps)
self.set_current_waypoint(0, check_afterwards=False)
self.change_mode('AUTO')
self.wait_ready_to_arm()
here = self.mav.location()
guided_loc = self.offset_location_ne(here, 500, -500)
self.arm_vehicle()
self.wait_current_waypoint(3)
self.reposition_to_loc(guided_loc)
self.send_cmd_do_set_mode('RTL')
self.wait_distance_to_home(100, 200, timeout=120)
self.wait_current_waypoint(7)
self.fly_home_land_and_disarm()
def WindEstimateConsistency(self):
'''test that DCM and EKF3 roughly agree on wind speed and direction'''
self.set_parameters({
'SIM_WIND_SPD': 10,
'SIM_WIND_DIR': 315,
})
self.change_mode('TAKEOFF')
self.wait_ready_to_arm()
self.arm_vehicle()
self.delay_sim_time(180)
mlog = self.dfreader_for_current_onboard_log()
self.fly_home_land_and_disarm()
self.progress("Inspecting dataflash log")
match_start_time = None
dcm = None
xkf2 = None
while True:
m = mlog.recv_match(
type=['DCM', 'XKF2'],
blocking=True,
)
if m is None:
raise NotAchievedException("Did not see wind estimates match")
m_type = m.get_type()
if m_type == 'DCM':
dcm = m
else:
xkf2 = m
if dcm is None or xkf2 is None:
continue
now = m.TimeUS * 1e-6
matches_east = abs(dcm.VWE-xkf2.VWE) < 1.5
matches_north = abs(dcm.VWN-xkf2.VWN) < 1.5
matches = matches_east and matches_north
if not matches:
match_start_time = None
continue
if match_start_time is None:
match_start_time = now
continue
if now - match_start_time > 60:
self.progress("Wind estimates correlated")
break
def QLoiterRecovery(self):
'''test QLOITER recovery from bad attitude'''
self.context_push()
self.install_example_script_context("sim_arming_pos.lua")
self.install_terrain_handlers_context()
self.set_parameters({
"SCR_ENABLE": 1,
"AHRS_EKF_TYPE": 10,
"EK3_ENABLE": 0,
"LOG_DISARMED": 1,
"Q_LAND_FINAL_SPD" : 2,
"HOME_RESET_ALT" : -1,
})
self.reboot_sitl(check_position=True)
self.context_collect('STATUSTEXT')
self.set_parameters({
"SIM_APOS_ENABLE" : 1,
"SIM_APOS_PIT" : -70,
"SIM_APOS_POS_D" : -200,
"SIM_APOS_POS_E" : 400,
"SIM_APOS_POS_N" : 200,
"SIM_APOS_RLL" : 150,
"SIM_APOS_VEL_X" : 40.0,
"SIM_APOS_VEL_Y" : 0.0,
"SIM_APOS_VEL_Z" : 0.0,
"SIM_APOS_YAW" : 250,
"SIM_APOS_GX" : 0,
"SIM_APOS_GY" : 0,
"SIM_APOS_GZ" : 0,
"SIM_APOS_MODE" : 19,
})
self.scripting_restart()
self.wait_text("Loaded arm pose", check_context=True)
self.wait_ready_to_arm()
self.set_rc(3, 2000)
self.set_parameter("Q_ASSIST_SPEED", 0)
NTESTS = 20
for t in range(NTESTS):
self.change_mode("FBWA")
self.delay_sim_time(3)
self.progress("Fast Recovery test %u" % t)
self.arm_vehicle(force=True)
self.wait_groundspeed(0, 2, timeout=15)
final_alt = self.assert_receive_message('TERRAIN_REPORT').current_height
if final_alt < 100:
raise NotAchievedException(f"Final alt {final_alt:.1f}")
self.disarm_vehicle(force=True)
self.progress("Setup for inverted slow recovery")
self.set_parameters({
"SIM_APOS_ENABLE" : 1,
"SIM_APOS_PIT" : 0,
"SIM_APOS_POS_D" : -200,
"SIM_APOS_POS_E" : 400,
"SIM_APOS_POS_N" : 200,
"SIM_APOS_RLL" : 180,
"SIM_APOS_VEL_X" : 0.0,
"SIM_APOS_VEL_Y" : 0.0,
"SIM_APOS_VEL_Z" : 0.0,
"SIM_APOS_GX" : 0,
"SIM_APOS_GY" : 0,
"SIM_APOS_GZ" : 0,
"SIM_APOS_MODE" : 19,
})
for t in range(NTESTS):
self.change_mode("FBWA")
self.delay_sim_time(3)
self.progress("Slow Recovery test %u" % t)
self.arm_vehicle(force=True)
self.wait_attitude(desroll=0, despitch=0, timeout=10, tolerance=5)
self.wait_groundspeed(0, 2, timeout=10)
final_alt = self.assert_receive_message('TERRAIN_REPORT').current_height
if final_alt < 100:
raise NotAchievedException(f"Final alt {final_alt:.1f}")
self.disarm_vehicle(force=True)
self.set_parameter("SIM_APOS_ENABLE", 0)
self.arm_vehicle(force=True)
self.change_mode("QLAND")
self.wait_disarmed(timeout=300)
self.context_pop()
def CruiseRecovery(self):
'''test QAssist recovery in CRUISE mode from bad attitude'''
self.context_push()
self.install_example_script_context("sim_arming_pos.lua")
self.install_terrain_handlers_context()
self.set_parameters({
"SCR_ENABLE": 1,
"AHRS_EKF_TYPE": 10,
"EK3_ENABLE": 0,
"LOG_DISARMED": 1,
"Q_LAND_FINAL_SPD" : 2,
"HOME_RESET_ALT" : -1,
})
self.reboot_sitl(check_position=True)
self.context_collect('STATUSTEXT')
self.set_parameters({
"SIM_APOS_ENABLE" : 1,
"SIM_APOS_PIT" : -70,
"SIM_APOS_POS_D" : -200,
"SIM_APOS_POS_E" : 400,
"SIM_APOS_POS_N" : 200,
"SIM_APOS_RLL" : 150,
"SIM_APOS_VEL_X" : 40.0,
"SIM_APOS_VEL_Y" : 0.0,
"SIM_APOS_VEL_Z" : 0.0,
"SIM_APOS_YAW" : 250,
"SIM_APOS_GX" : 0,
"SIM_APOS_GY" : 0,
"SIM_APOS_GZ" : 0,
"SIM_APOS_MODE" : 7,
})
self.scripting_restart()
self.wait_text("Loaded arm pose", check_context=True)
self.wait_ready_to_arm()
self.set_rc(3, 1500)
target_airspeed = self.get_parameter("AIRSPEED_CRUISE")
NTESTS = 20
for t in range(NTESTS):
self.change_mode("FBWA")
self.delay_sim_time(3)
self.progress("Fast CRUISE Recovery test %u" % t)
self.arm_vehicle(force=True)
self.delay_sim_time(3)
self.set_rc(2, 1600)
self.set_rc(2, 1500)
self.set_rc(1, 1600)
self.set_rc(1, 1500)
self.wait_attitude(desroll=0, despitch=0, timeout=10, tolerance=10)
self.wait_airspeed(target_airspeed-1, target_airspeed+1)
final_alt = self.assert_receive_message('TERRAIN_REPORT').current_height
if final_alt < 100:
raise NotAchievedException(f"Final alt {final_alt:.1f}")
self.disarm_vehicle(force=True)
self.progress("Setup for inverted slow recovery")
self.set_parameters({
"SIM_APOS_ENABLE" : 1,
"SIM_APOS_PIT" : 0,
"SIM_APOS_POS_D" : -200,
"SIM_APOS_POS_E" : 400,
"SIM_APOS_POS_N" : 200,
"SIM_APOS_RLL" : 180,
"SIM_APOS_VEL_X" : 0.0,
"SIM_APOS_VEL_Y" : 0.0,
"SIM_APOS_VEL_Z" : 0.0,
"SIM_APOS_GX" : 0,
"SIM_APOS_GY" : 0,
"SIM_APOS_GZ" : 0,
"SIM_APOS_MODE" : 7,
})
for t in range(NTESTS):
self.change_mode("FBWA")
self.delay_sim_time(3)
self.progress("Slow CRUISE Recovery test %u" % t)
self.arm_vehicle(force=True)
self.delay_sim_time(3)
self.set_rc(2, 1600)
self.set_rc(2, 1500)
self.set_rc(1, 1600)
self.set_rc(1, 1500)
self.wait_attitude(desroll=0, despitch=0, timeout=10, tolerance=10)
self.wait_airspeed(target_airspeed-1, target_airspeed+1)
final_alt = self.assert_receive_message('TERRAIN_REPORT').current_height
if final_alt < 100:
raise NotAchievedException(f"Final alt {final_alt:.1f}")
self.disarm_vehicle(force=True)
self.set_parameter("SIM_APOS_ENABLE", 0)
self.arm_vehicle(force=True)
self.change_mode("QLAND")
self.wait_disarmed(timeout=300)
self.context_pop()
def FastInvertedRecovery(self):
'''test recovery from inverted flight is fast'''
self.set_parameters({
"Q_A_ACCEL_R_MAX": 20000,
"Q_A_ACCEL_P_MAX": 20000,
"Q_A_ACCEL_Y_MAX": 20000,
"Q_A_RATE_R_MAX": 50,
"Q_A_RATE_P_MAX": 50,
"Q_A_RATE_Y_MAX": 50,
})
self.wait_ready_to_arm()
self.takeoff(60, mode='GUIDED', timeout=100)
self.context_collect('STATUSTEXT')
self.set_rc(3, 1500)
self.change_mode('CRUISE')
self.wait_statustext("Transition done", check_context=True)
self.progress("Go to inverted flight")
self.run_auxfunc(43, 2)
self.wait_roll(180, 3, absolute_value=True)
self.delay_sim_time(10)
initial_altitude = self.get_altitude(relative=True, timeout=2)
self.change_mode('QHOVER')
self.wait_roll(0, 3, absolute_value=True)
recovery_altitude = self.get_altitude(relative=True, timeout=2)
alt_change = initial_altitude - recovery_altitude
self.progress("Recovery AltChange %.1fm" % alt_change)
max_alt_change = 3
if alt_change > max_alt_change:
raise NotAchievedException("Recovery AltChange too high %.1f > %.1f" % (alt_change, max_alt_change))
self.fly_home_land_and_disarm()
def DoRepositionTerrain(self):
'''test handling of DO_REPOSITION with terrain alt'''
self.install_terrain_handlers_context()
self.start_subtest("test reposition with terrain alt")
self.wait_ready_to_arm()
dest = copy.copy(SITL_START_LOCATION)
dest.alt = 45
self.set_parameters({
'Q_GUIDED_MODE': 1,
})
self.takeoff(30, mode='GUIDED')
self.send_do_reposition(dest, frame=mavutil.mavlink.MAV_FRAME_GLOBAL_TERRAIN_ALT)
self.wait_location(
dest,
accuracy=200,
timeout=600,
height_accuracy=10,
)
self.delay_sim_time(20)
self.wait_altitude(
dest.alt-10,
dest.alt+10,
minimum_duration=10,
timeout=30,
relative=False,
altitude_source="TERRAIN_REPORT.current_height"
)
start_alt = self.assert_receive_message('TERRAIN_REPORT').current_height
terrain_height_min = start_alt
terrain_height_max = start_alt
def terrain_height_range(mav, m):
if m.get_type() == 'TERRAIN_REPORT':
nonlocal terrain_height_min, terrain_height_max
terrain_height_min = min(terrain_height_min, m.current_height)
terrain_height_max = max(terrain_height_max, m.current_height)
self.install_message_hook_context(terrain_height_range)
loc1 = copy.copy(dest)
self.location_offset_ne(loc1, -250, 0)
loc1.alt = 100
loc2 = copy.copy(dest)
self.location_offset_ne(loc2, 250, 0)
loc2.alt = 150
loc3 = copy.copy(loc2)
loc3.alt = 100
positions = [
("Loc1", loc1),
("Loc2", loc2),
("Loc1", loc1),
("Loc3", loc3),
("Loc1", loc1),
("Loc3", loc3),
("Loc1", loc1),
("Loc3", loc3),
("Loc2", loc2),
("Loc3", loc3),
("Loc2", loc2),
]
for (name, loc) in positions:
start_alt = self.assert_receive_message('TERRAIN_REPORT').current_height
terrain_height_min = start_alt
terrain_height_max = start_alt
self.progress(f"Flying to {name} at {loc.alt:.1f} from {start_alt:.1f}")
self.send_do_reposition(loc, frame=mavutil.mavlink.MAV_FRAME_GLOBAL_TERRAIN_ALT)
self.wait_location(
loc,
accuracy=10,
timeout=600,
height_accuracy=10,
)
self.delay_sim_time(10)
self.wait_altitude(
loc.alt-5,
loc.alt+5,
minimum_duration=10,
timeout=30,
relative=False,
altitude_source="TERRAIN_REPORT.current_height"
)
self.wait_groundspeed(0, 2)
self.wait_altitude(
loc.alt-5,
loc.alt+5,
minimum_duration=10,
timeout=30,
relative=False,
altitude_source="TERRAIN_REPORT.current_height"
)
min_alt_ok = min(start_alt, loc.alt) - 10
max_alt_ok = max(start_alt, loc.alt) + 10
self.progress(f"theight {terrain_height_min:.0f} to {terrain_height_max:.0f} accept {min_alt_ok:.0f}:{max_alt_ok:.0f}")
if terrain_height_min < min_alt_ok or terrain_height_max > max_alt_ok:
raise NotAchievedException(f"terrain range breach {start_alt:.1f} {terrain_height_min:.1f} {terrain_height_max:.1f}")
self.change_mode("QLAND")
self.mav.motors_disarmed_wait()
def DoRepositionTerrain2(self):
'''test handling of DO_REPOSITION terrain alt2'''
self.install_terrain_handlers_context()
self.start_subtest("test reposition terrain alt2")
takeoff_loc = mavutil.location(-35.28243788, 149.00502473, 583.7)
self.customise_SITL_commandline(
["--home", f"{takeoff_loc.lat},{takeoff_loc.lng},{takeoff_loc.alt},0"]
)
self.reboot_sitl(check_position=False)
self.wait_ready_to_arm()
dest = copy.copy(takeoff_loc)
dest.alt = 45
self.set_parameters({
'Q_GUIDED_MODE': 1,
})
self.takeoff(75, mode='GUIDED', timeout=60)
start_alt = self.assert_receive_message('TERRAIN_REPORT').current_height
terrain_height_min = start_alt
terrain_height_max = start_alt
def terrain_height_range(mav, m):
if m.get_type() == 'TERRAIN_REPORT':
nonlocal terrain_height_min, terrain_height_max
terrain_height_min = min(terrain_height_min, m.current_height)
terrain_height_max = max(terrain_height_max, m.current_height)
self.install_message_hook_context(terrain_height_range)
loc1 = mavutil.location(-35.27502040, 148.98635977, 75)
loc2 = mavutil.location(-35.28505202, 148.98604378, 75)
loc3 = mavutil.location(-35.27502040, 148.98635977, 120)
loc4 = mavutil.location(-35.28505202, 148.98604378, 120)
loc5 = mavutil.location(-35.28505202, 148.98604378, 100)
positions = [
("Loc1", loc1),
("Loc2", loc2),
("Loc1", loc1),
("Loc2", loc2),
("Loc1", loc1),
("Loc4", loc4),
("Loc3", loc3),
("Loc2", loc2),
("Loc5", loc5),
("Loc1", loc1),
("Loc5", loc5),
("Loc1", loc1),
]
for (name, loc) in positions:
start_alt = self.assert_receive_message('TERRAIN_REPORT').current_height
terrain_height_min = start_alt
terrain_height_max = start_alt
self.progress(f"Flying to {name} at {loc.alt:.1f} from {start_alt:.1f}")
self.send_do_reposition(loc, frame=mavutil.mavlink.MAV_FRAME_GLOBAL_TERRAIN_ALT)
self.wait_location(
loc,
accuracy=10,
timeout=600,
height_accuracy=10,
)
self.delay_sim_time(10)
self.wait_altitude(
loc.alt-5,
loc.alt+5,
minimum_duration=10,
timeout=30,
relative=False,
altitude_source="TERRAIN_REPORT.current_height"
)
self.wait_groundspeed(0, 2)
self.wait_altitude(
loc.alt-5,
loc.alt+5,
minimum_duration=10,
timeout=30,
relative=False,
altitude_source="TERRAIN_REPORT.current_height"
)
min_alt_ok = min(start_alt, loc.alt) - 10
max_alt_ok = max(start_alt, loc.alt) + 25
self.progress(f"theight {terrain_height_min:.0f} to {terrain_height_max:.0f} accept {min_alt_ok:.0f}:{max_alt_ok:.0f}")
if terrain_height_min < min_alt_ok or terrain_height_max > max_alt_ok:
raise NotAchievedException(f"terrain range breach {start_alt:.1f} {terrain_height_min:.1f} {terrain_height_max:.1f}")
self.change_mode("QLAND")
self.mav.motors_disarmed_wait()
self.reset_SITL_commandline()
def RudderArmedTakeoffRequiresNeutralThrottle(self):
'''check rudder must be neutral before VTOL takeoff allowed'''
self.upload_simple_relhome_mission([
(mavutil.mavlink.MAV_CMD_NAV_VTOL_TAKEOFF, 0, 0, 10),
(mavutil.mavlink.MAV_CMD_NAV_VTOL_LAND, 0, 0, 0),
])
self.upload_simple_relhome_mission
self.change_mode('AUTO')
self.wait_ready_to_arm()
self.set_rc(4, 2000)
self.wait_armed()
self.wait_altitude(-1, 1, relative=True, minimum_duration=10)
self.set_rc(4, 1500)
self.wait_altitude(5, 1000, relative=True)
self.zero_throttle()
self.wait_disarmed(timeout=60)
def RudderArmingWithARMING_CHECK_THROTTLEUnset(self) -> None:
'''check arming behaviour with ARMING_CHECK_THROTTLE unset'''
self.wait_ready_to_arm()
self.start_subtest("Should not be able to arm with mid-stick throttle")
self.set_rc(3, 1500)
self.set_rc(4, 2000)
w = vehicle_test_suite.WaitAndMaintainDisarmed(self, minimum_duration=10)
w.run()
self.set_rc(4, 1500)
self.disarm_vehicle()
self.clear_parameter_bit("RC_OPTIONS", 5)
self.start_subtest("Should be able to arm with mid-stick throttle")
self.set_rc(3, 1500)
self.set_rc(4, 2000)
self.wait_armed()
self.set_rc(4, 1500)
self.disarm_vehicle()
def ScriptedArmingChecksApplet(self):
""" Applet for Arming Checks will prevent a vehicle from arming based on scripted checks
"""
self.start_subtest("Scripted Arming Checks Applet validation")
self.context_collect("STATUSTEXT")
applet_script = "arming-checks.lua"
"""Initialize the FC"""
self.set_parameter("SCR_ENABLE", 1)
self.install_applet_script_context(applet_script)
self.reboot_sitl()
self.wait_ekf_happy()
self.wait_text("ArduPilot Ready", check_context=True)
self.wait_text("Arming Checks .* loaded", timeout=30, check_context=True, regex=True)
'''self.install_messageprinter_handlers_context(['PARAM_VALUE'])'''
self.start_subsubtest("ArmCk: Q_RTL_ALT must be legal")
self.set_parameter("SCALING_SPEED", 22)
self.set_parameter("Q_RTL_ALT", 150)
self.assert_prearm_failure("ArmCk: fail: Q_RTL_ALT too high", other_prearm_failures_fatal=False)
self.set_parameter("Q_RTL_ALT", 120)
self.wait_text("clear: Q_RTL_ALT", check_context=True)
self.start_subsubtest("ArmCk: Q_RTL vs QLand")
''' this is only a warning'''
self.set_parameter("Q_OPTIONS", 33)
self.wait_text("ArmCk: note: Q will RTL", check_context=True)
self.set_parameter("Q_OPTIONS", 1)
self.wait_text("ArmCk: note: Q will land", check_context=True)
def TerrainAvoidApplet(self):
'''Terrain Avoidance with CMTC'''
self.start_subtest("Terrain Avoidance Load and Start")
topofworld_loc = mavutil.location(64.1624778, -139.8402246, 1109.0)
self.customise_SITL_commandline(
["--home", f"{topofworld_loc.lat},{topofworld_loc.lng},{topofworld_loc.alt},0"]
)
self.context_collect("STATUSTEXT")
self.set_parameters({
"TERRAIN_ENABLE": 1,
"TERRAIN_SPACING": 30,
"TERRAIN_FOLLOW": 1,
"TERRAIN_OFS_MAX": 0,
})
self.install_terrain_handlers_context()
self.reboot_sitl(check_position=False)
self.set_parameters({
"SCR_ENABLE": 1,
"SIM_SPEEDUP": 20,
"RC7_OPTION": 305,
"RTL_AUTOLAND": 2,
"RNGFND1_TYPE": 100,
})
self.install_applet_script_context("quadplane_terrain_avoid.lua")
self.install_script_module(self.script_modules_source_path("mavlink_wrappers.lua"), "mavlink_wrappers.lua")
self.reboot_sitl(check_position=False)
self.wait_ready_to_arm()
self.wait_text("Terrain Avoid .* script loaded", regex=True, check_context=True)
self.set_parameters({
"TA_CMTC_ENABLE": 1,
"TA_CMTC_RAD": 80,
"TA_ALT_MAX": 250,
"WP_LOITER_RAD": 150,
"RNGFND1_SCALING": 10,
"RNGFND1_PIN": 0,
"RNGFND1_MAX": 100,
"SIM_SONAR_SCALE": 10,
})
filename = "TopOfTheWorldShort.waypoints"
self.progress("Flying mission %s" % filename)
num_wp = self.load_mission(filename)
self.progress("Mission items %d" % num_wp)
self.change_mode("AUTO")
loc = self.mav.location()
lng_int = int(loc.lng * 1e7)
lat_int = int(loc.lat * 1e7)
tstart = self.get_sim_time_cached()
last_terrain_report_pending = -1
while True:
now = self.get_sim_time_cached()
if now - tstart > 600:
raise NotAchievedException("Did not get correct terrain report")
self.mav.mav.terrain_check_send(lat_int, lng_int)
report = self.assert_receive_message('TERRAIN_REPORT', timeout=60)
self.progress(self.dump_message_verbose(report))
if report.spacing != 0:
break
if last_terrain_report_pending == -1:
last_terrain_report_pending = report.pending
elif report.pending < last_terrain_report_pending:
last_terrain_report_pending = report.pending
tstart = now
self.delay_sim_time(1)
self.progress(self.dump_message_verbose(report))
self.wait_ready_to_arm()
expected_terrain_height = 1101
if abs(report.terrain_height - expected_terrain_height) > 1.0:
raise NotAchievedException("Expected terrain height=%f got=%f" %
(expected_terrain_height, report.terrain_height))
self.set_rc(7, 1000)
self.wait_text("TerrAvoid: activated", check_context=True)
self.set_rc(7, 2000)
self.wait_text("TerrAvoid: deactivated", check_context=True)
self.set_rc(7, 1000)
self.wait_text("TerrAvoid: activated", check_context=True)
self.progress("TERRAIN_FOLLOW is %f" % self.get_parameter('TERRAIN_FOLLOW'))
self.progress("TERRAIN_LOOKAHD is %f" % self.get_parameter('TERRAIN_LOOKAHD'))
self.progress("TERRAIN_OFS_MAX is %f" % self.get_parameter('TERRAIN_OFS_MAX'))
self.progress("ROLL_LIMIT_DEG is %f" % self.get_parameter('ROLL_LIMIT_DEG'))
self.wait_ready_to_arm()
self.arm_vehicle()
self.wait_text("TerrAvoid: close to home", check_context=True)
self.wait_waypoint(2, 4, max_dist=100)
self.wait_text("TerrAvoid: away from home", check_context=True, regex=True)
self.wait_text("TerrAvoid: CMTC loiter left", check_context=True, regex=True)
self.progress("CMTC alt #1 is %f" % self.get_altitude(relative=False, timeout=2))
self.wait_altitude(1150, 1190, timeout=60, relative=False, minimum_duration=5)
self.wait_text("TerrAvoid: CMTC Done|TerrAvoid: Quading overrides CMTC", check_context=True, regex=True, timeout=60)
self.wait_text("TerrAvoid: CMTC loiter left", check_context=True, regex=True, timeout=60)
self.progress("CMTC alt #2 is %f" % self.get_altitude(relative=False, timeout=2))
self.wait_altitude(1070, 1180, timeout=60, relative=False, minimum_duration=5)
self.wait_text("TerrAvoid: CMTC Done|TerrAvoid: Quading overrides CMTC", check_context=True, regex=True)
self.wait_text("TerrAvoid: high terrain detected", check_context=True, regex=True, timeout=60)
self.wait_text("TerrAvoid: CMTC loiter left", check_context=True, regex=True)
self.progress("CMTC alt #3 is %f" % self.get_altitude(relative=False, timeout=2))
self.wait_altitude(1000, 1040, timeout=120, relative=False, minimum_duration=5)
self.wait_text("TerrAvoid: CMTC STOP", check_context=True, regex=True)
self.wait_text("TerrAvoid: CMTC Done", check_context=True, regex=True)
self.wait_text("TerrAvoid: CMTC loiter left", check_context=True, regex=True)
self.wait_text("TerrAvoid: CMTC STOP", check_context=True, regex=True)
self.wait_text("TerrAvoid: CMTC Done", check_context=True, regex=True)
self.wait_text("TerrAvoid: CMTC loiter left", check_context=True, regex=True)
self.wait_text("TerrAvoid: CMTC STOP", check_context=True, regex=True)
self.wait_text("TerrAvoid: CMTC Done", check_context=True, regex=True)
self.wait_text("TerrAvoid: high terrain detected", check_context=True, regex=True, timeout=60)
self.wait_text("TerrAvoid: CMTC loiter left", check_context=True, regex=True)
self.progress("CMTC alt #4 is %f" % self.get_altitude(relative=False, timeout=2))
self.wait_text("TerrAvoid: high terrain detected", check_context=True, regex=True, timeout=60)
self.wait_text("TerrAvoid: CMTC loiter left", check_context=True, regex=True)
self.progress("CMTC alt #6 is %f" % self.get_altitude(relative=False, timeout=2))
self.wait_text("TerrAvoid: CMTC STOP", check_context=True, regex=True)
self.wait_text("TerrAvoid: CMTC Done", check_context=True, regex=True)
self.progress("alt is %f" % self.get_altitude(relative=False, timeout=2))
self.wait_text("TerrAvoid: CMTC STOP", check_context=True, regex=True)
self.wait_text("TerrAvoid: CMTC Done", check_context=True, regex=True)
self.progress("#Pitching alt is %f" % self.get_altitude(relative=False, timeout=2))
self.wait_text("TerrAvoid: Pitching Started", check_context=True, regex=True, timeout=600)
self.wait_text("TerrAvoid: Terrain Ok", check_context=True, regex=True, timeout=60)
self.wait_text("TerrAvoid: CMTC loiter left", check_context=True, regex=True)
self.progress("CMTC alt #7 is %f" % self.get_altitude(relative=False, timeout=2))
self.wait_text("TerrAvoid: Pitching DONE", check_context=True, regex=True)
self.progress("#Pitching DONE alt is %f" % self.get_altitude(relative=False, timeout=2))
self.wait_altitude(1140, 1200, timeout=120, relative=False, minimum_duration=5)
self.wait_text("TerrAvoid: CMTC STOP", check_context=True, regex=True)
self.wait_text("TerrAvoid: CMTC Done", check_context=True, regex=True)
self.wait_text("TerrAvoid: CMTC Done", check_context=True, regex=True)
self.wait_text("TerrAvoid: Pitching started", check_context=True, regex=True, timeout=120)
self.progress("Pitching alt #1 is %f" % self.get_altitude(relative=False, timeout=2))
self.wait_text("TerrAvoid: Pitching DONE", check_context=True, regex=True)
self.progress("Pitching alt #2 is %f" % self.get_altitude(relative=False, timeout=2))
self.wait_text("TerrAvoid: CMTC Done", check_context=True, regex=True)
self.wait_statustext('Land complete', timeout=600)
self.wait_disarmed(timeout=120)
self.customise_SITL_commandline(
["--home", "-27.274439,151.290064,343.0,0"]
)
self.reboot_sitl()
self.remove_installed_script_module("mavlink_wrappers.lua")
def TakeoffCheck(self):
'''Test takeoff check - auto mode'''
self.set_parameters({
"AHRS_EKF_TYPE": 10,
'SIM_ESC_TELEM': 1,
})
self.start_subtest("Test blocking doesn't occur with in-range RPM")
self.context_push()
self.set_parameters({
'SIM_VIB_MOT_MAX': 150,
'SIM_ESC_ARM_RPM': 1000,
'Q_TKOFF_RPM_MIN': 900,
'Q_TKOFF_RPM_MAX': 1100,
})
self.upload_simple_relhome_mission([
(mavutil.mavlink.MAV_CMD_NAV_VTOL_TAKEOFF, 0, 0, 1),
(mavutil.mavlink.MAV_CMD_NAV_VTOL_LAND, 0, 0, 0),
])
self.change_mode('AUTO')
self.wait_ready_to_arm()
self.arm_vehicle()
self.wait_current_waypoint(2)
self.wait_disarmed()
self.set_current_waypoint(0, check_afterwards=False)
self.context_pop()
self.start_subtest("Ensure blocked if motors don't spool up")
self.context_push()
self.set_parameters({
'SIM_ESC_ARM_RPM': 500,
'Q_TKOFF_RPM_MIN': 1000,
})
self.upload_simple_relhome_mission([
(mavutil.mavlink.MAV_CMD_NAV_TAKEOFF, 0, 0, 30),
(mavutil.mavlink.MAV_CMD_NAV_RETURN_TO_LAUNCH, 0, 0, 0),
])
self.test_takeoff_check_mode("AUTO", force_disarm=True)
self.context_pop()
self.start_subtest("Ensure blocked if virtual motors are missing virtual props")
self.context_push()
self.set_parameters({
'Q_TKOFF_RPM_MIN': 1,
'Q_TKOFF_RPM_MAX': 3,
})
self.test_takeoff_check_mode("AUTO", force_disarm=True)
self.context_pop()
def tests(self):
'''return list of all tests'''
ret = super(AutoTestQuadPlane, self).tests()
ret.extend([
self.FwdThrInVTOL,
self.AirMode,
self.TestMotorMask,
self.PilotYaw,
self.ParameterChecks,
self.QAUTOTUNE,
self.TestLogDownload,
self.TestLogDownloadWrap,
self.EXTENDED_SYS_STATE,
self.Mission,
self.Weathervane,
self.QAssist,
self.GyroFFT,
self.Tailsitter,
self.CopterTailsitter,
self.ICEngine,
self.ICEngineMission,
self.ICEngineRPMGovernor,
self.MAV_CMD_DO_ENGINE_CONTROL,
self.MidAirDisarmDisallowed,
self.GUIDEDToAUTO,
self.BootInAUTO,
self.Ship,
self.WindEstimateConsistency,
self.MAV_CMD_NAV_LOITER_TO_ALT,
self.LoiterAltQLand,
self.VTOLLandSpiral,
self.VTOLQuicktune,
self.VTOLQuicktune_CPP,
self.PrecisionLanding,
self.ShipLanding,
Test(self.MotorTest, kwargs={
"mot1_servo_chan": 8,
"mot4_servo_chan": 6,
"wait_finish_text": False,
"quadplane": True,
}),
self.RCDisableAirspeedUse,
self.mission_MAV_CMD_DO_VTOL_TRANSITION,
self.mavlink_MAV_CMD_DO_VTOL_TRANSITION,
self.TransitionMinThrottle,
self.BackTransitionMinThrottle,
self.MAV_CMD_NAV_TAKEOFF,
self.Q_GUIDED_MODE,
self.DCMClimbRate,
self.RTL_AUTOLAND_1,
self.RTL_AUTOLAND_1_FROM_GUIDED,
self.AHRSFlyForwardFlag,
self.DoRepositionTerrain,
self.DoRepositionTerrain2,
self.QLoiterRecovery,
self.FastInvertedRecovery,
self.CruiseRecovery,
self.RudderArmedTakeoffRequiresNeutralThrottle,
self.RudderArmingWithARMING_CHECK_THROTTLEUnset,
self.ScriptedArmingChecksApplet,
self.TerrainAvoidApplet,
self.TakeoffCheck,
])
return ret