blob: b94c4f0163c6ea07e3640fac14666a9a5098183d [file] [log] [blame]
Austin Schuh2a671df2016-11-26 15:00:06 -08001#include "y2012/control_loops/drivetrain/drivetrain_base.h"
2
3#include "frc971/control_loops/drivetrain/drivetrain_config.h"
4
5#include "frc971/control_loops/state_feedback_loop.h"
6#include "y2012/control_loops/drivetrain/drivetrain_dog_motor_plant.h"
7#include "y2012/control_loops/drivetrain/polydrivetrain_dog_motor_plant.h"
8#include "y2012/control_loops/drivetrain/kalman_drivetrain_motor_plant.h"
9
10using ::frc971::control_loops::drivetrain::DrivetrainConfig;
11
12namespace y2012 {
13namespace control_loops {
14namespace drivetrain {
15
16using ::frc971::constants::ShifterHallEffect;
17
Sabina Davis415bb6c2017-10-16 23:30:52 -070018const ShifterHallEffect kThreeStateDriveShifter{0.0, 0.0, 0.25, 0.75};
Austin Schuh2a671df2016-11-26 15:00:06 -080019
Austin Schuhbcce26a2018-03-26 23:41:24 -070020const DrivetrainConfig<double> &GetDrivetrainConfig() {
21 static DrivetrainConfig<double> kDrivetrainConfig{
Austin Schuh2a671df2016-11-26 15:00:06 -080022 ::frc971::control_loops::drivetrain::ShifterType::NO_SHIFTER,
23 ::frc971::control_loops::drivetrain::LoopType::CLOSED_LOOP,
Campbell Crowley2527ed22017-02-17 21:10:02 -080024 ::frc971::control_loops::drivetrain::GyroType::SPARTAN_GYRO,
Diana Burgessd0180f12018-03-21 21:24:17 -070025 ::frc971::control_loops::drivetrain::IMUType::IMU_X,
Austin Schuh2a671df2016-11-26 15:00:06 -080026
27 ::y2012::control_loops::drivetrain::MakeDrivetrainLoop,
28 ::y2012::control_loops::drivetrain::MakeVelocityDrivetrainLoop,
29 ::y2012::control_loops::drivetrain::MakeKFDrivetrainLoop,
30
31 drivetrain::kDt,
32 drivetrain::kRobotRadius,
33 drivetrain::kWheelRadius,
34 drivetrain::kV,
35
36 drivetrain::kHighGearRatio,
37 drivetrain::kLowGearRatio,
38 kThreeStateDriveShifter,
39 kThreeStateDriveShifter,
Austin Schuhe8a54c02018-03-05 00:25:58 -080040 true /* default_high_gear */,
Austin Schuh2a671df2016-11-26 15:00:06 -080041 0.0,
Austin Schuhe8a54c02018-03-05 00:25:58 -080042 0.4 /* wheel_non_linearity */,
43 1.0 /* quickturn_wheel_multiplier */,
44 1.0 /* wheel_multiplier */};
Austin Schuh2a671df2016-11-26 15:00:06 -080045
46 return kDrivetrainConfig;
47};
48
49} // namespace drivetrain
50} // namespace control_loops
51} // namespace y2012