1 package org
.usfirst
.frc
.team3501
.robot
.subsystems
;
3 import org
.usfirst
.frc
.team3501
.robot
.Constants
;
4 import org
.usfirst
.frc
.team3501
.robot
.commands
.driving
.JoystickDrive
;
5 import org
.usfirst
.frc
.team3501
.robot
.utils
.BNO055
;
7 import com
.ctre
.CANTalon
;
9 import edu
.wpi
.first
.wpilibj
.Encoder
;
10 import edu
.wpi
.first
.wpilibj
.I2C
.Port
;
11 import edu
.wpi
.first
.wpilibj
.RobotDrive
;
12 import edu
.wpi
.first
.wpilibj
.command
.Subsystem
;
14 public class DriveTrain
extends Subsystem
{
15 public static double driveP
= 0.008, driveI
= 0.001, driveD
= -0.002;
16 public static double defaultGyroP
= 0.009, defaultGyroI
= 0.00000,
17 defaultGyroD
= -0.000;
18 private double gyroZero
= 0;
20 public static final double WHEEL_DIAMETER
= 6; // inches
21 public static final int ENCODER_PULSES_PER_REVOLUTION
= 256;
22 public static final double INCHES_PER_PULSE
= WHEEL_DIAMETER
* Math
.PI
23 / ENCODER_PULSES_PER_REVOLUTION
;
25 private static DriveTrain driveTrain
;
26 private final CANTalon frontLeft
, frontRight
, rearLeft
, rearRight
;
27 private final RobotDrive robotDrive
;
28 private final Encoder leftEncoder
, rightEncoder
;
32 private DriveTrain() {
34 frontLeft
= new CANTalon(Constants
.DriveTrain
.FRONT_LEFT
);
35 frontRight
= new CANTalon(Constants
.DriveTrain
.FRONT_RIGHT
);
36 rearLeft
= new CANTalon(Constants
.DriveTrain
.REAR_LEFT
);
37 rearRight
= new CANTalon(Constants
.DriveTrain
.REAR_RIGHT
);
40 leftEncoder
= new Encoder(Constants
.DriveTrain
.ENCODER_LEFT_A
,
41 Constants
.DriveTrain
.ENCODER_LEFT_B
, false, Encoder
.EncodingType
.k4X
);
42 rightEncoder
= new Encoder(Constants
.DriveTrain
.ENCODER_RIGHT_A
,
43 Constants
.DriveTrain
.ENCODER_RIGHT_B
, false, Encoder
.EncodingType
.k4X
);
45 leftEncoder
.setDistancePerPulse(INCHES_PER_PULSE
);
46 rightEncoder
.setDistancePerPulse(INCHES_PER_PULSE
);
49 robotDrive
= new RobotDrive(frontLeft
, rearLeft
, frontRight
, rearRight
);
51 this.imu
= BNO055
.getInstance(BNO055
.opmode_t
.OPERATION_MODE_IMUPLUS
,
52 BNO055
.vector_type_t
.VECTOR_EULER
, Port
.kOnboard
, (byte) 0x28);
53 gyroZero
= imu
.getHeading();
56 public static DriveTrain
getDriveTrain() {
57 if (driveTrain
== null) {
58 driveTrain
= new DriveTrain();
64 public void setMotorValues(final double left
, final double right
) {
68 frontRight
.set(-right
);
69 rearRight
.set(-right
);
72 public void joystickDrive(final double thrust
, final double twist
) {
73 robotDrive
.arcadeDrive(thrust
, twist
, true);
80 public double getFrontLeftMotorVal() {
81 return frontLeft
.get();
84 public double getFrontRightMotorVal() {
85 return frontRight
.get();
88 public double getRearLeftMotorVal() {
89 return frontLeft
.get();
92 public double getRearRightMotorVal() {
93 return frontLeft
.get();
98 public double getLeftEncoderDistance() {
99 return leftEncoder
.getDistance();
102 public double getRightEncoderDistance() {
103 return rightEncoder
.getDistance();
106 public void printEncoderOutput() {
107 System
.out
.println("left: " + getLeftEncoderDistance());
108 System
.out
.println("right: " + getRightEncoderDistance());
111 public double getAvgEncoderDistance() {
112 return (leftEncoder
.getDistance() + rightEncoder
.getDistance()) / 2;
115 public void resetEncoders() {
117 rightEncoder
.reset();
120 public double getLeftSpeed() {
121 return leftEncoder
.getRate();
124 public double getRightSpeed() {
125 return rightEncoder
.getRate();
128 public double getSpeed() {
129 return (getLeftSpeed() + getRightSpeed()) / 2.0;
132 // ------Gyro------//
133 public double getAngle() {
134 if (!this.imu
.isInitialized())
136 return this.imu
.getHeading() - this.gyroZero
;
139 public void resetGyro() {
140 this.gyroZero
= this.getAngle();
144 public double getZeroAngle() {
145 return this.gyroZero
;
149 protected void initDefaultCommand() {
150 setDefaultCommand(new JoystickDrive());