Austin Schuh | 010eb81 | 2014-10-25 18:06:49 -0700 | [diff] [blame] | 1 | #include <stdio.h> |
| 2 | #include <string.h> |
Austin Schuh | 010eb81 | 2014-10-25 18:06:49 -0700 | [diff] [blame] | 3 | #include <unistd.h> |
| 4 | #include <inttypes.h> |
| 5 | |
Brian Silverman | d8f403a | 2014-12-13 19:12:04 -0500 | [diff] [blame] | 6 | #include <thread> |
| 7 | #include <mutex> |
| 8 | #include <functional> |
| 9 | |
Austin Schuh | 010eb81 | 2014-10-25 18:06:49 -0700 | [diff] [blame] | 10 | #include "aos/common/logging/logging.h" |
| 11 | #include "aos/common/logging/queue_logging.h" |
Austin Schuh | 010eb81 | 2014-10-25 18:06:49 -0700 | [diff] [blame] | 12 | #include "aos/common/time.h" |
| 13 | #include "aos/common/util/log_interval.h" |
| 14 | #include "aos/common/util/phased_loop.h" |
| 15 | #include "aos/common/util/wrapping_counter.h" |
Brian Silverman | b073f24 | 2014-09-08 16:29:57 -0400 | [diff] [blame] | 16 | #include "aos/common/stl_mutex.h" |
Austin Schuh | 010eb81 | 2014-10-25 18:06:49 -0700 | [diff] [blame] | 17 | #include "aos/linux_code/init.h" |
Brian Silverman | 699f0cb | 2015-02-05 19:45:01 -0500 | [diff] [blame] | 18 | #include "aos/common/messages/robot_state.q.h" |
Austin Schuh | 010eb81 | 2014-10-25 18:06:49 -0700 | [diff] [blame] | 19 | |
Daniel Petti | a782741 | 2015-02-13 20:55:57 -0800 | [diff] [blame] | 20 | #include "frc971/constants.h" |
Brian Silverman | 335c20e | 2015-01-26 21:47:58 -0500 | [diff] [blame] | 21 | #include "frc971/control_loops/control_loops.q.h" |
Austin Schuh | 010eb81 | 2014-10-25 18:06:49 -0700 | [diff] [blame] | 22 | #include "frc971/control_loops/drivetrain/drivetrain.q.h" |
Daniel Petti | adf3843 | 2015-01-26 17:13:35 -0800 | [diff] [blame] | 23 | #include "frc971/control_loops/fridge/fridge.q.h" |
| 24 | #include "frc971/control_loops/claw/claw.q.h" |
Austin Schuh | 010eb81 | 2014-10-25 18:06:49 -0700 | [diff] [blame] | 25 | |
Brian Silverman | da45b6c | 2014-12-28 11:36:50 -0800 | [diff] [blame] | 26 | #include "frc971/wpilib/hall_effect.h" |
| 27 | #include "frc971/wpilib/joystick_sender.h" |
Brian Silverman | d8f403a | 2014-12-13 19:12:04 -0500 | [diff] [blame] | 28 | #include "frc971/wpilib/loop_output_handler.h" |
| 29 | #include "frc971/wpilib/buffered_solenoid.h" |
| 30 | #include "frc971/wpilib/buffered_pcm.h" |
Brian Silverman | 07ec88e | 2014-12-28 00:13:08 -0800 | [diff] [blame] | 31 | #include "frc971/wpilib/gyro_sender.h" |
Brian Silverman | ff7b347 | 2015-01-26 17:53:04 -0500 | [diff] [blame] | 32 | #include "frc971/wpilib/dma_edge_counting.h" |
Brian Silverman | 70ec719 | 2015-01-26 17:52:40 -0500 | [diff] [blame] | 33 | #include "frc971/wpilib/interrupt_edge_counting.h" |
Brian Silverman | 4da5807 | 2015-01-26 20:18:52 -0500 | [diff] [blame] | 34 | #include "frc971/wpilib/encoder_and_potentiometer.h" |
Brian Silverman | da45b6c | 2014-12-28 11:36:50 -0800 | [diff] [blame] | 35 | |
Brian Silverman | cb77f23 | 2014-12-19 21:48:36 -0800 | [diff] [blame] | 36 | #include "Encoder.h" |
Brian Silverman | cb77f23 | 2014-12-19 21:48:36 -0800 | [diff] [blame] | 37 | #include "Talon.h" |
| 38 | #include "DriverStation.h" |
| 39 | #include "AnalogInput.h" |
Brian Silverman | cb77f23 | 2014-12-19 21:48:36 -0800 | [diff] [blame] | 40 | #include "Compressor.h" |
| 41 | #include "RobotBase.h" |
Brian Silverman | 335c20e | 2015-01-26 21:47:58 -0500 | [diff] [blame] | 42 | #include "dma.h" |
Brian Silverman | 699f0cb | 2015-02-05 19:45:01 -0500 | [diff] [blame] | 43 | #include "ControllerPower.h" |
Austin Schuh | 010eb81 | 2014-10-25 18:06:49 -0700 | [diff] [blame] | 44 | |
| 45 | #ifndef M_PI |
| 46 | #define M_PI 3.14159265358979323846 |
| 47 | #endif |
| 48 | |
| 49 | using ::aos::util::SimpleLogInterval; |
Brian Silverman | ada5f2c | 2015-02-01 02:41:14 -0500 | [diff] [blame] | 50 | using ::frc971::control_loops::drivetrain_queue; |
Brian Silverman | 335c20e | 2015-01-26 21:47:58 -0500 | [diff] [blame] | 51 | using ::frc971::control_loops::fridge_queue; |
| 52 | using ::frc971::control_loops::claw_queue; |
Austin Schuh | 010eb81 | 2014-10-25 18:06:49 -0700 | [diff] [blame] | 53 | |
| 54 | namespace frc971 { |
Brian Silverman | da45b6c | 2014-12-28 11:36:50 -0800 | [diff] [blame] | 55 | namespace wpilib { |
Austin Schuh | 010eb81 | 2014-10-25 18:06:49 -0700 | [diff] [blame] | 56 | |
Austin Schuh | 010eb81 | 2014-10-25 18:06:49 -0700 | [diff] [blame] | 57 | double drivetrain_translate(int32_t in) { |
Austin Schuh | db51603 | 2014-12-28 00:12:38 -0800 | [diff] [blame] | 58 | return static_cast<double>(in) / |
Daniel Petti | adf3843 | 2015-01-26 17:13:35 -0800 | [diff] [blame] | 59 | (256.0 /*cpr*/ * 4.0 /*4x*/) * |
Daniel Petti | a782741 | 2015-02-13 20:55:57 -0800 | [diff] [blame] | 60 | constants::GetValues().drivetrain_encoder_ratio * |
Daniel Petti | adf3843 | 2015-01-26 17:13:35 -0800 | [diff] [blame] | 61 | (4 /*wheel diameter*/ * 2.54 / 100.0 * M_PI); |
| 62 | } |
| 63 | |
| 64 | double arm_translate(int32_t in) { |
Austin Schuh | 6246c54 | 2015-02-16 02:59:09 -0800 | [diff] [blame] | 65 | return -static_cast<double>(in) / |
Daniel Petti | adf3843 | 2015-01-26 17:13:35 -0800 | [diff] [blame] | 66 | (512.0 /*cpr*/ * 4.0 /*4x*/) * |
Daniel Petti | a782741 | 2015-02-13 20:55:57 -0800 | [diff] [blame] | 67 | constants::GetValues().arm_encoder_ratio * |
Daniel Petti | adf3843 | 2015-01-26 17:13:35 -0800 | [diff] [blame] | 68 | (2 * M_PI /*radians*/); |
| 69 | } |
| 70 | |
Brian Silverman | 5d712fc | 2015-02-15 03:39:31 -0500 | [diff] [blame] | 71 | double arm_potentiometer_translate(double voltage) { |
Austin Schuh | 35d0661 | 2015-02-15 23:35:23 -0800 | [diff] [blame] | 72 | return voltage * |
Daniel Petti | a782741 | 2015-02-13 20:55:57 -0800 | [diff] [blame] | 73 | constants::GetValues().arm_pot_ratio * |
Austin Schuh | 35d0661 | 2015-02-15 23:35:23 -0800 | [diff] [blame] | 74 | (5.0 /*turns*/ / 5.0 /*volts*/) * |
Daniel Petti | adf3843 | 2015-01-26 17:13:35 -0800 | [diff] [blame] | 75 | (2 * M_PI /*radians*/); |
| 76 | } |
| 77 | |
| 78 | double elevator_translate(int32_t in) { |
| 79 | return static_cast<double>(in) / |
| 80 | (512.0 /*cpr*/ * 4.0 /*4x*/) * |
Daniel Petti | a782741 | 2015-02-13 20:55:57 -0800 | [diff] [blame] | 81 | constants::GetValues().elev_encoder_ratio * |
| 82 | (2 * M_PI /*radians*/) * |
| 83 | constants::GetValues().elev_distance_per_radian; |
Daniel Petti | adf3843 | 2015-01-26 17:13:35 -0800 | [diff] [blame] | 84 | } |
| 85 | |
Brian Silverman | 5d712fc | 2015-02-15 03:39:31 -0500 | [diff] [blame] | 86 | double elevator_potentiometer_translate(double voltage) { |
Austin Schuh | 6246c54 | 2015-02-16 02:59:09 -0800 | [diff] [blame] | 87 | return -voltage * |
Daniel Petti | a782741 | 2015-02-13 20:55:57 -0800 | [diff] [blame] | 88 | constants::GetValues().elev_pot_ratio * |
| 89 | (2 * M_PI /*radians*/) * |
| 90 | constants::GetValues().elev_distance_per_radian * |
Austin Schuh | 35d0661 | 2015-02-15 23:35:23 -0800 | [diff] [blame] | 91 | (5.0 /*turns*/ / 5.0 /*volts*/); |
Daniel Petti | adf3843 | 2015-01-26 17:13:35 -0800 | [diff] [blame] | 92 | } |
| 93 | |
| 94 | double claw_translate(int32_t in) { |
| 95 | return static_cast<double>(in) / |
| 96 | (512.0 /*cpr*/ * 4.0 /*4x*/) * |
Daniel Petti | a782741 | 2015-02-13 20:55:57 -0800 | [diff] [blame] | 97 | constants::GetValues().claw_encoder_ratio * |
Daniel Petti | adf3843 | 2015-01-26 17:13:35 -0800 | [diff] [blame] | 98 | (2 * M_PI /*radians*/); |
| 99 | } |
| 100 | |
Brian Silverman | 5d712fc | 2015-02-15 03:39:31 -0500 | [diff] [blame] | 101 | double claw_potentiometer_translate(double voltage) { |
Austin Schuh | 6246c54 | 2015-02-16 02:59:09 -0800 | [diff] [blame] | 102 | return -voltage * |
Daniel Petti | a782741 | 2015-02-13 20:55:57 -0800 | [diff] [blame] | 103 | constants::GetValues().claw_pot_ratio * |
Austin Schuh | 35d0661 | 2015-02-15 23:35:23 -0800 | [diff] [blame] | 104 | (5.0 /*turns*/ / 5.0 /*volts*/) * |
Daniel Petti | adf3843 | 2015-01-26 17:13:35 -0800 | [diff] [blame] | 105 | (2 * M_PI /*radians*/); |
Austin Schuh | 010eb81 | 2014-10-25 18:06:49 -0700 | [diff] [blame] | 106 | } |
| 107 | |
Brian Silverman | 335c20e | 2015-01-26 21:47:58 -0500 | [diff] [blame] | 108 | static const double kMaximumEncoderPulsesPerSecond = |
| 109 | 19500.0 /* free speed RPM */ * 12.0 / 56.0 /* belt reduction */ / |
| 110 | 60.0 /* seconds / minute */ * 256.0 /* CPR */ * |
| 111 | 4.0 /* index pulse = 1/4 cycle */; |
| 112 | |
Austin Schuh | 010eb81 | 2014-10-25 18:06:49 -0700 | [diff] [blame] | 113 | class SensorReader { |
| 114 | public: |
Brian Silverman | 1f90d67 | 2015-01-26 20:20:45 -0500 | [diff] [blame] | 115 | SensorReader() { |
Brian Silverman | 335c20e | 2015-01-26 21:47:58 -0500 | [diff] [blame] | 116 | // Set it to filter out anything shorter than 1/4 of the minimum pulse width |
| 117 | // we should ever see. |
| 118 | filter_.SetPeriodNanoSeconds( |
| 119 | static_cast<int>(1 / 4.0 / kMaximumEncoderPulsesPerSecond * 1e9 + 0.5)); |
| 120 | } |
| 121 | |
| 122 | void set_arm_left_encoder(::std::unique_ptr<Encoder> encoder) { |
| 123 | filter_.Add(encoder.get()); |
| 124 | arm_left_encoder_.set_encoder(::std::move(encoder)); |
| 125 | } |
| 126 | |
| 127 | void set_arm_left_index(::std::unique_ptr<DigitalSource> index) { |
| 128 | filter_.Add(index.get()); |
| 129 | arm_left_encoder_.set_index(::std::move(index)); |
| 130 | } |
| 131 | |
| 132 | void set_arm_left_potentiometer( |
| 133 | ::std::unique_ptr<AnalogInput> potentiometer) { |
| 134 | arm_left_encoder_.set_potentiometer(::std::move(potentiometer)); |
| 135 | } |
| 136 | |
| 137 | void set_arm_right_encoder(::std::unique_ptr<Encoder> encoder) { |
| 138 | filter_.Add(encoder.get()); |
| 139 | arm_right_encoder_.set_encoder(::std::move(encoder)); |
| 140 | } |
| 141 | |
| 142 | void set_arm_right_index(::std::unique_ptr<DigitalSource> index) { |
| 143 | filter_.Add(index.get()); |
| 144 | arm_right_encoder_.set_index(::std::move(index)); |
| 145 | } |
| 146 | |
| 147 | void set_arm_right_potentiometer( |
| 148 | ::std::unique_ptr<AnalogInput> potentiometer) { |
| 149 | arm_right_encoder_.set_potentiometer(::std::move(potentiometer)); |
| 150 | } |
| 151 | |
| 152 | void set_elevator_left_encoder(::std::unique_ptr<Encoder> encoder) { |
| 153 | filter_.Add(encoder.get()); |
| 154 | elevator_left_encoder_.set_encoder(::std::move(encoder)); |
| 155 | } |
| 156 | |
| 157 | void set_elevator_left_index(::std::unique_ptr<DigitalSource> index) { |
| 158 | filter_.Add(index.get()); |
| 159 | elevator_left_encoder_.set_index(::std::move(index)); |
| 160 | } |
| 161 | |
| 162 | void set_elevator_left_potentiometer( |
| 163 | ::std::unique_ptr<AnalogInput> potentiometer) { |
| 164 | elevator_left_encoder_.set_potentiometer(::std::move(potentiometer)); |
| 165 | } |
| 166 | |
| 167 | void set_elevator_right_encoder(::std::unique_ptr<Encoder> encoder) { |
| 168 | filter_.Add(encoder.get()); |
| 169 | elevator_right_encoder_.set_encoder(::std::move(encoder)); |
| 170 | } |
| 171 | |
| 172 | void set_elevator_right_index(::std::unique_ptr<DigitalSource> index) { |
| 173 | filter_.Add(index.get()); |
| 174 | elevator_right_encoder_.set_index(::std::move(index)); |
| 175 | } |
| 176 | |
| 177 | void set_elevator_right_potentiometer( |
| 178 | ::std::unique_ptr<AnalogInput> potentiometer) { |
| 179 | elevator_right_encoder_.set_potentiometer(::std::move(potentiometer)); |
| 180 | } |
| 181 | |
| 182 | void set_wrist_encoder(::std::unique_ptr<Encoder> encoder) { |
| 183 | filter_.Add(encoder.get()); |
| 184 | wrist_encoder_.set_encoder(::std::move(encoder)); |
| 185 | } |
| 186 | |
| 187 | void set_wrist_index(::std::unique_ptr<DigitalSource> index) { |
| 188 | filter_.Add(index.get()); |
| 189 | wrist_encoder_.set_index(::std::move(index)); |
| 190 | } |
| 191 | |
| 192 | void set_wrist_potentiometer(::std::unique_ptr<AnalogInput> potentiometer) { |
| 193 | wrist_encoder_.set_potentiometer(::std::move(potentiometer)); |
Austin Schuh | 010eb81 | 2014-10-25 18:06:49 -0700 | [diff] [blame] | 194 | } |
| 195 | |
Brian Silverman | 1f90d67 | 2015-01-26 20:20:45 -0500 | [diff] [blame] | 196 | void set_left_encoder(::std::unique_ptr<Encoder> left_encoder) { |
| 197 | left_encoder_ = ::std::move(left_encoder); |
| 198 | } |
| 199 | |
| 200 | void set_right_encoder(::std::unique_ptr<Encoder> right_encoder) { |
| 201 | right_encoder_ = ::std::move(right_encoder); |
| 202 | } |
| 203 | |
Brian Silverman | 335c20e | 2015-01-26 21:47:58 -0500 | [diff] [blame] | 204 | // All of the DMA-related set_* calls must be made before this, and it doesn't |
| 205 | // hurt to do all of them. |
| 206 | void set_dma(::std::unique_ptr<DMA> dma) { |
| 207 | dma_synchronizer_.reset(new DMASynchronizer(::std::move(dma))); |
| 208 | dma_synchronizer_->Add(&arm_left_encoder_); |
Brian Silverman | 335c20e | 2015-01-26 21:47:58 -0500 | [diff] [blame] | 209 | dma_synchronizer_->Add(&elevator_left_encoder_); |
Austin Schuh | 58d8cdf | 2015-02-15 21:04:42 -0800 | [diff] [blame] | 210 | dma_synchronizer_->Add(&arm_right_encoder_); |
Brian Silverman | 335c20e | 2015-01-26 21:47:58 -0500 | [diff] [blame] | 211 | dma_synchronizer_->Add(&elevator_right_encoder_); |
| 212 | } |
| 213 | |
Austin Schuh | 010eb81 | 2014-10-25 18:06:49 -0700 | [diff] [blame] | 214 | void operator()() { |
Austin Schuh | 58d8cdf | 2015-02-15 21:04:42 -0800 | [diff] [blame] | 215 | LOG(INFO, "In sensor reader thread\n"); |
Brian Silverman | 2fe007c | 2014-12-28 12:20:01 -0800 | [diff] [blame] | 216 | ::aos::SetCurrentThreadName("SensorReader"); |
| 217 | |
Brian Silverman | 699f0cb | 2015-02-05 19:45:01 -0500 | [diff] [blame] | 218 | my_pid_ = getpid(); |
| 219 | ds_ = DriverStation::GetInstance(); |
| 220 | |
Brian Silverman | 335c20e | 2015-01-26 21:47:58 -0500 | [diff] [blame] | 221 | wrist_encoder_.Start(); |
| 222 | dma_synchronizer_->Start(); |
Austin Schuh | 58d8cdf | 2015-02-15 21:04:42 -0800 | [diff] [blame] | 223 | LOG(INFO, "Things are now started\n"); |
Austin Schuh | 010eb81 | 2014-10-25 18:06:49 -0700 | [diff] [blame] | 224 | |
Brian Silverman | 2fe007c | 2014-12-28 12:20:01 -0800 | [diff] [blame] | 225 | ::aos::SetCurrentThreadRealtimePriority(kPriority); |
Austin Schuh | 010eb81 | 2014-10-25 18:06:49 -0700 | [diff] [blame] | 226 | while (run_) { |
Austin Schuh | 58d8cdf | 2015-02-15 21:04:42 -0800 | [diff] [blame] | 227 | ::aos::time::PhasedLoopXMS(5, 4000); |
Austin Schuh | 010eb81 | 2014-10-25 18:06:49 -0700 | [diff] [blame] | 228 | RunIteration(); |
Austin Schuh | 010eb81 | 2014-10-25 18:06:49 -0700 | [diff] [blame] | 229 | } |
Brian Silverman | 335c20e | 2015-01-26 21:47:58 -0500 | [diff] [blame] | 230 | |
| 231 | wrist_encoder_.Stop(); |
Austin Schuh | 010eb81 | 2014-10-25 18:06:49 -0700 | [diff] [blame] | 232 | } |
| 233 | |
| 234 | void RunIteration() { |
Brian Silverman | 699f0cb | 2015-02-05 19:45:01 -0500 | [diff] [blame] | 235 | { |
| 236 | auto new_state = ::aos::robot_state.MakeMessage(); |
Austin Schuh | 010eb81 | 2014-10-25 18:06:49 -0700 | [diff] [blame] | 237 | |
Brian Silverman | 699f0cb | 2015-02-05 19:45:01 -0500 | [diff] [blame] | 238 | new_state->outputs_enabled = ds_->IsSysActive(); |
| 239 | new_state->browned_out = ds_->IsSysBrownedOut(); |
| 240 | |
| 241 | new_state->is_3v3_active = ControllerPower::GetEnabled3V3(); |
| 242 | new_state->is_5v_active = ControllerPower::GetEnabled5V(); |
| 243 | new_state->voltage_3v3 = ControllerPower::GetVoltage3V3(); |
| 244 | new_state->voltage_5v = ControllerPower::GetVoltage5V(); |
| 245 | |
| 246 | new_state->voltage_roborio_in = ControllerPower::GetInputVoltage(); |
| 247 | new_state->voltage_battery = ds_->GetBatteryVoltage(); |
| 248 | |
Brian Silverman | 4d882aa | 2015-02-20 13:40:42 -0500 | [diff] [blame^] | 249 | LOG_STRUCT(DEBUG, "robot_state", *new_state); |
| 250 | |
Brian Silverman | 699f0cb | 2015-02-05 19:45:01 -0500 | [diff] [blame] | 251 | new_state.Send(); |
Austin Schuh | 010eb81 | 2014-10-25 18:06:49 -0700 | [diff] [blame] | 252 | } |
| 253 | |
Austin Schuh | 35d0661 | 2015-02-15 23:35:23 -0800 | [diff] [blame] | 254 | { |
| 255 | auto drivetrain_message = drivetrain_queue.position.MakeMessage(); |
| 256 | drivetrain_message->right_encoder = |
| 257 | drivetrain_translate(right_encoder_->GetRaw()); |
| 258 | drivetrain_message->left_encoder = |
| 259 | -drivetrain_translate(left_encoder_->GetRaw()); |
| 260 | |
| 261 | drivetrain_message.Send(); |
| 262 | } |
Brian Silverman | 335c20e | 2015-01-26 21:47:58 -0500 | [diff] [blame] | 263 | |
| 264 | dma_synchronizer_->RunIteration(); |
| 265 | |
Philipp Schrader | 82c6507 | 2015-02-16 00:47:09 +0000 | [diff] [blame] | 266 | const auto &values = constants::GetValues(); |
Brian Silverman | 5d712fc | 2015-02-15 03:39:31 -0500 | [diff] [blame] | 267 | |
Brian Silverman | 335c20e | 2015-01-26 21:47:58 -0500 | [diff] [blame] | 268 | { |
| 269 | auto fridge_message = fridge_queue.position.MakeMessage(); |
| 270 | CopyPotAndIndexPosition(arm_left_encoder_, &fridge_message->arm.left, |
Brian Silverman | 5d712fc | 2015-02-15 03:39:31 -0500 | [diff] [blame] | 271 | arm_translate, arm_potentiometer_translate, false, |
Austin Schuh | 6246c54 | 2015-02-16 02:59:09 -0800 | [diff] [blame] | 272 | values.fridge.left_arm_potentiometer_offset); |
Brian Silverman | 5d712fc | 2015-02-15 03:39:31 -0500 | [diff] [blame] | 273 | CopyPotAndIndexPosition( |
| 274 | arm_right_encoder_, &fridge_message->arm.right, arm_translate, |
| 275 | arm_potentiometer_translate, true, |
Austin Schuh | 6246c54 | 2015-02-16 02:59:09 -0800 | [diff] [blame] | 276 | values.fridge.right_arm_potentiometer_offset); |
Brian Silverman | 335c20e | 2015-01-26 21:47:58 -0500 | [diff] [blame] | 277 | CopyPotAndIndexPosition( |
| 278 | elevator_left_encoder_, &fridge_message->elevator.left, |
Brian Silverman | 5d712fc | 2015-02-15 03:39:31 -0500 | [diff] [blame] | 279 | elevator_translate, elevator_potentiometer_translate, false, |
Austin Schuh | 6246c54 | 2015-02-16 02:59:09 -0800 | [diff] [blame] | 280 | values.fridge.left_elevator_potentiometer_offset); |
Brian Silverman | 5d712fc | 2015-02-15 03:39:31 -0500 | [diff] [blame] | 281 | CopyPotAndIndexPosition( |
| 282 | elevator_right_encoder_, &fridge_message->elevator.right, |
| 283 | elevator_translate, elevator_potentiometer_translate, true, |
Austin Schuh | 6246c54 | 2015-02-16 02:59:09 -0800 | [diff] [blame] | 284 | values.fridge.right_elevator_potentiometer_offset); |
Brian Silverman | 335c20e | 2015-01-26 21:47:58 -0500 | [diff] [blame] | 285 | fridge_message.Send(); |
| 286 | } |
| 287 | |
| 288 | { |
| 289 | auto claw_message = claw_queue.position.MakeMessage(); |
| 290 | CopyPotAndIndexPosition(wrist_encoder_, &claw_message->joint, |
Brian Silverman | 5d712fc | 2015-02-15 03:39:31 -0500 | [diff] [blame] | 291 | claw_translate, claw_potentiometer_translate, |
| 292 | false, values.claw.potentiometer_offset); |
Brian Silverman | 335c20e | 2015-01-26 21:47:58 -0500 | [diff] [blame] | 293 | claw_message.Send(); |
| 294 | } |
Austin Schuh | 010eb81 | 2014-10-25 18:06:49 -0700 | [diff] [blame] | 295 | } |
| 296 | |
| 297 | void Quit() { run_ = false; } |
| 298 | |
| 299 | private: |
Brian Silverman | 335c20e | 2015-01-26 21:47:58 -0500 | [diff] [blame] | 300 | static const int kPriority = 30; |
| 301 | static const int kInterruptPriority = 55; |
| 302 | |
Brian Silverman | 699f0cb | 2015-02-05 19:45:01 -0500 | [diff] [blame] | 303 | int32_t my_pid_; |
| 304 | DriverStation *ds_; |
| 305 | |
Brian Silverman | 335c20e | 2015-01-26 21:47:58 -0500 | [diff] [blame] | 306 | void CopyPotAndIndexPosition( |
| 307 | const DMAEncoderAndPotentiometer &encoder, PotAndIndexPosition *position, |
| 308 | ::std::function<double(int32_t)> encoder_translate, |
Brian Silverman | 5d712fc | 2015-02-15 03:39:31 -0500 | [diff] [blame] | 309 | ::std::function<double(double)> potentiometer_translate, bool reverse, |
| 310 | double potentiometer_offset) { |
Brian Silverman | 335c20e | 2015-01-26 21:47:58 -0500 | [diff] [blame] | 311 | const double multiplier = reverse ? -1.0 : 1.0; |
| 312 | position->encoder = |
| 313 | multiplier * encoder_translate(encoder.polled_encoder_value()); |
Brian Silverman | 5d712fc | 2015-02-15 03:39:31 -0500 | [diff] [blame] | 314 | position->pot = multiplier * potentiometer_translate( |
| 315 | encoder.polled_potentiometer_voltage()) + |
| 316 | potentiometer_offset; |
Brian Silverman | 335c20e | 2015-01-26 21:47:58 -0500 | [diff] [blame] | 317 | position->latched_encoder = |
| 318 | multiplier * encoder_translate(encoder.last_encoder_value()); |
| 319 | position->latched_pot = |
Brian Silverman | 5d712fc | 2015-02-15 03:39:31 -0500 | [diff] [blame] | 320 | multiplier * |
| 321 | potentiometer_translate(encoder.last_potentiometer_voltage()) + |
| 322 | potentiometer_offset; |
Brian Silverman | 335c20e | 2015-01-26 21:47:58 -0500 | [diff] [blame] | 323 | position->index_pulses = encoder.index_posedge_count(); |
| 324 | } |
| 325 | |
| 326 | void CopyPotAndIndexPosition( |
| 327 | const InterruptEncoderAndPotentiometer &encoder, |
| 328 | PotAndIndexPosition *position, |
| 329 | ::std::function<double(int32_t)> encoder_translate, |
Brian Silverman | 5d712fc | 2015-02-15 03:39:31 -0500 | [diff] [blame] | 330 | ::std::function<double(double)> potentiometer_translate, bool reverse, |
| 331 | double potentiometer_offset) { |
Brian Silverman | 335c20e | 2015-01-26 21:47:58 -0500 | [diff] [blame] | 332 | const double multiplier = reverse ? -1.0 : 1.0; |
| 333 | position->encoder = |
| 334 | multiplier * encoder_translate(encoder.encoder()->GetRaw()); |
Brian Silverman | 5d712fc | 2015-02-15 03:39:31 -0500 | [diff] [blame] | 335 | position->pot = multiplier * potentiometer_translate( |
| 336 | encoder.potentiometer()->GetVoltage()) + |
| 337 | potentiometer_offset; |
Brian Silverman | 335c20e | 2015-01-26 21:47:58 -0500 | [diff] [blame] | 338 | position->latched_encoder = |
| 339 | multiplier * encoder_translate(encoder.last_encoder_value()); |
| 340 | position->latched_pot = |
Brian Silverman | 5d712fc | 2015-02-15 03:39:31 -0500 | [diff] [blame] | 341 | multiplier * |
| 342 | potentiometer_translate(encoder.last_potentiometer_voltage()) + |
| 343 | potentiometer_offset; |
Brian Silverman | 335c20e | 2015-01-26 21:47:58 -0500 | [diff] [blame] | 344 | position->index_pulses = encoder.index_posedge_count(); |
| 345 | } |
| 346 | |
Brian Silverman | 335c20e | 2015-01-26 21:47:58 -0500 | [diff] [blame] | 347 | ::std::unique_ptr<DMASynchronizer> dma_synchronizer_; |
| 348 | |
| 349 | DMAEncoderAndPotentiometer arm_left_encoder_, arm_right_encoder_, |
| 350 | elevator_left_encoder_, elevator_right_encoder_; |
| 351 | |
| 352 | InterruptEncoderAndPotentiometer wrist_encoder_{kInterruptPriority}; |
| 353 | |
Austin Schuh | 010eb81 | 2014-10-25 18:06:49 -0700 | [diff] [blame] | 354 | ::std::unique_ptr<Encoder> left_encoder_; |
| 355 | ::std::unique_ptr<Encoder> right_encoder_; |
Austin Schuh | 010eb81 | 2014-10-25 18:06:49 -0700 | [diff] [blame] | 356 | |
Brian Silverman | 1f90d67 | 2015-01-26 20:20:45 -0500 | [diff] [blame] | 357 | ::std::atomic<bool> run_{true}; |
Austin Schuh | 010eb81 | 2014-10-25 18:06:49 -0700 | [diff] [blame] | 358 | DigitalGlitchFilter filter_; |
| 359 | }; |
| 360 | |
Brian Silverman | d8f403a | 2014-12-13 19:12:04 -0500 | [diff] [blame] | 361 | class SolenoidWriter { |
Austin Schuh | 010eb81 | 2014-10-25 18:06:49 -0700 | [diff] [blame] | 362 | public: |
Brian Silverman | d8f403a | 2014-12-13 19:12:04 -0500 | [diff] [blame] | 363 | SolenoidWriter(const ::std::unique_ptr<BufferedPcm> &pcm) |
Daniel Petti | adf3843 | 2015-01-26 17:13:35 -0800 | [diff] [blame] | 364 | : pcm_(pcm), |
| 365 | fridge_(".frc971.control_loops.fridge.output"), |
| 366 | claw_(".frc971.control_loops.claw.output") {} |
Brian Silverman | d8f403a | 2014-12-13 19:12:04 -0500 | [diff] [blame] | 367 | |
Daniel Petti | adf3843 | 2015-01-26 17:13:35 -0800 | [diff] [blame] | 368 | void set_fridge_grabbers_top_front(::std::unique_ptr<BufferedSolenoid> s) { |
| 369 | fridge_grabbers_top_front_ = ::std::move(s); |
Austin Schuh | 010eb81 | 2014-10-25 18:06:49 -0700 | [diff] [blame] | 370 | } |
| 371 | |
Daniel Petti | adf3843 | 2015-01-26 17:13:35 -0800 | [diff] [blame] | 372 | void set_fridge_grabbers_top_back(::std::unique_ptr<BufferedSolenoid> s) { |
| 373 | fridge_grabbers_top_back_ = ::std::move(s); |
| 374 | } |
| 375 | |
| 376 | void set_fridge_grabbers_bottom_front( |
| 377 | ::std::unique_ptr<BufferedSolenoid> s) { |
| 378 | fridge_grabbers_bottom_front_ = ::std::move(s); |
| 379 | } |
| 380 | |
| 381 | void set_fridge_grabbers_bottom_back( |
| 382 | ::std::unique_ptr<BufferedSolenoid> s) { |
| 383 | fridge_grabbers_bottom_back_ = ::std::move(s); |
| 384 | } |
| 385 | |
| 386 | void set_claw_pinchers(::std::unique_ptr<BufferedSolenoid> s) { |
| 387 | claw_pinchers_ = ::std::move(s); |
Brian Silverman | d8f403a | 2014-12-13 19:12:04 -0500 | [diff] [blame] | 388 | } |
Austin Schuh | 010eb81 | 2014-10-25 18:06:49 -0700 | [diff] [blame] | 389 | |
Brian Silverman | d8f403a | 2014-12-13 19:12:04 -0500 | [diff] [blame] | 390 | void operator()() { |
| 391 | ::aos::SetCurrentThreadName("Solenoids"); |
| 392 | ::aos::SetCurrentThreadRealtimePriority(30); |
| 393 | |
| 394 | while (run_) { |
| 395 | ::aos::time::PhasedLoopXMS(20, 1000); |
| 396 | |
| 397 | { |
Daniel Petti | adf3843 | 2015-01-26 17:13:35 -0800 | [diff] [blame] | 398 | fridge_.FetchLatest(); |
| 399 | if (fridge_.get()) { |
| 400 | LOG_STRUCT(DEBUG, "solenoids", *fridge_); |
| 401 | fridge_grabbers_top_front_->Set(fridge_->grabbers.top_front); |
| 402 | fridge_grabbers_top_back_->Set(fridge_->grabbers.top_back); |
| 403 | fridge_grabbers_bottom_front_->Set(fridge_->grabbers.bottom_front); |
| 404 | fridge_grabbers_bottom_back_->Set(fridge_->grabbers.bottom_back); |
| 405 | } |
| 406 | } |
| 407 | |
| 408 | { |
| 409 | claw_.FetchLatest(); |
| 410 | if (claw_.get()) { |
| 411 | LOG_STRUCT(DEBUG, "solenoids", *claw_); |
| 412 | claw_pinchers_->Set(claw_->rollers_closed); |
Brian Silverman | d8f403a | 2014-12-13 19:12:04 -0500 | [diff] [blame] | 413 | } |
| 414 | } |
| 415 | |
Brian Silverman | d8f403a | 2014-12-13 19:12:04 -0500 | [diff] [blame] | 416 | pcm_->Flush(); |
Austin Schuh | 010eb81 | 2014-10-25 18:06:49 -0700 | [diff] [blame] | 417 | } |
| 418 | } |
| 419 | |
Brian Silverman | d8f403a | 2014-12-13 19:12:04 -0500 | [diff] [blame] | 420 | void Quit() { run_ = false; } |
Austin Schuh | 010eb81 | 2014-10-25 18:06:49 -0700 | [diff] [blame] | 421 | |
Brian Silverman | d8f403a | 2014-12-13 19:12:04 -0500 | [diff] [blame] | 422 | private: |
| 423 | const ::std::unique_ptr<BufferedPcm> &pcm_; |
Daniel Petti | adf3843 | 2015-01-26 17:13:35 -0800 | [diff] [blame] | 424 | ::std::unique_ptr<BufferedSolenoid> fridge_grabbers_top_front_; |
| 425 | ::std::unique_ptr<BufferedSolenoid> fridge_grabbers_top_back_; |
| 426 | ::std::unique_ptr<BufferedSolenoid> fridge_grabbers_bottom_front_; |
| 427 | ::std::unique_ptr<BufferedSolenoid> fridge_grabbers_bottom_back_; |
| 428 | ::std::unique_ptr<BufferedSolenoid> claw_pinchers_; |
Austin Schuh | 010eb81 | 2014-10-25 18:06:49 -0700 | [diff] [blame] | 429 | |
Daniel Petti | adf3843 | 2015-01-26 17:13:35 -0800 | [diff] [blame] | 430 | ::aos::Queue<::frc971::control_loops::FridgeQueue::Output> fridge_; |
| 431 | ::aos::Queue<::frc971::control_loops::ClawQueue::Output> claw_; |
Austin Schuh | 010eb81 | 2014-10-25 18:06:49 -0700 | [diff] [blame] | 432 | |
Brian Silverman | d8f403a | 2014-12-13 19:12:04 -0500 | [diff] [blame] | 433 | ::std::atomic<bool> run_{true}; |
| 434 | }; |
| 435 | |
| 436 | class DrivetrainWriter : public LoopOutputHandler { |
| 437 | public: |
| 438 | void set_left_drivetrain_talon(::std::unique_ptr<Talon> t) { |
| 439 | left_drivetrain_talon_ = ::std::move(t); |
Austin Schuh | 010eb81 | 2014-10-25 18:06:49 -0700 | [diff] [blame] | 440 | } |
| 441 | |
Brian Silverman | d8f403a | 2014-12-13 19:12:04 -0500 | [diff] [blame] | 442 | void set_right_drivetrain_talon(::std::unique_ptr<Talon> t) { |
| 443 | right_drivetrain_talon_ = ::std::move(t); |
| 444 | } |
Austin Schuh | 010eb81 | 2014-10-25 18:06:49 -0700 | [diff] [blame] | 445 | |
Brian Silverman | d8f403a | 2014-12-13 19:12:04 -0500 | [diff] [blame] | 446 | private: |
| 447 | virtual void Read() override { |
Brian Silverman | ada5f2c | 2015-02-01 02:41:14 -0500 | [diff] [blame] | 448 | ::frc971::control_loops::drivetrain_queue.output.FetchAnother(); |
Brian Silverman | d8f403a | 2014-12-13 19:12:04 -0500 | [diff] [blame] | 449 | } |
| 450 | |
| 451 | virtual void Write() override { |
Brian Silverman | ada5f2c | 2015-02-01 02:41:14 -0500 | [diff] [blame] | 452 | auto &queue = ::frc971::control_loops::drivetrain_queue.output; |
Brian Silverman | d8f403a | 2014-12-13 19:12:04 -0500 | [diff] [blame] | 453 | LOG_STRUCT(DEBUG, "will output", *queue); |
Austin Schuh | a004b0c | 2015-02-16 17:06:30 -0800 | [diff] [blame] | 454 | left_drivetrain_talon_->Set(queue->left_voltage / 12.0); |
| 455 | right_drivetrain_talon_->Set(-queue->right_voltage / 12.0); |
Brian Silverman | d8f403a | 2014-12-13 19:12:04 -0500 | [diff] [blame] | 456 | } |
| 457 | |
| 458 | virtual void Stop() override { |
| 459 | LOG(WARNING, "drivetrain output too old\n"); |
| 460 | left_drivetrain_talon_->Disable(); |
| 461 | right_drivetrain_talon_->Disable(); |
| 462 | } |
| 463 | |
Austin Schuh | 010eb81 | 2014-10-25 18:06:49 -0700 | [diff] [blame] | 464 | ::std::unique_ptr<Talon> left_drivetrain_talon_; |
Brian Silverman | d8f403a | 2014-12-13 19:12:04 -0500 | [diff] [blame] | 465 | ::std::unique_ptr<Talon> right_drivetrain_talon_; |
| 466 | }; |
| 467 | |
Daniel Petti | adf3843 | 2015-01-26 17:13:35 -0800 | [diff] [blame] | 468 | class FridgeWriter : public LoopOutputHandler { |
| 469 | public: |
| 470 | void set_left_arm_talon(::std::unique_ptr<Talon> t) { |
| 471 | left_arm_talon_ = ::std::move(t); |
| 472 | } |
| 473 | |
| 474 | void set_right_arm_talon(::std::unique_ptr<Talon> t) { |
| 475 | right_arm_talon_ = ::std::move(t); |
| 476 | } |
| 477 | |
| 478 | void set_left_elevator_talon(::std::unique_ptr<Talon> t) { |
| 479 | left_elevator_talon_ = ::std::move(t); |
| 480 | } |
| 481 | |
| 482 | void set_right_elevator_talon(::std::unique_ptr<Talon> t) { |
| 483 | right_elevator_talon_ = ::std::move(t); |
| 484 | } |
| 485 | |
| 486 | private: |
| 487 | virtual void Read() override { |
| 488 | ::frc971::control_loops::fridge_queue.output.FetchAnother(); |
| 489 | } |
| 490 | |
| 491 | virtual void Write() override { |
| 492 | auto &queue = ::frc971::control_loops::fridge_queue.output; |
| 493 | LOG_STRUCT(DEBUG, "will output", *queue); |
Austin Schuh | 859a930 | 2015-02-16 15:45:45 -0800 | [diff] [blame] | 494 | left_arm_talon_->Set(queue->left_arm / 12.0); |
| 495 | right_arm_talon_->Set(-queue->right_arm / 12.0); |
| 496 | left_elevator_talon_->Set(queue->left_elevator / 12.0); |
| 497 | right_elevator_talon_->Set(-queue->right_elevator / 12.0); |
Daniel Petti | adf3843 | 2015-01-26 17:13:35 -0800 | [diff] [blame] | 498 | } |
| 499 | |
| 500 | virtual void Stop() override { |
| 501 | LOG(WARNING, "Fridge output too old.\n"); |
| 502 | left_arm_talon_->Disable(); |
| 503 | right_arm_talon_->Disable(); |
| 504 | left_elevator_talon_->Disable(); |
| 505 | right_elevator_talon_->Disable(); |
| 506 | } |
| 507 | |
| 508 | ::std::unique_ptr<Talon> left_arm_talon_; |
| 509 | ::std::unique_ptr<Talon> right_arm_talon_; |
| 510 | ::std::unique_ptr<Talon> left_elevator_talon_; |
| 511 | ::std::unique_ptr<Talon> right_elevator_talon_; |
| 512 | }; |
| 513 | |
| 514 | class ClawWriter : public LoopOutputHandler { |
| 515 | public: |
Austin Schuh | 58d8cdf | 2015-02-15 21:04:42 -0800 | [diff] [blame] | 516 | void set_left_intake_talon(::std::unique_ptr<Talon> t) { |
| 517 | left_intake_talon_ = ::std::move(t); |
| 518 | } |
| 519 | |
| 520 | void set_right_intake_talon(::std::unique_ptr<Talon> t) { |
| 521 | right_intake_talon_ = ::std::move(t); |
Daniel Petti | adf3843 | 2015-01-26 17:13:35 -0800 | [diff] [blame] | 522 | } |
| 523 | |
| 524 | void set_wrist_talon(::std::unique_ptr<Talon> t) { |
| 525 | wrist_talon_ = ::std::move(t); |
| 526 | } |
| 527 | |
| 528 | private: |
| 529 | virtual void Read() override { |
| 530 | ::frc971::control_loops::claw_queue.output.FetchAnother(); |
| 531 | } |
| 532 | |
| 533 | virtual void Write() override { |
| 534 | auto &queue = ::frc971::control_loops::claw_queue.output; |
| 535 | LOG_STRUCT(DEBUG, "will output", *queue); |
Austin Schuh | 58d8cdf | 2015-02-15 21:04:42 -0800 | [diff] [blame] | 536 | left_intake_talon_->Set(queue->intake_voltage / 12.0); |
Austin Schuh | 8a436e8 | 2015-02-16 23:31:28 -0800 | [diff] [blame] | 537 | right_intake_talon_->Set(-queue->intake_voltage / 12.0); |
| 538 | wrist_talon_->Set(-queue->voltage / 12.0); |
Daniel Petti | adf3843 | 2015-01-26 17:13:35 -0800 | [diff] [blame] | 539 | } |
| 540 | |
| 541 | virtual void Stop() override { |
| 542 | LOG(WARNING, "Claw output too old.\n"); |
Austin Schuh | 58d8cdf | 2015-02-15 21:04:42 -0800 | [diff] [blame] | 543 | left_intake_talon_->Disable(); |
| 544 | right_intake_talon_->Disable(); |
Daniel Petti | adf3843 | 2015-01-26 17:13:35 -0800 | [diff] [blame] | 545 | wrist_talon_->Disable(); |
| 546 | } |
| 547 | |
Austin Schuh | 58d8cdf | 2015-02-15 21:04:42 -0800 | [diff] [blame] | 548 | ::std::unique_ptr<Talon> left_intake_talon_; |
| 549 | ::std::unique_ptr<Talon> right_intake_talon_; |
Daniel Petti | adf3843 | 2015-01-26 17:13:35 -0800 | [diff] [blame] | 550 | ::std::unique_ptr<Talon> wrist_talon_; |
| 551 | }; |
| 552 | |
Brian Silverman | 1f90d67 | 2015-01-26 20:20:45 -0500 | [diff] [blame] | 553 | // TODO(brian): Replace this with ::std::make_unique once all our toolchains |
| 554 | // have support. |
| 555 | template <class T, class... U> |
| 556 | std::unique_ptr<T> make_unique(U &&... u) { |
| 557 | return std::unique_ptr<T>(new T(std::forward<U>(u)...)); |
| 558 | } |
| 559 | |
Austin Schuh | 010eb81 | 2014-10-25 18:06:49 -0700 | [diff] [blame] | 560 | class WPILibRobot : public RobotBase { |
| 561 | public: |
Austin Schuh | 58d8cdf | 2015-02-15 21:04:42 -0800 | [diff] [blame] | 562 | ::std::unique_ptr<Encoder> encoder(int index) { |
| 563 | return make_unique<Encoder>(10 + index * 2, 11 + index * 2, false, |
| 564 | Encoder::k4X); |
| 565 | } |
Austin Schuh | 010eb81 | 2014-10-25 18:06:49 -0700 | [diff] [blame] | 566 | virtual void StartCompetition() { |
Brian Silverman | d8f403a | 2014-12-13 19:12:04 -0500 | [diff] [blame] | 567 | ::aos::InitNRT(); |
Brian Silverman | 2fe007c | 2014-12-28 12:20:01 -0800 | [diff] [blame] | 568 | ::aos::SetCurrentThreadName("StartCompetition"); |
Brian Silverman | d8f403a | 2014-12-13 19:12:04 -0500 | [diff] [blame] | 569 | |
Brian Silverman | 98f6ee2 | 2015-01-26 17:50:12 -0500 | [diff] [blame] | 570 | JoystickSender joystick_sender; |
Austin Schuh | 010eb81 | 2014-10-25 18:06:49 -0700 | [diff] [blame] | 571 | ::std::thread joystick_thread(::std::ref(joystick_sender)); |
Austin Schuh | 58d8cdf | 2015-02-15 21:04:42 -0800 | [diff] [blame] | 572 | // TODO(austin): Compressor needs to use a spike. |
Brian Silverman | d8f403a | 2014-12-13 19:12:04 -0500 | [diff] [blame] | 573 | ::std::unique_ptr<Compressor> compressor(new Compressor()); |
| 574 | compressor->SetClosedLoopControl(true); |
| 575 | |
Brian Silverman | 98f6ee2 | 2015-01-26 17:50:12 -0500 | [diff] [blame] | 576 | SensorReader reader; |
Austin Schuh | 58d8cdf | 2015-02-15 21:04:42 -0800 | [diff] [blame] | 577 | LOG(INFO, "Creating the reader\n"); |
| 578 | reader.set_arm_left_encoder(encoder(1)); |
| 579 | reader.set_arm_left_index(make_unique<DigitalInput>(1)); |
| 580 | reader.set_arm_left_potentiometer(make_unique<AnalogInput>(1)); |
| 581 | |
| 582 | reader.set_arm_right_encoder(encoder(5)); |
| 583 | reader.set_arm_right_index(make_unique<DigitalInput>(5)); |
| 584 | reader.set_arm_right_potentiometer(make_unique<AnalogInput>(5)); |
| 585 | |
| 586 | reader.set_elevator_left_encoder(encoder(0)); |
| 587 | reader.set_elevator_left_index(make_unique<DigitalInput>(0)); |
| 588 | reader.set_elevator_left_potentiometer(make_unique<AnalogInput>(0)); |
| 589 | |
| 590 | reader.set_elevator_right_encoder(encoder(4)); |
| 591 | reader.set_elevator_right_index(make_unique<DigitalInput>(4)); |
| 592 | reader.set_elevator_right_potentiometer(make_unique<AnalogInput>(4)); |
| 593 | |
| 594 | reader.set_wrist_encoder(encoder(6)); |
| 595 | reader.set_wrist_index(make_unique<DigitalInput>(6)); |
| 596 | reader.set_wrist_potentiometer(make_unique<AnalogInput>(6)); |
| 597 | |
| 598 | reader.set_left_encoder(encoder(2)); |
| 599 | reader.set_right_encoder(encoder(3)); |
Brian Silverman | 335c20e | 2015-01-26 21:47:58 -0500 | [diff] [blame] | 600 | reader.set_dma(make_unique<DMA>()); |
Brian Silverman | 98f6ee2 | 2015-01-26 17:50:12 -0500 | [diff] [blame] | 601 | ::std::thread reader_thread(::std::ref(reader)); |
| 602 | GyroSender gyro_sender; |
| 603 | ::std::thread gyro_thread(::std::ref(gyro_sender)); |
| 604 | |
| 605 | DrivetrainWriter drivetrain_writer; |
Brian Silverman | d8f403a | 2014-12-13 19:12:04 -0500 | [diff] [blame] | 606 | drivetrain_writer.set_left_drivetrain_talon( |
Austin Schuh | 58d8cdf | 2015-02-15 21:04:42 -0800 | [diff] [blame] | 607 | ::std::unique_ptr<Talon>(new Talon(8))); |
Brian Silverman | d8f403a | 2014-12-13 19:12:04 -0500 | [diff] [blame] | 608 | drivetrain_writer.set_right_drivetrain_talon( |
Austin Schuh | 58d8cdf | 2015-02-15 21:04:42 -0800 | [diff] [blame] | 609 | ::std::unique_ptr<Talon>(new Talon(0))); |
Brian Silverman | d8f403a | 2014-12-13 19:12:04 -0500 | [diff] [blame] | 610 | ::std::thread drivetrain_writer_thread(::std::ref(drivetrain_writer)); |
| 611 | |
Daniel Petti | adf3843 | 2015-01-26 17:13:35 -0800 | [diff] [blame] | 612 | // TODO(sensors): Get real PWM output and relay numbers for the fridge and |
| 613 | // claw. |
| 614 | FridgeWriter fridge_writer; |
| 615 | fridge_writer.set_left_arm_talon( |
Austin Schuh | 58d8cdf | 2015-02-15 21:04:42 -0800 | [diff] [blame] | 616 | ::std::unique_ptr<Talon>(new Talon(6))); |
Daniel Petti | adf3843 | 2015-01-26 17:13:35 -0800 | [diff] [blame] | 617 | fridge_writer.set_right_arm_talon( |
Austin Schuh | 58d8cdf | 2015-02-15 21:04:42 -0800 | [diff] [blame] | 618 | ::std::unique_ptr<Talon>(new Talon(2))); |
Daniel Petti | adf3843 | 2015-01-26 17:13:35 -0800 | [diff] [blame] | 619 | fridge_writer.set_left_elevator_talon( |
Austin Schuh | 58d8cdf | 2015-02-15 21:04:42 -0800 | [diff] [blame] | 620 | ::std::unique_ptr<Talon>(new Talon(7))); |
Daniel Petti | adf3843 | 2015-01-26 17:13:35 -0800 | [diff] [blame] | 621 | fridge_writer.set_right_elevator_talon( |
Austin Schuh | 58d8cdf | 2015-02-15 21:04:42 -0800 | [diff] [blame] | 622 | ::std::unique_ptr<Talon>(new Talon(1))); |
Daniel Petti | adf3843 | 2015-01-26 17:13:35 -0800 | [diff] [blame] | 623 | ::std::thread fridge_writer_thread(::std::ref(fridge_writer)); |
| 624 | |
| 625 | ClawWriter claw_writer; |
Austin Schuh | 58d8cdf | 2015-02-15 21:04:42 -0800 | [diff] [blame] | 626 | claw_writer.set_left_intake_talon( |
| 627 | ::std::unique_ptr<Talon>(new Talon(5))); |
| 628 | claw_writer.set_right_intake_talon( |
| 629 | ::std::unique_ptr<Talon>(new Talon(3))); |
Daniel Petti | adf3843 | 2015-01-26 17:13:35 -0800 | [diff] [blame] | 630 | claw_writer.set_wrist_talon( |
Austin Schuh | 58d8cdf | 2015-02-15 21:04:42 -0800 | [diff] [blame] | 631 | ::std::unique_ptr<Talon>(new Talon(4))); |
Daniel Petti | adf3843 | 2015-01-26 17:13:35 -0800 | [diff] [blame] | 632 | ::std::thread claw_writer_thread(::std::ref(claw_writer)); |
| 633 | |
| 634 | ::std::unique_ptr<::frc971::wpilib::BufferedPcm> pcm( |
| 635 | new ::frc971::wpilib::BufferedPcm()); |
Brian Silverman | 98f6ee2 | 2015-01-26 17:50:12 -0500 | [diff] [blame] | 636 | SolenoidWriter solenoid_writer(pcm); |
Austin Schuh | 58d8cdf | 2015-02-15 21:04:42 -0800 | [diff] [blame] | 637 | solenoid_writer.set_fridge_grabbers_top_front(pcm->MakeSolenoid(1)); |
| 638 | solenoid_writer.set_fridge_grabbers_top_back(pcm->MakeSolenoid(1)); |
| 639 | solenoid_writer.set_fridge_grabbers_bottom_front(pcm->MakeSolenoid(2)); |
| 640 | solenoid_writer.set_fridge_grabbers_bottom_back(pcm->MakeSolenoid(3)); |
| 641 | solenoid_writer.set_claw_pinchers(pcm->MakeSolenoid(0)); |
Brian Silverman | d8f403a | 2014-12-13 19:12:04 -0500 | [diff] [blame] | 642 | ::std::thread solenoid_thread(::std::ref(solenoid_writer)); |
| 643 | |
| 644 | // Wait forever. Not much else to do... |
| 645 | PCHECK(select(0, nullptr, nullptr, nullptr, nullptr)); |
| 646 | |
Austin Schuh | 010eb81 | 2014-10-25 18:06:49 -0700 | [diff] [blame] | 647 | LOG(ERROR, "Exiting WPILibRobot\n"); |
Brian Silverman | 07ec88e | 2014-12-28 00:13:08 -0800 | [diff] [blame] | 648 | |
Austin Schuh | 010eb81 | 2014-10-25 18:06:49 -0700 | [diff] [blame] | 649 | joystick_sender.Quit(); |
| 650 | joystick_thread.join(); |
Brian Silverman | d8f403a | 2014-12-13 19:12:04 -0500 | [diff] [blame] | 651 | reader.Quit(); |
| 652 | reader_thread.join(); |
Brian Silverman | 07ec88e | 2014-12-28 00:13:08 -0800 | [diff] [blame] | 653 | gyro_sender.Quit(); |
| 654 | gyro_thread.join(); |
Brian Silverman | d8f403a | 2014-12-13 19:12:04 -0500 | [diff] [blame] | 655 | |
| 656 | drivetrain_writer.Quit(); |
| 657 | drivetrain_writer_thread.join(); |
Brian Silverman | d8f403a | 2014-12-13 19:12:04 -0500 | [diff] [blame] | 658 | solenoid_writer.Quit(); |
| 659 | solenoid_thread.join(); |
| 660 | |
Austin Schuh | 010eb81 | 2014-10-25 18:06:49 -0700 | [diff] [blame] | 661 | ::aos::Cleanup(); |
| 662 | } |
| 663 | }; |
| 664 | |
Brian Silverman | 98f6ee2 | 2015-01-26 17:50:12 -0500 | [diff] [blame] | 665 | } // namespace wpilib |
| 666 | } // namespace frc971 |
Austin Schuh | db51603 | 2014-12-28 00:12:38 -0800 | [diff] [blame] | 667 | |
Brian Silverman | 98f6ee2 | 2015-01-26 17:50:12 -0500 | [diff] [blame] | 668 | |
| 669 | START_ROBOT_CLASS(::frc971::wpilib::WPILibRobot); |