blob: 28685393e2eb6420c9d326a5c4110070c6e7647a [file] [log] [blame]
milind-u086d7262022-01-19 20:44:18 -08001#include <unistd.h>
2
3#include <array>
4#include <chrono>
5#include <cinttypes>
6#include <cmath>
7#include <cstdio>
8#include <cstring>
9#include <functional>
10#include <memory>
11#include <mutex>
12#include <thread>
13
14#include "ctre/phoenix/CANifier.h"
15#include "frc971/wpilib/ahal/AnalogInput.h"
16#include "frc971/wpilib/ahal/Counter.h"
17#include "frc971/wpilib/ahal/DigitalGlitchFilter.h"
18#include "frc971/wpilib/ahal/DriverStation.h"
19#include "frc971/wpilib/ahal/Encoder.h"
20#include "frc971/wpilib/ahal/TalonFX.h"
21#include "frc971/wpilib/ahal/VictorSP.h"
22#undef ERROR
23
24#include "aos/commonmath.h"
25#include "aos/events/event_loop.h"
26#include "aos/events/shm_event_loop.h"
27#include "aos/init.h"
28#include "aos/logging/logging.h"
29#include "aos/realtime.h"
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"
34#include "ctre/phoenix/motorcontrol/can/TalonFX.h"
35#include "ctre/phoenix/motorcontrol/can/TalonSRX.h"
36#include "frc971/autonomous/auto_mode_generated.h"
37#include "frc971/control_loops/drivetrain/drivetrain_position_generated.h"
38#include "frc971/input/robot_state_generated.h"
39#include "frc971/wpilib/ADIS16448.h"
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 "y2022/constants.h"
52#include "y2022/control_loops/superstructure/superstructure_output_generated.h"
53#include "y2022/control_loops/superstructure/superstructure_position_generated.h"
54
55using ::aos::monotonic_clock;
56using ::y2022::constants::Values;
57namespace superstructure = ::y2022::control_loops::superstructure;
58namespace chrono = ::std::chrono;
59using std::make_unique;
60
61namespace y2022 {
62namespace wpilib {
63namespace {
64
65constexpr 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
milind-u086d7262022-01-19 20:44:18 -080071double drivetrain_velocity_translate(double in) {
72 return (((1.0 / in) / Values::kDrivetrainCyclesPerRevolution()) *
73 (2.0 * M_PI)) *
74 Values::kDrivetrainEncoderRatio() *
75 control_loops::drivetrain::kWheelRadius;
76}
77
Siddhant Kanwar0e37f592022-02-21 19:26:50 -080078double climber_pot_translate(double voltage) {
79 return voltage * Values::kClimberPotRatio() *
80 (10.0 /*turns*/ / 5.0 /*volts*/) *
81 Values::kClimberPotMetersPerRevolution();
82}
83
Griffin Buibcbef482022-02-23 15:32:10 -080084double flipper_arms_pot_translate(double voltage) {
85 return voltage * Values::kFlipperArmsPotRatio() *
86 (3.0 /*turns*/ / 5.0 /*volts*/) * (2 * M_PI /*radians*/);
87}
88
Henry Speiser55aa3ba2022-02-21 23:21:12 -080089double intake_pot_translate(double voltage) {
90 return voltage * Values::kIntakePotRatio() * (3.0 /*turns*/ / 5.0 /*volts*/) *
91 (2 * M_PI /*radians*/);
92}
93
94double turret_pot_translate(double voltage) {
95 return voltage * Values::kTurretPotRatio() *
96 (10.0 /*turns*/ / 5.0 /*volts*/) * (2 * M_PI /*radians*/);
97}
98
milind-u086d7262022-01-19 20:44:18 -080099constexpr double kMaxFastEncoderPulsesPerSecond =
Henry Speiser55aa3ba2022-02-21 23:21:12 -0800100 std::max({Values::kMaxDrivetrainEncoderPulsesPerSecond(),
101 Values::kMaxIntakeEncoderPulsesPerSecond()});
milind-u086d7262022-01-19 20:44:18 -0800102static_assert(kMaxFastEncoderPulsesPerSecond <= 1300000,
103 "fast encoders are too fast");
104constexpr double kMaxMediumEncoderPulsesPerSecond =
Henry Speiser55aa3ba2022-02-21 23:21:12 -0800105 Values::kMaxTurretEncoderPulsesPerSecond();
milind-u086d7262022-01-19 20:44:18 -0800106
107static_assert(kMaxMediumEncoderPulsesPerSecond <= 400000,
108 "medium encoders are too fast");
109
Austin Schuh39f26f62022-02-24 21:34:46 -0800110double catapult_pot_translate(double voltage) {
111 return voltage * Values::kCatapultPotRatio() *
112 (3.0 /*turns*/ / 5.0 /*volts*/) * (2 * M_PI /*radians*/);
113}
114
milind-u086d7262022-01-19 20:44:18 -0800115} // namespace
116
117// Class to send position messages with sensor readings to our loops.
118class SensorReader : public ::frc971::wpilib::SensorReader {
119 public:
Henry Speiser55aa3ba2022-02-21 23:21:12 -0800120 SensorReader(::aos::ShmEventLoop *event_loop,
121 std::shared_ptr<const Values> values)
milind-u086d7262022-01-19 20:44:18 -0800122 : ::frc971::wpilib::SensorReader(event_loop),
Henry Speiser55aa3ba2022-02-21 23:21:12 -0800123 values_(std::move(values)),
milind-u086d7262022-01-19 20:44:18 -0800124 auto_mode_sender_(
125 event_loop->MakeSender<::frc971::autonomous::AutonomousMode>(
126 "/autonomous")),
127 superstructure_position_sender_(
128 event_loop->MakeSender<superstructure::Position>(
129 "/superstructure")),
130 drivetrain_position_sender_(
131 event_loop
132 ->MakeSender<::frc971::control_loops::drivetrain::Position>(
133 "/drivetrain")) {
134 // Set to filter out anything shorter than 1/4 of the minimum pulse width
135 // we should ever see.
136 UpdateFastEncoderFilterHz(kMaxFastEncoderPulsesPerSecond);
137 UpdateMediumEncoderFilterHz(kMaxMediumEncoderPulsesPerSecond);
138 }
139
140 // Auto mode switches.
141 void set_autonomous_mode(int i, ::std::unique_ptr<frc::DigitalInput> sensor) {
142 autonomous_modes_.at(i) = ::std::move(sensor);
143 }
144
Austin Schuh39f26f62022-02-24 21:34:46 -0800145 void set_catapult_encoder(::std::unique_ptr<frc::Encoder> encoder) {
146 medium_encoder_filter_.Add(encoder.get());
147 catapult_encoder_.set_encoder(::std::move(encoder));
148 }
149
150 void set_catapult_absolute_pwm(
151 ::std::unique_ptr<frc::DigitalInput> absolute_pwm) {
152 catapult_encoder_.set_absolute_pwm(::std::move(absolute_pwm));
153 }
154
155 void set_catapult_potentiometer(
156 ::std::unique_ptr<frc::AnalogInput> potentiometer) {
157 catapult_encoder_.set_potentiometer(::std::move(potentiometer));
158 }
159
milind-u086d7262022-01-19 20:44:18 -0800160 void RunIteration() override {
Henry Speiser55aa3ba2022-02-21 23:21:12 -0800161 {
162 auto builder = superstructure_position_sender_.MakeBuilder();
163
Austin Schuh39f26f62022-02-24 21:34:46 -0800164 frc971::PotAndAbsolutePositionT catapult;
165 CopyPosition(catapult_encoder_, &catapult,
166 Values::kCatapultEncoderCountsPerRevolution(),
167 Values::kCatapultEncoderRatio(), catapult_pot_translate,
168 true, values_->catapult.potentiometer_offset);
169 flatbuffers::Offset<frc971::PotAndAbsolutePosition> catapult_offset =
170 frc971::PotAndAbsolutePosition::Pack(*builder.fbb(), &catapult);
171
Henry Speiser55aa3ba2022-02-21 23:21:12 -0800172 frc971::RelativePositionT climber;
173 CopyPosition(*climber_potentiometer_, &climber, climber_pot_translate,
174 false, values_->climber.potentiometer_offset);
175 flatbuffers::Offset<frc971::RelativePosition> climber_offset =
176 frc971::RelativePosition::Pack(*builder.fbb(), &climber);
177
Griffin Buibcbef482022-02-23 15:32:10 -0800178 frc971::RelativePositionT flipper_arm_left;
179 CopyPosition(*flipper_arm_left_potentiometer_, &flipper_arm_left,
180 flipper_arms_pot_translate, false,
181 values_->flipper_arm_left.potentiometer_offset);
182
183 frc971::RelativePositionT flipper_arm_right;
184 CopyPosition(*flipper_arm_right_potentiometer_, &flipper_arm_right,
185 flipper_arms_pot_translate, false,
186 values_->flipper_arm_right.potentiometer_offset);
187
Henry Speiser55aa3ba2022-02-21 23:21:12 -0800188 // Intake
189 frc971::PotAndAbsolutePositionT intake_front;
Henry Speiser55aa3ba2022-02-21 23:21:12 -0800190 CopyPosition(intake_encoder_front_, &intake_front,
191 Values::kIntakeEncoderCountsPerRevolution(),
192 Values::kIntakeEncoderRatio(), intake_pot_translate, false,
193 values_->intake_front.potentiometer_offset);
Griffin Buibcbef482022-02-23 15:32:10 -0800194 frc971::PotAndAbsolutePositionT intake_back;
Henry Speiser55aa3ba2022-02-21 23:21:12 -0800195 CopyPosition(intake_encoder_back_, &intake_back,
196 Values::kIntakeEncoderCountsPerRevolution(),
197 Values::kIntakeEncoderRatio(), intake_pot_translate, false,
198 values_->intake_back.potentiometer_offset);
Griffin Buibcbef482022-02-23 15:32:10 -0800199 frc971::PotAndAbsolutePositionT turret;
Henry Speiser55aa3ba2022-02-21 23:21:12 -0800200 CopyPosition(turret_encoder_, &turret,
201 Values::kTurretEncoderCountsPerRevolution(),
202 Values::kTurretEncoderRatio(), turret_pot_translate, false,
203 values_->turret.potentiometer_offset);
204
205 flatbuffers::Offset<frc971::PotAndAbsolutePosition> intake_offset_front =
206 frc971::PotAndAbsolutePosition::Pack(*builder.fbb(), &intake_front);
207 flatbuffers::Offset<frc971::PotAndAbsolutePosition> intake_offset_back =
208 frc971::PotAndAbsolutePosition::Pack(*builder.fbb(), &intake_back);
209 flatbuffers::Offset<frc971::PotAndAbsolutePosition> turret_offset =
210 frc971::PotAndAbsolutePosition::Pack(*builder.fbb(), &turret);
Griffin Buibcbef482022-02-23 15:32:10 -0800211 flatbuffers::Offset<frc971::RelativePosition> flipper_arm_left_offset =
212 frc971::RelativePosition::Pack(*builder.fbb(), &flipper_arm_left);
213 flatbuffers::Offset<frc971::RelativePosition> flipper_arm_right_offset =
214 frc971::RelativePosition::Pack(*builder.fbb(), &flipper_arm_right);
Henry Speiser55aa3ba2022-02-21 23:21:12 -0800215
216 superstructure::Position::Builder position_builder =
217 builder.MakeBuilder<superstructure::Position>();
218 position_builder.add_climber(climber_offset);
Griffin Buibcbef482022-02-23 15:32:10 -0800219 position_builder.add_flipper_arm_left(flipper_arm_left_offset);
220 position_builder.add_flipper_arm_right(flipper_arm_right_offset);
Henry Speiser55aa3ba2022-02-21 23:21:12 -0800221 position_builder.add_intake_front(intake_offset_front);
222 position_builder.add_intake_back(intake_offset_back);
223 position_builder.add_turret(turret_offset);
Milo Lin4950ac52022-02-25 19:56:11 -0800224 position_builder.add_intake_beambreak_front(
225 intake_beambreak_front_->Get());
226 position_builder.add_intake_beambreak_back(intake_beambreak_back_->Get());
227 position_builder.add_turret_beambreak(turret_beambreak_->Get());
Austin Schuh39f26f62022-02-24 21:34:46 -0800228 position_builder.add_catapult(catapult_offset);
Henry Speiser55aa3ba2022-02-21 23:21:12 -0800229 builder.CheckOk(builder.Send(position_builder.Finish()));
230 }
Siddhant Kanwar0e37f592022-02-21 19:26:50 -0800231
milind-u086d7262022-01-19 20:44:18 -0800232 {
233 auto builder = drivetrain_position_sender_.MakeBuilder();
234 frc971::control_loops::drivetrain::Position::Builder drivetrain_builder =
235 builder.MakeBuilder<frc971::control_loops::drivetrain::Position>();
236 drivetrain_builder.add_left_encoder(
James Kuszmaul53507e12022-02-12 18:36:40 -0800237 constants::Values::DrivetrainEncoderToMeters(
238 drivetrain_left_encoder_->GetRaw()));
milind-u086d7262022-01-19 20:44:18 -0800239 drivetrain_builder.add_left_speed(
240 drivetrain_velocity_translate(drivetrain_left_encoder_->GetPeriod()));
241
242 drivetrain_builder.add_right_encoder(
James Kuszmaul53507e12022-02-12 18:36:40 -0800243 -constants::Values::DrivetrainEncoderToMeters(
244 drivetrain_right_encoder_->GetRaw()));
milind-u086d7262022-01-19 20:44:18 -0800245 drivetrain_builder.add_right_speed(-drivetrain_velocity_translate(
246 drivetrain_right_encoder_->GetPeriod()));
247
248 builder.CheckOk(builder.Send(drivetrain_builder.Finish()));
249 }
250
251 {
milind-u086d7262022-01-19 20:44:18 -0800252 auto builder = auto_mode_sender_.MakeBuilder();
253
254 uint32_t mode = 0;
255 for (size_t i = 0; i < autonomous_modes_.size(); ++i) {
256 if (autonomous_modes_[i] && autonomous_modes_[i]->Get()) {
257 mode |= 1 << i;
258 }
259 }
260
261 auto auto_mode_builder =
262 builder.MakeBuilder<frc971::autonomous::AutonomousMode>();
263
264 auto_mode_builder.add_mode(mode);
265
266 builder.CheckOk(builder.Send(auto_mode_builder.Finish()));
267 }
268 }
269
Siddhant Kanwar0e37f592022-02-21 19:26:50 -0800270 void set_climber_potentiometer(
271 ::std::unique_ptr<frc::AnalogInput> potentiometer) {
272 climber_potentiometer_ = ::std::move(potentiometer);
273 }
274
Griffin Buibcbef482022-02-23 15:32:10 -0800275 void set_flipper_arm_left_potentiometer(
276 ::std::unique_ptr<frc::AnalogInput> potentiometer) {
277 flipper_arm_left_potentiometer_ = ::std::move(potentiometer);
278 }
279
280 void set_flipper_arm_right_potentiometer(
281 ::std::unique_ptr<frc::AnalogInput> potentiometer) {
282 flipper_arm_right_potentiometer_ = ::std::move(potentiometer);
283 }
284
Henry Speiser55aa3ba2022-02-21 23:21:12 -0800285 void set_intake_encoder_front(::std::unique_ptr<frc::Encoder> encoder) {
286 fast_encoder_filter_.Add(encoder.get());
287 intake_encoder_front_.set_encoder(::std::move(encoder));
288 }
289
290 void set_intake_encoder_back(::std::unique_ptr<frc::Encoder> encoder) {
291 fast_encoder_filter_.Add(encoder.get());
292 intake_encoder_back_.set_encoder(::std::move(encoder));
293 }
294
295 void set_intake_front_absolute_pwm(
296 ::std::unique_ptr<frc::DigitalInput> absolute_pwm) {
297 intake_encoder_front_.set_absolute_pwm(::std::move(absolute_pwm));
298 }
299
300 void set_intake_front_potentiometer(
301 ::std::unique_ptr<frc::AnalogInput> potentiometer) {
302 intake_encoder_front_.set_potentiometer(::std::move(potentiometer));
303 }
304
305 void set_intake_back_absolute_pwm(
306 ::std::unique_ptr<frc::DigitalInput> absolute_pwm) {
307 intake_encoder_back_.set_absolute_pwm(::std::move(absolute_pwm));
308 }
309
310 void set_intake_back_potentiometer(
311 ::std::unique_ptr<frc::AnalogInput> potentiometer) {
312 intake_encoder_back_.set_potentiometer(::std::move(potentiometer));
313 }
314
315 void set_turret_encoder(::std::unique_ptr<frc::Encoder> encoder) {
316 medium_encoder_filter_.Add(encoder.get());
317 turret_encoder_.set_encoder(::std::move(encoder));
318 }
319
320 void set_turret_absolute_pwm(
321 ::std::unique_ptr<frc::DigitalInput> absolute_pwm) {
322 turret_encoder_.set_absolute_pwm(::std::move(absolute_pwm));
323 }
324
325 void set_turret_potentiometer(
326 ::std::unique_ptr<frc::AnalogInput> potentiometer) {
327 turret_encoder_.set_potentiometer(::std::move(potentiometer));
328 }
329
Milo Lin4950ac52022-02-25 19:56:11 -0800330 void set_intake_beambreak_front(::std::unique_ptr<frc::DigitalInput> sensor) {
331 intake_beambreak_front_ = ::std::move(sensor);
332 }
333 void set_intake_beambreak_back(::std::unique_ptr<frc::DigitalInput> sensor) {
334 intake_beambreak_back_ = ::std::move(sensor);
335 }
336 void set_turret_beambreak(::std::unique_ptr<frc::DigitalInput> sensor) {
337 turret_beambreak_ = ::std::move(sensor);
338 }
339
milind-u086d7262022-01-19 20:44:18 -0800340 private:
Henry Speiser55aa3ba2022-02-21 23:21:12 -0800341 std::shared_ptr<const Values> values_;
342
Siddhant Kanwar0e37f592022-02-21 19:26:50 -0800343 aos::Sender<frc971::autonomous::AutonomousMode> auto_mode_sender_;
344 aos::Sender<superstructure::Position> superstructure_position_sender_;
345 aos::Sender<frc971::control_loops::drivetrain::Position>
milind-u086d7262022-01-19 20:44:18 -0800346 drivetrain_position_sender_;
347
Siddhant Kanwar0e37f592022-02-21 19:26:50 -0800348 std::array<std::unique_ptr<frc::DigitalInput>, 2> autonomous_modes_;
Henry Speiser55aa3ba2022-02-21 23:21:12 -0800349
Milo Lin4950ac52022-02-25 19:56:11 -0800350 std::unique_ptr<frc::DigitalInput> intake_beambreak_front_,
351 intake_beambreak_back_, turret_beambreak_;
352
Griffin Buibcbef482022-02-23 15:32:10 -0800353 std::unique_ptr<frc::AnalogInput> climber_potentiometer_,
354 flipper_arm_right_potentiometer_, flipper_arm_left_potentiometer_;
Henry Speiser55aa3ba2022-02-21 23:21:12 -0800355 frc971::wpilib::AbsoluteEncoderAndPotentiometer intake_encoder_front_,
356 intake_encoder_back_, turret_encoder_;
Austin Schuh39f26f62022-02-24 21:34:46 -0800357
358 frc971::wpilib::AbsoluteEncoderAndPotentiometer catapult_encoder_;
milind-u086d7262022-01-19 20:44:18 -0800359};
360
361class SuperstructureWriter
362 : public ::frc971::wpilib::LoopOutputHandler<superstructure::Output> {
363 public:
Siddhant Kanwar0e37f592022-02-21 19:26:50 -0800364 SuperstructureWriter(aos::EventLoop *event_loop)
365 : frc971::wpilib::LoopOutputHandler<superstructure::Output>(
milind-u086d7262022-01-19 20:44:18 -0800366 event_loop, "/superstructure") {}
367
Henry Speiser55aa3ba2022-02-21 23:21:12 -0800368 void set_climber_falcon(std::unique_ptr<frc::TalonFX> t) {
Siddhant Kanwar0e37f592022-02-21 19:26:50 -0800369 climber_falcon_ = std::move(t);
Griffin Bui67abb912022-01-22 16:16:21 -0800370 }
371
Jacob Ismael322ebb92022-02-09 20:12:47 -0800372 void set_turret_falcon(::std::unique_ptr<::frc::TalonFX> t) {
373 turret_falcon_ = ::std::move(t);
374 }
375
376 void set_catapult_falcon_1(::std::unique_ptr<::frc::TalonFX> t) {
377 catapult_falcon_1_ = ::std::move(t);
378 }
379
380 void set_catapult_falcon_2(::std::unique_ptr<::frc::TalonFX> t) {
381 catapult_falcon_2_ = ::std::move(t);
382 }
383
Henry Speiser55aa3ba2022-02-21 23:21:12 -0800384 void set_intake_falcon_front(::std::unique_ptr<frc::TalonFX> t) {
385 intake_falcon_front_ = ::std::move(t);
Griffin Bui67abb912022-01-22 16:16:21 -0800386 }
milind-u086d7262022-01-19 20:44:18 -0800387
Henry Speiser55aa3ba2022-02-21 23:21:12 -0800388 void set_intake_falcon_back(::std::unique_ptr<frc::TalonFX> t) {
389 intake_falcon_back_ = ::std::move(t);
390 }
391
392 void set_roller_falcon_front(
393 ::std::unique_ptr<::ctre::phoenix::motorcontrol::can::TalonFX> t) {
394 roller_falcon_front_ = ::std::move(t);
395 roller_falcon_front_->ConfigSupplyCurrentLimit(
396 {true, Values::kIntakeRollerSupplyCurrentLimit(),
397 Values::kIntakeRollerSupplyCurrentLimit(), 0});
398 roller_falcon_front_->ConfigStatorCurrentLimit(
399 {true, Values::kIntakeRollerStatorCurrentLimit(),
400 Values::kIntakeRollerStatorCurrentLimit(), 0});
401 }
402
403 void set_roller_falcon_back(
404 ::std::unique_ptr<::ctre::phoenix::motorcontrol::can::TalonFX> t) {
405 roller_falcon_back_ = ::std::move(t);
406 roller_falcon_back_->ConfigSupplyCurrentLimit(
407 {true, Values::kIntakeRollerSupplyCurrentLimit(),
408 Values::kIntakeRollerSupplyCurrentLimit(), 0});
409 roller_falcon_back_->ConfigStatorCurrentLimit(
410 {true, Values::kIntakeRollerStatorCurrentLimit(),
411 Values::kIntakeRollerStatorCurrentLimit(), 0});
412 }
413
Griffin Buibcbef482022-02-23 15:32:10 -0800414 void set_flipper_arms_falcon(
415 ::std::unique_ptr<::ctre::phoenix::motorcontrol::can::TalonFX> t) {
416 flipper_arms_falcon_ = ::std::move(t);
417 flipper_arms_falcon_->ConfigSupplyCurrentLimit(
418 {true, Values::kFlipperArmSupplyCurrentLimit(),
419 Values::kFlipperArmSupplyCurrentLimit(), 0});
420 flipper_arms_falcon_->ConfigStatorCurrentLimit(
421 {true, Values::kFlipperArmStatorCurrentLimit(),
422 Values::kFlipperArmStatorCurrentLimit(), 0});
423 }
424
Henry Speiser55aa3ba2022-02-21 23:21:12 -0800425 void set_transfer_roller_victor(::std::unique_ptr<::frc::VictorSP> t) {
426 transfer_roller_victor_ = ::std::move(t);
427 }
428
429 private:
Griffin Bui67abb912022-01-22 16:16:21 -0800430 void Stop() override {
431 AOS_LOG(WARNING, "Superstructure output too old.\n");
Siddhant Kanwar0e37f592022-02-21 19:26:50 -0800432 climber_falcon_->SetDisabled();
Henry Speiser55aa3ba2022-02-21 23:21:12 -0800433 roller_falcon_front_->Set(
434 ctre::phoenix::motorcontrol::ControlMode::Disabled, 0);
435 roller_falcon_back_->Set(ctre::phoenix::motorcontrol::ControlMode::Disabled,
436 0);
Griffin Buibcbef482022-02-23 15:32:10 -0800437 flipper_arms_falcon_->Set(
438 ctre::phoenix::motorcontrol::ControlMode::Disabled, 0);
Henry Speiser55aa3ba2022-02-21 23:21:12 -0800439 intake_falcon_front_->SetDisabled();
440 intake_falcon_back_->SetDisabled();
441 transfer_roller_victor_->SetDisabled();
Jacob Ismael322ebb92022-02-09 20:12:47 -0800442 catapult_falcon_1_->SetDisabled();
443 catapult_falcon_2_->SetDisabled();
444 turret_falcon_->SetDisabled();
Griffin Bui67abb912022-01-22 16:16:21 -0800445 }
446
Henry Speiser55aa3ba2022-02-21 23:21:12 -0800447 void Write(const superstructure::Output &output) override {
448 WritePwm(output.climber_voltage(), climber_falcon_.get());
Griffin Buibcbef482022-02-23 15:32:10 -0800449
Henry Speiser55aa3ba2022-02-21 23:21:12 -0800450 WritePwm(output.intake_voltage_front(), intake_falcon_front_.get());
451 WritePwm(output.intake_voltage_back(), intake_falcon_back_.get());
452 WriteCan(output.roller_voltage_front(), roller_falcon_front_.get());
453 WriteCan(output.roller_voltage_back(), roller_falcon_back_.get());
454 WritePwm(output.transfer_roller_voltage(), transfer_roller_victor_.get());
Griffin Buibcbef482022-02-23 15:32:10 -0800455
456 WriteCan(output.flipper_arms_voltage(), flipper_arms_falcon_.get());
457
Henry Speiser55aa3ba2022-02-21 23:21:12 -0800458 WritePwm(output.catapult_voltage(), catapult_falcon_1_.get());
459 WritePwm(output.catapult_voltage(), catapult_falcon_2_.get());
Griffin Buibcbef482022-02-23 15:32:10 -0800460
Henry Speiser55aa3ba2022-02-21 23:21:12 -0800461 WritePwm(output.turret_voltage(), turret_falcon_.get());
462 }
463
464 static void WriteCan(const double voltage,
465 ::ctre::phoenix::motorcontrol::can::TalonFX *falcon) {
466 falcon->Set(
467 ctre::phoenix::motorcontrol::ControlMode::PercentOutput,
468 std::clamp(voltage, -kMaxBringupPower, kMaxBringupPower) / 12.0);
469 }
470
471 template <typename T>
472 static void WritePwm(const double voltage, T *motor) {
473 motor->SetSpeed(std::clamp(voltage, -kMaxBringupPower, kMaxBringupPower) /
474 12.0);
475 }
476
477 ::std::unique_ptr<frc::TalonFX> intake_falcon_front_, intake_falcon_back_;
478
479 ::std::unique_ptr<::ctre::phoenix::motorcontrol::can::TalonFX>
Griffin Buibcbef482022-02-23 15:32:10 -0800480 roller_falcon_front_, roller_falcon_back_, flipper_arms_falcon_;
Henry Speiser55aa3ba2022-02-21 23:21:12 -0800481
Jacob Ismael322ebb92022-02-09 20:12:47 -0800482 ::std::unique_ptr<::frc::TalonFX> turret_falcon_, catapult_falcon_1_,
Siddhant Kanwar0e37f592022-02-21 19:26:50 -0800483 catapult_falcon_2_, climber_falcon_;
Henry Speiser55aa3ba2022-02-21 23:21:12 -0800484 ::std::unique_ptr<::frc::VictorSP> transfer_roller_victor_;
milind-u086d7262022-01-19 20:44:18 -0800485};
486
487class WPILibRobot : public ::frc971::wpilib::WPILibRobotBase {
488 public:
489 ::std::unique_ptr<frc::Encoder> make_encoder(int index) {
490 return make_unique<frc::Encoder>(10 + index * 2, 11 + index * 2, false,
491 frc::Encoder::k4X);
492 }
493
494 void Run() override {
Henry Speiser55aa3ba2022-02-21 23:21:12 -0800495 std::shared_ptr<const Values> values =
496 std::make_shared<const Values>(constants::MakeValues());
497
milind-u086d7262022-01-19 20:44:18 -0800498 aos::FlatbufferDetachedBuffer<aos::Configuration> config =
Austin Schuhc5fa6d92022-02-25 14:36:28 -0800499 aos::configuration::ReadConfig("aos_config.json");
milind-u086d7262022-01-19 20:44:18 -0800500
501 // Thread 1.
502 ::aos::ShmEventLoop joystick_sender_event_loop(&config.message());
503 ::frc971::wpilib::JoystickSender joystick_sender(
504 &joystick_sender_event_loop);
505 AddLoop(&joystick_sender_event_loop);
506
507 // Thread 2.
508 ::aos::ShmEventLoop pdp_fetcher_event_loop(&config.message());
509 ::frc971::wpilib::PDPFetcher pdp_fetcher(&pdp_fetcher_event_loop);
510 AddLoop(&pdp_fetcher_event_loop);
511
512 // Thread 3.
513 ::aos::ShmEventLoop sensor_reader_event_loop(&config.message());
Henry Speiser55aa3ba2022-02-21 23:21:12 -0800514 SensorReader sensor_reader(&sensor_reader_event_loop, values);
milind-u086d7262022-01-19 20:44:18 -0800515 sensor_reader.set_drivetrain_left_encoder(make_encoder(0));
516 sensor_reader.set_drivetrain_right_encoder(make_encoder(1));
Henry Speiser55aa3ba2022-02-21 23:21:12 -0800517
518 sensor_reader.set_intake_encoder_front(make_encoder(2));
519 sensor_reader.set_intake_front_absolute_pwm(
520 make_unique<frc::DigitalInput>(2));
521 sensor_reader.set_intake_front_potentiometer(
522 make_unique<frc::AnalogInput>(2));
523
524 sensor_reader.set_intake_encoder_back(make_encoder(3));
525 sensor_reader.set_intake_back_absolute_pwm(
526 make_unique<frc::DigitalInput>(3));
527 sensor_reader.set_intake_back_potentiometer(
528 make_unique<frc::AnalogInput>(3));
529
530 sensor_reader.set_turret_encoder(make_encoder(4));
531 sensor_reader.set_turret_absolute_pwm(make_unique<frc::DigitalInput>(4));
532 sensor_reader.set_turret_potentiometer(make_unique<frc::AnalogInput>(4));
533
Milo Lin4950ac52022-02-25 19:56:11 -0800534 sensor_reader.set_intake_beambreak_front(make_unique<frc::DigitalInput>(5));
535 sensor_reader.set_intake_beambreak_back(make_unique<frc::DigitalInput>(6));
536 sensor_reader.set_turret_beambreak(make_unique<frc::DigitalInput>(7));
537
Henry Speiser55aa3ba2022-02-21 23:21:12 -0800538 sensor_reader.set_climber_potentiometer(make_unique<frc::AnalogInput>(5));
milind-u086d7262022-01-19 20:44:18 -0800539
Griffin Buibcbef482022-02-23 15:32:10 -0800540 sensor_reader.set_flipper_arm_left_potentiometer(
541 make_unique<frc::AnalogInput>(4));
542 sensor_reader.set_flipper_arm_right_potentiometer(
543 make_unique<frc::AnalogInput>(5));
544
Austin Schuh39f26f62022-02-24 21:34:46 -0800545 sensor_reader.set_catapult_encoder(
546 make_unique<frc::Encoder>(0, 1, false, frc::Encoder::k4X));
547 sensor_reader.set_catapult_absolute_pwm(std::make_unique<frc::DigitalInput>(2));
548 sensor_reader.set_catapult_potentiometer(std::make_unique<frc::AnalogInput>(2));
549
milind-u086d7262022-01-19 20:44:18 -0800550 AddLoop(&sensor_reader_event_loop);
551
552 // Thread 4.
553 ::aos::ShmEventLoop output_event_loop(&config.message());
554 ::frc971::wpilib::DrivetrainWriter drivetrain_writer(&output_event_loop);
555 drivetrain_writer.set_left_controller0(
Austin Schuh39f26f62022-02-24 21:34:46 -0800556 ::std::unique_ptr<::frc::VictorSP>(new ::frc::VictorSP(1)), true);
milind-u086d7262022-01-19 20:44:18 -0800557 drivetrain_writer.set_right_controller0(
Austin Schuh39f26f62022-02-24 21:34:46 -0800558 ::std::unique_ptr<::frc::VictorSP>(new ::frc::VictorSP(0)), false);
milind-u086d7262022-01-19 20:44:18 -0800559
560 SuperstructureWriter superstructure_writer(&output_event_loop);
561
Austin Schuh39f26f62022-02-24 21:34:46 -0800562 superstructure_writer.set_turret_falcon(make_unique<::frc::TalonFX>(8));
Henry Speiser55aa3ba2022-02-21 23:21:12 -0800563 superstructure_writer.set_roller_falcon_front(
Griffin Buibcbef482022-02-23 15:32:10 -0800564 make_unique<::ctre::phoenix::motorcontrol::can::TalonFX>(3));
Henry Speiser55aa3ba2022-02-21 23:21:12 -0800565 superstructure_writer.set_roller_falcon_back(
Griffin Buibcbef482022-02-23 15:32:10 -0800566 make_unique<::ctre::phoenix::motorcontrol::can::TalonFX>(4));
Henry Speiser55aa3ba2022-02-21 23:21:12 -0800567 superstructure_writer.set_transfer_roller_victor(
Austin Schuh39f26f62022-02-24 21:34:46 -0800568 make_unique<::frc::VictorSP>(9));
Henry Speiser55aa3ba2022-02-21 23:21:12 -0800569 superstructure_writer.set_intake_falcon_front(make_unique<frc::TalonFX>(5));
570 superstructure_writer.set_intake_falcon_back(make_unique<frc::TalonFX>(6));
571 superstructure_writer.set_climber_falcon(make_unique<frc::TalonFX>(7));
Griffin Buibcbef482022-02-23 15:32:10 -0800572 superstructure_writer.set_flipper_arms_falcon(
573 make_unique<::ctre::phoenix::motorcontrol::can::TalonFX>(5));
Griffin Bui67abb912022-01-22 16:16:21 -0800574
Austin Schuh39f26f62022-02-24 21:34:46 -0800575 superstructure_writer.set_catapult_falcon_1(make_unique<::frc::TalonFX>(2));
576 superstructure_writer.set_catapult_falcon_2(make_unique<::frc::TalonFX>(3));
577
milind-u086d7262022-01-19 20:44:18 -0800578 AddLoop(&output_event_loop);
579
580 RunLoops();
581 }
582};
583
584} // namespace wpilib
585} // namespace y2022
586
587AOS_ROBOT_CLASS(::y2022::wpilib::WPILibRobot);