blob: c0da838628ea3110cebd56476f03793a830e1ea1 [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;
James Kuszmaul5a46c8d2021-09-03 19:33:48 -070087 H.block<3, 3>(0, 0) =
88 Eigen::AngleAxis<double>(1.1, Eigen::Vector3d::UnitZ()) *
89 H.block<3, 3>(0, 0);
James Kuszmaul688a2b02020-02-19 18:26:33 -080090 locations.push_back(H);
91 return locations;
James Kuszmaul5398fae2020-02-17 16:44:03 -080092}
James Kuszmaul958b21e2020-02-26 21:51:40 -080093
Austin Schuhbe69cf32020-08-27 11:38:33 -070094constexpr std::chrono::seconds kPiTimeOffset(-10);
James Kuszmaul5398fae2020-02-17 16:44:03 -080095} // namespace
96
97namespace chrono = std::chrono;
James Kuszmaul5398fae2020-02-17 16:44:03 -080098using aos::monotonic_clock;
Brian Silverman1f345222020-09-24 21:14:48 -070099using frc971::control_loops::drivetrain::DrivetrainLoop;
100using frc971::control_loops::drivetrain::testing::DrivetrainSimulation;
James Kuszmaul5398fae2020-02-17 16:44:03 -0800101
James Kuszmaul61750662021-06-21 21:32:33 -0700102class LocalizedDrivetrainTest : public frc971::testing::ControlLoopTest {
James Kuszmaul5398fae2020-02-17 16:44:03 -0800103 protected:
104 // We must use the 2020 drivetrain config so that we don't have to deal
105 // with shifting:
106 LocalizedDrivetrainTest()
James Kuszmaul61750662021-06-21 21:32:33 -0700107 : frc971::testing::ControlLoopTest(
James Kuszmaul5398fae2020-02-17 16:44:03 -0800108 aos::configuration::ReadConfig(
109 "y2020/control_loops/drivetrain/simulation_config.json"),
110 GetTest2020DrivetrainConfig().dt),
Austin Schuh87dd3832021-01-01 23:07:31 -0800111 time_converter_(aos::configuration::NodesCount(configuration())),
Austin Schuh6aa77be2020-02-22 21:06:40 -0800112 roborio_(aos::configuration::GetNode(configuration(), "roborio")),
113 pi1_(aos::configuration::GetNode(configuration(), "pi1")),
114 test_event_loop_(MakeEventLoop("test", roborio_)),
James Kuszmaul5398fae2020-02-17 16:44:03 -0800115 drivetrain_goal_sender_(
116 test_event_loop_->MakeSender<Goal>("/drivetrain")),
117 drivetrain_goal_fetcher_(
118 test_event_loop_->MakeFetcher<Goal>("/drivetrain")),
119 localizer_control_sender_(
120 test_event_loop_->MakeSender<LocalizerControl>("/drivetrain")),
James Kuszmaul688a2b02020-02-19 18:26:33 -0800121 superstructure_status_sender_(
122 test_event_loop_->MakeSender<superstructure::Status>(
123 "/superstructure")),
James Kuszmaul958b21e2020-02-26 21:51:40 -0800124 server_statistics_sender_(
125 test_event_loop_->MakeSender<aos::message_bridge::ServerStatistics>(
126 "/aos")),
Austin Schuh6aa77be2020-02-22 21:06:40 -0800127 drivetrain_event_loop_(MakeEventLoop("drivetrain", roborio_)),
James Kuszmaul5398fae2020-02-17 16:44:03 -0800128 dt_config_(GetTest2020DrivetrainConfig()),
Austin Schuh6aa77be2020-02-22 21:06:40 -0800129 pi1_event_loop_(MakeEventLoop("test", pi1_)),
James Kuszmaul5398fae2020-02-17 16:44:03 -0800130 camera_sender_(
Austin Schuh6aa77be2020-02-22 21:06:40 -0800131 pi1_event_loop_->MakeSender<ImageMatchResult>("/pi1/camera")),
James Kuszmaul5398fae2020-02-17 16:44:03 -0800132 localizer_(drivetrain_event_loop_.get(), dt_config_),
133 drivetrain_(dt_config_, drivetrain_event_loop_.get(), &localizer_),
Austin Schuh6aa77be2020-02-22 21:06:40 -0800134 drivetrain_plant_event_loop_(MakeEventLoop("plant", roborio_)),
James Kuszmaul5398fae2020-02-17 16:44:03 -0800135 drivetrain_plant_(drivetrain_plant_event_loop_.get(), dt_config_),
136 last_frame_(monotonic_now()) {
Austin Schuh87dd3832021-01-01 23:07:31 -0800137 event_loop_factory()->SetTimeConverter(&time_converter_);
James Kuszmaul9c128122021-03-22 22:24:36 -0700138 CHECK_EQ(aos::configuration::GetNodeIndex(configuration(), roborio_), 6);
Austin Schuh87dd3832021-01-01 23:07:31 -0800139 CHECK_EQ(aos::configuration::GetNodeIndex(configuration(), pi1_), 1);
Austin Schuh66168842021-08-17 19:42:21 -0700140 time_converter_.AddMonotonic({BootTimestamp::epoch() + kPiTimeOffset,
141 BootTimestamp::epoch() + kPiTimeOffset,
142 BootTimestamp::epoch() + kPiTimeOffset,
143 BootTimestamp::epoch() + kPiTimeOffset,
144 BootTimestamp::epoch() + kPiTimeOffset,
145 BootTimestamp::epoch() + kPiTimeOffset,
146 BootTimestamp::epoch()});
James Kuszmaul5398fae2020-02-17 16:44:03 -0800147 set_team_id(frc971::control_loops::testing::kTeamNumber);
James Kuszmaul5398fae2020-02-17 16:44:03 -0800148 set_battery_voltage(12.0);
149
150 if (!FLAGS_output_file.empty()) {
Austin Schuh6aa77be2020-02-22 21:06:40 -0800151 logger_event_loop_ = MakeEventLoop("logger", roborio_);
Brian Silverman1f345222020-09-24 21:14:48 -0700152 logger_ = std::make_unique<aos::logger::Logger>(logger_event_loop_.get());
153 logger_->StartLoggingLocalNamerOnRun(FLAGS_output_file);
James Kuszmaul5398fae2020-02-17 16:44:03 -0800154 }
155
156 test_event_loop_->MakeWatcher(
157 "/drivetrain",
158 [this](const frc971::control_loops::drivetrain::Status &) {
159 // Needs to do camera updates right after we run the control loop.
160 if (enable_cameras_) {
161 SendDelayedFrames();
162 if (last_frame_ + std::chrono::milliseconds(100) <
163 monotonic_now()) {
164 CaptureFrames();
165 last_frame_ = monotonic_now();
166 }
167 }
Brian Silverman1f345222020-09-24 21:14:48 -0700168 });
James Kuszmaul688a2b02020-02-19 18:26:33 -0800169
170 test_event_loop_->AddPhasedLoop(
171 [this](int) {
James Kuszmaul958b21e2020-02-26 21:51:40 -0800172 auto builder = server_statistics_sender_.MakeBuilder();
173 auto name_offset = builder.fbb()->CreateString("pi1");
174 auto node_builder = builder.MakeBuilder<aos::Node>();
175 node_builder.add_name(name_offset);
176 auto node_offset = node_builder.Finish();
177 auto connection_builder =
178 builder.MakeBuilder<aos::message_bridge::ServerConnection>();
179 connection_builder.add_node(node_offset);
180 connection_builder.add_monotonic_offset(
Austin Schuhbe69cf32020-08-27 11:38:33 -0700181 chrono::duration_cast<chrono::nanoseconds>(kPiTimeOffset)
James Kuszmaul958b21e2020-02-26 21:51:40 -0800182 .count());
183 auto connection_offset = connection_builder.Finish();
184 auto connections_offset =
185 builder.fbb()->CreateVector(&connection_offset, 1);
186 auto statistics_builder =
187 builder.MakeBuilder<aos::message_bridge::ServerStatistics>();
188 statistics_builder.add_connections(connections_offset);
189 builder.Send(statistics_builder.Finish());
190 },
191 chrono::milliseconds(500));
192
193 test_event_loop_->AddPhasedLoop(
194 [this](int) {
James Kuszmaul688a2b02020-02-19 18:26:33 -0800195 // Also use the opportunity to send out turret messages.
196 UpdateTurretPosition();
197 auto builder = superstructure_status_sender_.MakeBuilder();
198 auto turret_builder =
199 builder
200 .MakeBuilder<frc971::control_loops::
201 PotAndAbsoluteEncoderProfiledJointStatus>();
202 turret_builder.add_position(turret_position_);
203 turret_builder.add_velocity(turret_velocity_);
204 auto turret_offset = turret_builder.Finish();
205 auto status_builder = builder.MakeBuilder<superstructure::Status>();
206 status_builder.add_turret(turret_offset);
207 builder.Send(status_builder.Finish());
208 },
209 chrono::milliseconds(5));
James Kuszmaul5398fae2020-02-17 16:44:03 -0800210
James Kuszmaul286b4282020-02-26 20:29:32 -0800211 test_event_loop_->OnRun([this]() { SetStartingPosition({3.0, 2.0, 0.0}); });
212
James Kuszmaul5398fae2020-02-17 16:44:03 -0800213 // Run for enough time to allow the gyro/imu zeroing code to run.
214 RunFor(std::chrono::seconds(10));
215 }
216
217 virtual ~LocalizedDrivetrainTest() override {}
218
219 void SetStartingPosition(const Eigen::Matrix<double, 3, 1> &xytheta) {
220 *drivetrain_plant_.mutable_state() << xytheta.x(), xytheta.y(),
221 xytheta(2, 0), 0.0, 0.0;
James Kuszmaulbcd96fc2020-10-12 20:29:32 -0700222 Eigen::Matrix<double, Localizer::HybridEkf::kNStates, 1> localizer_state;
James Kuszmaul5398fae2020-02-17 16:44:03 -0800223 localizer_state.setZero();
James Kuszmaulbcd96fc2020-10-12 20:29:32 -0700224 localizer_state.block<3, 1>(0, 0) = xytheta;
James Kuszmaul5398fae2020-02-17 16:44:03 -0800225 localizer_.Reset(monotonic_now(), localizer_state);
226 }
227
228 void VerifyNearGoal(double eps = 1e-3) {
229 drivetrain_goal_fetcher_.Fetch();
230 EXPECT_NEAR(drivetrain_goal_fetcher_->left_goal(),
231 drivetrain_plant_.GetLeftPosition(), eps);
232 EXPECT_NEAR(drivetrain_goal_fetcher_->right_goal(),
233 drivetrain_plant_.GetRightPosition(), eps);
234 }
235
James Kuszmaul688a2b02020-02-19 18:26:33 -0800236 ::testing::AssertionResult IsNear(double expected, double actual,
237 double epsilon) {
238 if (std::abs(expected - actual) < epsilon) {
239 return ::testing::AssertionSuccess();
240 } else {
241 return ::testing::AssertionFailure()
242 << "Expected " << expected << " but got " << actual
243 << " with a max difference of " << epsilon
244 << " and an actual difference of " << std::abs(expected - actual);
245 }
246 }
247 ::testing::AssertionResult VerifyEstimatorAccurate(double eps) {
James Kuszmaul5398fae2020-02-17 16:44:03 -0800248 const Eigen::Matrix<double, 5, 1> true_state = drivetrain_plant_.state();
James Kuszmaul688a2b02020-02-19 18:26:33 -0800249 ::testing::AssertionResult result(true);
250 if (!(result = IsNear(localizer_.x(), true_state(0), eps))) {
251 return result;
252 }
253 if (!(result = IsNear(localizer_.y(), true_state(1), eps))) {
254 return result;
255 }
256 if (!(result = IsNear(localizer_.theta(), true_state(2), eps))) {
257 return result;
258 }
259 if (!(result = IsNear(localizer_.left_velocity(), true_state(3), eps))) {
260 return result;
261 }
262 if (!(result = IsNear(localizer_.right_velocity(), true_state(4), eps))) {
263 return result;
264 }
265 return result;
James Kuszmaul5398fae2020-02-17 16:44:03 -0800266 }
267
268 // Goes through and captures frames on the camera(s), queueing them up to be
269 // sent by SendDelayedFrames().
270 void CaptureFrames() {
271 const frc971::control_loops::Pose robot_pose(
272 {drivetrain_plant_.GetPosition().x(),
273 drivetrain_plant_.GetPosition().y(), 0.0},
274 drivetrain_plant_.state()(2, 0));
275 std::unique_ptr<ImageMatchResultT> frame(new ImageMatchResultT());
276
James Kuszmaulc51dbfe2020-02-23 15:39:00 -0800277 for (const auto &H_field_target : TargetLocations()) {
James Kuszmaul5398fae2020-02-17 16:44:03 -0800278 std::unique_ptr<CameraPoseT> camera_target(new CameraPoseT());
279
280 camera_target->field_to_target.reset(new TransformationMatrixT());
James Kuszmaulc51dbfe2020-02-23 15:39:00 -0800281 camera_target->field_to_target->data = MatrixToVector(H_field_target);
James Kuszmaul688a2b02020-02-19 18:26:33 -0800282
James Kuszmaulc51dbfe2020-02-23 15:39:00 -0800283 Eigen::Matrix<double, 4, 4> H_turret_camera =
James Kuszmaul688a2b02020-02-19 18:26:33 -0800284 Eigen::Matrix<double, 4, 4>::Identity();
285 if (is_turreted_) {
James Kuszmaulc51dbfe2020-02-23 15:39:00 -0800286 H_turret_camera = frc971::control_loops::TransformationMatrixForYaw(
James Kuszmaul688a2b02020-02-19 18:26:33 -0800287 turret_position_) *
288 CameraTurretTransformation();
289 }
James Kuszmaul5398fae2020-02-17 16:44:03 -0800290
291 // TODO(james): Use non-zero turret angles.
292 camera_target->camera_to_target.reset(new TransformationMatrixT());
James Kuszmaulaca88782021-04-24 16:48:45 -0700293 camera_target->camera_to_target->data = MatrixToVector(
294 (robot_pose.AsTransformationMatrix() * TurretRobotTransformation() *
295 H_turret_camera * camera_calibration_offset_)
296 .inverse() *
297 H_field_target);
James Kuszmaul5398fae2020-02-17 16:44:03 -0800298
299 frame->camera_poses.emplace_back(std::move(camera_target));
300 }
301
302 frame->image_monotonic_timestamp_ns =
303 chrono::duration_cast<chrono::nanoseconds>(
James Kuszmaul958b21e2020-02-26 21:51:40 -0800304 event_loop_factory()
305 ->GetNodeEventLoopFactory(pi1_)
306 ->monotonic_now()
307 .time_since_epoch())
James Kuszmaul5398fae2020-02-17 16:44:03 -0800308 .count();
309 frame->camera_calibration.reset(new CameraCalibrationT());
310 {
311 frame->camera_calibration->fixed_extrinsics.reset(
312 new TransformationMatrixT());
James Kuszmaulc51dbfe2020-02-23 15:39:00 -0800313 TransformationMatrixT *H_robot_turret =
James Kuszmaul5398fae2020-02-17 16:44:03 -0800314 frame->camera_calibration->fixed_extrinsics.get();
James Kuszmaulc51dbfe2020-02-23 15:39:00 -0800315 H_robot_turret->data = MatrixToVector(TurretRobotTransformation());
James Kuszmaul5398fae2020-02-17 16:44:03 -0800316 }
James Kuszmaul688a2b02020-02-19 18:26:33 -0800317
318 if (is_turreted_) {
James Kuszmaul5398fae2020-02-17 16:44:03 -0800319 frame->camera_calibration->turret_extrinsics.reset(
320 new TransformationMatrixT());
James Kuszmaulc51dbfe2020-02-23 15:39:00 -0800321 TransformationMatrixT *H_turret_camera =
James Kuszmaul5398fae2020-02-17 16:44:03 -0800322 frame->camera_calibration->turret_extrinsics.get();
James Kuszmaulc51dbfe2020-02-23 15:39:00 -0800323 H_turret_camera->data = MatrixToVector(CameraTurretTransformation());
James Kuszmaul5398fae2020-02-17 16:44:03 -0800324 }
325
326 camera_delay_queue_.emplace(monotonic_now(), std::move(frame));
327 }
328
329 // Actually sends out all the camera frames.
330 void SendDelayedFrames() {
331 const std::chrono::milliseconds camera_latency(150);
332 while (!camera_delay_queue_.empty() &&
333 std::get<0>(camera_delay_queue_.front()) <
334 monotonic_now() - camera_latency) {
335 auto builder = camera_sender_.MakeBuilder();
336 ASSERT_TRUE(builder.Send(ImageMatchResult::Pack(
337 *builder.fbb(), std::get<1>(camera_delay_queue_.front()).get())));
338 camera_delay_queue_.pop();
339 }
340 }
341
Austin Schuh87dd3832021-01-01 23:07:31 -0800342 aos::message_bridge::TestingTimeConverter time_converter_;
343
Austin Schuh6aa77be2020-02-22 21:06:40 -0800344 const aos::Node *const roborio_;
345 const aos::Node *const pi1_;
346
James Kuszmaul5398fae2020-02-17 16:44:03 -0800347 std::unique_ptr<aos::EventLoop> test_event_loop_;
348 aos::Sender<Goal> drivetrain_goal_sender_;
349 aos::Fetcher<Goal> drivetrain_goal_fetcher_;
350 aos::Sender<LocalizerControl> localizer_control_sender_;
James Kuszmaul688a2b02020-02-19 18:26:33 -0800351 aos::Sender<superstructure::Status> superstructure_status_sender_;
James Kuszmaul958b21e2020-02-26 21:51:40 -0800352 aos::Sender<aos::message_bridge::ServerStatistics> server_statistics_sender_;
James Kuszmaul5398fae2020-02-17 16:44:03 -0800353
354 std::unique_ptr<aos::EventLoop> drivetrain_event_loop_;
Brian Silverman1f345222020-09-24 21:14:48 -0700355 const frc971::control_loops::drivetrain::DrivetrainConfig<double> dt_config_;
James Kuszmaul5398fae2020-02-17 16:44:03 -0800356
Austin Schuh6aa77be2020-02-22 21:06:40 -0800357 std::unique_ptr<aos::EventLoop> pi1_event_loop_;
James Kuszmaul5398fae2020-02-17 16:44:03 -0800358 aos::Sender<ImageMatchResult> camera_sender_;
359
360 Localizer localizer_;
361 DrivetrainLoop drivetrain_;
362
363 std::unique_ptr<aos::EventLoop> drivetrain_plant_event_loop_;
364 DrivetrainSimulation drivetrain_plant_;
365 monotonic_clock::time_point last_frame_;
366
367 // A queue of camera frames so that we can add a time delay to the data
368 // coming from the cameras.
369 std::queue<std::tuple<aos::monotonic_clock::time_point,
370 std::unique_ptr<ImageMatchResultT>>>
371 camera_delay_queue_;
372
James Kuszmaulaca88782021-04-24 16:48:45 -0700373 // Offset to add to the camera for actually taking the images, to simulate an
374 // inaccurate calibration.
375 Eigen::Matrix<double, 4, 4> camera_calibration_offset_ =
376 Eigen::Matrix<double, 4, 4>::Identity();
377
James Kuszmaul5398fae2020-02-17 16:44:03 -0800378 void set_enable_cameras(bool enable) { enable_cameras_ = enable; }
James Kuszmaul688a2b02020-02-19 18:26:33 -0800379 void set_camera_is_turreted(bool turreted) { is_turreted_ = turreted; }
380
381 void set_turret(double position, double velocity) {
382 turret_position_ = position;
383 turret_velocity_ = velocity;
384 }
385
386 void SendGoal(double left, double right) {
387 auto builder = drivetrain_goal_sender_.MakeBuilder();
388
389 Goal::Builder drivetrain_builder = builder.MakeBuilder<Goal>();
390 drivetrain_builder.add_controller_type(
391 frc971::control_loops::drivetrain::ControllerType::MOTION_PROFILE);
392 drivetrain_builder.add_left_goal(left);
393 drivetrain_builder.add_right_goal(right);
394
395 EXPECT_TRUE(builder.Send(drivetrain_builder.Finish()));
396 }
James Kuszmaul5398fae2020-02-17 16:44:03 -0800397
398 private:
James Kuszmaul688a2b02020-02-19 18:26:33 -0800399 void UpdateTurretPosition() {
400 turret_position_ +=
401 turret_velocity_ *
402 aos::time::DurationInSeconds(monotonic_now() - last_turret_update_);
403 last_turret_update_ = monotonic_now();
404 }
405
James Kuszmaul5398fae2020-02-17 16:44:03 -0800406 bool enable_cameras_ = false;
James Kuszmaul688a2b02020-02-19 18:26:33 -0800407 // Whether to make the camera be on the turret or not.
408 bool is_turreted_ = true;
409
410 // The time at which we last incremented turret_position_.
411 monotonic_clock::time_point last_turret_update_ = monotonic_clock::min_time;
412 // Current turret position and velocity. These are set directly by the user in
413 // the test, and if velocity is non-zero, then we will automatically increment
414 // turret_position_ with every timestep.
415 double turret_position_ = 0.0; // rad
416 double turret_velocity_ = 0.0; // rad / sec
James Kuszmaul5398fae2020-02-17 16:44:03 -0800417
418 std::unique_ptr<aos::EventLoop> logger_event_loop_;
James Kuszmaul5398fae2020-02-17 16:44:03 -0800419 std::unique_ptr<aos::logger::Logger> logger_;
420};
421
422// Tests that no camera updates, combined with a perfect model, results in no
423// error.
424TEST_F(LocalizedDrivetrainTest, NoCameraUpdate) {
425 SetEnabled(true);
426 set_enable_cameras(false);
James Kuszmaul688a2b02020-02-19 18:26:33 -0800427 EXPECT_TRUE(VerifyEstimatorAccurate(1e-7));
James Kuszmaul5398fae2020-02-17 16:44:03 -0800428
James Kuszmaul688a2b02020-02-19 18:26:33 -0800429 SendGoal(-1.0, 1.0);
James Kuszmaul5398fae2020-02-17 16:44:03 -0800430
James Kuszmaul5398fae2020-02-17 16:44:03 -0800431 RunFor(chrono::seconds(3));
432 VerifyNearGoal();
James Kuszmaul688a2b02020-02-19 18:26:33 -0800433 EXPECT_TRUE(VerifyEstimatorAccurate(5e-4));
James Kuszmaul5398fae2020-02-17 16:44:03 -0800434}
435
James Kuszmaul7f55f072020-03-01 10:21:26 -0800436// Tests that we can drive in a straight line and have things estimate
437// correctly.
438TEST_F(LocalizedDrivetrainTest, NoCameraUpdateStraightLine) {
439 SetEnabled(true);
440 set_enable_cameras(false);
441 EXPECT_TRUE(VerifyEstimatorAccurate(1e-7));
442
443 SendGoal(1.0, 1.0);
444
445 RunFor(chrono::seconds(1));
446 VerifyNearGoal();
447 // Due to accelerometer drift, the straight-line driving tends to be less
448 // accurate...
Austin Schuhcadb3292021-01-23 20:24:17 -0800449 EXPECT_TRUE(VerifyEstimatorAccurate(0.15));
James Kuszmaul7f55f072020-03-01 10:21:26 -0800450}
451
James Kuszmaul688a2b02020-02-19 18:26:33 -0800452// Tests that camera updates with a perfect models results in no errors.
James Kuszmaul5398fae2020-02-17 16:44:03 -0800453TEST_F(LocalizedDrivetrainTest, PerfectCameraUpdate) {
454 SetEnabled(true);
455 set_enable_cameras(true);
James Kuszmaul688a2b02020-02-19 18:26:33 -0800456 set_camera_is_turreted(true);
James Kuszmaul5398fae2020-02-17 16:44:03 -0800457
James Kuszmaul688a2b02020-02-19 18:26:33 -0800458 SendGoal(-1.0, 1.0);
James Kuszmaul5398fae2020-02-17 16:44:03 -0800459
James Kuszmaul5398fae2020-02-17 16:44:03 -0800460 RunFor(chrono::seconds(3));
461 VerifyNearGoal();
James Kuszmaul66efe832020-03-16 19:38:33 -0700462 EXPECT_TRUE(VerifyEstimatorAccurate(2e-3));
James Kuszmaul5398fae2020-02-17 16:44:03 -0800463}
464
James Kuszmaulaca88782021-04-24 16:48:45 -0700465// Tests that camera updates with a perfect model but incorrect camera pitch
466// results in no errors.
467TEST_F(LocalizedDrivetrainTest, PerfectCameraUpdateWithBadPitch) {
468 // Introduce some camera pitch.
469 camera_calibration_offset_.template block<3, 3>(0, 0) =
470 Eigen::AngleAxis<double>(0.1, Eigen::Vector3d::UnitY())
471 .toRotationMatrix();
472 SetEnabled(true);
473 set_enable_cameras(true);
474 set_camera_is_turreted(true);
475
476 SendGoal(-1.0, 1.0);
477
478 RunFor(chrono::seconds(3));
479 VerifyNearGoal();
480 EXPECT_TRUE(VerifyEstimatorAccurate(2e-3));
481}
482
James Kuszmaul688a2b02020-02-19 18:26:33 -0800483// Tests that camera updates with a constant initial error in the position
James Kuszmaul5398fae2020-02-17 16:44:03 -0800484// results in convergence.
485TEST_F(LocalizedDrivetrainTest, InitialPositionError) {
486 SetEnabled(true);
487 set_enable_cameras(true);
James Kuszmaul688a2b02020-02-19 18:26:33 -0800488 set_camera_is_turreted(true);
James Kuszmaul5398fae2020-02-17 16:44:03 -0800489 drivetrain_plant_.mutable_state()->topRows(3) +=
490 Eigen::Vector3d(0.1, 0.1, 0.01);
James Kuszmaul5398fae2020-02-17 16:44:03 -0800491
James Kuszmaul688a2b02020-02-19 18:26:33 -0800492 // Confirm that some translational movement does get handled correctly.
493 SendGoal(-0.9, 1.0);
James Kuszmaul5398fae2020-02-17 16:44:03 -0800494
James Kuszmaul5398fae2020-02-17 16:44:03 -0800495 // Give the filters enough time to converge.
496 RunFor(chrono::seconds(10));
497 VerifyNearGoal(5e-3);
James Kuszmaul58cb1fe2020-03-07 16:18:59 -0800498 EXPECT_TRUE(VerifyEstimatorAccurate(4e-2));
James Kuszmaul688a2b02020-02-19 18:26:33 -0800499}
500
501// Tests that camera updates using a non-turreted camera work.
502TEST_F(LocalizedDrivetrainTest, InitialPositionErrorNoTurret) {
503 SetEnabled(true);
504 set_enable_cameras(true);
505 set_camera_is_turreted(false);
506 drivetrain_plant_.mutable_state()->topRows(3) +=
507 Eigen::Vector3d(0.1, 0.1, 0.01);
508
509 SendGoal(-1.0, 1.0);
510
511 // Give the filters enough time to converge.
512 RunFor(chrono::seconds(10));
513 VerifyNearGoal(5e-3);
James Kuszmaul58cb1fe2020-03-07 16:18:59 -0800514 EXPECT_TRUE(VerifyEstimatorAccurate(4e-2));
James Kuszmaul688a2b02020-02-19 18:26:33 -0800515}
516
517// Tests that we are able to handle a constant, non-zero turret angle.
518TEST_F(LocalizedDrivetrainTest, NonZeroTurret) {
519 SetEnabled(true);
520 set_enable_cameras(true);
521 set_camera_is_turreted(true);
522 set_turret(1.0, 0.0);
523 drivetrain_plant_.mutable_state()->topRows(3) +=
524 Eigen::Vector3d(0.1, 0.1, 0.0);
525
526 SendGoal(-1.0, 1.0);
527
528 // Give the filters enough time to converge.
529 RunFor(chrono::seconds(10));
530 VerifyNearGoal(5e-3);
James Kuszmaul58cb1fe2020-03-07 16:18:59 -0800531 EXPECT_TRUE(VerifyEstimatorAccurate(1e-2));
James Kuszmaul688a2b02020-02-19 18:26:33 -0800532}
533
534// Tests that we are able to handle a constant velocity turret.
535TEST_F(LocalizedDrivetrainTest, MovingTurret) {
536 SetEnabled(true);
537 set_enable_cameras(true);
538 set_camera_is_turreted(true);
539 set_turret(0.0, 0.2);
540 drivetrain_plant_.mutable_state()->topRows(3) +=
541 Eigen::Vector3d(0.1, 0.1, 0.0);
542
543 SendGoal(-1.0, 1.0);
544
545 // Give the filters enough time to converge.
546 RunFor(chrono::seconds(10));
547 VerifyNearGoal(5e-3);
James Kuszmaul58cb1fe2020-03-07 16:18:59 -0800548 EXPECT_TRUE(VerifyEstimatorAccurate(1e-2));
James Kuszmaul688a2b02020-02-19 18:26:33 -0800549}
550
551// Tests that we reject camera measurements when the turret is spinning too
552// fast.
553TEST_F(LocalizedDrivetrainTest, TooFastTurret) {
554 SetEnabled(true);
555 set_enable_cameras(true);
556 set_camera_is_turreted(true);
557 set_turret(0.0, -10.0);
558 const Eigen::Vector3d disturbance(0.1, 0.1, 0.0);
559 drivetrain_plant_.mutable_state()->topRows(3) += disturbance;
560
561 SendGoal(-1.0, 1.0);
562
563 RunFor(chrono::seconds(10));
564 EXPECT_FALSE(VerifyEstimatorAccurate(1e-3));
565 // If we remove the disturbance, we should now be correct.
566 drivetrain_plant_.mutable_state()->topRows(3) -= disturbance;
567 VerifyNearGoal(5e-3);
Austin Schuhcadb3292021-01-23 20:24:17 -0800568 EXPECT_TRUE(VerifyEstimatorAccurate(2e-3));
James Kuszmaul688a2b02020-02-19 18:26:33 -0800569}
570
571// Tests that we don't reject camera measurements when the turret is spinning
572// too fast but we aren't using a camera attached to the turret.
573TEST_F(LocalizedDrivetrainTest, TooFastTurretDoesntAffectFixedCamera) {
574 SetEnabled(true);
575 set_enable_cameras(true);
576 set_camera_is_turreted(false);
577 set_turret(0.0, -10.0);
578 const Eigen::Vector3d disturbance(0.1, 0.1, 0.0);
579 drivetrain_plant_.mutable_state()->topRows(3) += disturbance;
580
581 SendGoal(-1.0, 1.0);
582
583 RunFor(chrono::seconds(10));
584 VerifyNearGoal(5e-3);
James Kuszmaul58cb1fe2020-03-07 16:18:59 -0800585 EXPECT_TRUE(VerifyEstimatorAccurate(1e-2));
James Kuszmaul5398fae2020-02-17 16:44:03 -0800586}
587
James Kuszmaulbcd96fc2020-10-12 20:29:32 -0700588// Tests that we don't blow up if we stop getting updates for an extended period
589// of time and fall behind on fetching fron the cameras.
590TEST_F(LocalizedDrivetrainTest, FetchersHandleTimeGap) {
591 set_enable_cameras(true);
592 set_send_delay(std::chrono::seconds(0));
593 event_loop_factory()->set_network_delay(std::chrono::nanoseconds(1));
594 test_event_loop_
595 ->AddTimer([this]() { drivetrain_plant_.set_send_messages(false); })
596 ->Setup(test_event_loop_->monotonic_now());
597 test_event_loop_->AddPhasedLoop(
598 [this](int) {
599 auto builder = camera_sender_.MakeBuilder();
600 ImageMatchResultT image;
601 ASSERT_TRUE(
602 builder.Send(ImageMatchResult::Pack(*builder.fbb(), &image)));
603 },
milind945708b2021-07-03 13:30:15 -0700604 std::chrono::milliseconds(40));
James Kuszmaulbcd96fc2020-10-12 20:29:32 -0700605 test_event_loop_
606 ->AddTimer([this]() {
607 drivetrain_plant_.set_send_messages(true);
608 SimulateSensorReset();
609 })
610 ->Setup(test_event_loop_->monotonic_now() + std::chrono::seconds(10));
611
612 RunFor(chrono::seconds(20));
613}
614
James Kuszmaul5398fae2020-02-17 16:44:03 -0800615} // namespace testing
616} // namespace drivetrain
617} // namespace control_loops
618} // namespace y2020