blob: 3b00248083386d170d4c144d1178782ac4dcc25c [file] [log] [blame]
Sabina Davis2243cab2019-02-05 21:45:08 -08001#include <inttypes.h>
Comran Morshed9a9948c2016-01-16 15:58:04 +00002#include <stdio.h>
3#include <string.h>
4#include <unistd.h>
Comran Morshed9a9948c2016-01-16 15:58:04 +00005
Brian Silverman68cb5c22016-03-20 18:11:14 -07006#include <array>
Sabina Davis2243cab2019-02-05 21:45:08 -08007#include <chrono>
8#include <functional>
9#include <mutex>
10#include <thread>
Comran Morshed9a9948c2016-01-16 15:58:04 +000011
Parker Schuhd3b7a8872018-02-19 16:42:27 -080012#include "frc971/wpilib/ahal/AnalogInput.h"
13#include "frc971/wpilib/ahal/Compressor.h"
14#include "frc971/wpilib/ahal/DigitalGlitchFilter.h"
15#include "frc971/wpilib/ahal/DriverStation.h"
16#include "frc971/wpilib/ahal/Encoder.h"
17#include "frc971/wpilib/ahal/Relay.h"
18#include "frc971/wpilib/ahal/Talon.h"
Comran Morshed9a9948c2016-01-16 15:58:04 +000019#include "frc971/wpilib/wpilib_robot_base.h"
Comran Morshed9a9948c2016-01-16 15:58:04 +000020#undef ERROR
21
Brian Silvermanf819b442019-01-20 16:51:04 -080022#include "aos/commonmath.h"
Austin Schuhdf6cbb12019-02-02 13:46:52 -080023#include "aos/events/shm-event-loop.h"
Brian Silvermanf819b442019-01-20 16:51:04 -080024#include "aos/init.h"
John Park33858a32018-09-28 23:05:48 -070025#include "aos/logging/logging.h"
26#include "aos/logging/queue_logging.h"
Brian Silvermanf819b442019-01-20 16:51:04 -080027#include "aos/make_unique.h"
28#include "aos/robot_state/robot_state.q.h"
29#include "aos/stl_mutex/stl_mutex.h"
John Park33858a32018-09-28 23:05:48 -070030#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"
Philipp Schrader4bd29b12017-02-22 04:42:27 +000034#include "frc971/autonomous/auto.q.h"
Comran Morshed225f0b92016-02-10 20:34:27 +000035#include "frc971/control_loops/control_loops.q.h"
Comran Morshed9a9948c2016-01-16 15:58:04 +000036#include "frc971/control_loops/drivetrain/drivetrain.q.h"
Austin Schuh54667ac2019-02-02 16:44:49 -080037#include "frc971/wpilib/ADIS16448.h"
38#include "frc971/wpilib/buffered_pcm.h"
39#include "frc971/wpilib/buffered_solenoid.h"
40#include "frc971/wpilib/dma.h"
41#include "frc971/wpilib/dma_edge_counting.h"
Sabina Davisb71bc282019-02-03 01:17:23 -080042#include "frc971/wpilib/drivetrain_writer.h"
Austin Schuh54667ac2019-02-02 16:44:49 -080043#include "frc971/wpilib/encoder_and_potentiometer.h"
44#include "frc971/wpilib/gyro_sender.h"
45#include "frc971/wpilib/interrupt_edge_counting.h"
46#include "frc971/wpilib/joystick_sender.h"
47#include "frc971/wpilib/logging.q.h"
48#include "frc971/wpilib/loop_output_handler.h"
49#include "frc971/wpilib/pdp_fetcher.h"
50#include "frc971/wpilib/sensor_reader.h"
Comran Morshed6c6a0a92016-01-17 12:45:16 +000051#include "y2016/constants.h"
Comran Morshed5bb12112016-02-16 13:48:57 +000052#include "y2016/control_loops/drivetrain/drivetrain_dog_motor_plant.h"
Comran Morshed225f0b92016-02-10 20:34:27 +000053#include "y2016/control_loops/shooter/shooter.q.h"
Austin Schuh54667ac2019-02-02 16:44:49 -080054#include "y2016/control_loops/shooter/shooter.q.h"
Comran Morshed225f0b92016-02-10 20:34:27 +000055#include "y2016/control_loops/superstructure/superstructure.q.h"
Comran Morshedaa0573c2016-03-05 19:05:54 +000056#include "y2016/queues/ball_detector.q.h"
Comran Morshed9a9948c2016-01-16 15:58:04 +000057
Comran Morshed9a9948c2016-01-16 15:58:04 +000058using ::frc971::control_loops::drivetrain_queue;
Comran Morshed225f0b92016-02-10 20:34:27 +000059using ::y2016::control_loops::shooter::shooter_queue;
60using ::y2016::control_loops::superstructure_queue;
Parker Schuhd3b7a8872018-02-19 16:42:27 -080061using namespace frc;
Brian Silvermanf819b442019-01-20 16:51:04 -080062using aos::make_unique;
Comran Morshed9a9948c2016-01-16 15:58:04 +000063
Comran Morshed6c6a0a92016-01-17 12:45:16 +000064namespace y2016 {
Comran Morshed9a9948c2016-01-16 15:58:04 +000065namespace wpilib {
Austin Schuha9992ff2016-02-28 21:59:23 -080066namespace {
67constexpr double kMaxBringupPower = 12.0;
68} // namespace
Comran Morshed9a9948c2016-01-16 15:58:04 +000069
70// TODO(Brian): Fix the interpretation of the result of GetRaw here and in the
71// DMA stuff and then removing the * 2.0 in *_translate.
72// The low bit is direction.
73
Comran Morshed225f0b92016-02-10 20:34:27 +000074// Translates for the sensor values to convert raw index pulses into something
75// with proper units.
76
77// TODO(comran): Template these methods since there is a lot of repetition here.
78double hall_translate(double in) {
79 // Turn voltage from our 3-state halls into a ratio that the loop can use.
80 return in / 5.0;
81}
82
Comran Morshed9a9948c2016-01-16 15:58:04 +000083double drivetrain_translate(int32_t in) {
Comran Morshed6c6a0a92016-01-17 12:45:16 +000084 return -static_cast<double>(in) / (256.0 /*cpr*/ * 4.0 /*4x*/) *
Comran Morshed225f0b92016-02-10 20:34:27 +000085 constants::Values::kDrivetrainEncoderRatio *
Austin Schuh9f77fd22016-02-21 02:53:58 -080086 control_loops::drivetrain::kWheelRadius * 2.0 * M_PI;
Comran Morshed9a9948c2016-01-16 15:58:04 +000087}
88
89double drivetrain_velocity_translate(double in) {
90 return (1.0 / in) / 256.0 /*cpr*/ *
Comran Morshed225f0b92016-02-10 20:34:27 +000091 constants::Values::kDrivetrainEncoderRatio *
Austin Schuh9f77fd22016-02-21 02:53:58 -080092 control_loops::drivetrain::kWheelRadius * 2.0 * M_PI;
Comran Morshed9a9948c2016-01-16 15:58:04 +000093}
94
Comran Morshed225f0b92016-02-10 20:34:27 +000095double shooter_translate(int32_t in) {
Comran Morshed5bb12112016-02-16 13:48:57 +000096 return -static_cast<double>(in) / (128.0 /*cpr*/ * 4.0 /*4x*/) *
Comran Morshed225f0b92016-02-10 20:34:27 +000097 constants::Values::kShooterEncoderRatio * (2 * M_PI /*radians*/);
98}
Comran Morshed9a9948c2016-01-16 15:58:04 +000099
Comran Morshed225f0b92016-02-10 20:34:27 +0000100double intake_translate(int32_t in) {
Austin Schuh9f77fd22016-02-21 02:53:58 -0800101 return static_cast<double>(in) / (512.0 /*cpr*/ * 4.0 /*4x*/) *
Comran Morshed225f0b92016-02-10 20:34:27 +0000102 constants::Values::kIntakeEncoderRatio * (2 * M_PI /*radians*/);
103}
104
105double shoulder_translate(int32_t in) {
106 return -static_cast<double>(in) / (512.0 /*cpr*/ * 4.0 /*4x*/) *
107 constants::Values::kShoulderEncoderRatio * (2 * M_PI /*radians*/);
108}
109
110double wrist_translate(int32_t in) {
111 return -static_cast<double>(in) / (512.0 /*cpr*/ * 4.0 /*4x*/) *
112 constants::Values::kWristEncoderRatio * (2 * M_PI /*radians*/);
113}
114
115double intake_pot_translate(double voltage) {
116 return voltage * constants::Values::kIntakePotRatio *
Comran Morshed5bb12112016-02-16 13:48:57 +0000117 (10.0 /*turns*/ / 5.0 /*volts*/) * (2 * M_PI /*radians*/);
Comran Morshed225f0b92016-02-10 20:34:27 +0000118}
119
120double shoulder_pot_translate(double voltage) {
121 return voltage * constants::Values::kShoulderPotRatio *
Comran Morshed5bb12112016-02-16 13:48:57 +0000122 (3.0 /*turns*/ / 5.0 /*volts*/) * (2 * M_PI /*radians*/);
Comran Morshed225f0b92016-02-10 20:34:27 +0000123}
124
125double wrist_pot_translate(double voltage) {
126 return voltage * constants::Values::kWristPotRatio *
Comran Morshed5bb12112016-02-16 13:48:57 +0000127 (3.0 /*turns*/ / 5.0 /*volts*/) * (2 * M_PI /*radians*/);
Comran Morshed9a9948c2016-01-16 15:58:04 +0000128}
129
Comran Morshed1b764322016-02-14 20:18:12 +0000130constexpr double kMaxDrivetrainEncoderPulsesPerSecond =
131 5600.0 /* CIM free speed RPM */ * 14.0 / 48.0 /* 1st reduction */ * 28.0 /
132 50.0 /* 2nd reduction (high gear) */ * 30.0 / 44.0 /* encoder gears */ /
133 60.0 /* seconds per minute */ * 256.0 /* CPR */ * 4 /* edges per cycle */;
134
135constexpr double kMaxShooterEncoderPulsesPerSecond =
136 18700.0 /* 775pro free speed RPM */ * 12.0 /
137 18.0 /* motor to encoder reduction */ / 60.0 /* seconds per minute */ *
138 128.0 /* CPR */ * 4 /* edges per cycle */;
139
140double kMaxDrivetrainShooterEncoderPulsesPerSecond = ::std::max(
141 kMaxDrivetrainEncoderPulsesPerSecond, kMaxShooterEncoderPulsesPerSecond);
142
143constexpr double kMaxSuperstructureEncoderPulsesPerSecond =
144 18700.0 /* 775pro free speed RPM */ * 12.0 /
145 56.0 /* motor to encoder reduction */ / 60.0 /* seconds per minute */ *
146 512.0 /* CPR */ * 4 /* index pulse every quarter cycle */;
Comran Morshed9a9948c2016-01-16 15:58:04 +0000147
Comran Morshed225f0b92016-02-10 20:34:27 +0000148// Class to send position messages with sensor readings to our loops.
Austin Schuh54667ac2019-02-02 16:44:49 -0800149class SensorReader : public ::frc971::wpilib::SensorReader {
Comran Morshed9a9948c2016-01-16 15:58:04 +0000150 public:
Austin Schuhdf6cbb12019-02-02 13:46:52 -0800151 SensorReader(::aos::EventLoop *event_loop)
Austin Schuh4b652c92019-05-27 13:22:27 -0700152 : ::frc971::wpilib::SensorReader(event_loop),
153 ball_detector_sender_(
154 event_loop->MakeSender<::y2016::sensors::BallDetector>(
Austin Schuha250b2d2019-05-27 16:14:02 -0700155 ".y2016.sensors.ball_detector")),
156 auto_mode_sender_(
157 event_loop->MakeSender<::frc971::autonomous::AutonomousMode>(
158 ".frc971.autonomous.auto_mode")) {
Comran Morshed9a9948c2016-01-16 15:58:04 +0000159 // Set it to filter out anything shorter than 1/4 of the minimum pulse width
160 // we should ever see.
Austin Schuh54667ac2019-02-02 16:44:49 -0800161 UpdateFastEncoderFilterHz(kMaxDrivetrainShooterEncoderPulsesPerSecond);
162 UpdateMediumEncoderFilterHz(kMaxSuperstructureEncoderPulsesPerSecond);
Comran Morshed9a9948c2016-01-16 15:58:04 +0000163 hall_filter_.SetPeriodNanoSeconds(100000);
164 }
165
Comran Morshed225f0b92016-02-10 20:34:27 +0000166 void set_drivetrain_left_hall(::std::unique_ptr<AnalogInput> analog) {
167 drivetrain_left_hall_ = ::std::move(analog);
Comran Morshed9a9948c2016-01-16 15:58:04 +0000168 }
169
Comran Morshed225f0b92016-02-10 20:34:27 +0000170 void set_drivetrain_right_hall(::std::unique_ptr<AnalogInput> analog) {
171 drivetrain_right_hall_ = ::std::move(analog);
Comran Morshed9a9948c2016-01-16 15:58:04 +0000172 }
173
Comran Morshed225f0b92016-02-10 20:34:27 +0000174 // Shooter setters.
175 void set_shooter_left_encoder(::std::unique_ptr<Encoder> encoder) {
Austin Schuh54667ac2019-02-02 16:44:49 -0800176 fast_encoder_filter_.Add(encoder.get());
Comran Morshed225f0b92016-02-10 20:34:27 +0000177 shooter_left_encoder_ = ::std::move(encoder);
Comran Morshed9a9948c2016-01-16 15:58:04 +0000178 }
179
Comran Morshed225f0b92016-02-10 20:34:27 +0000180 void set_shooter_right_encoder(::std::unique_ptr<Encoder> encoder) {
Austin Schuh54667ac2019-02-02 16:44:49 -0800181 fast_encoder_filter_.Add(encoder.get());
Comran Morshed225f0b92016-02-10 20:34:27 +0000182 shooter_right_encoder_ = ::std::move(encoder);
183 }
184
185 // Intake setters.
186 void set_intake_encoder(::std::unique_ptr<Encoder> encoder) {
Austin Schuh54667ac2019-02-02 16:44:49 -0800187 medium_encoder_filter_.Add(encoder.get());
Comran Morshed225f0b92016-02-10 20:34:27 +0000188 intake_encoder_.set_encoder(::std::move(encoder));
189 }
190
191 void set_intake_potentiometer(::std::unique_ptr<AnalogInput> potentiometer) {
192 intake_encoder_.set_potentiometer(::std::move(potentiometer));
193 }
194
195 void set_intake_index(::std::unique_ptr<DigitalInput> index) {
Austin Schuh54667ac2019-02-02 16:44:49 -0800196 medium_encoder_filter_.Add(index.get());
Comran Morshed225f0b92016-02-10 20:34:27 +0000197 intake_encoder_.set_index(::std::move(index));
198 }
199
200 // Shoulder setters.
201 void set_shoulder_encoder(::std::unique_ptr<Encoder> encoder) {
Austin Schuh54667ac2019-02-02 16:44:49 -0800202 medium_encoder_filter_.Add(encoder.get());
Comran Morshed225f0b92016-02-10 20:34:27 +0000203 shoulder_encoder_.set_encoder(::std::move(encoder));
204 }
205
206 void set_shoulder_potentiometer(
207 ::std::unique_ptr<AnalogInput> potentiometer) {
208 shoulder_encoder_.set_potentiometer(::std::move(potentiometer));
209 }
210
211 void set_shoulder_index(::std::unique_ptr<DigitalInput> index) {
Austin Schuh54667ac2019-02-02 16:44:49 -0800212 medium_encoder_filter_.Add(index.get());
Comran Morshed225f0b92016-02-10 20:34:27 +0000213 shoulder_encoder_.set_index(::std::move(index));
214 }
215
216 // Wrist setters.
217 void set_wrist_encoder(::std::unique_ptr<Encoder> encoder) {
Austin Schuh54667ac2019-02-02 16:44:49 -0800218 medium_encoder_filter_.Add(encoder.get());
Comran Morshed225f0b92016-02-10 20:34:27 +0000219 wrist_encoder_.set_encoder(::std::move(encoder));
220 }
221
222 void set_wrist_potentiometer(::std::unique_ptr<AnalogInput> potentiometer) {
223 wrist_encoder_.set_potentiometer(::std::move(potentiometer));
224 }
225
226 void set_wrist_index(::std::unique_ptr<DigitalInput> index) {
Austin Schuh54667ac2019-02-02 16:44:49 -0800227 medium_encoder_filter_.Add(index.get());
Comran Morshed225f0b92016-02-10 20:34:27 +0000228 wrist_encoder_.set_index(::std::move(index));
Comran Morshed9a9948c2016-01-16 15:58:04 +0000229 }
230
Comran Morshedaa0573c2016-03-05 19:05:54 +0000231 // Ball detector setter.
232 void set_ball_detector(::std::unique_ptr<AnalogInput> analog) {
233 ball_detector_ = ::std::move(analog);
234 }
235
Brian Silverman68cb5c22016-03-20 18:11:14 -0700236 // Autonomous mode switch setter.
237 void set_autonomous_mode(int i, ::std::unique_ptr<DigitalInput> sensor) {
238 autonomous_modes_.at(i) = ::std::move(sensor);
239 }
240
Comran Morshed9a9948c2016-01-16 15:58:04 +0000241 // All of the DMA-related set_* calls must be made before this, and it doesn't
242 // hurt to do all of them.
Comran Morshed9a9948c2016-01-16 15:58:04 +0000243
Austin Schuh54667ac2019-02-02 16:44:49 -0800244 void Start() {
245 AddToDMA(&intake_encoder_);
246 AddToDMA(&shoulder_encoder_);
247 AddToDMA(&wrist_encoder_);
Comran Morshed9a9948c2016-01-16 15:58:04 +0000248 }
249
250 void RunIteration() {
Comran Morshed9a9948c2016-01-16 15:58:04 +0000251 {
252 auto drivetrain_message = drivetrain_queue.position.MakeMessage();
253 drivetrain_message->right_encoder =
Austin Schuh9f77fd22016-02-21 02:53:58 -0800254 drivetrain_translate(-drivetrain_right_encoder_->GetRaw());
Comran Morshed9a9948c2016-01-16 15:58:04 +0000255 drivetrain_message->left_encoder =
256 -drivetrain_translate(drivetrain_left_encoder_->GetRaw());
257 drivetrain_message->left_speed =
258 drivetrain_velocity_translate(drivetrain_left_encoder_->GetPeriod());
259 drivetrain_message->right_speed =
260 drivetrain_velocity_translate(drivetrain_right_encoder_->GetPeriod());
261
Comran Morshed9a9948c2016-01-16 15:58:04 +0000262 drivetrain_message->left_shifter_position =
Comran Morshed225f0b92016-02-10 20:34:27 +0000263 hall_translate(drivetrain_left_hall_->GetVoltage());
Comran Morshed9a9948c2016-01-16 15:58:04 +0000264 drivetrain_message->right_shifter_position =
Comran Morshed225f0b92016-02-10 20:34:27 +0000265 hall_translate(drivetrain_right_hall_->GetVoltage());
Comran Morshed9a9948c2016-01-16 15:58:04 +0000266
267 drivetrain_message.Send();
268 }
Austin Schuh54667ac2019-02-02 16:44:49 -0800269 }
Comran Morshed9a9948c2016-01-16 15:58:04 +0000270
Austin Schuh54667ac2019-02-02 16:44:49 -0800271 void RunDmaIteration() {
272 const auto &values = constants::GetValues();
Comran Morshed225f0b92016-02-10 20:34:27 +0000273 {
274 auto shooter_message = shooter_queue.position.MakeMessage();
275 shooter_message->theta_left =
Austin Schuh9f77fd22016-02-21 02:53:58 -0800276 shooter_translate(-shooter_left_encoder_->GetRaw());
Comran Morshed225f0b92016-02-10 20:34:27 +0000277 shooter_message->theta_right =
278 shooter_translate(shooter_right_encoder_->GetRaw());
279 shooter_message.Send();
280 }
281
282 {
283 auto superstructure_message = superstructure_queue.position.MakeMessage();
Sabina Davis2243cab2019-02-05 21:45:08 -0800284 CopyPosition(intake_encoder_, &superstructure_message->intake,
285 intake_translate, intake_pot_translate, false,
286 values.intake.pot_offset);
287 CopyPosition(shoulder_encoder_, &superstructure_message->shoulder,
288 shoulder_translate, shoulder_pot_translate, false,
289 values.shoulder.pot_offset);
290 CopyPosition(wrist_encoder_, &superstructure_message->wrist,
291 wrist_translate, wrist_pot_translate, true,
292 values.wrist.pot_offset);
Comran Morshed225f0b92016-02-10 20:34:27 +0000293
294 superstructure_message.Send();
295 }
Comran Morshedaa0573c2016-03-05 19:05:54 +0000296
297 {
Austin Schuh4b652c92019-05-27 13:22:27 -0700298 auto ball_detector_message = ball_detector_sender_.MakeMessage();
Comran Morshedaa0573c2016-03-05 19:05:54 +0000299 ball_detector_message->voltage = ball_detector_->GetVoltage();
300 LOG_STRUCT(DEBUG, "ball detector", *ball_detector_message);
301 ball_detector_message.Send();
302 }
Brian Silverman68cb5c22016-03-20 18:11:14 -0700303
304 {
Austin Schuha250b2d2019-05-27 16:14:02 -0700305 auto auto_mode_message = auto_mode_sender_.MakeMessage();
Brian Silverman68cb5c22016-03-20 18:11:14 -0700306 auto_mode_message->mode = 0;
307 for (size_t i = 0; i < autonomous_modes_.size(); ++i) {
308 if (autonomous_modes_[i]->Get()) {
309 auto_mode_message->mode |= 1 << i;
310 }
311 }
312 LOG_STRUCT(DEBUG, "auto mode", *auto_mode_message);
313 auto_mode_message.Send();
314 }
Comran Morshed9a9948c2016-01-16 15:58:04 +0000315 }
316
Comran Morshed9a9948c2016-01-16 15:58:04 +0000317 private:
Austin Schuh4b652c92019-05-27 13:22:27 -0700318 ::aos::Sender<::y2016::sensors::BallDetector> ball_detector_sender_;
Austin Schuha250b2d2019-05-27 16:14:02 -0700319 ::aos::Sender<::frc971::autonomous::AutonomousMode> auto_mode_sender_;
Austin Schuh4b652c92019-05-27 13:22:27 -0700320
Comran Morshed225f0b92016-02-10 20:34:27 +0000321 ::std::unique_ptr<AnalogInput> drivetrain_left_hall_, drivetrain_right_hall_;
322
323 ::std::unique_ptr<Encoder> shooter_left_encoder_, shooter_right_encoder_;
Comran Morshedb79c4242016-02-06 18:27:26 +0000324 ::frc971::wpilib::DMAEncoderAndPotentiometer intake_encoder_,
325 shoulder_encoder_, wrist_encoder_;
Comran Morshedaa0573c2016-03-05 19:05:54 +0000326 ::std::unique_ptr<AnalogInput> ball_detector_;
Comran Morshed9a9948c2016-01-16 15:58:04 +0000327
Brian Silverman68cb5c22016-03-20 18:11:14 -0700328 ::std::array<::std::unique_ptr<DigitalInput>, 4> autonomous_modes_;
329
Austin Schuh54667ac2019-02-02 16:44:49 -0800330 DigitalGlitchFilter hall_filter_;
Comran Morshed9a9948c2016-01-16 15:58:04 +0000331};
332
333class SolenoidWriter {
334 public:
335 SolenoidWriter(const ::std::unique_ptr<::frc971::wpilib::BufferedPcm> &pcm)
336 : pcm_(pcm),
Comran Morshedb79c4242016-02-06 18:27:26 +0000337 drivetrain_(".frc971.control_loops.drivetrain_queue.output"),
Austin Schuh843412b2016-03-20 16:48:46 -0700338 shooter_(".y2016.control_loops.shooter.shooter_queue.output"),
Sabina Davis2243cab2019-02-05 21:45:08 -0800339 superstructure_(".y2016.control_loops.superstructure_queue.output") {}
Comran Morshed9a9948c2016-01-16 15:58:04 +0000340
Campbell Crowley1ab5fab2016-02-21 13:39:31 -0800341 void set_compressor(::std::unique_ptr<Compressor> compressor) {
342 compressor_ = ::std::move(compressor);
Comran Morshed9a9948c2016-01-16 15:58:04 +0000343 }
344
Comran Morshed71466fe2016-04-21 20:21:14 -0700345 void set_drivetrain_shifter(
Comran Morshed9a9948c2016-01-16 15:58:04 +0000346 ::std::unique_ptr<::frc971::wpilib::BufferedSolenoid> s) {
Comran Morshed71466fe2016-04-21 20:21:14 -0700347 drivetrain_shifter_ = ::std::move(s);
Comran Morshed9a9948c2016-01-16 15:58:04 +0000348 }
349
Comran Morshed71466fe2016-04-21 20:21:14 -0700350 void set_climber_trigger(
Comran Morshed9a9948c2016-01-16 15:58:04 +0000351 ::std::unique_ptr<::frc971::wpilib::BufferedSolenoid> s) {
Comran Morshed71466fe2016-04-21 20:21:14 -0700352 climber_trigger_ = ::std::move(s);
Comran Morshed9a9948c2016-01-16 15:58:04 +0000353 }
354
Sabina Davis2243cab2019-02-05 21:45:08 -0800355 void set_traverse(::std::unique_ptr<::frc971::wpilib::BufferedSolenoid> s) {
Austin Schuh843412b2016-03-20 16:48:46 -0700356 traverse_ = ::std::move(s);
357 }
358
359 void set_traverse_latch(
360 ::std::unique_ptr<::frc971::wpilib::BufferedSolenoid> s) {
361 traverse_latch_ = ::std::move(s);
362 }
363
Comran Morshedb79c4242016-02-06 18:27:26 +0000364 void set_shooter_clamp(
365 ::std::unique_ptr<::frc971::wpilib::BufferedSolenoid> s) {
366 shooter_clamp_ = ::std::move(s);
367 }
368
369 void set_shooter_pusher(
370 ::std::unique_ptr<::frc971::wpilib::BufferedSolenoid> s) {
371 shooter_pusher_ = ::std::move(s);
372 }
373
Sabina Davis2243cab2019-02-05 21:45:08 -0800374 void set_lights(::std::unique_ptr<::frc971::wpilib::BufferedSolenoid> s) {
Austin Schuhe0729a62016-03-12 21:54:17 -0800375 lights_ = ::std::move(s);
376 }
377
Austin Schuh8b89d332016-03-24 20:19:12 -0700378 void set_flashlight(::std::unique_ptr<Relay> relay) {
379 flashlight_ = ::std::move(relay);
380 }
381
Comran Morshed9a9948c2016-01-16 15:58:04 +0000382 void operator()() {
Campbell Crowley1ab5fab2016-02-21 13:39:31 -0800383 compressor_->Start();
Comran Morshed9a9948c2016-01-16 15:58:04 +0000384 ::aos::SetCurrentThreadName("Solenoids");
385 ::aos::SetCurrentThreadRealtimePriority(27);
386
Austin Schuh8aec1ed2016-05-01 13:29:20 -0700387 ::aos::time::PhasedLoop phased_loop(::std::chrono::milliseconds(20),
388 ::std::chrono::milliseconds(1));
Comran Morshed9a9948c2016-01-16 15:58:04 +0000389
390 while (run_) {
391 {
392 const int iterations = phased_loop.SleepUntilNext();
393 if (iterations != 1) {
394 LOG(DEBUG, "Solenoids skipped %d iterations\n", iterations - 1);
395 }
396 }
397
398 {
Comran Morshed9a9948c2016-01-16 15:58:04 +0000399 drivetrain_.FetchLatest();
400 if (drivetrain_.get()) {
401 LOG_STRUCT(DEBUG, "solenoids", *drivetrain_);
Comran Morshed71466fe2016-04-21 20:21:14 -0700402 drivetrain_shifter_->Set(
403 !(drivetrain_->left_high || drivetrain_->right_high));
Comran Morshed9a9948c2016-01-16 15:58:04 +0000404 }
405 }
406
407 {
Comran Morshedb79c4242016-02-06 18:27:26 +0000408 shooter_.FetchLatest();
409 if (shooter_.get()) {
410 LOG_STRUCT(DEBUG, "solenoids", *shooter_);
411 shooter_clamp_->Set(shooter_->clamp_open);
412 shooter_pusher_->Set(shooter_->push_to_shooter);
Austin Schuhe0729a62016-03-12 21:54:17 -0800413 lights_->Set(shooter_->lights_on);
Austin Schuhb2c33382016-04-03 16:09:17 -0700414 if (shooter_->forwards_flashlight) {
415 if (shooter_->backwards_flashlight) {
416 flashlight_->Set(Relay::kOn);
417 } else {
418 flashlight_->Set(Relay::kReverse);
419 }
420 } else {
421 if (shooter_->backwards_flashlight) {
422 flashlight_->Set(Relay::kForward);
423 } else {
424 flashlight_->Set(Relay::kOff);
425 }
426 }
Comran Morshedb79c4242016-02-06 18:27:26 +0000427 }
428 }
429
430 {
Austin Schuh843412b2016-03-20 16:48:46 -0700431 superstructure_.FetchLatest();
432 if (superstructure_.get()) {
Comran Morshed71466fe2016-04-21 20:21:14 -0700433 LOG_STRUCT(DEBUG, "solenoids", *superstructure_);
434
435 climber_trigger_->Set(superstructure_->unfold_climber);
436
Austin Schuh843412b2016-03-20 16:48:46 -0700437 traverse_->Set(superstructure_->traverse_down);
438 traverse_latch_->Set(superstructure_->traverse_unlatched);
439 }
440 }
441
442 {
Comran Morshed9a9948c2016-01-16 15:58:04 +0000443 ::frc971::wpilib::PneumaticsToLog to_log;
Sabina Davis2243cab2019-02-05 21:45:08 -0800444 { to_log.compressor_on = compressor_->Enabled(); }
Comran Morshed9a9948c2016-01-16 15:58:04 +0000445
446 pcm_->Flush();
447 to_log.read_solenoids = pcm_->GetAll();
448 LOG_STRUCT(DEBUG, "pneumatics info", to_log);
449 }
450 }
451 }
452
453 void Quit() { run_ = false; }
454
455 private:
456 const ::std::unique_ptr<::frc971::wpilib::BufferedPcm> &pcm_;
457
Comran Morshed71466fe2016-04-21 20:21:14 -0700458 ::std::unique_ptr<::frc971::wpilib::BufferedSolenoid> drivetrain_shifter_,
459 shooter_clamp_, shooter_pusher_, lights_, traverse_, traverse_latch_,
460 climber_trigger_;
Austin Schuh8b89d332016-03-24 20:19:12 -0700461 ::std::unique_ptr<Relay> flashlight_;
Campbell Crowley1ab5fab2016-02-21 13:39:31 -0800462 ::std::unique_ptr<Compressor> compressor_;
Comran Morshed9a9948c2016-01-16 15:58:04 +0000463
Comran Morshed9a9948c2016-01-16 15:58:04 +0000464 ::aos::Queue<::frc971::control_loops::DrivetrainQueue::Output> drivetrain_;
Comran Morshed3263e8f2016-02-14 17:55:45 +0000465 ::aos::Queue<::y2016::control_loops::shooter::ShooterQueue::Output> shooter_;
Sabina Davis2243cab2019-02-05 21:45:08 -0800466 ::aos::Queue<::y2016::control_loops::SuperstructureQueue::Output>
Austin Schuh843412b2016-03-20 16:48:46 -0700467 superstructure_;
Comran Morshed9a9948c2016-01-16 15:58:04 +0000468
469 ::std::atomic<bool> run_{true};
470};
471
Comran Morshed225f0b92016-02-10 20:34:27 +0000472class ShooterWriter : public ::frc971::wpilib::LoopOutputHandler {
473 public:
Austin Schuhdf6cbb12019-02-02 13:46:52 -0800474 ShooterWriter(::aos::EventLoop *event_loop)
475 : ::frc971::wpilib::LoopOutputHandler(event_loop) {}
476
Comran Morshed225f0b92016-02-10 20:34:27 +0000477 void set_shooter_left_talon(::std::unique_ptr<Talon> t) {
478 shooter_left_talon_ = ::std::move(t);
479 }
480
481 void set_shooter_right_talon(::std::unique_ptr<Talon> t) {
482 shooter_right_talon_ = ::std::move(t);
483 }
484
485 private:
486 virtual void Read() override {
487 ::y2016::control_loops::shooter::shooter_queue.output.FetchAnother();
488 }
489
490 virtual void Write() override {
491 auto &queue = ::y2016::control_loops::shooter::shooter_queue.output;
492 LOG_STRUCT(DEBUG, "will output", *queue);
Austin Schuhcaa1ee92016-02-27 14:45:37 -0800493
Brian Silverman6eaa2d82017-02-04 20:48:30 -0800494 shooter_left_talon_->SetSpeed(queue->voltage_left / 12.0);
495 shooter_right_talon_->SetSpeed(-queue->voltage_right / 12.0);
Comran Morshed225f0b92016-02-10 20:34:27 +0000496 }
497
498 virtual void Stop() override {
499 LOG(WARNING, "Shooter output too old.\n");
Brian Silverman6eaa2d82017-02-04 20:48:30 -0800500 shooter_left_talon_->SetDisabled();
501 shooter_right_talon_->SetDisabled();
Comran Morshed225f0b92016-02-10 20:34:27 +0000502 }
503
504 ::std::unique_ptr<Talon> shooter_left_talon_, shooter_right_talon_;
505};
506
507class SuperstructureWriter : public ::frc971::wpilib::LoopOutputHandler {
508 public:
Austin Schuhdf6cbb12019-02-02 13:46:52 -0800509 SuperstructureWriter(::aos::EventLoop *event_loop)
510 : ::frc971::wpilib::LoopOutputHandler(event_loop) {}
511
Comran Morshed225f0b92016-02-10 20:34:27 +0000512 void set_intake_talon(::std::unique_ptr<Talon> t) {
513 intake_talon_ = ::std::move(t);
514 }
515
516 void set_shoulder_talon(::std::unique_ptr<Talon> t) {
517 shoulder_talon_ = ::std::move(t);
518 }
519
520 void set_wrist_talon(::std::unique_ptr<Talon> t) {
521 wrist_talon_ = ::std::move(t);
522 }
523
Campbell Crowleyd4fd6552016-02-21 17:53:46 -0800524 void set_top_rollers_talon(::std::unique_ptr<Talon> t) {
525 top_rollers_talon_ = ::std::move(t);
526 }
527
528 void set_bottom_rollers_talon(::std::unique_ptr<Talon> t) {
529 bottom_rollers_talon_ = ::std::move(t);
Comran Morshedf4cd7642016-02-15 20:40:49 +0000530 }
531
Comran Morshed71466fe2016-04-21 20:21:14 -0700532 void set_climber_talon(::std::unique_ptr<Talon> t) {
533 climber_talon_ = ::std::move(t);
534 }
535
Comran Morshed225f0b92016-02-10 20:34:27 +0000536 private:
537 virtual void Read() override {
538 ::y2016::control_loops::superstructure_queue.output.FetchAnother();
539 }
540
541 virtual void Write() override {
542 auto &queue = ::y2016::control_loops::superstructure_queue.output;
543 LOG_STRUCT(DEBUG, "will output", *queue);
Brian Silverman6eaa2d82017-02-04 20:48:30 -0800544 intake_talon_->SetSpeed(::aos::Clip(queue->voltage_intake,
545 -kMaxBringupPower, kMaxBringupPower) /
546 12.0);
547 shoulder_talon_->SetSpeed(::aos::Clip(-queue->voltage_shoulder,
548 -kMaxBringupPower, kMaxBringupPower) /
549 12.0);
550 wrist_talon_->SetSpeed(
Austin Schuha9992ff2016-02-28 21:59:23 -0800551 ::aos::Clip(queue->voltage_wrist, -kMaxBringupPower, kMaxBringupPower) /
552 12.0);
Brian Silverman6eaa2d82017-02-04 20:48:30 -0800553 top_rollers_talon_->SetSpeed(-queue->voltage_top_rollers / 12.0);
554 bottom_rollers_talon_->SetSpeed(-queue->voltage_bottom_rollers / 12.0);
555 climber_talon_->SetSpeed(-queue->voltage_climber / 12.0);
Comran Morshed225f0b92016-02-10 20:34:27 +0000556 }
557
558 virtual void Stop() override {
559 LOG(WARNING, "Superstructure output too old.\n");
Brian Silverman6eaa2d82017-02-04 20:48:30 -0800560 intake_talon_->SetDisabled();
561 shoulder_talon_->SetDisabled();
562 wrist_talon_->SetDisabled();
Comran Morshed225f0b92016-02-10 20:34:27 +0000563 }
564
Comran Morshedf4cd7642016-02-15 20:40:49 +0000565 ::std::unique_ptr<Talon> intake_talon_, shoulder_talon_, wrist_talon_,
Comran Morshed71466fe2016-04-21 20:21:14 -0700566 top_rollers_talon_, bottom_rollers_talon_, climber_talon_;
Comran Morshed9a9948c2016-01-16 15:58:04 +0000567};
568
Comran Morshed9a9948c2016-01-16 15:58:04 +0000569class WPILibRobot : public ::frc971::wpilib::WPILibRobotBase {
570 public:
571 ::std::unique_ptr<Encoder> make_encoder(int index) {
572 return make_unique<Encoder>(10 + index * 2, 11 + index * 2, false,
573 Encoder::k4X);
574 }
575
576 void Run() override {
577 ::aos::InitNRT();
578 ::aos::SetCurrentThreadName("StartCompetition");
579
Austin Schuhdf6cbb12019-02-02 13:46:52 -0800580 ::aos::ShmEventLoop event_loop;
581
582 ::frc971::wpilib::JoystickSender joystick_sender(&event_loop);
Comran Morshed9a9948c2016-01-16 15:58:04 +0000583 ::std::thread joystick_thread(::std::ref(joystick_sender));
584
Austin Schuh0b545432019-05-12 15:46:12 -0700585 ::frc971::wpilib::PDPFetcher pdp_fetcher(&event_loop);
Comran Morshed9a9948c2016-01-16 15:58:04 +0000586 ::std::thread pdp_fetcher_thread(::std::ref(pdp_fetcher));
Austin Schuhdf6cbb12019-02-02 13:46:52 -0800587 SensorReader reader(&event_loop);
Comran Morshed9a9948c2016-01-16 15:58:04 +0000588
Austin Schuh9f77fd22016-02-21 02:53:58 -0800589 reader.set_drivetrain_left_encoder(make_encoder(5));
590 reader.set_drivetrain_right_encoder(make_encoder(6));
591 reader.set_drivetrain_left_hall(make_unique<AnalogInput>(5));
592 reader.set_drivetrain_right_hall(make_unique<AnalogInput>(6));
Comran Morshed225f0b92016-02-10 20:34:27 +0000593
Austin Schuhf0c05762016-04-03 16:06:49 -0700594 reader.set_shooter_left_encoder(make_encoder(7));
595 reader.set_shooter_right_encoder(make_encoder(-3));
Comran Morshed225f0b92016-02-10 20:34:27 +0000596
597 reader.set_intake_encoder(make_encoder(0));
598 reader.set_intake_index(make_unique<DigitalInput>(0));
599 reader.set_intake_potentiometer(make_unique<AnalogInput>(0));
600
Austin Schuhf0c05762016-04-03 16:06:49 -0700601 reader.set_shoulder_encoder(make_encoder(4));
Austin Schuh9f77fd22016-02-21 02:53:58 -0800602 reader.set_shoulder_index(make_unique<DigitalInput>(2));
603 reader.set_shoulder_potentiometer(make_unique<AnalogInput>(2));
Comran Morshed225f0b92016-02-10 20:34:27 +0000604
Austin Schuh9f77fd22016-02-21 02:53:58 -0800605 reader.set_wrist_encoder(make_encoder(1));
606 reader.set_wrist_index(make_unique<DigitalInput>(1));
607 reader.set_wrist_potentiometer(make_unique<AnalogInput>(1));
Comran Morshed9a9948c2016-01-16 15:58:04 +0000608
Comran Morshedaa0573c2016-03-05 19:05:54 +0000609 reader.set_ball_detector(make_unique<AnalogInput>(7));
610
Brian Silverman68cb5c22016-03-20 18:11:14 -0700611 reader.set_autonomous_mode(0, make_unique<DigitalInput>(9));
612 reader.set_autonomous_mode(1, make_unique<DigitalInput>(8));
613 reader.set_autonomous_mode(2, make_unique<DigitalInput>(7));
614 reader.set_autonomous_mode(3, make_unique<DigitalInput>(6));
615
Comran Morshed9a9948c2016-01-16 15:58:04 +0000616 ::std::thread reader_thread(::std::ref(reader));
617
Brian Silverman003a4732018-03-11 14:02:15 -0700618 // TODO(Brian): This interacts poorly with the SpiRxClearer in ADIS16448.
Austin Schuhdf6cbb12019-02-02 13:46:52 -0800619 ::frc971::wpilib::GyroSender gyro_sender(&event_loop);
Comran Morshed9a9948c2016-01-16 15:58:04 +0000620 ::std::thread gyro_thread(::std::ref(gyro_sender));
621
Austin Schuhf0c05762016-04-03 16:06:49 -0700622 auto imu_trigger = make_unique<DigitalInput>(3);
Austin Schuhdf6cbb12019-02-02 13:46:52 -0800623 ::frc971::wpilib::ADIS16448 imu(&event_loop, SPI::Port::kMXP,
624 imu_trigger.get());
Brian Silverman5f17a972016-02-28 01:49:32 -0500625 ::std::thread imu_thread(::std::ref(imu));
Brian Silverman5f17a972016-02-28 01:49:32 -0500626
Austin Schuhdf6cbb12019-02-02 13:46:52 -0800627 ::frc971::wpilib::DrivetrainWriter drivetrain_writer(&event_loop);
Sabina Davisb71bc282019-02-03 01:17:23 -0800628 drivetrain_writer.set_left_controller0(
629 ::std::unique_ptr<Talon>(new Talon(5)), false);
630 drivetrain_writer.set_right_controller0(
631 ::std::unique_ptr<Talon>(new Talon(4)), true);
Comran Morshed9a9948c2016-01-16 15:58:04 +0000632 ::std::thread drivetrain_writer_thread(::std::ref(drivetrain_writer));
633
Austin Schuhdf6cbb12019-02-02 13:46:52 -0800634 ShooterWriter shooter_writer(&event_loop);
Comran Morshed225f0b92016-02-10 20:34:27 +0000635 shooter_writer.set_shooter_left_talon(
Austin Schuh0c2b58c2016-02-21 17:23:46 -0800636 ::std::unique_ptr<Talon>(new Talon(9)));
Comran Morshed225f0b92016-02-10 20:34:27 +0000637 shooter_writer.set_shooter_right_talon(
Austin Schuh0c2b58c2016-02-21 17:23:46 -0800638 ::std::unique_ptr<Talon>(new Talon(8)));
Comran Morshed225f0b92016-02-10 20:34:27 +0000639 ::std::thread shooter_writer_thread(::std::ref(shooter_writer));
640
Austin Schuhdf6cbb12019-02-02 13:46:52 -0800641 SuperstructureWriter superstructure_writer(&event_loop);
Comran Morshed225f0b92016-02-10 20:34:27 +0000642 superstructure_writer.set_intake_talon(
Austin Schuh0c2b58c2016-02-21 17:23:46 -0800643 ::std::unique_ptr<Talon>(new Talon(3)));
Comran Morshed225f0b92016-02-10 20:34:27 +0000644 superstructure_writer.set_shoulder_talon(
Austin Schuh0c2b58c2016-02-21 17:23:46 -0800645 ::std::unique_ptr<Talon>(new Talon(6)));
Comran Morshed225f0b92016-02-10 20:34:27 +0000646 superstructure_writer.set_wrist_talon(
Austin Schuh0c2b58c2016-02-21 17:23:46 -0800647 ::std::unique_ptr<Talon>(new Talon(2)));
Campbell Crowleyd4fd6552016-02-21 17:53:46 -0800648 superstructure_writer.set_top_rollers_talon(
Austin Schuh0c2b58c2016-02-21 17:23:46 -0800649 ::std::unique_ptr<Talon>(new Talon(1)));
Austin Schuhcaa1ee92016-02-27 14:45:37 -0800650 superstructure_writer.set_bottom_rollers_talon(
651 ::std::unique_ptr<Talon>(new Talon(0)));
Comran Morshed71466fe2016-04-21 20:21:14 -0700652 superstructure_writer.set_climber_talon(
653 ::std::unique_ptr<Talon>(new Talon(7)));
Comran Morshed225f0b92016-02-10 20:34:27 +0000654 ::std::thread superstructure_writer_thread(
655 ::std::ref(superstructure_writer));
656
Comran Morshed9a9948c2016-01-16 15:58:04 +0000657 ::std::unique_ptr<::frc971::wpilib::BufferedPcm> pcm(
658 new ::frc971::wpilib::BufferedPcm());
659 SolenoidWriter solenoid_writer(pcm);
Comran Morshed71466fe2016-04-21 20:21:14 -0700660 solenoid_writer.set_drivetrain_shifter(pcm->MakeSolenoid(0));
Austin Schuh843412b2016-03-20 16:48:46 -0700661 solenoid_writer.set_traverse_latch(pcm->MakeSolenoid(2));
662 solenoid_writer.set_traverse(pcm->MakeSolenoid(3));
Austin Schuhcaa1ee92016-02-27 14:45:37 -0800663 solenoid_writer.set_shooter_clamp(pcm->MakeSolenoid(4));
664 solenoid_writer.set_shooter_pusher(pcm->MakeSolenoid(5));
Austin Schuhe0729a62016-03-12 21:54:17 -0800665 solenoid_writer.set_lights(pcm->MakeSolenoid(6));
Comran Morshed71466fe2016-04-21 20:21:14 -0700666 solenoid_writer.set_climber_trigger(pcm->MakeSolenoid(1));
Austin Schuh8b89d332016-03-24 20:19:12 -0700667 solenoid_writer.set_flashlight(make_unique<Relay>(0));
Comran Morshed9a9948c2016-01-16 15:58:04 +0000668
Campbell Crowley1ab5fab2016-02-21 13:39:31 -0800669 solenoid_writer.set_compressor(make_unique<Compressor>());
670
Comran Morshed9a9948c2016-01-16 15:58:04 +0000671 ::std::thread solenoid_thread(::std::ref(solenoid_writer));
672
673 // Wait forever. Not much else to do...
674 while (true) {
675 const int r = select(0, nullptr, nullptr, nullptr, nullptr);
676 if (r != 0) {
677 PLOG(WARNING, "infinite select failed");
678 } else {
679 PLOG(WARNING, "infinite select succeeded??\n");
680 }
681 }
682
683 LOG(ERROR, "Exiting WPILibRobot\n");
684
685 joystick_sender.Quit();
686 joystick_thread.join();
687 pdp_fetcher.Quit();
688 pdp_fetcher_thread.join();
689 reader.Quit();
690 reader_thread.join();
691 gyro_sender.Quit();
692 gyro_thread.join();
Brian Silverman5f17a972016-02-28 01:49:32 -0500693 imu.Quit();
694 imu_thread.join();
Comran Morshed9a9948c2016-01-16 15:58:04 +0000695
696 drivetrain_writer.Quit();
697 drivetrain_writer_thread.join();
Comran Morshed225f0b92016-02-10 20:34:27 +0000698 shooter_writer.Quit();
699 shooter_writer_thread.join();
700 superstructure_writer.Quit();
701 superstructure_writer_thread.join();
Comran Morshed9a9948c2016-01-16 15:58:04 +0000702 solenoid_writer.Quit();
703 solenoid_thread.join();
704
705 ::aos::Cleanup();
706 }
707};
708
709} // namespace wpilib
Comran Morshed6c6a0a92016-01-17 12:45:16 +0000710} // namespace y2016
Comran Morshed9a9948c2016-01-16 15:58:04 +0000711
Comran Morshed6c6a0a92016-01-17 12:45:16 +0000712AOS_ROBOT_CLASS(::y2016::wpilib::WPILibRobot);