blob: 0065199c17da2c7c7b3a1566c223cd0bbe56b2d5 [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 {
15
Comran Morshed225f0b92016-02-10 20:34:27 +000016using ::frc971::constants::ShifterHallEffect;
17
Brian Silvermanc6579472016-02-14 20:47:52 -050018const ShifterHallEffect kThreeStateDriveShifter{0.0, 0.0, 0.0, 0.0, 0.25, 0.75};
Comran Morshed225f0b92016-02-10 20:34:27 +000019
Comran Morshed9a9948c2016-01-16 15:58:04 +000020const DrivetrainConfig &GetDrivetrainConfig() {
21 static DrivetrainConfig kDrivetrainConfig{
22 ::frc971::control_loops::drivetrain::ShifterType::HALL_EFFECT_SHIFTER,
23
Comran Morshed6c6a0a92016-01-17 12:45:16 +000024 ::y2016::control_loops::drivetrain::MakeDrivetrainLoop,
25 ::y2016::control_loops::drivetrain::MakeVelocityDrivetrainLoop,
26 ::y2016::control_loops::drivetrain::MakeKFDrivetrainLoop,
Comran Morshed9a9948c2016-01-16 15:58:04 +000027
Comran Morshed6c6a0a92016-01-17 12:45:16 +000028 drivetrain::kDt,
Comran Morshed6c6a0a92016-01-17 12:45:16 +000029 drivetrain::kRobotRadius,
30 drivetrain::kWheelRadius,
Comran Morshed6c6a0a92016-01-17 12:45:16 +000031 drivetrain::kV,
Comran Morshed3708f092016-02-20 16:23:51 +000032
Austin Schuh33bc6842016-02-17 00:38:51 -080033 drivetrain::kHighGearRatio,
34 drivetrain::kLowGearRatio,
Comran Morshed225f0b92016-02-10 20:34:27 +000035 kThreeStateDriveShifter,
36 kThreeStateDriveShifter};
Comran Morshed9a9948c2016-01-16 15:58:04 +000037
38 return kDrivetrainConfig;
39};
40
41} // namespace control_loops
Comran Morshed6c6a0a92016-01-17 12:45:16 +000042} // namespace y2016