blob: a1b6121038081dfd1a3001d8187289f256568711 [file] [log] [blame]
James Kuszmaul61750662021-06-21 21:32:33 -07001#include "y2020/control_loops/drivetrain/localizer.h"
2
James Kuszmaul5398fae2020-02-17 16:44:03 -08003#include <queue>
4
Austin Schuhb06f03b2021-02-17 22:00:37 -08005#include "aos/events/logging/log_writer.h"
James Kuszmaul958b21e2020-02-26 21:51:40 -08006#include "aos/network/message_bridge_server_generated.h"
James Kuszmaul5398fae2020-02-17 16:44:03 -08007#include "aos/network/team_number.h"
Austin Schuh87dd3832021-01-01 23:07:31 -08008#include "aos/network/testing_time_converter.h"
James Kuszmaul61750662021-06-21 21:32:33 -07009#include "frc971/control_loops/control_loop_test.h"
James Kuszmaul5398fae2020-02-17 16:44:03 -080010#include "frc971/control_loops/drivetrain/drivetrain.h"
11#include "frc971/control_loops/drivetrain/drivetrain_test_lib.h"
12#include "frc971/control_loops/team_number_test_environment.h"
James Kuszmaul61750662021-06-21 21:32:33 -070013#include "gtest/gtest.h"
James Kuszmaul5398fae2020-02-17 16:44:03 -080014#include "y2020/control_loops/drivetrain/drivetrain_base.h"
James Kuszmaul688a2b02020-02-19 18:26:33 -080015#include "y2020/control_loops/superstructure/superstructure_status_generated.h"
James Kuszmaul5398fae2020-02-17 16:44:03 -080016
17DEFINE_string(output_file, "",
18 "If set, logs all channels to the provided logfile.");
19
20// This file tests that the full 2020 localizer behaves sanely.
21
22namespace y2020 {
23namespace control_loops {
24namespace drivetrain {
25namespace testing {
26
Austin Schuh66168842021-08-17 19:42:21 -070027using aos::logger::BootTimestamp;
James Kuszmaul5398fae2020-02-17 16:44:03 -080028using frc971::control_loops::drivetrain::DrivetrainConfig;
29using frc971::control_loops::drivetrain::Goal;
30using frc971::control_loops::drivetrain::LocalizerControl;
Brian Silverman1f345222020-09-24 21:14:48 -070031using frc971::vision::sift::CameraCalibrationT;
32using frc971::vision::sift::CameraPoseT;
James Kuszmaul5398fae2020-02-17 16:44:03 -080033using frc971::vision::sift::ImageMatchResult;
34using frc971::vision::sift::ImageMatchResultT;
James Kuszmaul5398fae2020-02-17 16:44:03 -080035using frc971::vision::sift::TransformationMatrixT;
36
37namespace {
38DrivetrainConfig<double> GetTest2020DrivetrainConfig() {
39 DrivetrainConfig<double> config = GetDrivetrainConfig();
40 return config;
41}
42
43// Copies an Eigen matrix into a row-major vector of the data.
44std::vector<float> MatrixToVector(const Eigen::Matrix<double, 4, 4> &H) {
45 std::vector<float> data;
46 for (int row = 0; row < 4; ++row) {
47 for (int col = 0; col < 4; ++col) {
48 data.push_back(H(row, col));
49 }
50 }
51 return data;
52}
53
54// Provides the location of the turret to use for simulation. Mostly we care
55// about providing a location that is not perfectly aligned with the robot's
56// origin.
57Eigen::Matrix<double, 4, 4> TurretRobotTransformation() {
58 Eigen::Matrix<double, 4, 4> H;
59 H.setIdentity();
James Kuszmaul688a2b02020-02-19 18:26:33 -080060 H.block<3, 1>(0, 3) << 1, 1.1, 0.9;
James Kuszmaul5398fae2020-02-17 16:44:03 -080061 return H;
62}
63
64// Provides the location of the camera on the turret.
65// TODO(james): Also simulate a fixed camera that is *not* on the turret.
66Eigen::Matrix<double, 4, 4> CameraTurretTransformation() {
67 Eigen::Matrix<double, 4, 4> H;
68 H.setIdentity();
69 H.block<3, 1>(0, 3) << 0.1, 0, 0;
70 // Introduce a bit of pitch to make sure that we're exercising all the code.
71 H.block<3, 3>(0, 0) =
James Kuszmaul688a2b02020-02-19 18:26:33 -080072 Eigen::AngleAxis<double>(0.1, Eigen::Vector3d::UnitY()) *
James Kuszmaul5398fae2020-02-17 16:44:03 -080073 H.block<3, 3>(0, 0);
74 return H;
75}
76
77// The absolute target location to use. Not meant to correspond with a
78// particular field target.
79// TODO(james): Make more targets.
James Kuszmaul688a2b02020-02-19 18:26:33 -080080std::vector<Eigen::Matrix<double, 4, 4>> TargetLocations() {
81 std::vector<Eigen::Matrix<double, 4, 4>> locations;
James Kuszmaul5398fae2020-02-17 16:44:03 -080082 Eigen::Matrix<double, 4, 4> H;
83 H.setIdentity();
84 H.block<3, 1>(0, 3) << 10.0, 0, 0;
James Kuszmaul688a2b02020-02-19 18:26:33 -080085 locations.push_back(H);
86 H.block<3, 1>(0, 3) << -10.0, 0, 0;
87 locations.push_back(H);
88 return locations;
James Kuszmaul5398fae2020-02-17 16:44:03 -080089}
James Kuszmaul958b21e2020-02-26 21:51:40 -080090
Austin Schuhbe69cf32020-08-27 11:38:33 -070091constexpr std::chrono::seconds kPiTimeOffset(-10);
James Kuszmaul5398fae2020-02-17 16:44:03 -080092} // namespace
93
94namespace chrono = std::chrono;
James Kuszmaul5398fae2020-02-17 16:44:03 -080095using aos::monotonic_clock;
Brian Silverman1f345222020-09-24 21:14:48 -070096using frc971::control_loops::drivetrain::DrivetrainLoop;
97using frc971::control_loops::drivetrain::testing::DrivetrainSimulation;
James Kuszmaul5398fae2020-02-17 16:44:03 -080098
James Kuszmaul61750662021-06-21 21:32:33 -070099class LocalizedDrivetrainTest : public frc971::testing::ControlLoopTest {
James Kuszmaul5398fae2020-02-17 16:44:03 -0800100 protected:
101 // We must use the 2020 drivetrain config so that we don't have to deal
102 // with shifting:
103 LocalizedDrivetrainTest()
James Kuszmaul61750662021-06-21 21:32:33 -0700104 : frc971::testing::ControlLoopTest(
James Kuszmaul5398fae2020-02-17 16:44:03 -0800105 aos::configuration::ReadConfig(
106 "y2020/control_loops/drivetrain/simulation_config.json"),
107 GetTest2020DrivetrainConfig().dt),
Austin Schuh87dd3832021-01-01 23:07:31 -0800108 time_converter_(aos::configuration::NodesCount(configuration())),
Austin Schuh6aa77be2020-02-22 21:06:40 -0800109 roborio_(aos::configuration::GetNode(configuration(), "roborio")),
110 pi1_(aos::configuration::GetNode(configuration(), "pi1")),
111 test_event_loop_(MakeEventLoop("test", roborio_)),
James Kuszmaul5398fae2020-02-17 16:44:03 -0800112 drivetrain_goal_sender_(
113 test_event_loop_->MakeSender<Goal>("/drivetrain")),
114 drivetrain_goal_fetcher_(
115 test_event_loop_->MakeFetcher<Goal>("/drivetrain")),
116 localizer_control_sender_(
117 test_event_loop_->MakeSender<LocalizerControl>("/drivetrain")),
James Kuszmaul688a2b02020-02-19 18:26:33 -0800118 superstructure_status_sender_(
119 test_event_loop_->MakeSender<superstructure::Status>(
120 "/superstructure")),
James Kuszmaul958b21e2020-02-26 21:51:40 -0800121 server_statistics_sender_(
122 test_event_loop_->MakeSender<aos::message_bridge::ServerStatistics>(
123 "/aos")),
Austin Schuh6aa77be2020-02-22 21:06:40 -0800124 drivetrain_event_loop_(MakeEventLoop("drivetrain", roborio_)),
James Kuszmaul5398fae2020-02-17 16:44:03 -0800125 dt_config_(GetTest2020DrivetrainConfig()),
Austin Schuh6aa77be2020-02-22 21:06:40 -0800126 pi1_event_loop_(MakeEventLoop("test", pi1_)),
James Kuszmaul5398fae2020-02-17 16:44:03 -0800127 camera_sender_(
Austin Schuh6aa77be2020-02-22 21:06:40 -0800128 pi1_event_loop_->MakeSender<ImageMatchResult>("/pi1/camera")),
James Kuszmaul5398fae2020-02-17 16:44:03 -0800129 localizer_(drivetrain_event_loop_.get(), dt_config_),
130 drivetrain_(dt_config_, drivetrain_event_loop_.get(), &localizer_),
Austin Schuh6aa77be2020-02-22 21:06:40 -0800131 drivetrain_plant_event_loop_(MakeEventLoop("plant", roborio_)),
James Kuszmaul5398fae2020-02-17 16:44:03 -0800132 drivetrain_plant_(drivetrain_plant_event_loop_.get(), dt_config_),
133 last_frame_(monotonic_now()) {
Austin Schuh87dd3832021-01-01 23:07:31 -0800134 event_loop_factory()->SetTimeConverter(&time_converter_);
James Kuszmaul9c128122021-03-22 22:24:36 -0700135 CHECK_EQ(aos::configuration::GetNodeIndex(configuration(), roborio_), 6);
Austin Schuh87dd3832021-01-01 23:07:31 -0800136 CHECK_EQ(aos::configuration::GetNodeIndex(configuration(), pi1_), 1);
Austin Schuh66168842021-08-17 19:42:21 -0700137 time_converter_.AddMonotonic({BootTimestamp::epoch() + kPiTimeOffset,
138 BootTimestamp::epoch() + kPiTimeOffset,
139 BootTimestamp::epoch() + kPiTimeOffset,
140 BootTimestamp::epoch() + kPiTimeOffset,
141 BootTimestamp::epoch() + kPiTimeOffset,
142 BootTimestamp::epoch() + kPiTimeOffset,
143 BootTimestamp::epoch()});
James Kuszmaul5398fae2020-02-17 16:44:03 -0800144 set_team_id(frc971::control_loops::testing::kTeamNumber);
James Kuszmaul5398fae2020-02-17 16:44:03 -0800145 set_battery_voltage(12.0);
146
147 if (!FLAGS_output_file.empty()) {
Austin Schuh6aa77be2020-02-22 21:06:40 -0800148 logger_event_loop_ = MakeEventLoop("logger", roborio_);
Brian Silverman1f345222020-09-24 21:14:48 -0700149 logger_ = std::make_unique<aos::logger::Logger>(logger_event_loop_.get());
150 logger_->StartLoggingLocalNamerOnRun(FLAGS_output_file);
James Kuszmaul5398fae2020-02-17 16:44:03 -0800151 }
152
153 test_event_loop_->MakeWatcher(
154 "/drivetrain",
155 [this](const frc971::control_loops::drivetrain::Status &) {
156 // Needs to do camera updates right after we run the control loop.
157 if (enable_cameras_) {
158 SendDelayedFrames();
159 if (last_frame_ + std::chrono::milliseconds(100) <
160 monotonic_now()) {
161 CaptureFrames();
162 last_frame_ = monotonic_now();
163 }
164 }
Brian Silverman1f345222020-09-24 21:14:48 -0700165 });
James Kuszmaul688a2b02020-02-19 18:26:33 -0800166
167 test_event_loop_->AddPhasedLoop(
168 [this](int) {
James Kuszmaul958b21e2020-02-26 21:51:40 -0800169 auto builder = server_statistics_sender_.MakeBuilder();
170 auto name_offset = builder.fbb()->CreateString("pi1");
171 auto node_builder = builder.MakeBuilder<aos::Node>();
172 node_builder.add_name(name_offset);
173 auto node_offset = node_builder.Finish();
174 auto connection_builder =
175 builder.MakeBuilder<aos::message_bridge::ServerConnection>();
176 connection_builder.add_node(node_offset);
177 connection_builder.add_monotonic_offset(
Austin Schuhbe69cf32020-08-27 11:38:33 -0700178 chrono::duration_cast<chrono::nanoseconds>(kPiTimeOffset)
James Kuszmaul958b21e2020-02-26 21:51:40 -0800179 .count());
180 auto connection_offset = connection_builder.Finish();
181 auto connections_offset =
182 builder.fbb()->CreateVector(&connection_offset, 1);
183 auto statistics_builder =
184 builder.MakeBuilder<aos::message_bridge::ServerStatistics>();
185 statistics_builder.add_connections(connections_offset);
186 builder.Send(statistics_builder.Finish());
187 },
188 chrono::milliseconds(500));
189
190 test_event_loop_->AddPhasedLoop(
191 [this](int) {
James Kuszmaul688a2b02020-02-19 18:26:33 -0800192 // Also use the opportunity to send out turret messages.
193 UpdateTurretPosition();
194 auto builder = superstructure_status_sender_.MakeBuilder();
195 auto turret_builder =
196 builder
197 .MakeBuilder<frc971::control_loops::
198 PotAndAbsoluteEncoderProfiledJointStatus>();
199 turret_builder.add_position(turret_position_);
200 turret_builder.add_velocity(turret_velocity_);
201 auto turret_offset = turret_builder.Finish();
202 auto status_builder = builder.MakeBuilder<superstructure::Status>();
203 status_builder.add_turret(turret_offset);
204 builder.Send(status_builder.Finish());
205 },
206 chrono::milliseconds(5));
James Kuszmaul5398fae2020-02-17 16:44:03 -0800207
James Kuszmaul286b4282020-02-26 20:29:32 -0800208 test_event_loop_->OnRun([this]() { SetStartingPosition({3.0, 2.0, 0.0}); });
209
James Kuszmaul5398fae2020-02-17 16:44:03 -0800210 // Run for enough time to allow the gyro/imu zeroing code to run.
211 RunFor(std::chrono::seconds(10));
212 }
213
214 virtual ~LocalizedDrivetrainTest() override {}
215
216 void SetStartingPosition(const Eigen::Matrix<double, 3, 1> &xytheta) {
217 *drivetrain_plant_.mutable_state() << xytheta.x(), xytheta.y(),
218 xytheta(2, 0), 0.0, 0.0;
James Kuszmaulbcd96fc2020-10-12 20:29:32 -0700219 Eigen::Matrix<double, Localizer::HybridEkf::kNStates, 1> localizer_state;
James Kuszmaul5398fae2020-02-17 16:44:03 -0800220 localizer_state.setZero();
James Kuszmaulbcd96fc2020-10-12 20:29:32 -0700221 localizer_state.block<3, 1>(0, 0) = xytheta;
James Kuszmaul5398fae2020-02-17 16:44:03 -0800222 localizer_.Reset(monotonic_now(), localizer_state);
223 }
224
225 void VerifyNearGoal(double eps = 1e-3) {
226 drivetrain_goal_fetcher_.Fetch();
227 EXPECT_NEAR(drivetrain_goal_fetcher_->left_goal(),
228 drivetrain_plant_.GetLeftPosition(), eps);
229 EXPECT_NEAR(drivetrain_goal_fetcher_->right_goal(),
230 drivetrain_plant_.GetRightPosition(), eps);
231 }
232
James Kuszmaul688a2b02020-02-19 18:26:33 -0800233 ::testing::AssertionResult IsNear(double expected, double actual,
234 double epsilon) {
235 if (std::abs(expected - actual) < epsilon) {
236 return ::testing::AssertionSuccess();
237 } else {
238 return ::testing::AssertionFailure()
239 << "Expected " << expected << " but got " << actual
240 << " with a max difference of " << epsilon
241 << " and an actual difference of " << std::abs(expected - actual);
242 }
243 }
244 ::testing::AssertionResult VerifyEstimatorAccurate(double eps) {
James Kuszmaul5398fae2020-02-17 16:44:03 -0800245 const Eigen::Matrix<double, 5, 1> true_state = drivetrain_plant_.state();
James Kuszmaul688a2b02020-02-19 18:26:33 -0800246 ::testing::AssertionResult result(true);
247 if (!(result = IsNear(localizer_.x(), true_state(0), eps))) {
248 return result;
249 }
250 if (!(result = IsNear(localizer_.y(), true_state(1), eps))) {
251 return result;
252 }
253 if (!(result = IsNear(localizer_.theta(), true_state(2), eps))) {
254 return result;
255 }
256 if (!(result = IsNear(localizer_.left_velocity(), true_state(3), eps))) {
257 return result;
258 }
259 if (!(result = IsNear(localizer_.right_velocity(), true_state(4), eps))) {
260 return result;
261 }
262 return result;
James Kuszmaul5398fae2020-02-17 16:44:03 -0800263 }
264
265 // Goes through and captures frames on the camera(s), queueing them up to be
266 // sent by SendDelayedFrames().
267 void CaptureFrames() {
268 const frc971::control_loops::Pose robot_pose(
269 {drivetrain_plant_.GetPosition().x(),
270 drivetrain_plant_.GetPosition().y(), 0.0},
271 drivetrain_plant_.state()(2, 0));
272 std::unique_ptr<ImageMatchResultT> frame(new ImageMatchResultT());
273
James Kuszmaulc51dbfe2020-02-23 15:39:00 -0800274 for (const auto &H_field_target : TargetLocations()) {
James Kuszmaul5398fae2020-02-17 16:44:03 -0800275 std::unique_ptr<CameraPoseT> camera_target(new CameraPoseT());
276
277 camera_target->field_to_target.reset(new TransformationMatrixT());
James Kuszmaulc51dbfe2020-02-23 15:39:00 -0800278 camera_target->field_to_target->data = MatrixToVector(H_field_target);
James Kuszmaul688a2b02020-02-19 18:26:33 -0800279
James Kuszmaulc51dbfe2020-02-23 15:39:00 -0800280 Eigen::Matrix<double, 4, 4> H_turret_camera =
James Kuszmaul688a2b02020-02-19 18:26:33 -0800281 Eigen::Matrix<double, 4, 4>::Identity();
282 if (is_turreted_) {
James Kuszmaulc51dbfe2020-02-23 15:39:00 -0800283 H_turret_camera = frc971::control_loops::TransformationMatrixForYaw(
James Kuszmaul688a2b02020-02-19 18:26:33 -0800284 turret_position_) *
285 CameraTurretTransformation();
286 }
James Kuszmaul5398fae2020-02-17 16:44:03 -0800287
288 // TODO(james): Use non-zero turret angles.
289 camera_target->camera_to_target.reset(new TransformationMatrixT());
James Kuszmaulaca88782021-04-24 16:48:45 -0700290 camera_target->camera_to_target->data = MatrixToVector(
291 (robot_pose.AsTransformationMatrix() * TurretRobotTransformation() *
292 H_turret_camera * camera_calibration_offset_)
293 .inverse() *
294 H_field_target);
James Kuszmaul5398fae2020-02-17 16:44:03 -0800295
296 frame->camera_poses.emplace_back(std::move(camera_target));
297 }
298
299 frame->image_monotonic_timestamp_ns =
300 chrono::duration_cast<chrono::nanoseconds>(
James Kuszmaul958b21e2020-02-26 21:51:40 -0800301 event_loop_factory()
302 ->GetNodeEventLoopFactory(pi1_)
303 ->monotonic_now()
304 .time_since_epoch())
James Kuszmaul5398fae2020-02-17 16:44:03 -0800305 .count();
306 frame->camera_calibration.reset(new CameraCalibrationT());
307 {
308 frame->camera_calibration->fixed_extrinsics.reset(
309 new TransformationMatrixT());
James Kuszmaulc51dbfe2020-02-23 15:39:00 -0800310 TransformationMatrixT *H_robot_turret =
James Kuszmaul5398fae2020-02-17 16:44:03 -0800311 frame->camera_calibration->fixed_extrinsics.get();
James Kuszmaulc51dbfe2020-02-23 15:39:00 -0800312 H_robot_turret->data = MatrixToVector(TurretRobotTransformation());
James Kuszmaul5398fae2020-02-17 16:44:03 -0800313 }
James Kuszmaul688a2b02020-02-19 18:26:33 -0800314
315 if (is_turreted_) {
James Kuszmaul5398fae2020-02-17 16:44:03 -0800316 frame->camera_calibration->turret_extrinsics.reset(
317 new TransformationMatrixT());
James Kuszmaulc51dbfe2020-02-23 15:39:00 -0800318 TransformationMatrixT *H_turret_camera =
James Kuszmaul5398fae2020-02-17 16:44:03 -0800319 frame->camera_calibration->turret_extrinsics.get();
James Kuszmaulc51dbfe2020-02-23 15:39:00 -0800320 H_turret_camera->data = MatrixToVector(CameraTurretTransformation());
James Kuszmaul5398fae2020-02-17 16:44:03 -0800321 }
322
323 camera_delay_queue_.emplace(monotonic_now(), std::move(frame));
324 }
325
326 // Actually sends out all the camera frames.
327 void SendDelayedFrames() {
328 const std::chrono::milliseconds camera_latency(150);
329 while (!camera_delay_queue_.empty() &&
330 std::get<0>(camera_delay_queue_.front()) <
331 monotonic_now() - camera_latency) {
332 auto builder = camera_sender_.MakeBuilder();
333 ASSERT_TRUE(builder.Send(ImageMatchResult::Pack(
334 *builder.fbb(), std::get<1>(camera_delay_queue_.front()).get())));
335 camera_delay_queue_.pop();
336 }
337 }
338
Austin Schuh87dd3832021-01-01 23:07:31 -0800339 aos::message_bridge::TestingTimeConverter time_converter_;
340
Austin Schuh6aa77be2020-02-22 21:06:40 -0800341 const aos::Node *const roborio_;
342 const aos::Node *const pi1_;
343
James Kuszmaul5398fae2020-02-17 16:44:03 -0800344 std::unique_ptr<aos::EventLoop> test_event_loop_;
345 aos::Sender<Goal> drivetrain_goal_sender_;
346 aos::Fetcher<Goal> drivetrain_goal_fetcher_;
347 aos::Sender<LocalizerControl> localizer_control_sender_;
James Kuszmaul688a2b02020-02-19 18:26:33 -0800348 aos::Sender<superstructure::Status> superstructure_status_sender_;
James Kuszmaul958b21e2020-02-26 21:51:40 -0800349 aos::Sender<aos::message_bridge::ServerStatistics> server_statistics_sender_;
James Kuszmaul5398fae2020-02-17 16:44:03 -0800350
351 std::unique_ptr<aos::EventLoop> drivetrain_event_loop_;
Brian Silverman1f345222020-09-24 21:14:48 -0700352 const frc971::control_loops::drivetrain::DrivetrainConfig<double> dt_config_;
James Kuszmaul5398fae2020-02-17 16:44:03 -0800353
Austin Schuh6aa77be2020-02-22 21:06:40 -0800354 std::unique_ptr<aos::EventLoop> pi1_event_loop_;
James Kuszmaul5398fae2020-02-17 16:44:03 -0800355 aos::Sender<ImageMatchResult> camera_sender_;
356
357 Localizer localizer_;
358 DrivetrainLoop drivetrain_;
359
360 std::unique_ptr<aos::EventLoop> drivetrain_plant_event_loop_;
361 DrivetrainSimulation drivetrain_plant_;
362 monotonic_clock::time_point last_frame_;
363
364 // A queue of camera frames so that we can add a time delay to the data
365 // coming from the cameras.
366 std::queue<std::tuple<aos::monotonic_clock::time_point,
367 std::unique_ptr<ImageMatchResultT>>>
368 camera_delay_queue_;
369
James Kuszmaulaca88782021-04-24 16:48:45 -0700370 // Offset to add to the camera for actually taking the images, to simulate an
371 // inaccurate calibration.
372 Eigen::Matrix<double, 4, 4> camera_calibration_offset_ =
373 Eigen::Matrix<double, 4, 4>::Identity();
374
James Kuszmaul5398fae2020-02-17 16:44:03 -0800375 void set_enable_cameras(bool enable) { enable_cameras_ = enable; }
James Kuszmaul688a2b02020-02-19 18:26:33 -0800376 void set_camera_is_turreted(bool turreted) { is_turreted_ = turreted; }
377
378 void set_turret(double position, double velocity) {
379 turret_position_ = position;
380 turret_velocity_ = velocity;
381 }
382
383 void SendGoal(double left, double right) {
384 auto builder = drivetrain_goal_sender_.MakeBuilder();
385
386 Goal::Builder drivetrain_builder = builder.MakeBuilder<Goal>();
387 drivetrain_builder.add_controller_type(
388 frc971::control_loops::drivetrain::ControllerType::MOTION_PROFILE);
389 drivetrain_builder.add_left_goal(left);
390 drivetrain_builder.add_right_goal(right);
391
392 EXPECT_TRUE(builder.Send(drivetrain_builder.Finish()));
393 }
James Kuszmaul5398fae2020-02-17 16:44:03 -0800394
395 private:
James Kuszmaul688a2b02020-02-19 18:26:33 -0800396 void UpdateTurretPosition() {
397 turret_position_ +=
398 turret_velocity_ *
399 aos::time::DurationInSeconds(monotonic_now() - last_turret_update_);
400 last_turret_update_ = monotonic_now();
401 }
402
James Kuszmaul5398fae2020-02-17 16:44:03 -0800403 bool enable_cameras_ = false;
James Kuszmaul688a2b02020-02-19 18:26:33 -0800404 // Whether to make the camera be on the turret or not.
405 bool is_turreted_ = true;
406
407 // The time at which we last incremented turret_position_.
408 monotonic_clock::time_point last_turret_update_ = monotonic_clock::min_time;
409 // Current turret position and velocity. These are set directly by the user in
410 // the test, and if velocity is non-zero, then we will automatically increment
411 // turret_position_ with every timestep.
412 double turret_position_ = 0.0; // rad
413 double turret_velocity_ = 0.0; // rad / sec
James Kuszmaul5398fae2020-02-17 16:44:03 -0800414
415 std::unique_ptr<aos::EventLoop> logger_event_loop_;
James Kuszmaul5398fae2020-02-17 16:44:03 -0800416 std::unique_ptr<aos::logger::Logger> logger_;
417};
418
419// Tests that no camera updates, combined with a perfect model, results in no
420// error.
421TEST_F(LocalizedDrivetrainTest, NoCameraUpdate) {
422 SetEnabled(true);
423 set_enable_cameras(false);
James Kuszmaul688a2b02020-02-19 18:26:33 -0800424 EXPECT_TRUE(VerifyEstimatorAccurate(1e-7));
James Kuszmaul5398fae2020-02-17 16:44:03 -0800425
James Kuszmaul688a2b02020-02-19 18:26:33 -0800426 SendGoal(-1.0, 1.0);
James Kuszmaul5398fae2020-02-17 16:44:03 -0800427
James Kuszmaul5398fae2020-02-17 16:44:03 -0800428 RunFor(chrono::seconds(3));
429 VerifyNearGoal();
James Kuszmaul688a2b02020-02-19 18:26:33 -0800430 EXPECT_TRUE(VerifyEstimatorAccurate(5e-4));
James Kuszmaul5398fae2020-02-17 16:44:03 -0800431}
432
James Kuszmaul7f55f072020-03-01 10:21:26 -0800433// Tests that we can drive in a straight line and have things estimate
434// correctly.
435TEST_F(LocalizedDrivetrainTest, NoCameraUpdateStraightLine) {
436 SetEnabled(true);
437 set_enable_cameras(false);
438 EXPECT_TRUE(VerifyEstimatorAccurate(1e-7));
439
440 SendGoal(1.0, 1.0);
441
442 RunFor(chrono::seconds(1));
443 VerifyNearGoal();
444 // Due to accelerometer drift, the straight-line driving tends to be less
445 // accurate...
Austin Schuhcadb3292021-01-23 20:24:17 -0800446 EXPECT_TRUE(VerifyEstimatorAccurate(0.15));
James Kuszmaul7f55f072020-03-01 10:21:26 -0800447}
448
James Kuszmaul688a2b02020-02-19 18:26:33 -0800449// Tests that camera updates with a perfect models results in no errors.
James Kuszmaul5398fae2020-02-17 16:44:03 -0800450TEST_F(LocalizedDrivetrainTest, PerfectCameraUpdate) {
451 SetEnabled(true);
452 set_enable_cameras(true);
James Kuszmaul688a2b02020-02-19 18:26:33 -0800453 set_camera_is_turreted(true);
James Kuszmaul5398fae2020-02-17 16:44:03 -0800454
James Kuszmaul688a2b02020-02-19 18:26:33 -0800455 SendGoal(-1.0, 1.0);
James Kuszmaul5398fae2020-02-17 16:44:03 -0800456
James Kuszmaul5398fae2020-02-17 16:44:03 -0800457 RunFor(chrono::seconds(3));
458 VerifyNearGoal();
James Kuszmaul66efe832020-03-16 19:38:33 -0700459 EXPECT_TRUE(VerifyEstimatorAccurate(2e-3));
James Kuszmaul5398fae2020-02-17 16:44:03 -0800460}
461
James Kuszmaulaca88782021-04-24 16:48:45 -0700462// Tests that camera updates with a perfect model but incorrect camera pitch
463// results in no errors.
464TEST_F(LocalizedDrivetrainTest, PerfectCameraUpdateWithBadPitch) {
465 // Introduce some camera pitch.
466 camera_calibration_offset_.template block<3, 3>(0, 0) =
467 Eigen::AngleAxis<double>(0.1, Eigen::Vector3d::UnitY())
468 .toRotationMatrix();
469 SetEnabled(true);
470 set_enable_cameras(true);
471 set_camera_is_turreted(true);
472
473 SendGoal(-1.0, 1.0);
474
475 RunFor(chrono::seconds(3));
476 VerifyNearGoal();
477 EXPECT_TRUE(VerifyEstimatorAccurate(2e-3));
478}
479
James Kuszmaul688a2b02020-02-19 18:26:33 -0800480// Tests that camera updates with a constant initial error in the position
James Kuszmaul5398fae2020-02-17 16:44:03 -0800481// results in convergence.
482TEST_F(LocalizedDrivetrainTest, InitialPositionError) {
483 SetEnabled(true);
484 set_enable_cameras(true);
James Kuszmaul688a2b02020-02-19 18:26:33 -0800485 set_camera_is_turreted(true);
James Kuszmaul5398fae2020-02-17 16:44:03 -0800486 drivetrain_plant_.mutable_state()->topRows(3) +=
487 Eigen::Vector3d(0.1, 0.1, 0.01);
James Kuszmaul5398fae2020-02-17 16:44:03 -0800488
James Kuszmaul688a2b02020-02-19 18:26:33 -0800489 // Confirm that some translational movement does get handled correctly.
490 SendGoal(-0.9, 1.0);
James Kuszmaul5398fae2020-02-17 16:44:03 -0800491
James Kuszmaul5398fae2020-02-17 16:44:03 -0800492 // Give the filters enough time to converge.
493 RunFor(chrono::seconds(10));
494 VerifyNearGoal(5e-3);
James Kuszmaul58cb1fe2020-03-07 16:18:59 -0800495 EXPECT_TRUE(VerifyEstimatorAccurate(4e-2));
James Kuszmaul688a2b02020-02-19 18:26:33 -0800496}
497
498// Tests that camera updates using a non-turreted camera work.
499TEST_F(LocalizedDrivetrainTest, InitialPositionErrorNoTurret) {
500 SetEnabled(true);
501 set_enable_cameras(true);
502 set_camera_is_turreted(false);
503 drivetrain_plant_.mutable_state()->topRows(3) +=
504 Eigen::Vector3d(0.1, 0.1, 0.01);
505
506 SendGoal(-1.0, 1.0);
507
508 // Give the filters enough time to converge.
509 RunFor(chrono::seconds(10));
510 VerifyNearGoal(5e-3);
James Kuszmaul58cb1fe2020-03-07 16:18:59 -0800511 EXPECT_TRUE(VerifyEstimatorAccurate(4e-2));
James Kuszmaul688a2b02020-02-19 18:26:33 -0800512}
513
514// Tests that we are able to handle a constant, non-zero turret angle.
515TEST_F(LocalizedDrivetrainTest, NonZeroTurret) {
516 SetEnabled(true);
517 set_enable_cameras(true);
518 set_camera_is_turreted(true);
519 set_turret(1.0, 0.0);
520 drivetrain_plant_.mutable_state()->topRows(3) +=
521 Eigen::Vector3d(0.1, 0.1, 0.0);
522
523 SendGoal(-1.0, 1.0);
524
525 // Give the filters enough time to converge.
526 RunFor(chrono::seconds(10));
527 VerifyNearGoal(5e-3);
James Kuszmaul58cb1fe2020-03-07 16:18:59 -0800528 EXPECT_TRUE(VerifyEstimatorAccurate(1e-2));
James Kuszmaul688a2b02020-02-19 18:26:33 -0800529}
530
531// Tests that we are able to handle a constant velocity turret.
532TEST_F(LocalizedDrivetrainTest, MovingTurret) {
533 SetEnabled(true);
534 set_enable_cameras(true);
535 set_camera_is_turreted(true);
536 set_turret(0.0, 0.2);
537 drivetrain_plant_.mutable_state()->topRows(3) +=
538 Eigen::Vector3d(0.1, 0.1, 0.0);
539
540 SendGoal(-1.0, 1.0);
541
542 // Give the filters enough time to converge.
543 RunFor(chrono::seconds(10));
544 VerifyNearGoal(5e-3);
James Kuszmaul58cb1fe2020-03-07 16:18:59 -0800545 EXPECT_TRUE(VerifyEstimatorAccurate(1e-2));
James Kuszmaul688a2b02020-02-19 18:26:33 -0800546}
547
548// Tests that we reject camera measurements when the turret is spinning too
549// fast.
550TEST_F(LocalizedDrivetrainTest, TooFastTurret) {
551 SetEnabled(true);
552 set_enable_cameras(true);
553 set_camera_is_turreted(true);
554 set_turret(0.0, -10.0);
555 const Eigen::Vector3d disturbance(0.1, 0.1, 0.0);
556 drivetrain_plant_.mutable_state()->topRows(3) += disturbance;
557
558 SendGoal(-1.0, 1.0);
559
560 RunFor(chrono::seconds(10));
561 EXPECT_FALSE(VerifyEstimatorAccurate(1e-3));
562 // If we remove the disturbance, we should now be correct.
563 drivetrain_plant_.mutable_state()->topRows(3) -= disturbance;
564 VerifyNearGoal(5e-3);
Austin Schuhcadb3292021-01-23 20:24:17 -0800565 EXPECT_TRUE(VerifyEstimatorAccurate(2e-3));
James Kuszmaul688a2b02020-02-19 18:26:33 -0800566}
567
568// Tests that we don't reject camera measurements when the turret is spinning
569// too fast but we aren't using a camera attached to the turret.
570TEST_F(LocalizedDrivetrainTest, TooFastTurretDoesntAffectFixedCamera) {
571 SetEnabled(true);
572 set_enable_cameras(true);
573 set_camera_is_turreted(false);
574 set_turret(0.0, -10.0);
575 const Eigen::Vector3d disturbance(0.1, 0.1, 0.0);
576 drivetrain_plant_.mutable_state()->topRows(3) += disturbance;
577
578 SendGoal(-1.0, 1.0);
579
580 RunFor(chrono::seconds(10));
581 VerifyNearGoal(5e-3);
James Kuszmaul58cb1fe2020-03-07 16:18:59 -0800582 EXPECT_TRUE(VerifyEstimatorAccurate(1e-2));
James Kuszmaul5398fae2020-02-17 16:44:03 -0800583}
584
James Kuszmaulbcd96fc2020-10-12 20:29:32 -0700585// Tests that we don't blow up if we stop getting updates for an extended period
586// of time and fall behind on fetching fron the cameras.
587TEST_F(LocalizedDrivetrainTest, FetchersHandleTimeGap) {
588 set_enable_cameras(true);
589 set_send_delay(std::chrono::seconds(0));
590 event_loop_factory()->set_network_delay(std::chrono::nanoseconds(1));
591 test_event_loop_
592 ->AddTimer([this]() { drivetrain_plant_.set_send_messages(false); })
593 ->Setup(test_event_loop_->monotonic_now());
594 test_event_loop_->AddPhasedLoop(
595 [this](int) {
596 auto builder = camera_sender_.MakeBuilder();
597 ImageMatchResultT image;
598 ASSERT_TRUE(
599 builder.Send(ImageMatchResult::Pack(*builder.fbb(), &image)));
600 },
milind945708b2021-07-03 13:30:15 -0700601 std::chrono::milliseconds(40));
James Kuszmaulbcd96fc2020-10-12 20:29:32 -0700602 test_event_loop_
603 ->AddTimer([this]() {
604 drivetrain_plant_.set_send_messages(true);
605 SimulateSensorReset();
606 })
607 ->Setup(test_event_loop_->monotonic_now() + std::chrono::seconds(10));
608
609 RunFor(chrono::seconds(20));
610}
611
James Kuszmaul5398fae2020-02-17 16:44:03 -0800612} // namespace testing
613} // namespace drivetrain
614} // namespace control_loops
615} // namespace y2020