blob: b351cbff5407c2632cd1e0fc645e4430d884ca5a [file] [log] [blame]
Diana Vandenberg223703d2017-01-28 17:39:53 -08001#include "y2017/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 "y2017/control_loops/drivetrain/drivetrain_dog_motor_plant.h"
7#include "y2017/control_loops/drivetrain/polydrivetrain_dog_motor_plant.h"
8#include "y2017/control_loops/drivetrain/kalman_drivetrain_motor_plant.h"
9#include "y2017/constants.h"
10
11using ::frc971::control_loops::drivetrain::DrivetrainConfig;
12
13namespace y2017 {
14namespace control_loops {
15namespace drivetrain {
16
17using ::frc971::constants::ShifterHallEffect;
18
Sabina Davis415bb6c2017-10-16 23:30:52 -070019const ShifterHallEffect kThreeStateDriveShifter{0.0, 0.0, 0.25, 0.75};
Diana Vandenberg223703d2017-01-28 17:39:53 -080020
Austin Schuhbcce26a2018-03-26 23:41:24 -070021const DrivetrainConfig<double> &GetDrivetrainConfig() {
22 static DrivetrainConfig<double> kDrivetrainConfig{
Austin Schuhaab05dc2017-02-04 22:13:38 -080023 ::frc971::control_loops::drivetrain::ShifterType::NO_SHIFTER,
Diana Vandenberg223703d2017-01-28 17:39:53 -080024 ::frc971::control_loops::drivetrain::LoopType::CLOSED_LOOP,
Austin Schuh9f9adb62017-03-05 01:01:37 -080025 ::frc971::control_loops::drivetrain::GyroType::IMU_Z_GYRO,
Diana Burgessd0180f12018-03-21 21:24:17 -070026 ::frc971::control_loops::drivetrain::IMUType::IMU_X,
Diana Vandenberg223703d2017-01-28 17:39:53 -080027
28 ::y2017::control_loops::drivetrain::MakeDrivetrainLoop,
29 ::y2017::control_loops::drivetrain::MakeVelocityDrivetrainLoop,
30 ::y2017::control_loops::drivetrain::MakeKFDrivetrainLoop,
31
32 drivetrain::kDt,
33 drivetrain::kRobotRadius,
34 drivetrain::kWheelRadius,
35 drivetrain::kV,
36
37 drivetrain::kHighGearRatio,
38 drivetrain::kHighGearRatio,
39 kThreeStateDriveShifter,
40 kThreeStateDriveShifter,
Austin Schuhe8a54c02018-03-05 00:25:58 -080041 true /* default_high_gear */,
Diana Vandenberg223703d2017-01-28 17:39:53 -080042 constants::GetValues().down_error,
Austin Schuhe8a54c02018-03-05 00:25:58 -080043 0.4 /* wheel_non_linearity */,
44 1.0 /* quickturn_wheel_multiplier */,
45 1.0 /* wheel_multiplier */,
46 };
Diana Vandenberg223703d2017-01-28 17:39:53 -080047
48 return kDrivetrainConfig;
49};
50
51} // namespace drivetrain
52} // namespace control_loops
53} // namespace y2017