blob: b469f773e73c1e59f947443fce1ed589a65327d2 [file] [log] [blame]
Comran Morshed6c6a0a92016-01-17 12:45:16 +00001#include "y2016/control_loops/drivetrain/drivetrain_base.h"
Comran Morshed9a9948c2016-01-16 15:58:04 +00002
3#include "frc971/control_loops/drivetrain/drivetrain_config.h"
4
Comran Morshed225f0b92016-02-10 20:34:27 +00005#include "frc971/control_loops/state_feedback_loop.h"
Comran Morshed6c6a0a92016-01-17 12:45:16 +00006#include "y2016/control_loops/drivetrain/drivetrain_dog_motor_plant.h"
7#include "y2016/control_loops/drivetrain/polydrivetrain_dog_motor_plant.h"
8#include "y2016/control_loops/drivetrain/kalman_drivetrain_motor_plant.h"
9#include "y2016/constants.h"
Comran Morshed9a9948c2016-01-16 15:58:04 +000010
11using ::frc971::control_loops::drivetrain::DrivetrainConfig;
12
Comran Morshed6c6a0a92016-01-17 12:45:16 +000013namespace y2016 {
Comran Morshed9a9948c2016-01-16 15:58:04 +000014namespace control_loops {
Austin Schuhba93d9e2016-03-18 22:38:57 -070015namespace drivetrain {
Comran Morshed9a9948c2016-01-16 15:58:04 +000016
Comran Morshed225f0b92016-02-10 20:34:27 +000017using ::frc971::constants::ShifterHallEffect;
18
Sabina Davis415bb6c2017-10-16 23:30:52 -070019const ShifterHallEffect kThreeStateDriveShifter{0.0, 0.0, 0.25, 0.75};
Comran Morshed225f0b92016-02-10 20:34:27 +000020
Austin Schuhbcce26a2018-03-26 23:41:24 -070021const DrivetrainConfig<double> &GetDrivetrainConfig() {
22 static DrivetrainConfig<double> kDrivetrainConfig{
Comran Morshed9a9948c2016-01-16 15:58:04 +000023 ::frc971::control_loops::drivetrain::ShifterType::HALL_EFFECT_SHIFTER,
Austin Schuh126924c2016-02-28 21:56:17 -080024 ::frc971::control_loops::drivetrain::LoopType::CLOSED_LOOP,
Campbell Crowley2527ed22017-02-17 21:10:02 -080025 ::frc971::control_loops::drivetrain::GyroType::SPARTAN_GYRO,
Diana Burgessd0180f12018-03-21 21:24:17 -070026 ::frc971::control_loops::drivetrain::IMUType::IMU_X,
Comran Morshed9a9948c2016-01-16 15:58:04 +000027
Comran Morshed6c6a0a92016-01-17 12:45:16 +000028 ::y2016::control_loops::drivetrain::MakeDrivetrainLoop,
29 ::y2016::control_loops::drivetrain::MakeVelocityDrivetrainLoop,
30 ::y2016::control_loops::drivetrain::MakeKFDrivetrainLoop,
Comran Morshed9a9948c2016-01-16 15:58:04 +000031
Comran Morshed6c6a0a92016-01-17 12:45:16 +000032 drivetrain::kDt,
Comran Morshed6c6a0a92016-01-17 12:45:16 +000033 drivetrain::kRobotRadius,
34 drivetrain::kWheelRadius,
Comran Morshed6c6a0a92016-01-17 12:45:16 +000035 drivetrain::kV,
Comran Morshed3708f092016-02-20 16:23:51 +000036
Austin Schuh33bc6842016-02-17 00:38:51 -080037 drivetrain::kHighGearRatio,
38 drivetrain::kLowGearRatio,
Comran Morshed225f0b92016-02-10 20:34:27 +000039 kThreeStateDriveShifter,
Adam Snaiderbc918b62016-02-27 21:03:39 -080040 kThreeStateDriveShifter,
Austin Schuhe8a54c02018-03-05 00:25:58 -080041 true /* default_high_gear */,
Adam Snaider94a52372016-10-19 20:06:01 -070042 constants::GetValues().down_error,
Austin Schuhe8a54c02018-03-05 00:25:58 -080043 0.25 /* wheel_non_linearity */,
44 1.0 /* quickturn_wheel_multiplier */,
45 1.0 /* wheel_multiplier */,
46 };
Comran Morshed9a9948c2016-01-16 15:58:04 +000047
48 return kDrivetrainConfig;
49};
50
Austin Schuhba93d9e2016-03-18 22:38:57 -070051} // namespace drivetrain
Comran Morshed9a9948c2016-01-16 15:58:04 +000052} // namespace control_loops
Comran Morshed6c6a0a92016-01-17 12:45:16 +000053} // namespace y2016