Stephan Massalt | d021f97 | 2020-01-05 20:41:23 -0800 | [diff] [blame] | 1 | #include <unistd.h> |
| 2 | |
| 3 | #include <array> |
| 4 | #include <chrono> |
Tyler Chatow | bf0609c | 2021-07-31 16:13:27 -0700 | [diff] [blame] | 5 | #include <cinttypes> |
Stephan Massalt | d021f97 | 2020-01-05 20:41:23 -0800 | [diff] [blame] | 6 | #include <cmath> |
Tyler Chatow | bf0609c | 2021-07-31 16:13:27 -0700 | [diff] [blame] | 7 | #include <cstdio> |
| 8 | #include <cstring> |
Stephan Massalt | d021f97 | 2020-01-05 20:41:23 -0800 | [diff] [blame] | 9 | #include <functional> |
Tyler Chatow | bf0609c | 2021-07-31 16:13:27 -0700 | [diff] [blame] | 10 | #include <memory> |
Stephan Massalt | d021f97 | 2020-01-05 20:41:23 -0800 | [diff] [blame] | 11 | #include <mutex> |
| 12 | #include <thread> |
| 13 | |
Stephan Massalt | d021f97 | 2020-01-05 20:41:23 -0800 | [diff] [blame] | 14 | #include "frc971/wpilib/ahal/AnalogInput.h" |
| 15 | #include "frc971/wpilib/ahal/Counter.h" |
| 16 | #include "frc971/wpilib/ahal/DigitalGlitchFilter.h" |
| 17 | #include "frc971/wpilib/ahal/DriverStation.h" |
| 18 | #include "frc971/wpilib/ahal/Encoder.h" |
Alex Perry | c4691f5 | 2020-02-17 19:20:01 -0800 | [diff] [blame] | 19 | #include "frc971/wpilib/ahal/TalonFX.h" |
Stephan Massalt | d021f97 | 2020-01-05 20:41:23 -0800 | [diff] [blame] | 20 | #include "frc971/wpilib/ahal/VictorSP.h" |
| 21 | #undef ERROR |
| 22 | |
| 23 | #include "aos/commonmath.h" |
| 24 | #include "aos/events/event_loop.h" |
| 25 | #include "aos/events/shm_event_loop.h" |
| 26 | #include "aos/init.h" |
| 27 | #include "aos/logging/logging.h" |
Austin Schuh | 83873c3 | 2020-02-22 14:58:39 -0800 | [diff] [blame] | 28 | #include "aos/network/team_number.h" |
Stephan Massalt | d021f97 | 2020-01-05 20:41:23 -0800 | [diff] [blame] | 29 | #include "aos/realtime.h" |
Stephan Massalt | d021f97 | 2020-01-05 20:41:23 -0800 | [diff] [blame] | 30 | #include "aos/time/time.h" |
| 31 | #include "aos/util/log_interval.h" |
| 32 | #include "aos/util/phased_loop.h" |
| 33 | #include "aos/util/wrapping_counter.h" |
Alex Perry | c4691f5 | 2020-02-17 19:20:01 -0800 | [diff] [blame] | 34 | #include "ctre/phoenix/motorcontrol/can/TalonFX.h" |
Ravago Jones | 3dda560 | 2021-03-10 00:33:13 -0800 | [diff] [blame] | 35 | #include "ctre/phoenix/motorcontrol/can/VictorSPX.h" |
Stephan Massalt | d021f97 | 2020-01-05 20:41:23 -0800 | [diff] [blame] | 36 | #include "frc971/autonomous/auto_mode_generated.h" |
| 37 | #include "frc971/control_loops/drivetrain/drivetrain_position_generated.h" |
James Kuszmaul | 7077d34 | 2021-06-09 20:23:58 -0700 | [diff] [blame] | 38 | #include "frc971/input/robot_state_generated.h" |
James Kuszmaul | a244a91 | 2020-01-18 13:50:50 -0800 | [diff] [blame] | 39 | #include "frc971/wpilib/ADIS16470.h" |
Stephan Massalt | d021f97 | 2020-01-05 20:41:23 -0800 | [diff] [blame] | 40 | #include "frc971/wpilib/buffered_pcm.h" |
| 41 | #include "frc971/wpilib/buffered_solenoid.h" |
| 42 | #include "frc971/wpilib/dma.h" |
| 43 | #include "frc971/wpilib/drivetrain_writer.h" |
| 44 | #include "frc971/wpilib/encoder_and_potentiometer.h" |
| 45 | #include "frc971/wpilib/joystick_sender.h" |
| 46 | #include "frc971/wpilib/logging_generated.h" |
| 47 | #include "frc971/wpilib/loop_output_handler.h" |
| 48 | #include "frc971/wpilib/pdp_fetcher.h" |
| 49 | #include "frc971/wpilib/sensor_reader.h" |
| 50 | #include "frc971/wpilib/wpilib_robot_base.h" |
| 51 | #include "y2020/constants.h" |
| 52 | #include "y2020/control_loops/superstructure/superstructure_output_generated.h" |
| 53 | #include "y2020/control_loops/superstructure/superstructure_position_generated.h" |
| 54 | |
| 55 | using ::aos::monotonic_clock; |
| 56 | using ::y2020::constants::Values; |
| 57 | namespace superstructure = ::y2020::control_loops::superstructure; |
| 58 | namespace chrono = ::std::chrono; |
Vinay Siva | e52a6b3 | 2021-07-10 15:19:26 -0700 | [diff] [blame] | 59 | using std::make_unique; |
Stephan Massalt | d021f97 | 2020-01-05 20:41:23 -0800 | [diff] [blame] | 60 | |
| 61 | namespace y2020 { |
| 62 | namespace wpilib { |
| 63 | namespace { |
| 64 | |
| 65 | constexpr double kMaxBringupPower = 12.0; |
| 66 | |
| 67 | // TODO(Brian): Fix the interpretation of the result of GetRaw here and in the |
| 68 | // DMA stuff and then removing the * 2.0 in *_translate. |
| 69 | // The low bit is direction. |
| 70 | |
Stephan Massalt | d021f97 | 2020-01-05 20:41:23 -0800 | [diff] [blame] | 71 | double drivetrain_translate(int32_t in) { |
| 72 | return ((static_cast<double>(in) / |
| 73 | Values::kDrivetrainEncoderCountsPerRevolution()) * |
| 74 | (2.0 * M_PI)) * |
| 75 | Values::kDrivetrainEncoderRatio() * |
| 76 | control_loops::drivetrain::kWheelRadius; |
| 77 | } |
| 78 | |
| 79 | double drivetrain_velocity_translate(double in) { |
| 80 | return (((1.0 / in) / Values::kDrivetrainCyclesPerRevolution()) * |
| 81 | (2.0 * M_PI)) * |
| 82 | Values::kDrivetrainEncoderRatio() * |
| 83 | control_loops::drivetrain::kWheelRadius; |
| 84 | } |
| 85 | |
Alex Perry | c4691f5 | 2020-02-17 19:20:01 -0800 | [diff] [blame] | 86 | double turret_pot_translate(double voltage) { |
| 87 | return voltage * Values::kTurretPotRatio() * |
| 88 | (10.0 /*turns*/ / 5.0 /*volts*/) * (2 * M_PI /*radians*/); |
| 89 | } |
| 90 | |
Stephan Massalt | d021f97 | 2020-01-05 20:41:23 -0800 | [diff] [blame] | 91 | constexpr double kMaxFastEncoderPulsesPerSecond = |
Austin Schuh | 9dcd520 | 2020-02-20 20:06:04 -0800 | [diff] [blame] | 92 | std::max({Values::kMaxControlPanelEncoderPulsesPerSecond(), |
| 93 | Values::kMaxFinisherEncoderPulsesPerSecond(), |
| 94 | Values::kMaxAcceleratorEncoderPulsesPerSecond()}); |
| 95 | static_assert(kMaxFastEncoderPulsesPerSecond <= 1000000.0, |
Stephan Massalt | d021f97 | 2020-01-05 20:41:23 -0800 | [diff] [blame] | 96 | "fast encoders are too fast"); |
Alex Perry | c4691f5 | 2020-02-17 19:20:01 -0800 | [diff] [blame] | 97 | constexpr double kMaxMediumEncoderPulsesPerSecond = |
Austin Schuh | 9dcd520 | 2020-02-20 20:06:04 -0800 | [diff] [blame] | 98 | std::max({Values::kMaxDrivetrainEncoderPulsesPerSecond(), |
| 99 | Values::kMaxHoodEncoderPulsesPerSecond(), |
| 100 | Values::kMaxIntakeEncoderPulsesPerSecond(), |
| 101 | Values::kMaxTurretEncoderPulsesPerSecond()}); |
Stephan Massalt | d021f97 | 2020-01-05 20:41:23 -0800 | [diff] [blame] | 102 | |
Austin Schuh | 9dcd520 | 2020-02-20 20:06:04 -0800 | [diff] [blame] | 103 | static_assert(kMaxMediumEncoderPulsesPerSecond <= 400000.0, |
Stephan Massalt | d021f97 | 2020-01-05 20:41:23 -0800 | [diff] [blame] | 104 | "medium encoders are too fast"); |
| 105 | |
| 106 | } // namespace |
| 107 | |
| 108 | // Class to send position messages with sensor readings to our loops. |
| 109 | class SensorReader : public ::frc971::wpilib::SensorReader { |
| 110 | public: |
| 111 | SensorReader(::aos::ShmEventLoop *event_loop) |
| 112 | : ::frc971::wpilib::SensorReader(event_loop), |
| 113 | auto_mode_sender_( |
| 114 | event_loop->MakeSender<::frc971::autonomous::AutonomousMode>( |
| 115 | "/autonomous")), |
| 116 | superstructure_position_sender_( |
| 117 | event_loop->MakeSender<superstructure::Position>( |
| 118 | "/superstructure")), |
| 119 | drivetrain_position_sender_( |
| 120 | event_loop |
| 121 | ->MakeSender<::frc971::control_loops::drivetrain::Position>( |
| 122 | "/drivetrain")) { |
| 123 | // Set to filter out anything shorter than 1/4 of the minimum pulse width |
| 124 | // we should ever see. |
| 125 | UpdateFastEncoderFilterHz(kMaxFastEncoderPulsesPerSecond); |
| 126 | UpdateMediumEncoderFilterHz(kMaxMediumEncoderPulsesPerSecond); |
| 127 | } |
| 128 | |
Alex Perry | c4691f5 | 2020-02-17 19:20:01 -0800 | [diff] [blame] | 129 | // Hood |
Alex Perry | c4691f5 | 2020-02-17 19:20:01 -0800 | [diff] [blame] | 130 | void set_hood_encoder(::std::unique_ptr<frc::Encoder> encoder) { |
| 131 | medium_encoder_filter_.Add(encoder.get()); |
| 132 | hood_encoder_.set_encoder(::std::move(encoder)); |
| 133 | } |
| 134 | |
| 135 | void set_hood_absolute_pwm( |
| 136 | ::std::unique_ptr<frc::DigitalInput> absolute_pwm) { |
| 137 | hood_encoder_.set_absolute_pwm(::std::move(absolute_pwm)); |
| 138 | } |
| 139 | |
Ravago Jones | 937587c | 2020-12-26 17:21:09 -0800 | [diff] [blame] | 140 | void set_hood_single_turn_absolute_pwm( |
| 141 | ::std::unique_ptr<frc::DigitalInput> absolute_pwm) { |
| 142 | hood_encoder_.set_single_turn_absolute_pwm(::std::move(absolute_pwm)); |
| 143 | } |
| 144 | |
Alex Perry | c4691f5 | 2020-02-17 19:20:01 -0800 | [diff] [blame] | 145 | // Intake |
| 146 | |
| 147 | void set_intake_encoder(::std::unique_ptr<frc::Encoder> encoder) { |
| 148 | medium_encoder_filter_.Add(encoder.get()); |
| 149 | intake_joint_encoder_.set_encoder(::std::move(encoder)); |
| 150 | } |
| 151 | |
| 152 | void set_intake_absolute_pwm( |
| 153 | ::std::unique_ptr<frc::DigitalInput> absolute_pwm) { |
| 154 | intake_joint_encoder_.set_absolute_pwm(::std::move(absolute_pwm)); |
| 155 | } |
| 156 | |
| 157 | // Turret |
| 158 | |
| 159 | void set_turret_encoder(::std::unique_ptr<frc::Encoder> encoder) { |
| 160 | medium_encoder_filter_.Add(encoder.get()); |
| 161 | turret_encoder_.set_encoder(::std::move(encoder)); |
| 162 | } |
| 163 | |
| 164 | void set_turret_absolute_pwm( |
| 165 | ::std::unique_ptr<frc::DigitalInput> absolute_pwm) { |
| 166 | turret_encoder_.set_absolute_pwm(::std::move(absolute_pwm)); |
| 167 | } |
| 168 | |
| 169 | void set_turret_potentiometer( |
| 170 | ::std::unique_ptr<frc::AnalogInput> potentiometer) { |
| 171 | turret_encoder_.set_potentiometer(::std::move(potentiometer)); |
| 172 | } |
| 173 | |
| 174 | // Shooter |
Sabina Davis | c2e4bdb | 2020-02-23 13:44:58 -0800 | [diff] [blame] | 175 | void set_finisher_encoder(::std::unique_ptr<frc::Encoder> encoder) { |
Alex Perry | c4691f5 | 2020-02-17 19:20:01 -0800 | [diff] [blame] | 176 | fast_encoder_filter_.Add(encoder.get()); |
Sabina Davis | c2e4bdb | 2020-02-23 13:44:58 -0800 | [diff] [blame] | 177 | finisher_encoder_ = ::std::move(encoder); |
| 178 | } |
| 179 | void set_left_accelerator_encoder(::std::unique_ptr<frc::Encoder> encoder) { |
| 180 | fast_encoder_filter_.Add(encoder.get()); |
| 181 | left_accelerator_encoder_ = ::std::move(encoder); |
Alex Perry | c4691f5 | 2020-02-17 19:20:01 -0800 | [diff] [blame] | 182 | } |
| 183 | |
Sabina Davis | c2e4bdb | 2020-02-23 13:44:58 -0800 | [diff] [blame] | 184 | void set_right_accelerator_encoder(::std::unique_ptr<frc::Encoder> encoder) { |
Alex Perry | c4691f5 | 2020-02-17 19:20:01 -0800 | [diff] [blame] | 185 | fast_encoder_filter_.Add(encoder.get()); |
Sabina Davis | c2e4bdb | 2020-02-23 13:44:58 -0800 | [diff] [blame] | 186 | right_accelerator_encoder_ = ::std::move(encoder); |
Alex Perry | c4691f5 | 2020-02-17 19:20:01 -0800 | [diff] [blame] | 187 | } |
| 188 | |
Stephan Massalt | d021f97 | 2020-01-05 20:41:23 -0800 | [diff] [blame] | 189 | // Auto mode switches. |
| 190 | void set_autonomous_mode(int i, ::std::unique_ptr<frc::DigitalInput> sensor) { |
Austin Schuh | 9dcd520 | 2020-02-20 20:06:04 -0800 | [diff] [blame] | 191 | medium_encoder_filter_.Add(sensor.get()); |
Stephan Massalt | d021f97 | 2020-01-05 20:41:23 -0800 | [diff] [blame] | 192 | autonomous_modes_.at(i) = ::std::move(sensor); |
| 193 | } |
| 194 | |
James Kuszmaul | a244a91 | 2020-01-18 13:50:50 -0800 | [diff] [blame] | 195 | void set_imu(frc971::wpilib::ADIS16470 *imu) { imu_ = imu; } |
| 196 | |
Stephan Massalt | d021f97 | 2020-01-05 20:41:23 -0800 | [diff] [blame] | 197 | void RunIteration() override { |
James Kuszmaul | a244a91 | 2020-01-18 13:50:50 -0800 | [diff] [blame] | 198 | CHECK_NOTNULL(imu_)->DoReads(); |
| 199 | |
Stephan Massalt | d021f97 | 2020-01-05 20:41:23 -0800 | [diff] [blame] | 200 | { |
| 201 | auto builder = drivetrain_position_sender_.MakeBuilder(); |
| 202 | frc971::control_loops::drivetrain::Position::Builder drivetrain_builder = |
| 203 | builder.MakeBuilder<frc971::control_loops::drivetrain::Position>(); |
| 204 | drivetrain_builder.add_left_encoder( |
| 205 | drivetrain_translate(drivetrain_left_encoder_->GetRaw())); |
| 206 | drivetrain_builder.add_left_speed( |
| 207 | drivetrain_velocity_translate(drivetrain_left_encoder_->GetPeriod())); |
| 208 | |
| 209 | drivetrain_builder.add_right_encoder( |
| 210 | -drivetrain_translate(drivetrain_right_encoder_->GetRaw())); |
| 211 | drivetrain_builder.add_right_speed(-drivetrain_velocity_translate( |
| 212 | drivetrain_right_encoder_->GetPeriod())); |
| 213 | |
| 214 | builder.Send(drivetrain_builder.Finish()); |
| 215 | } |
Alex Perry | c4691f5 | 2020-02-17 19:20:01 -0800 | [diff] [blame] | 216 | const auto values = constants::GetValues(); |
Stephan Massalt | d021f97 | 2020-01-05 20:41:23 -0800 | [diff] [blame] | 217 | |
| 218 | { |
| 219 | auto builder = superstructure_position_sender_.MakeBuilder(); |
Austin Schuh | 18ca45f | 2020-02-29 22:58:49 -0800 | [diff] [blame] | 220 | |
Alex Perry | c4691f5 | 2020-02-17 19:20:01 -0800 | [diff] [blame] | 221 | // TODO(alex): check new absolute encoder api. |
| 222 | // Hood |
Ravago Jones | 937587c | 2020-12-26 17:21:09 -0800 | [diff] [blame] | 223 | frc971::AbsoluteAndAbsolutePositionT hood; |
Alex Perry | c4691f5 | 2020-02-17 19:20:01 -0800 | [diff] [blame] | 224 | CopyPosition(hood_encoder_, &hood, |
| 225 | Values::kHoodEncoderCountsPerRevolution(), |
Ravago Jones | 937587c | 2020-12-26 17:21:09 -0800 | [diff] [blame] | 226 | Values::kHoodEncoderRatio(), |
| 227 | Values::kHoodSingleTurnEncoderRatio(), false); |
| 228 | flatbuffers::Offset<frc971::AbsoluteAndAbsolutePosition> hood_offset = |
| 229 | frc971::AbsoluteAndAbsolutePosition::Pack(*builder.fbb(), &hood); |
Alex Perry | c4691f5 | 2020-02-17 19:20:01 -0800 | [diff] [blame] | 230 | |
| 231 | // Intake |
| 232 | frc971::AbsolutePositionT intake_joint; |
| 233 | CopyPosition(intake_joint_encoder_, &intake_joint, |
| 234 | Values::kIntakeEncoderCountsPerRevolution(), |
| 235 | Values::kIntakeEncoderRatio(), false); |
| 236 | flatbuffers::Offset<frc971::AbsolutePosition> intake_joint_offset = |
| 237 | frc971::AbsolutePosition::Pack(*builder.fbb(), &intake_joint); |
| 238 | |
| 239 | // Turret |
| 240 | frc971::PotAndAbsolutePositionT turret; |
| 241 | CopyPosition(turret_encoder_, &turret, |
| 242 | Values::kTurretEncoderCountsPerRevolution(), |
Sabina Davis | f7afd11 | 2020-02-23 13:42:14 -0800 | [diff] [blame] | 243 | Values::kTurretEncoderRatio(), turret_pot_translate, true, |
Alex Perry | c4691f5 | 2020-02-17 19:20:01 -0800 | [diff] [blame] | 244 | values.turret.potentiometer_offset); |
| 245 | flatbuffers::Offset<frc971::PotAndAbsolutePosition> turret_offset = |
| 246 | frc971::PotAndAbsolutePosition::Pack(*builder.fbb(), &turret); |
| 247 | |
| 248 | // Shooter |
| 249 | y2020::control_loops::superstructure::ShooterPositionT shooter; |
| 250 | shooter.theta_finisher = |
Austin Schuh | 0ad31d7 | 2021-03-06 17:07:04 -0800 | [diff] [blame] | 251 | encoder_translate(-finisher_encoder_->GetRaw(), |
Alex Perry | c4691f5 | 2020-02-17 19:20:01 -0800 | [diff] [blame] | 252 | Values::kFinisherEncoderCountsPerRevolution(), |
| 253 | Values::kFinisherEncoderRatio()); |
Sabina Davis | c2e4bdb | 2020-02-23 13:44:58 -0800 | [diff] [blame] | 254 | |
Alex Perry | c4691f5 | 2020-02-17 19:20:01 -0800 | [diff] [blame] | 255 | shooter.theta_accelerator_left = |
Sabina Davis | c2e4bdb | 2020-02-23 13:44:58 -0800 | [diff] [blame] | 256 | encoder_translate(-left_accelerator_encoder_->GetRaw(), |
Alex Perry | c4691f5 | 2020-02-17 19:20:01 -0800 | [diff] [blame] | 257 | Values::kAcceleratorEncoderCountsPerRevolution(), |
| 258 | Values::kAcceleratorEncoderRatio()); |
| 259 | shooter.theta_accelerator_right = |
Sabina Davis | c2e4bdb | 2020-02-23 13:44:58 -0800 | [diff] [blame] | 260 | encoder_translate(right_accelerator_encoder_->GetRaw(), |
Alex Perry | c4691f5 | 2020-02-17 19:20:01 -0800 | [diff] [blame] | 261 | Values::kAcceleratorEncoderCountsPerRevolution(), |
| 262 | Values::kAcceleratorEncoderRatio()); |
| 263 | flatbuffers::Offset<y2020::control_loops::superstructure::ShooterPosition> |
| 264 | shooter_offset = |
| 265 | y2020::control_loops::superstructure::ShooterPosition::Pack( |
| 266 | *builder.fbb(), &shooter); |
| 267 | |
Austin Schuh | 18ca45f | 2020-02-29 22:58:49 -0800 | [diff] [blame] | 268 | superstructure::Position::Builder position_builder = |
| 269 | builder.MakeBuilder<superstructure::Position>(); |
Alex Perry | c4691f5 | 2020-02-17 19:20:01 -0800 | [diff] [blame] | 270 | position_builder.add_hood(hood_offset); |
| 271 | position_builder.add_intake_joint(intake_joint_offset); |
| 272 | position_builder.add_turret(turret_offset); |
| 273 | position_builder.add_shooter(shooter_offset); |
Alex Perry | c4691f5 | 2020-02-17 19:20:01 -0800 | [diff] [blame] | 274 | |
Stephan Massalt | d021f97 | 2020-01-05 20:41:23 -0800 | [diff] [blame] | 275 | builder.Send(position_builder.Finish()); |
| 276 | } |
| 277 | |
| 278 | { |
| 279 | auto builder = auto_mode_sender_.MakeBuilder(); |
| 280 | |
| 281 | uint32_t mode = 0; |
| 282 | for (size_t i = 0; i < autonomous_modes_.size(); ++i) { |
| 283 | if (autonomous_modes_[i] && autonomous_modes_[i]->Get()) { |
| 284 | mode |= 1 << i; |
| 285 | } |
| 286 | } |
| 287 | |
| 288 | auto auto_mode_builder = |
| 289 | builder.MakeBuilder<frc971::autonomous::AutonomousMode>(); |
| 290 | |
| 291 | auto_mode_builder.add_mode(mode); |
| 292 | |
| 293 | builder.Send(auto_mode_builder.Finish()); |
| 294 | } |
| 295 | } |
| 296 | |
| 297 | private: |
| 298 | ::aos::Sender<::frc971::autonomous::AutonomousMode> auto_mode_sender_; |
| 299 | ::aos::Sender<superstructure::Position> superstructure_position_sender_; |
| 300 | ::aos::Sender<::frc971::control_loops::drivetrain::Position> |
| 301 | drivetrain_position_sender_; |
| 302 | |
Alex Perry | c4691f5 | 2020-02-17 19:20:01 -0800 | [diff] [blame] | 303 | ::frc971::wpilib::AbsoluteEncoderAndPotentiometer turret_encoder_; |
| 304 | |
Ravago Jones | 937587c | 2020-12-26 17:21:09 -0800 | [diff] [blame] | 305 | ::frc971::wpilib::AbsoluteAndAbsoluteEncoder hood_encoder_; |
| 306 | |
| 307 | ::frc971::wpilib::AbsoluteEncoder intake_joint_encoder_; |
Alex Perry | c4691f5 | 2020-02-17 19:20:01 -0800 | [diff] [blame] | 308 | |
Sabina Davis | c2e4bdb | 2020-02-23 13:44:58 -0800 | [diff] [blame] | 309 | ::std::unique_ptr<::frc::Encoder> finisher_encoder_, |
milind-u | 3d68aa7 | 2021-09-26 12:19:03 -0700 | [diff] [blame^] | 310 | left_accelerator_encoder_, right_accelerator_encoder_; |
Stephan Massalt | d021f97 | 2020-01-05 20:41:23 -0800 | [diff] [blame] | 311 | ::std::array<::std::unique_ptr<frc::DigitalInput>, 2> autonomous_modes_; |
James Kuszmaul | a244a91 | 2020-01-18 13:50:50 -0800 | [diff] [blame] | 312 | |
| 313 | frc971::wpilib::ADIS16470 *imu_ = nullptr; |
Stephan Massalt | d021f97 | 2020-01-05 20:41:23 -0800 | [diff] [blame] | 314 | }; |
| 315 | |
| 316 | class SuperstructureWriter |
| 317 | : public ::frc971::wpilib::LoopOutputHandler<superstructure::Output> { |
| 318 | public: |
| 319 | SuperstructureWriter(::aos::EventLoop *event_loop) |
| 320 | : ::frc971::wpilib::LoopOutputHandler<superstructure::Output>( |
| 321 | event_loop, "/superstructure") {} |
| 322 | |
Alex Perry | c4691f5 | 2020-02-17 19:20:01 -0800 | [diff] [blame] | 323 | void set_hood_victor(::std::unique_ptr<::frc::VictorSP> t) { |
| 324 | hood_victor_ = ::std::move(t); |
| 325 | } |
Stephan Massalt | d021f97 | 2020-01-05 20:41:23 -0800 | [diff] [blame] | 326 | |
Alex Perry | c4691f5 | 2020-02-17 19:20:01 -0800 | [diff] [blame] | 327 | void set_intake_joint_victor(::std::unique_ptr<::frc::VictorSP> t) { |
| 328 | intake_joint_victor_ = ::std::move(t); |
| 329 | } |
| 330 | |
| 331 | void set_intake_roller_falcon( |
| 332 | ::std::unique_ptr<::ctre::phoenix::motorcontrol::can::TalonFX> t) { |
| 333 | intake_roller_falcon_ = ::std::move(t); |
| 334 | intake_roller_falcon_->ConfigSupplyCurrentLimit( |
| 335 | {true, Values::kIntakeRollerSupplyCurrentLimit(), |
| 336 | Values::kIntakeRollerSupplyCurrentLimit(), 0}); |
| 337 | intake_roller_falcon_->ConfigStatorCurrentLimit( |
| 338 | {true, Values::kIntakeRollerStatorCurrentLimit(), |
| 339 | Values::kIntakeRollerStatorCurrentLimit(), 0}); |
| 340 | } |
| 341 | |
| 342 | void set_turret_victor(::std::unique_ptr<::frc::VictorSP> t) { |
| 343 | turret_victor_ = ::std::move(t); |
| 344 | } |
| 345 | |
Ravago Jones | 3dda560 | 2021-03-10 00:33:13 -0800 | [diff] [blame] | 346 | void set_feeder_falcon( |
| 347 | ::std::unique_ptr<::ctre::phoenix::motorcontrol::can::TalonFX> t) { |
Alex Perry | c4691f5 | 2020-02-17 19:20:01 -0800 | [diff] [blame] | 348 | feeder_falcon_ = ::std::move(t); |
Austin Schuh | 3399d12 | 2021-03-20 14:29:20 -0700 | [diff] [blame] | 349 | { |
| 350 | auto result = feeder_falcon_->ConfigSupplyCurrentLimit( |
| 351 | {true, Values::kFeederSupplyCurrentLimit(), |
| 352 | Values::kFeederSupplyCurrentLimit(), 0}); |
| 353 | if (result != ctre::phoenix::OKAY) { |
| 354 | LOG(WARNING) << "Failed to configure feeder supply current limit: " |
| 355 | << result; |
| 356 | } |
| 357 | } |
| 358 | { |
| 359 | auto result = feeder_falcon_->ConfigStatorCurrentLimit( |
| 360 | {true, Values::kFeederStatorCurrentLimit(), |
| 361 | Values::kFeederStatorCurrentLimit(), 0}); |
| 362 | if (result != ctre::phoenix::OKAY) { |
| 363 | LOG(WARNING) << "Failed to configure feeder stator current limit: " |
| 364 | << result; |
| 365 | } |
| 366 | } |
Alex Perry | c4691f5 | 2020-02-17 19:20:01 -0800 | [diff] [blame] | 367 | } |
| 368 | |
Sabina Davis | c2e4bdb | 2020-02-23 13:44:58 -0800 | [diff] [blame] | 369 | void set_accelerator_left_falcon(::std::unique_ptr<::frc::TalonFX> t) { |
| 370 | accelerator_left_falcon_ = ::std::move(t); |
Alex Perry | c4691f5 | 2020-02-17 19:20:01 -0800 | [diff] [blame] | 371 | } |
| 372 | |
Sabina Davis | c2e4bdb | 2020-02-23 13:44:58 -0800 | [diff] [blame] | 373 | void set_accelerator_right_falcon(::std::unique_ptr<::frc::TalonFX> t) { |
| 374 | accelerator_right_falcon_ = ::std::move(t); |
Alex Perry | c4691f5 | 2020-02-17 19:20:01 -0800 | [diff] [blame] | 375 | } |
| 376 | |
Austin Schuh | 0ad31d7 | 2021-03-06 17:07:04 -0800 | [diff] [blame] | 377 | void set_finisher_falcon0(::std::unique_ptr<::frc::TalonFX> t) { |
| 378 | finisher_falcon0_ = ::std::move(t); |
| 379 | } |
| 380 | |
| 381 | void set_finisher_falcon1(::std::unique_ptr<::frc::TalonFX> t) { |
| 382 | finisher_falcon1_ = ::std::move(t); |
Alex Perry | c4691f5 | 2020-02-17 19:20:01 -0800 | [diff] [blame] | 383 | } |
| 384 | |
| 385 | void set_climber_falcon( |
| 386 | ::std::unique_ptr<::ctre::phoenix::motorcontrol::can::TalonFX> t) { |
| 387 | climber_falcon_ = ::std::move(t); |
| 388 | climber_falcon_->ConfigSupplyCurrentLimit( |
| 389 | {true, Values::kClimberSupplyCurrentLimit(), |
| 390 | Values::kClimberSupplyCurrentLimit(), 0}); |
| 391 | } |
| 392 | |
| 393 | private: |
| 394 | void Write(const superstructure::Output &output) override { |
Austin Schuh | 2efe168 | 2021-03-06 22:47:15 -0800 | [diff] [blame] | 395 | hood_victor_->SetSpeed(std::clamp(-output.hood_voltage(), -kMaxBringupPower, |
| 396 | kMaxBringupPower) / |
| 397 | 12.0); |
Alex Perry | c4691f5 | 2020-02-17 19:20:01 -0800 | [diff] [blame] | 398 | |
Sabina Davis | c2e4bdb | 2020-02-23 13:44:58 -0800 | [diff] [blame] | 399 | intake_joint_victor_->SetSpeed(std::clamp(-output.intake_joint_voltage(), |
| 400 | -kMaxBringupPower, |
| 401 | kMaxBringupPower) / |
Alex Perry | c4691f5 | 2020-02-17 19:20:01 -0800 | [diff] [blame] | 402 | 12.0); |
| 403 | |
| 404 | intake_roller_falcon_->Set( |
| 405 | ctre::phoenix::motorcontrol::ControlMode::PercentOutput, |
Sabina Davis | c2e4bdb | 2020-02-23 13:44:58 -0800 | [diff] [blame] | 406 | std::clamp(-output.intake_roller_voltage(), -kMaxBringupPower, |
| 407 | kMaxBringupPower) / |
Alex Perry | c4691f5 | 2020-02-17 19:20:01 -0800 | [diff] [blame] | 408 | 12.0); |
| 409 | |
Sabina Davis | c2e4bdb | 2020-02-23 13:44:58 -0800 | [diff] [blame] | 410 | turret_victor_->SetSpeed(std::clamp(-output.turret_voltage(), |
| 411 | -kMaxBringupPower, kMaxBringupPower) / |
Alex Perry | c4691f5 | 2020-02-17 19:20:01 -0800 | [diff] [blame] | 412 | 12.0); |
| 413 | |
Ravago Jones | 3dda560 | 2021-03-10 00:33:13 -0800 | [diff] [blame] | 414 | feeder_falcon_->Set(ctre::phoenix::motorcontrol::ControlMode::PercentOutput, |
| 415 | std::clamp(output.feeder_voltage(), -kMaxBringupPower, |
| 416 | kMaxBringupPower) / |
| 417 | 12.0); |
Alex Perry | c4691f5 | 2020-02-17 19:20:01 -0800 | [diff] [blame] | 418 | |
Sabina Davis | c2e4bdb | 2020-02-23 13:44:58 -0800 | [diff] [blame] | 419 | accelerator_left_falcon_->SetSpeed( |
| 420 | std::clamp(-output.accelerator_left_voltage(), -kMaxBringupPower, |
| 421 | kMaxBringupPower) / |
| 422 | 12.0); |
Alex Perry | c4691f5 | 2020-02-17 19:20:01 -0800 | [diff] [blame] | 423 | |
Sabina Davis | c2e4bdb | 2020-02-23 13:44:58 -0800 | [diff] [blame] | 424 | accelerator_right_falcon_->SetSpeed( |
Alex Perry | c4691f5 | 2020-02-17 19:20:01 -0800 | [diff] [blame] | 425 | std::clamp(output.accelerator_right_voltage(), -kMaxBringupPower, |
Sabina Davis | c2e4bdb | 2020-02-23 13:44:58 -0800 | [diff] [blame] | 426 | kMaxBringupPower) / |
Alex Perry | c4691f5 | 2020-02-17 19:20:01 -0800 | [diff] [blame] | 427 | 12.0); |
| 428 | |
Austin Schuh | 0ad31d7 | 2021-03-06 17:07:04 -0800 | [diff] [blame] | 429 | finisher_falcon1_->SetSpeed(std::clamp(output.finisher_voltage(), |
| 430 | -kMaxBringupPower, |
| 431 | kMaxBringupPower) / |
| 432 | 12.0); |
| 433 | finisher_falcon0_->SetSpeed(std::clamp(-output.finisher_voltage(), |
| 434 | -kMaxBringupPower, |
| 435 | kMaxBringupPower) / |
| 436 | 12.0); |
Alex Perry | c4691f5 | 2020-02-17 19:20:01 -0800 | [diff] [blame] | 437 | |
Sabina Davis | c2e4bdb | 2020-02-23 13:44:58 -0800 | [diff] [blame] | 438 | if (climber_falcon_) { |
| 439 | climber_falcon_->Set( |
| 440 | ctre::phoenix::motorcontrol::ControlMode::PercentOutput, |
Tyler Chatow | 1039e43 | 2020-02-28 21:37:50 -0800 | [diff] [blame] | 441 | std::clamp(-output.climber_voltage(), -kMaxBringupPower, |
Sabina Davis | c2e4bdb | 2020-02-23 13:44:58 -0800 | [diff] [blame] | 442 | kMaxBringupPower) / |
| 443 | 12.0); |
| 444 | } |
Alex Perry | c4691f5 | 2020-02-17 19:20:01 -0800 | [diff] [blame] | 445 | } |
| 446 | |
| 447 | void Stop() override { |
| 448 | AOS_LOG(WARNING, "Superstructure output too old.\n"); |
| 449 | hood_victor_->SetDisabled(); |
| 450 | intake_joint_victor_->SetDisabled(); |
| 451 | turret_victor_->SetDisabled(); |
Sabina Davis | c2e4bdb | 2020-02-23 13:44:58 -0800 | [diff] [blame] | 452 | accelerator_left_falcon_->SetDisabled(); |
| 453 | accelerator_right_falcon_->SetDisabled(); |
Austin Schuh | 0ad31d7 | 2021-03-06 17:07:04 -0800 | [diff] [blame] | 454 | finisher_falcon0_->SetDisabled(); |
| 455 | finisher_falcon1_->SetDisabled(); |
Ravago Jones | 3dda560 | 2021-03-10 00:33:13 -0800 | [diff] [blame] | 456 | feeder_falcon_->Set(ctre::phoenix::motorcontrol::ControlMode::Disabled, 0); |
| 457 | intake_roller_falcon_->Set( |
| 458 | ctre::phoenix::motorcontrol::ControlMode::Disabled, 0); |
| 459 | climber_falcon_->Set(ctre::phoenix::motorcontrol::ControlMode::Disabled, 0); |
Alex Perry | c4691f5 | 2020-02-17 19:20:01 -0800 | [diff] [blame] | 460 | } |
| 461 | |
| 462 | ::std::unique_ptr<::frc::VictorSP> hood_victor_, intake_joint_victor_, |
milind-u | 3d68aa7 | 2021-09-26 12:19:03 -0700 | [diff] [blame^] | 463 | turret_victor_; |
Alex Perry | c4691f5 | 2020-02-17 19:20:01 -0800 | [diff] [blame] | 464 | |
Ravago Jones | 3dda560 | 2021-03-10 00:33:13 -0800 | [diff] [blame] | 465 | ::std::unique_ptr<::frc::TalonFX> accelerator_left_falcon_, |
Austin Schuh | 0ad31d7 | 2021-03-06 17:07:04 -0800 | [diff] [blame] | 466 | accelerator_right_falcon_, finisher_falcon0_, finisher_falcon1_; |
Alex Perry | c4691f5 | 2020-02-17 19:20:01 -0800 | [diff] [blame] | 467 | |
| 468 | ::std::unique_ptr<::ctre::phoenix::motorcontrol::can::TalonFX> |
Ravago Jones | 3dda560 | 2021-03-10 00:33:13 -0800 | [diff] [blame] | 469 | intake_roller_falcon_, climber_falcon_, feeder_falcon_; |
Stephan Massalt | d021f97 | 2020-01-05 20:41:23 -0800 | [diff] [blame] | 470 | }; |
| 471 | |
| 472 | class WPILibRobot : public ::frc971::wpilib::WPILibRobotBase { |
| 473 | public: |
Alex Perry | c4691f5 | 2020-02-17 19:20:01 -0800 | [diff] [blame] | 474 | ::std::unique_ptr<frc::Encoder> make_encoder( |
| 475 | int index, frc::Encoder::EncodingType encodingType = frc::Encoder::k4X) { |
Stephan Massalt | d021f97 | 2020-01-05 20:41:23 -0800 | [diff] [blame] | 476 | return make_unique<frc::Encoder>(10 + index * 2, 11 + index * 2, false, |
Alex Perry | c4691f5 | 2020-02-17 19:20:01 -0800 | [diff] [blame] | 477 | encodingType); |
Stephan Massalt | d021f97 | 2020-01-05 20:41:23 -0800 | [diff] [blame] | 478 | } |
| 479 | |
| 480 | void Run() override { |
| 481 | aos::FlatbufferDetachedBuffer<aos::Configuration> config = |
| 482 | aos::configuration::ReadConfig("config.json"); |
| 483 | |
| 484 | // Thread 1. |
| 485 | ::aos::ShmEventLoop joystick_sender_event_loop(&config.message()); |
| 486 | ::frc971::wpilib::JoystickSender joystick_sender( |
| 487 | &joystick_sender_event_loop); |
| 488 | AddLoop(&joystick_sender_event_loop); |
| 489 | |
| 490 | // Thread 2. |
| 491 | ::aos::ShmEventLoop pdp_fetcher_event_loop(&config.message()); |
| 492 | ::frc971::wpilib::PDPFetcher pdp_fetcher(&pdp_fetcher_event_loop); |
| 493 | AddLoop(&pdp_fetcher_event_loop); |
| 494 | |
| 495 | // Thread 3. |
| 496 | ::aos::ShmEventLoop sensor_reader_event_loop(&config.message()); |
| 497 | SensorReader sensor_reader(&sensor_reader_event_loop); |
Austin Schuh | f7db58c | 2020-02-29 22:57:43 -0800 | [diff] [blame] | 498 | sensor_reader.set_pwm_trigger(true); |
Stephan Massalt | d021f97 | 2020-01-05 20:41:23 -0800 | [diff] [blame] | 499 | sensor_reader.set_drivetrain_left_encoder(make_encoder(0)); |
| 500 | sensor_reader.set_drivetrain_right_encoder(make_encoder(1)); |
Alex Perry | c4691f5 | 2020-02-17 19:20:01 -0800 | [diff] [blame] | 501 | // TODO: pin numbers |
Sabina Davis | f7afd11 | 2020-02-23 13:42:14 -0800 | [diff] [blame] | 502 | sensor_reader.set_hood_encoder( |
| 503 | make_unique<frc::Encoder>(22, 23, false, frc::Encoder::k4X)); |
Alex Perry | c4691f5 | 2020-02-17 19:20:01 -0800 | [diff] [blame] | 504 | |
Ravago Jones | 937587c | 2020-12-26 17:21:09 -0800 | [diff] [blame] | 505 | sensor_reader.set_hood_absolute_pwm(make_unique<frc::DigitalInput>(25)); |
| 506 | sensor_reader.set_hood_single_turn_absolute_pwm( |
| 507 | make_unique<frc::DigitalInput>(24)); |
Alex Perry | c4691f5 | 2020-02-17 19:20:01 -0800 | [diff] [blame] | 508 | |
Sabina Davis | f7afd11 | 2020-02-23 13:42:14 -0800 | [diff] [blame] | 509 | sensor_reader.set_intake_encoder(make_encoder(5)); |
| 510 | sensor_reader.set_intake_absolute_pwm(make_unique<frc::DigitalInput>(1)); |
Alex Perry | c4691f5 | 2020-02-17 19:20:01 -0800 | [diff] [blame] | 511 | |
Sabina Davis | f7afd11 | 2020-02-23 13:42:14 -0800 | [diff] [blame] | 512 | sensor_reader.set_turret_encoder(make_encoder(2)); |
| 513 | sensor_reader.set_turret_absolute_pwm(make_unique<frc::DigitalInput>(0)); |
| 514 | sensor_reader.set_turret_potentiometer(make_unique<frc::AnalogInput>(0)); |
Alex Perry | c4691f5 | 2020-02-17 19:20:01 -0800 | [diff] [blame] | 515 | |
Sabina Davis | f7afd11 | 2020-02-23 13:42:14 -0800 | [diff] [blame] | 516 | sensor_reader.set_finisher_encoder( |
| 517 | make_unique<frc::Encoder>(3, 2, false, frc::Encoder::k4X)); |
| 518 | sensor_reader.set_left_accelerator_encoder(make_encoder(4)); |
| 519 | sensor_reader.set_right_accelerator_encoder(make_encoder(3)); |
| 520 | |
James Kuszmaul | 022d40e | 2020-02-11 17:06:18 -0800 | [diff] [blame] | 521 | // Note: If ADIS16470 is plugged in directly to the roboRIO SPI port without |
| 522 | // the Spartan Board, then trigger is on 26, reset 27, and chip select is |
| 523 | // CS0. |
Austin Schuh | 83873c3 | 2020-02-22 14:58:39 -0800 | [diff] [blame] | 524 | frc::SPI::Port spi_port = frc::SPI::Port::kOnboardCS2; |
| 525 | std::unique_ptr<frc::DigitalInput> imu_trigger; |
| 526 | std::unique_ptr<frc::DigitalOutput> imu_reset; |
| 527 | if (::aos::network::GetTeamNumber() == |
| 528 | constants::Values::kCodingRobotTeamNumber) { |
| 529 | imu_trigger = make_unique<frc::DigitalInput>(26); |
| 530 | imu_reset = make_unique<frc::DigitalOutput>(27); |
| 531 | spi_port = frc::SPI::Port::kOnboardCS0; |
| 532 | } else { |
Sabina Davis | f7afd11 | 2020-02-23 13:42:14 -0800 | [diff] [blame] | 533 | imu_trigger = make_unique<frc::DigitalInput>(9); |
| 534 | imu_reset = make_unique<frc::DigitalOutput>(8); |
Austin Schuh | 83873c3 | 2020-02-22 14:58:39 -0800 | [diff] [blame] | 535 | } |
| 536 | auto spi = make_unique<frc::SPI>(spi_port); |
James Kuszmaul | a244a91 | 2020-01-18 13:50:50 -0800 | [diff] [blame] | 537 | frc971::wpilib::ADIS16470 imu(&sensor_reader_event_loop, spi.get(), |
| 538 | imu_trigger.get(), imu_reset.get()); |
| 539 | sensor_reader.set_imu(&imu); |
Stephan Massalt | d021f97 | 2020-01-05 20:41:23 -0800 | [diff] [blame] | 540 | AddLoop(&sensor_reader_event_loop); |
| 541 | |
| 542 | // Thread 4. |
| 543 | ::aos::ShmEventLoop output_event_loop(&config.message()); |
James Kuszmaul | 57c2baa | 2020-01-19 14:52:52 -0800 | [diff] [blame] | 544 | output_event_loop.set_name("output_writer"); |
Stephan Massalt | d021f97 | 2020-01-05 20:41:23 -0800 | [diff] [blame] | 545 | ::frc971::wpilib::DrivetrainWriter drivetrain_writer(&output_event_loop); |
| 546 | drivetrain_writer.set_left_controller0( |
| 547 | ::std::unique_ptr<::frc::VictorSP>(new ::frc::VictorSP(0)), true); |
| 548 | drivetrain_writer.set_right_controller0( |
| 549 | ::std::unique_ptr<::frc::VictorSP>(new ::frc::VictorSP(1)), false); |
| 550 | |
| 551 | SuperstructureWriter superstructure_writer(&output_event_loop); |
Sabina Davis | c2e4bdb | 2020-02-23 13:44:58 -0800 | [diff] [blame] | 552 | superstructure_writer.set_hood_victor(make_unique<frc::VictorSP>(8)); |
Alex Perry | c4691f5 | 2020-02-17 19:20:01 -0800 | [diff] [blame] | 553 | superstructure_writer.set_intake_joint_victor( |
Sabina Davis | c2e4bdb | 2020-02-23 13:44:58 -0800 | [diff] [blame] | 554 | make_unique<frc::VictorSP>(2)); |
Alex Perry | c4691f5 | 2020-02-17 19:20:01 -0800 | [diff] [blame] | 555 | superstructure_writer.set_intake_roller_falcon( |
Sabina Davis | c2e4bdb | 2020-02-23 13:44:58 -0800 | [diff] [blame] | 556 | make_unique<::ctre::phoenix::motorcontrol::can::TalonFX>(0)); |
| 557 | superstructure_writer.set_turret_victor(make_unique<frc::VictorSP>(7)); |
Ravago Jones | 3dda560 | 2021-03-10 00:33:13 -0800 | [diff] [blame] | 558 | superstructure_writer.set_feeder_falcon( |
| 559 | make_unique<ctre::phoenix::motorcontrol::can::TalonFX>(1)); |
Sabina Davis | c2e4bdb | 2020-02-23 13:44:58 -0800 | [diff] [blame] | 560 | superstructure_writer.set_accelerator_left_falcon( |
| 561 | make_unique<::frc::TalonFX>(5)); |
| 562 | superstructure_writer.set_accelerator_right_falcon( |
| 563 | make_unique<::frc::TalonFX>(4)); |
Austin Schuh | 0ad31d7 | 2021-03-06 17:07:04 -0800 | [diff] [blame] | 564 | superstructure_writer.set_finisher_falcon0(make_unique<::frc::TalonFX>(9)); |
| 565 | superstructure_writer.set_finisher_falcon1(make_unique<::frc::TalonFX>(3)); |
Sabina Davis | c2e4bdb | 2020-02-23 13:44:58 -0800 | [diff] [blame] | 566 | // TODO: check port |
Tyler Chatow | 1039e43 | 2020-02-28 21:37:50 -0800 | [diff] [blame] | 567 | superstructure_writer.set_climber_falcon( |
Ravago Jones | 3dda560 | 2021-03-10 00:33:13 -0800 | [diff] [blame] | 568 | make_unique<::ctre::phoenix::motorcontrol::can::TalonFX>(2)); |
Stephan Massalt | d021f97 | 2020-01-05 20:41:23 -0800 | [diff] [blame] | 569 | |
| 570 | AddLoop(&output_event_loop); |
| 571 | |
| 572 | RunLoops(); |
| 573 | } |
| 574 | }; |
| 575 | |
| 576 | } // namespace wpilib |
| 577 | } // namespace y2020 |
| 578 | |
| 579 | AOS_ROBOT_CLASS(::y2020::wpilib::WPILibRobot); |