Add constant-heading localizer reset button
Change-Id: I25afc8985dbce2199f6fc4e7eb91c58b1e9bbc5d
diff --git a/frc971/control_loops/drivetrain/drivetrain.cc b/frc971/control_loops/drivetrain/drivetrain.cc
index 2af6765..dd6b05c 100644
--- a/frc971/control_loops/drivetrain/drivetrain.cc
+++ b/frc971/control_loops/drivetrain/drivetrain.cc
@@ -243,10 +243,11 @@
// simulation.
if (localizer_control_fetcher_.Fetch()) {
LOG_STRUCT(DEBUG, "localizer_control", *localizer_control_fetcher_);
- localizer_->ResetPosition(monotonic_now, localizer_control_fetcher_->x,
- localizer_control_fetcher_->y,
- localizer_control_fetcher_->theta,
- localizer_control_fetcher_->theta_uncertainty);
+ localizer_->ResetPosition(
+ monotonic_now, localizer_control_fetcher_->x,
+ localizer_control_fetcher_->y, localizer_control_fetcher_->theta,
+ localizer_control_fetcher_->theta_uncertainty,
+ !localizer_control_fetcher_->keep_current_theta);
}
localizer_->Update({last_last_left_voltage_, last_last_right_voltage_},
monotonic_now, position->left_encoder,